Files
v2ray_simple/udp_test.go
e1732a364fed f28f0d0bee 修订代码, 默认loglevel 改为 Log_info.
对一般用户而言,还是需要使用Info等级 来了解一下 一般的 日志情况,等到使用熟练之后,且确认运行没有错误后, 可以自行调为 warning 来提升性能

发现 bubble包 还自己引入了 命令行参数,这十分不可取,所以我们还是直接使用其代码。

将其它包中 的 命令行参数 统一 移动 到 cmd/verysimple 中;tls lazy 特性因为还在 调试阶段,所以 命令行参数 仍然放到 v2ray_simple 包中。
2022-04-26 13:22:18 +08:00

321 lines
7.2 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

package v2ray_simple
import (
"fmt"
"net"
"testing"
"github.com/e1732a364fed/v2ray_simple/netLayer"
"github.com/e1732a364fed/v2ray_simple/proxy"
"github.com/e1732a364fed/v2ray_simple/proxy/socks5"
"github.com/e1732a364fed/v2ray_simple/utils"
"github.com/miekg/dns"
)
func TestUDP_vless(t *testing.T) {
testUDP(t, "vless", 0, "tcp", false, false, false)
}
//v0 没有fullcone
func TestUDP_vless_v1(t *testing.T) {
testUDP(t, "vless", 1, "tcp", false, false, false)
}
func TestUDP_vless_v1_fullcone(t *testing.T) {
testUDP(t, "vless", 1, "tcp", false, true, false)
}
func TestUDP_vless_v1_udpMulti(t *testing.T) {
testUDP(t, "vless", 1, "tcp", true, false, false)
}
func TestUDP_vless_v1_udpMulti_fullcone(t *testing.T) {
testUDP(t, "vless", 1, "tcp", true, true, false)
}
func TestUDP_trojan(t *testing.T) {
testUDP(t, "trojan", 0, "tcp", false, false, false)
}
func TestUDP_trojan_mux(t *testing.T) {
testUDP(t, "trojan", 0, "tcp", false, false, true)
}
func TestUDP_trojan_fullcone(t *testing.T) {
testUDP(t, "trojan", 0, "tcp", false, true, false)
}
func TestUDP_trojan_through_udp(t *testing.T) {
testUDP(t, "trojan", 0, "udp", false, false, false)
}
//udp测试我们直接使用dns请求来测试.
func testUDP(t *testing.T, protocol string, version int, network string, multi bool, fullcone bool, mux bool) {
utils.LogLevel = utils.Log_debug
utils.InitLog()
//同时监听两个dokodemo, 发向不同raddr, 这样就可以模拟 多raddr目标时的 情况
//vless v1的udp_multi的dialfunc 需要单一 client 拨号多个raddr 才能被触发, 所以还要使用socks5测试两次
isTestInChina := utils.IsTimezoneCN()
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://%s: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(true, true)
clientListen2Port := netLayer.RandPortStr(true, true)
clientDialPort := netLayer.RandPortStr(true, false)
socks5Port, socks5PortStr := netLayer.RandPort_andStr(true, false)
var secondDnsServerStr string
if isTestInChina {
secondDnsServerStr = "114.114.114.114"
} else {
secondDnsServerStr = "1.1.1.1"
}
testClientConfStr := fmt.Sprintf(testClientConfFormatStr, clientListenPort,
clientListen2Port, secondDnsServerStr, 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 := proxy.LoadTomlConfStr(testClientConfStr)
if err != nil {
t.Log(err)
t.FailNow()
}
serverConf, err := proxy.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, nil)
ListenSer(clientEndInServer2, clientEndOutClient, nil)
ListenSer(clientEndInServer3, clientEndOutClient, nil)
ListenSer(serverEndInServer, serverEndOutClient, nil)
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)
}
}
if isTestInChina {
socks5ClientConn.WriteUDP_Target.IP = net.IPv4(114, 114, 114, 114)
} else {
socks5ClientConn.WriteUDP_Target.IP = net.IPv4(1, 1, 1, 1)
}
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)
}
}
}