Perform proper goodby on lockdown shutdown.

1 files changed, 55 insertions, 4 deletions

diff --git a/src/iphone.c b/src/iphone.c
index 32d27f6..1f68180 100644
--- a/src/iphone.c
+++ b/src/iphone.c
@@ -28,6 +28,49 @@
 #include <stdlib.h>
 #include <string.h>
 
+/**
+ * This function sets the configuration of the given device to 3
+ * and claims the interface 1. If usb_set_configuration fails, it detaches
+ * the kernel driver that blocks the device, and retries configuration.
+ *
+ * @param phone which device to configure
+ */
+static void iphone_config_usb_device(iphone_device_t phone)
+{
+        int ret;
+
+        log_debug_msg("setting configuration... ");
+        ret = usb_set_configuration(phone->device, 3);
+        if (ret != 0) {
+                log_debug_msg("Hm, usb_set_configuration returned %d: %s, trying to fix:\n", ret, strerror(-ret));
+                log_debug_msg("-> detaching kernel driver... ");
+                ret =
+                        usb_detach_kernel_driver_np(phone->device,
+                                                                                phone->__device->config->interface->altsetting->bInterfaceNumber);
+                if (ret != 0) {
+                        log_debug_msg("usb_detach_kernel_driver_np returned %d: %s\n", ret, strerror(-ret));
+                } else {
+                        log_debug_msg("done.\n");
+                        log_debug_msg("setting configuration again... ");
+                        ret = usb_set_configuration(phone->device, 3);
+                        if (ret != 0) {
+                                log_debug_msg("Error: usb_set_configuration returned %d: %s\n", ret, strerror(-ret));
+                        } else {
+                                log_debug_msg("done.\n");
+                        }
+                }
+        } else {
+                log_debug_msg("done.\n");
+        }
+
+        log_debug_msg("claiming interface... ");
+        ret = usb_claim_interface(phone->device, 1);
+        if (ret != 0) {
+                log_debug_msg("Error: usb_claim_interface returned %d: %s\n", ret, strerror(-ret));
+        } else {
+                log_debug_msg("done.\n");
+        }
+}
 
 /**
  * Given a USB bus and device number, returns a device handle to the iPhone on
@@ -73,8 +116,7 @@ static iphone_error_t iphone_get_specific_device(unsigned int bus_n, int dev_n,
                                 if (dev->devnum == dev_n) {
                                         phone->__device = dev;
                                         phone->device = usb_open(phone->__device);
-                                        usb_set_configuration(phone->device, 3);
-                                        usb_claim_interface(phone->device, 1);
+                                        iphone_config_usb_device(phone);
                                         goto found;
                                 }
 
@@ -115,9 +157,10 @@ static iphone_error_t iphone_get_specific_device(unsigned int bus_n, int dev_n,
                 return IPHONE_E_SUCCESS;
         } else {
                 // Bad header
+                log_debug_msg("get_iPhone(): Received a bad header/invalid version number.\n");
+                log_debug_buffer((char *) version, sizeof(*version));
                 iphone_free_device(phone);
                 free(version);
-                log_debug_msg("get_iPhone(): Received a bad header/invalid version number.");
                 return IPHONE_E_BAD_HEADER;
         }
 
@@ -173,13 +216,21 @@ iphone_error_t iphone_free_device(iphone_device_t device)
         if (!device)
                 return IPHONE_E_INVALID_ARG;
         iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
+        int bytes;
+        unsigned char buf[512];
+
+        // read final package
+        bytes = usb_bulk_read(device->device, BULKIN, (void *) &buf, 512, 1000);
+        if (bytes > 0) {
+                log_debug_msg("iphone_free_device: final read returned\n");
+                log_debug_buffer(buf, bytes);
+        }
 
         if (device->buffer) {
                 free(device->buffer);
         }
         if (device->device) {
                 usb_release_interface(device->device, 1);
-                usb_reset(device->device);
                 usb_close(device->device);
                 ret = IPHONE_E_SUCCESS;
         }