mirror of https://github.com/dnomd343/ProxyC
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.8 KiB
78 lines
2.8 KiB
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import json
|
|
from Basis.Test import Settings
|
|
from Builder import ShadowsocksR
|
|
from Basis.Logger import logging
|
|
from Basis.Process import Process
|
|
from Basis.Functions import genFlag, getAvailablePort
|
|
from Basis.Constant import ssrMethods, ssrProtocols, ssrObfuscations
|
|
|
|
|
|
def loadServer(configFile: str, proxyInfo: dict) -> Process: # load server process
|
|
ssrConfig = {
|
|
'server': proxyInfo['server'],
|
|
'server_port': proxyInfo['port'], # type -> int
|
|
'password': proxyInfo['passwd'],
|
|
'method': proxyInfo['method'],
|
|
'protocol': proxyInfo['protocol'],
|
|
'protocol_param': proxyInfo['protocolParam'],
|
|
'obfs': proxyInfo['obfs'],
|
|
'obfs_param': proxyInfo['obfsParam'],
|
|
}
|
|
serverFile = os.path.join(Settings['workDir'], configFile)
|
|
return Process(Settings['workDir'], cmd = ['ssr-server', '-vv', '-c', serverFile], file = {
|
|
'path': serverFile,
|
|
'content': json.dumps(ssrConfig)
|
|
}, isStart = False)
|
|
|
|
|
|
def loadClient(configFile: str, proxyInfo: dict, socksInfo: dict) -> Process: # load client process
|
|
clientFile = os.path.join(Settings['workDir'], configFile)
|
|
ssrCommand, ssrConfig, _ = ShadowsocksR.load(proxyInfo, socksInfo, clientFile)
|
|
return Process(Settings['workDir'], cmd = ssrCommand, file = {
|
|
'path': clientFile,
|
|
'content': ssrConfig
|
|
}, isStart = False)
|
|
|
|
|
|
def loadTest(method: str, protocol: str, obfs: str) -> dict:
|
|
proxyInfo = { # connection info
|
|
'server': Settings['serverBind'],
|
|
'port': getAvailablePort(),
|
|
'passwd': genFlag(length = 8), # random password
|
|
'method': method,
|
|
'protocol': protocol,
|
|
'protocolParam': '',
|
|
'obfs': obfs,
|
|
'obfsParam': '',
|
|
}
|
|
socksInfo = { # socks5 interface for test
|
|
'addr': Settings['clientBind'],
|
|
'port': getAvailablePort()
|
|
}
|
|
configName = 'ssr_%s_%s_%s' % (method, protocol, obfs) # prefix of config file name
|
|
testInfo = { # release test info
|
|
'caption': 'ShadowsocksR test: method = %s | protocol = %s | obfs = %s' % (method, protocol, obfs),
|
|
'client': loadClient(configName + '_client.json', proxyInfo, socksInfo),
|
|
'server': loadServer(configName + '_server.json', proxyInfo),
|
|
'socks': socksInfo, # exposed socks5 address
|
|
'interface': {
|
|
'addr': proxyInfo['server'],
|
|
'port': proxyInfo['port'],
|
|
}
|
|
}
|
|
logging.debug('New ShadowsocksR test -> %s' % testInfo)
|
|
return testInfo
|
|
|
|
|
|
def load():
|
|
for method in ssrMethods:
|
|
yield loadTest(method, 'origin', 'plain')
|
|
for protocol in ssrProtocols:
|
|
yield loadTest('aes-128-ctr', protocol, 'plain')
|
|
for obfs in ssrObfuscations:
|
|
yield loadTest('aes-128-ctr', 'origin', obfs)
|
|
logging.info('ShadowsocksR test yield complete')
|
|
|