From 3050e61588bfc79b3b360fa25db32c81e3264efb Mon Sep 17 00:00:00 2001
From: Nikias Bassen
Date: Wed, 25 Sep 2019 19:23:20 +0200
Subject: Use global event handlers for normale/restore and recovery/dfu mode
 devices

---
 src/recovery.c | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src/recovery.c')

diff --git a/src/recovery.c b/src/recovery.c
index 436261f..c079d90 100644
--- a/src/recovery.c
+++ b/src/recovery.c
@@ -231,7 +231,13 @@ int recovery_enter_restore(struct idevicerestore_client_t* client, plist_t build
 		return -1;
 	}
 
-	client->mode = &idevicerestore_modes[MODE_RESTORE];
+	debug("DEBUG: Waiting for device to disconnect...\n");
+	WAIT_FOR(client->mode != &idevicerestore_modes[MODE_RECOVERY] || (client->flags & FLAG_QUIT), 30);
+	if (client->mode == &idevicerestore_modes[MODE_RECOVERY]) {
+		error("ERROR: Failed to place device in restore mode\n");
+		return -1;
+	}
+
 	return 0;
 }
 
-- 
cgit v1.1-32-gdbae