[fio.git] / json.c

#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <stdarg.h>
#include "json.h"
#include "log.h"

struct json_object *json_create_object(void)
{
	return calloc(1, sizeof(struct json_object));
}

struct json_array *json_create_array(void)
{
	return calloc(1, sizeof(struct json_array));
}

static struct json_pair *json_create_pair(const char *name, struct json_value *value)
{
	struct json_pair *pair = malloc(sizeof(struct json_pair));
	if (pair) {
		pair->name = strdup(name);
		pair->value = value;

		value->parent_type = JSON_PARENT_TYPE_PAIR;
		value->parent_pair = pair;
	}
	return pair;
}

static struct json_value *json_create_value_int(long long number)
{
	struct json_value *value = malloc(sizeof(struct json_value));

	if (value) {
		value->type = JSON_TYPE_INTEGER;
		value->integer_number = number;
	}
	return value;
}

static struct json_value *json_create_value_float(double number)
{
	struct json_value *value = malloc(sizeof(struct json_value));

	if (value) {
		value->type = JSON_TYPE_FLOAT;
		value->float_number = number;
	}
	return value;
}

static char *strdup_escape(const char *str)
{
	const char *input = str;
	char *p, *ret;
	int escapes;

	if (!strlen(str))
		return NULL;

	escapes = 0;
	while ((input = strpbrk(input, "\\\"")) != NULL) {
		escapes++;
		input++;
	}

	p = ret = malloc(strlen(str) + escapes + 1);
	while (*str) {
		if (*str == '\\' || *str == '\"')
			*p++ = '\\';
		*p++ = *str++;
	}
	*p = '\0';

	return ret;
}

/*
 * Valid JSON strings must escape '"' and '/' with a preceding '/'
 */
static struct json_value *json_create_value_string(const char *str)
{
	struct json_value *value = malloc(sizeof(struct json_value));

	if (value) {
		value->type = JSON_TYPE_STRING;
		value->string = strdup_escape(str);
		if (!value->string) {
			free(value);
			value = NULL;
		}
	}
	return value;
}

static struct json_value *json_create_value_object(struct json_object *obj)
{
	struct json_value *value = malloc(sizeof(struct json_value));

	if (value) {
		value->type = JSON_TYPE_OBJECT;
		value->object = obj;
		obj->parent = value;
	}
	return value;
}

static struct json_value *json_create_value_array(struct json_array *array)
{
	struct json_value *value = malloc(sizeof(struct json_value));

	if (value) {
		value->type = JSON_TYPE_ARRAY;
		value->array = array;
		array->parent = value;
	}
	return value;
}

static void json_free_pair(struct json_pair *pair);
static void json_free_value(struct json_value *value);

void json_free_object(struct json_object *obj)
{
	int i;

	for (i = 0; i < obj->pair_cnt; i++)
		json_free_pair(obj->pairs[i]);
	free(obj->pairs);
	free(obj);
}

static void json_free_array(struct json_array *array)
{
	int i;

	for (i = 0; i < array->value_cnt; i++)
		json_free_value(array->values[i]);
	free(array->values);
	free(array);
}

static void json_free_pair(struct json_pair *pair)
{
	json_free_value(pair->value);
	free(pair->name);
	free(pair);
}

static void json_free_value(struct json_value *value)
{
	switch (value->type) {
	case JSON_TYPE_STRING:
		free(value->string);
		break;
	case JSON_TYPE_OBJECT:
		json_free_object(value->object);
		break;
	case JSON_TYPE_ARRAY:
		json_free_array(value->array);
		break;
	}
	free(value);
}

static int json_array_add_value(struct json_array *array, struct json_value *value)
{
	struct json_value **values = realloc(array->values,
		sizeof(struct json_value *) * (array->value_cnt + 1));

	if (!values)
		return ENOMEM;
	values[array->value_cnt] = value;
	array->value_cnt++;
	array->values = values;

	value->parent_type = JSON_PARENT_TYPE_ARRAY;
	value->parent_array = array;
	return 0;
}

static int json_object_add_pair(struct json_object *obj, struct json_pair *pair)
{
	struct json_pair **pairs = realloc(obj->pairs,
		sizeof(struct json_pair *) * (obj->pair_cnt + 1));
	if (!pairs)
		return ENOMEM;
	pairs[obj->pair_cnt] = pair;
	obj->pair_cnt++;
	obj->pairs = pairs;

	pair->parent = obj;
	return 0;
}

