From c6a69b4912ee817c997e43fd45eb7018d99315f9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= Date: Wed, 2 Aug 2023 18:59:22 +0800 Subject: [PATCH] Fix "Fix http proxy server" --- protocol/http/handshake.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/protocol/http/handshake.go b/protocol/http/handshake.go index 78b2479..455de7c 100644 --- a/protocol/http/handshake.go +++ b/protocol/http/handshake.go @@ -28,11 +28,6 @@ func HandleConnection(ctx context.Context, conn net.Conn, reader *std_bufio.Read return E.Cause(err, "read http request") } - if hostStr := request.Header.Get("Host"); hostStr != "" { - request.Host = hostStr - request.URL.Host = hostStr - } - if authenticator != nil { var authOk bool authorization := request.Header.Get("Proxy-Authorization") @@ -89,6 +84,12 @@ func HandleConnection(ctx context.Context, conn net.Conn, reader *std_bufio.Read removeHopByHopHeaders(request.Header) removeExtraHTTPHostPort(request) + if hostStr := request.Header.Get("Host"); hostStr != "" { + if hostStr != request.URL.Host { + request.Host = hostStr + } + } + if request.URL.Scheme == "" || request.URL.Host == "" { return responseWith(request, http.StatusBadRequest).Write(conn) }