summaryrefslogtreecommitdiffstats
path: root/tokio/src/io/util/write_buf.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tokio/src/io/util/write_buf.rs')
-rw-r--r--tokio/src/io/util/write_buf.rs13
1 files changed, 5 insertions, 8 deletions
diff --git a/tokio/src/io/util/write_buf.rs b/tokio/src/io/util/write_buf.rs
index e49282fe..cedfde64 100644
--- a/tokio/src/io/util/write_buf.rs
+++ b/tokio/src/io/util/write_buf.rs
@@ -20,7 +20,7 @@ cfg_io_util! {
/// asynchronous manner, returning a future.
pub(crate) fn write_buf<'a, W, B>(writer: &'a mut W, buf: &'a mut B) -> WriteBuf<'a, W, B>
where
- W: AsyncWrite,
+ W: AsyncWrite + Unpin,
B: Buf,
{
WriteBuf { writer, buf }
@@ -28,16 +28,13 @@ where
impl<W, B> Future for WriteBuf<'_, W, B>
where
- W: AsyncWrite,
+ W: AsyncWrite + Unpin,
B: Buf,
{
type Output = io::Result<usize>;
- fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<usize>> {
- // safety: no data is moved from self
- unsafe {
- let me = self.get_unchecked_mut();
- Pin::new_unchecked(&mut *me.writer).poll_write_buf(cx, &mut me.buf)
- }
+ fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<usize>> {
+ let me = &mut *self;
+ Pin::new(&mut *me.writer).poll_write_buf(cx, me.buf)
}
}