feat: filter of Trojan-Go

2 years ago · 65baad2be1
4 changed files with 326 additions and 31 deletions
--- a/ProxyFilter/TrojanGo.py
+++ b/ProxyFilter/TrojanGo.py
@ -0,0 +1,140 @@
 #!/usr/bin/python
 # -*- coding:utf-8 -*-
 from ProxyFilter import baseFunc
 from ProxyFilter import Plugin
 trojanGoFilterRules = {
    'rootObject': {
        'remark': {
            'optional': False,
            'default': '',
            'type': str,
            'format': baseFunc.toStr
        },
        'server': {
            'optional': True,
            'type': str,
            'format': baseFunc.toStrTidy,
            'filter': baseFunc.isHost,
            'errMsg': 'Illegal server address'
        },
        'port': {
            'optional': True,
            'type': int,
            'format': baseFunc.toInt,
            'filter': baseFunc.isPort,
            'errMsg': 'Illegal port number'
        },
        'passwd': {
            'optional': True,
            'type': str,
            'format': baseFunc.toStr
        },
        'sni': {
            'optional': False,
            'default': '',
            'type': str,
            'format': baseFunc.toStr
        },
        'alpn': {
            'optional': False,
            'default': None,
            'allowNone': True,
            'type': str,
            'format': baseFunc.toStrTidy,
            'filter': lambda alpn: alpn in ['h2', 'http/1.1', 'h2,http/1.1'],
            'errMsg': 'Illegal alpn option'
        },
        'verify': {
            'optional': False,
            'default': True,
            'type': bool,
            'format': baseFunc.toBool
        },
        'ws': {
            'optional': False,
            'default': None,
            'allowNone': True,
            'type': 'wsObject'
        },
        'ss': {
            'optional': False,
            'default': None,
            'allowNone': True,
            'type': 'ssObject'
        },
        'plugin': {
            'optional': False,
            'default': None,
            'allowNone': True,
            'type': 'pluginObject'
        }
    },
    'ssObject': {
        'method': {
            'optional': False,
            'default': 'AES-128-GCM',
            'type': str,
            'format': lambda s: baseFunc.toStrTidy(s).replace('_', '-').upper(),
            'filter': lambda method: method in ['AES-128-GCM', 'AES-256-GCM', 'CHACHA20-IETF-POLY1305'],
            'errMsg': 'Unknown Shadowsocks method'
        },
        'passwd': {
            'optional': True,
            'type': str,
            'format': baseFunc.toStr
        },
    },
    'wsObject': {
        'host': {
            'optional': False,
            'default': '',
            'type': str,
            'format': baseFunc.toStr
        },
        'path': {
            'optional': False,
            'default': '/',
            'type': str,
            'format': baseFunc.toStr
        }
    },
    'pluginObject': {
        'type': {
            'optional': True,
            'type': str,
            'format': lambda pluginType: Plugin.pluginFormat(baseFunc.toStrTidy(pluginType)),
            'filter': Plugin.isPlugin,
            'errMsg': 'Unknown SIP003 plugin'
        },
        'param': {
            'optional': False,
            'default': '',
            'type': str,
            'format': baseFunc.toStr
        }
    }
 }
 def trojanGoFilter(rawInfo: dict, isExtra: bool) -> tuple[bool, str or dict]:
    """
    Trojan-Go节点合法性检查
        不合法:
            return False, {reason}
        合法:
            return True, {
                'type': 'trojan-go',
                ...
            }
    """
    try:
        if not isExtra: # 去除非必要参数
            trojanGoFilterRules['rootObject'].pop('remark')
        return baseFunc.ruleFilter(rawInfo, trojanGoFilterRules, {
            'type': 'trojan-go'
        })
    except:
        return False, 'Unknown error'
