From a18ad835cb84268fa07118cf6abed01d1e93856a Mon Sep 17 00:00:00 2001
From: Nikias Bassen
Date: Thu, 26 Sep 2019 04:36:39 +0200
Subject: Make sure CTRL+C is working at specific stages of the process

---
 src/normal.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/normal.c')

diff --git a/src/normal.c b/src/normal.c
index 58d907c..14fa0f6 100644
--- a/src/normal.c
+++ b/src/normal.c
@@ -239,14 +239,14 @@ int normal_enter_recovery(struct idevicerestore_client_t* client)
 
 	debug("DEBUG: Waiting for device to disconnect...\n");
 	WAIT_FOR(client->mode != &idevicerestore_modes[MODE_NORMAL] || (client->flags & FLAG_QUIT), 30);
-	if (client->mode == &idevicerestore_modes[MODE_NORMAL]) {
+	if (client->mode == &idevicerestore_modes[MODE_NORMAL] || (client->flags & FLAG_QUIT)) {
 		error("ERROR: Failed to place device in recovery mode\n");
 		return -1;
 	}
 
 	debug("DEBUG: Waiting for device to connect in recovery mode...\n");
 	WAIT_FOR(client->mode == &idevicerestore_modes[MODE_RECOVERY] || (client->flags & FLAG_QUIT), 30);
-	if (client->mode != &idevicerestore_modes[MODE_RECOVERY]) {
+	if (client->mode != &idevicerestore_modes[MODE_RECOVERY] || (client->flags & FLAG_QUIT)) {
 		error("ERROR: Failed to enter recovery mode\n");
 		return -1;
 	}
-- 
cgit v1.1-32-gdbae