Initial backup implementation, device is sending the backup now

1 files changed, 360 insertions, 144 deletions

diff --git a/tools/iphonebackup.c b/tools/iphonebackup.c
index 8e29160..611c0f2 100644
--- a/tools/iphonebackup.c
+++ b/tools/iphonebackup.c
@@ -24,6 +24,7 @@
 #include <errno.h>
 #include <stdlib.h>
 #include <signal.h>
+#include <glib.h>
 
 #include <libiphone/libiphone.h>
 #include <libiphone/lockdown.h>
@@ -31,6 +32,10 @@
 
 #define MOBILEBACKUP_SERVICE_NAME "com.apple.mobilebackup"
 
+static mobilebackup_client_t mobilebackup = NULL;
+static lockdownd_client_t client = NULL;
+static iphone_device_t phone = NULL;
+
 static int quit_flag = 0;
 
 enum cmd_mode {
@@ -38,43 +43,68 @@ enum cmd_mode {
         CMD_RESTORE
 };
 
-/*
-Backup Process Communication:
---------------------------------------------------------------------------------
-* Check lockdown value for domain com.apple.mobile.backup key RequiresEncrypt and WillEncrypt
-* Verify battery on AC enough battery remaining
-* ValidatePair with device for TrustedHost ability
-> DLMessageVersionExchange
-< DLMessageVersionExchange - DLVersionsOk
-> DLMessageDeviceReady
-< DLMessageProcessMessage: BackupMessageTypeKey: BackupMessageBackupRequest
-> DLMessageProcessMessage: BackupMessageTypeKey: BackupMessageBackupReplyOK
-...
-> DLSendFile
-< DLMessageProcessMessage: BackupMessageTypeKey: BackupMessageBackupFileReceived
-...
-> DLMessageProcessMessage: BackupMessageTypeKey: BackupMessageBackupFinished
+static plist_t mobilebackup_factory_info_plist()
+{
+        /* gather data from lockdown */
+        GTimeVal tv = {0, 0};
+        plist_t value_node = NULL;
+        plist_t root_node = NULL;
+        char *uuid = NULL;
+        char *uuid_uppercase = NULL;
 
+        plist_t ret = plist_new_dict();
 
-*/
+        /* get basic device information in one go */
+        lockdownd_get_value(client, NULL, NULL, &root_node);
 
-/*
-Restore Process Communication:
---------------------------------------------------------------------------------
-* Verify battery on AC enough battery remaining
-* ValidatePair with device for TrustedHost ability
-> DLMessageVersionExchange
-< DLMessageVersionExchange - DLVersionsOk
-> DLMessageDeviceReady
-< DLMessageProcessMessage - BackupMessageTypeKey: BackupMessageRestoreMigrate
-...
-DLSendFile
-...
-< DLMessageProcessMessage - BackupMessageTypeKey: BackupMessageRestoreReplyOK
-*/
+        /* set fields we understand */
+        value_node = plist_dict_get_item(root_node, "BuildVersion");
+        plist_dict_insert_item(ret, "Build Version", plist_copy(value_node));
+
+        value_node = plist_dict_get_item(root_node, "DeviceName");
+        plist_dict_insert_item(ret, "Device Name", plist_copy(value_node));
+        plist_dict_insert_item(ret, "Display Name", plist_copy(value_node));
+
+        /* FIXME: How is the GUID generated? */
+        plist_dict_insert_item(ret, "GUID", plist_new_string("---"));
+
+        value_node = plist_dict_get_item(root_node, "InternationalMobileEquipmentIdentity");
+        if (value_node)
+                plist_dict_insert_item(ret, "IMEI", plist_copy(value_node));
+
+        g_get_current_time(&tv);
+        plist_dict_insert_item(ret, "Last Backup Date", plist_new_date(tv.tv_sec, tv.tv_usec));
 
-static plist_t mobilebackup_factory_metadata()
+        value_node = plist_dict_get_item(root_node, "ProductType");
+        plist_dict_insert_item(ret, "Product Type", plist_copy(value_node));
+
+        value_node = plist_dict_get_item(root_node, "ProductVersion");
+        plist_dict_insert_item(ret, "Product Version", plist_copy(value_node));
+
+        value_node = plist_dict_get_item(root_node, "SerialNumber");
+        plist_dict_insert_item(ret, "Serial Number", plist_copy(value_node));
+
+        value_node = plist_dict_get_item(root_node, "UniqueDeviceID");
+        iphone_device_get_uuid(phone, &uuid);
+        plist_dict_insert_item(ret, "Target Identifier", plist_new_string(uuid));
+
+        /* uppercase */
+        uuid_uppercase = g_ascii_strup(uuid, -1);
+        plist_dict_insert_item(ret, "Unique Identifier", plist_new_string(uuid_uppercase));
+        free(uuid_uppercase);
+        free(uuid);
+
+        plist_t files = plist_new_dict();
+        /* FIXME: Embed files as <data> nodes */
+        plist_dict_insert_item(ret, "iTunes Files", files);
+        plist_dict_insert_item(ret, "iTunes Version", plist_new_string("9.0.2"));
+
+        return ret;
+}
+
+static plist_t mobilebackup_factory_metadata_plist()
 {
+        plist_t ret = NULL;
 /*
 Metadata key is:
 <dict>
@@ -106,6 +136,7 @@ Metadata key is:
         <false/>
 </dict>
 */
+        return ret;
 }
 
 /**
@@ -113,92 +144,165 @@ Metadata key is:
  */
 static plist_t mobilebackup_factory_manifest_data_plist()
 {
-        plist_t manifest_data = plist_dict_new();
-        /*
-        File hash is:
-        sha1(<Domain>-<Relative File Path>)
-        */
-        /*
-        Data hash is:
-        sha1(<file>)
-        */
+        plist_t ret = NULL;
+        plist_t value_node = NULL;
+        char *uuid = NULL;
+        GTimeVal tv = {0, 0};
 
-/*
-<dict>
-        <key>DeviceId</key>
-        <string>7a7b570ee169f02c43d3893f0d661cf7a32e1cf5</string>
-        <key>Version</key>
-        <string>6.2</string>
-        <key>Files</key>
-        <dict>
-                <key>3d0d7e5fb2ce288813306e4d4636395e047a3d28</key>
-                <dict>
-                        <key>ModificationTime</key>
-                        <date>2009-12-29T02:12:17Z</date>
-                        <key>FileLength</key>
-                        <integer>131072</integer>
-                        <key>Domain</key>
-                        <string>HomeDomain</string>
-                        <key>DataHash</key>
-                        <data>
-                        MfpSk+qw+RAJqLNTJI81tntvrwc=
-                        </data>
-                        <key>Group ID</key>
-                        <integer>501</integer>
-                        <key>User ID</key>
-                        <integer>501</integer>
-                        <key>Mode</key>
-                        <integer>420</integer>
-                </dict>
-        </dict>
-        <key>DeviceICCID</key>
-        <string>89492060399209300736</string>
-</dict>
-*/
+        ret = plist_new_dict();
 
-        return manifest_data;
+        /* get basic device information in one go */
+        lockdownd_get_value(client, NULL, "IntegratedCircuitCardIdentity", &value_node);
+
+        iphone_device_get_uuid(phone, &uuid);
+        plist_dict_insert_item(ret, "DeviceId", plist_new_string(uuid));
+        free(uuid);
+
+        plist_dict_insert_item(ret, "Version", plist_new_string("6.2"));
+
+        /* TODO: add all Applications */
+
+        /* TODO: add all Files */
+        plist_t files = plist_new_dict();
+
+        /* single file entry */
+        plist_t info_node = plist_new_dict();
+        g_get_current_time(&tv);
+        plist_dict_insert_item(info_node, "ModificationTime", plist_new_date(tv.tv_sec, tv.tv_usec));
+        plist_dict_insert_item(info_node, "FileLength", plist_new_uint(131072));
+        plist_dict_insert_item(info_node, "Domain", plist_new_string("HomeDomain"));
+
+        /* FIXME: calculate correct data hash */
+        /* Data hash is: sha1(<file>) */
+        plist_dict_insert_item(info_node, "DataHash", plist_new_data(NULL, 0));
+        plist_dict_insert_item(info_node, "Group ID", plist_new_uint(501));
+        plist_dict_insert_item(info_node, "User ID", plist_new_uint(501));
+        plist_dict_insert_item(info_node, "Mode ID", plist_new_uint(420));
+
+        /* FIXME: calculate correct file hash */
+        /* File hash is: sha1(<Domain>-<Relative File Path>) */
+        plist_dict_insert_item(files, "3d0d7e5fb2ce288813306e4d4636395e047a3d28", info_node);
+        plist_dict_insert_item(ret, "Files", files);
+
+        /* last node with ICCID */
+        if (value_node)
+                plist_dict_insert_item(ret, "DeviceICCID", &value_node);
+
+        return ret;
 }
 
 /**
  * Generates a manifest plist with all needed information and hashes
  */
