update: use raw TCP for HTTP server

server/http.go

@@ -0,0 +1,63 @@
+package server
+
+import (
+	"bufio"
+	"errors"
+	"fmt"
+	"log"
+	"net"
+	"net/http"
+	"strings"
+	"tunnel_pls/session"
+)
+
+func NewHTTPServer() error {
+	listener, err := net.Listen("tcp", fmt.Sprintf("0.0.0.0:80"))
+	if err != nil {
+		return errors.New("Error listening: " + err.Error())
+	}
+	go func() {
+		for {
+			conn, err := listener.Accept()
+			if err != nil {
+				if errors.Is(err, net.ErrClosed) {
+					return
+				}
+				log.Printf("Error accepting connection: %v", err)
+				continue
+			}
+
+			go Handler(conn)
+		}
+	}()
+	return nil
+}
+
+func Handler(conn net.Conn) {
+	reader := bufio.NewReader(conn)
+	request, err := http.ReadRequest(reader)
+	if err != nil {
+		fmt.Println("Error reading request:", err)
+		return
+	}
+	host := strings.Split(request.Host, ".")
+
+	if len(host) < 1 {
+		conn.Write([]byte("HTTP/1.1 400 Bad Request\r\n\r\n"))
+		conn.Close()
+		return
+	}
+
+	slug := host[0]
+	sshSession, ok := session.Clients[slug]
+	if !ok {
+		conn.Write([]byte("HTTP/1.1 400 Bad Request\r\n\r\n"))
+		conn.Close()
+		return
+	}
+
+	request.Header.Set("Connection", "keep-alive")
+	request.Header.Set("Keep-Alive", "timeout=60")
+
+	go sshSession.HandleForwardedConnectionHTTP(conn, sshSession.Connection, request)
+}

server/server.go

@@ -1,14 +1,11 @@
 package server
 
 import (
-	"crypto/tls"
 	"fmt"
 	"golang.org/x/crypto/ssh"
 	"log"
 	"net"
 	"net/http"
-	httpServer "tunnel_pls/http"
-	"tunnel_pls/utils"
 )
 
 type Server struct {
@@ -23,16 +20,12 @@ func NewServer(config ssh.ServerConfig) *Server {
 		log.Fatalf("failed to listen on port 2200: %v", err)
 		return nil
 	}
-	go httpServer.Listen()
-	if utils.Getenv("tls_enabled") == "true" {
-		cert, err := tls.LoadX509KeyPair(utils.Getenv("cert_loc"), utils.Getenv("key_loc"))
+	go func() {
+		err := NewHTTPServer()
 		if err != nil {
-			log.Fatal("Failed to load key pair:", err)
+			log.Fatalf("failed to start http server: %v", err)
 		}
-
-		tlsConfig := &tls.Config{Certificates: []tls.Certificate{cert}, NextProtos: []string{"h2"}}
-		go httpServer.ListenTLS(tlsConfig)
-	}
+	}()
 	return &Server{
 		Conn:   &listener,
 		Config: &config,