mirror of
https://github.com/vimagick/dockerfiles.git
synced 2024-11-24 08:52:31 +02:00
129 lines
3.4 KiB
Python
Executable File
129 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# sstest - shadowsocks tester
|
|
#
|
|
# # Long Form
|
|
# sstest ss://aes-256-cfb:secret@datageek.info:8388
|
|
# # Short Form
|
|
# sstest password@datageek.info
|
|
# # Hash Form
|
|
# sstest c3M6Ly9hZXMtMjU2LWNmYjpzZWNyZXRAZGF0YWdlZWsuaW5mbzo4Mzg4
|
|
# # File Form
|
|
# sstest /etc/sslocal.json
|
|
#
|
|
|
|
|
|
from sh import sslocal, curl
|
|
from urlparse import urlparse
|
|
import argparse
|
|
import base64
|
|
import json
|
|
import os.path
|
|
import socket
|
|
import sys
|
|
import time
|
|
|
|
|
|
def random_port():
|
|
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
s.bind(('localhost', 0))
|
|
addr, port = s.getsockname()
|
|
s.close()
|
|
return port
|
|
|
|
|
|
def parse_uri(uri):
|
|
|
|
obj = urlparse(uri)
|
|
assert obj.scheme == 'ss'
|
|
if '@' not in uri:
|
|
b64 = obj.netloc
|
|
b64 += "=" * ((4 - len(b64) % 4) % 4)
|
|
uri = base64.decodestring(b64)
|
|
obj = urlparse('ss://' + uri)
|
|
param = dict(
|
|
method = obj.username or 'aes-256-cfb',
|
|
password = obj.password,
|
|
server = obj.hostname,
|
|
server_port = obj.port or 8388,
|
|
)
|
|
if not param['password']:
|
|
param['method'], param['password'] = 'aes-256-cfb', param['method']
|
|
return param
|
|
|
|
|
|
def run_sslocal(args):
|
|
|
|
lport = random_port()
|
|
proc = sslocal(l=lport, m=args['method'], s=args['server'], p=args['server_port'], k=args['password'], _bg=True)
|
|
return proc, lport
|
|
|
|
|
|
def run_curl(lport, timeout=5):
|
|
|
|
try:
|
|
proc = curl('-sSI', '-m', timeout, '-x', 'socks5h://localhost:%d' % lport, 'https://www.youtube.com/')
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
|
|
def gen_config(args, bind, port, enc):
|
|
|
|
if enc:
|
|
uri = '{0[method]}:{0[password]}@{0[server]}:{0[server_port]}'.format(args)
|
|
return 'ss://' + base64.encodestring(uri).rstrip('=\n')
|
|
else:
|
|
return json.dumps({
|
|
'local_address': bind,
|
|
'local_port': port or random_port(),
|
|
'server': args['server'],
|
|
'server_port': args['server_port'],
|
|
'method': args['method'],
|
|
'password': args['password'],
|
|
}, indent=2)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-t', '--timeout', type=int, default=5)
|
|
parser.add_argument('-w', '--wait', type=int, default=1)
|
|
parser.add_argument('-e', '--encode', action='store_true', default=False)
|
|
parser.add_argument('-n', '--dryrun', action='store_true', default=False)
|
|
parser.add_argument('-q', '--quiet', action='store_true', default=False)
|
|
parser.add_argument('-b', '--bind', type=str, default='127.0.0.1')
|
|
parser.add_argument('-p', '--port', type=int, default=0)
|
|
parser.add_argument('uri')
|
|
args = parser.parse_args()
|
|
|
|
uri = args.uri
|
|
if os.path.exists(uri):
|
|
obj = json.load(open(uri))
|
|
uri = 'ss://{0[method]}:{0[password]}@{0[server]}:{0[server_port]}'.format(obj)
|
|
args.bind = obj.get('local_address', '127.0.0.1')
|
|
args.port = obj.get('local_port', 0)
|
|
if not uri.startswith('ss://'):
|
|
uri = 'ss://' + uri
|
|
|
|
param = parse_uri(uri)
|
|
|
|
if args.dryrun:
|
|
ok = True
|
|
else:
|
|
proc, lport = run_sslocal(param)
|
|
time.sleep(args.wait)
|
|
ok = run_curl(lport, args.timeout)
|
|
proc.process.kill()
|
|
|
|
if ok:
|
|
print gen_config(param, args.bind, args.port, args.encode)
|
|
|
|
if not args.quiet:
|
|
print >> sys.stderr, '[{}] is {}'.format(param['server'], 'good' if ok else 'bad')
|
|
|
|
exit(0 if ok else 1)
|
|
|