summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorGravatar Jonathan Beck2008-12-10 23:22:12 +0100
committerGravatar Jonathan Beck2008-12-10 23:22:12 +0100
commit625633203a27f569bea8890cb269132fea83b497 (patch)
tree884d6e4855755ffc1986f661985dc6fa3b81c221 /src
parent1a06347d27ca51283de3a9ff21e138a3ea9ba9b6 (diff)
downloadlibimobiledevice-625633203a27f569bea8890cb269132fea83b497.tar.gz
libimobiledevice-625633203a27f569bea8890cb269132fea83b497.tar.bz2
add bplist writting capability.
Diffstat (limited to 'src')
-rw-r--r--src/plist.c383
-rw-r--r--src/plist.h23
2 files changed, 356 insertions, 50 deletions
diff --git a/src/plist.c b/src/plist.c
index c691c16..431c64a 100644
--- a/src/plist.c
+++ b/src/plist.c
@@ -25,6 +25,8 @@
#include "utils.h"
#include "plist.h"
#include <wchar.h>
+#include <stdlib.h>
+#include <stdio.h>
/**********************************************
* *
@@ -127,7 +129,7 @@ void plist_new_plist(plist_t * plist)
if (*plist != NULL)
return;
struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
- data->type = PLIST_PLIST;
+ data->type = PLIST_DICT;
*plist = g_node_new(data);
}
@@ -191,7 +193,6 @@ void plist_add_dict_element(dict_t dict, char *key, plist_type type, void *value
case PLIST_ARRAY:
case PLIST_DICT:
case PLIST_DATE:
- case PLIST_PLIST:
default:
break;
}
@@ -314,10 +315,6 @@ void node_to_xml(GNode * node, gpointer xml_struct)
tag = "dict";
isStruct = TRUE;
break;
- case PLIST_PLIST:
- tag = "plist";
- isStruct = TRUE;
- break;
case PLIST_DATE: //TODO : handle date tag
default:
break;
@@ -332,8 +329,7 @@ void node_to_xml(GNode * node, gpointer xml_struct)
g_free(val);
//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)
+ if (node_data->type == PLIST_ARRAY || node_data->type == PLIST_DICT || node_data->type == PLIST_DATA)
xmlNodeAddContent(child_node, "\n");
if (isStruct) {
@@ -341,8 +337,7 @@ void node_to_xml(GNode * node, gpointer xml_struct)
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) {
+ if (node_data->type == PLIST_ARRAY || node_data->type == PLIST_DICT || node_data->type == PLIST_DATA) {
for (i = 0; i < xstruct->depth; i++) {
xmlNodeAddContent(child_node, "\t");
@@ -446,7 +441,7 @@ void xml_to_plist(const char *plist_xml, uint32_t length, plist_t * plist)
struct plist_data *data = (struct plist_data *) calloc(sizeof(struct plist_data), 1);
*plist = g_node_new(data);
- data->type = PLIST_PLIST;
+ data->type = PLIST_DICT;
xml_to_node(root_node, *plist);
}
@@ -512,6 +507,9 @@ void byte_convert(char *address, size_t size)
#define be64dec(x) bswap_64( *(uint64_t*)(x) )
+#define get_needed_bytes(x) (x <= 1<<8 ? 1 : ( x <= 1<<16 ? 2 : ( x <= 1<<32 ? 4 : 8)))
+#define get_real_bytes(x) (x >> 32 ? 4 : 8)
+
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);
@@ -638,7 +636,6 @@ uint64_t plist_get_node_uint_val(plist_t node)
return 0;
}
-
GNode *parse_bin_node(char *object, uint8_t dict_size, char **next_object)
{
if (!object)
@@ -738,13 +735,6 @@ GNode *parse_bin_node(char *object, uint8_t dict_size, char **next_object)
return NULL;
}
-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;
@@ -770,7 +760,6 @@ gpointer copy_plist_data(gconstpointer src, gpointer data)
case PLIST_UNICODE:
dstdata->unicodeval = wcsdup(srcdata->unicodeval);
break;
- case PLIST_PLIST:
case PLIST_DATA:
case PLIST_ARRAY:
case PLIST_DICT:
@@ -856,8 +845,6 @@ void bin_to_plist(const char *plist_bin, uint32_t length, plist_t * plist)
//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]);
@@ -895,7 +882,6 @@ void bin_to_plist(const char *plist_bin, uint32_t length, plist_t * plist)
*plist = nodeslist[root_object];
}
-
GNode *find_query_node(plist_t plist, char *key, char *request)
{
if (!plist)
@@ -912,7 +898,7 @@ GNode *find_query_node(plist_t plist, char *key, char *request)
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) {
+ if (data->type == PLIST_DICT || data->type == PLIST_ARRAY) {
GNode *sub = find_query_node(current, key, request);
if (sub)
return sub;
@@ -947,7 +933,6 @@ char compare_node_value(plist_type type, struct plist_data *data, void *value)
case PLIST_ARRAY:
case PLIST_DICT:
case PLIST_DATE:
- case PLIST_PLIST:
default:
break;
}
@@ -967,7 +952,7 @@ GNode *find_node(plist_t plist, plist_type type, void *value)
if (data->type == type && compare_node_value(type, data, value)) {
return current;
}
- if (data->type == PLIST_DICT || data->type == PLIST_ARRAY || data->type == PLIST_PLIST) {
+ if (data->type == PLIST_DICT || data->type == PLIST_ARRAY) {
GNode *sub = find_node(current, type, value);
if (sub)
return sub;
@@ -1008,8 +993,352 @@ void get_type_and_value(GNode * node, plist_type * type, void *value)
case PLIST_ARRAY:
case PLIST_DICT:
case PLIST_DATE:
- case PLIST_PLIST:
default:
break;
}
}
+
+guint plist_data_hash(gconstpointer key)
+{
+ struct plist_data *data = (struct plist_data *) ((GNode *) key)->data;
+
+ guint hash = data->type;
+ guint i = 0;
+
+ char *buff = NULL;
+ guint size = 0;
+
+ switch (data->type) {
+ case PLIST_BOOLEAN:
+ case PLIST_UINT:
+ case PLIST_REAL:
+ buff = (char *) &data->intval;
+ size = 8;
+
+ case PLIST_KEY:
+ case PLIST_STRING:
+ buff = data->strval;
+ size = strlen(buff);
+
+ case PLIST_UNICODE:
+ buff = data->unicodeval;
+ size = strlen(buff) * sizeof(wchar_t);
+
+ case PLIST_DATA:
+ case PLIST_ARRAY:
+ case PLIST_DICT:
+ //for these types only hash pointer
+ buff = &key;
+ size = sizeof(gconstpointer);
+ break;
+ case PLIST_DATE:
+ default:
+ break;
+ }
+
+ //now perform hash
+ for (i = 0; i < size; buff++, i++)
+ hash = hash << 7 ^ (*buff);
+
+ return hash;
+}
+
+gboolean plist_data_compare(gconstpointer a, gconstpointer b)
+{
+ if (!a || !b)
+ return FALSE;
+
+ if (!((GNode *) a)->data || !((GNode *) b)->data)
+ return FALSE;
+
+ struct plist_data *val_a = (struct plist_data *) ((GNode *) a)->data;
+ struct plist_data *val_b = (struct plist_data *) ((GNode *) b)->data;
+
+ if (val_a->type != val_b->type)
+ return FALSE;
+
+ switch (val_a->type) {
+ case PLIST_BOOLEAN:
+ case PLIST_UINT:
+ case PLIST_REAL:
+ if (val_a->intval == val_b->intval) //it is an union so this is sufficient
+ return TRUE;
+ else
+ return FALSE;
+
+ case PLIST_KEY:
+ case PLIST_STRING:
+ if (!strcmp(val_a->strval, val_b->strval))
+ return TRUE;
+ else
+ return FALSE;
+ case PLIST_UNICODE:
+ if (!strcmp(val_a->unicodeval, val_b->unicodeval))
+ return TRUE;
+ else
+ return FALSE;
+
+ case PLIST_DATA:
+ case PLIST_ARRAY:
+ case PLIST_DICT:
+ //compare pointer
+ if (a == b)
+ return TRUE;
+ else
+ return FALSE;
+ break;
+ case PLIST_DATE:
+ default:
+ break;
+ }
+ return FALSE;
+}
+
+struct serialize_s {
+ GPtrArray *objects;
+ GHashTable *ref_table;
+};
+
+void serialize_plist(GNode * node, gpointer data)
+{
+ struct serialize_s *ser = (struct serialize_s *) data;
+ uint64_t current_index = ser->objects->len;
+
+ //first check that node is not yet in objects
+ gpointer val = g_hash_table_lookup(ser->ref_table, node);
+ if (val) {
+ //data is already in table
+ return;
+ }
+ //insert new ref
+ g_hash_table_insert(ser->ref_table, node, GUINT_TO_POINTER(current_index));
+
+ //now append current node to object array
+ g_ptr_array_add(ser->objects, node);
+
+ //now recurse on children
+ g_node_children_foreach(node, G_TRAVERSE_ALL, serialize_plist, data);
+ return;
+}
+
+
+
+void write_int(GByteArray * bplist, uint64_t val)
+{
+ uint64_t size = get_needed_bytes(val);
+ uint8_t *buff = (uint8_t *) malloc(sizeof(uint8_t) + size);
+ buff[0] = BPLIST_UINT | size >> 1;
+ memcpy(buff + 1, &val, size);
+ swap_n_bytes(buff + 1, size);
+ g_byte_array_append(bplist, buff, sizeof(uint8_t) + size);
+ free(buff);
+}
+
+void write_real(GByteArray * bplist, double val)
+{
+ uint64_t size = get_real_bytes(*((uint64_t *) & val)); //cheat to know used space
+ uint8_t *buff = (uint8_t *) malloc(sizeof(uint8_t) + size);
+ buff[0] = BPLIST_REAL | size >> 1;
+ memcpy(buff + 1, &val, size);
+ swap_n_bytes(buff + 1, size);
+ g_byte_array_append(bplist, buff, sizeof(uint8_t) + size);
+ free(buff);
+}
+
+void write_raw_data(GByteArray * bplist, uint8_t mark, uint8_t * val, uint64_t size)
+{
+ uint8_t marker = mark | (size < 15 ? size : 0xf);
+ g_byte_array_append(bplist, &marker, sizeof(uint8_t));
+ if (size >= 15) {
+ GByteArray *int_buff = g_byte_array_new();
+ write_int(int_buff, size);
+ g_byte_array_append(bplist, int_buff->data, int_buff->len);
+ g_byte_array_free(int_buff, TRUE);
+ }
+ uint8_t *buff = (uint8_t *) malloc(size);
+ memcpy(buff, val, size);
+ g_byte_array_append(bplist, buff, size);
+ free(buff);
+}
+
+void write_data(GByteArray * bplist, uint8_t * val, uint64_t size)
+{
+ write_raw_data(bplist, BPLIST_DATA, val, size);
+}
+
+void write_string(GByteArray * bplist, char *val)
+{
+ uint64_t size = strlen(val);
+ write_raw_data(bplist, BPLIST_STRING, val, size);
+}
+
+void write_array(GByteArray * bplist, GNode * node, GHashTable * ref_table, uint8_t dict_param_size)
+{
+ uint64_t size = g_node_n_children(node);
+ uint8_t marker = BPLIST_ARRAY | (size < 15 ? size : 0xf);
+ g_byte_array_append(bplist, &marker, sizeof(uint8_t));
+ if (size >= 15) {
+ GByteArray *int_buff = g_byte_array_new();
+ write_int(int_buff, size);
+ g_byte_array_append(bplist, int_buff->data, int_buff->len);
+ g_byte_array_free(int_buff, TRUE);
+ }
+
+ uint64_t idx = 0;
+ uint8_t *buff = (uint8_t *) malloc(size * dict_param_size);
+
+ GNode *cur = NULL;
+ int i = 0;
+ for (i = 0, cur = node->children; cur && i < size; cur = cur->next, i++) {
+ idx = GPOINTER_TO_UINT(g_hash_table_lookup(ref_table, cur));
+ memcpy(buff + i * dict_param_size, &idx, dict_param_size);
+ swap_n_bytes(buff + i * dict_param_size, dict_param_size);
+ }
+
+ //now append to bplist
+ g_byte_array_append(bplist, buff, size * dict_param_size);
+ free(buff);
+
+}
+
+void write_dict(GByteArray * bplist, GNode * node, GHashTable * ref_table, uint8_t dict_param_size)
+{
+ uint64_t size = g_node_n_children(node) / 2;
+ uint8_t marker = BPLIST_ARRAY | (size < 15 ? size : 0xf);
+ g_byte_array_append(bplist, &marker, sizeof(uint8_t));
+ if (size >= 15) {
+ GByteArray *int_buff = g_byte_array_new();
+ write_int(int_buff, size);
+ g_byte_array_append(bplist, int_buff->data, int_buff->len);
+ g_byte_array_free(int_buff, TRUE);
+ }
+
+ uint64_t idx1 = 0;
+ uint64_t idx2 = 0;
+ uint8_t *buff = (uint8_t *) malloc(size * 2 * dict_param_size);
+
+ GNode *cur = NULL;
+ int i = 0;
+ for (i = 0, cur = node->children; cur && i < size; cur = cur->next->next, i++) {
+ idx1 = GPOINTER_TO_UINT(g_hash_table_lookup(ref_table, cur));
+ memcpy(buff + i * dict_param_size, &idx1, dict_param_size);
+ swap_n_bytes(buff + i * dict_param_size, dict_param_size);
+
+ idx2 = GPOINTER_TO_UINT(g_hash_table_lookup(ref_table, cur->next));
+ memcpy(buff + (i + size) * dict_param_size, &idx2, dict_param_size);
+ swap_n_bytes(buff + (i + size) * dict_param_size, dict_param_size);
+ }
+
+ //now append to bplist
+ g_byte_array_append(bplist, buff, size * dict_param_size);
+ free(buff);
+
+}
+
+void plist_to_bin(plist_t plist, char **plist_bin, uint32_t * length)
+{
+ //first serialize tree
+
+ //list of objects
+ GPtrArray *objects = g_ptr_array_new();
+ //hashtable to write only once same nodes
+ GHashTable *ref_table = g_hash_table_new(plist_data_hash, plist_data_compare);
+
+ //serialize plist
+ struct serialize_s ser_s = { objects, ref_table };
+ g_node_children_foreach(plist, G_TRAVERSE_ALL, serialize_plist, &ser_s);
+
+ //now stream to output buffer
+ uint8_t offset_size = 0; //unknown yet
+ uint8_t dict_param_size = get_needed_bytes(objects->len);
+ uint64_t num_objects = objects->len;
+ uint64_t root_object = 0; //root is first in list
+ uint64_t offset_table_index = 0; //unknown yet
+
+ //setup a dynamic bytes array to store bplist in
+ GByteArray *bplist_buff = g_byte_array_new();
+
+ //set magic number and version
+ g_byte_array_append(bplist_buff, BPLIST_MAGIC, BPLIST_MAGIC_SIZE);
+ g_byte_array_append(bplist_buff, BPLIST_VERSION, BPLIST_VERSION_SIZE);
+
+ //write objects and table
+ int i = 0;
+ uint8_t *buff = NULL;
+ uint8_t size = 0;
+ uint64_t offsets[num_objects];
+ for (i = 0; i <= num_objects; i++) {
+
+ offsets[i] = bplist_buff->len;
+ struct plist_data *data = (struct plist_data *) ((GNode *) g_ptr_array_index(objects, i))->data;
+
+ switch (data->type) {
+ case PLIST_BOOLEAN:
+ buff = (uint8_t *) malloc(sizeof(uint8_t));
+ buff[0] = data->boolval ? BPLIST_TRUE : BPLIST_FALSE;
+ g_byte_array_append(bplist_buff, buff, sizeof(uint8_t));
+ free(buff);
+ break;
+
+ case PLIST_UINT:
+ write_int(bplist_buff, data->intval);
+ break;
+
+ case PLIST_REAL:
+ write_real(bplist_buff, data->realval);
+ break;
+
+ case PLIST_KEY:
+ case PLIST_STRING:
+ write_string(bplist_buff, data->strval);
+ break;
+ case PLIST_UNICODE:
+ //TODO
+ break;
+ case PLIST_DATA:
+ write_data(bplist_buff, data->strval, data->length);
+ case PLIST_ARRAY:
+ write_array(bplist_buff, g_ptr_array_index(objects, i), ref_table, dict_param_size);
+ break;
+ case PLIST_DICT:
+ write_dict(bplist_buff, g_ptr_array_index(objects, i), ref_table, dict_param_size);
+ break;
+ case PLIST_DATE:
+ //TODO
+ break;
+ default:
+ break;
+ }
+ }
+
+ //write offsets
+ offset_size = get_needed_bytes(bplist_buff->len);
+ for (i = 0; i <= num_objects; i++) {
+ uint8_t *buff = (uint8_t *) malloc(offset_size);
+ memcpy(buff, offsets + i, offset_size);
+ swap_n_bytes(buff, offset_size);
+ g_byte_array_append(bplist_buff, buff, offset_size);
+ free(buff);
+ }
+
+ //setup trailer
+ num_objects = bswap_64(num_objects);
+ root_object = bswap_64(root_object);
+ offset_table_index = bswap_64(offset_table_index);
+
+ char trailer[BPLIST_TRL_SIZE];
+ memcpy(trailer + BPLIST_TRL_OFFSIZE_IDX, &offset_size, sizeof(uint8_t));
+ memcpy(trailer + BPLIST_TRL_PARMSIZE_IDX, &dict_param_size, sizeof(uint8_t));
+ memcpy(trailer + BPLIST_TRL_NUMOBJ_IDX, &num_objects, sizeof(uint64_t));
+ memcpy(trailer + BPLIST_TRL_ROOTOBJ_IDX, &root_object, sizeof(uint64_t));
+ memcpy(trailer + BPLIST_TRL_OFFTAB_IDX, &offset_table_index, sizeof(uint64_t));
+
+ g_byte_array_append(bplist_buff, trailer, BPLIST_TRL_SIZE);
+
+ //duplicate buffer
+ *plist_bin = (char *) malloc(bplist_buff->len);
+ memcpy(*plist_bin, bplist_buff->data, bplist_buff->len);
+ *length = bplist_buff->len;
+
+ g_byte_array_free(bplist_buff, TRUE);
+}
diff --git a/src/plist.h b/src/plist.h
index df1d3e4..63f67f7 100644
--- a/src/plist.h
+++ b/src/plist.h
@@ -33,28 +33,6 @@
char *format_string(const char *buf, int cols, int depth);
-/* Binary plist stuff */
-
-/*
-typedef enum {
- PLIST_BOOLEAN,
- PLIST_UINT8,
- PLIST_UINT16,
- PLIST_UINT32,
- PLIST_UINT64,
- PLIST_FLOAT32,
- PLIST_FLOAT64,
- PLIST_STRING,
- PLIST_UNICODE,
- PLIST_ARRAY,
- PLIST_DICT,
- PLIST_DATE,
- PLIST_DATA,
- PLIST_PLIST,
- PLIST_KEY,
-} plist_type;
-*/
-
typedef enum {
PLIST_BOOLEAN,
PLIST_UINT,
@@ -65,7 +43,6 @@ typedef enum {
PLIST_DICT,
PLIST_DATE,
PLIST_DATA,
- PLIST_PLIST,
PLIST_KEY,
} plist_type;