summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorGravatar Nikias Bassen2014-03-24 03:24:00 +0100
committerGravatar Nikias Bassen2014-03-24 03:24:00 +0100
commit185eb2b935ea307d9ed1ad2d00240c953132e558 (patch)
tree177868fd1c84caea53137bcd302c5abfea9f091a /src
parent4e4dec264adcc1b847808663205d44b84168481f (diff)
downloadlibusbmuxd-185eb2b935ea307d9ed1ad2d00240c953132e558.tar.gz
libusbmuxd-185eb2b935ea307d9ed1ad2d00240c953132e558.tar.bz2
Properly handle segmented send/receive situations
Diffstat (limited to 'src')
-rw-r--r--src/libusbmuxd.c24
1 files changed, 20 insertions, 4 deletions
diff --git a/src/libusbmuxd.c b/src/libusbmuxd.c
index 2d27ba3..11af687 100644
--- a/src/libusbmuxd.c
+++ b/src/libusbmuxd.c
@@ -179,8 +179,16 @@ static int receive_packet(int sfd, struct usbmuxd_header *header, void **payload
uint32_t payload_size = hdr.length - sizeof(hdr);
if (payload_size > 0) {
payload_loc = (char*)malloc(payload_size);
- if (socket_receive_timeout(sfd, payload_loc, payload_size, 0, 5000) != (int)payload_size) {
- DEBUG(1, "%s: Error receiving payload of size %d\n", __func__, payload_size);
+ uint32_t rsize = 0;
+ do {
+ int res = socket_receive_timeout(sfd, payload_loc + rsize, payload_size - rsize, 0, 5000);
+ if (res < 0) {
+ break;
+ }
+ rsize += res;
+ } while (rsize < payload_size);
+ if (rsize != payload_size) {
+ DEBUG(1, "%s: Error receiving payload of size %d (bytes received: %d)\n", __func__, payload_size, rsize);
free(payload_loc);
return -EBADMSG;
}
@@ -344,10 +352,18 @@ static int send_packet(int sfd, uint32_t message, uint32_t tag, void *payload, u
return -1;
}
if (payload && (payload_size > 0)) {
- sent += socket_send(sfd, payload, payload_size);
+ uint32_t ssize = 0;
+ do {
+ int res = socket_send(sfd, (char*)payload + ssize, payload_size - ssize);
+ if (res < 0) {
+ break;
+ }
+ ssize += res;
+ } while (ssize < payload_size);
+ sent += ssize;
}
if (sent != (int)header.length) {
- DEBUG(1, "%s: ERROR: could not send whole packet\n", __func__);
+ DEBUG(1, "%s: ERROR: could not send whole packet (sent %d of %d)\n", __func__, sent, header.length);
socket_close(sfd);
return -1;
}