idevicebackup: Finish implementing sending application data for restore

1 files changed, 68 insertions, 3 deletions

diff --git a/tools/idevicebackup.c b/tools/idevicebackup.c
index 87a404e..b9fb81a 100644
--- a/tools/idevicebackup.c
+++ b/tools/idevicebackup.c
@@ -1378,9 +1378,74 @@ int main(int argc, char *argv[])
                         }
                         /* TODO: observe notification_proxy id com.apple.mobile.application_installed */
                         /* TODO: loop over Applications entries in Manifest data plist */
-                                /* send AppInfo entries */
-                                /* receive com.apple.mobile.application_installed notification */
-                                /* receive BackupMessageRestoreApplicationReceived from device */
+                        plist_t applications = plist_dict_get_item(backup_data, "Applications");
+                        if (applications && (plist_get_node_type(applications) == PLIST_DICT) && restore_ok) {
+                                plist_dict_iter iter = NULL;
+                                plist_dict_new_iter(applications, &iter);
+                                if (iter) {
+                                        /* loop over Application entries in Manifest data plist */
+                                        char *hash = NULL;
+                                        int total_files = plist_dict_get_size(applications);
+                                        int cur_file = 1;
+                                        plist_t tmp_node = NULL;
+                                        plist_t dict = NULL;
+                                        plist_t array = NULL;
+                                        node = NULL;
+                                        plist_dict_next_item(applications, iter, &hash, &node);
+                                        while (node) {
+                                                printf("Restoring Application %s %d/%d (%d%%)...", hash, cur_file, total_files, (cur_file*100/total_files));
+                                                /* FIXME: receive com.apple.mobile.application_installed notification */
+                                                /* send AppInfo entry */
+                                                tmp_node = plist_dict_get_item(node, "AppInfo");
+
+                                                dict = plist_new_dict();
+                                                plist_dict_insert_item(dict, "AppInfo", plist_copy(tmp_node));
+                                                plist_dict_insert_item(dict, "BackupMessageTypeKey", plist_new_string("BackupMessageRestoreApplicationSent"));
+
+                                                array = plist_new_array();
+                                                plist_array_append_item(array, plist_new_string("DLMessageProcessMessage"));
+                                                plist_array_append_item(array, dict);
+
+                                                err = mobilebackup_send(mobilebackup, array);
+                                                if (err != MOBILEBACKUP_E_SUCCESS) {
+                                                        printf("ERROR: Unable to restore application %s due to error %d. Aborting...\n", hash, err);
+                                                        restore_ok = 0;
+                                                }
+
+                                                plist_free(array);
+                                                array = NULL;
+                                                dict = NULL;
+
+                                                /* receive BackupMessageRestoreApplicationReceived from device */
+                                                if (restore_ok) {
+                                                        err = mobilebackup_receive_restore_application_received(mobilebackup, NULL);
+                                                        if (err != MOBILEBACKUP_E_SUCCESS) {
+                                                                printf("ERROR: Failed to receive an ack from the device for this application due to error %d. Aborting...\n", err);
+                                                                restore_ok = 0;
+                                                        }
+                                                }
+
+                                                tmp_node = NULL;
+                                                node = NULL;
+                                                free(hash);
+                                                hash = NULL;
+
+                                                if (restore_ok) {
+                                                        printf("DONE\n");
+                                                        cur_file++;
+                                                        plist_dict_next_item(applications, iter, &hash, &node);
+                                                } else
+                                                        break;
+                                        }
+                                        free(iter);
+
+                                        if (restore_ok)
+                                                printf("All applications restored.\n");
+                                        else
+                                                printf("Failed to restore applications.\n");
+                                }
+                        }
+
                         plist_free(backup_data);
 
                         /* signal restore finished message to device; BackupMessageRestoreComplete */