Free lockdownd connection early for both backup and restore commands

1 files changed, 6 insertions, 8 deletions

diff --git a/tools/idevicebackup.c b/tools/idevicebackup.c
index 8f47433..6626020 100644
--- a/tools/idevicebackup.c
+++ b/tools/idevicebackup.c
@@ -857,6 +857,12 @@ int main(int argc, char *argv[])
 			}
 		}
 
+		/* close down the lockdown connection as it is no longer needed */
+		if (client) {
+			lockdownd_client_free(client);
+			client = NULL;
+		}
+
 		mobilebackup_error_t err;
 
 		/* Manifest.plist (backup manifest (backup state)) */
@@ -896,12 +902,6 @@ int main(int argc, char *argv[])
 			plist_free(info_plist);
 			info_plist = NULL;
 
-			/* close down the lockdown connection as it is no longer needed */
-			if (client) {
-				lockdownd_client_free(client);
-				client = NULL;
-			}
-
 			/* create Status.plist with failed status for now */
 			mobilebackup_write_status(backup_directory, 0);
 
@@ -1288,8 +1288,6 @@ int main(int argc, char *argv[])
 				printf("ERROR: Could not send BackupMessageRestoreComplete, error code %d\n", err);
 			}
 			/* TODO: close down notification_proxy connection */
-			/* close down lockdown connection as it is no longer needed */
-			lockdownd_client_free(client);
 			client = NULL;
 			break;
 			case CMD_LEAVE: