78 lines
1.4 KiB
Go
78 lines
1.4 KiB
Go
package httpServer
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
"tunnel_pls/session"
|
|
)
|
|
|
|
func Listen() {
|
|
server, err := net.Listen("tcp", ":80")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
return
|
|
}
|
|
defer server.Close()
|
|
log.Println("Listening on :80")
|
|
for {
|
|
conn, err := server.Accept()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
return
|
|
}
|
|
|
|
go handleRequest(conn)
|
|
}
|
|
}
|
|
|
|
func handleRequest(conn net.Conn) {
|
|
defer conn.Close()
|
|
var rawRequest string
|
|
|
|
reader := bufio.NewReader(conn)
|
|
r, err := http.ReadRequest(reader)
|
|
if err != nil {
|
|
fmt.Println("Error reading request:", err)
|
|
return
|
|
}
|
|
|
|
slug := strings.Split(r.Host, ".")[0]
|
|
if slug == "" {
|
|
fmt.Println("Error parsing slug: ", r.Host)
|
|
return
|
|
}
|
|
|
|
sshSession, ok := session.Clients[slug]
|
|
if !ok {
|
|
fmt.Println("Error finding ssh session: ", slug)
|
|
return
|
|
}
|
|
|
|
rawRequest += fmt.Sprintf("%s %s %s\r\n", r.Method, r.URL.RequestURI(), r.Proto)
|
|
rawRequest += fmt.Sprintf("Host: %s\r\n", r.Host)
|
|
|
|
for k, v := range r.Header {
|
|
rawRequest += fmt.Sprintf("%s: %s\r\n", k, v[0])
|
|
}
|
|
rawRequest += "\r\n"
|
|
|
|
if r.Body != nil {
|
|
body, err := io.ReadAll(r.Body)
|
|
if err != nil {
|
|
log.Println("Error reading request body:", err)
|
|
} else {
|
|
rawRequest += string(body)
|
|
}
|
|
}
|
|
|
|
payload := []byte(rawRequest)
|
|
|
|
host, originPort := session.ParseAddr(conn.RemoteAddr().String())
|
|
sshSession.GetForwardedConnection(conn, host, sshSession.Connection, payload, originPort, 80)
|
|
}
|