func Test_listenWSAndDial_TLS(t *testing.T) { assert := assert.On(t) go func() { <-time.After(time.Second * 5) assert.Fail("Too slow") }() listen, err := ListenWS(v2net.DomainAddress("localhost"), 13143, internet.ListenOptions{ Stream: &internet.StreamConfig{ SecurityType: serial.GetMessageType(new(v2tls.Config)), SecuritySettings: []*serial.TypedMessage{serial.ToTypedMessage(&v2tls.Config{ Certificate: []*v2tls.Certificate{tlsgen.GenerateCertificateForTest()}, })}, Protocol: internet.TransportProtocol_WebSocket, TransportSettings: []*internet.TransportConfig{ { Protocol: internet.TransportProtocol_WebSocket, Settings: serial.ToTypedMessage(&Config{ Path: "wss", ConnectionReuse: &ConnectionReuse{ Enable: true, }, }), }, }, }, }) assert.Error(err).IsNil() go func() { conn, err := listen.Accept() assert.Error(err).IsNil() conn.Close() listen.Close() }() ctx := internet.ContextWithTransportSettings(context.Background(), &Config{ Path: "wss", ConnectionReuse: &ConnectionReuse{ Enable: true, }, }) ctx = internet.ContextWithSecuritySettings(ctx, &v2tls.Config{ AllowInsecure: true, }) conn, err := Dial(ctx, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13143)) assert.Error(err).IsNil() conn.Close() }
func Test_listenWSAndDial(t *testing.T) { assert := assert.On(t) listen, err := ListenWS(v2net.DomainAddress("localhost"), 13146, internet.ListenOptions{ Stream: &internet.StreamConfig{ Protocol: internet.TransportProtocol_WebSocket, TransportSettings: []*internet.TransportConfig{ { Protocol: internet.TransportProtocol_WebSocket, Settings: serial.ToTypedMessage(&Config{ Path: "ws", }), }, }, }, }) assert.Error(err).IsNil() go func() { for { conn, err := listen.Accept() if err != nil { break } go func() { defer conn.Close() var b [1024]byte n, err := conn.Read(b[:]) //assert.Error(err).IsNil() if err != nil { conn.SetReusable(false) return } assert.Bool(bytes.HasPrefix(b[:n], []byte("Test connection"))).IsTrue() _, err = conn.Write([]byte("Response")) assert.Error(err).IsNil() }() } }() ctx := internet.ContextWithTransportSettings(context.Background(), &Config{Path: "ws"}) conn, err := Dial(ctx, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13146)) assert.Error(err).IsNil() _, err = conn.Write([]byte("Test connection 1")) assert.Error(err).IsNil() var b [1024]byte n, err := conn.Read(b[:]) assert.Error(err).IsNil() assert.String(string(b[:n])).Equals("Response") assert.Error(conn.Close()).IsNil() <-time.After(time.Second * 5) conn, err = Dial(ctx, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13146)) assert.Error(err).IsNil() _, err = conn.Write([]byte("Test connection 2")) assert.Error(err).IsNil() n, err = conn.Read(b[:]) assert.Error(err).IsNil() assert.String(string(b[:n])).Equals("Response") assert.Error(conn.Close()).IsNil() <-time.After(time.Second * 15) conn, err = Dial(ctx, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13146)) assert.Error(err).IsNil() _, err = conn.Write([]byte("Test connection 3")) assert.Error(err).IsNil() n, err = conn.Read(b[:]) assert.Error(err).IsNil() assert.String(string(b[:n])).Equals("Response") assert.Error(conn.Close()).IsNil() assert.Error(listen.Close()).IsNil() }
func TestDialAndListen(t *testing.T) { assert := assert.On(t) listerner, err := NewListener(v2net.LocalHostIP, v2net.Port(0), internet.ListenOptions{ Stream: &internet.StreamConfig{ Protocol: internet.TransportProtocol_MKCP, TransportSettings: []*internet.TransportConfig{ { Protocol: internet.TransportProtocol_MKCP, Settings: serial.ToTypedMessage(&Config{}), }, }, }, }) assert.Error(err).IsNil() port := v2net.Port(listerner.Addr().(*net.UDPAddr).Port) go func() { for { conn, err := listerner.Accept() if err != nil { break } go func() { payload := make([]byte, 4096) for { nBytes, err := conn.Read(payload) if err != nil { break } for idx, b := range payload[:nBytes] { payload[idx] = b ^ 'c' } conn.Write(payload[:nBytes]) } conn.Close() }() } }() ctx := internet.ContextWithTransportSettings(context.Background(), &Config{}) wg := new(sync.WaitGroup) for i := 0; i < 10; i++ { clientConn, err := DialKCP(ctx, v2net.UDPDestination(v2net.LocalHostIP, port)) assert.Error(err).IsNil() wg.Add(1) go func() { clientSend := make([]byte, 1024*1024) rand.Read(clientSend) go clientConn.Write(clientSend) clientReceived := make([]byte, 1024*1024) nBytes, _ := io.ReadFull(clientConn, clientReceived) assert.Int(nBytes).Equals(len(clientReceived)) clientConn.Close() clientExpected := make([]byte, 1024*1024) for idx, b := range clientSend { clientExpected[idx] = b ^ 'c' } assert.Bytes(clientReceived).Equals(clientExpected) wg.Done() }() } wg.Wait() for i := 0; i < 60 && listerner.ActiveConnections() > 0; i++ { time.Sleep(500 * time.Millisecond) } assert.Int(listerner.ActiveConnections()).Equals(0) listerner.Close() }