From 75a3f5f5c6ec5b89659f4e4aacd5d629a3912f22 Mon Sep 17 00:00:00 2001
From: Joshua Hill
Date: Mon, 31 May 2010 18:36:41 -0400
Subject: Added code to automatically detect when connection is lost
---
src/idevicerestore.c | 8 +++++---
1 file 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;
}
}
--
cgit v1.1-32-gdbae