Annotation of libelwix/src/json.c, revision 1.4

1.2       misho       1: /*************************************************************************
                      2: * (C) 2017 AITNET ltd - Sofia/Bulgaria - <misho@aitnet.org>
                      3: *  by Michael Pounov <misho@elwix.org>
                      4: *
                      5: * $Author: misho $
1.4     ! misho       6: * $Id: json.c,v 1.3.4.1 2018/03/07 12:26:38 misho Exp $
1.2       misho       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: 
1.4     ! misho      15: Copyright 2004 - 2018
1.2       misho      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
1.4     ! misho     490:  * @type = Search key for particular token type, if is J_UNDEF this mean any type
1.2       misho     491:  * @toks = Parsed tokens
                    492:  * @toksnum = Number of parsed tokens
                    493:  * return: =NULL error or !=NULL data token found 
                    494:  */
                    495: jtok_t *
1.4     ! misho     496: json_findbykey(const char *jstr, const char *key, jtype_t type, jtok_t * __restrict toks, int toksnum)
1.2       misho     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)) {
1.4     ! misho     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:                        }
1.2       misho     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;
1.3       misho     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;
1.2       misho    1043: }

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