uloop: fix a regression in timeout handling
[project/libubox.git] / jshn.c
diff --git a/jshn.c b/jshn.c
index dc002c1..4989099 100644 (file)
--- a/jshn.c
+++ b/jshn.c
@@ -1,4 +1,24 @@
-#include <json/json.h>
+/*
+ * Copyright (C) 2011-2013 Felix Fietkau <nbd@openwrt.org>
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+#ifdef JSONC
+        #include <json.h>
+#else
+        #include <json/json.h>
+#endif
+
 #include <string.h>
 #include <stdlib.h>
 #include <stdio.h>
 #include <getopt.h>
 #include "list.h"
 
+#include "avl.h"
+#include "blob.h"
+#include "blobmsg_json.h"
+
 #define MAX_VARLEN     256
 
+static struct avl_tree env_vars;
+static struct blob_buf b = { 0 };
+
 static const char *var_prefix = "";
 static int var_prefix_len = 0;
 
 static int add_json_element(const char *key, json_object *obj);
 
+struct env_var {
+       struct avl_node avl;
+       char *val;
+};
+
 static int add_json_object(json_object *obj)
 {
        int ret = 0;
@@ -92,6 +124,9 @@ static int add_json_element(const char *key, json_object *obj)
        case json_type_int:
                type = "int";
                break;
+       case json_type_double:
+               type = "double";
+               break;
        default:
                return -1;
        }
@@ -121,6 +156,9 @@ static int add_json_element(const char *key, json_object *obj)
        case json_type_int:
                fprintf(stdout, "' %d;\n", json_object_get_int(obj));
                break;
+       case json_type_double:
+               fprintf(stdout, "' %lf;\n", json_object_get_double(obj));
+               break;
        default:
                return -1;
        }
@@ -133,7 +171,7 @@ static int jshn_parse(const char *str)
        json_object *obj;
 
        obj = json_tokener_parse(str);
-       if (is_error(obj) || json_object_get_type(obj) != json_type_object) {
+       if (!obj || json_object_get_type(obj) != json_type_object) {
                fprintf(stderr, "Failed to parse message data\n");
                return 1;
        }
@@ -144,29 +182,35 @@ static int jshn_parse(const char *str)
        return 0;
 }
 
+static char *getenv_avl(const char *key)
+{
+       struct env_var *var = avl_find_element(&env_vars, key, var, avl);
+       return var ? var->val : NULL;
+}
+
 static char *get_keys(const char *prefix)
 {
        char *keys;
 
-       keys = alloca(var_prefix_len + strlen(prefix) + sizeof("KEYS_") + 1);
-       sprintf(keys, "%sKEYS_%s", var_prefix, prefix);
-       return getenv(keys);
+       keys = alloca(var_prefix_len + strlen(prefix) + sizeof("K_") + 1);
+       sprintf(keys, "%sK_%s", var_prefix, prefix);
+       return getenv_avl(keys);
 }
 
 static void get_var(const char *prefix, const char **name, char **var, char **type)
 {
        char *tmpname, *varname;
 
-       tmpname = alloca(var_prefix_len + strlen(prefix) + 1 + strlen(*name) + 1 + sizeof("TYPE_"));
+       tmpname = alloca(var_prefix_len + strlen(prefix) + 1 + strlen(*name) + 1 + sizeof("T_"));
 
        sprintf(tmpname, "%s%s_%s", var_prefix, prefix, *name);
-       *var = getenv(tmpname);
+       *var = getenv_avl(tmpname);
 
-       sprintf(tmpname, "%sTYPE_%s_%s", var_prefix, prefix, *name);
-       *type = getenv(tmpname);
+       sprintf(tmpname, "%sT_%s_%s", var_prefix, prefix, *name);
+       *type = getenv_avl(tmpname);
 
-       sprintf(tmpname, "%sNAME_%s_%s", var_prefix, prefix, *name);
-       varname = getenv(tmpname);
+       sprintf(tmpname, "%sN_%s_%s", var_prefix, prefix, *name);
+       varname = getenv_avl(tmpname);
        if (varname)
                *name = varname;
 }
@@ -192,6 +236,8 @@ static void jshn_add_object_var(json_object *obj, bool array, const char *prefix
                new = json_object_new_string(var);
        } else if (!strcmp(type, "int")) {
                new = json_object_new_int(atoi(var));
+       } else if (!strcmp(type, "double")) {
+               new = json_object_new_double(strtod(var, NULL));
        } else if (!strcmp(type, "boolean")) {
                new = json_object_new_boolean(!!atoi(var));
        } else {
@@ -221,30 +267,94 @@ out:
        return obj;
 }
 
-static int jshn_format(bool no_newline)
+static int jshn_format(bool no_newline, bool indent)
 {
        json_object *obj;
+       const char *output;
+       char *blobmsg_output = NULL;
+       int ret = -1;
 
-       obj = json_object_new_object();
-       jshn_add_objects(obj, "JSON_VAR", false);
-       fprintf(stdout, "%s%s", json_object_to_json_string(obj),
-               no_newline ? "" : "\n");
+       if (!(obj = json_object_new_object()))
+               return -1;
+
+       jshn_add_objects(obj, "J_V", false);
+       if (!(output = json_object_to_json_string(obj)))
+               goto out;
+
+       if (indent) {
+               blob_buf_init(&b, 0);
+               if (!blobmsg_add_json_from_string(&b, output))
+                       goto out;
+               if (!(blobmsg_output = blobmsg_format_json_indent(b.head, 1, 0)))
+                       goto out;
+               output = blobmsg_output;
+       }
+       fprintf(stdout, "%s%s", output, no_newline ? "" : "\n");
+       free(blobmsg_output);
+       ret = 0;
+
+out:
        json_object_put(obj);
-       return 0;
+       return ret;
 }
 
 static int usage(const char *progname)
 {
-       fprintf(stderr, "Usage: %s [-n] -r <message>|-w\n", progname);
+       fprintf(stderr, "Usage: %s [-n] [-i] -r <message>|-w\n", progname);
        return 2;
 }
 
+static int avl_strcmp_var(const void *k1, const void *k2, void *ptr)
+{
+       const char *s1 = k1;
+       const char *s2 = k2;
+       char c1, c2;
+
+       while (*s1 && *s1 == *s2) {
+               s1++;
+               s2++;
+       }
+
+       c1 = *s1;
+       c2 = *s2;
+       if (c1 == '=')
+               c1 = 0;
+       if (c2 == '=')
+               c2 = 0;
+
+       return c1 - c2;
+}
+
 int main(int argc, char **argv)
 {
+       extern char **environ;
        bool no_newline = false;
+       bool indent = false;
+       struct env_var *vars;
+       int i;
        int ch;
 
-       while ((ch = getopt(argc, argv, "p:nr:w")) != -1) {
+       avl_init(&env_vars, avl_strcmp_var, false, NULL);
+       for (i = 0; environ[i]; i++);
+
+       vars = calloc(i, sizeof(*vars));
+       if (!vars) {
+               fprintf(stderr, "%m\n");
+               return -1;
+       }
+       for (i = 0; environ[i]; i++) {
+               char *c;
+
+               vars[i].avl.key = environ[i];
+               c = strchr(environ[i], '=');
+               if (!c)
+                       continue;
+
+               vars[i].val = c + 1;
+               avl_insert(&env_vars, &vars[i].avl);
+       }
+
+       while ((ch = getopt(argc, argv, "p:nir:w")) != -1) {
                switch(ch) {
                case 'p':
                        var_prefix = optarg;
@@ -253,10 +363,13 @@ int main(int argc, char **argv)
                case 'r':
                        return jshn_parse(optarg);
                case 'w':
-                       return jshn_format(no_newline);
+                       return jshn_format(no_newline, indent);
                case 'n':
                        no_newline = true;
                        break;
+               case 'i':
+                       indent = true;
+                       break;
                default:
                        return usage(argv[0]);
                }