int json_object_add_value_type(struct json_object *obj, const char *name,
			       const struct json_value *arg)
{
	struct json_value *value;
	struct json_pair *pair;
	int ret;

	switch (arg->type) {
	case JSON_TYPE_STRING:
		value = json_create_value_string(arg->string);
		break;
	case JSON_TYPE_INTEGER:
		value = json_create_value_int(arg->integer_number);
		break;
	case JSON_TYPE_FLOAT:
		value = json_create_value_float(arg->float_number);
		break;
	case JSON_TYPE_OBJECT:
		value = json_create_value_object(arg->object);
		break;
	default:
	case JSON_TYPE_ARRAY:
		value = json_create_value_array(arg->array);
		break;
	}

	if (!value)
		return ENOMEM;

	pair = json_create_pair(name, value);
	if (!pair) {
		json_free_value(value);
		return ENOMEM;
	}
	ret = json_object_add_pair(obj, pair);
	if (ret) {
		json_free_pair(pair);
		return ENOMEM;
	}
	return 0;
}

int json_array_add_value_type(struct json_array *array,
			      const struct json_value *arg)
{
	struct json_value *value;
	int ret;

	switch (arg->type) {
	case JSON_TYPE_STRING:
		value = json_create_value_string(arg->string);
		break;
	case JSON_TYPE_INTEGER:
		value = json_create_value_int(arg->integer_number);
		break;
	case JSON_TYPE_FLOAT:
		value = json_create_value_float(arg->float_number);
		break;
	case JSON_TYPE_OBJECT:
		value = json_create_value_object(arg->object);
		break;
	default:
	case JSON_TYPE_ARRAY:
		value = json_create_value_array(arg->array);
		break;
	}

	if (!value)
		return ENOMEM;

	ret = json_array_add_value(array, value);
	if (ret) {
		json_free_value(value);
		return ENOMEM;
	}
	return 0;
}

static int json_value_level(struct json_value *value);
static int json_pair_level(struct json_pair *pair);
static int json_array_level(struct json_array *array);
static int json_object_level(struct json_object *object)
{
	if (object->parent == NULL)
		return 0;
	return json_value_level(object->parent);
}

static int json_pair_level(struct json_pair *pair)
{
	return json_object_level(pair->parent) + 1;
}

static int json_array_level(struct json_array *array)
{
	return json_value_level(array->parent);
}

static int json_value_level(struct json_value *value)
{
	if (value->parent_type == JSON_PARENT_TYPE_PAIR)
		return json_pair_level(value->parent_pair);
	else
		return json_array_level(value->parent_array) + 1;
}

static void json_print_level(int level, struct buf_output *out)
{
	while (level-- > 0)
		log_buf(out, "  ");
}

static void json_print_pair(struct json_pair *pair, struct buf_output *);
static void json_print_value(struct json_value *value, struct buf_output *);

void json_print_object(struct json_object *obj, struct buf_output *out)
{
	int i;

	log_buf(out, "{\n");
	for (i = 0; i < obj->pair_cnt; i++) {
		if (i > 0)
			log_buf(out, ",\n");
		json_print_pair(obj->pairs[i], out);
	}
	log_buf(out, "\n");
	json_print_level(json_object_level(obj), out);
	log_buf(out, "}");
}

static void json_print_pair(struct json_pair *pair, struct buf_output *out)
{
	json_print_level(json_pair_level(pair), out);
	log_buf(out, "\"%s\" : ", pair->name);
	json_print_value(pair->value, out);
}

static void json_print_array(struct json_array *array, struct buf_output *out)
{
	int i;

	log_buf(out, "[\n");
	for (i = 0; i < array->value_cnt; i++) {
		if (i > 0)
			log_buf(out, ",\n");
		json_print_level(json_value_level(array->values[i]), out);
		json_print_value(array->values[i], out);
	}
	log_buf(out, "\n");
	json_print_level(json_array_level(array), out);
	log_buf(out, "]");
}

