minor fixes

justcoding121 · justcoding121 · commit e1cd5fcd5a4c · 2016-05-20T20:52:00.000-04:00
diff --git a/Titanium.Web.Proxy/Network/TcpConnectionManager.cs b/Titanium.Web.Proxy/Network/TcpConnectionManager.cs
@@ -68,7 +68,7 @@ internal static async Task<TcpConnection> GetClient(SessionEventArgs sessionArgs
             x.IsSecure == isSecure && x.TcpClient.Connected && x.Version.Equals(version)).Count() < 2)
             {
                 var task = CreateClient(sessionArgs, hostname, port, isSecure, version)
-                            .ContinueWith(x => ReleaseClient(x.Result));
+                            .ContinueWith(x => { if (x.Status == TaskStatus.RanToCompletion) ReleaseClient(x.Result); });
             }
 
             return cached;
diff --git a/Titanium.Web.Proxy/ResponseHandler.cs b/Titanium.Web.Proxy/ResponseHandler.cs
@@ -67,8 +67,10 @@ public static async Task HandleHttpSessionResponse(SessionEventArgs args)
                     {
                         args.WebSession.Response.ResponseBody = await GetCompressedResponseBody(contentEncoding, args.WebSession.Response.ResponseBody).ConfigureAwait(false);
 
-                        if (isChunked==false)
-                         args.WebSession.Response.ContentLength = args.WebSession.Response.ResponseBody.Length;
+                        if (isChunked == false)
+                            args.WebSession.Response.ContentLength = args.WebSession.Response.ResponseBody.Length;
+                        else
+                            args.WebSession.Response.ContentLength = -1;
                     }
 
                     await WriteResponseHeaders(args.Client.ClientStreamWriter, args.WebSession.Response.ResponseHeaders).ConfigureAwait(false);

Original file line number	Diff line number	Diff line change
`@@ -68,7 +68,7 @@ internal static async Task<TcpConnection> GetClient(SessionEventArgs sessionArgs`
`68`	`68`	`x.IsSecure == isSecure && x.TcpClient.Connected && x.Version.Equals(version)).Count() < 2)`
`69`	`69`	`{`
`70`	`70`	`var task = CreateClient(sessionArgs, hostname, port, isSecure, version)`
`71`		`- .ContinueWith(x => ReleaseClient(x.Result));`
	`71`	`+ .ContinueWith(x => { if (x.Status == TaskStatus.RanToCompletion) ReleaseClient(x.Result); });`
`72`	`72`	`}`
`73`	`73`
`74`	`74`	`return cached;`
Original file line number	Diff line number	Diff line change
`@@ -67,8 +67,10 @@ public static async Task HandleHttpSessionResponse(SessionEventArgs args)`
`67`	`67`	`{`
`68`	`68`	`args.WebSession.Response.ResponseBody = await GetCompressedResponseBody(contentEncoding, args.WebSession.Response.ResponseBody).ConfigureAwait(false);`
`69`	`69`
`70`		`- if (isChunked==false)`
`71`		`- args.WebSession.Response.ContentLength = args.WebSession.Response.ResponseBody.Length;`
	`70`	`+ if (isChunked == false)`
	`71`	`+ args.WebSession.Response.ContentLength = args.WebSession.Response.ResponseBody.Length;`
	`72`	`+ else`
	`73`	`+ args.WebSession.Response.ContentLength = -1;`
`72`	`74`	`}`
`73`	`75`
`74`	`76`	`await WriteResponseHeaders(args.Client.ClientStreamWriter, args.WebSession.Response.ResponseHeaders).ConfigureAwait(false);`