1 files changed, 41 insertions, 19 deletions

diff --git a/src/bplist.c b/src/bplist.c
index 2a82b9f..6b2d2f3 100644
--- a/src/bplist.c
+++ b/src/bplist.c
@@ -24,6 +24,8 @@
 #include <stdio.h>
 #include <string.h>
 
+#include <libxml/tree.h>
+
 #include <plist/plist.h>
 #include "plist.h"
 
@@ -184,13 +186,28 @@ 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 = (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((uint8_t*)(data->unicodeval + i), sizeof(gunichar2));
+        gunichar2 *unicodestr = NULL;
+        gchar *tmpstr = NULL;
+        int type = 0;
+        glong items_read = 0;
+        glong items_written = 0;
+        GError *error = NULL;
+
+        data->type = PLIST_STRING;
+        unicodestr = (gunichar2 *) malloc(sizeof(gunichar2) * size);
+        memcpy(unicodestr, bnode, sizeof(gunichar2) * size);
+        for (i = 0; i < size; i++)
+                byte_convert((uint8_t*)(unicodestr + i), sizeof(gunichar2));
+
+        tmpstr = g_utf16_to_utf8(unicodestr, size, &items_read, &items_written, &error);
+        free(unicodestr);
+
+        data->type = PLIST_STRING;
+        data->strval = (char *) malloc(sizeof(char) * (items_written + 1));
+        memcpy(data->strval, tmpstr, items_written);
+        data->strval[items_written] = '\0';
+        data->length = strlen(data->strval);
+        g_free(tmpstr);
         return g_node_new(data);
 }
 
@@ -364,10 +381,6 @@ static gpointer copy_plist_data(gconstpointer src, gpointer data)
         case PLIST_STRING:
                 dstdata->strval = strdup(srcdata->strval);
                 break;
-        case PLIST_UNICODE:
-                dstdata->unicodeval = (gunichar2 *) malloc(srcdata->length * sizeof(gunichar2));
-                memcpy(dstdata->unicodeval, srcdata->unicodeval, srcdata->length * sizeof(gunichar2));
-                break;
         case PLIST_DATA:
         case PLIST_ARRAY:
                 dstdata->buff = (uint8_t *) malloc(sizeof(uint8_t *) * srcdata->length);
@@ -520,10 +533,6 @@ static guint plist_data_hash(gconstpointer key)
                 buff = data->strval;
                 size = strlen(buff);
                 break;
-        case PLIST_UNICODE:
-                buff = (char *) data->unicodeval;
-                size = data->length;
-                break;
         case PLIST_DATA:
         case PLIST_ARRAY:
         case PLIST_DICT:
@@ -752,6 +761,13 @@ void plist_to_bin(plist_t plist, char **plist_bin, uint32_t * length)
         uint64_t *offsets = NULL;
         uint8_t pad[6] = { 0, 0, 0, 0, 0, 0 };
         uint8_t trailer[BPLIST_TRL_SIZE];
+        //for string
+        glong len = 0;
+        int type = 0;
+        glong items_read = 0;
+        glong items_written = 0;
+        GError *error = NULL;
+        gunichar2 *unicodestr = NULL;
 
         //check for valid input
         if (!plist || !plist_bin || *plist_bin || !length)
@@ -806,10 +822,16 @@ void plist_to_bin(plist_t plist, char **plist_bin, uint32_t * length)
 
                 case PLIST_KEY:
                 case PLIST_STRING:
-                        write_string(bplist_buff, data->strval);
-                        break;
-                case PLIST_UNICODE:
-                        write_unicode(bplist_buff, data->unicodeval, data->length);
+                        len = strlen(data->strval);
+                        type = xmlDetectCharEncoding(data->strval, len);
+                        if (XML_CHAR_ENCODING_UTF8 == type) {
+                                unicodestr = g_utf8_to_utf16(data->strval, len, &items_read, &items_written, &error);
+                                write_unicode(bplist_buff, unicodestr, items_written);
+                                g_free(unicodestr);
+                        }
+                        else if (XML_CHAR_ENCODING_ASCII == type || XML_CHAR_ENCODING_NONE == type) {
+                                write_string(bplist_buff, data->strval);
+                        }
                         break;
                 case PLIST_DATA:
                         write_data(bplist_buff, data->buff, data->length);