summaryrefslogtreecommitdiffstats
path: root/fluent-bit/tests/internal/regex.c
diff options
context:
space:
mode:
Diffstat (limited to 'fluent-bit/tests/internal/regex.c')
-rw-r--r--fluent-bit/tests/internal/regex.c330
1 files changed, 330 insertions, 0 deletions
diff --git a/fluent-bit/tests/internal/regex.c b/fluent-bit/tests/internal/regex.c
new file mode 100644
index 000000000..77045a38c
--- /dev/null
+++ b/fluent-bit/tests/internal/regex.c
@@ -0,0 +1,330 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+/* Fluent Bit
+ * ==========
+ * Copyright (C) 2019-2022 The Fluent Bit Authors
+ * Copyright (C) 2015-2018 Treasure Data Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <string.h>
+#include <fluent-bit/flb_regex.h>
+#include "flb_tests_internal.h"
+
+struct kv_list {
+ int index;
+ size_t size;
+ char **lists;
+};
+static void cb_kv(const char *name, const char *value,
+ size_t vlen, void *data)
+{
+ size_t len;
+ struct kv_list *l = (struct kv_list*)data;
+
+ if (!TEST_CHECK(name != NULL && value != NULL && data != NULL)) {
+ TEST_MSG("input is NULL");
+ return;
+ }
+ len = strlen(name);
+
+ if (!TEST_CHECK(len == strlen(l->lists[l->index * 2]))) {
+ TEST_MSG("name: lenght error. got:%zu expect:%zu", len, strlen(l->lists[l->index * 2]));
+ TEST_MSG("name: got:%s expect:%s", name, l->lists[l->index * 2]);
+ goto cb_kv_end;
+ }
+ if (!TEST_CHECK(strncmp(name, l->lists[l->index * 2], len) == 0)) {
+ TEST_MSG("name: mismatch. got:%s expect:%s", name, l->lists[l->index * 2]);
+ goto cb_kv_end;
+ }
+
+ if (!TEST_CHECK(vlen == strlen(l->lists[l->index * 2+1]))) {
+ TEST_MSG("value: lenght error. got:%zu expect:%zu", vlen, strlen(l->lists[l->index * 2+1]));
+ TEST_MSG("value: got:%s expect:%s", value, l->lists[l->index * 2+1]);
+ goto cb_kv_end;
+ }
+ if (!TEST_CHECK(strncmp(value, l->lists[l->index * 2+1], vlen) == 0)) {
+ TEST_MSG("value: mismatch. got:%s expect:%s", value, l->lists[l->index * 2+1]);
+ goto cb_kv_end;
+ }
+ cb_kv_end:
+ l->index++;
+}
+
+static void test_basic()
+{
+ struct flb_regex *regex = NULL;
+ struct flb_regex_search result;
+ int ret;
+ ssize_t size;
+ const char *input = "string 1234 2022/10/24";
+
+ char *expected_strs[] = {"str","string", "num","1234", "time","2022/10/24"};
+ struct kv_list expected = {
+ .index = 0,
+ .size = sizeof(expected_strs)/sizeof(char *),
+ .lists = &expected_strs[0],
+ };
+
+ regex = flb_regex_create("/(?<str>[a-z]+) (?<num>\\d+) (?<time>\\d{4}/\\d{2}/\\d{2})/");
+ if (!TEST_CHECK(regex != NULL)) {
+ TEST_MSG("flb_regex_create failed");
+ exit(1);
+ }
+
+ size = flb_regex_do(regex, input, strlen(input), &result);
+ if (!TEST_CHECK(size >= 0)) {
+ TEST_MSG("flb_regex_do failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_parse(regex, &result, cb_kv, &expected);
+ if (!TEST_CHECK(ret != -1)) {
+ TEST_MSG("flb_regex_parse failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_destroy(regex);
+ if (!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_regex_destroy failed");
+ exit(1);
+ }
+}
+
+/* test uri to check if misunderstanding "/pattern/option" */
+static void test_uri()
+{
+ struct flb_regex *regex = NULL;
+ struct flb_regex_search result;
+ int ret;
+ ssize_t size;
+ const char *input = "/uri/is/hoge";
+
+ char *expected_strs[] = {"middle","is"};
+ struct kv_list expected = {
+ .index = 0,
+ .size = sizeof(expected_strs)/sizeof(char *),
+ .lists = &expected_strs[0],
+ };
+
+ regex = flb_regex_create("/uri/(?<middle>[a-z]+)/hoge");
+ if (!TEST_CHECK(regex != NULL)) {
+ TEST_MSG("flb_regex_create failed");
+ exit(1);
+ }
+
+ size = flb_regex_do(regex, input, strlen(input), &result);
+ if (!TEST_CHECK(size >= 0)) {
+ TEST_MSG("flb_regex_do failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_parse(regex, &result, cb_kv, &expected);
+ if (!TEST_CHECK(ret != -1)) {
+ TEST_MSG("flb_regex_parse failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_destroy(regex);
+ if (!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_regex_destroy failed");
+ exit(1);
+ }
+}
+
+/* test "/pattern/i" */
+static void test_option_ignore_case()
+{
+ struct flb_regex *regex = NULL;
+ struct flb_regex_search result;
+ int ret;
+ ssize_t size;
+ const char *input = "STRING";
+
+ char *expected_strs[] = {"str","STRING"};
+ struct kv_list expected = {
+ .index = 0,
+ .size = sizeof(expected_strs)/sizeof(char *),
+ .lists = &expected_strs[0],
+ };
+
+ regex = flb_regex_create("/(?<str>[a-z]+)/i");
+ if (!TEST_CHECK(regex != NULL)) {
+ TEST_MSG("flb_regex_create failed");
+ exit(1);
+ }
+
+ size = flb_regex_do(regex, input, strlen(input), &result);
+ if (!TEST_CHECK(size >= 0)) {
+ TEST_MSG("flb_regex_do failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_parse(regex, &result, cb_kv, &expected);
+ if (!TEST_CHECK(ret != -1)) {
+ TEST_MSG("flb_regex_parse failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_destroy(regex);
+ if (!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_regex_destroy failed");
+ exit(1);
+ }
+}
+
+/* test "/pattern/m" */
+static void test_option_multiline()
+{
+ struct flb_regex *regex = NULL;
+ struct flb_regex_search result;
+ int ret;
+ ssize_t size;
+ const char *input = "string\n1234\nstring";
+
+ char *expected_strs[] = {"full_str","string\n1234\nstring"};
+ struct kv_list expected = {
+ .index = 0,
+ .size = sizeof(expected_strs)/sizeof(char *),
+ .lists = &expected_strs[0],
+ };
+
+ regex = flb_regex_create("/(?<full_str>.+)/m");
+ if (!TEST_CHECK(regex != NULL)) {
+ TEST_MSG("flb_regex_create failed");
+ exit(1);
+ }
+
+ size = flb_regex_do(regex, input, strlen(input), &result);
+ if (!TEST_CHECK(size >= 0)) {
+ TEST_MSG("flb_regex_do failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_parse(regex, &result, cb_kv, &expected);
+ if (!TEST_CHECK(ret != -1)) {
+ TEST_MSG("flb_regex_parse failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_destroy(regex);
+ if (!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_regex_destroy failed");
+ exit(1);
+ }
+}
+
+/* test "/pattern/x" */
+static void test_option_extend()
+{
+ struct flb_regex *regex = NULL;
+ struct flb_regex_search result;
+ int ret;
+ ssize_t size;
+ const char *input = "3.14";
+
+ char *expected_strs[] = {"pi","3.14"};
+ struct kv_list expected = {
+ .index = 0,
+ .size = sizeof(expected_strs)/sizeof(char *),
+ .lists = &expected_strs[0],
+ };
+
+ regex = flb_regex_create("/(?<pi>\\d \\. 14)/x");
+ if (!TEST_CHECK(regex != NULL)) {
+ TEST_MSG("flb_regex_create failed");
+ exit(1);
+ }
+
+ size = flb_regex_do(regex, input, strlen(input), &result);
+ if (!TEST_CHECK(size >= 0)) {
+ TEST_MSG("flb_regex_do failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_parse(regex, &result, cb_kv, &expected);
+ if (!TEST_CHECK(ret != -1)) {
+ TEST_MSG("flb_regex_parse failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_destroy(regex);
+ if (!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_regex_destroy failed");
+ exit(1);
+ }
+}
+
+/* test "/pattern/ix" */
+static void test_option_i_x()
+{
+ struct flb_regex *regex = NULL;
+ struct flb_regex_search result;
+ int ret;
+ ssize_t size;
+ const char *input = "3.14pi";
+
+ char *expected_strs[] = {"full_str","3.14pi"};
+ struct kv_list expected = {
+ .index = 0,
+ .size = sizeof(expected_strs)/sizeof(char *),
+ .lists = &expected_strs[0],
+ };
+
+ regex = flb_regex_create("/(?<full_str>\\d \\. 14PI)/ix");
+ if (!TEST_CHECK(regex != NULL)) {
+ TEST_MSG("flb_regex_create failed");
+ exit(1);
+ }
+
+ size = flb_regex_do(regex, input, strlen(input), &result);
+ if (!TEST_CHECK(size >= 0)) {
+ TEST_MSG("flb_regex_do failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_parse(regex, &result, cb_kv, &expected);
+ if (!TEST_CHECK(ret != -1)) {
+ TEST_MSG("flb_regex_parse failed");
+ flb_regex_destroy(regex);
+ exit(1);
+ }
+
+ ret = flb_regex_destroy(regex);
+ if (!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_regex_destroy failed");
+ exit(1);
+ }
+}
+
+TEST_LIST = {
+ { "basic" , test_basic},
+ { "uri" , test_uri},
+ { "option_ignore_case", test_option_ignore_case},
+ { "option_multiline" , test_option_multiline},
+ { "option_extend" , test_option_extend},
+ { "option_i_x" , test_option_i_x},
+ { 0 }
+};
generated by cgit v1.2.3 (git 2.39.1) at 2025-01-20 13:17:00 +0000