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

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