Guard to make sure we can't send data while the transport is closing

zigpy · Sep 10, 2024 · 4dbe60a · 4dbe60a
1 parent d384755
commit 4dbe60a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bellows/ash.py b/bellows/ash.py
@@ -579,6 +579,9 @@ def _write_frame(
         prefix: tuple[Reserved] = (),
         suffix: tuple[Reserved] = (Reserved.FLAG,),
     ) -> None:
+        if self._transport is None or self._transport.is_closing():
+            raise NcpFailure("Transport is closed, cannot send frame")
+
         if _LOGGER.isEnabledFor(logging.DEBUG):
             prefix_str = "".join([f"{r.name} + " for r in prefix])
             suffix_str = "".join([f" + {r.name}" for r in suffix])