Raise an EOFException on short reads

polypheny · Oct 8, 2024 · 56ff490 · 56ff490
1 parent 7794722
commit 56ff490
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/polypheny/rpc.py b/polypheny/rpc.py
@@ -57,6 +57,8 @@ def send_msg(self, serialized):
 
     def recv_msg(self):
         bl = self.con.recv(8)
+        if len(bl) != 8:
+            raise EOFError
         n = int.from_bytes(bl, 'little')
         raw = self.con.recv(n)
         if len(raw) != n: