mirror of
https://github.com/e1732a364fed/v2ray_simple.git
synced 2025-10-31 20:12:45 +08:00
343 lines
7.7 KiB
Go
343 lines
7.7 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"net"
|
||
"testing"
|
||
|
||
"github.com/hahahrfool/v2ray_simple/netLayer"
|
||
"github.com/hahahrfool/v2ray_simple/proxy"
|
||
"github.com/hahahrfool/v2ray_simple/proxy/socks5"
|
||
"github.com/hahahrfool/v2ray_simple/utils"
|
||
"github.com/miekg/dns"
|
||
)
|
||
|
||
/*
|
||
nc 模拟dns请求
|
||
https://unix.stackexchange.com/questions/600194/create-dns-query-with-netcat-or-dev-udp
|
||
|
||
echo cfc9 0100 0001 0000 0000 0000 0a64 7563 6b64 7563 6b67 6f03 636f 6d00 0001 0001 |
|
||
xxd -p -r | nc -u -v 114.114.114.114 53
|
||
|
||
不过为了灵活我们还是引用 miekg/dns 包
|
||
参考 https://zhengyinyong.com/post/go-dns-library/
|
||
|
||
虽然net.Resolver也能用，
|
||
https://stackoverflow.com/questions/59889882/specifying-dns-server-for-lookup-in-go
|
||
|
||
但是我还是喜欢 miekg/dns;
|
||
|
||
func TestDNSLookup_CN(t *testing.T) {
|
||
m := new(dns.Msg)
|
||
m.SetQuestion(dns.Fqdn("www.qq.com"), dns.TypeA)
|
||
c := new(dns.Client)
|
||
|
||
r, _, err := c.Exchange(m, "114.114.114.114:53")
|
||
if r == nil {
|
||
t.Log("*** error: ", err.Error())
|
||
t.FailNow()
|
||
}
|
||
|
||
if r.Rcode != dns.RcodeSuccess {
|
||
t.Log("*** err2 ", r.Rcode, r)
|
||
t.FailNow()
|
||
}
|
||
|
||
for _, a := range r.Answer {
|
||
t.Log(a)
|
||
}
|
||
}
|
||
*/
|
||
|
||
func TestUDP_vless(t *testing.T) {
|
||
testUDP("vless", 0, "tcp", false, false, false, t)
|
||
}
|
||
|
||
//v0 没有fullcone
|
||
|
||
func TestUDP_vless_v1(t *testing.T) {
|
||
testUDP("vless", 1, "tcp", false, false, false, t)
|
||
}
|
||
|
||
func TestUDP_vless_v1_fullcone(t *testing.T) {
|
||
testUDP("vless", 1, "tcp", false, true, false, t)
|
||
}
|
||
|
||
func TestUDP_vless_v1_udpMulti(t *testing.T) {
|
||
testUDP("vless", 1, "tcp", true, false, false, t)
|
||
}
|
||
|
||
func TestUDP_vless_v1_udpMulti_fullcone(t *testing.T) {
|
||
testUDP("vless", 1, "tcp", true, true, false, t)
|
||
}
|
||
|
||
func TestUDP_trojan(t *testing.T) {
|
||
testUDP("trojan", 0, "tcp", false, false, false, t)
|
||
}
|
||
|
||
func TestUDP_trojan_mux(t *testing.T) {
|
||
testUDP("trojan", 0, "tcp", false, false, true, t)
|
||
}
|
||
|
||
func TestUDP_trojan_fullcone(t *testing.T) {
|
||
testUDP("trojan", 0, "tcp", false, true, false, t)
|
||
}
|
||
|
||
func TestUDP_trojan_through_udp(t *testing.T) {
|
||
testUDP("trojan", 0, "udp", false, false, false, t)
|
||
}
|
||
|
||
//udp测试我们直接使用dns请求来测试.
|
||
func testUDP(protocol string, version int, network string, multi bool, fullcone bool, mux bool, t *testing.T) {
|
||
utils.LogLevel = utils.Log_debug
|
||
utils.InitLog()
|
||
|
||
//同时监听两个dokodemo, 发向不同raddr, 这样就可以模拟 多raddr目标时的 情况
|
||
//vless v1的udp_multi的dialfunc 需要单一 client 拨号多个raddr 才能被触发, 所以还要使用socks5测试两次
|
||
|
||
var testClientConfFormatStr = `
|
||
[[listen]]
|
||
protocol = "dokodemo"
|
||
network = "udp"
|
||
host = "127.0.0.1"
|
||
port = %s
|
||
target = "udp://8.8.8.8:53"
|
||
|
||
[[listen]]
|
||
protocol = "dokodemo"
|
||
network = "udp"
|
||
host = "127.0.0.1"
|
||
port = %s
|
||
target = "udp://114.114.114.114:53"
|
||
|
||
[[listen]]
|
||
protocol = "socks5"
|
||
host = "127.0.0.1"
|
||
port = %s
|
||
|
||
[[dial]]
|
||
protocol = "%s"
|
||
uuid = "a684455c-b14f-11ea-bf0d-42010aaa0003"
|
||
host = "127.0.0.1"
|
||
port = %s
|
||
version = %d
|
||
insecure = true
|
||
network = "%s"
|
||
`
|
||
if mux {
|
||
testClientConfFormatStr += "\nuse_mux = true\n"
|
||
}
|
||
|
||
if multi {
|
||
testClientConfFormatStr += "\nextra = { vless1_udp_multi = true }\n"
|
||
}
|
||
|
||
clientListenPort := netLayer.RandPortStr()
|
||
clientListen2Port := netLayer.RandPortStr()
|
||
clientDialPort := netLayer.RandPortStr()
|
||
socks5Port, socks5PortStr := netLayer.RandPort_andStr()
|
||
|
||
testClientConfStr := fmt.Sprintf(testClientConfFormatStr, clientListenPort,
|
||
clientListen2Port, socks5PortStr, protocol, clientDialPort, version, network)
|
||
|
||
var testServerConfFormatStr = `
|
||
[[listen]]
|
||
protocol = "%s"
|
||
uuid = "a684455c-b14f-11ea-bf0d-42010aaa0003"
|
||
host = "127.0.0.1"
|
||
port = %s
|
||
version = %d
|
||
insecure = true
|
||
cert = "cert.pem"
|
||
key = "cert.key"
|
||
network = "%s"
|
||
|
||
[[dial]]
|
||
protocol = "direct"
|
||
|
||
`
|
||
if fullcone {
|
||
testServerConfFormatStr += "fullcone = true"
|
||
|
||
}
|
||
|
||
testServerConfStr := fmt.Sprintf(testServerConfFormatStr, protocol, clientDialPort, version, network)
|
||
|
||
clientConf, err := LoadTomlConfStr(testClientConfStr)
|
||
if err != nil {
|
||
t.Log(err)
|
||
t.FailNow()
|
||
}
|
||
serverConf, err := LoadTomlConfStr(testServerConfStr)
|
||
if err != nil {
|
||
t.Log(err)
|
||
t.FailNow()
|
||
}
|
||
|
||
//先建立服务端监听和客户端监听，最后自定义dns查询 并导向 客户端的 dokodemo监听端口
|
||
|
||
//domodemo in
|
||
clientEndInServer, err := proxy.NewServer(clientConf.Listen[0])
|
||
if err != nil {
|
||
t.Log("can not create clientEndInServer: ", err)
|
||
t.FailNow()
|
||
}
|
||
|
||
//domodemo in2
|
||
clientEndInServer2, err := proxy.NewServer(clientConf.Listen[1])
|
||
if err != nil {
|
||
t.Log("can not create clientEndInServer: ", err)
|
||
t.FailNow()
|
||
}
|
||
|
||
//socks5 in
|
||
clientEndInServer3, err := proxy.NewServer(clientConf.Listen[2])
|
||
if err != nil {
|
||
t.Log("can not create clientEndInServer: ", err)
|
||
t.FailNow()
|
||
}
|
||
|
||
// vless out
|
||
clientEndOutClient, err := proxy.NewClient(clientConf.Dial[0])
|
||
if err != nil {
|
||
t.Log("can not create clientEndOutClient: ", err)
|
||
t.FailNow()
|
||
}
|
||
|
||
//vless in
|
||
serverEndInServer, err := proxy.NewServer(serverConf.Listen[0])
|
||
if err != nil {
|
||
t.Log("can not create serverEndInServer: ", err)
|
||
t.FailNow()
|
||
}
|
||
// direct out
|
||
serverEndOutClient, err := proxy.NewClient(serverConf.Dial[0])
|
||
if err != nil {
|
||
t.Log("can not create serverEndOutClient: ", err)
|
||
t.FailNow()
|
||
}
|
||
|
||
listenSer(clientEndInServer, clientEndOutClient, false)
|
||
listenSer(clientEndInServer2, clientEndOutClient, false)
|
||
listenSer(clientEndInServer3, clientEndOutClient, false)
|
||
listenSer(serverEndInServer, serverEndOutClient, false)
|
||
|
||
m := new(dns.Msg)
|
||
m.SetQuestion(dns.Fqdn("www.qq.com"), dns.TypeA)
|
||
c := new(dns.Client)
|
||
|
||
// server 1 测试 /////////////////////////////////////////
|
||
r, _, err := c.Exchange(m, "127.0.0.1:"+clientListenPort)
|
||
if r == nil {
|
||
t.Log("error: ", err.Error())
|
||
t.FailNow()
|
||
}
|
||
|
||
if r.Rcode != dns.RcodeSuccess {
|
||
t.Log("err2 ", r.Rcode, r)
|
||
t.FailNow()
|
||
}
|
||
|
||
for _, a := range r.Answer {
|
||
t.Log("header is", a.Header())
|
||
t.Log("string is", a.String())
|
||
t.Log("a is ", a)
|
||
|
||
if aa, ok := a.(*dns.A); ok {
|
||
t.Log("arecord is ", aa.A)
|
||
}
|
||
}
|
||
|
||
// server 2 测试 /////////////////////////////////////////
|
||
|
||
r, _, err = c.Exchange(m, "127.0.0.1:"+clientListen2Port)
|
||
if r == nil {
|
||
t.Log("test2, error: ", err.Error())
|
||
t.FailNow()
|
||
}
|
||
|
||
if r.Rcode != dns.RcodeSuccess {
|
||
t.Log("test2, err2 ", r.Rcode, r)
|
||
t.FailNow()
|
||
}
|
||
|
||
for _, a := range r.Answer {
|
||
t.Log("test2, header is", a.Header())
|
||
t.Log("test2, string is", a.String())
|
||
t.Log("test2, a is ", a)
|
||
|
||
if aa, ok := a.(*dns.A); ok {
|
||
t.Log("test2, arecord is ", aa.A)
|
||
}
|
||
}
|
||
|
||
// server 3 socks5 udp 测试 /////////////////////////////////////////
|
||
//向不同地址写入两次，以测试 vless v1 udp multi
|
||
|
||
socks5ClientConn := &socks5.ClientUDPConn{
|
||
ServerAddr: &net.TCPAddr{
|
||
IP: net.IPv4(127, 0, 0, 1),
|
||
Port: socks5Port,
|
||
},
|
||
WriteUDP_Target: &net.UDPAddr{
|
||
IP: net.IPv4(8, 8, 8, 8),
|
||
Port: 53,
|
||
},
|
||
}
|
||
err = socks5ClientConn.Associate()
|
||
if err != nil {
|
||
t.Log("test3, Associate error: ", err.Error())
|
||
t.FailNow()
|
||
}
|
||
|
||
r, _, err = c.ExchangeWithConn(m, &dns.Conn{
|
||
Conn: socks5ClientConn,
|
||
})
|
||
|
||
if r == nil {
|
||
t.Log("test3, error: ", err.Error())
|
||
t.FailNow()
|
||
}
|
||
|
||
if r.Rcode != dns.RcodeSuccess {
|
||
t.Log("test3, err2 ", r.Rcode, r)
|
||
t.FailNow()
|
||
}
|
||
|
||
for _, a := range r.Answer {
|
||
t.Log("test3, header is", a.Header())
|
||
t.Log("test3, string is", a.String())
|
||
t.Log("test3, a is ", a)
|
||
|
||
if aa, ok := a.(*dns.A); ok {
|
||
t.Log("test3, arecord is ", aa.A)
|
||
}
|
||
}
|
||
|
||
socks5ClientConn.WriteUDP_Target.IP = net.IPv4(114, 114, 114, 114)
|
||
|
||
r, _, err = c.ExchangeWithConn(m, &dns.Conn{
|
||
Conn: socks5ClientConn,
|
||
})
|
||
|
||
if r == nil {
|
||
t.Log("test3_2, error: ", err.Error())
|
||
t.FailNow()
|
||
}
|
||
|
||
if r.Rcode != dns.RcodeSuccess {
|
||
t.Log("test3_2, err2 ", r.Rcode, r)
|
||
t.FailNow()
|
||
}
|
||
|
||
for _, a := range r.Answer {
|
||
t.Log("test3_2, header is", a.Header())
|
||
t.Log("test3_2, string is", a.String())
|
||
t.Log("test3_2, a is ", a)
|
||
|
||
if aa, ok := a.(*dns.A); ok {
|
||
t.Log("test3_2, arecord is ", aa.A)
|
||
}
|
||
}
|
||
}
|