log2journal-yaml.c 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838
  1. // SPDX-License-Identifier: GPL-3.0-or-later
  2. #include "log2journal.h"
  3. // ----------------------------------------------------------------------------
  4. // yaml configuration file
  5. #ifdef HAVE_LIBYAML
  6. static const char *yaml_event_name(yaml_event_type_t type) {
  7. switch (type) {
  8. case YAML_NO_EVENT:
  9. return "YAML_NO_EVENT";
  10. case YAML_SCALAR_EVENT:
  11. return "YAML_SCALAR_EVENT";
  12. case YAML_ALIAS_EVENT:
  13. return "YAML_ALIAS_EVENT";
  14. case YAML_MAPPING_START_EVENT:
  15. return "YAML_MAPPING_START_EVENT";
  16. case YAML_MAPPING_END_EVENT:
  17. return "YAML_MAPPING_END_EVENT";
  18. case YAML_SEQUENCE_START_EVENT:
  19. return "YAML_SEQUENCE_START_EVENT";
  20. case YAML_SEQUENCE_END_EVENT:
  21. return "YAML_SEQUENCE_END_EVENT";
  22. case YAML_STREAM_START_EVENT:
  23. return "YAML_STREAM_START_EVENT";
  24. case YAML_STREAM_END_EVENT:
  25. return "YAML_STREAM_END_EVENT";
  26. case YAML_DOCUMENT_START_EVENT:
  27. return "YAML_DOCUMENT_START_EVENT";
  28. case YAML_DOCUMENT_END_EVENT:
  29. return "YAML_DOCUMENT_END_EVENT";
  30. default:
  31. return "UNKNOWN";
  32. }
  33. }
  34. #define yaml_error(parser, event, fmt, args...) yaml_error_with_trace(parser, event, __LINE__, __FUNCTION__, __FILE__, fmt, ##args)
  35. 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)));
  36. 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, ...) {
  37. char buf[1024] = ""; // Initialize buf to an empty string
  38. const char *type = "";
  39. if(event) {
  40. type = yaml_event_name(event->type);
  41. switch (event->type) {
  42. case YAML_SCALAR_EVENT:
  43. copy_to_buffer(buf, sizeof(buf), (char *)event->data.scalar.value, event->data.scalar.length);
  44. break;
  45. case YAML_ALIAS_EVENT:
  46. snprintf(buf, sizeof(buf), "%s", event->data.alias.anchor);
  47. break;
  48. default:
  49. break;
  50. }
  51. }
  52. fprintf(stderr, "YAML %zu@%s, %s(): (line %d, column %d, %s%s%s): ",
  53. line, file, function,
  54. (int)(parser->mark.line + 1), (int)(parser->mark.column + 1),
  55. type, buf[0]? ", near ": "", buf);
  56. va_list args;
  57. va_start(args, format);
  58. vfprintf(stderr, format, args);
  59. va_end(args);
  60. fprintf(stderr, "\n");
  61. }
  62. #define yaml_parse(parser, event) yaml_parse_with_trace(parser, event, __LINE__, __FUNCTION__, __FILE__)
  63. static bool yaml_parse_with_trace(yaml_parser_t *parser, yaml_event_t *event, size_t line, const char *function, const char *file) {
  64. if (!yaml_parser_parse(parser, event)) {
  65. yaml_error(parser, NULL, "YAML parser error %d", parser->error);
  66. return false;
  67. }
  68. // fprintf(stderr, ">>> %s >>> %.*s\n",
  69. // yaml_event_name(event->type),
  70. // event->type == YAML_SCALAR_EVENT ? event->data.scalar.length : 0,
  71. // event->type == YAML_SCALAR_EVENT ? (char *)event->data.scalar.value : "");
  72. return true;
  73. }
  74. #define yaml_parse_expect_event(parser, type) yaml_parse_expect_event_with_trace(parser, type, __LINE__, __FUNCTION__, __FILE__)
  75. 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) {
  76. yaml_event_t event;
  77. if (!yaml_parse(parser, &event))
  78. return false;
  79. bool ret = true;
  80. if(event.type != type) {
  81. yaml_error_with_trace(parser, &event, line, function, file, "unexpected event - expecting: %s", yaml_event_name(type));
  82. ret = false;
  83. }
  84. // else
  85. // fprintf(stderr, "OK (%zu@%s, %s()\n", line, file, function);
  86. yaml_event_delete(&event);
  87. return ret;
  88. }
  89. #define yaml_scalar_matches(event, s, len) yaml_scalar_matches_with_trace(event, s, len, __LINE__, __FUNCTION__, __FILE__)
  90. 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) {
  91. if(event->type != YAML_SCALAR_EVENT)
  92. return false;
  93. if(len != event->data.scalar.length)
  94. return false;
  95. // else
  96. // fprintf(stderr, "OK (%zu@%s, %s()\n", line, file, function);
  97. return strcmp((char *)event->data.scalar.value, s) == 0;
  98. }
  99. // ----------------------------------------------------------------------------
  100. static struct key_dup *yaml_parse_duplicate_key(struct log_job *jb, yaml_parser_t *parser) {
  101. yaml_event_t event;
  102. if (!yaml_parse(parser, &event))
  103. return false;
  104. struct key_dup *kd = NULL;
  105. if(event.type == YAML_SCALAR_EVENT) {
  106. kd = log_job_add_duplication_to_job(jb, (char *) event.data.scalar.value, event.data.scalar.length);
  107. }
  108. else
  109. yaml_error(parser, &event, "duplicate key must be a scalar.");
  110. yaml_event_delete(&event);
  111. return kd;
  112. }
  113. static size_t yaml_parse_duplicate_from(struct log_job *jb, yaml_parser_t *parser, struct key_dup *kd) {
  114. size_t errors = 0;
  115. yaml_event_t event;
  116. if (!yaml_parse(parser, &event))
  117. return 1;
  118. bool ret = true;
  119. if(event.type == YAML_SCALAR_EVENT)
  120. ret = log_job_add_key_to_duplication(kd, (char *) event.data.scalar.value, event.data.scalar.length);
  121. else if(event.type == YAML_SEQUENCE_START_EVENT) {
  122. bool finished = false;
  123. while(!errors && !finished) {
  124. yaml_event_t sub_event;
  125. if (!yaml_parse(parser, &sub_event))
  126. return errors++;
  127. else {
  128. if (sub_event.type == YAML_SCALAR_EVENT)
  129. log_job_add_key_to_duplication(kd, (char *) sub_event.data.scalar.value
  130. , sub_event.data.scalar.length
  131. );
  132. else if (sub_event.type == YAML_SEQUENCE_END_EVENT)
  133. finished = true;
  134. yaml_event_delete(&sub_event);
  135. }
  136. }
  137. }
  138. else
  139. yaml_error(parser, &event, "not expected event type");
  140. yaml_event_delete(&event);
  141. return errors;
  142. }
  143. static size_t yaml_parse_filename_injection(yaml_parser_t *parser, struct log_job *jb) {
  144. yaml_event_t event;
  145. size_t errors = 0;
  146. if(!yaml_parse_expect_event(parser, YAML_MAPPING_START_EVENT))
  147. return 1;
  148. if (!yaml_parse(parser, &event))
  149. return 1;
  150. if (yaml_scalar_matches(&event, "key", strlen("key"))) {
  151. yaml_event_t sub_event;
  152. if (!yaml_parse(parser, &sub_event))
  153. errors++;
  154. else {
  155. if (event.type == YAML_SCALAR_EVENT) {
  156. if(!log_job_add_filename_key(jb, (char *)sub_event.data.scalar.value, sub_event.data.scalar.length))
  157. errors++;
  158. }
  159. else {
  160. yaml_error(parser, &sub_event, "expected the filename as %s", yaml_event_name(YAML_SCALAR_EVENT));
  161. errors++;
  162. }
  163. yaml_event_delete(&sub_event);
  164. }
  165. }
  166. if(!yaml_parse_expect_event(parser, YAML_MAPPING_END_EVENT))
  167. errors++;
  168. yaml_event_delete(&event);
  169. return errors;
  170. }
  171. static size_t yaml_parse_duplicates_injection(yaml_parser_t *parser, struct log_job *jb) {
  172. if (!yaml_parse_expect_event(parser, YAML_SEQUENCE_START_EVENT))
  173. return 1;
  174. struct key_dup *kd = NULL;
  175. // Expecting a key-value pair for each duplicate
  176. bool finished;
  177. size_t errors = 0;
  178. while (!errors && !finished) {
  179. yaml_event_t event;
  180. if (!yaml_parse(parser, &event)) {
  181. errors++;
  182. break;
  183. }
  184. if(event.type == YAML_MAPPING_START_EVENT) {
  185. ;
  186. }
  187. if (event.type == YAML_SEQUENCE_END_EVENT) {
  188. finished = true;
  189. }
  190. else if(event.type == YAML_SCALAR_EVENT) {
  191. if (yaml_scalar_matches(&event, "key", strlen("key"))) {
  192. kd = yaml_parse_duplicate_key(jb, parser);
  193. if (!kd)
  194. errors++;
  195. else {
  196. while (!errors && kd) {
  197. yaml_event_t sub_event;
  198. if (!yaml_parse(parser, &sub_event)) {
  199. errors++;
  200. break;
  201. }
  202. if (sub_event.type == YAML_MAPPING_END_EVENT) {
  203. kd = NULL;
  204. } else if (sub_event.type == YAML_SCALAR_EVENT) {
  205. if (yaml_scalar_matches(&sub_event, "values_of", strlen("values_of"))) {
  206. if (!kd) {
  207. yaml_error(parser, &sub_event, "Found 'values_of' but the 'key' is not set.");
  208. errors++;
  209. } else
  210. errors += yaml_parse_duplicate_from(jb, parser, kd);
  211. } else {
  212. yaml_error(parser, &sub_event, "unknown scalar");
  213. errors++;
  214. }
  215. } else {
  216. yaml_error(parser, &sub_event, "unexpected event type");
  217. errors++;
  218. }
  219. // Delete the event after processing
  220. yaml_event_delete(&event);
  221. }
  222. }
  223. } else {
  224. yaml_error(parser, &event, "unknown scalar");
  225. errors++;
  226. }
  227. }
  228. yaml_event_delete(&event);
  229. }
  230. return errors;
  231. }
  232. static bool yaml_parse_constant_field_injection(yaml_parser_t *parser, struct log_job *jb, bool unmatched) {
  233. yaml_event_t event;
  234. if (!yaml_parse(parser, &event) || event.type != YAML_SCALAR_EVENT) {
  235. yaml_error(parser, &event, "Expected scalar for constant field injection key");
  236. yaml_event_delete(&event);
  237. return false;
  238. }
  239. char *key = strndupz((char *)event.data.scalar.value, event.data.scalar.length);
  240. char *value = NULL;
  241. bool ret = false;
  242. yaml_event_delete(&event);
  243. if (!yaml_parse(parser, &event) || event.type != YAML_SCALAR_EVENT) {
  244. yaml_error(parser, &event, "Expected scalar for constant field injection value");
  245. goto cleanup;
  246. }
  247. if(!yaml_scalar_matches(&event, "value", strlen("value"))) {
  248. yaml_error(parser, &event, "Expected scalar 'value'");
  249. goto cleanup;
  250. }
  251. if (!yaml_parse(parser, &event) || event.type != YAML_SCALAR_EVENT) {
  252. yaml_error(parser, &event, "Expected scalar for constant field injection value");
  253. goto cleanup;
  254. }
  255. value = strndupz((char *)event.data.scalar.value, event.data.scalar.length);
  256. if(!log_job_add_injection(jb, key, strlen(key), value, strlen(value), unmatched))
  257. ret = false;
  258. else
  259. ret = true;
  260. ret = true;
  261. cleanup:
  262. yaml_event_delete(&event);
  263. freez(key);
  264. freez(value);
  265. return !ret ? 1 : 0;
  266. }
  267. static bool yaml_parse_injection_mapping(yaml_parser_t *parser, struct log_job *jb, bool unmatched) {
  268. yaml_event_t event;
  269. size_t errors = 0;
  270. bool finished = false;
  271. while (!errors && !finished) {
  272. if (!yaml_parse(parser, &event)) {
  273. errors++;
  274. continue;
  275. }
  276. switch (event.type) {
  277. case YAML_SCALAR_EVENT:
  278. if (yaml_scalar_matches(&event, "key", strlen("key"))) {
  279. errors += yaml_parse_constant_field_injection(parser, jb, unmatched);
  280. } else {
  281. yaml_error(parser, &event, "Unexpected scalar in injection mapping");
  282. errors++;
  283. }
  284. break;
  285. case YAML_MAPPING_END_EVENT:
  286. finished = true;
  287. break;
  288. default:
  289. yaml_error(parser, &event, "Unexpected event in injection mapping");
  290. errors++;
  291. break;
  292. }
  293. yaml_event_delete(&event);
  294. }
  295. return errors == 0;
  296. }
  297. static size_t yaml_parse_injections(yaml_parser_t *parser, struct log_job *jb, bool unmatched) {
  298. yaml_event_t event;
  299. size_t errors = 0;
  300. bool finished = false;
  301. if (!yaml_parse_expect_event(parser, YAML_SEQUENCE_START_EVENT))
  302. return 1;
  303. while (!errors && !finished) {
  304. if (!yaml_parse(parser, &event)) {
  305. errors++;
  306. continue;
  307. }
  308. switch (event.type) {
  309. case YAML_MAPPING_START_EVENT:
  310. if (!yaml_parse_injection_mapping(parser, jb, unmatched))
  311. errors++;
  312. break;
  313. case YAML_SEQUENCE_END_EVENT:
  314. finished = true;
  315. break;
  316. default:
  317. yaml_error(parser, &event, "Unexpected event in injections sequence");
  318. errors++;
  319. break;
  320. }
  321. yaml_event_delete(&event);
  322. }
  323. return errors;
  324. }
  325. static size_t yaml_parse_unmatched(yaml_parser_t *parser, struct log_job *jb) {
  326. size_t errors = 0;
  327. bool finished = false;
  328. if (!yaml_parse_expect_event(parser, YAML_MAPPING_START_EVENT))
  329. return 1;
  330. while (!errors && !finished) {
  331. yaml_event_t event;
  332. if (!yaml_parse(parser, &event)) {
  333. errors++;
  334. continue;
  335. }
  336. switch (event.type) {
  337. case YAML_SCALAR_EVENT:
  338. if (yaml_scalar_matches(&event, "key", strlen("key"))) {
  339. yaml_event_t sub_event;
  340. if (!yaml_parse(parser, &sub_event)) {
  341. errors++;
  342. } else {
  343. if (sub_event.type == YAML_SCALAR_EVENT) {
  344. jb->unmatched.key = strndupz((char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
  345. } else {
  346. yaml_error(parser, &sub_event, "expected a scalar value for 'key'");
  347. errors++;
  348. }
  349. yaml_event_delete(&sub_event);
  350. }
  351. } else if (yaml_scalar_matches(&event, "inject", strlen("inject"))) {
  352. errors += yaml_parse_injections(parser, jb, true);
  353. } else {
  354. yaml_error(parser, &event, "Unexpected scalar in unmatched section");
  355. errors++;
  356. }
  357. break;
  358. case YAML_MAPPING_END_EVENT:
  359. finished = true;
  360. break;
  361. default:
  362. yaml_error(parser, &event, "Unexpected event in unmatched section");
  363. errors++;
  364. break;
  365. }
  366. yaml_event_delete(&event);
  367. }
  368. return errors;
  369. }
  370. static size_t yaml_parse_rewrites(yaml_parser_t *parser, struct log_job *jb) {
  371. size_t errors = 0;
  372. if (!yaml_parse_expect_event(parser, YAML_SEQUENCE_START_EVENT))
  373. return 1;
  374. bool finished = false;
  375. while (!errors && !finished) {
  376. yaml_event_t event;
  377. if (!yaml_parse(parser, &event)) {
  378. errors++;
  379. continue;
  380. }
  381. switch (event.type) {
  382. case YAML_MAPPING_START_EVENT:
  383. {
  384. struct key_rewrite rw = {0};
  385. bool mapping_finished = false;
  386. while (!errors && !mapping_finished) {
  387. yaml_event_t sub_event;
  388. if (!yaml_parse(parser, &sub_event)) {
  389. errors++;
  390. continue;
  391. }
  392. switch (sub_event.type) {
  393. case YAML_SCALAR_EVENT:
  394. if (yaml_scalar_matches(&sub_event, "key", strlen("key"))) {
  395. if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
  396. yaml_error(parser, &sub_event, "Expected scalar for rewrite key");
  397. errors++;
  398. } else {
  399. rw.key = strndupz((char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
  400. yaml_event_delete(&sub_event);
  401. }
  402. } else if (yaml_scalar_matches(&sub_event, "search", strlen("search"))) {
  403. if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
  404. yaml_error(parser, &sub_event, "Expected scalar for rewrite search pattern");
  405. errors++;
  406. } else {
  407. rw.search_pattern = strndupz((char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
  408. yaml_event_delete(&sub_event);
  409. }
  410. } else if (yaml_scalar_matches(&sub_event, "replace", strlen("replace"))) {
  411. if (!yaml_parse(parser, &sub_event) || sub_event.type != YAML_SCALAR_EVENT) {
  412. yaml_error(parser, &sub_event, "Expected scalar for rewrite replace pattern");
  413. errors++;
  414. } else {
  415. rw.replace_pattern = strndupz((char *)sub_event.data.scalar.value, sub_event.data.scalar.length);
  416. yaml_event_delete(&sub_event);
  417. }
  418. } else {
  419. yaml_error(parser, &sub_event, "Unexpected scalar in rewrite mapping");
  420. errors++;
  421. }
  422. break;
  423. case YAML_MAPPING_END_EVENT:
  424. if(rw.key && rw.search_pattern && rw.replace_pattern) {
  425. if (!log_job_add_rewrite(jb, rw.key, rw.search_pattern, rw.replace_pattern))
  426. errors++;
  427. }
  428. freez(rw.key);
  429. freez(rw.search_pattern);
  430. freez(rw.replace_pattern);
  431. memset(&rw, 0, sizeof(rw));
  432. mapping_finished = true;
  433. break;
  434. default:
  435. yaml_error(parser, &sub_event, "Unexpected event in rewrite mapping");
  436. errors++;
  437. break;
  438. }
  439. yaml_event_delete(&sub_event);
  440. }
  441. }
  442. break;
  443. case YAML_SEQUENCE_END_EVENT:
  444. finished = true;
  445. break;
  446. default:
  447. yaml_error(parser, &event, "Unexpected event in rewrites sequence");
  448. errors++;
  449. break;
  450. }
  451. yaml_event_delete(&event);
  452. }
  453. return errors;
  454. }
  455. static size_t yaml_parse_pattern(yaml_parser_t *parser, struct log_job *jb) {
  456. yaml_event_t event;
  457. size_t errors = 0;
  458. if (!yaml_parse(parser, &event))
  459. return 1;
  460. if(event.type == YAML_SCALAR_EVENT)
  461. jb->pattern = strndupz((char *)event.data.scalar.value, event.data.scalar.length);
  462. else {
  463. yaml_error(parser, &event, "unexpected event type");
  464. errors++;
  465. }
  466. yaml_event_delete(&event);
  467. return errors;
  468. }
  469. static size_t yaml_parse_initialized(yaml_parser_t *parser, struct log_job *jb) {
  470. size_t errors = 0;
  471. if(!yaml_parse_expect_event(parser, YAML_STREAM_START_EVENT)) {
  472. errors++;
  473. goto cleanup;
  474. }
  475. if(!yaml_parse_expect_event(parser, YAML_DOCUMENT_START_EVENT)) {
  476. errors++;
  477. goto cleanup;
  478. }
  479. if(!yaml_parse_expect_event(parser, YAML_MAPPING_START_EVENT)) {
  480. errors++;
  481. goto cleanup;
  482. }
  483. bool finished = false;
  484. while (!errors && !finished) {
  485. yaml_event_t event;
  486. if(!yaml_parse(parser, &event)) {
  487. errors++;
  488. continue;
  489. }
  490. switch(event.type) {
  491. default:
  492. yaml_error(parser, &event, "unexpected type");
  493. errors++;
  494. break;
  495. case YAML_MAPPING_END_EVENT:
  496. finished = true;
  497. break;
  498. case YAML_SCALAR_EVENT:
  499. if (yaml_scalar_matches(&event, "pattern", strlen("pattern")))
  500. errors += yaml_parse_pattern(parser, jb);
  501. else if (yaml_scalar_matches(&event, "filename", strlen("filename")))
  502. errors += yaml_parse_filename_injection(parser, jb);
  503. else if (yaml_scalar_matches(&event, "duplicate", strlen("duplicate")))
  504. errors += yaml_parse_duplicates_injection(parser, jb);
  505. else if (yaml_scalar_matches(&event, "inject", strlen("inject")))
  506. errors += yaml_parse_injections(parser, jb, false);
  507. else if (yaml_scalar_matches(&event, "unmatched", strlen("unmatched")))
  508. errors += yaml_parse_unmatched(parser, jb);
  509. else if (yaml_scalar_matches(&event, "rewrite", strlen("rewrite")))
  510. errors += yaml_parse_rewrites(parser, jb);
  511. else {
  512. yaml_error(parser, &event, "unexpected scalar");
  513. errors++;
  514. }
  515. break;
  516. }
  517. yaml_event_delete(&event);
  518. }
  519. if(!yaml_parse_expect_event(parser, YAML_DOCUMENT_END_EVENT)) {
  520. errors++;
  521. goto cleanup;
  522. }
  523. if(!yaml_parse_expect_event(parser, YAML_STREAM_END_EVENT)) {
  524. errors++;
  525. goto cleanup;
  526. }
  527. cleanup:
  528. return errors;
  529. }
  530. bool yaml_parse_file(const char *config_file_path, struct log_job *jb) {
  531. if(!config_file_path || !*config_file_path) {
  532. log2stderr("yaml configuration filename cannot be empty.");
  533. return false;
  534. }
  535. FILE *fp = fopen(config_file_path, "r");
  536. if (!fp) {
  537. log2stderr("Error opening config file: %s", config_file_path);
  538. return false;
  539. }
  540. yaml_parser_t parser;
  541. yaml_parser_initialize(&parser);
  542. yaml_parser_set_input_file(&parser, fp);
  543. size_t errors = yaml_parse_initialized(&parser, jb);
  544. yaml_parser_delete(&parser);
  545. fclose(fp);
  546. return errors == 0;
  547. }
  548. bool yaml_parse_config(const char *config_name, struct log_job *jb) {
  549. char filename[FILENAME_MAX + 1];
  550. snprintf(filename, sizeof(filename), "%s/%s.yaml", LOG2JOURNAL_CONFIG_PATH, config_name);
  551. return yaml_parse_file(filename, jb);
  552. }
  553. #endif // HAVE_LIBYAML
  554. // ----------------------------------------------------------------------------
  555. // printing yaml
  556. static void yaml_print_multiline_value(const char *s, size_t depth) {
  557. if (!s)
  558. s = "";
  559. do {
  560. const char* next = strchr(s, '\n');
  561. if(next) next++;
  562. size_t len = next ? (size_t)(next - s) : strlen(s);
  563. char buf[len + 1];
  564. strncpy(buf, s, len);
  565. buf[len] = '\0';
  566. fprintf(stderr, "%.*s%s%s",
  567. (int)(depth * 2), " ",
  568. buf, next ? "" : "\n");
  569. s = next;
  570. } while(s && *s);
  571. }
  572. static bool needs_quotes_in_yaml(const char *str) {
  573. // Lookup table for special YAML characters
  574. static bool special_chars[256] = { false };
  575. static bool table_initialized = false;
  576. if (!table_initialized) {
  577. // Initialize the lookup table
  578. const char *special_chars_str = ":{}[],&*!|>'\"%@`^";
  579. for (const char *c = special_chars_str; *c; ++c) {
  580. special_chars[(unsigned char)*c] = true;
  581. }
  582. table_initialized = true;
  583. }
  584. while (*str) {
  585. if (special_chars[(unsigned char)*str]) {
  586. return true;
  587. }
  588. str++;
  589. }
  590. return false;
  591. }
  592. static void yaml_print_node(const char *key, const char *value, size_t depth, bool dash) {
  593. if(depth > 10) depth = 10;
  594. const char *quote = "\"";
  595. const char *second_line = NULL;
  596. if(value && strchr(value, '\n')) {
  597. second_line = value;
  598. value = "|";
  599. quote = "";
  600. }
  601. else if(!value || !needs_quotes_in_yaml(value))
  602. quote = "";
  603. fprintf(stderr, "%.*s%s%s%s%s%s%s\n",
  604. (int)(depth * 2), " ", dash ? "- ": "",
  605. key ? key : "", key ? ": " : "",
  606. quote, value ? value : "", quote);
  607. if(second_line) {
  608. yaml_print_multiline_value(second_line, depth + 1);
  609. }
  610. }
  611. void log_job_to_yaml(struct log_job *jb) {
  612. if(jb->pattern)
  613. yaml_print_node("pattern", jb->pattern, 0, false);
  614. if(jb->prefix) {
  615. fprintf(stderr, "\n");
  616. yaml_print_node("prefix", jb->prefix, 0, false);
  617. }
  618. if(jb->filename.key) {
  619. fprintf(stderr, "\n");
  620. yaml_print_node("filename", NULL, 0, false);
  621. yaml_print_node("key", jb->filename.key, 1, false);
  622. }
  623. if(jb->dups.used) {
  624. fprintf(stderr, "\n");
  625. yaml_print_node("duplicate", NULL, 0, false);
  626. for(size_t i = 0; i < jb->dups.used ;i++) {
  627. struct key_dup *kd = &jb->dups.array[i];
  628. yaml_print_node("key", kd->target, 1, true);
  629. yaml_print_node("values_of", NULL, 2, false);
  630. for(size_t k = 0; k < kd->used ;k++)
  631. yaml_print_node(NULL, kd->keys[k], 3, true);
  632. }
  633. }
  634. if(jb->injections.used) {
  635. fprintf(stderr, "\n");
  636. yaml_print_node("inject", NULL, 0, false);
  637. for (size_t i = 0; i < jb->injections.used; i++) {
  638. yaml_print_node("key", jb->injections.keys[i].key, 1, true);
  639. yaml_print_node("value", jb->injections.keys[i].value.s, 2, false);
  640. }
  641. }
  642. if(jb->rewrites.used) {
  643. fprintf(stderr, "\n");
  644. yaml_print_node("rewrite", NULL, 0, false);
  645. for(size_t i = 0; i < jb->rewrites.used ;i++) {
  646. yaml_print_node("key", jb->rewrites.array[i].key, 1, true);
  647. yaml_print_node("search", jb->rewrites.array[i].search_pattern, 2, false);
  648. yaml_print_node("replace", jb->rewrites.array[i].replace_pattern, 2, false);
  649. }
  650. }
  651. if(jb->renames.used) {
  652. fprintf(stderr, "\n");
  653. yaml_print_node("rename", NULL, 0, false);
  654. for(size_t i = 0; i < jb->renames.used ;i++) {
  655. yaml_print_node("new_key", jb->renames.array[i].new_key, 1, true);
  656. yaml_print_node("old_key", jb->renames.array[i].old_key, 2, false);
  657. }
  658. }
  659. if(jb->unmatched.key || jb->unmatched.injections.used) {
  660. fprintf(stderr, "\n");
  661. yaml_print_node("unmatched", NULL, 0, false);
  662. if(jb->unmatched.key)
  663. yaml_print_node("key", jb->unmatched.key, 1, false);
  664. if(jb->unmatched.injections.used) {
  665. fprintf(stderr, "\n");
  666. yaml_print_node("inject", NULL, 1, false);
  667. for (size_t i = 0; i < jb->unmatched.injections.used; i++) {
  668. yaml_print_node("key", jb->unmatched.injections.keys[i].key, 2, true);
  669. yaml_print_node("value", jb->unmatched.injections.keys[i].value.s, 3, false);
  670. }
  671. }
  672. }
  673. }