123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964 |
- // SPDX-License-Identifier: GPL-3.0-or-later
- #include "log2journal.h"
- // ----------------------------------------------------------------------------
- // yaml configuration file
- #ifdef HAVE_LIBYAML
- static const char *yaml_event_name(yaml_event_type_t type) {
- switch (type) {
- case YAML_NO_EVENT:
- return "YAML_NO_EVENT";
- case YAML_SCALAR_EVENT:
- return "YAML_SCALAR_EVENT";
- case YAML_ALIAS_EVENT:
- return "YAML_ALIAS_EVENT";
- case YAML_MAPPING_START_EVENT:
- return "YAML_MAPPING_START_EVENT";
- case YAML_MAPPING_END_EVENT:
- return "YAML_MAPPING_END_EVENT";
- case YAML_SEQUENCE_START_EVENT:
- return "YAML_SEQUENCE_START_EVENT";
- case YAML_SEQUENCE_END_EVENT:
- return "YAML_SEQUENCE_END_EVENT";
- case YAML_STREAM_START_EVENT:
- return "YAML_STREAM_START_EVENT";
- case YAML_STREAM_END_EVENT:
- return "YAML_STREAM_END_EVENT";
- case YAML_DOCUMENT_START_EVENT:
- return "YAML_DOCUMENT_START_EVENT";
- case YAML_DOCUMENT_END_EVENT:
- return "YAML_DOCUMENT_END_EVENT";
- default:
- return "UNKNOWN";
- }
- }
- #define yaml_error(parser, event, fmt, args...) yaml_error_with_trace(parser, event, __LINE__, __FUNCTION__, __FILE__, fmt, ##args)
- static void yaml_error_with_trace(yaml_parser_t *parser, yaml_event_t *event, size_t line, const char *function, const char *file, const char *format, ...) __attribute__ ((format(__printf__, 6, 7)));
- static void yaml_error_with_trace(yaml_parser_t *parser, yaml_event_t *event, size_t line, const char *function, const char *file, const char *format, ...) {
- char buf[1024] = ""; // Initialize buf to an empty string
- const char *type = "";
- if(event) {
- type = yaml_event_name(event->type);
- switch (event->type) {
- case YAML_SCALAR_EVENT:
- copy_to_buffer(buf, sizeof(buf), (char *)event->data.scalar.value, event->data.scalar.length);
- break;
- case YAML_ALIAS_EVENT:
- snprintf(buf, sizeof(buf), "%s", event->data.alias.anchor);
- break;
- default:
- break;
- }
- }
- fprintf(stderr, "YAML %zu@%s, %s(): (line %d, column %d, %s%s%s): ",
- line, file, function,
- (int)(parser->mark.line + 1), (int)(parser->mark.column + 1),
- type, buf[0]? ", near ": "", buf);
- va_list args;
- va_start(args, format);
- vfprintf(stderr, format, args);
- va_end(args);
- fprintf(stderr, "\n");
- }
- #define yaml_parse(parser, event) yaml_parse_with_trace(parser, event, __LINE__, __FUNCTION__, __FILE__)
- static bool yaml_parse_with_trace(yaml_parser_t *parser, yaml_event_t *event, size_t line __maybe_unused, const char *function __maybe_unused, const char *file __maybe_unused) {
- if (!yaml_parser_parse(parser, event)) {
- yaml_error(parser, NULL, "YAML parser error %d", parser->error);
- return false;
- }
- // fprintf(stderr, ">>> %s >>> %.*s\n",
- // yaml_event_name(event->type),
- // event->type == YAML_SCALAR_EVENT ? event->data.scalar.length : 0,
- // event->type == YAML_SCALAR_EVENT ? (char *)event->data.scalar.value : "");
- return true;
- }
- #define yaml_parse_expect_event(parser, type) yaml_parse_expect_event_with_trace(parser, type, __LINE__, __FUNCTION__, __FILE__)
- static bool yaml_parse_expect_event_with_trace(yaml_parser_t *parser, yaml_event_type_t type, size_t line, const char *function, const char *file) {
- yaml_event_t event;
- if (!yaml_parse(parser, &event))
- return false;
- bool ret = true;
- if(event.type != type) {
- yaml_error_with_trace(parser, &event, line, function, file, "unexpected event - expecting: %s", yaml_event_name(type));
- ret = false;
- }
- // else
- // fprintf(stderr, "OK (%zu@%s, %s()\n", line, file, function);
- yaml_event_delete(&event);
- return ret;
- }
- #define yaml_scalar_matches(event, s, len) yaml_scalar_matches_with_trace(event, s, len, __LINE__, __FUNCTION__, __FILE__)
- static bool yaml_scalar_matches_with_trace(yaml_event_t *event, const char *s, size_t len, size_t line __maybe_unused, const char *function __maybe_unused, const char *file __maybe_unused) {
- if(event->type != YAML_SCALAR_EVENT)
- return false;
- if(len != event->data.scalar.length)
- return false;
- // else
- // fprintf(stderr, "OK (%zu@%s, %s()\n", line, file, function);
- return strcmp((char *)event->data.scalar.value, s) == 0;
- }
- // ----------------------------------------------------------------------------
- static size_t yaml_parse_filename_injection(yaml_parser_t *parser, LOG_JOB *jb) {
- yaml_event_t event;
- size_t errors = 0;
- if(!yaml_parse_expect_event(parser, YAML_MAPPING_START_EVENT))
- return 1;
- if (!yaml_parse(parser, &event))
- return 1;
- if (yaml_scalar_matches(&event, "key", strlen("key"))) {
- yaml_event_t sub_event;
- if (!yaml_parse(parser, &sub_event))
- errors++;
- else {
- if (sub_event.type == YAML_SCALAR_EVENT) {
- if(!log_job_filename_key_set(jb, (char *) sub_event.data.scalar.value,
- sub_event.data.scalar.length))
- errors++;
- }
- else {
- yaml_error(parser, &sub_event, "expected the filename as %s", yaml_event_name(YAML_SCALAR_EVENT));
- errors++;
- }
- yaml_event_delete(&sub_event);
- }
- }
- if(!yaml_parse_expect_event(parser, YAML_MAPPING_END_EVENT))
- errors++;
- yaml_event_delete(&event);
- return errors;
- }
- static size_t yaml_parse_filters(yaml_parser_t *parser, LOG_JOB *jb) {
- if(!yaml_parse_expect_event(parser, YAML_MAPPING_START_EVENT))
- return 1;
- size_t errors = 0;
- bool finished = false;
- while(!errors && !finished) {
- yaml_event_t event;
- if(!yaml_parse(parser, &event))
- return 1;
- if(event.type == YAML_SCALAR_EVENT) {
- if(yaml_scalar_matches(&event, "include", strlen("include"))) {
- yaml_event_t sub_event;
- if(!yaml_parse(parser, &sub_event))
- errors++;
- else {
- if(sub_event.type == YAML_SCALAR_EVENT) {
- if(!log_job_include_pattern_set(jb, (char *) sub_event.data.scalar.value,
- sub_event.data.scalar.length))
- errors++;
- }
- else {
- yaml_error(parser, &sub_event, "expected the include as %s",
- yaml_event_name(YAML_SCALAR_EVENT));
- errors++;
- }
- yaml_event_delete(&sub_event);
- }
- }
- else if(yaml_scalar_matches(&event, "exclude", strlen("exclude"))) {
- yaml_event_t sub_event;
- if(!yaml_parse(parser, &sub_event))
- errors++;
- else {
- if(sub_event.type == YAML_SCALAR_EVENT) {
- if(!log_job_exclude_pattern_set(jb,(char *) sub_event.data.scalar.value,
- sub_event.data.scalar.length))
- errors++;
- }
- else {
- yaml_error(parser, &sub_event, "expected the exclude as %s",
- yaml_event_name(YAML_SCALAR_EVENT));
- errors++;
- }
- yaml_event_delete(&sub_event);
- }
- }
- }
- else if(event.type == YAML_MAPPING_END_EVENT)
- finished = true;
- else {
- yaml_error(parser, &event, "expected %s or %s",
- yaml_event_name(YAML_SCALAR_EVENT),
- yaml_event_name(YAML_MAPPING_END_EVENT));
- errors++;
- }
- yaml_event_delete(&event);
- }
- return errors;
- }
- static size_t yaml_parse_prefix(yaml_parser_t *parser, LOG_JOB *jb) {
- yaml_event_t event;
- size_t errors = 0;
- if (!yaml_parse(parser, &event))
- return 1;
- if (event.type == YAML_SCALAR_EVENT) {
- if(!log_job_key_prefix_set(jb, (char *) event.data.scalar.value, event.data.scalar.length))
- errors++;
- }
- yaml_event_delete(&event);
- return errors;
- }
- static bool yaml_parse_constant_field_injection(yaml_parser_t *parser, LOG_JOB *jb, bool unmatched) {
- yaml_event_t event;
- if (!yaml_parse(parser, &event) || event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &event, "Expected scalar for constant field injection key");
- yaml_event_delete(&event);
- return false;
- }
- char *key = strndupz((char *)event.data.scalar.value, event.data.scalar.length);
- char *value = NULL;
- bool ret = false;
- yaml_event_delete(&event);
- if (!yaml_parse(parser, &event) || event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &event, "Expected scalar for constant field injection value");
- goto cleanup;
- }
- if(!yaml_scalar_matches(&event, "value", strlen("value"))) {
- yaml_error(parser, &event, "Expected scalar 'value'");
- goto cleanup;
- }
- if (!yaml_parse(parser, &event) || event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &event, "Expected scalar for constant field injection value");
- goto cleanup;
- }
- value = strndupz((char *)event.data.scalar.value, event.data.scalar.length);
- if(!log_job_injection_add(jb, key, strlen(key), value, strlen(value), unmatched))
- ret = false;
- else
- ret = true;
- ret = true;
- cleanup:
- yaml_event_delete(&event);
- freez(key);
- freez(value);
- return !ret ? 1 : 0;
- }
- static bool yaml_parse_injection_mapping(yaml_parser_t *parser, LOG_JOB *jb, bool unmatched) {
- yaml_event_t event;
- size_t errors = 0;
- bool finished = false;
- while (!errors && !finished) {
- if (!yaml_parse(parser, &event)) {
- errors++;
- continue;
- }
- switch (event.type) {
- case YAML_SCALAR_EVENT:
- if (yaml_scalar_matches(&event, "key", strlen("key"))) {
- errors += yaml_parse_constant_field_injection(parser, jb, unmatched);
- } else {
- yaml_error(parser, &event, "Unexpected scalar in injection mapping");
- errors++;
- }
- break;
- case YAML_MAPPING_END_EVENT:
- finished = true;
- break;
- default:
- yaml_error(parser, &event, "Unexpected event in injection mapping");
- errors++;
- break;
- }
- yaml_event_delete(&event);
- }
- return errors == 0;
- }
- static size_t yaml_parse_injections(yaml_parser_t *parser, LOG_JOB *jb, bool unmatched) {
- yaml_event_t event;
- size_t errors = 0;
- bool finished = false;
- if (!yaml_parse_expect_event(parser, YAML_SEQUENCE_START_EVENT))
- return 1;
- while (!errors && !finished) {
- if (!yaml_parse(parser, &event)) {
- errors++;
- continue;
- }
- switch (event.type) {
- case YAML_MAPPING_START_EVENT:
- if (!yaml_parse_injection_mapping(parser, jb, unmatched))
- errors++;
- break;
- case YAML_SEQUENCE_END_EVENT:
- finished = true;
- break;
- default:
- yaml_error(parser, &event, "Unexpected event in injections sequence");
- errors++;
- break;
- }
- yaml_event_delete(&event);
- }
- return errors;
- }
- static size_t yaml_parse_unmatched(yaml_parser_t *parser, LOG_JOB *jb) {
- size_t errors = 0;
- bool finished = false;
- if (!yaml_parse_expect_event(parser, YAML_MAPPING_START_EVENT))
- return 1;
- while (!errors && !finished) {
- yaml_event_t event;
- if (!yaml_parse(parser, &event)) {
- errors++;
- continue;
- }
- switch (event.type) {
- case YAML_SCALAR_EVENT:
- if (yaml_scalar_matches(&event, "key", strlen("key"))) {
- yaml_event_t sub_event;
- if (!yaml_parse(parser, &sub_event)) {
- errors++;
- } else {
- if (sub_event.type == YAML_SCALAR_EVENT) {
- hashed_key_len_set(&jb->unmatched.key, (char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
- } else {
- yaml_error(parser, &sub_event, "expected a scalar value for 'key'");
- errors++;
- }
- yaml_event_delete(&sub_event);
- }
- } else if (yaml_scalar_matches(&event, "inject", strlen("inject"))) {
- errors += yaml_parse_injections(parser, jb, true);
- } else {
- yaml_error(parser, &event, "Unexpected scalar in unmatched section");
- errors++;
- }
- break;
- case YAML_MAPPING_END_EVENT:
- finished = true;
- break;
- default:
- yaml_error(parser, &event, "Unexpected event in unmatched section");
- errors++;
- break;
- }
- yaml_event_delete(&event);
- }
- return errors;
- }
- static size_t yaml_parse_rewrites(yaml_parser_t *parser, LOG_JOB *jb) {
- size_t errors = 0;
- if (!yaml_parse_expect_event(parser, YAML_SEQUENCE_START_EVENT))
- return 1;
- bool finished = false;
- while (!errors && !finished) {
- yaml_event_t event;
- if (!yaml_parse(parser, &event)) {
- errors++;
- continue;
- }
- switch (event.type) {
- case YAML_MAPPING_START_EVENT:
- {
- RW_FLAGS flags = RW_NONE;
- char *key = NULL;
- char *search_pattern = NULL;
- char *replace_pattern = NULL;
- bool mapping_finished = false;
- while (!errors && !mapping_finished) {
- yaml_event_t sub_event;
- if (!yaml_parse(parser, &sub_event)) {
- errors++;
- continue;
- }
- switch (sub_event.type) {
- case YAML_SCALAR_EVENT:
- if (yaml_scalar_matches(&sub_event, "key", strlen("key"))) {
- if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &sub_event, "Expected scalar for rewrite key");
- errors++;
- } else {
- key = strndupz((char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
- yaml_event_delete(&sub_event);
- }
- } else if (yaml_scalar_matches(&sub_event, "match", strlen("match"))) {
- if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &sub_event, "Expected scalar for rewrite match PCRE2 pattern");
- errors++;
- }
- else {
- if(search_pattern)
- freez(search_pattern);
- flags |= RW_MATCH_PCRE2;
- flags &= ~RW_MATCH_NON_EMPTY;
- search_pattern = strndupz((char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
- yaml_event_delete(&sub_event);
- }
- } else if (yaml_scalar_matches(&sub_event, "not_empty", strlen("not_empty"))) {
- if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &sub_event, "Expected scalar for rewrite not empty condition");
- errors++;
- }
- else {
- if(search_pattern)
- freez(search_pattern);
- flags |= RW_MATCH_NON_EMPTY;
- flags &= ~RW_MATCH_PCRE2;
- search_pattern = strndupz((char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
- yaml_event_delete(&sub_event);
- }
- } else if (yaml_scalar_matches(&sub_event, "value", strlen("value"))) {
- if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &sub_event, "Expected scalar for rewrite value");
- errors++;
- } else {
- replace_pattern = strndupz((char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
- yaml_event_delete(&sub_event);
- }
- } else if (yaml_scalar_matches(&sub_event, "stop", strlen("stop"))) {
- if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &sub_event, "Expected scalar for rewrite stop boolean");
- errors++;
- } else {
- if(strncmp((char*)sub_event.data.scalar.value, "no", 2) == 0 ||
- strncmp((char*)sub_event.data.scalar.value, "false", 5) == 0)
- flags |= RW_DONT_STOP;
- else
- flags &= ~RW_DONT_STOP;
- yaml_event_delete(&sub_event);
- }
- } else if (yaml_scalar_matches(&sub_event, "inject", strlen("inject"))) {
- if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &sub_event, "Expected scalar for rewrite inject boolean");
- errors++;
- } else {
- if(strncmp((char*)sub_event.data.scalar.value, "yes", 3) == 0 ||
- strncmp((char*)sub_event.data.scalar.value, "true", 4) == 0)
- flags |= RW_INJECT;
- else
- flags &= ~RW_INJECT;
- yaml_event_delete(&sub_event);
- }
- } else {
- yaml_error(parser, &sub_event, "Unexpected scalar in rewrite mapping");
- errors++;
- }
- break;
- case YAML_MAPPING_END_EVENT:
- if(key) {
- if (!log_job_rewrite_add(jb, key, flags, search_pattern, replace_pattern))
- errors++;
- }
- freez(key);
- key = NULL;
- freez(search_pattern);
- search_pattern = NULL;
- freez(replace_pattern);
- replace_pattern = NULL;
- flags = RW_NONE;
- mapping_finished = true;
- break;
- default:
- yaml_error(parser, &sub_event, "Unexpected event in rewrite mapping");
- errors++;
- break;
- }
- yaml_event_delete(&sub_event);
- }
- }
- break;
- case YAML_SEQUENCE_END_EVENT:
- finished = true;
- break;
- default:
- yaml_error(parser, &event, "Unexpected event in rewrites sequence");
- errors++;
- break;
- }
- yaml_event_delete(&event);
- }
- return errors;
- }
- static size_t yaml_parse_renames(yaml_parser_t *parser, LOG_JOB *jb) {
- size_t errors = 0;
- if (!yaml_parse_expect_event(parser, YAML_SEQUENCE_START_EVENT))
- return 1;
- bool finished = false;
- while (!errors && !finished) {
- yaml_event_t event;
- if (!yaml_parse(parser, &event)) {
- errors++;
- continue;
- }
- switch (event.type) {
- case YAML_MAPPING_START_EVENT:
- {
- struct key_rename rn = { 0 };
- bool mapping_finished = false;
- while (!errors && !mapping_finished) {
- yaml_event_t sub_event;
- if (!yaml_parse(parser, &sub_event)) {
- errors++;
- continue;
- }
- switch (sub_event.type) {
- case YAML_SCALAR_EVENT:
- if (yaml_scalar_matches(&sub_event, "new_key", strlen("new_key"))) {
- if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &sub_event, "Expected scalar for rename new_key");
- errors++;
- } else {
- hashed_key_len_set(&rn.new_key, (char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
- yaml_event_delete(&sub_event);
- }
- } else if (yaml_scalar_matches(&sub_event, "old_key", strlen("old_key"))) {
- if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
- yaml_error(parser, &sub_event, "Expected scalar for rename old_key");
- errors++;
- } else {
- hashed_key_len_set(&rn.old_key, (char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
- yaml_event_delete(&sub_event);
- }
- } else {
- yaml_error(parser, &sub_event, "Unexpected scalar in rewrite mapping");
- errors++;
- }
- break;
- case YAML_MAPPING_END_EVENT:
- if(rn.old_key.key && rn.new_key.key) {
- if (!log_job_rename_add(jb, rn.new_key.key, rn.new_key.len,
- rn.old_key.key, rn.old_key.len))
- errors++;
- }
- rename_cleanup(&rn);
- mapping_finished = true;
- break;
- default:
- yaml_error(parser, &sub_event, "Unexpected event in rewrite mapping");
- errors++;
- break;
- }
- yaml_event_delete(&sub_event);
- }
- }
- break;
- case YAML_SEQUENCE_END_EVENT:
- finished = true;
- break;
- default:
- yaml_error(parser, &event, "Unexpected event in rewrites sequence");
- errors++;
- break;
- }
- yaml_event_delete(&event);
- }
- return errors;
- }
- static size_t yaml_parse_pattern(yaml_parser_t *parser, LOG_JOB *jb) {
- yaml_event_t event;
- size_t errors = 0;
- if (!yaml_parse(parser, &event))
- return 1;
- if(event.type == YAML_SCALAR_EVENT)
- log_job_pattern_set(jb, (char *) event.data.scalar.value, event.data.scalar.length);
- else {
- yaml_error(parser, &event, "unexpected event type");
- errors++;
- }
- yaml_event_delete(&event);
- return errors;
- }
- static size_t yaml_parse_initialized(yaml_parser_t *parser, LOG_JOB *jb) {
- size_t errors = 0;
- if(!yaml_parse_expect_event(parser, YAML_STREAM_START_EVENT)) {
- errors++;
- goto cleanup;
- }
- if(!yaml_parse_expect_event(parser, YAML_DOCUMENT_START_EVENT)) {
- errors++;
- goto cleanup;
- }
- if(!yaml_parse_expect_event(parser, YAML_MAPPING_START_EVENT)) {
- errors++;
- goto cleanup;
- }
- bool finished = false;
- while (!errors && !finished) {
- yaml_event_t event;
- if(!yaml_parse(parser, &event)) {
- errors++;
- continue;
- }
- switch(event.type) {
- default:
- yaml_error(parser, &event, "unexpected type");
- errors++;
- break;
- case YAML_MAPPING_END_EVENT:
- finished = true;
- break;
- case YAML_SCALAR_EVENT:
- if (yaml_scalar_matches(&event, "pattern", strlen("pattern")))
- errors += yaml_parse_pattern(parser, jb);
- else if (yaml_scalar_matches(&event, "prefix", strlen("prefix")))
- errors += yaml_parse_prefix(parser, jb);
- else if (yaml_scalar_matches(&event, "filename", strlen("filename")))
- errors += yaml_parse_filename_injection(parser, jb);
- else if (yaml_scalar_matches(&event, "filter", strlen("filter")))
- errors += yaml_parse_filters(parser, jb);
- else if (yaml_scalar_matches(&event, "inject", strlen("inject")))
- errors += yaml_parse_injections(parser, jb, false);
- else if (yaml_scalar_matches(&event, "unmatched", strlen("unmatched")))
- errors += yaml_parse_unmatched(parser, jb);
- else if (yaml_scalar_matches(&event, "rewrite", strlen("rewrite")))
- errors += yaml_parse_rewrites(parser, jb);
- else if (yaml_scalar_matches(&event, "rename", strlen("rename")))
- errors += yaml_parse_renames(parser, jb);
- else {
- yaml_error(parser, &event, "unexpected scalar");
- errors++;
- }
- break;
- }
- yaml_event_delete(&event);
- }
- if(!errors && !yaml_parse_expect_event(parser, YAML_DOCUMENT_END_EVENT)) {
- errors++;
- goto cleanup;
- }
- if(!errors && !yaml_parse_expect_event(parser, YAML_STREAM_END_EVENT)) {
- errors++;
- goto cleanup;
- }
- cleanup:
- return errors;
- }
- bool yaml_parse_file(const char *config_file_path, LOG_JOB *jb) {
- if(!config_file_path || !*config_file_path) {
- log2stderr("yaml configuration filename cannot be empty.");
- return false;
- }
- FILE *fp = fopen(config_file_path, "r");
- if (!fp) {
- log2stderr("Error opening config file: %s", config_file_path);
- return false;
- }
- yaml_parser_t parser;
- yaml_parser_initialize(&parser);
- yaml_parser_set_input_file(&parser, fp);
- size_t errors = yaml_parse_initialized(&parser, jb);
- yaml_parser_delete(&parser);
- fclose(fp);
- return errors == 0;
- }
- bool yaml_parse_config(const char *config_name, LOG_JOB *jb) {
- char filename[FILENAME_MAX + 1];
- snprintf(filename, sizeof(filename), "%s/%s.yaml", LOG2JOURNAL_CONFIG_PATH, config_name);
- return yaml_parse_file(filename, jb);
- }
- #endif // HAVE_LIBYAML
- // ----------------------------------------------------------------------------
- // printing yaml
- static void yaml_print_multiline_value(const char *s, size_t depth) {
- if (!s)
- s = "";
- do {
- const char* next = strchr(s, '\n');
- if(next) next++;
- size_t len = next ? (size_t)(next - s) : strlen(s);
- char buf[len + 1];
- copy_to_buffer(buf, sizeof(buf), s, len);
- fprintf(stderr, "%.*s%s%s",
- (int)(depth * 2), " ",
- buf, next ? "" : "\n");
- s = next;
- } while(s && *s);
- }
- static bool needs_quotes_in_yaml(const char *str) {
- // Lookup table for special YAML characters
- static bool special_chars[256] = { false };
- static bool table_initialized = false;
- if (!table_initialized) {
- // Initialize the lookup table
- const char *special_chars_str = ":{}[],&*!|>'\"%@`^";
- for (const char *c = special_chars_str; *c; ++c) {
- special_chars[(unsigned char)*c] = true;
- }
- table_initialized = true;
- }
- while (*str) {
- if (special_chars[(unsigned char)*str]) {
- return true;
- }
- str++;
- }
- return false;
- }
- static void yaml_print_node(const char *key, const char *value, size_t depth, bool dash) {
- if(depth > 10) depth = 10;
- const char *quote = "'";
- const char *second_line = NULL;
- if(value && strchr(value, '\n')) {
- second_line = value;
- value = "|";
- quote = "";
- }
- else if(!value || !needs_quotes_in_yaml(value))
- quote = "";
- fprintf(stderr, "%.*s%s%s%s%s%s%s\n",
- (int)(depth * 2), " ", dash ? "- ": "",
- key ? key : "", key ? ": " : "",
- quote, value ? value : "", quote);
- if(second_line) {
- yaml_print_multiline_value(second_line, depth + 1);
- }
- }
- void log_job_configuration_to_yaml(LOG_JOB *jb) {
- if(jb->pattern)
- yaml_print_node("pattern", jb->pattern, 0, false);
- if(jb->prefix) {
- fprintf(stderr, "\n");
- yaml_print_node("prefix", jb->prefix, 0, false);
- }
- if(jb->filename.key.key) {
- fprintf(stderr, "\n");
- yaml_print_node("filename", NULL, 0, false);
- yaml_print_node("key", jb->filename.key.key, 1, false);
- }
- if(jb->filter.include.pattern || jb->filter.exclude.pattern) {
- fprintf(stderr, "\n");
- yaml_print_node("filter", NULL, 0, false);
- if(jb->filter.include.pattern)
- yaml_print_node("include", jb->filter.include.pattern, 1, false);
- if(jb->filter.exclude.pattern)
- yaml_print_node("exclude", jb->filter.exclude.pattern, 1, false);
- }
- if(jb->renames.used) {
- fprintf(stderr, "\n");
- yaml_print_node("rename", NULL, 0, false);
- for(size_t i = 0; i < jb->renames.used ;i++) {
- yaml_print_node("new_key", jb->renames.array[i].new_key.key, 1, true);
- yaml_print_node("old_key", jb->renames.array[i].old_key.key, 2, false);
- }
- }
- if(jb->injections.used) {
- fprintf(stderr, "\n");
- yaml_print_node("inject", NULL, 0, false);
- for (size_t i = 0; i < jb->injections.used; i++) {
- yaml_print_node("key", jb->injections.keys[i].key.key, 1, true);
- yaml_print_node("value", jb->injections.keys[i].value.pattern, 2, false);
- }
- }
- if(jb->rewrites.used) {
- fprintf(stderr, "\n");
- yaml_print_node("rewrite", NULL, 0, false);
- for(size_t i = 0; i < jb->rewrites.used ;i++) {
- REWRITE *rw = &jb->rewrites.array[i];
- yaml_print_node("key", rw->key.key, 1, true);
- if(rw->flags & RW_MATCH_PCRE2)
- yaml_print_node("match", rw->match_pcre2.pattern, 2, false);
- else if(rw->flags & RW_MATCH_NON_EMPTY)
- yaml_print_node("not_empty", rw->match_non_empty.pattern, 2, false);
- yaml_print_node("value", rw->value.pattern, 2, false);
- if(rw->flags & RW_INJECT)
- yaml_print_node("inject", "yes", 2, false);
- if(rw->flags & RW_DONT_STOP)
- yaml_print_node("stop", "no", 2, false);
- }
- }
- if(jb->unmatched.key.key || jb->unmatched.injections.used) {
- fprintf(stderr, "\n");
- yaml_print_node("unmatched", NULL, 0, false);
- if(jb->unmatched.key.key)
- yaml_print_node("key", jb->unmatched.key.key, 1, false);
- if(jb->unmatched.injections.used) {
- fprintf(stderr, "\n");
- yaml_print_node("inject", NULL, 1, false);
- for (size_t i = 0; i < jb->unmatched.injections.used; i++) {
- yaml_print_node("key", jb->unmatched.injections.keys[i].key.key, 2, true);
- yaml_print_node("value", jb->unmatched.injections.keys[i].value.pattern, 3, false);
- }
- }
- }
- }
|