dnomd343
/
ProxyC
mirror of https://github.com/dnomd343/ProxyC
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

refactor: new ProxyBuilder

master
Dnomd343 3 years ago
parent
3641c7a05a
commit
3f1f3223d7
5 changed files with 429 additions and 265 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 3
      .gitignore
  2. 471
      ProxyBuilder/Shadowsocks.py
  3. 114
      ProxyBuilder/ShadowsocksR.py
  4. 2
      ProxyBuilder/__init__.py
  5. 102
      ProxyBuilder/builder.py

3
.gitignore
View File

@ -1 +1,2 @@
**/__pycache__ **/__pycache__/
/.idea/

471
ProxyBuilder/Shadowsocks.py
View File

@ -1,185 +1,226 @@
#!/usr/bin/python #!/usr/bin/python
# -*- coding:utf-8 -*- # -*- coding:utf-8 -*-
"""
# Shadowsocks构建器
## 节点格式
```json
{
"server": "...",
"port": ...,
"method": "...",
"passwd": "...",
"plugin": pluginObject
}
```
+ server (str) -> 必选, 服务器地址 (IPv4 / IPv6 / Domain)
+ port (int) -> 必选, 服务器端口 (1 ~ 65535)
+ method (str) -> 必选, Shadowsocks加密方式 (`aes-128-gcm`,`aes-192-gcm`,`aes-256-gcm`,
`aes-128-ctr`,`aes-192-ctr`,`aes-256-ctr`,
`aes-128-ocb`,`aes-192-ocb`,`aes-256-ocb`,
`aes-128-ofb`,`aes-192-ofb`,`aes-256-ofb`,
`aes-128-cfb`,`aes-192-cfb`,`aes-256-cfb`,
`aes-128-cfb1`,`aes-192-cfb1`,`aes-256-cfb1`,
`aes-128-cfb8`,`aes-192-cfb8`,`aes-256-cfb8`,
`aes-128-cfb128`,`aes-192-cfb128`,`aes-256-cfb128`,
`camellia-128-cfb`,`camellia-192-cfb`,`camellia-256-cfb`,
`camellia-128-cfb128`,`camellia-192-cfb128`,`camellia-256-cfb128`,
`plain`,`none`,`table`,`rc4`,`rc4-md5`,`rc2-cfb`,`bf-cfb`,
`cast5-cfb`,`des-cfb`,`idea-cfb`,`seed-cfb`,`salsa20`,`salsa20-ctr`,
`xchacha20`,`chacha20`,`chacha20-ietf`,`chacha20-poly1305`,
`chacha20-ietf-poly1305`,`xchacha20-ietf-poly1305`)
+ passwd (str) -> 必选, Shadowsocks连接密码
+ plugin (pluginObject) -> 必选, Shadowsocks连接插件
### pluginObject (None / dict)
无插件: None
带插件:
```json
{
"type": "...",
"param": "..."
}
```
+ type (str) -> 插件名称
+ param (str) -> 插件参数
"""
import json import json
ssMethodList = { # shadowsocks各版本支持的加密方式 ssMethodList = { # Shadowsocks各版本加密方式支持
"ss-python": [ 'ss-python': [
"aes-128-gcm", 'aes-128-gcm',
"aes-192-gcm", 'aes-192-gcm',
"aes-256-gcm", 'aes-256-gcm',
"aes-128-ctr", 'aes-128-ctr',
"aes-192-ctr", 'aes-192-ctr',
"aes-256-ctr", 'aes-256-ctr',
"aes-128-ocb", 'aes-128-ocb',
"aes-192-ocb", 'aes-192-ocb',
"aes-256-ocb", 'aes-256-ocb',
"aes-128-ofb", 'aes-128-ofb',
"aes-192-ofb", 'aes-192-ofb',
"aes-256-ofb", 'aes-256-ofb',
"aes-128-cfb", 'aes-128-cfb',
"aes-192-cfb", 'aes-192-cfb',
"aes-256-cfb", 'aes-256-cfb',
"aes-128-cfb1", 'aes-128-cfb1',
"aes-192-cfb1", 'aes-192-cfb1',
"aes-256-cfb1", 'aes-256-cfb1',
"aes-128-cfb8", 'aes-128-cfb8',
"aes-192-cfb8", 'aes-192-cfb8',
"aes-256-cfb8", 'aes-256-cfb8',
"aes-128-cfb128", 'aes-128-cfb128',
"aes-192-cfb128", 'aes-192-cfb128',
"aes-256-cfb128", 'aes-256-cfb128',
"camellia-128-cfb", 'camellia-128-cfb',
"camellia-192-cfb", 'camellia-192-cfb',
"camellia-256-cfb", 'camellia-256-cfb',
"camellia-128-cfb128", 'camellia-128-cfb128',
"camellia-192-cfb128", 'camellia-192-cfb128',
"camellia-256-cfb128", 'camellia-256-cfb128',
"table", 'table',
"rc4", 'rc4',
"rc4-md5", 'rc4-md5',
"rc2-cfb", 'rc2-cfb',
"bf-cfb", 'bf-cfb',
"cast5-cfb", 'cast5-cfb',
"des-cfb", 'des-cfb',
"idea-cfb", 'idea-cfb',
"seed-cfb", 'seed-cfb',
"salsa20", 'salsa20',
"xchacha20", 'xchacha20',
"chacha20", 'chacha20',
"chacha20-ietf", 'chacha20-ietf',
"chacha20-poly1305", 'chacha20-poly1305',
"chacha20-ietf-poly1305", 'chacha20-ietf-poly1305',
"xchacha20-ietf-poly1305", 'xchacha20-ietf-poly1305',
], ],
"ss-python-legacy": [ 'ss-python-legacy': [
"aes-128-ctr", 'aes-128-ctr',
"aes-192-ctr", 'aes-192-ctr',
"aes-256-ctr", 'aes-256-ctr',
"aes-128-ofb", 'aes-128-ofb',
"aes-192-ofb", 'aes-192-ofb',
"aes-256-ofb", 'aes-256-ofb',
"aes-128-cfb", 'aes-128-cfb',
"aes-192-cfb", 'aes-192-cfb',
"aes-256-cfb", 'aes-256-cfb',
"aes-128-cfb1", 'aes-128-cfb1',
"aes-192-cfb1", 'aes-192-cfb1',
"aes-256-cfb1", 'aes-256-cfb1',
"aes-128-cfb8", 'aes-128-cfb8',
"aes-192-cfb8", 'aes-192-cfb8',
"aes-256-cfb8", 'aes-256-cfb8',
"camellia-128-cfb", 'camellia-128-cfb',
"camellia-192-cfb", 'camellia-192-cfb',
"camellia-256-cfb", 'camellia-256-cfb',
"table", 'table',
"rc4", 'rc4',
"rc4-md5", 'rc4-md5',
"rc2-cfb", 'rc2-cfb',
"bf-cfb", 'bf-cfb',
"cast5-cfb", 'cast5-cfb',
"des-cfb", 'des-cfb',
"idea-cfb", 'idea-cfb',
"seed-cfb", 'seed-cfb',
"salsa20", 'salsa20',
"salsa20-ctr", 'salsa20-ctr',
"chacha20", 'chacha20',
], ],
"ss-libev": [ 'ss-libev': [
"aes-128-gcm", 'aes-128-gcm',
"aes-192-gcm", 'aes-192-gcm',
"aes-256-gcm", 'aes-256-gcm',
"aes-128-ctr", 'aes-128-ctr',
"aes-192-ctr", 'aes-192-ctr',
"aes-256-ctr", 'aes-256-ctr',
"aes-128-cfb", 'aes-128-cfb',
"aes-192-cfb", 'aes-192-cfb',
"aes-256-cfb", 'aes-256-cfb',
"camellia-128-cfb", 'camellia-128-cfb',
"camellia-192-cfb", 'camellia-192-cfb',
"camellia-256-cfb", 'camellia-256-cfb',
"rc4", 'rc4',
"rc4-md5", 'rc4-md5',
"bf-cfb", 'bf-cfb',
"salsa20", 'salsa20',
"chacha20", 'chacha20',
"chacha20-ietf", 'chacha20-ietf',
"chacha20-ietf-poly1305", 'chacha20-ietf-poly1305',
"xchacha20-ietf-poly1305", 'xchacha20-ietf-poly1305',
], ],
"ss-libev-legacy": [ 'ss-libev-legacy': [
"aes-128-ctr", 'aes-128-ctr',
"aes-192-ctr", 'aes-192-ctr',
"aes-256-ctr", 'aes-256-ctr',
"aes-128-cfb", 'aes-128-cfb',
"aes-192-cfb", 'aes-192-cfb',
"aes-256-cfb", 'aes-256-cfb',
"camellia-128-cfb", 'camellia-128-cfb',
"camellia-192-cfb", 'camellia-192-cfb',
"camellia-256-cfb", 'camellia-256-cfb',
"table", 'table',
"rc4", 'rc4',
"rc4-md5", 'rc4-md5',
"rc2-cfb", 'rc2-cfb',
"bf-cfb", 'bf-cfb',
"cast5-cfb", 'cast5-cfb',
"des-cfb", 'des-cfb',
"idea-cfb", 'idea-cfb',
"seed-cfb", 'seed-cfb',
"salsa20", 'salsa20',
"chacha20", 'chacha20',
"chacha20-ietf", 'chacha20-ietf',
], ],
"ss-rust": [ 'ss-rust': [
"aes-128-gcm", 'aes-128-gcm',
"aes-256-gcm", 'aes-256-gcm',
"plain", 'plain',
"none", 'none',
"chacha20-ietf-poly1305", 'chacha20-ietf-poly1305',
] ]
} }
def __baseJSON(proxyInfo, socksPort): # 生成JSON基本结构 def __baseConfig(proxyInfo: dict, socksPort: int) -> dict: # 生成基本配置
jsonContent = { config = {
'server': proxyInfo['server'], 'server': proxyInfo['server'],
'server_port': int(proxyInfo['port']), 'server_port': proxyInfo['port'],
'local_address': '127.0.0.1', 'local_address': '127.0.0.1',
'local_port': int(socksPort), 'local_port': socksPort,
'password': proxyInfo['password'],
'method': proxyInfo['method'], 'method': proxyInfo['method'],
'password': proxyInfo['passwd'],
} }
if proxyInfo['plugin'] != '': if proxyInfo['plugin'] is not None: # 带插件
jsonContent['plugin'] = proxyInfo['plugin'] config['plugin'] = proxyInfo['plugin']['type']
jsonContent['plugin_opts'] = proxyInfo['pluginParam'] config['plugin_opts'] = proxyInfo['plugin']['param']
return jsonContent return config
def __pluginUdpCheck(plugin, pluginParam): # 插件是否使用UDP通讯 def __pluginWithUdp(plugin: str, pluginParam: str) -> bool: # 插件是否使用UDP通讯
if plugin == '': # 无插件 if plugin in ['obfs-local', 'simple-tls', 'ck-client', 'gq-client',
return False 'mtt-client', 'rabbit-plugin', 'gun-plugin']: # 不使用UDP通讯的插件
noUdpPlugin = [ # 不使用UDP通讯的插件
'obfs-local',
'simple-tls',
'ck-client',
'gq-client',
'mtt-client',
'rabbit-plugin',
'gun-plugin',
]
onlyUdpPlugin = [ # 仅使用UDP通讯的插件
'kcptun-client',
'qtun-client',
]
if plugin in noUdpPlugin:
return False return False
if plugin in onlyUdpPlugin: if plugin in ['v2ray-plugin', 'xray-plugin', 'gost-plugin']:
return True if 'mode=quic' not in pluginParam.split(';'): # 非quic模式不使用UDP通讯
if plugin == 'v2ray-plugin' or plugin == 'xray-plugin' or plugin == 'gost-plugin':
if not 'mode=quic' in pluginParam.split(';'):
return False return False
return True # 默认假定占用UDP return True # 默认假定占用UDP
def __ssPython(proxyInfo, socksPort, isLegacy = False): # ss-python配置文件生成 def __ssPython(proxyInfo: dict, socksPort: int, isLegacy: bool = False) -> tuple[dict, str]: # ss-python配置生成
jsonContent = __baseJSON(proxyInfo, socksPort) config = __baseConfig(proxyInfo, socksPort)
specialMethods = [ mbedtlsMethods = [
'aes-128-cfb128', 'aes-128-cfb128',
'aes-192-cfb128', 'aes-192-cfb128',
'aes-256-cfb128', 'aes-256-cfb128',
@ -187,46 +228,80 @@ def __ssPython(proxyInfo, socksPort, isLegacy = False): # ss-python配置文件
'camellia-192-cfb128', 'camellia-192-cfb128',
'camellia-256-cfb128', 'camellia-256-cfb128',
] ]
if isLegacy == False: # 仅新版本支持 if not isLegacy: # 新版本特性
if jsonContent['method'] in specialMethods: if config['method'] in mbedtlsMethods: # mbedtls库流加密
jsonContent['method'] = 'mbedtls:' + jsonContent['method'] config['method'] = 'mbedtls:' + config['method']
if jsonContent['method'] == 'idea-cfb' or jsonContent['method'] == 'seed-cfb': if config['method'] in ['idea-cfb', 'seed-cfb']: # 仅openssl旧版本支持
jsonContent['extra_opts'] = '--libopenssl=libcrypto.so.1.0.0' config['extra_opts'] = '--libopenssl=libcrypto.so.1.0.0'
if proxyInfo['udp'] != True: if not proxyInfo['udp']:
jsonContent['no_udp'] = True config['no_udp'] = True # 关闭UDP代理
if isLegacy == True: config['shadowsocks'] = 'ss-python-legacy-local' if isLegacy else 'ss-python-local'
jsonContent['shadowsocks'] = 'ss-python-legacy-local' return config, 'ss-bootstrap-local'
else:
jsonContent['shadowsocks'] = 'ss-python-local' def __ssLibev(proxyInfo: dict, socksPort: int, isLegacy: bool = False) -> tuple[dict, str]: # ss-libev配置生成
return jsonContent, 'ss-bootstrap-local' config = __baseConfig(proxyInfo, socksPort)
if proxyInfo['udp']:
def __ssLibev(proxyInfo, socksPort, isLegacy = False): # ss-libev配置文件生成 config['mode'] = 'tcp_and_udp'
jsonContent = __baseJSON(proxyInfo, socksPort) return config, 'ss-libev-legacy-local' if isLegacy else 'ss-libev-local'
if proxyInfo['udp'] == True:
jsonContent['mode'] = 'tcp_and_udp' def __ssRust(proxyInfo: dict, socksPort: int) -> tuple[dict, str]: # ss-rust配置生成
if isLegacy == True: config = __baseConfig(proxyInfo, socksPort)
return jsonContent, 'ss-libev-legacy-local' if proxyInfo['udp']:
config['mode'] = 'tcp_and_udp'
return config, 'ss-rust-local'
def __ssFormatCheck(proxyInfo: dict) -> bool: # Shadowsocks参数检查
if 'server' not in proxyInfo or not isinstance(proxyInfo['server'], str): # server -> str
return False
if 'port' not in proxyInfo or not isinstance(proxyInfo['port'], int): # port -> int
return False
if 'method' not in proxyInfo or not isinstance(proxyInfo['method'], str): # method -> str
return False
if 'passwd' not in proxyInfo or not isinstance(proxyInfo['passwd'], str): # passwd -> str
return False
if 'plugin' not in proxyInfo:
return False
plugin = proxyInfo['plugin']
if isinstance(plugin, dict): # plugin -> dict / None
if 'type' not in plugin or not isinstance(plugin['type'], str): # plugin.type -> str
return False
if 'param' not in plugin or not isinstance(plugin['param'], str): # plugin.param -> str
return False
elif plugin is not None:
return False
return True
def load(proxyInfo: dict, socksPort: int, configFile: str) -> tuple[list or None, str or None, dict or None]:
"""
Shadowsocks配置载入
proxyInfo: 节点信息
socksPort: 本地通讯端口
configFile: 配置文件路径
节点有误:
return None, None, None
载入成功:
return startCommand, fileContent, envVar
"""
if not __ssFormatCheck(proxyInfo): # 参数有误
return None, None, None
if proxyInfo['plugin'] is None: # 无插件时启用UDP
proxyInfo['udp'] = True
else: else:
return jsonContent, 'ss-libev-local' proxyInfo['udp'] = not __pluginWithUdp( # 获取插件UDP冲突状态
proxyInfo['plugin']['type'], proxyInfo['plugin']['param']
def __ssRust(proxyInfo, socksPort): # ss-rust配置文件生成 )
jsonContent = __baseJSON(proxyInfo, socksPort) if proxyInfo['method'] in ssMethodList['ss-libev']: # 按序匹配客户端
if proxyInfo['udp'] == True: config, ssFile = __ssLibev(proxyInfo, socksPort)
jsonContent['mode'] = 'tcp_and_udp'
return jsonContent, 'ss-rust-local'
def load(proxyInfo, socksPort, configFile): # Shadowsocks配置载入
proxyInfo['udp'] = not __pluginUdpCheck(proxyInfo['plugin'], proxyInfo['pluginParam'])
if proxyInfo['method'] in ssMethodList['ss-libev']:
jsonContent, ssFile = __ssLibev(proxyInfo, socksPort)
elif proxyInfo['method'] in ssMethodList['ss-libev-legacy']: elif proxyInfo['method'] in ssMethodList['ss-libev-legacy']:
jsonContent, ssFile = __ssLibev(proxyInfo, socksPort, isLegacy = True) config, ssFile = __ssLibev(proxyInfo, socksPort, isLegacy = True)
elif proxyInfo['method'] in ssMethodList['ss-python']: elif proxyInfo['method'] in ssMethodList['ss-python']:
jsonContent, ssFile = __ssPython(proxyInfo, socksPort) config, ssFile = __ssPython(proxyInfo, socksPort)
elif proxyInfo['method'] in ssMethodList['ss-python-legacy']: elif proxyInfo['method'] in ssMethodList['ss-python-legacy']:
jsonContent, ssFile = __ssPython(proxyInfo, socksPort, isLegacy = True) config, ssFile = __ssPython(proxyInfo, socksPort, isLegacy = True)
elif proxyInfo['method'] in ssMethodList['ss-rust']: elif proxyInfo['method'] in ssMethodList['ss-rust']:
jsonContent, ssFile = __ssRust(proxyInfo, socksPort) config, ssFile = __ssRust(proxyInfo, socksPort)
else: else:
return None, None # 匹配不到加密方式 return None, None, None # 无匹配加密方式
return [ ssFile, '-c', configFile ], json.dumps(jsonContent) return [ssFile, '-c', configFile], json.dumps(config), {}

114
ProxyBuilder/ShadowsocksR.py
View File

@ -1,9 +1,59 @@
#!/usr/bin/python #!/usr/bin/python
# -*- coding:utf-8 -*- # -*- coding:utf-8 -*-
"""
# ShadowsocksR构建器
## 节点格式
```json
{
"server": "...",
"port": ...,
"method": "...",
"passwd": "...",
"protocol": "...",
"protocolParam": "...",
"obfs": "...",
"obfsParam": "..."
}
```
+ server (str) -> 必选, 服务器地址 (IPv4 / IPv6 / Domain)
+ port (int) -> 必选, 服务器端口 (1 ~ 65535)
+ method (str) -> 必选, ShadowsocksR加密方式 (`aes-128-cfb`,`aes-192-cfb`,`aes-256-cfb`,
`aes-128-cfb1`,`aes-192-cfb1`,`aes-256-cfb1`,
`aes-128-cfb8`,`aes-192-cfb8`,`aes-256-cfb8`,
`aes-128-ctr`,`aes-192-ctr`,`aes-256-ctr`,
`aes-128-gcm`,`aes-192-gcm`,`aes-256-gcm`,
`aes-128-ofb`,`aes-192-ofb`,`aes-256-ofb`,
`camellia-128-cfb`,`camellia-192-cfb`,`camellia-256-cfb`,
`none`,`table`,`rc4`,`rc4-md5`,`rc4-md5-6`,`bf-cfb`,
`cast5-cfb`,`des-cfb`,`idea-cfb`,`seed-cfb`,`rc2-cfb`,
`salsa20`,`xsalsa20`,`chacha20`,`xchacha20`,`chacha20-ietf`)
+ passwd (str) -> 必选, ShadowsocksR连接密码
+ protocol (str) -> 必选, ShadowsocksR连接协议 (`origin`,`verify_sha1`,`verify_simple`,`verify_deflate`,
`auth_simple`,`auth_sha1`,`auth_sha1_v2`,`auth_sha1_v4`,
`auth_aes128`,`auth_aes128_md5`,`auth_aes128_sha1`,
`auth_chain_a`,`auth_chain_b`,`auth_chain_c`,
`auth_chain_d`,`auth_chain_e`,`auth_chain_f`)
+ protocolParam (str) -> 必选, ShadowsocksR协议参数
+ obfs (str) -> 必选, ShadowsocksR混淆方式 (`plain`,`http_post`,`http_simple`,`tls_simple`,
`tls1.2_ticket_auth`,`tls1.2_ticket_fastauth`,`random_head`)
+ obfsParam (str) -> 必选, ShadowsocksR混淆参数
"""
import json import json
ssrMethodList = [ # 加密方法 ssrMethodList = [ # ShadowsocksR加密方式
"aes-128-cfb", "aes-128-cfb",
"aes-192-cfb", "aes-192-cfb",
"aes-256-cfb", "aes-256-cfb",
@ -43,7 +93,7 @@ ssrMethodList = [ # 加密方法
"chacha20-ietf", "chacha20-ietf",
] ]
ssrProtocolList = [ # 协议 ssrProtocolList = [ # ShadowsocksR协议列表
"origin", "origin",
"verify_sha1", "verify_sha1",
"verify_simple", "verify_simple",
@ -63,7 +113,7 @@ ssrProtocolList = [ # 协议
"auth_chain_f", "auth_chain_f",
] ]
ssrObfsList = [ # 混淆方式 ssrObfsList = [ # ShadowsocksR混淆方式
"plain", "plain",
"http_post", "http_post",
"http_simple", "http_simple",
@ -73,22 +123,56 @@ ssrObfsList = [ # 混淆方式
"random_head", "random_head",
] ]
def load(proxyInfo, socksPort, configFile): # ShadowsocksR配置载入 def __ssrFormatCheck(proxyInfo: dict) -> bool: # ShadowsocksR参数检查
if not proxyInfo['method'] in ssrMethodList: if 'server' not in proxyInfo or not isinstance(proxyInfo['server'], str): # server -> str
return None, None # 匹配不到加密方法 return False
if not proxyInfo['protocol'] in ssrProtocolList: if 'port' not in proxyInfo or not isinstance(proxyInfo['port'], int): # port -> int
return None, None # 匹配不到协议 return False
if not proxyInfo['obfs'] in ssrObfsList: if 'method' not in proxyInfo or not isinstance(proxyInfo['method'], str): # method -> str
return None, None # 匹配不到混淆方式 return False
return [ 'ssr-local', '-c', configFile ], json.dumps({ if 'passwd' not in proxyInfo or not isinstance(proxyInfo['passwd'], str): # passwd -> str
return False
if 'protocol' not in proxyInfo or not isinstance(proxyInfo['protocol'], str): # protocol -> str
return False
if 'protocolParam' not in proxyInfo or not isinstance(proxyInfo['protocolParam'], str): # protocolParam -> str
return False
if 'obfs' not in proxyInfo or not isinstance(proxyInfo['obfs'], str): # obfs -> str
return False
if 'obfsParam' not in proxyInfo or not isinstance(proxyInfo['obfsParam'], str): # obfsParam -> str
return False
return True
def load(proxyInfo: dict, socksPort: int, configFile: str) -> tuple[list or None, str or None, dict or None]:
"""
ShadowsocksR配置载入
proxyInfo: 节点信息
socksPort: 本地通讯端口
configFile: 配置文件路径
节点有误:
return None, None, None
载入成功:
return startCommand, fileContent, envVar
"""
if not __ssrFormatCheck(proxyInfo): # 参数有误
return None, None, None
if not proxyInfo['method'] in ssrMethodList: # 匹配不到加密方法
return None, None, None
if not proxyInfo['protocol'] in ssrProtocolList: # 匹配不到协议
return None, None, None
if not proxyInfo['obfs'] in ssrObfsList: # 匹配不到混淆方式
return None, None, None
config = {
'server': proxyInfo['server'], 'server': proxyInfo['server'],
'server_port': int(proxyInfo['port']), 'server_port': proxyInfo['port'],
'local_address': '127.0.0.1', 'local_address': '127.0.0.1',
'local_port': int(socksPort), 'local_port': socksPort,
'password': proxyInfo['password'], 'password': proxyInfo['passwd'],
'method': proxyInfo['method'], 'method': proxyInfo['method'],
'protocol': proxyInfo['protocol'], 'protocol': proxyInfo['protocol'],
'protocol_param': proxyInfo['protocolParam'], 'protocol_param': proxyInfo['protocolParam'],
'obfs': proxyInfo['obfs'], 'obfs': proxyInfo['obfs'],
'obfs_param': proxyInfo['obfsParam'] 'obfs_param': proxyInfo['obfsParam']
}) }
return ['ssr-local', '-c', configFile], json.dumps(config), {}

2
ProxyBuilder/__init__.py
View File

@ -3,4 +3,4 @@
from ProxyBuilder.builder import * from ProxyBuilder.builder import *
__all__ = [ 'build', 'check', 'destroy' ] __all__ = ['build', 'check', 'destroy']

102
ProxyBuilder/builder.py
View File

@ -13,13 +13,17 @@ from ProxyBuilder import ShadowsocksR
libcPaths = [ libcPaths = [
'/usr/lib64/libc.so.6', # CentOS '/usr/lib64/libc.so.6', # CentOS
'/lib/libc.musl-x86_64.so.1', # Alpine
'/lib/i386-linux-gnu/libc.so.6', # Debian / Ubuntu '/lib/i386-linux-gnu/libc.so.6', # Debian / Ubuntu
'/lib/x86_64-linux-gnu/libc.so.6', '/lib/x86_64-linux-gnu/libc.so.6',
'/lib/aarch64-linux-gnu/libc.so.6', '/lib/aarch64-linux-gnu/libc.so.6',
'/lib/libc.musl-x86_64.so.1', # Alpine
] ]
def __checkPortAvailable(port): # 检测端口可用性 def __checkPortAvailable(port: int) -> bool: # 检测端口可用性
ipv4_tcp = None
ipv4_udp = None
ipv6_tcp = None
ipv6_udp = None
try: try:
ipv4_tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM) ipv4_tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
ipv4_udp = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) ipv4_udp = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
@ -50,8 +54,8 @@ def __checkPortAvailable(port): # 检测端口可用性
ipv6_udp.close() ipv6_udp.close()
except: pass except: pass
def __genTaskFlag(length = 16): # 生成任务代号 def __genTaskFlag(length: int = 16) -> str: # 生成任务代号
flag = "" flag = ''
for i in range(0, length): for i in range(0, length):
tmp = random.randint(0, 15) tmp = random.randint(0, 15)
if tmp >= 10: if tmp >= 10:
@ -60,8 +64,8 @@ def __genTaskFlag(length = 16): # 生成任务代号
flag += str(tmp) # 0 ~ 9 flag += str(tmp) # 0 ~ 9
return flag return flag
def __getAvailablePort(rangeStart, rangeEnd): # 获取一个空闲端口 def __getAvailablePort(rangeStart: int, rangeEnd: int) -> int or None: # 获取一个空闲端口
if rangeStart > rangeEnd: if rangeStart > rangeEnd or rangeStart < 0 or rangeEnd > 65535:
return None return None
while True: while True:
port = random.randint(rangeStart, rangeEnd) # 随机选取 port = random.randint(rangeStart, rangeEnd) # 随机选取
@ -69,8 +73,9 @@ def __getAvailablePort(rangeStart, rangeEnd): # 获取一个空闲端口
return port return port
time.sleep(0.1) # 100ms time.sleep(0.1) # 100ms
def build(proxyInfo, configDir, portRangeStart = 1024, portRangeEnd = 65535): def build(proxyInfo: dict, configDir: str,
''' portRangeStart: int = 1024, portRangeEnd: int = 65535) -> tuple[bool or None, str or dict]:
"""
创建代理节点客户端 创建代理节点客户端
程序内部错误: 程序内部错误:
@ -86,45 +91,54 @@ def build(proxyInfo, configDir, portRangeStart = 1024, portRangeEnd = 65535):
'file': configFile, 'file': configFile,
'process': process 'process': process
} }
''' """
taskFlag = __genTaskFlag() # 生成测试标志 taskFlag = __genTaskFlag() # 生成测试标志
socksPort = __getAvailablePort(portRangeStart, portRangeEnd) # 获取Socks5测试端口 socksPort = __getAvailablePort(portRangeStart, portRangeEnd) # 获取Socks5测试端口
if not 'type' in proxyInfo: # 未指定节点类型 if 'type' not in proxyInfo: # 未指定节点类型
return False, 'Proxy type not specified' return False, 'Proxy type not specified'
proxyType = proxyInfo['type'] # 节点类型 proxyType = proxyInfo['type'] # 节点类型
proxyInfo.pop('type') proxyInfo.pop('type')
configFile = configDir + '/' + taskFlag + '.json' # 配置文件路径
if proxyType == 'ss': # Shadowsocks节点 if proxyType == 'ss': # Shadowsocks节点
startCommand, fileContent = Shadowsocks.load(proxyInfo, socksPort, configFile) clientObj = Shadowsocks
elif proxyType == 'ssr': # ShadowsocksR节点 elif proxyType == 'ssr': # ShadowsocksR节点
startCommand, fileContent = ShadowsocksR.load(proxyInfo, socksPort, configFile) clientObj = ShadowsocksR
else: # 未知类型 else: # 未知类型
return False, 'Unknown proxy type' return False, 'Unknown proxy type'
if startCommand == None: # 格式出错
configFile = configDir + '/' + taskFlag + '.json' # 配置文件路径
startCommand, fileContent, envVar = clientObj.load(proxyInfo, socksPort, configFile) # 载入配置
if startCommand is None: # 格式出错
return False, 'Format error with ' + str(proxyType) return False, 'Format error with ' + str(proxyType)
try: try:
with open(configFile, 'w') as fileObject: with open(configFile, 'w') as fileObject: # 保存配置文件
fileObject.write(fileContent) # 保存配置文件 fileObject.write(fileContent)
except: # 配置文件写入失败 except: # 配置文件写入失败
return None, "Unable write to file " + str(configFile) return None, "Unable write to file " + str(configFile)
process = None
try: # 子进程形式启动 try: # 子进程形式启动
for libcPath in libcPaths: for libcPath in libcPaths:
if os.path.exists(libcPath): # 定位libc.so文件 if os.path.exists(libcPath): # 定位libc.so文件
break break
exitWithMe = lambda: ctypes.CDLL(libcPath).prctl(1, 15) # SIGTERM process = subprocess.Popen( # 启动子进程
process = subprocess.Popen(startCommand, startCommand,
env = envVar,
stdout = subprocess.DEVNULL, stdout = subprocess.DEVNULL,
stderr = subprocess.DEVNULL, stderr = subprocess.DEVNULL,
preexec_fn = exitWithMe) # 子进程跟随退出 preexec_fn = lambda: ctypes.CDLL(libcPath).prctl(1, 15) # 子进程跟随退出
)
except: except:
try: try:
process = subprocess.Popen(startCommand, process = subprocess.Popen( # prctl失败 回退正常启动
startCommand,
env = envVar,
stdout = subprocess.DEVNULL, stdout = subprocess.DEVNULL,
stderr = subprocess.DEVNULL) # prctl失败 回退正常启动 stderr = subprocess.DEVNULL
except: pass )
if not 'process' in vars(): # 启动失败 except:
pass
if process is None: # 启动失败
return None, 'Subprocess start failed by `' + ' '.join(startCommand) + '`' return None, 'Subprocess start failed by `' + ' '.join(startCommand) + '`'
return True, { # 返回连接参数 return True, { # 返回连接参数
@ -134,8 +148,8 @@ def build(proxyInfo, configDir, portRangeStart = 1024, portRangeEnd = 65535):
'process': process 'process': process
} }
def check(client): def check(client: dict) -> bool or None:
''' """
检查客户端是否正常运行 检查客户端是否正常运行
检测出错: return None 检测出错: return None
@ -143,44 +157,34 @@ def check(client):
工作异常: return False 工作异常: return False
工作正常: return True 工作正常: return True
''' """
if client == None:
return None
try: try:
if client['process'].poll() != None: return client['process'].poll() is None
return False # 死亡
else:
return True # 正常
except: except:
return None # 异常 return None # 异常
def destroy(client): def destroy(client: dict) -> bool:
''' """
结束客户端并清理 结束客户端并清理
销毁异常: return None 销毁异常: return False
客户端退出: return True 销毁成功: return True
''' """
if client == None:
return None
try: try:
process = client['process'] process = client['process']
if process.poll() == None: # 未死亡 if process.poll() is None: # 未死亡
while process.poll() is None: # 等待退出
process.terminate() # SIGTERM process.terminate() # SIGTERM
while process.poll() == None: # 等待退出 time.sleep(0.2)
time.sleep(1)
process.terminate()
except: except:
return None return False
try: try:
file = client['file'] file = client['file']
if os.path.exists(file) and os.path.isfile(file): if os.path.exists(file) and os.path.isfile(file):
os.remove(file) # 删除配置文件 os.remove(file) # 删除配置文件
else: return True # 销毁成功
return None
except: except:
return None pass
return False
return True # 销毁完成

Write Preview
Loading…
Cancel
Save