|
|
@ -12,180 +12,175 @@ import base64 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class MuJsonLoader(object): |
|
|
class MuJsonLoader(object): |
|
|
def __init__(self): |
|
|
def __init__(self): |
|
|
self.json = None |
|
|
self.json = None |
|
|
|
|
|
|
|
|
def load(self, path): |
|
|
def load(self, path): |
|
|
l = "[]" |
|
|
l = "[]" |
|
|
try: |
|
|
try: |
|
|
with open(path, 'rb+') as f: |
|
|
with open(path, 'rb+') as f: |
|
|
l = f.read().decode('utf8') |
|
|
l = f.read().decode('utf8') |
|
|
except: |
|
|
except: |
|
|
pass |
|
|
pass |
|
|
self.json = json.loads(l) |
|
|
self.json = json.loads(l) |
|
|
|
|
|
|
|
|
def save(self, path): |
|
|
def save(self, path): |
|
|
if self.json: |
|
|
if self.json: |
|
|
output = json.dumps(self.json, sort_keys=True, indent=4, separators=(',', ': ')) |
|
|
output = json.dumps(self.json, sort_keys=True, indent=4, separators=(',', ': ')) |
|
|
with open(path, 'wb') as f: |
|
|
with open(path, 'wb') as f: |
|
|
f.write(output.encode('utf8')) |
|
|
f.write(output.encode('utf8')) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class MuMgr(object): |
|
|
class MuMgr(object): |
|
|
def __init__(self): |
|
|
def __init__(self): |
|
|
self.config_path = get_config().MUDB_FILE |
|
|
self.config_path = get_config().MUDB_FILE |
|
|
try: |
|
|
try: |
|
|
self.server_addr = get_config().SERVER_PUB_ADDR |
|
|
self.server_addr = get_config().SERVER_PUB_ADDR |
|
|
except: |
|
|
except: |
|
|
self.server_addr = '127.0.0.1' |
|
|
self.server_addr = '127.0.0.1' |
|
|
self.data = MuJsonLoader() |
|
|
self.data = MuJsonLoader() |
|
|
|
|
|
|
|
|
if self.server_addr == '127.0.0.1': |
|
|
if self.server_addr == '127.0.0.1': |
|
|
self.server_addr = self.getipaddr() |
|
|
self.server_addr = self.getipaddr() |
|
|
|
|
|
|
|
|
def getipaddr(self, ifname='eth0'): |
|
|
def getipaddr(self, ifname='eth0'): |
|
|
import socket |
|
|
import socket |
|
|
import struct |
|
|
import struct |
|
|
ret = '127.0.0.1' |
|
|
ret = '127.0.0.1' |
|
|
try: |
|
|
try: |
|
|
ret = socket.gethostbyname(socket.getfqdn(socket.gethostname())) |
|
|
ret = socket.gethostbyname(socket.getfqdn(socket.gethostname())) |
|
|
except: |
|
|
except: |
|
|
pass |
|
|
pass |
|
|
if ret == '127.0.0.1': |
|
|
if ret == '127.0.0.1': |
|
|
try: |
|
|
try: |
|
|
import fcntl |
|
|
import fcntl |
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) |
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) |
|
|
ret = socket.inet_ntoa(fcntl.ioctl(s.fileno(), 0x8915, struct.pack('256s', ifname[:15]))[20:24]) |
|
|
ret = socket.inet_ntoa(fcntl.ioctl(s.fileno(), 0x8915, struct.pack('256s', ifname[:15]))[20:24]) |
|
|
except: |
|
|
except: |
|
|
pass |
|
|
pass |
|
|
return ret |
|
|
return ret |
|
|
|
|
|
|
|
|
def ssrlink(self, user, encode): |
|
|
def ssrlink(self, user, encode): |
|
|
protocol = user.get('protocol', '') |
|
|
protocol = user.get('protocol', '') |
|
|
obfs = user.get('obfs', '') |
|
|
obfs = user.get('obfs', '') |
|
|
protocol = protocol.replace("_compatible", "") |
|
|
protocol = protocol.replace("_compatible", "") |
|
|
obfs = obfs.replace("_compatible", "") |
|
|
obfs = obfs.replace("_compatible", "") |
|
|
link = "%s:%s:%s:%s:%s:%s" % (self.server_addr, user['port'], protocol, user['method'], obfs, |
|
|
link = "%s:%s:%s:%s:%s:%s" % (self.server_addr, user['port'], protocol, user['method'], obfs, common.to_str(base64.urlsafe_b64encode(common.to_bytes(user['passwd']))).replace("=", "")) |
|
|
common.to_str(base64.urlsafe_b64encode(common.to_bytes(user['passwd']))).replace( |
|
|
return "ssr://" + (encode and common.to_str(base64.urlsafe_b64encode(common.to_bytes(link))).replace("=", "") or link) |
|
|
"=", "")) |
|
|
|
|
|
return "ssr://" + ( |
|
|
|
|
|
encode and common.to_str(base64.urlsafe_b64encode(common.to_bytes(link))).replace("=", "") or link) |
|
|
|
|
|
|
|
|
|
|
|
def userinfo(self, user): |
|
|
def userinfo(self, user): |
|
|
ret = "" |
|
|
ret = "" |
|
|
for key in sorted(user): |
|
|
for key in sorted(user): |
|
|
if key in ['enable']: |
|
|
if key in ['enable']: |
|
|
continue |
|
|
continue |
|
|
ret += '\n' |
|
|
ret += '\n' |
|
|
if key in ['transfer_enable', 'u', 'd']: |
|
|
if key in ['transfer_enable', 'u', 'd']: |
|
|
val = user[key] |
|
|
val = user[key] |
|
|
if val / 1024 < 4: |
|
|
if val / 1024 < 4: |
|
|
ret += " %s : %s" % (key, val) |
|
|
ret += " %s : %s" % (key, val) |
|
|
elif val / 1024 ** 2 < 4: |
|
|
elif val / 1024 ** 2 < 4: |
|
|
val /= float(1024) |
|
|
val /= float(1024) |
|
|
ret += " %s : %s K Bytes" % (key, val) |
|
|
ret += " %s : %s K Bytes" % (key, val) |
|
|
elif val / 1024 ** 3 < 4: |
|
|
elif val / 1024 ** 3 < 4: |
|
|
val /= float(1024 ** 2) |
|
|
val /= float(1024 ** 2) |
|
|
ret += " %s : %s M Bytes" % (key, val) |
|
|
ret += " %s : %s M Bytes" % (key, val) |
|
|
else: |
|
|
else: |
|
|
val /= float(1024 ** 3) |
|
|
val /= float(1024 ** 3) |
|
|
ret += " %s : %s G Bytes" % (key, val) |
|
|
ret += " %s : %s G Bytes" % (key, val) |
|
|
else: |
|
|
else: |
|
|
ret += " %s : %s" % (key, user[key]) |
|
|
ret += " %s : %s" % (key, user[key]) |
|
|
ret += "\n " + self.ssrlink(user, False) |
|
|
ret += "\n " + self.ssrlink(user, False) |
|
|
ret += "\n " + self.ssrlink(user, True) |
|
|
ret += "\n " + self.ssrlink(user, True) |
|
|
return ret |
|
|
return ret |
|
|
|
|
|
|
|
|
def rand_pass(self): |
|
|
def rand_pass(self): |
|
|
return ''.join( |
|
|
return ''.join([random.choice('''ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789~-_=+(){}[]^&%$@''') for i in range(8)]) |
|
|
[random.choice('''ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789~-_=+(){}[]^&%$@''') for i |
|
|
|
|
|
in range(8)]) |
|
|
|
|
|
|
|
|
|
|
|
def add(self, user): |
|
|
def add(self, user): |
|
|
up = {'enable': 1, 'u': 0, 'd': 0, 'method': "aes-128-cfb", |
|
|
up = {'enable': 1, 'u': 0, 'd': 0, 'method': "aes-128-cfb", |
|
|
'protocol': "auth_sha1_v4", |
|
|
'protocol': "auth_sha1_v4", |
|
|
'obfs': "tls1.2_ticket_auth_compatible", |
|
|
'obfs': "tls1.2_ticket_auth_compatible", |
|
|
'transfer_enable': 1125899906842624} |
|
|
'transfer_enable': 1125899906842624} |
|
|
up['passwd'] = self.rand_pass() |
|
|
up['passwd'] = self.rand_pass() |
|
|
up.update(user) |
|
|
up.update(user) |
|
|
|
|
|
|
|
|
self.data.load(self.config_path) |
|
|
self.data.load(self.config_path) |
|
|
for row in self.data.json: |
|
|
for row in self.data.json: |
|
|
match = False |
|
|
match = False |
|
|
if 'user' in user and row['user'] == user['user']: |
|
|
if 'user' in user and row['user'] == user['user']: |
|
|
match = True |
|
|
match = True |
|
|
if 'port' in user and row['port'] == user['port']: |
|
|
if 'port' in user and row['port'] == user['port']: |
|
|
match = True |
|
|
match = True |
|
|
if match: |
|
|
if match: |
|
|
print("user [%s] port [%s] already exist" % (row['user'], row['port'])) |
|
|
print("user [%s] port [%s] already exist" % (row['user'], row['port'])) |
|
|
return |
|
|
return |
|
|
self.data.json.append(up) |
|
|
self.data.json.append(up) |
|
|
print("### add user info %s" % self.userinfo(up)) |
|
|
print("### add user info %s" % self.userinfo(up)) |
|
|
self.data.save(self.config_path) |
|
|
self.data.save(self.config_path) |
|
|
|
|
|
|
|
|
def edit(self, user): |
|
|
def edit(self, user): |
|
|
self.data.load(self.config_path) |
|
|
self.data.load(self.config_path) |
|
|
for row in self.data.json: |
|
|
for row in self.data.json: |
|
|
match = True |
|
|
match = True |
|
|
if 'user' in user and row['user'] != user['user']: |
|
|
if 'user' in user and row['user'] != user['user']: |
|
|
match = False |
|
|
match = False |
|
|
if 'port' in user and row['port'] != user['port']: |
|
|
if 'port' in user and row['port'] != user['port']: |
|
|
match = False |
|
|
match = False |
|
|
if match: |
|
|
if match: |
|
|
print("edit user [%s]" % (row['user'],)) |
|
|
print("edit user [%s]" % (row['user'],)) |
|
|
row.update(user) |
|
|
row.update(user) |
|
|
print("### new user info %s" % self.userinfo(row)) |
|
|
print("### new user info %s" % self.userinfo(row)) |
|
|
break |
|
|
break |
|
|
self.data.save(self.config_path) |
|
|
self.data.save(self.config_path) |
|
|
|
|
|
|
|
|
def delete(self, user): |
|
|
def delete(self, user): |
|
|
self.data.load(self.config_path) |
|
|
self.data.load(self.config_path) |
|
|
index = 0 |
|
|
index = 0 |
|
|
for row in self.data.json: |
|
|
for row in self.data.json: |
|
|
match = True |
|
|
match = True |
|
|
if 'user' in user and row['user'] != user['user']: |
|
|
if 'user' in user and row['user'] != user['user']: |
|
|
match = False |
|
|
match = False |
|
|
if 'port' in user and row['port'] != user['port']: |
|
|
if 'port' in user and row['port'] != user['port']: |
|
|
match = False |
|
|
match = False |
|
|
if match: |
|
|
if match: |
|
|
print("delete user [%s]" % row['user']) |
|
|
print("delete user [%s]" % row['user']) |
|
|
del self.data.json[index] |
|
|
del self.data.json[index] |
|
|
break |
|
|
break |
|
|
index += 1 |
|
|
index += 1 |
|
|
self.data.save(self.config_path) |
|
|
self.data.save(self.config_path) |
|
|
|
|
|
|
|
|
def clear_ud(self, user): |
|
|
def clear_ud(self, user): |
|
|
up = {'u': 0, 'd': 0} |
|
|
up = {'u': 0, 'd': 0} |
|
|
self.data.load(self.config_path) |
|
|
self.data.load(self.config_path) |
|
|
for row in self.data.json: |
|
|
for row in self.data.json: |
|
|
match = True |
|
|
match = True |
|
|
if 'user' in user and row['user'] != user['user']: |
|
|
if 'user' in user and row['user'] != user['user']: |
|
|
match = False |
|
|
match = False |
|
|
if 'port' in user and row['port'] != user['port']: |
|
|
if 'port' in user and row['port'] != user['port']: |
|
|
match = False |
|
|
match = False |
|
|
if match: |
|
|
if match: |
|
|
row.update(up) |
|
|
row.update(up) |
|
|
print("clear user [%s]" % row['user']) |
|
|
print("clear user [%s]" % row['user']) |
|
|
self.data.save(self.config_path) |
|
|
self.data.save(self.config_path) |
|
|
|
|
|
|
|
|
def list_user(self, user): |
|
|
def list_user(self, user): |
|
|
self.data.load(self.config_path) |
|
|
self.data.load(self.config_path) |
|
|
if not user: |
|
|
if not user: |
|
|
for row in self.data.json: |
|
|
for row in self.data.json: |
|
|
print("user [%s] port %s" % (row['user'], row['port'])) |
|
|
print("user [%s] port %s" % (row['user'], row['port'])) |
|
|
return |
|
|
return |
|
|
for row in self.data.json: |
|
|
for row in self.data.json: |
|
|
match = True |
|
|
match = True |
|
|
if 'user' in user and row['user'] != user['user']: |
|
|
if 'user' in user and row['user'] != user['user']: |
|
|
match = False |
|
|
match = False |
|
|
if 'port' in user and row['port'] != user['port']: |
|
|
if 'port' in user and row['port'] != user['port']: |
|
|
match = False |
|
|
match = False |
|
|
if match: |
|
|
if match: |
|
|
print("### user [%s] info %s" % (row['user'], self.userinfo(row))) |
|
|
print("### user [%s] info %s" % (row['user'], self.userinfo(row))) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def print_server_help(): |
|
|
def print_server_help(): |
|
|
print('''usage: python mujson_manage.py -a|-d|-e|-c|-l [OPTION]... |
|
|
print('''usage: python mujson_manage.py -a|-d|-e|-c|-l [OPTION]... |
|
|
|
|
|
|
|
|
Actions: |
|
|
Actions: |
|
|
-a ADD add/edit a user |
|
|
-a ADD add/edit a user |
|
|
@ -212,118 +207,117 @@ General options: |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def main(): |
|
|
def main(): |
|
|
shortopts = 'adeclu:p:k:O:o:G:g:m:t:f:h' |
|
|
shortopts = 'adeclu:p:k:O:o:G:g:m:t:f:h' |
|
|
longopts = ['help'] |
|
|
longopts = ['help'] |
|
|
action = None |
|
|
action = None |
|
|
user = {} |
|
|
user = {} |
|
|
fast_set_obfs = {'0': 'plain', |
|
|
fast_set_obfs = {'0': 'plain', |
|
|
'+1': 'http_simple_compatible', |
|
|
'+1': 'http_simple_compatible', |
|
|
'1': 'http_simple', |
|
|
'1': 'http_simple', |
|
|
'+2': 'tls1.2_ticket_auth_compatible', |
|
|
'+2': 'tls1.2_ticket_auth_compatible', |
|
|
'2': 'tls1.2_ticket_auth'} |
|
|
'2': 'tls1.2_ticket_auth'} |
|
|
fast_set_protocol = {'0': 'origin', |
|
|
fast_set_protocol = {'0': 'origin', |
|
|
'+1': 'verify_sha1_compatible', |
|
|
'+1': 'verify_sha1_compatible', |
|
|
'1': 'verify_sha1', |
|
|
'1': 'verify_sha1', |
|
|
'2': 'auth_sha1', |
|
|
'2': 'auth_sha1', |
|
|
'3': 'auth_sha1_v2', |
|
|
'3': 'auth_sha1_v2', |
|
|
'4': 'auth_sha1_v4', |
|
|
'4': 'auth_sha1_v4', |
|
|
'am': 'auth_aes128_md5', |
|
|
'am': 'auth_aes128_md5', |
|
|
'as': 'auth_aes128_sha1', |
|
|
'as': 'auth_aes128_sha1', |
|
|
} |
|
|
} |
|
|
fast_set_method = {'a0': 'aes-128-cfb', |
|
|
fast_set_method = {'a0': 'aes-128-cfb', |
|
|
'a1': 'aes-192-cfb', |
|
|
'a1': 'aes-192-cfb', |
|
|
'a2': 'aes-256-cfb', |
|
|
'a2': 'aes-256-cfb', |
|
|
'r': 'rc4-md5', |
|
|
'r': 'rc4-md5', |
|
|
'r6': 'rc4-md5-6', |
|
|
'r6': 'rc4-md5-6', |
|
|
'c': 'chacha20', |
|
|
'c': 'chacha20', |
|
|
'ci': 'chacha20-ietf', |
|
|
'ci': 'chacha20-ietf', |
|
|
's': 'salsa20', |
|
|
's': 'salsa20', |
|
|
'b': 'bf-cfb', |
|
|
'b': 'bf-cfb', |
|
|
'm0': 'camellia-128-cfb', |
|
|
'm0': 'camellia-128-cfb', |
|
|
'm1': 'camellia-192-cfb', |
|
|
'm1': 'camellia-192-cfb', |
|
|
'm2': 'camellia-256-cfb', |
|
|
'm2': 'camellia-256-cfb', |
|
|
'a0t': 'aes-128-ctr', |
|
|
'a0t': 'aes-128-ctr', |
|
|
'a1t': 'aes-192-ctr', |
|
|
'a1t': 'aes-192-ctr', |
|
|
'a2t': 'aes-256-ctr'} |
|
|
'a2t': 'aes-256-ctr'} |
|
|
try: |
|
|
try: |
|
|
optlist, args = getopt.getopt(sys.argv[1:], shortopts, longopts) |
|
|
optlist, args = getopt.getopt(sys.argv[1:], shortopts, longopts) |
|
|
for key, value in optlist: |
|
|
for key, value in optlist: |
|
|
if key == '-a': |
|
|
if key == '-a': |
|
|
action = 1 |
|
|
action = 1 |
|
|
elif key == '-d': |
|
|
elif key == '-d': |
|
|
action = 2 |
|
|
action = 2 |
|
|
elif key == '-e': |
|
|
elif key == '-e': |
|
|
action = 3 |
|
|
action = 3 |
|
|
elif key == '-l': |
|
|
elif key == '-l': |
|
|
action = 4 |
|
|
action = 4 |
|
|
elif key == '-c': |
|
|
elif key == '-c': |
|
|
action = 0 |
|
|
action = 0 |
|
|
elif key == '-u': |
|
|
elif key == '-u': |
|
|
user['user'] = value |
|
|
user['user'] = value |
|
|
elif key == '-p': |
|
|
elif key == '-p': |
|
|
user['port'] = int(value) |
|
|
user['port'] = int(value) |
|
|
elif key == '-k': |
|
|
elif key == '-k': |
|
|
user['passwd'] = value |
|
|
user['passwd'] = value |
|
|
elif key == '-o': |
|
|
elif key == '-o': |
|
|
if value in fast_set_obfs: |
|
|
if value in fast_set_obfs: |
|
|
user['obfs'] = fast_set_obfs[value] |
|
|
user['obfs'] = fast_set_obfs[value] |
|
|
else: |
|
|
else: |
|
|
user['obfs'] = value |
|
|
user['obfs'] = value |
|
|
elif key == '-O': |
|
|
elif key == '-O': |
|
|
if value in fast_set_protocol: |
|
|
if value in fast_set_protocol: |
|
|
user['protocol'] = fast_set_protocol[value] |
|
|
user['protocol'] = fast_set_protocol[value] |
|
|
else: |
|
|
else: |
|
|
user['protocol'] = value |
|
|
user['protocol'] = value |
|
|
elif key == '-g': |
|
|
elif key == '-g': |
|
|
user['obfs_param'] = value |
|
|
user['obfs_param'] = value |
|
|
elif key == '-G': |
|
|
elif key == '-G': |
|
|
user['protocol_param'] = value |
|
|
user['protocol_param'] = value |
|
|
elif key == '-m': |
|
|
elif key == '-m': |
|
|
if value in fast_set_method: |
|
|
if value in fast_set_method: |
|
|
user['method'] = fast_set_method[value] |
|
|
user['method'] = fast_set_method[value] |
|
|
else: |
|
|
else: |
|
|
user['method'] = value |
|
|
user['method'] = value |
|
|
elif key == '-f': |
|
|
elif key == '-f': |
|
|
user['forbidden_port'] = value |
|
|
user['forbidden_port'] = value |
|
|
elif key == '-t': |
|
|
elif key == '-t': |
|
|
val = float(value) |
|
|
val = float(value) |
|
|
try: |
|
|
try: |
|
|
val = int(value) |
|
|
val = int(value) |
|
|
except: |
|
|
except: |
|
|
pass |
|
|
pass |
|
|
user['transfer_enable'] = val * (1024 ** 3) |
|
|
user['transfer_enable'] = val * (1024 ** 3) |
|
|
elif key in ('-h', '--help'): |
|
|
elif key in ('-h', '--help'): |
|
|
print_server_help() |
|
|
print_server_help() |
|
|
sys.exit(0) |
|
|
sys.exit(0) |
|
|
except getopt.GetoptError as e: |
|
|
except getopt.GetoptError as e: |
|
|
print(e) |
|
|
print(e) |
|
|
sys.exit(2) |
|
|
sys.exit(2) |
|
|
|
|
|
|
|
|
manage = MuMgr() |
|
|
|
|
|
if action == 0: |
|
|
|
|
|
manage.clear_ud(user) |
|
|
|
|
|
elif action == 1: |
|
|
|
|
|
if 'user' not in user and 'port' in user: |
|
|
|
|
|
user['user'] = str(user['port']) |
|
|
|
|
|
if 'user' in user and 'port' in user: |
|
|
|
|
|
manage.add(user) |
|
|
|
|
|
else: |
|
|
|
|
|
print("You have to set the port with -p") |
|
|
|
|
|
elif action == 2: |
|
|
|
|
|
if 'user' in user or 'port' in user: |
|
|
|
|
|
manage.delete(user) |
|
|
|
|
|
else: |
|
|
|
|
|
print("You have to set the user name or port with -u/-p") |
|
|
|
|
|
elif action == 3: |
|
|
|
|
|
if 'user' in user or 'port' in user: |
|
|
|
|
|
manage.edit(user) |
|
|
|
|
|
else: |
|
|
|
|
|
print("You have to set the user name or port with -u/-p") |
|
|
|
|
|
elif action == 4: |
|
|
|
|
|
manage.list_user(user) |
|
|
|
|
|
elif action is None: |
|
|
|
|
|
print_server_help() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
manage = MuMgr() |
|
|
|
|
|
if action == 0: |
|
|
|
|
|
manage.clear_ud(user) |
|
|
|
|
|
elif action == 1: |
|
|
|
|
|
if 'user' not in user and 'port' in user: |
|
|
|
|
|
user['user'] = str(user['port']) |
|
|
|
|
|
if 'user' in user and 'port' in user: |
|
|
|
|
|
manage.add(user) |
|
|
|
|
|
else: |
|
|
|
|
|
print("You have to set the port with -p") |
|
|
|
|
|
elif action == 2: |
|
|
|
|
|
if 'user' in user or 'port' in user: |
|
|
|
|
|
manage.delete(user) |
|
|
|
|
|
else: |
|
|
|
|
|
print("You have to set the user name or port with -u/-p") |
|
|
|
|
|
elif action == 3: |
|
|
|
|
|
if 'user' in user or 'port' in user: |
|
|
|
|
|
manage.edit(user) |
|
|
|
|
|
else: |
|
|
|
|
|
print("You have to set the user name or port with -u/-p") |
|
|
|
|
|
elif action == 4: |
|
|
|
|
|
manage.list_user(user) |
|
|
|
|
|
elif action is None: |
|
|
|
|
|
print_server_help() |
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
if __name__ == '__main__': |
|
|
main() |
|
|
main() |
|
|
|
noisyfox
8 years ago