summaryrefslogtreecommitdiffstats
path: root/fluent-bit/tests/internal/flb_time.c
diff options
context:
space:
mode:
Diffstat (limited to 'fluent-bit/tests/internal/flb_time.c')
-rw-r--r--fluent-bit/tests/internal/flb_time.c335
1 files changed, 335 insertions, 0 deletions
diff --git a/fluent-bit/tests/internal/flb_time.c b/fluent-bit/tests/internal/flb_time.c
new file mode 100644
index 00000000..a4e2980b
--- /dev/null
+++ b/fluent-bit/tests/internal/flb_time.c
@@ -0,0 +1,335 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+/* Fluent Bit
+ * ==========
+ * Copyright (C) 2019-2021 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 <fluent-bit.h>
+#include <fluent-bit/flb_time.h>
+#include <fluent-bit/flb_pack.h>
+#include <mpack/mpack.h>
+#include <msgpack.h>
+#include <msgpack/timestamp.h>
+#include "flb_tests_internal.h"
+
+#define SEC_32BIT 1647061992 /* 0x622c2be8 */
+#define NSEC_32BIT 123000000 /* 123ms 0x0754d4c0 */
+#define D_SEC 1647061992.123;
+const char eventtime[8] = {0x62, 0x2c, 0x2b, 0xe8, 0x07, 0x54, 0xd4, 0xc0 };
+
+void test_to_nanosec()
+{
+ uint64_t expect = 123000000456;
+ uint64_t ret;
+ struct flb_time tm;
+
+ flb_time_set(&tm, 123, 456);
+
+ ret = flb_time_to_nanosec(&tm);
+ if (!TEST_CHECK(ret == expect)) {
+ TEST_MSG("given =%" PRIu64, ret);
+ TEST_MSG("expect =%" PRIu64, expect);
+ }
+}
+
+/* https://github.com/fluent/fluent-bit/issues/5215 */
+void test_append_to_mpack_v1() {
+ mpack_writer_t writer;
+ char *data;
+ size_t size;
+ struct flb_time tm;
+ int ret;
+
+ msgpack_zone mempool;
+ msgpack_object ret_obj;
+ size_t off = 0;
+
+ flb_time_set(&tm, 123, 456);
+ mpack_writer_init_growable(&writer, &data, &size);
+
+ ret = flb_time_append_to_mpack(&writer, &tm, FLB_TIME_ETFMT_V1_FIXEXT);
+ if (!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_time_append_to_mpack failed");
+ mpack_writer_destroy(&writer);
+ flb_free(data);
+ exit(EXIT_FAILURE);
+ }
+ mpack_writer_destroy(&writer);
+
+ msgpack_zone_init(&mempool, 1024);
+ ret = msgpack_unpack(data, size, &off, &mempool, &ret_obj);
+ if (!TEST_CHECK(ret == MSGPACK_UNPACK_SUCCESS)) {
+ TEST_MSG("unpack failed ret = %d", ret);
+ msgpack_zone_destroy(&mempool);
+ flb_free(data);
+ exit(EXIT_FAILURE);
+ }
+
+ if (!TEST_CHECK(ret_obj.type == MSGPACK_OBJECT_EXT)) {
+ TEST_MSG("data type is not ext. type=%d", ret_obj.type);
+ msgpack_zone_destroy(&mempool);
+ flb_free(data);
+ exit(EXIT_FAILURE);
+ }
+ if (!TEST_CHECK(ret_obj.via.ext.type == 0)) {
+ TEST_MSG("ext type is not 0. ext type=%d", ret_obj.via.ext.type);
+ msgpack_zone_destroy(&mempool);
+ flb_free(data);
+ exit(EXIT_FAILURE);
+ }
+ msgpack_zone_destroy(&mempool);
+ flb_free(data);
+}
+
+void test_msgpack_to_time_int()
+{
+ struct flb_time tm;
+ int64_t expect = SEC_32BIT;
+ int ret;
+
+ msgpack_packer mp_pck;
+ msgpack_sbuffer mp_sbuf;
+ msgpack_unpacked result;
+
+ msgpack_object tm_obj;
+
+ /* create int object*/
+ msgpack_sbuffer_init(&mp_sbuf);
+ msgpack_packer_init(&mp_pck, &mp_sbuf, msgpack_sbuffer_write);
+ msgpack_pack_int(&mp_pck, expect);
+
+ msgpack_unpacked_init(&result);
+ msgpack_unpack_next(&result, mp_sbuf.data, mp_sbuf.size, NULL);
+
+ tm_obj = result.data;
+ ret = flb_time_msgpack_to_time(&tm, &tm_obj);
+ if(!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_time_msgpack_to_time failed");
+ exit(EXIT_FAILURE);
+ }
+
+ if (!TEST_CHECK(tm.tm.tv_sec == expect && tm.tm.tv_nsec == 0)) {
+ TEST_MSG("got %ld.%ld, expect %ld.%d", tm.tm.tv_sec, tm.tm.tv_nsec, expect, 0);
+ }
+
+ msgpack_sbuffer_destroy(&mp_sbuf);
+ msgpack_unpacked_destroy(&result);
+}
+
+void test_msgpack_to_time_double()
+{
+ struct flb_time tm;
+ double d_time = D_SEC;
+ int64_t expect_sec = SEC_32BIT;
+ int64_t expect_nsec = NSEC_32BIT;
+
+ int ret;
+
+ msgpack_packer mp_pck;
+ msgpack_sbuffer mp_sbuf;
+ msgpack_unpacked result;
+
+ msgpack_object tm_obj;
+
+ /* create int object*/
+ msgpack_sbuffer_init(&mp_sbuf);
+ msgpack_packer_init(&mp_pck, &mp_sbuf, msgpack_sbuffer_write);
+ msgpack_pack_double(&mp_pck, d_time);
+
+ msgpack_unpacked_init(&result);
+ msgpack_unpack_next(&result, mp_sbuf.data, mp_sbuf.size, NULL);
+
+ tm_obj = result.data;
+ ret = flb_time_msgpack_to_time(&tm, &tm_obj);
+ if(!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_time_msgpack_to_time failed");
+ exit(EXIT_FAILURE);
+ }
+
+ if (!TEST_CHECK(tm.tm.tv_sec == expect_sec &&
+ llabs(tm.tm.tv_nsec - expect_nsec ) < 10000 /* 10us*/)) {
+ TEST_MSG("got %ld.%ld, expect %ld.%ld", tm.tm.tv_sec, tm.tm.tv_nsec, expect_sec, expect_nsec);
+ }
+
+ msgpack_sbuffer_destroy(&mp_sbuf);
+ msgpack_unpacked_destroy(&result);
+}
+
+void test_msgpack_to_time_eventtime()
+{
+ struct flb_time tm;
+ int64_t expect_sec = SEC_32BIT;
+ int64_t expect_nsec = NSEC_32BIT;
+ char ext_data[8] = {0};
+ int ret;
+
+ msgpack_packer mp_pck;
+ msgpack_sbuffer mp_sbuf;
+ msgpack_unpacked result;
+
+ msgpack_object tm_obj;
+
+ memcpy(&ext_data[0], &eventtime[0], 8);
+
+ /* create int object*/
+ msgpack_sbuffer_init(&mp_sbuf);
+ msgpack_packer_init(&mp_pck, &mp_sbuf, msgpack_sbuffer_write);
+
+ /* https://github.com/fluent/fluentd/wiki/Forward-Protocol-Specification-v1#eventtime-ext-format */
+ msgpack_pack_ext(&mp_pck, 8/*fixext8*/, 0);
+ msgpack_pack_ext_body(&mp_pck, ext_data, sizeof(ext_data));
+
+ msgpack_unpacked_init(&result);
+ msgpack_unpack_next(&result, mp_sbuf.data, mp_sbuf.size, NULL);
+
+ tm_obj = result.data;
+ ret = flb_time_msgpack_to_time(&tm, &tm_obj);
+ if(!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_time_msgpack_to_time failed");
+ exit(EXIT_FAILURE);
+ }
+
+ if (!TEST_CHECK(tm.tm.tv_sec == expect_sec &&
+ llabs(tm.tm.tv_nsec - expect_nsec ) < 10000 /* 10us*/)) {
+ TEST_MSG("got %ld.%ld, expect %ld.%ld", tm.tm.tv_sec, tm.tm.tv_nsec, expect_sec, expect_nsec);
+ }
+
+ msgpack_sbuffer_destroy(&mp_sbuf);
+ msgpack_unpacked_destroy(&result);
+}
+
+void test_msgpack_to_time_invalid()
+{
+ struct flb_time tm;
+ char ext_data[8] = {0x00, 0x11, 0x22, 0xaa, 0xbb, 0xcc, 0xdd, 0xee};
+ int ret;
+
+ msgpack_packer mp_pck;
+ msgpack_sbuffer mp_sbuf;
+ msgpack_unpacked result;
+
+
+ msgpack_object tm_obj;
+
+ /* create int object*/
+ msgpack_sbuffer_init(&mp_sbuf);
+ msgpack_packer_init(&mp_pck, &mp_sbuf, msgpack_sbuffer_write);
+
+ msgpack_pack_ext(&mp_pck, 5 /* invalid size */, 0);
+ msgpack_pack_ext_body(&mp_pck, ext_data, 5);
+
+ msgpack_unpacked_init(&result);
+ msgpack_unpack_next(&result, mp_sbuf.data, mp_sbuf.size, NULL);
+
+ tm_obj = result.data;
+
+ /* Check if ext */
+ TEST_CHECK(tm_obj.type == MSGPACK_OBJECT_EXT);
+ TEST_CHECK(tm_obj.via.ext.type == 0);
+ TEST_CHECK(tm_obj.via.ext.size == 5);
+
+ ret = flb_time_msgpack_to_time(&tm, &tm_obj);
+ if(!TEST_CHECK(ret != 0)) {
+ TEST_MSG("flb_time_msgpack_to_time should fail");
+ exit(EXIT_FAILURE);
+ }
+
+ msgpack_sbuffer_destroy(&mp_sbuf);
+ msgpack_unpacked_destroy(&result);
+
+
+ /* create int object*/
+ msgpack_sbuffer_init(&mp_sbuf);
+ msgpack_packer_init(&mp_pck, &mp_sbuf, msgpack_sbuffer_write);
+
+ msgpack_pack_ext(&mp_pck, 8, 10 /* invalid type */);
+ msgpack_pack_ext_body(&mp_pck, ext_data, 8);
+
+ msgpack_unpacked_init(&result);
+ msgpack_unpack_next(&result, mp_sbuf.data, mp_sbuf.size, NULL);
+
+ tm_obj = result.data;
+
+ /* Check if ext */
+ TEST_CHECK(tm_obj.type == MSGPACK_OBJECT_EXT);
+ TEST_CHECK(tm_obj.via.ext.type == 10);
+ TEST_CHECK(tm_obj.via.ext.size == 8);
+
+ ret = flb_time_msgpack_to_time(&tm, &tm_obj);
+ if(!TEST_CHECK(ret != 0)) {
+ TEST_MSG("flb_time_msgpack_to_time should fail");
+ exit(EXIT_FAILURE);
+ }
+
+ msgpack_sbuffer_destroy(&mp_sbuf);
+ msgpack_unpacked_destroy(&result);
+}
+
+void test_append_to_msgpack_eventtime()
+{
+ struct flb_time tm;
+ int ret;
+ char expect_data[8] = {0};
+
+ msgpack_packer mp_pck;
+ msgpack_sbuffer mp_sbuf;
+ msgpack_unpacked result;
+
+ msgpack_object tm_obj;
+
+ memcpy(&expect_data[0], &eventtime[0], 8);
+
+ tm.tm.tv_sec = SEC_32BIT;
+ tm.tm.tv_nsec = NSEC_32BIT;
+
+ /* create int object*/
+ msgpack_sbuffer_init(&mp_sbuf);
+ msgpack_packer_init(&mp_pck, &mp_sbuf, msgpack_sbuffer_write);
+
+ ret = flb_time_append_to_msgpack(&tm, &mp_pck, FLB_TIME_ETFMT_V1_FIXEXT);
+ if(!TEST_CHECK(ret == 0)) {
+ TEST_MSG("flb_time_append_to_msgpack failed");
+ exit(EXIT_FAILURE);
+ }
+ msgpack_unpacked_init(&result);
+ msgpack_unpack_next(&result, mp_sbuf.data, mp_sbuf.size, NULL);
+
+ tm_obj = result.data;
+
+ /* Check if Eventtime */
+ TEST_CHECK(tm_obj.type == MSGPACK_OBJECT_EXT);
+ TEST_CHECK(tm_obj.via.ext.type == 0);
+ TEST_CHECK(tm_obj.via.ext.size == 8);
+
+ if (!TEST_CHECK(memcmp(&expect_data[0], tm_obj.via.ext.ptr, 8) == 0) ) {
+ TEST_MSG("got 0x%x, expect 0x%x", *(uint32_t*)tm_obj.via.ext.ptr, *((uint32_t*)&expect_data[0]));
+ }
+
+ msgpack_sbuffer_destroy(&mp_sbuf);
+ msgpack_unpacked_destroy(&result);
+}
+
+TEST_LIST = {
+ { "flb_time_to_nanosec" , test_to_nanosec},
+ { "flb_time_append_to_mpack_v1" , test_append_to_mpack_v1},
+ { "msgpack_to_time_int" , test_msgpack_to_time_int},
+ { "msgpack_to_time_double" , test_msgpack_to_time_double},
+ { "msgpack_to_time_eventtime" , test_msgpack_to_time_eventtime},
+ { "msgpack_to_time_invalid" , test_msgpack_to_time_invalid},
+ { "append_to_msgpack_eventtime" , test_append_to_msgpack_eventtime},
+ { NULL, NULL }
+};
generated by cgit v1.2.3 (git 2.39.1) at 2024-09-29 18:10:12 +0000