libuv 2.13.1

A safe rust wrapper for libuv
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

//! To run this example, you'll need to compile cgi-tick first:
//!
//! ```bash
//! cargo build --example cgi-tick
//! cargo run --example cgi
//! ```
//!
//! Then connect to localhost:7000:
//!
//! ```bash
//! nc localhost 7000
//! ```

extern crate libuv;
use libuv::prelude::*;
use libuv::{
    exepath, ProcessHandle, ProcessOptions, StdioContainer, StdioFlags, StdioType, TcpBindFlags,
    TcpHandle,
};
use std::net::Ipv4Addr;
use std::path::PathBuf;

fn invoke_cgi_script(client: TcpHandle) -> Result<(), Box<dyn std::error::Error>> {
    let mut path: PathBuf = exepath()?.into();
    if cfg!(windows) {
        path.set_file_name("cgi-tick.exe");
    } else {
        path.set_file_name("cgi-tick");
    }

    let child_stdio = [
        Default::default(),
        StdioContainer {
            flags: StdioFlags::INHERIT_STREAM,
            data: StdioType::Stream(client.to_stream()),
        },
        Default::default(),
    ];

    let path = path.to_string_lossy().into_owned();
    let args: [&str; 1] = [&path];
    let mut options = ProcessOptions::new(&args);
    options.stdio = &child_stdio;

    let mut client_clone = client.clone();
    options.exit_cb = (move |mut handle: ProcessHandle, exit_status: i64, term_signal: i32| {
        println!(
            "Process exited with status {}, signal {}",
            exit_status, term_signal
        );
        handle.close(());
        client_clone.close(());
    })
    .into();

    client.get_loop().spawn_process(options)?;

    Ok(())
}

fn on_new_connection(mut server: StreamHandle, status: libuv::Result<u32>) {
    if let Err(e) = status {
        eprintln!("New connection error: {}", e);
        return;
    }

    if let Ok(mut client) = server.get_loop().tcp() {
        match server.accept(&mut client.to_stream()) {
            Ok(_) => {
                if let Err(e) = invoke_cgi_script(client) {
                    eprintln!("Error invoking CGI script: {}", e);
                }
            }
            Err(_) => {
                client.close(());
            }
        }
    }
}

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let mut r#loop = Loop::default()?;

    let mut server = r#loop.tcp()?;
    let addr = (Ipv4Addr::UNSPECIFIED, 7000).into();
    server.bind(&addr, TcpBindFlags::empty())?;
    server.listen(128, on_new_connection)?;

    r#loop.run(RunMode::Default)?;

    Ok(())
}