diff options
| -rw-r--r-- | src/idevicerestore.c | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/idevicerestore.c b/src/idevicerestore.c index be8c9c7..fa685bd 100644 --- a/src/idevicerestore.c +++ b/src/idevicerestore.c @@ -43,6 +43,7 @@ int idevicerestore_debug = 0; static int idevicerestore_mode = 0; +static int idevicerestore_quit = 0; static int idevicerestore_custom = 0; void usage(int argc, char* argv[]); @@ -344,12 +345,11 @@ int main(int argc, char* argv[]) { info("Device has successfully entered restore mode\n"); /* start restore process */ - int quit_flag = 0; char* kernelcache = NULL; info("Restore protocol version is %llu.\n", version); restore_error = restored_start_restore(restore); if (restore_error == RESTORE_E_SUCCESS) { - while (!quit_flag) { + while (!idevicerestore_quit) { plist_t message = NULL; restore_error = restored_receive(restore, &message); plist_t msgtype_node = plist_dict_get_item(message, "MsgType"); @@ -398,7 +398,7 @@ int main(int argc, char* argv[]) { if (RESTORE_E_SUCCESS != restore_error) { error("Invalid return status %d\n", restore_error); - //quit_flag = 1; + //idevicerestore_quit = 1; } plist_free(message); @@ -417,6 +417,8 @@ int main(int argc, char* argv[]) { void device_callback(const idevice_event_t* event, void *user_data) { if (event->event == IDEVICE_DEVICE_ADD) { idevicerestore_mode = RESTORE_MODE; + } else if(event->event == IDEVICE_DEVICE_REMOVE) { + idevicerestore_quit = 1; } } |