mirror/Easytier
1
0
Fork 0
mirror of https://github.com/EasyTier/EasyTier.git synced 2026-09-03 01:25:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support configurable TCP STUN servers (#2314)

Browse Source 
tcp_stun_servers explicitly controls TCP STUN servers.
If tcp_stun_servers is not configured, TCP STUN falls back to configured stun_servers.
If neither is configured, TCP STUN uses the built-in default TCP STUN list.
Empty lists explicitly disable the corresponding STUN server list.
Empty CLI/env overrides now clear existing configured STUN servers instead of appending nothing.
This commit is contained in:
fanyang
2026-08-13 09:56:44 +08:00
committed by GitHub
parent 52bb26680a
commit 8c15941c44
4 changed files with 207 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
easytier-core/src/config/toml.rs
+39 -18
View File
@@ -12,8 +12,6 @@ use anyhow::Context;
use ariadne::{CharSet, Config as AriadneConfig, IndexType, Label, Report, ReportKind, Source};
use serde::{Deserialize, Serialize};
#[cfg(feature = "config-write")]
use crate::config::{DEFAULT_UDP_STUN_SERVERS, DEFAULT_UDP_V6_STUN_SERVERS, default_stun_servers};
use crate::proto::{
acl::Acl,
common::{CompressionAlgoPb, SecureModeConfig},
@@ -27,16 +25,6 @@ pub(crate) fn default_instance_name() -> String {
"default".to_owned()
}
#[cfg(feature = "config-write")]
fn default_udp_stun_servers() -> Vec<String> {
default_stun_servers(DEFAULT_UDP_STUN_SERVERS)
}
#[cfg(feature = "config-write")]
fn default_udp_v6_stun_servers() -> Vec<String> {
default_stun_servers(DEFAULT_UDP_V6_STUN_SERVERS)
}
pub fn gen_default_flags() -> Flags {
#[allow(deprecated)]
Flags {
@@ -266,6 +254,11 @@ pub trait ConfigLoader: Send + Sync {
fn get_stun_servers(&self) -> Option<Vec<String>>;
fn set_stun_servers(&self, servers: Option<Vec<String>>);
fn get_tcp_stun_servers(&self) -> Option<Vec<String>> {
None
}
fn set_tcp_stun_servers(&self, _servers: Option<Vec<String>>) {}
fn get_stun_servers_v6(&self) -> Option<Vec<String>>;
fn set_stun_servers_v6(&self, servers: Option<Vec<String>>);
@@ -489,6 +482,7 @@ struct Config {
tcp_whitelist: Option<Vec<String>>,
udp_whitelist: Option<Vec<String>>,
stun_servers: Option<Vec<String>>,
tcp_stun_servers: Option<Vec<String>>,
stun_servers_v6: Option<Vec<String>>,
credential_file: Option<PathBuf>,
@@ -981,6 +975,14 @@ impl ConfigLoader for TomlConfig {
self.config.lock().unwrap().stun_servers = servers;
}
fn get_tcp_stun_servers(&self) -> Option<Vec<String>> {
self.config.lock().unwrap().tcp_stun_servers.clone()
}
fn set_tcp_stun_servers(&self, servers: Option<Vec<String>>) {
self.config.lock().unwrap().tcp_stun_servers = servers;
}
fn get_stun_servers_v6(&self) -> Option<Vec<String>> {
self.config.lock().unwrap().stun_servers_v6.clone()
}
@@ -1028,12 +1030,6 @@ impl ConfigLoader for TomlConfig {
let mut config = self.config.lock().unwrap().clone();
Self::normalize_config_source(&mut config);
config.flags = Some(flags_diff_from_default(&self.get_flags()));
if config.stun_servers == Some(default_udp_stun_servers()) {
config.stun_servers = None;
}
if config.stun_servers_v6 == Some(default_udp_v6_stun_servers()) {
config.stun_servers_v6 = None;
}
toml::to_string_pretty(&config).unwrap()
}
#[cfg(not(feature = "config-write"))]
@@ -1298,6 +1294,7 @@ socket_mark = 66
let config = TomlConfigLoader::default();
let stun_servers = config.get_stun_servers();
assert!(stun_servers.is_none());
assert!(config.get_tcp_stun_servers().is_none());
// Test setting custom stun servers
let custom_servers = vec!["txt:stun.easytier.cn".to_string()];
@@ -1305,6 +1302,12 @@ socket_mark = 66
let retrieved_servers = config.get_stun_servers();
assert_eq!(retrieved_servers.unwrap(), custom_servers);
let custom_tcp_servers = vec!["tcp-stun.example.com:3478".to_string()];
config.set_tcp_stun_servers(Some(custom_tcp_servers.clone()));
let retrieved_tcp_servers = config.get_tcp_stun_servers();
assert_eq!(retrieved_tcp_servers.unwrap(), custom_tcp_servers);
}
#[test]
@@ -1315,15 +1318,33 @@ stun_servers = [
"stun.l.google.com:19302",
"stun1.l.google.com:19302",
"txt:stun.easytier.cn"
]
tcp_stun_servers = [
"tcp-stun.example.com:3478"
]"#;
let config = TomlConfigLoader::new_from_str(config_str).unwrap();
let stun_servers = config.get_stun_servers().unwrap();
let tcp_stun_servers = config.get_tcp_stun_servers().unwrap();
assert_eq!(stun_servers.len(), 3);
assert_eq!(stun_servers[0], "stun.l.google.com:19302");
assert_eq!(stun_servers[1], "stun1.l.google.com:19302");
assert_eq!(stun_servers[2], "txt:stun.easytier.cn");
assert_eq!(tcp_stun_servers, ["tcp-stun.example.com:3478"]);
}
#[test]
fn test_empty_tcp_stun_servers_toml_parsing() {
let config = TomlConfigLoader::new_from_str(
r#"
instance_name = "test"
tcp_stun_servers = []
"#,
)
.unwrap();
assert_eq!(config.get_tcp_stun_servers(), Some(Vec::new()));
}
#[cfg(feature = "config-write")]
easytier-core/src/instance/config.rs
+100 -3
View File
@@ -323,6 +323,7 @@ impl CoreInstanceConfig {
provider_supported: host.public_ipv6_provider_supported,
},
};
let stun_servers = config.get_stun_servers();
Ok(Self {
instance_name: config.get_inst_name(),
@@ -335,13 +336,17 @@ impl CoreInstanceConfig {
gateway: host.gateway_enabled,
},
stun: StunServerConfig {
udp_servers: config
.get_stun_servers()
udp_servers: stun_servers
.clone()
.unwrap_or_else(|| StunServerConfig::default().udp_servers),
tcp_servers: config
.get_tcp_stun_servers()
.or_else(|| stun_servers.clone())
.unwrap_or_else(|| StunServerConfig::default().tcp_servers),
udp_v6_servers: config
.get_stun_servers_v6()
.or_else(|| stun_servers.as_ref().map(|_| Vec::new()))
.unwrap_or_else(|| StunServerConfig::default().udp_v6_servers),
..StunServerConfig::default()
},
endpoint_discovery: ManualEndpointDiscoveryConfig {
user_agent: format!("easytier/{}", host.easytier_version),
@@ -377,6 +382,98 @@ impl CoreInstanceConfig {
mod tests {
use super::*;
#[test]
fn tcp_stun_servers_follow_toml_override_rules() {
let fallback = TomlConfig::new_from_str(
r#"
stun_servers = ["fallback.example.com:3478"]
"#,
)
.unwrap();
let normalized = CoreInstanceConfig::from_toml(&fallback).unwrap();
assert_eq!(
normalized.connectivity.stun.tcp_servers,
["fallback.example.com:3478"]
);
let overridden = TomlConfig::new_from_str(
r#"
stun_servers = ["fallback.example.com:3478"]
tcp_stun_servers = ["tcp.example.com:3478"]
"#,
)
.unwrap();
let normalized = CoreInstanceConfig::from_toml(&overridden).unwrap();
assert_eq!(
normalized.connectivity.stun.tcp_servers,
["tcp.example.com:3478"]
);
let disabled = TomlConfig::new_from_str(
r#"
stun_servers = ["fallback.example.com:3478"]
tcp_stun_servers = []
"#,
)
.unwrap();
let normalized = CoreInstanceConfig::from_toml(&disabled).unwrap();
assert!(normalized.connectivity.stun.tcp_servers.is_empty());
}
#[test]
fn custom_udp_stun_servers_disable_default_ipv6_servers() {
let config = TomlConfig::new_from_str(
r#"
stun_servers = ["custom.example.com:3478"]
"#,
)
.unwrap();
let normalized = CoreInstanceConfig::from_toml(&config).unwrap();
assert!(normalized.connectivity.stun.udp_v6_servers.is_empty());
let config = TomlConfig::new_from_str(
r#"
stun_servers = ["custom.example.com:3478"]
stun_servers_v6 = ["custom-v6.example.com:3478"]
"#,
)
.unwrap();
let normalized = CoreInstanceConfig::from_toml(&config).unwrap();
assert_eq!(
normalized.connectivity.stun.udp_v6_servers,
["custom-v6.example.com:3478"]
);
}
#[cfg(feature = "config-write")]
#[test]
fn explicit_stun_servers_survive_dump_reload() {
let assert_roundtrip = |config: TomlConfig| {
let before = CoreInstanceConfig::from_toml(&config)
.unwrap()
.connectivity
.stun;
let reloaded = TomlConfig::new_from_str(&config.dump()).unwrap();
let after = CoreInstanceConfig::from_toml(&reloaded)
.unwrap()
.connectivity
.stun;
assert_eq!(after, before);
};
let defaults = StunServerConfig::default();
let config = TomlConfig::default();
config.set_stun_servers(Some(defaults.udp_servers.clone()));
assert_roundtrip(config);
let config = TomlConfig::default();
config.set_stun_servers(Some(vec!["custom.example.com:3478".to_string()]));
config.set_tcp_stun_servers(Some(defaults.tcp_servers));
config.set_stun_servers_v6(Some(defaults.udp_v6_servers));
assert_roundtrip(config);
}
#[test]
fn shared_toml_normalizes_instance_identity_and_connectivity() {
let config = TomlConfig::new_from_str(