idevice: [OpenSSL] Handle non-blocking SSL_write

1 files changed, 13 insertions, 0 deletions

diff --git a/src/idevice.c b/src/idevice.c
index 7b3625e..d3f28c2 100644
--- a/src/idevice.c
+++ b/src/idevice.c
@@ -549,7 +549,20 @@ LIBIMOBILEDEVICE_API idevice_error_t idevice_connection_send(idevice_connection_
                 uint32_t sent = 0;
                 while (sent < len) {
 #ifdef HAVE_OPENSSL
+                        int c = socket_check_fd((int)(long)connection->data, FDM_WRITE, 100);
+                        if (c < 0) {
+                                break;
+                        } else if (c == 0) {
+                                continue;
+                        }
                         int s = SSL_write(connection->ssl_data->session, (const void*)(data+sent), (int)(len-sent));
+                        if (s <= 0) {
+                                int sslerr = SSL_get_error(connection->ssl_data->session, s);
+                                if (sslerr == SSL_ERROR_WANT_WRITE) {
+                                        continue;
+                                }
+                                break;
+                        }
 #else
                         ssize_t s = gnutls_record_send(connection->ssl_data->session, (void*)(data+sent), (size_t)(len-sent));
 #endif