sing-box/test/box_test.go

130 lines
4.7 KiB
Go
Raw Normal View History

2022-07-08 16:01:23 +00:00
package main
import (
"context"
"net"
"testing"
2022-07-10 11:17:44 +00:00
"time"
2022-07-08 16:01:23 +00:00
"github.com/sagernet/sing-box"
"github.com/sagernet/sing-box/option"
2022-08-16 15:37:51 +00:00
"github.com/sagernet/sing/common/bufio"
"github.com/sagernet/sing/common/debug"
2022-07-08 16:01:23 +00:00
M "github.com/sagernet/sing/common/metadata"
N "github.com/sagernet/sing/common/network"
"github.com/sagernet/sing/protocol/socks"
"github.com/stretchr/testify/require"
2022-09-23 05:14:31 +00:00
"go.uber.org/goleak"
2022-07-08 16:01:23 +00:00
)
2022-09-23 05:14:31 +00:00
func TestMain(m *testing.M) {
goleak.VerifyTestMain(m)
}
func startInstance(t *testing.T, options option.Options) *box.Box {
if debug.Enabled {
options.Log = &option.LogOptions{
Level: "trace",
}
2022-09-07 07:41:19 +00:00
} else {
options.Log = &option.LogOptions{
Level: "warning",
}
}
2022-09-23 05:14:31 +00:00
// ctx := context.Background()
ctx, cancel := context.WithCancel(context.Background())
2022-08-01 04:23:34 +00:00
var instance *box.Box
2022-07-18 12:40:14 +00:00
var err error
for retry := 0; retry < 3; retry++ {
instance, err = box.New(box.Options{
Context: ctx,
Options: options,
})
2022-07-18 12:40:14 +00:00
require.NoError(t, err)
err = instance.Start()
if err != nil {
2022-08-11 15:59:22 +00:00
time.Sleep(time.Second)
2022-07-18 12:40:14 +00:00
continue
}
2022-08-01 04:23:34 +00:00
break
2022-07-18 12:40:14 +00:00
}
2022-07-08 16:01:23 +00:00
require.NoError(t, err)
2022-08-01 04:23:34 +00:00
t.Cleanup(func() {
instance.Close()
2022-09-23 05:14:31 +00:00
cancel()
2022-08-01 04:23:34 +00:00
})
2022-09-23 05:14:31 +00:00
return instance
2022-07-08 16:01:23 +00:00
}
func testSuit(t *testing.T, clientPort uint16, testPort uint16) {
dialer := socks.NewClient(N.SystemDialer, M.ParseSocksaddrHostPort("127.0.0.1", clientPort), socks.Version5, "", "")
dialTCP := func() (net.Conn, error) {
return dialer.DialContext(context.Background(), "tcp", M.ParseSocksaddrHostPort("127.0.0.1", testPort))
}
dialUDP := func() (net.PacketConn, error) {
return dialer.ListenPacket(context.Background(), M.ParseSocksaddrHostPort("127.0.0.1", testPort))
}
2022-09-12 09:30:54 +00:00
require.NoError(t, testPingPongWithConn(t, testPort, dialTCP))
require.NoError(t, testPingPongWithPacketConn(t, testPort, dialUDP))
require.NoError(t, testLargeDataWithConn(t, testPort, dialTCP))
require.NoError(t, testLargeDataWithPacketConn(t, testPort, dialUDP))
// require.NoError(t, testPacketConnTimeout(t, dialUDP))
}
2022-08-10 12:19:16 +00:00
func testTCP(t *testing.T, clientPort uint16, testPort uint16) {
dialer := socks.NewClient(N.SystemDialer, M.ParseSocksaddrHostPort("127.0.0.1", clientPort), socks.Version5, "", "")
dialTCP := func() (net.Conn, error) {
return dialer.DialContext(context.Background(), "tcp", M.ParseSocksaddrHostPort("127.0.0.1", testPort))
}
2022-08-22 04:02:16 +00:00
require.NoError(t, testPingPongWithConn(t, testPort, dialTCP))
2022-08-24 02:21:56 +00:00
require.NoError(t, testLargeDataWithConn(t, testPort, dialTCP))
2022-08-10 12:19:16 +00:00
}
2022-08-29 02:10:41 +00:00
func testSuitSimple(t *testing.T, clientPort uint16, testPort uint16) {
2022-07-08 16:01:23 +00:00
dialer := socks.NewClient(N.SystemDialer, M.ParseSocksaddrHostPort("127.0.0.1", clientPort), socks.Version5, "", "")
2022-07-18 04:32:31 +00:00
dialTCP := func() (net.Conn, error) {
return dialer.DialContext(context.Background(), "tcp", M.ParseSocksaddrHostPort("127.0.0.1", testPort))
2022-07-08 16:01:23 +00:00
}
2022-07-18 04:32:31 +00:00
dialUDP := func() (net.PacketConn, error) {
return dialer.ListenPacket(context.Background(), M.ParseSocksaddrHostPort("127.0.0.1", testPort))
2022-07-08 16:01:23 +00:00
}
require.NoError(t, testPingPongWithConn(t, testPort, dialTCP))
require.NoError(t, testPingPongWithPacketConn(t, testPort, dialUDP))
2022-09-12 09:30:54 +00:00
require.NoError(t, testPingPongWithConn(t, testPort, dialTCP))
require.NoError(t, testPingPongWithPacketConn(t, testPort, dialUDP))
2022-08-16 15:37:51 +00:00
}
2022-09-13 02:41:10 +00:00
func testSuitSimple1(t *testing.T, clientPort uint16, testPort uint16) {
dialer := socks.NewClient(N.SystemDialer, M.ParseSocksaddrHostPort("127.0.0.1", clientPort), socks.Version5, "", "")
dialTCP := func() (net.Conn, error) {
return dialer.DialContext(context.Background(), "tcp", M.ParseSocksaddrHostPort("127.0.0.1", testPort))
}
dialUDP := func() (net.PacketConn, error) {
return dialer.ListenPacket(context.Background(), M.ParseSocksaddrHostPort("127.0.0.1", testPort))
}
require.NoError(t, testPingPongWithConn(t, testPort, dialTCP))
require.NoError(t, testPingPongWithPacketConn(t, testPort, dialUDP))
require.NoError(t, testPingPongWithConn(t, testPort, dialTCP))
require.NoError(t, testLargeDataWithPacketConn(t, testPort, dialUDP))
}
2022-08-16 15:37:51 +00:00
func testSuitWg(t *testing.T, clientPort uint16, testPort uint16) {
dialer := socks.NewClient(N.SystemDialer, M.ParseSocksaddrHostPort("127.0.0.1", clientPort), socks.Version5, "", "")
dialTCP := func() (net.Conn, error) {
return dialer.DialContext(context.Background(), "tcp", M.ParseSocksaddrHostPort("10.0.0.1", testPort))
}
dialUDP := func() (net.PacketConn, error) {
conn, err := dialer.DialContext(context.Background(), "udp", M.ParseSocksaddrHostPort("10.0.0.1", testPort))
if err != nil {
return nil, err
}
return bufio.NewUnbindPacketConn(conn), nil
}
2022-09-13 02:41:10 +00:00
require.NoError(t, testPingPongWithConn(t, testPort, dialTCP))
require.NoError(t, testPingPongWithPacketConn(t, testPort, dialUDP))
2022-08-01 04:23:34 +00:00
require.NoError(t, testLargeDataWithConn(t, testPort, dialTCP))
require.NoError(t, testLargeDataWithPacketConn(t, testPort, dialUDP))
2022-07-08 16:01:23 +00:00
}