complete xml plist abstraction and migrate lockdownd_hello to new plist API.

3 files changed, 166 insertions, 49 deletions

diff --git a/src/lockdown.c b/src/lockdown.c
index 6b8f298..2906fdf 100644
--- a/src/lockdown.c
+++ b/src/lockdown.c
@@ -201,49 +201,57 @@ iphone_error_t lockdownd_hello(iphone_lckd_client_t control)
 {
 	if (!control)
 		return IPHONE_E_INVALID_ARG;
-	xmlDocPtr plist = new_plist();
-	xmlNode *dict, *key;
-	char **dictionary;
+
 	int bytes = 0, i = 0;
 	iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
 
+	plist_t plist = NULL;
+	plist_new_plist(&plist);
+
+	dict_t dict = NULL;
+	plist_new_dict_in_plist(plist, &dict);
+
+	plist_add_dict_element(dict, "Request", PLIST_STRING, (void *) "QueryType");
+
 	log_debug_msg("lockdownd_hello() called\n");
-	dict = add_child_to_plist(plist, "dict", "\n", NULL, 0);
-	key = add_key_str_dict_element(plist, dict, "Request", "QueryType", 1);
-	char *XML_content;
-	uint32_t length;
+	char *XML_content = NULL;
+	uint32_t length = 0;
 
-	xmlDocDumpMemory(plist, (xmlChar **) & XML_content, &length);
+	plist_to_xml(plist, &XML_content, &length);
+	log_debug_msg("Send msg :\nsize : %i\nxml : %s", length, XML_content);
 	ret = iphone_lckd_send(control, XML_content, length, &bytes);
 
 	xmlFree(XML_content);
-	xmlFreeDoc(plist);
+	XML_content = NULL;
+	plist_free(plist);
 	plist = NULL;
+
 	ret = iphone_lckd_recv(control, &XML_content, &bytes);
+	log_debug_msg("Receive msg :\nsize : %i\nxml : %s", bytes, XML_content);
+	xml_to_plist(XML_content, bytes, &plist);
 
-	plist = xmlReadMemory(XML_content, bytes, NULL, NULL, 0);
 	if (!plist)
 		return IPHONE_E_PLIST_ERROR;
-	dict = xmlDocGetRootElement(plist);
-	for (dict = dict->children; dict; dict = dict->next) {
-		if (!xmlStrcmp(dict->name, "dict"))
-			break;
-	}
-	if (!dict)
-		return IPHONE_E_DICT_ERROR;
-	dictionary = read_dict_element_strings(dict);
-	xmlFreeDoc(plist);
-	free(XML_content);
 
-	for (i = 0; dictionary[i]; i += 2) {
-		if (!strcmp(dictionary[i], "Result") && !strcmp(dictionary[i + 1], "Success")) {
-			log_debug_msg("lockdownd_hello(): success\n");
-			ret = IPHONE_E_SUCCESS;
-			break;
-		}
+	plist_t query_node = find_query_node(plist, "Request", "QueryType");
+	plist_t result_node = g_node_next_sibling(query_node);
+	plist_t value_node = g_node_next_sibling(result_node);
+
+	plist_type result_type;
+	plist_type value_type;
+
+	char *result_value = NULL;
+	char *value_value = NULL;
+
+	get_type_and_value(result_node, &result_type, (void *) (&result_value));
+	get_type_and_value(value_node, &value_type, (void *) (&value_value));
+
+	if (result_type == PLIST_KEY &&
+		value_type == PLIST_STRING && !strcmp(result_value, "Result") && !strcmp(value_value, "Success")) {
+		log_debug_msg("lockdownd_hello(): success\n");
+		ret = IPHONE_E_SUCCESS;
 	}
 
-	free_dictionary(dictionary);
 	return ret;
 }
 
diff --git a/src/plist.c b/src/plist.c
index 377646d..5b61570 100644
--- a/src/plist.c
+++ b/src/plist.c
@@ -286,8 +286,6 @@ void byte_convert(char *address, size_t size)
 	}
 }
 
-
-
 void print_bytes(char *val, size_t size)
 {
 	int i = 0;
@@ -421,11 +419,17 @@ void plist_free(plist_t plist)
 	g_node_destroy(plist);
 }
 
