feat: test of Trojan-Go

ProxyBuilder/TrojanGo.py

@@ -73,4 +73,4 @@ def load(proxyInfo: dict, socksPort: int, configFile: str) -> tuple[list, str, d
         'shadowsocks': __ssConfig(proxyInfo),
         'transport_plugin': __pluginConfig(proxyInfo)
     }
-    return ['trojan-go', '-config', configFile], json.dumps(config), {}
+    return ['trojan-go', '-config', configFile], json.dumps(config), {'PATH': '/usr/bin'}

ProxyBuilder/builder.py

@@ -6,6 +6,7 @@ import time
 import ctypes
 import random
 import socket
+import signal
 import subprocess
 
 from ProxyBuilder import Shadowsocks
@@ -163,6 +164,9 @@ def check(client: dict) -> bool or None:
     """
     return client['process'].poll() is None
 
+def skipDestroy() -> None:
+    pass
+
 def destroy(client: dict) -> bool:
     """
     结束客户端并清理
@@ -171,12 +175,21 @@ def destroy(client: dict) -> bool:
 
         销毁成功: return True
     """
+    signal.signal(signal.SIGINT, skipDestroy) # 捕获SIGINT
+    signal.signal(signal.SIGTERM, skipDestroy) # 捕获SIGTERM
     try:
         process = client['process']
+        os.killpg(os.getpgid(process.pid), signal.SIGTERM) # 杀进程组
+        time.sleep(0.2)
+        maxWait = 100
         if process.poll() is None: # 未死亡
             while process.poll() is None: # 等待退出
                 process.terminate() # SIGTERM
                 time.sleep(0.2)
+                maxWait -= 1
+                if maxWait < 0:
+                    process.kill() # SIGKILL
+                    time.sleep(0.5)
     except:
         return False
 

ProxyTester/TrojanGo.py

@@ -0,0 +1,164 @@
+#!/usr/bin/python
+# -*- coding:utf-8 -*-
+
+import copy
+import json
+
+from ProxyTester import Plugin
+
+config = {}
+
+trojanGoMethod = [
+    'AES-128-GCM',
+    'AES-256-GCM',
+    'CHACHA20-IETF-POLY1305'
+]
+
+sip003PluginList = [ # SIP003插件列表
+    'obfs-local',
+    'simple-tls',
+    'v2ray-plugin',
+    'xray-plugin',
+    'kcptun-client',
+    'gost-plugin',
+    'ck-client',
+    'gq-client',
+    'mtt-client',
+    'rabbit-plugin',
+    'qtun-client',
+    'gun-plugin'
+]
+
+def loadTrojanGo(isWs: bool, ssMethod: str or None) -> dict:
+    caption = 'Trojan-Go original'
+    serverConfig = {
+        'run_type': 'server',
+        'local_addr': '127.0.0.1',
+        'local_port': config['port'],
+        'remote_addr': '127.0.0.1', # only for shadowsocks fallback
+        'remote_port': 343,
+        'password': [
+            config['passwd']
+        ],
+        'disable_http_check': True,
+        'ssl': {
+            'cert': config['cert'],
+            'key': config['key']
+        }
+    }
+    proxyInfo = {
+        'type': 'trojan-go',
+        'server': '127.0.0.1',
+        'port': config['port'],
+        'passwd': config['passwd'],
+        'sni': config['host'],
+    }
+    if ssMethod is not None: # add Shadowsocks encrypt
+        caption += ' ' + ssMethod + ' encrypt'
+        serverConfig['shadowsocks'] = {
+            'enabled': True,
+            'method': ssMethod,
+            'password': config['passwd']
+        }
+        proxyInfo['ss'] = {
+            'method': ssMethod,
+            'passwd': config['passwd']
+        }
+    if isWs: # add WebSocket config
+        caption += ' (websocket)'
+        serverConfig['websocket'] = {
+            'enabled': True,
+            'host': config['host'],
+            'path': config['path']
+        }
+        proxyInfo['ws'] = {
+            'host': config['host'],
+            'path': config['path']
+        }
+    return {
+        'caption': caption,
+        'client': proxyInfo,
+        'server': serverConfig,
+        'file': None,
+        'path': None
+    }
+
+def loadTrojanGoPlugin(plugin: str) -> list:
+    result = []
+    rabbitPort = 20191
+    trojanBaseConfig = loadTrojanGo(False, None)
+
+    if plugin == 'rabbit-plugin': # rabbit-tcp
+        trojanBaseConfig['caption'] = 'Trojan-Go rabbit-plugin (basic mode)'
+        trojanBaseConfig['client']['port'] = rabbitPort
+        trojanBaseConfig['client']['plugin'] = {
+            'type': 'rabbit-plugin',
+            'param': 'serviceAddr=127.0.0.1:' + str(config['port']) + ';password=' + config['passwd']
+        }
+        trojanBaseConfig['server']['transport_plugin'] = {
+            'enabled': True,
+            'type': 'other',
+            'command': 'rabbit',
+            'arg': [
+                '-mode', 's',
+                '-password', config['passwd'],
+                '-rabbit-addr', ':' + str(rabbitPort)
+            ]
+        }
+        trojanBaseConfig['file'] = None
+        trojanBaseConfig['path'] = None
+        return [trojanBaseConfig]
+
+    # other plugin
+    pluginConfig = Plugin.loadPluginConfig(plugin, config['host'], config['cert'], config['key'])  # 载入插件配置
+    for pluginOption in pluginConfig:
+        trojanConfig = copy.deepcopy(trojanBaseConfig)
+        trojanConfig['caption'] = 'Trojan-Go plugin ' + plugin + ' (' + pluginOption['caption'] + ')'
+        trojanConfig['client']['plugin'] = pluginOption['client']
+        trojanConfig['server']['transport_plugin'] = {
+            'enabled': True,
+            'type': 'shadowsocks',
+            'command': pluginOption['server']['type'],
+            'option': pluginOption['server']['param']
+        }
+        trojanConfig['file'] = pluginOption['file']
+        trojanConfig['path'] = pluginOption['path']
+        result.append(trojanConfig)
+    return result
+
+def loadTrojanGoConfig(trojanGoConfigList: list) -> list:
+    result = []
+    for trojanGoConfig in trojanGoConfigList:
+        result.append({
+            'caption': trojanGoConfig['caption'],
+            'proxy': trojanGoConfig['client'],
+            'server': {
+                'startCommand': ['trojan-go', '-config', config['file']],
+                'fileContent': json.dumps(trojanGoConfig['server']),
+                'filePath': config['file'],
+                'envVar': {'PATH': '/usr/bin'}
+            },
+            'aider': {
+                'startCommand': None,
+                'fileContent': trojanGoConfig['file'],
+                'filePath': trojanGoConfig['path'],
+                'envVar': {}
+            }
+        })
+    return result
+
+def trojanGoTest(trojanGoConfig: dict) -> list:
+    result = []
+    for key, value in trojanGoConfig.items(): # trojanGoConfig -> config
+        config[key] = value
+
+    result += loadTrojanGoConfig([loadTrojanGo(False, None)])
+    result += loadTrojanGoConfig([loadTrojanGo(True, None)])
+    for ssMethod in trojanGoMethod:
+        result += loadTrojanGoConfig([loadTrojanGo(False, ssMethod)])
+        result += loadTrojanGoConfig([loadTrojanGo(True, ssMethod)])
+
+    for plugin in sip003PluginList:
+        result += loadTrojanGoConfig(loadTrojanGoPlugin(plugin))
+
+    return result

ProxyTester/tester.py

@@ -6,6 +6,7 @@ from ProxyTester import ShadowsocksR
 from ProxyTester import VMess
 from ProxyTester import VLESS
 from ProxyTester import Trojan
+from ProxyTester import TrojanGo
 
 def test(key: str, config: dict) -> list:
     if key in ['ss', 'shadowsocks']:
@@ -18,5 +19,7 @@ def test(key: str, config: dict) -> list:
         return VLESS.vlessTest(config)
     elif key == 'trojan':
         return Trojan.trojanTest(config)
+    elif key == 'trojan-go':
+        return TrojanGo.trojanGoTest(config)
     else:
         return []

Test.py

@@ -12,12 +12,12 @@ import ProxyTester as Tester
 testConfig = {
     'port': 12345,
     'passwd': 'dnomd343',
-    'host': 'dns.343.re',
+    'host': 'local.343.re',
     'path': '/test',
     'service': 'dnomd343',
-    'file': '/tmp/proxyc-test.json',
-    'cert': '/etc/ssl/certs/dns.343.re/certificate.crt',
-    'key': '/etc/ssl/certs/dns.343.re/private.key',
+    'file': '/tmp/proxycTest.json',
+    'cert': '/etc/ssl/certs/343.re/fullchain.pem',
+    'key': '/etc/ssl/certs/343.re/privkey.pem',
     'id': '1f7aa040-94d8-4b53-ae85-af6946d550bb',
 }
 
@@ -25,6 +25,8 @@ def testBuild(config: dict): # load file and start process
     if config['filePath'] is not None:
         with open(config['filePath'], 'w') as fileObject:  # save file
             fileObject.write(config['fileContent'])
+    if config['startCommand'] is None:
+        return None
     return subprocess.Popen( # start process
         config['startCommand'],
         env = config['envVar'],
@@ -33,7 +35,7 @@ def testBuild(config: dict): # load file and start process
     )
 
 def testDestroy(config: dict, process): # remove file and kill process
-    if process.poll() is None:  # still alive
+    if process is not None and process.poll() is None:  # still alive
         while process.poll() is None:  # wait for exit
             process.terminate()  # SIGTERM
             time.sleep(0.2)

demo.py

@@ -6,31 +6,21 @@ import Check as Checker
 info = {
     'type': 'trojan-go',
     'server': '127.0.0.1',
-    'port': '12345',
+    'port': 12345,
     'passwd': 'dnomd343',
     'sni': 'local.343.re',
-    'alpn': 'h2',
-    'verify': False,
-    # 'ws': {
-    #     'host': 'local.343.re',
-    #     'path': '/test'
-    # },
-    # 'ss': {
-    #     'method': 'chacha20-ietf-poly1305',
-    #     'passwd': 'dnomd343'
-    # },
-    # 'plugin': {
+    'plugin': {
         # 'type': 'obfs-local',
-    #     'param': 'obfs=http'
-    # }
+        # 'param': 'obfs=http;obfs-host=www.bing.com'
+        'type': 'simple-tls',
+        'param': 'n=local.343.re;no-verify'
+    }
 }
 
 status, ret = Filter.filte(info, isExtra = True)
 print(status)
 print(ret)
 
-# Builder.build(ret, '/tmp/ProxyC')
-
 data = Checker.proxyTest({
     'check': ['http'],
     'info': ret