From 03a30a450af09a27de8ab1b9e110d0b09d59bf3c Mon Sep 17 00:00:00 2001 From: Shunsuke Shibayama Date: Wed, 26 Apr 2023 23:59:23 +0900 Subject: [PATCH] Update repl_server.py --- src/scripts/repl_server.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/src/scripts/repl_server.py b/src/scripts/repl_server.py index bcfd1693..0759f343 100644 --- a/src/scripts/repl_server.py +++ b/src/scripts/repl_server.py @@ -26,16 +26,10 @@ class MessageStream: def recv_msg(self): self._read_buf.clear() - # requires at least 3 bytes as metadata - while len(self._read_buf) < 3: - self._read_buf.extend(self.socket.recv(1024)) - + self._read_buf.extend(self.socket.recv(3)) inst = int.from_bytes(self._read_buf[:1], 'big') data_len = int.from_bytes(self._read_buf[1:3], 'big') - - # until all data has been read - while len(self._read_buf) < 3 + data_len: - self._read_buf.extend(self.socket.recv(1024)) + self._read_buf.extend(self.socket.recv(data_len)) return (inst, self._read_buf[3:].decode())