Add shadowsocks tests

2025-04-04 20:37:37 +03:00 · 2022-07-09 00:01:23 +08:00 · 2022-07-09 00:01:23 +08:00 · 7f8c9ffa30
commit 7f8c9ffa30
parent f448b6b977
10 changed files with 1063 additions and 2 deletions
--- a/test/shadowsocks_test.go
+++ b/test/shadowsocks_test.go
@ -0,0 +1,189 @@
+package main
+
+import (
+	"crypto/rand"
+	"encoding/base64"
+	"net/netip"
+	"testing"
+
+	C "github.com/sagernet/sing-box/constant"
+	"github.com/sagernet/sing-box/option"
+	F "github.com/sagernet/sing/common/format"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestShadowsocks(t *testing.T) {
+	for _, method16 := range []string{
+		"2022-blake3-aes-128-gcm",
+	} {
+		t.Run(method16+"-inbound", func(t *testing.T) {
+			testShadowsocksInboundWithShadowsocksRust(t, method16, mkBase64(t, 16))
+		})
+		t.Run(method16+"-outbound", func(t *testing.T) {
+			testShadowsocksOutboundWithShadowsocksRust(t, method16, mkBase64(t, 16))
+		})
+		t.Run(method16+"-self", func(t *testing.T) {
+			testShadowsocksSelf(t, method16, mkBase64(t, 16))
+		})
+	}
+	for _, method32 := range []string{
+		"2022-blake3-aes-256-gcm",
+		"2022-blake3-chacha20-poly1305",
+	} {
+		t.Run(method32+"-inbound", func(t *testing.T) {
+			testShadowsocksInboundWithShadowsocksRust(t, method32, mkBase64(t, 32))
+		})
+		t.Run(method32+"-outbound", func(t *testing.T) {
+			testShadowsocksOutboundWithShadowsocksRust(t, method32, mkBase64(t, 32))
+		})
+		t.Run(method32+"-self", func(t *testing.T) {
+			testShadowsocksSelf(t, method32, mkBase64(t, 32))
+		})
+	}
+}
+
+func testShadowsocksInboundWithShadowsocksRust(t *testing.T, method string, password string) {
+	t.Parallel()
+	serverPort := mkPort(t)
+	clientPort := mkPort(t)
+	testPort := mkPort(t)
+	startDockerContainer(t, DockerOptions{
+		Image:      ImageShadowsocksRustClient,
+		EntryPoint: "sslocal",
+		Ports:      []uint16{serverPort, clientPort},
+		Cmd:        []string{"-s", F.ToString("127.0.0.1:", serverPort), "-b", F.ToString("0.0.0.0:", clientPort), "-m", method, "-k", password, "-U"},
+	})
+	startInstance(t, option.Options{
+		Log: &option.LogOption{
+			Disabled: true,
+		},
+		Inbounds: []option.Inbound{
+			{
+				Type: C.TypeShadowsocks,
+				ShadowsocksOptions: option.ShadowsocksInboundOptions{
+					ListenOptions: option.ListenOptions{
+						Listen:     option.ListenAddress(netip.IPv4Unspecified()),
+						ListenPort: serverPort,
+					},
+					Method:   method,
+					Password: password,
+				},
+			},
+		},
+	})
+	testSuit(t, clientPort, testPort)
+}
+
+func testShadowsocksOutboundWithShadowsocksRust(t *testing.T, method string, password string) {
+	t.Parallel()
+	serverPort := mkPort(t)
+	clientPort := mkPort(t)
+	testPort := mkPort(t)
+	startDockerContainer(t, DockerOptions{
+		Image:      ImageShadowsocksRustServer,
+		EntryPoint: "ssserver",
+		Ports:      []uint16{serverPort, testPort},
+		Cmd:        []string{"-s", F.ToString("0.0.0.0:", serverPort), "-m", method, "-k", password, "-U"},
+	})
+	startInstance(t, option.Options{
+		Log: &option.LogOption{
+			Disabled: true,
+		},
+		Inbounds: []option.Inbound{
+			{
+				Type: C.TypeMixed,
+				MixedOptions: option.SimpleInboundOptions{
+					ListenOptions: option.ListenOptions{
+						Listen:     option.ListenAddress(netip.IPv4Unspecified()),
+						ListenPort: clientPort,
+					},
+				},
+			},
+		},
+		Outbounds: []option.Outbound{
+			{
+				Type: C.TypeShadowsocks,
+				ShadowsocksOptions: option.ShadowsocksOutboundOptions{
+					ServerOptions: option.ServerOptions{
+						Server:     "127.0.0.1",
+						ServerPort: serverPort,
+					},
+					Method:   method,
+					Password: password,
+				},
+			},
+		},
+	})
+	testSuit(t, clientPort, testPort)
+}
+
+func testShadowsocksSelf(t *testing.T, method string, password string) {
+	t.Parallel()
+	serverPort := mkPort(t)
+	clientPort := mkPort(t)
+	testPort := mkPort(t)
+	startInstance(t, option.Options{
+		Log: &option.LogOption{
+			Disabled: true,
+		},
+		Inbounds: []option.Inbound{
+			{
+				Type: C.TypeMixed,
+				Tag:  "mixed-in",
+				MixedOptions: option.SimpleInboundOptions{
+					ListenOptions: option.ListenOptions{
+						Listen:     option.ListenAddress(netip.IPv4Unspecified()),
+						ListenPort: clientPort,
+					},
+				},
+			},
+			{
+				Type: C.TypeShadowsocks,
+				ShadowsocksOptions: option.ShadowsocksInboundOptions{
+					ListenOptions: option.ListenOptions{
+						Listen:     option.ListenAddress(netip.IPv4Unspecified()),
+						ListenPort: serverPort,
+					},
+					Method:   method,
+					Password: password,
+				},
+			},
+		},
+		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,
+				},
+			},
+		},
+		Route: &option.RouteOptions{
+			Rules: []option.Rule{
+				{
+					DefaultOptions: option.DefaultRule{
+						Inbound:  []string{"mixed-in"},
+						Outbound: "ss-out",
+					},
+				},
+			},
+		},
+	})
+	testSuit(t, clientPort, testPort)
+}
+
+func mkBase64(t *testing.T, length int) string {
+	psk := make([]byte, length)
+	_, err := rand.Read(psk)
+	require.NoError(t, err)
+	return base64.StdEncoding.EncodeToString(psk)
+}