tab-pty-process 0.2.0

the tab multiplexer fork of tokio-pty-process. Interact with a child process through a pseudo-TTY, asynchronously using Tokio
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
106
107
108
109
110
111
112
113
114
115
116
117

// Copyright 2018 Peter Williams, Tokio Contributors
// Copyright 2019 Fabian Freyer
// Licensed under both the MIT License and the Apache-2.0 license.

//! This module is a clone of
//! <https://github.com/tokio-rs/tokio/blob/master/tokio-io/src/split.rs>
//! (commit 1119d57), modified to refer to our AsyncPtyMaster types. We need
//! to implement the splitting ourselves in order to be able to implement
//! AsRawFd for the split types.

use futures::{lock::BiLock, ready};
use std::io::{self};
use std::{
    os::unix::io::{AsRawFd, RawFd},
    task::{Context, Poll},
};
use tokio::io::{AsyncRead, AsyncWrite};

use crate::{AsAsyncPtyFd, AsyncPtyMaster};

pub fn split(master: AsyncPtyMaster) -> (AsyncPtyMasterReadHalf, AsyncPtyMasterWriteHalf) {
    let (a, b) = BiLock::new(master);
    (
        AsyncPtyMasterReadHalf { handle: a },
        AsyncPtyMasterWriteHalf { handle: b },
    )
}

/// Read half of a AsyncPtyMaster, created with AsyncPtyMaster::split.
pub struct AsyncPtyMasterReadHalf {
    handle: BiLock<AsyncPtyMaster>,
}

/// Write half of a AsyncPtyMaster, created with AsyncPtyMaster::split.
pub struct AsyncPtyMasterWriteHalf {
    handle: BiLock<AsyncPtyMaster>,
}

impl AsAsyncPtyFd for AsyncPtyMasterReadHalf {
    fn as_async_pty_fd(&self, cx: &mut Context<'_>) -> Poll<RawFd> {
        let l = ready!(self.handle.poll_lock(cx));
        Poll::Ready(l.as_raw_fd())
    }
}

impl AsAsyncPtyFd for &AsyncPtyMasterReadHalf {
    fn as_async_pty_fd(&self, cx: &mut Context<'_>) -> Poll<RawFd> {
        let l = ready!(self.handle.poll_lock(cx));
        Poll::Ready(l.as_raw_fd())
    }
}

impl AsAsyncPtyFd for &mut AsyncPtyMasterReadHalf {
    fn as_async_pty_fd(&self, cx: &mut Context<'_>) -> Poll<RawFd> {
        let l = ready!(self.handle.poll_lock(cx));
        Poll::Ready(l.as_raw_fd())
    }
}

impl AsAsyncPtyFd for AsyncPtyMasterWriteHalf {
    fn as_async_pty_fd(&self, cx: &mut Context<'_>) -> Poll<RawFd> {
        let l = ready!(self.handle.poll_lock(cx));
        Poll::Ready(l.as_raw_fd())
    }
}

impl AsAsyncPtyFd for &AsyncPtyMasterWriteHalf {
    fn as_async_pty_fd(&self, cx: &mut Context<'_>) -> Poll<RawFd> {
        let l = ready!(self.handle.poll_lock(cx));
        Poll::Ready(l.as_raw_fd())
    }
}

impl AsAsyncPtyFd for &mut AsyncPtyMasterWriteHalf {
    fn as_async_pty_fd(&self, cx: &mut Context<'_>) -> Poll<RawFd> {
        let l = ready!(self.handle.poll_lock(cx));
        Poll::Ready(l.as_raw_fd())
    }
}

impl AsyncRead for AsyncPtyMasterReadHalf {
    fn poll_read(
        self: std::pin::Pin<&mut Self>,
        cx: &mut std::task::Context<'_>,
        buf: &mut [u8],
    ) -> std::task::Poll<io::Result<usize>> {
        let mut l = ready!(self.handle.poll_lock(cx));
        l.as_pin_mut().poll_read(cx, buf)
    }
}

impl AsyncWrite for AsyncPtyMasterWriteHalf {
    fn poll_write(
        self: std::pin::Pin<&mut Self>,
        cx: &mut std::task::Context<'_>,
        buf: &[u8],
    ) -> Poll<Result<usize, io::Error>> {
        let mut l = ready!(self.handle.poll_lock(cx));
        l.as_pin_mut().poll_write(cx, buf)
    }

    fn poll_flush(
        self: std::pin::Pin<&mut Self>,
        cx: &mut std::task::Context<'_>,
    ) -> Poll<Result<(), io::Error>> {
        let mut l = ready!(self.handle.poll_lock(cx));
        l.as_pin_mut().poll_flush(cx)
    }

    fn poll_shutdown(
        self: std::pin::Pin<&mut Self>,
        cx: &mut std::task::Context<'_>,
    ) -> Poll<Result<(), io::Error>> {
        let mut l = ready!(self.handle.poll_lock(cx));
        l.as_pin_mut().poll_shutdown(cx)
    }
}