From d9f255921404ee576eda0de46597c91580d51d68 Mon Sep 17 00:00:00 2001
From: HystericalDragon <HystericalDragons@proton.me>
Date: Mon, 17 Jun 2024 12:37:09 +0800
Subject: [PATCH] Fix http socks close

---
 protocol/http/client.go  | 2 ++
 protocol/socks/client.go | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/protocol/http/client.go b/protocol/http/client.go
index 727d4d9..9622352 100644
--- a/protocol/http/client.go
+++ b/protocol/http/client.go
@@ -80,6 +80,7 @@ func (c *Client) DialContext(ctx context.Context, network string, destination M.
 	}
 	if c.host != "" && c.host != destination.Fqdn {
 		if c.path != "" {
+			_ = conn.Close()
 			return nil, E.New("Host header and path are not allowed at the same time")
 		}
 		request.Host = c.host
@@ -90,6 +91,7 @@ func (c *Client) DialContext(ctx context.Context, network string, destination M.
 	if c.path != "" {
 		err = URLSetPath(request.URL, c.path)
 		if err != nil {
+			_ = conn.Close()
 			return nil, err
 		}
 	}
diff --git a/protocol/socks/client.go b/protocol/socks/client.go
index 6004f83..9e4a1f5 100644
--- a/protocol/socks/client.go
+++ b/protocol/socks/client.go
@@ -149,6 +149,7 @@ func (c *Client) DialContext(ctx context.Context, network string, address M.Sock
 		}
 		return NewAssociatePacketConn(udpConn, address, tcpConn), nil
 	}
+	_ = tcpConn.Close()
 	return nil, os.ErrInvalid
 }
 
@@ -181,5 +182,6 @@ func (c *Client) BindContext(ctx context.Context, address M.Socksaddr) (net.Conn
 		}
 		return tcpConn, nil
 	}
+	_ = tcpConn.Close()
 	return nil, os.ErrInvalid
 }