skeleton

Author: dignifiedquire

src/std_adapter.rs

@@ -1,9 +1,9 @@
-use std::io::{self, Read, Write};
+use std::io::{self, BufRead, Read, Write};
 use std::marker::Unpin;
 use std::pin::Pin;
 use std::task::{Context, Poll};
 
-use futures_io::{AsyncRead, AsyncWrite};
+use futures_io::{AsyncBufRead, AsyncRead, AsyncWrite};
 
 #[derive(Debug)]
 pub(crate) struct StdAdapter<S> {
@@ -43,6 +43,24 @@ where
     }
 }
 
+impl<S> BufRead for StdAdapter<S>
+where
+    S: AsyncBufRead + Unpin,
+{
+    fn fill_buf(&mut self) -> io::Result<&[u8]> {
+        unimplemented!()
+        // TODO: make it compile
+        // match self.with_context(|ctx, stream| stream.poll_fill_buf(ctx)) {
+        //     Poll::Ready(buf) => buf,
+        //     Poll::Pending => Err(io::Error::from(io::ErrorKind::WouldBlock)),
+        // }
+    }
+
+    fn consume(&mut self, amt: usize) {
+        self.consume(amt)
+    }
+}
+
 impl<S> Write for StdAdapter<S>
 where
     S: AsyncWrite + Unpin,

src/tls_stream.rs

@@ -1,10 +1,10 @@
-use std::io::{self, Read, Write};
+use std::io::{self, BufRead, Read, Write};
 use std::marker::Unpin;
 use std::pin::Pin;
 use std::ptr::null_mut;
 use std::task::{Context, Poll};
 
-use futures_io::{AsyncRead, AsyncWrite};
+use futures_io::{AsyncBufRead, AsyncRead, AsyncWrite};
 
 use crate::std_adapter::StdAdapter;
 
@@ -64,6 +64,19 @@ where
     }
 }
 
+impl<S> AsyncBufRead for TlsStream<S>
+where
+    S: AsyncBufRead + AsyncWrite + Unpin,
+{
+    fn poll_fill_buf(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<&[u8]>> {
+        self.with_context(cx, |s| cvt(s.fill_buf()))
+    }
+
+    fn consume(self: Pin<&mut Self>, amt: usize) {
+        self.consume(amt)
+    }
+}
+
 impl<S> AsyncWrite for TlsStream<S>
 where
     S: AsyncRead + AsyncWrite + Unpin,