File:  [ELWIX - Embedded LightWeight unIX -] / libelwix / src / json.c
Revision 1.4: download - view: text, annotated - select for diffs - revision graph
Wed Mar 7 12:29:28 2018 UTC (6 years, 2 months ago) by misho
Branches: MAIN
CVS tags: elwix4_9, elwix4_11, elwix4_10, HEAD, ELWIX4_9, ELWIX4_8, ELWIX4_10
ver 4.8

    1: /*************************************************************************
    2: * (C) 2017 AITNET ltd - Sofia/Bulgaria - <misho@aitnet.org>
    3: *  by Michael Pounov <misho@elwix.org>
    4: *
    5: * $Author: misho $
    6: * $Id: json.c,v 1.4 2018/03/07 12:29:28 misho Exp $
    7: *
    8: **************************************************************************
    9: The ELWIX and AITNET software is distributed under the following
   10: terms:
   11: 
   12: All of the documentation and software included in the ELWIX and AITNET
   13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
   14: 
   15: Copyright 2004 - 2018
   16: 	by Michael Pounov <misho@elwix.org>.  All rights reserved.
   17: 
   18: Redistribution and use in source and binary forms, with or without
   19: modification, are permitted provided that the following conditions
   20: are met:
   21: 1. Redistributions of source code must retain the above copyright
   22:    notice, this list of conditions and the following disclaimer.
   23: 2. Redistributions in binary form must reproduce the above copyright
   24:    notice, this list of conditions and the following disclaimer in the
   25:    documentation and/or other materials provided with the distribution.
   26: 3. All advertising materials mentioning features or use of this software
   27:    must display the following acknowledgement:
   28: This product includes software developed by Michael Pounov <misho@elwix.org>
   29: ELWIX - Embedded LightWeight unIX and its contributors.
   30: 4. Neither the name of AITNET nor the names of its contributors
   31:    may be used to endorse or promote products derived from this software
   32:    without specific prior written permission.
   33: 
   34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
   35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   37: ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
   38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
   39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
   40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
   41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
   42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
   43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
   44: SUCH DAMAGE.
   45: */
   46: #include "global.h"
   47: 
   48: 
   49: /* JSON error strings */
   50: const char *jerrstr[] = {
   51: 	"No error",
   52: 	"Not enough tokens were provided",
   53: 	"Invalid character",
   54: 	"JSON string isn't full",
   55: 	"Invalid parameter", 
   56: 	NULL
   57: };
   58: 
   59: 
   60: /*
   61:  * json_init() - Initialize JSON handler
   62:  *
   63:  * @json = JSON handler, if there is NULL then dynamically will be allocated
   64:  * @jstrict = JSON strict mode, when we select strict mode every unquoted value is error
   65:  * return: =NULL error or !=NULL ready for use JSON handler and should be free with json_free()
   66:  */
   67: json_t *
   68: json_init(json_t * __restrict json, int jstrict)
   69: {
   70: 	json_t *j = json;
   71: 
   72: 	if (!j) {
   73: 		j = e_malloc(sizeof(json_t));
   74: 		if (!j) {
   75: 			LOGERR;
   76: 			return NULL;
   77: 		}
   78: 	}
   79: 
   80: 	memset(j, 0, sizeof(json_t));
   81: 	j->h_parent = -1;
   82: 	j->h_strict = jstrict;
   83: 
   84: 	/* handler is dynamically allocated! */
   85: 	if (!json)
   86: 		j->h_alloc = j;
   87: 
   88: 	return j;
   89: }
   90: 
   91: /*
   92:  * json_free() - Free JSON handler
   93:  *
   94:  * @json = JSON handler
   95:  * return: none
   96:  */
   97: void
   98: json_free(json_t * __restrict json)
   99: {
  100: 	if (json) {
  101: 		if (json->h_alloc)
  102: 			e_free(json);
  103: 		else
  104: 			memset(json, 0, sizeof(json_t));
  105: 	}
  106: }
  107: 
  108: static jtok_t *
  109: json_gettoken(json_t * __restrict json, jtok_t * __restrict jtoks, u_int toksnum)
  110: {
  111: 	jtok_t *tok;
  112: 
  113: 	assert(json || !(!jtoks && toksnum));
  114: 
  115: 	if (json->h_next >= toksnum) {
  116: 		elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  117: 		return NULL;
  118: 	} else
  119: 		tok = &jtoks[json->h_next];
  120: 	tok->tok_idx = json->h_next++;
  121: 	tok->tok_start = tok->tok_end = tok->tok_parent = -1;
  122: 	tok->tok_size = 0;
  123: 
  124: 	return tok;
  125: }
  126: 
  127: inline void
  128: json_filltoken(jtok_t * __restrict tok, jtype_t type, long start, long end, long parent)
  129: {
  130: 	assert(tok);
  131: 
  132: 	tok->tok_type = type;
  133: 	tok->tok_start = start;
  134: 	tok->tok_end = end;
  135: 	tok->tok_parent = parent;
  136: 	tok->tok_size = 0;
  137: }
  138: 
  139: static int
  140: json_parse_string(json_t * __restrict json, const char *jstr, size_t jlen, jtok_t * __restrict jtoks, u_int toksnum)
  141: {
  142: 	jtok_t	*tok;
  143: 	u_long pos;
  144: 	char ch;
  145: 	register int i;
  146: 
  147: 	assert(json || jstr || !(!jtoks && toksnum));
  148: 
  149: 	for (pos = json->h_pos++; json->h_pos < jlen && jstr[json->h_pos]; json->h_pos++) {
  150: 		ch = jstr[json->h_pos];
  151: 
  152: 		if (ch == '\"') {
  153: 			if (!jtoks)
  154: 				return 0;
  155: 			if (!(tok = json_gettoken(json, jtoks, toksnum))) {
  156: 				json->h_pos = pos;
  157: 				return -1;
  158: 			} else
  159: 				json_filltoken(tok, J_STRING, pos + 1, json->h_pos, json->h_parent);
  160: 			return 0;
  161: 		}
  162: 
  163: 		if (ch == '\\' && json->h_pos + 1 < jlen) {
  164: 			switch (jstr[++json->h_pos]) {
  165: 				case '\"':
  166: 				case '/':
  167: 				case '\\':
  168: 				case 'b':
  169: 				case 'f':
  170: 				case 'r':
  171: 				case 'n':
  172: 				case 't':
  173: 					/* Allowed escaped symbols */
  174: 					break;
  175: 				case 'u':
  176: 					/* Allows escaped symbol \uXXXX */
  177: 					json->h_pos++;
  178: 					for (i = 0; i < 4 && json->h_pos < jlen && jstr[json->h_pos]; i++, json->h_pos++) {
  179: 						/* If it isn't a hex character we have an error */
  180: 						if (!((jstr[json->h_pos] >= 48 && jstr[json->h_pos] <= 57) || /* 0-9 */
  181: 								(jstr[json->h_pos] >= 65 && jstr[json->h_pos] <= 70) || /* A-F */
  182: 								(jstr[json->h_pos] >= 97 && jstr[json->h_pos] <= 102))) { /* a-f */
  183: 							json->h_pos = pos;
  184: 							elwix_SetErr(J_ERR_INVAL, "%s", jerrstr[J_ERR_INVAL]);
  185: 							return -1;
  186: 						}
  187: 					}
  188: 					json->h_pos--;
  189: 					break;
  190: 				default:
  191: 					/* Unexpected symbol */
  192: 					json->h_pos = pos;
  193: 					elwix_SetErr(J_ERR_INVAL, "%s", jerrstr[J_ERR_INVAL]);
  194: 					return -1;
  195: 			}
  196: 		}
  197: 	}
  198: 
  199: 	json->h_pos = pos;
  200: 	elwix_SetErr(J_ERR_PART, "%s", jerrstr[J_ERR_PART]);
  201: 	return -1;
  202: }
  203: 
  204: static int
  205: json_parse_value(json_t * __restrict json, const char *jstr, size_t jlen, jtok_t * __restrict jtoks, u_int toksnum)
  206: {
  207: 	jtok_t	*tok;
  208: 	u_long pos;
  209: 
  210: 	assert(json || jstr || !(!jtoks && toksnum));
  211: 
  212: 	for (pos = json->h_pos; json->h_pos < jlen && jstr[json->h_pos]; json->h_pos++) {
  213: 		switch (jstr[json->h_pos]) {
  214: 			case ':':
  215: 				if (json->h_strict)
  216: 					goto found;
  217: 				break;
  218: 			case '\t':
  219: 			case '\r':
  220: 			case '\n':
  221: 			case ' ':
  222: 			case ',':
  223: 			case ']':
  224: 			case '}':
  225: 				goto found;
  226: 		}
  227: 		if (jstr[json->h_pos] < 32 || jstr[json->h_pos] > 127) {
  228: 			json->h_pos = pos;
  229: 			elwix_SetErr(J_ERR_INVAL, "%s", jerrstr[J_ERR_INVAL]);
  230: 			return -1;
  231: 		}
  232: 	}
  233: 
  234: 	if (json->h_strict) {
  235: 		json->h_pos = pos;
  236: 		elwix_SetErr(J_ERR_PART, "%s", jerrstr[J_ERR_PART]);
  237: 		return -1;
  238: 	}
  239: found:
  240: 	if (jtoks) {
  241: 		if (!(tok = json_gettoken(json, jtoks, toksnum))) {
  242: 			json->h_pos = pos;
  243: 			return -1;
  244: 		} else
  245: 			json_filltoken(tok, J_VALUE, pos, json->h_pos, json->h_parent);
  246: 	}
  247: 
  248: 	json->h_pos--;
  249: 	return 0;
  250: }
  251: 
  252: /*
  253:  * json_parse() - Parse JSON string
  254:  *
  255:  * @json = JSON handler
  256:  * @jstr = JSON string
  257:  * @jlen = JSON string length
  258:  * @jtoks = Token array
  259:  * @toksnum = Token array size, return number of allocated tokens in array
  260:  * return: -1 error or number of found tokens 
  261:  */
  262: u_int
  263: json_parse(json_t * __restrict json, const char *jstr, size_t jlen, jtok_t * __restrict jtoks, u_int toksnum)
  264: {
  265: 	register int i;
  266: 	register u_int cx;
  267: 	jtype_t type;
  268: 	jtok_t *tok;
  269: 	char ch;
  270: 
  271: 	if (!json || !jstr || (!jtoks && toksnum)) {
  272: 		elwix_SetErr(J_ERR_PARAM, "%s", jerrstr[J_ERR_PARAM]);
  273: 		return (u_int) -1;
  274: 	}
  275: 
  276: 	for (cx = json->h_next; json->h_pos < jlen && jstr[json->h_pos]; json->h_pos++) {
  277: 		switch ((ch = jstr[json->h_pos])) {
  278: 			case '{':
  279: 			case '[':
  280: 				cx++;	/* start new object/array token */
  281: 				if (!jtoks)
  282: 					break;
  283: 
  284: 				tok = json_gettoken(json, jtoks, toksnum);
  285: 				if (!tok)
  286: 					return (u_int) -1;
  287: 				if (json->h_parent != -1) {
  288: 					jtoks[json->h_parent].tok_size++;
  289: 					tok->tok_parent = json->h_parent;
  290: 				}
  291: 				tok->tok_type = (ch == '{' ? J_OBJECT : J_ARRAY);
  292: 				tok->tok_start = json->h_pos;
  293: 				json->h_parent = json->h_next - 1;
  294: 				break;
  295: 			case '}':
  296: 			case ']':
  297: 				if (!jtoks)
  298: 					break;
  299: 
  300: 				if (json->h_next < 1) {
  301: 					elwix_SetErr(J_ERR_INVAL, "%s", jerrstr[J_ERR_INVAL]);
  302: 					return (u_int) -1;
  303: 				}
  304: 
  305: 				type = (ch == '}' ? J_OBJECT : J_ARRAY);
  306: 				tok = &jtoks[json->h_next - 1];
  307: 				while (42) {
  308: 					if (tok->tok_start != -1 && tok->tok_end == -1) {
  309: 						if (tok->tok_type != type) {
  310: 							elwix_SetErr(J_ERR_INVAL, "%s", jerrstr[J_ERR_INVAL]);
  311: 							return (u_int) -1;
  312: 						}
  313: 						tok->tok_end = json->h_pos + 1;
  314: 						json->h_parent = tok->tok_parent;
  315: 						break;
  316: 					}
  317: 					if (tok->tok_parent == -1) {
  318: 						if (tok->tok_type != type || json->h_parent == -1) {
  319: 							elwix_SetErr(J_ERR_INVAL, "%s", jerrstr[J_ERR_INVAL]);
  320: 							return (u_int) -1;
  321: 						}
  322: 						break;
  323: 					}
  324: 					tok = &jtoks[tok->tok_parent];
  325: 				}
  326: 				break;
  327: 			case '\"':
  328: 				if (json_parse_string(json, jstr, jlen, jtoks, toksnum) == -1)
  329: 					return (u_int) -1;
  330: 				cx++;	/* start new string token */
  331: 				if (jtoks && json->h_parent != -1)
  332: 					jtoks[json->h_parent].tok_size++;
  333: 				break;
  334: 			case '\t':
  335: 			case '\r':
  336: 			case '\n':
  337: 			case ' ':
  338: 				/* whitespace, skip */
  339: 				break;
  340: 			case ':':
  341: 				json->h_parent = json->h_next - 1;
  342: 				break;
  343: 			case ',':
  344: 				if (jtoks && json->h_parent != -1 && 
  345: 						jtoks[json->h_parent].tok_type != J_OBJECT && 
  346: 						jtoks[json->h_parent].tok_type != J_ARRAY)
  347: 					json->h_parent = jtoks[json->h_parent].tok_parent;
  348: 				break;
  349: 			case '-':
  350: 			case '0':
  351: 			case '1':
  352: 			case '2':
  353: 			case '3':
  354: 			case '4':
  355: 			case '5':
  356: 			case '6':
  357: 			case '7':
  358: 			case '8':
  359: 			case '9':
  360: 			case 't':
  361: 			case 'f':
  362: 			case 'n':
  363: 				if (json->h_strict) {
  364: 					if (jtoks && json->h_parent != -1) {
  365: 						/* they must not be keys of the object */
  366: 						if (jtoks[json->h_parent].tok_type == J_OBJECT || 
  367: 								(jtoks[json->h_parent].tok_type == J_STRING && 
  368: 								 jtoks[json->h_parent].tok_size)) {
  369: 							elwix_SetErr(J_ERR_INVAL, "%s", jerrstr[J_ERR_INVAL]);
  370: 							return (u_int) -1;
  371: 						}
  372: 					}
  373: 
  374: 					if (json_parse_value(json, jstr, jlen, jtoks, toksnum) == -1)
  375: 						return (u_int) -1;
  376: 					cx++;	/* start new value token */
  377: 					if (jtoks && json->h_parent != -1)
  378: 						jtoks[json->h_parent].tok_size++;
  379: 					break;
  380: 				}
  381: 			default:
  382: 				if (json->h_strict) {
  383: 					elwix_SetErr(J_ERR_INVAL, "%s", jerrstr[J_ERR_INVAL]);
  384: 					return (u_int) -1;
  385: 				}
  386: 
  387: 				if (json_parse_value(json, jstr, jlen, jtoks, toksnum) == -1)
  388: 					return (u_int) -1;
  389: 				cx++;	/* start new value token */
  390: 				if (jtoks && json->h_parent != -1)
  391: 					jtoks[json->h_parent].tok_size++;
  392: 				break;
  393: 		}
  394: 	}
  395: 
  396: 	if (jtoks) {
  397: 		for (i = json->h_next - 1; i >= 0; i--) {
  398: 			/* unmatched opened object or array */
  399: 			if (jtoks[i].tok_start != -1 && jtoks[i].tok_end == -1) {
  400: 				elwix_SetErr(J_ERR_PART, "%s", jerrstr[J_ERR_PART]);
  401: 				return (u_int) -1;
  402: 			}
  403: 		}
  404: 	} else
  405: 		cx++;	/* increment needed tokens number for termination empty token */
  406: 
  407: 	return cx;
  408: }
  409: 
  410: /*
  411:  * json_token2val() - Return token to AIT variable
  412:  *
  413:  * @jstr = JSON string
  414:  * @tok = Token for convert
  415:  * @return =NULL error or !=NULL allocated variable, after use should be ait_freeVar()
  416:  */
  417: ait_val_t *
  418: json_token2val(const char *jstr, jtok_t * __restrict tok)
  419: {
  420: 	ait_val_t *v = NULL;
  421: 
  422: 	if (!jstr || !tok)
  423: 		return NULL;
  424: 
  425: 	v = ait_allocVar();
  426: 	if (!v)
  427: 		return NULL;
  428: 
  429: 	AIT_SET_STRSIZ(v, json_toklen(tok));
  430: 	strncpy(AIT_GET_STR(v), json_tokstr(jstr, tok), AIT_LEN(v) - 1);
  431: 
  432: 	return v;
  433: }
  434: 
  435: /*
  436:  * json_token2str() - Return token to string
  437:  *
  438:  * @jstr = JSON string
  439:  * @tok = Token for convert
  440:  * @return =NULL error or !=NULL allocated str, after use should be e_free()
  441:  */
  442: char *
  443: json_token2str(const char *jstr, jtok_t * __restrict tok)
  444: {
  445: 	char *str = NULL;
  446: 	size_t len;
  447: 
  448: 	if (!jstr || !tok)
  449: 		return NULL;
  450: 
  451: 	len = json_toklen(tok);
  452: 	str = e_malloc(len + 1);
  453: 	if (!str)
  454: 		return NULL;
  455: 	else {
  456: 		strncpy(str, json_tokstr(jstr, tok), len);
  457: 		str[len] = 0;
  458: 	}
  459: 
  460: 	return str;
  461: }
  462: 
  463: /*
  464:  * json_token2num() - Return token to numeric
  465:  *
  466:  * @jstr = JSON string
  467:  * @tok = Token for convert
  468:  * @return number
  469:  */
  470: long
  471: json_token2num(const char *jstr, jtok_t * __restrict tok)
  472: {
  473: 	long ret = 0;
  474: 	char *str;
  475: 
  476: 	str = json_token2str(jstr, tok);
  477: 	if (!str)
  478: 		return 0;
  479: 
  480: 	ret = strtol(str, NULL, 0);
  481: 	e_free(str);
  482: 	return ret;
  483: }
  484: 
  485: /*
  486:  * json_findbykey() - Find data by key
  487:  *
  488:  * @jstr = JSON string
  489:  * @key = Search key
  490:  * @type = Search key for particular token type, if is J_UNDEF this mean any type
  491:  * @toks = Parsed tokens
  492:  * @toksnum = Number of parsed tokens
  493:  * return: =NULL error or !=NULL data token found 
  494:  */
  495: jtok_t *
  496: json_findbykey(const char *jstr, const char *key, jtype_t type, jtok_t * __restrict toks, int toksnum)
  497: {
  498: 	jtok_t *tok = NULL;
  499: 	register int i;
  500: 	int klen;
  501: 
  502: 	if (!jstr || !key || !toks)
  503: 		return NULL;
  504: 	else
  505: 		klen = strlen(key);
  506: 
  507: 	for (i = 1; i < toksnum; i++) {
  508: 		if (toks[i].tok_type == J_STRING && 
  509: 				klen == toks[i].tok_end - toks[i].tok_start && 
  510: 				!strncmp(jstr + toks[i].tok_start, key, klen)) {
  511: 			if (type != J_UNDEF) {
  512: 				if (toks[i + 1].tok_type == type) {
  513: 					tok = toks + i + 1;
  514: 					break;
  515: 				}
  516: 			} else {
  517: 				tok = toks + i + 1;
  518: 				break;
  519: 			}
  520: 		}
  521: 	}
  522: 
  523: 	return tok;
  524: }
  525: 
  526: /*
  527:  * json_token2array() - Convert token to string array
  528:  *
  529:  * @jstr = JSON string
  530:  * @tok = Token for convert
  531:  * return: =NULL error or !=NULL allocated array with string variables, 
  532:  * 		after use should be ait_freeVars()
  533:  */
  534: array_t *
  535: json_token2array(const char *jstr, jtok_t * __restrict tok)
  536: {
  537: 	array_t *arr = NULL;
  538: 	register int i;
  539: 	int siz;
  540: 	ait_val_t *v;
  541: 	jtok_t *t;
  542: 
  543: 	if (!jstr || !tok)
  544: 		return NULL;
  545: 
  546: 	siz = tok->tok_size;
  547: 	if (!siz && json_toktype(tok) != J_ARRAY && json_toktype(tok) != J_OBJECT)
  548: 		siz++;
  549: 
  550: 	arr = ait_allocVars(siz);
  551: 	if (!arr)
  552: 		return NULL;
  553: 
  554: 	if (tok->tok_type == J_STRING || tok->tok_type == J_VALUE) {
  555: 		v = ait_getVars(&arr, 0);
  556: 		AIT_SET_STRSIZ(v, json_toklen(tok) + 1);
  557: 		json_tokstrcpy(AIT_GET_STR(v), jstr, tok);
  558: 	} else if (tok->tok_type == J_ARRAY) {
  559: 		for (i = 0; i < tok->tok_size; i++) {
  560: 			t = &tok[i + 1];
  561: 			v = ait_getVars(&arr, i);
  562: 			AIT_SET_STRSIZ(v, json_toklen(t) + 1);
  563: 			json_tokstrcpy(AIT_GET_STR(v), jstr, t);
  564: 		}
  565: 	} else if (tok->tok_type == J_OBJECT) {
  566: 		for (i = 0; tok->tok_idx <= tok[i + 1].tok_parent; i++) {
  567: 			t = &tok[i + 1];
  568: 			v = ait_getVars(&arr, i);
  569: 			AIT_SET_STRSIZ(v, json_toklen(t) + 1);
  570: 			json_tokstrcpy(AIT_GET_STR(v), jstr, t);
  571: 		}
  572: 	} else {
  573: 		elwix_SetErr(J_ERR_PARAM, "%s", jerrstr[J_ERR_PARAM]);
  574: 		ait_freeVars(&arr);
  575: 		return NULL;
  576: 	}
  577: 
  578: 	return arr;
  579: }
  580: 
  581: 
  582: 
  583: /*
  584:  * json_add_begin_object() - Adds begin of object {
  585:  *
  586:  * @jstr = JSON string
  587:  * @jlen = JSON string length
  588:  * @wspace = whitespace include
  589:  * return: -1 error or !=-1 actual JSON string length
  590:  */
  591: int
  592: json_add_begin_object(char * __restrict jstr, int jlen, int wspace)
  593: {
  594: 	int len;
  595: 	size_t eos;
  596: 
  597: 	if (!jstr)
  598: 		return -1;
  599: 	else
  600: 		eos = strlen(jstr);
  601: 
  602: 
  603: 	if (wspace)
  604: 		len = strlcat(jstr, "{ ", jlen);
  605: 	else
  606: 		len = strlcat(jstr, "{", jlen);
  607: 
  608: 	if (len >= jlen) {
  609: 		elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  610: 		jstr[eos] = 0;
  611: 		return -1;
  612: 	}
  613: 
  614: 	return len;
  615: }
  616: 
  617: /*
  618:  * json_add_end_object() - Adds end of object }
  619:  *
  620:  * @jstr = JSON string
  621:  * @jlen = JSON string length
  622:  * @wspace = whitespace include
  623:  * return: -1 error or !=-1 actual JSON string length
  624:  */
  625: int
  626: json_add_end_object(char * __restrict jstr, int jlen, int wspace)
  627: {
  628: 	int len;
  629: 	size_t eos;
  630: 
  631: 	if (!jstr)
  632: 		return -1;
  633: 	else
  634: 		eos = strlen(jstr);
  635: 
  636: 	if (wspace)
  637: 		len = strlcat(jstr, " }", jlen);
  638: 	else
  639: 		len = strlcat(jstr, "}", jlen);
  640: 
  641: 	if (len >= jlen) {
  642: 		elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  643: 		jstr[eos] = 0;
  644: 		return -1;
  645: 	}
  646: 
  647: 	return len;
  648: }
  649: 
  650: /*
  651:  * json_add_begin_array() - Adds begin of array [
  652:  *
  653:  * @jstr = JSON string
  654:  * @jlen = JSON string length
  655:  * @wspace = whitespace include
  656:  * return: -1 error or !=-1 actual JSON string length
  657:  */
  658: int
  659: json_add_begin_array(char * __restrict jstr, int jlen, int wspace)
  660: {
  661: 	int len;
  662: 	size_t eos;
  663: 
  664: 	if (!jstr)
  665: 		return -1;
  666: 	else
  667: 		eos = strlen(jstr);
  668: 
  669: 	if (wspace)
  670: 		len = strlcat(jstr, "[ ", jlen);
  671: 	else
  672: 		len = strlcat(jstr, "[", jlen);
  673: 
  674: 	if (len >= jlen) {
  675: 		elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  676: 		jstr[eos] = 0;
  677: 		return -1;
  678: 	}
  679: 
  680: 	return len;
  681: }
  682: 
  683: /*
  684:  * json_add_end_array() - Adds end of array ]
  685:  *
  686:  * @jstr = JSON string
  687:  * @jlen = JSON string length
  688:  * @wspace = whitespace include
  689:  * return: -1 error or !=-1 actual JSON string length
  690:  */
  691: int
  692: json_add_end_array(char * __restrict jstr, int jlen, int wspace)
  693: {
  694: 	int len;
  695: 	size_t eos;
  696: 
  697: 	if (!jstr)
  698: 		return -1;
  699: 	else
  700: 		eos = strlen(jstr);
  701: 
  702: 	if (wspace)
  703: 		len = strlcat(jstr, " ]", jlen);
  704: 	else
  705: 		len = strlcat(jstr, "]", jlen);
  706: 
  707: 	if (len >= jlen) {
  708: 		elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  709: 		jstr[eos] = 0;
  710: 		return -1;
  711: 	}
  712: 
  713: 	return len;
  714: }
  715: 
  716: /*
  717:  * json_add_char() - Adds character
  718:  *
  719:  * @jstr = JSON string
  720:  * @jlen = JSON string length
  721:  * @ch = Character
  722:  * return: -1 error or !=-1 actual JSON string length
  723:  */
  724: int
  725: json_add_char(char * __restrict jstr, int jlen, u_char ch)
  726: {
  727: 	int len;
  728: 
  729: 	if (!jstr)
  730: 		return -1;
  731: 
  732: 	len = strlen(jstr) + 1;
  733: 	if (len >= jlen) {
  734: 		elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  735: 		return -1;
  736: 	} else {
  737: 		jstr[len++] = (char) ch;
  738: 		jstr[len] = 0;
  739: 	}
  740: 
  741: 	return len;
  742: }
  743: 
  744: /*
  745:  * json_add_colon() - Adds key/value pair delimiter colon :
  746:  *
  747:  * @jstr = JSON string
  748:  * @jlen = JSON string length
  749:  * @wspace = whitespace include
  750:  * return: -1 error or !=-1 actual JSON string length
  751:  */
  752: int
  753: json_add_colon(char * __restrict jstr, int jlen, int wspace)
  754: {
  755: 	int len;
  756: 	size_t eos;
  757: 
  758: 	if (!jstr)
  759: 		return -1;
  760: 	else
  761: 		eos = strlen(jstr);
  762: 
  763: 	if (wspace)
  764: 		len = strlcat(jstr, ": ", jlen);
  765: 	else
  766: 		len = strlcat(jstr, ":", jlen);
  767: 
  768: 	if (len >= jlen) {
  769: 		elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  770: 		jstr[eos] = 0;
  771: 		return -1;
  772: 	}
  773: 
  774: 	return len;
  775: }
  776: 
  777: /*
  778:  * json_add_comma() - Adds value delimiter comma ,
  779:  *
  780:  * @jstr = JSON string
  781:  * @jlen = JSON string length
  782:  * @wspace = whitespace include
  783:  * return: -1 error or !=-1 actual JSON string length
  784:  */
  785: int
  786: json_add_comma(char * __restrict jstr, int jlen, int wspace)
  787: {
  788: 	int len;
  789: 	size_t eos;
  790: 
  791: 	if (!jstr)
  792: 		return -1;
  793: 	else
  794: 		eos = strlen(jstr);
  795: 
  796: 	if (wspace)
  797: 		len = strlcat(jstr, ", ", jlen);
  798: 	else
  799: 		len = strlcat(jstr, ",", jlen);
  800: 
  801: 	if (len >= jlen) {
  802: 		elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  803: 		jstr[eos] = 0;
  804: 		return -1;
  805: 	}
  806: 
  807: 	return len;
  808: }
  809: 
  810: /*
  811:  * json_add_string() - Adds string
  812:  *
  813:  * @jstr = JSON string
  814:  * @jlen = JSON string length
  815:  * @unquot = Unquoted string
  816:  * @str = String, it can't be NULL
  817:  * return: -1 error or !=-1 actual JSON string length
  818:  */
  819: int
  820: json_add_string(char * __restrict jstr, int jlen, int unquot, const char *str)
  821: {
  822: 	int len;
  823: 	size_t eos;
  824: 
  825: 	if (!jstr || !str)
  826: 		return -1;
  827: 	else
  828: 		eos = strlen(jstr);
  829: 
  830: 	if (!unquot) {
  831: 		len = strlcat(jstr, "\"", jlen);
  832: 		if (len >= jlen) {
  833: 			elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  834: 			jstr[eos] = 0;
  835: 			return -1;
  836: 		}
  837: 	}
  838: 	len = strlcat(jstr, str, jlen);
  839: 	if (len >= jlen) {
  840: 		elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  841: 		jstr[eos] = 0;
  842: 		return -1;
  843: 	}
  844: 	if (!unquot) {
  845: 		len = strlcat(jstr, "\"", jlen);
  846: 		if (len >= jlen) {
  847: 			elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  848: 			jstr[eos] = 0;
  849: 			return -1;
  850: 		}
  851: 	}
  852: 
  853: 	return len;
  854: }
  855: 
  856: /*
  857:  * json_add_value() - Adds value
  858:  *
  859:  * @jstr = JSON string
  860:  * @jlen = JSON string length
  861:  * @unquot = Unquoted number
  862:  * @num = Number
  863:  * return: -1 error or !=-1 actual JSON string length
  864:  */
  865: int
  866: json_add_value(char * __restrict jstr, int jlen, int unquot, long num)
  867: {
  868: 	int len;
  869: 	char wrk[STRSIZ] = { [0 ... STRSIZ - 1] = 0 };
  870: 	size_t eos;
  871: 
  872: 	if (!jstr)
  873: 		return -1;
  874: 	else
  875: 		eos = strlen(jstr);
  876: 
  877: 	if (!unquot) {
  878: 		len = strlcat(jstr, "\"", jlen);
  879: 		if (len >= jlen) {
  880: 			elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  881: 			jstr[eos] = 0;
  882: 			return -1;
  883: 		}
  884: 	}
  885: 	snprintf(wrk, sizeof wrk, "%ld", num);
  886: 	len = strlcat(jstr, wrk, jlen);
  887: 	if (len >= jlen) {
  888: 		elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  889: 		jstr[eos] = 0;
  890: 		return -1;
  891: 	}
  892: 	if (!unquot) {
  893: 		len = strlcat(jstr, "\"", jlen);
  894: 		if (len >= jlen) {
  895: 			elwix_SetErr(J_ERR_NOMEM, "%s", jerrstr[J_ERR_NOMEM]);
  896: 			jstr[eos] = 0;
  897: 			return -1;
  898: 		}
  899: 	}
  900: 
  901: 	return len;
  902: }
  903: 
  904: /*
  905:  * json_add_pair() - Adds key/value pair
  906:  *
  907:  * @jstr = JSON string
  908:  * @jlen = JSON string length
  909:  * @wspace = whitespace include
  910:  * @key = Key string
  911:  * @val = Value string
  912:  * return: -1 error or !=-1 actual JSON string length
  913:  */
  914: int
  915: json_add_pair(char * __restrict jstr, int jlen, int wspace, const char *key, const char *val)
  916: {
  917: 	int len = -1;
  918: 	size_t eos;
  919: 
  920: 	if (!jstr || !key || !val)
  921: 		return -1;
  922: 	else
  923: 		eos = strlen(jstr);
  924: 
  925: 	if (json_add_string(jstr, jlen, 0, key) == -1) {
  926: 		jstr[eos] = 0;
  927: 		return -1;
  928: 	}
  929: 	if (json_add_colon(jstr, jlen, wspace) == -1) {
  930: 		jstr[eos] = 0;
  931: 		return -1;
  932: 	}
  933: 	if ((len = json_add_string(jstr, jlen, 0, val)) == -1) {
  934: 		jstr[eos] = 0;
  935: 		return -1;
  936: 	}
  937: 
  938: 	return len;
  939: }
  940: 
  941: /*
  942:  * json_add_array() - Adds array
  943:  *
  944:  * @jstr = JSON string
  945:  * @jlen = JSON string length
  946:  * @wspace = whitespace include
  947:  * @arr = Array with variables
  948:  * return: -1 error or !=-1 actual JSON string length
  949:  */
  950: int
  951: json_add_array(char * __restrict jstr, int jlen, int wspace, array_t * __restrict arr)
  952: {
  953: 	int len = -1;
  954: 	register int i;
  955: 	ait_val_t *v;
  956: 	size_t eos;
  957: 
  958: 	if (!jstr || !arr)
  959: 		return -1;
  960: 	else
  961: 		eos = strlen(jstr);
  962: 
  963: 	if (json_add_begin_array(jstr, jlen, wspace) == -1) {
  964: 		jstr[eos] = 0;
  965: 		return -1;
  966: 	}
  967: 	for (i = 0; i < array_Size(arr); i++) {
  968: 		v = array(arr, i, ait_val_t*);
  969: 		if (v) {
  970: 			if (AIT_TYPE(v) == string) {
  971: 				if (json_add_string(jstr, jlen, 0, AIT_GET_STR(v)) == -1) {
  972: 					jstr[eos] = 0;
  973: 					return -1;
  974: 				}
  975: 			} else {
  976: 				if (json_add_value(jstr, jlen, 0, AIT_GET_LIKE(v, long)) == -1) {
  977: 					jstr[eos] = 0;
  978: 					return -1;
  979: 				}
  980: 			}
  981: 			if (i < array_Size(arr) - 1 && json_add_comma(jstr, jlen, wspace) == -1) {
  982: 				jstr[eos] = 0;
  983: 				return -1;
  984: 			}
  985: 		}
  986: 	}
  987: 	if ((len = json_add_end_array(jstr, jlen, wspace)) == -1) {
  988: 		jstr[eos] = 0;
  989: 		return -1;
  990: 	}
  991: 
  992: 	return len;
  993: }
  994: 
  995: /*
  996:  * json_dump_yaml() - Dump parsed JSON string to YAML format
  997:  *
  998:  * @f = Output handler
  999:  * @jstr = JSON string
 1000:  * @toks = JSON tokens
 1001:  * @toksnum = Number of tokens
 1002:  * @indent = Start indent spaces
 1003:  * return: 0 done and 1 added one more item
 1004:  */
 1005: int
 1006: json_dump_yaml(FILE *f, const char *jstr, jtok_t *toks, int toksnum, int indent)
 1007: {
 1008: 	register int i, j, k;
 1009: 
 1010: 	if (!toksnum)
 1011: 		return 0;
 1012: 
 1013: 	if (toks->tok_type == J_VALUE) {
 1014: 		fprintf(f, "%.*s", (int) json_toklen(toks), json_tokstr(jstr, toks));
 1015: 		return 1;
 1016: 	} else if (toks->tok_type == J_STRING) {
 1017: 		fprintf(f, "%.*s", (int) json_toklen(toks), json_tokstr(jstr, toks));
 1018: 		return 1;
 1019: 	} else if (toks->tok_type == J_OBJECT) {
 1020: 		fprintf(f, "\n");
 1021: 		for (j = i = 0; i < json_toksize(toks); i++) {
 1022: 			for (k = 0; k < indent; k++)
 1023: 				fprintf(f, "  ");
 1024: 			j += json_dump_yaml(f, jstr, toks + j + 1, toksnum - j, indent + 1);
 1025: 			fprintf(f, ": ");
 1026: 			j += json_dump_yaml(f, jstr, toks + j + 1, toksnum - j, indent + 1);
 1027: 			fprintf(f, "\n");
 1028: 		}
 1029: 		return j + 1;
 1030: 	} else if (toks->tok_type == J_ARRAY) {
 1031: 		fprintf(f, "\n");
 1032: 		for (j = i = 0; i < json_toksize(toks); i++) {
 1033: 			for (k = 0; k < indent - 1; k++)
 1034: 				fprintf(f, "  ");
 1035: 			fprintf(f, "   - ");
 1036: 			j += json_dump_yaml(f, jstr, toks + j + 1, toksnum - j, indent + 1);
 1037: 			fprintf(f, "\n");
 1038: 		}
 1039: 		return j + 1;
 1040: 	}
 1041: 
 1042: 	return 0;
 1043: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>