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         bool array;
 290
 291         s.len = blob_len(attr);
 292         s.buf = malloc(s.len);
 293         s.pos = 0;
 294         s.custom_format = cb;
 295         s.priv = priv;
 296         s.indent = false;
 297
 298         if (indent >= 0) {
 299                 s.indent = true;
 300                 s.indent_level = indent;
 301         }
 302
 303         array = blob_is_extended(attr) &&
 304                 blobmsg_type(attr) == BLOBMSG_TYPE_ARRAY;
 305
 306         if (list)
 307                 blobmsg_format_json_list(&s, blobmsg_data(attr), blobmsg_data_len(attr), array);
 308         else
 309                 blobmsg_format_element(&s, attr, false, false);
 310
 311         if (!s.len) {
 312                 free(s.buf);
 313                 return NULL;
 314         }
 315
 316         s.buf = realloc(s.buf, s.pos + 1);
 317         s.buf[s.pos] = 0;
 318
 319         return s.buf;
 320 }