From 002fc6fe535a79a3d3646b22e4356e6ed50053c4 Mon Sep 17 00:00:00 2001
From: dnomd343 <i@343.re>
Date: Wed, 6 Jul 2022 11:31:44 +0800
Subject: [PATCH] refactor: shadowsocks proxy builder

---
 Builder/Shadowsocks.py | 84 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 Builder/Shadowsocks.py

diff --git a/Builder/Shadowsocks.py b/Builder/Shadowsocks.py
new file mode 100644
index 0000000..e893c96
--- /dev/null
+++ b/Builder/Shadowsocks.py
@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import json
+from Basis.Methods import ssMethods
+
+
+def loadConfig(proxyInfo: dict, socksInfo: dict) -> dict:  # load basic config option
+    config = {
+        'server': proxyInfo['server'],
+        'server_port': proxyInfo['port'],  # type -> int
+        'local_address': socksInfo['addr'],
+        'local_port': socksInfo['port'],  # type -> int
+        'method': proxyInfo['method'],
+        'password': proxyInfo['passwd'],
+    }
+    if proxyInfo['plugin'] is not None:  # with plugin
+        config['plugin'] = proxyInfo['plugin']['type']
+        config['plugin_opts'] = proxyInfo['plugin']['param']
+    return config
+
+
+def pluginUdp(plugin: str, pluginParam: str) -> bool:  # whether the plugin uses UDP
+    if plugin in ['obfs-local', 'simple-tls', 'ck-client', 'gq-client', 'mtt-client', 'rabbit-plugin']:
+        return False  # UDP is not used
+    if plugin in ['v2ray-plugin', 'xray-plugin', 'gost-plugin']:
+        if 'mode=quic' not in pluginParam.split(';'):  # non-quic mode does not use UDP
+            return False
+    return True  # UDP is assumed by default
+
+
+def ssRust(proxyInfo: dict, socksInfo: dict, isUdp: bool, isLegacy: bool = False) -> tuple[dict, list]:
+    config = loadConfig(proxyInfo, socksInfo)
+    if isUdp:  # Proxy UDP traffic
+        config['mode'] = 'tcp_and_udp'
+    return config, ['ss-rust-local', '-v']
+
+
+def ssLibev(proxyInfo: dict, socksInfo: dict, isUdp: bool, isLegacy: bool = False) -> tuple[dict, list]:
+    config = loadConfig(proxyInfo, socksInfo)
+    if isUdp:  # Proxy UDP traffic
+        config['mode'] = 'tcp_and_udp'
+    return config, ['ss-libev-legacy-local' if isLegacy else 'ss-libev-local', '-v']
+
+
+def ssPython(proxyInfo: dict, socksInfo: dict, isUdp: bool, isLegacy: bool = False) -> tuple[dict, list]:
+    config = loadConfig(proxyInfo, socksInfo)
+    mbedtlsMethods = [
+        'aes-128-cfb128',
+        'aes-192-cfb128',
+        'aes-256-cfb128',
+        'camellia-128-cfb128',
+        'camellia-192-cfb128',
+        'camellia-256-cfb128',
+    ]
+    if not isLegacy:  # only for latest version
+        if config['method'] in mbedtlsMethods:  # mbedtls methods should use prefix `mbedtls:`
+            config['method'] = 'mbedtls:' + config['method']
+        if config['method'] in ['idea-cfb', 'seed-cfb']:  # Only older versions of openssl are supported
+            config['extra_opts'] = '--libopenssl=libcrypto.so.1.0.0'
+    if not isUdp:
+        config['no_udp'] = True  # UDP traffic is not proxied
+    config['shadowsocks'] = 'ss-python-legacy-local' if isLegacy else 'ss-python-local'
+    return config, ['ss-bootstrap-local', '--debug', '-vv']
+
+
+def load(proxyInfo: dict, socksInfo: dict, configFile: str) -> tuple[list, str, None]:
+    if proxyInfo['plugin'] is None:  # UDP is enabled when server without plugin
+        isUdp = True
+    else:
+        isUdp = not pluginUdp(  # check the UDP conflict status of plugins
+            proxyInfo['plugin']['type'], proxyInfo['plugin']['param']
+        )
+    if proxyInfo['method'] not in ssMethods['all']:  # unknown shadowsocks method
+        raise RuntimeError('Unknown shadowsocks method')
+    for client in ssMethods:  # traverse all shadowsocks client
+        if proxyInfo['method'] not in ssMethods[client] or client != 'all':
+            continue
+        ssLoadConfig = None
+        if 'rust' in client: ssLoadConfig = ssRust
+        if 'libev' in client: ssLoadConfig = ssLibev
+        if 'python' in client: ssLoadConfig = ssPython
+        ssConfig, ssClient = ssLoadConfig(proxyInfo, socksInfo, isUdp, 'legacy' in client)  # generate config file
+        return ssClient + ['-c', configFile], json.dumps(ssConfig), None  # tuple[command, fileContent, envVar]