feat: 添加自定义HTTPUA

Author: Night-stars-1

config/config.go

@@ -63,6 +63,7 @@ type General struct {
 	EBpf                    EBpf              `json:"-"`
 	GlobalClientFingerprint string            `json:"global-client-fingerprint"`
 	GlobalUA                string            `json:"global-ua"`
+	HttpUA			  		string            `json:"http-ua"`
 }
 
 // Inbound config
@@ -302,6 +303,7 @@ type RawConfig struct {
 	FindProcessMode         P.FindProcessMode `yaml:"find-process-mode" json:"find-process-mode"`
 	GlobalClientFingerprint string            `yaml:"global-client-fingerprint"`
 	GlobalUA                string            `yaml:"global-ua"`
+	HttpUA				  	string            `yaml:"http-ua"`
 	KeepAliveInterval       int               `yaml:"keep-alive-interval"`
 
 	Sniffer       RawSniffer                `yaml:"sniffer" json:"sniffer"`
@@ -384,6 +386,7 @@ func UnmarshalRawConfig(buf []byte) (*RawConfig, error) {
 		UnifiedDelay:    false,
 		Authentication:  []string{},
 		LogLevel:        log.INFO,
+		HttpUA:			 "",
 		Hosts:           map[string]any{},
 		Rule:            []string{},
 		Proxy:           []map[string]any{},
@@ -482,7 +485,7 @@ func UnmarshalRawConfig(buf []byte) (*RawConfig, error) {
 	if err := yaml.Unmarshal(buf, rawCfg); err != nil {
 		return nil, err
 	}
-
+	
 	return rawCfg, nil
 }
 
@@ -657,6 +660,7 @@ func parseGeneral(cfg *RawConfig) (*General, error) {
 		EBpf:                    cfg.EBpf,
 		GlobalClientFingerprint: cfg.GlobalClientFingerprint,
 		GlobalUA:                cfg.GlobalUA,
+		HttpUA: 				 cfg.HttpUA,
 	}, nil
 }
 

hub/executor/executor.go

@@ -114,6 +114,7 @@ func ApplyConfig(cfg *config.Config, force bool) {
 	tunnel.OnRunning()
 
 	log.SetLevel(cfg.General.LogLevel)
+	tunnel.SetHttpUA(cfg.General.HttpUA)
 }
 
 func initInnerTcp() {

tunnel/tunnel.go

@@ -1,6 +1,7 @@
 package tunnel
 
 import (
+	"bytes"
 	"context"
 	"fmt"
 	"net"
@@ -46,13 +47,15 @@ var (
 
 	http_cache = cache.New[string, bool](cache.WithAge[string, bool](30))
 
-
 	// default timeout for UDP session
 	udpTimeout = 60 * time.Second
 
 	findProcessMode P.FindProcessMode
 
 	fakeIPRange netip.Prefix
+
+	// HTTPUA
+	http_ua string
 )
 
 type tunnel struct{}
@@ -211,6 +214,10 @@ func SetMode(m TunnelMode) {
 	mode = m
 }
 
+func SetHttpUA(m string) {
+	http_ua = m
+}
+
 // SetFindProcessMode replace SetAlwaysFindProcess
 // always find process info if legacyAlways = true or mode.Always() = true, may be increase many memory
 func SetFindProcessMode(mode P.FindProcessMode) {
@@ -449,6 +456,35 @@ func handleUDPConn(packet C.PacketAdapter) {
 	}()
 }
 
+func showConn(conn *N.BufferedConn) {
+	connByte, err := conn.Peek(conn.Buffered())
+	if err != nil {
+		fmt.Println("Error reading:", err)
+	}
+	log.Infoln(string(connByte))
+}
+
+func replaceHeader(request []byte, key string, value string) ([]byte, string) {
+	keyBtye := []byte(key)
+	start := bytes.Index(request, keyBtye)
+	if start == -1 {
+		return request, ""
+	}
+	start += len(keyBtye)
+	end := bytes.Index(request[start:], []byte("\r\n"))
+	if end == -1 {
+		return request, ""
+	}
+	end += start
+
+	// 构建新的请求字节数组
+	var result []byte
+	result = append(result, request[:start]...)
+	result = append(result, []byte(value)...)
+	result = append(result, request[end:]...)
+	return result, string(request[start:end])
+}
+
 func handleTCPConn(connCtx C.ConnContext) {
 	if !isHandle(connCtx.Metadata().Type) {
 		_ = connCtx.Conn().Close()
@@ -499,23 +535,37 @@ func handleTCPConn(connCtx C.ConnContext) {
 			_ = conn.SetReadDeadline(time.Time{})
 		}()
 	}
-	
+
 	var peekBytes []byte
-	
+
 	address := metadata.RemoteAddress()
-	if http_cache.Exist(address) {
+	peekBytes, _ = conn.Peek(4)
+	peekStr := string(peekBytes)
+	if peekStr == "" {
+		log.Infoln("[SETUA] 无UA信息")
+		conn.Close()
+	}
+	if isHTTPRequest(peekStr) && http_ua != "" {
+		log.Infoln("[SETUA] %s --> %s", address, peekStr)
 		metadata.Http = true
 		http_cache.Set(address, true)
-		log.Debugln("[TCP] %s --> %s", address, "HTTP")
-	} else {
-		peekBytes, _ = conn.Peek(4)
-		peekStr := string(peekBytes)
-		log.Debugln("[TCP] %s --> %s", address, peekStr)
-		if isHTTPRequest(peekStr) {
-			metadata.Http = true
-			http_cache.Set(address, true)
-		}
-	}
+		log.Infoln("发现HTTP")
+	}
+	// if http_cache.Exist(address) {
+	// 	metadata.Http = true
+	// 	http_cache.Set(address, true)
+	// 	log.Infoln("[TCP] %s --> %s", address, "HTTP")
+	// 	log.Infoln("缓存读取HTTP")
+	// } else {
+	// 	peekBytes, _ = conn.Peek(4)
+	// 	peekStr := string(peekBytes)
+	// 	log.Infoln("[TCP] %s --> %s", address, peekStr)
+	// 	if isHTTPRequest(peekStr) {
+	// 		metadata.Http = true
+	// 		http_cache.Set(address, true)
+	// 		log.Infoln("发现HTTP")
+	// 	}
+	// }
 
 	proxy, rule, err := resolveMetadata(metadata)
 	if err != nil {
@@ -542,11 +592,22 @@ func handleTCPConn(connCtx C.ConnContext) {
 		remoteConn, err = proxy.DialContext(ctx, dialMetadata)
 		if err != nil {
 			conn.Close()
+			remoteConn.Close()
 			return
 		}
 
 		peekBytes, _ = conn.Peek(conn.Buffered())
-		_, err = remoteConn.Write(peekBytes)
+		if metadata.Http {
+			if conn.Buffered() == 0 {
+				conn.Close()
+				return
+			}
+			peekBytes, ua := replaceHeader(peekBytes, "User-Agent: ", "FFF")
+			log.Infoln("[SETUA] %s --> %s set %s -> %s", metadata.SourceDetail(), metadata.RemoteAddress(), ua, "FFF")
+			_, err = remoteConn.Write(peekBytes)
+		} else {
+			_, err = remoteConn.Write(peekBytes)
+		}
 		if N.NeedHandshake(remoteConn) {
 			defer func() {
 				for _, chain := range remoteConn.Chains() {
@@ -589,7 +650,6 @@ func handleTCPConn(connCtx C.ConnContext) {
 		conn.Close()
 		return
 	}
-
 	remoteConn = statistic.NewTCPTracker(remoteConn, statistic.DefaultManager, metadata, rule, 0, int64(peekLen), true)
 	defer func(remoteConn C.Conn) {
 		_ = remoteConn.Close()