updating proxy server

Author: debuggerboy

.gitignore

@@ -24,6 +24,8 @@ go.work
 detour2-devel
 server.crt
 server.key
+detour.yaml
+detour-mappings.yaml
 old-ssl/
 
 # Backup

detour.yaml.sample

@@ -0,0 +1,4 @@
+port: ":443"
+certificate: "path/to/server.crt"
+keyfile: "path/to/server.key"
+

go.mod

@@ -2,7 +2,10 @@ module pulsiot/detour2-devel
 
 go 1.19
 
-require github.com/valyala/fasthttp v1.48.0
+require (
+	github.com/valyala/fasthttp v1.48.0
+	gopkg.in/yaml.v2 v2.4.0
+)
 
 require (
 	github.com/andybalholm/brotli v1.0.5 // indirect

go.sum

@@ -6,3 +6,6 @@ github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6Kllzaw
 github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
 github.com/valyala/fasthttp v1.48.0 h1:oJWvHb9BIZToTQS3MuQ2R3bJZiNSa2KiNdeI8A+79Tc=
 github.com/valyala/fasthttp v1.48.0/go.mod h1:k2zXd82h/7UZc3VOdJ2WaUqt1uZ/XpXAfE9i+HBC3lA=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
+gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=

main.go

@@ -2,13 +2,40 @@ package main
 
 import (
 	"crypto/tls"
+	"fmt"
+	"io/ioutil"
 	"log"
 	"net/url"
 
 	"github.com/valyala/fasthttp"
+	"gopkg.in/yaml.v2"
 )
 
+// Mapping struct to hold the proxy domain and target URL
+type Mapping struct {
+	Domain    string `yaml:"domain"`
+	TargetURL string `yaml:"targetURL"`
+}
+
 func main() {
+	// Read the detour.yaml file
+	detourData, err := ioutil.ReadFile("detour.yaml")
+	if err != nil {
+		log.Fatal("Failed to read detour.yaml:", err)
+	}
+
+	// Parse the YAML data
+	var detourConfig struct {
+		Port      int       `yaml:"port"`
+		CertFile  string    `yaml:"certFile"`
+		KeyFile   string    `yaml:"keyFile"`
+		Mappings  []Mapping `yaml:"mappings"`
+	}
+	err = yaml.Unmarshal(detourData, &detourConfig)
+	if err != nil {
+		log.Fatal("Failed to parse detour.yaml:", err)
+	}
+
 	// Create a new TLS configuration
 	tlsConfig := &tls.Config{
 		MinVersion:               tls.VersionTLS12, // Set the minimum TLS version
@@ -17,7 +44,10 @@ func main() {
 
 	// Create a new HTTP client with TLS support
 	client := &fasthttp.Client{
-		TLSConfig: tlsConfig,
+		TLSConfig:           tlsConfig,
+		MaxResponseBodySize: 10 * 1024 * 1024, // 10MB max response size
+		ReadBufferSize:      4096,
+		WriteBufferSize:     4096,
 	}
 
 	// Create a new fasthttp server
@@ -26,15 +56,16 @@ func main() {
 			// Extract the requested host
 			host := string(ctx.Host())
 
-			// Define the mapping of domains to target URLs
-			targetURLs := map[string]string{
-				"domain.com":        "https://www.google.com",
-				"sales.domain.com":  "https://debuggerboy.com",
+			// Find the target URL for the requested host in the mappings
+			var targetURL string
+			for _, mapping := range detourConfig.Mappings {
+				if mapping.Domain == host {
+					targetURL = mapping.TargetURL
+					break
+				}
 			}
 
-			// Get the target URL for the requested host
-			targetURL, ok := targetURLs[host]
-			if !ok {
+			if targetURL == "" {
 				ctx.Error("Unknown host", fasthttp.StatusNotFound)
 				return
 			}
@@ -50,6 +81,7 @@ func main() {
 			ctx.URI().SetScheme(target.Scheme)
 			ctx.URI().SetHost(target.Host)
 			ctx.URI().SetPath(string(target.Path) + string(ctx.Path()))
+			ctx.URI().SetQueryStringBytes(ctx.QueryArgs().QueryString())
 
 			// Set the host header to the target host
 			ctx.Request.Header.SetHost(target.Host)
@@ -70,9 +102,8 @@ func main() {
 	}
 
 	// Listen for incoming HTTPS/TLS connections
-	err := server.ListenAndServeTLS(":443", "server.crt", "server.key")
+	err = server.ListenAndServeTLS(fmt.Sprintf(":%d", detourConfig.Port), detourConfig.CertFile, detourConfig.KeyFile)
 	if err != nil {
 		log.Fatal(err)
 	}
 }
-