idevice: properly handle partial SSL writes

1 files changed, 15 insertions, 8 deletions

diff --git a/src/idevice.c b/src/idevice.c
index 90b531d..06991c5 100644
--- a/src/idevice.c
+++ b/src/idevice.c
@@ -384,18 +384,25 @@ LIBIMOBILEDEVICE_API idevice_error_t idevice_connection_send(idevice_connection_
         }
 
         if (connection->ssl_data) {
+                uint32_t sent = 0;
+                while (sent < len) {
 #ifdef HAVE_OPENSSL
-                int sent = SSL_write(connection->ssl_data->session, (const void*)data, (int)len);
-                debug_info("SSL_write %d, sent %d", len, sent);
+                        int s = SSL_write(connection->ssl_data->session, (const void*)(data+sent), (int)(len-sent));
 #else
-                ssize_t sent = gnutls_record_send(connection->ssl_data->session, (void*)data, (size_t)len);
+                        ssize_t s = gnutls_record_send(connection->ssl_data->session, (void*)(data+sent), (size_t)(len-sent));
 #endif
-                if ((uint32_t)sent == (uint32_t)len) {
-                        *sent_bytes = sent;
-                        return IDEVICE_E_SUCCESS;
+                        if (s < 0) {
+                                break;
+                        }
+                        sent += s;
                 }
-                *sent_bytes = 0;
-                return IDEVICE_E_SSL_ERROR;
+                debug_info("SSL_write %d, sent %d", len, sent);
+                if (sent < len) {
+                        *sent_bytes = 0;
+                        return IDEVICE_E_SSL_ERROR;
+                }
+                *sent_bytes = sent;
+                return IDEVICE_E_SUCCESS;
         }
         return internal_connection_send(connection, data, len, sent_bytes);
 }