Check if stdout at EOF before read

fixes #321
carlmontanari · Apr 9, 2024 · b725386 · b725386
1 parent ac14468
commit b725386
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scrapli/transport/plugins/asyncssh/transport.py b/scrapli/transport/plugins/asyncssh/transport.py
@@ -259,6 +259,9 @@ async def read(self) -> bytes:
         if not self.stdout:
             raise ScrapliConnectionNotOpened
 
+        if self.stdout.at_eof():
+            raise ScrapliConnectionError("transport at EOF; no more data to be read")
+
         try:
             buf: bytes = await self.stdout.read(65535)
         except ConnectionLost as exc: