optional prefetch

Author: justcoding121

examples/Titanium.Web.Proxy.Examples.Basic/ProxyTestController.cs

@@ -24,7 +24,6 @@ public class ProxyTestController
         public ProxyTestController()
         {
             proxyServer = new ProxyServer();
-            proxyServer.EnableConnectionPool = true;
             // generate root certificate without storing it in file system
             //proxyServer.CertificateManager.CreateRootCertificate(false);
 

src/Titanium.Web.Proxy/ExplicitClientHandler.cs

@@ -152,11 +152,14 @@ await clientStreamWriter.WriteResponseAsync(connectArgs.WebSession.Response,
 
                         SslStream sslStream = null;
 
-                        //don't pass cancellation token here
-                        //it could cause floating server connections when client exits
-                        prefetchConnectionTask = tcpConnectionFactory.GetServerConnection(this, connectArgs,
-                                                isConnect: true, applicationProtocols: null, noCache: false,
-                                                cancellationToken: CancellationToken.None);
+                        if (EnableTcpServerConnectionPrefetch)
+                        {
+                            //don't pass cancellation token here
+                            //it could cause floating server connections when client exits
+                            prefetchConnectionTask = tcpConnectionFactory.GetServerConnection(this, connectArgs,
+                                                    isConnect: true, applicationProtocols: null, noCache: false,
+                                                    cancellationToken: CancellationToken.None);
+                        }
 
                         try
                         {
@@ -204,10 +207,10 @@ await clientStreamWriter.WriteResponseAsync(connectArgs.WebSession.Response,
                             decryptSsl = false;
                         }
 
-                        if(!decryptSsl)
+                        if (!decryptSsl)
                         {
                             await tcpConnectionFactory.Release(prefetchConnectionTask, true);
-                            prefetchConnectionTask = null; 
+                            prefetchConnectionTask = null;
                         }
                     }
 

src/Titanium.Web.Proxy/ProxyServer.cs

@@ -161,9 +161,19 @@ public ProxyServer(string rootCertificateName, string rootCertificateIssuerName,
 
         /// <summary>
         ///     Should we enable experimental server connection pool?
-        ///     Defaults to disable.
+        ///     Defaults to true.
         /// </summary>
-        public bool EnableConnectionPool { get; set; }
+        public bool EnableConnectionPool { get; set; } = true;
+
+        /// <summary>
+        ///     Should we enable tcp server connection prefetching?
+        ///     When enabled, as soon as we receive a client connection we concurrently initiate 
+        ///     corresponding server connection process using CONNECT hostname or SNI hostname on a separate task so that after parsing client request
+        ///     we will have the server connection immediately ready or in the process of getting ready.
+        ///     If a server connection is available in cache then this prefetch task will immediatly return with the available connection from cache.
+        ///     Defaults to true.
+        /// </summary>
+        public bool EnableTcpServerConnectionPrefetch { get; set; } = true;
 
         /// <summary>
         ///     Buffer size in bytes used throughout this proxy.
@@ -186,14 +196,14 @@ public ProxyServer(string rootCertificateName, string rootCertificateIssuerName,
         public int MaxCachedConnections { get; set; } = 2;
 
         /// <summary>
-        /// Number of seconds to linger when Tcp connection is in TIME_WAIT state.
-        /// Default value is 30.
+        ///     Number of seconds to linger when Tcp connection is in TIME_WAIT state.
+        ///     Default value is 30.
         /// </summary>
         public int TcpTimeWaitSeconds { get; set; } = 30;
 
         /// <summary>
-        /// Should we reuse client/server tcp sockets.
-        /// Default is true (disabled for linux/macOS due to bug in .Net core).
+        ///     Should we reuse client/server tcp sockets.
+        ///     Default is true (disabled for linux/macOS due to bug in .Net core).
         /// </summary>
         public bool ReuseSocket { get; set; } = true;
 

src/Titanium.Web.Proxy/TransparentClientHandler.cs

@@ -63,13 +63,16 @@ private async Task handleClient(TransparentProxyEndPoint endPoint, TcpClientConn
 
                     if (endPoint.DecryptSsl && args.DecryptSsl)
                     {
-                        //don't pass cancellation token here
-                        //it could cause floating server connections when client exits
-                        prefetchConnectionTask = tcpConnectionFactory.GetServerConnection(httpsHostName, endPoint.Port,
-                                httpVersion: null, isHttps: true, applicationProtocols: null, isConnect: false,
-                                proxyServer: this, upStreamEndPoint: UpStreamEndPoint, externalProxy: UpStreamHttpsProxy,
-                                noCache: false, cancellationToken: CancellationToken.None);
-
+                        if(EnableTcpServerConnectionPrefetch)
+                        {
+                            //don't pass cancellation token here
+                            //it could cause floating server connections when client exits
+                            prefetchConnectionTask = tcpConnectionFactory.GetServerConnection(httpsHostName, endPoint.Port,
+                                    httpVersion: null, isHttps: true, applicationProtocols: null, isConnect: false,
+                                    proxyServer: this, upStreamEndPoint: UpStreamEndPoint, externalProxy: UpStreamHttpsProxy,
+                                    noCache: false, cancellationToken: CancellationToken.None);
+                        }
+                        
                         SslStream sslStream = null;
 
                         //do client authentication using fake certificate