| @@ -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. | |||||