From c875a4ffab1a54fc1834eca05298d67e6586121e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Fri, 3 Mar 2023 13:20:48 +0800
Subject: [PATCH] Fix http client

---
 protocol/http/client.go | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/protocol/http/client.go b/protocol/http/client.go
index 8277a65..bc85fd2 100644
--- a/protocol/http/client.go
+++ b/protocol/http/client.go
@@ -65,30 +65,36 @@ func (c *Client) DialContext(ctx context.Context, network string, destination M.
 	}
 	err = request.Write(conn)
 	if err != nil {
+		conn.Close()
 		return nil, err
 	}
 	reader := std_bufio.NewReader(conn)
 	response, err := http.ReadResponse(reader, request)
 	if err != nil {
+		conn.Close()
 		return nil, err
 	}
-	switch response.StatusCode {
-	case http.StatusOK:
+	if response.StatusCode == http.StatusOK {
 		if reader.Buffered() > 0 {
 			buffer := buf.NewSize(reader.Buffered())
 			_, err = buffer.ReadFullFrom(reader, buffer.FreeLen())
 			if err != nil {
+				conn.Close()
 				return nil, err
 			}
 			conn = bufio.NewCachedConn(conn, buffer)
 		}
 		return conn, nil
-	case http.StatusProxyAuthRequired:
-		return nil, E.New("authentication required")
-	case http.StatusMethodNotAllowed:
-		return nil, E.New("method not allowed")
-	default:
-		return nil, E.New("unexpected status: ", response.Status)
+	} else {
+		conn.Close()
+		switch response.StatusCode {
+		case http.StatusProxyAuthRequired:
+			return nil, E.New("authentication required")
+		case http.StatusMethodNotAllowed:
+			return nil, E.New("method not allowed")
+		default:
+			return nil, E.New("unexpected status: ", response.Status)
+		}
 	}
 }