Test adding Connection: close header, see #7

ecovaci · Apr 8, 2022 · c944602 · c944602
1 parent 988b99e
commit c944602
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/java/org/kpax/winfoom/proxy/processor/NonConnectClientConnectionProcessor.java b/src/main/java/org/kpax/winfoom/proxy/processor/NonConnectClientConnectionProcessor.java
@@ -20,6 +20,7 @@
 import org.apache.http.conn.ConnectTimeoutException;
 import org.apache.http.conn.HttpHostConnectException;
 import org.apache.http.impl.client.CloseableHttpClient;
+import org.apache.http.protocol.HTTP;
 import org.apache.http.util.EntityUtils;
 import org.kpax.winfoom.annotation.ThreadSafe;
 import org.kpax.winfoom.config.ProxyConfig;
@@ -88,6 +89,8 @@ void handleRequest(final ClientConnection clientConnection, final ProxyInfo prox
 
                     response.removeHeaders(HttpHeaders.VIA);
                     response.removeHeaders(HttpHeaders.PROXY_AUTHENTICATE);
+                    response.removeHeaders(HttpHeaders.CONNECTION);// Remove the Connection header
+                                                                   // in order to add Connection: close header later.
 
                     for (Header header : response.getAllHeaders()) {
                         if (HttpHeaders.TRANSFER_ENCODING.equals(header.getName())) {
@@ -109,6 +112,10 @@ void handleRequest(final ClientConnection clientConnection, final ProxyInfo prox
                         }
                     }
 
+                    clientConnection.write(
+                            HttpUtils.createHttpHeader(HttpHeaders.CONNECTION,
+                                    HTTP.CONN_CLOSE));
+
                     // Empty line marking the end
                     // of header's section
                     clientConnection.writeln();