Fix WLANHandler.write returning None

bessman · bessman · commit 10fd7554888f · 2025-02-19T18:41:45.000+01:00
diff --git a/pslab/connection/wlan.py b/pslab/connection/wlan.py
@@ -101,7 +101,16 @@ def write(self, data: bytes) -> int:
         numbytes : int
             Number of bytes written.
         """
-        return self._sock.sendall(data)
+        buf_size = 4096
+        remaining = len(data)
+        sent = 0
+
+        while remaining > 0:
+            chunk = data[sent : min(remaining, buf_size)]
+            sent += self._sock.send(chunk)
+            remaining -= len(chunk)
+
+        return sent
 
     def __repr__(self) -> str:  # noqa
         return f"{self.__class__.__name__}[{self.host}:{self.port}]"
