1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-01-19 15:57:04 -05:00

test tfo on listener

This commit is contained in:
Darien Raymond 2018-09-10 21:19:59 +02:00
parent c8a12bc35d
commit b9e4a4882c
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
2 changed files with 12 additions and 3 deletions

View File

@ -1,12 +1,14 @@
package tcp package tcp
import ( import (
"context"
"fmt" "fmt"
"io" "io"
"v2ray.com/core/common/buf" "v2ray.com/core/common/buf"
"v2ray.com/core/common/net" "v2ray.com/core/common/net"
"v2ray.com/core/common/task" "v2ray.com/core/common/task"
"v2ray.com/core/transport/internet"
"v2ray.com/core/transport/pipe" "v2ray.com/core/transport/pipe"
) )
@ -20,14 +22,17 @@ type Server struct {
} }
func (server *Server) Start() (net.Destination, error) { func (server *Server) Start() (net.Destination, error) {
return server.StartContext(context.Background())
}
func (server *Server) StartContext(ctx context.Context) (net.Destination, error) {
listenerAddr := server.Listen listenerAddr := server.Listen
if listenerAddr == nil { if listenerAddr == nil {
listenerAddr = net.LocalHostIP listenerAddr = net.LocalHostIP
} }
listener, err := net.ListenTCP("tcp", &net.TCPAddr{ listener, err := internet.ListenSystemTCP(ctx, &net.TCPAddr{
IP: listenerAddr.IP(), IP: listenerAddr.IP(),
Port: int(server.Port), Port: int(server.Port),
Zone: "",
}) })
if err != nil { if err != nil {
return net.Destination{}, err return net.Destination{}, err

View File

@ -17,7 +17,11 @@ func TestTCPFastOpen(t *testing.T) {
return b return b
}, },
} }
dest, err := tcpServer.Start() dest, err := tcpServer.StartContext(ContextWithStreamSettings(context.Background(), &MemoryStreamConfig{
SocketSettings: &SocketConfig{
Tfo: SocketConfig_Enable,
},
}))
common.Must(err) common.Must(err)
defer tcpServer.Close() defer tcpServer.Close()