| @@ -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) | |||
| { | |||
| /* 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; | |||
| 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; | |||
| 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; | |||
| } | |||
| 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; | |||
| } | |||
| 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; | |||
| @@ -1393,7 +1453,10 @@ int json_object_deep_copy(struct json_object *src, struct json_object **dst) | |||
| 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) { | |||
| json_object_put(*dst); | |||
| *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, | |||
| 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. | |||
| * 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 | |||
| */ | |||
| 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 | |||
| } | |||
| #endif | |||
| @@ -7,6 +7,7 @@ | |||
| #include <time.h> | |||
| #include "json.h" | |||
| #include "printbuf.h" | |||
| 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) | |||
| { | |||
| struct json_object *src1, *src2, *src3; | |||
| @@ -99,17 +125,17 @@ int main(int argc, char **argv) | |||
| printf("PASSED - loaded input data\n"); | |||
| /* 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"); | |||
| assert(-1 == json_object_deep_copy(src1, &dst1)); | |||
| assert(-1 == json_object_deep_copy(src1, &dst1, NULL)); | |||
| assert(errno == EINVAL); | |||
| assert(-1 == json_object_deep_copy(src2, &dst2)); | |||
| assert(-1 == json_object_deep_copy(src2, &dst2, NULL)); | |||
| assert(errno == EINVAL); | |||
| assert(-1 == json_object_deep_copy(src3, &dst3)); | |||
| assert(-1 == json_object_deep_copy(src3, &dst3, NULL)); | |||
| assert(errno == EINVAL); | |||
| 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"); | |||
| json_object_get(dst1); | |||
| assert(-1 == json_object_deep_copy(src1, &dst1)); | |||
| assert(-1 == json_object_deep_copy(src1, &dst1, NULL)); | |||
| assert(errno == EINVAL); | |||
| json_object_put(dst1); | |||
| @@ -139,24 +165,44 @@ int main(int argc, char **argv) | |||
| printf("\nPrinting JSON objects for visual inspection\n"); | |||
| printf("------------------------------------------------\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(" 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(" JSON3\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"); | |||
| json_object_put(dst1); | |||
| json_object_put(dst2); | |||
| 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) | |||
| { | |||
| do_benchmark(src2); | |||
| @@ -177,7 +223,7 @@ static void do_benchmark(json_object *src2) | |||
| /** | |||
| * 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 '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; | |||
| @@ -193,10 +239,10 @@ static void do_benchmark(json_object *src2) | |||
| start = time(NULL); | |||
| dst2 = NULL; | |||
| for (ii = 0; ii < iterations; ii++) { | |||
| json_object_deep_copy(src2, &dst2); | |||
| json_object_deep_copy(src2, &dst2, NULL); | |||
| json_object_put(dst2); | |||
| 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 | |||
| { | |||
| @@ -112,7 +113,8 @@ Printing JSON objects for visual inspection | |||
| } | |||
| ] | |||
| } | |||
| }------------------------------------------------ | |||
| } | |||
| ------------------------------------------------ | |||
| ------------------------------------------------ | |||
| 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. | |||