From 59fdd010aa845557b0eb90b98959c0cd112b638c Mon Sep 17 00:00:00 2001
From: Joshua Hill
Date: Tue, 8 Jun 2010 20:34:28 -0400
Subject: started to add support multiple build identities

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

(limited to 'src/restore.c')

diff --git a/src/restore.c b/src/restore.c
index fc22b75..fa2ccdd 100644
--- a/src/restore.c
+++ b/src/restore.c
@@ -144,6 +144,7 @@ void restore_device_callback(const idevice_event_t* event, void* user_data) {
 
 	} else if (event->event == IDEVICE_DEVICE_REMOVE) {
 		restore_device_connected = 0;
+		idevicerestore_quit = 1;
 	}
 }
 
@@ -498,7 +499,11 @@ int restore_handle_data_request_msg(idevice_t device, restored_client_t restore,
 		}
 
 		else if (!strcmp(type, "NORData")) {
-			restore_send_nor(restore, ipsw, tss);
+			if(!idevicerestore_exclude) {
+				restore_send_nor(restore, ipsw, tss);
+			} else {
+				idevicerestore_quit = 1;
+			}
 
 		} else {
 			// Unknown DataType!!
-- 
cgit v1.1-32-gdbae