io-tubes 0.1.1

provides functionality like pwntools tube for async io in rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use std::{
    ffi::OsStr,
    io::{self, Error, ErrorKind},
    pin::Pin,
    process::Stdio,
    task::{Context, Poll},
};
use tokio::{
    io::{AsyncRead, AsyncWrite, ReadBuf},
    process::{Child, ChildStdin, ChildStdout, Command},
};

/// A tube-like struct that allows easy access to spawned process's stdin and stdout.
#[derive(Debug)]
pub struct ProcessTube {
    inner: Child,
    stdin: ChildStdin,
    stdout: ChildStdout,
}

impl ProcessTube {
    /// Create a new ProcessTube by launching a program
    pub fn new<S: AsRef<OsStr>>(program: S) -> io::Result<Self> {
        Command::new(program).try_into()
    }

    /// Create a new ProcessTube using the specified command
    pub fn from_command(cmd: Command) -> io::Result<Self> {
        cmd.try_into()
    }
}

impl TryFrom<Command> for ProcessTube {
    type Error = io::Error;

    fn try_from(mut value: Command) -> Result<Self, Self::Error> {
        value
            .stdin(Stdio::piped())
            .stdout(Stdio::piped())
            .spawn()?
            .try_into()
    }
}

impl TryFrom<Child> for ProcessTube {
    type Error = io::Error;

    fn try_from(mut inner: Child) -> Result<Self, Self::Error> {
        let stdin = inner.stdin.take().ok_or_else(|| {
            Error::new(ErrorKind::BrokenPipe, "Unable to extract stdin from child")
        })?;
        let stdout = inner.stdout.take().ok_or_else(|| {
            Error::new(ErrorKind::BrokenPipe, "Unable to extract stdout from child")
        })?;
        Ok(ProcessTube {
            inner,
            stdin,
            stdout,
        })
    }
}

impl From<ProcessTube> for Child {
    fn from(mut tube: ProcessTube) -> Self {
        tube.inner.stdin = Some(tube.stdin);
        tube.inner.stdout = Some(tube.stdout);
        tube.inner
    }
}

impl AsyncRead for ProcessTube {
    fn poll_read(
        self: Pin<&mut Self>,
        cx: &mut Context,
        buf: &mut ReadBuf,
    ) -> Poll<io::Result<()>> {
        Pin::new(&mut self.get_mut().stdout).poll_read(cx, buf)
    }
}

impl AsyncWrite for ProcessTube {
    fn poll_write(self: Pin<&mut Self>, cx: &mut Context, buf: &[u8]) -> Poll<io::Result<usize>> {
        Pin::new(&mut self.get_mut().stdin).poll_write(cx, buf)
    }

    fn poll_flush(self: Pin<&mut Self>, cx: &mut Context) -> Poll<io::Result<()>> {
        Pin::new(&mut self.get_mut().stdin).poll_flush(cx)
    }

    fn poll_shutdown(self: Pin<&mut Self>, cx: &mut Context) -> Poll<io::Result<()>> {
        Pin::new(&mut self.get_mut().stdin).poll_shutdown(cx)
    }

    fn poll_write_vectored(
        self: Pin<&mut Self>,
        cx: &mut Context,
        bufs: &[io::IoSlice],
    ) -> Poll<Result<usize, io::Error>> {
        Pin::new(&mut self.get_mut().stdin).poll_write_vectored(cx, bufs)
    }

    fn is_write_vectored(&self) -> bool {
        self.stdin.is_write_vectored()
    }
}