idevice: add error checking to internal_ssl_write()

Returning 0 bytes sent upon error causes an infinite loop
within the calling gnutls code. Returning -1 as an error
code allows gnutls to properly detect and recover.

1 files changed, 5 insertions, 1 deletions

diff --git a/src/idevice.c b/src/idevice.c
index cad1431..af87e61 100644
--- a/src/idevice.c
+++ b/src/idevice.c
@@ -504,9 +504,13 @@ static ssize_t internal_ssl_read(gnutls_transport_ptr_t transport, char *buffer,
 static ssize_t internal_ssl_write(gnutls_transport_ptr_t transport, char *buffer, size_t length)
 {
 	uint32_t bytes = 0;
+	idevice_error_t res;
 	idevice_connection_t connection = (idevice_connection_t)transport;
 	debug_info("pre-send length = %zi", length);
-	internal_connection_send(connection, buffer, length, &bytes);
+	if ((res = internal_connection_send(connection, buffer, length, &bytes)) != IDEVICE_E_SUCCESS) {
+		debug_info("ERROR: internal_connection_send returned %d", res);
+		return -1;
+	}
 	debug_info("post-send sent %i bytes", bytes);
 	return bytes;
 }