fix connection state leak

Author: provinzkraut

src/hypercorn/asyncio/tcp_server.py

@@ -61,7 +61,7 @@ async def run(self) -> None:
                     self.config,
                     self.context,
                     task_group,
-                    ConnectionState(self.state.copy()),
+                    ConnectionState(self.state),
                     ssl,
                     client,
                     server,

src/hypercorn/protocol/http_stream.py

@@ -97,7 +97,7 @@ async def handle(self, event: Event) -> None:
                 "headers": event.headers,
                 "client": self.client,
                 "server": self.server,
-                "state": event.state,
+                "state": event.state.copy(),
                 "extensions": {},
             }
 

src/hypercorn/protocol/ws_stream.py

@@ -219,7 +219,7 @@ async def handle(self, event: Event) -> None:
                 "headers": event.headers,
                 "client": self.client,
                 "server": self.server,
-                "state": event.state,
+                "state": event.state.copy(),
                 "subprotocols": self.handshake.subprotocols or [],
                 "extensions": {"websocket.http.response": {}},
             }

src/hypercorn/trio/tcp_server.py

@@ -64,7 +64,7 @@ async def run(self) -> None:
                     self.config,
                     self.context,
                     task_group,
-                    ConnectionState(self.state.copy()),
+                    ConnectionState(self.state),
                     ssl,
                     client,
                     server,

tests/e2e/test_httpx.py

@@ -55,3 +55,46 @@ async def serve() -> None:
             result.raise_for_status()
 
         shutdown.set()
+
+
+@pytest.mark.trio
+async def test_handle_isolate_state():
+    config = Config()
+    config.bind = ["0.0.0.0:1234"]
+    config.accesslog = "-"  # Log to stdout/err
+    config.errorlog = "-"
+
+    async def app(scope, receive, send):
+        assert scope["type"] == "http"
+
+        await send(
+            {
+                "type": "http.response.start",
+                "status": 200,
+                "headers": [[b"content-type", b"text/plain"]],
+            }
+        )
+        await send(
+            {"type": "http.response.body", "body": scope["state"].get("key", b"")}
+        )
+        scope["state"]["key"] = b"one"
+
+    async with trio.open_nursery() as nursery:
+        shutdown = trio.Event()
+
+        async def serve() -> None:
+            await hypercorn.trio.serve(app, config, shutdown_trigger=shutdown.wait)
+
+        nursery.start_soon(serve)
+
+        await trio.testing.wait_all_tasks_blocked()
+
+        client = httpx.AsyncClient()
+
+        result = await client.get("http://0.0.0.0:1234/")
+        assert result.content == b""
+
+        result = await client.get("http://0.0.0.0:1234/")
+        assert result.content == b""
+
+        shutdown.set()