Add a deep copy function and value setters for nodes.

1 files changed, 116 insertions, 3 deletions

diff --git a/src/plist.c b/src/plist.c
index 7949bce..e43d50d 100644
--- a/src/plist.c
+++ b/src/plist.c
@@ -109,7 +109,7 @@ static plist_t plist_add_sub_element(plist_t node, plist_type type, const void *
                                 data->strval = strdup((char *) value);
                                 break;
                         case PLIST_DATA:
-                                data->buff = (uint8_t*)malloc(length);
+                                data->buff = (uint8_t *) malloc(length);
                                 memcpy(data->buff, value, length);
                                 break;
                         case PLIST_DATE:
@@ -138,6 +138,28 @@ void plist_free(plist_t plist)
         g_node_destroy(plist);
 }
 
+static void plist_copy_node(GNode * node, gpointer parent_node)
+{
+        plist_t newnode = NULL;
+        plist_data_t data = plist_get_data(node);
+        plist_data_t newdata = plist_new_plist_data();
+
+        assert(data);                           // plist should always have data
+
+        memcpy(newdata, data, sizeof(struct plist_data_s));
+        newnode = plist_new_node(newdata);
+
+        if (parent_node) {
+                g_node_append(parent_node, newnode);
+        }
+        g_node_children_foreach(node, G_TRAVERSE_ALL, plist_copy_node, newnode);
+}
+
+plist_t plist_copy(plist_t node)
+{
+        plist_copy_node(node, NULL);
+}
+
 plist_t plist_get_first_child(plist_t node)
 {
         return (plist_t) g_node_first_child((GNode *) node);
@@ -160,7 +182,7 @@ plist_t plist_get_array_nth_el(plist_t node, uint32_t n)
                 uint32_t i = 0;
                 plist_t temp = plist_get_first_child(node);
 
-                while ( i <= n && temp) {
+                while (i <= n && temp) {
                         if (i == n)
                                 ret = temp;
                         temp = plist_get_next_sibling(temp);
@@ -465,5 +487,96 @@ gboolean plist_data_compare(gconstpointer a, gconstpointer b)
 
 char plist_compare_node_value(plist_t node_l, plist_t node_r)
 {
-        return plist_data_compare( node_l , node_r );
+        return plist_data_compare(node_l, node_r);
+}
+
+static plist_t plist_set_element_val(plist_t node, plist_type type, const void *value, uint64_t length)
+{
+        //free previous allocated buffer
+        plist_data_t data = plist_get_data(node);
+        assert(data);                           // a node should always have data attached
+
+        switch (data->type) {
+        case PLIST_KEY:
+        case PLIST_STRING:
+                free(data->strval);
+                data->strval = NULL;
+                break;
+        case PLIST_DATA:
+                free(data->buff);
+                data->buff = NULL;
+                break;
+        default:
+                break;
+        }
+
+        //now handle value
+
+        data->type = type;
+        data->length = length;
+
+        switch (type) {
+        case PLIST_BOOLEAN:
+                data->boolval = *((char *) value);
+                break;
+        case PLIST_UINT:
+                data->intval = *((uint64_t *) value);
+                break;
+        case PLIST_REAL:
+                data->realval = *((double *) value);
+                break;
+        case PLIST_KEY:
+        case PLIST_STRING:
+                data->strval = strdup((char *) value);
+                break;
+        case PLIST_DATA:
+                data->buff = (uint8_t *) malloc(length);
+                memcpy(data->buff, value, length);
+                break;
+        case PLIST_DATE:
+                data->timeval.tv_sec = ((GTimeVal *) value)->tv_sec;
+                data->timeval.tv_usec = ((GTimeVal *) value)->tv_usec;
+                break;
+        case PLIST_ARRAY:
+        case PLIST_DICT:
+        default:
+                break;
+        }
+}
+
+
+void plist_set_key_val(plist_t node, const char *val)
+{
+        plist_set_element_val(node, PLIST_KEY, val, strlen(val));
+}
+
+void plist_set_string_val(plist_t node, const char *val)
+{
+        plist_set_element_val(node, PLIST_STRING, val, strlen(val));
+}
+
+void plist_set_bool_val(plist_t node, uint8_t val)
+{
+        plist_set_element_val(node, PLIST_BOOLEAN, &val, sizeof(uint8_t));
+}
+
+void plist_set_uint_val(plist_t node, uint64_t val)
+{
+        plist_set_element_val(node, PLIST_UINT, &val, sizeof(uint64_t));
+}
+
+void plist_set_real_val(plist_t node, double val)
+{
+        plist_set_element_val(node, PLIST_REAL, &val, sizeof(double));
+}
+
+void plist_set_data_val(plist_t node, const char *val, uint64_t length)
+{
+        plist_set_element_val(node, PLIST_DATA, val, length);
+}
+
+void plist_set_date_val(plist_t node, int32_t sec, int32_t usec)
+{
+        GTimeVal val = { sec, usec };
+        plist_set_element_val(node, PLIST_DATE, &val, sizeof(GTimeVal));
 }