idevicebackup2: updated for compatibility with iOS6

1 files changed, 13 insertions, 2 deletions

diff --git a/tools/idevicebackup2.c b/tools/idevicebackup2.c
index 4b7e79e..d2ec792 100644
--- a/tools/idevicebackup2.c
+++ b/tools/idevicebackup2.c
@@ -30,6 +30,7 @@
 #include <libgen.h>
 #include <ctype.h>
 #include <time.h>
+#include <sys/statvfs.h>
 
 #include <libimobiledevice/libimobiledevice.h>
 #include <libimobiledevice/lockdown.h>
@@ -1571,13 +1572,23 @@ checkpoint:
 				} else if (!strcmp(dlmsg, "DLMessageUploadFiles")) {
 					/* device wants to send files to the computer */
 					file_count += mb2_handle_receive_files(message, backup_directory);
+				} else if (!strcmp(dlmsg, "DLMessageGetFreeDiskSpace")) {
+					/* device wants to know how much disk space is available on the computer */
+					struct statvfs fs;
+					memset(&fs, '\0', sizeof(fs));
+					int res = statvfs(backup_directory, &fs);
+					uint64_t freespace = 0;
+					if (res == 0) {
+						freespace = fs.f_bavail * fs.f_bsize;
+					}
+					mobilebackup2_send_status_response(mobilebackup2, res, NULL, plist_new_uint(freespace));
 				} else if (!strcmp(dlmsg, "DLContentsOfDirectory")) {
 					/* list directory contents */
 					mb2_handle_list_directory(message, backup_directory);
 				} else if (!strcmp(dlmsg, "DLMessageCreateDirectory")) {
 					/* make a directory */
 					mb2_handle_make_directory(message, backup_directory);
-				} else if (!strcmp(dlmsg, "DLMessageMoveFiles")) {
+				} else if (!strcmp(dlmsg, "DLMessageMoveFiles") || !strcmp(dlmsg, "DLMessageMoveItems")) {
 					/* perform a series of rename operations */
 					plist_t moves = plist_array_get_item(message, 1);
 					uint32_t cnt = plist_dict_get_size(moves);
@@ -1623,7 +1634,7 @@ checkpoint:
 					if (err != MOBILEBACKUP2_E_SUCCESS) {
 						printf("Could not send status response, error %d\n", err);
 					}
-				} else if (!strcmp(dlmsg, "DLMessageRemoveFiles")) {
+				} else if (!strcmp(dlmsg, "DLMessageRemoveFiles") || !strcmp(dlmsg, "DLMessageRemoveItems")) {
 					plist_t removes = plist_array_get_item(message, 1);
 					uint32_t cnt = plist_array_get_size(removes);
 					PRINT_VERBOSE(1, "Removing %d file%s\n", cnt, (cnt == 1) ? "" : "s");