mirror of
https://github.com/v2fly/v2ray-core.git
synced 2025-01-02 15:36:41 -05:00
fix sockopt tests
This commit is contained in:
parent
af65048868
commit
b1e900a506
@ -1,48 +0,0 @@
|
|||||||
package internet_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"syscall"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"v2ray.com/core/common"
|
|
||||||
"v2ray.com/core/common/net"
|
|
||||||
"v2ray.com/core/testing/servers/tcp"
|
|
||||||
. "v2ray.com/core/transport/internet"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestSockOptMark(t *testing.T) {
|
|
||||||
t.Skip("requires CAP_NET_ADMIN")
|
|
||||||
|
|
||||||
tcpServer := tcp.Server{
|
|
||||||
MsgProcessor: func(b []byte) []byte {
|
|
||||||
return b
|
|
||||||
},
|
|
||||||
}
|
|
||||||
dest, err := tcpServer.Start()
|
|
||||||
common.Must(err)
|
|
||||||
defer tcpServer.Close()
|
|
||||||
|
|
||||||
const mark = 1
|
|
||||||
ctx := context.Background()
|
|
||||||
ctx = ContextWithStreamSettings(ctx, &MemoryStreamConfig{
|
|
||||||
SocketSettings: &SocketConfig{
|
|
||||||
Mark: mark,
|
|
||||||
},
|
|
||||||
})
|
|
||||||
dialer := DefaultSystemDialer{}
|
|
||||||
conn, err := dialer.Dial(ctx, nil, dest)
|
|
||||||
common.Must(err)
|
|
||||||
defer conn.Close()
|
|
||||||
|
|
||||||
rawConn, err := conn.(*net.TCPConn).SyscallConn()
|
|
||||||
common.Must(err)
|
|
||||||
err = rawConn.Control(func(fd uintptr) {
|
|
||||||
m, err := syscall.GetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_MARK)
|
|
||||||
common.Must(err)
|
|
||||||
if mark != m {
|
|
||||||
t.Fatal("unexpected conneciton mark", m, " want ", mark)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
common.Must(err)
|
|
||||||
}
|
|
@ -2,16 +2,18 @@ package internet_test
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"syscall"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"v2ray.com/core/common"
|
"v2ray.com/core/common"
|
||||||
"v2ray.com/core/common/buf"
|
"v2ray.com/core/common/buf"
|
||||||
"v2ray.com/core/common/compare"
|
"v2ray.com/core/common/compare"
|
||||||
|
"v2ray.com/core/common/net"
|
||||||
"v2ray.com/core/testing/servers/tcp"
|
"v2ray.com/core/testing/servers/tcp"
|
||||||
. "v2ray.com/core/transport/internet"
|
. "v2ray.com/core/transport/internet"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestSockOptMark(t *testing.T) {
|
func TestTCPFastOpen(t *testing.T) {
|
||||||
tcpServer := tcp.Server{
|
tcpServer := tcp.Server{
|
||||||
MsgProcessor: func(b []byte) []byte {
|
MsgProcessor: func(b []byte) []byte {
|
||||||
return b
|
return b
|
||||||
@ -41,3 +43,39 @@ func TestSockOptMark(t *testing.T) {
|
|||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestSockOptMark(t *testing.T) {
|
||||||
|
t.Skip("requires CAP_NET_ADMIN")
|
||||||
|
|
||||||
|
tcpServer := tcp.Server{
|
||||||
|
MsgProcessor: func(b []byte) []byte {
|
||||||
|
return b
|
||||||
|
},
|
||||||
|
}
|
||||||
|
dest, err := tcpServer.Start()
|
||||||
|
common.Must(err)
|
||||||
|
defer tcpServer.Close()
|
||||||
|
|
||||||
|
const mark = 1
|
||||||
|
ctx := context.Background()
|
||||||
|
ctx = ContextWithStreamSettings(ctx, &MemoryStreamConfig{
|
||||||
|
SocketSettings: &SocketConfig{
|
||||||
|
Mark: mark,
|
||||||
|
},
|
||||||
|
})
|
||||||
|
dialer := DefaultSystemDialer{}
|
||||||
|
conn, err := dialer.Dial(ctx, nil, dest)
|
||||||
|
common.Must(err)
|
||||||
|
defer conn.Close()
|
||||||
|
|
||||||
|
rawConn, err := conn.(*net.TCPConn).SyscallConn()
|
||||||
|
common.Must(err)
|
||||||
|
err = rawConn.Control(func(fd uintptr) {
|
||||||
|
m, err := syscall.GetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_MARK)
|
||||||
|
common.Must(err)
|
||||||
|
if mark != m {
|
||||||
|
t.Fatal("unexpected conneciton mark", m, " want ", mark)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
common.Must(err)
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user