1 files changed, 8 insertions, 1 deletions

diff --git a/server.go b/server.go
index 1faf33d..9a9d897 100644
--- a/server.go
+++ b/server.go
@@ -42,5 +42,12 @@ func (h *ServerHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 
         log.Println("Connected to SSH server")
 
-        serviceBoth(wsconn, proxyconn, r.Context())
+        errc := make(chan error)
+        ws := &WebsocketReadWriter{wsconn}
+
+        go serviceBoth(proxyconn, ws, errc)
+        go serviceBoth(ws, proxyconn, errc)
+
+        <-errc
+        log.Println("Closing client server")
 }