diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 02:57:58 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 02:57:58 +0000 |
commit | be1c7e50e1e8809ea56f2c9d472eccd8ffd73a97 (patch) | |
tree | 9754ff1ca740f6346cf8483ec915d4054bc5da2d /web/server/h2o/libh2o/deps/yoml/yoml-parser.h | |
parent | Initial commit. (diff) | |
download | netdata-be1c7e50e1e8809ea56f2c9d472eccd8ffd73a97.tar.xz netdata-be1c7e50e1e8809ea56f2c9d472eccd8ffd73a97.zip |
Adding upstream version 1.44.3.upstream/1.44.3upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'web/server/h2o/libh2o/deps/yoml/yoml-parser.h')
-rw-r--r-- | web/server/h2o/libh2o/deps/yoml/yoml-parser.h | 391 |
1 files changed, 391 insertions, 0 deletions
diff --git a/web/server/h2o/libh2o/deps/yoml/yoml-parser.h b/web/server/h2o/libh2o/deps/yoml/yoml-parser.h new file mode 100644 index 00000000..78cdd446 --- /dev/null +++ b/web/server/h2o/libh2o/deps/yoml/yoml-parser.h @@ -0,0 +1,391 @@ +/* + * Copyright (c) 2014 DeNA Co., Ltd. + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to + * deal in the Software without restriction, including without limitation the + * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or + * sell copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in + * all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING + * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS + * IN THE SOFTWARE. + */ +#ifndef yoml_parser_h +#define yoml_parser_h + +#ifdef __cplusplus +extern "C" { +#endif + +#include <assert.h> +#include <stddef.h> +#include <stdlib.h> +#include <string.h> +#include <yaml.h> +#include "yoml.h" + +typedef struct st_yoml_parse_args_t { + const char *filename; + void *(*mem_set)(void *, int, size_t); + struct { + yoml_t *(*cb)(const char *tag, yoml_t *node, void *cb_arg); + void *cb_arg; + } resolve_tag; +} yoml_parse_args_t; + +static yoml_t *yoml__parse_node(yaml_parser_t *parser, yaml_event_type_t *last_event, yoml_parse_args_t *parse_args); + +static inline char *yoml__strdup(yaml_char_t *s) +{ + return strdup((char *)s); +} + +static inline yoml_t *yoml__new_node(const char *filename, yoml_type_t type, size_t sz, yaml_char_t *anchor, yaml_char_t *tag, yaml_event_t *event) +{ + yoml_t *node = malloc(sz); + node->filename = filename != NULL ? strdup(filename) : NULL; + node->type = type; + node->line = event->start_mark.line; + if (type == YOML_TYPE_SCALAR) { + switch (event->data.scalar.style) { + case YAML_LITERAL_SCALAR_STYLE: + case YAML_FOLDED_SCALAR_STYLE: + ++node->line; + break; + default: + break; + } + } + node->column = event->start_mark.column; + node->anchor = anchor != NULL ? yoml__strdup(anchor) : NULL; + node->tag = tag != NULL ? yoml__strdup(tag) : NULL; + node->_refcnt = 1; + return node; +} + +static inline yoml_t *yoml__parse_sequence(yaml_parser_t *parser, yaml_event_t *event, yoml_parse_args_t *parse_args) +{ + yoml_t *seq = yoml__new_node(parse_args->filename, YOML_TYPE_SEQUENCE, offsetof(yoml_t, data.sequence.elements), + event->data.sequence_start.anchor, event->data.sequence_start.tag, event); + + seq->data.sequence.size = 0; + + while (1) { + yoml_t *new_node; + yaml_event_type_t unhandled; + if ((new_node = yoml__parse_node(parser, &unhandled, parse_args)) == NULL) { + if (unhandled == YAML_SEQUENCE_END_EVENT) { + break; + } else { + yoml_free(seq, parse_args->mem_set); + seq = NULL; + break; + } + } + seq = realloc(seq, offsetof(yoml_t, data.sequence.elements) + sizeof(yoml_t *) * (seq->data.sequence.size + 1)); + seq->data.sequence.elements[seq->data.sequence.size++] = new_node; + } + + return seq; +} + +static inline yoml_t *yoml__parse_mapping(yaml_parser_t *parser, yaml_event_t *event, yoml_parse_args_t *parse_args) +{ + yoml_t *map = yoml__new_node(parse_args->filename, YOML_TYPE_MAPPING, offsetof(yoml_t, data.mapping.elements), + event->data.mapping_start.anchor, event->data.mapping_start.tag, event); + + map->data.mapping.size = 0; + + while (1) { + yoml_t *key, *value; + yaml_event_type_t unhandled; + if ((key = yoml__parse_node(parser, &unhandled, parse_args)) == NULL) { + if (unhandled == YAML_MAPPING_END_EVENT) { + break; + } else { + yoml_free(map, parse_args->mem_set); + map = NULL; + break; + } + } + if ((value = yoml__parse_node(parser, NULL, parse_args)) == NULL) { + yoml_free(map, parse_args->mem_set); + map = NULL; + break; + } + map = realloc(map, offsetof(yoml_t, data.mapping.elements) + sizeof(yoml_mapping_element_t) * (map->data.mapping.size + 1)); + map->data.mapping.elements[map->data.mapping.size].key = key; + map->data.mapping.elements[map->data.mapping.size].value = value; + ++map->data.mapping.size; + } + + return map; +} + +static yoml_t *yoml__parse_node(yaml_parser_t *parser, yaml_event_type_t *unhandled, yoml_parse_args_t *parse_args) +{ + yoml_t *node; + yaml_event_t event; + + if (unhandled != NULL) + *unhandled = YAML_NO_EVENT; + + /* wait for a node that is not a stream/doc start event */ + while (1) { + if (!yaml_parser_parse(parser, &event)) + return NULL; + if (!(event.type == YAML_STREAM_START_EVENT || event.type == YAML_DOCUMENT_START_EVENT)) + break; + yaml_event_delete(&event); + } + + switch (event.type) { + case YAML_ALIAS_EVENT: + node = yoml__new_node(parse_args->filename, YOML__TYPE_UNRESOLVED_ALIAS, sizeof(*node), NULL, NULL, &event); + node->data.alias = yoml__strdup(event.data.alias.anchor); + break; + case YAML_SCALAR_EVENT: + node = yoml__new_node(parse_args->filename, YOML_TYPE_SCALAR, sizeof(*node), event.data.scalar.anchor, event.data.scalar.tag, &event); + node->data.scalar = yoml__strdup(event.data.scalar.value); + if (parse_args->mem_set != NULL) + parse_args->mem_set(event.data.scalar.value, 'A', strlen(node->data.scalar)); + break; + case YAML_SEQUENCE_START_EVENT: + node = yoml__parse_sequence(parser, &event, parse_args); + break; + case YAML_MAPPING_START_EVENT: + node = yoml__parse_mapping(parser, &event, parse_args); + break; + default: + node = NULL; + if (unhandled != NULL) + *unhandled = event.type; + break; + } + + yaml_event_delete(&event); + + return node; +} + +static inline int yoml__merge(yoml_t **dest, size_t offset, yoml_t *src) +{ + yoml_t *key, *value; + size_t i, j; + + if (src->type != YOML_TYPE_MAPPING) + return -1; + + for (i = 0; i != src->data.mapping.size; ++i) { + key = src->data.mapping.elements[i].key; + value = src->data.mapping.elements[i].value; + if (key->type == YOML_TYPE_SCALAR) { + for (j = offset; j != (*dest)->data.mapping.size; ++j) { + if ((*dest)->data.mapping.elements[j].key->type == YOML_TYPE_SCALAR && + strcmp((*dest)->data.mapping.elements[j].key->data.scalar, key->data.scalar) == 0) + goto Skip; + } + } + *dest = realloc(*dest, offsetof(yoml_t, data.mapping.elements) + + ((*dest)->data.mapping.size + 1) * sizeof((*dest)->data.mapping.elements[0])); + memmove((*dest)->data.mapping.elements + offset + 1, (*dest)->data.mapping.elements + offset, + ((*dest)->data.mapping.size - offset) * sizeof((*dest)->data.mapping.elements[0])); + (*dest)->data.mapping.elements[offset].key = key; + ++key->_refcnt; + (*dest)->data.mapping.elements[offset].value = value; + ++value->_refcnt; + ++(*dest)->data.mapping.size; + ++offset; + Skip: + ; + } + + return 0; +} + +static inline int yoml__resolve_merge(yoml_t **target, yaml_parser_t *parser, yoml_parse_args_t *parse_args) +{ + size_t i, j; + + switch ((*target)->type) { + case YOML_TYPE_SCALAR: + break; + case YOML_TYPE_SEQUENCE: + for (i = 0; i != (*target)->data.sequence.size; ++i) { + if (yoml__resolve_merge((*target)->data.sequence.elements + i, parser, parse_args) != 0) + return -1; + } + break; + case YOML_TYPE_MAPPING: + if ((*target)->data.mapping.size != 0) { + i = (*target)->data.mapping.size; + do { + --i; + if (yoml__resolve_merge(&(*target)->data.mapping.elements[i].key, parser, parse_args) != 0) + return -1; + if (yoml__resolve_merge(&(*target)->data.mapping.elements[i].value, parser, parse_args) != 0) + return -1; + if ((*target)->data.mapping.elements[i].key->type == YOML_TYPE_SCALAR && + strcmp((*target)->data.mapping.elements[i].key->data.scalar, "<<") == 0) { + /* erase the slot (as well as preserving the values) */ + yoml_mapping_element_t src = (*target)->data.mapping.elements[i]; + memmove((*target)->data.mapping.elements + i, (*target)->data.mapping.elements + i + 1, + ((*target)->data.mapping.size - i - 1) * sizeof((*target)->data.mapping.elements[0])); + --(*target)->data.mapping.size; + /* merge */ + if (src.value->type == YOML_TYPE_SEQUENCE) { + for (j = 0; j != src.value->data.sequence.size; ++j) + if (yoml__merge(target, i, src.value->data.sequence.elements[j]) != 0) { + MergeError: + if (parser != NULL) { + parser->problem = "value of the merge key MUST be a mapping or a sequence of mappings"; + parser->problem_mark.line = src.key->line; + parser->problem_mark.column = src.key->column; + } + return -1; + } + } else { + if (yoml__merge(target, i, src.value) != 0) + goto MergeError; + } + /* cleanup */ + yoml_free(src.key, parse_args->mem_set); + yoml_free(src.value, parse_args->mem_set); + } + } while (i != 0); + } + break; + case YOML__TYPE_UNRESOLVED_ALIAS: + assert(!"unreachable"); + break; + } + + return 0; +} + + +static inline int yoml__resolve_alias(yoml_t **target, yoml_t *doc, yaml_parser_t *parser, yoml_parse_args_t *parse_args) +{ + size_t i; + + switch ((*target)->type) { + case YOML_TYPE_SCALAR: + break; + case YOML_TYPE_SEQUENCE: + for (i = 0; i != (*target)->data.sequence.size; ++i) { + if (yoml__resolve_alias((*target)->data.sequence.elements + i, doc, parser, parse_args) != 0) + return -1; + } + break; + case YOML_TYPE_MAPPING: + for (i = 0; i != (*target)->data.mapping.size; ++i) { + if (yoml__resolve_alias(&(*target)->data.mapping.elements[i].key, doc, parser, parse_args) != 0) + return -1; + if (yoml__resolve_alias(&(*target)->data.mapping.elements[i].value, doc, parser, parse_args) != 0) + return -1; + } + break; + case YOML__TYPE_UNRESOLVED_ALIAS: { + yoml_t *node = yoml_find_anchor(doc, (*target)->data.alias); + if (node == NULL) { + if (parser != NULL) { + parser->problem = "could not resolve the alias"; + parser->problem_mark.line = (*target)->line; + parser->problem_mark.column = (*target)->column; + } + return -1; + } + yoml_free(*target, parse_args->mem_set); + *target = node; + ++node->_refcnt; + } break; + } + + return 0; +} + +static inline int yoml__resolve_tag(yoml_t **target, yaml_parser_t *parser, yoml_parse_args_t *parse_args) +{ + size_t i; + + if (parse_args->resolve_tag.cb == NULL) + return 0; + + if ((*target)->tag != NULL) { + yoml_t *resolved = parse_args->resolve_tag.cb((*target)->tag, *target, parse_args->resolve_tag.cb_arg); + if (resolved == NULL) { + if (parser != NULL) { + parser->problem = "tag resolution failed"; + parser->problem_mark.line = (*target)->line; + parser->problem_mark.column = (*target)->column; + } + return -1; + } + yoml_free(*target, parse_args->mem_set); + *target = resolved; + } + + switch ((*target)->type) { + case YOML_TYPE_SCALAR: + break; + case YOML_TYPE_SEQUENCE: + for (i = 0; i != (*target)->data.sequence.size; ++i) { + if (yoml__resolve_tag((*target)->data.sequence.elements + i, parser, parse_args) != 0) + return -1; + } + break; + case YOML_TYPE_MAPPING: + for (i = 0; i != (*target)->data.mapping.size; ++i) { + if (yoml__resolve_tag(&(*target)->data.mapping.elements[i].key, parser, parse_args) != 0) + return -1; + if (yoml__resolve_tag(&(*target)->data.mapping.elements[i].value, parser, parse_args) != 0) + return -1; + } + break; + case YOML__TYPE_UNRESOLVED_ALIAS: + break; + } + + return 0; +} + +static inline yoml_t *yoml_parse_document(yaml_parser_t *parser, yaml_event_type_t *unhandled, yoml_parse_args_t *parse_args) +{ + yoml_t *doc; + + /* parse */ + if ((doc = yoml__parse_node(parser, unhandled, parse_args)) == NULL) { + return NULL; + } + if (unhandled != NULL) + *unhandled = YAML_NO_EVENT; + + /* resolve tags, aliases and merge */ + if (yoml__resolve_tag(&doc, parser, parse_args) != 0) + goto Error; + if (yoml__resolve_alias(&doc, doc, parser, parse_args) != 0) + goto Error; + if (yoml__resolve_merge(&doc, parser, parse_args) != 0) + goto Error; + + return doc; + +Error: + yoml_free(doc, parse_args->mem_set); + return NULL; +} + +#ifdef __cplusplus +} +#endif + +#endif |