-static plist_t mobilebackup_factory_manifest_plist()
+static plist_t mobilebackup_factory_manifest_plist(plist_t manifest_data)
 {
-        plist_t manifest_data = mobilebackup_factory_manifest_data_plist();
-        plist_t manifest = plist_dict_new();
+        char *buffer = NULL;
+        char *s = NULL;
+        uint32_t length;
+        unsigned char sha1[20];
+        gsize sha1_len;
+        GChecksum *checksum;
+        plist_t ret = NULL;
 
-        /*
-        AuthSignature Hash is:
-        sha1(<manifest_data>)
-        */
+        if (!manifest_data)
+                return ret;
 
-/*
-<dict>
-                <key>BackupManifestKey</key>
-                <dict>
-                        <key>AuthVersion</key>
-                        <string>2.0</string>
-                        <key>AuthSignature</key>
-                        <data>
-                        WBdfjcZWg/u/Bpn7aKDJC68UZF4=
-                        </data>
-                        <key>IsEncrypted</key>
-                        <integer>0</integer>
-                        <key>Data</key>
-                        <data><!-- binary plist -->
-                        ...
-                        </data>
-                </dict>
-                <key>BackupComputerBasePathKey</key>
-                <string>/</string>
-                <key>BackupMessageTypeKey</key>
-                <string>BackupMessageBackupRequest</string>
-                <key>BackupProtocolVersion</key>
-                <string>1.6</string>
-</dict>
-*/
+        ret = plist_new_dict();
+        plist_dict_insert_item(ret, "AuthVersion", plist_new_string("2.0"));
 
-        return manifest;
+        /* AuthSignature Hash is: sha1(<manifest_data>) */
+        plist_to_bin(manifest_data, &buffer, &length);
+
+        sha1_len = g_checksum_type_get_length(G_CHECKSUM_SHA1);
+        checksum = g_checksum_new(G_CHECKSUM_SHA1);
+        g_checksum_update(checksum, (guchar *)buffer, length);
+        g_checksum_get_digest(checksum, sha1, &sha1_len);
+        s = (char *)g_checksum_get_string(checksum);
+        printf("SHA1 AuthSignature: %s\n", s);
+        plist_dict_insert_item(ret, "AuthSignature", plist_new_data((char*)sha1, sha1_len));
+        g_checksum_free(checksum);
+
+
+        plist_dict_insert_item(ret, "IsEncrypted", plist_new_uint(0));
+        plist_dict_insert_item(ret, "Data", plist_new_data(buffer, length));
+
+        free(buffer);
+
+        return ret;
+}
+
+enum plist_format_t {
+        PLIST_FORMAT_XML,
+        PLIST_FORMAT_BINARY
+};
+
+static int plist_read_from_filename(char *filename, plist_t *plist)
+{
+        return 1;
+}
+
+static int plist_write_to_filename(plist_t plist, char *filename, enum plist_format_t format)
+{
+        char *buffer = NULL;
+        uint32_t length;
+        FILE *f;
+
+        if (!plist || !filename)
+                return 0;
+
+        if (format == PLIST_FORMAT_XML)
+                plist_to_xml(plist, &buffer, &length);
+        else if (format == PLIST_FORMAT_BINARY)
+                plist_to_bin(plist, &buffer, &length);
+        else
+                return 0;
+
+        f = fopen(filename, "wb");
+        fwrite(buffer, sizeof(char), length, f);
+        fclose(f);
+
+        free(buffer);
+
+        return 1;
+}
+
+static int plist_strcmp(plist_t node, const char *str)
+{
+        char *buffer = NULL;
+        int ret = 0;
+
+        if (plist_get_node_type(node) != PLIST_STRING)
+                return ret;
+
+        plist_get_string_val(node, &buffer);
+        ret = strcmp(buffer, str);
+        free(buffer);
+
+        return ret;
+}
+
+static void mobilebackup_write_status(char *path, int status)
+{
+        plist_t status_plist = plist_new_dict();
+        plist_dict_insert_item(status_plist, "Backup Success", plist_new_bool(status));
+        char *file_path = g_build_path(G_DIR_SEPARATOR_S, path, "Status.plist", NULL);
+        plist_write_to_filename(status_plist, file_path, PLIST_FORMAT_XML);
+        g_free(file_path);
+        plist_free(status_plist);
+}
+
+static void debug_plist(plist_t plist)
+{
+        char *buffer = NULL;
+        uint32_t length = 0;
+
+        if (!plist)
+                return;
+
+        plist_to_xml(plist, &buffer, &length);
+
+        printf("Printing %i bytes plist:\n%s\n", length, buffer);
+        free(buffer);
 }
 
 /**
@@ -226,10 +330,6 @@ static void print_usage(int argc, char **argv)
         printf("\n");
 }
 
-static mobilebackup_client_t mobilebackup = NULL;
-static lockdownd_client_t client = NULL;
-static iphone_device_t phone = NULL;
-
 int main(int argc, char *argv[])
 {
         iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
@@ -239,6 +339,8 @@ int main(int argc, char *argv[])
         uuid[0] = 0;
         int cmd = -1;
         char *backup_directory = NULL;
+        struct stat st;
+        plist_t node = NULL;
 
         /* we need to exit cleanly on running backups and restores or we cause havok */
         signal(SIGINT, clean_exit);
