test: add client-server connect tests

Author: asciimoth

.golangci.yml

@@ -83,7 +83,6 @@ linters:
     - tagliatelle
     - testifylint
     - testpackage
-    - tparallel
     - unconvert
     - unparam
     - usestdlibvars
@@ -95,3 +94,4 @@ linters:
   disable:
     - unused
     - contextcheck
+    - tparallel

pair_client_server_test.go

@@ -0,0 +1,175 @@
+package socksgo_test
+
+import (
+	"net/http"
+	"testing"
+	"time"
+
+	"github.com/asciimoth/bufpool"
+	socksgo "github.com/asciimoth/socksgo"
+)
+
+/*
+ TODO:
+- Bind
+- Mbind
+- UDP Assoc dial
+- USP Assoc listen
+- x2 for UDP Tun
+- Resolve, ResolvePtr with custom resolver
+*/
+
+func runClientHttp(t *testing.T, srv string, urls []string) {
+	socks, err := socksgo.ClientFromURL(srv)
+	if err != nil {
+		t.Fatal()
+	}
+	transport := &http.Transport{
+		DialContext:         socks.Dial,
+		MaxIdleConns:        100,
+		IdleConnTimeout:     60 * time.Second,
+		TLSHandshakeTimeout: 60 * time.Second,
+	}
+	client := &http.Client{
+		Transport: transport,
+		Timeout:   60 * time.Second,
+	}
+	for _, url := range urls {
+		var req *http.Request
+		req, err = http.NewRequest("GET", url, nil) //nolint
+		if err != nil {
+			continue
+		}
+		resp, err := client.Do(req)
+		if err == nil {
+			_ = resp.Body.Close()
+			break
+		}
+	}
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func runClients45Http(
+	t *testing.T, addr string, urls []string, tls, ws bool,
+) {
+	postfix := ""
+	if tls {
+		if ws {
+			postfix = "+wss"
+		} else {
+			postfix = "+tls"
+		}
+	} else {
+		if ws {
+			postfix = "+ws"
+		}
+	}
+	t.Run("group", func(t *testing.T) {
+		t.Run("socks4", func(t *testing.T) {
+			t.Parallel()
+			runClientHttp(t, "socks4"+postfix+"://"+addr, urls)
+		})
+		t.Run("socks4 user", func(t *testing.T) {
+			t.Parallel()
+			runClientHttp(t, "socks4"+postfix+"://user@"+addr, urls)
+		})
+		t.Run("socks4a", func(t *testing.T) {
+			t.Parallel()
+			runClientHttp(t, "socks4a"+postfix+"://"+addr, urls)
+		})
+		t.Run("socks4a user", func(t *testing.T) {
+			t.Parallel()
+			runClientHttp(t, "socks4a"+postfix+"://user@"+addr, urls)
+		})
+		t.Run("socks5", func(t *testing.T) {
+			t.Parallel()
+			runClientHttp(t, "socks5"+postfix+"://"+addr, urls)
+		})
+		t.Run("socks5 pass", func(t *testing.T) {
+			t.Parallel()
+			runClientHttp(t, "socks5"+postfix+"://user:pass@"+addr, urls)
+		})
+	})
+}
+
+func TestClientServerConnect(t *testing.T) {
+	t.Parallel()
+
+	pool := bufpool.NewTestDebugPool(t)
+	pool.OnLog = nil // Too verbose
+	defer pool.Close()
+
+	cfg := GetEnvConfig()
+
+	cancel, addr, err := runTCPServer(&socksgo.Server{
+		Pool: pool,
+	}, cfg.Host)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer cancel()
+
+	runClients45Http(t, addr.String(), cfg.CurlURLs, false, false)
+}
+
+func TestClientServerTLSConnect(t *testing.T) {
+	t.Parallel()
+
+	pool := bufpool.NewTestDebugPool(t)
+	pool.OnLog = nil // Too verbose
+	defer pool.Close()
+
+	cfg := GetEnvConfig()
+
+	cancel, addr, err := runTLSServer(&socksgo.Server{
+		Pool: pool,
+	}, cfg.Host)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer cancel()
+
+	runClients45Http(t, addr.String(), cfg.CurlURLs, true, false)
+}
+
+func TestClientWSServerConnect(t *testing.T) {
+	t.Parallel()
+
+	pool := bufpool.NewTestDebugPool(t)
+	pool.OnLog = nil // Too verbose
+	defer pool.Close()
+
+	cfg := GetEnvConfig()
+
+	cancel, addr, err := runWSServer(&socksgo.Server{
+		Pool: pool,
+	}, cfg.Host, false)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer cancel()
+
+	runClients45Http(t, addr.String(), cfg.CurlURLs, false, true)
+}
+
+func TestClientWSSServerConnect(t *testing.T) {
+	t.Parallel()
+
+	pool := bufpool.NewTestDebugPool(t)
+	pool.OnLog = nil // Too verbose
+	defer pool.Close()
+
+	cfg := GetEnvConfig()
+
+	cancel, addr, err := runWSServer(&socksgo.Server{
+		Pool: pool,
+	}, cfg.Host, true)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer cancel()
+
+	runClients45Http(t, addr.String(), cfg.CurlURLs, true, true)
+}