func Test_Connect_wss_guess_reuse(t *testing.T) { assert := assert.On(t) (&Config{Pto: "", Path: "", ConnectionReuse: true}).Apply() i := 3 for i != 0 { conn, err := Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("echo.websocket.org"), 443)) assert.Error(err).IsNil() conn.Write([]byte("echo")) s := make(chan int) go func() { buf := make([]byte, 4) conn.Read(buf) str := string(buf) if str != "echo" { assert.Fail("Data mismatch") } s <- 0 }() <-s if i == 0 { conn.SetDeadline(time.Now()) conn.SetReadDeadline(time.Now()) conn.SetWriteDeadline(time.Now()) conn.SetReusable(false) } conn.Close() i-- } }
func TestRecivingWindow(t *testing.T) { assert := assert.On(t) window := NewReceivingWindow(3) seg0 := &DataSegment{} seg1 := &DataSegment{} seg2 := &DataSegment{} seg3 := &DataSegment{} assert.Bool(window.Set(0, seg0)).IsTrue() assert.Pointer(window.RemoveFirst()).Equals(seg0) e := window.RemoveFirst() if e != nil { assert.Fail("Expecting nil.") } assert.Bool(window.Set(1, seg1)).IsTrue() assert.Bool(window.Set(2, seg2)).IsTrue() window.Advance() assert.Bool(window.Set(2, seg3)).IsTrue() assert.Pointer(window.RemoveFirst()).Equals(seg1) assert.Pointer(window.Remove(1)).Equals(seg2) assert.Pointer(window.Remove(2)).Equals(seg3) }
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_Connect_wss_guess(t *testing.T) { assert := assert.On(t) (&Config{Pto: "", Path: ""}).Apply() conn, err := Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("echo.websocket.org"), 443)) assert.Error(err).IsNil() conn.Write([]byte("echo")) s := make(chan int) go func() { buf := make([]byte, 4) conn.Read(buf) str := string(buf) if str != "echo" { assert.Fail("Data mismatch") } s <- 0 }() <-s conn.Close() }
func Test_listenWSAndDial_TLS(t *testing.T) { assert := assert.On(t) go func() { <-time.After(time.Second * 5) assert.Fail("Too slow") }() (&Config{Pto: "wss", Path: "wss", ConnectionReuse: true, DeveloperInsecureSkipVerify: true, PrivKey: "./../../../testing/tls/key.pem", Cert: "./../../../testing/tls/cert.pem"}).Apply() listen, err := ListenWS(v2net.DomainAddress("localhost"), 13143) assert.Error(err).IsNil() go func() { conn, err := listen.Accept() assert.Error(err).IsNil() conn.Close() listen.Close() }() conn, err := Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13143)) assert.Error(err).IsNil() conn.Close() }
func Test_listenWSAndDial_TLS(t *testing.T) { assert := assert.On(t) go func() { <-time.After(time.Second * 5) assert.Fail("Too slow") }() tlsSettings := &v2tls.Config{ AllowInsecure: true, Certificate: []*v2tls.Certificate{ { Certificate: ReadFile("./../../../testing/tls/cert.pem", assert), Key: ReadFile("./../../../testing/tls/key.pem", assert), }, }, } listen, err := ListenWS(v2net.DomainAddress("localhost"), 13143, internet.ListenOptions{ Stream: &internet.StreamConfig{ SecurityType: loader.GetType(new(v2tls.Config)), SecuritySettings: []*loader.TypedSettings{loader.NewTypedSettings(tlsSettings)}, Network: v2net.Network_WebSocket, NetworkSettings: []*internet.NetworkSettings{ { Network: v2net.Network_WebSocket, Settings: loader.NewTypedSettings(&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() }() conn, err := Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13143), internet.DialerOptions{ Stream: &internet.StreamConfig{ SecurityType: loader.GetType(new(v2tls.Config)), SecuritySettings: []*loader.TypedSettings{loader.NewTypedSettings(tlsSettings)}, Network: v2net.Network_WebSocket, NetworkSettings: []*internet.NetworkSettings{ { Network: v2net.Network_WebSocket, Settings: loader.NewTypedSettings(&Config{ Path: "wss", ConnectionReuse: &ConnectionReuse{ Enable: true, }, }), }, }, }, }) assert.Error(err).IsNil() conn.Close() }