radare2/libr/util/json_indent.c

274 lines
4.7 KiB
C
Raw Normal View History

/* radare2 (from sdb) - MIT - Copyright 2012-2017 - pancake */
#include <r_util.h>
#include <r_print.h>
2017-03-17 11:09:17 +00:00
static void doIndent(int idt, char** o, const char *tab) {
int i;
char *x;
for (i = 0; i < idt; i++) {
for (x = (char*) tab; *x; x++) {
*(*o)++ = *x;
}
}
}
2017-05-19 11:43:33 +00:00
#define EMIT_ESC(s, code) do { \
if (color) { \
const char *p = code; \
while (*p) { \
*s++ = *p++; \
} \
} \
} while (0);
enum {
JC_FALSE, // 31m
JC_TRUE, // 32m
JC_KEY, // 33m
JC_VAL, // 34m
JC_RESET,
};
static const char *origColors[] = {
"\x1b[31m",
"\x1b[32m",
"\x1b[33m",
"\x1b[34m",
"\x1b[0m",
};
// static const char colors
2017-10-16 11:31:13 +00:00
R_API char* r_print_json_path(const char* s, int pos) {
int indent = 0;
#define DSZ 128
const char *words[DSZ] = { NULL };
int lengths[DSZ] = { 0 };
int indexs[DSZ] = { 0 };
int instr = 0;
bool isarr = false;
if (!s) {
return NULL;
}
int arrpos = 0;
const char *os = s;
int osz = (1 + strlen (s)) * 20;
if (osz < 1) {
return NULL;
}
const char *str_a = NULL;
2017-12-08 14:47:48 +00:00
for (; *s; s++) {
2017-10-16 11:31:13 +00:00
if (instr) {
if (s[0] == '"') {
instr = 0;
ut64 cur = str_a - os;
if (cur > pos) {
break;
}
2017-10-21 09:33:51 +00:00
if (indent < DSZ) {
words[indent - 1] = str_a;
lengths[indent - 1] = s - str_a;
indexs[indent - 1] = 0;
}
2017-10-16 11:31:13 +00:00
}
continue;
}
if (s[0] == '"') {
instr = 1;
str_a = s + 1;
}
if (*s == '\n' || *s == '\r' || *s == '\t' || *s == ' ') {
continue;
}
switch (*s) {
case ':':
break;
case ',':
if (isarr) {
arrpos ++;
2017-10-21 09:33:51 +00:00
if (indent < DSZ) {
indexs[indent - 1] = arrpos;
lengths[indent - 1] = (s - os);
}
2017-10-16 11:31:13 +00:00
}
break;
case '{':
case '[':
if (*s == '[') {
isarr = true;
arrpos = 0;
}
if (indent > 128) {
eprintf ("JSON indentation is too deep\n");
indent = 0;
} else {
indent++;
}
break;
case '}':
case ']':
if (*s == ']') {
isarr = false;
}
indent--;
break;
}
}
int i;
ut64 opos = 0;
2017-10-21 09:33:51 +00:00
for (i = 0; i < DSZ && i < indent; i++) {
2017-12-03 17:22:47 +00:00
if ((int)(size_t)words[i] < DSZ) {
2017-10-16 11:31:13 +00:00
ut64 cur = lengths[i];
if (cur < opos) {
continue;
}
opos = cur;
if (cur > pos) {
break;
}
eprintf ("0x%08"PFMT64x" %d [%d]\n", cur, i, indexs[i]);
} else {
char *a = r_str_ndup (words[i], lengths[i]);
ut64 cur = words[i] - os - 1;
if (cur < opos) {
continue;
}
opos = cur;
if (cur > pos) {
break;
}
char *q = strchr (a, '"');
if (q) {
*q = 0;
}
eprintf ("0x%08"PFMT64x" %d %s\n", cur, i, a);
free (a);
}
}
// TODO return something
return NULL;
}
R_API char* r_print_json_indent(const char* s, bool color, const char* tab, const char **palette) {
int indent = 0;
const int indentSize = strlen (tab);
2017-03-17 11:09:17 +00:00
int instr = 0;
bool isValue = false;
char *o, *OE, *tmp;
if (!s) {
return NULL;
}
const char **colors = palette ? palette: origColors;
int osz = (1 + strlen (s)) * 20;
if (osz < 1) {
return NULL;
}
char *O = malloc (osz);
if (!O) {
return NULL;
}
OE = O + osz;
for (o = O; *s; s++) {
if (o + (indent * indentSize) + 10 > OE) {
int delta = o - O;
osz += 0x1000 + (indent * indentSize);
if (osz < 1) {
free (O);
return NULL;
}
tmp = realloc (O, osz);
if (!tmp) {
free (O);
return NULL;
}
O = tmp;
OE = tmp + osz;
o = O + delta;
}
if (instr) {
if (s[0] == '"') {
instr = 0;
} else if (s[0] == '\\' && s[1] == '"') {
*o++ = *s++;
}
if (instr) {
if (isValue) {
// TODO: do not emit color in every char
EMIT_ESC (o, colors[JC_VAL]);
} else {
EMIT_ESC (o, colors[JC_KEY]);
}
} else {
EMIT_ESC (o, colors[JC_RESET]);
}
*o++ = *s;
continue;
}
if (indent <= 0) {
// non-JSON part
if (s[0] != '{' && s[0] != '[') {
*o++ = *s;
continue;
}
}
if (s[0] == '"') {
instr = 1;
}
if (*s == '\n' || *s == '\r' || *s == '\t' || *s == ' ') {
continue;
}
switch (*s) {
case ':':
*o++ = *s;
*o++ = ' ';
if (!strncmp (s + 1, "true", 4)) {
EMIT_ESC (o, colors[JC_TRUE]);
} else if (!strncmp (s + 1, "false", 5)) {
EMIT_ESC (o, colors[JC_FALSE]);
}
isValue = true;
break;
case ',':
EMIT_ESC (o, colors[JC_RESET]);
*o++ = *s;
*o++ = '\n';
isValue = false;
2017-03-17 11:09:17 +00:00
doIndent (indent, &o, tab);
break;
case '{':
case '[':
isValue = false;
*o++ = *s;
*o++ = (indent != -1)? '\n': ' ';
if (indent > 128) {
eprintf ("JSON indentation is too deep\n");
indent = 0;
} else {
indent++;
}
2017-03-17 11:09:17 +00:00
doIndent (indent, &o, tab);
break;
case '}':
case ']':
EMIT_ESC (o, colors[JC_RESET]);
isValue = false;
*o++ = '\n';
2017-03-17 11:09:17 +00:00
indent--;
doIndent (indent, &o, tab);
*o++ = *s;
break;
default:
*o++ = *s;
}
}
*o = 0;
return O;
}
#undef EMIT_ESC