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

1 files changed, 123 insertions, 18 deletions

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;
+	}
+}