README.md

@@ -27,46 +27,6 @@ You'll need to edit the kubeconfig file and change the api host to be your local
 
 
 
-## How to use it
-
-### Pod
-
-Launch a pod (through a deployment, or sts, or something else) on your cluster.
-You can use the following image `forge.k3s.fr/frank/kube-escape:latest`
-
-Don't forget to give it the following env values:
-
-- WEBSOCKET_ROOT_URL
-- WS_ID: facultative, can auto generate itself
-
-Then look at its logs and you'll see the ws url to use when connecting to it
-
-### WS Proxy
-
-You should spawn a WS proxy that will receive connections from the client and the pod.
-It should be accessible by both.
-
-You can override the command of the image and use `./proxy.py`
-
-### Client
-
-Launch your client with:
-```bash
-./client.py <ws_URL_given_by_pod>
-```
-
-This will open a listening socket on localhost port 6443
-
-### Kubectl
-
-Change your kubeconfig's server to `https://localhost:6443`
-
-And then, enjoy!
-
-
-## Considerations
-
-
 ### Security
 
 I guess you could proxy your websockets through an HTTPs endpoint. Wouldn't be bad.
@@ -75,4 +35,4 @@ However, the kubeapi proto is already over TLS, so it wouldn't add much value.
 ### Compression
 
 Sadly it's not really possible (efficient-wise) to compress TLS data as it looks
-random-ish.
+random-ish.

pod.py

@@ -16,7 +16,7 @@ async def handle_socket_read(socketid, tcpreader, ws):
         print(f"New socket: {socketid}. Waiting on recv")
         while True:
             data = await tcpreader.read(2024)
-            #print(f"TCP@{socketid} Received {len(data)} bytes")
+            print(f"TCP@{socketid} Received {len(data)} bytes")
             if data == b'':
                 print(f"TCP@{socketid} Connection closed")
                 c = conn.WSMsg(socketid, conn.MsgType.DISCONNECT)
@@ -24,7 +24,7 @@ async def handle_socket_read(socketid, tcpreader, ws):
                 break
 
             c = conn.WSMsg(socketid, conn.MsgType.DATA, data)
-            #print(f'TCP>WS: {c}')
+            print(f'TCP>WS: {c}')
             await ws.send(c.to_bytes())
 
     except Exception as e:
@@ -62,8 +62,9 @@ async def handle_ws_incoming(cfg, ws, sockets):
 
     elif c.msg == conn.MsgType.DATA:
         tcpreader, tcpwriter = sockets[socketid]
-        #print(f'WS>TCP: {c}')
+        print(f'WS>TCP: {c}')
         tcpwriter.write(c.payload)
+        print('written')
 
 
 def get_config():

proxy.py

@@ -29,12 +29,12 @@ async def handler(websocket):
         print(f"Connection closed: {e}")
     finally:
         # Unregister the client
-        connected_clients.get(websocket.request.path, set()).remove(websocket)
+        connected_clients.remove(websocket)
 
 async def main():
     # Start the WebSocket server
     ws_port = os.environ.get("WS_PORT", 9999)
-    server = await websockets.asyncio.server.serve(handler, "", ws_port)
+    server = await websockets.asyncio.server.serve(handler, "::", ws_port)
     print(f"WebSocket server listening on ws://[::]:{ws_port}")
     await server.wait_closed()