mirror of https://github.com/cbeuw/Cloak
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
741 B
Go
41 lines
741 B
Go
package common
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func BenchmarkTLSConn_Write(b *testing.B) {
|
|
const bufSize = 16 * 1024
|
|
addrCh := make(chan string, 1)
|
|
go func() {
|
|
listener, err := net.Listen("tcp", "127.0.0.1:0")
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
addrCh <- listener.Addr().String()
|
|
conn, err := listener.Accept()
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
readBuf := make([]byte, bufSize*2)
|
|
for {
|
|
_, err = conn.Read(readBuf)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
data := make([]byte, bufSize)
|
|
discardConn, _ := net.Dial("tcp", <-addrCh)
|
|
tlsConn := NewTLSConn(discardConn)
|
|
defer tlsConn.Close()
|
|
b.SetBytes(bufSize)
|
|
b.ResetTimer()
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
tlsConn.Write(data)
|
|
}
|
|
})
|
|
}
|