[proxy]如果客户端请求的Host为空，则自动指定为当前连接的服务器地址

Author: iwind

teaproxy/listener.go

@@ -389,6 +389,18 @@ func (this *Listener) handleHTTP(writer http.ResponseWriter, rawRequest *http.Re
 
 	// 域名
 	reqHost := rawRequest.Host
+
+	// 防止空Host
+	if len(reqHost) == 0 {
+		ctx := rawRequest.Context()
+		if ctx != nil {
+			addr := ctx.Value(http.LocalAddrContextKey)
+			if addr != nil {
+				reqHost = addr.(net.Addr).String()
+			}
+		}
+	}
+
 	domain, _, err := net.SplitHostPort(reqHost)
 	if err != nil {
 		domain = reqHost

teaproxy/request_backend.go

@@ -104,7 +104,7 @@ func (this *Request) callBackend(writer *ResponseWriter) error {
 			}
 
 			// 支持修改Host
-			if header.Name == "Host" {
+			if header.Name == "Host" && len(header.Value) > 0 {
 				this.raw.Host = header.Value
 			}
 		}