chuan
72 lines
3.1 KiB
Python
72 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
from pyxray.libs.nodes import parse_node_link
|
|
from pyxray.libs.xray_config import XrayConfigSettings, generate_tinytun_config, write_tinytun_config_file
|
|
|
|
|
|
def test_generate_tinytun_config_matches_v2raya_defaults() -> None:
|
|
settings = XrayConfigSettings()
|
|
node = parse_node_link(_ss_link("198.51.100.10"))
|
|
|
|
config = generate_tinytun_config(node, settings, geosite_file="/data/geosite.dat")
|
|
|
|
assert config["tun"] == {
|
|
"name": "tun0",
|
|
"ip": "198.18.0.1",
|
|
"netmask": "255.255.255.255",
|
|
"ipv6_mode": "auto",
|
|
"ipv6": "fd00::1",
|
|
"ipv6_prefix": 128,
|
|
"auto_route": True,
|
|
"mtu": 1500,
|
|
}
|
|
assert config["socks5"] == {"name": "proxy", "address": "127.0.0.1:52345"}
|
|
assert "198.51.100.10" in config["filtering"]["skip_ips"]
|
|
assert config["filtering"]["skip_networks"][:5] == ["127.0.0.0/8", "169.254.0.0/16", "::1/128", "fc00::/7", "fe80::/10"]
|
|
assert config["filtering"]["block_ports"] == [22, 23, 25, 110, 143]
|
|
assert config["filtering"]["allow_ports"] == [80, 443, 53]
|
|
assert config["route"] == {"auto_detect_interface": True}
|
|
|
|
|
|
def test_generate_tinytun_dns_groups_and_routing_from_dns_rules() -> None:
|
|
settings = XrayConfigSettings()
|
|
node = parse_node_link(_ss_link("ss.example.net"))
|
|
|
|
config = generate_tinytun_config(node, settings, geosite_file="/data/geosite.dat")
|
|
dns = config["dns"]
|
|
|
|
assert dns["groups"][0] == {"name": "direct", "servers": ["223.5.5.5:53"], "strategy": "concurrent", "upstream": "direct", "protocol": "udp"}
|
|
assert dns["groups"][1] == {"name": "proxy", "servers": ["8.8.8.8:53"], "strategy": "concurrent", "upstream": "proxy", "protocol": "udp"}
|
|
assert "match(geosite:private),direct" in dns["routing"]["rules"]
|
|
assert "match(geosite:cn),direct" in dns["routing"]["rules"]
|
|
assert dns["routing"]["fallback_group"] == "proxy"
|
|
assert dns["routing"]["geosite_file"] == "/data/geosite.dat"
|
|
assert dns["routing"]["enable_cache"] is True
|
|
assert dns["routing"]["cache_capacity"] == 4096
|
|
assert dns["hijack"] == {"enabled": False, "mark": 1, "table_id": 100, "capture_tcp": True}
|
|
|
|
|
|
def test_write_tinytun_config_file_only_for_tun_mode(tmp_path) -> None:
|
|
settings = XrayConfigSettings()
|
|
node = parse_node_link(_ss_link("198.51.100.10"))
|
|
|
|
assert write_tinytun_config_file(node, settings, tmp_path) is None
|
|
|
|
settings.transparent.mode = "proxy"
|
|
settings.transparent.type = "tun"
|
|
settings.transparent.tun_auto_route = False
|
|
settings.transparent.tun_bypass_interfaces = "172.17.0.0/16"
|
|
settings.transparent.tun_exclude_processes = "/usr/bin/xray, pyxray"
|
|
path = write_tinytun_config_file(node, settings, tmp_path, geosite_file="/data/geosite.dat")
|
|
content = path.read_text(encoding="utf-8")
|
|
|
|
assert path.name == "tinytun.yaml"
|
|
assert "auto_route: false" in content
|
|
assert "- 172.17.0.0/16" in content
|
|
assert "- xray" in content
|
|
assert "- pyxray" in content
|
|
|
|
|
|
def _ss_link(host: str) -> str:
|
|
return f"ss://Y2hhY2hhMjAtaWV0Zi1wb2x5MTMwNTpzZWNyZXQ@{host}:8388#ss-node"
|