/jansson/src/dump.c

http://github.com/nicolasff/webdis · C · 461 lines · 364 code · 80 blank · 17 comment · 66 complexity · 689f39b79ab59041a4ca87b82d3a4e64 MD5 · raw file

  1. /*
  2. * Copyright (c) 2009, 2010 Petri Lehtinen <petri@digip.org>
  3. *
  4. * Jansson is free software; you can redistribute it and/or modify
  5. * it under the terms of the MIT license. See LICENSE for details.
  6. */
  7. #define _GNU_SOURCE
  8. #include <stdio.h>
  9. #include <stdlib.h>
  10. #include <string.h>
  11. #include <assert.h>
  12. #include <jansson.h>
  13. #include "jansson_private.h"
  14. #include "strbuffer.h"
  15. #include "utf.h"
  16. #define MAX_INTEGER_STR_LENGTH 100
  17. #define MAX_REAL_STR_LENGTH 100
  18. typedef int (*dump_func)(const char *buffer, int size, void *data);
  19. struct string
  20. {
  21. char *buffer;
  22. int length;
  23. int size;
  24. };
  25. static int dump_to_strbuffer(const char *buffer, int size, void *data)
  26. {
  27. return strbuffer_append_bytes((strbuffer_t *)data, buffer, size);
  28. }
  29. static int dump_to_file(const char *buffer, int size, void *data)
  30. {
  31. FILE *dest = (FILE *)data;
  32. if(fwrite(buffer, size, 1, dest) != 1)
  33. return -1;
  34. return 0;
  35. }
  36. /* 32 spaces (the maximum indentation size) */
  37. static char whitespace[] = " ";
  38. static int dump_indent(size_t flags, int depth, int space, dump_func dump, void *data)
  39. {
  40. if(JSON_INDENT(flags) > 0)
  41. {
  42. int i, ws_count = JSON_INDENT(flags);
  43. if(dump("\n", 1, data))
  44. return -1;
  45. for(i = 0; i < depth; i++)
  46. {
  47. if(dump(whitespace, ws_count, data))
  48. return -1;
  49. }
  50. }
  51. else if(space && !(flags & JSON_COMPACT))
  52. {
  53. return dump(" ", 1, data);
  54. }
  55. return 0;
  56. }
  57. static int dump_string(const char *str, int ascii, dump_func dump, void *data)
  58. {
  59. const char *pos, *end;
  60. int32_t codepoint;
  61. if(dump("\"", 1, data))
  62. return -1;
  63. end = pos = str;
  64. while(1)
  65. {
  66. const char *text;
  67. char seq[13];
  68. int length;
  69. while(*end)
  70. {
  71. end = utf8_iterate(pos, &codepoint);
  72. if(!end)
  73. return -1;
  74. /* mandatory escape or control char */
  75. if(codepoint == '\\' || codepoint == '"' || codepoint < 0x20)
  76. break;
  77. /* non-ASCII */
  78. if(ascii && codepoint > 0x7F)
  79. break;
  80. pos = end;
  81. }
  82. if(pos != str) {
  83. if(dump(str, pos - str, data))
  84. return -1;
  85. }
  86. if(end == pos)
  87. break;
  88. /* handle \, ", and control codes */
  89. length = 2;
  90. switch(codepoint)
  91. {
  92. case '\\': text = "\\\\"; break;
  93. case '\"': text = "\\\""; break;
  94. case '\b': text = "\\b"; break;
  95. case '\f': text = "\\f"; break;
  96. case '\n': text = "\\n"; break;
  97. case '\r': text = "\\r"; break;
  98. case '\t': text = "\\t"; break;
  99. default:
  100. {
  101. /* codepoint is in BMP */
  102. if(codepoint < 0x10000)
  103. {
  104. sprintf(seq, "\\u%04x", codepoint);
  105. length = 6;
  106. }
  107. /* not in BMP -> construct a UTF-16 surrogate pair */
  108. else
  109. {
  110. int32_t first, last;
  111. codepoint -= 0x10000;
  112. first = 0xD800 | ((codepoint & 0xffc00) >> 10);
  113. last = 0xDC00 | (codepoint & 0x003ff);
  114. sprintf(seq, "\\u%04x\\u%04x", first, last);
  115. length = 12;
  116. }
  117. text = seq;
  118. break;
  119. }
  120. }
  121. if(dump(text, length, data))
  122. return -1;
  123. str = pos = end;
  124. }
  125. return dump("\"", 1, data);
  126. }
  127. static int object_key_compare_keys(const void *key1, const void *key2)
  128. {
  129. return strcmp((*(const object_key_t **)key1)->key,
  130. (*(const object_key_t **)key2)->key);
  131. }
  132. static int object_key_compare_serials(const void *key1, const void *key2)
  133. {
  134. return (*(const object_key_t **)key1)->serial -
  135. (*(const object_key_t **)key2)->serial;
  136. }
  137. static int do_dump(const json_t *json, size_t flags, int depth,
  138. dump_func dump, void *data)
  139. {
  140. int ascii = flags & JSON_ENSURE_ASCII ? 1 : 0;
  141. switch(json_typeof(json)) {
  142. case JSON_NULL:
  143. return dump("null", 4, data);
  144. case JSON_TRUE:
  145. return dump("true", 4, data);
  146. case JSON_FALSE:
  147. return dump("false", 5, data);
  148. case JSON_INTEGER:
  149. {
  150. char buffer[MAX_INTEGER_STR_LENGTH];
  151. int size;
  152. size = snprintf(buffer, MAX_INTEGER_STR_LENGTH,
  153. "%" JSON_INTEGER_FORMAT,
  154. json_integer_value(json));
  155. if(size >= MAX_INTEGER_STR_LENGTH)
  156. return -1;
  157. return dump(buffer, size, data);
  158. }
  159. case JSON_REAL:
  160. {
  161. char buffer[MAX_REAL_STR_LENGTH];
  162. int size;
  163. size = snprintf(buffer, MAX_REAL_STR_LENGTH, "%.17g",
  164. json_real_value(json));
  165. if(size >= MAX_REAL_STR_LENGTH)
  166. return -1;
  167. /* Make sure there's a dot or 'e' in the output. Otherwise
  168. a real is converted to an integer when decoding */
  169. if(strchr(buffer, '.') == NULL &&
  170. strchr(buffer, 'e') == NULL)
  171. {
  172. if(size + 2 >= MAX_REAL_STR_LENGTH) {
  173. /* No space to append ".0" */
  174. return -1;
  175. }
  176. buffer[size] = '.';
  177. buffer[size + 1] = '0';
  178. size += 2;
  179. }
  180. return dump(buffer, size, data);
  181. }
  182. case JSON_STRING:
  183. return dump_string(json_string_value(json), ascii, dump, data);
  184. case JSON_ARRAY:
  185. {
  186. int i;
  187. int n;
  188. json_array_t *array;
  189. /* detect circular references */
  190. array = json_to_array(json);
  191. if(array->visited)
  192. goto array_error;
  193. array->visited = 1;
  194. n = json_array_size(json);
  195. if(dump("[", 1, data))
  196. goto array_error;
  197. if(n == 0) {
  198. array->visited = 0;
  199. return dump("]", 1, data);
  200. }
  201. if(dump_indent(flags, depth + 1, 0, dump, data))
  202. goto array_error;
  203. for(i = 0; i < n; ++i) {
  204. if(do_dump(json_array_get(json, i), flags, depth + 1,
  205. dump, data))
  206. goto array_error;
  207. if(i < n - 1)
  208. {
  209. if(dump(",", 1, data) ||
  210. dump_indent(flags, depth + 1, 1, dump, data))
  211. goto array_error;
  212. }
  213. else
  214. {
  215. if(dump_indent(flags, depth, 0, dump, data))
  216. goto array_error;
  217. }
  218. }
  219. array->visited = 0;
  220. return dump("]", 1, data);
  221. array_error:
  222. array->visited = 0;
  223. return -1;
  224. }
  225. case JSON_OBJECT:
  226. {
  227. json_object_t *object;
  228. void *iter;
  229. const char *separator;
  230. int separator_length;
  231. if(flags & JSON_COMPACT) {
  232. separator = ":";
  233. separator_length = 1;
  234. }
  235. else {
  236. separator = ": ";
  237. separator_length = 2;
  238. }
  239. /* detect circular references */
  240. object = json_to_object(json);
  241. if(object->visited)
  242. goto object_error;
  243. object->visited = 1;
  244. iter = json_object_iter((json_t *)json);
  245. if(dump("{", 1, data))
  246. goto object_error;
  247. if(!iter) {
  248. object->visited = 0;
  249. return dump("}", 1, data);
  250. }
  251. if(dump_indent(flags, depth + 1, 0, dump, data))
  252. goto object_error;
  253. if(flags & JSON_SORT_KEYS || flags & JSON_PRESERVE_ORDER)
  254. {
  255. const object_key_t **keys;
  256. size_t size, i;
  257. int (*cmp_func)(const void *, const void *);
  258. size = json_object_size(json);
  259. keys = malloc(size * sizeof(object_key_t *));
  260. if(!keys)
  261. goto object_error;
  262. i = 0;
  263. while(iter)
  264. {
  265. keys[i] = jsonp_object_iter_fullkey(iter);
  266. iter = json_object_iter_next((json_t *)json, iter);
  267. i++;
  268. }
  269. assert(i == size);
  270. if(flags & JSON_SORT_KEYS)
  271. cmp_func = object_key_compare_keys;
  272. else
  273. cmp_func = object_key_compare_serials;
  274. qsort(keys, size, sizeof(object_key_t *), cmp_func);
  275. for(i = 0; i < size; i++)
  276. {
  277. const char *key;
  278. json_t *value;
  279. key = keys[i]->key;
  280. value = json_object_get(json, key);
  281. assert(value);
  282. dump_string(key, ascii, dump, data);
  283. if(dump(separator, separator_length, data) ||
  284. do_dump(value, flags, depth + 1, dump, data))
  285. {
  286. free(keys);
  287. goto object_error;
  288. }
  289. if(i < size - 1)
  290. {
  291. if(dump(",", 1, data) ||
  292. dump_indent(flags, depth + 1, 1, dump, data))
  293. {
  294. free(keys);
  295. goto object_error;
  296. }
  297. }
  298. else
  299. {
  300. if(dump_indent(flags, depth, 0, dump, data))
  301. {
  302. free(keys);
  303. goto object_error;
  304. }
  305. }
  306. }
  307. free(keys);
  308. }
  309. else
  310. {
  311. /* Don't sort keys */
  312. while(iter)
  313. {
  314. void *next = json_object_iter_next((json_t *)json, iter);
  315. dump_string(json_object_iter_key(iter), ascii, dump, data);
  316. if(dump(separator, separator_length, data) ||
  317. do_dump(json_object_iter_value(iter), flags, depth + 1,
  318. dump, data))
  319. goto object_error;
  320. if(next)
  321. {
  322. if(dump(",", 1, data) ||
  323. dump_indent(flags, depth + 1, 1, dump, data))
  324. goto object_error;
  325. }
  326. else
  327. {
  328. if(dump_indent(flags, depth, 0, dump, data))
  329. goto object_error;
  330. }
  331. iter = next;
  332. }
  333. }
  334. object->visited = 0;
  335. return dump("}", 1, data);
  336. object_error:
  337. object->visited = 0;
  338. return -1;
  339. }
  340. default:
  341. /* not reached */
  342. return -1;
  343. }
  344. }
  345. char *json_dumps(const json_t *json, size_t flags)
  346. {
  347. strbuffer_t strbuff;
  348. char *result;
  349. if(!json_is_array(json) && !json_is_object(json))
  350. return NULL;
  351. if(strbuffer_init(&strbuff))
  352. return NULL;
  353. if(do_dump(json, flags, 0, dump_to_strbuffer, (void *)&strbuff)) {
  354. strbuffer_close(&strbuff);
  355. return NULL;
  356. }
  357. result = strdup(strbuffer_value(&strbuff));
  358. strbuffer_close(&strbuff);
  359. return result;
  360. }
  361. int json_dumpf(const json_t *json, FILE *output, size_t flags)
  362. {
  363. if(!json_is_array(json) && !json_is_object(json))
  364. return -1;
  365. return do_dump(json, flags, 0, dump_to_file, (void *)output);
  366. }
  367. int json_dump_file(const json_t *json, const char *path, size_t flags)
  368. {
  369. int result;
  370. FILE *output = fopen(path, "w");
  371. if(!output)
  372. return -1;
  373. result = json_dumpf(json, output, flags);
  374. fclose(output);
  375. return result;
  376. }