static int
yaml_parser_append_tag_directive(yaml_parser_t *parser,
yaml_tag_directive_t value, int allow_duplicates, yaml_mark_t mark)
{
yaml_tag_directive_t *tag_directive;
yaml_tag_directive_t copy = { NULL, NULL };
for (tag_directive = parser->tag_directives.start;
tag_directive != parser->tag_directives.top; tag_directive ++) {
if (strcmp((char *)value.handle, (char *)tag_directive->handle) == 0) {
if (allow_duplicates)
return 1;
return yaml_parser_set_parser_error(parser,
"found duplicate %TAG directive", mark);
}
}
copy.handle = yaml_strdup(value.handle);
copy.prefix = yaml_strdup(value.prefix);
if (!copy.handle || !copy.prefix) {
parser->error = YAML_MEMORY_ERROR;
goto error;
}
if (!PUSH(parser, parser->tag_directives, copy))
goto error;
return 1;
error:
yaml_free(copy.handle);
yaml_free(copy.prefix);
return 0;
}
static int yaml_parser_parse_stream_start (yaml_parser_t * parser, yaml_event_t * event)
{
yaml_token_t *token;
token = PEEK_TOKEN (parser);
if (!token)
return 0;
if (token->type != YAML_STREAM_START_TOKEN)
{
return yaml_parser_set_parser_error (parser, "did not find expected <stream-start>", token->start_mark);
}
parser->state = YAML_PARSE_IMPLICIT_DOCUMENT_START_STATE;
STREAM_START_EVENT_INIT (*event, token->data.stream_start.encoding, token->start_mark, token->start_mark);
SKIP_TOKEN (parser);
return 1;
}
static int
yaml_parser_parse_document_start(yaml_parser_t *parser, yaml_event_t *event,
int implicit)
{
yaml_token_t *token;
yaml_version_directive_t *version_directive = NULL;
struct {
yaml_tag_directive_t *start;
yaml_tag_directive_t *end;
} tag_directives = { NULL, NULL };
token = PEEK_TOKEN(parser);
if (!token) return 0;
/* Parse extra document end indicators. */
if (!implicit)
{
while (token->type == YAML_DOCUMENT_END_TOKEN) {
SKIP_TOKEN(parser);
token = PEEK_TOKEN(parser);
if (!token) return 0;
}
}
/* Parse an implicit document. */
if (implicit && token->type != YAML_VERSION_DIRECTIVE_TOKEN &&
token->type != YAML_TAG_DIRECTIVE_TOKEN &&
token->type != YAML_DOCUMENT_START_TOKEN &&
token->type != YAML_STREAM_END_TOKEN)
{
if (!yaml_parser_process_directives(parser, NULL, NULL, NULL))
return 0;
if (!PUSH(parser, parser->states, YAML_PARSE_DOCUMENT_END_STATE))
return 0;
parser->state = YAML_PARSE_BLOCK_NODE_STATE;
DOCUMENT_START_EVENT_INIT(*event, NULL, NULL, NULL, 1,
token->start_mark, token->start_mark);
return 1;
}
/* Parse an explicit document. */
else if (token->type != YAML_STREAM_END_TOKEN)
{
yaml_mark_t start_mark, end_mark;
start_mark = token->start_mark;
if (!yaml_parser_process_directives(parser, &version_directive,
&tag_directives.start, &tag_directives.end))
return 0;
token = PEEK_TOKEN(parser);
if (!token) goto error;
if (token->type != YAML_DOCUMENT_START_TOKEN) {
yaml_parser_set_parser_error(parser,
"did not find expected <document start>", token->start_mark);
goto error;
}
if (!PUSH(parser, parser->states, YAML_PARSE_DOCUMENT_END_STATE))
goto error;
parser->state = YAML_PARSE_DOCUMENT_CONTENT_STATE;
end_mark = token->end_mark;
DOCUMENT_START_EVENT_INIT(*event, version_directive,
tag_directives.start, tag_directives.end, 0,
start_mark, end_mark);
SKIP_TOKEN(parser);
version_directive = NULL;
tag_directives.start = tag_directives.end = NULL;
return 1;
}
/* Parse the stream end. */
else
{
parser->state = YAML_PARSE_END_STATE;
STREAM_END_EVENT_INIT(*event, token->start_mark, token->end_mark);
SKIP_TOKEN(parser);
return 1;
}
error:
yaml_free(version_directive);
while (tag_directives.start != tag_directives.end) {
yaml_free(tag_directives.end[-1].handle);
yaml_free(tag_directives.end[-1].prefix);
tag_directives.end --;
}
yaml_free(tag_directives.start);
return 0;
}
static int
yaml_parser_process_directives(yaml_parser_t *parser,
yaml_version_directive_t **version_directive_ref,
yaml_tag_directive_t **tag_directives_start_ref,
yaml_tag_directive_t **tag_directives_end_ref)
{
yaml_tag_directive_t default_tag_directives[] = {
{(yaml_char_t *)"!", (yaml_char_t *)"!"},
{(yaml_char_t *)"!!", (yaml_char_t *)"tag:yaml.org,2002:"},
{NULL, NULL}
};
yaml_tag_directive_t *default_tag_directive;
yaml_version_directive_t *version_directive = NULL;
struct {
yaml_tag_directive_t *start;
yaml_tag_directive_t *end;
yaml_tag_directive_t *top;
} tag_directives = { NULL, NULL, NULL };
yaml_token_t *token;
if (!STACK_INIT(parser, tag_directives, INITIAL_STACK_SIZE))
goto error;
token = PEEK_TOKEN(parser);
if (!token) goto error;
while (token->type == YAML_VERSION_DIRECTIVE_TOKEN ||
token->type == YAML_TAG_DIRECTIVE_TOKEN)
{
if (token->type == YAML_VERSION_DIRECTIVE_TOKEN) {
if (version_directive) {
yaml_parser_set_parser_error(parser,
"found duplicate %YAML directive", token->start_mark);
goto error;
}
if (token->data.version_directive.major != 1
|| token->data.version_directive.minor != 1) {
yaml_parser_set_parser_error(parser,
"found incompatible YAML document", token->start_mark);
goto error;
}
version_directive = yaml_malloc(sizeof(yaml_version_directive_t));
if (!version_directive) {
parser->error = YAML_MEMORY_ERROR;
goto error;
}
version_directive->major = token->data.version_directive.major;
version_directive->minor = token->data.version_directive.minor;
}
else if (token->type == YAML_TAG_DIRECTIVE_TOKEN) {
yaml_tag_directive_t value;
value.handle = token->data.tag_directive.handle;
value.prefix = token->data.tag_directive.prefix;
if (!yaml_parser_append_tag_directive(parser, value, 0,
token->start_mark))
goto error;
if (!PUSH(parser, tag_directives, value))
goto error;
}
SKIP_TOKEN(parser);
token = PEEK_TOKEN(parser);
if (!token) goto error;
}
for (default_tag_directive = default_tag_directives;
default_tag_directive->handle; default_tag_directive++) {
if (!yaml_parser_append_tag_directive(parser, *default_tag_directive, 1,
token->start_mark))
goto error;
}
if (version_directive_ref) {
*version_directive_ref = version_directive;
}
if (tag_directives_start_ref) {
if (STACK_EMPTY(parser, tag_directives)) {
*tag_directives_start_ref = *tag_directives_end_ref = NULL;
STACK_DEL(parser, tag_directives);
}
else {
*tag_directives_start_ref = tag_directives.start;
*tag_directives_end_ref = tag_directives.top;
}
}
else {
STACK_DEL(parser, tag_directives);
}
return 1;
error:
yaml_free(version_directive);
while (!STACK_EMPTY(parser, tag_directives)) {
yaml_tag_directive_t tag_directive = POP(parser, tag_directives);
yaml_free(tag_directive.handle);
yaml_free(tag_directive.prefix);
}
STACK_DEL(parser, tag_directives);
return 0;
}
