Allow access to underlying IClient instance through IHttpContext.Client

uhttpsharp/Clients/ClientSslDecoerator.cs → uhttpsharp/Clients/ClientSslDecorator.cs

@@ -21,7 +21,7 @@ public ClientSslDecorator(IClient child, X509Certificate certificate)
             _sslStream = new SslStream(_child.Stream);
         }
 
-        public async Task AuthenticateAsServer()
+        public async Task InitializeStream()
         {
             Task timeout = Task.Delay(TimeSpan.FromSeconds(10));
             if (timeout == await Task.WhenAny(_sslStream.AuthenticateAsServerAsync(_certificate, false, SslProtocols.Tls, true), timeout).ConfigureAwait(false))

uhttpsharp/Clients/IClient.cs

@@ -1,5 +1,6 @@
 using System.IO;
 using System.Net;
+using System.Threading.Tasks;
 
 namespace uhttpsharp.Clients
 {
@@ -14,7 +15,7 @@ public interface IClient
 
         EndPoint RemoteEndPoint { get; }
 
-
+        Task InitializeStream();
 
     }
 }

uhttpsharp/Clients/TcpClientAdapter.cs

@@ -1,6 +1,7 @@
 using System.IO;
 using System.Net;
 using System.Net.Sockets;
+using System.Threading.Tasks;
 
 namespace uhttpsharp.Clients
 {
@@ -45,5 +46,9 @@ public EndPoint RemoteEndPoint
         {
             get { return _client.Client.RemoteEndPoint; }
         }
+
+        public Task InitializeStream() {
+            return Task.Factory.GetCompleted();
+        }
     }
 }

uhttpsharp/HttpClient.cs

@@ -60,21 +60,13 @@ public HttpClientHandler(IClient client, Func<IHttpContext, Task> requestHandler
             UpdateLastOperationTime();
         }
 
-        private async Task InitializeStream()
-        {
-            if (Client is ClientSslDecorator)
-            {
-                await ((ClientSslDecorator)Client).AuthenticateAsServer().ConfigureAwait(false);
-            }
-
-            _stream = new BufferedStream(_client.Stream, 8096);
-        }
-
         private async void Process()
         {
             try
             {
-                await InitializeStream();
+                await _client.InitializeStream().ConfigureAwait(false);
+
+                _stream = new BufferedStream(_client.Stream, 8096);
 
                 while (_client.Connected)
                 {
@@ -89,7 +81,7 @@ private async void Process()
                     {
                         UpdateLastOperationTime();
 
-                        var context = new HttpContext(request, _client.RemoteEndPoint);
+                        var context = new HttpContext(request, _client);
 
                         Logger.InfoFormat("{1} : Got request {0}", request.Uri, _client.RemoteEndPoint);
 

uhttpsharp/HttpContext.cs

@@ -1,19 +1,20 @@
 using System.Dynamic;
 using System.Net;
+using uhttpsharp.Clients;
 using uhttpsharp.Headers;
 
 namespace uhttpsharp
 {
     internal class HttpContext : IHttpContext
     {
         private readonly IHttpRequest _request;
-        private readonly EndPoint _remoteEndPoint;
+        private readonly IClient _client;
         private readonly ICookiesStorage _cookies;
         private readonly ExpandoObject _state = new ExpandoObject();
-        public HttpContext(IHttpRequest request, EndPoint remoteEndPoint)
+        public HttpContext(IHttpRequest request, IClient client)
         {
             _request = request;
-            _remoteEndPoint = remoteEndPoint;
+            _client = client;
             _cookies = new CookiesStorage(_request.Headers.GetByNameOrDefault("cookie", string.Empty));
         }
 
@@ -29,14 +30,18 @@ public ICookiesStorage Cookies
             get { return _cookies; }
         }
 
-
         public dynamic State
         {
             get { return _state; }
         }
+
+        public IClient Client {
+            get { return _client; }
+        }
+
         public EndPoint RemoteEndPoint
         {
-            get { return _remoteEndPoint; }
+            get { return _client.RemoteEndPoint; }
         }
     }
 }

uhttpsharp/IHttpContext.cs

@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using System.Net;
 using System.Text;
+using uhttpsharp.Clients;
 using uhttpsharp.Headers;
 
 namespace uhttpsharp
@@ -16,6 +17,8 @@ public interface IHttpContext
 
         dynamic State { get; }
 
+        IClient Client { get; }
+
         EndPoint RemoteEndPoint { get; }
     }
 

uhttpsharp/uhttpsharp.csproj

@@ -56,7 +56,7 @@
     <Compile Include="App_Packages\LibLog.3.1\LibLog.cs" />
     <Compile Include="Attributes\IModelBinding.cs" />
     <Compile Include="Attributes\HttpMethodAttribute.cs" />
-    <Compile Include="Clients\ClientSslDecoerator.cs" />
+    <Compile Include="Clients\ClientSslDecorator.cs" />
     <Compile Include="Clients\IClient.cs" />
     <Compile Include="Controllers\ErrorContainer.cs" />
     <Compile Include="Controllers\IErrorContainer.cs" />
@@ -102,7 +102,7 @@
     <Compile Include="IHttpMethodProvider.cs" />
     <Compile Include="LimitedStream.cs" />
     <Compile Include="Listeners\IHttpListener.cs" />
-    <Compile Include="Listeners\SslListenerDecoerator.cs" />
+    <Compile Include="Listeners\SslListenerDecorator.cs" />
     <Compile Include="Listeners\TcpListenerAdapter.cs" />
     <Compile Include="ModelBinders\IModelBinder.cs" />
     <Compile Include="ModelBinders\JsonModelBinder.cs" />