implement irecv_set_interface for win32

1 files changed, 22 insertions, 4 deletions

diff --git a/libirecovery.c b/libirecovery.c
index 3b839f9..04845ad 100644
--- a/libirecovery.c
+++ b/libirecovery.c
@@ -618,6 +618,21 @@ irecv_error_t irecv_open(irecv_client_t* pclient, unsigned long long ecid) {
 	return IRECV_E_UNABLE_TO_CONNECT;
 #else
 	int ret = mobiledevice_connect(pclient, ecid);
+	if (ret == IRECV_E_SUCCESS) {
+		irecv_client_t client = *pclient;
+		int error = IRECV_E_SUCCESS;
+		if ((client->mode != kDfuMode) && (client->mode != kWTFMode)) {
+			error = irecv_set_interface(client, 0, 0);
+			if (client->mode > kRecoveryMode2) {
+				error = irecv_set_interface(client, 1, 1);
+			}
+		} else {
+			error = irecv_set_interface(client, 0, 0);
+		}
+		if (error != IRECV_E_SUCCESS) {
+			debug("WARNING: set interface failed, error %d\n", error);
+		}
+	}
 	return ret;
 #endif
 }
@@ -644,12 +659,12 @@ irecv_error_t irecv_set_configuration(irecv_client_t client, int configuration)
 
 irecv_error_t irecv_set_interface(irecv_client_t client, int interface, int alt_interface) {
 	if (check_context(client) != IRECV_E_SUCCESS) return IRECV_E_NO_DEVICE;
-	
+
+	debug("Setting to interface %d:%d\n", interface, alt_interface);
 #ifndef WIN32
 	// pod2g 2011-01-07: we may want to claim multiple interfaces
 	//libusb_release_interface(client->handle, client->interface);
 
-	debug("Setting to interface %d:%d\n", interface, alt_interface);
 	if (libusb_claim_interface(client->handle, interface) < 0) {
 		return IRECV_E_USB_INTERFACE;
 	}
@@ -657,10 +672,13 @@ irecv_error_t irecv_set_interface(irecv_client_t client, int interface, int alt_
 	if (libusb_set_interface_alt_setting(client->handle, interface, alt_interface) < 0) {
 		return IRECV_E_USB_INTERFACE;
 	}
-
+#else
+	if (irecv_control_transfer(client, 0, 0x0B, alt_interface, interface, NULL, 0, USB_TIMEOUT) < 0) {
+		return IRECV_E_USB_INTERFACE;
+	}
+#endif
 	client->interface = interface;
 	client->alt_interface = alt_interface;
-#endif
 
 	return IRECV_E_SUCCESS;
 }