loona 0.3.4

HTTP 1 and 2 on top of io_uring
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

use std::{any::Any, net::SocketAddr, path::PathBuf, process::Stdio};

use b_x::BxForResults;
#[cfg(target_os = "linux")]
use libc::{prctl, PR_SET_PDEATHSIG, SIGKILL};

use tokio::{
    io::{AsyncBufReadExt, BufReader},
    process::Command,
};
use tracing::debug;

#[cfg(target_os = "windows")]
const EXE_FILE_EXT: &str = ".exe";

#[cfg(not(target_os = "windows"))]
const EXE_FILE_EXT: &str = "";

pub async fn start() -> b_x::Result<(SocketAddr, impl Any)> {
    let (addr_tx, addr_rx) = tokio::sync::oneshot::channel::<SocketAddr>();

    let manifest_dir = std::env::var("CARGO_MANIFEST_DIR").unwrap();
    let manifest_dir = PathBuf::from(manifest_dir);
    let project_dir = manifest_dir.parent().unwrap().parent().unwrap();

    let exe_name = format!("httpwg-hyper{EXE_FILE_EXT}");
    let binary_path = project_dir.join("target").join("release").join(exe_name);
    eprintln!("Using testbed binary: {}", binary_path.display());
    let mut cmd = Command::new(binary_path);
    cmd.stdout(Stdio::piped());
    cmd.env("PROTO", "h1");

    // Only Linux gets the nice "I'm taking you with me" feature for now.
    #[cfg(target_os = "linux")]
    unsafe {
        cmd.pre_exec(|| {
            prctl(PR_SET_PDEATHSIG, SIGKILL);
            Ok(())
        })
    };

    cmd.kill_on_drop(true);

    let mut child = cmd.spawn()?;
    let stdout = child.stdout.take().unwrap();
    let mut addr_tx = Some(addr_tx);

    loona::buffet::spawn(async move {
        let stdout = BufReader::new(stdout);
        let mut lines = stdout.lines();
        while let Some(line) = lines.next_line().await.unwrap() {
            let settings = httpwg_harness::Settings::from_env().unwrap();
            if let Ok(Some(addr)) = settings.decode_listen_line(&line) {
                if let Some(addr_tx) = addr_tx.take() {
                    addr_tx.send(addr).unwrap();
                }
            } else {
                debug!("[upstream] {}", line);
            }
        }
    });

    let upstream_addr = addr_rx.await.bx()?;
    Ok((upstream_addr, child))
}