allow to terminate server connection

Author: honfika

Titanium.Web.Proxy/RequestHandler.cs

@@ -46,7 +46,6 @@ private async Task HandleClient(ExplicitProxyEndPoint endPoint, TcpClient tcpCli
             try
             {
                 string connectHostname = null;
-
                 ConnectRequest connectRequest = null;
 
                 //Client wants to create a secure tcp tunnel (probably its a HTTPS or Websocket request)
@@ -297,7 +296,7 @@ private async Task HandleHttpSessionRequest(TcpClient client, CustomBufferedStre
                     string httpCmd = await clientStreamReader.ReadLineAsync();
                     if (string.IsNullOrEmpty(httpCmd))
                     {
-                        break;
+                        return;
                     }
 
                     var args = new SessionEventArgs(BufferSize, endPoint, ExceptionFunc)
@@ -364,7 +363,7 @@ private async Task HandleHttpSessionRequest(TcpClient client, CustomBufferedStre
 
                                 //send the response
                                 await clientStreamWriter.WriteResponseAsync(args.WebSession.Response);
-                                break;
+                                return;
                             }
 
                             if (!isTransparentEndPoint)
@@ -397,7 +396,7 @@ private async Task HandleHttpSessionRequest(TcpClient client, CustomBufferedStre
 
                                 if (!response.KeepAlive)
                                 {
-                                    break;
+                                    return;
                                 }
 
                                 continue;
@@ -450,16 +449,22 @@ await TcpHelper.SendRaw(clientStream, connection.Stream, BufferSize,
                                     (buffer, offset, count) => { args.OnDataReceived(buffer, offset, count); },
                                     ExceptionFunc);
 
-                                break;
+                                return;
                             }
 
                             //construct the web request that we are going to issue on behalf of the client.
                             await HandleHttpSessionRequestInternal(connection, args);
 
+                            if (args.WebSession.ServerConnection == null)
+                            {
+                                //server connection was closed
+                                return;
+                            }
+
                             //if connection is closing exit
                             if (!response.KeepAlive)
                             {
-                                break;
+                                return;
                             }
                         }
                         catch (Exception e) when (!(e is ProxyHttpException))

Titanium.Web.Proxy/ResponseHandler.cs

@@ -57,6 +57,11 @@ private async Task HandleHttpSessionResponse(SessionEventArgs args)
                         //syphon out the response body from server before setting the new body
                         await args.SyphonOutBodyAsync(false);
                     }
+                    else
+                    {
+                        args.WebSession.ServerConnection.Dispose();
+                        args.WebSession.ServerConnection = null;
+                    }
 
                     return;
                 }