connection timeout for web socket requests

Author: justcoding121

Titanium.Web.Proxy/Helpers/Tcp.cs

@@ -28,7 +28,7 @@ internal class TcpHelper
         /// <param name="isHttps"></param>
         /// <returns></returns>
         internal static async Task SendRaw(Stream clientStream, string httpCmd, Dictionary<string, HttpHeader> requestHeaders, string hostName,
-            int tunnelPort, bool isHttps, SslProtocols supportedProtocols)
+            int tunnelPort, bool isHttps, SslProtocols supportedProtocols, int connectionTimeOutSeconds)
         {
             //prepare the prefix content
             StringBuilder sb = null;
@@ -83,6 +83,12 @@ internal static async Task SendRaw(Stream clientStream, string httpCmd, Dictiona
                     }
                 }
 
+                tunnelClient.SendTimeout = connectionTimeOutSeconds * 1000;
+                tunnelClient.ReceiveTimeout = connectionTimeOutSeconds * 1000;
+
+                tunnelStream.ReadTimeout = connectionTimeOutSeconds * 1000;
+                tunnelStream.WriteTimeout = connectionTimeOutSeconds * 1000;
+
                 Task sendRelay;
 
                 //Now async relay all server=>client & client=>server data

Titanium.Web.Proxy/Network/TcpConnectionFactory.cs

@@ -129,6 +129,9 @@ private async Task<TcpConnection> CreateClient(string hostname, int port, bool i
             client.ReceiveTimeout = connectionTimeOutSeconds * 1000;
             client.SendTimeout = connectionTimeOutSeconds * 1000;
 
+            stream.ReadTimeout = connectionTimeOutSeconds * 1000;
+            stream.WriteTimeout = connectionTimeOutSeconds * 1000;
+
             return new TcpConnection()
             {
                 HostName = hostname,

Titanium.Web.Proxy/RequestHandler.cs

@@ -127,7 +127,7 @@ await sslStream.AuthenticateAsServerAsync(certificate, false,
 
                     //Just relay the request/response without decrypting it
                     await TcpHelper.SendRaw(clientStream, null, null, httpRemoteUri.Host, httpRemoteUri.Port,
-                        false, SupportedSslProtocols);
+                        false, SupportedSslProtocols, ConnectionTimeOutSeconds);
 
                     Dispose(client, clientStream, clientStreamReader, clientStreamWriter, null);
                     return;
@@ -311,7 +311,8 @@ private async Task HandleHttpSessionRequest(TcpClient client, string httpCmd, St
                     if (args.WebSession.Request.UpgradeToWebSocket)
                     {
                         await TcpHelper.SendRaw(clientStream, httpCmd, args.WebSession.Request.RequestHeaders,
-                                 httpRemoteUri.Host, httpRemoteUri.Port, args.IsHttps, SupportedSslProtocols);
+                                 httpRemoteUri.Host, httpRemoteUri.Port, args.IsHttps, SupportedSslProtocols, ConnectionTimeOutSeconds);
+
                         Dispose(client, clientStream, clientStreamReader, clientStreamWriter, args);
                         break;
                     }