1 files changed, 4 insertions, 1 deletions

diff --git a/relay/server.go b/relay/server.go
index 085929c..d4a1edd 100644
--- a/relay/server.go
+++ b/relay/server.go
@@ -98,7 +98,10 @@ func (s *Server) handleWS(w http.ResponseWriter, r *http.Request) {
         s.mu.Add(1)
         go func() {
                 defer s.mu.Done()
-                ctx := r.Context()
+                // r.Context() is cancelled by the HTTP server when Hijack is called,
+                // so we use a fresh context. The connection manages its own lifecycle
+                // via the ping loop and WebSocket close frames.
+                ctx := context.Background()
                 h.serve(ctx)
                 if err := c.CloseConn(); err != nil {
                         // Ignore close errors — connection may already be gone.