Allow server to receive files more than initial receive buffer size

Author: denis.kashitsyn

python/mujinasync/asynctcp.py

@@ -364,19 +364,30 @@ def SpinOnce(self, timeout=0):
                 continue
 
             serverClient, connection = socketConnections[rsocket]
-            try:
-                received = rsocket.recv_into(connection.receiveBuffer.writeView)
-            except Exception as e:
-                connection.closeType = 'Immediate'
-                log.exception('error while trying to receive from connection %s: %s', connection, e)
+            totalReceived = 0
+            while True:
+                try:
+                    received = rsocket.recv_into(connection.receiveBuffer.writeView)
+                    if received == 0:
+                        break
+                    connection.receiveBuffer.size += received
+                    totalReceived += received
+                    if connection.receiveBuffer.capacity - connection.receiveBuffer.size < connection.receiveBuffer.capacity:
+                        connection.receiveBuffer.capacity *= 2
+                except socket.error as e:
+                    if e.errno not in (errno.EAGAIN, errno.EWOULDBLOCK):
+                        connection.closeType = 'Immediate'
+                        log.exception('error while trying to receive from connection %s: %s', connection, e)
+                    break
+
+            if connection.closeType == 'Immediate':
                 continue
 
-            if received == 0:
+            if totalReceived == 0:
                 connection.closeType = 'AfterSend'
                 log.debug('received nothing from connection, maybe closed: %s', connection)
                 continue
 
-            connection.receiveBuffer.size += received
             receivedConnections.append((serverClient, connection))
 
         # handle sockets that can write