childflow 0.3.0

Forces DNS/proxy/interface for a child process tree and captures only its packets
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

// Copyright (c) 2026 Blacknon. All rights reserved.
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.

use anyhow::{bail, Result};

use crate::cli::{Cli, OutputView, ProxyScheme, ProxySpec};

use super::types::ProxyEnvVar;

pub fn render_proxy_uri(cli: &Cli) -> Result<Option<String>> {
    let Some(proxy) = cli.proxy.as_ref() else {
        return Ok(None);
    };

    let auth = match (&cli.proxy_user, &cli.proxy_password) {
        (Some(user), Some(password)) => Some(format!("{user}:{password}@")),
        (None, None) => None,
        _ => bail!("`--proxy-user` and `--proxy-password` must be provided together"),
    };

    let host = render_proxy_host(proxy);
    Ok(Some(format!(
        "{}://{}{host}:{}",
        render_scheme(proxy.scheme),
        auth.as_deref().unwrap_or(""),
        proxy.port
    )))
}

pub fn build_explicit_proxy_env(spec: &ProxySpec, uri: &str) -> Vec<ProxyEnvVar> {
    let mut vars = vec![
        ProxyEnvVar::new("ALL_PROXY", uri),
        ProxyEnvVar::new("all_proxy", uri),
    ];

    if matches!(spec.scheme, ProxyScheme::Http | ProxyScheme::Https) {
        vars.push(ProxyEnvVar::new("HTTP_PROXY", uri));
        vars.push(ProxyEnvVar::new("http_proxy", uri));
        vars.push(ProxyEnvVar::new("HTTPS_PROXY", uri));
        vars.push(ProxyEnvVar::new("https_proxy", uri));
    }

    vars
}

fn render_proxy_host(proxy: &ProxySpec) -> String {
    if proxy.host.contains(':') && !proxy.host.starts_with('[') {
        format!("[{}]", proxy.host)
    } else {
        proxy.host.clone()
    }
}

fn render_scheme(scheme: ProxyScheme) -> &'static str {
    match scheme {
        ProxyScheme::Http => "http",
        ProxyScheme::Https => "https",
        ProxyScheme::Socks5 => "socks5",
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::cli::Cli;
    use crate::network::NetworkBackend;

    fn base_cli() -> Cli {
        Cli {
            output: None,
            output_view: OutputView::Child,
            root: false,
            doctor: false,
            network_backend: NetworkBackend::RootlessInternal,
            dns: None,
            hosts_file: None,
            proxy: Some("http://127.0.0.1:8080".parse().unwrap()),
            proxy_user: None,
            proxy_password: None,
            proxy_insecure: false,
            iface: None,
            command: vec!["curl".into()],
        }
    }

    #[test]
    fn render_proxy_uri_embeds_credentials() {
        let mut cli = base_cli();
        cli.proxy = Some("https://proxy.example.com:443".parse().unwrap());
        cli.proxy_user = Some("alice".into());
        cli.proxy_password = Some("secret".into());

        assert_eq!(
            render_proxy_uri(&cli).unwrap(),
            Some("https://alice:secret@proxy.example.com:443".into())
        );
    }

    #[test]
    fn build_explicit_proxy_env_sets_expected_http_vars() {
        let spec: ProxySpec = "http://127.0.0.1:8080".parse().unwrap();
        let vars = build_explicit_proxy_env(&spec, "http://127.0.0.1:8080");
        let keys: Vec<_> = vars.iter().map(|var| var.key.as_str()).collect();

        assert!(keys.contains(&"HTTP_PROXY"));
        assert!(keys.contains(&"HTTPS_PROXY"));
        assert!(keys.contains(&"ALL_PROXY"));
    }
}