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,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)
+	}
+}