breakwa11
10 years ago
4 changed files with 17 additions and 300 deletions
@ -1,278 +0,0 @@ |
|||||
#!/usr/bin/python |
|
||||
# -*- coding: utf-8 -*- |
|
||||
|
|
||||
# Copyright (c) 2014 clowwindy |
|
||||
# |
|
||||
# Permission is hereby granted, free of charge, to any person obtaining a copy |
|
||||
# of this software and associated documentation files (the "Software"), to deal |
|
||||
# in the Software without restriction, including without limitation the rights |
|
||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
|
||||
# copies of the Software, and to permit persons to whom the Software is |
|
||||
# furnished to do so, subject to the following conditions: |
|
||||
# |
|
||||
# The above copyright notice and this permission notice shall be included in |
|
||||
# all copies or substantial portions of the Software. |
|
||||
# |
|
||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
|
||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
|
||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
|
||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
|
||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
|
||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
|
||||
# SOFTWARE. |
|
||||
|
|
||||
import os |
|
||||
import json |
|
||||
import sys |
|
||||
import getopt |
|
||||
import logging |
|
||||
|
|
||||
|
|
||||
VERBOSE_LEVEL = 5 |
|
||||
|
|
||||
|
|
||||
def check_python(): |
|
||||
info = sys.version_info |
|
||||
if not (info[0] == 2 and info[1] >= 6): |
|
||||
print 'Python 2.6 or 2.7 required' |
|
||||
sys.exit(1) |
|
||||
|
|
||||
|
|
||||
def print_shadowsocks(): |
|
||||
version = '' |
|
||||
try: |
|
||||
import pkg_resources |
|
||||
version = pkg_resources.get_distribution('shadowsocks').version |
|
||||
except Exception: |
|
||||
pass |
|
||||
print 'shadowsocks %s' % version |
|
||||
|
|
||||
|
|
||||
def find_config(): |
|
||||
config_path = 'config.json' |
|
||||
if os.path.exists(config_path): |
|
||||
return config_path |
|
||||
config_path = os.path.join(os.path.dirname(__file__), '../', 'config.json') |
|
||||
if os.path.exists(config_path): |
|
||||
return config_path |
|
||||
return None |
|
||||
|
|
||||
|
|
||||
def check_config(config): |
|
||||
if config.get('local_address', '') in ['0.0.0.0']: |
|
||||
logging.warn('warning: local set to listen 0.0.0.0, which is not safe') |
|
||||
if config.get('server', '') in ['127.0.0.1', 'localhost']: |
|
||||
logging.warn('warning: server set to listen %s:%s, are you sure?' % |
|
||||
(config['server'], config['server_port'])) |
|
||||
if (config.get('method', '') or '').lower() == '': |
|
||||
logging.warn('warning: table is not safe; please use a safer cipher, ' |
|
||||
'like AES-256-CFB') |
|
||||
if (config.get('method', '') or '').lower() == 'rc4': |
|
||||
logging.warn('warning: RC4 is not safe; please use a safer cipher, ' |
|
||||
'like AES-256-CFB') |
|
||||
if config.get('timeout', 300) < 100: |
|
||||
logging.warn('warning: your timeout %d seems too short' % |
|
||||
int(config.get('timeout'))) |
|
||||
if config.get('timeout', 300) > 600: |
|
||||
logging.warn('warning: your timeout %d seems too long' % |
|
||||
int(config.get('timeout'))) |
|
||||
if config.get('password') in ['mypassword', 'barfoo!']: |
|
||||
logging.error('DON\'T USE DEFAULT PASSWORD! Please change it in your ' |
|
||||
'config.json!') |
|
||||
exit(1) |
|
||||
|
|
||||
|
|
||||
def get_config(is_local): |
|
||||
logging.basicConfig(level=logging.INFO, |
|
||||
format='%(levelname)-s: %(message)s', filemode='a+') |
|
||||
if is_local: |
|
||||
shortopts = 'hs:b:p:k:l:m:c:t:vq' |
|
||||
longopts = ['fast-open'] |
|
||||
else: |
|
||||
shortopts = 'hs:p:k:m:c:t:vq' |
|
||||
longopts = ['fast-open', 'workers:'] |
|
||||
try: |
|
||||
config_path = find_config() |
|
||||
optlist, args = getopt.getopt(sys.argv[1:], shortopts, longopts) |
|
||||
for key, value in optlist: |
|
||||
if key == '-c': |
|
||||
config_path = value |
|
||||
|
|
||||
if config_path: |
|
||||
logging.info('loading config from %s' % config_path) |
|
||||
with open(config_path, 'rb') as f: |
|
||||
try: |
|
||||
config = json.load(f, object_hook=_decode_dict) |
|
||||
except ValueError as e: |
|
||||
logging.error('found an error in config.json: %s', |
|
||||
e.message) |
|
||||
sys.exit(1) |
|
||||
else: |
|
||||
config = {} |
|
||||
|
|
||||
optlist, args = getopt.getopt(sys.argv[1:], shortopts, longopts) |
|
||||
v_count = 0 |
|
||||
for key, value in optlist: |
|
||||
if key == '-p': |
|
||||
config['server_port'] = int(value) |
|
||||
elif key == '-k': |
|
||||
config['password'] = value |
|
||||
elif key == '-l': |
|
||||
config['local_port'] = int(value) |
|
||||
elif key == '-s': |
|
||||
config['server'] = value |
|
||||
elif key == '-m': |
|
||||
config['method'] = value |
|
||||
elif key == '-b': |
|
||||
config['local_address'] = value |
|
||||
elif key == '-v': |
|
||||
v_count += 1 |
|
||||
# '-vv' turns on more verbose mode |
|
||||
config['verbose'] = v_count |
|
||||
elif key == '-t': |
|
||||
config['timeout'] = int(value) |
|
||||
elif key == '--fast-open': |
|
||||
config['fast_open'] = True |
|
||||
elif key == '--workers': |
|
||||
config['workers'] = value |
|
||||
elif key == '-h': |
|
||||
if is_local: |
|
||||
print_local_help() |
|
||||
else: |
|
||||
print_server_help() |
|
||||
sys.exit(0) |
|
||||
elif key == '-q': |
|
||||
v_count -= 1 |
|
||||
config['verbose'] = v_count |
|
||||
except getopt.GetoptError as e: |
|
||||
print >>sys.stderr, e |
|
||||
print_help(is_local) |
|
||||
sys.exit(2) |
|
||||
|
|
||||
if not config: |
|
||||
logging.error('config not specified') |
|
||||
print_help(is_local) |
|
||||
sys.exit(2) |
|
||||
|
|
||||
config['password'] = config.get('password', None) |
|
||||
config['method'] = config.get('method', 'aes-256-cfb') |
|
||||
config['port_password'] = config.get('port_password', None) |
|
||||
config['timeout'] = int(config.get('timeout', 300)) |
|
||||
config['fast_open'] = config.get('fast_open', False) |
|
||||
config['workers'] = config.get('workers', 1) |
|
||||
config['verbose'] = config.get('verbose', False) |
|
||||
config['local_address'] = config.get('local_address', '127.0.0.1') |
|
||||
config['local_port'] = config.get('local_port', 1080) |
|
||||
if is_local: |
|
||||
if config.get('server', None) is None: |
|
||||
logging.error('server addr not specified') |
|
||||
print_local_help() |
|
||||
sys.exit(2) |
|
||||
else: |
|
||||
config['server'] = config.get('server', '0.0.0.0') |
|
||||
config['server_port'] = config.get('server_port', 8388) |
|
||||
|
|
||||
if not ('password' in config and config['password']): |
|
||||
logging.error('password not specified') |
|
||||
print_help(is_local) |
|
||||
sys.exit(2) |
|
||||
|
|
||||
logging.getLogger('').handlers = [] |
|
||||
logging.addLevelName(VERBOSE_LEVEL, 'VERBOSE') |
|
||||
if config['verbose'] >= 2: |
|
||||
level = VERBOSE_LEVEL |
|
||||
elif config['verbose'] == 1: |
|
||||
level = logging.DEBUG |
|
||||
elif config['verbose'] == -1: |
|
||||
level = logging.WARN |
|
||||
elif config['verbose'] <= -2: |
|
||||
level = logging.ERROR |
|
||||
else: |
|
||||
level = logging.INFO |
|
||||
logging.basicConfig(level=level, |
|
||||
format='%(asctime)s %(levelname)-8s %(message)s', |
|
||||
datefmt='%Y-%m-%d %H:%M:%S', filemode='a+') |
|
||||
|
|
||||
check_config(config) |
|
||||
|
|
||||
return config |
|
||||
|
|
||||
|
|
||||
def print_help(is_local): |
|
||||
if is_local: |
|
||||
print_local_help() |
|
||||
else: |
|
||||
print_server_help() |
|
||||
|
|
||||
|
|
||||
def print_local_help(): |
|
||||
print '''usage: sslocal [-h] -s SERVER_ADDR [-p SERVER_PORT] |
|
||||
[-b LOCAL_ADDR] [-l LOCAL_PORT] -k PASSWORD [-m METHOD] |
|
||||
[-t TIMEOUT] [-c CONFIG] [--fast-open] [-v] [-q] |
|
||||
|
|
||||
optional arguments: |
|
||||
-h, --help show this help message and exit |
|
||||
-s SERVER_ADDR server address |
|
||||
-p SERVER_PORT server port, default: 8388 |
|
||||
-b LOCAL_ADDR local binding address, default: 127.0.0.1 |
|
||||
-l LOCAL_PORT local port, default: 1080 |
|
||||
-k PASSWORD password |
|
||||
-m METHOD encryption method, default: aes-256-cfb |
|
||||
-t TIMEOUT timeout in seconds, default: 300 |
|
||||
-c CONFIG path to config file |
|
||||
--fast-open use TCP_FASTOPEN, requires Linux 3.7+ |
|
||||
-v, -vv verbose mode |
|
||||
-q, -qq quiet mode, only show warnings/errors |
|
||||
|
|
||||
Online help: <https://github.com/clowwindy/shadowsocks> |
|
||||
''' |
|
||||
|
|
||||
|
|
||||
def print_server_help(): |
|
||||
print '''usage: ssserver [-h] [-s SERVER_ADDR] [-p SERVER_PORT] -k PASSWORD |
|
||||
-m METHOD [-t TIMEOUT] [-c CONFIG] [--fast-open] |
|
||||
[--workers WORKERS] [-v] [-q] |
|
||||
|
|
||||
optional arguments: |
|
||||
-h, --help show this help message and exit |
|
||||
-s SERVER_ADDR server address, default: 0.0.0.0 |
|
||||
-p SERVER_PORT server port, default: 8388 |
|
||||
-k PASSWORD password |
|
||||
-m METHOD encryption method, default: aes-256-cfb |
|
||||
-t TIMEOUT timeout in seconds, default: 300 |
|
||||
-c CONFIG path to config file |
|
||||
--fast-open use TCP_FASTOPEN, requires Linux 3.7+ |
|
||||
--workers WORKERS number of workers, available on Unix/Linux |
|
||||
-v, -vv verbose mode |
|
||||
-q, -qq quiet mode, only show warnings/errors |
|
||||
|
|
||||
Online help: <https://github.com/clowwindy/shadowsocks> |
|
||||
''' |
|
||||
|
|
||||
|
|
||||
def _decode_list(data): |
|
||||
rv = [] |
|
||||
for item in data: |
|
||||
if isinstance(item, unicode): |
|
||||
item = item.encode('utf-8') |
|
||||
elif isinstance(item, list): |
|
||||
item = _decode_list(item) |
|
||||
elif isinstance(item, dict): |
|
||||
item = _decode_dict(item) |
|
||||
rv.append(item) |
|
||||
return rv |
|
||||
|
|
||||
|
|
||||
def _decode_dict(data): |
|
||||
rv = {} |
|
||||
for key, value in data.iteritems(): |
|
||||
if isinstance(key, unicode): |
|
||||
key = key.encode('utf-8') |
|
||||
if isinstance(value, unicode): |
|
||||
value = value.encode('utf-8') |
|
||||
elif isinstance(value, list): |
|
||||
value = _decode_list(value) |
|
||||
elif isinstance(value, dict): |
|
||||
value = _decode_dict(value) |
|
||||
rv[key] = value |
|
||||
return rv |
|
||||
Loading…
Reference in new issue