From 01510de60e81c93636265888c88ace8944a1e633 Mon Sep 17 00:00:00 2001 From: meejah Date: Tue, 27 Sep 2022 15:48:26 -0600 Subject: [PATCH] register producers/consumers for TCP and WebSocket --- src/wormhole_transit_relay/transit_server.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/wormhole_transit_relay/transit_server.py b/src/wormhole_transit_relay/transit_server.py index 4b7b0b5..ad24d6c 100644 --- a/src/wormhole_transit_relay/transit_server.py +++ b/src/wormhole_transit_relay/transit_server.py @@ -47,6 +47,7 @@ class TransitConnection(LineReceiver): ITransitClient API """ self._buddy = other + self._buddy._client.transport.registerProducer(self.transport, True) def disconnect_partner(self): """ @@ -198,6 +199,7 @@ class WebSocketTransitConnection(WebSocketServerProtocol): ITransitClient API """ self._buddy = other + self._buddy._client.transport.registerProducer(self.transport, True) def disconnect_partner(self): """