Display iBoot boot stage

This helps debugging cases where the iDevice does not go into stage 2
because of a missing firmware

2 files changed, 29 insertions, 0 deletions

diff --git a/src/dfu.c b/src/dfu.c
index 0c3c480..4487ade 100644
--- a/src/dfu.c
+++ b/src/dfu.c
@@ -473,6 +473,21 @@ int dfu_enter_recovery(struct idevicerestore_client_t* client, plist_t build_ide
 				return -1;
 			}
 
+			char *value = NULL;
+			unsigned long boot_stage = 0;
+			irecv_getenv(client->dfu->client, "boot-stage", &value);
+			if (value) {
+				boot_stage = strtoul(value, NULL, 0);
+			}
+			if (boot_stage > 0) {
+				info("iBoot boot-stage=%s\n", value);
+				free(value);
+				value = NULL;
+				if (boot_stage != 1) {
+					error("ERROR: iBoot should be at boot stage 1, continuing anyway...\n");
+				}
+			}
+
 			if (dfu_send_iboot_stage1_components(client, build_identity) < 0) {
 				mutex_unlock(&client->device_event_mutex);
 				error("ERROR: Unable to send iBoot stage 1 components to device\n");
diff --git a/src/recovery.c b/src/recovery.c
index a9d6b61..02f5689 100644
--- a/src/recovery.c
+++ b/src/recovery.c
@@ -158,6 +158,20 @@ int recovery_enter_restore(struct idevicerestore_client_t* client, plist_t build
 	free(value);
 	value = NULL;
 
+	unsigned long boot_stage = 0;
+	irecv_getenv(client->recovery->client, "boot-stage", &value);
+	if (value) {
+		boot_stage = strtoul(value, NULL, 0);
+	}
+	if (boot_stage > 0) {
+		info("iBoot boot-stage=%s\n", value);
+		free(value);
+		value = NULL;
+		if (boot_stage != 2) {
+			error("ERROR: iBoot should be at boot stage 2, continuing anyway...\n");
+		}
+	}
+
 	unsigned long radio_error = 0;
 	irecv_getenv(client->recovery->client, "radio-error", &value);
 	if (value) {