mirror of https://github.com/ginuerzh/gost
ginuerzh
8 years ago
6 changed files with 529 additions and 26 deletions
@ -0,0 +1,223 @@ |
|||||
|
package gost |
||||
|
|
||||
|
import ( |
||||
|
"crypto/rand" |
||||
|
"crypto/tls" |
||||
|
"net/http/httptest" |
||||
|
"net/url" |
||||
|
"testing" |
||||
|
) |
||||
|
|
||||
|
func autoHTTPProxyRoundtrip(targetURL string, data []byte, clientInfo *url.Userinfo, serverInfo []*url.Userinfo) error { |
||||
|
ln, err := TCPListener("") |
||||
|
if err != nil { |
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
client := &Client{ |
||||
|
Connector: HTTPConnector(clientInfo), |
||||
|
Transporter: TCPTransporter(), |
||||
|
} |
||||
|
|
||||
|
server := &Server{ |
||||
|
Listener: ln, |
||||
|
Handler: AutoHandler( |
||||
|
UsersHandlerOption(serverInfo...), |
||||
|
), |
||||
|
} |
||||
|
|
||||
|
go server.Run() |
||||
|
defer server.Close() |
||||
|
|
||||
|
return proxyRoundtrip(client, server, targetURL, data) |
||||
|
} |
||||
|
|
||||
|
func TestAutoHTTPProxy(t *testing.T) { |
||||
|
httpSrv := httptest.NewServer(httpTestHandler) |
||||
|
defer httpSrv.Close() |
||||
|
|
||||
|
sendData := make([]byte, 128) |
||||
|
rand.Read(sendData) |
||||
|
|
||||
|
for i, tc := range httpProxyTests { |
||||
|
err := autoHTTPProxyRoundtrip(httpSrv.URL, sendData, tc.cliUser, tc.srvUsers) |
||||
|
if err == nil { |
||||
|
if tc.errStr != "" { |
||||
|
t.Errorf("#%d should failed with error %s", i, tc.errStr) |
||||
|
} |
||||
|
} else { |
||||
|
if tc.errStr == "" { |
||||
|
t.Errorf("#%d got error %v", i, err) |
||||
|
} |
||||
|
if err.Error() != tc.errStr { |
||||
|
t.Errorf("#%d got error %v, want %v", i, err, tc.errStr) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func autoSocks5ProxyRoundtrip(targetURL string, data []byte, clientInfo *url.Userinfo, serverInfo []*url.Userinfo) error { |
||||
|
ln, err := TCPListener("") |
||||
|
if err != nil { |
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
client := &Client{ |
||||
|
Connector: SOCKS5Connector(clientInfo), |
||||
|
Transporter: TCPTransporter(), |
||||
|
} |
||||
|
|
||||
|
server := &Server{ |
||||
|
Handler: AutoHandler(UsersHandlerOption(serverInfo...)), |
||||
|
Listener: ln, |
||||
|
} |
||||
|
|
||||
|
go server.Run() |
||||
|
defer server.Close() |
||||
|
|
||||
|
return proxyRoundtrip(client, server, targetURL, data) |
||||
|
} |
||||
|
|
||||
|
func TestAutoSOCKS5Proxy(t *testing.T) { |
||||
|
cert, err := GenCertificate() |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
DefaultTLSConfig = &tls.Config{ |
||||
|
Certificates: []tls.Certificate{cert}, |
||||
|
} |
||||
|
|
||||
|
httpSrv := httptest.NewServer(httpTestHandler) |
||||
|
defer httpSrv.Close() |
||||
|
|
||||
|
sendData := make([]byte, 128) |
||||
|
rand.Read(sendData) |
||||
|
|
||||
|
for i, tc := range socks5ProxyTests { |
||||
|
err := autoSocks5ProxyRoundtrip(httpSrv.URL, sendData, |
||||
|
tc.cliUser, |
||||
|
tc.srvUsers, |
||||
|
) |
||||
|
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 autoSOCKS4ProxyRoundtrip(targetURL string, data []byte) error { |
||||
|
ln, err := TCPListener("") |
||||
|
if err != nil { |
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
client := &Client{ |
||||
|
Connector: SOCKS4Connector(), |
||||
|
Transporter: TCPTransporter(), |
||||
|
} |
||||
|
|
||||
|
server := &Server{ |
||||
|
Listener: ln, |
||||
|
Handler: AutoHandler(), |
||||
|
} |
||||
|
go server.Run() |
||||
|
defer server.Close() |
||||
|
|
||||
|
return proxyRoundtrip(client, server, targetURL, data) |
||||
|
} |
||||
|
|
||||
|
func TestAutoSOCKS4Proxy(t *testing.T) { |
||||
|
httpSrv := httptest.NewServer(httpTestHandler) |
||||
|
defer httpSrv.Close() |
||||
|
|
||||
|
sendData := make([]byte, 128) |
||||
|
rand.Read(sendData) |
||||
|
|
||||
|
err := autoSOCKS4ProxyRoundtrip(httpSrv.URL, sendData) |
||||
|
// t.Logf("#%d %v", i, err)
|
||||
|
if err != nil { |
||||
|
t.Errorf("got error: %v", err) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func autoSocks4aProxyRoundtrip(targetURL string, data []byte) error { |
||||
|
ln, err := TCPListener("") |
||||
|
if err != nil { |
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
client := &Client{ |
||||
|
Connector: SOCKS4AConnector(), |
||||
|
Transporter: TCPTransporter(), |
||||
|
} |
||||
|
|
||||
|
server := &Server{ |
||||
|
Listener: ln, |
||||
|
Handler: AutoHandler(), |
||||
|
} |
||||
|
|
||||
|
go server.Run() |
||||
|
defer server.Close() |
||||
|
|
||||
|
return proxyRoundtrip(client, server, targetURL, data) |
||||
|
} |
||||
|
|
||||
|
func TestAutoSOCKS4AProxy(t *testing.T) { |
||||
|
httpSrv := httptest.NewServer(httpTestHandler) |
||||
|
defer httpSrv.Close() |
||||
|
|
||||
|
sendData := make([]byte, 128) |
||||
|
rand.Read(sendData) |
||||
|
|
||||
|
err := autoSocks4aProxyRoundtrip(httpSrv.URL, sendData) |
||||
|
// t.Logf("#%d %v", i, err)
|
||||
|
if err != nil { |
||||
|
t.Errorf("got error: %v", err) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func autoSSProxyRoundtrip(targetURL string, data []byte, clientInfo *url.Userinfo, serverInfo *url.Userinfo) error { |
||||
|
ln, err := TCPListener("") |
||||
|
if err != nil { |
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
client := &Client{ |
||||
|
Connector: ShadowConnector(clientInfo), |
||||
|
Transporter: TCPTransporter(), |
||||
|
} |
||||
|
|
||||
|
server := &Server{ |
||||
|
Handler: AutoHandler(UsersHandlerOption(serverInfo)), |
||||
|
Listener: ln, |
||||
|
} |
||||
|
|
||||
|
go server.Run() |
||||
|
defer server.Close() |
||||
|
|
||||
|
return proxyRoundtrip(client, server, targetURL, data) |
||||
|
} |
||||
|
|
||||
|
func TestAutoSSProxy(t *testing.T) { |
||||
|
httpSrv := httptest.NewServer(httpTestHandler) |
||||
|
defer httpSrv.Close() |
||||
|
|
||||
|
sendData := make([]byte, 128) |
||||
|
rand.Read(sendData) |
||||
|
|
||||
|
for i, tc := range ssTests { |
||||
|
err := autoSSProxyRoundtrip(httpSrv.URL, sendData, |
||||
|
tc.clientCipher, |
||||
|
tc.serverCipher, |
||||
|
) |
||||
|
if err == nil { |
||||
|
t.Errorf("#%d should failed", i) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,121 @@ |
|||||
|
package gost |
||||
|
|
||||
|
import ( |
||||
|
"bytes" |
||||
|
"io" |
||||
|
"net" |
||||
|
"testing" |
||||
|
"time" |
||||
|
) |
||||
|
|
||||
|
var hostsLookupTests = []struct { |
||||
|
hosts []Host |
||||
|
host string |
||||
|
ip net.IP |
||||
|
}{ |
||||
|
{nil, "", nil}, |
||||
|
{nil, "example.com", nil}, |
||||
|
{[]Host{}, "", nil}, |
||||
|
{[]Host{}, "example.com", nil}, |
||||
|
{[]Host{NewHost(nil, "")}, "", nil}, |
||||
|
{[]Host{NewHost(nil, "example.com")}, "example.com", nil}, |
||||
|
{[]Host{NewHost(net.IPv4(192, 168, 1, 1), "")}, "", nil}, |
||||
|
{[]Host{NewHost(net.IPv4(192, 168, 1, 1), "example.com")}, "example.com", net.IPv4(192, 168, 1, 1)}, |
||||
|
{[]Host{NewHost(net.IPv4(192, 168, 1, 1), "example.com")}, "example", nil}, |
||||
|
{[]Host{NewHost(net.IPv4(192, 168, 1, 1), "example.com", "example", "examples")}, "example", net.IPv4(192, 168, 1, 1)}, |
||||
|
{[]Host{NewHost(net.IPv4(192, 168, 1, 1), "example.com", "example", "examples")}, "examples", net.IPv4(192, 168, 1, 1)}, |
||||
|
} |
||||
|
|
||||
|
func TestHostsLookup(t *testing.T) { |
||||
|
for i, tc := range hostsLookupTests { |
||||
|
hosts := NewHosts(tc.hosts...) |
||||
|
ip := hosts.Lookup(tc.host) |
||||
|
if !ip.Equal(tc.ip) { |
||||
|
t.Errorf("#%d test failed: lookup should be %s, got %s", i, tc.ip, ip) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
var HostsReloadTests = []struct { |
||||
|
r io.Reader |
||||
|
period time.Duration |
||||
|
host string |
||||
|
ip net.IP |
||||
|
stopped bool |
||||
|
}{ |
||||
|
{ |
||||
|
r: nil, |
||||
|
period: 0, |
||||
|
host: "", |
||||
|
ip: nil, |
||||
|
}, |
||||
|
{ |
||||
|
r: bytes.NewBufferString(""), |
||||
|
period: 0, |
||||
|
host: "example.com", |
||||
|
ip: nil, |
||||
|
}, |
||||
|
{ |
||||
|
r: bytes.NewBufferString("reload 10s"), |
||||
|
period: 10 * time.Second, |
||||
|
host: "example.com", |
||||
|
ip: nil, |
||||
|
}, |
||||
|
{ |
||||
|
r: bytes.NewBufferString("reload 10s\n192.168.1.1"), |
||||
|
period: 10 * time.Second, |
||||
|
host: "", |
||||
|
ip: nil, |
||||
|
}, |
||||
|
{ |
||||
|
r: bytes.NewBufferString("#reload 10s\n192.168.1.1 example.com"), |
||||
|
period: 0, |
||||
|
host: "example.com", |
||||
|
ip: net.IPv4(192, 168, 1, 1), |
||||
|
}, |
||||
|
{ |
||||
|
r: bytes.NewBufferString("#reload 10s\n#192.168.1.1 example.com"), |
||||
|
period: 0, |
||||
|
host: "example.com", |
||||
|
ip: nil, |
||||
|
stopped: true, |
||||
|
}, |
||||
|
{ |
||||
|
r: bytes.NewBufferString("#reload 10s\n192.168.1.1 example.com example examples"), |
||||
|
period: 0, |
||||
|
host: "example", |
||||
|
ip: net.IPv4(192, 168, 1, 1), |
||||
|
stopped: true, |
||||
|
}, |
||||
|
{ |
||||
|
r: bytes.NewBufferString("#reload 10s\n192.168.1.1 example.com example examples"), |
||||
|
period: 0, |
||||
|
host: "examples", |
||||
|
ip: net.IPv4(192, 168, 1, 1), |
||||
|
stopped: true, |
||||
|
}, |
||||
|
} |
||||
|
|
||||
|
func TestHostsReload(t *testing.T) { |
||||
|
for i, tc := range HostsReloadTests { |
||||
|
hosts := NewHosts() |
||||
|
if err := hosts.Reload(tc.r); err != nil { |
||||
|
t.Error(err) |
||||
|
} |
||||
|
if hosts.Period() != tc.period { |
||||
|
t.Errorf("#%d test failed: period value should be %v, got %v", |
||||
|
i, tc.period, hosts.Period()) |
||||
|
} |
||||
|
ip := hosts.Lookup(tc.host) |
||||
|
if !ip.Equal(tc.ip) { |
||||
|
t.Errorf("#%d test failed: lookup should be %s, got %s", i, tc.ip, ip) |
||||
|
} |
||||
|
if tc.stopped { |
||||
|
hosts.Stop() |
||||
|
} |
||||
|
if hosts.Stopped() != tc.stopped { |
||||
|
t.Errorf("#%d test failed: stopped value should be %v, got %v", |
||||
|
i, tc.stopped, hosts.Stopped()) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue