Hunter0x7c7
2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
//go:build linux
// +build linux
 
package tcp_test
 
import (
    "context"
    "strings"
    "testing"
 
    "github.com/v2fly/v2ray-core/v5/common"
    "github.com/v2fly/v2ray-core/v5/testing/servers/tcp"
    "github.com/v2fly/v2ray-core/v5/transport/internet"
    . "github.com/v2fly/v2ray-core/v5/transport/internet/tcp"
)
 
func TestGetOriginalDestination(t *testing.T) {
    tcpServer := tcp.Server{}
    dest, err := tcpServer.Start()
    common.Must(err)
    defer tcpServer.Close()
 
    config, err := internet.ToMemoryStreamConfig(nil)
    common.Must(err)
    conn, err := Dial(context.Background(), dest, config)
    common.Must(err)
    defer conn.Close()
 
    originalDest, err := GetOriginalDestination(conn)
    if !(dest == originalDest || strings.Contains(err.Error(), "failed to call getsockopt")) {
        t.Error("unexpected state")
    }
}