afc: Add afc_get_file_info_plist and afc_get_device_info_plist functions

These functions are like afc_get_file_info/afc_get_device_info but the
data is returned as a plist_t dictionary instead of a string list.

1 files changed, 116 insertions, 2 deletions

diff --git a/src/afc.c b/src/afc.c
index c0314bc..c7eed5c 100644
--- a/src/afc.c
+++ b/src/afc.c
@@ -402,6 +402,50 @@ static char **make_strings_list(char *tokens, uint32_t length)
 	return list;
 }
 
+static plist_t *make_dictionary(char *tokens, size_t length)
+{
+	size_t j = 0;
+	plist_t dict = NULL;
+
+	if (!tokens || !length)
+		return NULL;
+
+	dict = plist_new_dict();
+
+	while (j < length) {
+		size_t key_len = strnlen(tokens + j, length - j);
+		if (j + key_len >= length) {
+			plist_free(dict);
+			return NULL;
+		}
+		char* key = tokens + j;
+		j += key_len + 1;
+
+		if (j >= length) {
+			plist_free(dict);
+			return NULL;
+		}
+
+		size_t val_len = strnlen(tokens + j, length - j);
+		if (j + val_len >= length) {
+			plist_free(dict);
+			return NULL;
+		}
+		char* val = tokens + j;
+		j += val_len + 1;
+
+		char* endp = NULL;
+		unsigned long long u64val = strtoull(val, &endp, 10);
+		if (endp && *endp == '\0') {
+			plist_dict_set_item(dict, key, plist_new_uint(u64val));
+		} else {
+			plist_dict_set_item(dict, key, plist_new_string(val));
+		}
+	}
+
+	return dict;
+}
+
 static int _afc_check_packet_buffer(afc_client_t client, uint32_t data_len)
 {
 	if (data_len > client->packet_extra) {
@@ -498,6 +542,40 @@ afc_error_t afc_get_device_info(afc_client_t client, char ***device_information)
 	return ret;
 }
 
+afc_error_t afc_get_device_info_plist(afc_client_t client, plist_t *device_information)
+{
+	uint32_t bytes = 0;
+	char *data = NULL;
+	afc_error_t ret = AFC_E_UNKNOWN_ERROR;
+
+	if (!client || !device_information)
+		return AFC_E_INVALID_ARG;
+
+	afc_lock(client);
+
+	/* Send the command */
+	ret = afc_dispatch_packet(client, AFC_OP_GET_DEVINFO, 0, NULL, 0, &bytes);
+	if (ret != AFC_E_SUCCESS) {
+		afc_unlock(client);
+		return AFC_E_NOT_ENOUGH_DATA;
+	}
+	/* Receive the data */
+	ret = afc_receive_data(client, &data, &bytes);
+	if (ret != AFC_E_SUCCESS) {
+		if (data)
+			free(data);
+		afc_unlock(client);
+		return ret;
+	}
+	/* Parse the data */
+	*device_information = make_dictionary(data, bytes);
+	free(data);
+
+	afc_unlock(client);
+
+	return ret;
+}
+
 afc_error_t afc_get_device_info_key(afc_client_t client, const char *key, char **value)
 {
 	afc_error_t ret = AFC_E_INTERNAL_ERROR;
@@ -647,8 +725,6 @@ afc_error_t afc_get_file_info(afc_client_t client, const char *path, char ***fil
 		return AFC_E_NO_MEM;
 	}
 
-	debug_info("We got %p and %p", client->afc_packet, AFC_PACKET_DATA_PTR);
-
 	/* Send command */
 	memcpy(AFC_PACKET_DATA_PTR, path, data_len);
 	ret = afc_dispatch_packet(client, AFC_OP_GET_FILE_INFO, data_len, NULL, 0, &bytes);
@@ -669,6 +745,44 @@ afc_error_t afc_get_file_info(afc_client_t client, const char *path, char ***fil
 	return ret;
 }
 
+afc_error_t afc_get_file_info_plist(afc_client_t client, const char *path, plist_t *file_information)
+{
+	char *received = NULL;
+	uint32_t bytes = 0;
+	afc_error_t ret = AFC_E_UNKNOWN_ERROR;
+
+	if (!client || !path || !file_information)
+		return AFC_E_INVALID_ARG;
+
+	afc_lock(client);
+
+	uint32_t data_len = (uint32_t)strlen(path)+1;
+	if (_afc_check_packet_buffer(client, data_len) < 0) {
+		afc_unlock(client);
+		debug_info("Failed to realloc packet buffer");
+		return AFC_E_NO_MEM;
+	}
+
+	/* Send command */
+	memcpy(AFC_PACKET_DATA_PTR, path, data_len);
+	ret = afc_dispatch_packet(client, AFC_OP_GET_FILE_INFO, data_len, NULL, 0, &bytes);
+	if (ret != AFC_E_SUCCESS) {
+		afc_unlock(client);
+		return AFC_E_NOT_ENOUGH_DATA;
+	}
+
+	/* Receive data */
+	ret = afc_receive_data(client, &received, &bytes);
+	if (received) {
+		*file_information = make_dictionary(received, bytes);
+		free(received);
+	}
+
+	afc_unlock(client);
+
+	return ret;
+}
+
 afc_error_t afc_file_open(afc_client_t client, const char *filename, afc_file_mode_t file_mode, uint64_t *handle)
 {
 	if (!client || !client->parent || !client->afc_packet)