leyline-http 0.1.1

HTTP client that mimics browsers on the wire
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

#![forbid(unsafe_code)]
use tokio::net::TcpStream;

use crate::tls::error::TlsError;
use crate::tls::{SessionCache, TlsIo, TlsStream};

pub(crate) mod http;
#[cfg(feature = "socks")]
pub(crate) mod socks5;

const SOCKS5_DEFAULT_PORT: u16 = 1080;

pub(crate) async fn connect_to_proxy<C: crate::tls::TlsHandshake>(
    connector: &C,
    proxy: &url::Url,
) -> Result<TcpStream, TlsError> {
    let host = proxy
        .host_str()
        .ok_or_else(|| TlsError::proxy(format!("{} proxy has no host", proxy.scheme())))?;
    let port = proxy.port_or_known_default().unwrap_or(SOCKS5_DEFAULT_PORT);
    connector
        .dial_tcp(host, port)
        .await
        .map_err(TlsError::into_proxy)
}

pub(crate) enum Tunnel {
    Plain(TcpStream),
    Tls(TlsIo),
}

pub(crate) fn parse(proxy_url: &str) -> Result<url::Url, TlsError> {
    url::Url::parse(proxy_url).map_err(|e| TlsError::proxy(format!("invalid proxy URL: {e}")))
}

pub(crate) async fn open_tunnel<C: crate::tls::TlsHandshake>(
    connector: &C,
    host: &str,
    port: u16,
    proxy: &url::Url,
) -> Result<Tunnel, TlsError> {
    match proxy.scheme() {
        "socks5" | "socks5h" => {
            #[cfg(feature = "socks")]
            return socks5::tunnel(connector, host, port, proxy)
                .await
                .map(Tunnel::Plain);
            #[cfg(not(feature = "socks"))]
            return Err(TlsError::proxy(
                "SOCKS proxy support requires the `socks` feature",
            ));
        }
        "http" => http::tunnel(connector, host, port, proxy)
            .await
            .map(Tunnel::Plain),
        "https" => http::tunnel_via_tls(connector, host, port, proxy)
            .await
            .map(Tunnel::Tls),
        other => Err(TlsError::proxy(format!(
            "unsupported proxy scheme `{other}`: use an http, https, socks5, or socks5h proxy URL"
        ))),
    }
}

pub(crate) async fn handshake<C: crate::tls::TlsHandshake>(
    connector: &C,
    tunnel: Tunnel,
    host: &str,
    port: u16,
    proxy: &url::Url,
    include_alps: bool,
) -> Result<TlsStream, TlsError> {
    let session_key = SessionCache::key(host, port, Some(proxy));
    match tunnel {
        Tunnel::Plain(tcp) => {
            connector
                .do_tls_handshake(tcp, host, &session_key, include_alps)
                .await
        }
        Tunnel::Tls(io) => {
            connector
                .do_tls_handshake_nested(io, host, &session_key, include_alps)
                .await
        }
    }
}

#[cfg(test)]
mod tests;