Implement "Sync in progress..." screen with abort detection

1 files changed, 90 insertions, 2 deletions

diff --git a/tools/iphonebackup.c b/tools/iphonebackup.c
index 781f9d3..f212fc8 100644
--- a/tools/iphonebackup.c
+++ b/tools/iphonebackup.c
@@ -2,7 +2,8 @@
  * iphonebackup.c
  * Command line interface to use the device's backup and restore service
  *
- * Copyright (c) 2009 Martin Szulecki All Rights Reserved.
+ * Copyright (c) 2009-2010 Martin Szulecki All Rights Reserved.
+ * Copyright (c) 2010      Nikias Bassen All Rights Reserved.
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
@@ -29,8 +30,11 @@
 #include <libiphone/libiphone.h>
 #include <libiphone/lockdown.h>
 #include <libiphone/mobilebackup.h>
+#include <libiphone/notification_proxy.h>
+#include <libiphone/afc.h>
 
 #define MOBILEBACKUP_SERVICE_NAME "com.apple.mobilebackup"
+#define NP_SERVICE_NAME "com.apple.mobile.notification_proxy"
 
 static mobilebackup_client_t mobilebackup = NULL;
 static lockdownd_client_t client = NULL;
@@ -55,6 +59,16 @@ enum device_link_file_status_t {
 	DEVICE_LINK_FILE_STATUS_LAST_HUNK
 };
 
+static void notify_cb(const char *notification)
+{
+	if (!strcmp(notification, NP_SYNC_CANCEL_REQUEST)) {
+		printf("User has aborted on-device\n");
+		quit_flag++;
+	} else {
+		printf("unhandled notification '%s' (TODO: implement)\n", notification);
+	}
+}
+
 static plist_t mobilebackup_factory_info_plist_new()
 {
 	/* gather data from lockdown */
@@ -345,6 +359,29 @@ static int mobilebackup_delete_backup_file_by_hash(const char *backup_directory,
 	return ret;
 }
 
+static void do_post_notification(const char *notification)
+{
+	uint16_t nport = 0;
+	np_client_t np;
+
+	if (!client) {
+		if (lockdownd_client_new_with_handshake(phone, &client, "iphonebackup") != LOCKDOWN_E_SUCCESS) {
+			return;
+		}
+	}
+
+	lockdownd_start_service(client, NP_SERVICE_NAME, &nport);
+	if (nport) {
+		np_client_new(phone, nport, &np);
+		if (np) {
+			np_post_notification(np, notification);
+			np_client_free(np);
+		}
+	} else {
+		printf("Could not start %s\n", NP_SERVICE_NAME);
+	}
+}
+
 /**
  * signal handler function for cleaning up properly
  */
@@ -483,7 +520,34 @@ int main(int argc, char *argv[])
 		return -1;
 	}
 
+	/* start notification_proxy */
+	np_client_t np = NULL;
+	ret = lockdownd_start_service(client, NP_SERVICE_NAME, &port);
+	if ((ret == LOCKDOWN_E_SUCCESS) && port) {
+		np_client_new(phone, port, &np);
+		np_set_notify_callback(np, notify_cb);
+		const char *noties[5] = {
+			NP_SYNC_CANCEL_REQUEST,
+			NP_SYNC_SUSPEND_REQUEST,
+			NP_SYNC_RESUME_REQUEST,
+			NP_BACKUP_DOMAIN_CHANGED,
+			NULL
+		};
+		np_observe_notifications(np, noties);
+	} else {
+		printf("ERROR: Could not start service %s.\n", NP_SERVICE_NAME);
+	}
+
+	/* start AFC, we need this for the lock file */
+	afc_client_t afc = NULL;
+	port = 0;
+	ret = lockdownd_start_service(client, "com.apple.afc", &port);
+	if ((ret == LOCKDOWN_E_SUCCESS) && port) {
+		afc_client_new(phone, port, &afc);
+	}
+
 	/* start syslog_relay service and retrieve port */
+	port = 0;
 	ret = lockdownd_start_service(client, MOBILEBACKUP_SERVICE_NAME, &port);
 	if ((ret == LOCKDOWN_E_SUCCESS) && port) {
 		printf("Started \"%s\" service on port %d.\n", MOBILEBACKUP_SERVICE_NAME, port);
@@ -494,11 +558,23 @@ int main(int argc, char *argv[])
 			cmd = CMD_LEAVE;
 		}
 
+		do_post_notification(NP_SYNC_WILL_START);
+		uint64_t lockfile = 0;
+		afc_file_open(afc, "/com.apple.itunes.lock_sync", AFC_FOPEN_RW, &lockfile);
+		if (lockfile) {
+			do_post_notification(NP_SYNC_LOCK_REQUEST);
+			if (afc_file_lock(afc, lockfile, AFC_LOCK_EX) == AFC_E_SUCCESS) {
+				do_post_notification(NP_SYNC_DID_START);
+			} else {
+				afc_file_close(afc, lockfile);
+				lockfile = 0;
+			}
+		}
 		switch(cmd) {
 			case CMD_BACKUP:
 			printf("Starting backup...\n");
 			/* TODO: check domain com.apple.mobile.backup key RequiresEncrypt and WillEncrypt with lockdown */
-			/* TODO: verify battery on AC enough battery remaining */
+			/* TODO: verify battery on AC enough battery remaining */	
 
 			/* Info.plist (Device infos, IC-Info.sidb, photos, app_ids, iTunesPrefs) */
 
@@ -815,6 +891,12 @@ int main(int argc, char *argv[])
 			default:
 			break;
 		}
+		if (lockfile) {
+			afc_file_lock(afc, lockfile, AFC_LOCK_UN);
+			afc_file_close(afc, lockfile);
+			lockfile = 0;
+			do_post_notification(NP_SYNC_DID_FINISH);
+		}
 	} else {
 		printf("ERROR: Could not start service %s.\n", MOBILEBACKUP_SERVICE_NAME);
 		lockdownd_client_free(client);
@@ -826,6 +908,12 @@ int main(int argc, char *argv[])
 		client = NULL;
 	}
 
+	if (afc)
+		afc_client_free(afc);
+
+	if (np)
+		np_client_free(np);
+
 	if (mobilebackup)
 		mobilebackup_client_free(mobilebackup);