From 185f01093783b1356d3d135eaa1475b3584b723e Mon Sep 17 00:00:00 2001
From: PZ Read <netfirewall@gmail.com>
Date: Fri, 26 May 2017 16:38:02 +0800
Subject: [PATCH] Add async builders for custom session.

---
 src/lib.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 8728219..87f461f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -52,6 +52,15 @@ impl ClientConfigExt for Arc<ClientConfig> {
     }
 }
 
+pub fn connect_async_with_session<S>(stream: S, session: ClientSession)
+    -> ConnectAsync<S>
+    where S: AsyncRead + AsyncWrite
+{
+    ConnectAsync(MidHandshake {
+        inner: Some(TlsStream::new(stream, session))
+    })
+}
+
 impl ServerConfigExt for Arc<ServerConfig> {
     fn accept_async<S>(&self, stream: S)
         -> AcceptAsync<S>
@@ -63,6 +72,15 @@ impl ServerConfigExt for Arc<ServerConfig> {
     }
 }
 
+pub fn accept_async_with_session<S>(stream: S, session: ServerSession)
+    -> AcceptAsync<S>
+    where S: AsyncRead + AsyncWrite
+{
+    AcceptAsync(MidHandshake {
+        inner: Some(TlsStream::new(stream, session))
+    })
+}
+
 impl<S: AsyncRead + AsyncWrite> Future for ConnectAsync<S> {
     type Item = TlsStream<S, ClientSession>;
     type Error = io::Error;