68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
|
""" TEST_CONFIG
|
|||
|
|
{
|
|||
|
|
"name": "base_deny_test",
|
|||
|
|
"network": {
|
|||
|
|
"allowInternet": false,
|
|||
|
|
"allowLocalNetwork": false,
|
|||
|
|
"allowListen": [9990],
|
|||
|
|
"allowList": ["111.63.65.247:80"]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
"""
|
|||
|
|
import os, sys, json, socket, platform
|
|||
|
|
|
|||
|
|
def run_test():
|
|||
|
|
is_darwin = platform.system().lower() == "darwin"
|
|||
|
|
|
|||
|
|
results = {
|
|||
|
|
"listen_9990_ok": False,
|
|||
|
|
"listen_9991_denied": False,
|
|||
|
|
"whitelist_ip_port_ok": False,
|
|||
|
|
"whitelist_ip_wrong_port_denied": False,
|
|||
|
|
"wrong_ip_denied": True # Mac 默认设为 True，跳过检查
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 1. 监听对比测试
|
|||
|
|
# 9990 (Allowed)
|
|||
|
|
try:
|
|||
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
|||
|
|
s.bind(('0.0.0.0', 9990))
|
|||
|
|
results["listen_9990_ok"] = True
|
|||
|
|
except: pass
|
|||
|
|
|
|||
|
|
# 9991 (Denied)
|
|||
|
|
try:
|
|||
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
|||
|
|
s.bind(('0.0.0.0', 9991))
|
|||
|
|
except:
|
|||
|
|
results["listen_9991_denied"] = True
|
|||
|
|
|
|||
|
|
# 2. 网络访问对比测试
|
|||
|
|
# 111.63.65.247:80 (Allowed)
|
|||
|
|
try:
|
|||
|
|
with socket.create_connection(("111.63.65.247", 80), timeout=1):
|
|||
|
|
results["whitelist_ip_port_ok"] = True
|
|||
|
|
except: pass
|
|||
|
|
|
|||
|
|
# 111.63.65.247:443 (Wrong Port - Denied)
|
|||
|
|
try:
|
|||
|
|
with socket.create_connection(("111.63.65.247", 443), timeout=1):
|
|||
|
|
pass
|
|||
|
|
except:
|
|||
|
|
results["whitelist_ip_wrong_port_denied"] = True
|
|||
|
|
|
|||
|
|
# 3. 跨 IP 拦截测试 (仅 Linux)
|
|||
|
|
if not is_darwin:
|
|||
|
|
results["wrong_ip_denied"] = False
|
|||
|
|
try:
|
|||
|
|
# 61.135.164.50:80 不在白名单
|
|||
|
|
with socket.create_connection(("61.135.164.50", 80), timeout=1):
|
|||
|
|
pass
|
|||
|
|
except:
|
|||
|
|
results["wrong_ip_denied"] = True
|
|||
|
|
|
|||
|
|
test_success = all(results.values())
|
|||
|
|
print(json.dumps({"testSuccess": test_success, "details": results}, indent=2))
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
run_test()
|