--- a/ProxyFilter/filter.py
+++ b/ProxyFilter/filter.py
@ -6,6 +6,7 @@ from ProxyFilter import ShadowsocksR
 from ProxyFilter import VMess
 from ProxyFilter import VLESS
 from ProxyFilter import Trojan
 from ProxyFilter import TrojanGo
 def filte(raw: dict, isExtra: bool = False) -> tuple[bool, str or dict]:
    """
@ -34,6 +35,8 @@ def filte(raw: dict, isExtra: bool = False) -> tuple[bool, str or dict]:
            return VLESS.vlessFilter(raw, isExtra)
        elif raw['type'] == 'trojan':
            return Trojan.trojanFilter(raw, isExtra)
        elif raw['type'] == 'trojan-go':
            return TrojanGo.trojanGoFilter(raw, isExtra)
        else:
            return False, 'Unknown proxy type'
    except:
--- a/demo.py
+++ b/demo.py
@ -2,37 +2,28 @@ import ProxyDecoder as Decoder
 import ProxyFilter as Filter
 import Check as Checker
-# url = 'trojan://dnomd343@127.0.0.1:12345?security=tls&sni=local.343.re'
+info = {
-url = 'trojan://dnomd343@local.343.re:12345'
+    'type': 'trojan-go',
    'server': '127.0.0.1',
    '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': {
        'type': 'obfs-local',
        'param': 'obfs=http'
    }
 }
-ret = Decoder.decode(url)
+status, ret = Filter.filte(info, isExtra = True)
 print(ret)
 #
 status, ret = Filter.filte(ret, isExtra = True)
 print(status)
 print(ret)
 # info = {
 #     'type': 'vmess',
 #     'server': 'baidu.com',
 #     'port': 12345,
 #     'id': 'a859f794-1fcb-422e-bcad-3264dcea1f12',
 #     'aid': 0,
 #     'stream': {
 #         'type': 'ws',
 #         'host': 'host.343.re',
 #         'secure': {
 #             # 'sni': 'sni.343.re'
 #         }
 #     }
 # }
 # status, ret = Filter.filte(info, isExtra = True)
 # print(status)
 # print(ret)
 data = Checker.proxyTest({
    'check': ['http'],
    'info': ret
 })
 print(data)
--- a/docs/ProxyObject/TrojanGo.md
+++ b/docs/ProxyObject/TrojanGo.md
@ -0,0 +1,161 @@
 ## Trojan-Go
 > **remark**
 > 
 > + 类型：*str*
 > + 说明：节点备注名称
 > + 缺省：''
 > + 可选值：不限
 ```
 {
    'type': 'trojan-go',
    'server': ...,
    'port': ...,
    'passwd': ...,
    'sni': ...,
    'alpn': ...,
    'verify': ...,
    'ws': ...,
    'ss': ...,
    'plugin': ...
 }
 ```
 **server**
 + 类型：*str*
 + 说明：服务器地址
 + 缺省：必选
 + 可选值：合法的IP地址或域名
 **port**
 + 类型：*int*
 + 说明：服务器端口
 + 缺省：必选
 + 可选值：1 ～ 65535
 **passwd**
 + 类型：*str*
 + 说明：Trojan连接密码
 + 缺省：必选
 + 可选值：不限
 **sni**
 + 类型：*str*
 + 说明：TLS握手SNI字段
 + 缺省： ''
 + 可选值：不限
 + 建议值：合法域名
 **alpn**
 + 类型：*None* / *str*
 + 说明：TLS握手协商协议
 + 缺省：None
 + 可选值：`h2`,`http/1.1`,`h2,http/1.1`
 + 建议值：'h2,http/1.1'
 **verify**
 + 类型：*bool*
 + 说明：是否验证服务端证书
 + 缺省：True
 + 可选值：不限
 + 建议值：True
 **ws**
 + 类型：*None* / *wsObject*
 + 说明：WebSocket连接配置
 + 缺省：None
 + 可选值：不限
 **ss**
 + 类型：*None* / *ssObject*
 + 说明：Shadowsocks加密配置
 + 缺省：None
 + 可选值：不限
 **plugin**
 + 类型：*None* / *pluginObject*
 + 说明：SIP003插件
 + 缺省：None
 + 可选值：不限
 ### wsObject
 ```
 {
    'host': ...,
    'path': ...
 }
 ```
 **host**
 + 类型：*str*
 + 说明：Websocket连接域名
 + 缺省：''
 + 可选值：不限
 + 建议值：合法域名
 **path**
 + 类型：*str*
 + 说明：Websocket连接路径
 + 缺省：'/'
 + 可选值：不限
 + 建议值：以`/`开头的合法路径
 ### ssObject
 ```
 {
    'method': ...,
    'passwd': ...
 }
 ```
 **method**
 + 类型：*str*
 + 说明：Shadowsocks流加密方式
 + 缺省：'AES-128-GCM'
 + 可选值：`AES-128-GCM`,`AES-256-GCM`,`CHACHA20-IETF-POLY1305`
 + 建议值：x86平台使用AES方式，ARM平台使用CHACHA20方式
 **passwd**
 + 类型：*str*
 + 说明：Shadowsocks密码
 + 缺省：''
 + 可选值：不限
 ### pluginObject
 ```
 {
    'type': ...,
    'param': ...
 }
 ```
 **type**
 + 类型：*str*
 + 说明：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`
 **param**
 + 类型：*str*
 + 说明：SIP003插件参数
 + 缺省：''
 + 可选值：不限