sing-box/test/mux_test.go

179 lines
3.8 KiB
Go
Raw Normal View History

2022-07-29 16:29:22 +00:00
package main
import (
"net/netip"
"testing"
C "github.com/sagernet/sing-box/constant"
"github.com/sagernet/sing-box/option"
"github.com/sagernet/sing-shadowsocks/shadowaead_2022"
2022-08-04 02:38:20 +00:00
2023-04-09 07:37:06 +00:00
"github.com/gofrs/uuid/v5"
2022-07-29 16:29:22 +00:00
)
2023-07-11 06:03:55 +00:00
var muxProtocols = []string{
"h2mux",
"smux",
"yamux",
2022-08-04 02:38:20 +00:00
}
2022-09-23 05:14:31 +00:00
func TestVMessSMux(t *testing.T) {
testVMessMux(t, option.OutboundMultiplexOptions{
2023-04-20 03:16:22 +00:00
Enabled: true,
2023-07-11 06:03:55 +00:00
Protocol: "smux",
2023-04-20 03:16:22 +00:00
})
2022-08-03 13:51:34 +00:00
}
2022-09-23 05:14:31 +00:00
func TestShadowsocksMux(t *testing.T) {
2022-08-04 02:38:20 +00:00
for _, protocol := range muxProtocols {
2023-07-11 06:03:55 +00:00
t.Run(protocol, func(t *testing.T) {
testShadowsocksMux(t, option.OutboundMultiplexOptions{
2023-04-20 03:16:22 +00:00
Enabled: true,
2023-07-11 06:03:55 +00:00
Protocol: protocol,
2023-04-20 03:16:22 +00:00
})
2022-08-04 02:38:20 +00:00
})
}
}
2023-04-20 03:16:22 +00:00
func TestShadowsockH2Mux(t *testing.T) {
testShadowsocksMux(t, option.OutboundMultiplexOptions{
2023-04-20 03:16:22 +00:00
Enabled: true,
2023-07-11 06:03:55 +00:00
Protocol: "h2mux",
2023-04-20 03:16:22 +00:00
Padding: true,
})
}
func TestShadowsockSMuxPadding(t *testing.T) {
testShadowsocksMux(t, option.OutboundMultiplexOptions{
2023-04-20 03:16:22 +00:00
Enabled: true,
2023-07-11 06:03:55 +00:00
Protocol: "smux",
2023-04-20 03:16:22 +00:00
Padding: true,
})
}
func testShadowsocksMux(t *testing.T, options option.OutboundMultiplexOptions) {
2022-07-29 16:29:22 +00:00
method := shadowaead_2022.List[0]
password := mkBase64(t, 16)
startInstance(t, option.Options{
Inbounds: []option.Inbound{
{
Type: C.TypeMixed,
Tag: "mixed-in",
MixedOptions: option.HTTPMixedInboundOptions{
ListenOptions: option.ListenOptions{
2023-03-19 12:46:22 +00:00
Listen: option.NewListenAddress(netip.IPv4Unspecified()),
2022-07-29 16:29:22 +00:00
ListenPort: clientPort,
},
},
},
{
Type: C.TypeShadowsocks,
ShadowsocksOptions: option.ShadowsocksInboundOptions{
ListenOptions: option.ListenOptions{
2023-03-19 12:46:22 +00:00
Listen: option.NewListenAddress(netip.IPv4Unspecified()),
2022-07-29 16:29:22 +00:00
ListenPort: serverPort,
},
Method: method,
Password: password,
2023-12-27 02:30:04 +00:00
Multiplex: &option.InboundMultiplexOptions{
Enabled: true,
},
2022-07-29 16:29:22 +00:00
},
},
},
Outbounds: []option.Outbound{
{
Type: C.TypeDirect,
},
{
Type: C.TypeShadowsocks,
Tag: "ss-out",
ShadowsocksOptions: option.ShadowsocksOutboundOptions{
ServerOptions: option.ServerOptions{
Server: "127.0.0.1",
ServerPort: serverPort,
},
Method: method,
Password: password,
Multiplex: &options,
2022-07-29 16:29:22 +00:00
},
},
},
Route: &option.RouteOptions{
Rules: []option.Rule{
{
DefaultOptions: option.DefaultRule{
Inbound: []string{"mixed-in"},
Outbound: "ss-out",
},
},
},
},
})
testSuit(t, clientPort, testPort)
}
2022-08-04 02:38:20 +00:00
func testVMessMux(t *testing.T, options option.OutboundMultiplexOptions) {
2022-08-04 02:38:20 +00:00
user, _ := uuid.NewV4()
startInstance(t, option.Options{
Inbounds: []option.Inbound{
{
Type: C.TypeMixed,
Tag: "mixed-in",
MixedOptions: option.HTTPMixedInboundOptions{
ListenOptions: option.ListenOptions{
2023-03-19 12:46:22 +00:00
Listen: option.NewListenAddress(netip.IPv4Unspecified()),
2022-08-04 02:38:20 +00:00
ListenPort: clientPort,
},
},
},
{
Type: C.TypeVMess,
VMessOptions: option.VMessInboundOptions{
ListenOptions: option.ListenOptions{
2023-03-19 12:46:22 +00:00
Listen: option.NewListenAddress(netip.IPv4Unspecified()),
2022-08-04 02:38:20 +00:00
ListenPort: serverPort,
},
Users: []option.VMessUser{
{
UUID: user.String(),
},
},
Multiplex: &option.InboundMultiplexOptions{
Enabled: true,
},
2022-08-04 02:38:20 +00:00
},
},
},
Outbounds: []option.Outbound{
{
Type: C.TypeDirect,
},
{
Type: C.TypeVMess,
Tag: "vmess-out",
VMessOptions: option.VMessOutboundOptions{
ServerOptions: option.ServerOptions{
Server: "127.0.0.1",
ServerPort: serverPort,
},
2023-04-20 03:16:22 +00:00
Security: "auto",
UUID: user.String(),
Multiplex: &options,
2022-08-04 02:38:20 +00:00
},
},
},
Route: &option.RouteOptions{
Rules: []option.Rule{
{
DefaultOptions: option.DefaultRule{
Inbound: []string{"mixed-in"},
Outbound: "vmess-out",
},
},
},
},
})
testSuit(t, clientPort, testPort)
}