Add Unicode support.

1 files changed, 25 insertions, 13 deletions

diff --git a/src/bplist.c b/src/bplist.c
index fd0280c..67d7f72 100644
--- a/src/bplist.c
+++ b/src/bplist.c
@@ -21,7 +21,6 @@
 
 
 #include "plist.h"
-#include <wchar.h>
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
@@ -152,7 +151,7 @@ static plist_t parse_date_node(char *bnode, uint8_t size)
 	return node;
 }
 
-static plist_t parse_string_node(char *bnode, uint8_t size)
+static plist_t parse_string_node(char *bnode, uint64_t size)
 {
 	plist_data_t data = plist_new_plist_data();
 
@@ -165,16 +164,17 @@ static plist_t parse_string_node(char *bnode, uint8_t size)
 	return g_node_new(data);
 }
 
-static plist_t parse_unicode_node(char *bnode, uint8_t size)
+static plist_t parse_unicode_node(char *bnode, uint64_t size)
 {
 	plist_data_t data = plist_new_plist_data();
-
+	uint64_t i = 0;
 	data->type = PLIST_UNICODE;
-	data->unicodeval = (wchar_t *) malloc(sizeof(wchar_t) * (size + 1));
-	memcpy(data->unicodeval, bnode, size);
-	data->unicodeval[size] = '\0';
-	data->length = wcslen(data->unicodeval);
-
+	data->unicodeval = (gunichar2 *) malloc(sizeof(gunichar2) * (size + 1));
+	memcpy(data->unicodeval, bnode, sizeof(gunichar2) * size);
+	data->unicodeval[sizeof(gunichar2) * size] = '\0';
+	data->length = size;
+	for (i = 0; i <= size; i++)
+		byte_convert(data->unicodeval + i, sizeof(gunichar2));
 	return g_node_new(data);
 }
 
@@ -338,7 +338,8 @@ static gpointer copy_plist_data(gconstpointer src, gpointer data)
 		dstdata->strval = strdup(srcdata->strval);
 		break;
 	case PLIST_UNICODE:
-		dstdata->unicodeval = wcsdup(srcdata->unicodeval);
+		dstdata->unicodeval = (gunichar2*) malloc(srcdata->length * sizeof(gunichar2));
+		memcpy(dstdata->unicodeval, srcdata->unicodeval, srcdata->length * sizeof(gunichar2));
 		break;
 	case PLIST_DATA:
 	case PLIST_ARRAY:
@@ -493,7 +494,7 @@ static guint plist_data_hash(gconstpointer key)
 		break;
 	case PLIST_UNICODE:
 		buff = (char *) data->unicodeval;
-		size = strlen(buff) * sizeof(wchar_t);
+		size = data->length;
 		break;
 	case PLIST_DATA:
 	case PLIST_ARRAY:
@@ -544,7 +545,7 @@ static gboolean plist_data_compare(gconstpointer a, gconstpointer b)
 		else
 			return FALSE;
 	case PLIST_UNICODE:
-		if (!wcscmp(val_a->unicodeval, val_b->unicodeval))
+		if (!memcmp(val_a->unicodeval, val_b->unicodeval,val_a->length))
 			return TRUE;
 		else
 			return FALSE;
@@ -662,6 +663,17 @@ static void write_string(GByteArray * bplist, char *val)
 	write_raw_data(bplist, BPLIST_STRING, (uint8_t *) val, size);
 }
 
+static void write_unicode(GByteArray * bplist, gunichar2 *val, uint64_t size)
+{
+	uint64_t i = 0;
+	uint64_t size2 = size * sizeof(gunichar2);
+	uint8_t* buff = (uint8_t*) malloc(size2);
+	memcpy(buff, val, size2);
+	for (i = 0; i < size; i++)
+		byte_convert(buff + i * sizeof(gunichar2), sizeof(gunichar2));
+	write_raw_data(bplist, BPLIST_STRING, buff, size2);
+}
+
 static void write_array(GByteArray * bplist, GNode * node, GHashTable * ref_table, uint8_t dict_param_size)
 {
 	uint64_t size = g_node_n_children(node);
@@ -784,7 +796,7 @@ void plist_to_bin(plist_t plist, char **plist_bin, uint32_t * length)
 			write_string(bplist_buff, data->strval);
 			break;
 		case PLIST_UNICODE:
-			//TODO
+			write_unicode(bplist_buff, data->unicodeval, data->length);
 			break;
 		case PLIST_DATA:
 			write_data(bplist_buff, data->buff, data->length);