idevice_connection_send: Make sure send works with non-blocking sockets

1 files changed, 17 insertions, 1 deletions

diff --git a/src/idevice.c b/src/idevice.c
index 418ee86..7b3625e 100644
--- a/src/idevice.c
+++ b/src/idevice.c
@@ -565,8 +565,24 @@ LIBIMOBILEDEVICE_API idevice_error_t idevice_connection_send(idevice_connection_
 		}
 		*sent_bytes = sent;
 		return IDEVICE_E_SUCCESS;
+	} else {
+		uint32_t sent = 0;
+		while (sent < len) {
+			uint32_t bytes = 0;
+			int s = internal_connection_send(connection, data+sent, len-sent, &bytes);
+			if (s < 0) {
+				break;
+			}
+			sent += bytes;
+		}
+		debug_info("internal_connection_send %d, sent %d", len, sent);
+		if (sent < len) {
+			*sent_bytes = 0;
+			return IDEVICE_E_NOT_ENOUGH_DATA;
+		}
+		*sent_bytes = sent;
+		return IDEVICE_E_SUCCESS;
 	}
-	return internal_connection_send(connection, data, len, sent_bytes);
 }
 
 static idevice_error_t socket_recv_to_idevice_error(int conn_error, uint32_t len, uint32_t received)