1 files changed, 45 insertions, 0 deletions

diff --git a/client.go b/client.go
new file mode 100644
index 0000000..00455ab
--- /dev/null
+++ b/client.go
@@ -0,0 +1,45 @@
+package main
+
+import (
+        "log"
+        "net"
+
+        "github.com/gorilla/websocket"
+)
+
+type ClientHandler struct {
+        SocketListenOn  string
+        WebsocketServer string
+}
+
+func (h *ClientHandler) ServiceConnection(proxyconn net.Conn) {
+        defer proxyconn.Close()
+
+        wsconn, _, err := websocket.DefaultDialer.Dial(h.WebsocketServer, nil)
+        if err != nil {
+                log.Println(err)
+                return
+        }
+        defer wsconn.Close()
+
+        log.Println("Connected to server")
+
+        serviceBoth(wsconn, proxyconn)
+}
+
+func (h *ClientHandler) Run() {
+        listener, err := net.Listen("tcp", h.SocketListenOn)
+        if err != nil {
+                log.Printf("error: Run: %s", err)
+                return
+        }
+
+        for {
+                conn, err := listener.Accept()
+                if err != nil {
+                        log.Printf("error: Run: %s", err)
+                        continue
+                }
+                go h.ServiceConnection(conn)
+        }
+}