avl: add AVL_TREE macro to define an initialized struct avl_tree
[project/libubox.git] / blobmsg_json.c
1 /*
2 * Copyright (C) 2010-2012 Felix Fietkau <nbd@openwrt.org>
3 *
4 * Permission to use, copy, modify, and/or distribute this software for any
5 * purpose with or without fee is hereby granted, provided that the above
6 * copyright notice and this permission notice appear in all copies.
7 *
8 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 */
16 #include "blobmsg.h"
17 #include "blobmsg_json.h"
18
19 bool blobmsg_add_object(struct blob_buf *b, json_object *obj)
20 {
21 json_object_object_foreach(obj, key, val) {
22 if (!blobmsg_add_json_element(b, key, val))
23 return false;
24 }
25 return true;
26 }
27
28 static bool blobmsg_add_array(struct blob_buf *b, struct array_list *a)
29 {
30 int i, len;
31
32 for (i = 0, len = array_list_length(a); i < len; i++) {
33 if (!blobmsg_add_json_element(b, NULL, array_list_get_idx(a, i)))
34 return false;
35 }
36
37 return true;
38 }
39
40 bool blobmsg_add_json_element(struct blob_buf *b, const char *name, json_object *obj)
41 {
42 bool ret = true;
43 void *c;
44
45 if (!obj)
46 return false;
47
48 switch (json_object_get_type(obj)) {
49 case json_type_object:
50 c = blobmsg_open_table(b, name);
51 ret = blobmsg_add_object(b, obj);
52 blobmsg_close_table(b, c);
53 break;
54 case json_type_array:
55 c = blobmsg_open_array(b, name);
56 ret = blobmsg_add_array(b, json_object_get_array(obj));
57 blobmsg_close_array(b, c);
58 break;
59 case json_type_string:
60 blobmsg_add_string(b, name, json_object_get_string(obj));
61 break;
62 case json_type_boolean:
63 blobmsg_add_u8(b, name, json_object_get_boolean(obj));
64 break;
65 case json_type_int:
66 blobmsg_add_u32(b, name, json_object_get_int(obj));
67 break;
68 default:
69 return false;
70 }
71 return ret;
72 }
73
74 static bool __blobmsg_add_json(struct blob_buf *b, json_object *obj)
75 {
76 bool ret = false;
77
78 if (is_error(obj))
79 return false;
80
81 if (json_object_get_type(obj) != json_type_object)
82 goto out;
83
84 ret = blobmsg_add_object(b, obj);
85
86 out:
87 json_object_put(obj);
88 return ret;
89 }
90
91 bool blobmsg_add_json_from_file(struct blob_buf *b, const char *file)
92 {
93 return __blobmsg_add_json(b, json_object_from_file(file));
94 }
95
96 bool blobmsg_add_json_from_string(struct blob_buf *b, const char *str)
97 {
98 return __blobmsg_add_json(b, json_tokener_parse(str));
99 }
100
101
102 struct strbuf {
103 int len;
104 int pos;
105 char *buf;
106
107 blobmsg_json_format_t custom_format;
108 void *priv;
109 bool indent;
110 int indent_level;
111 };
112
113 static bool blobmsg_puts(struct strbuf *s, const char *c, int len)
114 {
115 if (len <= 0)
116 return true;
117
118 if (s->pos + len >= s->len) {
119 s->len += 16 + len;
120 s->buf = realloc(s->buf, s->len);
121 if (!s->buf)
122 return false;
123 }
124 memcpy(s->buf + s->pos, c, len);
125 s->pos += len;
126 return true;
127 }
128
129 static void add_separator(struct strbuf *s)
130 {
131 static char indent_chars[17] = "\n\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t";
132 int indent;
133 char *start;
134
135 if (!s->indent)
136 return;
137
138 indent = s->indent_level;
139 if (indent > 16)
140 indent = 16;
141
142 start = &indent_chars[sizeof(indent_chars) - indent - 1];
143 *start = '\n';
144 blobmsg_puts(s, start, indent + 1);
145 *start = '\t';
146 }
147
148
149 static void blobmsg_format_string(struct strbuf *s, const char *str)
150 {
151 const unsigned char *p, *last, *end;
152 char buf[8] = "\\u00";
153
154 end = (unsigned char *) str + strlen(str);
155 blobmsg_puts(s, "\"", 1);
156 for (p = (unsigned char *) str, last = p; *p; p++) {
157 char escape = '\0';
158 int len;
159
160 switch(*p) {
161 case '\b':
162 escape = 'b';
163 break;
164 case '\n':
165 escape = 'n';
166 break;
167 case '\t':
168 escape = 't';
169 break;
170 case '\r':
171 escape = 'r';
172 break;
173 case '"':
174 case '\\':
175 case '/':
176 escape = *p;
177 break;
178 default:
179 if (*p < ' ')
180 escape = 'u';
181 break;
182 }
183
184 if (!escape)
185 continue;
186
187 if (p > last)
188 blobmsg_puts(s, (char *) last, p - last);
189 last = p + 1;
190 buf[1] = escape;
191
192 if (escape == 'u') {
193 sprintf(buf + 4, "%02x", (unsigned char) *p);
194 len = 6;
195 } else {
196 len = 2;
197 }
198 blobmsg_puts(s, buf, len);
199 }
200
201 blobmsg_puts(s, (char *) last, end - last);
202 blobmsg_puts(s, "\"", 1);
203 }
204
205 static void blobmsg_format_json_list(struct strbuf *s, struct blob_attr *attr, int len, bool array);
206
207 static void blobmsg_format_element(struct strbuf *s, struct blob_attr *attr, bool array, bool head)
208 {
209 const char *data_str;
210 char buf[32];
211 void *data;
212 int len;
213
214 if (!blobmsg_check_attr(attr, false))
215 return;
216
217 if (!array && blobmsg_name(attr)[0]) {
218 blobmsg_format_string(s, blobmsg_name(attr));
219 blobmsg_puts(s, ": ", s->indent ? 2 : 1);
220 }
221
222 data = blobmsg_data(attr);
223 len = blobmsg_data_len(attr);
224
225 if (!head && s->custom_format) {
226 data_str = s->custom_format(s->priv, attr);
227 if (data_str)
228 goto out;
229 }
230
231 data_str = buf;
232 switch(blob_id(attr)) {
233 case BLOBMSG_TYPE_UNSPEC:
234 sprintf(buf, "null");
235 break;
236 case BLOBMSG_TYPE_BOOL:
237 sprintf(buf, "%s", *(uint8_t *)data ? "true" : "false");
238 break;
239 case BLOBMSG_TYPE_INT16:
240 sprintf(buf, "%d", be16_to_cpu(*(uint16_t *)data));
241 break;
242 case BLOBMSG_TYPE_INT32:
243 sprintf(buf, "%d", (int32_t) be32_to_cpu(*(uint32_t *)data));
244 break;
245 case BLOBMSG_TYPE_INT64:
246 sprintf(buf, "%lld", (long long int) be64_to_cpu(*(uint64_t *)data));
247 break;
248 case BLOBMSG_TYPE_STRING:
249 blobmsg_format_string(s, data);
250 return;
251 case BLOBMSG_TYPE_ARRAY:
252 blobmsg_format_json_list(s, data, len, true);
253 return;
254 case BLOBMSG_TYPE_TABLE:
255 blobmsg_format_json_list(s, data, len, false);
256 return;
257 }
258
259 out:
260 blobmsg_puts(s, data_str, strlen(data_str));
261 }
262
263 static void blobmsg_format_json_list(struct strbuf *s, struct blob_attr *attr, int len, bool array)
264 {
265 struct blob_attr *pos;
266 bool first = true;
267 int rem = len;
268
269 blobmsg_puts(s, (array ? "[" : "{" ), 1);
270 s->indent_level++;
271 add_separator(s);
272 __blob_for_each_attr(pos, attr, rem) {
273 if (!first) {
274 blobmsg_puts(s, ",", 1);
275 add_separator(s);
276 }
277
278 blobmsg_format_element(s, pos, array, false);
279 first = false;
280 }
281 s->indent_level--;
282 add_separator(s);
283 blobmsg_puts(s, (array ? "]" : "}"), 1);
284 }
285
286 char *blobmsg_format_json_with_cb(struct blob_attr *attr, bool list, blobmsg_json_format_t cb, void *priv, int indent)
287 {
288 struct strbuf s;
289
290 s.len = blob_len(attr);
291 s.buf = malloc(s.len);
292 s.pos = 0;
293 s.custom_format = cb;
294 s.priv = priv;
295 s.indent = false;
296
297 if (indent >= 0) {
298 s.indent = true;
299 s.indent_level = indent;
300 }
301
302 if (list)
303 blobmsg_format_json_list(&s, blobmsg_data(attr), blobmsg_data_len(attr), false);
304 else
305 blobmsg_format_element(&s, attr, false, false);
306
307 if (!s.len)
308 return NULL;
309
310 s.buf = realloc(s.buf, s.pos + 1);
311 s.buf[s.pos] = 0;
312
313 return s.buf;
314 }