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