diff --git a/src/config/load.rs b/src/config/load.rs index 811fe6b..21932ec 100644 --- a/src/config/load.rs +++ b/src/config/load.rs @@ -1,14 +1,12 @@ #![allow(deprecated)] use std::collections::{BTreeSet, HashMap, HashSet}; -use std::hash::{DefaultHasher, Hash, Hasher}; use std::net::{IpAddr, SocketAddr}; use std::path::{Path, PathBuf}; use std::sync::Arc; use rand::RngExt; use serde::{Deserialize, Serialize}; -use shadowsocks::config::ServerConfig as ShadowsocksServerConfig; use tracing::warn; use crate::error::{ProxyError, Result}; @@ -16,7 +14,30 @@ use crate::error::{ProxyError, Result}; use super::defaults::*; use super::types::*; -const ACCESS_SECRET_BYTES: usize = 16; +// Domain names, mask targets, and legacy scalar normalization helpers. +mod normalize; +// Include preprocessing and rendered config metadata helpers. +mod includes; +// Strict-config unknown key detection and suggestions. +mod strict_keys; +// Precomputed user authentication data for handshake hot paths. +mod runtime_auth; +// Post-deserialization validation helpers. +mod validation; + +use self::includes::{hash_rendered_snapshot, normalize_config_path, preprocess_includes}; +use self::normalize::{ + is_valid_ad_tag, is_valid_tls_domain_name, normalize_domain_to_ascii, + normalize_exclusive_mask_target, normalize_mask_host_to_ascii, parse_exclusive_mask_target, + push_unique_nonempty, sanitize_ad_tag, +}; +pub(crate) use self::runtime_auth::UserAuthSnapshot; +use self::strict_keys::handle_unknown_config_keys; +use self::validation::{ + normalize_upstream_family_policy, validate_logging_config, validate_network_cfg, + validate_upstreams, +}; + const MAX_ME_WRITER_CMD_CHANNEL_CAPACITY: usize = 16_384; const MAX_ME_ROUTE_CHANNEL_CAPACITY: usize = 8_192; const MAX_ME_C2ME_CHANNEL_CAPACITY: usize = 8_192; @@ -24,784 +45,6 @@ const MIN_MAX_CLIENT_FRAME_BYTES: usize = 4 * 1024; const MAX_MAX_CLIENT_FRAME_BYTES: usize = 16 * 1024 * 1024; const MAX_API_REQUEST_BODY_LIMIT_BYTES: usize = 1024 * 1024; -fn is_valid_tls_domain_name(domain: &str) -> bool { - !domain.is_empty() - && !domain - .chars() - .any(|ch| ch.is_whitespace() || matches!(ch, '/' | '\\')) -} - -fn normalize_domain_to_ascii(domain: &str, field: &str) -> Result { - let domain = domain.trim(); - if !is_valid_tls_domain_name(domain) { - return Err(ProxyError::Config(format!( - "Invalid {field}: '{}'. Must be a valid domain name", - domain - ))); - } - - let parsed = url::Url::parse(&format!("https://{domain}/")).map_err(|error| { - ProxyError::Config(format!( - "Invalid {field}: '{}'. IDNA conversion failed: {error}", - domain - )) - })?; - let host = parsed.host_str().ok_or_else(|| { - ProxyError::Config(format!("Invalid {field}: '{}'. Host is empty", domain)) - })?; - Ok(host.to_ascii_lowercase()) -} - -fn normalize_mask_host_to_ascii(host: &str, field: &str) -> Result { - let host = host.trim(); - if host.starts_with('[') && host.ends_with(']') { - let inner = &host[1..host.len() - 1]; - let ip = inner.parse::().map_err(|_| { - ProxyError::Config(format!( - "Invalid {field}: '{}'. IPv6 literal is invalid", - host - )) - })?; - return match ip { - std::net::IpAddr::V6(v6) => Ok(format!("[{v6}]")), - std::net::IpAddr::V4(v4) => Ok(v4.to_string()), - }; - } - if let Ok(ip) = host.parse::() { - return match ip { - std::net::IpAddr::V4(v4) => Ok(v4.to_string()), - std::net::IpAddr::V6(v6) => Ok(format!("[{v6}]")), - }; - } - - normalize_domain_to_ascii(host, field) -} - -fn parse_exclusive_mask_target(target: &str) -> Option<(&str, u16)> { - let target = target.trim(); - if target.is_empty() { - return None; - } - - if target.starts_with('[') { - let end = target.find(']')?; - if target.get(end + 1..end + 2)? != ":" { - return None; - } - let host = &target[..=end]; - let port = target[end + 2..].parse::().ok()?; - return (port > 0).then_some((host, port)); - } - - let (host, port) = target.rsplit_once(':')?; - if host.is_empty() || host.contains(':') { - return None; - } - let port = port.parse::().ok()?; - (port > 0).then_some((host, port)) -} - -fn normalize_exclusive_mask_target(target: &str, field: &str) -> Result { - let (host, port) = parse_exclusive_mask_target(target).ok_or_else(|| { - ProxyError::Config(format!( - "Invalid {field}: '{}'. Expected host:port with port > 0", - target - )) - })?; - let host = normalize_mask_host_to_ascii(host, field)?; - Ok(format!("{host}:{port}")) -} - -const TOP_LEVEL_CONFIG_KEYS: &[&str] = &[ - "general", - "logging", - "network", - "server", - "timeouts", - "censorship", - "access", - "upstreams", - "show_link", - "dc_overrides", - "default_dc", - "beobachten", - "beobachten_minutes", - "beobachten_flush_secs", - "beobachten_file", - "include", -]; - -const GENERAL_CONFIG_KEYS: &[&str] = &[ - "data_path", - "quota_state_path", - "config_strict", - "modes", - "prefer_ipv6", - "fast_mode", - "use_middle_proxy", - "proxy_secret_path", - "proxy_secret_url", - "proxy_config_v4_cache_path", - "proxy_config_v4_url", - "proxy_config_v6_cache_path", - "proxy_config_v6_url", - "ad_tag", - "middle_proxy_nat_ip", - "middle_proxy_nat_probe", - "middle_proxy_nat_stun", - "middle_proxy_nat_stun_servers", - "stun_nat_probe_concurrency", - "middle_proxy_pool_size", - "middle_proxy_warm_standby", - "me_init_retry_attempts", - "me2dc_fallback", - "me2dc_fast", - "me_keepalive_enabled", - "me_keepalive_interval_secs", - "me_keepalive_jitter_secs", - "me_keepalive_payload_random", - "rpc_proxy_req_every", - "me_writer_cmd_channel_capacity", - "me_route_channel_capacity", - "me_c2me_channel_capacity", - "me_c2me_send_timeout_ms", - "me_reader_route_data_wait_ms", - "me_d2c_flush_batch_max_frames", - "me_d2c_flush_batch_max_bytes", - "me_d2c_flush_batch_max_delay_us", - "me_d2c_ack_flush_immediate", - "me_quota_soft_overshoot_bytes", - "me_d2c_frame_buf_shrink_threshold_bytes", - "direct_relay_copy_buf_c2s_bytes", - "direct_relay_copy_buf_s2c_bytes", - "crypto_pending_buffer", - "max_client_frame", - "desync_all_full", - "beobachten", - "beobachten_minutes", - "beobachten_flush_secs", - "beobachten_file", - "hardswap", - "me_warmup_stagger_enabled", - "me_warmup_step_delay_ms", - "me_warmup_step_jitter_ms", - "me_reconnect_max_concurrent_per_dc", - "me_reconnect_backoff_base_ms", - "me_reconnect_backoff_cap_ms", - "me_reconnect_fast_retry_count", - "me_single_endpoint_shadow_writers", - "me_single_endpoint_outage_mode_enabled", - "me_single_endpoint_outage_disable_quarantine", - "me_single_endpoint_outage_backoff_min_ms", - "me_single_endpoint_outage_backoff_max_ms", - "me_single_endpoint_shadow_rotate_every_secs", - "me_floor_mode", - "me_adaptive_floor_idle_secs", - "me_adaptive_floor_min_writers_single_endpoint", - "me_adaptive_floor_min_writers_multi_endpoint", - "me_adaptive_floor_recover_grace_secs", - "me_adaptive_floor_writers_per_core_total", - "me_adaptive_floor_cpu_cores_override", - "me_adaptive_floor_max_extra_writers_single_per_core", - "me_adaptive_floor_max_extra_writers_multi_per_core", - "me_adaptive_floor_max_active_writers_per_core", - "me_adaptive_floor_max_warm_writers_per_core", - "me_adaptive_floor_max_active_writers_global", - "me_adaptive_floor_max_warm_writers_global", - "upstream_connect_retry_attempts", - "upstream_connect_retry_backoff_ms", - "upstream_connect_budget_ms", - "tg_connect", - "upstream_unhealthy_fail_threshold", - "upstream_connect_failfast_hard_errors", - "stun_iface_mismatch_ignore", - "unknown_dc_log_path", - "unknown_dc_file_log_enabled", - "log_level", - "disable_colors", - "telemetry", - "me_socks_kdf_policy", - "me_route_backpressure_enabled", - "me_route_fairshare_enabled", - "me_route_backpressure_base_timeout_ms", - "me_route_backpressure_high_timeout_ms", - "me_route_backpressure_high_watermark_pct", - "me_health_interval_ms_unhealthy", - "me_health_interval_ms_healthy", - "me_admission_poll_ms", - "me_warn_rate_limit_ms", - "me_route_no_writer_mode", - "me_route_no_writer_wait_ms", - "me_route_hybrid_max_wait_ms", - "me_route_blocking_send_timeout_ms", - "me_route_inline_recovery_attempts", - "me_route_inline_recovery_wait_ms", - "links", - "fast_mode_min_tls_record", - "update_every", - "me_reinit_every_secs", - "me_hardswap_warmup_delay_min_ms", - "me_hardswap_warmup_delay_max_ms", - "me_hardswap_warmup_extra_passes", - "me_hardswap_warmup_pass_backoff_base_ms", - "me_config_stable_snapshots", - "me_config_apply_cooldown_secs", - "me_snapshot_require_http_2xx", - "me_snapshot_reject_empty_map", - "me_snapshot_min_proxy_for_lines", - "proxy_secret_stable_snapshots", - "proxy_secret_rotate_runtime", - "me_secret_atomic_snapshot", - "proxy_secret_len_max", - "me_pool_drain_ttl_secs", - "me_instadrain", - "me_pool_drain_threshold", - "me_pool_drain_soft_evict_enabled", - "me_pool_drain_soft_evict_grace_secs", - "me_pool_drain_soft_evict_per_writer", - "me_pool_drain_soft_evict_budget_per_core", - "me_pool_drain_soft_evict_cooldown_ms", - "me_bind_stale_mode", - "me_bind_stale_ttl_secs", - "me_pool_min_fresh_ratio", - "me_reinit_drain_timeout_secs", - "proxy_secret_auto_reload_secs", - "proxy_config_auto_reload_secs", - "me_reinit_singleflight", - "me_reinit_trigger_channel", - "me_reinit_coalesce_window_ms", - "me_deterministic_writer_sort", - "me_writer_pick_mode", - "me_writer_pick_sample_size", - "ntp_check", - "ntp_servers", - "auto_degradation_enabled", - "degradation_min_unavailable_dc_groups", - "rst_on_close", -]; - -const NETWORK_CONFIG_KEYS: &[&str] = &[ - "ipv4", - "ipv6", - "prefer", - "multipath", - "stun_use", - "stun_servers", - "stun_tcp_fallback", - "http_ip_detect_urls", - "cache_public_ip_path", - "dns_overrides", -]; - -const SERVER_CONFIG_KEYS: &[&str] = &[ - "port", - "listen_addr_ipv4", - "listen_addr_ipv6", - "listen_unix_sock", - "listen_unix_sock_perm", - "listen_tcp", - "client_mss", - "client_mss_bulk", - "proxy_protocol", - "proxy_protocol_header_timeout_ms", - "proxy_protocol_trusted_cidrs", - "metrics_port", - "metrics_listen", - "metrics_whitelist", - "api", - "admin_api", - "listeners", - "listen_backlog", - "max_connections", - "accept_permit_timeout_ms", - "conntrack_control", -]; - -const API_CONFIG_KEYS: &[&str] = &[ - "enabled", - "listen", - "whitelist", - "gray_action", - "auth_header", - "request_body_limit_bytes", - "minimal_runtime_enabled", - "minimal_runtime_cache_ttl_ms", - "runtime_edge_enabled", - "runtime_edge_cache_ttl_ms", - "runtime_edge_top_n", - "runtime_edge_events_capacity", - "read_only", -]; - -const CONNTRACK_CONTROL_CONFIG_KEYS: &[&str] = &[ - "inline_conntrack_control", - "mode", - "backend", - "profile", - "hybrid_listener_ips", - "pressure_high_watermark_pct", - "pressure_low_watermark_pct", - "delete_budget_per_sec", -]; - -const LISTENER_CONFIG_KEYS: &[&str] = &[ - "ip", - "port", - "client_mss", - "synlimit", - "synlimit_seconds", - "synlimit_hitcount", - "synlimit_burst", - "synlimit_ios_seconds", - "synlimit_ios_hitcount", - "synlimit_ios_burst", - "synlimit_hashlimit_expire_ms", - "synlimit_hashlimit_size", - "announce", - "announce_ip", - "proxy_protocol", - "reuse_allow", -]; - -const TIMEOUTS_CONFIG_KEYS: &[&str] = &[ - "client_first_byte_idle_secs", - "client_handshake", - "relay_idle_policy_v2_enabled", - "relay_client_idle_soft_secs", - "relay_client_idle_hard_secs", - "relay_idle_grace_after_downstream_activity_secs", - "client_keepalive", - "client_ack", - "me_one_retry", - "me_one_timeout_ms", -]; - -const CENSORSHIP_CONFIG_KEYS: &[&str] = &[ - "tls_domain", - "tls_domains", - "unknown_sni_action", - "tls_fetch_scope", - "tls_fetch", - "mask", - "mask_dynamic", - "mask_host", - "mask_port", - "exclusive_mask", - "mask_unix_sock", - "fake_cert_len", - "tls_emulation", - "tls_front_dir", - "server_hello_delay_min_ms", - "server_hello_delay_max_ms", - "tls_new_session_tickets", - "serverhello_compact", - "tls_full_cert_ttl_secs", - "alpn_enforce", - "mask_proxy_protocol", - "mask_shape_hardening", - "mask_shape_hardening_aggressive_mode", - "mask_shape_bucket_floor_bytes", - "mask_shape_bucket_cap_bytes", - "mask_shape_above_cap_blur", - "mask_shape_above_cap_blur_max_bytes", - "mask_relay_max_bytes", - "mask_relay_timeout_ms", - "mask_relay_idle_timeout_ms", - "mask_classifier_prefetch_timeout_ms", - "mask_timing_normalization_enabled", - "mask_timing_normalization_floor_ms", - "mask_timing_normalization_ceiling_ms", -]; - -const TLS_FETCH_CONFIG_KEYS: &[&str] = &[ - "profiles", - "strict_route", - "attempt_timeout_ms", - "total_budget_ms", - "grease_enabled", - "deterministic", - "profile_cache_ttl_secs", -]; - -const ACCESS_CONFIG_KEYS: &[&str] = &[ - "users", - "user_enabled", - "user_ad_tags", - "user_max_tcp_conns", - "user_max_tcp_conns_global_each", - "user_expirations", - "user_data_quota", - "user_rate_limits", - "cidr_rate_limits", - "user_max_unique_ips", - "user_max_unique_ips_global_each", - "user_max_unique_ips_mode", - "user_max_unique_ips_window_secs", - "replay_check_len", - "replay_window_secs", - "ignore_time_skew", -]; - -const RATE_LIMIT_BPS_CONFIG_KEYS: &[&str] = &["up_bps", "down_bps"]; - -const UPSTREAM_CONFIG_KEYS: &[&str] = &[ - "type", - "interface", - "bind_addresses", - "bindtodevice", - "force_bind", - "address", - "user_id", - "username", - "password", - "url", - "weight", - "enabled", - "scopes", - "ipv4", - "ipv6", -]; - -const PROXY_MODES_CONFIG_KEYS: &[&str] = &["classic", "secure", "tls"]; -const TELEMETRY_CONFIG_KEYS: &[&str] = &["core_enabled", "user_enabled", "me_level"]; -const LINKS_CONFIG_KEYS: &[&str] = &["show", "public_host", "public_port"]; -const LOGGING_CONFIG_KEYS: &[&str] = &[ - "destination", - "path", - "rotation", - "max_size_bytes", - "max_files", - "max_age_secs", -]; - -#[derive(Debug)] -struct UnknownConfigKey { - path: String, - suggestion: Option, -} - -fn table_at<'a>(value: &'a toml::Value, path: &[&str]) -> Option<&'a toml::Table> { - let mut current = value; - for segment in path { - current = current.get(*segment)?; - } - current.as_table() -} - -fn is_strict_config(parsed_toml: &toml::Value) -> bool { - table_at(parsed_toml, &["general"]) - .and_then(|table| table.get("config_strict")) - .and_then(toml::Value::as_bool) - .unwrap_or(false) -} - -fn known_config_keys_for_suggestion() -> Vec<&'static str> { - let mut keys = Vec::new(); - for group in [ - TOP_LEVEL_CONFIG_KEYS, - GENERAL_CONFIG_KEYS, - NETWORK_CONFIG_KEYS, - SERVER_CONFIG_KEYS, - API_CONFIG_KEYS, - CONNTRACK_CONTROL_CONFIG_KEYS, - LISTENER_CONFIG_KEYS, - TIMEOUTS_CONFIG_KEYS, - CENSORSHIP_CONFIG_KEYS, - TLS_FETCH_CONFIG_KEYS, - ACCESS_CONFIG_KEYS, - RATE_LIMIT_BPS_CONFIG_KEYS, - UPSTREAM_CONFIG_KEYS, - PROXY_MODES_CONFIG_KEYS, - TELEMETRY_CONFIG_KEYS, - LINKS_CONFIG_KEYS, - LOGGING_CONFIG_KEYS, - ] { - keys.extend_from_slice(group); - } - keys -} - -fn levenshtein_distance(a: &str, b: &str) -> usize { - let b_chars: Vec = b.chars().collect(); - let mut prev: Vec = (0..=b_chars.len()).collect(); - let mut curr = vec![0usize; b_chars.len() + 1]; - - for (i, ca) in a.chars().enumerate() { - curr[0] = i + 1; - for (j, cb) in b_chars.iter().enumerate() { - let replace = if ca == *cb { prev[j] } else { prev[j] + 1 }; - curr[j + 1] = (prev[j + 1] + 1).min(curr[j] + 1).min(replace); - } - std::mem::swap(&mut prev, &mut curr); - } - - prev[b_chars.len()] -} - -fn unknown_key_suggestion(key: &str, known_keys: &[&'static str]) -> Option { - let normalized = key.to_ascii_lowercase(); - let mut best: Option<(&str, usize)> = None; - for known in known_keys { - let distance = levenshtein_distance(&normalized, known); - let is_better = match best { - Some((_, best_distance)) => distance < best_distance, - None => true, - }; - if distance <= 4 && is_better { - best = Some((known, distance)); - } - } - best.map(|(known, _)| known.to_string()) -} - -fn push_unknown_keys( - unknown: &mut Vec, - known_for_suggestion: &[&'static str], - path: &str, - table: &toml::Table, - allowed: &[&str], -) { - for key in table.keys() { - if !allowed.contains(&key.as_str()) { - let full_path = if path.is_empty() { - key.clone() - } else { - format!("{path}.{key}") - }; - unknown.push(UnknownConfigKey { - path: full_path, - suggestion: unknown_key_suggestion(key, known_for_suggestion), - }); - } - } -} - -fn check_known_table( - parsed_toml: &toml::Value, - unknown: &mut Vec, - known_for_suggestion: &[&'static str], - path: &[&str], - allowed: &[&str], -) { - if let Some(table) = table_at(parsed_toml, path) { - push_unknown_keys( - unknown, - known_for_suggestion, - &path.join("."), - table, - allowed, - ); - } -} - -fn check_nested_table_value( - unknown: &mut Vec, - known_for_suggestion: &[&'static str], - path: String, - value: &toml::Value, - allowed: &[&str], -) { - if let Some(table) = value.as_table() { - push_unknown_keys(unknown, known_for_suggestion, &path, table, allowed); - } -} - -fn collect_unknown_config_keys(parsed_toml: &toml::Value) -> Vec { - let known_for_suggestion = known_config_keys_for_suggestion(); - let mut unknown = Vec::new(); - - if let Some(root) = parsed_toml.as_table() { - push_unknown_keys( - &mut unknown, - &known_for_suggestion, - "", - root, - TOP_LEVEL_CONFIG_KEYS, - ); - } - - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["general"], - GENERAL_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["general", "modes"], - PROXY_MODES_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["general", "telemetry"], - TELEMETRY_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["general", "links"], - LINKS_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["logging"], - LOGGING_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["network"], - NETWORK_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["server"], - SERVER_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["server", "api"], - API_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["server", "admin_api"], - API_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["server", "conntrack_control"], - CONNTRACK_CONTROL_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["timeouts"], - TIMEOUTS_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["censorship"], - CENSORSHIP_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["censorship", "tls_fetch"], - TLS_FETCH_CONFIG_KEYS, - ); - check_known_table( - parsed_toml, - &mut unknown, - &known_for_suggestion, - &["access"], - ACCESS_CONFIG_KEYS, - ); - - if let Some(listeners) = table_at(parsed_toml, &["server"]) - .and_then(|table| table.get("listeners")) - .and_then(toml::Value::as_array) - { - for (idx, listener) in listeners.iter().enumerate() { - check_nested_table_value( - &mut unknown, - &known_for_suggestion, - format!("server.listeners[{idx}]"), - listener, - LISTENER_CONFIG_KEYS, - ); - } - } - - if let Some(upstreams) = parsed_toml.get("upstreams").and_then(toml::Value::as_array) { - for (idx, upstream) in upstreams.iter().enumerate() { - check_nested_table_value( - &mut unknown, - &known_for_suggestion, - format!("upstreams[{idx}]"), - upstream, - UPSTREAM_CONFIG_KEYS, - ); - } - } - - for access_map in ["user_rate_limits", "cidr_rate_limits"] { - if let Some(table) = table_at(parsed_toml, &["access"]) - .and_then(|access| access.get(access_map)) - .and_then(toml::Value::as_table) - { - for (entry_name, value) in table { - check_nested_table_value( - &mut unknown, - &known_for_suggestion, - format!("access.{access_map}.{entry_name}"), - value, - RATE_LIMIT_BPS_CONFIG_KEYS, - ); - } - } - } - - unknown -} - -fn handle_unknown_config_keys(parsed_toml: &toml::Value) -> Result<()> { - let unknown = collect_unknown_config_keys(parsed_toml); - if unknown.is_empty() { - return Ok(()); - } - - for item in &unknown { - if let Some(suggestion) = item.suggestion.as_deref() { - warn!( - key = %item.path, - suggestion = %suggestion, - "Unknown config key ignored; did you mean the suggested key?" - ); - } else { - warn!(key = %item.path, "Unknown config key ignored"); - } - } - - if is_strict_config(parsed_toml) { - let mut paths = Vec::with_capacity(unknown.len()); - for item in unknown { - if let Some(suggestion) = item.suggestion { - paths.push(format!("{} (did you mean `{}`?)", item.path, suggestion)); - } else { - paths.push(item.path); - } - } - return Err(ProxyError::Config(format!( - "unknown config keys are not allowed when general.config_strict=true: {}", - paths.join(", ") - ))); - } - - Ok(()) -} - #[derive(Debug, Clone)] pub(crate) struct LoadedConfig { pub(crate) config: ProxyConfig, @@ -809,299 +52,10 @@ pub(crate) struct LoadedConfig { pub(crate) rendered_hash: u64, } -/// Precomputed, immutable user authentication data used by handshake hot paths. -#[derive(Debug, Clone, Default)] -pub(crate) struct UserAuthSnapshot { - entries: Vec, - by_name: HashMap, - sni_index: HashMap>, - sni_initial_index: HashMap>, -} - -#[derive(Debug, Clone)] -pub(crate) struct UserAuthEntry { - pub(crate) user: String, - pub(crate) secret: [u8; ACCESS_SECRET_BYTES], -} - -impl UserAuthSnapshot { - fn from_users(users: &HashMap) -> Result { - let mut entries = Vec::with_capacity(users.len()); - let mut by_name = HashMap::with_capacity(users.len()); - let mut sni_index = HashMap::with_capacity(users.len()); - let mut sni_initial_index = HashMap::with_capacity(users.len()); - - for (user, secret_hex) in users { - let decoded = hex::decode(secret_hex).map_err(|_| ProxyError::InvalidSecret { - user: user.clone(), - reason: "Must be 32 hex characters".to_string(), - })?; - if decoded.len() != ACCESS_SECRET_BYTES { - return Err(ProxyError::InvalidSecret { - user: user.clone(), - reason: "Must be 32 hex characters".to_string(), - }); - } - - let user_id = u32::try_from(entries.len()).map_err(|_| { - ProxyError::Config("Too many users for runtime auth snapshot".to_string()) - })?; - - let mut secret = [0u8; ACCESS_SECRET_BYTES]; - secret.copy_from_slice(&decoded); - entries.push(UserAuthEntry { - user: user.clone(), - secret, - }); - by_name.insert(user.clone(), user_id); - sni_index - .entry(Self::sni_lookup_hash(user)) - .or_insert_with(Vec::new) - .push(user_id); - if let Some(initial) = user - .as_bytes() - .first() - .map(|byte| byte.to_ascii_lowercase()) - { - sni_initial_index - .entry(initial) - .or_insert_with(Vec::new) - .push(user_id); - } - } - - Ok(Self { - entries, - by_name, - sni_index, - sni_initial_index, - }) - } - - pub(crate) fn entries(&self) -> &[UserAuthEntry] { - &self.entries - } - - pub(crate) fn user_id_by_name(&self, user: &str) -> Option { - self.by_name.get(user).copied() - } - - pub(crate) fn entry_by_id(&self, user_id: u32) -> Option<&UserAuthEntry> { - let idx = usize::try_from(user_id).ok()?; - self.entries.get(idx) - } - - pub(crate) fn sni_candidates(&self, sni: &str) -> Option<&[u32]> { - self.sni_index - .get(&Self::sni_lookup_hash(sni)) - .map(Vec::as_slice) - } - - pub(crate) fn sni_initial_candidates(&self, sni: &str) -> Option<&[u32]> { - let initial = sni - .as_bytes() - .first() - .map(|byte| byte.to_ascii_lowercase())?; - self.sni_initial_index.get(&initial).map(Vec::as_slice) - } - - fn sni_lookup_hash(value: &str) -> u64 { - let mut hasher = DefaultHasher::new(); - for byte in value.bytes() { - hasher.write_u8(byte.to_ascii_lowercase()); - } - hasher.finish() - } -} - -fn normalize_config_path(path: &Path) -> PathBuf { - path.canonicalize().unwrap_or_else(|_| { - if path.is_absolute() { - path.to_path_buf() - } else { - std::env::current_dir() - .map(|cwd| cwd.join(path)) - .unwrap_or_else(|_| path.to_path_buf()) - } - }) -} - -fn hash_rendered_snapshot(rendered: &str) -> u64 { - let mut hasher = DefaultHasher::new(); - rendered.hash(&mut hasher); - hasher.finish() -} - -fn preprocess_includes( - content: &str, - base_dir: &Path, - depth: u8, - source_files: &mut BTreeSet, -) -> Result { - if depth > 10 { - return Err(ProxyError::Config("Include depth > 10".into())); - } - let mut output = String::with_capacity(content.len()); - for line in content.lines() { - let trimmed = line.trim(); - if let Some(rest) = trimmed.strip_prefix("include") { - let rest = rest.trim(); - if let Some(rest) = rest.strip_prefix('=') { - let path_str = rest.trim().trim_matches('"'); - let resolved = base_dir.join(path_str); - source_files.insert(normalize_config_path(&resolved)); - let included = std::fs::read_to_string(&resolved) - .map_err(|e| ProxyError::Config(e.to_string()))?; - let included_dir = resolved.parent().unwrap_or(base_dir); - output.push_str(&preprocess_includes( - &included, - included_dir, - depth + 1, - source_files, - )?); - output.push('\n'); - continue; - } - } - output.push_str(line); - output.push('\n'); - } - Ok(output) -} - -fn validate_network_cfg(net: &mut NetworkConfig) -> Result<()> { - if !net.ipv4 && matches!(net.ipv6, Some(false)) { - return Err(ProxyError::Config( - "Both ipv4 and ipv6 are disabled in [network]".to_string(), - )); - } - - if net.prefer != 4 && net.prefer != 6 { - return Err(ProxyError::Config( - "network.prefer must be 4 or 6".to_string(), - )); - } - - if !net.ipv4 && net.prefer == 4 { - warn!("prefer=4 but ipv4=false; forcing prefer=6"); - net.prefer = 6; - } - - if matches!(net.ipv6, Some(false)) && net.prefer == 6 { - warn!("prefer=6 but ipv6=false; forcing prefer=4"); - net.prefer = 4; - } - - Ok(()) -} - -fn push_unique_nonempty(target: &mut Vec, value: String) { - let trimmed = value.trim(); - if trimmed.is_empty() { - return; - } - if !target.iter().any(|existing| existing == trimmed) { - target.push(trimmed.to_string()); - } -} - -fn is_valid_ad_tag(tag: &str) -> bool { - tag.len() == 32 && tag.chars().all(|ch| ch.is_ascii_hexdigit()) -} - -fn sanitize_ad_tag(ad_tag: &mut Option) { - let Some(tag) = ad_tag.as_ref() else { - return; - }; - - if !is_valid_ad_tag(tag) { - warn!("Invalid general.ad_tag value, expected exactly 32 hex chars; ad_tag is disabled"); - *ad_tag = None; - } -} - -fn validate_logging_config(logging: &LoggingConfig) -> Result<()> { - if let Some(path) = logging.path.as_ref() - && path.trim().is_empty() - { - return Err(ProxyError::Config( - "logging.path cannot be empty when provided".to_string(), - )); - } - - if matches!(logging.destination, LoggingDestination::File) && logging.path.is_none() { - return Err(ProxyError::Config( - "logging.path must be set when logging.destination=\"file\"".to_string(), - )); - } - - Ok(()) -} - -fn validate_upstreams(config: &ProxyConfig) -> Result<()> { - let has_enabled_shadowsocks = config.upstreams.iter().any(|upstream| { - upstream.enabled && matches!(upstream.upstream_type, UpstreamType::Shadowsocks { .. }) - }); - - if has_enabled_shadowsocks && config.general.use_middle_proxy { - return Err(ProxyError::Config( - "shadowsocks upstreams require general.use_middle_proxy = false".to_string(), - )); - } - - for upstream in &config.upstreams { - if matches!(upstream.ipv4, Some(false)) && matches!(upstream.ipv6, Some(false)) { - return Err(ProxyError::Config( - "upstream.ipv4 and upstream.ipv6 cannot both be false".to_string(), - )); - } - if let Some(prefer) = upstream.prefer - && prefer != 4 - && prefer != 6 - { - return Err(ProxyError::Config( - "upstream.prefer must be 4 or 6".to_string(), - )); - } - - if let UpstreamType::Shadowsocks { url, .. } = &upstream.upstream_type { - let parsed = ShadowsocksServerConfig::from_url(url) - .map_err(|error| ProxyError::Config(format!("invalid shadowsocks url: {error}")))?; - if parsed.plugin().is_some() { - return Err(ProxyError::Config( - "shadowsocks plugins are not supported".to_string(), - )); - } - } - } - - Ok(()) -} - -fn normalize_upstream_family_policy(config: &mut ProxyConfig) { - for (idx, upstream) in config.upstreams.iter_mut().enumerate() { - if matches!(upstream.ipv4, Some(false)) && upstream.prefer == Some(4) { - warn!( - upstream = idx, - "upstream.prefer=4 but upstream.ipv4=false; forcing prefer=6" - ); - upstream.prefer = Some(6); - } - - if matches!(upstream.ipv6, Some(false)) && upstream.prefer == Some(6) { - warn!( - upstream = idx, - "upstream.prefer=6 but upstream.ipv6=false; forcing prefer=4" - ); - upstream.prefer = Some(4); - } - } -} - -// Main runtime configuration loaded from TOML. - +/// Main runtime configuration loaded from TOML. #[derive(Debug, Clone, Serialize, Deserialize, Default)] pub struct ProxyConfig { + /// General runtime options shared across proxy subsystems. #[serde(default)] pub general: GeneralConfig, @@ -1109,24 +63,31 @@ pub struct ProxyConfig { #[serde(default)] pub logging: LoggingConfig, + /// Network binding, routing, and socket-level configuration. #[serde(default)] pub network: NetworkConfig, + /// Server-side listener, fallback, and API configuration. #[serde(default)] pub server: ServerConfig, + /// Timeout values used by client, fallback, and upstream operations. #[serde(default)] pub timeouts: TimeoutsConfig, + /// Anti-censorship behavior and traffic shaping configuration. #[serde(default)] pub censorship: AntiCensorshipConfig, + /// User authentication secrets and admission policy. #[serde(default)] pub access: AccessConfig, + /// Telegram upstream endpoint configuration. #[serde(default)] pub upstreams: Vec, + /// Optional proxy link rendering controls. #[serde(default)] pub show_link: ShowLink, @@ -1151,6 +112,7 @@ pub struct ProxyConfig { } impl ProxyConfig { + /// Loads runtime configuration from a TOML file. pub fn load>(path: P) -> Result { Self::load_with_metadata(path).map(|loaded| loaded.config) } @@ -2388,6 +1350,7 @@ impl ProxyConfig { self.runtime_user_auth.as_deref() } + /// Validates cross-field configuration invariants after deserialization. pub fn validate(&self) -> Result<()> { if self.access.users.is_empty() { return Err(ProxyError::Config("No users configured".to_string())); @@ -2470,2347 +1433,5 @@ mod load_mask_classifier_prefetch_timeout_security_tests; mod load_memory_envelope_tests; #[cfg(test)] -mod tests { - use super::*; - - const TEST_SHADOWSOCKS_URL: &str = - "ss://2022-blake3-aes-256-gcm:MDEyMzQ1Njc4OTAxMjM0NTY3ODkwMTIzNDU2Nzg5MDE=@127.0.0.1:8388"; - - fn load_config_from_temp_toml(toml: &str) -> ProxyConfig { - let nonce = std::time::SystemTime::now() - .duration_since(std::time::UNIX_EPOCH) - .unwrap() - .as_nanos(); - let dir = std::env::temp_dir().join(format!("telemt_load_cfg_{nonce}")); - std::fs::create_dir_all(&dir).unwrap(); - let path = dir.join("config.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - let _ = std::fs::remove_file(path); - let _ = std::fs::remove_dir(dir); - cfg - } - - fn load_config_error_from_temp_toml(toml: &str) -> String { - let nonce = std::time::SystemTime::now() - .duration_since(std::time::UNIX_EPOCH) - .unwrap() - .as_nanos(); - let dir = std::env::temp_dir().join(format!("telemt_load_cfg_error_{nonce}")); - std::fs::create_dir_all(&dir).unwrap(); - let path = dir.join("config.toml"); - std::fs::write(&path, toml).unwrap(); - let error = ProxyConfig::load(&path).unwrap_err().to_string(); - let _ = std::fs::remove_file(path); - let _ = std::fs::remove_dir(dir); - error - } - - #[test] - fn synlimit_synfix_defaults_are_loaded_for_listener() { - let cfg = load_config_from_temp_toml( - r#" - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - - [[server.listeners]] - ip = "0.0.0.0" - port = 443 - synlimit = "iptables" - "#, - ); - - let listener = &cfg.server.listeners[0]; - assert_eq!(listener.synlimit_seconds, 60); - assert_eq!(listener.synlimit_hitcount, 48); - assert_eq!(listener.synlimit_burst, 1); - assert_eq!(listener.synlimit_ios_seconds, 1); - assert_eq!(listener.synlimit_ios_hitcount, 12); - assert_eq!(listener.synlimit_ios_burst, 24); - assert_eq!(listener.synlimit_hashlimit_expire_ms, 60_000); - assert_eq!(listener.synlimit_hashlimit_size, 32_768); - } - - #[test] - fn synlimit_synfix_zero_values_are_rejected() { - for (field, expected) in [ - ( - "synlimit_ios_seconds", - "server.listeners[0].synlimit_ios_seconds must be > 0", - ), - ( - "synlimit_ios_hitcount", - "server.listeners[0].synlimit_ios_hitcount must be > 0", - ), - ( - "synlimit_ios_burst", - "server.listeners[0].synlimit_ios_burst must be > 0", - ), - ( - "synlimit_hashlimit_expire_ms", - "server.listeners[0].synlimit_hashlimit_expire_ms must be > 0", - ), - ( - "synlimit_hashlimit_size", - "server.listeners[0].synlimit_hashlimit_size must be > 0", - ), - ] { - let toml = format!( - r#" - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - - [[server.listeners]] - ip = "0.0.0.0" - port = 443 - synlimit = "iptables" - {field} = 0 - "# - ); - let error = load_config_error_from_temp_toml(&toml); - assert!(error.contains(expected), "{field}: {error}"); - } - } - - #[test] - fn serde_defaults_remain_unchanged_for_present_sections() { - let toml = r#" - [network] - [general] - [server] - [access] - "#; - let cfg: ProxyConfig = toml::from_str(toml).unwrap(); - - assert_eq!(cfg.logging, LoggingConfig::default()); - assert_eq!(cfg.network.ipv6, default_network_ipv6()); - assert_eq!(cfg.network.stun_use, default_true()); - assert_eq!(cfg.network.stun_tcp_fallback, default_stun_tcp_fallback()); - assert_eq!( - cfg.general.middle_proxy_warm_standby, - default_middle_proxy_warm_standby() - ); - assert_eq!( - cfg.general.me_reconnect_max_concurrent_per_dc, - default_me_reconnect_max_concurrent_per_dc() - ); - assert_eq!( - cfg.general.me_reconnect_fast_retry_count, - default_me_reconnect_fast_retry_count() - ); - assert_eq!( - cfg.general.me_init_retry_attempts, - default_me_init_retry_attempts() - ); - assert_eq!(cfg.general.me2dc_fallback, default_me2dc_fallback()); - assert_eq!(cfg.general.me2dc_fast, default_me2dc_fast()); - assert_eq!( - cfg.general.proxy_config_v4_cache_path, - default_proxy_config_v4_cache_path() - ); - assert_eq!( - cfg.general.proxy_config_v6_cache_path, - default_proxy_config_v6_cache_path() - ); - assert_eq!( - cfg.general.me_single_endpoint_shadow_writers, - default_me_single_endpoint_shadow_writers() - ); - assert_eq!( - cfg.general.me_single_endpoint_outage_mode_enabled, - default_me_single_endpoint_outage_mode_enabled() - ); - assert_eq!( - cfg.general.me_single_endpoint_outage_disable_quarantine, - default_me_single_endpoint_outage_disable_quarantine() - ); - assert_eq!( - cfg.general.me_single_endpoint_outage_backoff_min_ms, - default_me_single_endpoint_outage_backoff_min_ms() - ); - assert_eq!( - cfg.general.me_single_endpoint_outage_backoff_max_ms, - default_me_single_endpoint_outage_backoff_max_ms() - ); - assert_eq!( - cfg.general.me_single_endpoint_shadow_rotate_every_secs, - default_me_single_endpoint_shadow_rotate_every_secs() - ); - assert_eq!(cfg.general.me_floor_mode, MeFloorMode::default()); - assert_eq!( - cfg.general.me_adaptive_floor_idle_secs, - default_me_adaptive_floor_idle_secs() - ); - assert_eq!( - cfg.general.me_adaptive_floor_min_writers_single_endpoint, - default_me_adaptive_floor_min_writers_single_endpoint() - ); - assert_eq!( - cfg.general.me_adaptive_floor_recover_grace_secs, - default_me_adaptive_floor_recover_grace_secs() - ); - assert_eq!( - cfg.general.upstream_connect_retry_attempts, - default_upstream_connect_retry_attempts() - ); - assert_eq!( - cfg.general.upstream_connect_retry_backoff_ms, - default_upstream_connect_retry_backoff_ms() - ); - assert_eq!( - cfg.general.upstream_unhealthy_fail_threshold, - default_upstream_unhealthy_fail_threshold() - ); - assert_eq!( - cfg.general.upstream_connect_failfast_hard_errors, - default_upstream_connect_failfast_hard_errors() - ); - assert_eq!( - cfg.general.rpc_proxy_req_every, - default_rpc_proxy_req_every() - ); - assert_eq!(cfg.general.beobachten_file, default_beobachten_file()); - assert_eq!(cfg.general.update_every, default_update_every()); - assert_eq!(cfg.server.listen_addr_ipv4, default_listen_addr_ipv4()); - assert_eq!(cfg.server.listen_addr_ipv6, default_listen_addr_ipv6_opt()); - assert_eq!(cfg.server.client_mss_value(), Ok(None)); - assert_eq!( - cfg.server.proxy_protocol_trusted_cidrs, - default_proxy_protocol_trusted_cidrs() - ); - assert_eq!(cfg.censorship.unknown_sni_action, UnknownSniAction::Drop); - assert_eq!(cfg.server.api.listen, default_api_listen()); - assert_eq!(cfg.server.api.whitelist, default_api_whitelist()); - assert_eq!(cfg.server.api.gray_action, ApiGrayAction::Drop); - assert_eq!( - cfg.server.api.request_body_limit_bytes, - default_api_request_body_limit_bytes() - ); - assert_eq!( - cfg.server.api.minimal_runtime_enabled, - default_api_minimal_runtime_enabled() - ); - assert_eq!( - cfg.server.api.minimal_runtime_cache_ttl_ms, - default_api_minimal_runtime_cache_ttl_ms() - ); - assert_eq!( - cfg.server.api.runtime_edge_enabled, - default_api_runtime_edge_enabled() - ); - assert_eq!( - cfg.server.api.runtime_edge_cache_ttl_ms, - default_api_runtime_edge_cache_ttl_ms() - ); - assert_eq!( - cfg.server.api.runtime_edge_top_n, - default_api_runtime_edge_top_n() - ); - assert_eq!( - cfg.server.api.runtime_edge_events_capacity, - default_api_runtime_edge_events_capacity() - ); - assert_eq!( - cfg.server.conntrack_control.inline_conntrack_control, - default_conntrack_control_enabled() - ); - assert_eq!(cfg.server.conntrack_control.mode, ConntrackMode::default()); - assert_eq!( - cfg.server.conntrack_control.backend, - ConntrackBackend::default() - ); - assert_eq!( - cfg.server.conntrack_control.profile, - ConntrackPressureProfile::default() - ); - assert_eq!( - cfg.server.conntrack_control.pressure_high_watermark_pct, - default_conntrack_pressure_high_watermark_pct() - ); - assert_eq!( - cfg.server.conntrack_control.pressure_low_watermark_pct, - default_conntrack_pressure_low_watermark_pct() - ); - assert_eq!( - cfg.server.conntrack_control.delete_budget_per_sec, - default_conntrack_delete_budget_per_sec() - ); - assert_eq!(cfg.access.users, default_access_users()); - assert_eq!( - cfg.access.user_max_tcp_conns_global_each, - default_user_max_tcp_conns_global_each() - ); - assert_eq!( - cfg.access.user_max_unique_ips_mode, - UserMaxUniqueIpsMode::default() - ); - assert_eq!( - cfg.access.user_max_unique_ips_window_secs, - default_user_max_unique_ips_window_secs() - ); - } - - #[test] - fn logging_config_is_loaded_from_strict_config() { - let cfg = load_config_from_temp_toml( - r#" - [general] - config_strict = true - - [general.modes] - classic = false - secure = false - tls = true - - [logging] - destination = "file" - path = "/tmp/telemt.log" - rotation = "daily" - max_size_bytes = 1024 - max_files = 3 - max_age_secs = 60 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#, - ); - - assert_eq!(cfg.logging.destination, LoggingDestination::File); - assert_eq!(cfg.logging.path.as_deref(), Some("/tmp/telemt.log")); - assert_eq!(cfg.logging.rotation, LogRotation::Daily); - assert_eq!(cfg.logging.max_size_bytes, 1024); - assert_eq!(cfg.logging.max_files, 3); - assert_eq!(cfg.logging.max_age_secs, 60); - } - - #[test] - fn file_logging_requires_path() { - let error = load_config_error_from_temp_toml( - r#" - [general.modes] - classic = false - secure = false - tls = true - - [logging] - destination = "file" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#, - ); - - assert!(error.contains("logging.path must be set")); - } - - #[test] - fn impl_defaults_are_sourced_from_default_helpers() { - let network = NetworkConfig::default(); - assert_eq!(network.ipv6, default_network_ipv6()); - assert_eq!(network.stun_use, default_true()); - assert_eq!(network.stun_tcp_fallback, default_stun_tcp_fallback()); - - let general = GeneralConfig::default(); - assert_eq!( - general.middle_proxy_warm_standby, - default_middle_proxy_warm_standby() - ); - assert_eq!( - general.me_reconnect_max_concurrent_per_dc, - default_me_reconnect_max_concurrent_per_dc() - ); - assert_eq!( - general.me_reconnect_fast_retry_count, - default_me_reconnect_fast_retry_count() - ); - assert_eq!( - general.me_init_retry_attempts, - default_me_init_retry_attempts() - ); - assert_eq!(general.me2dc_fallback, default_me2dc_fallback()); - assert_eq!(general.me2dc_fast, default_me2dc_fast()); - assert_eq!( - general.proxy_config_v4_cache_path, - default_proxy_config_v4_cache_path() - ); - assert_eq!( - general.proxy_config_v6_cache_path, - default_proxy_config_v6_cache_path() - ); - assert_eq!( - general.me_single_endpoint_shadow_writers, - default_me_single_endpoint_shadow_writers() - ); - assert_eq!( - general.me_single_endpoint_outage_mode_enabled, - default_me_single_endpoint_outage_mode_enabled() - ); - assert_eq!( - general.me_single_endpoint_outage_disable_quarantine, - default_me_single_endpoint_outage_disable_quarantine() - ); - assert_eq!( - general.me_single_endpoint_outage_backoff_min_ms, - default_me_single_endpoint_outage_backoff_min_ms() - ); - assert_eq!( - general.me_single_endpoint_outage_backoff_max_ms, - default_me_single_endpoint_outage_backoff_max_ms() - ); - assert_eq!( - general.me_single_endpoint_shadow_rotate_every_secs, - default_me_single_endpoint_shadow_rotate_every_secs() - ); - assert_eq!(general.me_floor_mode, MeFloorMode::default()); - assert_eq!( - general.me_adaptive_floor_idle_secs, - default_me_adaptive_floor_idle_secs() - ); - assert_eq!( - general.me_adaptive_floor_min_writers_single_endpoint, - default_me_adaptive_floor_min_writers_single_endpoint() - ); - assert_eq!( - general.me_adaptive_floor_recover_grace_secs, - default_me_adaptive_floor_recover_grace_secs() - ); - assert_eq!( - general.upstream_connect_retry_attempts, - default_upstream_connect_retry_attempts() - ); - assert_eq!( - general.upstream_connect_retry_backoff_ms, - default_upstream_connect_retry_backoff_ms() - ); - assert_eq!( - general.upstream_unhealthy_fail_threshold, - default_upstream_unhealthy_fail_threshold() - ); - assert_eq!( - general.upstream_connect_failfast_hard_errors, - default_upstream_connect_failfast_hard_errors() - ); - assert_eq!(general.rpc_proxy_req_every, default_rpc_proxy_req_every()); - assert_eq!(general.beobachten_file, default_beobachten_file()); - assert_eq!(general.update_every, default_update_every()); - - let server = ServerConfig::default(); - assert_eq!(server.listen_addr_ipv6, Some(default_listen_addr_ipv6())); - assert_eq!( - server.proxy_protocol_trusted_cidrs, - default_proxy_protocol_trusted_cidrs() - ); - assert_eq!( - AntiCensorshipConfig::default().unknown_sni_action, - UnknownSniAction::Drop - ); - assert_eq!(server.api.listen, default_api_listen()); - assert_eq!(server.api.whitelist, default_api_whitelist()); - assert_eq!(server.api.gray_action, ApiGrayAction::Drop); - assert_eq!( - server.api.request_body_limit_bytes, - default_api_request_body_limit_bytes() - ); - assert_eq!( - server.api.minimal_runtime_enabled, - default_api_minimal_runtime_enabled() - ); - assert_eq!( - server.api.minimal_runtime_cache_ttl_ms, - default_api_minimal_runtime_cache_ttl_ms() - ); - assert_eq!( - server.api.runtime_edge_enabled, - default_api_runtime_edge_enabled() - ); - assert_eq!( - server.api.runtime_edge_cache_ttl_ms, - default_api_runtime_edge_cache_ttl_ms() - ); - assert_eq!( - server.api.runtime_edge_top_n, - default_api_runtime_edge_top_n() - ); - assert_eq!( - server.api.runtime_edge_events_capacity, - default_api_runtime_edge_events_capacity() - ); - assert_eq!( - server.conntrack_control.inline_conntrack_control, - default_conntrack_control_enabled() - ); - assert_eq!(server.conntrack_control.mode, ConntrackMode::default()); - assert_eq!( - server.conntrack_control.backend, - ConntrackBackend::default() - ); - assert_eq!( - server.conntrack_control.profile, - ConntrackPressureProfile::default() - ); - assert_eq!( - server.conntrack_control.pressure_high_watermark_pct, - default_conntrack_pressure_high_watermark_pct() - ); - assert_eq!( - server.conntrack_control.pressure_low_watermark_pct, - default_conntrack_pressure_low_watermark_pct() - ); - assert_eq!( - server.conntrack_control.delete_budget_per_sec, - default_conntrack_delete_budget_per_sec() - ); - - let access = AccessConfig::default(); - assert_eq!(access.users, default_access_users()); - assert_eq!( - access.user_max_tcp_conns_global_each, - default_user_max_tcp_conns_global_each() - ); - } - - #[test] - fn proxy_protocol_trusted_cidrs_missing_uses_trust_all_but_explicit_empty_stays_empty() { - let cfg_missing: ProxyConfig = toml::from_str( - r#" - [server] - [general] - [network] - [access] - "#, - ) - .unwrap(); - assert_eq!( - cfg_missing.server.proxy_protocol_trusted_cidrs, - default_proxy_protocol_trusted_cidrs() - ); - - let cfg_explicit_empty: ProxyConfig = toml::from_str( - r#" - [server] - proxy_protocol_trusted_cidrs = [] - - [general] - [network] - [access] - "#, - ) - .unwrap(); - assert!( - cfg_explicit_empty - .server - .proxy_protocol_trusted_cidrs - .is_empty() - ); - } - - #[test] - fn conntrack_inline_explicit_flag_is_false_when_omitted() { - let cfg = load_config_from_temp_toml( - r#" - [general] - [network] - [server] - [server.conntrack_control] - [access] - "#, - ); - assert!( - !cfg.server - .conntrack_control - .inline_conntrack_control_explicit - ); - } - - #[test] - fn conntrack_inline_explicit_flag_is_true_when_present() { - let cfg = load_config_from_temp_toml( - r#" - [general] - [network] - [server] - [server.conntrack_control] - inline_conntrack_control = true - [access] - "#, - ); - assert!( - cfg.server - .conntrack_control - .inline_conntrack_control_explicit - ); - } - - #[test] - fn unknown_sni_action_parses_and_defaults_to_drop() { - let cfg_default: ProxyConfig = toml::from_str( - r#" - [server] - [general] - [network] - [access] - [censorship] - "#, - ) - .unwrap(); - assert_eq!( - cfg_default.censorship.unknown_sni_action, - UnknownSniAction::Drop - ); - - let cfg_mask: ProxyConfig = toml::from_str( - r#" - [server] - [general] - [network] - [access] - [censorship] - unknown_sni_action = "mask" - "#, - ) - .unwrap(); - assert_eq!( - cfg_mask.censorship.unknown_sni_action, - UnknownSniAction::Mask - ); - - let cfg_accept: ProxyConfig = toml::from_str( - r#" - [server] - [general] - [network] - [access] - [censorship] - unknown_sni_action = "accept" - "#, - ) - .unwrap(); - assert_eq!( - cfg_accept.censorship.unknown_sni_action, - UnknownSniAction::Accept - ); - - let cfg_reject: ProxyConfig = toml::from_str( - r#" - [server] - [general] - [network] - [access] - [censorship] - unknown_sni_action = "reject_handshake" - "#, - ) - .unwrap(); - assert_eq!( - cfg_reject.censorship.unknown_sni_action, - UnknownSniAction::RejectHandshake - ); - } - - #[test] - fn exclusive_mask_parses_domain_target_map() { - let cfg = load_config_from_temp_toml( - r#" - [general] - [network] - [server] - [access] - [censorship] - tls_domain = "weißbiergärten.de" - tls_domains = ["bürgeramt.de"] - [censorship.exclusive_mask] - "bürgeramt.de" = "rindfleischetikettierungsüberwachungsaufgabenübertragungsgesetz.de:443" - "ipv6.example" = "[::1]:443" - "#, - ); - - assert!(cfg.censorship.tls_domain.is_ascii()); - assert!(cfg.censorship.tls_domain.contains("xn--")); - assert_eq!(cfg.censorship.tls_domains.len(), 1); - let normalized_extra = &cfg.censorship.tls_domains[0]; - assert!(normalized_extra.is_ascii()); - assert!(normalized_extra.contains("xn--")); - - let normalized_target = cfg - .censorship - .exclusive_mask - .get(normalized_extra) - .expect("exclusive_mask key must match normalized tls_domains entry"); - assert!(normalized_target.is_ascii()); - assert!(normalized_target.contains("xn--")); - assert!(normalized_target.ends_with(":443")); - assert_eq!( - cfg.censorship.exclusive_mask.get("ipv6.example"), - Some(&"[::1]:443".to_string()) - ); - } - - #[test] - fn api_gray_action_parses_and_defaults_to_drop() { - let cfg_default: ProxyConfig = toml::from_str( - r#" - [server] - [general] - [network] - [access] - "#, - ) - .unwrap(); - assert_eq!(cfg_default.server.api.gray_action, ApiGrayAction::Drop); - - let cfg_api: ProxyConfig = toml::from_str( - r#" - [server] - [general] - [network] - [access] - [server.api] - gray_action = "api" - "#, - ) - .unwrap(); - assert_eq!(cfg_api.server.api.gray_action, ApiGrayAction::Api); - - let cfg_200: ProxyConfig = toml::from_str( - r#" - [server] - [general] - [network] - [access] - [server.api] - gray_action = "200" - "#, - ) - .unwrap(); - assert_eq!(cfg_200.server.api.gray_action, ApiGrayAction::Ok200); - - let cfg_drop: ProxyConfig = toml::from_str( - r#" - [server] - [general] - [network] - [access] - [server.api] - gray_action = "drop" - "#, - ) - .unwrap(); - assert_eq!(cfg_drop.server.api.gray_action, ApiGrayAction::Drop); - } - - #[test] - fn top_level_beobachten_keys_migrate_to_general_when_general_not_explicit() { - let cfg = load_config_from_temp_toml( - r#" - beobachten = false - beobachten_minutes = 7 - beobachten_flush_secs = 3 - beobachten_file = "tmp/legacy-beob.txt" - - [server] - [general] - [network] - [access] - "#, - ); - - assert!(!cfg.general.beobachten); - assert_eq!(cfg.general.beobachten_minutes, 7); - assert_eq!(cfg.general.beobachten_flush_secs, 3); - assert_eq!(cfg.general.beobachten_file, "tmp/legacy-beob.txt"); - } - - #[test] - fn general_beobachten_keys_have_priority_over_legacy_top_level() { - let cfg = load_config_from_temp_toml( - r#" - beobachten = true - beobachten_minutes = 30 - beobachten_flush_secs = 30 - beobachten_file = "tmp/legacy-beob.txt" - - [server] - [general] - beobachten = false - beobachten_minutes = 5 - beobachten_flush_secs = 2 - beobachten_file = "tmp/general-beob.txt" - [network] - [access] - "#, - ); - - assert!(!cfg.general.beobachten); - assert_eq!(cfg.general.beobachten_minutes, 5); - assert_eq!(cfg.general.beobachten_flush_secs, 2); - assert_eq!(cfg.general.beobachten_file, "tmp/general-beob.txt"); - } - - #[test] - fn dc_overrides_allow_string_and_array() { - let toml = r#" - [dc_overrides] - "201" = "149.154.175.50:443" - "202" = ["149.154.167.51:443", "149.154.175.100:443"] - "#; - let cfg: ProxyConfig = toml::from_str(toml).unwrap(); - assert_eq!(cfg.dc_overrides["201"], vec!["149.154.175.50:443"]); - assert_eq!( - cfg.dc_overrides["202"], - vec!["149.154.167.51:443", "149.154.175.100:443"] - ); - } - - #[test] - fn load_with_metadata_collects_include_files() { - let nonce = std::time::SystemTime::now() - .duration_since(std::time::UNIX_EPOCH) - .unwrap() - .as_nanos(); - let dir = std::env::temp_dir().join(format!("telemt_load_metadata_{nonce}")); - std::fs::create_dir_all(&dir).unwrap(); - let main_path = dir.join("config.toml"); - let include_path = dir.join("included.toml"); - - std::fs::write( - &include_path, - r#" - [access.users] - user = "00000000000000000000000000000000" - "#, - ) - .unwrap(); - std::fs::write( - &main_path, - r#" - include = "included.toml" - - [censorship] - tls_domain = "example.com" - "#, - ) - .unwrap(); - - let loaded = ProxyConfig::load_with_metadata(&main_path).unwrap(); - let main_normalized = normalize_config_path(&main_path); - let include_normalized = normalize_config_path(&include_path); - - assert!(loaded.source_files.contains(&main_normalized)); - assert!(loaded.source_files.contains(&include_normalized)); - - let _ = std::fs::remove_file(main_path); - let _ = std::fs::remove_file(include_path); - let _ = std::fs::remove_dir(dir); - } - - #[test] - fn dc_overrides_inject_dc203_default() { - let toml = r#" - [general] - use_middle_proxy = false - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_dc_override_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert!( - cfg.dc_overrides - .get("203") - .map(|v| v.contains(&"91.105.192.100:443".to_string())) - .unwrap_or(false) - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn update_every_overrides_legacy_fields() { - let toml = r#" - [general] - update_every = 123 - proxy_secret_auto_reload_secs = 700 - proxy_config_auto_reload_secs = 800 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_update_every_override_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!(cfg.general.effective_update_every_secs(), 123); - let _ = std::fs::remove_file(path); - } - - #[test] - fn update_every_fallback_to_legacy_min() { - let toml = r#" - [general] - proxy_secret_auto_reload_secs = 600 - proxy_config_auto_reload_secs = 120 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_update_every_legacy_min_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!(cfg.general.update_every, None); - assert_eq!(cfg.general.effective_update_every_secs(), 120); - let _ = std::fs::remove_file(path); - } - - #[test] - fn update_every_zero_is_rejected() { - let toml = r#" - [general] - update_every = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_update_every_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.update_every must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn stun_nat_probe_concurrency_zero_is_rejected() { - let toml = r#" - [general] - stun_nat_probe_concurrency = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_stun_nat_probe_concurrency_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.stun_nat_probe_concurrency must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_reinit_every_default_is_set() { - let toml = r#" - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_reinit_every_default_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!( - cfg.general.me_reinit_every_secs, - default_me_reinit_every_secs() - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_reinit_every_zero_is_rejected() { - let toml = r#" - [general] - me_reinit_every_secs = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_reinit_every_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_reinit_every_secs must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_single_endpoint_outage_backoff_range_is_validated() { - let toml = r#" - [general] - me_single_endpoint_outage_backoff_min_ms = 4000 - me_single_endpoint_outage_backoff_max_ms = 3000 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_single_endpoint_outage_backoff_range_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains( - "general.me_single_endpoint_outage_backoff_min_ms must be <= general.me_single_endpoint_outage_backoff_max_ms" - )); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_single_endpoint_shadow_writers_too_large_is_rejected() { - let toml = r#" - [general] - me_single_endpoint_shadow_writers = 33 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_single_endpoint_shadow_writers_limit_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_single_endpoint_shadow_writers must be within [0, 32]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_adaptive_floor_min_writers_out_of_range_is_rejected() { - let toml = r#" - [general] - me_adaptive_floor_min_writers_single_endpoint = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_adaptive_floor_min_writers_out_of_range_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains( - "general.me_adaptive_floor_min_writers_single_endpoint must be within [1, 32]" - )); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_floor_mode_adaptive_is_parsed() { - let toml = r#" - [general] - me_floor_mode = "adaptive" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_floor_mode_adaptive_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!(cfg.general.me_floor_mode, MeFloorMode::Adaptive); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_adaptive_floor_max_active_writers_per_core_zero_is_rejected() { - let toml = r#" - [general] - me_adaptive_floor_max_active_writers_per_core = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_adaptive_floor_max_active_per_core_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_adaptive_floor_max_active_writers_per_core must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_adaptive_floor_max_warm_writers_global_zero_is_rejected() { - let toml = r#" - [general] - me_adaptive_floor_max_warm_writers_global = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_adaptive_floor_max_warm_global_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_adaptive_floor_max_warm_writers_global must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn upstream_connect_retry_attempts_zero_is_rejected() { - let toml = r#" - [general] - upstream_connect_retry_attempts = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_upstream_connect_retry_attempts_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.upstream_connect_retry_attempts must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn upstream_unhealthy_fail_threshold_zero_is_rejected() { - let toml = r#" - [general] - upstream_unhealthy_fail_threshold = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_upstream_unhealthy_fail_threshold_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.upstream_unhealthy_fail_threshold must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn tg_connect_zero_is_rejected() { - let toml = r#" - [general] - tg_connect = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_tg_connect_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.tg_connect must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn rpc_proxy_req_every_out_of_range_is_rejected() { - let toml = r#" - [general] - rpc_proxy_req_every = 9 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_rpc_proxy_req_every_out_of_range_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.rpc_proxy_req_every must be 0 or within [10, 300]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn rpc_proxy_req_every_zero_and_valid_range_are_accepted() { - let toml_zero = r#" - [general] - rpc_proxy_req_every = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path_zero = dir.join("telemt_rpc_proxy_req_every_zero_ok_test.toml"); - std::fs::write(&path_zero, toml_zero).unwrap(); - let cfg_zero = ProxyConfig::load(&path_zero).unwrap(); - assert_eq!(cfg_zero.general.rpc_proxy_req_every, 0); - let _ = std::fs::remove_file(path_zero); - - let toml_valid = r#" - [general] - rpc_proxy_req_every = 40 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let path_valid = dir.join("telemt_rpc_proxy_req_every_valid_ok_test.toml"); - std::fs::write(&path_valid, toml_valid).unwrap(); - let cfg_valid = ProxyConfig::load(&path_valid).unwrap(); - assert_eq!(cfg_valid.general.rpc_proxy_req_every, 40); - let _ = std::fs::remove_file(path_valid); - } - - #[test] - fn me_route_backpressure_base_timeout_ms_out_of_range_is_rejected() { - let toml = r#" - [general] - me_route_backpressure_base_timeout_ms = 5001 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_route_backpressure_base_timeout_ms_out_of_range_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!( - err.contains("general.me_route_backpressure_base_timeout_ms must be within [1, 5000]") - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_route_backpressure_high_timeout_ms_out_of_range_is_rejected() { - let toml = r#" - [general] - me_route_backpressure_base_timeout_ms = 100 - me_route_backpressure_high_timeout_ms = 5001 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_route_backpressure_high_timeout_ms_out_of_range_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!( - err.contains("general.me_route_backpressure_high_timeout_ms must be within [1, 5000]") - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_route_no_writer_wait_ms_out_of_range_is_rejected() { - let toml = r#" - [general] - me_route_no_writer_wait_ms = 5 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_route_no_writer_wait_ms_out_of_range_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_route_no_writer_wait_ms must be within [10, 5000]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_route_blocking_send_timeout_ms_zero_is_rejected() { - let toml = r#" - [general] - me_route_blocking_send_timeout_ms = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_route_blocking_send_timeout_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_route_blocking_send_timeout_ms must be within [1, 5000]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_route_no_writer_mode_is_parsed() { - let toml = r#" - [general] - me_route_no_writer_mode = "inline_recovery_legacy" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_route_no_writer_mode_parse_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!( - cfg.general.me_route_no_writer_mode, - crate::config::MeRouteNoWriterMode::InlineRecoveryLegacy - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn proxy_config_cache_paths_empty_are_rejected() { - let toml = r#" - [general] - proxy_config_v4_cache_path = " " - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_proxy_config_v4_cache_path_empty_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.proxy_config_v4_cache_path cannot be empty")); - let _ = std::fs::remove_file(path); - - let toml_v6 = r#" - [general] - proxy_config_v6_cache_path = "" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let path_v6 = dir.join("telemt_proxy_config_v6_cache_path_empty_test.toml"); - std::fs::write(&path_v6, toml_v6).unwrap(); - let err_v6 = ProxyConfig::load(&path_v6).unwrap_err().to_string(); - assert!(err_v6.contains("general.proxy_config_v6_cache_path cannot be empty")); - let _ = std::fs::remove_file(path_v6); - } - - #[test] - fn me_hardswap_warmup_defaults_are_set() { - let toml = r#" - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_hardswap_warmup_defaults_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!( - cfg.general.me_hardswap_warmup_delay_min_ms, - default_me_hardswap_warmup_delay_min_ms() - ); - assert_eq!( - cfg.general.me_hardswap_warmup_delay_max_ms, - default_me_hardswap_warmup_delay_max_ms() - ); - assert_eq!( - cfg.general.me_hardswap_warmup_extra_passes, - default_me_hardswap_warmup_extra_passes() - ); - assert_eq!( - cfg.general.me_hardswap_warmup_pass_backoff_base_ms, - default_me_hardswap_warmup_pass_backoff_base_ms() - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_hardswap_warmup_delay_range_is_validated() { - let toml = r#" - [general] - me_hardswap_warmup_delay_min_ms = 2001 - me_hardswap_warmup_delay_max_ms = 2000 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_hardswap_warmup_delay_range_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains( - "general.me_hardswap_warmup_delay_min_ms must be <= general.me_hardswap_warmup_delay_max_ms" - )); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_hardswap_warmup_delay_max_zero_is_rejected() { - let toml = r#" - [general] - me_hardswap_warmup_delay_max_ms = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_hardswap_warmup_delay_max_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_hardswap_warmup_delay_max_ms must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_hardswap_warmup_extra_passes_out_of_range_is_rejected() { - let toml = r#" - [general] - me_hardswap_warmup_extra_passes = 11 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_hardswap_warmup_extra_passes_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_hardswap_warmup_extra_passes must be within [0, 10]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_hardswap_warmup_pass_backoff_zero_is_rejected() { - let toml = r#" - [general] - me_hardswap_warmup_pass_backoff_base_ms = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_hardswap_warmup_backoff_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_hardswap_warmup_pass_backoff_base_ms must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_config_stable_snapshots_zero_is_rejected() { - let toml = r#" - [general] - me_config_stable_snapshots = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_config_stable_snapshots_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_config_stable_snapshots must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn proxy_secret_stable_snapshots_zero_is_rejected() { - let toml = r#" - [general] - proxy_secret_stable_snapshots = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_proxy_secret_stable_snapshots_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.proxy_secret_stable_snapshots must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn proxy_secret_len_max_out_of_range_is_rejected() { - let toml = r#" - [general] - proxy_secret_len_max = 16 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_proxy_secret_len_max_out_of_range_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.proxy_secret_len_max must be within [32, 4096]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn me_pool_min_fresh_ratio_out_of_range_is_rejected() { - let toml = r#" - [general] - me_pool_min_fresh_ratio = 1.5 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_me_pool_min_ratio_invalid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("general.me_pool_min_fresh_ratio must be within [0.0, 1.0]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn api_minimal_runtime_cache_ttl_out_of_range_is_rejected() { - let toml = r#" - [server.api] - enabled = true - listen = "127.0.0.1:9091" - minimal_runtime_cache_ttl_ms = 70000 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_api_minimal_runtime_cache_ttl_invalid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("server.api.minimal_runtime_cache_ttl_ms must be within [0, 60000]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn client_mss_presets_and_listener_override_are_resolved() { - let toml = r#" - [server] - client_mss = "tspu" - - [[server.listeners]] - ip = "127.0.0.1" - port = 1443 - - [[server.listeners]] - ip = "127.0.0.2" - port = 1444 - client_mss = "2in8" - - [[server.listeners]] - ip = "127.0.0.3" - port = 1445 - client_mss = "" - - [[server.listeners]] - ip = "127.0.0.4" - port = 1446 - client_mss = "extreme-low" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_client_mss_valid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - - assert_eq!(cfg.server.client_mss_value(), Ok(Some(92))); - assert_eq!( - cfg.server.listeners[0].effective_client_mss(&cfg.server), - Ok(Some(92)) - ); - assert_eq!( - cfg.server.listeners[1].effective_client_mss(&cfg.server), - Ok(Some(256)) - ); - assert_eq!( - cfg.server.listeners[2].effective_client_mss(&cfg.server), - Ok(None) - ); - assert_eq!( - cfg.server.listeners[3].effective_client_mss(&cfg.server), - Ok(Some(88)) - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn client_mss_custom_value_is_accepted() { - let toml = r#" - [server] - client_mss = "4096" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_client_mss_custom_valid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - - assert_eq!(cfg.server.client_mss_value(), Ok(Some(4096))); - let _ = std::fs::remove_file(path); - } - - #[test] - fn client_mss_out_of_range_is_rejected() { - for value in ["87", "4097"] { - let toml = format!( - r#" - [server] - client_mss = "{value}" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "# - ); - let dir = std::env::temp_dir(); - let path = dir.join(format!("telemt_client_mss_out_of_range_{value}_test.toml")); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - - assert!(err.contains("server.client_mss custom value must be within [88, 4096]")); - let _ = std::fs::remove_file(path); - } - } - - #[test] - fn client_mss_unquoted_number_is_rejected() { - let toml = r#" - [server] - client_mss = 256 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_client_mss_unquoted_number_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - - assert!(err.contains("client_mss")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn listener_client_mss_invalid_preset_is_rejected() { - let toml = r#" - [[server.listeners]] - ip = "127.0.0.1" - port = 1443 - client_mss = "tiny" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_listener_client_mss_invalid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - - assert!(err.contains("server.listeners[0].client_mss")); - assert!(err.contains("must be \"\", extreme-low, tspu, 2in8")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn api_runtime_edge_cache_ttl_out_of_range_is_rejected() { - let toml = r#" - [server.api] - enabled = true - listen = "127.0.0.1:9091" - runtime_edge_cache_ttl_ms = 70000 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_api_runtime_edge_cache_ttl_invalid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("server.api.runtime_edge_cache_ttl_ms must be within [0, 60000]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn api_runtime_edge_top_n_out_of_range_is_rejected() { - let toml = r#" - [server.api] - enabled = true - listen = "127.0.0.1:9091" - runtime_edge_top_n = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_api_runtime_edge_top_n_invalid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("server.api.runtime_edge_top_n must be within [1, 1000]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn api_runtime_edge_events_capacity_out_of_range_is_rejected() { - let toml = r#" - [server.api] - enabled = true - listen = "127.0.0.1:9091" - runtime_edge_events_capacity = 8 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_api_runtime_edge_events_capacity_invalid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("server.api.runtime_edge_events_capacity must be within [16, 4096]")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn conntrack_pressure_high_watermark_out_of_range_is_rejected() { - let toml = r#" - [server.conntrack_control] - pressure_high_watermark_pct = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_conntrack_high_watermark_invalid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains( - "server.conntrack_control.pressure_high_watermark_pct must be within [1, 100]" - )); - let _ = std::fs::remove_file(path); - } - - #[test] - fn conntrack_pressure_low_watermark_must_be_below_high() { - let toml = r#" - [server.conntrack_control] - pressure_high_watermark_pct = 50 - pressure_low_watermark_pct = 50 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_conntrack_low_watermark_invalid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!( - err.contains( - "server.conntrack_control.pressure_low_watermark_pct must be < pressure_high_watermark_pct" - ) - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn conntrack_delete_budget_zero_is_rejected() { - let toml = r#" - [server.conntrack_control] - delete_budget_per_sec = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_conntrack_delete_budget_invalid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("server.conntrack_control.delete_budget_per_sec must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn conntrack_hybrid_mode_requires_listener_allow_list() { - let toml = r#" - [server.conntrack_control] - mode = "hybrid" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_conntrack_hybrid_requires_ips_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains( - "server.conntrack_control.hybrid_listener_ips must be non-empty in mode=hybrid" - )); - let _ = std::fs::remove_file(path); - } - - #[test] - fn conntrack_profile_is_loaded_from_config() { - let toml = r#" - [server.conntrack_control] - profile = "aggressive" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_conntrack_profile_parse_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!( - cfg.server.conntrack_control.profile, - ConntrackPressureProfile::Aggressive - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn force_close_default_matches_drain_ttl() { - let toml = r#" - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_force_close_default_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!(cfg.general.me_reinit_drain_timeout_secs, 90); - assert_eq!(cfg.general.effective_me_pool_force_close_secs(), 90); - let _ = std::fs::remove_file(path); - } - - #[test] - fn force_close_zero_uses_runtime_safety_fallback() { - let toml = r#" - [general] - me_reinit_drain_timeout_secs = 0 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_force_close_zero_fallback_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!(cfg.general.me_reinit_drain_timeout_secs, 0); - assert_eq!(cfg.general.effective_me_pool_force_close_secs(), 300); - let _ = std::fs::remove_file(path); - } - - #[test] - fn force_close_bumped_when_below_drain_ttl() { - let toml = r#" - [general] - me_pool_drain_ttl_secs = 90 - me_reinit_drain_timeout_secs = 30 - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_force_close_bump_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!(cfg.general.me_reinit_drain_timeout_secs, 90); - let _ = std::fs::remove_file(path); - } - - #[test] - fn tls_fetch_scope_default_is_empty() { - let toml = r#" - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_tls_fetch_scope_default_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert!(cfg.censorship.tls_fetch_scope.is_empty()); - let _ = std::fs::remove_file(path); - } - - #[test] - fn tls_fetch_scope_is_trimmed_during_load() { - let toml = r#" - [censorship] - tls_domain = "example.com" - tls_fetch_scope = " me " - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_tls_fetch_scope_trim_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!(cfg.censorship.tls_fetch_scope, "me"); - let _ = std::fs::remove_file(path); - } - - #[test] - fn tls_fetch_scope_whitespace_becomes_empty() { - let toml = r#" - [censorship] - tls_domain = "example.com" - tls_fetch_scope = " " - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_tls_fetch_scope_blank_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert!(cfg.censorship.tls_fetch_scope.is_empty()); - let _ = std::fs::remove_file(path); - } - - #[test] - fn tls_fetch_defaults_are_applied() { - let toml = r#" - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_tls_fetch_defaults_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!( - cfg.censorship.tls_fetch.profiles, - TlsFetchConfig::default().profiles - ); - assert!(cfg.censorship.tls_fetch.strict_route); - assert_eq!(cfg.censorship.tls_fetch.attempt_timeout_ms, 5_000); - assert_eq!(cfg.censorship.tls_fetch.total_budget_ms, 15_000); - assert_eq!(cfg.censorship.tls_fetch.profile_cache_ttl_secs, 600); - let _ = std::fs::remove_file(path); - } - - #[test] - fn tls_fetch_profiles_are_deduplicated_preserving_order() { - let toml = r#" - [censorship] - tls_domain = "example.com" - [censorship.tls_fetch] - profiles = ["compat_tls12", "modern_chrome_like", "compat_tls12", "legacy_minimal"] - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_tls_fetch_profiles_dedup_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!( - cfg.censorship.tls_fetch.profiles, - vec![ - TlsFetchProfile::CompatTls12, - TlsFetchProfile::ModernChromeLike, - TlsFetchProfile::LegacyMinimal - ] - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn tls_fetch_attempt_timeout_zero_is_rejected() { - let toml = r#" - [censorship] - tls_domain = "example.com" - [censorship.tls_fetch] - attempt_timeout_ms = 0 - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_tls_fetch_attempt_timeout_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("censorship.tls_fetch.attempt_timeout_ms must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn tls_fetch_total_budget_zero_is_rejected() { - let toml = r#" - [censorship] - tls_domain = "example.com" - [censorship.tls_fetch] - total_budget_ms = 0 - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_tls_fetch_total_budget_zero_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("censorship.tls_fetch.total_budget_ms must be > 0")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn invalid_ad_tag_is_disabled_during_load() { - let toml = r#" - [general] - ad_tag = "not_hex" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_invalid_ad_tag_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert!(cfg.general.ad_tag.is_none()); - let _ = std::fs::remove_file(path); - } - - #[test] - fn valid_ad_tag_is_preserved_during_load() { - let toml = r#" - [general] - ad_tag = "00112233445566778899aabbccddeeff" - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_valid_ad_tag_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!( - cfg.general.ad_tag.as_deref(), - Some("00112233445566778899aabbccddeeff") - ); - let _ = std::fs::remove_file(path); - } - - #[test] - fn shadowsocks_upstream_url_loads_successfully() { - let toml = format!( - r#" - [general] - use_middle_proxy = false - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - - [[upstreams]] - type = "shadowsocks" - url = "{url}" - interface = "127.0.0.2" - "#, - url = TEST_SHADOWSOCKS_URL, - ); - let dir = std::env::temp_dir(); - let path = dir.join("telemt_shadowsocks_valid_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - - assert!(matches!( - &cfg.upstreams[0].upstream_type, - UpstreamType::Shadowsocks { url, interface } - if url == TEST_SHADOWSOCKS_URL && interface.as_deref() == Some("127.0.0.2") - )); - - let _ = std::fs::remove_file(path); - } - - #[test] - fn shadowsocks_requires_direct_mode() { - let toml = format!( - r#" - [general] - use_middle_proxy = true - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - - [[upstreams]] - type = "shadowsocks" - url = "{url}" - "#, - url = TEST_SHADOWSOCKS_URL, - ); - let dir = std::env::temp_dir(); - let path = dir.join("telemt_shadowsocks_me_reject_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - - assert!(err.contains("shadowsocks upstreams require general.use_middle_proxy = false")); - - let _ = std::fs::remove_file(path); - } - - #[test] - fn invalid_shadowsocks_url_is_rejected() { - let toml = r#" - [general] - use_middle_proxy = false - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - - [[upstreams]] - type = "shadowsocks" - url = "not-a-valid-ss-url" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_shadowsocks_invalid_url_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - - assert!(err.contains("invalid shadowsocks url")); - - let _ = std::fs::remove_file(path); - } - - #[test] - fn shadowsocks_plugins_are_rejected() { - let toml = format!( - r#" - [general] - use_middle_proxy = false - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - - [[upstreams]] - type = "shadowsocks" - url = "{url}?plugin=obfs-local%3Bobfs%3Dhttp" - "#, - url = TEST_SHADOWSOCKS_URL, - ); - let dir = std::env::temp_dir(); - let path = dir.join("telemt_shadowsocks_plugin_reject_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - - assert!(err.contains("shadowsocks plugins are not supported")); - - let _ = std::fs::remove_file(path); - } - - #[test] - fn invalid_user_ad_tag_reports_access_user_ad_tags_key() { - let toml = r#" - [censorship] - tls_domain = "example.com" - - [access.users] - alice = "00000000000000000000000000000000" - - [access.user_ad_tags] - alice = "not_hex" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_invalid_user_ad_tag_message_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - let err = cfg.validate().unwrap_err().to_string(); - assert!(err.contains("access.user_ad_tags['alice'] must be exactly 32 hex characters")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn invalid_dns_override_is_rejected() { - let toml = r#" - [network] - dns_overrides = ["example.com:443:2001:db8::10"] - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_invalid_dns_override_test.toml"); - std::fs::write(&path, toml).unwrap(); - let err = ProxyConfig::load(&path).unwrap_err().to_string(); - assert!(err.contains("must be bracketed")); - let _ = std::fs::remove_file(path); - } - - #[test] - fn valid_dns_override_is_accepted() { - let toml = r#" - [network] - dns_overrides = ["example.com:443:127.0.0.1", "example.net:443:[2001:db8::10]"] - - [censorship] - tls_domain = "example.com" - - [access.users] - user = "00000000000000000000000000000000" - "#; - let dir = std::env::temp_dir(); - let path = dir.join("telemt_valid_dns_override_test.toml"); - std::fs::write(&path, toml).unwrap(); - let cfg = ProxyConfig::load(&path).unwrap(); - assert_eq!(cfg.network.dns_overrides.len(), 2); - let _ = std::fs::remove_file(path); - } -} +#[path = "tests/load_basic_tests.rs"] +mod tests; diff --git a/src/config/load/includes.rs b/src/config/load/includes.rs new file mode 100644 index 0000000..222887b --- /dev/null +++ b/src/config/load/includes.rs @@ -0,0 +1,60 @@ +use std::collections::BTreeSet; +use std::hash::{DefaultHasher, Hash, Hasher}; +use std::path::{Path, PathBuf}; + +use crate::error::{ProxyError, Result}; + +pub(super) fn normalize_config_path(path: &Path) -> PathBuf { + path.canonicalize().unwrap_or_else(|_| { + if path.is_absolute() { + path.to_path_buf() + } else { + std::env::current_dir() + .map(|cwd| cwd.join(path)) + .unwrap_or_else(|_| path.to_path_buf()) + } + }) +} + +pub(super) fn hash_rendered_snapshot(rendered: &str) -> u64 { + let mut hasher = DefaultHasher::new(); + rendered.hash(&mut hasher); + hasher.finish() +} + +pub(super) fn preprocess_includes( + content: &str, + base_dir: &Path, + depth: u8, + source_files: &mut BTreeSet, +) -> Result { + if depth > 10 { + return Err(ProxyError::Config("Include depth > 10".into())); + } + let mut output = String::with_capacity(content.len()); + for line in content.lines() { + let trimmed = line.trim(); + if let Some(rest) = trimmed.strip_prefix("include") { + let rest = rest.trim(); + if let Some(rest) = rest.strip_prefix('=') { + let path_str = rest.trim().trim_matches('"'); + let resolved = base_dir.join(path_str); + source_files.insert(normalize_config_path(&resolved)); + let included = std::fs::read_to_string(&resolved) + .map_err(|e| ProxyError::Config(e.to_string()))?; + let included_dir = resolved.parent().unwrap_or(base_dir); + output.push_str(&preprocess_includes( + &included, + included_dir, + depth + 1, + source_files, + )?); + output.push('\n'); + continue; + } + } + output.push_str(line); + output.push('\n'); + } + Ok(output) +} diff --git a/src/config/load/normalize.rs b/src/config/load/normalize.rs new file mode 100644 index 0000000..21103fb --- /dev/null +++ b/src/config/load/normalize.rs @@ -0,0 +1,115 @@ +use crate::error::{ProxyError, Result}; +use tracing::warn; + +pub(super) fn is_valid_tls_domain_name(domain: &str) -> bool { + !domain.is_empty() + && !domain + .chars() + .any(|ch| ch.is_whitespace() || matches!(ch, '/' | '\\')) +} + +pub(super) fn normalize_domain_to_ascii(domain: &str, field: &str) -> Result { + let domain = domain.trim(); + if !is_valid_tls_domain_name(domain) { + return Err(ProxyError::Config(format!( + "Invalid {field}: '{}'. Must be a valid domain name", + domain + ))); + } + + let parsed = url::Url::parse(&format!("https://{domain}/")).map_err(|error| { + ProxyError::Config(format!( + "Invalid {field}: '{}'. IDNA conversion failed: {error}", + domain + )) + })?; + let host = parsed.host_str().ok_or_else(|| { + ProxyError::Config(format!("Invalid {field}: '{}'. Host is empty", domain)) + })?; + Ok(host.to_ascii_lowercase()) +} + +pub(super) fn normalize_mask_host_to_ascii(host: &str, field: &str) -> Result { + let host = host.trim(); + if host.starts_with('[') && host.ends_with(']') { + let inner = &host[1..host.len() - 1]; + let ip = inner.parse::().map_err(|_| { + ProxyError::Config(format!( + "Invalid {field}: '{}'. IPv6 literal is invalid", + host + )) + })?; + return match ip { + std::net::IpAddr::V6(v6) => Ok(format!("[{v6}]")), + std::net::IpAddr::V4(v4) => Ok(v4.to_string()), + }; + } + if let Ok(ip) = host.parse::() { + return match ip { + std::net::IpAddr::V4(v4) => Ok(v4.to_string()), + std::net::IpAddr::V6(v6) => Ok(format!("[{v6}]")), + }; + } + + normalize_domain_to_ascii(host, field) +} + +pub(super) fn parse_exclusive_mask_target(target: &str) -> Option<(&str, u16)> { + let target = target.trim(); + if target.is_empty() { + return None; + } + + if target.starts_with('[') { + let end = target.find(']')?; + if target.get(end + 1..end + 2)? != ":" { + return None; + } + let host = &target[..=end]; + let port = target[end + 2..].parse::().ok()?; + return (port > 0).then_some((host, port)); + } + + let (host, port) = target.rsplit_once(':')?; + if host.is_empty() || host.contains(':') { + return None; + } + let port = port.parse::().ok()?; + (port > 0).then_some((host, port)) +} + +pub(super) fn normalize_exclusive_mask_target(target: &str, field: &str) -> Result { + let (host, port) = parse_exclusive_mask_target(target).ok_or_else(|| { + ProxyError::Config(format!( + "Invalid {field}: '{}'. Expected host:port with port > 0", + target + )) + })?; + let host = normalize_mask_host_to_ascii(host, field)?; + Ok(format!("{host}:{port}")) +} + +pub(super) fn push_unique_nonempty(target: &mut Vec, value: String) { + let trimmed = value.trim(); + if trimmed.is_empty() { + return; + } + if !target.iter().any(|existing| existing == trimmed) { + target.push(trimmed.to_string()); + } +} + +pub(super) fn is_valid_ad_tag(tag: &str) -> bool { + tag.len() == 32 && tag.chars().all(|ch| ch.is_ascii_hexdigit()) +} + +pub(super) fn sanitize_ad_tag(ad_tag: &mut Option) { + let Some(tag) = ad_tag.as_ref() else { + return; + }; + + if !is_valid_ad_tag(tag) { + warn!("Invalid general.ad_tag value, expected exactly 32 hex chars; ad_tag is disabled"); + *ad_tag = None; + } +} diff --git a/src/config/load/runtime_auth.rs b/src/config/load/runtime_auth.rs new file mode 100644 index 0000000..01d1c4b --- /dev/null +++ b/src/config/load/runtime_auth.rs @@ -0,0 +1,112 @@ +use std::collections::HashMap; +use std::collections::hash_map::DefaultHasher; +use std::hash::Hasher; + +use crate::error::{ProxyError, Result}; + +const ACCESS_SECRET_BYTES: usize = 16; + +/// Precomputed, immutable user authentication data used by handshake hot paths. +#[derive(Debug, Clone, Default)] +pub(crate) struct UserAuthSnapshot { + entries: Vec, + by_name: HashMap, + sni_index: HashMap>, + sni_initial_index: HashMap>, +} + +#[derive(Debug, Clone)] +pub(crate) struct UserAuthEntry { + pub(crate) user: String, + pub(crate) secret: [u8; ACCESS_SECRET_BYTES], +} + +impl UserAuthSnapshot { + pub(super) fn from_users(users: &HashMap) -> Result { + let mut entries = Vec::with_capacity(users.len()); + let mut by_name = HashMap::with_capacity(users.len()); + let mut sni_index = HashMap::with_capacity(users.len()); + let mut sni_initial_index = HashMap::with_capacity(users.len()); + + for (user, secret_hex) in users { + let decoded = hex::decode(secret_hex).map_err(|_| ProxyError::InvalidSecret { + user: user.clone(), + reason: "Must be 32 hex characters".to_string(), + })?; + if decoded.len() != ACCESS_SECRET_BYTES { + return Err(ProxyError::InvalidSecret { + user: user.clone(), + reason: "Must be 32 hex characters".to_string(), + }); + } + + let user_id = u32::try_from(entries.len()).map_err(|_| { + ProxyError::Config("Too many users for runtime auth snapshot".to_string()) + })?; + + let mut secret = [0u8; ACCESS_SECRET_BYTES]; + secret.copy_from_slice(&decoded); + entries.push(UserAuthEntry { + user: user.clone(), + secret, + }); + by_name.insert(user.clone(), user_id); + sni_index + .entry(Self::sni_lookup_hash(user)) + .or_insert_with(Vec::new) + .push(user_id); + if let Some(initial) = user + .as_bytes() + .first() + .map(|byte| byte.to_ascii_lowercase()) + { + sni_initial_index + .entry(initial) + .or_insert_with(Vec::new) + .push(user_id); + } + } + + Ok(Self { + entries, + by_name, + sni_index, + sni_initial_index, + }) + } + + pub(crate) fn entries(&self) -> &[UserAuthEntry] { + &self.entries + } + + pub(crate) fn user_id_by_name(&self, user: &str) -> Option { + self.by_name.get(user).copied() + } + + pub(crate) fn entry_by_id(&self, user_id: u32) -> Option<&UserAuthEntry> { + let idx = usize::try_from(user_id).ok()?; + self.entries.get(idx) + } + + pub(crate) fn sni_candidates(&self, sni: &str) -> Option<&[u32]> { + self.sni_index + .get(&Self::sni_lookup_hash(sni)) + .map(Vec::as_slice) + } + + pub(crate) fn sni_initial_candidates(&self, sni: &str) -> Option<&[u32]> { + let initial = sni + .as_bytes() + .first() + .map(|byte| byte.to_ascii_lowercase())?; + self.sni_initial_index.get(&initial).map(Vec::as_slice) + } + + fn sni_lookup_hash(value: &str) -> u64 { + let mut hasher = DefaultHasher::new(); + for byte in value.bytes() { + hasher.write_u8(byte.to_ascii_lowercase()); + } + hasher.finish() + } +} diff --git a/src/config/load/strict_keys.rs b/src/config/load/strict_keys.rs new file mode 100644 index 0000000..9d832e4 --- /dev/null +++ b/src/config/load/strict_keys.rs @@ -0,0 +1,693 @@ +use tracing::warn; + +use crate::error::{ProxyError, Result}; + +const TOP_LEVEL_CONFIG_KEYS: &[&str] = &[ + "general", + "logging", + "network", + "server", + "timeouts", + "censorship", + "access", + "upstreams", + "show_link", + "dc_overrides", + "default_dc", + "beobachten", + "beobachten_minutes", + "beobachten_flush_secs", + "beobachten_file", + "include", +]; + +const GENERAL_CONFIG_KEYS: &[&str] = &[ + "data_path", + "quota_state_path", + "config_strict", + "modes", + "prefer_ipv6", + "fast_mode", + "use_middle_proxy", + "proxy_secret_path", + "proxy_secret_url", + "proxy_config_v4_cache_path", + "proxy_config_v4_url", + "proxy_config_v6_cache_path", + "proxy_config_v6_url", + "ad_tag", + "middle_proxy_nat_ip", + "middle_proxy_nat_probe", + "middle_proxy_nat_stun", + "middle_proxy_nat_stun_servers", + "stun_nat_probe_concurrency", + "middle_proxy_pool_size", + "middle_proxy_warm_standby", + "me_init_retry_attempts", + "me2dc_fallback", + "me2dc_fast", + "me_keepalive_enabled", + "me_keepalive_interval_secs", + "me_keepalive_jitter_secs", + "me_keepalive_payload_random", + "rpc_proxy_req_every", + "me_writer_cmd_channel_capacity", + "me_route_channel_capacity", + "me_c2me_channel_capacity", + "me_c2me_send_timeout_ms", + "me_reader_route_data_wait_ms", + "me_d2c_flush_batch_max_frames", + "me_d2c_flush_batch_max_bytes", + "me_d2c_flush_batch_max_delay_us", + "me_d2c_ack_flush_immediate", + "me_quota_soft_overshoot_bytes", + "me_d2c_frame_buf_shrink_threshold_bytes", + "direct_relay_copy_buf_c2s_bytes", + "direct_relay_copy_buf_s2c_bytes", + "crypto_pending_buffer", + "max_client_frame", + "desync_all_full", + "beobachten", + "beobachten_minutes", + "beobachten_flush_secs", + "beobachten_file", + "hardswap", + "me_warmup_stagger_enabled", + "me_warmup_step_delay_ms", + "me_warmup_step_jitter_ms", + "me_reconnect_max_concurrent_per_dc", + "me_reconnect_backoff_base_ms", + "me_reconnect_backoff_cap_ms", + "me_reconnect_fast_retry_count", + "me_single_endpoint_shadow_writers", + "me_single_endpoint_outage_mode_enabled", + "me_single_endpoint_outage_disable_quarantine", + "me_single_endpoint_outage_backoff_min_ms", + "me_single_endpoint_outage_backoff_max_ms", + "me_single_endpoint_shadow_rotate_every_secs", + "me_floor_mode", + "me_adaptive_floor_idle_secs", + "me_adaptive_floor_min_writers_single_endpoint", + "me_adaptive_floor_min_writers_multi_endpoint", + "me_adaptive_floor_recover_grace_secs", + "me_adaptive_floor_writers_per_core_total", + "me_adaptive_floor_cpu_cores_override", + "me_adaptive_floor_max_extra_writers_single_per_core", + "me_adaptive_floor_max_extra_writers_multi_per_core", + "me_adaptive_floor_max_active_writers_per_core", + "me_adaptive_floor_max_warm_writers_per_core", + "me_adaptive_floor_max_active_writers_global", + "me_adaptive_floor_max_warm_writers_global", + "upstream_connect_retry_attempts", + "upstream_connect_retry_backoff_ms", + "upstream_connect_budget_ms", + "tg_connect", + "upstream_unhealthy_fail_threshold", + "upstream_connect_failfast_hard_errors", + "stun_iface_mismatch_ignore", + "unknown_dc_log_path", + "unknown_dc_file_log_enabled", + "log_level", + "disable_colors", + "telemetry", + "me_socks_kdf_policy", + "me_route_backpressure_enabled", + "me_route_fairshare_enabled", + "me_route_backpressure_base_timeout_ms", + "me_route_backpressure_high_timeout_ms", + "me_route_backpressure_high_watermark_pct", + "me_health_interval_ms_unhealthy", + "me_health_interval_ms_healthy", + "me_admission_poll_ms", + "me_warn_rate_limit_ms", + "me_route_no_writer_mode", + "me_route_no_writer_wait_ms", + "me_route_hybrid_max_wait_ms", + "me_route_blocking_send_timeout_ms", + "me_route_inline_recovery_attempts", + "me_route_inline_recovery_wait_ms", + "links", + "fast_mode_min_tls_record", + "update_every", + "me_reinit_every_secs", + "me_hardswap_warmup_delay_min_ms", + "me_hardswap_warmup_delay_max_ms", + "me_hardswap_warmup_extra_passes", + "me_hardswap_warmup_pass_backoff_base_ms", + "me_config_stable_snapshots", + "me_config_apply_cooldown_secs", + "me_snapshot_require_http_2xx", + "me_snapshot_reject_empty_map", + "me_snapshot_min_proxy_for_lines", + "proxy_secret_stable_snapshots", + "proxy_secret_rotate_runtime", + "me_secret_atomic_snapshot", + "proxy_secret_len_max", + "me_pool_drain_ttl_secs", + "me_instadrain", + "me_pool_drain_threshold", + "me_pool_drain_soft_evict_enabled", + "me_pool_drain_soft_evict_grace_secs", + "me_pool_drain_soft_evict_per_writer", + "me_pool_drain_soft_evict_budget_per_core", + "me_pool_drain_soft_evict_cooldown_ms", + "me_bind_stale_mode", + "me_bind_stale_ttl_secs", + "me_pool_min_fresh_ratio", + "me_reinit_drain_timeout_secs", + "proxy_secret_auto_reload_secs", + "proxy_config_auto_reload_secs", + "me_reinit_singleflight", + "me_reinit_trigger_channel", + "me_reinit_coalesce_window_ms", + "me_deterministic_writer_sort", + "me_writer_pick_mode", + "me_writer_pick_sample_size", + "ntp_check", + "ntp_servers", + "auto_degradation_enabled", + "degradation_min_unavailable_dc_groups", + "rst_on_close", +]; + +const NETWORK_CONFIG_KEYS: &[&str] = &[ + "ipv4", + "ipv6", + "prefer", + "multipath", + "stun_use", + "stun_servers", + "stun_tcp_fallback", + "http_ip_detect_urls", + "cache_public_ip_path", + "dns_overrides", +]; + +const SERVER_CONFIG_KEYS: &[&str] = &[ + "port", + "listen_addr_ipv4", + "listen_addr_ipv6", + "listen_unix_sock", + "listen_unix_sock_perm", + "listen_tcp", + "client_mss", + "client_mss_bulk", + "proxy_protocol", + "proxy_protocol_header_timeout_ms", + "proxy_protocol_trusted_cidrs", + "metrics_port", + "metrics_listen", + "metrics_whitelist", + "api", + "admin_api", + "listeners", + "listen_backlog", + "max_connections", + "accept_permit_timeout_ms", + "conntrack_control", +]; + +const API_CONFIG_KEYS: &[&str] = &[ + "enabled", + "listen", + "whitelist", + "gray_action", + "auth_header", + "request_body_limit_bytes", + "minimal_runtime_enabled", + "minimal_runtime_cache_ttl_ms", + "runtime_edge_enabled", + "runtime_edge_cache_ttl_ms", + "runtime_edge_top_n", + "runtime_edge_events_capacity", + "read_only", +]; + +const CONNTRACK_CONTROL_CONFIG_KEYS: &[&str] = &[ + "inline_conntrack_control", + "mode", + "backend", + "profile", + "hybrid_listener_ips", + "pressure_high_watermark_pct", + "pressure_low_watermark_pct", + "delete_budget_per_sec", +]; + +const LISTENER_CONFIG_KEYS: &[&str] = &[ + "ip", + "port", + "client_mss", + "synlimit", + "synlimit_seconds", + "synlimit_hitcount", + "synlimit_burst", + "synlimit_ios_seconds", + "synlimit_ios_hitcount", + "synlimit_ios_burst", + "synlimit_hashlimit_expire_ms", + "synlimit_hashlimit_size", + "announce", + "announce_ip", + "proxy_protocol", + "reuse_allow", +]; + +const TIMEOUTS_CONFIG_KEYS: &[&str] = &[ + "client_first_byte_idle_secs", + "client_handshake", + "relay_idle_policy_v2_enabled", + "relay_client_idle_soft_secs", + "relay_client_idle_hard_secs", + "relay_idle_grace_after_downstream_activity_secs", + "client_keepalive", + "client_ack", + "me_one_retry", + "me_one_timeout_ms", +]; + +const CENSORSHIP_CONFIG_KEYS: &[&str] = &[ + "tls_domain", + "tls_domains", + "unknown_sni_action", + "tls_fetch_scope", + "tls_fetch", + "mask", + "mask_dynamic", + "mask_host", + "mask_port", + "exclusive_mask", + "mask_unix_sock", + "fake_cert_len", + "tls_emulation", + "tls_front_dir", + "server_hello_delay_min_ms", + "server_hello_delay_max_ms", + "tls_new_session_tickets", + "serverhello_compact", + "tls_full_cert_ttl_secs", + "alpn_enforce", + "mask_proxy_protocol", + "mask_shape_hardening", + "mask_shape_hardening_aggressive_mode", + "mask_shape_bucket_floor_bytes", + "mask_shape_bucket_cap_bytes", + "mask_shape_above_cap_blur", + "mask_shape_above_cap_blur_max_bytes", + "mask_relay_max_bytes", + "mask_relay_timeout_ms", + "mask_relay_idle_timeout_ms", + "mask_classifier_prefetch_timeout_ms", + "mask_timing_normalization_enabled", + "mask_timing_normalization_floor_ms", + "mask_timing_normalization_ceiling_ms", +]; + +const TLS_FETCH_CONFIG_KEYS: &[&str] = &[ + "profiles", + "strict_route", + "attempt_timeout_ms", + "total_budget_ms", + "grease_enabled", + "deterministic", + "profile_cache_ttl_secs", +]; + +const ACCESS_CONFIG_KEYS: &[&str] = &[ + "users", + "user_enabled", + "user_ad_tags", + "user_max_tcp_conns", + "user_max_tcp_conns_global_each", + "user_expirations", + "user_data_quota", + "user_rate_limits", + "cidr_rate_limits", + "user_max_unique_ips", + "user_max_unique_ips_global_each", + "user_max_unique_ips_mode", + "user_max_unique_ips_window_secs", + "replay_check_len", + "replay_window_secs", + "ignore_time_skew", +]; + +const RATE_LIMIT_BPS_CONFIG_KEYS: &[&str] = &["up_bps", "down_bps"]; + +const UPSTREAM_CONFIG_KEYS: &[&str] = &[ + "type", + "interface", + "bind_addresses", + "bindtodevice", + "force_bind", + "address", + "user_id", + "username", + "password", + "url", + "weight", + "enabled", + "scopes", + "ipv4", + "ipv6", +]; + +const PROXY_MODES_CONFIG_KEYS: &[&str] = &["classic", "secure", "tls"]; +const TELEMETRY_CONFIG_KEYS: &[&str] = &["core_enabled", "user_enabled", "me_level"]; +const LINKS_CONFIG_KEYS: &[&str] = &["show", "public_host", "public_port"]; +const LOGGING_CONFIG_KEYS: &[&str] = &[ + "destination", + "path", + "rotation", + "max_size_bytes", + "max_files", + "max_age_secs", +]; + +#[derive(Debug)] +struct UnknownConfigKey { + path: String, + suggestion: Option, +} + +fn table_at<'a>(value: &'a toml::Value, path: &[&str]) -> Option<&'a toml::Table> { + let mut current = value; + for segment in path { + current = current.get(*segment)?; + } + current.as_table() +} + +fn is_strict_config(parsed_toml: &toml::Value) -> bool { + table_at(parsed_toml, &["general"]) + .and_then(|table| table.get("config_strict")) + .and_then(toml::Value::as_bool) + .unwrap_or(false) +} + +fn known_config_keys_for_suggestion() -> Vec<&'static str> { + let mut keys = Vec::new(); + for group in [ + TOP_LEVEL_CONFIG_KEYS, + GENERAL_CONFIG_KEYS, + NETWORK_CONFIG_KEYS, + SERVER_CONFIG_KEYS, + API_CONFIG_KEYS, + CONNTRACK_CONTROL_CONFIG_KEYS, + LISTENER_CONFIG_KEYS, + TIMEOUTS_CONFIG_KEYS, + CENSORSHIP_CONFIG_KEYS, + TLS_FETCH_CONFIG_KEYS, + ACCESS_CONFIG_KEYS, + RATE_LIMIT_BPS_CONFIG_KEYS, + UPSTREAM_CONFIG_KEYS, + PROXY_MODES_CONFIG_KEYS, + TELEMETRY_CONFIG_KEYS, + LINKS_CONFIG_KEYS, + LOGGING_CONFIG_KEYS, + ] { + keys.extend_from_slice(group); + } + keys +} + +fn levenshtein_distance(a: &str, b: &str) -> usize { + let b_chars: Vec = b.chars().collect(); + let mut prev: Vec = (0..=b_chars.len()).collect(); + let mut curr = vec![0usize; b_chars.len() + 1]; + + for (i, ca) in a.chars().enumerate() { + curr[0] = i + 1; + for (j, cb) in b_chars.iter().enumerate() { + let replace = if ca == *cb { prev[j] } else { prev[j] + 1 }; + curr[j + 1] = (prev[j + 1] + 1).min(curr[j] + 1).min(replace); + } + std::mem::swap(&mut prev, &mut curr); + } + + prev[b_chars.len()] +} + +fn unknown_key_suggestion(key: &str, known_keys: &[&'static str]) -> Option { + let normalized = key.to_ascii_lowercase(); + let mut best: Option<(&str, usize)> = None; + for known in known_keys { + let distance = levenshtein_distance(&normalized, known); + let is_better = match best { + Some((_, best_distance)) => distance < best_distance, + None => true, + }; + if distance <= 4 && is_better { + best = Some((known, distance)); + } + } + best.map(|(known, _)| known.to_string()) +} + +fn push_unknown_keys( + unknown: &mut Vec, + known_for_suggestion: &[&'static str], + path: &str, + table: &toml::Table, + allowed: &[&str], +) { + for key in table.keys() { + if !allowed.contains(&key.as_str()) { + let full_path = if path.is_empty() { + key.clone() + } else { + format!("{path}.{key}") + }; + unknown.push(UnknownConfigKey { + path: full_path, + suggestion: unknown_key_suggestion(key, known_for_suggestion), + }); + } + } +} + +fn check_known_table( + parsed_toml: &toml::Value, + unknown: &mut Vec, + known_for_suggestion: &[&'static str], + path: &[&str], + allowed: &[&str], +) { + if let Some(table) = table_at(parsed_toml, path) { + push_unknown_keys( + unknown, + known_for_suggestion, + &path.join("."), + table, + allowed, + ); + } +} + +fn check_nested_table_value( + unknown: &mut Vec, + known_for_suggestion: &[&'static str], + path: String, + value: &toml::Value, + allowed: &[&str], +) { + if let Some(table) = value.as_table() { + push_unknown_keys(unknown, known_for_suggestion, &path, table, allowed); + } +} + +fn collect_unknown_config_keys(parsed_toml: &toml::Value) -> Vec { + let known_for_suggestion = known_config_keys_for_suggestion(); + let mut unknown = Vec::new(); + + if let Some(root) = parsed_toml.as_table() { + push_unknown_keys( + &mut unknown, + &known_for_suggestion, + "", + root, + TOP_LEVEL_CONFIG_KEYS, + ); + } + + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["general"], + GENERAL_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["general", "modes"], + PROXY_MODES_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["general", "telemetry"], + TELEMETRY_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["general", "links"], + LINKS_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["logging"], + LOGGING_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["network"], + NETWORK_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["server"], + SERVER_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["server", "api"], + API_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["server", "admin_api"], + API_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["server", "conntrack_control"], + CONNTRACK_CONTROL_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["timeouts"], + TIMEOUTS_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["censorship"], + CENSORSHIP_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["censorship", "tls_fetch"], + TLS_FETCH_CONFIG_KEYS, + ); + check_known_table( + parsed_toml, + &mut unknown, + &known_for_suggestion, + &["access"], + ACCESS_CONFIG_KEYS, + ); + + if let Some(listeners) = table_at(parsed_toml, &["server"]) + .and_then(|table| table.get("listeners")) + .and_then(toml::Value::as_array) + { + for (idx, listener) in listeners.iter().enumerate() { + check_nested_table_value( + &mut unknown, + &known_for_suggestion, + format!("server.listeners[{idx}]"), + listener, + LISTENER_CONFIG_KEYS, + ); + } + } + + if let Some(upstreams) = parsed_toml.get("upstreams").and_then(toml::Value::as_array) { + for (idx, upstream) in upstreams.iter().enumerate() { + check_nested_table_value( + &mut unknown, + &known_for_suggestion, + format!("upstreams[{idx}]"), + upstream, + UPSTREAM_CONFIG_KEYS, + ); + } + } + + for access_map in ["user_rate_limits", "cidr_rate_limits"] { + if let Some(table) = table_at(parsed_toml, &["access"]) + .and_then(|access| access.get(access_map)) + .and_then(toml::Value::as_table) + { + for (entry_name, value) in table { + check_nested_table_value( + &mut unknown, + &known_for_suggestion, + format!("access.{access_map}.{entry_name}"), + value, + RATE_LIMIT_BPS_CONFIG_KEYS, + ); + } + } + } + + unknown +} + +pub(super) fn handle_unknown_config_keys(parsed_toml: &toml::Value) -> Result<()> { + let unknown = collect_unknown_config_keys(parsed_toml); + if unknown.is_empty() { + return Ok(()); + } + + for item in &unknown { + if let Some(suggestion) = item.suggestion.as_deref() { + warn!( + key = %item.path, + suggestion = %suggestion, + "Unknown config key ignored; did you mean the suggested key?" + ); + } else { + warn!(key = %item.path, "Unknown config key ignored"); + } + } + + if is_strict_config(parsed_toml) { + let mut paths = Vec::with_capacity(unknown.len()); + for item in unknown { + if let Some(suggestion) = item.suggestion { + paths.push(format!("{} (did you mean `{}`?)", item.path, suggestion)); + } else { + paths.push(item.path); + } + } + return Err(ProxyError::Config(format!( + "unknown config keys are not allowed when general.config_strict=true: {}", + paths.join(", ") + ))); + } + + Ok(()) +} diff --git a/src/config/load/validation.rs b/src/config/load/validation.rs new file mode 100644 index 0000000..b95f73b --- /dev/null +++ b/src/config/load/validation.rs @@ -0,0 +1,111 @@ +use shadowsocks::config::ServerConfig as ShadowsocksServerConfig; +use tracing::warn; + +use crate::error::{ProxyError, Result}; + +use super::super::types::{LoggingConfig, LoggingDestination, NetworkConfig, UpstreamType}; +use super::ProxyConfig; + +pub(super) fn validate_network_cfg(net: &mut NetworkConfig) -> Result<()> { + if !net.ipv4 && matches!(net.ipv6, Some(false)) { + return Err(ProxyError::Config( + "Both ipv4 and ipv6 are disabled in [network]".to_string(), + )); + } + + if net.prefer != 4 && net.prefer != 6 { + return Err(ProxyError::Config( + "network.prefer must be 4 or 6".to_string(), + )); + } + + if !net.ipv4 && net.prefer == 4 { + warn!("prefer=4 but ipv4=false; forcing prefer=6"); + net.prefer = 6; + } + + if matches!(net.ipv6, Some(false)) && net.prefer == 6 { + warn!("prefer=6 but ipv6=false; forcing prefer=4"); + net.prefer = 4; + } + + Ok(()) +} + +pub(super) fn validate_logging_config(logging: &LoggingConfig) -> Result<()> { + if let Some(path) = logging.path.as_ref() + && path.trim().is_empty() + { + return Err(ProxyError::Config( + "logging.path cannot be empty when provided".to_string(), + )); + } + + if matches!(logging.destination, LoggingDestination::File) && logging.path.is_none() { + return Err(ProxyError::Config( + "logging.path must be set when logging.destination=\"file\"".to_string(), + )); + } + + Ok(()) +} + +pub(super) fn validate_upstreams(config: &ProxyConfig) -> Result<()> { + let has_enabled_shadowsocks = config.upstreams.iter().any(|upstream| { + upstream.enabled && matches!(upstream.upstream_type, UpstreamType::Shadowsocks { .. }) + }); + + if has_enabled_shadowsocks && config.general.use_middle_proxy { + return Err(ProxyError::Config( + "shadowsocks upstreams require general.use_middle_proxy = false".to_string(), + )); + } + + for upstream in &config.upstreams { + if matches!(upstream.ipv4, Some(false)) && matches!(upstream.ipv6, Some(false)) { + return Err(ProxyError::Config( + "upstream.ipv4 and upstream.ipv6 cannot both be false".to_string(), + )); + } + if let Some(prefer) = upstream.prefer + && prefer != 4 + && prefer != 6 + { + return Err(ProxyError::Config( + "upstream.prefer must be 4 or 6".to_string(), + )); + } + + if let UpstreamType::Shadowsocks { url, .. } = &upstream.upstream_type { + let parsed = ShadowsocksServerConfig::from_url(url) + .map_err(|error| ProxyError::Config(format!("invalid shadowsocks url: {error}")))?; + if parsed.plugin().is_some() { + return Err(ProxyError::Config( + "shadowsocks plugins are not supported".to_string(), + )); + } + } + } + + Ok(()) +} + +pub(super) fn normalize_upstream_family_policy(config: &mut ProxyConfig) { + for (idx, upstream) in config.upstreams.iter_mut().enumerate() { + if matches!(upstream.ipv4, Some(false)) && upstream.prefer == Some(4) { + warn!( + upstream = idx, + "upstream.prefer=4 but upstream.ipv4=false; forcing prefer=6" + ); + upstream.prefer = Some(6); + } + + if matches!(upstream.ipv6, Some(false)) && upstream.prefer == Some(6) { + warn!( + upstream = idx, + "upstream.prefer=6 but upstream.ipv6=false; forcing prefer=4" + ); + upstream.prefer = Some(4); + } + } +} diff --git a/src/config/tests/load_basic_tests.rs b/src/config/tests/load_basic_tests.rs new file mode 100644 index 0000000..4e46bfe --- /dev/null +++ b/src/config/tests/load_basic_tests.rs @@ -0,0 +1,2342 @@ +use super::*; + +const TEST_SHADOWSOCKS_URL: &str = + "ss://2022-blake3-aes-256-gcm:MDEyMzQ1Njc4OTAxMjM0NTY3ODkwMTIzNDU2Nzg5MDE=@127.0.0.1:8388"; + +fn load_config_from_temp_toml(toml: &str) -> ProxyConfig { + let nonce = std::time::SystemTime::now() + .duration_since(std::time::UNIX_EPOCH) + .unwrap() + .as_nanos(); + let dir = std::env::temp_dir().join(format!("telemt_load_cfg_{nonce}")); + std::fs::create_dir_all(&dir).unwrap(); + let path = dir.join("config.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + let _ = std::fs::remove_file(path); + let _ = std::fs::remove_dir(dir); + cfg +} + +fn load_config_error_from_temp_toml(toml: &str) -> String { + let nonce = std::time::SystemTime::now() + .duration_since(std::time::UNIX_EPOCH) + .unwrap() + .as_nanos(); + let dir = std::env::temp_dir().join(format!("telemt_load_cfg_error_{nonce}")); + std::fs::create_dir_all(&dir).unwrap(); + let path = dir.join("config.toml"); + std::fs::write(&path, toml).unwrap(); + let error = ProxyConfig::load(&path).unwrap_err().to_string(); + let _ = std::fs::remove_file(path); + let _ = std::fs::remove_dir(dir); + error +} + +#[test] +fn synlimit_synfix_defaults_are_loaded_for_listener() { + let cfg = load_config_from_temp_toml( + r#" + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + + [[server.listeners]] + ip = "0.0.0.0" + port = 443 + synlimit = "iptables" + "#, + ); + + let listener = &cfg.server.listeners[0]; + assert_eq!(listener.synlimit_seconds, 60); + assert_eq!(listener.synlimit_hitcount, 48); + assert_eq!(listener.synlimit_burst, 1); + assert_eq!(listener.synlimit_ios_seconds, 1); + assert_eq!(listener.synlimit_ios_hitcount, 12); + assert_eq!(listener.synlimit_ios_burst, 24); + assert_eq!(listener.synlimit_hashlimit_expire_ms, 60_000); + assert_eq!(listener.synlimit_hashlimit_size, 32_768); +} + +#[test] +fn synlimit_synfix_zero_values_are_rejected() { + for (field, expected) in [ + ( + "synlimit_ios_seconds", + "server.listeners[0].synlimit_ios_seconds must be > 0", + ), + ( + "synlimit_ios_hitcount", + "server.listeners[0].synlimit_ios_hitcount must be > 0", + ), + ( + "synlimit_ios_burst", + "server.listeners[0].synlimit_ios_burst must be > 0", + ), + ( + "synlimit_hashlimit_expire_ms", + "server.listeners[0].synlimit_hashlimit_expire_ms must be > 0", + ), + ( + "synlimit_hashlimit_size", + "server.listeners[0].synlimit_hashlimit_size must be > 0", + ), + ] { + let toml = format!( + r#" + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + + [[server.listeners]] + ip = "0.0.0.0" + port = 443 + synlimit = "iptables" + {field} = 0 + "# + ); + let error = load_config_error_from_temp_toml(&toml); + assert!(error.contains(expected), "{field}: {error}"); + } +} + +#[test] +fn serde_defaults_remain_unchanged_for_present_sections() { + let toml = r#" + [network] + [general] + [server] + [access] + "#; + let cfg: ProxyConfig = toml::from_str(toml).unwrap(); + + assert_eq!(cfg.logging, LoggingConfig::default()); + assert_eq!(cfg.network.ipv6, default_network_ipv6()); + assert_eq!(cfg.network.stun_use, default_true()); + assert_eq!(cfg.network.stun_tcp_fallback, default_stun_tcp_fallback()); + assert_eq!( + cfg.general.middle_proxy_warm_standby, + default_middle_proxy_warm_standby() + ); + assert_eq!( + cfg.general.me_reconnect_max_concurrent_per_dc, + default_me_reconnect_max_concurrent_per_dc() + ); + assert_eq!( + cfg.general.me_reconnect_fast_retry_count, + default_me_reconnect_fast_retry_count() + ); + assert_eq!( + cfg.general.me_init_retry_attempts, + default_me_init_retry_attempts() + ); + assert_eq!(cfg.general.me2dc_fallback, default_me2dc_fallback()); + assert_eq!(cfg.general.me2dc_fast, default_me2dc_fast()); + assert_eq!( + cfg.general.proxy_config_v4_cache_path, + default_proxy_config_v4_cache_path() + ); + assert_eq!( + cfg.general.proxy_config_v6_cache_path, + default_proxy_config_v6_cache_path() + ); + assert_eq!( + cfg.general.me_single_endpoint_shadow_writers, + default_me_single_endpoint_shadow_writers() + ); + assert_eq!( + cfg.general.me_single_endpoint_outage_mode_enabled, + default_me_single_endpoint_outage_mode_enabled() + ); + assert_eq!( + cfg.general.me_single_endpoint_outage_disable_quarantine, + default_me_single_endpoint_outage_disable_quarantine() + ); + assert_eq!( + cfg.general.me_single_endpoint_outage_backoff_min_ms, + default_me_single_endpoint_outage_backoff_min_ms() + ); + assert_eq!( + cfg.general.me_single_endpoint_outage_backoff_max_ms, + default_me_single_endpoint_outage_backoff_max_ms() + ); + assert_eq!( + cfg.general.me_single_endpoint_shadow_rotate_every_secs, + default_me_single_endpoint_shadow_rotate_every_secs() + ); + assert_eq!(cfg.general.me_floor_mode, MeFloorMode::default()); + assert_eq!( + cfg.general.me_adaptive_floor_idle_secs, + default_me_adaptive_floor_idle_secs() + ); + assert_eq!( + cfg.general.me_adaptive_floor_min_writers_single_endpoint, + default_me_adaptive_floor_min_writers_single_endpoint() + ); + assert_eq!( + cfg.general.me_adaptive_floor_recover_grace_secs, + default_me_adaptive_floor_recover_grace_secs() + ); + assert_eq!( + cfg.general.upstream_connect_retry_attempts, + default_upstream_connect_retry_attempts() + ); + assert_eq!( + cfg.general.upstream_connect_retry_backoff_ms, + default_upstream_connect_retry_backoff_ms() + ); + assert_eq!( + cfg.general.upstream_unhealthy_fail_threshold, + default_upstream_unhealthy_fail_threshold() + ); + assert_eq!( + cfg.general.upstream_connect_failfast_hard_errors, + default_upstream_connect_failfast_hard_errors() + ); + assert_eq!( + cfg.general.rpc_proxy_req_every, + default_rpc_proxy_req_every() + ); + assert_eq!(cfg.general.beobachten_file, default_beobachten_file()); + assert_eq!(cfg.general.update_every, default_update_every()); + assert_eq!(cfg.server.listen_addr_ipv4, default_listen_addr_ipv4()); + assert_eq!(cfg.server.listen_addr_ipv6, default_listen_addr_ipv6_opt()); + assert_eq!(cfg.server.client_mss_value(), Ok(None)); + assert_eq!( + cfg.server.proxy_protocol_trusted_cidrs, + default_proxy_protocol_trusted_cidrs() + ); + assert_eq!(cfg.censorship.unknown_sni_action, UnknownSniAction::Drop); + assert_eq!(cfg.server.api.listen, default_api_listen()); + assert_eq!(cfg.server.api.whitelist, default_api_whitelist()); + assert_eq!(cfg.server.api.gray_action, ApiGrayAction::Drop); + assert_eq!( + cfg.server.api.request_body_limit_bytes, + default_api_request_body_limit_bytes() + ); + assert_eq!( + cfg.server.api.minimal_runtime_enabled, + default_api_minimal_runtime_enabled() + ); + assert_eq!( + cfg.server.api.minimal_runtime_cache_ttl_ms, + default_api_minimal_runtime_cache_ttl_ms() + ); + assert_eq!( + cfg.server.api.runtime_edge_enabled, + default_api_runtime_edge_enabled() + ); + assert_eq!( + cfg.server.api.runtime_edge_cache_ttl_ms, + default_api_runtime_edge_cache_ttl_ms() + ); + assert_eq!( + cfg.server.api.runtime_edge_top_n, + default_api_runtime_edge_top_n() + ); + assert_eq!( + cfg.server.api.runtime_edge_events_capacity, + default_api_runtime_edge_events_capacity() + ); + assert_eq!( + cfg.server.conntrack_control.inline_conntrack_control, + default_conntrack_control_enabled() + ); + assert_eq!(cfg.server.conntrack_control.mode, ConntrackMode::default()); + assert_eq!( + cfg.server.conntrack_control.backend, + ConntrackBackend::default() + ); + assert_eq!( + cfg.server.conntrack_control.profile, + ConntrackPressureProfile::default() + ); + assert_eq!( + cfg.server.conntrack_control.pressure_high_watermark_pct, + default_conntrack_pressure_high_watermark_pct() + ); + assert_eq!( + cfg.server.conntrack_control.pressure_low_watermark_pct, + default_conntrack_pressure_low_watermark_pct() + ); + assert_eq!( + cfg.server.conntrack_control.delete_budget_per_sec, + default_conntrack_delete_budget_per_sec() + ); + assert_eq!(cfg.access.users, default_access_users()); + assert_eq!( + cfg.access.user_max_tcp_conns_global_each, + default_user_max_tcp_conns_global_each() + ); + assert_eq!( + cfg.access.user_max_unique_ips_mode, + UserMaxUniqueIpsMode::default() + ); + assert_eq!( + cfg.access.user_max_unique_ips_window_secs, + default_user_max_unique_ips_window_secs() + ); +} + +#[test] +fn logging_config_is_loaded_from_strict_config() { + let cfg = load_config_from_temp_toml( + r#" + [general] + config_strict = true + + [general.modes] + classic = false + secure = false + tls = true + + [logging] + destination = "file" + path = "/tmp/telemt.log" + rotation = "daily" + max_size_bytes = 1024 + max_files = 3 + max_age_secs = 60 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#, + ); + + assert_eq!(cfg.logging.destination, LoggingDestination::File); + assert_eq!(cfg.logging.path.as_deref(), Some("/tmp/telemt.log")); + assert_eq!(cfg.logging.rotation, LogRotation::Daily); + assert_eq!(cfg.logging.max_size_bytes, 1024); + assert_eq!(cfg.logging.max_files, 3); + assert_eq!(cfg.logging.max_age_secs, 60); +} + +#[test] +fn file_logging_requires_path() { + let error = load_config_error_from_temp_toml( + r#" + [general.modes] + classic = false + secure = false + tls = true + + [logging] + destination = "file" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#, + ); + + assert!(error.contains("logging.path must be set")); +} + +#[test] +fn impl_defaults_are_sourced_from_default_helpers() { + let network = NetworkConfig::default(); + assert_eq!(network.ipv6, default_network_ipv6()); + assert_eq!(network.stun_use, default_true()); + assert_eq!(network.stun_tcp_fallback, default_stun_tcp_fallback()); + + let general = GeneralConfig::default(); + assert_eq!( + general.middle_proxy_warm_standby, + default_middle_proxy_warm_standby() + ); + assert_eq!( + general.me_reconnect_max_concurrent_per_dc, + default_me_reconnect_max_concurrent_per_dc() + ); + assert_eq!( + general.me_reconnect_fast_retry_count, + default_me_reconnect_fast_retry_count() + ); + assert_eq!( + general.me_init_retry_attempts, + default_me_init_retry_attempts() + ); + assert_eq!(general.me2dc_fallback, default_me2dc_fallback()); + assert_eq!(general.me2dc_fast, default_me2dc_fast()); + assert_eq!( + general.proxy_config_v4_cache_path, + default_proxy_config_v4_cache_path() + ); + assert_eq!( + general.proxy_config_v6_cache_path, + default_proxy_config_v6_cache_path() + ); + assert_eq!( + general.me_single_endpoint_shadow_writers, + default_me_single_endpoint_shadow_writers() + ); + assert_eq!( + general.me_single_endpoint_outage_mode_enabled, + default_me_single_endpoint_outage_mode_enabled() + ); + assert_eq!( + general.me_single_endpoint_outage_disable_quarantine, + default_me_single_endpoint_outage_disable_quarantine() + ); + assert_eq!( + general.me_single_endpoint_outage_backoff_min_ms, + default_me_single_endpoint_outage_backoff_min_ms() + ); + assert_eq!( + general.me_single_endpoint_outage_backoff_max_ms, + default_me_single_endpoint_outage_backoff_max_ms() + ); + assert_eq!( + general.me_single_endpoint_shadow_rotate_every_secs, + default_me_single_endpoint_shadow_rotate_every_secs() + ); + assert_eq!(general.me_floor_mode, MeFloorMode::default()); + assert_eq!( + general.me_adaptive_floor_idle_secs, + default_me_adaptive_floor_idle_secs() + ); + assert_eq!( + general.me_adaptive_floor_min_writers_single_endpoint, + default_me_adaptive_floor_min_writers_single_endpoint() + ); + assert_eq!( + general.me_adaptive_floor_recover_grace_secs, + default_me_adaptive_floor_recover_grace_secs() + ); + assert_eq!( + general.upstream_connect_retry_attempts, + default_upstream_connect_retry_attempts() + ); + assert_eq!( + general.upstream_connect_retry_backoff_ms, + default_upstream_connect_retry_backoff_ms() + ); + assert_eq!( + general.upstream_unhealthy_fail_threshold, + default_upstream_unhealthy_fail_threshold() + ); + assert_eq!( + general.upstream_connect_failfast_hard_errors, + default_upstream_connect_failfast_hard_errors() + ); + assert_eq!(general.rpc_proxy_req_every, default_rpc_proxy_req_every()); + assert_eq!(general.beobachten_file, default_beobachten_file()); + assert_eq!(general.update_every, default_update_every()); + + let server = ServerConfig::default(); + assert_eq!(server.listen_addr_ipv6, Some(default_listen_addr_ipv6())); + assert_eq!( + server.proxy_protocol_trusted_cidrs, + default_proxy_protocol_trusted_cidrs() + ); + assert_eq!( + AntiCensorshipConfig::default().unknown_sni_action, + UnknownSniAction::Drop + ); + assert_eq!(server.api.listen, default_api_listen()); + assert_eq!(server.api.whitelist, default_api_whitelist()); + assert_eq!(server.api.gray_action, ApiGrayAction::Drop); + assert_eq!( + server.api.request_body_limit_bytes, + default_api_request_body_limit_bytes() + ); + assert_eq!( + server.api.minimal_runtime_enabled, + default_api_minimal_runtime_enabled() + ); + assert_eq!( + server.api.minimal_runtime_cache_ttl_ms, + default_api_minimal_runtime_cache_ttl_ms() + ); + assert_eq!( + server.api.runtime_edge_enabled, + default_api_runtime_edge_enabled() + ); + assert_eq!( + server.api.runtime_edge_cache_ttl_ms, + default_api_runtime_edge_cache_ttl_ms() + ); + assert_eq!( + server.api.runtime_edge_top_n, + default_api_runtime_edge_top_n() + ); + assert_eq!( + server.api.runtime_edge_events_capacity, + default_api_runtime_edge_events_capacity() + ); + assert_eq!( + server.conntrack_control.inline_conntrack_control, + default_conntrack_control_enabled() + ); + assert_eq!(server.conntrack_control.mode, ConntrackMode::default()); + assert_eq!( + server.conntrack_control.backend, + ConntrackBackend::default() + ); + assert_eq!( + server.conntrack_control.profile, + ConntrackPressureProfile::default() + ); + assert_eq!( + server.conntrack_control.pressure_high_watermark_pct, + default_conntrack_pressure_high_watermark_pct() + ); + assert_eq!( + server.conntrack_control.pressure_low_watermark_pct, + default_conntrack_pressure_low_watermark_pct() + ); + assert_eq!( + server.conntrack_control.delete_budget_per_sec, + default_conntrack_delete_budget_per_sec() + ); + + let access = AccessConfig::default(); + assert_eq!(access.users, default_access_users()); + assert_eq!( + access.user_max_tcp_conns_global_each, + default_user_max_tcp_conns_global_each() + ); +} + +#[test] +fn proxy_protocol_trusted_cidrs_missing_uses_trust_all_but_explicit_empty_stays_empty() { + let cfg_missing: ProxyConfig = toml::from_str( + r#" + [server] + [general] + [network] + [access] + "#, + ) + .unwrap(); + assert_eq!( + cfg_missing.server.proxy_protocol_trusted_cidrs, + default_proxy_protocol_trusted_cidrs() + ); + + let cfg_explicit_empty: ProxyConfig = toml::from_str( + r#" + [server] + proxy_protocol_trusted_cidrs = [] + + [general] + [network] + [access] + "#, + ) + .unwrap(); + assert!( + cfg_explicit_empty + .server + .proxy_protocol_trusted_cidrs + .is_empty() + ); +} + +#[test] +fn conntrack_inline_explicit_flag_is_false_when_omitted() { + let cfg = load_config_from_temp_toml( + r#" + [general] + [network] + [server] + [server.conntrack_control] + [access] + "#, + ); + assert!( + !cfg.server + .conntrack_control + .inline_conntrack_control_explicit + ); +} + +#[test] +fn conntrack_inline_explicit_flag_is_true_when_present() { + let cfg = load_config_from_temp_toml( + r#" + [general] + [network] + [server] + [server.conntrack_control] + inline_conntrack_control = true + [access] + "#, + ); + assert!( + cfg.server + .conntrack_control + .inline_conntrack_control_explicit + ); +} + +#[test] +fn unknown_sni_action_parses_and_defaults_to_drop() { + let cfg_default: ProxyConfig = toml::from_str( + r#" + [server] + [general] + [network] + [access] + [censorship] + "#, + ) + .unwrap(); + assert_eq!( + cfg_default.censorship.unknown_sni_action, + UnknownSniAction::Drop + ); + + let cfg_mask: ProxyConfig = toml::from_str( + r#" + [server] + [general] + [network] + [access] + [censorship] + unknown_sni_action = "mask" + "#, + ) + .unwrap(); + assert_eq!( + cfg_mask.censorship.unknown_sni_action, + UnknownSniAction::Mask + ); + + let cfg_accept: ProxyConfig = toml::from_str( + r#" + [server] + [general] + [network] + [access] + [censorship] + unknown_sni_action = "accept" + "#, + ) + .unwrap(); + assert_eq!( + cfg_accept.censorship.unknown_sni_action, + UnknownSniAction::Accept + ); + + let cfg_reject: ProxyConfig = toml::from_str( + r#" + [server] + [general] + [network] + [access] + [censorship] + unknown_sni_action = "reject_handshake" + "#, + ) + .unwrap(); + assert_eq!( + cfg_reject.censorship.unknown_sni_action, + UnknownSniAction::RejectHandshake + ); +} + +#[test] +fn exclusive_mask_parses_domain_target_map() { + let cfg = load_config_from_temp_toml( + r#" + [general] + [network] + [server] + [access] + [censorship] + tls_domain = "weißbiergärten.de" + tls_domains = ["bürgeramt.de"] + [censorship.exclusive_mask] + "bürgeramt.de" = "rindfleischetikettierungsüberwachungsaufgabenübertragungsgesetz.de:443" + "ipv6.example" = "[::1]:443" + "#, + ); + + assert!(cfg.censorship.tls_domain.is_ascii()); + assert!(cfg.censorship.tls_domain.contains("xn--")); + assert_eq!(cfg.censorship.tls_domains.len(), 1); + let normalized_extra = &cfg.censorship.tls_domains[0]; + assert!(normalized_extra.is_ascii()); + assert!(normalized_extra.contains("xn--")); + + let normalized_target = cfg + .censorship + .exclusive_mask + .get(normalized_extra) + .expect("exclusive_mask key must match normalized tls_domains entry"); + assert!(normalized_target.is_ascii()); + assert!(normalized_target.contains("xn--")); + assert!(normalized_target.ends_with(":443")); + assert_eq!( + cfg.censorship.exclusive_mask.get("ipv6.example"), + Some(&"[::1]:443".to_string()) + ); +} + +#[test] +fn api_gray_action_parses_and_defaults_to_drop() { + let cfg_default: ProxyConfig = toml::from_str( + r#" + [server] + [general] + [network] + [access] + "#, + ) + .unwrap(); + assert_eq!(cfg_default.server.api.gray_action, ApiGrayAction::Drop); + + let cfg_api: ProxyConfig = toml::from_str( + r#" + [server] + [general] + [network] + [access] + [server.api] + gray_action = "api" + "#, + ) + .unwrap(); + assert_eq!(cfg_api.server.api.gray_action, ApiGrayAction::Api); + + let cfg_200: ProxyConfig = toml::from_str( + r#" + [server] + [general] + [network] + [access] + [server.api] + gray_action = "200" + "#, + ) + .unwrap(); + assert_eq!(cfg_200.server.api.gray_action, ApiGrayAction::Ok200); + + let cfg_drop: ProxyConfig = toml::from_str( + r#" + [server] + [general] + [network] + [access] + [server.api] + gray_action = "drop" + "#, + ) + .unwrap(); + assert_eq!(cfg_drop.server.api.gray_action, ApiGrayAction::Drop); +} + +#[test] +fn top_level_beobachten_keys_migrate_to_general_when_general_not_explicit() { + let cfg = load_config_from_temp_toml( + r#" + beobachten = false + beobachten_minutes = 7 + beobachten_flush_secs = 3 + beobachten_file = "tmp/legacy-beob.txt" + + [server] + [general] + [network] + [access] + "#, + ); + + assert!(!cfg.general.beobachten); + assert_eq!(cfg.general.beobachten_minutes, 7); + assert_eq!(cfg.general.beobachten_flush_secs, 3); + assert_eq!(cfg.general.beobachten_file, "tmp/legacy-beob.txt"); +} + +#[test] +fn general_beobachten_keys_have_priority_over_legacy_top_level() { + let cfg = load_config_from_temp_toml( + r#" + beobachten = true + beobachten_minutes = 30 + beobachten_flush_secs = 30 + beobachten_file = "tmp/legacy-beob.txt" + + [server] + [general] + beobachten = false + beobachten_minutes = 5 + beobachten_flush_secs = 2 + beobachten_file = "tmp/general-beob.txt" + [network] + [access] + "#, + ); + + assert!(!cfg.general.beobachten); + assert_eq!(cfg.general.beobachten_minutes, 5); + assert_eq!(cfg.general.beobachten_flush_secs, 2); + assert_eq!(cfg.general.beobachten_file, "tmp/general-beob.txt"); +} + +#[test] +fn dc_overrides_allow_string_and_array() { + let toml = r#" + [dc_overrides] + "201" = "149.154.175.50:443" + "202" = ["149.154.167.51:443", "149.154.175.100:443"] + "#; + let cfg: ProxyConfig = toml::from_str(toml).unwrap(); + assert_eq!(cfg.dc_overrides["201"], vec!["149.154.175.50:443"]); + assert_eq!( + cfg.dc_overrides["202"], + vec!["149.154.167.51:443", "149.154.175.100:443"] + ); +} + +#[test] +fn load_with_metadata_collects_include_files() { + let nonce = std::time::SystemTime::now() + .duration_since(std::time::UNIX_EPOCH) + .unwrap() + .as_nanos(); + let dir = std::env::temp_dir().join(format!("telemt_load_metadata_{nonce}")); + std::fs::create_dir_all(&dir).unwrap(); + let main_path = dir.join("config.toml"); + let include_path = dir.join("included.toml"); + + std::fs::write( + &include_path, + r#" + [access.users] + user = "00000000000000000000000000000000" + "#, + ) + .unwrap(); + std::fs::write( + &main_path, + r#" + include = "included.toml" + + [censorship] + tls_domain = "example.com" + "#, + ) + .unwrap(); + + let loaded = ProxyConfig::load_with_metadata(&main_path).unwrap(); + let main_normalized = normalize_config_path(&main_path); + let include_normalized = normalize_config_path(&include_path); + + assert!(loaded.source_files.contains(&main_normalized)); + assert!(loaded.source_files.contains(&include_normalized)); + + let _ = std::fs::remove_file(main_path); + let _ = std::fs::remove_file(include_path); + let _ = std::fs::remove_dir(dir); +} + +#[test] +fn dc_overrides_inject_dc203_default() { + let toml = r#" + [general] + use_middle_proxy = false + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_dc_override_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert!( + cfg.dc_overrides + .get("203") + .map(|v| v.contains(&"91.105.192.100:443".to_string())) + .unwrap_or(false) + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn update_every_overrides_legacy_fields() { + let toml = r#" + [general] + update_every = 123 + proxy_secret_auto_reload_secs = 700 + proxy_config_auto_reload_secs = 800 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_update_every_override_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!(cfg.general.effective_update_every_secs(), 123); + let _ = std::fs::remove_file(path); +} + +#[test] +fn update_every_fallback_to_legacy_min() { + let toml = r#" + [general] + proxy_secret_auto_reload_secs = 600 + proxy_config_auto_reload_secs = 120 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_update_every_legacy_min_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!(cfg.general.update_every, None); + assert_eq!(cfg.general.effective_update_every_secs(), 120); + let _ = std::fs::remove_file(path); +} + +#[test] +fn update_every_zero_is_rejected() { + let toml = r#" + [general] + update_every = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_update_every_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.update_every must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn stun_nat_probe_concurrency_zero_is_rejected() { + let toml = r#" + [general] + stun_nat_probe_concurrency = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_stun_nat_probe_concurrency_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.stun_nat_probe_concurrency must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_reinit_every_default_is_set() { + let toml = r#" + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_reinit_every_default_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!( + cfg.general.me_reinit_every_secs, + default_me_reinit_every_secs() + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_reinit_every_zero_is_rejected() { + let toml = r#" + [general] + me_reinit_every_secs = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_reinit_every_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_reinit_every_secs must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_single_endpoint_outage_backoff_range_is_validated() { + let toml = r#" + [general] + me_single_endpoint_outage_backoff_min_ms = 4000 + me_single_endpoint_outage_backoff_max_ms = 3000 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_single_endpoint_outage_backoff_range_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains( + "general.me_single_endpoint_outage_backoff_min_ms must be <= general.me_single_endpoint_outage_backoff_max_ms" + )); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_single_endpoint_shadow_writers_too_large_is_rejected() { + let toml = r#" + [general] + me_single_endpoint_shadow_writers = 33 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_single_endpoint_shadow_writers_limit_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_single_endpoint_shadow_writers must be within [0, 32]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_adaptive_floor_min_writers_out_of_range_is_rejected() { + let toml = r#" + [general] + me_adaptive_floor_min_writers_single_endpoint = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_adaptive_floor_min_writers_out_of_range_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!( + err.contains( + "general.me_adaptive_floor_min_writers_single_endpoint must be within [1, 32]" + ) + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_floor_mode_adaptive_is_parsed() { + let toml = r#" + [general] + me_floor_mode = "adaptive" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_floor_mode_adaptive_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!(cfg.general.me_floor_mode, MeFloorMode::Adaptive); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_adaptive_floor_max_active_writers_per_core_zero_is_rejected() { + let toml = r#" + [general] + me_adaptive_floor_max_active_writers_per_core = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_adaptive_floor_max_active_per_core_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_adaptive_floor_max_active_writers_per_core must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_adaptive_floor_max_warm_writers_global_zero_is_rejected() { + let toml = r#" + [general] + me_adaptive_floor_max_warm_writers_global = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_adaptive_floor_max_warm_global_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_adaptive_floor_max_warm_writers_global must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn upstream_connect_retry_attempts_zero_is_rejected() { + let toml = r#" + [general] + upstream_connect_retry_attempts = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_upstream_connect_retry_attempts_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.upstream_connect_retry_attempts must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn upstream_unhealthy_fail_threshold_zero_is_rejected() { + let toml = r#" + [general] + upstream_unhealthy_fail_threshold = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_upstream_unhealthy_fail_threshold_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.upstream_unhealthy_fail_threshold must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn tg_connect_zero_is_rejected() { + let toml = r#" + [general] + tg_connect = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_tg_connect_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.tg_connect must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn rpc_proxy_req_every_out_of_range_is_rejected() { + let toml = r#" + [general] + rpc_proxy_req_every = 9 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_rpc_proxy_req_every_out_of_range_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.rpc_proxy_req_every must be 0 or within [10, 300]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn rpc_proxy_req_every_zero_and_valid_range_are_accepted() { + let toml_zero = r#" + [general] + rpc_proxy_req_every = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path_zero = dir.join("telemt_rpc_proxy_req_every_zero_ok_test.toml"); + std::fs::write(&path_zero, toml_zero).unwrap(); + let cfg_zero = ProxyConfig::load(&path_zero).unwrap(); + assert_eq!(cfg_zero.general.rpc_proxy_req_every, 0); + let _ = std::fs::remove_file(path_zero); + + let toml_valid = r#" + [general] + rpc_proxy_req_every = 40 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let path_valid = dir.join("telemt_rpc_proxy_req_every_valid_ok_test.toml"); + std::fs::write(&path_valid, toml_valid).unwrap(); + let cfg_valid = ProxyConfig::load(&path_valid).unwrap(); + assert_eq!(cfg_valid.general.rpc_proxy_req_every, 40); + let _ = std::fs::remove_file(path_valid); +} + +#[test] +fn me_route_backpressure_base_timeout_ms_out_of_range_is_rejected() { + let toml = r#" + [general] + me_route_backpressure_base_timeout_ms = 5001 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_route_backpressure_base_timeout_ms_out_of_range_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_route_backpressure_base_timeout_ms must be within [1, 5000]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_route_backpressure_high_timeout_ms_out_of_range_is_rejected() { + let toml = r#" + [general] + me_route_backpressure_base_timeout_ms = 100 + me_route_backpressure_high_timeout_ms = 5001 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_route_backpressure_high_timeout_ms_out_of_range_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_route_backpressure_high_timeout_ms must be within [1, 5000]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_route_no_writer_wait_ms_out_of_range_is_rejected() { + let toml = r#" + [general] + me_route_no_writer_wait_ms = 5 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_route_no_writer_wait_ms_out_of_range_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_route_no_writer_wait_ms must be within [10, 5000]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_route_blocking_send_timeout_ms_zero_is_rejected() { + let toml = r#" + [general] + me_route_blocking_send_timeout_ms = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_route_blocking_send_timeout_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_route_blocking_send_timeout_ms must be within [1, 5000]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_route_no_writer_mode_is_parsed() { + let toml = r#" + [general] + me_route_no_writer_mode = "inline_recovery_legacy" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_route_no_writer_mode_parse_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!( + cfg.general.me_route_no_writer_mode, + crate::config::MeRouteNoWriterMode::InlineRecoveryLegacy + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn proxy_config_cache_paths_empty_are_rejected() { + let toml = r#" + [general] + proxy_config_v4_cache_path = " " + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_proxy_config_v4_cache_path_empty_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.proxy_config_v4_cache_path cannot be empty")); + let _ = std::fs::remove_file(path); + + let toml_v6 = r#" + [general] + proxy_config_v6_cache_path = "" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let path_v6 = dir.join("telemt_proxy_config_v6_cache_path_empty_test.toml"); + std::fs::write(&path_v6, toml_v6).unwrap(); + let err_v6 = ProxyConfig::load(&path_v6).unwrap_err().to_string(); + assert!(err_v6.contains("general.proxy_config_v6_cache_path cannot be empty")); + let _ = std::fs::remove_file(path_v6); +} + +#[test] +fn me_hardswap_warmup_defaults_are_set() { + let toml = r#" + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_hardswap_warmup_defaults_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!( + cfg.general.me_hardswap_warmup_delay_min_ms, + default_me_hardswap_warmup_delay_min_ms() + ); + assert_eq!( + cfg.general.me_hardswap_warmup_delay_max_ms, + default_me_hardswap_warmup_delay_max_ms() + ); + assert_eq!( + cfg.general.me_hardswap_warmup_extra_passes, + default_me_hardswap_warmup_extra_passes() + ); + assert_eq!( + cfg.general.me_hardswap_warmup_pass_backoff_base_ms, + default_me_hardswap_warmup_pass_backoff_base_ms() + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_hardswap_warmup_delay_range_is_validated() { + let toml = r#" + [general] + me_hardswap_warmup_delay_min_ms = 2001 + me_hardswap_warmup_delay_max_ms = 2000 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_hardswap_warmup_delay_range_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains( + "general.me_hardswap_warmup_delay_min_ms must be <= general.me_hardswap_warmup_delay_max_ms" + )); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_hardswap_warmup_delay_max_zero_is_rejected() { + let toml = r#" + [general] + me_hardswap_warmup_delay_max_ms = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_hardswap_warmup_delay_max_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_hardswap_warmup_delay_max_ms must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_hardswap_warmup_extra_passes_out_of_range_is_rejected() { + let toml = r#" + [general] + me_hardswap_warmup_extra_passes = 11 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_hardswap_warmup_extra_passes_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_hardswap_warmup_extra_passes must be within [0, 10]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_hardswap_warmup_pass_backoff_zero_is_rejected() { + let toml = r#" + [general] + me_hardswap_warmup_pass_backoff_base_ms = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_hardswap_warmup_backoff_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_hardswap_warmup_pass_backoff_base_ms must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_config_stable_snapshots_zero_is_rejected() { + let toml = r#" + [general] + me_config_stable_snapshots = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_config_stable_snapshots_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_config_stable_snapshots must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn proxy_secret_stable_snapshots_zero_is_rejected() { + let toml = r#" + [general] + proxy_secret_stable_snapshots = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_proxy_secret_stable_snapshots_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.proxy_secret_stable_snapshots must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn proxy_secret_len_max_out_of_range_is_rejected() { + let toml = r#" + [general] + proxy_secret_len_max = 16 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_proxy_secret_len_max_out_of_range_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.proxy_secret_len_max must be within [32, 4096]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn me_pool_min_fresh_ratio_out_of_range_is_rejected() { + let toml = r#" + [general] + me_pool_min_fresh_ratio = 1.5 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_me_pool_min_ratio_invalid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("general.me_pool_min_fresh_ratio must be within [0.0, 1.0]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn api_minimal_runtime_cache_ttl_out_of_range_is_rejected() { + let toml = r#" + [server.api] + enabled = true + listen = "127.0.0.1:9091" + minimal_runtime_cache_ttl_ms = 70000 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_api_minimal_runtime_cache_ttl_invalid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("server.api.minimal_runtime_cache_ttl_ms must be within [0, 60000]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn client_mss_presets_and_listener_override_are_resolved() { + let toml = r#" + [server] + client_mss = "tspu" + + [[server.listeners]] + ip = "127.0.0.1" + port = 1443 + + [[server.listeners]] + ip = "127.0.0.2" + port = 1444 + client_mss = "2in8" + + [[server.listeners]] + ip = "127.0.0.3" + port = 1445 + client_mss = "" + + [[server.listeners]] + ip = "127.0.0.4" + port = 1446 + client_mss = "extreme-low" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_client_mss_valid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + + assert_eq!(cfg.server.client_mss_value(), Ok(Some(92))); + assert_eq!( + cfg.server.listeners[0].effective_client_mss(&cfg.server), + Ok(Some(92)) + ); + assert_eq!( + cfg.server.listeners[1].effective_client_mss(&cfg.server), + Ok(Some(256)) + ); + assert_eq!( + cfg.server.listeners[2].effective_client_mss(&cfg.server), + Ok(None) + ); + assert_eq!( + cfg.server.listeners[3].effective_client_mss(&cfg.server), + Ok(Some(88)) + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn client_mss_custom_value_is_accepted() { + let toml = r#" + [server] + client_mss = "4096" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_client_mss_custom_valid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + + assert_eq!(cfg.server.client_mss_value(), Ok(Some(4096))); + let _ = std::fs::remove_file(path); +} + +#[test] +fn client_mss_out_of_range_is_rejected() { + for value in ["87", "4097"] { + let toml = format!( + r#" + [server] + client_mss = "{value}" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "# + ); + let dir = std::env::temp_dir(); + let path = dir.join(format!("telemt_client_mss_out_of_range_{value}_test.toml")); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + + assert!(err.contains("server.client_mss custom value must be within [88, 4096]")); + let _ = std::fs::remove_file(path); + } +} + +#[test] +fn client_mss_unquoted_number_is_rejected() { + let toml = r#" + [server] + client_mss = 256 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_client_mss_unquoted_number_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + + assert!(err.contains("client_mss")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn listener_client_mss_invalid_preset_is_rejected() { + let toml = r#" + [[server.listeners]] + ip = "127.0.0.1" + port = 1443 + client_mss = "tiny" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_listener_client_mss_invalid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + + assert!(err.contains("server.listeners[0].client_mss")); + assert!(err.contains("must be \"\", extreme-low, tspu, 2in8")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn api_runtime_edge_cache_ttl_out_of_range_is_rejected() { + let toml = r#" + [server.api] + enabled = true + listen = "127.0.0.1:9091" + runtime_edge_cache_ttl_ms = 70000 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_api_runtime_edge_cache_ttl_invalid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("server.api.runtime_edge_cache_ttl_ms must be within [0, 60000]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn api_runtime_edge_top_n_out_of_range_is_rejected() { + let toml = r#" + [server.api] + enabled = true + listen = "127.0.0.1:9091" + runtime_edge_top_n = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_api_runtime_edge_top_n_invalid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("server.api.runtime_edge_top_n must be within [1, 1000]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn api_runtime_edge_events_capacity_out_of_range_is_rejected() { + let toml = r#" + [server.api] + enabled = true + listen = "127.0.0.1:9091" + runtime_edge_events_capacity = 8 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_api_runtime_edge_events_capacity_invalid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("server.api.runtime_edge_events_capacity must be within [16, 4096]")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn conntrack_pressure_high_watermark_out_of_range_is_rejected() { + let toml = r#" + [server.conntrack_control] + pressure_high_watermark_pct = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_conntrack_high_watermark_invalid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!( + err.contains( + "server.conntrack_control.pressure_high_watermark_pct must be within [1, 100]" + ) + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn conntrack_pressure_low_watermark_must_be_below_high() { + let toml = r#" + [server.conntrack_control] + pressure_high_watermark_pct = 50 + pressure_low_watermark_pct = 50 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_conntrack_low_watermark_invalid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains( + "server.conntrack_control.pressure_low_watermark_pct must be < pressure_high_watermark_pct" + )); + let _ = std::fs::remove_file(path); +} + +#[test] +fn conntrack_delete_budget_zero_is_rejected() { + let toml = r#" + [server.conntrack_control] + delete_budget_per_sec = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_conntrack_delete_budget_invalid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("server.conntrack_control.delete_budget_per_sec must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn conntrack_hybrid_mode_requires_listener_allow_list() { + let toml = r#" + [server.conntrack_control] + mode = "hybrid" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_conntrack_hybrid_requires_ips_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!( + err.contains( + "server.conntrack_control.hybrid_listener_ips must be non-empty in mode=hybrid" + ) + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn conntrack_profile_is_loaded_from_config() { + let toml = r#" + [server.conntrack_control] + profile = "aggressive" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_conntrack_profile_parse_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!( + cfg.server.conntrack_control.profile, + ConntrackPressureProfile::Aggressive + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn force_close_default_matches_drain_ttl() { + let toml = r#" + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_force_close_default_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!(cfg.general.me_reinit_drain_timeout_secs, 90); + assert_eq!(cfg.general.effective_me_pool_force_close_secs(), 90); + let _ = std::fs::remove_file(path); +} + +#[test] +fn force_close_zero_uses_runtime_safety_fallback() { + let toml = r#" + [general] + me_reinit_drain_timeout_secs = 0 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_force_close_zero_fallback_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!(cfg.general.me_reinit_drain_timeout_secs, 0); + assert_eq!(cfg.general.effective_me_pool_force_close_secs(), 300); + let _ = std::fs::remove_file(path); +} + +#[test] +fn force_close_bumped_when_below_drain_ttl() { + let toml = r#" + [general] + me_pool_drain_ttl_secs = 90 + me_reinit_drain_timeout_secs = 30 + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_force_close_bump_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!(cfg.general.me_reinit_drain_timeout_secs, 90); + let _ = std::fs::remove_file(path); +} + +#[test] +fn tls_fetch_scope_default_is_empty() { + let toml = r#" + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_tls_fetch_scope_default_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert!(cfg.censorship.tls_fetch_scope.is_empty()); + let _ = std::fs::remove_file(path); +} + +#[test] +fn tls_fetch_scope_is_trimmed_during_load() { + let toml = r#" + [censorship] + tls_domain = "example.com" + tls_fetch_scope = " me " + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_tls_fetch_scope_trim_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!(cfg.censorship.tls_fetch_scope, "me"); + let _ = std::fs::remove_file(path); +} + +#[test] +fn tls_fetch_scope_whitespace_becomes_empty() { + let toml = r#" + [censorship] + tls_domain = "example.com" + tls_fetch_scope = " " + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_tls_fetch_scope_blank_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert!(cfg.censorship.tls_fetch_scope.is_empty()); + let _ = std::fs::remove_file(path); +} + +#[test] +fn tls_fetch_defaults_are_applied() { + let toml = r#" + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_tls_fetch_defaults_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!( + cfg.censorship.tls_fetch.profiles, + TlsFetchConfig::default().profiles + ); + assert!(cfg.censorship.tls_fetch.strict_route); + assert_eq!(cfg.censorship.tls_fetch.attempt_timeout_ms, 5_000); + assert_eq!(cfg.censorship.tls_fetch.total_budget_ms, 15_000); + assert_eq!(cfg.censorship.tls_fetch.profile_cache_ttl_secs, 600); + let _ = std::fs::remove_file(path); +} + +#[test] +fn tls_fetch_profiles_are_deduplicated_preserving_order() { + let toml = r#" + [censorship] + tls_domain = "example.com" + [censorship.tls_fetch] + profiles = ["compat_tls12", "modern_chrome_like", "compat_tls12", "legacy_minimal"] + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_tls_fetch_profiles_dedup_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!( + cfg.censorship.tls_fetch.profiles, + vec![ + TlsFetchProfile::CompatTls12, + TlsFetchProfile::ModernChromeLike, + TlsFetchProfile::LegacyMinimal + ] + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn tls_fetch_attempt_timeout_zero_is_rejected() { + let toml = r#" + [censorship] + tls_domain = "example.com" + [censorship.tls_fetch] + attempt_timeout_ms = 0 + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_tls_fetch_attempt_timeout_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("censorship.tls_fetch.attempt_timeout_ms must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn tls_fetch_total_budget_zero_is_rejected() { + let toml = r#" + [censorship] + tls_domain = "example.com" + [censorship.tls_fetch] + total_budget_ms = 0 + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_tls_fetch_total_budget_zero_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("censorship.tls_fetch.total_budget_ms must be > 0")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn invalid_ad_tag_is_disabled_during_load() { + let toml = r#" + [general] + ad_tag = "not_hex" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_invalid_ad_tag_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert!(cfg.general.ad_tag.is_none()); + let _ = std::fs::remove_file(path); +} + +#[test] +fn valid_ad_tag_is_preserved_during_load() { + let toml = r#" + [general] + ad_tag = "00112233445566778899aabbccddeeff" + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_valid_ad_tag_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!( + cfg.general.ad_tag.as_deref(), + Some("00112233445566778899aabbccddeeff") + ); + let _ = std::fs::remove_file(path); +} + +#[test] +fn shadowsocks_upstream_url_loads_successfully() { + let toml = format!( + r#" + [general] + use_middle_proxy = false + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + + [[upstreams]] + type = "shadowsocks" + url = "{url}" + interface = "127.0.0.2" + "#, + url = TEST_SHADOWSOCKS_URL, + ); + let dir = std::env::temp_dir(); + let path = dir.join("telemt_shadowsocks_valid_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + + assert!(matches!( + &cfg.upstreams[0].upstream_type, + UpstreamType::Shadowsocks { url, interface } + if url == TEST_SHADOWSOCKS_URL && interface.as_deref() == Some("127.0.0.2") + )); + + let _ = std::fs::remove_file(path); +} + +#[test] +fn shadowsocks_requires_direct_mode() { + let toml = format!( + r#" + [general] + use_middle_proxy = true + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + + [[upstreams]] + type = "shadowsocks" + url = "{url}" + "#, + url = TEST_SHADOWSOCKS_URL, + ); + let dir = std::env::temp_dir(); + let path = dir.join("telemt_shadowsocks_me_reject_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + + assert!(err.contains("shadowsocks upstreams require general.use_middle_proxy = false")); + + let _ = std::fs::remove_file(path); +} + +#[test] +fn invalid_shadowsocks_url_is_rejected() { + let toml = r#" + [general] + use_middle_proxy = false + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + + [[upstreams]] + type = "shadowsocks" + url = "not-a-valid-ss-url" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_shadowsocks_invalid_url_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + + assert!(err.contains("invalid shadowsocks url")); + + let _ = std::fs::remove_file(path); +} + +#[test] +fn shadowsocks_plugins_are_rejected() { + let toml = format!( + r#" + [general] + use_middle_proxy = false + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + + [[upstreams]] + type = "shadowsocks" + url = "{url}?plugin=obfs-local%3Bobfs%3Dhttp" + "#, + url = TEST_SHADOWSOCKS_URL, + ); + let dir = std::env::temp_dir(); + let path = dir.join("telemt_shadowsocks_plugin_reject_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + + assert!(err.contains("shadowsocks plugins are not supported")); + + let _ = std::fs::remove_file(path); +} + +#[test] +fn invalid_user_ad_tag_reports_access_user_ad_tags_key() { + let toml = r#" + [censorship] + tls_domain = "example.com" + + [access.users] + alice = "00000000000000000000000000000000" + + [access.user_ad_tags] + alice = "not_hex" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_invalid_user_ad_tag_message_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + let err = cfg.validate().unwrap_err().to_string(); + assert!(err.contains("access.user_ad_tags['alice'] must be exactly 32 hex characters")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn invalid_dns_override_is_rejected() { + let toml = r#" + [network] + dns_overrides = ["example.com:443:2001:db8::10"] + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_invalid_dns_override_test.toml"); + std::fs::write(&path, toml).unwrap(); + let err = ProxyConfig::load(&path).unwrap_err().to_string(); + assert!(err.contains("must be bracketed")); + let _ = std::fs::remove_file(path); +} + +#[test] +fn valid_dns_override_is_accepted() { + let toml = r#" + [network] + dns_overrides = ["example.com:443:127.0.0.1", "example.net:443:[2001:db8::10]"] + + [censorship] + tls_domain = "example.com" + + [access.users] + user = "00000000000000000000000000000000" + "#; + let dir = std::env::temp_dir(); + let path = dir.join("telemt_valid_dns_override_test.toml"); + std::fs::write(&path, toml).unwrap(); + let cfg = ProxyConfig::load(&path).unwrap(); + assert_eq!(cfg.network.dns_overrides.len(), 2); + let _ = std::fs::remove_file(path); +}