package main import ( "context" "net" "testing" "time" "github.com/sagernet/sing-box" "github.com/sagernet/sing-box/option" "github.com/sagernet/sing/common/control" F "github.com/sagernet/sing/common/format" 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" ) func mkPort(t *testing.T) uint16 { var lc net.ListenConfig lc.Control = control.ReuseAddr() for { tcpListener, err := lc.Listen(context.Background(), "tcp", ":0") require.NoError(t, err) listenPort := M.SocksaddrFromNet(tcpListener.Addr()).Port tcpListener.Close() udpListener, err := lc.Listen(context.Background(), "tcp", F.ToString(":", listenPort)) if err != nil { continue } udpListener.Close() return listenPort } } func startInstance(t *testing.T, options option.Options) { var instance *box.Box var err error for retry := 0; retry < 3; retry++ { instance, err = box.New(context.Background(), options) require.NoError(t, err) err = instance.Start() if err != nil { time.Sleep(5 * time.Millisecond) continue } break } require.NoError(t, err) t.Cleanup(func() { time.Sleep(500 * time.Millisecond) instance.Close() }) } 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)) } /*t.Run("tcp", func(t *testing.T) { t.Parallel() var err error for retry := 0; retry < 3; retry++ { err = testLargeDataWithConn(t, testPort, dialTCP) if err == nil { break } } require.NoError(t, err) }) t.Run("udp", func(t *testing.T) { t.Parallel() var err error for retry := 0; retry < 3; retry++ { err = testLargeDataWithPacketConn(t, testPort, dialUDP) if err == nil { break } } require.NoError(t, err) })*/ require.NoError(t, testLargeDataWithConn(t, testPort, dialTCP)) require.NoError(t, testLargeDataWithPacketConn(t, testPort, dialUDP)) // require.NoError(t, testPingPongWithConn(t, testPort, dialTCP)) // require.NoError(t, testPingPongWithPacketConn(t, testPort, dialUDP)) // require.NoError(t, testPacketConnTimeout(t, dialUDP)) }