From 1a06347d27ca51283de3a9ff21e138a3ea9ba9b6 Mon Sep 17 00:00:00 2001
From: Jonathan Beck
Date: Mon, 8 Dec 2008 22:47:02 +0100
Subject: cleanup binary parsing and move stuff around.

---
 src/lockdown.c |   2 +-
 src/plist.c    | 854 +++++++++++++++++++++++++++++++--------------------------
 src/plist.h    |  18 +-
 3 files changed, 489 insertions(+), 385 deletions(-)

diff --git a/src/lockdown.c b/src/lockdown.c
index 55e2e65..ae077b7 100644
--- a/src/lockdown.c
+++ b/src/lockdown.c
@@ -963,7 +963,7 @@ iphone_error_t iphone_lckd_start_service(iphone_lckd_client_t client, const char
 		if (result_key_type == PLIST_KEY &&
 			result_value_type == PLIST_STRING &&
 			port_key_type == PLIST_KEY &&
-			port_value_type == PLIST_UINT64 &&
+			port_value_type == PLIST_UINT &&
 			!strcmp(result_key, "Result") && !strcmp(result_value, "Success") && !strcmp(port_key, "Port")) {
 			port_loc = port_value;
 			ret = IPHONE_E_SUCCESS;
diff --git a/src/plist.c b/src/plist.c
index 5d8fc0e..c691c16 100644
--- a/src/plist.c
+++ b/src/plist.c
@@ -19,19 +19,24 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA 
  */
 
-#include <libxml/parser.h>
-#include <libxml/tree.h>
+
 #include <string.h>
 #include <assert.h>
 #include "utils.h"
 #include "plist.h"
 #include <wchar.h>
 
+/**********************************************
+*                                             *
+*           Abstract Plist stuff              *
+*                                             *
+**********************************************/
+
+
+
+
+
 
-const char *plist_base = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n\
-<!DOCTYPE plist PUBLIC \"-//Apple Computer//DTD PLIST 1.0//EN\" \"http://www.apple.com/DTDs/PropertyList-1.0.dtd\">\n\
-<plist version=\"1.0\">\n\
-</plist>\0";
 
 /** Formats a block of text to be a given indentation and width.
  * 
@@ -72,34 +77,7 @@ char *format_string(const char *buf, int cols, int depth)
 	return new_buf;
 }
 
-/** Creates a new plist XML document.
- * 
- * @return The plist XML document.
- */
-xmlDocPtr new_plist()
-{
-	char *plist = strdup(plist_base);
-	xmlDocPtr plist_xml = xmlReadMemory(plist, strlen(plist), NULL, NULL, 0);
-
-	if (!plist_xml)
-		return NULL;
-
-	free(plist);
-
-	return plist_xml;
-}
-
-/** Destroys a previously created XML document.
- *
- * @param plist The XML document to destroy.
- */
-void free_plist(xmlDocPtr plist)
-{
-	if (!plist)
-		return;
 
-	xmlFreeDoc(plist);
-}
 
 
 /*
@@ -117,29 +95,7 @@ void free_plist(xmlDocPtr plist)
  * 		- parse_nodes() will return the first node it encounters, which is usually the "root" node. 
  */
 
-uint32_t uipow(uint32_t value, uint32_t power)
-{
-	if (!power)
-		return 1;
-	int i = 0, oVal = value;
-	for (i = 1; i < power; i++) {
-		value *= oVal;
-	}
-	return value;
-}
 
-void byte_convert(char *address, size_t size)
-{
-	int i = 0, j = 0;
-	char tmp = '\0';
-
-	for (i = 0; i < (size / 2); i++) {
-		tmp = address[i];
-		j = ((size - 1) + 0) - i;
-		address[i] = address[j];
-		address[j] = tmp;
-	}
-}
 
 void print_bytes(char *val, size_t size)
 {
@@ -154,39 +110,17 @@ void print_bytes(char *val, size_t size)
 struct plist_data {
 	union {
 		char boolval;
-		uint8_t intval8;
-		uint16_t intval16;
-		uint32_t intval32;
-		uint64_t intval64;
-		float realval32;
-		double realval64;
+		uint64_t intval;
+		double realval;
 		char *strval;
 		wchar_t *unicodeval;
-		struct {
-			char *buff;
-			uint8_t ref_size;
-		};
+		char *buff;
 	};
 	uint64_t length;
 	plist_type type;
 };
 
-enum {
-	BPLIST_TRUE = 0x08,
-	BPLIST_FALSE = 0x09,
-	BPLIST_FILL = 0x0F,			/* will be used for length grabbing */
-	BPLIST_INT = 0x10,
-	BPLIST_REAL = 0x20,
-	BPLIST_DATE = 0x33,
-	BPLIST_DATA = 0x40,
-	BPLIST_STRING = 0x50,
-	BPLIST_UNICODE = 0x60,
-	BPLIST_UID = 0x70,
-	BPLIST_ARRAY = 0xA0,
-	BPLIST_SET = 0xC0,
-	BPLIST_DICT = 0xD0,
-	BPLIST_MASK = 0xF0
-};
+
 
 void plist_new_plist(plist_t * plist)
 {
@@ -239,23 +173,11 @@ void plist_add_dict_element(dict_t dict, char *key, plist_type type, void *value
 	case PLIST_BOOLEAN:
 		val->boolval = *((char *) value);
 		break;
-	case PLIST_UINT8:
-		val->intval8 = *((uint8_t *) value);
-		break;
-	case PLIST_UINT16:
-		val->intval16 = *((uint16_t *) value);
-		break;
-	case PLIST_UINT32:
-		val->intval32 = *((uint32_t *) value);
-		break;
-	case PLIST_UINT64:
-		val->intval64 = *((uint64_t *) value);
+	case PLIST_UINT:
+		val->intval = *((uint64_t *) value);
 		break;
-	case PLIST_FLOAT32:
-		val->realval32 = *((float *) value);
-		break;
-	case PLIST_FLOAT64:
-		val->realval64 = *((double *) value);
+	case PLIST_REAL:
+		val->realval = *((double *) value);
 		break;
 	case PLIST_STRING:
 		val->strval = strdup((char *) value);
@@ -282,11 +204,55 @@ void plist_free(plist_t plist)
 	g_node_destroy(plist);
 }
 
+/**********************************************
+*                                             *
+*              Xml Plist stuff                *
+*                                             *
+**********************************************/
+
+#include <libxml/parser.h>
+#include <libxml/tree.h>
+
+
+const char *plist_base = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n\
+<!DOCTYPE plist PUBLIC \"-//Apple Computer//DTD PLIST 1.0//EN\" \"http://www.apple.com/DTDs/PropertyList-1.0.dtd\">\n\
+<plist version=\"1.0\">\n\
+</plist>\0";
+
 struct xml_node {
 	xmlNodePtr xml;
 	uint32_t depth;
 };
 
+/** Creates a new plist XML document.
+ * 
+ * @return The plist XML document.
+ */
+xmlDocPtr new_plist()
+{
+	char *plist = strdup(plist_base);
+	xmlDocPtr plist_xml = xmlReadMemory(plist, strlen(plist), NULL, NULL, 0);
+
+	if (!plist_xml)
+		return NULL;
+
+	free(plist);
+
+	return plist_xml;
+}
+
+/** Destroys a previously created XML document.
+ *
+ * @param plist The XML document to destroy.
+ */
+void free_plist(xmlDocPtr plist)
+{
+	if (!plist)
+		return;
+
+	xmlFreeDoc(plist);
+}
+
 void node_to_xml(GNode * node, gpointer xml_struct)
 {
 	if (!node)
@@ -311,34 +277,14 @@ void node_to_xml(GNode * node, gpointer xml_struct)
 		}
 		break;
 
-	case PLIST_UINT8:
-		tag = "integer";
-		val = g_strdup_printf("%u", node_data->intval8);
-		break;
-
-	case PLIST_UINT16:
-		tag = "integer";
-		val = g_strdup_printf("%u", node_data->intval16);
-		break;
-
-	case PLIST_UINT32:
-		tag = "integer";
-		val = g_strdup_printf("%u", node_data->intval32);
-		break;
-
-	case PLIST_UINT64:
+	case PLIST_UINT:
 		tag = "integer";
-		val = g_strdup_printf("%lu", (long unsigned int) node_data->intval64);
-		break;
-
-	case PLIST_FLOAT32:
-		tag = "real";
-		val = g_strdup_printf("%f", node_data->realval32);
+		val = g_strdup_printf("%lu", (long unsigned int) node_data->intval);
 		break;
 
-	case PLIST_FLOAT64:
+	case PLIST_REAL:
 		tag = "real";
-		val = g_strdup_printf("%Lf", (long double) node_data->intval64);
+		val = g_strdup_printf("%Lf", (long double) node_data->realval);
 		break;
 
 	case PLIST_STRING:
@@ -358,7 +304,7 @@ void node_to_xml(GNode * node, gpointer xml_struct)
 
 	case PLIST_DATA:
 		tag = "data";
-		val = format_string(node_data->buff, 60, 0);
+		val = format_string(node_data->buff, 60, xstruct->depth);
 		break;
 	case PLIST_ARRAY:
 		tag = "array";
@@ -435,15 +381,15 @@ void xml_to_node(xmlNodePtr xml_node, GNode * plist_node)
 
 		if (!xmlStrcmp(node->name, "integer")) {
 			char *strval = xmlNodeGetContent(node);
-			data->intval64 = atoi(strval);
-			data->type = PLIST_UINT64;
+			data->intval = atoi(strval);
+			data->type = PLIST_UINT;
 			continue;
 		}
 
 		if (!xmlStrcmp(node->name, "real")) {
 			char *strval = xmlNodeGetContent(node);
-			data->realval64 = atof(strval);
-			data->type = PLIST_FLOAT64;
+			data->realval = atof(strval);
+			data->type = PLIST_REAL;
 			continue;
 		}
 
@@ -493,294 +439,460 @@ void plist_to_xml(plist_t plist, char **plist_xml, uint32_t * length)
 	xmlDocDumpMemory(plist_doc, (xmlChar **) plist_xml, length);
 }
 
-GNode *parse_raw_node(const char *bpbuffer, uint32_t bplength, uint32_t * position, uint8_t ref_size)
+void xml_to_plist(const char *plist_xml, uint32_t length, plist_t * plist)
 {
-	if (!position || !bpbuffer || !bplength)
-		return NULL;
+	xmlDocPtr plist_doc = xmlReadMemory(plist_xml, length, NULL, NULL, 0);
+	xmlNodePtr root_node = xmlDocGetRootElement(plist_doc);
 
-	uint8_t modifier = 0;
 	struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
-	GNode *new_node = g_node_new(data);
-	GNode *length_stupidity = NULL;
+	*plist = g_node_new(data);
+	data->type = PLIST_PLIST;
+	xml_to_node(root_node, *plist);
+}
 
-	int myPos = *position;
-	if (myPos == bplength || (myPos + 1) == bplength) {
-		g_node_destroy(new_node);
-		return NULL;
-	}							// end of string
 
-	uint32_t length = 0;
-	if (!myPos) {
-		if (strncmp(bpbuffer, "bplist00", strlen("bplist00"))) {
-			return NULL;		// badness!
-		}
-		myPos += strlen("bplist00");
-	}
-	// Get the node's type.
-	if (bpbuffer[myPos] == BPLIST_DATE) {	// handle date separately, but do it as a real
-		// better handling of date; basically interpret as real or double
-		data->type = BPLIST_DATE;
-		length = 8;				// always 8 for "date" (Apple intended it, not me)
-		myPos++;
-		memcpy(&data->realval64, bpbuffer + myPos, sizeof(data->realval64));
-		byte_convert((char *) &data->realval64, sizeof(data->realval64));
-		myPos += length;
-		*position = myPos;
-		return new_node;
-	}
 
-	int type = bpbuffer[myPos] & BPLIST_MASK;
-	data->length = bpbuffer[myPos] & BPLIST_FILL;
+/**********************************************
+*                                             *
+*            Binary Plist stuff               *
+*                                             *
+**********************************************/
 
-	if (!type) {
-		// what? check if it's a boolean.
-		if (bpbuffer[myPos] == BPLIST_TRUE || bpbuffer[myPos] == BPLIST_FALSE) {
-			// okay, so it is. Carry on.
-			data->type = PLIST_BOOLEAN;
-			data->boolval = TRUE;
-			data->length = 0;
-		} else {
-			// er, what? we have a bad type here. Return NULL.
-			g_node_destroy(new_node);
-			//printf("parse_raw_node: lol type: type given %x\n", bpbuffer[myPos]);
-			return NULL;
-		}
-	}
+/* Magic marker and size. */
+#define BPLIST_MAGIC		"bplist"
+#define BPLIST_MAGIC_SIZE	6
 
-	myPos++;					// puts us in the data.
-	if (length == BPLIST_FILL) {	// Data happens to contain length...
-		// what? you're going to make me parse an int for the length. You suck.
-		*position = myPos;
-		length_stupidity = parse_raw_node(bpbuffer, bplength, &myPos, ref_size);
-		switch (((struct plist_data *) length_stupidity->data)->type) {
-		case PLIST_UINT8:
-			data->length = ((struct plist_data *) length_stupidity->data)->intval8;
-			break;
-		case PLIST_UINT16:
-			data->length = ((struct plist_data *) length_stupidity->data)->intval16;
-			break;
-		case PLIST_UINT32:
-			data->length = ((struct plist_data *) length_stupidity->data)->intval32;
-			break;
-		case PLIST_UINT64:
-			data->length = ((struct plist_data *) length_stupidity->data)->intval64;
-			break;
-		default:
-			g_node_destroy(new_node);
-			g_node_destroy(length_stupidity);
-			return NULL;
-		}
-		// There, we have our fucking length now.
-		*position = myPos;
-		g_node_destroy(length_stupidity);	// cleanup
+#define BPLIST_VERSION		"00"
+#define BPLIST_VERSION_SIZE	2
+
+
+#define BPLIST_TRL_SIZE 	26
+#define BPLIST_TRL_OFFSIZE_IDX 	0
+#define BPLIST_TRL_PARMSIZE_IDX 1
+#define BPLIST_TRL_NUMOBJ_IDX 	2
+#define BPLIST_TRL_ROOTOBJ_IDX 	10
+#define BPLIST_TRL_OFFTAB_IDX 	18
+
+enum {
+	BPLIST_NULL = 0x00,
+	BPLIST_TRUE = 0x08,
+	BPLIST_FALSE = 0x09,
+	BPLIST_FILL = 0x0F,			/* will be used for length grabbing */
+	BPLIST_UINT = 0x10,
+	BPLIST_REAL = 0x20,
+	BPLIST_DATE = 0x30,
+	BPLIST_DATA = 0x40,
+	BPLIST_STRING = 0x50,
+	BPLIST_UNICODE = 0x60,
+	BPLIST_UID = 0x70,
+	BPLIST_ARRAY = 0xA0,
+	BPLIST_SET = 0xC0,
+	BPLIST_DICT = 0xD0,
+	BPLIST_MASK = 0xF0
+};
+
+void byte_convert(char *address, size_t size)
+{
+	int i = 0, j = 0;
+	char tmp = '\0';
+
+	for (i = 0; i < (size / 2); i++) {
+		tmp = address[i];
+		j = ((size - 1) + 0) - i;
+		address[i] = address[j];
+		address[j] = tmp;
 	}
-	// Now we're in the data. 
-	// Error-checking sorta
-	if ((myPos + data->length) >= bplength) {
-		data->length = bplength - myPos;	// truncate the object
+}
+
+#include <byteswap.h>
+#define swap_n_bytes(x, n) \
+		n == 8 ? bswap_64(*(uint64_t *)(x)) : \
+		(n == 4 ? bswap_32(*(uint32_t *)(x)) : \
+		(n == 2 ? bswap_16(*(uint16_t *)(x)) : *(x) ))
+
+#define be64dec(x) bswap_64( *(uint64_t*)(x) )
+
+GNode *parse_uint_node(char *bnode, uint8_t size, char **next_object)
+{
+	struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+
+	size = 1 << size;			// make length less misleading
+	switch (size) {
+	case sizeof(uint8_t):
+		data->intval = bnode[0];
+		break;
+	case sizeof(uint16_t):
+		memcpy(&data->intval, bnode, size);
+		data->intval = ntohs(data->intval);
+		break;
+	case sizeof(uint32_t):
+		memcpy(&data->intval, bnode, size);
+		data->intval = ntohl(data->intval);
+		break;
+	case sizeof(uint64_t):
+		memcpy(&data->intval, bnode, size);
+		byte_convert((char *) &data->intval, size);
+		break;
+	default:
+		free(data);
+		return NULL;
+	};
+
+	*next_object = bnode + size;
+	data->type = PLIST_UINT;
+	return g_node_new(data);
+}
+
+GNode *parse_real_node(char *bnode, uint8_t size)
+{
+	struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+
+	size = 1 << size;			// make length less misleading
+	switch (size) {
+	case sizeof(float):
+		memcpy(&data->realval, bnode, size);
+		byte_convert((char *) &data->realval, size);
+		break;
+	case sizeof(double):
+		memcpy(&data->realval, bnode, size);
+		byte_convert((char *) &data->realval, size);
+		break;
+	default:
+		free(data);
+		return NULL;
 	}
-	// And now for the greatest show on earth: the giant fucking switch statement.
+	data->type = PLIST_REAL;
+	return g_node_new(data);
+}
+
+GNode *parse_string_node(char *bnode, uint8_t size)
+{
+	struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+
+	data->type = PLIST_STRING;
+	data->strval = (char *) malloc(sizeof(char) * (size + 1));
+	memcpy(data->strval, bnode, size);
+	data->strval[size] = '\0';
+
+	return g_node_new(data);
+}
+
+GNode *parse_unicode_node(char *bnode, uint8_t size)
+{
+	struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+
+	data->type = PLIST_UNICODE;
+	data->unicodeval = (wchar_t *) malloc(sizeof(wchar_t) * (size + 1));
+	memcpy(data->unicodeval, bnode, size);
+	data->unicodeval[size] = '\0';
+
+	return g_node_new(data);
+}
+
+GNode *parse_data_node(char *bnode, uint64_t size, uint32_t ref_size)
+{
+	struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+
+	data->type = PLIST_DATA;
+	data->length = size;
+	data->buff = (char *) malloc(sizeof(char) * size);
+	memcpy(data->buff, bnode, sizeof(char) * size);
+
+	return g_node_new(data);
+}
+
+GNode *parse_dict_node(char *bnode, uint64_t size, uint32_t ref_size)
+{
+	struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+
+	data->type = PLIST_DICT;
+	data->length = size;
+	data->buff = (char *) malloc(sizeof(char) * size * ref_size * 2);
+	memcpy(data->buff, bnode, sizeof(char) * size * ref_size * 2);
+
+	return g_node_new(data);
+}
+
+GNode *parse_array_node(char *bnode, uint64_t size, uint32_t ref_size)
+{
+	struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+
+	data->type = PLIST_ARRAY;
+	data->length = size;
+	data->buff = (char *) malloc(sizeof(char) * size * ref_size);
+	memcpy(data->buff, bnode, sizeof(char) * size * ref_size);
+
+	return g_node_new(data);
+}
+
+plist_type plist_get_node_type(plist_t node)
+{
+	return ((struct plist_data *) node->data)->type;
+}
+
+uint64_t plist_get_node_uint_val(plist_t node)
+{
+	if (PLIST_UINT == plist_get_node_type(node))
+		return ((struct plist_data *) node->data)->intval;
+	else
+		return 0;
+}
+
+
+GNode *parse_bin_node(char *object, uint8_t dict_size, char **next_object)
+{
+	if (!object)
+		return NULL;
+
+	uint16_t type = *object & 0xF0;
+	uint64_t size = *object & 0x0F;
+	object++;
+
 	switch (type) {
-	case BPLIST_INT:
-		data->length = uipow(2, data->length);	// make length less misleading
-		switch (data->length) {
-		case sizeof(uint8_t):
-			data->type = PLIST_UINT8;
-			data->intval8 = bpbuffer[myPos];
-			break;
-		case sizeof(uint16_t):
-			data->type = PLIST_UINT16;
-			memcpy(&data->intval16, bpbuffer + myPos, sizeof(uint16_t));
-			data->intval16 = ntohs(data->intval16);
-			break;
-		case sizeof(uint32_t):
-			data->type = PLIST_UINT32;
-			memcpy(&data->intval32, bpbuffer + myPos, sizeof(uint32_t));
-			data->intval32 = ntohl(data->intval32);
-			break;
-		case sizeof(uint64_t):
-			data->type = PLIST_UINT64;
-			memcpy(&data->intval64, bpbuffer + myPos, sizeof(uint64_t));
-			byte_convert((char *) &data->intval64, sizeof(uint64_t));
-			break;
+
+	case BPLIST_NULL:
+		switch (size) {
+
+		case BPLIST_TRUE:
+			{
+				struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+				data->type = PLIST_BOOLEAN;
+				data->boolval = TRUE;
+				return g_node_new(data);
+			}
+
+		case BPLIST_FALSE:
+			{
+				struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+				data->type = PLIST_BOOLEAN;
+				data->boolval = FALSE;
+				return g_node_new(data);
+			}
+
+		case BPLIST_NULL:
 		default:
-			g_node_destroy(new_node);
-			printf("parse_raw_node: lol: invalid int: size given %lu\n", (long unsigned int) length);
-			printf("parse_raw_node: lol: by the way sizeof(uint64) = %i\n", sizeof(uint64_t));
 			return NULL;
 		}
-		break;
+
+	case BPLIST_UINT:
+		return parse_uint_node(object, size, next_object);
 
 	case BPLIST_REAL:
-		data->length = uipow(2, data->length);
-		switch (data->length) {
-		case sizeof(float):
-			data->type = PLIST_FLOAT32;
-			memcpy(&data->realval32, bpbuffer + myPos, data->length);
-			byte_convert((char *) &data->realval32, sizeof(float));	//necessary ??
-			break;
-		case sizeof(double):
-			data->type = PLIST_FLOAT64;
-			memcpy(&data->realval64, bpbuffer + myPos, data->length);
-			byte_convert((char *) &data->realval64, sizeof(double));
-			break;
-		default:
-			g_node_destroy(new_node);
-			printf("parse_raw_node: lol: invalid real: size given %lu\n", (long unsigned int) length);
-			printf("parse_raw_node: lol: by the way sizeof(uint64) = %i\n", sizeof(uint64_t));
+		return parse_real_node(object, size);
+
+	case BPLIST_DATE:
+		if (3 != size)
 			return NULL;
+		else
+			return parse_real_node(object, size);
+
+	case BPLIST_DATA:
+		if (0x0F == size) {
+			plist_t size_node = parse_bin_node(object, dict_size, &object);
+			if (plist_get_node_type(size_node) != PLIST_UINT)
+				return NULL;
+			size = plist_get_node_uint_val(size_node);
 		}
-		break;
+		return parse_data_node(object, size, dict_size);
 
 	case BPLIST_STRING:
-		data->type = PLIST_STRING;
-		data->strval = (char *) malloc(sizeof(char) * data->length);
-		memcpy(data->strval, bpbuffer + myPos, data->length);
-		break;
+		if (0x0F == size) {
+			plist_t size_node = parse_bin_node(object, dict_size, &object);
+			if (plist_get_node_type(size_node) != PLIST_UINT)
+				return NULL;
+			size = plist_get_node_uint_val(size_node);
+		}
+		return parse_string_node(object, size);
 
 	case BPLIST_UNICODE:
-		data->type = PLIST_UNICODE;
-		data->unicodeval = (wchar_t *) malloc(sizeof(wchar_t) * data->length);
-		memcpy(data->unicodeval, bpbuffer + myPos, data->length);
-		break;
+		if (0x0F == size) {
+			plist_t size_node = parse_bin_node(object, dict_size, &object);
+			if (plist_get_node_type(size_node) != PLIST_UINT)
+				return NULL;
+			size = plist_get_node_uint_val(size_node);
+		}
+		return parse_unicode_node(object, size);
+
+	case BPLIST_UID:
+	case BPLIST_ARRAY:
+		if (0x0F == size) {
+			plist_t size_node = parse_bin_node(object, dict_size, &object);
+			if (plist_get_node_type(size_node) != PLIST_UINT)
+				return NULL;
+			size = plist_get_node_uint_val(size_node);
+		}
+		return parse_array_node(object, size, dict_size);
+
+	case BPLIST_SET:
+	case BPLIST_DICT:
+		if (0x0F == size) {
+			plist_t size_node = parse_bin_node(object, dict_size, &object);
+			if (plist_get_node_type(size_node) != PLIST_UINT)
+				return NULL;
+			object++;
+			size = plist_get_node_uint_val(size_node);
+		}
+		return parse_dict_node(object, size, dict_size);
 
-	case BPLIST_DICT:			/* returning a raw dict, it forward-references, so. */
-		data->length = data->length * 2;	// dicts lie
-		data->type = PLIST_DICT;
+	}
+	return NULL;
+}
 
-	case BPLIST_ARRAY:			/* returning a raw array, it forward-references, so. */
-		data->ref_size = ref_size;	// in arrays and dicts, the "ref size" alluded to in the trailer applies, and should be stored in intval8 so as to save space.
-		if (data->type == 0)
-			data->type = PLIST_ARRAY;
-	case BPLIST_DATA:
-		if (data->type == 0)
-			data->type = PLIST_DATA;
-	default:					/* made to hold raw data. */
-		modifier = (data->ref_size > 0) ? data->ref_size : 1;
-		data->buff = (char *) malloc(sizeof(char) * (data->length * modifier));
-		memcpy(data->buff, bpbuffer + myPos, (data->length * modifier));
+void plist_to_bin(plist_t plist, char **plist_bin, uint32_t * length)
+{
+	uint64_t num_objects = g_node_n_nodes(plist, G_TRAVERSE_ALL);
+}
+
+
+
+gpointer copy_plist_data(gconstpointer src, gpointer data)
+{
+	struct plist_data *srcdata = (struct plist_data *) src;
+	struct plist_data *dstdata = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
+
+	dstdata->type = srcdata->type;
+	dstdata->length = srcdata->length;
+	switch (dstdata->type) {
+	case PLIST_BOOLEAN:
+		dstdata->boolval = srcdata->boolval;
+		break;
+	case PLIST_UINT:
+		dstdata->intval = srcdata->intval;
+		break;
+	case PLIST_DATE:
+	case PLIST_REAL:
+		dstdata->realval = srcdata->realval;
+		break;
+	case PLIST_KEY:
+	case PLIST_STRING:
+		dstdata->strval = strdup(srcdata->strval);
+		break;
+	case PLIST_UNICODE:
+		dstdata->unicodeval = wcsdup(srcdata->unicodeval);
+		break;
+	case PLIST_PLIST:
+	case PLIST_DATA:
+	case PLIST_ARRAY:
+	case PLIST_DICT:
+		dstdata->buff = (char *) malloc(sizeof(char *) * srcdata->length);
+		memcpy(dstdata->buff, srcdata->buff, sizeof(char *) * srcdata->length);
 		break;
 
+	default:
+		break;
 	}
 
-	myPos += data->length;
-	*position = myPos;
-	return new_node;
+	return dstdata;
 }
 
-plist_t parse_nodes(const char *bpbuffer, uint32_t bplength, uint32_t * position)
+void bin_to_plist(const char *plist_bin, uint32_t length, plist_t * plist)
 {
-	plist_t *nodeslist = NULL, *newaddr = NULL;
-	plist_t new_node = NULL, root_node = NULL;
+	//first check we have enough data
+	if (!(length >= BPLIST_MAGIC_SIZE + BPLIST_VERSION_SIZE + BPLIST_TRL_SIZE))
+		return;
+	//check that plist_bin in actually a plist
+	if (memcmp(plist_bin, BPLIST_MAGIC, BPLIST_MAGIC_SIZE) != 0)
+		return;
+	//check for known version
+	if (memcmp(plist_bin + BPLIST_MAGIC_SIZE, BPLIST_VERSION, BPLIST_VERSION_SIZE) != 0)
+		return;
 
-	uint32_t nodeslength = 0;
-	uint8_t offset_size = 0, dict_param_size = 0;
-	offset_size = bpbuffer[bplength - 26];
-	dict_param_size = bpbuffer[bplength - 25];
-	uint64_t current_offset = 0;
-	uint64_t num_objects = 0, root_object = 0, offset_table_index = 0;
-	memcpy(&num_objects, bpbuffer + bplength - 24, sizeof(uint64_t));
-	memcpy(&root_object, bpbuffer + bplength - 16, sizeof(uint64_t));
-	memcpy(&offset_table_index, bpbuffer + bplength - 8, sizeof(uint64_t));
-	byte_convert((char *) &num_objects, sizeof(uint64_t));
-	byte_convert((char *) &root_object, sizeof(uint64_t));
-	byte_convert((char *) &offset_table_index, sizeof(uint64_t));
-
-	log_debug_msg("Offset size: %i\nGiven: %i\n", offset_size, bpbuffer[bplength - 26]);
-	log_debug_msg("Ref size: %i\nGiven: %i\n", dict_param_size, bpbuffer[bplength - 25]);
-	log_debug_msg("Number of objects: %lli\nGiven: %llu\n", num_objects, *(bpbuffer + bplength - 24));
-	log_debug_msg("Root object index: %lli\nGiven: %llu\n", root_object, *(bpbuffer + bplength - 16));
-	log_debug_msg("Offset table index: %lli\nGiven: %llu\n", offset_table_index, *(bpbuffer + bplength - 8));
-	log_debug_msg("Size of uint64: %i\n", sizeof(uint64_t));
-
-	int i = 0, j = 0, k = 0, str_i = 0, str_j = 0;
-	uint32_t index1 = 0, index2 = 0;
+	//now parse trailer
+	const char *trailer = plist_bin + (length - BPLIST_TRL_SIZE);
+
+	uint8_t offset_size = trailer[BPLIST_TRL_OFFSIZE_IDX];
+	uint8_t dict_param_size = trailer[BPLIST_TRL_PARMSIZE_IDX];
+	uint64_t num_objects = be64dec(trailer + BPLIST_TRL_NUMOBJ_IDX);
+	uint64_t root_object = be64dec(trailer + BPLIST_TRL_ROOTOBJ_IDX);
+	uint64_t offset_table_index = be64dec(trailer + BPLIST_TRL_OFFTAB_IDX);
 
+	log_debug_msg("Offset size: %i\n", offset_size);
+	log_debug_msg("Ref size: %i\n", dict_param_size);
+	log_debug_msg("Number of objects: %lli\n", num_objects);
+	log_debug_msg("Root object index: %lli\n", root_object);
+	log_debug_msg("Offset table index: %lli\n", offset_table_index);
+
+	if (num_objects == 0)
+		return;
+
+	//allocate serialized array of nodes
+	plist_t *nodeslist = NULL;
 	nodeslist = (plist_t *) malloc(sizeof(plist_t) * num_objects);
+
 	if (!nodeslist)
-		return NULL;
+		return;
 
+	//parse serialized nodes
+	uint64_t i = 0;
+	uint64_t current_offset = 0;
+	const char *offset_table = plist_bin + offset_table_index;
 	for (i = 0; i < num_objects; i++) {
-		memcpy(&current_offset, bpbuffer + (offset_table_index + (i * offset_size)), offset_size);
-		byte_convert((char *) &current_offset,
-					 (offset_size <= sizeof(current_offset)) ? offset_size : sizeof(current_offset));
-		log_debug_msg("parse_nodes: current_offset = %x\n", current_offset);
-		nodeslist[i] = parse_raw_node(bpbuffer, bplength, (uint32_t *) & current_offset, dict_param_size);
+		current_offset = swap_n_bytes(offset_table + i * offset_size, offset_size);
+
+		log_debug_msg("parse_nodes: current_offset = %i\n", current_offset);
+		char *obj = plist_bin + current_offset;
+		nodeslist[i] = parse_bin_node(obj, dict_param_size, &obj);
 		log_debug_msg("parse_nodes: parse_raw_node done\n");
 	}
 
+	//setup children for structured types
+	int j = 0, str_i = 0, str_j = 0;
+	uint32_t index1 = 0, index2 = 0;
 
 	for (i = 0; i < num_objects; i++) {
-		// set elements for dicts and arrays and leave the rest alone
+
 		log_debug_msg("parse_nodes: on node %i\n", i);
 		struct plist_data *data = (struct plist_data *) nodeslist[i]->data;
 
 		switch (data->type) {
 		case PLIST_DICT:
 			log_debug_msg("parse_nodes: dictionary found\n");
-			for (j = 0; j < (data->length / 2); j++) {
-				str_i = j * data->ref_size;
-				str_j = (j + (data->length / 2)) * data->ref_size;
-
-				memcpy(&index1, data->buff + str_i, data->ref_size);
-				memcpy(&index2, data->buff + str_j, data->ref_size);
-
-				byte_convert((char *) &index1, (dict_param_size <= sizeof(index1)) ? dict_param_size : sizeof(index2));
-				byte_convert((char *) &index2, (dict_param_size <= sizeof(index2)) ? dict_param_size : sizeof(index2));
-
-				g_node_append(nodeslist[i], nodeslist[index1]);
-				g_node_append(nodeslist[i], nodeslist[index2]);
+			for (j = 0; j < data->length; j++) {
+				str_i = j * dict_param_size;
+				str_j = (j + data->length) * dict_param_size;
+
+				index1 = swap_n_bytes(data->buff + str_i, dict_param_size);
+				index2 = swap_n_bytes(data->buff + str_j, dict_param_size);
+
+				//first one is actually a key
+				((struct plist_data *) nodeslist[index1]->data)->type = PLIST_KEY;
+				//g_node_append(nodeslist[i], nodeslist[index1]);
+				//g_node_append(nodeslist[i], nodeslist[index2]);
+
+				if (G_NODE_IS_ROOT(nodeslist[index1]))
+					g_node_append(nodeslist[i], nodeslist[index1]);
+				else
+					g_node_append(nodeslist[i], g_node_copy_deep(nodeslist[index1], copy_plist_data, NULL));
+
+				if (G_NODE_IS_ROOT(nodeslist[index2]))
+					g_node_append(nodeslist[i], nodeslist[index2]);
+				else
+					g_node_append(nodeslist[i], g_node_copy_deep(nodeslist[index2], copy_plist_data, NULL));
 			}
 
-			data->length = data->length / 2;
 			free(data->buff);
-			k = 0;
 			break;
 
 		case PLIST_ARRAY:
 			log_debug_msg("parse_nodes: array found\n");
 			for (j = 0; j < data->length; j++) {
-				log_debug_msg("parse_nodes: array index %i\n", j);
-				str_j = j * data->ref_size;
-				memcpy(&index1, data->buff + str_j, data->ref_size);
-				log_debug_msg("parse_nodes: post-memcpy\n");
-				byte_convert((char *) &index1, (dict_param_size <= sizeof(index1)) ? dict_param_size : sizeof(index1));
-				log_debug_msg("parse_nodes: post-ntohl\nindex1 = %i\n", index1);
-				g_node_append(nodeslist[i], nodeslist[index1]);
-				log_debug_msg("parse_nodes: post-assignment\n");
+				str_j = j * dict_param_size;
+				index1 = swap_n_bytes(data->buff + str_j, dict_param_size);
+
+				//g_node_append(nodeslist[i], nodeslist[index1]);
+				if (G_NODE_IS_ROOT(nodeslist[index1]))
+					g_node_append(nodeslist[i], nodeslist[index1]);
+				else
+					g_node_append(nodeslist[i], g_node_copy_deep(nodeslist[index1], copy_plist_data, NULL));
 			}
 			free(data->buff);
 			break;
 		default:
-			//printf("lol... type %x\n", nodeslist[i]->type);
 			break;
-		}						// those are the only two we need to correct for.
+		}
 	}
 
-	root_node = nodeslist[root_object];
-	return root_node;
-}
-
-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)
-{
-	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);
-	*plist = g_node_new(data);
-	data->type = PLIST_PLIST;
-	xml_to_node(root_node, *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);
+	*plist = nodeslist[root_object];
 }
 
 
@@ -816,23 +928,11 @@ char compare_node_value(plist_type type, struct plist_data *data, void *value)
 	case PLIST_BOOLEAN:
 		res = data->boolval == *((char *) value) ? TRUE : FALSE;
 		break;
-	case PLIST_UINT8:
-		res = data->intval8 == *((uint8_t *) value) ? TRUE : FALSE;
-		break;
-	case PLIST_UINT16:
-		res = data->intval16 == *((uint16_t *) value) ? TRUE : FALSE;
-		break;
-	case PLIST_UINT32:
-		res = data->intval32 == *((uint32_t *) value) ? TRUE : FALSE;
-		break;
-	case PLIST_UINT64:
-		res = data->intval64 == *((uint64_t *) value) ? TRUE : FALSE;
+	case PLIST_UINT:
+		res = data->intval == *((uint64_t *) value) ? TRUE : FALSE;
 		break;
-	case PLIST_FLOAT32:
-		res = data->realval32 == *((float *) value) ? TRUE : FALSE;
-		break;
-	case PLIST_FLOAT64:
-		res = data->realval64 == *((double *) value) ? TRUE : FALSE;
+	case PLIST_REAL:
+		res = data->realval == *((double *) value) ? TRUE : FALSE;
 		break;
 	case PLIST_KEY:
 	case PLIST_STRING:
@@ -889,23 +989,11 @@ void get_type_and_value(GNode * node, plist_type * type, void *value)
 	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;
+	case PLIST_UINT:
+		*((uint64_t *) value) = data->intval;
 		break;
-	case PLIST_FLOAT64:
-		*((double *) value) = data->realval64;
+	case PLIST_REAL:
+		*((double *) value) = data->realval;
 		break;
 	case PLIST_STRING:
 		*((char **) value) = strdup(data->strval);
diff --git a/src/plist.h b/src/plist.h
index ed3d2b2..df1d3e4 100644
--- a/src/plist.h
+++ b/src/plist.h
@@ -35,7 +35,7 @@ char *format_string(const char *buf, int cols, int depth);
 
 /* Binary plist stuff */
 
-
+/*
 typedef enum {
 	PLIST_BOOLEAN,
 	PLIST_UINT8,
@@ -53,6 +53,22 @@ typedef enum {
 	PLIST_PLIST,
 	PLIST_KEY,
 } plist_type;
+*/
+
+typedef enum {
+	PLIST_BOOLEAN,
+	PLIST_UINT,
+	PLIST_REAL,
+	PLIST_STRING,
+	PLIST_UNICODE,
+	PLIST_ARRAY,
+	PLIST_DICT,
+	PLIST_DATE,
+	PLIST_DATA,
+	PLIST_PLIST,
+	PLIST_KEY,
+} plist_type;
+
 
 
 typedef GNode *plist_t;
-- 
cgit v1.1-32-gdbae