durian/durian_client.py at master · arthurlib/durian · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import json
import os
import sys

from lib.cipher import get_cipher
from lib.config import client_config__handler

base_path = os.path.dirname(os.path.realpath(sys.argv[0]))
sys.path.append(os.path.join(base_path, "./"))
import tornado

from service.client import Socks5Client, TunnelClient

cfgs = {}


def read_config():
    global cfgs
    with open(os.path.join(base_path, "config.json"), "r") as f:
        content = f.read()
        cfgs = json.loads(content)
    cfgs = client_config__handler(cfgs)


async def run():
    global cfgs
    cfg = cfgs.get('Socks5Client', None)
    if cfg:
        socks5_client = Socks5Client()
        socks5_client.set_remote_address(cfg['remote_host'], cfg['remote_port'])
        if cfg.get('key'):
            socks5_client.set_cipher(get_cipher('caesar', cfg['key']))
        socks5_client.listen(cfg['local_port'], cfg['local_host'] if cfg.get('local_host') else '')

    cfg = cfgs.get('TunnelClient', None)
    if cfg:
        tunnel_client = TunnelClient(cfg['key'])
        tunnel_client.set_remote_address(cfg['remote_host'], cfg['remote_port'])
        await tunnel_client.start()


def run2():
    global cfgs
    cfg = cfgs.get('Socks5Client', None)
    if cfg:
        socks5_client = Socks5Client()
        socks5_client.set_remote_address(cfg['remote_host'], cfg['remote_port'])
        if cfg.get('key'):
            socks5_client.set_cipher(get_cipher('caesar', cfg['key']))
        socks5_client.listen(cfg['local_port'], cfg['local_host'] if cfg.get('local_host') else '')

    tornado.ioloop.IOLoop.current().start()
if __name__ == '__main__':
    read_config()
    if cfgs.get('TunnelClient', None):
        tornado.ioloop.IOLoop.current().run_sync(run)
    else:
        run2()