@@ -293,6 +395,22 @@ int main(int argc, char *argv[])
                 return -1;
         }
 
+        /* verify if passed backup directory exists */
+        if (stat(backup_directory, &st) != 0) {
+                printf("ERROR: Backup directory \"%s\" does not exist!\n", backup_directory);
+                return -1;
+        }
+
+        /* restore directory must contain an Info.plist */
+        char *info_path = g_build_path(G_DIR_SEPARATOR_S, backup_directory, "Info.plist", NULL);
+        if (cmd == CMD_RESTORE) {
+                if (stat(info_path, &st) != 0) {
+                        g_free(info_path);
+                        printf("ERROR: Backup directory \"%s\" is invalid. No Info.plist found.\n", backup_directory);
+                        return -1;
+                }
+        }
+
         printf("Backup directory is \"%s\"\n", backup_directory);
 
         if (uuid[0] != 0) {
@@ -322,46 +440,140 @@ int main(int argc, char *argv[])
                 printf("Started \"%s\" service on port %d.\n", MOBILEBACKUP_SERVICE_NAME, port);
                 mobilebackup_client_new(phone, port, &mobilebackup);
 
-                /* TODO: Command implementations */
                 switch(cmd) {
                         case CMD_BACKUP:
-                        printf("TODO: Creating backup...\n");
-/*
-Create target directory:
-MobileSync/Backup/<uuid>-YYYYMMDD-HHMMSS/
-*/
-
-/*
-Create: Info.plist (Device infos, IC-Info.sidb, photos, app_ids, iTunesPrefs)
-Create:Manifest.plist (backup manifest (backup state))
-*/
+                        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 */
+
+                        /* ????: create target directory: MobileSync/Backup/<uuid>-YYYYMMDD-HHMMSS/ */
+
+                        /* create Info.plist (Device infos, IC-Info.sidb, photos, app_ids, iTunesPrefs) */
+                        printf("Creating \"%s/Info.plist\".\n", backup_directory);
+                        plist_t info_plist = mobilebackup_factory_info_plist();
+                        plist_write_to_filename(info_plist, info_path, PLIST_FORMAT_XML);
+                        g_free(info_path);
+
+                        /* create Manifest.plist (backup manifest (backup state)) */
+                        printf("Creating \"%s/Manifest.plist\".\n", backup_directory);
+                        char *manifest_path = g_build_path(G_DIR_SEPARATOR_S, backup_directory, "Manifest.plist", NULL);
+                        plist_t manifest_data = mobilebackup_factory_manifest_data_plist();
+                        plist_t manifest_plist = mobilebackup_factory_manifest_plist(manifest_data);
+                        plist_write_to_filename(manifest_plist, manifest_path, PLIST_FORMAT_XML);
+                        g_free(manifest_path);
+
+                        /* create Status.plist with failed status for now */
+                        mobilebackup_write_status(backup_directory, 0);
+
+                        /* close down lockdown connection as it is no longer needed */
                         lockdownd_client_free(client);
-/*
-Receive:
-...
-<hash>.mddata (Raw filedata)
-<hash>.mdinfo (backup file information)
-...
-Create: Status.plist (Info on how the backup process turned out)
-*/
+                        client = NULL;
+
+                        /* request backup from device with manifest */
+                        printf("Sending manifest and requesting backup.\n");
+
+                        node = plist_new_dict();
+                        plist_dict_insert_item(node, "BackupManifestKey", manifest_plist);
+                        plist_dict_insert_item(node, "BackupComputerBasePathKey", plist_new_string("/"));
+                        plist_dict_insert_item(node, "BackupMessageTypeKey", plist_new_string("BackupMessageBackupRequest"));
+                        plist_dict_insert_item(node, "BackupProtocolVersion", plist_new_string("1.6"));
+
+                        plist_t message = plist_new_array();
+                        plist_array_append_item(message, plist_new_string("DLMessageProcessMessage"));
+                        plist_array_append_item(message, node);
+
+                        mobilebackup_send(mobilebackup, message);
+                        plist_free(message);
+                        message = NULL;
+
+                        /* get response */
+                        int backup_ok = 0;
+                        mobilebackup_receive(mobilebackup, &message);
+                        node = plist_array_get_item(message, 0);
+                        if (!plist_strcmp(node, "DLMessageProcessMessage")) {
+                                node = plist_array_get_item(message, 1);
+                                node = plist_dict_get_item(node, "BackupMessageTypeKey");
+                                if (node && !plist_strcmp(node, "BackupMessageBackupReplyOK")) {
+                                        printf("Device accepts manifest and will send backup data now...\n");
+                                        backup_ok = 1;
+                                        printf("Acknowledging...\n");
+                                        /* send it back for ACK */
+                                        mobilebackup_send(mobilebackup, message);
+                                }
+                        } else {
+                                printf("Unhandled message received!\n");
+                                debug_plist(message);
+                        }
+                        plist_free(message);
+                        message = NULL;
+
+                        if (!backup_ok) {
+                                printf("ERROR: Device rejected to start the backup process.\n");
+                                break;
+                        }
+
+                        /* receive and save DLSendFile files and metadata, ACK each */
+                        int file_index = 0;
+                        do {
+                                mobilebackup_receive(mobilebackup, &message);
+                                node = plist_array_get_item(message, 0);
+                                if (plist_strcmp(node, "DLSendFile"))
+                                        break;
+
+                                printf("Receiving file %d...\n", file_index);
+                                /* TODO: save <hash>.mdinfo */
+                                /* TODO: save <hash>.mddata */
+                                debug_plist(message);
+                                plist_free(message);
+                                message = NULL;
+
+                                if (quit_flag) {
+                                        /* FIXME: need to cancel the backup here */
+                                        break;
+                                }
+
+                                /* acknowlegdge that we received the file */
+                                node = plist_new_dict();
+                                plist_dict_insert_item(node, "BackupMessageTypeKey", plist_new_string("kBackupMessageBackupFileReceived"));
+
+                                message = plist_new_array();
+                                plist_array_append_item(message, plist_new_string("DLMessageProcessMessage"));
+                                plist_array_append_item(message, node);
+                                mobilebackup_send(mobilebackup, message);
+
+                                plist_free(message);
+                                message = NULL;
+
+                                file_index++;
+                        } while (!plist_strcmp(node, "DLSendFile"));
+
+                        printf("Received %d files from device.\n", file_index);
+
+                        if (!plist_strcmp(node, "DLMessageProcessMessage")) {
+                                node = plist_array_get_item(message, 1);
+                                node = plist_dict_get_item(node, "BackupMessageTypeKey");
+                                /* wait until received final backup finished message */
+                                if (node && !plist_strcmp(node, "BackupMessageBackupFinished")) {
+                                        /* backup finished */
+                                        /* create: Status.plist (Info on how the backup process turned out) */
+                                        printf("Backup Successful.\n");
+                                        mobilebackup_write_status(backup_directory, 1);
+                                }
+                        }
+
+                        if (node)
+                                plist_free(node);
 
-/*
-- Check lockdown value for domain com.apple.mobile.backup key RequiresEncrypt and WillEncrypt
-- Verify battery on AC enough battery remaining
-- Request backup from device with manifest (BackupMessageBackupRequest)
-- Receive and save DLSendFile files and metadata, ACK each
-- Wait until received final backup finished message
-*/
                         break;
                         case CMD_RESTORE:
-                        printf("TODO: Restoring backup...\n");
-/*
-- Verify battery on AC enough battery remaining
-- Request restore from device (BackupMessageRestoreMigrate)
-- Read mddata files and send to devices using DLSendFile
-- Signal restore finished message to device
-*/
+                        printf("Restoring backup...\n");
+                        /* verify battery on AC enough battery remaining */
+                        /* request restore from device (BackupMessageRestoreMigrate) */
+                        /* read mddata files and send to devices using DLSendFile */
+                        /* signal restore finished message to device */
+                        /* close down lockdown connection as it is no longer needed */
                         lockdownd_client_free(client);
+                        client = NULL;
                         break;
                         default:
                         break;
@@ -369,8 +581,12 @@ Create: Status.plist (Info on how the backup process turned out)
         } else {
                 printf("ERROR: Could not start service %s.\n", MOBILEBACKUP_SERVICE_NAME);
                 lockdownd_client_free(client);
+                client = NULL;
         }
 
+        if (client)
+                lockdownd_client_free(client);
+
         if (mobilebackup)
                 mobilebackup_client_free(mobilebackup);
         iphone_device_free(phone);