Add lockdownd_get_data_sync_classes() helper for mobilesync logic

2 files changed, 113 insertions, 0 deletions

diff --git a/include/libimobiledevice/lockdown.h b/include/libimobiledevice/lockdown.h
index 4504ef1..9a02e69 100644
--- a/include/libimobiledevice/lockdown.h
+++ b/include/libimobiledevice/lockdown.h
@@ -96,6 +96,8 @@ lockdownd_error_t lockdownd_goodbye(lockdownd_client_t client);
 void lockdownd_client_set_label(lockdownd_client_t client, const char *label);
 lockdownd_error_t lockdownd_get_device_uuid(lockdownd_client_t control, char **uuid);
 lockdownd_error_t lockdownd_get_device_name(lockdownd_client_t client, char **device_name);
+lockdownd_error_t lockdownd_get_sync_data_classes(lockdownd_client_t client, char ***classes, int *count);
+lockdownd_error_t lockdownd_data_classes_free(char **classes);
 
 #ifdef __cplusplus
 }
diff --git a/src/lockdown.c b/src/lockdown.c
index 0623469..11f3c85 100644
--- a/src/lockdown.c
+++ b/src/lockdown.c
@@ -2,6 +2,7 @@
  * lockdown.c
  * com.apple.mobile.lockdownd service implementation.
  *
+ * Copyright (c) 2010 Bryan Forbes All Rights Reserved.
  * Copyright (c) 2008 Zach C. All Rights Reserved.
  *
  * This library is free software; you can redistribute it and/or
@@ -21,6 +22,10 @@
 
 #include <string.h>
 #include <stdlib.h>
+#define _GNU_SOURCE 1
+#define __USE_GNU 1
+#include <stdio.h>
+#include <ctype.h>
 #include <glib.h>
 #include <libtasn1.h>
 #include <gnutls/x509.h>
@@ -1506,3 +1511,109 @@ lockdownd_error_t lockdownd_deactivate(lockdownd_client_t client)
 	return ret;
 }
 
+static void str_remove_spaces(char *source)
+{
+	char *dest = source;
+	while (*source != 0) {
+		if (!isspace(*source)) {
+			*dest++ = *source; /* copy */
+		}
+		source++;
+	}
+	*dest = 0;
+}
+
+lockdownd_error_t lockdownd_get_sync_data_classes(lockdownd_client_t client, char ***classes, int *count)
+{
+	if (!client)
+		return LOCKDOWN_E_INVALID_ARG;
+
+	if (!client->session_id)
+		return LOCKDOWN_E_NO_RUNNING_SESSION;
+
+	plist_t dict = NULL;
+	plist_dict_iter iter = NULL;
+	lockdownd_error_t err = LOCKDOWN_E_UNKNOWN_ERROR;
+
+	char *key = NULL;
+	plist_t value = NULL;
+
+	char **newlist = NULL;
+	int newcount = 0;
+
+	*classes = NULL;
+	*count = 0;
+
+	err = lockdownd_get_value(client, "com.apple.mobile.tethered_sync", NULL, &dict);
+	if (err != LOCKDOWN_E_SUCCESS) {
+		if (dict) {
+			plist_free(dict);
+		}
+		return err;
+	}
+
+	if (plist_get_node_type(dict) != PLIST_DICT) {
+		plist_free(dict);
+		return LOCKDOWN_E_PLIST_ERROR;
+	}
+
+	plist_dict_new_iter(dict, &iter);
+
+	plist_dict_next_item(dict, iter, &key, &value);
+	while (key && value) {
+		int add_to_list = 0;
+		plist_t disabled = NULL;
+
+		disabled = plist_dict_get_item(value, "DisableTethered");
+
+		if (!disabled) {
+			add_to_list = 1;
+		} else {
+			if (plist_get_node_type(disabled) == PLIST_BOOLEAN) {
+				uint8_t val = 0;
+				plist_get_bool_val(disabled, &val);
+				add_to_list = val > 0 ? 0 : 1;
+			} else {
+				uint64_t val = 0;
+				plist_get_uint_val(disabled, &val);
+				add_to_list = val > 0 ? 0 : 1;
+			}
+		}
+
+		if (add_to_list) {
+			newlist = realloc(*classes, sizeof(char*) * (newcount+1));
+			str_remove_spaces(key);
+			asprintf(&newlist[newcount++], "com.apple.%s", key);
+			*classes = newlist;
+		}
+		free(key);
+		key = NULL;
+		value = NULL;
+		plist_dict_next_item(dict, iter, &key, &value);
+	}
+
+	*count = newcount;
+	newlist = realloc(*classes, sizeof(char*) * (newcount+1));
+	newlist[newcount] = NULL;
+	*classes = newlist;
+
+	if (iter) {
+		free(iter);
+	}
+	if (dict) {
+		plist_free(dict);
+	}
+	return LOCKDOWN_E_SUCCESS;
+}
+
+lockdownd_error_t lockdownd_data_classes_free(char **classes)
+{
+	if (classes) {
+		int i = 0;
+		while (classes[i++]) {
+			free(classes[i]);
+		}
+		free(classes);
+	}
+	return LOCKDOWN_E_SUCCESS;
+}