static void json_print_value(struct json_value *value, struct buf_output *out)
{
	switch (value->type) {
	case JSON_TYPE_STRING:
		log_buf(out, "\"%s\"", value->string);
		break;
	case JSON_TYPE_INTEGER:
		log_buf(out, "%lld", value->integer_number);
		break;
	case JSON_TYPE_FLOAT:
		log_buf(out, "%f", value->float_number);
		break;
	case JSON_TYPE_OBJECT:
		json_print_object(value->object, out);
		break;
	case JSON_TYPE_ARRAY:
		json_print_array(value->array, out);
		break;
	}
}
Commit	Line	Data
cc372b17 SL	1	#include <stdlib.h>
cc372b17 SL	2	#include <string.h>
cc372b17 SL	3	#include <errno.h>
	4	#include <stdarg.h>
	5	#include "json.h"
	6	#include "log.h"
	7
	8	struct json_object *json_create_object(void)
	9	{
e75cc8f3	10	return calloc(1, sizeof(struct json_object));
cc372b17 SL	11	}
	12
	13	struct json_array *json_create_array(void)
	14	{
e75cc8f3	15	return calloc(1, sizeof(struct json_array));
cc372b17 SL	16	}
	17
	18	static struct json_pair json_create_pair(const char name, struct json_value *value)
	19	{
	20	struct json_pair *pair = malloc(sizeof(struct json_pair));
	21	if (pair) {
	22	pair->name = strdup(name);
	23	pair->value = value;
	24
	25	value->parent_type = JSON_PARENT_TYPE_PAIR;
	26	value->parent_pair = pair;
	27	}
	28	return pair;
	29	}
	30
ee2e5717	31	static struct json_value *json_create_value_int(long long number)
cc372b17 SL	32	{
	33	struct json_value *value = malloc(sizeof(struct json_value));
	34
	35	if (value) {
	36	value->type = JSON_TYPE_INTEGER;
	37	value->integer_number = number;
	38	}
	39	return value;
	40	}
	41
a2972f96	42	static struct json_value *json_create_value_float(double number)
cc372b17 SL	43	{
	44	struct json_value *value = malloc(sizeof(struct json_value));
	45
	46	if (value) {
	47	value->type = JSON_TYPE_FLOAT;
	48	value->float_number = number;
	49	}
	50	return value;
	51	}
	52
d366a95e JA	53	static char strdup_escape(const char str)
	54	{
	55	const char *input = str;
	56	char p, ret;
	57	int escapes;
	58
a5725153 JA	59	if (!strlen(str))
	60	return NULL;
	61
d366a95e JA	62	escapes = 0;
	63	while ((input = strpbrk(input, "\\\"")) != NULL) {
	64	escapes++;
	65	input++;
	66	}
	67
a5725153	68	p = ret = malloc(strlen(str) + escapes + 1);
d366a95e JA	69	while (*str) {
	70	if (str == '\\' \|\| str == '\"')
	71	*p++ = '\\';
	72	p++ = str++;
	73	}
a5725153	74	*p = '\0';
d366a95e JA	75
	76	return ret;
	77	}
	78
	79	/*
de8f6de9	80	* Valid JSON strings must escape '"' and '/' with a preceding '/'
d366a95e	81	*/
cc372b17 SL	82	static struct json_value json_create_value_string(const char str)
	83	{
	84	struct json_value *value = malloc(sizeof(struct json_value));
	85
	86	if (value) {
	87	value->type = JSON_TYPE_STRING;
d366a95e	88	value->string = strdup_escape(str);
cc372b17 SL	89	if (!value->string) {
	90	free(value);
	91	value = NULL;
	92	}
	93	}
	94	return value;
	95	}
	96
	97	static struct json_value json_create_value_object(struct json_object obj)
	98	{
	99	struct json_value *value = malloc(sizeof(struct json_value));
	100
	101	if (value) {
	102	value->type = JSON_TYPE_OBJECT;
	103	value->object = obj;
	104	obj->parent = value;
	105	}
	106	return value;
	107	}
	108
	109	static struct json_value json_create_value_array(struct json_array array)
	110	{
	111	struct json_value *value = malloc(sizeof(struct json_value));
	112
	113	if (value) {
	114	value->type = JSON_TYPE_ARRAY;
	115	value->array = array;
	116	array->parent = value;
	117	}
	118	return value;
	119	}
	120
