mirror of https://github.com/ginuerzh/gost
ginuerzh
6 years ago
6 changed files with 159 additions and 840 deletions
@ -1,229 +0,0 @@ |
|||
package gost |
|||
|
|||
import ( |
|||
"bytes" |
|||
"encoding/binary" |
|||
"fmt" |
|||
"io" |
|||
"net" |
|||
"net/url" |
|||
"time" |
|||
|
|||
"github.com/ginuerzh/gosocks5" |
|||
"github.com/go-log/log" |
|||
"github.com/shadowsocks/go-shadowsocks2/core" |
|||
) |
|||
|
|||
type shadow2Connector struct { |
|||
Cipher *url.Userinfo |
|||
} |
|||
|
|||
// Shadow2Connector creates a Connector for go-shadowsocks2 proxy client.
|
|||
// It accepts a cipher info for shadowsocks data encryption/decryption.
|
|||
// The cipher must not be nil.
|
|||
func Shadow2Connector(cipher *url.Userinfo) Connector { |
|||
return &shadow2Connector{Cipher: cipher} |
|||
} |
|||
|
|||
func (c *shadow2Connector) Connect(conn net.Conn, addr string, options ...ConnectOption) (net.Conn, error) { |
|||
opts := &ConnectOptions{} |
|||
for _, option := range options { |
|||
option(opts) |
|||
} |
|||
|
|||
timeout := opts.Timeout |
|||
if timeout <= 0 { |
|||
timeout = ConnectTimeout |
|||
} |
|||
|
|||
conn.SetDeadline(time.Now().Add(timeout)) |
|||
defer conn.SetDeadline(time.Time{}) |
|||
|
|||
socksAddr, err := gosocks5.NewAddr(addr) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
rawaddr := sPool.Get().([]byte) |
|||
defer sPool.Put(rawaddr) |
|||
|
|||
n, err := socksAddr.Encode(rawaddr) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
var method, password string |
|||
cp := opts.User |
|||
if cp == nil { |
|||
cp = c.Cipher |
|||
} |
|||
if cp != nil { |
|||
method = cp.Username() |
|||
password, _ = cp.Password() |
|||
} |
|||
|
|||
cipher, err := core.PickCipher(method, nil, password) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
conn = cipher.StreamConn(conn) |
|||
if _, err := conn.Write(rawaddr[:n]); err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
return conn, nil |
|||
} |
|||
|
|||
type shadow2Handler struct { |
|||
options *HandlerOptions |
|||
} |
|||
|
|||
// Shadow2Handler creates a server Handler for go-shadowsocks2 proxy server.
|
|||
func Shadow2Handler(opts ...HandlerOption) Handler { |
|||
h := &shadow2Handler{} |
|||
h.Init(opts...) |
|||
|
|||
return h |
|||
} |
|||
|
|||
func (h *shadow2Handler) Init(options ...HandlerOption) { |
|||
if h.options == nil { |
|||
h.options = &HandlerOptions{} |
|||
} |
|||
|
|||
for _, opt := range options { |
|||
opt(h.options) |
|||
} |
|||
} |
|||
|
|||
func (h *shadow2Handler) Handle(conn net.Conn) { |
|||
defer conn.Close() |
|||
|
|||
var method, password string |
|||
users := h.options.Users |
|||
if len(users) > 0 { |
|||
method = users[0].Username() |
|||
password, _ = users[0].Password() |
|||
} |
|||
|
|||
cipher, err := core.PickCipher(method, nil, password) |
|||
if err != nil { |
|||
log.Logf("[ss2] %s -> %s : %s", |
|||
conn.RemoteAddr(), conn.LocalAddr(), err) |
|||
return |
|||
} |
|||
|
|||
conn = cipher.StreamConn(conn) |
|||
conn.SetReadDeadline(time.Now().Add(ReadTimeout)) |
|||
|
|||
addr, err := readSocksAddr(conn) |
|||
if err != nil { |
|||
log.Logf("[ss2] %s -> %s : %s", |
|||
conn.RemoteAddr(), conn.LocalAddr(), err) |
|||
return |
|||
} |
|||
|
|||
// clear timer
|
|||
conn.SetReadDeadline(time.Time{}) |
|||
|
|||
host := addr.String() |
|||
log.Logf("[ss2] %s -> %s -> %s", |
|||
conn.RemoteAddr(), h.options.Node.String(), host) |
|||
|
|||
if !Can("tcp", host, h.options.Whitelist, h.options.Blacklist) { |
|||
log.Logf("[ss2] %s - %s : Unauthorized to tcp connect to %s", |
|||
conn.RemoteAddr(), conn.LocalAddr(), host) |
|||
return |
|||
} |
|||
|
|||
if h.options.Bypass.Contains(host) { |
|||
log.Logf("[ss2] %s - %s : Bypass %s", |
|||
conn.RemoteAddr(), conn.LocalAddr(), host) |
|||
return |
|||
} |
|||
|
|||
retries := 1 |
|||
if h.options.Chain != nil && h.options.Chain.Retries > 0 { |
|||
retries = h.options.Chain.Retries |
|||
} |
|||
if h.options.Retries > 0 { |
|||
retries = h.options.Retries |
|||
} |
|||
|
|||
var cc net.Conn |
|||
var route *Chain |
|||
for i := 0; i < retries; i++ { |
|||
route, err = h.options.Chain.selectRouteFor(host) |
|||
if err != nil { |
|||
log.Logf("[ss2] %s -> %s : %s", |
|||
conn.RemoteAddr(), conn.LocalAddr(), err) |
|||
continue |
|||
} |
|||
|
|||
buf := bytes.Buffer{} |
|||
fmt.Fprintf(&buf, "%s -> %s -> ", |
|||
conn.RemoteAddr(), h.options.Node.String()) |
|||
for _, nd := range route.route { |
|||
fmt.Fprintf(&buf, "%d@%s -> ", nd.ID, nd.String()) |
|||
} |
|||
fmt.Fprintf(&buf, "%s", host) |
|||
log.Log("[route]", buf.String()) |
|||
|
|||
cc, err = route.Dial(host, |
|||
TimeoutChainOption(h.options.Timeout), |
|||
HostsChainOption(h.options.Hosts), |
|||
ResolverChainOption(h.options.Resolver), |
|||
) |
|||
if err == nil { |
|||
break |
|||
} |
|||
log.Logf("[ss2] %s -> %s : %s", |
|||
conn.RemoteAddr(), conn.LocalAddr(), err) |
|||
} |
|||
|
|||
if err != nil { |
|||
return |
|||
} |
|||
defer cc.Close() |
|||
|
|||
log.Logf("[ss2] %s <-> %s", conn.RemoteAddr(), host) |
|||
transport(conn, cc) |
|||
log.Logf("[ss2] %s >-< %s", conn.RemoteAddr(), host) |
|||
} |
|||
|
|||
func readSocksAddr(r io.Reader) (*gosocks5.Addr, error) { |
|||
addr := &gosocks5.Addr{} |
|||
b := sPool.Get().([]byte) |
|||
defer sPool.Put(b) |
|||
|
|||
_, err := io.ReadFull(r, b[:1]) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
addr.Type = b[0] |
|||
|
|||
switch addr.Type { |
|||
case gosocks5.AddrIPv4: |
|||
_, err = io.ReadFull(r, b[:net.IPv4len]) |
|||
addr.Host = net.IP(b[0:net.IPv4len]).String() |
|||
case gosocks5.AddrIPv6: |
|||
_, err = io.ReadFull(r, b[:net.IPv6len]) |
|||
addr.Host = net.IP(b[0:net.IPv6len]).String() |
|||
case gosocks5.AddrDomain: |
|||
if _, err = io.ReadFull(r, b[:1]); err != nil { |
|||
return nil, err |
|||
} |
|||
addrlen := int(b[0]) |
|||
_, err = io.ReadFull(r, b[:addrlen]) |
|||
addr.Host = string(b[:addrlen]) |
|||
default: |
|||
return nil, gosocks5.ErrBadAddrType |
|||
} |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
_, err = io.ReadFull(r, b[:2]) |
|||
addr.Port = binary.BigEndian.Uint16(b[:2]) |
|||
return addr, err |
|||
} |
|||
@ -1,425 +0,0 @@ |
|||
package gost |
|||
|
|||
import ( |
|||
"crypto/rand" |
|||
"fmt" |
|||
"net/http/httptest" |
|||
"net/url" |
|||
"testing" |
|||
) |
|||
|
|||
func init() { |
|||
// ss.Debug = true
|
|||
} |
|||
|
|||
var ss2Tests = []struct { |
|||
clientCipher *url.Userinfo |
|||
serverCipher *url.Userinfo |
|||
pass bool |
|||
}{ |
|||
{nil, nil, false}, |
|||
{&url.Userinfo{}, &url.Userinfo{}, false}, |
|||
{url.User("abc"), url.User("abc"), false}, |
|||
{url.UserPassword("abc", "def"), url.UserPassword("abc", "def"), false}, |
|||
|
|||
{url.User("aes-128-cfb"), url.User("aes-128-cfb"), true}, |
|||
{url.User("aes-128-cfb"), url.UserPassword("aes-128-cfb", "123456"), false}, |
|||
{url.UserPassword("aes-128-cfb", "123456"), url.User("aes-128-cfb"), false}, |
|||
{url.UserPassword("aes-128-cfb", "123456"), url.UserPassword("aes-128-cfb", "abc"), false}, |
|||
{url.UserPassword("aes-128-cfb", "123456"), url.UserPassword("aes-128-cfb", "123456"), true}, |
|||
|
|||
{url.User("aes-192-cfb"), url.User("aes-192-cfb"), true}, |
|||
{url.User("aes-192-cfb"), url.UserPassword("aes-192-cfb", "123456"), false}, |
|||
{url.UserPassword("aes-192-cfb", "123456"), url.User("aes-192-cfb"), false}, |
|||
{url.UserPassword("aes-192-cfb", "123456"), url.UserPassword("aes-192-cfb", "abc"), false}, |
|||
{url.UserPassword("aes-192-cfb", "123456"), url.UserPassword("aes-192-cfb", "123456"), true}, |
|||
|
|||
{url.User("aes-256-cfb"), url.User("aes-256-cfb"), true}, |
|||
{url.User("aes-256-cfb"), url.UserPassword("aes-256-cfb", "123456"), false}, |
|||
{url.UserPassword("aes-256-cfb", "123456"), url.User("aes-256-cfb"), false}, |
|||
{url.UserPassword("aes-256-cfb", "123456"), url.UserPassword("aes-256-cfb", "abc"), false}, |
|||
{url.UserPassword("aes-256-cfb", "123456"), url.UserPassword("aes-256-cfb", "123456"), true}, |
|||
|
|||
{url.User("aes-128-ctr"), url.User("aes-128-ctr"), true}, |
|||
{url.User("aes-128-ctr"), url.UserPassword("aes-128-ctr", "123456"), false}, |
|||
{url.UserPassword("aes-128-ctr", "123456"), url.User("aes-128-ctr"), false}, |
|||
{url.UserPassword("aes-128-ctr", "123456"), url.UserPassword("aes-128-ctr", "abc"), false}, |
|||
{url.UserPassword("aes-128-ctr", "123456"), url.UserPassword("aes-128-ctr", "123456"), true}, |
|||
|
|||
{url.User("aes-192-ctr"), url.User("aes-192-ctr"), true}, |
|||
{url.User("aes-192-ctr"), url.UserPassword("aes-192-ctr", "123456"), false}, |
|||
{url.UserPassword("aes-192-ctr", "123456"), url.User("aes-192-ctr"), false}, |
|||
{url.UserPassword("aes-192-ctr", "123456"), url.UserPassword("aes-192-ctr", "abc"), false}, |
|||
{url.UserPassword("aes-192-ctr", "123456"), url.UserPassword("aes-192-ctr", "123456"), true}, |
|||
|
|||
{url.User("aes-256-ctr"), url.User("aes-256-ctr"), true}, |
|||
{url.User("aes-256-ctr"), url.UserPassword("aes-256-ctr", "123456"), false}, |
|||
{url.UserPassword("aes-256-ctr", "123456"), url.User("aes-256-ctr"), false}, |
|||
{url.UserPassword("aes-256-ctr", "123456"), url.UserPassword("aes-256-ctr", "abc"), false}, |
|||
{url.UserPassword("aes-256-ctr", "123456"), url.UserPassword("aes-256-ctr", "123456"), true}, |
|||
|
|||
{url.User("chacha20-ietf"), url.User("chacha20-ietf"), true}, |
|||
{url.User("chacha20-ietf"), url.UserPassword("chacha20-ietf", "123456"), false}, |
|||
{url.UserPassword("chacha20-ietf", "123456"), url.User("chacha20-ietf"), false}, |
|||
{url.UserPassword("chacha20-ietf", "123456"), url.UserPassword("chacha20-ietf", "abc"), false}, |
|||
{url.UserPassword("chacha20-ietf", "123456"), url.UserPassword("chacha20-ietf", "123456"), true}, |
|||
|
|||
{url.User("xchacha20"), url.User("xchacha20"), true}, |
|||
{url.User("xchacha20"), url.UserPassword("xchacha20", "123456"), false}, |
|||
{url.UserPassword("xchacha20", "123456"), url.User("xchacha20"), false}, |
|||
{url.UserPassword("xchacha20", "123456"), url.UserPassword("xchacha20", "abc"), false}, |
|||
{url.UserPassword("xchacha20", "123456"), url.UserPassword("xchacha20", "123456"), true}, |
|||
|
|||
{url.User("AEAD_AES_128_GCM"), url.User("AEAD_AES_128_GCM"), true}, |
|||
{url.User("AEAD_AES_128_GCM"), url.UserPassword("AEAD_AES_128_GCM", "123456"), false}, |
|||
{url.UserPassword("AEAD_AES_128_GCM", "123456"), url.User("AEAD_AES_128_GCM"), false}, |
|||
{url.UserPassword("AEAD_AES_128_GCM", "123456"), url.UserPassword("AEAD_AES_128_GCM", "abc"), false}, |
|||
{url.UserPassword("AEAD_AES_128_GCM", "123456"), url.UserPassword("AEAD_AES_128_GCM", "123456"), true}, |
|||
|
|||
{url.User("AEAD_AES_192_GCM"), url.User("AEAD_AES_192_GCM"), true}, |
|||
{url.User("AEAD_AES_192_GCM"), url.UserPassword("AEAD_AES_192_GCM", "123456"), false}, |
|||
{url.UserPassword("AEAD_AES_192_GCM", "123456"), url.User("AEAD_AES_192_GCM"), false}, |
|||
{url.UserPassword("AEAD_AES_192_GCM", "123456"), url.UserPassword("AEAD_AES_192_GCM", "abc"), false}, |
|||
{url.UserPassword("AEAD_AES_192_GCM", "123456"), url.UserPassword("AEAD_AES_192_GCM", "123456"), true}, |
|||
|
|||
{url.User("AEAD_AES_256_GCM"), url.User("AEAD_AES_256_GCM"), true}, |
|||
{url.User("AEAD_AES_256_GCM"), url.UserPassword("AEAD_AES_256_GCM", "123456"), false}, |
|||
{url.UserPassword("AEAD_AES_256_GCM", "123456"), url.User("AEAD_AES_256_GCM"), false}, |
|||
{url.UserPassword("AEAD_AES_256_GCM", "123456"), url.UserPassword("AEAD_AES_256_GCM", "abc"), false}, |
|||
{url.UserPassword("AEAD_AES_256_GCM", "123456"), url.UserPassword("AEAD_AES_256_GCM", "123456"), true}, |
|||
|
|||
{url.User("AEAD_CHACHA20_POLY1305"), url.User("AEAD_CHACHA20_POLY1305"), true}, |
|||
{url.User("AEAD_CHACHA20_POLY1305"), url.UserPassword("AEAD_CHACHA20_POLY1305", "123456"), false}, |
|||
{url.UserPassword("AEAD_CHACHA20_POLY1305", "123456"), url.User("AEAD_CHACHA20_POLY1305"), false}, |
|||
{url.UserPassword("AEAD_CHACHA20_POLY1305", "123456"), url.UserPassword("AEAD_CHACHA20_POLY1305", "abc"), false}, |
|||
{url.UserPassword("AEAD_CHACHA20_POLY1305", "123456"), url.UserPassword("AEAD_CHACHA20_POLY1305", "123456"), true}, |
|||
} |
|||
|
|||
var ss2ProxyTests = []struct { |
|||
clientCipher *url.Userinfo |
|||
serverCipher *url.Userinfo |
|||
pass bool |
|||
}{ |
|||
{nil, nil, false}, |
|||
{&url.Userinfo{}, &url.Userinfo{}, false}, |
|||
{url.User("abc"), url.User("abc"), false}, |
|||
{url.UserPassword("abc", "def"), url.UserPassword("abc", "def"), false}, |
|||
|
|||
{url.User("aes-128-cfb"), url.User("aes-128-cfb"), false}, |
|||
{url.User("aes-128-cfb"), url.UserPassword("aes-128-cfb", "123456"), false}, |
|||
{url.UserPassword("aes-128-cfb", "123456"), url.User("aes-128-cfb"), false}, |
|||
// {url.UserPassword("aes-128-cfb", "123456"), url.UserPassword("aes-128-cfb", "abc"), false},
|
|||
{url.UserPassword("aes-128-cfb", "123456"), url.UserPassword("aes-128-cfb", "123456"), true}, |
|||
} |
|||
|
|||
func ss2ProxyRoundtrip(targetURL string, data []byte, clientInfo *url.Userinfo, serverInfo *url.Userinfo) error { |
|||
ln, err := TCPListener("") |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
client := &Client{ |
|||
Connector: Shadow2Connector(clientInfo), |
|||
Transporter: TCPTransporter(), |
|||
} |
|||
|
|||
server := &Server{ |
|||
Handler: Shadow2Handler(UsersHandlerOption(serverInfo)), |
|||
Listener: ln, |
|||
} |
|||
|
|||
go server.Run() |
|||
defer server.Close() |
|||
|
|||
return proxyRoundtrip(client, server, targetURL, data) |
|||
} |
|||
|
|||
func TestSS2Proxy(t *testing.T) { |
|||
httpSrv := httptest.NewServer(httpTestHandler) |
|||
defer httpSrv.Close() |
|||
|
|||
sendData := make([]byte, 128) |
|||
rand.Read(sendData) |
|||
|
|||
for i, tc := range ss2Tests { |
|||
tc := tc |
|||
t.Run(fmt.Sprintf("#%d", i), func(t *testing.T) { |
|||
err := ss2ProxyRoundtrip(httpSrv.URL, sendData, |
|||
tc.clientCipher, |
|||
tc.serverCipher, |
|||
) |
|||
if err == nil { |
|||
if !tc.pass { |
|||
t.Errorf("#%d should failed", i) |
|||
} |
|||
} else { |
|||
// t.Logf("#%d %v", i, err)
|
|||
if tc.pass { |
|||
t.Errorf("#%d got error: %v", i, err) |
|||
} |
|||
} |
|||
}) |
|||
} |
|||
} |
|||
|
|||
func BenchmarkSS2Proxy_AES256(b *testing.B) { |
|||
httpSrv := httptest.NewServer(httpTestHandler) |
|||
defer httpSrv.Close() |
|||
|
|||
sendData := make([]byte, 128) |
|||
rand.Read(sendData) |
|||
|
|||
ln, err := TCPListener("") |
|||
if err != nil { |
|||
b.Error(err) |
|||
} |
|||
|
|||
client := &Client{ |
|||
Connector: Shadow2Connector(url.UserPassword("aes-256-cfb", "123456")), |
|||
Transporter: TCPTransporter(), |
|||
} |
|||
|
|||
server := &Server{ |
|||
Handler: Shadow2Handler(UsersHandlerOption(url.UserPassword("aes-256-cfb", "123456"))), |
|||
Listener: ln, |
|||
} |
|||
|
|||
go server.Run() |
|||
defer server.Close() |
|||
|
|||
for i := 0; i < b.N; i++ { |
|||
if err := proxyRoundtrip(client, server, httpSrv.URL, sendData); err != nil { |
|||
b.Error(err) |
|||
} |
|||
} |
|||
} |
|||
|
|||
func BenchmarkSS2Proxy_XChacha20(b *testing.B) { |
|||
httpSrv := httptest.NewServer(httpTestHandler) |
|||
defer httpSrv.Close() |
|||
|
|||
sendData := make([]byte, 128) |
|||
rand.Read(sendData) |
|||
|
|||
ln, err := TCPListener("") |
|||
if err != nil { |
|||
b.Error(err) |
|||
} |
|||
|
|||
client := &Client{ |
|||
Connector: Shadow2Connector(url.UserPassword("xchacha20", "123456")), |
|||
Transporter: TCPTransporter(), |
|||
} |
|||
|
|||
server := &Server{ |
|||
Handler: Shadow2Handler(UsersHandlerOption(url.UserPassword("xchacha20", "123456"))), |
|||
Listener: ln, |
|||
} |
|||
|
|||
go server.Run() |
|||
defer server.Close() |
|||
|
|||
for i := 0; i < b.N; i++ { |
|||
if err := proxyRoundtrip(client, server, httpSrv.URL, sendData); err != nil { |
|||
b.Error(err) |
|||
} |
|||
} |
|||
} |
|||
|
|||
func BenchmarkSS2Proxy_Chacha20_ietf(b *testing.B) { |
|||
httpSrv := httptest.NewServer(httpTestHandler) |
|||
defer httpSrv.Close() |
|||
|
|||
sendData := make([]byte, 128) |
|||
rand.Read(sendData) |
|||
|
|||
ln, err := TCPListener("") |
|||
if err != nil { |
|||
b.Error(err) |
|||
} |
|||
|
|||
client := &Client{ |
|||
Connector: Shadow2Connector(url.UserPassword("chacha20-ietf", "123456")), |
|||
Transporter: TCPTransporter(), |
|||
} |
|||
|
|||
server := &Server{ |
|||
Handler: Shadow2Handler(UsersHandlerOption(url.UserPassword("chacha20-ietf", "123456"))), |
|||
Listener: ln, |
|||
} |
|||
|
|||
go server.Run() |
|||
defer server.Close() |
|||
|
|||
for i := 0; i < b.N; i++ { |
|||
if err := proxyRoundtrip(client, server, httpSrv.URL, sendData); err != nil { |
|||
b.Error(err) |
|||
} |
|||
} |
|||
} |
|||
|
|||
func BenchmarkSS2Proxy_CHACHA20_IETF_Parallel(b *testing.B) { |
|||
httpSrv := httptest.NewServer(httpTestHandler) |
|||
defer httpSrv.Close() |
|||
|
|||
sendData := make([]byte, 128) |
|||
rand.Read(sendData) |
|||
|
|||
ln, err := TCPListener("") |
|||
if err != nil { |
|||
b.Error(err) |
|||
} |
|||
|
|||
client := &Client{ |
|||
Connector: Shadow2Connector(url.UserPassword("chacha20-ietf", "123456")), |
|||
Transporter: TCPTransporter(), |
|||
} |
|||
|
|||
server := &Server{ |
|||
Handler: Shadow2Handler(UsersHandlerOption(url.UserPassword("chacha20-ietf", "123456"))), |
|||
Listener: ln, |
|||
} |
|||
|
|||
go server.Run() |
|||
defer server.Close() |
|||
|
|||
b.RunParallel(func(pb *testing.PB) { |
|||
for pb.Next() { |
|||
if err := proxyRoundtrip(client, server, httpSrv.URL, sendData); err != nil { |
|||
b.Error(err) |
|||
} |
|||
} |
|||
}) |
|||
} |
|||
|
|||
func BenchmarkSS2Proxy_AEAD_AES_256_GCM(b *testing.B) { |
|||
httpSrv := httptest.NewServer(httpTestHandler) |
|||
defer httpSrv.Close() |
|||
|
|||
sendData := make([]byte, 128) |
|||
rand.Read(sendData) |
|||
|
|||
ln, err := TCPListener("") |
|||
if err != nil { |
|||
b.Error(err) |
|||
} |
|||
|
|||
client := &Client{ |
|||
Connector: Shadow2Connector(url.UserPassword("AEAD_AES_256_GCM", "123456")), |
|||
Transporter: TCPTransporter(), |
|||
} |
|||
|
|||
server := &Server{ |
|||
Handler: Shadow2Handler(UsersHandlerOption(url.UserPassword("AEAD_AES_256_GCM", "123456"))), |
|||
Listener: ln, |
|||
} |
|||
|
|||
go server.Run() |
|||
defer server.Close() |
|||
|
|||
for i := 0; i < b.N; i++ { |
|||
if err := proxyRoundtrip(client, server, httpSrv.URL, sendData); err != nil { |
|||
b.Error(err) |
|||
} |
|||
} |
|||
} |
|||
|
|||
func BenchmarkSS2Proxy_AEAD_AES_256_GCM_Parallel(b *testing.B) { |
|||
httpSrv := httptest.NewServer(httpTestHandler) |
|||
defer httpSrv.Close() |
|||
|
|||
sendData := make([]byte, 128) |
|||
rand.Read(sendData) |
|||
|
|||
ln, err := TCPListener("") |
|||
if err != nil { |
|||
b.Error(err) |
|||
} |
|||
|
|||
client := &Client{ |
|||
Connector: Shadow2Connector(url.UserPassword("AEAD_AES_256_GCM", "123456")), |
|||
Transporter: TCPTransporter(), |
|||
} |
|||
|
|||
server := &Server{ |
|||
Handler: Shadow2Handler(UsersHandlerOption(url.UserPassword("AEAD_AES_256_GCM", "123456"))), |
|||
Listener: ln, |
|||
} |
|||
|
|||
go server.Run() |
|||
defer server.Close() |
|||
|
|||
b.RunParallel(func(pb *testing.PB) { |
|||
for pb.Next() { |
|||
if err := proxyRoundtrip(client, server, httpSrv.URL, sendData); err != nil { |
|||
b.Error(err) |
|||
} |
|||
} |
|||
}) |
|||
} |
|||
|
|||
func BenchmarkSS2Proxy_AEAD_CHACHA20_POLY1305(b *testing.B) { |
|||
httpSrv := httptest.NewServer(httpTestHandler) |
|||
defer httpSrv.Close() |
|||
|
|||
sendData := make([]byte, 128) |
|||
rand.Read(sendData) |
|||
|
|||
ln, err := TCPListener("") |
|||
if err != nil { |
|||
b.Error(err) |
|||
} |
|||
|
|||
client := &Client{ |
|||
Connector: Shadow2Connector(url.UserPassword("AEAD_CHACHA20_POLY1305", "123456")), |
|||
Transporter: TCPTransporter(), |
|||
} |
|||
|
|||
server := &Server{ |
|||
Handler: Shadow2Handler(UsersHandlerOption(url.UserPassword("AEAD_CHACHA20_POLY1305", "123456"))), |
|||
Listener: ln, |
|||
} |
|||
|
|||
go server.Run() |
|||
defer server.Close() |
|||
|
|||
for i := 0; i < b.N; i++ { |
|||
if err := proxyRoundtrip(client, server, httpSrv.URL, sendData); err != nil { |
|||
b.Error(err) |
|||
} |
|||
} |
|||
} |
|||
|
|||
func BenchmarkSS2Proxy_AEAD_CHACHA20_POLY1305_Parallel(b *testing.B) { |
|||
httpSrv := httptest.NewServer(httpTestHandler) |
|||
defer httpSrv.Close() |
|||
|
|||
sendData := make([]byte, 128) |
|||
rand.Read(sendData) |
|||
|
|||
ln, err := TCPListener("") |
|||
if err != nil { |
|||
b.Error(err) |
|||
} |
|||
|
|||
client := &Client{ |
|||
Connector: Shadow2Connector(url.UserPassword("AEAD_CHACHA20_POLY1305", "123456")), |
|||
Transporter: TCPTransporter(), |
|||
} |
|||
|
|||
server := &Server{ |
|||
Handler: Shadow2Handler(UsersHandlerOption(url.UserPassword("AEAD_CHACHA20_POLY1305", "123456"))), |
|||
Listener: ln, |
|||
} |
|||
|
|||
go server.Run() |
|||
defer server.Close() |
|||
|
|||
b.RunParallel(func(pb *testing.PB) { |
|||
for pb.Next() { |
|||
if err := proxyRoundtrip(client, server, httpSrv.URL, sendData); err != nil { |
|||
b.Error(err) |
|||
} |
|||
} |
|||
}) |
|||
} |
|||
Loading…
Reference in new issue