aboutsummaryrefslogtreecommitdiffstats
path: root/examples/basic.rs
blob: 39ef088588165941f632ac8f01589c04132fde47 (plain) (blame)
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
use std::io::{Read as _, Write as _};
use std::os::unix::io::AsRawFd as _;

fn main() {
    let mut cmd = pty_process::Command::new("cat").unwrap();
    // cmd.args(&["--color=auto"]);
    let mut child = cmd.spawn().unwrap();
    let mut buf = [0_u8; 4096];
    let pty = cmd.pty().as_raw_fd();
    let stdin = std::io::stdin().as_raw_fd();
    loop {
        let mut set = nix::sys::select::FdSet::new();
        set.insert(pty);
        set.insert(stdin);
        match nix::sys::select::select(None, Some(&mut set), None, None, None)
        {
            Ok(n) => {
                if n > 0 {
                    if set.contains(pty) {
                        match cmd.pty().read(&mut buf) {
                            Ok(bytes) => {
                                let buf = &buf[..bytes];
                                print!(
                                    "got {} bytes: '{}'",
                                    bytes,
                                    std::str::from_utf8(buf).unwrap()
                                );
                                std::io::stdout().flush().unwrap();
                            }
                            Err(e) => {
                                eprintln!("pty read failed: {:?}", e);
                                break;
                            }
                        };
                    }
                    if set.contains(stdin) {
                        match std::io::stdin().read(&mut buf) {
                            Ok(bytes) => {
                                let buf = &buf[..bytes];
                                cmd.pty().write_all(buf).unwrap();
                            }
                            Err(e) => {
                                eprintln!("stdin read failed: {:?}", e);
                                break;
                            }
                        }
                    }
                }
            }
            Err(e) => println!("select failed: {:?}", e),
        }
    }
    child.wait().unwrap();
}