From f48479a92c3b3931de1e8b82c052532c04edadbe Mon Sep 17 00:00:00 2001 From: htliang128 Date: Sat, 25 May 2024 12:17:46 +0800 Subject: [PATCH] Update of Socks Proxy adding headers --- .../github/jwdeveloper/tiktok/http/HttpProxyClient.java | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/Client/src/main/java/io/github/jwdeveloper/tiktok/http/HttpProxyClient.java b/Client/src/main/java/io/github/jwdeveloper/tiktok/http/HttpProxyClient.java index 92d9ace..4be4107 100644 --- a/Client/src/main/java/io/github/jwdeveloper/tiktok/http/HttpProxyClient.java +++ b/Client/src/main/java/io/github/jwdeveloper/tiktok/http/HttpProxyClient.java @@ -105,16 +105,19 @@ public class HttpProxyClient extends HttpClient { socksConnection.setSSLSocketFactory(sc.getSocketFactory()); socksConnection.setConnectTimeout(httpClientSettings.getTimeout().toMillisPart()); socksConnection.setReadTimeout(httpClientSettings.getTimeout().toMillisPart()); - + Map requestHeaders = httpClientSettings.getHeaders(); + for (Map.Entry entry : requestHeaders.entrySet()) { + socksConnection.setRequestProperty(entry.getKey(), entry.getValue()); + } byte[] body = socksConnection.getInputStream().readAllBytes(); - Map> headers = socksConnection.getHeaderFields() + Map> responseHeaders = socksConnection.getHeaderFields() .entrySet() .stream() .filter(entry -> entry.getKey() != null) .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue)); - var responseInfo = createResponseInfo(socksConnection.getResponseCode(), headers); + var responseInfo = createResponseInfo(socksConnection.getResponseCode(), responseHeaders); var response = createHttpResponse(body, toUrl(), responseInfo);