LCOV - differential code coverage report
Current view: top level - src/common - jsonapi.c (source / functions) Coverage Total Hit UBC CBC
Current: c70b6db34ffeab48beef1fb4ce61bcad3772b8dd vs 06473f5a344df8c9594ead90a609b86f6724cff8 Lines: 89.2 % 963 859 104 859
Current Date: 2025-09-06 07:49:51 +0900 Functions: 100.0 % 31 31 31
Baseline: lcov-20250906-005545-baseline Branches: 77.6 % 781 606 175 606
Baseline Date: 2025-09-05 08:21:35 +0100 Line coverage date bins:
Legend: Lines:     hit not hit
Branches: + taken - not taken # not executed
(30,360] days: 77.0 % 187 144 43 144
(360..) days: 92.1 % 776 715 61 715
Function coverage date bins:
(30,360] days: 100.0 % 2 2 2
(360..) days: 100.0 % 29 29 29
Branch coverage date bins:
(30,360] days: 57.6 % 170 98 72 98
(360..) days: 83.1 % 611 508 103 508

 Age         Owner                    Branch data    TLA  Line data    Source code
                                  1                 :                : /*-------------------------------------------------------------------------
                                  2                 :                :  *
                                  3                 :                :  * jsonapi.c
                                  4                 :                :  *      JSON parser and lexer interfaces
                                  5                 :                :  *
                                  6                 :                :  * Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group
                                  7                 :                :  * Portions Copyright (c) 1994, Regents of the University of California
                                  8                 :                :  *
                                  9                 :                :  * IDENTIFICATION
                                 10                 :                :  *    src/common/jsonapi.c
                                 11                 :                :  *
                                 12                 :                :  *-------------------------------------------------------------------------
                                 13                 :                :  */
                                 14                 :                : #ifndef FRONTEND
                                 15                 :                : #include "postgres.h"
                                 16                 :                : #else
                                 17                 :                : #include "postgres_fe.h"
                                 18                 :                : #endif
                                 19                 :                : 
                                 20                 :                : #include "common/jsonapi.h"
                                 21                 :                : #include "mb/pg_wchar.h"
                                 22                 :                : #include "port/pg_lfind.h"
                                 23                 :                : 
                                 24                 :                : #ifdef JSONAPI_USE_PQEXPBUFFER
                                 25                 :                : #include "pqexpbuffer.h"
                                 26                 :                : #else
                                 27                 :                : #include "lib/stringinfo.h"
                                 28                 :                : #include "miscadmin.h"
                                 29                 :                : #endif
                                 30                 :                : 
                                 31                 :                : /*
                                 32                 :                :  * By default, we will use palloc/pfree along with StringInfo.  In libpq,
                                 33                 :                :  * use malloc and PQExpBuffer, and return JSON_OUT_OF_MEMORY on out-of-memory.
                                 34                 :                :  */
                                 35                 :                : #ifdef JSONAPI_USE_PQEXPBUFFER
                                 36                 :                : 
                                 37                 :                : #define STRDUP(s) strdup(s)
                                 38                 :                : #define ALLOC(size) malloc(size)
                                 39                 :                : #define ALLOC0(size) calloc(1, size)
                                 40                 :                : #define REALLOC realloc
                                 41                 :                : #define FREE(s) free(s)
                                 42                 :                : 
                                 43                 :                : #define jsonapi_appendStringInfo            appendPQExpBuffer
                                 44                 :                : #define jsonapi_appendBinaryStringInfo      appendBinaryPQExpBuffer
                                 45                 :                : #define jsonapi_appendStringInfoChar        appendPQExpBufferChar
                                 46                 :                : /* XXX should we add a macro version to PQExpBuffer? */
                                 47                 :                : #define jsonapi_appendStringInfoCharMacro   appendPQExpBufferChar
                                 48                 :                : #define jsonapi_makeStringInfo              createPQExpBuffer
                                 49                 :                : #define jsonapi_initStringInfo              initPQExpBuffer
                                 50                 :                : #define jsonapi_resetStringInfo             resetPQExpBuffer
                                 51                 :                : #define jsonapi_termStringInfo              termPQExpBuffer
                                 52                 :                : #define jsonapi_destroyStringInfo           destroyPQExpBuffer
                                 53                 :                : 
                                 54                 :                : #else                           /* !JSONAPI_USE_PQEXPBUFFER */
                                 55                 :                : 
                                 56                 :                : #define STRDUP(s) pstrdup(s)
                                 57                 :                : #define ALLOC(size) palloc(size)
                                 58                 :                : #define ALLOC0(size) palloc0(size)
                                 59                 :                : #define REALLOC repalloc
                                 60                 :                : 
                                 61                 :                : #ifdef FRONTEND
                                 62                 :                : #define FREE pfree
                                 63                 :                : #else
                                 64                 :                : /*
                                 65                 :                :  * Backend pfree() doesn't handle NULL pointers like the frontend's does; smooth
                                 66                 :                :  * that over to reduce mental gymnastics. Avoid multiple evaluation of the macro
                                 67                 :                :  * argument to avoid future hair-pulling.
                                 68                 :                :  */
                                 69                 :                : #define FREE(s) do {    \
                                 70                 :                :     void *__v = (s);    \
                                 71                 :                :     if (__v)            \
                                 72                 :                :         pfree(__v);     \
                                 73                 :                : } while (0)
                                 74                 :                : #endif
                                 75                 :                : 
                                 76                 :                : #define jsonapi_appendStringInfo            appendStringInfo
                                 77                 :                : #define jsonapi_appendBinaryStringInfo      appendBinaryStringInfo
                                 78                 :                : #define jsonapi_appendStringInfoChar        appendStringInfoChar
                                 79                 :                : #define jsonapi_appendStringInfoCharMacro   appendStringInfoCharMacro
                                 80                 :                : #define jsonapi_makeStringInfo              makeStringInfo
                                 81                 :                : #define jsonapi_initStringInfo              initStringInfo
                                 82                 :                : #define jsonapi_resetStringInfo             resetStringInfo
                                 83                 :                : #define jsonapi_termStringInfo(s)           pfree((s)->data)
                                 84                 :                : #define jsonapi_destroyStringInfo           destroyStringInfo
                                 85                 :                : 
                                 86                 :                : #endif                          /* JSONAPI_USE_PQEXPBUFFER */
                                 87                 :                : 
                                 88                 :                : /*
                                 89                 :                :  * The context of the parser is maintained by the recursive descent
                                 90                 :                :  * mechanism, but is passed explicitly to the error reporting routine
                                 91                 :                :  * for better diagnostics.
                                 92                 :                :  */
                                 93                 :                : typedef enum                    /* contexts of JSON parser */
                                 94                 :                : {
                                 95                 :                :     JSON_PARSE_VALUE,           /* expecting a value */
                                 96                 :                :     JSON_PARSE_STRING,          /* expecting a string (for a field name) */
                                 97                 :                :     JSON_PARSE_ARRAY_START,     /* saw '[', expecting value or ']' */
                                 98                 :                :     JSON_PARSE_ARRAY_NEXT,      /* saw array element, expecting ',' or ']' */
                                 99                 :                :     JSON_PARSE_OBJECT_START,    /* saw '{', expecting label or '}' */
                                100                 :                :     JSON_PARSE_OBJECT_LABEL,    /* saw object label, expecting ':' */
                                101                 :                :     JSON_PARSE_OBJECT_NEXT,     /* saw object value, expecting ',' or '}' */
                                102                 :                :     JSON_PARSE_OBJECT_COMMA,    /* saw object ',', expecting next label */
                                103                 :                :     JSON_PARSE_END,             /* saw the end of a document, expect nothing */
                                104                 :                : } JsonParseContext;
                                105                 :                : 
                                106                 :                : /*
                                107                 :                :  * Setup for table-driven parser.
                                108                 :                :  * These enums need to be separate from the JsonTokenType and from each other
                                109                 :                :  * so we can have all of them on the prediction stack, which consists of
                                110                 :                :  * tokens, non-terminals, and semantic action markers.
                                111                 :                :  */
                                112                 :                : 
                                113                 :                : enum JsonNonTerminal
                                114                 :                : {
                                115                 :                :     JSON_NT_JSON = 32,
                                116                 :                :     JSON_NT_ARRAY_ELEMENTS,
                                117                 :                :     JSON_NT_MORE_ARRAY_ELEMENTS,
                                118                 :                :     JSON_NT_KEY_PAIRS,
                                119                 :                :     JSON_NT_MORE_KEY_PAIRS,
                                120                 :                : };
                                121                 :                : 
                                122                 :                : enum JsonParserSem
                                123                 :                : {
                                124                 :                :     JSON_SEM_OSTART = 64,
                                125                 :                :     JSON_SEM_OEND,
                                126                 :                :     JSON_SEM_ASTART,
                                127                 :                :     JSON_SEM_AEND,
                                128                 :                :     JSON_SEM_OFIELD_INIT,
                                129                 :                :     JSON_SEM_OFIELD_START,
                                130                 :                :     JSON_SEM_OFIELD_END,
                                131                 :                :     JSON_SEM_AELEM_START,
                                132                 :                :     JSON_SEM_AELEM_END,
                                133                 :                :     JSON_SEM_SCALAR_INIT,
                                134                 :                :     JSON_SEM_SCALAR_CALL,
                                135                 :                : };
                                136                 :                : 
                                137                 :                : /*
                                138                 :                :  * struct containing the 3 stacks used in non-recursive parsing,
                                139                 :                :  * and the token and value for scalars that need to be preserved
                                140                 :                :  * across calls.
                                141                 :                :  *
                                142                 :                :  * typedef appears in jsonapi.h
                                143                 :                :  */
                                144                 :                : struct JsonParserStack
                                145                 :                : {
                                146                 :                :     int         stack_size;
                                147                 :                :     char       *prediction;
                                148                 :                :     size_t      pred_index;
                                149                 :                :     /* these two are indexed by lex_level */
                                150                 :                :     char      **fnames;
                                151                 :                :     bool       *fnull;
                                152                 :                :     JsonTokenType scalar_tok;
                                153                 :                :     char       *scalar_val;
                                154                 :                : };
                                155                 :                : 
                                156                 :                : /*
                                157                 :                :  * struct containing state used when there is a possible partial token at the
                                158                 :                :  * end of a json chunk when we are doing incremental parsing.
                                159                 :                :  *
                                160                 :                :  * typedef appears in jsonapi.h
                                161                 :                :  */
                                162                 :                : struct JsonIncrementalState
                                163                 :                : {
                                164                 :                :     bool        started;
                                165                 :                :     bool        is_last_chunk;
                                166                 :                :     bool        partial_completed;
                                167                 :                :     jsonapi_StrValType partial_token;
                                168                 :                : };
                                169                 :                : 
                                170                 :                : /*
                                171                 :                :  * constants and macros used in the nonrecursive parser
                                172                 :                :  */
                                173                 :                : #define JSON_NUM_TERMINALS 13
                                174                 :                : #define JSON_NUM_NONTERMINALS 5
                                175                 :                : #define JSON_NT_OFFSET JSON_NT_JSON
                                176                 :                : /* for indexing the table */
                                177                 :                : #define OFS(NT) (NT) - JSON_NT_OFFSET
                                178                 :                : /* classify items we get off the stack */
                                179                 :                : #define IS_SEM(x) ((x) & 0x40)
                                180                 :                : #define IS_NT(x)  ((x) & 0x20)
                                181                 :                : 
                                182                 :                : /*
                                183                 :                :  * These productions are stored in reverse order right to left so that when
                                184                 :                :  * they are pushed on the stack what we expect next is at the top of the stack.
                                185                 :                :  */
                                186                 :                : static char JSON_PROD_EPSILON[] = {0};  /* epsilon - an empty production */
                                187                 :                : 
                                188                 :                : /* JSON -> string */
                                189                 :                : static char JSON_PROD_SCALAR_STRING[] = {JSON_SEM_SCALAR_CALL, JSON_TOKEN_STRING, JSON_SEM_SCALAR_INIT, 0};
                                190                 :                : 
                                191                 :                : /* JSON -> number */
                                192                 :                : static char JSON_PROD_SCALAR_NUMBER[] = {JSON_SEM_SCALAR_CALL, JSON_TOKEN_NUMBER, JSON_SEM_SCALAR_INIT, 0};
                                193                 :                : 
                                194                 :                : /* JSON -> 'true' */
                                195                 :                : static char JSON_PROD_SCALAR_TRUE[] = {JSON_SEM_SCALAR_CALL, JSON_TOKEN_TRUE, JSON_SEM_SCALAR_INIT, 0};
                                196                 :                : 
                                197                 :                : /* JSON -> 'false' */
                                198                 :                : static char JSON_PROD_SCALAR_FALSE[] = {JSON_SEM_SCALAR_CALL, JSON_TOKEN_FALSE, JSON_SEM_SCALAR_INIT, 0};
                                199                 :                : 
                                200                 :                : /* JSON -> 'null' */
                                201                 :                : static char JSON_PROD_SCALAR_NULL[] = {JSON_SEM_SCALAR_CALL, JSON_TOKEN_NULL, JSON_SEM_SCALAR_INIT, 0};
                                202                 :                : 
                                203                 :                : /* JSON -> '{' KEY_PAIRS '}' */
                                204                 :                : static char JSON_PROD_OBJECT[] = {JSON_SEM_OEND, JSON_TOKEN_OBJECT_END, JSON_NT_KEY_PAIRS, JSON_TOKEN_OBJECT_START, JSON_SEM_OSTART, 0};
                                205                 :                : 
                                206                 :                : /* JSON -> '[' ARRAY_ELEMENTS ']' */
                                207                 :                : static char JSON_PROD_ARRAY[] = {JSON_SEM_AEND, JSON_TOKEN_ARRAY_END, JSON_NT_ARRAY_ELEMENTS, JSON_TOKEN_ARRAY_START, JSON_SEM_ASTART, 0};
                                208                 :                : 
                                209                 :                : /* ARRAY_ELEMENTS -> JSON MORE_ARRAY_ELEMENTS */
                                210                 :                : static char JSON_PROD_ARRAY_ELEMENTS[] = {JSON_NT_MORE_ARRAY_ELEMENTS, JSON_SEM_AELEM_END, JSON_NT_JSON, JSON_SEM_AELEM_START, 0};
                                211                 :                : 
                                212                 :                : /* MORE_ARRAY_ELEMENTS -> ',' JSON MORE_ARRAY_ELEMENTS */
                                213                 :                : static char JSON_PROD_MORE_ARRAY_ELEMENTS[] = {JSON_NT_MORE_ARRAY_ELEMENTS, JSON_SEM_AELEM_END, JSON_NT_JSON, JSON_SEM_AELEM_START, JSON_TOKEN_COMMA, 0};
                                214                 :                : 
                                215                 :                : /* KEY_PAIRS -> string ':' JSON MORE_KEY_PAIRS */
                                216                 :                : static char JSON_PROD_KEY_PAIRS[] = {JSON_NT_MORE_KEY_PAIRS, JSON_SEM_OFIELD_END, JSON_NT_JSON, JSON_SEM_OFIELD_START, JSON_TOKEN_COLON, JSON_TOKEN_STRING, JSON_SEM_OFIELD_INIT, 0};
                                217                 :                : 
                                218                 :                : /* MORE_KEY_PAIRS -> ',' string ':'  JSON MORE_KEY_PAIRS */
                                219                 :                : static char JSON_PROD_MORE_KEY_PAIRS[] = {JSON_NT_MORE_KEY_PAIRS, JSON_SEM_OFIELD_END, JSON_NT_JSON, JSON_SEM_OFIELD_START, JSON_TOKEN_COLON, JSON_TOKEN_STRING, JSON_SEM_OFIELD_INIT, JSON_TOKEN_COMMA, 0};
                                220                 :                : 
                                221                 :                : /*
                                222                 :                :  * Note: there are also epsilon productions for ARRAY_ELEMENTS,
                                223                 :                :  * MORE_ARRAY_ELEMENTS, KEY_PAIRS and MORE_KEY_PAIRS
                                224                 :                :  * They are all the same as none require any semantic actions.
                                225                 :                :  */
                                226                 :                : 
                                227                 :                : /*
                                228                 :                :  * Table connecting the productions with their director sets of
                                229                 :                :  * terminal symbols.
                                230                 :                :  * Any combination not specified here represents an error.
                                231                 :                :  */
                                232                 :                : 
                                233                 :                : typedef struct
                                234                 :                : {
                                235                 :                :     size_t      len;
                                236                 :                :     char       *prod;
                                237                 :                : } td_entry;
                                238                 :                : 
                                239                 :                : #define TD_ENTRY(PROD) { sizeof(PROD) - 1, (PROD) }
                                240                 :                : 
                                241                 :                : static td_entry td_parser_table[JSON_NUM_NONTERMINALS][JSON_NUM_TERMINALS] =
                                242                 :                : {
                                243                 :                :     /* JSON */
                                244                 :                :     [OFS(JSON_NT_JSON)][JSON_TOKEN_STRING] = TD_ENTRY(JSON_PROD_SCALAR_STRING),
                                245                 :                :     [OFS(JSON_NT_JSON)][JSON_TOKEN_NUMBER] = TD_ENTRY(JSON_PROD_SCALAR_NUMBER),
                                246                 :                :     [OFS(JSON_NT_JSON)][JSON_TOKEN_TRUE] = TD_ENTRY(JSON_PROD_SCALAR_TRUE),
                                247                 :                :     [OFS(JSON_NT_JSON)][JSON_TOKEN_FALSE] = TD_ENTRY(JSON_PROD_SCALAR_FALSE),
                                248                 :                :     [OFS(JSON_NT_JSON)][JSON_TOKEN_NULL] = TD_ENTRY(JSON_PROD_SCALAR_NULL),
                                249                 :                :     [OFS(JSON_NT_JSON)][JSON_TOKEN_ARRAY_START] = TD_ENTRY(JSON_PROD_ARRAY),
                                250                 :                :     [OFS(JSON_NT_JSON)][JSON_TOKEN_OBJECT_START] = TD_ENTRY(JSON_PROD_OBJECT),
                                251                 :                :     /* ARRAY_ELEMENTS */
                                252                 :                :     [OFS(JSON_NT_ARRAY_ELEMENTS)][JSON_TOKEN_ARRAY_START] = TD_ENTRY(JSON_PROD_ARRAY_ELEMENTS),
                                253                 :                :     [OFS(JSON_NT_ARRAY_ELEMENTS)][JSON_TOKEN_OBJECT_START] = TD_ENTRY(JSON_PROD_ARRAY_ELEMENTS),
                                254                 :                :     [OFS(JSON_NT_ARRAY_ELEMENTS)][JSON_TOKEN_STRING] = TD_ENTRY(JSON_PROD_ARRAY_ELEMENTS),
                                255                 :                :     [OFS(JSON_NT_ARRAY_ELEMENTS)][JSON_TOKEN_NUMBER] = TD_ENTRY(JSON_PROD_ARRAY_ELEMENTS),
                                256                 :                :     [OFS(JSON_NT_ARRAY_ELEMENTS)][JSON_TOKEN_TRUE] = TD_ENTRY(JSON_PROD_ARRAY_ELEMENTS),
                                257                 :                :     [OFS(JSON_NT_ARRAY_ELEMENTS)][JSON_TOKEN_FALSE] = TD_ENTRY(JSON_PROD_ARRAY_ELEMENTS),
                                258                 :                :     [OFS(JSON_NT_ARRAY_ELEMENTS)][JSON_TOKEN_NULL] = TD_ENTRY(JSON_PROD_ARRAY_ELEMENTS),
                                259                 :                :     [OFS(JSON_NT_ARRAY_ELEMENTS)][JSON_TOKEN_ARRAY_END] = TD_ENTRY(JSON_PROD_EPSILON),
                                260                 :                :     /* MORE_ARRAY_ELEMENTS */
                                261                 :                :     [OFS(JSON_NT_MORE_ARRAY_ELEMENTS)][JSON_TOKEN_COMMA] = TD_ENTRY(JSON_PROD_MORE_ARRAY_ELEMENTS),
                                262                 :                :     [OFS(JSON_NT_MORE_ARRAY_ELEMENTS)][JSON_TOKEN_ARRAY_END] = TD_ENTRY(JSON_PROD_EPSILON),
                                263                 :                :     /* KEY_PAIRS */
                                264                 :                :     [OFS(JSON_NT_KEY_PAIRS)][JSON_TOKEN_STRING] = TD_ENTRY(JSON_PROD_KEY_PAIRS),
                                265                 :                :     [OFS(JSON_NT_KEY_PAIRS)][JSON_TOKEN_OBJECT_END] = TD_ENTRY(JSON_PROD_EPSILON),
                                266                 :                :     /* MORE_KEY_PAIRS */
                                267                 :                :     [OFS(JSON_NT_MORE_KEY_PAIRS)][JSON_TOKEN_COMMA] = TD_ENTRY(JSON_PROD_MORE_KEY_PAIRS),
                                268                 :                :     [OFS(JSON_NT_MORE_KEY_PAIRS)][JSON_TOKEN_OBJECT_END] = TD_ENTRY(JSON_PROD_EPSILON),
                                269                 :                : };
                                270                 :                : 
                                271                 :                : /* the GOAL production. Not stored in the table, but will be the initial contents of the prediction stack */
                                272                 :                : static char JSON_PROD_GOAL[] = {JSON_TOKEN_END, JSON_NT_JSON, 0};
                                273                 :                : 
                                274                 :                : static inline JsonParseErrorType json_lex_string(JsonLexContext *lex);
                                275                 :                : static inline JsonParseErrorType json_lex_number(JsonLexContext *lex, const char *s,
                                276                 :                :                                                  bool *num_err, size_t *total_len);
                                277                 :                : static inline JsonParseErrorType parse_scalar(JsonLexContext *lex, const JsonSemAction *sem);
                                278                 :                : static JsonParseErrorType parse_object_field(JsonLexContext *lex, const JsonSemAction *sem);
                                279                 :                : static JsonParseErrorType parse_object(JsonLexContext *lex, const JsonSemAction *sem);
                                280                 :                : static JsonParseErrorType parse_array_element(JsonLexContext *lex, const JsonSemAction *sem);
                                281                 :                : static JsonParseErrorType parse_array(JsonLexContext *lex, const JsonSemAction *sem);
                                282                 :                : static JsonParseErrorType report_parse_error(JsonParseContext ctx, JsonLexContext *lex);
                                283                 :                : static bool allocate_incremental_state(JsonLexContext *lex);
                                284                 :                : static inline void set_fname(JsonLexContext *lex, char *fname);
                                285                 :                : 
                                286                 :                : /* the null action object used for pure validation */
                                287                 :                : const JsonSemAction nullSemAction =
                                288                 :                : {
                                289                 :                :     NULL, NULL, NULL, NULL, NULL,
                                290                 :                :     NULL, NULL, NULL, NULL, NULL
                                291                 :                : };
                                292                 :                : 
                                293                 :                : /* sentinels used for out-of-memory conditions */
                                294                 :                : static JsonLexContext failed_oom;
                                295                 :                : static JsonIncrementalState failed_inc_oom;
                                296                 :                : 
                                297                 :                : /* Parser support routines */
                                298                 :                : 
                                299                 :                : /*
                                300                 :                :  * lex_peek
                                301                 :                :  *
                                302                 :                :  * what is the current look_ahead token?
                                303                 :                : */
                                304                 :                : static inline JsonTokenType
 2052 rhaas@postgresql.org      305                 :CBC     7349348 : lex_peek(JsonLexContext *lex)
                                306                 :                : {
                                307                 :        7349348 :     return lex->token_type;
                                308                 :                : }
                                309                 :                : 
                                310                 :                : /*
                                311                 :                :  * lex_expect
                                312                 :                :  *
                                313                 :                :  * move the lexer to the next token if the current look_ahead token matches
                                314                 :                :  * the parameter token. Otherwise, report an error.
                                315                 :                :  */
                                316                 :                : static inline JsonParseErrorType
                                317                 :         236216 : lex_expect(JsonParseContext ctx, JsonLexContext *lex, JsonTokenType token)
                                318                 :                : {
 2059                           319         [ +  + ]:         236216 :     if (lex_peek(lex) == token)
 2049                           320                 :         236162 :         return json_lex(lex);
                                321                 :                :     else
                                322                 :             54 :         return report_parse_error(ctx, lex);
                                323                 :                : }
                                324                 :                : 
                                325                 :                : /* chars to consider as part of an alphanumeric token */
                                326                 :                : #define JSON_ALPHANUMERIC_CHAR(c)  \
                                327                 :                :     (((c) >= 'a' && (c) <= 'z') || \
                                328                 :                :      ((c) >= 'A' && (c) <= 'Z') || \
                                329                 :                :      ((c) >= '0' && (c) <= '9') || \
                                330                 :                :      (c) == '_' || \
                                331                 :                :      IS_HIGHBIT_SET(c))
                                332                 :                : 
                                333                 :                : /*
                                334                 :                :  * Utility function to check if a string is a valid JSON number.
                                335                 :                :  *
                                336                 :                :  * str is of length len, and need not be null-terminated.
                                337                 :                :  */
                                338                 :                : bool
  442 peter@eisentraut.org      339                 :             23 : IsValidJsonNumber(const char *str, size_t len)
                                340                 :                : {
                                341                 :                :     bool        numeric_error;
                                342                 :                :     size_t      total_len;
  360                           343                 :             23 :     JsonLexContext dummy_lex = {0};
                                344                 :                : 
 2052 rhaas@postgresql.org      345         [ -  + ]:             23 :     if (len <= 0)
 2052 rhaas@postgresql.org      346                 :UBC           0 :         return false;
                                347                 :                : 
                                348                 :                :     /*
                                349                 :                :      * json_lex_number expects a leading  '-' to have been eaten already.
                                350                 :                :      *
                                351                 :                :      * having to cast away the constness of str is ugly, but there's not much
                                352                 :                :      * easy alternative.
                                353                 :                :      */
 2052 rhaas@postgresql.org      354         [ +  + ]:CBC          23 :     if (*str == '-')
                                355                 :                :     {
  442 peter@eisentraut.org      356                 :              2 :         dummy_lex.input = str + 1;
 2052 rhaas@postgresql.org      357                 :              2 :         dummy_lex.input_length = len - 1;
                                358                 :                :     }
                                359                 :                :     else
                                360                 :                :     {
  442 peter@eisentraut.org      361                 :             21 :         dummy_lex.input = str;
 2052 rhaas@postgresql.org      362                 :             21 :         dummy_lex.input_length = len;
                                363                 :                :     }
                                364                 :                : 
  519 andrew@dunslane.net       365                 :             23 :     dummy_lex.token_start = dummy_lex.input;
                                366                 :                : 
 2052 rhaas@postgresql.org      367                 :             23 :     json_lex_number(&dummy_lex, dummy_lex.input, &numeric_error, &total_len);
                                368                 :                : 
                                369   [ +  +  +  + ]:             23 :     return (!numeric_error) && (total_len == dummy_lex.input_length);
                                370                 :                : }
                                371                 :                : 
                                372                 :                : /*
                                373                 :                :  * makeJsonLexContextCstringLen
                                374                 :                :  *      Initialize the given JsonLexContext object, or create one
                                375                 :                :  *
                                376                 :                :  * If a valid 'lex' pointer is given, it is initialized.  This can
                                377                 :                :  * be used for stack-allocated structs, saving overhead.  If NULL is
                                378                 :                :  * given, a new struct is allocated.
                                379                 :                :  *
                                380                 :                :  * If need_escapes is true, ->strval stores the unescaped lexemes.
                                381                 :                :  * Unescaping is expensive, so only request it when necessary.
                                382                 :                :  *
                                383                 :                :  * If need_escapes is true or lex was given as NULL, then caller is
                                384                 :                :  * responsible for freeing the returned struct, either by calling
                                385                 :                :  * freeJsonLexContext() or (in backend environment) via memory context
                                386                 :                :  * cleanup.
                                387                 :                :  *
                                388                 :                :  * In shlib code, any out-of-memory failures will be deferred to time
                                389                 :                :  * of use; this function is guaranteed to return a valid JsonLexContext.
                                390                 :                :  */
                                391                 :                : JsonLexContext *
  442 peter@eisentraut.org      392                 :          19865 : makeJsonLexContextCstringLen(JsonLexContext *lex, const char *json,
                                393                 :                :                              size_t len, int encoding, bool need_escapes)
                                394                 :                : {
  702 alvherre@alvh.no-ip.      395         [ +  + ]:          19865 :     if (lex == NULL)
                                396                 :                :     {
  360 peter@eisentraut.org      397                 :           2828 :         lex = ALLOC0(sizeof(JsonLexContext));
                                398         [ -  + ]:           2828 :         if (!lex)
  360 peter@eisentraut.org      399                 :UBC           0 :             return &failed_oom;
  702 alvherre@alvh.no-ip.      400                 :CBC        2828 :         lex->flags |= JSONLEX_FREE_STRUCT;
                                401                 :                :     }
                                402                 :                :     else
                                403                 :          17037 :         memset(lex, 0, sizeof(JsonLexContext));
                                404                 :                : 
  538 dgustafsson@postgres      405                 :          19865 :     lex->errormsg = NULL;
 2052 rhaas@postgresql.org      406                 :          19865 :     lex->input = lex->token_terminator = lex->line_start = json;
                                407                 :          19865 :     lex->line_number = 1;
                                408                 :          19865 :     lex->input_length = len;
 2047                           409                 :          19865 :     lex->input_encoding = encoding;
  360 peter@eisentraut.org      410                 :          19865 :     lex->need_escapes = need_escapes;
 2052 rhaas@postgresql.org      411         [ +  + ]:          19865 :     if (need_escapes)
                                412                 :                :     {
                                413                 :                :         /*
                                414                 :                :          * This call can fail in shlib code. We defer error handling to time
                                415                 :                :          * of use (json_lex_string()) since we might not need to parse any
                                416                 :                :          * strings anyway.
                                417                 :                :          */
  360 peter@eisentraut.org      418                 :          15590 :         lex->strval = jsonapi_makeStringInfo();
  702 alvherre@alvh.no-ip.      419                 :          15590 :         lex->flags |= JSONLEX_FREE_STRVAL;
                                420                 :                :     }
                                421                 :                : 
 2052 rhaas@postgresql.org      422                 :          19865 :     return lex;
                                423                 :                : }
                                424                 :                : 
                                425                 :                : /*
                                426                 :                :  * Allocates the internal bookkeeping structures for incremental parsing. This
                                427                 :                :  * can only fail in-band with shlib code.
                                428                 :                :  */
                                429                 :                : #define JS_STACK_CHUNK_SIZE 64
                                430                 :                : #define JS_MAX_PROD_LEN 10      /* more than we need */
                                431                 :                : #define JSON_TD_MAX_STACK 6400  /* hard coded for now - this is a REALLY high
                                432                 :                :                                  * number */
                                433                 :                : static bool
  360 peter@eisentraut.org      434                 :           2086 : allocate_incremental_state(JsonLexContext *lex)
                                435                 :                : {
                                436                 :                :     void       *pstack,
                                437                 :                :                *prediction,
                                438                 :                :                *fnames,
                                439                 :                :                *fnull;
                                440                 :                : 
                                441                 :           2086 :     lex->inc_state = ALLOC0(sizeof(JsonIncrementalState));
  283 andrew@dunslane.net       442                 :           2086 :     pstack = ALLOC0(sizeof(JsonParserStack));
  360 peter@eisentraut.org      443                 :           2086 :     prediction = ALLOC(JS_STACK_CHUNK_SIZE * JS_MAX_PROD_LEN);
                                444                 :           2086 :     fnames = ALLOC(JS_STACK_CHUNK_SIZE * sizeof(char *));
                                445                 :           2086 :     fnull = ALLOC(JS_STACK_CHUNK_SIZE * sizeof(bool));
                                446                 :                : 
                                447                 :                : #ifdef JSONAPI_USE_PQEXPBUFFER
                                448         [ +  - ]:            980 :     if (!lex->inc_state
                                449         [ +  - ]:            980 :         || !pstack
                                450         [ +  - ]:            980 :         || !prediction
                                451         [ +  - ]:            980 :         || !fnames
                                452         [ -  + ]:            980 :         || !fnull)
                                453                 :                :     {
  360 peter@eisentraut.org      454                 :UBC           0 :         FREE(lex->inc_state);
                                455                 :              0 :         FREE(pstack);
                                456                 :              0 :         FREE(prediction);
                                457                 :              0 :         FREE(fnames);
                                458                 :              0 :         FREE(fnull);
                                459                 :                : 
                                460                 :              0 :         lex->inc_state = &failed_inc_oom;
                                461                 :              0 :         return false;
                                462                 :                :     }
                                463                 :                : #endif
                                464                 :                : 
  360 peter@eisentraut.org      465                 :CBC        2086 :     jsonapi_initStringInfo(&(lex->inc_state->partial_token));
                                466                 :           2086 :     lex->pstack = pstack;
                                467                 :           2086 :     lex->pstack->stack_size = JS_STACK_CHUNK_SIZE;
                                468                 :           2086 :     lex->pstack->prediction = prediction;
                                469                 :           2086 :     lex->pstack->fnames = fnames;
                                470                 :           2086 :     lex->pstack->fnull = fnull;
                                471                 :                : 
                                472                 :                :     /*
                                473                 :                :      * fnames between 0 and lex_level must always be defined so that
                                474                 :                :      * freeJsonLexContext() can handle them safely. inc/dec_lex_level() handle
                                475                 :                :      * the rest.
                                476                 :                :      */
  283 andrew@dunslane.net       477         [ -  + ]:           2086 :     Assert(lex->lex_level == 0);
                                478                 :           2086 :     lex->pstack->fnames[0] = NULL;
                                479                 :                : 
  360 peter@eisentraut.org      480                 :           2086 :     lex->incremental = true;
                                481                 :           2086 :     return true;
                                482                 :                : }
                                483                 :                : 
                                484                 :                : 
                                485                 :                : /*
                                486                 :                :  * makeJsonLexContextIncremental
                                487                 :                :  *
                                488                 :                :  * Similar to above but set up for use in incremental parsing. That means we
                                489                 :                :  * need explicit stacks for predictions, field names and null indicators, but
                                490                 :                :  * we don't need the input, that will be handed in bit by bit to the
                                491                 :                :  * parse routine. We also need an accumulator for partial tokens in case
                                492                 :                :  * the boundary between chunks happens to fall in the middle of a token.
                                493                 :                :  *
                                494                 :                :  * In shlib code, any out-of-memory failures will be deferred to time of use;
                                495                 :                :  * this function is guaranteed to return a valid JsonLexContext.
                                496                 :                :  */
                                497                 :                : JsonLexContext *
  545 andrew@dunslane.net       498                 :           2086 : makeJsonLexContextIncremental(JsonLexContext *lex, int encoding,
                                499                 :                :                               bool need_escapes)
                                500                 :                : {
                                501         [ +  + ]:           2086 :     if (lex == NULL)
                                502                 :                :     {
  360 peter@eisentraut.org      503                 :              1 :         lex = ALLOC0(sizeof(JsonLexContext));
                                504         [ -  + ]:              1 :         if (!lex)
  360 peter@eisentraut.org      505                 :UBC           0 :             return &failed_oom;
                                506                 :                : 
  545 andrew@dunslane.net       507                 :CBC           1 :         lex->flags |= JSONLEX_FREE_STRUCT;
                                508                 :                :     }
                                509                 :                :     else
                                510                 :           2085 :         memset(lex, 0, sizeof(JsonLexContext));
                                511                 :                : 
                                512                 :           2086 :     lex->line_number = 1;
                                513                 :           2086 :     lex->input_encoding = encoding;
                                514                 :                : 
  360 peter@eisentraut.org      515         [ -  + ]:           2086 :     if (!allocate_incremental_state(lex))
                                516                 :                :     {
  360 peter@eisentraut.org      517         [ #  # ]:UBC           0 :         if (lex->flags & JSONLEX_FREE_STRUCT)
                                518                 :                :         {
                                519         [ #  # ]:              0 :             FREE(lex);
                                520                 :              0 :             return &failed_oom;
                                521                 :                :         }
                                522                 :                : 
                                523                 :                :         /* lex->inc_state tracks the OOM failure; we can return here. */
                                524                 :              0 :         return lex;
                                525                 :                :     }
                                526                 :                : 
  360 peter@eisentraut.org      527                 :CBC        2086 :     lex->need_escapes = need_escapes;
  545 andrew@dunslane.net       528         [ +  + ]:           2086 :     if (need_escapes)
                                529                 :                :     {
                                530                 :                :         /*
                                531                 :                :          * This call can fail in shlib code. We defer error handling to time
                                532                 :                :          * of use (json_lex_string()) since we might not need to parse any
                                533                 :                :          * strings anyway.
                                534                 :                :          */
  360 peter@eisentraut.org      535                 :            129 :         lex->strval = jsonapi_makeStringInfo();
  545 andrew@dunslane.net       536                 :            129 :         lex->flags |= JSONLEX_FREE_STRVAL;
                                537                 :                :     }
                                538                 :                : 
                                539                 :           2086 :     return lex;
                                540                 :                : }
                                541                 :                : 
                                542                 :                : void
  283                           543                 :           2166 : setJsonLexContextOwnsTokens(JsonLexContext *lex, bool owned_by_context)
                                544                 :                : {
                                545   [ +  +  -  + ]:           2166 :     if (lex->incremental && lex->inc_state->started)
                                546                 :                :     {
                                547                 :                :         /*
                                548                 :                :          * Switching this flag after parsing has already started is a
                                549                 :                :          * programming error.
                                550                 :                :          */
  283 andrew@dunslane.net       551                 :UBC           0 :         Assert(false);
                                552                 :                :         return;
                                553                 :                :     }
                                554                 :                : 
  283 andrew@dunslane.net       555         [ +  + ]:CBC        2166 :     if (owned_by_context)
                                556                 :           1186 :         lex->flags |= JSONLEX_CTX_OWNS_TOKENS;
                                557                 :                :     else
                                558                 :            980 :         lex->flags &= ~JSONLEX_CTX_OWNS_TOKENS;
                                559                 :                : }
                                560                 :                : 
                                561                 :                : static inline bool
  545                           562                 :        2594233 : inc_lex_level(JsonLexContext *lex)
                                563                 :                : {
  360 peter@eisentraut.org      564   [ +  -  +  + ]:        2594233 :     if (lex->incremental && (lex->lex_level + 1) >= lex->pstack->stack_size)
                                565                 :                :     {
                                566                 :                :         size_t      new_stack_size;
                                567                 :                :         char       *new_prediction;
                                568                 :                :         char      **new_fnames;
                                569                 :                :         bool       *new_fnull;
                                570                 :                : 
                                571                 :          38400 :         new_stack_size = lex->pstack->stack_size + JS_STACK_CHUNK_SIZE;
                                572                 :                : 
                                573                 :          38400 :         new_prediction = REALLOC(lex->pstack->prediction,
                                574                 :                :                                  new_stack_size * JS_MAX_PROD_LEN);
                                575                 :                : #ifdef JSONAPI_USE_PQEXPBUFFER
  348 tgl@sss.pgh.pa.us         576         [ -  + ]:          19200 :         if (!new_prediction)
  348 tgl@sss.pgh.pa.us         577                 :UBC           0 :             return false;
                                578                 :                : #endif
  348 tgl@sss.pgh.pa.us         579                 :CBC       38400 :         lex->pstack->prediction = new_prediction;
                                580                 :                : 
  360 peter@eisentraut.org      581                 :          38400 :         new_fnames = REALLOC(lex->pstack->fnames,
                                582                 :                :                              new_stack_size * sizeof(char *));
                                583                 :                : #ifdef JSONAPI_USE_PQEXPBUFFER
  348 tgl@sss.pgh.pa.us         584         [ -  + ]:          19200 :         if (!new_fnames)
  348 tgl@sss.pgh.pa.us         585                 :UBC           0 :             return false;
                                586                 :                : #endif
  348 tgl@sss.pgh.pa.us         587                 :CBC       38400 :         lex->pstack->fnames = new_fnames;
                                588                 :                : 
                                589                 :          38400 :         new_fnull = REALLOC(lex->pstack->fnull, new_stack_size * sizeof(bool));
                                590                 :                : #ifdef JSONAPI_USE_PQEXPBUFFER
                                591         [ -  + ]:          19200 :         if (!new_fnull)
  360 peter@eisentraut.org      592                 :UBC           0 :             return false;
                                593                 :                : #endif
  348 tgl@sss.pgh.pa.us         594                 :CBC       38400 :         lex->pstack->fnull = new_fnull;
                                595                 :                : 
  360 peter@eisentraut.org      596                 :          38400 :         lex->pstack->stack_size = new_stack_size;
                                597                 :                :     }
                                598                 :                : 
                                599                 :        2594233 :     lex->lex_level += 1;
                                600                 :                : 
  283 andrew@dunslane.net       601         [ +  - ]:        2594233 :     if (lex->incremental)
                                602                 :                :     {
                                603                 :                :         /*
                                604                 :                :          * Ensure freeJsonLexContext() remains safe even if no fname is
                                605                 :                :          * assigned at this level.
                                606                 :                :          */
                                607                 :        2594233 :         lex->pstack->fnames[lex->lex_level] = NULL;
                                608                 :                :     }
                                609                 :                : 
  360 peter@eisentraut.org      610                 :        2594233 :     return true;
                                611                 :                : }
                                612                 :                : 
                                613                 :                : static inline void
  545 andrew@dunslane.net       614                 :         955332 : dec_lex_level(JsonLexContext *lex)
                                615                 :                : {
  283                           616                 :         955332 :     set_fname(lex, NULL);       /* free the current level's fname, if needed */
  545                           617                 :         955332 :     lex->lex_level -= 1;
                                618                 :         955332 : }
                                619                 :                : 
                                620                 :                : static inline void
                                621                 :        7439399 : push_prediction(JsonParserStack *pstack, td_entry entry)
                                622                 :                : {
                                623                 :        7439399 :     memcpy(pstack->prediction + pstack->pred_index, entry.prod, entry.len);
                                624                 :        7439399 :     pstack->pred_index += entry.len;
                                625                 :        7439399 : }
                                626                 :                : 
                                627                 :                : static inline char
                                628                 :       23927820 : pop_prediction(JsonParserStack *pstack)
                                629                 :                : {
                                630         [ -  + ]:       23927820 :     Assert(pstack->pred_index > 0);
                                631                 :       23927820 :     return pstack->prediction[--pstack->pred_index];
                                632                 :                : }
                                633                 :                : 
                                634                 :                : static inline char
                                635                 :             76 : next_prediction(JsonParserStack *pstack)
                                636                 :                : {
                                637         [ -  + ]:             76 :     Assert(pstack->pred_index > 0);
                                638                 :             76 :     return pstack->prediction[pstack->pred_index - 1];
                                639                 :                : }
                                640                 :                : 
                                641                 :                : static inline bool
                                642                 :       24230130 : have_prediction(JsonParserStack *pstack)
                                643                 :                : {
                                644                 :       24230130 :     return pstack->pred_index > 0;
                                645                 :                : }
                                646                 :                : 
                                647                 :                : static inline void
                                648                 :        1607906 : set_fname(JsonLexContext *lex, char *fname)
                                649                 :                : {
  283                           650         [ +  + ]:        1607906 :     if (lex->flags & JSONLEX_CTX_OWNS_TOKENS)
                                651                 :                :     {
                                652                 :                :         /*
                                653                 :                :          * Don't leak prior fnames. If one hasn't been assigned yet,
                                654                 :                :          * inc_lex_level ensured that it's NULL (and therefore safe to free).
                                655                 :                :          */
                                656         [ #  # ]:         436862 :         FREE(lex->pstack->fnames[lex->lex_level]);
                                657                 :                :     }
                                658                 :                : 
  545                           659                 :        1607906 :     lex->pstack->fnames[lex->lex_level] = fname;
                                660                 :        1607906 : }
                                661                 :                : 
                                662                 :                : static inline char *
                                663                 :         612818 : get_fname(JsonLexContext *lex)
                                664                 :                : {
                                665                 :         612818 :     return lex->pstack->fnames[lex->lex_level];
                                666                 :                : }
                                667                 :                : 
                                668                 :                : static inline void
                                669                 :        3240086 : set_fnull(JsonLexContext *lex, bool fnull)
                                670                 :                : {
                                671                 :        3240086 :     lex->pstack->fnull[lex->lex_level] = fnull;
                                672                 :        3240086 : }
                                673                 :                : 
                                674                 :                : static inline bool
                                675                 :            744 : get_fnull(JsonLexContext *lex)
                                676                 :                : {
                                677                 :            744 :     return lex->pstack->fnull[lex->lex_level];
                                678                 :                : }
                                679                 :                : 
                                680                 :                : /*
                                681                 :                :  * Free memory in a JsonLexContext.
                                682                 :                :  *
                                683                 :                :  * There's no need for this if a *lex pointer was given when the object was
                                684                 :                :  * made, need_escapes was false, and json_errdetail() was not called; or if (in
                                685                 :                :  * backend environment) a memory context delete/reset is imminent.
                                686                 :                :  */
                                687                 :                : void
  702 alvherre@alvh.no-ip.      688                 :           5429 : freeJsonLexContext(JsonLexContext *lex)
                                689                 :                : {
                                690                 :                :     static const JsonLexContext empty = {0};
                                691                 :                : 
  360 peter@eisentraut.org      692   [ +  -  -  + ]:           5429 :     if (!lex || lex == &failed_oom)
  360 peter@eisentraut.org      693                 :UBC           0 :         return;
                                694                 :                : 
  702 alvherre@alvh.no-ip.      695         [ +  + ]:CBC        5429 :     if (lex->flags & JSONLEX_FREE_STRVAL)
  360 peter@eisentraut.org      696                 :           3278 :         jsonapi_destroyStringInfo(lex->strval);
                                697                 :                : 
  538 dgustafsson@postgres      698         [ +  + ]:           5429 :     if (lex->errormsg)
  360 peter@eisentraut.org      699                 :            953 :         jsonapi_destroyStringInfo(lex->errormsg);
                                700                 :                : 
  515 andrew@dunslane.net       701         [ +  + ]:           5429 :     if (lex->incremental)
                                702                 :                :     {
  360 peter@eisentraut.org      703                 :           2083 :         jsonapi_termStringInfo(&lex->inc_state->partial_token);
                                704         [ +  - ]:           2083 :         FREE(lex->inc_state);
                                705         [ +  - ]:           2083 :         FREE(lex->pstack->prediction);
                                706                 :                : 
  283 andrew@dunslane.net       707         [ +  + ]:           2083 :         if (lex->flags & JSONLEX_CTX_OWNS_TOKENS)
                                708                 :                :         {
                                709                 :                :             int         i;
                                710                 :                : 
                                711                 :                :             /* Clean up any tokens that were left behind. */
                                712         [ +  + ]:         821410 :             for (i = 0; i <= lex->lex_level; i++)
                                713         [ #  # ]:         820430 :                 FREE(lex->pstack->fnames[i]);
                                714                 :                :         }
                                715                 :                : 
  360 peter@eisentraut.org      716         [ +  - ]:           2083 :         FREE(lex->pstack->fnames);
                                717         [ +  - ]:           2083 :         FREE(lex->pstack->fnull);
  283 andrew@dunslane.net       718         [ -  + ]:           2083 :         FREE(lex->pstack->scalar_val);
  360 peter@eisentraut.org      719         [ +  - ]:           2083 :         FREE(lex->pstack);
                                720                 :                :     }
                                721                 :                : 
  515 andrew@dunslane.net       722         [ +  + ]:           5429 :     if (lex->flags & JSONLEX_FREE_STRUCT)
  360 peter@eisentraut.org      723         [ +  - ]:           2718 :         FREE(lex);
                                724                 :                :     else
                                725                 :           2711 :         *lex = empty;
                                726                 :                : }
                                727                 :                : 
                                728                 :                : /*
                                729                 :                :  * pg_parse_json
                                730                 :                :  *
                                731                 :                :  * Publicly visible entry point for the JSON parser.
                                732                 :                :  *
                                733                 :                :  * lex is a lexing context, set up for the json to be processed by calling
                                734                 :                :  * makeJsonLexContext(). sem is a structure of function pointers to semantic
                                735                 :                :  * action routines to be called at appropriate spots during parsing, and a
                                736                 :                :  * pointer to a state object to be passed to those routines.
                                737                 :                :  *
                                738                 :                :  * If FORCE_JSON_PSTACK is defined then the routine will call the non-recursive
                                739                 :                :  * JSON parser. This is a useful way to validate that it's doing the right
                                740                 :                :  * thing at least for non-incremental cases. If this is on we expect to see
                                741                 :                :  * regression diffs relating to error messages about stack depth, but no
                                742                 :                :  * other differences.
                                743                 :                :  */
                                744                 :                : JsonParseErrorType
  396 heikki.linnakangas@i      745                 :          19499 : pg_parse_json(JsonLexContext *lex, const JsonSemAction *sem)
                                746                 :                : {
                                747                 :                : #ifdef FORCE_JSON_PSTACK
                                748                 :                :     /*
                                749                 :                :      * We don't need partial token processing, there is only one chunk. But we
                                750                 :                :      * still need to init the partial token string so that freeJsonLexContext
                                751                 :                :      * works, so perform the full incremental initialization.
                                752                 :                :      */
                                753                 :                :     if (!allocate_incremental_state(lex))
                                754                 :                :         return JSON_OUT_OF_MEMORY;
                                755                 :                : 
                                756                 :                :     return pg_parse_json_incremental(lex, sem, lex->input, lex->input_length, true);
                                757                 :                : 
                                758                 :                : #else
                                759                 :                : 
                                760                 :                :     JsonTokenType tok;
                                761                 :                :     JsonParseErrorType result;
                                762                 :                : 
  360 peter@eisentraut.org      763         [ -  + ]:          19499 :     if (lex == &failed_oom)
  360 peter@eisentraut.org      764                 :UBC           0 :         return JSON_OUT_OF_MEMORY;
  545 andrew@dunslane.net       765         [ -  + ]:CBC       19499 :     if (lex->incremental)
  545 andrew@dunslane.net       766                 :UBC           0 :         return JSON_INVALID_LEXER_TYPE;
                                767                 :                : 
                                768                 :                :     /* get the initial token */
 2049 rhaas@postgresql.org      769                 :CBC       19499 :     result = json_lex(lex);
                                770         [ +  + ]:          19499 :     if (result != JSON_SUCCESS)
                                771                 :            123 :         return result;
                                772                 :                : 
 2052                           773                 :          19376 :     tok = lex_peek(lex);
                                774                 :                : 
                                775                 :                :     /* parse by recursive descent */
                                776      [ +  +  + ]:          19376 :     switch (tok)
                                777                 :                :     {
                                778                 :          10584 :         case JSON_TOKEN_OBJECT_START:
 2049                           779                 :          10584 :             result = parse_object(lex, sem);
 2052                           780                 :          10550 :             break;
                                781                 :           3703 :         case JSON_TOKEN_ARRAY_START:
 2049                           782                 :           3703 :             result = parse_array(lex, sem);
 2052                           783                 :           3651 :             break;
                                784                 :           5089 :         default:
 1941 tgl@sss.pgh.pa.us         785                 :           5089 :             result = parse_scalar(lex, sem);    /* json can be a bare scalar */
                                786                 :                :     }
                                787                 :                : 
 2049 rhaas@postgresql.org      788         [ +  + ]:          19254 :     if (result == JSON_SUCCESS)
                                789                 :          19030 :         result = lex_expect(JSON_PARSE_END, lex, JSON_TOKEN_END);
                                790                 :                : 
                                791                 :          19254 :     return result;
                                792                 :                : #endif
                                793                 :                : }
                                794                 :                : 
                                795                 :                : /*
                                796                 :                :  * json_count_array_elements
                                797                 :                :  *
                                798                 :                :  * Returns number of array elements in lex context at start of array token
                                799                 :                :  * until end of array token at same nesting level.
                                800                 :                :  *
                                801                 :                :  * Designed to be called from array_start routines.
                                802                 :                :  */
                                803                 :                : JsonParseErrorType
                                804                 :              3 : json_count_array_elements(JsonLexContext *lex, int *elements)
                                805                 :                : {
                                806                 :                :     JsonLexContext copylex;
                                807                 :                :     int         count;
                                808                 :                :     JsonParseErrorType result;
                                809                 :                : 
  360 peter@eisentraut.org      810         [ -  + ]:              3 :     if (lex == &failed_oom)
  360 peter@eisentraut.org      811                 :UBC           0 :         return JSON_OUT_OF_MEMORY;
                                812                 :                : 
                                813                 :                :     /*
                                814                 :                :      * It's safe to do this with a shallow copy because the lexical routines
                                815                 :                :      * don't scribble on the input. They do scribble on the other pointers
                                816                 :                :      * etc, so doing this with a copy makes that safe.
                                817                 :                :      */
 2052 rhaas@postgresql.org      818                 :CBC           3 :     memcpy(&copylex, lex, sizeof(JsonLexContext));
  360 peter@eisentraut.org      819                 :              3 :     copylex.need_escapes = false;   /* not interested in values here */
 2052 rhaas@postgresql.org      820                 :              3 :     copylex.lex_level++;
                                821                 :                : 
                                822                 :              3 :     count = 0;
 2049                           823                 :              3 :     result = lex_expect(JSON_PARSE_ARRAY_START, &copylex,
                                824                 :                :                         JSON_TOKEN_ARRAY_START);
                                825         [ -  + ]:              3 :     if (result != JSON_SUCCESS)
 2049 rhaas@postgresql.org      826                 :UBC           0 :         return result;
 2052 rhaas@postgresql.org      827         [ +  - ]:CBC           3 :     if (lex_peek(&copylex) != JSON_TOKEN_ARRAY_END)
                                828                 :                :     {
                                829                 :                :         while (1)
                                830                 :                :         {
                                831                 :             24 :             count++;
 2049                           832                 :             24 :             result = parse_array_element(&copylex, &nullSemAction);
                                833         [ -  + ]:             24 :             if (result != JSON_SUCCESS)
 2049 rhaas@postgresql.org      834                 :UBC           0 :                 return result;
 2059 rhaas@postgresql.org      835         [ +  + ]:CBC          24 :             if (copylex.token_type != JSON_TOKEN_COMMA)
                                836                 :              3 :                 break;
 2049                           837                 :             21 :             result = json_lex(&copylex);
                                838         [ -  + ]:             21 :             if (result != JSON_SUCCESS)
 2049 rhaas@postgresql.org      839                 :UBC           0 :                 return result;
                                840                 :                :         }
                                841                 :                :     }
 2049 rhaas@postgresql.org      842                 :CBC           3 :     result = lex_expect(JSON_PARSE_ARRAY_NEXT, &copylex,
                                843                 :                :                         JSON_TOKEN_ARRAY_END);
                                844         [ -  + ]:              3 :     if (result != JSON_SUCCESS)
 2049 rhaas@postgresql.org      845                 :UBC           0 :         return result;
                                846                 :                : 
 2049 rhaas@postgresql.org      847                 :CBC           3 :     *elements = count;
                                848                 :              3 :     return JSON_SUCCESS;
                                849                 :                : }
                                850                 :                : 
                                851                 :                : /*
                                852                 :                :  * pg_parse_json_incremental
                                853                 :                :  *
                                854                 :                :  * Routine for incremental parsing of json. This uses the non-recursive top
                                855                 :                :  * down method of the Dragon Book Algorithm 4.3. It's somewhat slower than
                                856                 :                :  * the Recursive Descent pattern used above, so we only use it for incremental
                                857                 :                :  * parsing of JSON.
                                858                 :                :  *
                                859                 :                :  * The lexing context needs to be set up by a call to
                                860                 :                :  * makeJsonLexContextIncremental(). sem is a structure of function pointers
                                861                 :                :  * to semantic action routines, which should function exactly as those used
                                862                 :                :  * in the recursive descent parser.
                                863                 :                :  *
                                864                 :                :  * This routine can be called repeatedly with chunks of JSON. On the final
                                865                 :                :  * chunk is_last must be set to true. len is the length of the json chunk,
                                866                 :                :  * which does not need to be null terminated.
                                867                 :                :  */
                                868                 :                : JsonParseErrorType
  545 andrew@dunslane.net       869                 :         372958 : pg_parse_json_incremental(JsonLexContext *lex,
                                870                 :                :                           const JsonSemAction *sem,
                                871                 :                :                           const char *json,
                                872                 :                :                           size_t len,
                                873                 :                :                           bool is_last)
                                874                 :                : {
                                875                 :                :     JsonTokenType tok;
                                876                 :                :     JsonParseErrorType result;
                                877                 :         372958 :     JsonParseContext ctx = JSON_PARSE_VALUE;
                                878                 :         372958 :     JsonParserStack *pstack = lex->pstack;
                                879                 :                : 
  360 peter@eisentraut.org      880   [ +  -  -  + ]:         372958 :     if (lex == &failed_oom || lex->inc_state == &failed_inc_oom)
  360 peter@eisentraut.org      881                 :UBC           0 :         return JSON_OUT_OF_MEMORY;
  545 andrew@dunslane.net       882         [ -  + ]:CBC      372958 :     if (!lex->incremental)
  545 andrew@dunslane.net       883                 :UBC           0 :         return JSON_INVALID_LEXER_TYPE;
                                884                 :                : 
  545 andrew@dunslane.net       885                 :CBC      372958 :     lex->input = lex->token_terminator = lex->line_start = json;
                                886                 :         372958 :     lex->input_length = len;
                                887                 :         372958 :     lex->inc_state->is_last_chunk = is_last;
  283                           888                 :         372958 :     lex->inc_state->started = true;
                                889                 :                : 
                                890                 :                :     /* get the initial token */
  545                           891                 :         372958 :     result = json_lex(lex);
                                892         [ +  + ]:         372958 :     if (result != JSON_SUCCESS)
                                893                 :          71780 :         return result;
                                894                 :                : 
                                895                 :         301178 :     tok = lex_peek(lex);
                                896                 :                : 
                                897                 :                :     /* use prediction stack for incremental parsing */
                                898                 :                : 
                                899         [ +  + ]:         301178 :     if (!have_prediction(pstack))
                                900                 :                :     {
                                901                 :           1902 :         td_entry    goal = TD_ENTRY(JSON_PROD_GOAL);
                                902                 :                : 
                                903                 :           1902 :         push_prediction(pstack, goal);
                                904                 :                :     }
                                905                 :                : 
                                906         [ +  + ]:       23928952 :     while (have_prediction(pstack))
                                907                 :                :     {
                                908                 :       23927820 :         char        top = pop_prediction(pstack);
                                909                 :                :         td_entry    entry;
                                910                 :                : 
                                911                 :                :         /*
                                912                 :                :          * these first two branches are the guts of the Table Driven method
                                913                 :                :          */
                                914         [ +  + ]:       23927820 :         if (top == tok)
                                915                 :                :         {
                                916                 :                :             /*
                                917                 :                :              * tok can only be a terminal symbol, so top must be too. the
                                918                 :                :              * token matches the top of the stack, so get the next token.
                                919                 :                :              */
                                920         [ +  + ]:        6150828 :             if (tok < JSON_TOKEN_END)
                                921                 :                :             {
                                922                 :        6149696 :                 result = json_lex(lex);
                                923         [ +  + ]:        6149696 :                 if (result != JSON_SUCCESS)
                                924                 :         300045 :                     return result;
                                925                 :        5850348 :                 tok = lex_peek(lex);
                                926                 :                :             }
                                927                 :                :         }
                                928   [ +  +  +  + ]:       17776992 :         else if (IS_NT(top) && (entry = td_parser_table[OFS(top)][tok]).prod != NULL)
                                929                 :                :         {
                                930                 :                :             /*
                                931                 :                :              * the token is in the director set for a production of the
                                932                 :                :              * non-terminal at the top of the stack, so push the reversed RHS
                                933                 :                :              * of the production onto the stack.
                                934                 :                :              */
                                935                 :        7437497 :             push_prediction(pstack, entry);
                                936                 :                :         }
                                937         [ +  + ]:       10339495 :         else if (IS_SEM(top))
                                938                 :                :         {
                                939                 :                :             /*
                                940                 :                :              * top is a semantic action marker, so take action accordingly.
                                941                 :                :              * It's important to have these markers in the prediction stack
                                942                 :                :              * before any token they might need so we don't advance the token
                                943                 :                :              * prematurely. Note in a couple of cases we need to do something
                                944                 :                :              * both before and after the token.
                                945                 :                :              */
                                946   [ +  +  +  +  :       10339054 :             switch (top)
                                     +  +  +  +  +  
                                           +  +  - ]
                                947                 :                :             {
                                948                 :         133201 :                 case JSON_SEM_OSTART:
                                949                 :                :                     {
                                950                 :         133201 :                         json_struct_action ostart = sem->object_start;
                                951                 :                : 
                                952         [ -  + ]:         133201 :                         if (lex->lex_level >= JSON_TD_MAX_STACK)
  545 andrew@dunslane.net       953                 :UBC           0 :                             return JSON_NESTING_TOO_DEEP;
                                954                 :                : 
  545 andrew@dunslane.net       955         [ +  + ]:CBC      133201 :                         if (ostart != NULL)
                                956                 :                :                         {
                                957                 :         122525 :                             result = (*ostart) (sem->semstate);
                                958         [ -  + ]:         122525 :                             if (result != JSON_SUCCESS)
  545 andrew@dunslane.net       959                 :UBC           0 :                                 return result;
                                960                 :                :                         }
                                961                 :                : 
  360 peter@eisentraut.org      962         [ -  + ]:CBC      133201 :                         if (!inc_lex_level(lex))
  360 peter@eisentraut.org      963                 :UBC           0 :                             return JSON_OUT_OF_MEMORY;
                                964                 :                :                     }
  545 andrew@dunslane.net       965                 :CBC      133201 :                     break;
                                966                 :         132836 :                 case JSON_SEM_OEND:
                                967                 :                :                     {
                                968                 :         132836 :                         json_struct_action oend = sem->object_end;
                                969                 :                : 
                                970                 :         132836 :                         dec_lex_level(lex);
                                971         [ +  + ]:         132836 :                         if (oend != NULL)
                                972                 :                :                         {
                                973                 :         122524 :                             result = (*oend) (sem->semstate);
                                974         [ -  + ]:         122524 :                             if (result != JSON_SUCCESS)
  545 andrew@dunslane.net       975                 :UBC           0 :                                 return result;
                                976                 :                :                         }
                                977                 :                :                     }
  545 andrew@dunslane.net       978                 :CBC      132836 :                     break;
                                979                 :        2461288 :                 case JSON_SEM_ASTART:
                                980                 :                :                     {
                                981                 :        2461288 :                         json_struct_action astart = sem->array_start;
                                982                 :                : 
                                983         [ +  + ]:        2461288 :                         if (lex->lex_level >= JSON_TD_MAX_STACK)
                                984                 :            256 :                             return JSON_NESTING_TOO_DEEP;
                                985                 :                : 
                                986         [ +  + ]:        2461032 :                         if (astart != NULL)
                                987                 :                :                         {
                                988                 :            292 :                             result = (*astart) (sem->semstate);
                                989         [ -  + ]:            292 :                             if (result != JSON_SUCCESS)
  545 andrew@dunslane.net       990                 :UBC           0 :                                 return result;
                                991                 :                :                         }
                                992                 :                : 
  360 peter@eisentraut.org      993         [ -  + ]:CBC     2461032 :                         if (!inc_lex_level(lex))
  360 peter@eisentraut.org      994                 :UBC           0 :                             return JSON_OUT_OF_MEMORY;
                                995                 :                :                     }
  545 andrew@dunslane.net       996                 :CBC     2461032 :                     break;
                                997                 :         822496 :                 case JSON_SEM_AEND:
                                998                 :                :                     {
                                999                 :         822496 :                         json_struct_action aend = sem->array_end;
                               1000                 :                : 
                               1001                 :         822496 :                         dec_lex_level(lex);
                               1002         [ +  + ]:         822496 :                         if (aend != NULL)
                               1003                 :                :                         {
                               1004                 :            292 :                             result = (*aend) (sem->semstate);
                               1005         [ -  + ]:            292 :                             if (result != JSON_SUCCESS)
  545 andrew@dunslane.net      1006                 :UBC           0 :                                 return result;
                               1007                 :                :                         }
                               1008                 :                :                     }
  545 andrew@dunslane.net      1009                 :CBC      822496 :                     break;
                               1010                 :         652574 :                 case JSON_SEM_OFIELD_INIT:
                               1011                 :                :                     {
                               1012                 :                :                         /*
                               1013                 :                :                          * all we do here is save out the field name. We have
                               1014                 :                :                          * to wait to get past the ':' to see if the next
                               1015                 :                :                          * value is null so we can call the semantic routine
                               1016                 :                :                          */
                               1017                 :         652574 :                         char       *fname = NULL;
                               1018                 :         652574 :                         json_ofield_action ostart = sem->object_field_start;
                               1019                 :         652574 :                         json_ofield_action oend = sem->object_field_end;
                               1020                 :                : 
  360 peter@eisentraut.org     1021   [ +  +  -  +  :         652574 :                         if ((ostart != NULL || oend != NULL) && lex->need_escapes)
                                              +  - ]
                               1022                 :                :                         {
                               1023                 :         612194 :                             fname = STRDUP(lex->strval->data);
                               1024         [ -  + ]:         612194 :                             if (fname == NULL)
  360 peter@eisentraut.org     1025                 :UBC           0 :                                 return JSON_OUT_OF_MEMORY;
                               1026                 :                :                         }
  545 andrew@dunslane.net      1027                 :CBC      652574 :                         set_fname(lex, fname);
                               1028                 :                :                     }
                               1029                 :         652574 :                     break;
                               1030                 :         652446 :                 case JSON_SEM_OFIELD_START:
                               1031                 :                :                     {
                               1032                 :                :                         /*
                               1033                 :                :                          * the current token should be the first token of the
                               1034                 :                :                          * value
                               1035                 :                :                          */
                               1036                 :         652446 :                         bool        isnull = tok == JSON_TOKEN_NULL;
                               1037                 :         652446 :                         json_ofield_action ostart = sem->object_field_start;
                               1038                 :                : 
                               1039                 :         652446 :                         set_fnull(lex, isnull);
                               1040                 :                : 
                               1041         [ +  + ]:         652446 :                         if (ostart != NULL)
                               1042                 :                :                         {
                               1043                 :         612194 :                             char       *fname = get_fname(lex);
                               1044                 :                : 
                               1045                 :         612194 :                             result = (*ostart) (sem->semstate, fname, isnull);
                               1046         [ -  + ]:         612194 :                             if (result != JSON_SUCCESS)
  545 andrew@dunslane.net      1047                 :UBC           0 :                                 return result;
                               1048                 :                :                         }
                               1049                 :                :                     }
  545 andrew@dunslane.net      1050                 :CBC      652446 :                     break;
                               1051                 :         652409 :                 case JSON_SEM_OFIELD_END:
                               1052                 :                :                     {
                               1053                 :         652409 :                         json_ofield_action oend = sem->object_field_end;
                               1054                 :                : 
                               1055         [ +  + ]:         652409 :                         if (oend != NULL)
                               1056                 :                :                         {
                               1057                 :            624 :                             char       *fname = get_fname(lex);
                               1058                 :            624 :                             bool        isnull = get_fnull(lex);
                               1059                 :                : 
                               1060                 :            624 :                             result = (*oend) (sem->semstate, fname, isnull);
                               1061         [ -  + ]:            624 :                             if (result != JSON_SUCCESS)
  545 andrew@dunslane.net      1062                 :UBC           0 :                                 return result;
                               1063                 :                :                         }
                               1064                 :                :                     }
  545 andrew@dunslane.net      1065                 :CBC      652409 :                     break;
                               1066                 :        2587640 :                 case JSON_SEM_AELEM_START:
                               1067                 :                :                     {
                               1068                 :        2587640 :                         json_aelem_action astart = sem->array_element_start;
                               1069                 :        2587640 :                         bool        isnull = tok == JSON_TOKEN_NULL;
                               1070                 :                : 
                               1071                 :        2587640 :                         set_fnull(lex, isnull);
                               1072                 :                : 
                               1073         [ +  + ]:        2587640 :                         if (astart != NULL)
                               1074                 :                :                         {
                               1075                 :            120 :                             result = (*astart) (sem->semstate, isnull);
                               1076         [ -  + ]:            120 :                             if (result != JSON_SUCCESS)
  545 andrew@dunslane.net      1077                 :UBC           0 :                                 return result;
                               1078                 :                :                         }
                               1079                 :                :                     }
  545 andrew@dunslane.net      1080                 :CBC     2587640 :                     break;
                               1081                 :         949240 :                 case JSON_SEM_AELEM_END:
                               1082                 :                :                     {
                               1083                 :         949240 :                         json_aelem_action aend = sem->array_element_end;
                               1084                 :                : 
                               1085         [ +  + ]:         949240 :                         if (aend != NULL)
                               1086                 :                :                         {
                               1087                 :            120 :                             bool        isnull = get_fnull(lex);
                               1088                 :                : 
                               1089                 :            120 :                             result = (*aend) (sem->semstate, isnull);
                               1090         [ -  + ]:            120 :                             if (result != JSON_SUCCESS)
  545 andrew@dunslane.net      1091                 :UBC           0 :                                 return result;
                               1092                 :                :                         }
                               1093                 :                :                     }
  545 andrew@dunslane.net      1094                 :CBC      949240 :                     break;
                               1095                 :         647462 :                 case JSON_SEM_SCALAR_INIT:
                               1096                 :                :                     {
                               1097                 :         647462 :                         json_scalar_action sfunc = sem->scalar;
                               1098                 :                : 
                               1099                 :         647462 :                         pstack->scalar_val = NULL;
                               1100                 :                : 
                               1101         [ +  + ]:         647462 :                         if (sfunc != NULL)
                               1102                 :                :                         {
                               1103                 :                :                             /*
                               1104                 :                :                              * extract the de-escaped string value, or the raw
                               1105                 :                :                              * lexeme
                               1106                 :                :                              */
                               1107                 :                :                             /*
                               1108                 :                :                              * XXX copied from RD parser but looks like a
                               1109                 :                :                              * buglet
                               1110                 :                :                              */
                               1111         [ +  + ]:         611866 :                             if (tok == JSON_TOKEN_STRING)
                               1112                 :                :                             {
  360 peter@eisentraut.org     1113         [ +  - ]:         489256 :                                 if (lex->need_escapes)
                               1114                 :                :                                 {
                               1115                 :         489256 :                                     pstack->scalar_val = STRDUP(lex->strval->data);
                               1116         [ -  + ]:         489256 :                                     if (pstack->scalar_val == NULL)
  360 peter@eisentraut.org     1117                 :UBC           0 :                                         return JSON_OUT_OF_MEMORY;
                               1118                 :                :                                 }
                               1119                 :                :                             }
                               1120                 :                :                             else
                               1121                 :                :                             {
  442 peter@eisentraut.org     1122                 :CBC      122610 :                                 ptrdiff_t   tlen = (lex->token_terminator - lex->token_start);
                               1123                 :                : 
  360                          1124                 :         122610 :                                 pstack->scalar_val = ALLOC(tlen + 1);
                               1125         [ -  + ]:         122610 :                                 if (pstack->scalar_val == NULL)
  360 peter@eisentraut.org     1126                 :UBC           0 :                                     return JSON_OUT_OF_MEMORY;
                               1127                 :                : 
  545 andrew@dunslane.net      1128                 :CBC      122610 :                                 memcpy(pstack->scalar_val, lex->token_start, tlen);
                               1129                 :         122610 :                                 pstack->scalar_val[tlen] = '\0';
                               1130                 :                :                             }
                               1131                 :         611866 :                             pstack->scalar_tok = tok;
                               1132                 :                :                         }
                               1133                 :                :                     }
                               1134                 :         647462 :                     break;
                               1135                 :         647462 :                 case JSON_SEM_SCALAR_CALL:
                               1136                 :                :                     {
                               1137                 :                :                         /*
                               1138                 :                :                          * We'd like to be able to get rid of this business of
                               1139                 :                :                          * two bits of scalar action, but we can't. It breaks
                               1140                 :                :                          * certain semantic actions which expect that when
                               1141                 :                :                          * called the lexer has consumed the item. See for
                               1142                 :                :                          * example get_scalar() in jsonfuncs.c.
                               1143                 :                :                          */
                               1144                 :         647462 :                         json_scalar_action sfunc = sem->scalar;
                               1145                 :                : 
                               1146         [ +  + ]:         647462 :                         if (sfunc != NULL)
                               1147                 :                :                         {
                               1148                 :         611866 :                             result = (*sfunc) (sem->semstate, pstack->scalar_val, pstack->scalar_tok);
                               1149                 :                : 
                               1150                 :                :                             /*
                               1151                 :                :                              * Either ownership of the token passed to the
                               1152                 :                :                              * callback, or we need to free it now. Either
                               1153                 :                :                              * way, clear our pointer to it so it doesn't get
                               1154                 :                :                              * freed in the future.
                               1155                 :                :                              */
  283                          1156         [ +  + ]:         611865 :                             if (lex->flags & JSONLEX_CTX_OWNS_TOKENS)
                               1157         [ #  # ]:            272 :                                 FREE(pstack->scalar_val);
                               1158                 :         611865 :                             pstack->scalar_val = NULL;
                               1159                 :                : 
  545                          1160         [ -  + ]:         611865 :                             if (result != JSON_SUCCESS)
  545 andrew@dunslane.net      1161                 :UBC           0 :                                 return result;
                               1162                 :                :                         }
                               1163                 :                :                     }
  545 andrew@dunslane.net      1164                 :CBC      647461 :                     break;
  545 andrew@dunslane.net      1165                 :UBC           0 :                 default:
                               1166                 :                :                     /* should not happen */
                               1167                 :              0 :                     break;
                               1168                 :                :             }
                               1169                 :                :         }
                               1170                 :                :         else
                               1171                 :                :         {
                               1172                 :                :             /*
                               1173                 :                :              * The token didn't match the stack top if it's a terminal nor a
                               1174                 :                :              * production for the stack top if it's a non-terminal.
                               1175                 :                :              *
                               1176                 :                :              * Various cases here are Asserted to be not possible, as the
                               1177                 :                :              * token would not appear at the top of the prediction stack
                               1178                 :                :              * unless the lookahead matched.
                               1179                 :                :              */
  545 andrew@dunslane.net      1180   [ +  -  -  -  :CBC         441 :             switch (top)
                                     -  +  +  +  +  
                                           +  +  + ]
                               1181                 :                :             {
                               1182                 :             76 :                 case JSON_TOKEN_STRING:
                               1183         [ +  - ]:             76 :                     if (next_prediction(pstack) == JSON_TOKEN_COLON)
                               1184                 :             76 :                         ctx = JSON_PARSE_STRING;
                               1185                 :                :                     else
                               1186                 :                :                     {
  545 andrew@dunslane.net      1187                 :UBC           0 :                         Assert(false);
                               1188                 :                :                         ctx = JSON_PARSE_VALUE;
                               1189                 :                :                     }
  545 andrew@dunslane.net      1190                 :CBC          76 :                     break;
  545 andrew@dunslane.net      1191                 :UBC           0 :                 case JSON_TOKEN_NUMBER:
                               1192                 :                :                 case JSON_TOKEN_TRUE:
                               1193                 :                :                 case JSON_TOKEN_FALSE:
                               1194                 :                :                 case JSON_TOKEN_NULL:
                               1195                 :                :                 case JSON_TOKEN_ARRAY_START:
                               1196                 :                :                 case JSON_TOKEN_OBJECT_START:
                               1197                 :              0 :                     Assert(false);
                               1198                 :                :                     ctx = JSON_PARSE_VALUE;
                               1199                 :                :                     break;
                               1200                 :              0 :                 case JSON_TOKEN_ARRAY_END:
                               1201                 :              0 :                     Assert(false);
                               1202                 :                :                     ctx = JSON_PARSE_ARRAY_NEXT;
                               1203                 :                :                     break;
                               1204                 :              0 :                 case JSON_TOKEN_OBJECT_END:
                               1205                 :              0 :                     Assert(false);
                               1206                 :                :                     ctx = JSON_PARSE_OBJECT_NEXT;
                               1207                 :                :                     break;
                               1208                 :              0 :                 case JSON_TOKEN_COMMA:
                               1209                 :              0 :                     Assert(false);
                               1210                 :                :                     if (next_prediction(pstack) == JSON_TOKEN_STRING)
                               1211                 :                :                         ctx = JSON_PARSE_OBJECT_NEXT;
                               1212                 :                :                     else
                               1213                 :                :                         ctx = JSON_PARSE_ARRAY_NEXT;
                               1214                 :                :                     break;
  545 andrew@dunslane.net      1215                 :CBC          52 :                 case JSON_TOKEN_COLON:
                               1216                 :             52 :                     ctx = JSON_PARSE_OBJECT_LABEL;
                               1217                 :             52 :                     break;
                               1218                 :             12 :                 case JSON_TOKEN_END:
                               1219                 :             12 :                     ctx = JSON_PARSE_END;
                               1220                 :             12 :                     break;
                               1221                 :             36 :                 case JSON_NT_MORE_ARRAY_ELEMENTS:
                               1222                 :             36 :                     ctx = JSON_PARSE_ARRAY_NEXT;
                               1223                 :             36 :                     break;
                               1224                 :             28 :                 case JSON_NT_ARRAY_ELEMENTS:
                               1225                 :             28 :                     ctx = JSON_PARSE_ARRAY_START;
                               1226                 :             28 :                     break;
                               1227                 :            140 :                 case JSON_NT_MORE_KEY_PAIRS:
                               1228                 :            140 :                     ctx = JSON_PARSE_OBJECT_NEXT;
                               1229                 :            140 :                     break;
                               1230                 :             60 :                 case JSON_NT_KEY_PAIRS:
                               1231                 :             60 :                     ctx = JSON_PARSE_OBJECT_START;
                               1232                 :             60 :                     break;
                               1233                 :             37 :                 default:
                               1234                 :             37 :                     ctx = JSON_PARSE_VALUE;
                               1235                 :                :             }
                               1236                 :            441 :             return report_parse_error(ctx, lex);
                               1237                 :                :         }
                               1238                 :                :     }
                               1239                 :                : 
                               1240                 :           1132 :     return JSON_SUCCESS;
                               1241                 :                : }
                               1242                 :                : 
                               1243                 :                : /*
                               1244                 :                :  *  Recursive Descent parse routines. There is one for each structural
                               1245                 :                :  *  element in a json document:
                               1246                 :                :  *    - scalar (string, number, true, false, null)
                               1247                 :                :  *    - array  ( [ ] )
                               1248                 :                :  *    - array element
                               1249                 :                :  *    - object ( { } )
                               1250                 :                :  *    - object field
                               1251                 :                :  */
                               1252                 :                : static inline JsonParseErrorType
  396 heikki.linnakangas@i     1253                 :         168674 : parse_scalar(JsonLexContext *lex, const JsonSemAction *sem)
                               1254                 :                : {
 2052 rhaas@postgresql.org     1255                 :         168674 :     char       *val = NULL;
                               1256                 :         168674 :     json_scalar_action sfunc = sem->scalar;
                               1257                 :         168674 :     JsonTokenType tok = lex_peek(lex);
                               1258                 :                :     JsonParseErrorType result;
                               1259                 :                : 
                               1260                 :                :     /* a scalar must be a string, a number, true, false, or null */
 2059                          1261   [ +  +  +  +  :         168674 :     if (tok != JSON_TOKEN_STRING && tok != JSON_TOKEN_NUMBER &&
                                              +  + ]
                               1262   [ +  +  +  + ]:          16546 :         tok != JSON_TOKEN_TRUE && tok != JSON_TOKEN_FALSE &&
                               1263                 :                :         tok != JSON_TOKEN_NULL)
 2049                          1264                 :             94 :         return report_parse_error(JSON_PARSE_VALUE, lex);
                               1265                 :                : 
                               1266                 :                :     /* if no semantic function, just consume the token */
 2059                          1267         [ +  + ]:         168580 :     if (sfunc == NULL)
 2049                          1268                 :           5912 :         return json_lex(lex);
                               1269                 :                : 
                               1270                 :                :     /* extract the de-escaped string value, or the raw lexeme */
 2059                          1271         [ +  + ]:         162668 :     if (lex_peek(lex) == JSON_TOKEN_STRING)
                               1272                 :                :     {
  360 peter@eisentraut.org     1273         [ +  + ]:          38673 :         if (lex->need_escapes)
                               1274                 :                :         {
                               1275                 :          35835 :             val = STRDUP(lex->strval->data);
                               1276         [ -  + ]:          35835 :             if (val == NULL)
  360 peter@eisentraut.org     1277                 :UBC           0 :                 return JSON_OUT_OF_MEMORY;
                               1278                 :                :         }
                               1279                 :                :     }
                               1280                 :                :     else
                               1281                 :                :     {
 2059 rhaas@postgresql.org     1282                 :CBC      123995 :         int         len = (lex->token_terminator - lex->token_start);
                               1283                 :                : 
  360 peter@eisentraut.org     1284                 :         123995 :         val = ALLOC(len + 1);
                               1285         [ -  + ]:         123995 :         if (val == NULL)
  360 peter@eisentraut.org     1286                 :UBC           0 :             return JSON_OUT_OF_MEMORY;
                               1287                 :                : 
 2059 rhaas@postgresql.org     1288                 :CBC      123995 :         memcpy(val, lex->token_start, len);
                               1289                 :         123995 :         val[len] = '\0';
                               1290                 :                :     }
                               1291                 :                : 
                               1292                 :                :     /* consume the token */
 2049                          1293                 :         162668 :     result = json_lex(lex);
                               1294         [ -  + ]:         162668 :     if (result != JSON_SUCCESS)
                               1295                 :                :     {
  283 andrew@dunslane.net      1296         [ #  # ]:UBC           0 :         FREE(val);
 2049 rhaas@postgresql.org     1297                 :              0 :         return result;
                               1298                 :                :     }
                               1299                 :                : 
                               1300                 :                :     /*
                               1301                 :                :      * invoke the callback, which may take ownership of val. For string
                               1302                 :                :      * values, val is NULL if need_escapes is false.
                               1303                 :                :      */
 1000 tgl@sss.pgh.pa.us        1304                 :CBC      162668 :     result = (*sfunc) (sem->semstate, val, tok);
                               1305                 :                : 
  283 andrew@dunslane.net      1306         [ +  + ]:         162620 :     if (lex->flags & JSONLEX_CTX_OWNS_TOKENS)
                               1307         [ #  # ]:            888 :         FREE(val);
                               1308                 :                : 
 1000 tgl@sss.pgh.pa.us        1309                 :         162620 :     return result;
                               1310                 :                : }
                               1311                 :                : 
                               1312                 :                : static JsonParseErrorType
  396 heikki.linnakangas@i     1313                 :         163529 : parse_object_field(JsonLexContext *lex, const JsonSemAction *sem)
                               1314                 :                : {
                               1315                 :                :     /*
                               1316                 :                :      * An object field is "fieldname" : value where value can be a scalar,
                               1317                 :                :      * object or array.  Note: in user-facing docs and error messages, we
                               1318                 :                :      * generally call a field name a "key".
                               1319                 :                :      */
                               1320                 :                : 
  283 andrew@dunslane.net      1321                 :         163529 :     char       *fname = NULL;
 2052 rhaas@postgresql.org     1322                 :         163529 :     json_ofield_action ostart = sem->object_field_start;
                               1323                 :         163529 :     json_ofield_action oend = sem->object_field_end;
                               1324                 :                :     bool        isnull;
                               1325                 :                :     JsonTokenType tok;
                               1326                 :                :     JsonParseErrorType result;
                               1327                 :                : 
 2059                          1328         [ +  + ]:         163529 :     if (lex_peek(lex) != JSON_TOKEN_STRING)
 2049                          1329                 :              6 :         return report_parse_error(JSON_PARSE_STRING, lex);
  360 peter@eisentraut.org     1330   [ +  +  -  +  :         163523 :     if ((ostart != NULL || oend != NULL) && lex->need_escapes)
                                              +  - ]
                               1331                 :                :     {
                               1332                 :                :         /* fname is NULL if need_escapes is false */
                               1333                 :         129434 :         fname = STRDUP(lex->strval->data);
                               1334         [ -  + ]:         129434 :         if (fname == NULL)
  360 peter@eisentraut.org     1335                 :UBC           0 :             return JSON_OUT_OF_MEMORY;
                               1336                 :                :     }
 2049 rhaas@postgresql.org     1337                 :CBC      163523 :     result = json_lex(lex);
                               1338         [ +  + ]:         163523 :     if (result != JSON_SUCCESS)
                               1339                 :                :     {
  283 andrew@dunslane.net      1340         [ +  + ]:              6 :         FREE(fname);
 2049 rhaas@postgresql.org     1341                 :              6 :         return result;
                               1342                 :                :     }
                               1343                 :                : 
                               1344                 :         163517 :     result = lex_expect(JSON_PARSE_OBJECT_LABEL, lex, JSON_TOKEN_COLON);
                               1345         [ +  + ]:         163517 :     if (result != JSON_SUCCESS)
                               1346                 :                :     {
  283 andrew@dunslane.net      1347         [ +  + ]:             45 :         FREE(fname);
 2049 rhaas@postgresql.org     1348                 :             45 :         return result;
                               1349                 :                :     }
                               1350                 :                : 
 2052                          1351                 :         163472 :     tok = lex_peek(lex);
                               1352                 :         163472 :     isnull = tok == JSON_TOKEN_NULL;
                               1353                 :                : 
                               1354         [ +  + ]:         163472 :     if (ostart != NULL)
                               1355                 :                :     {
 1000 tgl@sss.pgh.pa.us        1356                 :         129392 :         result = (*ostart) (sem->semstate, fname, isnull);
                               1357         [ -  + ]:         129388 :         if (result != JSON_SUCCESS)
  283 andrew@dunslane.net      1358                 :UBC           0 :             goto ofield_cleanup;
                               1359                 :                :     }
                               1360                 :                : 
 2052 rhaas@postgresql.org     1361      [ +  +  + ]:CBC      163468 :     switch (tok)
                               1362                 :                :     {
                               1363                 :           5642 :         case JSON_TOKEN_OBJECT_START:
 2049                          1364                 :           5642 :             result = parse_object(lex, sem);
 2052                          1365                 :           2190 :             break;
                               1366                 :           7846 :         case JSON_TOKEN_ARRAY_START:
 2049                          1367                 :           7846 :             result = parse_array(lex, sem);
 2052                          1368                 :           7829 :             break;
                               1369                 :         149980 :         default:
 2049                          1370                 :         149980 :             result = parse_scalar(lex, sem);
                               1371                 :                :     }
                               1372         [ +  + ]:         159996 :     if (result != JSON_SUCCESS)
  283 andrew@dunslane.net      1373                 :             38 :         goto ofield_cleanup;
                               1374                 :                : 
 2052 rhaas@postgresql.org     1375         [ +  + ]:         159958 :     if (oend != NULL)
                               1376                 :                :     {
 1000 tgl@sss.pgh.pa.us        1377                 :          91705 :         result = (*oend) (sem->semstate, fname, isnull);
                               1378         [ +  - ]:          91705 :         if (result != JSON_SUCCESS)
  283 andrew@dunslane.net      1379                 :UBC           0 :             goto ofield_cleanup;
                               1380                 :                :     }
                               1381                 :                : 
  283 andrew@dunslane.net      1382                 :CBC      159958 : ofield_cleanup:
                               1383         [ +  + ]:         159996 :     if (lex->flags & JSONLEX_CTX_OWNS_TOKENS)
                               1384         [ #  # ]:            872 :         FREE(fname);
                               1385                 :         159996 :     return result;
                               1386                 :                : }
                               1387                 :                : 
                               1388                 :                : static JsonParseErrorType
  396 heikki.linnakangas@i     1389                 :          25674 : parse_object(JsonLexContext *lex, const JsonSemAction *sem)
                               1390                 :                : {
                               1391                 :                :     /*
                               1392                 :                :      * an object is a possibly empty sequence of object fields, separated by
                               1393                 :                :      * commas and surrounded by curly braces.
                               1394                 :                :      */
 2052 rhaas@postgresql.org     1395                 :          25674 :     json_struct_action ostart = sem->object_start;
                               1396                 :          25674 :     json_struct_action oend = sem->object_end;
                               1397                 :                :     JsonTokenType tok;
                               1398                 :                :     JsonParseErrorType result;
                               1399                 :                : 
                               1400                 :                : #ifndef FRONTEND
                               1401                 :                : 
                               1402                 :                :     /*
                               1403                 :                :      * TODO: clients need some way to put a bound on stack growth. Parse level
                               1404                 :                :      * limits maybe?
                               1405                 :                :      */
                               1406                 :          22475 :     check_stack_depth();
                               1407                 :                : #endif
                               1408                 :                : 
                               1409         [ +  + ]:          25668 :     if (ostart != NULL)
                               1410                 :                :     {
 1000 tgl@sss.pgh.pa.us        1411                 :          15942 :         result = (*ostart) (sem->semstate);
                               1412         [ +  + ]:          15932 :         if (result != JSON_SUCCESS)
                               1413                 :              1 :             return result;
                               1414                 :                :     }
                               1415                 :                : 
                               1416                 :                :     /*
                               1417                 :                :      * Data inside an object is at a higher nesting level than the object
                               1418                 :                :      * itself. Note that we increment this after we call the semantic routine
                               1419                 :                :      * for the object start and restore it before we call the routine for the
                               1420                 :                :      * object end.
                               1421                 :                :      */
 2052 rhaas@postgresql.org     1422                 :          25657 :     lex->lex_level++;
                               1423                 :                : 
 2059                          1424         [ -  + ]:          25657 :     Assert(lex_peek(lex) == JSON_TOKEN_OBJECT_START);
 2049                          1425                 :          25657 :     result = json_lex(lex);
                               1426         [ +  + ]:          25657 :     if (result != JSON_SUCCESS)
                               1427                 :             31 :         return result;
                               1428                 :                : 
 2052                          1429                 :          25626 :     tok = lex_peek(lex);
                               1430      [ +  +  + ]:          25626 :     switch (tok)
                               1431                 :                :     {
                               1432                 :          24137 :         case JSON_TOKEN_STRING:
 2049                          1433                 :          24137 :             result = parse_object_field(lex, sem);
                               1434   [ +  +  +  + ]:         160053 :             while (result == JSON_SUCCESS && lex_peek(lex) == JSON_TOKEN_COMMA)
                               1435                 :                :             {
                               1436                 :         139392 :                 result = json_lex(lex);
                               1437         [ -  + ]:         139392 :                 if (result != JSON_SUCCESS)
 2049 rhaas@postgresql.org     1438                 :UBC           0 :                     break;
 2049 rhaas@postgresql.org     1439                 :CBC      139392 :                 result = parse_object_field(lex, sem);
                               1440                 :                :             }
 2052                          1441                 :          20661 :             break;
                               1442                 :           1482 :         case JSON_TOKEN_OBJECT_END:
                               1443                 :           1482 :             break;
                               1444                 :              7 :         default:
                               1445                 :                :             /* case of an invalid initial token inside the object */
 2049                          1446                 :              7 :             result = report_parse_error(JSON_PARSE_OBJECT_START, lex);
                               1447                 :                :     }
                               1448         [ +  + ]:          22150 :     if (result != JSON_SUCCESS)
                               1449                 :            102 :         return result;
                               1450                 :                : 
                               1451                 :          22048 :     result = lex_expect(JSON_PARSE_OBJECT_NEXT, lex, JSON_TOKEN_OBJECT_END);
                               1452         [ +  + ]:          22048 :     if (result != JSON_SUCCESS)
                               1453                 :             18 :         return result;
                               1454                 :                : 
 2052                          1455                 :          22030 :     lex->lex_level--;
                               1456                 :                : 
                               1457         [ +  + ]:          22030 :     if (oend != NULL)
                               1458                 :                :     {
 1000 tgl@sss.pgh.pa.us        1459                 :          12944 :         result = (*oend) (sem->semstate);
                               1460         [ -  + ]:          12921 :         if (result != JSON_SUCCESS)
 1000 tgl@sss.pgh.pa.us        1461                 :UBC           0 :             return result;
                               1462                 :                :     }
                               1463                 :                : 
 2049 rhaas@postgresql.org     1464                 :CBC       22007 :     return JSON_SUCCESS;
                               1465                 :                : }
                               1466                 :                : 
                               1467                 :                : static JsonParseErrorType
  396 heikki.linnakangas@i     1468                 :          29534 : parse_array_element(JsonLexContext *lex, const JsonSemAction *sem)
                               1469                 :                : {
 2052 rhaas@postgresql.org     1470                 :          29534 :     json_aelem_action astart = sem->array_element_start;
                               1471                 :          29534 :     json_aelem_action aend = sem->array_element_end;
                               1472                 :          29534 :     JsonTokenType tok = lex_peek(lex);
                               1473                 :                :     JsonParseErrorType result;
                               1474                 :                :     bool        isnull;
                               1475                 :                : 
                               1476                 :          29534 :     isnull = tok == JSON_TOKEN_NULL;
                               1477                 :                : 
                               1478         [ +  + ]:          29534 :     if (astart != NULL)
                               1479                 :                :     {
 1000 tgl@sss.pgh.pa.us        1480                 :           3902 :         result = (*astart) (sem->semstate, isnull);
                               1481         [ -  + ]:           3902 :         if (result != JSON_SUCCESS)
 1000 tgl@sss.pgh.pa.us        1482                 :UBC           0 :             return result;
                               1483                 :                :     }
                               1484                 :                : 
                               1485                 :                :     /* an array element is any object, array or scalar */
 2052 rhaas@postgresql.org     1486      [ +  +  + ]:CBC       29534 :     switch (tok)
                               1487                 :                :     {
                               1488                 :           9448 :         case JSON_TOKEN_OBJECT_START:
 2049                          1489                 :           9448 :             result = parse_object(lex, sem);
 2052                          1490                 :           9419 :             break;
                               1491                 :           6481 :         case JSON_TOKEN_ARRAY_START:
 2049                          1492                 :           6481 :             result = parse_array(lex, sem);
 2052                          1493                 :           2156 :             break;
                               1494                 :          13605 :         default:
 2049                          1495                 :          13605 :             result = parse_scalar(lex, sem);
                               1496                 :                :     }
                               1497                 :                : 
                               1498         [ +  + ]:          25171 :     if (result != JSON_SUCCESS)
                               1499                 :             48 :         return result;
                               1500                 :                : 
 2052                          1501         [ +  + ]:          25123 :     if (aend != NULL)
                               1502                 :                :     {
 1000 tgl@sss.pgh.pa.us        1503                 :           3614 :         result = (*aend) (sem->semstate, isnull);
                               1504         [ -  + ]:           3608 :         if (result != JSON_SUCCESS)
 1000 tgl@sss.pgh.pa.us        1505                 :UBC           0 :             return result;
                               1506                 :                :     }
                               1507                 :                : 
 2049 rhaas@postgresql.org     1508                 :CBC       25117 :     return JSON_SUCCESS;
                               1509                 :                : }
                               1510                 :                : 
                               1511                 :                : static JsonParseErrorType
  396 heikki.linnakangas@i     1512                 :          18030 : parse_array(JsonLexContext *lex, const JsonSemAction *sem)
                               1513                 :                : {
                               1514                 :                :     /*
                               1515                 :                :      * an array is a possibly empty sequence of array elements, separated by
                               1516                 :                :      * commas and surrounded by square brackets.
                               1517                 :                :      */
 2052 rhaas@postgresql.org     1518                 :          18030 :     json_struct_action astart = sem->array_start;
                               1519                 :          18030 :     json_struct_action aend = sem->array_end;
                               1520                 :                :     JsonParseErrorType result;
                               1521                 :                : 
                               1522                 :                : #ifndef FRONTEND
                               1523                 :          17779 :     check_stack_depth();
                               1524                 :                : #endif
                               1525                 :                : 
                               1526         [ +  + ]:          18024 :     if (astart != NULL)
                               1527                 :                :     {
 1000 tgl@sss.pgh.pa.us        1528                 :           9068 :         result = (*astart) (sem->semstate);
                               1529         [ +  + ]:           9061 :         if (result != JSON_SUCCESS)
                               1530                 :              1 :             return result;
                               1531                 :                :     }
                               1532                 :                : 
                               1533                 :                :     /*
                               1534                 :                :      * Data inside an array is at a higher nesting level than the array
                               1535                 :                :      * itself. Note that we increment this after we call the semantic routine
                               1536                 :                :      * for the array start and restore it before we call the routine for the
                               1537                 :                :      * array end.
                               1538                 :                :      */
 2052 rhaas@postgresql.org     1539                 :          18016 :     lex->lex_level++;
                               1540                 :                : 
 2049                          1541                 :          18016 :     result = lex_expect(JSON_PARSE_ARRAY_START, lex, JSON_TOKEN_ARRAY_START);
                               1542   [ +  -  +  + ]:          18016 :     if (result == JSON_SUCCESS && lex_peek(lex) != JSON_TOKEN_ARRAY_END)
                               1543                 :                :     {
                               1544                 :          14280 :         result = parse_array_element(lex, sem);
                               1545                 :                : 
                               1546   [ +  +  +  + ]:          25141 :         while (result == JSON_SUCCESS && lex_peek(lex) == JSON_TOKEN_COMMA)
                               1547                 :                :         {
                               1548                 :          15230 :             result = json_lex(lex);
                               1549         [ -  + ]:          15230 :             if (result != JSON_SUCCESS)
 2049 rhaas@postgresql.org     1550                 :UBC           0 :                 break;
 2049 rhaas@postgresql.org     1551                 :CBC       15230 :             result = parse_array_element(lex, sem);
                               1552                 :                :         }
                               1553                 :                :     }
                               1554         [ +  + ]:          13647 :     if (result != JSON_SUCCESS)
                               1555                 :             48 :         return result;
                               1556                 :                : 
                               1557                 :          13599 :     result = lex_expect(JSON_PARSE_ARRAY_NEXT, lex, JSON_TOKEN_ARRAY_END);
                               1558         [ +  + ]:          13599 :     if (result != JSON_SUCCESS)
                               1559                 :             12 :         return result;
                               1560                 :                : 
 2052                          1561                 :          13587 :     lex->lex_level--;
                               1562                 :                : 
                               1563         [ +  + ]:          13587 :     if (aend != NULL)
                               1564                 :                :     {
 1000 tgl@sss.pgh.pa.us        1565                 :           5787 :         result = (*aend) (sem->semstate);
                               1566         [ -  + ]:           5775 :         if (result != JSON_SUCCESS)
 1000 tgl@sss.pgh.pa.us        1567                 :UBC           0 :             return result;
                               1568                 :                :     }
                               1569                 :                : 
 2049 rhaas@postgresql.org     1570                 :CBC       13575 :     return JSON_SUCCESS;
                               1571                 :                : }
                               1572                 :                : 
                               1573                 :                : /*
                               1574                 :                :  * Lex one token from the input stream.
                               1575                 :                :  *
                               1576                 :                :  * When doing incremental parsing, we can reach the end of the input string
                               1577                 :                :  * without having (or knowing we have) a complete token. If it's not the
                               1578                 :                :  * final chunk of input, the partial token is then saved to the lex
                               1579                 :                :  * structure's ptok StringInfo. On subsequent calls input is appended to this
                               1580                 :                :  * buffer until we have something that we think is a complete token,
                               1581                 :                :  * which is then lexed using a recursive call to json_lex. Processing then
                               1582                 :                :  * continues as normal on subsequent calls.
                               1583                 :                :  *
                               1584                 :                :  * Note than when doing incremental processing, the lex.prev_token_terminator
                               1585                 :                :  * should not be relied on. It could point into a previous input chunk or
                               1586                 :                :  * worse.
                               1587                 :                :  */
                               1588                 :                : JsonParseErrorType
 2052                          1589                 :        7325846 : json_lex(JsonLexContext *lex)
                               1590                 :                : {
                               1591                 :                :     const char *s;
  442 peter@eisentraut.org     1592                 :        7325846 :     const char *const end = lex->input + lex->input_length;
                               1593                 :                :     JsonParseErrorType result;
                               1594                 :                : 
  360                          1595   [ +  -  -  + ]:        7325846 :     if (lex == &failed_oom || lex->inc_state == &failed_inc_oom)
  360 peter@eisentraut.org     1596                 :UBC           0 :         return JSON_OUT_OF_MEMORY;
                               1597                 :                : 
  360 peter@eisentraut.org     1598         [ +  + ]:CBC     7325846 :     if (lex->incremental)
                               1599                 :                :     {
                               1600         [ +  + ]:        6522654 :         if (lex->inc_state->partial_completed)
                               1601                 :                :         {
                               1602                 :                :             /*
                               1603                 :                :              * We just lexed a completed partial token on the last call, so
                               1604                 :                :              * reset everything
                               1605                 :                :              */
                               1606                 :          34466 :             jsonapi_resetStringInfo(&(lex->inc_state->partial_token));
                               1607                 :          34466 :             lex->token_terminator = lex->input;
                               1608                 :          34466 :             lex->inc_state->partial_completed = false;
                               1609                 :                :         }
                               1610                 :                : 
                               1611                 :                : #ifdef JSONAPI_USE_PQEXPBUFFER
                               1612                 :                :         /* Make sure our partial token buffer is valid before using it below. */
                               1613         [ -  + ]:        1915760 :         if (PQExpBufferDataBroken(lex->inc_state->partial_token))
  360 peter@eisentraut.org     1614                 :UBC           0 :             return JSON_OUT_OF_MEMORY;
                               1615                 :                : #endif
                               1616                 :                :     }
                               1617                 :                : 
 2052 rhaas@postgresql.org     1618                 :CBC     7325846 :     s = lex->token_terminator;
                               1619                 :                : 
  545 andrew@dunslane.net      1620   [ +  +  +  + ]:        7325846 :     if (lex->incremental && lex->inc_state->partial_token.len)
                               1621                 :                :     {
                               1622                 :                :         /*
                               1623                 :                :          * We have a partial token. Extend it and if completed lex it by a
                               1624                 :                :          * recursive call
                               1625                 :                :          */
  360 peter@eisentraut.org     1626                 :          83238 :         jsonapi_StrValType *ptok = &(lex->inc_state->partial_token);
  442                          1627                 :          83238 :         size_t      added = 0;
  545 andrew@dunslane.net      1628                 :          83238 :         bool        tok_done = false;
  324 peter@eisentraut.org     1629                 :          83238 :         JsonLexContext dummy_lex = {0};
                               1630                 :                :         JsonParseErrorType partial_result;
                               1631                 :                : 
  545 andrew@dunslane.net      1632         [ +  + ]:          83238 :         if (ptok->data[0] == '"')
                               1633                 :                :         {
                               1634                 :                :             /*
                               1635                 :                :              * It's a string. Accumulate characters until we reach an
                               1636                 :                :              * unescaped '"'.
                               1637                 :                :              */
                               1638                 :          80109 :             int         escapes = 0;
                               1639                 :                : 
                               1640         [ +  + ]:          81393 :             for (int i = ptok->len - 1; i > 0; i--)
                               1641                 :                :             {
                               1642                 :                :                 /* count the trailing backslashes on the partial token */
                               1643         [ +  + ]:          75698 :                 if (ptok->data[i] == '\\')
                               1644                 :           1284 :                     escapes++;
                               1645                 :                :                 else
                               1646                 :          74414 :                     break;
                               1647                 :                :             }
                               1648                 :                : 
  442 peter@eisentraut.org     1649         [ +  + ]:         607638 :             for (size_t i = 0; i < lex->input_length; i++)
                               1650                 :                :             {
  545 andrew@dunslane.net      1651                 :         560066 :                 char        c = lex->input[i];
                               1652                 :                : 
  360 peter@eisentraut.org     1653         [ -  + ]:         560066 :                 jsonapi_appendStringInfoCharMacro(ptok, c);
  545 andrew@dunslane.net      1654                 :         560066 :                 added++;
                               1655   [ +  +  +  + ]:         560066 :                 if (c == '"' && escapes % 2 == 0)
                               1656                 :                :                 {
                               1657                 :          32537 :                     tok_done = true;
                               1658                 :          32537 :                     break;
                               1659                 :                :                 }
                               1660         [ +  + ]:         527529 :                 if (c == '\\')
                               1661                 :           2048 :                     escapes++;
                               1662                 :                :                 else
                               1663                 :         525481 :                     escapes = 0;
                               1664                 :                :             }
                               1665                 :                :         }
                               1666                 :                :         else
                               1667                 :                :         {
                               1668                 :                :             /* not a string */
                               1669                 :           3129 :             char        c = ptok->data[0];
                               1670                 :                : 
                               1671   [ +  -  +  -  :           3129 :             if (c == '-' || (c >= '0' && c <= '9'))
                                              +  + ]
                               1672                 :                :             {
                               1673                 :                :                 /* for numbers look for possible numeric continuations */
                               1674                 :                : 
                               1675                 :            661 :                 bool        numend = false;
                               1676                 :                : 
  442 peter@eisentraut.org     1677   [ +  +  +  + ]:           1865 :                 for (size_t i = 0; i < lex->input_length && !numend; i++)
                               1678                 :                :                 {
  545 andrew@dunslane.net      1679                 :           1204 :                     char        cc = lex->input[i];
                               1680                 :                : 
                               1681         [ +  + ]:           1204 :                     switch (cc)
                               1682                 :                :                     {
                               1683                 :            811 :                         case '+':
                               1684                 :                :                         case '-':
                               1685                 :                :                         case 'e':
                               1686                 :                :                         case 'E':
                               1687                 :                :                         case '0':
                               1688                 :                :                         case '1':
                               1689                 :                :                         case '2':
                               1690                 :                :                         case '3':
                               1691                 :                :                         case '4':
                               1692                 :                :                         case '5':
                               1693                 :                :                         case '6':
                               1694                 :                :                         case '7':
                               1695                 :                :                         case '8':
                               1696                 :                :                         case '9':
                               1697                 :                :                             {
  360 peter@eisentraut.org     1698         [ -  + ]:            811 :                                 jsonapi_appendStringInfoCharMacro(ptok, cc);
  545 andrew@dunslane.net      1699                 :            811 :                                 added++;
                               1700                 :                :                             }
                               1701                 :            811 :                             break;
                               1702                 :            393 :                         default:
                               1703                 :            393 :                             numend = true;
                               1704                 :                :                     }
                               1705                 :                :                 }
                               1706                 :                :             }
                               1707                 :                : 
                               1708                 :                :             /*
                               1709                 :                :              * Add any remaining alphanumeric chars. This takes care of the
                               1710                 :                :              * {null, false, true} literals as well as any trailing
                               1711                 :                :              * alphanumeric junk on non-string tokens.
                               1712                 :                :              */
  442 peter@eisentraut.org     1713         [ +  + ]:           6453 :             for (size_t i = added; i < lex->input_length; i++)
                               1714                 :                :             {
  545 andrew@dunslane.net      1715                 :           5473 :                 char        cc = lex->input[i];
                               1716                 :                : 
                               1717   [ +  +  +  +  :           5473 :                 if (JSON_ALPHANUMERIC_CHAR(cc))
                                     +  +  +  -  +  
                                     +  +  -  +  -  
                                              -  + ]
                               1718                 :                :                 {
  360 peter@eisentraut.org     1719         [ -  + ]:           3324 :                     jsonapi_appendStringInfoCharMacro(ptok, cc);
  545 andrew@dunslane.net      1720                 :           3324 :                     added++;
                               1721                 :                :                 }
                               1722                 :                :                 else
                               1723                 :                :                 {
                               1724                 :           2149 :                     tok_done = true;
                               1725                 :           2149 :                     break;
                               1726                 :                :                 }
                               1727                 :                :             }
                               1728         [ +  + ]:           3129 :             if (added == lex->input_length &&
                               1729         [ +  + ]:            980 :                 lex->inc_state->is_last_chunk)
                               1730                 :                :             {
                               1731                 :             76 :                 tok_done = true;
                               1732                 :                :             }
                               1733                 :                :         }
                               1734                 :                : 
                               1735         [ +  + ]:          83238 :         if (!tok_done)
                               1736                 :                :         {
                               1737                 :                :             /* We should have consumed the whole chunk in this case. */
                               1738         [ -  + ]:          48476 :             Assert(added == lex->input_length);
                               1739                 :                : 
                               1740         [ +  + ]:          48476 :             if (!lex->inc_state->is_last_chunk)
                               1741                 :          48436 :                 return JSON_INCOMPLETE;
                               1742                 :                : 
                               1743                 :                :             /* json_errdetail() needs access to the accumulated token. */
                               1744                 :             40 :             lex->token_start = ptok->data;
                               1745                 :             40 :             lex->token_terminator = ptok->data + ptok->len;
                               1746                 :             40 :             return JSON_INVALID_TOKEN;
                               1747                 :                :         }
                               1748                 :                : 
                               1749                 :                :         /*
                               1750                 :                :          * Everything up to lex->input[added] has been added to the partial
                               1751                 :                :          * token, so move the input past it.
                               1752                 :                :          */
                               1753                 :          34762 :         lex->input += added;
                               1754                 :          34762 :         lex->input_length -= added;
                               1755                 :                : 
                               1756                 :          34762 :         dummy_lex.input = dummy_lex.token_terminator =
                               1757                 :          34762 :             dummy_lex.line_start = ptok->data;
                               1758                 :          34762 :         dummy_lex.line_number = lex->line_number;
                               1759                 :          34762 :         dummy_lex.input_length = ptok->len;
                               1760                 :          34762 :         dummy_lex.input_encoding = lex->input_encoding;
                               1761                 :          34762 :         dummy_lex.incremental = false;
  360 peter@eisentraut.org     1762                 :          34762 :         dummy_lex.need_escapes = lex->need_escapes;
  545 andrew@dunslane.net      1763                 :          34762 :         dummy_lex.strval = lex->strval;
                               1764                 :                : 
                               1765                 :          34762 :         partial_result = json_lex(&dummy_lex);
                               1766                 :                : 
                               1767                 :                :         /*
                               1768                 :                :          * We either have a complete token or an error. In either case we need
                               1769                 :                :          * to point to the partial token data for the semantic or error
                               1770                 :                :          * routines. If it's not an error we'll readjust on the next call to
                               1771                 :                :          * json_lex.
                               1772                 :                :          */
                               1773                 :          34762 :         lex->token_type = dummy_lex.token_type;
                               1774                 :          34762 :         lex->line_number = dummy_lex.line_number;
                               1775                 :                : 
                               1776                 :                :         /*
                               1777                 :                :          * We know the prev_token_terminator must be back in some previous
                               1778                 :                :          * piece of input, so we just make it NULL.
                               1779                 :                :          */
                               1780                 :          34762 :         lex->prev_token_terminator = NULL;
                               1781                 :                : 
                               1782                 :                :         /*
                               1783                 :                :          * Normally token_start would be ptok->data, but it could be later,
                               1784                 :                :          * see json_lex_string's handling of invalid escapes.
                               1785                 :                :          */
                               1786                 :          34762 :         lex->token_start = dummy_lex.token_start;
                               1787                 :          34762 :         lex->token_terminator = dummy_lex.token_terminator;
                               1788         [ +  + ]:          34762 :         if (partial_result == JSON_SUCCESS)
                               1789                 :                :         {
                               1790                 :                :             /* make sure we've used all the input */
                               1791         [ -  + ]:          34654 :             if (lex->token_terminator - lex->token_start != ptok->len)
                               1792                 :                :             {
  545 andrew@dunslane.net      1793                 :UBC           0 :                 Assert(false);
                               1794                 :                :                 return JSON_INVALID_TOKEN;
                               1795                 :                :             }
                               1796                 :                : 
  545 andrew@dunslane.net      1797                 :CBC       34654 :             lex->inc_state->partial_completed = true;
                               1798                 :                :         }
                               1799                 :          34762 :         return partial_result;
                               1800                 :                :         /* end of partial token processing */
                               1801                 :                :     }
                               1802                 :                : 
                               1803                 :                :     /* Skip leading whitespace. */
 1156 john.naylor@postgres     1804   [ +  +  +  +  :       11668953 :     while (s < end && (*s == ' ' || *s == '\t' || *s == '\n' || *s == '\r'))
                                     +  +  +  +  -  
                                                 + ]
                               1805                 :                :     {
 1650 tgl@sss.pgh.pa.us        1806         [ +  + ]:        4426345 :         if (*s++ == '\n')
                               1807                 :                :         {
 2052 rhaas@postgresql.org     1808                 :         295163 :             ++lex->line_number;
 1650 tgl@sss.pgh.pa.us        1809                 :         295163 :             lex->line_start = s;
                               1810                 :                :         }
                               1811                 :                :     }
 2052 rhaas@postgresql.org     1812                 :        7242608 :     lex->token_start = s;
                               1813                 :                : 
                               1814                 :                :     /* Determine token type. */
 1156 john.naylor@postgres     1815         [ +  + ]:        7242608 :     if (s >= end)
                               1816                 :                :     {
 2052 rhaas@postgresql.org     1817                 :         326985 :         lex->token_start = NULL;
                               1818                 :         326985 :         lex->prev_token_terminator = lex->token_terminator;
                               1819                 :         326985 :         lex->token_terminator = s;
                               1820                 :         326985 :         lex->token_type = JSON_TOKEN_END;
                               1821                 :                :     }
                               1822                 :                :     else
                               1823                 :                :     {
                               1824   [ +  +  +  +  :        6915623 :         switch (*s)
                                     +  +  +  +  +  
                                                 + ]
                               1825                 :                :         {
                               1826                 :                :                 /* Single-character token, some kind of punctuation mark. */
                               1827                 :         159048 :             case '{':
                               1828                 :         159048 :                 lex->prev_token_terminator = lex->token_terminator;
                               1829                 :         159048 :                 lex->token_terminator = s + 1;
                               1830                 :         159048 :                 lex->token_type = JSON_TOKEN_OBJECT_START;
                               1831                 :         159048 :                 break;
                               1832                 :         154948 :             case '}':
                               1833                 :         154948 :                 lex->prev_token_terminator = lex->token_terminator;
                               1834                 :         154948 :                 lex->token_terminator = s + 1;
                               1835                 :         154948 :                 lex->token_type = JSON_TOKEN_OBJECT_END;
                               1836                 :         154948 :                 break;
                               1837                 :        2479387 :             case '[':
                               1838                 :        2479387 :                 lex->prev_token_terminator = lex->token_terminator;
                               1839                 :        2479387 :                 lex->token_terminator = s + 1;
                               1840                 :        2479387 :                 lex->token_type = JSON_TOKEN_ARRAY_START;
                               1841                 :        2479387 :                 break;
                               1842                 :         836170 :             case ']':
                               1843                 :         836170 :                 lex->prev_token_terminator = lex->token_terminator;
                               1844                 :         836170 :                 lex->token_terminator = s + 1;
                               1845                 :         836170 :                 lex->token_type = JSON_TOKEN_ARRAY_END;
                               1846                 :         836170 :                 break;
                               1847                 :         802396 :             case ',':
                               1848                 :         802396 :                 lex->prev_token_terminator = lex->token_terminator;
                               1849                 :         802396 :                 lex->token_terminator = s + 1;
                               1850                 :         802396 :                 lex->token_type = JSON_TOKEN_COMMA;
                               1851                 :         802396 :                 break;
                               1852                 :         815963 :             case ':':
                               1853                 :         815963 :                 lex->prev_token_terminator = lex->token_terminator;
                               1854                 :         815963 :                 lex->token_terminator = s + 1;
                               1855                 :         815963 :                 lex->token_type = JSON_TOKEN_COLON;
                               1856                 :         815963 :                 break;
                               1857                 :        1405895 :             case '"':
                               1858                 :                :                 /* string */
 2049                          1859                 :        1405895 :                 result = json_lex_string(lex);
                               1860         [ +  + ]:        1405895 :                 if (result != JSON_SUCCESS)
                               1861                 :          32766 :                     return result;
 2052                          1862                 :        1373129 :                 lex->token_type = JSON_TOKEN_STRING;
                               1863                 :        1373129 :                 break;
                               1864                 :            177 :             case '-':
                               1865                 :                :                 /* Negative number. */
 2049                          1866                 :            177 :                 result = json_lex_number(lex, s + 1, NULL, NULL);
                               1867         [ -  + ]:            177 :                 if (result != JSON_SUCCESS)
 2049 rhaas@postgresql.org     1868                 :UBC           0 :                     return result;
 2052 rhaas@postgresql.org     1869                 :CBC         177 :                 lex->token_type = JSON_TOKEN_NUMBER;
                               1870                 :            177 :                 break;
                               1871                 :         231575 :             case '0':
                               1872                 :                :             case '1':
                               1873                 :                :             case '2':
                               1874                 :                :             case '3':
                               1875                 :                :             case '4':
                               1876                 :                :             case '5':
                               1877                 :                :             case '6':
                               1878                 :                :             case '7':
                               1879                 :                :             case '8':
                               1880                 :                :             case '9':
                               1881                 :                :                 /* Positive number. */
 2049                          1882                 :         231575 :                 result = json_lex_number(lex, s, NULL, NULL);
                               1883         [ +  + ]:         231575 :                 if (result != JSON_SUCCESS)
                               1884                 :            457 :                     return result;
 2052                          1885                 :         231118 :                 lex->token_type = JSON_TOKEN_NUMBER;
                               1886                 :         231118 :                 break;
                               1887                 :          30064 :             default:
                               1888                 :                :                 {
                               1889                 :                :                     const char *p;
                               1890                 :                : 
                               1891                 :                :                     /*
                               1892                 :                :                      * We're not dealing with a string, number, legal
                               1893                 :                :                      * punctuation mark, or end of string.  The only legal
                               1894                 :                :                      * tokens we might find here are true, false, and null,
                               1895                 :                :                      * but for error reporting purposes we scan until we see a
                               1896                 :                :                      * non-alphanumeric character.  That way, we can report
                               1897                 :                :                      * the whole word as an unexpected token, rather than just
                               1898                 :                :                      * some unintuitive prefix thereof.
                               1899                 :                :                      */
 1156 john.naylor@postgres     1900   [ +  +  +  +  :         162172 :                     for (p = s; p < end && JSON_ALPHANUMERIC_CHAR(*p); p++)
                                     +  +  +  +  -  
                                     +  +  +  -  +  
                                        -  +  -  + ]
                               1901                 :                :                          /* skip */ ;
                               1902                 :                : 
                               1903                 :                :                     /*
                               1904                 :                :                      * We got some sort of unexpected punctuation or an
                               1905                 :                :                      * otherwise unexpected character, so just complain about
                               1906                 :                :                      * that one character.
                               1907                 :                :                      */
 2052 rhaas@postgresql.org     1908         [ +  + ]:          30064 :                     if (p == s)
                               1909                 :                :                     {
                               1910                 :             40 :                         lex->prev_token_terminator = lex->token_terminator;
                               1911                 :             40 :                         lex->token_terminator = s + 1;
 2049                          1912                 :             40 :                         return JSON_INVALID_TOKEN;
                               1913                 :                :                     }
                               1914                 :                : 
  545 andrew@dunslane.net      1915   [ +  +  +  + ]:          30024 :                     if (lex->incremental && !lex->inc_state->is_last_chunk &&
                               1916         [ +  + ]:           7832 :                         p == lex->input + lex->input_length)
                               1917                 :                :                     {
  360 peter@eisentraut.org     1918                 :           1832 :                         jsonapi_appendBinaryStringInfo(&(lex->inc_state->partial_token), s, end - s);
  545 andrew@dunslane.net      1919                 :           1832 :                         return JSON_INCOMPLETE;
                               1920                 :                :                     }
                               1921                 :                : 
                               1922                 :                :                     /*
                               1923                 :                :                      * We've got a real alphanumeric token here.  If it
                               1924                 :                :                      * happens to be true, false, or null, all is well.  If
                               1925                 :                :                      * not, error out.
                               1926                 :                :                      */
 2052 rhaas@postgresql.org     1927                 :          28192 :                     lex->prev_token_terminator = lex->token_terminator;
                               1928                 :          28192 :                     lex->token_terminator = p;
                               1929         [ +  + ]:          28192 :                     if (p - s == 4)
                               1930                 :                :                     {
                               1931         [ +  + ]:          12988 :                         if (memcmp(s, "true", 4) == 0)
                               1932                 :           3688 :                             lex->token_type = JSON_TOKEN_TRUE;
                               1933         [ +  + ]:           9300 :                         else if (memcmp(s, "null", 4) == 0)
                               1934                 :           9294 :                             lex->token_type = JSON_TOKEN_NULL;
                               1935                 :                :                         else
 2049                          1936                 :              6 :                             return JSON_INVALID_TOKEN;
                               1937                 :                :                     }
 2052                          1938   [ +  +  +  + ]:          15204 :                     else if (p - s == 5 && memcmp(s, "false", 5) == 0)
                               1939                 :          15085 :                         lex->token_type = JSON_TOKEN_FALSE;
                               1940                 :                :                     else
 2049                          1941                 :            119 :                         return JSON_INVALID_TOKEN;
                               1942                 :                :                 }
                               1943                 :                :         }                       /* end of switch */
                               1944                 :                :     }
                               1945                 :                : 
  545 andrew@dunslane.net      1946   [ +  +  +  +  :        7207388 :     if (lex->incremental && lex->token_type == JSON_TOKEN_END && !lex->inc_state->is_last_chunk)
                                              +  + ]
                               1947                 :         287634 :         return JSON_INCOMPLETE;
                               1948                 :                :     else
                               1949                 :        6919754 :         return JSON_SUCCESS;
                               1950                 :                : }
                               1951                 :                : 
                               1952                 :                : /*
                               1953                 :                :  * The next token in the input stream is known to be a string; lex it.
                               1954                 :                :  *
                               1955                 :                :  * If lex->strval isn't NULL, fill it with the decoded string.
                               1956                 :                :  * Set lex->token_terminator to the end of the decoded input, and in
                               1957                 :                :  * success cases, transfer its previous value to lex->prev_token_terminator.
                               1958                 :                :  * Return JSON_SUCCESS or an error code.
                               1959                 :                :  *
                               1960                 :                :  * Note: be careful that all error exits advance lex->token_terminator
                               1961                 :                :  * to the point after the character we detected the error on.
                               1962                 :                :  */
                               1963                 :                : static inline JsonParseErrorType
 2052 rhaas@postgresql.org     1964                 :        1405895 : json_lex_string(JsonLexContext *lex)
                               1965                 :                : {
                               1966                 :                :     const char *s;
  442 peter@eisentraut.org     1967                 :        1405895 :     const char *const end = lex->input + lex->input_length;
 2052 rhaas@postgresql.org     1968                 :        1405895 :     int         hi_surrogate = -1;
                               1969                 :                : 
                               1970                 :                :     /* Convenience macros for error exits */
                               1971                 :                : #define FAIL_OR_INCOMPLETE_AT_CHAR_START(code) \
                               1972                 :                :     do { \
                               1973                 :                :         if (lex->incremental && !lex->inc_state->is_last_chunk) \
                               1974                 :                :         { \
                               1975                 :                :             jsonapi_appendBinaryStringInfo(&lex->inc_state->partial_token, \
                               1976                 :                :                                            lex->token_start, \
                               1977                 :                :                                            end - lex->token_start); \
                               1978                 :                :             return JSON_INCOMPLETE; \
                               1979                 :                :         } \
                               1980                 :                :         lex->token_terminator = s; \
                               1981                 :                :         return code; \
                               1982                 :                :     } while (0)
                               1983                 :                : #define FAIL_AT_CHAR_END(code) \
                               1984                 :                :     do { \
                               1985                 :                :         ptrdiff_t   remaining = end - s; \
                               1986                 :                :         int         charlen; \
                               1987                 :                :         charlen = pg_encoding_mblen_or_incomplete(lex->input_encoding, \
                               1988                 :                :                                                   s, remaining); \
                               1989                 :                :         lex->token_terminator = (charlen <= remaining) ? s + charlen : end; \
                               1990                 :                :         return code; \
                               1991                 :                :     } while (0)
                               1992                 :                : 
  360 peter@eisentraut.org     1993         [ +  + ]:        1405895 :     if (lex->need_escapes)
                               1994                 :                :     {
                               1995                 :                : #ifdef JSONAPI_USE_PQEXPBUFFER
                               1996                 :                :         /* make sure initialization succeeded */
                               1997         [ -  + ]:           2337 :         if (lex->strval == NULL)
  360 peter@eisentraut.org     1998                 :UBC           0 :             return JSON_OUT_OF_MEMORY;
                               1999                 :                : #endif
  360 peter@eisentraut.org     2000                 :CBC     1273763 :         jsonapi_resetStringInfo(lex->strval);
                               2001                 :                :     }
                               2002                 :                : 
 2052 rhaas@postgresql.org     2003         [ -  + ]:        1405895 :     Assert(lex->input_length > 0);
                               2004                 :        1405895 :     s = lex->token_start;
                               2005                 :                :     for (;;)
                               2006                 :                :     {
                               2007                 :        2811995 :         s++;
                               2008                 :                :         /* Premature end of the string. */
 1156 john.naylor@postgres     2009         [ +  + ]:        2811995 :         if (s >= end)
  545 andrew@dunslane.net      2010   [ +  +  +  + ]:          32363 :             FAIL_OR_INCOMPLETE_AT_CHAR_START(JSON_INVALID_TOKEN);
 1152 john.naylor@postgres     2011         [ +  + ]:        2779632 :         else if (*s == '"')
                               2012                 :        1373129 :             break;
 2052 rhaas@postgresql.org     2013         [ +  + ]:        1406503 :         else if (*s == '\\')
                               2014                 :                :         {
                               2015                 :                :             /* OK, we have an escape character. */
                               2016                 :           5129 :             s++;
 1156 john.naylor@postgres     2017         [ +  + ]:           5129 :             if (s >= end)
  545 andrew@dunslane.net      2018   [ +  -  +  - ]:             96 :                 FAIL_OR_INCOMPLETE_AT_CHAR_START(JSON_INVALID_TOKEN);
 2052 rhaas@postgresql.org     2019         [ +  + ]:           5033 :             else if (*s == 'u')
                               2020                 :                :             {
                               2021                 :                :                 int         i;
                               2022                 :           1943 :                 int         ch = 0;
                               2023                 :                : 
                               2024         [ +  + ]:           9341 :                 for (i = 1; i <= 4; i++)
                               2025                 :                :                 {
                               2026                 :           7545 :                     s++;
 1156 john.naylor@postgres     2027         [ +  + ]:           7545 :                     if (s >= end)
  545 andrew@dunslane.net      2028   [ +  -  +  - ]:            128 :                         FAIL_OR_INCOMPLETE_AT_CHAR_START(JSON_INVALID_TOKEN);
 2052 rhaas@postgresql.org     2029   [ +  +  +  + ]:           7417 :                     else if (*s >= '0' && *s <= '9')
                               2030                 :           4615 :                         ch = (ch * 16) + (*s - '0');
                               2031   [ +  +  +  + ]:           2802 :                     else if (*s >= 'a' && *s <= 'f')
                               2032                 :           2771 :                         ch = (ch * 16) + (*s - 'a') + 10;
                               2033   [ +  +  +  + ]:             31 :                     else if (*s >= 'A' && *s <= 'F')
                               2034                 :             12 :                         ch = (ch * 16) + (*s - 'A') + 10;
                               2035                 :                :                     else
  908 tgl@sss.pgh.pa.us        2036         [ +  + ]:             19 :                         FAIL_AT_CHAR_END(JSON_UNICODE_ESCAPE_FORMAT);
                               2037                 :                :                 }
  360 peter@eisentraut.org     2038         [ +  + ]:           1796 :                 if (lex->need_escapes)
                               2039                 :                :                 {
                               2040                 :                :                     /*
                               2041                 :                :                      * Combine surrogate pairs.
                               2042                 :                :                      */
 2010 tgl@sss.pgh.pa.us        2043         [ +  + ]:            130 :                     if (is_utf16_surrogate_first(ch))
                               2044                 :                :                     {
 2052 rhaas@postgresql.org     2045         [ +  + ]:             36 :                         if (hi_surrogate != -1)
  908 tgl@sss.pgh.pa.us        2046         [ +  - ]:              6 :                             FAIL_AT_CHAR_END(JSON_UNICODE_HIGH_SURROGATE);
 2010                          2047                 :             30 :                         hi_surrogate = ch;
 2052 rhaas@postgresql.org     2048                 :             30 :                         continue;
                               2049                 :                :                     }
 2010 tgl@sss.pgh.pa.us        2050         [ +  + ]:             94 :                     else if (is_utf16_surrogate_second(ch))
                               2051                 :                :                     {
 2052 rhaas@postgresql.org     2052         [ +  + ]:             30 :                         if (hi_surrogate == -1)
  908 tgl@sss.pgh.pa.us        2053         [ +  - ]:             12 :                             FAIL_AT_CHAR_END(JSON_UNICODE_LOW_SURROGATE);
 2010                          2054                 :             18 :                         ch = surrogate_pair_to_codepoint(hi_surrogate, ch);
 2052 rhaas@postgresql.org     2055                 :             18 :                         hi_surrogate = -1;
                               2056                 :                :                     }
                               2057                 :                : 
                               2058         [ -  + ]:             82 :                     if (hi_surrogate != -1)
  908 tgl@sss.pgh.pa.us        2059         [ #  # ]:UBC           0 :                         FAIL_AT_CHAR_END(JSON_UNICODE_LOW_SURROGATE);
                               2060                 :                : 
                               2061                 :                :                     /*
                               2062                 :                :                      * Reject invalid cases.  We can't have a value above
                               2063                 :                :                      * 0xFFFF here (since we only accepted 4 hex digits
                               2064                 :                :                      * above), so no need to test for out-of-range chars.
                               2065                 :                :                      */
 2052 rhaas@postgresql.org     2066         [ +  + ]:CBC          82 :                     if (ch == 0)
                               2067                 :                :                     {
                               2068                 :                :                         /* We can't allow this, since our TEXT type doesn't */
  908 tgl@sss.pgh.pa.us        2069         [ +  - ]:             12 :                         FAIL_AT_CHAR_END(JSON_UNICODE_CODE_POINT_ZERO);
                               2070                 :                :                     }
                               2071                 :                : 
                               2072                 :                :                     /*
                               2073                 :                :                      * Add the represented character to lex->strval.  In the
                               2074                 :                :                      * backend, we can let pg_unicode_to_server_noerror()
                               2075                 :                :                      * handle any required character set conversion; in
                               2076                 :                :                      * frontend, we can only deal with trivial conversions.
                               2077                 :                :                      */
                               2078                 :                : #ifndef FRONTEND
                               2079                 :                :                     {
                               2080                 :                :                         char        cbuf[MAX_UNICODE_EQUIVALENT_STRING + 1];
                               2081                 :                : 
 1000                          2082         [ -  + ]:             42 :                         if (!pg_unicode_to_server_noerror(ch, (unsigned char *) cbuf))
  908 tgl@sss.pgh.pa.us        2083         [ #  # ]:UBC           0 :                             FAIL_AT_CHAR_END(JSON_UNICODE_UNTRANSLATABLE);
 2010 tgl@sss.pgh.pa.us        2084                 :CBC          42 :                         appendStringInfoString(lex->strval, cbuf);
                               2085                 :                :                     }
                               2086                 :                : #else
                               2087         [ +  - ]:             28 :                     if (lex->input_encoding == PG_UTF8)
                               2088                 :                :                     {
                               2089                 :                :                         /* OK, we can map the code point to UTF8 easily */
                               2090                 :                :                         char        utf8str[5];
                               2091                 :                :                         int         utf8len;
                               2092                 :                : 
 2052 rhaas@postgresql.org     2093                 :             28 :                         unicode_to_utf8(ch, (unsigned char *) utf8str);
                               2094                 :             28 :                         utf8len = pg_utf_mblen((unsigned char *) utf8str);
  360 peter@eisentraut.org     2095                 :             28 :                         jsonapi_appendBinaryStringInfo(lex->strval, utf8str, utf8len);
                               2096                 :                :                     }
 2052 rhaas@postgresql.org     2097         [ #  # ]:UBC           0 :                     else if (ch <= 0x007f)
                               2098                 :                :                     {
                               2099                 :                :                         /* The ASCII range is the same in all encodings */
  360 peter@eisentraut.org     2100                 :              0 :                         jsonapi_appendStringInfoChar(lex->strval, (char) ch);
                               2101                 :                :                     }
                               2102                 :                :                     else
  908 tgl@sss.pgh.pa.us        2103         [ #  # ]:              0 :                         FAIL_AT_CHAR_END(JSON_UNICODE_HIGH_ESCAPE);
                               2104                 :                : #endif                          /* FRONTEND */
                               2105                 :                :                 }
                               2106                 :                :             }
  360 peter@eisentraut.org     2107         [ +  + ]:CBC        3090 :             else if (lex->need_escapes)
                               2108                 :                :             {
 2052 rhaas@postgresql.org     2109         [ -  + ]:            303 :                 if (hi_surrogate != -1)
  908 tgl@sss.pgh.pa.us        2110         [ #  # ]:UBC           0 :                     FAIL_AT_CHAR_END(JSON_UNICODE_LOW_SURROGATE);
                               2111                 :                : 
 2052 rhaas@postgresql.org     2112   [ +  +  +  +  :CBC         303 :                 switch (*s)
                                           +  +  + ]
                               2113                 :                :                 {
                               2114                 :            193 :                     case '"':
                               2115                 :                :                     case '\\':
                               2116                 :                :                     case '/':
  360 peter@eisentraut.org     2117                 :            193 :                         jsonapi_appendStringInfoChar(lex->strval, *s);
 2052 rhaas@postgresql.org     2118                 :            193 :                         break;
                               2119                 :             22 :                     case 'b':
  360 peter@eisentraut.org     2120                 :             22 :                         jsonapi_appendStringInfoChar(lex->strval, '\b');
 2052 rhaas@postgresql.org     2121                 :             22 :                         break;
                               2122                 :              4 :                     case 'f':
  360 peter@eisentraut.org     2123                 :              4 :                         jsonapi_appendStringInfoChar(lex->strval, '\f');
 2052 rhaas@postgresql.org     2124                 :              4 :                         break;
                               2125                 :             31 :                     case 'n':
  360 peter@eisentraut.org     2126                 :             31 :                         jsonapi_appendStringInfoChar(lex->strval, '\n');
 2052 rhaas@postgresql.org     2127                 :             31 :                         break;
                               2128                 :              4 :                     case 'r':
  360 peter@eisentraut.org     2129                 :              4 :                         jsonapi_appendStringInfoChar(lex->strval, '\r');
 2052 rhaas@postgresql.org     2130                 :              4 :                         break;
                               2131                 :             46 :                     case 't':
  360 peter@eisentraut.org     2132                 :             46 :                         jsonapi_appendStringInfoChar(lex->strval, '\t');
 2052 rhaas@postgresql.org     2133                 :             46 :                         break;
                               2134                 :              3 :                     default:
                               2135                 :                : 
                               2136                 :                :                         /*
                               2137                 :                :                          * Not a valid string escape, so signal error.  We
                               2138                 :                :                          * adjust token_start so that just the escape sequence
                               2139                 :                :                          * is reported, not the whole string.
                               2140                 :                :                          */
 2049                          2141                 :              3 :                         lex->token_start = s;
  908 tgl@sss.pgh.pa.us        2142         [ +  - ]:              3 :                         FAIL_AT_CHAR_END(JSON_ESCAPING_INVALID);
                               2143                 :                :                 }
                               2144                 :                :             }
 2052 rhaas@postgresql.org     2145         [ +  + ]:           2787 :             else if (strchr("\"\\/bfnrt", *s) == NULL)
                               2146                 :                :             {
                               2147                 :                :                 /*
                               2148                 :                :                  * Simpler processing if we're not bothered about de-escaping
                               2149                 :                :                  *
                               2150                 :                :                  * It's very tempting to remove the strchr() call here and
                               2151                 :                :                  * replace it with a switch statement, but testing so far has
                               2152                 :                :                  * shown it's not a performance win.
                               2153                 :                :                  */
 2049                          2154                 :             63 :                 lex->token_start = s;
  908 tgl@sss.pgh.pa.us        2155         [ +  + ]:             63 :                 FAIL_AT_CHAR_END(JSON_ESCAPING_INVALID);
                               2156                 :                :             }
                               2157                 :                :         }
                               2158                 :                :         else
                               2159                 :                :         {
  442 peter@eisentraut.org     2160                 :        1401374 :             const char *p = s;
                               2161                 :                : 
 2052 rhaas@postgresql.org     2162         [ +  + ]:        1401374 :             if (hi_surrogate != -1)
  908 tgl@sss.pgh.pa.us        2163         [ +  - ]:              6 :                 FAIL_AT_CHAR_END(JSON_UNICODE_LOW_SURROGATE);
                               2164                 :                : 
                               2165                 :                :             /*
                               2166                 :                :              * Skip to the first byte that requires special handling, so we
                               2167                 :                :              * can batch calls to jsonapi_appendBinaryStringInfo.
                               2168                 :                :              */
 1102 john.naylor@postgres     2169                 :        1401368 :             while (p < end - sizeof(Vector8) &&
                               2170         [ +  + ]:        1721054 :                    !pg_lfind8('\\', (uint8 *) p, sizeof(Vector8)) &&
                               2171   [ +  +  +  + ]:        3530893 :                    !pg_lfind8('"', (uint8 *) p, sizeof(Vector8)) &&
                               2172         [ +  - ]:         410280 :                    !pg_lfind8_le(31, (uint8 *) p, sizeof(Vector8)))
                               2173                 :         410280 :                 p += sizeof(Vector8);
                               2174                 :                : 
                               2175         [ +  + ]:       10868689 :             for (; p < end; p++)
                               2176                 :                :             {
 1163                          2177   [ +  +  +  + ]:       10841925 :                 if (*p == '\\' || *p == '"')
                               2178                 :                :                     break;
 1102                          2179         [ +  + ]:        9467379 :                 else if ((unsigned char) *p <= 31)
                               2180                 :                :                 {
                               2181                 :                :                     /* Per RFC4627, these characters MUST be escaped. */
                               2182                 :                :                     /*
                               2183                 :                :                      * Since *p isn't printable, exclude it from the context
                               2184                 :                :                      * string
                               2185                 :                :                      */
 1163                          2186                 :             58 :                     lex->token_terminator = p;
                               2187                 :             58 :                     return JSON_ESCAPING_REQUIRED;
                               2188                 :                :                 }
                               2189                 :                :             }
                               2190                 :                : 
  360 peter@eisentraut.org     2191         [ +  + ]:        1401310 :             if (lex->need_escapes)
                               2192                 :        1273813 :                 jsonapi_appendBinaryStringInfo(lex->strval, s, p - s);
                               2193                 :                : 
                               2194                 :                :             /*
                               2195                 :                :              * s will be incremented at the top of the loop, so set it to just
                               2196                 :                :              * behind our lookahead position
                               2197                 :                :              */
 1163 john.naylor@postgres     2198                 :        1401310 :             s = p - 1;
                               2199                 :                :         }
                               2200                 :                :     }
                               2201                 :                : 
 1152                          2202         [ -  + ]:        1373129 :     if (hi_surrogate != -1)
                               2203                 :                :     {
  908 tgl@sss.pgh.pa.us        2204                 :UBC           0 :         lex->token_terminator = s + 1;
 1152 john.naylor@postgres     2205                 :              0 :         return JSON_UNICODE_LOW_SURROGATE;
                               2206                 :                :     }
                               2207                 :                : 
                               2208                 :                : #ifdef JSONAPI_USE_PQEXPBUFFER
  360 peter@eisentraut.org     2209   [ +  +  +  -  :CBC       36117 :     if (lex->need_escapes && PQExpBufferBroken(lex->strval))
                                              -  + ]
  360 peter@eisentraut.org     2210                 :UBC           0 :         return JSON_OUT_OF_MEMORY;
                               2211                 :                : #endif
                               2212                 :                : 
                               2213                 :                :     /* Hooray, we found the end of the string! */
 1152 john.naylor@postgres     2214                 :CBC     1373129 :     lex->prev_token_terminator = lex->token_terminator;
                               2215                 :        1373129 :     lex->token_terminator = s + 1;
                               2216                 :        1373129 :     return JSON_SUCCESS;
                               2217                 :                : 
                               2218                 :                : #undef FAIL_OR_INCOMPLETE_AT_CHAR_START
                               2219                 :                : #undef FAIL_AT_CHAR_END
                               2220                 :                : }
                               2221                 :                : 
                               2222                 :                : /*
                               2223                 :                :  * The next token in the input stream is known to be a number; lex it.
                               2224                 :                :  *
                               2225                 :                :  * In JSON, a number consists of four parts:
                               2226                 :                :  *
                               2227                 :                :  * (1) An optional minus sign ('-').
                               2228                 :                :  *
                               2229                 :                :  * (2) Either a single '0', or a string of one or more digits that does not
                               2230                 :                :  *     begin with a '0'.
                               2231                 :                :  *
                               2232                 :                :  * (3) An optional decimal part, consisting of a period ('.') followed by
                               2233                 :                :  *     one or more digits.  (Note: While this part can be omitted
                               2234                 :                :  *     completely, it's not OK to have only the decimal point without
                               2235                 :                :  *     any digits afterwards.)
                               2236                 :                :  *
                               2237                 :                :  * (4) An optional exponent part, consisting of 'e' or 'E', optionally
                               2238                 :                :  *     followed by '+' or '-', followed by one or more digits.  (Note:
                               2239                 :                :  *     As with the decimal part, if 'e' or 'E' is present, it must be
                               2240                 :                :  *     followed by at least one digit.)
                               2241                 :                :  *
                               2242                 :                :  * The 's' argument to this function points to the ostensible beginning
                               2243                 :                :  * of part 2 - i.e. the character after any optional minus sign, or the
                               2244                 :                :  * first character of the string if there is none.
                               2245                 :                :  *
                               2246                 :                :  * If num_err is not NULL, we return an error flag to *num_err rather than
                               2247                 :                :  * raising an error for a badly-formed number.  Also, if total_len is not NULL
                               2248                 :                :  * the distance from lex->input to the token end+1 is returned to *total_len.
                               2249                 :                :  */
                               2250                 :                : static inline JsonParseErrorType
  442 peter@eisentraut.org     2251                 :         231775 : json_lex_number(JsonLexContext *lex, const char *s,
                               2252                 :                :                 bool *num_err, size_t *total_len)
                               2253                 :                : {
 2052 rhaas@postgresql.org     2254                 :         231775 :     bool        error = false;
                               2255                 :         231775 :     int         len = s - lex->input;
                               2256                 :                : 
                               2257                 :                :     /* Part (1): leading sign indicator. */
                               2258                 :                :     /* Caller already did this for us; so do nothing. */
                               2259                 :                : 
                               2260                 :                :     /* Part (2): parse main digit string. */
                               2261   [ +  -  +  + ]:         231775 :     if (len < lex->input_length && *s == '0')
                               2262                 :                :     {
                               2263                 :          64393 :         s++;
                               2264                 :          64393 :         len++;
                               2265                 :                :     }
                               2266   [ +  -  +  -  :         167382 :     else if (len < lex->input_length && *s >= '1' && *s <= '9')
                                              +  + ]
                               2267                 :                :     {
                               2268                 :                :         do
                               2269                 :                :         {
                               2270                 :         578926 :             s++;
                               2271                 :         578926 :             len++;
                               2272   [ +  +  +  +  :         578926 :         } while (len < lex->input_length && *s >= '0' && *s <= '9');
                                              +  + ]
                               2273                 :                :     }
                               2274                 :                :     else
                               2275                 :              1 :         error = true;
                               2276                 :                : 
                               2277                 :                :     /* Part (3): parse optional decimal portion. */
                               2278   [ +  +  +  + ]:         231775 :     if (len < lex->input_length && *s == '.')
                               2279                 :                :     {
                               2280                 :          22694 :         s++;
                               2281                 :          22694 :         len++;
                               2282   [ +  -  +  -  :          22694 :         if (len == lex->input_length || *s < '0' || *s > '9')
                                              +  + ]
                               2283                 :              6 :             error = true;
                               2284                 :                :         else
                               2285                 :                :         {
                               2286                 :                :             do
                               2287                 :                :             {
                               2288                 :          53797 :                 s++;
                               2289                 :          53797 :                 len++;
                               2290   [ +  +  +  +  :          53797 :             } while (len < lex->input_length && *s >= '0' && *s <= '9');
                                              +  + ]
                               2291                 :                :         }
                               2292                 :                :     }
                               2293                 :                : 
                               2294                 :                :     /* Part (4): parse optional exponent. */
                               2295   [ +  +  +  +  :         231775 :     if (len < lex->input_length && (*s == 'e' || *s == 'E'))
                                              +  + ]
                               2296                 :                :     {
                               2297                 :             47 :         s++;
                               2298                 :             47 :         len++;
                               2299   [ +  -  +  +  :             47 :         if (len < lex->input_length && (*s == '+' || *s == '-'))
                                              +  + ]
                               2300                 :                :         {
                               2301                 :              5 :             s++;
                               2302                 :              5 :             len++;
                               2303                 :                :         }
                               2304   [ +  -  +  -  :             47 :         if (len == lex->input_length || *s < '0' || *s > '9')
                                              +  + ]
                               2305                 :              6 :             error = true;
                               2306                 :                :         else
                               2307                 :                :         {
                               2308                 :                :             do
                               2309                 :                :             {
                               2310                 :            142 :                 s++;
                               2311                 :            142 :                 len++;
                               2312   [ +  +  +  -  :            142 :             } while (len < lex->input_length && *s >= '0' && *s <= '9');
                                              +  + ]
                               2313                 :                :         }
                               2314                 :                :     }
                               2315                 :                : 
                               2316                 :                :     /*
                               2317                 :                :      * Check for trailing garbage.  As in json_lex(), any alphanumeric stuff
                               2318                 :                :      * here should be considered part of the token for error-reporting
                               2319                 :                :      * purposes.
                               2320                 :                :      */
                               2321   [ +  +  +  +  :         231949 :     for (; len < lex->input_length && JSON_ALPHANUMERIC_CHAR(*s); s++, len++)
                                     +  +  +  +  -  
                                     +  +  +  +  +  
                                        -  +  -  + ]
                               2322                 :            174 :         error = true;
                               2323                 :                : 
                               2324         [ +  + ]:         231775 :     if (total_len != NULL)
                               2325                 :             23 :         *total_len = len;
                               2326                 :                : 
  545 andrew@dunslane.net      2327   [ +  +  +  + ]:         231775 :     if (lex->incremental && !lex->inc_state->is_last_chunk &&
                               2328         [ +  + ]:          66223 :         len >= lex->input_length)
                               2329                 :                :     {
  360 peter@eisentraut.org     2330                 :            393 :         jsonapi_appendBinaryStringInfo(&lex->inc_state->partial_token,
                               2331                 :            393 :                                        lex->token_start, s - lex->token_start);
  519 andrew@dunslane.net      2332         [ -  + ]:            393 :         if (num_err != NULL)
  519 andrew@dunslane.net      2333                 :UBC           0 :             *num_err = error;
                               2334                 :                : 
  545 andrew@dunslane.net      2335                 :CBC         393 :         return JSON_INCOMPLETE;
                               2336                 :                :     }
                               2337         [ +  + ]:         231382 :     else if (num_err != NULL)
                               2338                 :                :     {
                               2339                 :                :         /* let the caller handle any error */
 2052 rhaas@postgresql.org     2340                 :             23 :         *num_err = error;
                               2341                 :                :     }
                               2342                 :                :     else
                               2343                 :                :     {
                               2344                 :                :         /* return token endpoint */
                               2345                 :         231359 :         lex->prev_token_terminator = lex->token_terminator;
                               2346                 :         231359 :         lex->token_terminator = s;
                               2347                 :                :         /* handle error if any */
                               2348         [ +  + ]:         231359 :         if (error)
 2049                          2349                 :             64 :             return JSON_INVALID_TOKEN;
                               2350                 :                :     }
                               2351                 :                : 
                               2352                 :         231318 :     return JSON_SUCCESS;
                               2353                 :                : }
                               2354                 :                : 
                               2355                 :                : /*
                               2356                 :                :  * Report a parse error.
                               2357                 :                :  *
                               2358                 :                :  * lex->token_start and lex->token_terminator must identify the current token.
                               2359                 :                :  */
                               2360                 :                : static JsonParseErrorType
 2052                          2361                 :            602 : report_parse_error(JsonParseContext ctx, JsonLexContext *lex)
                               2362                 :                : {
                               2363                 :                :     /* Handle case where the input ended prematurely. */
                               2364   [ +  +  -  + ]:            602 :     if (lex->token_start == NULL || lex->token_type == JSON_TOKEN_END)
 2049                          2365                 :            133 :         return JSON_EXPECTED_MORE;
                               2366                 :                : 
                               2367                 :                :     /* Otherwise choose the error type based on the parsing context. */
                               2368   [ +  +  +  +  :            469 :     switch (ctx)
                                     +  +  +  +  -  
                                                 - ]
                               2369                 :                :     {
                               2370                 :             24 :         case JSON_PARSE_END:
                               2371                 :             24 :             return JSON_EXPECTED_END;
                               2372                 :             87 :         case JSON_PARSE_VALUE:
                               2373                 :             87 :             return JSON_EXPECTED_JSON;
                               2374                 :             82 :         case JSON_PARSE_STRING:
                               2375                 :             82 :             return JSON_EXPECTED_STRING;
                               2376                 :             28 :         case JSON_PARSE_ARRAY_START:
                               2377                 :             28 :             return JSON_EXPECTED_ARRAY_FIRST;
                               2378                 :             36 :         case JSON_PARSE_ARRAY_NEXT:
                               2379                 :             36 :             return JSON_EXPECTED_ARRAY_NEXT;
                               2380                 :             62 :         case JSON_PARSE_OBJECT_START:
                               2381                 :             62 :             return JSON_EXPECTED_OBJECT_FIRST;
                               2382                 :             64 :         case JSON_PARSE_OBJECT_LABEL:
                               2383                 :             64 :             return JSON_EXPECTED_COLON;
                               2384                 :             86 :         case JSON_PARSE_OBJECT_NEXT:
                               2385                 :             86 :             return JSON_EXPECTED_OBJECT_NEXT;
 2049 rhaas@postgresql.org     2386                 :UBC           0 :         case JSON_PARSE_OBJECT_COMMA:
                               2387                 :              0 :             return JSON_EXPECTED_STRING;
                               2388                 :                :     }
                               2389                 :                : 
                               2390                 :                :     /*
                               2391                 :                :      * We don't use a default: case, so that the compiler will warn about
                               2392                 :                :      * unhandled enum values.
                               2393                 :                :      */
 1527 michael@paquier.xyz      2394                 :              0 :     Assert(false);
                               2395                 :                :     return JSON_SUCCESS;        /* silence stupider compilers */
                               2396                 :                : }
                               2397                 :                : 
                               2398                 :                : /*
                               2399                 :                :  * Construct an (already translated) detail message for a JSON error.
                               2400                 :                :  *
                               2401                 :                :  * The returned pointer should not be freed, the allocation is either static
                               2402                 :                :  * or owned by the JsonLexContext.
                               2403                 :                :  */
                               2404                 :                : char *
 2049 rhaas@postgresql.org     2405                 :CBC        1179 : json_errdetail(JsonParseErrorType error, JsonLexContext *lex)
                               2406                 :                : {
  360 peter@eisentraut.org     2407   [ +  -  -  + ]:           1179 :     if (error == JSON_OUT_OF_MEMORY || lex == &failed_oom)
                               2408                 :                :     {
                               2409                 :                :         /* Short circuit. Allocating anything for this case is unhelpful. */
  360 peter@eisentraut.org     2410                 :UBC           0 :         return _("out of memory");
                               2411                 :                :     }
                               2412                 :                : 
  538 dgustafsson@postgres     2413         [ -  + ]:CBC        1179 :     if (lex->errormsg)
  360 peter@eisentraut.org     2414                 :UBC           0 :         jsonapi_resetStringInfo(lex->errormsg);
                               2415                 :                :     else
  360 peter@eisentraut.org     2416                 :CBC        1179 :         lex->errormsg = jsonapi_makeStringInfo();
                               2417                 :                : 
                               2418                 :                :     /*
                               2419                 :                :      * A helper for error messages that should print the current token. The
                               2420                 :                :      * format must contain exactly one %.*s specifier.
                               2421                 :                :      */
                               2422                 :                : #define json_token_error(lex, format) \
                               2423                 :                :     jsonapi_appendStringInfo((lex)->errormsg, _(format), \
                               2424                 :                :                              (int) ((lex)->token_terminator - (lex)->token_start), \
                               2425                 :                :                              (lex)->token_start);
                               2426                 :                : 
 2049 rhaas@postgresql.org     2427   [ -  -  +  +  :           1179 :     switch (error)
                                     +  +  +  +  +  
                                     +  +  +  +  +  
                                     +  -  +  +  -  
                                        -  +  +  -  
                                                 - ]
                               2428                 :                :     {
  545 andrew@dunslane.net      2429                 :UBC           0 :         case JSON_INCOMPLETE:
                               2430                 :                :         case JSON_SUCCESS:
                               2431                 :                :             /* fall through to the error code after switch */
 2049 rhaas@postgresql.org     2432                 :              0 :             break;
  545 andrew@dunslane.net      2433                 :              0 :         case JSON_INVALID_LEXER_TYPE:
                               2434         [ #  # ]:              0 :             if (lex->incremental)
  471 peter@eisentraut.org     2435                 :              0 :                 return _("Recursive descent parser cannot use incremental lexer.");
                               2436                 :                :             else
                               2437                 :              0 :                 return _("Incremental parser requires incremental lexer.");
  545 andrew@dunslane.net      2438                 :CBC         256 :         case JSON_NESTING_TOO_DEEP:
  471 peter@eisentraut.org     2439                 :            256 :             return (_("JSON nested too deep, maximum permitted depth is 6400."));
 2049 rhaas@postgresql.org     2440                 :             66 :         case JSON_ESCAPING_INVALID:
  480 peter@eisentraut.org     2441                 :             66 :             json_token_error(lex, "Escape sequence \"\\%.*s\" is invalid.");
  538 dgustafsson@postgres     2442                 :             66 :             break;
 2049 rhaas@postgresql.org     2443                 :             58 :         case JSON_ESCAPING_REQUIRED:
  360 peter@eisentraut.org     2444                 :             58 :             jsonapi_appendStringInfo(lex->errormsg,
                               2445                 :             58 :                                      _("Character with value 0x%02x must be escaped."),
                               2446                 :             58 :                                      (unsigned char) *(lex->token_terminator));
  538 dgustafsson@postgres     2447                 :             58 :             break;
 2049 rhaas@postgresql.org     2448                 :             24 :         case JSON_EXPECTED_END:
  480 peter@eisentraut.org     2449                 :             24 :             json_token_error(lex, "Expected end of input, but found \"%.*s\".");
  538 dgustafsson@postgres     2450                 :             24 :             break;
 2049 rhaas@postgresql.org     2451                 :             28 :         case JSON_EXPECTED_ARRAY_FIRST:
  480 peter@eisentraut.org     2452                 :             28 :             json_token_error(lex, "Expected array element or \"]\", but found \"%.*s\".");
  538 dgustafsson@postgres     2453                 :             28 :             break;
 2049 rhaas@postgresql.org     2454                 :             36 :         case JSON_EXPECTED_ARRAY_NEXT:
  480 peter@eisentraut.org     2455                 :             36 :             json_token_error(lex, "Expected \",\" or \"]\", but found \"%.*s\".");
  538 dgustafsson@postgres     2456                 :             36 :             break;
 2049 rhaas@postgresql.org     2457                 :             64 :         case JSON_EXPECTED_COLON:
  480 peter@eisentraut.org     2458                 :             64 :             json_token_error(lex, "Expected \":\", but found \"%.*s\".");
  538 dgustafsson@postgres     2459                 :             64 :             break;
 2049 rhaas@postgresql.org     2460                 :             60 :         case JSON_EXPECTED_JSON:
  480 peter@eisentraut.org     2461                 :             60 :             json_token_error(lex, "Expected JSON value, but found \"%.*s\".");
  538 dgustafsson@postgres     2462                 :             60 :             break;
 2049 rhaas@postgresql.org     2463                 :             95 :         case JSON_EXPECTED_MORE:
                               2464                 :             95 :             return _("The input string ended unexpectedly.");
                               2465                 :             62 :         case JSON_EXPECTED_OBJECT_FIRST:
  480 peter@eisentraut.org     2466                 :             62 :             json_token_error(lex, "Expected string or \"}\", but found \"%.*s\".");
  538 dgustafsson@postgres     2467                 :             62 :             break;
 2049 rhaas@postgresql.org     2468                 :             86 :         case JSON_EXPECTED_OBJECT_NEXT:
  480 peter@eisentraut.org     2469                 :             86 :             json_token_error(lex, "Expected \",\" or \"}\", but found \"%.*s\".");
  538 dgustafsson@postgres     2470                 :             86 :             break;
 2049 rhaas@postgresql.org     2471                 :             82 :         case JSON_EXPECTED_STRING:
  480 peter@eisentraut.org     2472                 :             82 :             json_token_error(lex, "Expected string, but found \"%.*s\".");
  538 dgustafsson@postgres     2473                 :             82 :             break;
 2049 rhaas@postgresql.org     2474                 :            207 :         case JSON_INVALID_TOKEN:
  480 peter@eisentraut.org     2475                 :            207 :             json_token_error(lex, "Token \"%.*s\" is invalid.");
  538 dgustafsson@postgres     2476                 :            207 :             break;
  360 peter@eisentraut.org     2477                 :UBC           0 :         case JSON_OUT_OF_MEMORY:
                               2478                 :                :             /* should have been handled above; use the error path */
                               2479                 :              0 :             break;
 2049 rhaas@postgresql.org     2480                 :CBC          12 :         case JSON_UNICODE_CODE_POINT_ZERO:
                               2481                 :             12 :             return _("\\u0000 cannot be converted to text.");
                               2482                 :             19 :         case JSON_UNICODE_ESCAPE_FORMAT:
                               2483                 :             19 :             return _("\"\\u\" must be followed by four hexadecimal digits.");
 2049 rhaas@postgresql.org     2484                 :UBC           0 :         case JSON_UNICODE_HIGH_ESCAPE:
                               2485                 :                :             /* note: this case is only reachable in frontend not backend */
 2010 tgl@sss.pgh.pa.us        2486                 :              0 :             return _("Unicode escape values cannot be used for code point values above 007F when the encoding is not UTF8.");
 1000                          2487                 :              0 :         case JSON_UNICODE_UNTRANSLATABLE:
                               2488                 :                : 
                               2489                 :                :             /*
                               2490                 :                :              * Note: this case is only reachable in backend and not frontend.
                               2491                 :                :              * #ifdef it away so the frontend doesn't try to link against
                               2492                 :                :              * backend functionality.
                               2493                 :                :              */
                               2494                 :                : #ifndef FRONTEND
                               2495                 :              0 :             return psprintf(_("Unicode escape value could not be translated to the server's encoding %s."),
                               2496                 :                :                             GetDatabaseEncodingName());
                               2497                 :                : #else
  538 dgustafsson@postgres     2498                 :              0 :             Assert(false);
                               2499                 :                :             break;
                               2500                 :                : #endif
 2049 rhaas@postgresql.org     2501                 :CBC           6 :         case JSON_UNICODE_HIGH_SURROGATE:
                               2502                 :              6 :             return _("Unicode high surrogate must not follow a high surrogate.");
                               2503                 :             18 :         case JSON_UNICODE_LOW_SURROGATE:
                               2504                 :             18 :             return _("Unicode low surrogate must follow a high surrogate.");
 1000 tgl@sss.pgh.pa.us        2505                 :UBC           0 :         case JSON_SEM_ACTION_FAILED:
                               2506                 :                :             /* fall through to the error code after switch */
                               2507                 :              0 :             break;
                               2508                 :                :     }
                               2509                 :                : #undef json_token_error
                               2510                 :                : 
                               2511                 :                :     /* Note that lex->errormsg can be NULL in shlib code. */
  360 peter@eisentraut.org     2512   [ +  -  -  + ]:CBC         773 :     if (lex->errormsg && lex->errormsg->len == 0)
                               2513                 :                :     {
                               2514                 :                :         /*
                               2515                 :                :          * We don't use a default: case, so that the compiler will warn about
                               2516                 :                :          * unhandled enum values.  But this needs to be here anyway to cover
                               2517                 :                :          * the possibility of an incorrect input.
                               2518                 :                :          */
  360 peter@eisentraut.org     2519                 :UBC           0 :         jsonapi_appendStringInfo(lex->errormsg,
                               2520                 :                :                                  "unexpected json parse error type: %d",
                               2521                 :                :                                  (int) error);
                               2522                 :                :     }
                               2523                 :                : 
                               2524                 :                : #ifdef JSONAPI_USE_PQEXPBUFFER
  360 peter@eisentraut.org     2525   [ +  -  -  + ]:CBC         316 :     if (PQExpBufferBroken(lex->errormsg))
  360 peter@eisentraut.org     2526                 :UBC           0 :         return _("out of memory while constructing error description");
                               2527                 :                : #endif
                               2528                 :                : 
  538 dgustafsson@postgres     2529                 :CBC         773 :     return lex->errormsg->data;
                               2530                 :                : }
        

Generated by: LCOV version 2.4-beta