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.

37 lines
1.7 KiB

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from Utils.Common import v6AddBracket
def loadOrigin(proxyInfo: dict) -> list: # origin stream
return ['client'] + [
'--server', '%s:%i' % (v6AddBracket(proxyInfo['server']), proxyInfo['port']),
'--password', proxyInfo['passwd'],
] + (['--udpovertcp'] if proxyInfo['stream']['uot'] else []) # add uot option
def loadWebsocket(proxyInfo: dict) -> list: # websocket stream
isTls = proxyInfo['stream']['secure'] is not None # ws or wss
wsAddress = (('wss' if isTls else 'ws') + '://%s:%i%s') % ( # websocket address
v6AddBracket(proxyInfo['stream']['host']), proxyInfo['port'], proxyInfo['stream']['path']
)
brookCommand = [
'wssclient' if isTls else 'wsclient',
'--address', '%s:%i' % (v6AddBracket(proxyInfo['server']), proxyInfo['port']), # real address
'--password', proxyInfo['passwd'],
] + (['--withoutBrookProtocol'] if proxyInfo['stream']['raw'] else []) # raw transmission on ws or wss
if not isTls:
return brookCommand + ['--wsserver', wsAddress]
return brookCommand + ['--wssserver', wsAddress] + (
[] if proxyInfo['stream']['secure']['verify'] else ['--insecure'] # add tls options
)
def load(proxyInfo: dict, socksInfo: dict, configFile: str) -> tuple[list, str, dict]:
brookCommand = ['brook', '--debug', '--listen', ':'] + { # debug module listen on random port
'origin': loadOrigin,
'ws': loadWebsocket,
}[proxyInfo['stream']['type']](proxyInfo) # choose origin or websocket stream
brookCommand += ['--socks5', '%s:%i' % (v6AddBracket(socksInfo['addr']), socksInfo['port'])]
return brookCommand, 'Config file %s no need' % configFile, {} # command, fileContent, envVar