-void node_to_xml(GNode * node, gpointer data)
+struct xml_node {
+	xmlNodePtr xml;
+	uint32_t depth;
+};
+
+void node_to_xml(GNode * node, gpointer xml_struct)
 {
 	if (!node)
 		return;
 
+	struct xml_node *xstruct = (struct xml_node *) xml_struct;
 	struct plist_data *node_data = (struct plist_data *) node->data;
 
 	xmlNodePtr child_node = NULL;
@@ -510,12 +514,31 @@ void node_to_xml(GNode * node, gpointer data)
 		break;
 	}
 
-	child_node = xmlNewChild(data, NULL, tag, val);
-	xmlNodeAddContent(child_node, "\n");
+	int i = 0;
+	for (i = 0; i < xstruct->depth; i++) {
+		xmlNodeAddContent(xstruct->xml, "\t");
+	}
+	child_node = xmlNewChild(xstruct->xml, NULL, tag, val);
+	xmlNodeAddContent(xstruct->xml, "\n");
 	g_free(val);
 
-	if (isStruct)
-		g_node_children_foreach(node, G_TRAVERSE_ALL, node_to_xml, child_node);
+	//add return for structured types
+	if (node_data->type == PLIST_ARRAY ||
+		node_data->type == PLIST_DICT || node_data->type == PLIST_DATA || node_data->type == PLIST_PLIST)
+		xmlNodeAddContent(child_node, "\n");
+
+	if (isStruct) {
+		struct xml_node child = { child_node, xstruct->depth + 1 };
+		g_node_children_foreach(node, G_TRAVERSE_ALL, node_to_xml, &child);
+	}
+	//fix indent for structured types
+	if (node_data->type == PLIST_ARRAY ||
+		node_data->type == PLIST_DICT || node_data->type == PLIST_DATA || node_data->type == PLIST_PLIST) {
+
+		for (i = 0; i < xstruct->depth; i++) {
+			xmlNodeAddContent(child_node, "\t");
+		}
+	}
 
 	return;
 }
