mirror of
https://github.com/zhom/donutbrowser.git
synced 2026-08-13 14:40:39 +02:00
refactor: cleanup
This commit is contained in:
@@ -0,0 +1,511 @@
|
||||
//! Whether a profile's exit node can be reached from somewhere that is not this
|
||||
//! machine.
|
||||
//!
|
||||
//! Remote execution — an interactive remote session or a Cookie Bot night — runs
|
||||
//! the browser on a leased fleet host, but the PROFILE (and its proxy, and its
|
||||
//! VPN config) is pulled from the user's sync namespace. Nothing in that
|
||||
//! handover rewrites addresses, so a proxy recorded as `127.0.0.1:8080` arrives
|
||||
//! on the fleet host meaning *the fleet host's own loopback*.
|
||||
//!
|
||||
//! That is the whole bug this module exists to prevent. The server already
|
||||
//! refuses a profile with NO exit (`proxy_required`), because a night browsed
|
||||
//! from the fleet's datacenter address damages an identity rather than building
|
||||
//! it — but it was asking whether an exit was *configured*, never whether it was
|
||||
//! *reachable*. A local proxy satisfied the first question and failed the
|
||||
//! second, so the run was accepted, dispatched, and burned a leased host either
|
||||
//! erroring out or (worse) egressing direct from the datacenter: exactly the
|
||||
//! outcome `proxy_required` exists to stop, reached by the one route it did not
|
||||
//! check.
|
||||
//!
|
||||
//! Local proxies are not an exotic case. A local MITM proxy, an SSH tunnel, a
|
||||
//! locally-run SOCKS client and Donut's own VLESS support all present to the
|
||||
//! browser as `127.0.0.1:<port>`.
|
||||
//!
|
||||
//! This module is the single answer, shared by every caller, and it FAILS
|
||||
//! CLOSED: anything it cannot parse is reported as unreachable. Refusing a
|
||||
//! working setup costs the user one support question; accepting a broken one
|
||||
//! costs a burned hour and a damaged profile identity.
|
||||
|
||||
use std::net::{IpAddr, Ipv4Addr, Ipv6Addr};
|
||||
|
||||
/// Whether a leased fleet host could dial this profile's exit.
|
||||
#[derive(Debug, Clone, PartialEq, Eq)]
|
||||
pub enum ExitReachability {
|
||||
/// No proxy and no VPN. The caller's existing "no exit" refusal applies.
|
||||
None,
|
||||
/// An address a host elsewhere on the internet can reach.
|
||||
Remote,
|
||||
/// An address that only means anything on this machine or this LAN.
|
||||
LocalOnly {
|
||||
/// The offending host, for a message the user can act on.
|
||||
host: String,
|
||||
/// Which part of the config it came from: "proxy" or "VPN".
|
||||
source: &'static str,
|
||||
},
|
||||
/// Configured, but this code could not determine the host.
|
||||
///
|
||||
/// Treated as unreachable by [`ExitReachability::is_remote`] — see the
|
||||
/// fail-closed note in the module docs.
|
||||
Unknown {
|
||||
reason: String,
|
||||
source: &'static str,
|
||||
},
|
||||
}
|
||||
|
||||
impl ExitReachability {
|
||||
/// Whether remote execution may proceed.
|
||||
pub fn is_remote(&self) -> bool {
|
||||
matches!(self, ExitReachability::Remote)
|
||||
}
|
||||
|
||||
/// A one-line reason for a refusal, or None when there is nothing to refuse.
|
||||
pub fn refusal_detail(&self) -> Option<String> {
|
||||
match self {
|
||||
ExitReachability::Remote | ExitReachability::None => None,
|
||||
ExitReachability::LocalOnly { host, source } => Some(format!(
|
||||
"The {source} for this profile points at {host}, which only exists on this computer. \
|
||||
Remote runs happen on our hosts and cannot reach it."
|
||||
)),
|
||||
ExitReachability::Unknown { reason, source } => Some(format!(
|
||||
"The {source} for this profile could not be read ({reason}), so we cannot confirm a \
|
||||
remote host could use it."
|
||||
)),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Whether a hostname or IP literal is reachable from another machine.
|
||||
///
|
||||
/// Rejects, in order: empty/whitespace, unparsable-as-either, and every IP
|
||||
/// range that is scoped to a machine or a private network. Hostnames that are
|
||||
/// not IP literals are accepted unless they use a name suffix that is
|
||||
/// definitionally local — a public DNS name cannot be validated here without a
|
||||
/// lookup, and doing a lookup would make this impure and slow on a hot path.
|
||||
pub fn host_is_remote_reachable(host: &str) -> bool {
|
||||
let host = normalize_host(host);
|
||||
if host.is_empty() {
|
||||
return false;
|
||||
}
|
||||
|
||||
if let Ok(ip) = host.parse::<IpAddr>() {
|
||||
return ip_is_remote_reachable(ip);
|
||||
}
|
||||
|
||||
let lower = host.to_ascii_lowercase();
|
||||
|
||||
// `localhost` and anything under it resolve to loopback everywhere.
|
||||
if lower == "localhost" || lower.ends_with(".localhost") {
|
||||
return false;
|
||||
}
|
||||
|
||||
// Suffixes reserved for local/private name resolution (RFC 6762 mDNS, RFC
|
||||
// 8375, and the names router vendors hand out on a LAN). A fleet host
|
||||
// resolving one of these gets its own network's answer, not the user's.
|
||||
const LOCAL_SUFFIXES: [&str; 7] = [
|
||||
".local",
|
||||
".localdomain",
|
||||
".internal",
|
||||
".home",
|
||||
".home.arpa",
|
||||
".lan",
|
||||
".intranet",
|
||||
];
|
||||
if LOCAL_SUFFIXES.iter().any(|suffix| lower.ends_with(suffix)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// A bare single-label name ("my-proxy", "router") is only resolvable through
|
||||
// a local search domain, so it is no more use to a fleet host than `.local`.
|
||||
if !lower.contains('.') {
|
||||
return false;
|
||||
}
|
||||
|
||||
true
|
||||
}
|
||||
|
||||
/// Whether an IP literal is routable from another machine.
|
||||
fn ip_is_remote_reachable(ip: IpAddr) -> bool {
|
||||
match ip {
|
||||
IpAddr::V4(v4) => ipv4_is_remote_reachable(v4),
|
||||
IpAddr::V6(v6) => ipv6_is_remote_reachable(v6),
|
||||
}
|
||||
}
|
||||
|
||||
fn ipv4_is_remote_reachable(ip: Ipv4Addr) -> bool {
|
||||
// `is_private`/`is_loopback`/`is_link_local` are stable; the rest are not, so
|
||||
// the remaining ranges are spelled out rather than gated behind a nightly
|
||||
// feature.
|
||||
if ip.is_loopback() || ip.is_private() || ip.is_link_local() || ip.is_unspecified() {
|
||||
return false;
|
||||
}
|
||||
if ip.is_broadcast() || ip.is_multicast() || ip.is_documentation() {
|
||||
return false;
|
||||
}
|
||||
let [a, b, ..] = ip.octets();
|
||||
// 100.64.0.0/10 — carrier-grade NAT (RFC 6598). Reachable inside one
|
||||
// carrier's network and nowhere else.
|
||||
if a == 100 && (64..128).contains(&b) {
|
||||
return false;
|
||||
}
|
||||
// 0.0.0.0/8 "this network", and 240.0.0.0/4 reserved.
|
||||
if a == 0 || a >= 240 {
|
||||
return false;
|
||||
}
|
||||
true
|
||||
}
|
||||
|
||||
fn ipv6_is_remote_reachable(ip: Ipv6Addr) -> bool {
|
||||
if ip.is_loopback() || ip.is_unspecified() || ip.is_multicast() {
|
||||
return false;
|
||||
}
|
||||
// An IPv4 address wearing an IPv6 hat is still that IPv4 address — classify
|
||||
// it as one, or `::ffff:127.0.0.1` walks straight through.
|
||||
if let Some(v4) = ip.to_ipv4_mapped() {
|
||||
return ipv4_is_remote_reachable(v4);
|
||||
}
|
||||
if let Some(v4) = ip.to_ipv4() {
|
||||
return ipv4_is_remote_reachable(v4);
|
||||
}
|
||||
let segments = ip.segments();
|
||||
// fc00::/7 unique-local, fe80::/10 link-local.
|
||||
if (segments[0] & 0xfe00) == 0xfc00 {
|
||||
return false;
|
||||
}
|
||||
if (segments[0] & 0xffc0) == 0xfe80 {
|
||||
return false;
|
||||
}
|
||||
true
|
||||
}
|
||||
|
||||
/// Strip the decoration a host can arrive wrapped in: whitespace, `[...]`
|
||||
/// around an IPv6 literal, a trailing dot on an FQDN, and any `user@` or
|
||||
/// `:port` that came along from a URI.
|
||||
fn normalize_host(raw: &str) -> String {
|
||||
let mut host = raw.trim();
|
||||
if host.is_empty() {
|
||||
return String::new();
|
||||
}
|
||||
|
||||
// `user:pass@host` — take what follows the LAST '@', since a password may
|
||||
// itself contain one.
|
||||
if let Some(at) = host.rfind('@') {
|
||||
host = &host[at + 1..];
|
||||
}
|
||||
|
||||
// Bracketed IPv6, optionally with a port: `[::1]:1080`.
|
||||
if let Some(stripped) = host.strip_prefix('[') {
|
||||
if let Some(end) = stripped.find(']') {
|
||||
return stripped[..end].trim().to_string();
|
||||
}
|
||||
return stripped.trim().to_string();
|
||||
}
|
||||
|
||||
// `host:port`, but only when there is exactly one colon — more than one means
|
||||
// a bare IPv6 literal, whose colons are part of the address.
|
||||
if host.matches(':').count() == 1 {
|
||||
if let Some((left, _port)) = host.split_once(':') {
|
||||
host = left;
|
||||
}
|
||||
}
|
||||
|
||||
host.trim().trim_end_matches('.').to_string()
|
||||
}
|
||||
|
||||
/// The host a VLESS URI actually dials.
|
||||
///
|
||||
/// Load-bearing because of an asymmetry that is easy to get backwards: a VLESS
|
||||
/// proxy presents to the browser as `127.0.0.1:<port>` — Donut runs a local xray
|
||||
/// worker and points the browser at it — but the address that decides whether
|
||||
/// anyone else could use this config is the SERVER inside the URI. The local
|
||||
/// port is an implementation detail of this machine; the URI is the exit.
|
||||
pub fn vless_uri_host(uri: &str) -> Option<String> {
|
||||
let rest = uri.trim().strip_prefix("vless://")?;
|
||||
// Cut the fragment (`#label`) and query (`?type=...`) before looking for the
|
||||
// authority — either may contain '@' or ':'.
|
||||
let rest = rest.split('#').next()?;
|
||||
let rest = rest.split('?').next()?;
|
||||
// `uuid@host:port/...`
|
||||
let authority = rest.split('/').next()?;
|
||||
let host_port = authority
|
||||
.rsplit_once('@')
|
||||
.map(|(_, h)| h)
|
||||
.unwrap_or(authority);
|
||||
let host = normalize_host(host_port);
|
||||
if host.is_empty() {
|
||||
None
|
||||
} else {
|
||||
Some(host)
|
||||
}
|
||||
}
|
||||
|
||||
/// The exit host a stored proxy represents, as a remote host would have to dial
|
||||
/// it.
|
||||
pub fn proxy_exit_host(settings: &crate::browser::ProxySettings) -> Result<String, String> {
|
||||
if settings.proxy_type.eq_ignore_ascii_case("vless") {
|
||||
let uri = settings
|
||||
.vless_uri
|
||||
.as_deref()
|
||||
.filter(|uri| !uri.trim().is_empty())
|
||||
.ok_or_else(|| "VLESS proxy has no server URI".to_string())?;
|
||||
return vless_uri_host(uri).ok_or_else(|| "VLESS server URI is malformed".to_string());
|
||||
}
|
||||
|
||||
let host = normalize_host(&settings.host);
|
||||
if host.is_empty() {
|
||||
return Err("proxy has no host".to_string());
|
||||
}
|
||||
Ok(host)
|
||||
}
|
||||
|
||||
/// Classify a stored proxy.
|
||||
pub fn classify_proxy(settings: &crate::browser::ProxySettings) -> ExitReachability {
|
||||
match proxy_exit_host(settings) {
|
||||
Err(reason) => ExitReachability::Unknown {
|
||||
reason,
|
||||
source: "proxy",
|
||||
},
|
||||
Ok(host) => {
|
||||
if host_is_remote_reachable(&host) {
|
||||
ExitReachability::Remote
|
||||
} else {
|
||||
ExitReachability::LocalOnly {
|
||||
host,
|
||||
source: "proxy",
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Classify a WireGuard peer endpoint (`host:port`).
|
||||
pub fn classify_wireguard_endpoint(peer_endpoint: &str) -> ExitReachability {
|
||||
let host = normalize_host(peer_endpoint);
|
||||
if host.is_empty() {
|
||||
return ExitReachability::Unknown {
|
||||
reason: "VPN config has no peer endpoint".to_string(),
|
||||
source: "VPN",
|
||||
};
|
||||
}
|
||||
if host_is_remote_reachable(&host) {
|
||||
ExitReachability::Remote
|
||||
} else {
|
||||
ExitReachability::LocalOnly {
|
||||
host,
|
||||
source: "VPN",
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::browser::ProxySettings;
|
||||
|
||||
fn proxy(proxy_type: &str, host: &str) -> ProxySettings {
|
||||
ProxySettings {
|
||||
proxy_type: proxy_type.to_string(),
|
||||
host: host.to_string(),
|
||||
port: 8080,
|
||||
username: None,
|
||||
password: None,
|
||||
vless_uri: None,
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn loopback_in_every_spelling_is_local() {
|
||||
// The literal case the bug was reported for, plus the spellings that reach
|
||||
// the same place. `::ffff:127.0.0.1` is the one a naive IPv6 check misses.
|
||||
for host in [
|
||||
"127.0.0.1",
|
||||
"127.1.2.3",
|
||||
"localhost",
|
||||
"LOCALHOST",
|
||||
"foo.localhost",
|
||||
"::1",
|
||||
"[::1]",
|
||||
"::ffff:127.0.0.1",
|
||||
"0.0.0.0",
|
||||
"::",
|
||||
] {
|
||||
assert!(
|
||||
!host_is_remote_reachable(host),
|
||||
"{host} should not be remote-reachable"
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn private_and_carrier_ranges_are_local() {
|
||||
for host in [
|
||||
"10.0.0.1",
|
||||
"192.168.1.1",
|
||||
"172.16.0.1",
|
||||
"172.31.255.254",
|
||||
"169.254.1.1", // link-local / APIPA
|
||||
"100.64.0.1", // CGNAT
|
||||
"100.127.255.1",
|
||||
"fd00::1", // unique-local
|
||||
"fe80::1", // link-local
|
||||
"240.0.0.1",
|
||||
"0.1.2.3",
|
||||
] {
|
||||
assert!(
|
||||
!host_is_remote_reachable(host),
|
||||
"{host} should not be remote-reachable"
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn public_addresses_and_names_are_reachable() {
|
||||
for host in [
|
||||
"1.1.1.1",
|
||||
"8.8.8.8",
|
||||
"172.15.0.1", // just outside 172.16/12
|
||||
"172.32.0.1",
|
||||
"100.63.255.255", // just outside 100.64/10
|
||||
"100.128.0.1",
|
||||
"2606:4700:4700::1111",
|
||||
"proxy.example.com",
|
||||
"gate.smartproxy.net.",
|
||||
"residential.example.co.uk",
|
||||
] {
|
||||
assert!(
|
||||
host_is_remote_reachable(host),
|
||||
"{host} should be remote-reachable"
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn lan_only_names_are_local() {
|
||||
// A fleet host resolving these gets ITS network's answer, not the user's —
|
||||
// which is worse than failing, because it may well succeed against
|
||||
// something unrelated.
|
||||
for host in [
|
||||
"my-proxy", // single label: needs a search domain
|
||||
"router.local",
|
||||
"nas.home.arpa",
|
||||
"proxy.lan",
|
||||
"box.internal",
|
||||
"server.localdomain",
|
||||
"gateway.intranet",
|
||||
] {
|
||||
assert!(
|
||||
!host_is_remote_reachable(host),
|
||||
"{host} should not be remote-reachable"
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn host_port_and_credentials_are_stripped_before_classifying() {
|
||||
assert!(!host_is_remote_reachable("127.0.0.1:8080"));
|
||||
assert!(!host_is_remote_reachable("user:pass@127.0.0.1:8080"));
|
||||
assert!(!host_is_remote_reachable("[::1]:1080"));
|
||||
assert!(host_is_remote_reachable("user:p@ss@proxy.example.com:8080"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn a_vless_proxy_is_judged_by_its_server_not_its_local_port() {
|
||||
// THE asymmetry. Donut points the browser at a local xray worker, so the
|
||||
// browser-facing address of every VLESS proxy is 127.0.0.1 — but the stored
|
||||
// config names a real server, and that is what a fleet host would dial.
|
||||
// Classifying VLESS off `settings.host` would refuse every VLESS profile.
|
||||
let mut settings = proxy("vless", "127.0.0.1");
|
||||
settings.vless_uri =
|
||||
Some("vless://6d6e21a1-4829-4d2b-bc7f-1b25707b61e4@vpn.example.com:443?type=tcp#node".into());
|
||||
|
||||
assert_eq!(classify_proxy(&settings), ExitReachability::Remote);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn a_vless_uri_pointing_at_loopback_is_still_local() {
|
||||
let mut settings = proxy("vless", "127.0.0.1");
|
||||
settings.vless_uri = Some("vless://uuid@127.0.0.1:443?type=tcp".into());
|
||||
|
||||
assert_eq!(
|
||||
classify_proxy(&settings),
|
||||
ExitReachability::LocalOnly {
|
||||
host: "127.0.0.1".to_string(),
|
||||
source: "proxy",
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn vless_host_parsing_survives_query_and_fragment() {
|
||||
assert_eq!(
|
||||
vless_uri_host("vless://uuid@example.com:443?sni=a@b.com&x=1#my@label"),
|
||||
Some("example.com".to_string())
|
||||
);
|
||||
assert_eq!(
|
||||
vless_uri_host("vless://uuid@[2606:4700::1111]:443?type=ws"),
|
||||
Some("2606:4700::1111".to_string())
|
||||
);
|
||||
assert_eq!(vless_uri_host("not-a-vless-uri"), None);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn an_unreadable_config_fails_closed() {
|
||||
// Unknown must never be treated as usable: the point of the check is that
|
||||
// we could not confirm reachability, and guessing "yes" reintroduces the
|
||||
// exact failure it prevents.
|
||||
let mut settings = proxy("vless", "");
|
||||
settings.vless_uri = None;
|
||||
let verdict = classify_proxy(&settings);
|
||||
|
||||
assert!(matches!(verdict, ExitReachability::Unknown { .. }));
|
||||
assert!(!verdict.is_remote());
|
||||
assert!(verdict.refusal_detail().is_some());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn ordinary_proxies_are_classified_by_host() {
|
||||
assert_eq!(
|
||||
classify_proxy(&proxy("socks5", "gate.example.com")),
|
||||
ExitReachability::Remote
|
||||
);
|
||||
assert_eq!(
|
||||
classify_proxy(&proxy("http", "192.168.0.10")),
|
||||
ExitReachability::LocalOnly {
|
||||
host: "192.168.0.10".to_string(),
|
||||
source: "proxy",
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn wireguard_endpoints_are_classified_by_their_peer() {
|
||||
assert_eq!(
|
||||
classify_wireguard_endpoint("vpn.example.com:51820"),
|
||||
ExitReachability::Remote
|
||||
);
|
||||
assert_eq!(
|
||||
classify_wireguard_endpoint("10.0.0.1:51820"),
|
||||
ExitReachability::LocalOnly {
|
||||
host: "10.0.0.1".to_string(),
|
||||
source: "VPN",
|
||||
}
|
||||
);
|
||||
assert!(matches!(
|
||||
classify_wireguard_endpoint(" "),
|
||||
ExitReachability::Unknown { .. }
|
||||
));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn only_remote_permits_a_run() {
|
||||
assert!(ExitReachability::Remote.is_remote());
|
||||
assert!(!ExitReachability::None.is_remote());
|
||||
assert!(!ExitReachability::LocalOnly {
|
||||
host: "127.0.0.1".into(),
|
||||
source: "proxy"
|
||||
}
|
||||
.is_remote());
|
||||
// `None` has no detail: the caller's existing "no exit at all" refusal is
|
||||
// the better message, and two refusals for one condition read as a bug.
|
||||
assert!(ExitReachability::None.refusal_detail().is_none());
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user