Improve check for BackupTotalSizeKey and improve message reporting total size

1 files changed, 6 insertions, 4 deletions

diff --git a/tools/iphonebackup.c b/tools/iphonebackup.c
index 3fe3a3f..2628871 100644
--- a/tools/iphonebackup.c
+++ b/tools/iphonebackup.c
@@ -445,10 +445,12 @@ int main(int argc, char *argv[])
 				/* first message contains total backup size */
 				if (file_index == 0) {
 					node = plist_dict_get_item(node_tmp, "BackupTotalSizeKey");
-					plist_get_uint_val(node, &backup_total_size);
-					format_size = g_format_size_for_display(backup_total_size);
-					printf("Backup will need %s on disk.\n", format_size);
-					g_free(format_size);
+					if (node) {
+						plist_get_uint_val(node, &backup_total_size);
+						format_size = g_format_size_for_display(backup_total_size);
+						printf("Backup data requires %s on the disk.\n", format_size);
+						g_free(format_size);
+					}
 				}
 
 				/* print out "received" if DLFileStatusKey is 2 (last file piece) */