@@ -523,12 +546,18 @@ void node_to_xml(GNode * node, gpointer data)
 void xml_to_node(xmlNodePtr xml_node, GNode * plist_node)
 {
 	xmlNodePtr node = NULL;
-	struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
-	GNode *subnode = g_node_new(data);
-	g_node_append(plist_node, subnode);
 
 	for (node = xml_node->children; node; node = node->next) {
 
+		while (node && !xmlStrcmp(node->name, "text"))
+			node = node->next;
+		if (!node)
+			break;
+
+		struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+		GNode *subnode = g_node_new(data);
+		g_node_append(plist_node, subnode);
+
 		if (!xmlStrcmp(node->name, "true")) {
 			data->boolval = 1;
 			data->type = PLIST_BOOLEAN;
@@ -590,15 +619,15 @@ void xml_to_node(xmlNodePtr xml_node, GNode * plist_node)
 	}
 }
 
-void plist_to_xml(plist_t plist, char **plist_xml)
+void plist_to_xml(plist_t plist, char **plist_xml, uint32_t * length)
 {
 	if (!plist || !plist_xml || *plist_xml)
 		return;
 	xmlDocPtr plist_doc = new_plist();
 	xmlNodePtr root_node = xmlDocGetRootElement(plist_doc);
-	g_node_children_foreach(plist, G_TRAVERSE_ALL, node_to_xml, root_node);
-	int size = 0;
-	xmlDocDumpMemory(plist_doc, (xmlChar **) plist_xml, &size);
+	struct xml_node root = { root_node, 0 };
+	g_node_children_foreach(plist, G_TRAVERSE_ALL, node_to_xml, &root);
+	xmlDocDumpMemory(plist_doc, (xmlChar **) plist_xml, length);
 }
 
 GNode *parse_raw_node(const char *bpbuffer, uint32_t bplength, uint32_t * position, uint8_t ref_size)
@@ -869,13 +898,13 @@ plist_t parse_nodes(const char *bpbuffer, uint32_t bplength, uint32_t * position
 	return root_node;
 }
 
-void plist_to_bin(plist_t plist, char **plist_bin, int *length)
+void plist_to_bin(plist_t plist, char **plist_bin, uint32_t * length)
 {
 }
 
-void xml_to_plist(const char *plist_xml, plist_t * plist)
+void xml_to_plist(const char *plist_xml, uint32_t length, plist_t * plist)
 {
-	xmlDocPtr plist_doc = xmlReadMemory(plist_xml, strlen(plist_xml), NULL, NULL, 0);
+	xmlDocPtr plist_doc = xmlReadMemory(plist_xml, length, NULL, NULL, 0);
 	xmlNodePtr root_node = xmlDocGetRootElement(plist_doc);
 
 	struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
@@ -885,8 +914,84 @@ void xml_to_plist(const char *plist_xml, plist_t * plist)
 
 }
 
-void bin_to_plist(const char *plist_bin, int length, plist_t * plist)
+void bin_to_plist(const char *plist_bin, uint32_t length, plist_t * plist)
 {
 	uint32_t pos = 0;
 	*plist = parse_nodes(plist_bin, length, &pos);
 }
+
+
+GNode *find_query_node(plist_t plist, char *key, char *request)
+{
+	if (!plist)
+		return NULL;
+
+	GNode *current = NULL;
+	for (current = plist->children; current; current = current->next) {
+
+		struct plist_data *data = (struct plist_data *) current->data;
+
+		if (data->type == PLIST_KEY && !strcmp(data->strval, key) && current->next) {
+
+			data = (struct plist_data *) current->next->data;
+			if (data->type == PLIST_STRING && !strcmp(data->strval, request))
+				return current->next;
+		}
+		if (data->type == PLIST_DICT || data->type == PLIST_ARRAY || data->type == PLIST_PLIST) {
+			GNode *sub = find_query_node(current, key, request);
+			if (sub)
+				return sub;
+		}
+	}
+	return NULL;
+}
+
+void get_type_and_value(GNode * node, plist_type * type, void *value)
+{
+	if (!node)
+		return;
+
+	struct plist_data *data = (struct plist_data *) node->data;
+
+	*type = data->type;
+
+	switch (*type) {
+	case PLIST_BOOLEAN:
+		*((char *) value) = data->boolval;
+		break;
+	case PLIST_UINT8:
+		*((uint8_t *) value) = data->intval8;
+		break;
+	case PLIST_UINT16:
+		*((uint16_t *) value) = data->intval16;
+		break;
+	case PLIST_UINT32:
+		*((uint32_t *) value) = data->intval32;
+		break;
+	case PLIST_UINT64:
+		*((uint64_t *) value) = data->intval64;
+		break;
+	case PLIST_FLOAT32:
+		*((float *) value) = data->realval32;
+		break;
+	case PLIST_FLOAT64:
+		*((double *) value) = data->realval64;
+		break;
+	case PLIST_STRING:
+		*((char **) value) = strdup(data->strval);
+		break;
+	case PLIST_UNICODE:
+		*((wchar_t **) value) = wcsdup(data->unicodeval);
+		break;
+	case PLIST_KEY:
+		*((char **) value) = strdup(data->strval);
+		break;
+	case PLIST_DATA:
+	case PLIST_ARRAY:
+	case PLIST_DICT:
+	case PLIST_DATE:
+	case PLIST_PLIST:
+	default:
+		break;
+	}
+}
diff --git a/src/plist.h b/src/plist.h
index 1ca55f9..ffc00e4 100644
--- a/src/plist.h
+++ b/src/plist.h
@@ -70,15 +70,19 @@ typedef GNode *plist_t;
 typedef GNode *dict_t;
 typedef GNode *array_t;
 
+
 void plist_new_plist(plist_t * plist);
 void plist_new_dict_in_plist(plist_t plist, dict_t * dict);
 void plist_new_array_in_plist(plist_t plist, int length, plist_type type, void **values, array_t * array);
 void plist_add_dict_element(dict_t dict, char *key, plist_type type, void *value);
 void plist_free(plist_t plist);
 
-void plist_to_xml(plist_t plist, char **plist_xml);
-void plist_to_bin(plist_t plist, char **plist_bin, int *length);
+void plist_to_xml(plist_t plist, char **plist_xml, uint32_t * length);
+void plist_to_bin(plist_t plist, char **plist_bin, uint32_t * length);
+
+void xml_to_plist(const char *plist_xml, uint32_t length, plist_t * plist);
+void bin_to_plist(const char *plist_bin, uint32_t length, plist_t * plist);
 
-void xml_to_plist(const char *plist_xml, plist_t * plist);
-void bin_to_plist(const char *plist_bin, int length, plist_t * plist);
+GNode *find_query_node(plist_t plist, char *key, char *request);
+void get_type_and_value(GNode * node, plist_type * type, void *value);
 #endif