From 9308e665a2741690b4fa50fa494d7568dda95a28 Mon Sep 17 00:00:00 2001
From: Nikias Bassen
Date: Wed, 20 Nov 2013 11:32:26 +0100
Subject: refactor component personalization and add support for Image4
 stitching

---
 src/dfu.c | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

(limited to 'src/dfu.c')

diff --git a/src/dfu.c b/src/dfu.c
index 5fdbfd5..6050ed0 100644
--- a/src/dfu.c
+++ b/src/dfu.c
@@ -179,11 +179,23 @@ int dfu_send_component(struct idevicerestore_client_t* client, plist_t build_ide
 		}
 	}
 
-	if (ipsw_get_component_by_path(client->ipsw, client->tss, component, path, &data, &size) < 0) {
-		error("ERROR: Unable to get component: %s\n", component);
+	unsigned char* component_data = NULL;
+	unsigned int component_size = 0;
+
+	if (extract_component(client->ipsw, path, &component_data, &component_size) < 0) {
+		error("ERROR: Unable to extract component: %s\n", component);
+		free(path);
+		return -1;
+	}
+
+	if (personalize_component(component, component_data, component_size, client->tss, &data, &size) < 0) {
+		error("ERROR: Unable to get personalized component: %s\n", component);
+		free(component_data);
 		free(path);
 		return -1;
 	}
+	free(component_data);
+	component_data = NULL;
 
 	if (!(client->flags & FLAG_CUSTOM) && (strcmp(component, "iBEC") == 0)) {
 		unsigned char* ticket = NULL;
-- 
cgit v1.1-32-gdbae