@@ -1301,89 +1301,149 @@ int json_object_equal(struct json_object* jso1, struct json_object* jso2) | |||||
static int json_object_copy_serializer_data(struct json_object *src, struct json_object *dst) | static int json_object_copy_serializer_data(struct json_object *src, struct json_object *dst) | ||||
{ | { | ||||
/* FIXME: this is shared between copies ; maybe add a `_user_copy` cb here */ | |||||
dst->_userdata = src->_userdata; | |||||
dst->_to_json_string = src->_to_json_string; | |||||
if (!src->_userdata && !src->_user_delete) | |||||
return 0; | |||||
if (dst->_to_json_string == json_object_userdata_to_json_string) | |||||
{ | |||||
dst->_userdata = strdup(src->_userdata); | |||||
} | |||||
// else if ... other supported serializers ... | |||||
else | |||||
{ | |||||
_json_c_set_last_err("json_object_deep_copy: unable to copy unknown serializer data: %p\n", dst->_to_json_string); | |||||
return -1; | |||||
} | |||||
dst->_user_delete = src->_user_delete; | dst->_user_delete = src->_user_delete; | ||||
return 0; | return 0; | ||||
} | } | ||||
static int json_object_deep_copy_recursive(struct json_object *src, struct json_object **dst) | |||||
/** | |||||
* The default shallow copy implementation. Simply creates a new object of the same | |||||
* type but does *not* copy over _userdata nor retain any custom serializer. | |||||
* If custom serializers are in use, json_object_deep_copy() must be passed a shallow copy | |||||
* implementation that is aware of how to copy them. | |||||
* | |||||
* This always returns -1 or 1. It will never return 2 since it does not copy the serializer. | |||||
*/ | |||||
int json_c_shallow_copy_default(json_object *src, json_object *parent, const char *key, size_t index, json_object **dst) | |||||
{ | { | ||||
struct json_object *jso = NULL; | |||||
struct json_object_iter iter; | |||||
size_t i; | |||||
switch (src->o_type) { | |||||
case json_type_boolean: | |||||
*dst = json_object_new_boolean(src->o.c_boolean); | |||||
break; | |||||
if (!src || !dst) { | |||||
case json_type_double: | |||||
*dst = json_object_new_double(src->o.c_double); | |||||
break; | |||||
case json_type_int: | |||||
*dst = json_object_new_int64(src->o.c_int64); | |||||
break; | |||||
case json_type_string: | |||||
*dst = json_object_new_string(get_string_component(src)); | |||||
break; | |||||
case json_type_object: | |||||
*dst = json_object_new_object(); | |||||
break; | |||||
case json_type_array: | |||||
*dst = json_object_new_array(); | |||||
break; | |||||
default: | |||||
errno = EINVAL; | errno = EINVAL; | ||||
return -1; | return -1; | ||||
} | } | ||||
switch (src->o_type) { | |||||
case json_type_boolean: | |||||
*dst = json_object_new_boolean(src->o.c_boolean); | |||||
break; | |||||
case json_type_double: | |||||
*dst = json_object_new_double(src->o.c_double); | |||||
break; | |||||
if (!*dst) { | |||||
errno = ENOMEM; | |||||
return -1; | |||||
} | |||||
(*dst)->_to_json_string = src->_to_json_string; | |||||
// _userdata and _user_delete are copied later | |||||
return 1; | |||||
} | |||||
case json_type_int: | |||||
*dst = json_object_new_int64(src->o.c_int64); | |||||
break; | |||||
/* | |||||
* The actual guts of json_object_deep_copy(), with a few additional args | |||||
* needed so we can keep track of where we are within the object tree. | |||||
* | |||||
* Note: caller is responsible for freeing *dst if this fails and returns -1. | |||||
*/ | |||||
static int json_object_deep_copy_recursive(struct json_object *src, struct json_object *parent, const char *key_in_parent, size_t index_in_parent, struct json_object **dst, json_c_shallow_copy_fn shallow_copy) | |||||
{ | |||||
struct json_object_iter iter; | |||||
size_t src_array_len, ii; | |||||
case json_type_string: | |||||
*dst = json_object_new_string(get_string_component(src)); | |||||
break; | |||||
int shallow_copy_rc = 0; | |||||
shallow_copy_rc = shallow_copy(src, parent, key_in_parent, index_in_parent, dst); | |||||
/* -1=error, 1=object created ok, 2=userdata set */ | |||||
if (shallow_copy_rc < 1) | |||||
{ | |||||
errno = EINVAL; | |||||
return -1; | |||||
} | |||||
assert(*dst != NULL); | |||||
case json_type_object: | |||||
*dst = json_object_new_object(); | |||||
if (!*dst) { | |||||
errno = ENOMEM; | |||||
switch (src->o_type) { | |||||
case json_type_object: | |||||
json_object_object_foreachC(src, iter) { | |||||
struct json_object *jso = NULL; | |||||
/* This handles the `json_type_null` case */ | |||||
if (!iter.val) | |||||
jso = NULL; | |||||
else if (json_object_deep_copy_recursive(iter.val, src, iter.key, -1, &jso, shallow_copy) < 0) | |||||
{ | |||||
json_object_put(jso); | |||||
return -1; | return -1; | ||||
} | } | ||||
json_object_object_foreachC(src, iter) { | |||||
/* This handles the `json_type_null` case */ | |||||
if (!iter.val) | |||||
jso = NULL; | |||||
else if (json_object_deep_copy_recursive(iter.val, &jso) < 0) | |||||
return -1; | |||||
if (json_object_object_add(*dst, iter.key, jso) < 0) | |||||
return -1; | |||||
if (json_object_object_add(*dst, iter.key, jso) < 0) | |||||
{ | |||||
json_object_put(jso); | |||||
return -1; | |||||
} | } | ||||
break; | |||||
} | |||||
break; | |||||
case json_type_array: | |||||
*dst = json_object_new_array(); | |||||
if (!*dst) { | |||||
errno = ENOMEM; | |||||
case json_type_array: | |||||
src_array_len = json_object_array_length(src); | |||||
for (ii = 0; ii < src_array_len; ii++) { | |||||
struct json_object *jso = NULL; | |||||
struct json_object *jso1 = json_object_array_get_idx(src, ii); | |||||
/* This handles the `json_type_null` case */ | |||||
if (!jso1) | |||||
jso = NULL; | |||||
else if (json_object_deep_copy_recursive(jso1, src, NULL, ii, &jso, shallow_copy) < 0) | |||||
{ | |||||
json_object_put(jso); | |||||
return -1; | return -1; | ||||
} | } | ||||
for (i = 0; i < json_object_array_length(src); i++) { | |||||
struct json_object *jso1 = json_object_array_get_idx(src, i); | |||||
/* This handles the `json_type_null` case */ | |||||
if (!jso1) | |||||
jso = NULL; | |||||
else if (json_object_deep_copy_recursive(jso1, &jso) < 0) | |||||
return -1; | |||||
if (json_object_array_add(*dst, jso) < 0) | |||||
return -1; | |||||
if (json_object_array_add(*dst, jso) < 0) | |||||
{ | |||||
json_object_put(jso); | |||||
return -1; | |||||
} | } | ||||
break; | |||||
} | |||||
break; | |||||
default: | |||||
errno = EINVAL; | |||||
return -1; | |||||
}; | |||||
default: | |||||
break; | |||||
/* else, nothing to do, shallow_copy already did. */ | |||||
} | |||||
/* errno should be set by the function that faulted */ | |||||
if (!*dst) | |||||
return -1; | |||||
if (shallow_copy_rc != 2) | |||||
return json_object_copy_serializer_data(src, *dst); | |||||
return json_object_copy_serializer_data(src, *dst); | |||||
return 0; | |||||
} | } | ||||
int json_object_deep_copy(struct json_object *src, struct json_object **dst) | |||||
int json_object_deep_copy(struct json_object *src, struct json_object **dst, json_c_shallow_copy_fn shallow_copy) | |||||
{ | { | ||||
int rc; | int rc; | ||||
@@ -1393,7 +1453,10 @@ int json_object_deep_copy(struct json_object *src, struct json_object **dst) | |||||
return -1; | return -1; | ||||
} | } | ||||
rc = json_object_deep_copy_recursive(src, dst); | |||||
if (shallow_copy == NULL) | |||||
shallow_copy = json_c_shallow_copy_default; | |||||
rc = json_object_deep_copy_recursive(src, NULL, NULL, -1, dst, shallow_copy); | |||||
if (rc < 0) { | if (rc < 0) { | ||||
json_object_put(*dst); | json_object_put(*dst); | ||||
*dst = NULL; | *dst = NULL; | ||||
@@ -963,6 +963,36 @@ JSON_EXPORT int json_object_set_string_len(json_object* obj, const char* new_val | |||||
JSON_EXPORT int json_object_equal(struct json_object *obj1, | JSON_EXPORT int json_object_equal(struct json_object *obj1, | ||||
struct json_object *obj2); | struct json_object *obj2); | ||||
/** | |||||
* Perform a shallow copy of src into *dst as part of an overall json_object_deep_copy(). | |||||
* | |||||
* If src is part of a containing object or array, parent will be non-NULL, | |||||
* and key or index will be provided. | |||||
* When shallow_copy is called *dst will be NULL, and must be non-NULL when it returns. | |||||
* src will never be NULL. | |||||
* | |||||
* If shallow_copy sets the serializer on an object, return 2 to indicate to | |||||
* json_object_deep_copy that it should not attempt to use the standard userdata | |||||
* copy function. | |||||
* | |||||
* @return On success 1 or 2, -1 on errors | |||||
*/ | |||||
typedef int (json_c_shallow_copy_fn)(json_object *src, json_object *parent, const char *key, size_t index, json_object **dst); | |||||
/** | |||||
* The default shallow copy implementation for use with json_object_deep_copy(). | |||||
* This simply calls the appropriate json_object_new_<type>() function and | |||||
* copies over the serializer function (_to_json_string internal field of | |||||
* the json_object structure) but not any _userdata or _user_delete values. | |||||
* | |||||
* If you're writing a custom shallow_copy function, perhaps because you're using | |||||
* your own custom serializer, you can call this first to create the new object | |||||
* before customizing it with json_object_set_serializer(). | |||||
* | |||||
* @return 1 on success, -1 on errors, but never 2. | |||||
*/ | |||||
json_c_shallow_copy_fn json_c_shallow_copy_default; | |||||
/** | /** | ||||
* Copy the contents of the JSON object. | * Copy the contents of the JSON object. | ||||
* The destination object must be initialized to NULL, | * The destination object must be initialized to NULL, | ||||
@@ -979,8 +1009,7 @@ JSON_EXPORT int json_object_equal(struct json_object *obj1, | |||||
* or if the destination pointer is non-NULL | * or if the destination pointer is non-NULL | ||||
*/ | */ | ||||
extern int json_object_deep_copy(struct json_object *src, struct json_object **dst); | |||||
JSON_EXPORT int json_object_deep_copy(struct json_object *src, struct json_object **dst, json_c_shallow_copy_fn shallow_copy); | |||||
#ifdef __cplusplus | #ifdef __cplusplus | ||||
} | } | ||||
#endif | #endif | ||||
@@ -7,6 +7,7 @@ | |||||
#include <time.h> | #include <time.h> | ||||
#include "json.h" | #include "json.h" | ||||
#include "printbuf.h" | |||||
static void do_benchmark(json_object *src1); | static void do_benchmark(json_object *src1); | ||||
@@ -77,6 +78,31 @@ static const char *json_str3 = | |||||
" }" | " }" | ||||
"}}"; | "}}"; | ||||
json_object_to_json_string_fn my_custom_serializer; | |||||
int my_custom_serializer(struct json_object *jso, struct printbuf *pb, int level, int flags) | |||||
{ | |||||
sprintbuf(pb, "OTHER"); | |||||
return 0; | |||||
} | |||||
json_c_shallow_copy_fn my_shallow_copy; | |||||
int my_shallow_copy(json_object *src, json_object *parent, const char *key, size_t index, json_object **dst) | |||||
{ | |||||
int rc; | |||||
rc = json_c_shallow_copy_default(src, parent, key, index, dst); | |||||
if (rc < 0) | |||||
return rc; | |||||
if (key != NULL && strcmp(key, "with_serializer") == 0) | |||||
{ | |||||
printf("CALLED: my_shallow_copy on with_serializer object\n"); | |||||
void *userdata = json_object_get_userdata(src); | |||||
json_object_set_serializer(*dst, my_custom_serializer, userdata, NULL); | |||||
return 2; | |||||
} | |||||
return rc; | |||||
} | |||||
int main(int argc, char **argv) | int main(int argc, char **argv) | ||||
{ | { | ||||
struct json_object *src1, *src2, *src3; | struct json_object *src1, *src2, *src3; | ||||
@@ -99,17 +125,17 @@ int main(int argc, char **argv) | |||||
printf("PASSED - loaded input data\n"); | printf("PASSED - loaded input data\n"); | ||||
/* do this 3 times to make sure overwriting it works */ | /* do this 3 times to make sure overwriting it works */ | ||||
assert(0 == json_object_deep_copy(src1, &dst1)); | |||||
assert(0 == json_object_deep_copy(src2, &dst2)); | |||||
assert(0 == json_object_deep_copy(src3, &dst3)); | |||||
assert(0 == json_object_deep_copy(src1, &dst1, NULL)); | |||||
assert(0 == json_object_deep_copy(src2, &dst2, NULL)); | |||||
assert(0 == json_object_deep_copy(src3, &dst3, NULL)); | |||||
printf("PASSED - all json_object_deep_copy() returned succesful\n"); | printf("PASSED - all json_object_deep_copy() returned succesful\n"); | ||||
assert(-1 == json_object_deep_copy(src1, &dst1)); | |||||
assert(-1 == json_object_deep_copy(src1, &dst1, NULL)); | |||||
assert(errno == EINVAL); | assert(errno == EINVAL); | ||||
assert(-1 == json_object_deep_copy(src2, &dst2)); | |||||
assert(-1 == json_object_deep_copy(src2, &dst2, NULL)); | |||||
assert(errno == EINVAL); | assert(errno == EINVAL); | ||||
assert(-1 == json_object_deep_copy(src3, &dst3)); | |||||
assert(-1 == json_object_deep_copy(src3, &dst3, NULL)); | |||||
assert(errno == EINVAL); | assert(errno == EINVAL); | ||||
printf("PASSED - all json_object_deep_copy() returned EINVAL for non-null pointer\n"); | printf("PASSED - all json_object_deep_copy() returned EINVAL for non-null pointer\n"); | ||||
@@ -130,7 +156,7 @@ int main(int argc, char **argv) | |||||
printf("PASSED - comparison of string output\n"); | printf("PASSED - comparison of string output\n"); | ||||
json_object_get(dst1); | json_object_get(dst1); | ||||
assert(-1 == json_object_deep_copy(src1, &dst1)); | |||||
assert(-1 == json_object_deep_copy(src1, &dst1, NULL)); | |||||
assert(errno == EINVAL); | assert(errno == EINVAL); | ||||
json_object_put(dst1); | json_object_put(dst1); | ||||
@@ -139,24 +165,44 @@ int main(int argc, char **argv) | |||||
printf("\nPrinting JSON objects for visual inspection\n"); | printf("\nPrinting JSON objects for visual inspection\n"); | ||||
printf("------------------------------------------------\n"); | printf("------------------------------------------------\n"); | ||||
printf(" JSON1\n"); | printf(" JSON1\n"); | ||||
printf("%s", json_object_to_json_string_ext(dst1, JSON_C_TO_STRING_PRETTY)); | |||||
printf("%s\n", json_object_to_json_string_ext(dst1, JSON_C_TO_STRING_PRETTY)); | |||||
printf("------------------------------------------------\n"); | printf("------------------------------------------------\n"); | ||||
printf("------------------------------------------------\n"); | printf("------------------------------------------------\n"); | ||||
printf(" JSON2\n"); | printf(" JSON2\n"); | ||||
printf("%s", json_object_to_json_string_ext(dst2, JSON_C_TO_STRING_PRETTY)); | |||||
printf("%s\n", json_object_to_json_string_ext(dst2, JSON_C_TO_STRING_PRETTY)); | |||||
printf("------------------------------------------------\n"); | printf("------------------------------------------------\n"); | ||||
printf("------------------------------------------------\n"); | printf("------------------------------------------------\n"); | ||||
printf(" JSON3\n"); | printf(" JSON3\n"); | ||||
printf("------------------------------------------------\n"); | printf("------------------------------------------------\n"); | ||||
printf("%s", json_object_to_json_string_ext(dst3, JSON_C_TO_STRING_PRETTY)); | |||||
printf("%s\n", json_object_to_json_string_ext(dst3, JSON_C_TO_STRING_PRETTY)); | |||||
printf("------------------------------------------------\n"); | printf("------------------------------------------------\n"); | ||||
json_object_put(dst1); | json_object_put(dst1); | ||||
json_object_put(dst2); | json_object_put(dst2); | ||||
json_object_put(dst3); | json_object_put(dst3); | ||||
printf("\nTesting deep_copy with a custom serializer set\n"); | |||||
json_object *with_serializer = json_object_new_string("notemitted"); | |||||
json_object_set_serializer(with_serializer, my_custom_serializer, "dummy userdata", NULL); | |||||
json_object_object_add(src1, "with_serializer", with_serializer); | |||||
dst1 = NULL; | |||||
/* With a custom serializer in use, a custom shallow_copy function must also be used */ | |||||
assert(-1 == json_object_deep_copy(src1, &dst1, NULL)); | |||||
assert(0 == json_object_deep_copy(src1, &dst1, my_shallow_copy)); | |||||
json_object *dest_with_serializer = json_object_object_get(dst1, "with_serializer"); | |||||
assert(dest_with_serializer != NULL); | |||||
char *dst_userdata = json_object_get_userdata(dest_with_serializer); | |||||
assert(strcmp(dst_userdata, "dummy userdata") == 0); | |||||
const char *special_output = json_object_to_json_string(dest_with_serializer); | |||||
assert(strcmp(special_output, "OTHER") == 0); | |||||
printf("\ndeep_copy with custom serializer worked OK.\n"); | |||||
json_object_put(dst1); | |||||
if (benchmark) | if (benchmark) | ||||
{ | { | ||||
do_benchmark(src2); | do_benchmark(src2); | ||||
@@ -177,7 +223,7 @@ static void do_benchmark(json_object *src2) | |||||
/** | /** | ||||
* The numbers that I got are: | * The numbers that I got are: | ||||
* BENCHMARK - 1000000 iterations of 'dst2 = json_tokener_parse(json_object_get_string(src2))' took 71 seconds | * BENCHMARK - 1000000 iterations of 'dst2 = json_tokener_parse(json_object_get_string(src2))' took 71 seconds | ||||
* BENCHMARK - 1000000 iterations of 'json_object_deep_copy(src2, &dst2)' took 29 seconds | |||||
* BENCHMARK - 1000000 iterations of 'json_object_deep_copy(src2, &dst2, NULL)' took 29 seconds | |||||
*/ | */ | ||||
int iterations = 1000000; | int iterations = 1000000; | ||||
@@ -193,10 +239,10 @@ static void do_benchmark(json_object *src2) | |||||
start = time(NULL); | start = time(NULL); | ||||
dst2 = NULL; | dst2 = NULL; | ||||
for (ii = 0; ii < iterations; ii++) { | for (ii = 0; ii < iterations; ii++) { | ||||
json_object_deep_copy(src2, &dst2); | |||||
json_object_deep_copy(src2, &dst2, NULL); | |||||
json_object_put(dst2); | json_object_put(dst2); | ||||
dst2 = NULL; | dst2 = NULL; | ||||
} | } | ||||
printf("BENCHMARK - %d iterations of 'json_object_deep_copy(src2, &dst2)' took %d seconds\n", iterations, (int)(time(NULL) - start)); | |||||
printf("BENCHMARK - %d iterations of 'json_object_deep_copy(src2, &dst2, NULL)' took %d seconds\n", iterations, (int)(time(NULL) - start)); | |||||
} | } | ||||
@@ -32,7 +32,8 @@ Printing JSON objects for visual inspection | |||||
} | } | ||||
} | } | ||||
} | } | ||||
}------------------------------------------------ | |||||
} | |||||
------------------------------------------------ | |||||
------------------------------------------------ | ------------------------------------------------ | ||||
JSON2 | JSON2 | ||||
{ | { | ||||
@@ -112,7 +113,8 @@ Printing JSON objects for visual inspection | |||||
} | } | ||||
] | ] | ||||
} | } | ||||
}------------------------------------------------ | |||||
} | |||||
------------------------------------------------ | |||||
------------------------------------------------ | ------------------------------------------------ | ||||
JSON3 | JSON3 | ||||
------------------------------------------------ | ------------------------------------------------ | ||||
@@ -137,4 +139,10 @@ Printing JSON objects for visual inspection | |||||
] | ] | ||||
} | } | ||||
} | } | ||||
}------------------------------------------------ | |||||
} | |||||
------------------------------------------------ | |||||
Testing deep_copy with a custom serializer set | |||||
CALLED: my_shallow_copy on with_serializer object | |||||
deep_copy with custom serializer worked OK. |