handling keepalive properly when user respond in the BeforeRequest handler

Author: honfika

Titanium.Web.Proxy/RequestHandler.cs

@@ -429,10 +429,18 @@ private async Task HandleHttpSessionRequest(TcpClient client, CustomBufferedStre
                             await BeforeRequest.InvokeAsync(this, args, ExceptionFunc);
                         }
 
+                        var response = args.WebSession.Response;
+
                         if (args.WebSession.Request.CancelRequest)
                         {
                             await HandleHttpSessionResponse(args);
-                            break;
+
+                            if (!response.KeepAlive)
+                            {
+                                break;
+                            }
+
+                            continue;
                         }
 
                         //create a new connection if hostname/upstream end point changes
@@ -450,8 +458,6 @@ private async Task HandleHttpSessionRequest(TcpClient client, CustomBufferedStre
                             connection = await GetServerConnection(args, false);
                         }
 
-                        var response = args.WebSession.Response;
-
                         //if upgrading to websocket then relay the requet without reading the contents
                         if (args.WebSession.Request.UpgradeToWebSocket)
                         {
@@ -491,7 +497,7 @@ await TcpHelper.SendRaw(clientStream, connection.Stream, BufferSize,
                         await HandleHttpSessionRequestInternal(connection, args);
 
                         //if connection is closing exit
-                        if (response.KeepAlive == false)
+                        if (!response.KeepAlive)
                         {
                             break;
                         }