From 245bc9b6510a8900c69c2fd543434bae1bc6ca6a Mon Sep 17 00:00:00 2001
From: BALATON Zoltan
Date: Tue, 5 May 2015 00:58:02 +0200
Subject: Clean up free() usage and plug some potential memory leaks

---
 src/recovery.c | 38 +++++++++++++++-----------------------
 1 file changed, 15 insertions(+), 23 deletions(-)

(limited to 'src/recovery.c')

diff --git a/src/recovery.c b/src/recovery.c
index 91c43f3..5930db6 100644
--- a/src/recovery.c
+++ b/src/recovery.c
@@ -172,16 +172,14 @@ int recovery_enter_restore(struct idevicerestore_client_t* client, plist_t build
 	char* value = NULL;
 	irecv_getenv(client->recovery->client, "build-version", &value);
 	info("iBoot build-version=%s\n", (value) ? value : "(unknown)");
-	if (value) {
-		free(value);
-		value = NULL;
-	}
+	free(value);
+	value = NULL;
+
 	irecv_getenv(client->recovery->client, "build-style", &value);
 	info("iBoot build-style=%s\n", (value) ? value : "(unknown)");
-	if (value) {
-		free(value);
-		value = NULL;
-	}
+	free(value);
+	value = NULL;
+
 	unsigned long radio_error = 0;
 	irecv_getenv(client->recovery->client, "radio-error", &value);
 	if (value) {
@@ -242,12 +240,11 @@ int recovery_send_ticket(struct idevicerestore_client_t* client)
 
 	info("Sending APTicket (%d bytes)\n", size);
 	irecv_error_t err = irecv_send_buffer(client->recovery->client, data, size, 0);
+	free(data);
 	if (err != IRECV_E_SUCCESS) {
 		error("ERROR: Unable to send APTicket: %s\n", irecv_strerror(err));
-		free(data);
 		return -1;
 	}
-	free(data);
 
 	err = irecv_send_command(client->recovery->client, "ticket");
 	if (err != IRECV_E_SUCCESS) {
@@ -272,42 +269,37 @@ int recovery_send_component(struct idevicerestore_client_t* client, plist_t buil
 	if (!path) {
 		if (build_identity_get_component_path(build_identity, component, &path) < 0) {
 			error("ERROR: Unable to get path for component '%s'\n", component);
-			if (path)
-				free(path);
+			free(path);
 			return -1;
 		}
 	}
 
 	unsigned char* component_data = NULL;
 	unsigned int component_size = 0;
-
-	if (extract_component(client->ipsw, path, &component_data, &component_size) < 0) {
+	int ret = extract_component(client->ipsw, path, &component_data, &component_size);
+	free(path);
+	if (ret < 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) {
+	ret = personalize_component(component, component_data, component_size, client->tss, &data, &size);
+	free(component_data);
+	if (ret < 0) {
 		error("ERROR: Unable to get personalized component: %s\n", component);
-		free(component_data);
-		free(path);
 		return -1;
 	}
-	free(component_data);
-	component_data = NULL;	
 
 	info("Sending %s (%d bytes)...\n", component, size);
 
 	// FIXME: Did I do this right????
 	err = irecv_send_buffer(client->recovery->client, data, size, 0);
-	free(path);
+	free(data);
 	if (err != IRECV_E_SUCCESS) {
 		error("ERROR: Unable to send %s component: %s\n", component, irecv_strerror(err));
-		free(data);
 		return -1;
 	}
 
-	free(data);
 	return 0;
 }
 
-- 
cgit v1.1-32-gdbae