From a91e336c24a0d741e47be7adf0cd0b2beb20e5ab Mon Sep 17 00:00:00 2001
From: Joshua Hill
Date: Mon, 7 Jun 2010 02:24:08 -0400
Subject: I really need to put more descriptive messages here, but im doing
 stuff all over the place

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

(limited to 'src/recovery.c')

diff --git a/src/recovery.c b/src/recovery.c
index e67b10d..52f4802 100644
--- a/src/recovery.c
+++ b/src/recovery.c
@@ -31,6 +31,13 @@
 #include "recovery.h"
 #include "idevicerestore.h"
 
+int recovery_progress_callback(irecv_client_t client, const irecv_event_t* event) {
+	if (event->type == IRECV_PROGRESS) {
+		print_progress_bar(event->data, event->progress);
+	}
+	return 0;
+}
+
 int recovery_check_mode() {
 	irecv_client_t recovery = NULL;
 	irecv_error_t recovery_error = IRECV_E_SUCCESS;
@@ -92,8 +99,9 @@ int recovery_enter_restore(const char* uuid, const char* ipsw, plist_t tss) {
 		error("ERROR: Unable to connect to device in restore mode\n");
 		return -1;
 	}
-	restore_close(device, restore);
 
+	restore_close(device, restore);
+	idevicerestore_mode = MODE_RESTORE;
 	return 0;
 }
 
@@ -149,6 +157,7 @@ int recovery_open_with_timeout(irecv_client_t* client) {
 		debug("Retrying connection...\n");
 	}
 
+	irecv_event_subscribe(recovery, IRECV_PROGRESS, &recovery_progress_callback, NULL);
 	*client = recovery;
 	return 0;
 }
-- 
cgit v1.1-32-gdbae