mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-06-29 18:45:23 +00:00
103 lines
2.7 KiB
Go
103 lines
2.7 KiB
Go
package websocket_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"v2ray.com/core/common/net"
|
|
tlsgen "v2ray.com/core/testing/tls"
|
|
"v2ray.com/core/transport/internet"
|
|
v2tls "v2ray.com/core/transport/internet/tls"
|
|
. "v2ray.com/core/transport/internet/websocket"
|
|
. "v2ray.com/ext/assert"
|
|
)
|
|
|
|
func Test_listenWSAndDial(t *testing.T) {
|
|
assert := With(t)
|
|
listen, err := ListenWS(internet.ContextWithTransportSettings(context.Background(), &Config{
|
|
Path: "ws",
|
|
}), net.DomainAddress("localhost"), 13146, func(ctx context.Context, conn internet.Connection) bool {
|
|
go func(c internet.Connection) {
|
|
defer c.Close()
|
|
|
|
var b [1024]byte
|
|
n, err := c.Read(b[:])
|
|
//assert(err, IsNil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
assert(bytes.HasPrefix(b[:n], []byte("Test connection")), IsTrue)
|
|
|
|
_, err = c.Write([]byte("Response"))
|
|
assert(err, IsNil)
|
|
}(conn)
|
|
return true
|
|
})
|
|
assert(err, IsNil)
|
|
|
|
ctx := internet.ContextWithTransportSettings(context.Background(), &Config{Path: "ws"})
|
|
conn, err := Dial(ctx, net.TCPDestination(net.DomainAddress("localhost"), 13146))
|
|
|
|
assert(err, IsNil)
|
|
_, err = conn.Write([]byte("Test connection 1"))
|
|
assert(err, IsNil)
|
|
|
|
var b [1024]byte
|
|
n, err := conn.Read(b[:])
|
|
assert(err, IsNil)
|
|
assert(string(b[:n]), Equals, "Response")
|
|
|
|
assert(conn.Close(), IsNil)
|
|
<-time.After(time.Second * 5)
|
|
conn, err = Dial(ctx, net.TCPDestination(net.DomainAddress("localhost"), 13146))
|
|
assert(err, IsNil)
|
|
_, err = conn.Write([]byte("Test connection 2"))
|
|
assert(err, IsNil)
|
|
n, err = conn.Read(b[:])
|
|
assert(err, IsNil)
|
|
assert(string(b[:n]), Equals, "Response")
|
|
assert(conn.Close(), IsNil)
|
|
<-time.After(time.Second * 15)
|
|
conn, err = Dial(ctx, net.TCPDestination(net.DomainAddress("localhost"), 13146))
|
|
assert(err, IsNil)
|
|
_, err = conn.Write([]byte("Test connection 3"))
|
|
assert(err, IsNil)
|
|
n, err = conn.Read(b[:])
|
|
assert(err, IsNil)
|
|
assert(string(b[:n]), Equals, "Response")
|
|
assert(conn.Close(), IsNil)
|
|
|
|
assert(listen.Close(), IsNil)
|
|
}
|
|
|
|
func Test_listenWSAndDial_TLS(t *testing.T) {
|
|
assert := With(t)
|
|
|
|
start := time.Now()
|
|
|
|
ctx := internet.ContextWithTransportSettings(context.Background(), &Config{
|
|
Path: "wss",
|
|
})
|
|
ctx = internet.ContextWithSecuritySettings(ctx, &v2tls.Config{
|
|
AllowInsecure: true,
|
|
Certificate: []*v2tls.Certificate{tlsgen.GenerateCertificateForTest()},
|
|
})
|
|
listen, err := ListenWS(ctx, net.DomainAddress("localhost"), 13143, func(ctx context.Context, conn internet.Connection) bool {
|
|
go func() {
|
|
_ = conn.Close()
|
|
}()
|
|
return true
|
|
})
|
|
assert(err, IsNil)
|
|
defer listen.Close()
|
|
|
|
conn, err := Dial(ctx, net.TCPDestination(net.DomainAddress("localhost"), 13143))
|
|
assert(err, IsNil)
|
|
_ = conn.Close()
|
|
|
|
end := time.Now()
|
|
assert(end.Before(start.Add(time.Second*5)), IsTrue)
|
|
}
|