v2ray_simple/tcp_test.go

package main

import (
	"crypto/tls"
	"fmt"
	"io/ioutil"
	"net/http"
	"net/url"
	"testing"

	"github.com/hahahrfool/v2ray_simple/netLayer"
	"github.com/hahahrfool/v2ray_simple/proxy"
	"github.com/hahahrfool/v2ray_simple/utils"
)

func TestTCP_vless(t *testing.T) {
	testTCP("vless", 0, "tcp", false, t)
}

func TestTCP_trojan(t *testing.T) {
	testTCP("trojan", 0, "tcp", false, t)
}

func TestTCP_trojan_mux(t *testing.T) {
	testTCP("trojan", 0, "tcp", true, t)
}

//tcp测试我们直接使用http请求来测试
func testTCP(protocol string, version int, network string, innermux bool, t *testing.T) {
	utils.LogLevel = utils.Log_debug
	utils.InitLog()

	var testClientConfFormatStr = `
[[listen]]
protocol = "http"
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 innermux {
		testClientConfFormatStr += "use_mux = true"
	}

	const 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"

`

	clientListenPort := netLayer.RandPortStr(true, false)
	clientDialPort := netLayer.RandPortStr(true, false)

	testClientConfStr := fmt.Sprintf(testClientConfFormatStr, clientListenPort, protocol, clientDialPort, version, network)

	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监听端口

	//http in
	clientEndInServer, err := proxy.NewServer(clientConf.Listen[0])
	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(serverEndInServer, serverEndOutClient, false)

	proxyurl := "http://127.0.0.1:" + clientListenPort

	url_proxy, e2 := url.Parse(proxyurl)
	if e2 != nil {
		fmt.Println("proxyurl given was wrong,", proxyurl, e2)
		return
	}

	client := &http.Client{
		Transport: &http.Transport{
			Proxy:           http.ProxyURL(url_proxy),
			TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
		},
	}

	//tryGetHttp(client, "http://www.baidu.com", t)
	//tryGetHttp(client, "https://www.qq.com", t)
	tryGetHttp(client, "http://captive.apple.com", t)
	tryGetHttp(client, "http://www.msftconnecttest.com/connecttest.txt", t)
}

func tryGetHttp(client *http.Client, path string, t *testing.T) {
	t.Log("start dial", path)
	resp, err := client.Get(path)
	if err != nil {
		t.Log("get http failed", err)
		t.FailNow()
	}

	t.Log("Got,start read")

	bs, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		t.Log("get http failed read", err)
		t.FailNow()
	}
	resp.Body.Close()

	t.Log("got len", len(bs))
	if len(bs) > 5 {
		t.Log("first 5:", string(bs[:5]))

	}

}