Merge pull request #322 from forrejam/patch-1

Check if stdout at EOF before read
carlmontanari · Apr 9, 2024 · e0334ad · e0334ad
2 parents ac14468 + b725386
commit e0334ad
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: