i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-03 09:50:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(tunnel): allow explicit false IP family flags

Browse Source
This commit is contained in:
RWDai
2026-05-20 13:50:58 +08:00
parent edded777e7
commit f4c79c80ac
1 changed files with 32 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
apps/aether-tunnel/src/config.rs
View File

@@ -572,21 +572,11 @@ pub struct Config {
pub tunnel_connect_timeout_ms: u64, pub tunnel_connect_timeout_ms: u64,
/// Force direct WebSocket tunnel TCP connects, or Aether outbound proxy endpoint connects, to IPv4 addresses only. /// Force direct WebSocket tunnel TCP connects, or Aether outbound proxy endpoint connects, to IPv4 addresses only.
#[arg( #[arg(long, env = "AETHER_TUNNEL_IPV4_ONLY", default_value_t = false)]
long,
env = "AETHER_TUNNEL_IPV4_ONLY",
default_value_t = false,
conflicts_with = "tunnel_ipv6_only"
)]
pub tunnel_ipv4_only: bool, pub tunnel_ipv4_only: bool,
/// Force direct WebSocket tunnel TCP connects, or Aether outbound proxy endpoint connects, to IPv6 addresses only. /// Force direct WebSocket tunnel TCP connects, or Aether outbound proxy endpoint connects, to IPv6 addresses only.
#[arg( #[arg(long, env = "AETHER_TUNNEL_IPV6_ONLY", default_value_t = false)]
long,
env = "AETHER_TUNNEL_IPV6_ONLY",
default_value_t = false,
conflicts_with = "tunnel_ipv4_only"
)]
pub tunnel_ipv6_only: bool, pub tunnel_ipv6_only: bool,
/// WebSocket tunnel TCP keepalive in seconds (0 disables) /// WebSocket tunnel TCP keepalive in seconds (0 disables)
@@ -1643,8 +1633,8 @@ node_name = "tunnel-test"
} }
#[test] #[test]
fn cli_rejects_conflicting_tunnel_ip_family_flags() { fn cli_parses_conflicting_tunnel_ip_family_flags_before_validation() {
let error = Config::try_parse_from([ let config = Config::parse_from([
"aether-tunnel", "aether-tunnel",
"--aether-url", "--aether-url",
"https://example.com", "https://example.com",
@@ -1654,10 +1644,35 @@ node_name = "tunnel-test"
"tunnel-test", "tunnel-test",
"--tunnel-ipv4-only", "--tunnel-ipv4-only",
"--tunnel-ipv6-only", "--tunnel-ipv6-only",
]) ]);
.expect_err("conflicting tunnel IP-family flags should fail");
assert_eq!(error.kind(), clap::error::ErrorKind::ArgumentConflict); assert!(config.tunnel_ipv4_only);
assert!(config.tunnel_ipv6_only);
let error = config
.validate()
.expect_err("conflicting tunnel IP-family flags should fail validation");
assert!(error.to_string().contains("tunnel_ipv4_only"));
}
#[test]
fn cli_accepts_explicit_false_tunnel_ip_family_flags() {
let config = Config::parse_from([
"aether-tunnel",
"--aether-url",
"https://example.com",
"--management-token",
"ae_test",
"--node-name",
"tunnel-test",
"--tunnel-ipv4-only=false",
"--tunnel-ipv6-only=false",
]);
assert!(!config.tunnel_ipv4_only);
assert!(!config.tunnel_ipv6_only);
config
.validate()
.expect("explicit false family flags should be valid");
} }
#[test] #[test]