refactor(web): move all server files into server directory

2023-01-27 01:11:24 +03:00
parent 2981a4b0ca
commit 9a4817d6ad
3 changed files with 0 additions and 0 deletions
@@ -0,0 +1,3 @@
+module tweb
+
+go 1.13
@@ -0,0 +1,49 @@
+package main
+
+import (
+	"bytes"
+	"net"
+)
+
+// TDBMS connection struct
+type TDBMSConnection struct {
+	conn net.Conn
+}
+
+// Connect to TDBMS server
+func (tdbms *TDBMSConnection) Connect(domain, addr string) error {
+	c, err := net.Dial(domain, addr)
+	tdbms.conn = c
+	return err
+}
+
+// Close connection to TDBMS server
+func (tdbms *TDBMSConnection) Close() error {
+	return tdbms.conn.Close()
+}
+
+// Execute a TDB request
+func (tdbms *TDBMSConnection) Query(db_name string, trc uint8, request_body string) []byte {
+	var err error
+	buffer := []byte{trc}
+	buffer = append(buffer, db_name...)
+	buffer = append(buffer, 0)
+	buffer = append(buffer, request_body...)
+	buffer = append(buffer, 0, 4)
+	_, err = tdbms.conn.Write(buffer)
+	if err != nil {
+		return nil
+	}
+	var response []byte
+	var resp_size int = -1
+	buffer = make([]byte, 8192)
+	for resp_size <= 0 {
+		_, err = tdbms.conn.Read(buffer)
+		if err != nil {
+			return nil
+		}
+		response = append(response, buffer...)
+		resp_size = bytes.IndexByte(response, 4)
+	}
+	return response[:resp_size-1]
+}
@@ -0,0 +1,161 @@
+package main
+
+import (
+	"bytes"
+	"crypto/sha256"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"path"
+	"strconv"
+	"time"
+)
+
+type JSON struct {
+	Status bool   `json:"status,omitempty"`
+	Token  string `json:"token,omitempty"`
+	TRC    uint8  `json:"trc,omitempty"`
+	TRDB   string `json:"trdb,omitempty"`
+	TRB    string `json:"trb,omitempty"`
+}
+
+var tdbms TDBMSConnection
+
+const TOKEN_VALIDTIME = 604800
+
+var SID int64 = 0
+var TOKEN = ""
+
+func TokenGenerate(seed []byte) {
+	SID = time.Now().Unix()
+	value := SID
+	for i, char := range seed {
+		value += int64(char) << (i * 8)
+	}
+	TOKEN = fmt.Sprintf("%x", sha256.Sum256([]byte(strconv.FormatInt(value, 16))))
+}
+
+func TokenValidate(token string) bool {
+	if time.Now().Unix()-SID >= TOKEN_VALIDTIME || token != TOKEN {
+		return false
+	}
+	return true
+}
+
+func HandlerAuth(w http.ResponseWriter, r *http.Request) {
+	var buffer = make([]byte, sha256.Size)
+	var response = JSON{Status: false}
+	var passhash = make([]byte, sha256.Size)
+	var hash [sha256.Size]byte
+	var passlen = sha256.Size
+	var err error
+	passfile, err := os.Open("/etc/tanabata/.htpasswd")
+	if err != nil {
+		log.Fatalf("Failed to open password file: %s\n", err)
+	}
+	read, err := passfile.Read(passhash)
+	if err != nil {
+		log.Fatalf("Failed to read password file: %s\n", err)
+	}
+	if read != sha256.Size {
+		log.Fatalln("Invalid password file")
+	}
+	err = passfile.Close()
+	if err != nil {
+		log.Fatalf("Failed to close password file: %s\n", err)
+	}
+	_, err = r.Body.Read(buffer)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+	for i := 0; i < sha256.Size; i++ {
+		if buffer[i] == 0 {
+			passlen = i
+			break
+		}
+	}
+	hash = sha256.Sum256(buffer[:passlen])
+	if bytes.Equal(hash[:], passhash) {
+		TokenGenerate(buffer)
+		response.Status = true
+		response.Token = TOKEN
+	}
+	w.Header().Set("Content-Type", "application/json")
+	jsonData, err := json.Marshal(response)
+	if err != nil {
+		log.Fatalln(err)
+	}
+	_, err = w.Write(jsonData)
+	if err != nil {
+		log.Fatalln(err)
+	}
+}
+
+func HandlerTDBMS(w http.ResponseWriter, r *http.Request) {
+	var request JSON
+	var response []byte
+	var err error
+	r.Body = http.MaxBytesReader(w, r.Body, 1048576)
+	json_decoder := json.NewDecoder(r.Body)
+	json_decoder.DisallowUnknownFields()
+	err = json_decoder.Decode(&request)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+	if !TokenValidate(request.Token) {
+		http.Error(w, "Invalid token", http.StatusBadRequest)
+		return
+	}
+	response = tdbms.Query(request.TRDB, request.TRC, request.TRB)
+	if response == nil {
+		http.Error(w, "Failed to execute request", http.StatusInternalServerError)
+		return
+	}
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	_, err = w.Write(response)
+	if err != nil {
+		log.Println(err)
+	}
+}
+
+func main() {
+	var err error
+	log.Println("Connecting to TDBMS server...")
+	err = tdbms.Connect("unix", "/tmp/tdbms.sock")
+	if err != nil {
+		log.Fatalln("Failed to connect to TDBMS server")
+	}
+	defer func(tdbms TDBMSConnection) {
+		err := tdbms.Close()
+		if err != nil {
+			log.Fatalln(err)
+		}
+	}(tdbms)
+	log.Println("Initializing...")
+	server := &http.Server{
+		Addr: ":42776",
+	}
+	public_fs := http.FileServer(http.Dir("/srv/www/tanabata"))
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path != "/" && path.Ext(r.URL.Path) == "" {
+			r.URL.Path += ".html"
+		}
+		public_fs.ServeHTTP(w, r)
+	})
+	http.HandleFunc("/AUTH", HandlerAuth)
+	http.HandleFunc("/TDBMS", HandlerTDBMS)
+	tfm_fs := http.FileServer(http.Dir("/srv/data/tfm"))
+	http.Handle("/tfm/", http.StripPrefix("/tfm", tfm_fs))
+	log.Println("Running...")
+	err = server.ListenAndServeTLS("/etc/ssl/certs/web-global.crt", "/etc/ssl/private/web-global.key")
+	if errors.Is(err, http.ErrServerClosed) {
+		log.Fatalln("Server closed")
+	} else if err != nil {
+		log.Fatalf("Error starting server: %s\n", err)
+	}
+}