client web ui

Author: lzjluzijie

config/config.go

@@ -1,6 +1,7 @@
 package config
 
 import (
+	"github.com/lzjluzijie/websocks/core/client"
 	"github.com/urfave/cli"
 )
 
@@ -43,7 +44,7 @@ var Command = cli.Command{
 					Usage: "InsecureSkipVerify: true",
 				},
 			},
-			Action: GenerateClientConfig,
+			Action: client.GenerateClientConfig,
 		},
 		{
 			Name:    "server",

config/server.go

@@ -7,7 +7,6 @@ import (
 
 	"github.com/gorilla/websocket"
 	"github.com/lzjluzijie/websocks/core"
-	"github.com/pkg/errors"
 	"github.com/urfave/cli"
 )
 
@@ -47,10 +46,10 @@ func GenerateServerConfig(c *cli.Context) (err error) {
 	keyPath := c.String("key")
 	reverseProxy := c.String("reverse-proxy")
 
-	if []byte(pattern)[0] != '/' {
-		err = errors.New("pattern does not start with '/'")
-		return
-	}
+	//if []byte(pattern)[0] != '/' {
+	//	err = errors.New("pattern does not start with '/'")
+	//	return
+	//}
 
 	config := &core.ServerConfig{
 		Pattern:      pattern,

core/client.go

@@ -0,0 +1,187 @@
+package client
+
+import (
+	"io"
+	"net"
+	"time"
+
+	"net/url"
+
+	"crypto/tls"
+
+	"github.com/gorilla/websocket"
+	"github.com/lzjluzijie/websocks/core"
+	"github.com/sirupsen/logrus"
+)
+
+var log = logrus.New()
+
+type WebSocksClientConfig struct {
+	ListenAddr string
+	ServerURL  string
+
+	SNI          string
+	InsecureCert bool
+
+	Mux bool
+}
+
+type WebSocksClient struct {
+	ServerURL  *url.URL
+	ListenAddr *net.TCPAddr
+	Dialer     *websocket.Dialer
+	muxWS      *core.MuxWebSocket
+
+	//todo enable mux
+	Mux bool
+
+	stopC chan int
+
+	//statistics
+	CreatedAt time.Time
+}
+
+func NewWebSocksClient(config *WebSocksClientConfig) (client *WebSocksClient) {
+	serverURL, err := url.Parse(config.ServerURL)
+	if err != nil {
+		return
+	}
+
+	laddr, err := net.ResolveTCPAddr("tcp", config.ListenAddr)
+	if err != nil {
+		return
+	}
+
+	tlsConfig := &tls.Config{
+		InsecureSkipVerify: config.InsecureCert,
+		ServerName:         config.SNI,
+	}
+
+	client = &WebSocksClient{
+		ServerURL:  serverURL,
+		ListenAddr: laddr,
+		Dialer: &websocket.Dialer{
+			ReadBufferSize:   4 * 1024,
+			WriteBufferSize:  4 * 1024,
+			HandshakeTimeout: 10 * time.Second,
+			TLSClientConfig:  tlsConfig,
+		},
+
+		CreatedAt: time.Now(),
+	}
+
+	return
+}
+
+func (client *WebSocksClient) Listen() (err error) {
+	listener, err := net.ListenTCP("tcp", client.ListenAddr)
+	if err != nil {
+		return err
+	}
+
+	log.Infof("Start to listen at %s", client.ListenAddr.String())
+
+	defer listener.Close()
+
+	if client.Mux {
+		err := client.OpenMux()
+		if err != nil {
+			log.Debugf(err.Error())
+			return err
+		}
+
+		go client.ListenMuxWS(client.muxWS)
+	}
+
+	go func() {
+		client.stopC = make(chan int)
+		<-client.stopC
+		err = listener.Close()
+		if err != nil {
+			log.Errorf(err.Error())
+			return
+		}
+
+		log.Infof("stopped")
+	}()
+
+	for {
+		conn, err := listener.AcceptTCP()
+		if err != nil {
+			log.Debugf(err.Error())
+			break
+		}
+
+		go client.handleConn(conn)
+	}
+	return nil
+}
+
+func (client *WebSocksClient) Stop() {
+	client.stopC <- 1911
+	return
+}
+
+func (client *WebSocksClient) handleConn(conn *net.TCPConn) {
+	defer conn.Close()
+
+	conn.SetLinger(0)
+
+	err := handShake(conn)
+	if err != nil {
+		log.Debugf(err.Error())
+		return
+	}
+
+	_, host, err := getRequest(conn)
+	if err != nil {
+		log.Debugf(err.Error())
+		return
+	}
+
+	_, err = conn.Write([]byte{0x05, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x08, 0x43})
+	if err != nil {
+		log.Debugf(err.Error())
+		return
+	}
+
+	if client.Mux {
+		client.DialMuxConn(host, conn)
+	} else {
+		client.DialWSConn(host, conn)
+	}
+
+	return
+}
+
+func (client *WebSocksClient) DialWSConn(host string, conn *net.TCPConn) {
+	wsConn, _, err := client.Dialer.Dial(client.ServerURL.String(), map[string][]string{
+		"WebSocks-Host": {host},
+	})
+
+	if err != nil {
+		log.Errorf(err.Error())
+		return
+	}
+	defer wsConn.Close()
+
+	log.Debugf("dialed ws for %s", host)
+
+	ws := core.NewWebSocket(wsConn)
+
+	go func() {
+		_, err = io.Copy(ws, conn)
+		if err != nil {
+			log.Debugf(err.Error())
+			return
+		}
+		return
+	}()
+
+	_, err = io.Copy(conn, ws)
+	if err != nil {
+		log.Debugf(err.Error())
+		return
+	}
+	return
+}