cc372b17 SL	121	static void json_free_pair(struct json_pair *pair);
cc372b17 SL	122	static void json_free_value(struct json_value *value);
f3afa57e	123
cc372b17 SL	124	void json_free_object(struct json_object *obj)
	125	{
	126	int i;
	127
	128	for (i = 0; i < obj->pair_cnt; i++)
	129	json_free_pair(obj->pairs[i]);
	130	free(obj->pairs);
	131	free(obj);
	132	}
	133
	134	static void json_free_array(struct json_array *array)
	135	{
	136	int i;
	137
	138	for (i = 0; i < array->value_cnt; i++)
	139	json_free_value(array->values[i]);
	140	free(array->values);
	141	free(array);
	142	}
	143
	144	static void json_free_pair(struct json_pair *pair)
	145	{
	146	json_free_value(pair->value);
	147	free(pair->name);
	148	free(pair);
	149	}
	150
	151	static void json_free_value(struct json_value *value)
	152	{
	153	switch (value->type) {
	154	case JSON_TYPE_STRING:
	155	free(value->string);
	156	break;
	157	case JSON_TYPE_OBJECT:
	158	json_free_object(value->object);
	159	break;
	160	case JSON_TYPE_ARRAY:
	161	json_free_array(value->array);
	162	break;
	163	}
	164	free(value);
	165	}
	166
	167	static int json_array_add_value(struct json_array array, struct json_value value)
	168	{
	169	struct json_value **values = realloc(array->values,
	170	sizeof(struct json_value ) (array->value_cnt + 1));
	171
	172	if (!values)
	173	return ENOMEM;
	174	values[array->value_cnt] = value;
	175	array->value_cnt++;
	176	array->values = values;
	177
	178	value->parent_type = JSON_PARENT_TYPE_ARRAY;
	179	value->parent_array = array;
	180	return 0;
	181	}
	182
	183	static int json_object_add_pair(struct json_object obj, struct json_pair pair)
	184	{
	185	struct json_pair **pairs = realloc(obj->pairs,
	186	sizeof(struct json_pair ) (obj->pair_cnt + 1));
	187	if (!pairs)
188	return ENOMEM;
189	pairs[obj->pair_cnt] = pair;
190	obj->pair_cnt++;
191	obj->pairs = pairs;
192
193	pair->parent = obj;
194	return 0;
195	}
196
eb2f29b7 BVA	197	int json_object_add_value_type(struct json_object obj, const char name,
eb2f29b7 BVA	198	const struct json_value *arg)
cc372b17 SL	199	{
	200	struct json_value *value;
	201	struct json_pair *pair;
cc372b17 SL	202	int ret;
cc372b17 SL	203
eb2f29b7 BVA	204	switch (arg->type) {
	205	case JSON_TYPE_STRING:
	206	value = json_create_value_string(arg->string);
	207	break;
	208	case JSON_TYPE_INTEGER:
	209	value = json_create_value_int(arg->integer_number);
	210	break;
	211	case JSON_TYPE_FLOAT:
	212	value = json_create_value_float(arg->float_number);
	213	break;
	214	case JSON_TYPE_OBJECT:
	215	value = json_create_value_object(arg->object);
	216	break;
	217	default:
	218	case JSON_TYPE_ARRAY:
	219	value = json_create_value_array(arg->array);
	220	break;
	221	}
cc372b17 SL	222
	223	if (!value)
	224	return ENOMEM;
	225
	226	pair = json_create_pair(name, value);
	227	if (!pair) {
	228	json_free_value(value);
	229	return ENOMEM;
	230	}
	231	ret = json_object_add_pair(obj, pair);
	232	if (ret) {
	233	json_free_pair(pair);
	234	return ENOMEM;
	235	}
	236	return 0;
	237	}
	238
