xmengnet/mihomo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: disable tcp_test

Browse source
This commit is contained in:
metacubex 2022-08-15 15:52:03 +08:00
parent b725c91b05
commit 4611fbfe0c
1 changed files with 176 additions and 180 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

356
transport/hysteria/conns/faketcp/tcp_test.go
View file

@ -4,11 +4,7 @@
package faketcp package faketcp
import ( import (
"log"
"net"
"net/http"
_ "net/http/pprof" _ "net/http/pprof"
"testing"
) )
//const testPortStream = "127.0.0.1:3456" //const testPortStream = "127.0.0.1:3456"
@ -18,179 +14,179 @@ const testPortStream = "127.0.0.1:3456"
const portServerPacket = "[::]:3457" const portServerPacket = "[::]:3457"
const portRemotePacket = "127.0.0.1:3457" const portRemotePacket = "127.0.0.1:3457"
func init() { //func init() {
startTCPServer() // startTCPServer()
startTCPRawServer() // startTCPRawServer()
go func() { // go func() {
log.Println(http.ListenAndServe("0.0.0.0:6060", nil)) // log.Println(http.ListenAndServe("0.0.0.0:6060", nil))
}() // }()
} //}
//
func startTCPServer() net.Listener { //func startTCPServer() net.Listener {
l, err := net.Listen("tcp", testPortStream) // l, err := net.Listen("tcp", testPortStream)
if err != nil { // if err != nil {
log.Panicln(err) // log.Panicln(err)
} // }
//
go func() { // go func() {
defer l.Close() // defer l.Close()
for { // for {
conn, err := l.Accept() // conn, err := l.Accept()
if err != nil { // if err != nil {
log.Println(err) // log.Println(err)
return // return
} // }
//
go handleRequest(conn) // go handleRequest(conn)
} // }
}() // }()
return l // return l
} //}
//
func startTCPRawServer() *TCPConn { //func startTCPRawServer() *TCPConn {
conn, err := Listen("tcp", portServerPacket) // conn, err := Listen("tcp", portServerPacket)
if err != nil { // if err != nil {
log.Panicln(err) // log.Panicln(err)
} // }
err = conn.SetReadBuffer(1024 * 1024) // err = conn.SetReadBuffer(1024 * 1024)
if err != nil { // if err != nil {
log.Println(err) // log.Println(err)
} // }
err = conn.SetWriteBuffer(1024 * 1024) // err = conn.SetWriteBuffer(1024 * 1024)
if err != nil { // if err != nil {
log.Println(err) // log.Println(err)
} // }
//
go func() { // go func() {
defer conn.Close() // defer conn.Close()
buf := make([]byte, 1024) // buf := make([]byte, 1024)
for { // for {
n, addr, err := conn.ReadFrom(buf) // n, addr, err := conn.ReadFrom(buf)
if err != nil { // if err != nil {
log.Println("server readfrom:", err) // log.Println("server readfrom:", err)
return // return
} // }
//echo // //echo
n, err = conn.WriteTo(buf[:n], addr) // n, err = conn.WriteTo(buf[:n], addr)
if err != nil { // if err != nil {
log.Println("server writeTo:", err) // log.Println("server writeTo:", err)
return // return
} // }
} // }
}() // }()
return conn // return conn
} //}
//
func handleRequest(conn net.Conn) { //func handleRequest(conn net.Conn) {
defer conn.Close() // defer conn.Close()
//
for { // for {
buf := make([]byte, 1024) // buf := make([]byte, 1024)
size, err := conn.Read(buf) // size, err := conn.Read(buf)
if err != nil { // if err != nil {
log.Println("handleRequest:", err) // log.Println("handleRequest:", err)
return // return
} // }
data := buf[:size] // data := buf[:size]
conn.Write(data) // conn.Write(data)
} // }
} //}
//
func TestDialTCPStream(t *testing.T) { //func TestDialTCPStream(t *testing.T) {
conn, err := Dial("tcp", testPortStream) // conn, err := Dial("tcp", testPortStream)
if err != nil { // if err != nil {
t.Fatal(err) // t.Fatal(err)
} // }
defer conn.Close() // defer conn.Close()
//
addr, err := net.ResolveTCPAddr("tcp", testPortStream) // addr, err := net.ResolveTCPAddr("tcp", testPortStream)
if err != nil { // if err != nil {
t.Fatal(err) // t.Fatal(err)
} // }
//
n, err := conn.WriteTo([]byte("abc"), addr) // n, err := conn.WriteTo([]byte("abc"), addr)
if err != nil { // if err != nil {
t.Fatal(n, err) // t.Fatal(n, err)
} // }
//
buf := make([]byte, 1024) // buf := make([]byte, 1024)
if n, addr, err := conn.ReadFrom(buf); err != nil { // if n, addr, err := conn.ReadFrom(buf); err != nil {
t.Fatal(n, addr, err) // t.Fatal(n, addr, err)
} else { // } else {
log.Println(string(buf[:n]), "from:", addr) // log.Println(string(buf[:n]), "from:", addr)
} // }
} //}
//
func TestDialToTCPPacket(t *testing.T) { //func TestDialToTCPPacket(t *testing.T) {
conn, err := Dial("tcp", portRemotePacket) // conn, err := Dial("tcp", portRemotePacket)
if err != nil { // if err != nil {
t.Fatal(err) // t.Fatal(err)
} // }
defer conn.Close() // defer conn.Close()
//
addr, err := net.ResolveTCPAddr("tcp", portRemotePacket) // addr, err := net.ResolveTCPAddr("tcp", portRemotePacket)
if err != nil { // if err != nil {
t.Fatal(err) // t.Fatal(err)
} // }
//
n, err := conn.WriteTo([]byte("abc"), addr) // n, err := conn.WriteTo([]byte("abc"), addr)
if err != nil { // if err != nil {
t.Fatal(n, err) // t.Fatal(n, err)
} // }
log.Println("written") // log.Println("written")
//
buf := make([]byte, 1024) // buf := make([]byte, 1024)
log.Println("readfrom buf") // log.Println("readfrom buf")
if n, addr, err := conn.ReadFrom(buf); err != nil { // if n, addr, err := conn.ReadFrom(buf); err != nil {
log.Println(err) // log.Println(err)
t.Fatal(n, addr, err) // t.Fatal(n, addr, err)
} else { // } else {
log.Println(string(buf[:n]), "from:", addr) // log.Println(string(buf[:n]), "from:", addr)
} // }
//
log.Println("complete") // log.Println("complete")
} //}
//
func TestSettings(t *testing.T) { //func TestSettings(t *testing.T) {
conn, err := Dial("tcp", portRemotePacket) // conn, err := Dial("tcp", portRemotePacket)
if err != nil { // if err != nil {
t.Fatal(err) // t.Fatal(err)
} // }
defer conn.Close() // defer conn.Close()
if err := conn.SetDSCP(46); err != nil { // if err := conn.SetDSCP(46); err != nil {
log.Fatal("SetDSCP:", err) // log.Fatal("SetDSCP:", err)
} // }
if err := conn.SetReadBuffer(4096); err != nil { // if err := conn.SetReadBuffer(4096); err != nil {
log.Fatal("SetReaderBuffer:", err) // log.Fatal("SetReaderBuffer:", err)
} // }
if err := conn.SetWriteBuffer(4096); err != nil { // if err := conn.SetWriteBuffer(4096); err != nil {
log.Fatal("SetWriteBuffer:", err) // log.Fatal("SetWriteBuffer:", err)
} // }
} //}
//
func BenchmarkEcho(b *testing.B) { //func BenchmarkEcho(b *testing.B) {
conn, err := Dial("tcp", portRemotePacket) // conn, err := Dial("tcp", portRemotePacket)
if err != nil { // if err != nil {
b.Fatal(err) // b.Fatal(err)
} // }
defer conn.Close() // defer conn.Close()
//
addr, err := net.ResolveTCPAddr("tcp", portRemotePacket) // addr, err := net.ResolveTCPAddr("tcp", portRemotePacket)
if err != nil { // if err != nil {
b.Fatal(err) // b.Fatal(err)
} // }
//
buf := make([]byte, 1024) // buf := make([]byte, 1024)
b.ReportAllocs() // b.ReportAllocs()
b.SetBytes(int64(len(buf))) // b.SetBytes(int64(len(buf)))
for i := 0; i < b.N; i++ { // for i := 0; i < b.N; i++ {
n, err := conn.WriteTo(buf, addr) // n, err := conn.WriteTo(buf, addr)
if err != nil { // if err != nil {
b.Fatal(n, err) // b.Fatal(n, err)
} // }
//
if n, addr, err := conn.ReadFrom(buf); err != nil { // if n, addr, err := conn.ReadFrom(buf); err != nil {
b.Fatal(n, addr, err) // b.Fatal(n, addr, err)
} // }
} // }
} //}