3 files changed, 33 insertions, 5 deletions

diff --git a/Makefile b/Makefile
index 2036ddc..3971798 100644
--- a/Makefile
+++ b/Makefile
@@ -1,5 +1,5 @@
 TARGETS=usbmuxd iproxy testclient
-CFLAGS=-Wall -g
+CFLAGS=-Wall -g -DDEBUG
 LIBS=-lpthread -lusb -lrt
 LDFLAGS=
 
diff --git a/iphone.c b/iphone.c
index bf0d5de..0a2b6cc 100644
--- a/iphone.c
+++ b/iphone.c
@@ -449,9 +449,11 @@ int send_to_phone(iphone_device_t phone, char *data, int datalen)
     int bytes = 0;
 
 #ifdef DEBUG
+    #ifdef DEBUG_MORE
     printf("===============================\n%s: trying to send\n", __func__);
     print_buffer(data, datalen);
     printf("===============================\n");
+    #endif
 #endif
     do {
         if (retrycount > 3) {
@@ -980,6 +982,7 @@ uint32 append_receive_buffer(iphone_umux_client_t client, char* packet)
 
     // ensure there is enough space, either by first malloc or realloc
     if (datalen > 0) {
+        fprintf(stderr, "%s: putting %d bytes into client's recv_buffer\n", __func__, datalen);
         if (client->r_len == 0) dobroadcast = 1;
 
         if (client->recv_buffer == NULL) {
@@ -1077,6 +1080,7 @@ void iphone_mux_pullbulk(iphone_device_t phone)
         // to construct a full packet, including its data
         uint32 packetlen = ntohl(header->length);
         if (usbReceive.leftover < packetlen) {
+            printf("%s: not enough data to construct a full packet\n", __func__);
             break;
         }
 
@@ -1087,6 +1091,7 @@ void iphone_mux_pullbulk(iphone_device_t phone)
         }
         else {
             // stuff the data
+            fprintf(stderr, "%s: found client, calling append_receive_buffer\n", __func__);
             append_receive_buffer(client, cursor);
         }
 
@@ -1104,6 +1109,7 @@ void iphone_mux_pullbulk(iphone_device_t phone)
     // if there are no leftovers, we just leave the datastructure as is,
     // and re-use the block next time.
     if (usbReceive.leftover > 0 && cursor != usbReceive.buffer) {
+        fprintf(stderr, "%s: we got a leftover, so handle it\n", __func__);
         char* newbuff = malloc(DEFAULT_CAPACITY);
         memcpy(newbuff, cursor, usbReceive.leftover);
         free(usbReceive.buffer);
diff --git a/main.c b/main.c
index ec9b007..8fa2b81 100644
--- a/main.c
+++ b/main.c
@@ -74,6 +74,7 @@ struct client_data {
 static struct device_use_info **device_use_list = NULL;
 static int device_use_count = 0;
 static pthread_mutex_t usbmux_mutex = PTHREAD_MUTEX_INITIALIZER;
+static pthread_mutex_t usb_mutex = PTHREAD_MUTEX_INITIALIZER;
 
 #ifdef DEBUG
 /**
@@ -167,6 +168,7 @@ static int usbmuxd_get_request(int fd, void **data, size_t len)
 static int usbmuxd_send_result(int fd, uint32_t tag, uint32_t result_code)
 {
     struct usbmuxd_result res;
+    int ret;
 
     res.header.length = sizeof(res);
     res.header.reserved = 0;
@@ -176,7 +178,9 @@ static int usbmuxd_send_result(int fd, uint32_t tag, uint32_t result_code)
 
     fprintf(stderr, "%s: tag=%d result=%d\n", __func__, res.header.tag, res.result);
 
-    return send_buf(fd, &res, sizeof(res));
+    ret = send_buf(fd, &res, sizeof(res));
+    fsync(fd); // let's get it sent
+    return ret;
 }
 
 /**
@@ -278,7 +282,7 @@ static int usbmuxd_handleConnectResult(struct client_data *cdata)
         }
     } else {
         result = 0;
-        err = iphone_mux_recv_timeout(cdata->muxclient, buffer, maxlen, &rlen, DEFAULT_TIMEOUT);
+        err = iphone_mux_recv_timeout(cdata->muxclient, buffer, maxlen, &rlen, 1000);
         if (err != 0) {
             fprintf(stderr, "%s: encountered USB read error: %d\n", __func__, err);
             usbmuxd_send_result(cdata->socket, cdata->tag, -err);
@@ -505,11 +509,17 @@ static void *usbmuxd_client_init_thread(void *arg)
         goto leave;
     }
 
+    pthread_mutex_lock(&usb_mutex);
+    fprintf(stderr, "%s: usb init\n", __func__);
     // gather data about all iPhones/iPods attached
     usb_init();
+    fprintf(stderr, "%s: usb find busses\n", __func__);
     usb_find_busses();
+    fprintf(stderr, "%s: usb find devices\n", __func__);
     usb_find_devices();
 
+    fprintf(stderr, "%s: Looking for attached devices...\n", __func__);
+
     for (bus = usb_get_busses(); bus; bus = bus->next) {
         for (dev = bus->devices; dev; dev = dev->next) {
             if (dev->descriptor.idVendor == 0x05ac
@@ -548,13 +558,16 @@ static void *usbmuxd_client_init_thread(void *arg)
             }
         }
     }
+    pthread_mutex_unlock(&usb_mutex);
 
-    // now wait for connect request
     if (found <= 0) {
         fprintf(stderr, "%s: No attached iPhone/iPod devices found.\n", __func__);
         goto leave;
     }
 
+    fprintf(stderr, "%s: Waiting for connect request\n", __func__);
+
+    // now wait for connect request
     //memset(&c_req, 0, sizeof(c_req));
     if ((recv_len = usbmuxd_get_request(cdata->socket, (void**)&c_req, 0)) <= 0) {
         fprintf(stderr, "%s: Did not receive any connect request.\n", __func__);
@@ -590,12 +603,19 @@ connect:
     }
     if (!phone) {
         // if not found, make a new connection
+        fprintf(stderr, "%s: creating new usb connection, device_id=%d\n", __func__, c_req->device_id);
+
+        pthread_mutex_lock(&usb_mutex);
         if (iphone_get_specific_device(0, c_req->device_id, &phone) != IPHONE_E_SUCCESS) {
+            pthread_mutex_unlock(&usb_mutex);
             fprintf(stderr, "%s: device_id %d could not be opened\n", __func__, c_req->device_id);
             usbmuxd_send_result(cdata->socket, c_req->header.tag, ENODEV);
             goto leave;
         }
+        pthread_mutex_unlock(&usb_mutex);
+        
         // add to device list
+        fprintf(stderr, "%s: add to device list\n", __func__);
         cur_dev = (struct device_use_info*)malloc(sizeof(struct device_use_info));
         memset(cur_dev, 0, sizeof(struct device_use_info));
         cur_dev->use_count = 1;
@@ -615,7 +635,7 @@ connect:
         }
         pthread_mutex_unlock(&usbmux_mutex);
     } else {
-        fprintf(stderr, "%s: reusing usb connection device_id %d\n", __func__, c_req->device_id);
+        fprintf(stderr, "%s: reusing usb connection, device_id=%d\n", __func__, c_req->device_id);
     }
 
     // setup connection to iPhone/iPod
@@ -697,7 +717,9 @@ leave:
             cur_dev->use_count = 0;
             pthread_mutex_unlock(&cur_dev->mutex);
             pthread_join(cur_dev->bulk_reader, NULL);
+            pthread_mutex_lock(&usb_mutex);
             iphone_free_device(cur_dev->phone);
+            pthread_mutex_unlock(&usb_mutex);
             pthread_mutex_destroy(&cur_dev->writer_mutex);
             pthread_mutex_destroy(&cur_dev->mutex);
             free(cur_dev);