eb2f29b7 BVA	239	int json_array_add_value_type(struct json_array *array,
eb2f29b7 BVA	240	const struct json_value *arg)
cc372b17 SL	241	{
cc372b17 SL	242	struct json_value *value;
cc372b17 SL	243	int ret;
cc372b17 SL	244
eb2f29b7 BVA	245	switch (arg->type) {
	246	case JSON_TYPE_STRING:
	247	value = json_create_value_string(arg->string);
	248	break;
	249	case JSON_TYPE_INTEGER:
	250	value = json_create_value_int(arg->integer_number);
	251	break;
	252	case JSON_TYPE_FLOAT:
	253	value = json_create_value_float(arg->float_number);
	254	break;
	255	case JSON_TYPE_OBJECT:
	256	value = json_create_value_object(arg->object);
	257	break;
	258	default:
	259	case JSON_TYPE_ARRAY:
	260	value = json_create_value_array(arg->array);
	261	break;
	262	}
cc372b17 SL	263
	264	if (!value)
	265	return ENOMEM;
	266
	267	ret = json_array_add_value(array, value);
	268	if (ret) {
	269	json_free_value(value);
	270	return ENOMEM;
	271	}
	272	return 0;
	273	}
	274
	275	static int json_value_level(struct json_value *value);
	276	static int json_pair_level(struct json_pair *pair);
	277	static int json_array_level(struct json_array *array);
	278	static int json_object_level(struct json_object *object)
	279	{
	280	if (object->parent == NULL)
	281	return 0;
	282	return json_value_level(object->parent);
	283	}
	284
	285	static int json_pair_level(struct json_pair *pair)
	286	{
	287	return json_object_level(pair->parent) + 1;
	288	}
	289
	290	static int json_array_level(struct json_array *array)
	291	{
	292	return json_value_level(array->parent);
	293	}
	294
	295	static int json_value_level(struct json_value *value)
	296	{
	297	if (value->parent_type == JSON_PARENT_TYPE_PAIR)
	298	return json_pair_level(value->parent_pair);
	299	else
	300	return json_array_level(value->parent_array) + 1;
	301	}
	302
a666cab8	303	static void json_print_level(int level, struct buf_output *out)
cc372b17 SL	304	{
cc372b17 SL	305	while (level-- > 0)
a666cab8	306	log_buf(out, " ");
cc372b17 SL	307	}
cc372b17 SL	308
a666cab8	309	static void json_print_pair(struct json_pair pair, struct buf_output );
a666cab8	310	static void json_print_value(struct json_value value, struct buf_output );
038d3e5b	311
a666cab8	312	void json_print_object(struct json_object obj, struct buf_output out)
cc372b17 SL	313	{
	314	int i;
	315
a666cab8	316	log_buf(out, "{\n");
cc372b17 SL	317	for (i = 0; i < obj->pair_cnt; i++) {
cc372b17 SL	318	if (i > 0)
a666cab8 JA	319	log_buf(out, ",\n");
a666cab8 JA	320	json_print_pair(obj->pairs[i], out);
cc372b17	321	}
a666cab8 JA	322	log_buf(out, "\n");
	323	json_print_level(json_object_level(obj), out);
	324	log_buf(out, "}");
cc372b17 SL	325	}
cc372b17 SL	326
a666cab8	327	static void json_print_pair(struct json_pair pair, struct buf_output out)
cc372b17	328	{
a666cab8 JA	329	json_print_level(json_pair_level(pair), out);
	330	log_buf(out, "\"%s\" : ", pair->name);
	331	json_print_value(pair->value, out);
cc372b17 SL	332	}
cc372b17 SL	333
a666cab8	334	static void json_print_array(struct json_array array, struct buf_output out)
cc372b17 SL	335	{
	336	int i;
	337
a666cab8	338	log_buf(out, "[\n");
cc372b17 SL	339	for (i = 0; i < array->value_cnt; i++) {
cc372b17 SL	340	if (i > 0)
a666cab8 JA	341	log_buf(out, ",\n");
	342	json_print_level(json_value_level(array->values[i]), out);
	343	json_print_value(array->values[i], out);
cc372b17	344	}
a666cab8 JA	345	log_buf(out, "\n");
	346	json_print_level(json_array_level(array), out);
	347	log_buf(out, "]");
cc372b17 SL	348	}
cc372b17 SL	349
a666cab8	350	static void json_print_value(struct json_value value, struct buf_output out)
cc372b17 SL	351	{
cc372b17 SL	352	switch (value->type) {
2374148c JA	353	case JSON_TYPE_STRING:
	354	log_buf(out, "\"%s\"", value->string);
	355	break;
cc372b17	356	case JSON_TYPE_INTEGER:
a666cab8	357	log_buf(out, "%lld", value->integer_number);
cc372b17 SL	358	break;
cc372b17 SL	359	case JSON_TYPE_FLOAT:
817720e1	360	log_buf(out, "%f", value->float_number);
cc372b17 SL	361	break;
cc372b17 SL	362	case JSON_TYPE_OBJECT:
a666cab8	363	json_print_object(value->object, out);
cc372b17 SL	364	break;
cc372b17 SL	365	case JSON_TYPE_ARRAY:
a666cab8	366	json_print_array(value->array, out);
cc372b17 SL	367	break;
	368	}
	369	}