wip: kcp client

1 files changed, 8 insertions, 0 deletions

diff --git a/client/main.go b/client/main.go
index 0c7edcc..daf6ce4 100644
--- a/client/main.go
+++ b/client/main.go
@@ -10,6 +10,8 @@ import (
 
 	"0xacab.org/leap/obfsvpn"
 	socks5 "github.com/armon/go-socks5"
+
+	"github.com/xtaci/kcp-go"
 )
 
 func main() {
@@ -49,6 +51,12 @@ func main() {
 		logger.Fatalf("cannot get dialer: %v", err)
 	}
 
+	if os.Getenv("KCP") == "1" {
+		dialer.DialFunc = func(network, address string) (net.Conn, error) {
+			return kcp.Dial(address)
+		}
+	}
+
 	socksConf := &socks5.Config{
 		Dial: dialer.Dial,
 	}