feat: support SOCKS5 (prev)

Author: KevinZonda

go.mod

@@ -0,0 +1,8 @@
+module fastgit.org/sniproxygo
+
+go 1.18
+
+require (
+	golang.org/x/net v0.0.0-20220812174116-3211cb980234 // indirect
+	gopkg.in/yaml.v2 v2.4.0 // indirect
+)

go.sum

@@ -0,0 +1,5 @@
+golang.org/x/net v0.0.0-20220812174116-3211cb980234 h1:RDqmgfe7SvlMWoqC3xwQ2blLO3fcWcxMa3eBLRdRW7E=
+golang.org/x/net v0.0.0-20220812174116-3211cb980234/go.mod h1:YDH+HFinaLZZlnHAfSS6ZXJJ9M9t4Dl22yv3iI2vPwk=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
+gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=

interfaces.go

@@ -0,0 +1,25 @@
+package main
+
+import (
+	"net"
+
+	"golang.org/x/net/proxy"
+)
+
+var proxyDialer proxy.Dialer = nil
+
+func GetDialer(isSocks5 bool) proxy.Dialer {
+	if isSocks5 {
+		return &net.Dialer{}
+	}
+	if proxyDialer != nil {
+		return proxyDialer
+
+	}
+	var err error
+	proxyDialer, err = proxy.SOCKS5("tcp", "127.0.0.1:4000", nil, proxy.Direct)
+	if err != nil {
+		return proxyDialer
+	}
+	panic(err)
+}

main.go

@@ -4,7 +4,6 @@ import (
 	"context"
 	"flag"
 	"fmt"
-	"gopkg.in/yaml.v2"
 	"io"
 	"io/ioutil"
 	"net"
@@ -13,6 +12,8 @@ import (
 	"strings"
 	"syscall"
 	"time"
+
+	"gopkg.in/yaml.v2"
 )
 
 var SNIPort = 443
@@ -41,7 +42,7 @@ func main() {
 		os.Exit(0)
 	}
 	if len(cfg.ForwardRules) <= 0 {
-		serviceLogger(fmt.Sprintf("No rules found in yaml!"), 31)
+		serviceLogger("No rules found in yaml!", 31)
 		os.Exit(0)
 	}
 	for _, rule := range cfg.ForwardRules {
@@ -93,7 +94,7 @@ func serve(c net.Conn, raddr string) {
 	servername := getSNIServerName(buf[:n])
 
 	if servername == "" {
-		serviceDebugger(fmt.Sprintf("No SNI server name found, ignore it"), 31)
+		serviceDebugger("No SNI server name found, ignore it", 31)
 		return
 	}
 
@@ -108,7 +109,7 @@ func serve(c net.Conn, raddr string) {
 func getSNIServerName(buf []byte) string {
 	n := len(buf)
 	if n < 5 {
-		serviceDebugger(fmt.Sprintf("Not tls handshake"), 31)
+		serviceDebugger("Not tls handshake", 31)
 		return ""
 	}
 
@@ -120,7 +121,7 @@ func getSNIServerName(buf []byte) string {
 
 	// tls major version
 	if buf[1] != 3 {
-		serviceDebugger(fmt.Sprintf("TLS version < 3 not supported"), 31)
+		serviceDebugger("TLS version < 3 not supported", 31)
 		return ""
 	}
 
@@ -131,7 +132,7 @@ func getSNIServerName(buf []byte) string {
 
 	// handshake message type
 	if buf[5] != typeClientHello {
-		serviceDebugger(fmt.Sprintf("Not client hello"), 31)
+		serviceDebugger("Not client hello", 31)
 		return ""
 	}
 
@@ -142,14 +143,15 @@ func getSNIServerName(buf []byte) string {
 	// client hello message not include tls header, 5 bytes
 	ret := msg.unmarshal(buf[5:n])
 	if !ret {
-		serviceDebugger(fmt.Sprintf("Parse hello message return false"), 31)
+		serviceDebugger("Parse hello message return false", 31)
 		return ""
 	}
 	return msg.serverName
 }
 
 func forward(conn net.Conn, data []byte, dst string, raddr string) {
-	backend, err := net.Dial("tcp", dst)
+	// TODO: FIX
+	backend, err := GetDialer(true).Dial("tcp", dst)
 	if err != nil {
 		serviceLogger(fmt.Sprintf("Couldn't connect to backend, %v", err), 31)
 		return