src/dump.c

   1 /*
   2  * Copyright (c) 2009 Petri Lehtinen <petri@digip.org>
   3  *
   4  * Jansson is free software; you can redistribute it and/or modify
   5  * it under the terms of the MIT license. See LICENSE for details.
   6  */
   7
   8 #define _GNU_SOURCE
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11 #include <string.h>
  12
  13 #include <jansson.h>
  14 #include "strbuffer.h"
  15
  16 typedef int (*dump_func)(const char *buffer, int size, void *data);
  17
  18 struct string
  19 {
  20     char *buffer;
  21     int length;
  22     int size;
  23 };
  24
  25 static int dump_to_strbuffer(const char *buffer, int size, void *data)
  26 {
  27     return strbuffer_append_bytes((strbuffer_t *)data, buffer, size);
  28 }
  29
  30 static int dump_to_file(const char *buffer, int size, void *data)
  31 {
  32     FILE *dest = (FILE *)data;
  33     if(fwrite(buffer, size, 1, dest) != 1)
  34         return -1;
  35     return 0;
  36 }
  37
  38 /* 256 spaces (the maximum indentation size) */
  39 static char whitespace[] = "                                                                                                                                                                                                                                                                ";
  40
  41 static int dump_indent(unsigned long flags, int depth, dump_func dump, void *data)
  42 {
  43     if(JSON_INDENT(flags) > 0)
  44     {
  45         int i, ws_count = JSON_INDENT(flags);
  46
  47         if(dump("\n", 1, data))
  48             return -1;
  49
  50         for(i = 0; i < depth; i++)
  51         {
  52             if(dump(whitespace, ws_count, data))
  53                 return -1;
  54         }
  55     }
  56     return 0;
  57 }
  58
  59 static int dump_string(const char *str, dump_func dump, void *data)
  60 {
  61     const char *end;
  62
  63     if(dump("\"", 1, data))
  64         return -1;
  65
  66     end = str;
  67     while(1)
  68     {
  69         const char *text;
  70         char seq[7];
  71         int length;
  72
  73         while(*end && *end != '\\' && *end != '"' && (unsigned char)*end > 0x1F)
  74             end++;
  75
  76         if(end != str) {
  77             if(dump(str, end - str, data))
  78                 return -1;
  79         }
  80
  81         if(!*end)
  82             break;
  83
  84         /* handle \, ", and control codes */
  85         length = 2;
  86         switch(*end)
  87         {
  88             case '\\': text = "\\\\"; break;
  89             case '\"': text = "\\\""; break;
  90             case '\b': text = "\\b"; break;
  91             case '\f': text = "\\f"; break;
  92             case '\n': text = "\\n"; break;
  93             case '\r': text = "\\r"; break;
  94             case '\t': text = "\\t"; break;
  95             default:
  96             {
  97                 sprintf(seq, "\\u00%02x", *end);
  98                 text = seq;
  99                 length = 6;
 100                 break;
 101             }
 102         }
 103
 104         if(dump(text, length, data))
 105             return -1;
 106
 107         end++;
 108         str = end;
 109     }
 110
 111     return dump("\"", 1, data);
 112 }
 113
 114 static int do_dump(const json_t *json, unsigned long flags, int depth,
 115                    dump_func dump, void *data)
 116 {
 117     switch(json_typeof(json)) {
 118         case JSON_NULL:
 119             return dump("null", 4, data);
 120
 121         case JSON_TRUE:
 122             return dump("true", 4, data);
 123
 124         case JSON_FALSE:
 125             return dump("false", 5, data);
 126
 127         case JSON_INTEGER:
 128         {
 129             char *buffer;
 130             int size, ret;
 131
 132             size = asprintf(&buffer, "%d", json_integer_value(json));
 133             if(size == -1)
 134                 return -1;
 135
 136             ret = dump(buffer, size, data);
 137             free(buffer);
 138             return ret;
 139         }
 140
 141         case JSON_REAL:
 142         {
 143             char *buffer;
 144             int size, ret;
 145
 146             size = asprintf(&buffer, "%.17f", json_real_value(json));
 147             if(size == -1)
 148                 return -1;
 149
 150             ret = dump(buffer, size, data);
 151             free(buffer);
 152             return ret;
 153         }
 154
 155         case JSON_STRING:
 156             return dump_string(json_string_value(json), dump, data);
 157
 158         case JSON_ARRAY:
 159         {
 160             int i;
 161             int n = json_array_size(json);
 162
 163             if(dump("[", 1, data))
 164                 return -1;
 165             if(n == 0)
 166                 return dump("]", 1, data);
 167             if(dump_indent(flags, depth + 1, dump, data))
 168                 return -1;
 169
 170             for(i = 0; i < n; ++i) {
 171                 if(do_dump(json_array_get(json, i), flags, depth + 1,
 172                            dump, data))
 173                     return -1;
 174
 175                 if(i < n - 1)
 176                 {
 177                     if(dump(",", 1, data) ||
 178                        dump_indent(flags, depth + 1, dump, data))
 179                         return -1;
 180                 }
 181                 else
 182                 {
 183                     if(dump_indent(flags, depth, dump, data))
 184                         return -1;
 185                 }
 186             }
 187             return dump("]", 1, data);
 188         }
 189
 190         case JSON_OBJECT:
 191         {
 192             void *iter = json_object_iter((json_t *)json);
 193
 194             if(dump("{", 1, data))
 195                 return -1;
 196             if(!iter)
 197                 return dump("}", 1, data);
 198             if(dump_indent(flags, depth + 1, dump, data))
 199                 return -1;
 200
 201             while(iter)
 202             {
 203                 void *next = json_object_iter_next((json_t *)json, iter);
 204
 205                 dump_string(json_object_iter_key(iter), dump, data);
 206                 if(dump(": ", 2, data) ||
 207                    do_dump(json_object_iter_value(iter), flags, depth + 1,
 208                            dump, data))
 209                     return -1;
 210
 211                 if(next)
 212                 {
 213                     if(dump(",", 1, data) ||
 214                        dump_indent(flags, depth + 1, dump, data))
 215                         return -1;
 216                 }
 217                 else
 218                 {
 219                     if(dump_indent(flags, depth, dump, data))
 220                         return -1;
 221                 }
 222
 223                 iter = next;
 224             }
 225             return dump("}", 1, data);
 226         }
 227
 228         default:
 229             /* not reached */
 230             return -1;
 231     }
 232 }
 233
 234
 235 char *json_dumps(const json_t *json, unsigned long flags)
 236 {
 237     strbuffer_t strbuff;
 238     char *result;
 239
 240     if(!json_is_array(json) && !json_is_object(json))
 241         return NULL;
 242
 243     if(strbuffer_init(&strbuff))
 244       return NULL;
 245
 246     if(do_dump(json, flags, 0, dump_to_strbuffer, (void *)&strbuff))
 247         return NULL;
 248
 249     if(dump_to_strbuffer("\n", 1, (void *)&strbuff))
 250         return NULL;
 251
 252     result = strdup(strbuffer_value(&strbuff));
 253     strbuffer_close(&strbuff);
 254
 255     return result;
 256 }
 257
 258 int json_dumpf(const json_t *json, FILE *output, unsigned long flags)
 259 {
 260     if(!json_is_array(json) && !json_is_object(json))
 261         return -1;
 262
 263     if(do_dump(json, flags, 0, dump_to_file, (void *)output))
 264         return -1;
 265     return dump_to_file("\n", 1, (void *)output);
 266 }
 267
 268 int json_dump_file(const json_t *json, const char *path, unsigned long flags)
 269 {
 270     int result;
 271
 272     FILE *output = fopen(path, "w");
 273     if(!output)
 274         return -1;
 275
 276     result = json_dumpf(json, output, flags);
 277
 278     fclose(output);
 279     return result;
 280 }