core: Don't crash if a stream is missing a side.

7 years ago · 0a4711bce9
parent 43c673b230
commit 0a4711bce9
1 changed files with 4 additions and 2 deletions
--- a/mitogen/core.py
+++ b/mitogen/core.py
@ -619,8 +619,10 @@ class BasicStream(object):
        LOG.debug('%r.on_disconnect()', self)
        broker.stop_receive(self)
        broker.stop_transmit(self)
-        self.receive_side.close()
-        self.transmit_side.close()
+        if self.receive_side:
+            self.receive_side.close()
+        if self.transmit_side:
+            self.transmit_side.close()
        fire(self, 'disconnect')

    def on_shutdown(self, broker):