uloop: Fix race condition in SIGCHLD handling
[project/libubox.git] / jshn.c
1 /*
2  * Copyright (C) 2011-2013 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 #ifdef JSONC
17         #include <json.h>
18 #else
19         #include <json/json.h>
20 #endif
21
22 #include <string.h>
23 #include <stdlib.h>
24 #include <stdio.h>
25 #include <stdbool.h>
26 #include <ctype.h>
27 #include <getopt.h>
28 #include "list.h"
29
30 #include "avl.h"
31 #include "blob.h"
32 #include "blobmsg_json.h"
33
34 #define MAX_VARLEN      256
35
36 static struct avl_tree env_vars;
37 static struct blob_buf b = { 0 };
38
39 static const char *var_prefix = "";
40 static int var_prefix_len = 0;
41
42 static int add_json_element(const char *key, json_object *obj);
43
44 struct env_var {
45         struct avl_node avl;
46         char *val;
47 };
48
49 static int add_json_object(json_object *obj)
50 {
51         int ret = 0;
52
53         json_object_object_foreach(obj, key, val) {
54                 ret = add_json_element(key, val);
55                 if (ret)
56                         break;
57         }
58         return ret;
59 }
60
61 static int add_json_array(struct array_list *a)
62 {
63         char seq[12];
64         int i, len;
65         int ret;
66
67         for (i = 0, len = array_list_length(a); i < len; i++) {
68                 sprintf(seq, "%d", i);
69                 ret = add_json_element(seq, array_list_get_idx(a, i));
70                 if (ret)
71                         return ret;
72         }
73
74         return 0;
75 }
76
77 static void add_json_string(const char *str)
78 {
79         char *ptr = (char *) str;
80         int len;
81         char *c;
82
83         while ((c = strchr(ptr, '\'')) != NULL) {
84                 len = c - ptr;
85                 if (len > 0)
86                         fwrite(ptr, len, 1, stdout);
87                 ptr = c + 1;
88                 c = "'\\''";
89                 fwrite(c, strlen(c), 1, stdout);
90         }
91         len = strlen(ptr);
92         if (len > 0)
93                 fwrite(ptr, len, 1, stdout);
94 }
95
96 static void write_key_string(const char *key)
97 {
98         while (*key) {
99                 putc(isalnum(*key) ? *key : '_', stdout);
100                 key++;
101         }
102 }
103
104 static int add_json_element(const char *key, json_object *obj)
105 {
106         char *type;
107
108         if (!obj)
109                 return -1;
110
111         switch (json_object_get_type(obj)) {
112         case json_type_object:
113                 type = "object";
114                 break;
115         case json_type_array:
116                 type = "array";
117                 break;
118         case json_type_string:
119                 type = "string";
120                 break;
121         case json_type_boolean:
122                 type = "boolean";
123                 break;
124         case json_type_int:
125                 type = "int";
126                 break;
127         case json_type_double:
128                 type = "double";
129                 break;
130         default:
131                 return -1;
132         }
133
134         fprintf(stdout, "json_add_%s '", type);
135         write_key_string(key);
136
137         switch (json_object_get_type(obj)) {
138         case json_type_object:
139                 fprintf(stdout, "';\n");
140                 add_json_object(obj);
141                 fprintf(stdout, "json_close_object;\n");
142                 break;
143         case json_type_array:
144                 fprintf(stdout, "';\n");
145                 add_json_array(json_object_get_array(obj));
146                 fprintf(stdout, "json_close_array;\n");
147                 break;
148         case json_type_string:
149                 fprintf(stdout, "' '");
150                 add_json_string(json_object_get_string(obj));
151                 fprintf(stdout, "';\n");
152                 break;
153         case json_type_boolean:
154                 fprintf(stdout, "' %d;\n", json_object_get_boolean(obj));
155                 break;
156         case json_type_int:
157                 fprintf(stdout, "' %d;\n", json_object_get_int(obj));
158                 break;
159         case json_type_double:
160                 fprintf(stdout, "' %lf;\n", json_object_get_double(obj));
161                 break;
162         default:
163                 return -1;
164         }
165
166         return 0;
167 }
168
169 static int jshn_parse(const char *str)
170 {
171         json_object *obj;
172
173         obj = json_tokener_parse(str);
174         if (!obj || json_object_get_type(obj) != json_type_object) {
175                 fprintf(stderr, "Failed to parse message data\n");
176                 return 1;
177         }
178         fprintf(stdout, "json_init;\n");
179         add_json_object(obj);
180         fflush(stdout);
181
182         return 0;
183 }
184
185 static char *getenv_avl(const char *key)
186 {
187         struct env_var *var = avl_find_element(&env_vars, key, var, avl);
188         return var ? var->val : NULL;
189 }
190
191 static char *get_keys(const char *prefix)
192 {
193         char *keys;
194
195         keys = alloca(var_prefix_len + strlen(prefix) + sizeof("K_") + 1);
196         sprintf(keys, "%sK_%s", var_prefix, prefix);
197         return getenv_avl(keys);
198 }
199
200 static void get_var(const char *prefix, const char **name, char **var, char **type)
201 {
202         char *tmpname, *varname;
203
204         tmpname = alloca(var_prefix_len + strlen(prefix) + 1 + strlen(*name) + 1 + sizeof("T_"));
205
206         sprintf(tmpname, "%s%s_%s", var_prefix, prefix, *name);
207         *var = getenv_avl(tmpname);
208
209         sprintf(tmpname, "%sT_%s_%s", var_prefix, prefix, *name);
210         *type = getenv_avl(tmpname);
211
212         sprintf(tmpname, "%sN_%s_%s", var_prefix, prefix, *name);
213         varname = getenv_avl(tmpname);
214         if (varname)
215                 *name = varname;
216 }
217
218 static json_object *jshn_add_objects(json_object *obj, const char *prefix, bool array);
219
220 static void jshn_add_object_var(json_object *obj, bool array, const char *prefix, const char *name)
221 {
222         json_object *new;
223         char *var, *type;
224
225         get_var(prefix, &name, &var, &type);
226         if (!var || !type)
227                 return;
228
229         if (!strcmp(type, "array")) {
230                 new = json_object_new_array();
231                 jshn_add_objects(new, var, true);
232         } else if (!strcmp(type, "object")) {
233                 new = json_object_new_object();
234                 jshn_add_objects(new, var, false);
235         } else if (!strcmp(type, "string")) {
236                 new = json_object_new_string(var);
237         } else if (!strcmp(type, "int")) {
238                 new = json_object_new_int(atoi(var));
239         } else if (!strcmp(type, "double")) {
240                 new = json_object_new_double(strtod(var, NULL));
241         } else if (!strcmp(type, "boolean")) {
242                 new = json_object_new_boolean(!!atoi(var));
243         } else {
244                 return;
245         }
246
247         if (array)
248                 json_object_array_add(obj, new);
249         else
250                 json_object_object_add(obj, name, new);
251 }
252
253 static json_object *jshn_add_objects(json_object *obj, const char *prefix, bool array)
254 {
255         char *keys, *key, *brk;
256
257         keys = get_keys(prefix);
258         if (!keys || !obj)
259                 goto out;
260
261         for (key = strtok_r(keys, " ", &brk); key;
262              key = strtok_r(NULL, " ", &brk)) {
263                 jshn_add_object_var(obj, array, prefix, key);
264         }
265
266 out:
267         return obj;
268 }
269
270 static int jshn_format(bool no_newline, bool indent)
271 {
272         json_object *obj;
273         const char *output;
274         char *blobmsg_output = NULL;
275         int ret = -1;
276
277         if (!(obj = json_object_new_object()))
278                 return -1;
279
280         jshn_add_objects(obj, "J_V", false);
281         if (!(output = json_object_to_json_string(obj)))
282                 goto out;
283
284         if (indent) {
285                 blob_buf_init(&b, 0);
286                 if (!blobmsg_add_json_from_string(&b, output))
287                         goto out;
288                 if (!(blobmsg_output = blobmsg_format_json_indent(b.head, 1, 0)))
289                         goto out;
290                 output = blobmsg_output;
291         }
292         fprintf(stdout, "%s%s", output, no_newline ? "" : "\n");
293         free(blobmsg_output);
294         ret = 0;
295
296 out:
297         json_object_put(obj);
298         return ret;
299 }
300
301 static int usage(const char *progname)
302 {
303         fprintf(stderr, "Usage: %s [-n] [-i] -r <message>|-w\n", progname);
304         return 2;
305 }
306
307 static int avl_strcmp_var(const void *k1, const void *k2, void *ptr)
308 {
309         const char *s1 = k1;
310         const char *s2 = k2;
311         char c1, c2;
312
313         while (*s1 && *s1 == *s2) {
314                 s1++;
315                 s2++;
316         }
317
318         c1 = *s1;
319         c2 = *s2;
320         if (c1 == '=')
321                 c1 = 0;
322         if (c2 == '=')
323                 c2 = 0;
324
325         return c1 - c2;
326 }
327
328 int main(int argc, char **argv)
329 {
330         extern char **environ;
331         bool no_newline = false;
332         bool indent = false;
333         struct env_var *vars;
334         int i;
335         int ch;
336
337         avl_init(&env_vars, avl_strcmp_var, false, NULL);
338         for (i = 0; environ[i]; i++);
339
340         vars = calloc(i, sizeof(*vars));
341         if (!vars) {
342                 fprintf(stderr, "%m\n");
343                 return -1;
344         }
345         for (i = 0; environ[i]; i++) {
346                 char *c;
347
348                 vars[i].avl.key = environ[i];
349                 c = strchr(environ[i], '=');
350                 if (!c)
351                         continue;
352
353                 vars[i].val = c + 1;
354                 avl_insert(&env_vars, &vars[i].avl);
355         }
356
357         while ((ch = getopt(argc, argv, "p:nir:w")) != -1) {
358                 switch(ch) {
359                 case 'p':
360                         var_prefix = optarg;
361                         var_prefix_len = strlen(var_prefix);
362                         break;
363                 case 'r':
364                         return jshn_parse(optarg);
365                 case 'w':
366                         return jshn_format(no_newline, indent);
367                 case 'n':
368                         no_newline = true;
369                         break;
370                 case 'i':
371                         indent = true;
372                         break;
373                 default:
374                         return usage(argv[0]);
375                 }
376         }
377         return usage(argv[0]);
378 }