Show response data as image in test app, try-catch added to http2 test

Author: honfika

examples/Titanium.Web.Proxy.Examples.Wpf/MainWindow.xaml

@@ -36,13 +36,20 @@
         </ListView>
         <TabControl Grid.Column="2" Grid.Row="0">
             <TabItem Header="Session">
-                <Grid Background="Red" HorizontalAlignment="Stretch" VerticalAlignment="Stretch">
+                <Grid HorizontalAlignment="Stretch" VerticalAlignment="Stretch">
                     <Grid.RowDefinitions>
                         <RowDefinition />
                         <RowDefinition />
                     </Grid.RowDefinitions>
                     <TextBox x:Name="TextBoxRequest" Grid.Row="0" />
-                    <TextBox x:Name="TextBoxResponse" Grid.Row="1" />
+                    <TabControl Grid.Row="1">
+                        <TabItem Header="Text">
+                            <TextBox x:Name="TextBoxResponse" />
+                        </TabItem>
+                        <TabItem Header="Image">
+                            <Image x:Name="ImageResponse" HorizontalAlignment="Center"  VerticalAlignment="Center" Stretch="None" />
+                        </TabItem>
+                    </TabControl>
                 </Grid>
             </TabItem>
         </TabControl>

examples/Titanium.Web.Proxy.Examples.Wpf/MainWindow.xaml.cs

@@ -1,13 +1,15 @@
 using System;
 using System.Collections.Generic;
 using System.Collections.ObjectModel;
+using System.IO;
 using System.Linq;
 using System.Net;
 using System.Text;
 using System.Threading.Tasks;
 using System.Windows;
 using System.Windows.Controls;
 using System.Windows.Input;
+using System.Windows.Media.Imaging;
 using Titanium.Web.Proxy.EventArguments;
 using Titanium.Web.Proxy.Http;
 using Titanium.Web.Proxy.Models;
@@ -297,7 +299,8 @@ private void selectedSessionChanged()
 
             var session = SelectedSession.HttpClient;
             var request = session.Request;
-            var data = (request.IsBodyRead ? request.Body : null) ?? new byte[0];
+            var fullData = (request.IsBodyRead ? request.Body : null) ?? new byte[0];
+            var data = fullData;
             bool truncated = data.Length > truncateLimit;
             if (truncated)
             {
@@ -313,7 +316,8 @@ private void selectedSessionChanged()
             TextBoxRequest.Text = sb.ToString();
 
             var response = session.Response;
-            data = (response.IsBodyRead ? response.Body : null) ?? new byte[0];
+            fullData = (response.IsBodyRead ? response.Body : null) ?? new byte[0];
+            data = fullData;
             truncated = data.Length > truncateLimit;
             if (truncated)
             {
@@ -333,6 +337,19 @@ private void selectedSessionChanged()
             }
 
             TextBoxResponse.Text = sb.ToString();
+
+            try
+            {
+                using (MemoryStream stream = new MemoryStream(fullData))
+                {
+                    ImageResponse.Source =
+                        BitmapFrame.Create(stream, BitmapCreateOptions.None, BitmapCacheOption.OnLoad);
+                }
+            }
+            catch
+            {
+                ImageResponse.Source = null;
+            }
         }
     }
 }

src/StreamExtended/Network/CustomBufferedStream.cs

@@ -245,8 +245,7 @@ public override int ReadByte()
             {
                 return -1;
             }
-
-
+            
             return streamBuffer[bufferPos + index];
         }
 
@@ -491,7 +490,7 @@ public bool FillBuffer()
             if (bufferLength > 0)
             {
                 //normally we fill the buffer only when it is empty, but sometimes we need more data
-                //move the remanining data to the beginning of the buffer 
+                //move the remaining data to the beginning of the buffer 
                 Buffer.BlockCopy(streamBuffer, bufferPos, streamBuffer, 0, bufferLength);
             }
 
@@ -516,7 +515,6 @@ public bool FillBuffer()
             }
 
             return result;
-
         }
 
         /// <summary>

src/Titanium.Web.Proxy/ExplicitClientHandler.cs

@@ -47,8 +47,7 @@ private async Task handleClient(ExplicitProxyEndPoint endPoint, TcpClientConnect
             {
                 string connectHostname = null;
                 TunnelConnectSessionEventArgs connectArgs = null;
-
-
+                
                 // Client wants to create a secure tcp tunnel (probably its a HTTPS or Websocket request)
                 if (await HttpHelper.IsConnectMethod(clientStream) == 1)
                 {
@@ -142,15 +141,22 @@ await clientStreamWriter.WriteResponseAsync(connectArgs.HttpClient.Response,
                         if (alpn != null && alpn.Contains(SslApplicationProtocol.Http2))
                         {
                             // test server HTTP/2 support
-                            // todo: this is a hack, because Titanium does not support HTTP protocol changing currently
-                            var connection = await tcpConnectionFactory.GetServerConnection(this, connectArgs,
-                                                    isConnect: true, applicationProtocols: SslExtensions.Http2ProtocolAsList,
-                                                    noCache: true, cancellationToken: cancellationToken);
-
-                            http2Supported = connection.NegotiatedApplicationProtocol == SslApplicationProtocol.Http2;
-
-                            //release connection back to pool instead of closing when connection pool is enabled.
-                            await tcpConnectionFactory.Release(connection, true);
+                            try
+                            {
+                                // todo: this is a hack, because Titanium does not support HTTP protocol changing currently
+                                var connection = await tcpConnectionFactory.GetServerConnection(this, connectArgs,
+                                    isConnect: true, applicationProtocols: SslExtensions.Http2ProtocolAsList,
+                                    noCache: true, cancellationToken: cancellationToken);
+
+                                http2Supported = connection.NegotiatedApplicationProtocol ==
+                                                 SslApplicationProtocol.Http2;
+                                //release connection back to pool instead of closing when connection pool is enabled.
+                                await tcpConnectionFactory.Release(connection, true);
+                            }
+                            catch (Exception ex)
+                            {
+                                // ignore
+                            }
                         }
 
                         if (EnableTcpServerConnectionPrefetch)