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 }