From 121c0b14e45793f9175fabf4929e17bfd6b00077 Mon Sep 17 00:00:00 2001
From: Adlyq <2833154405@qq.com>
Date: Wed, 5 Apr 2023 19:34:52 +0800
Subject: [PATCH] Add headers option to http connect client

---
 protocol/http/client.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/protocol/http/client.go b/protocol/http/client.go
index bc85fd2..c6f595e 100644
--- a/protocol/http/client.go
+++ b/protocol/http/client.go
@@ -23,14 +23,16 @@ type Client struct {
 	serverAddr M.Socksaddr
 	username   string
 	password   string
+	headers    http.Header
 }
 
-func NewClient(dialer N.Dialer, serverAddr M.Socksaddr, username string, password string) *Client {
+func NewClient(dialer N.Dialer, serverAddr M.Socksaddr, username string, password string, headers http.Header) *Client {
 	return &Client{
 		dialer,
 		serverAddr,
 		username,
 		password,
+		headers,
 	}
 }
 
@@ -59,6 +61,12 @@ func (c *Client) DialContext(ctx context.Context, network string, destination M.
 			"Proxy-Connection": []string{"Keep-Alive"},
 		},
 	}
+	for key, valueList := range c.headers {
+		request.Header.Set(key, valueList[0])
+		for _, value := range valueList[1:] {
+			request.Header.Add(key, value)
+		}
+	}
 	if c.username != "" {
 		auth := c.username + ":" + c.password
 		request.Header.Add("Proxy-Authorization", "Basic "+base64.StdEncoding.EncodeToString([]byte(auth)))