tests: internal: aws_util: remove timezone dependency

Signed-off-by: Takahiro Yamashita <[email protected]>

Author: nokute78

tests/internal/aws_util.c

@@ -3,6 +3,7 @@
 #include <fluent-bit/flb_sds.h>
 #include <fluent-bit/flb_aws_util.h>
 #include <fluent-bit/flb_mem.h>
+#include <fluent-bit/flb_pthread.h>
 
 #include "flb_tests_internal.h"
 
@@ -47,6 +48,60 @@
 #define VALID_SEQ_INDEX 0
 
 
+pthread_mutex_t env_mutex = PTHREAD_MUTEX_INITIALIZER;
+static int mktime_utc(struct tm *day, time_t *tm)
+{
+    int ret;
+    char *tzvar = NULL;
+    char orig_tz[256] = {0};
+    time_t t;
+
+    if (!TEST_CHECK(day != NULL)) {
+        TEST_MSG("struct tm is null");
+        return -1;
+    }
+    if (!TEST_CHECK(tm != NULL)) {
+        TEST_MSG("time_t is null");
+        return -1;
+    }
+
+    pthread_mutex_lock(&env_mutex);
+
+    /* save current TZ var */
+    tzvar = getenv("TZ");
+    if (tzvar != NULL) {
+        if (!TEST_CHECK(strlen(tzvar) <= sizeof(orig_tz))) {
+            TEST_MSG("TZ is large. len=%ld TZ=%s", strlen(tzvar), tzvar);
+            pthread_mutex_unlock(&env_mutex);
+            return -1;
+        }
+        strncpy(&orig_tz[0], tzvar, sizeof(orig_tz));
+    }
+
+    /* setenv is not thread safe */
+    ret = setenv("TZ", "UTC", 1);
+    if (!TEST_CHECK(ret == 0)) {
+        TEST_MSG("setenv failed");
+        pthread_mutex_unlock(&env_mutex);
+        return -1;
+    }
+
+    t = mktime(day);
+    *tm = t;
+
+    /* restore TZ */
+    if (tzvar != NULL) {
+        ret = setenv("TZ", &orig_tz[0], 1);
+    }
+    else {
+        ret = unsetenv("TZ");
+    }
+
+    pthread_mutex_unlock(&env_mutex);
+
+    return ret;
+}
+
 static void test_flb_aws_error()
 {
     flb_sds_t error_type;
@@ -92,7 +147,9 @@ static void test_flb_get_s3_key_multi_tag_exists()
 {
     flb_sds_t s3_key_format = NULL;
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
+    time_t t;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_TAG_PART, t, TAG, TAG_DELIMITER, 0);
     TEST_CHECK(strcmp(s3_key_format, S3_OBJECT_KEY_TAG_PART) == 0);
 
@@ -103,7 +160,9 @@ static void test_flb_get_s3_key_full_tag()
 {
     flb_sds_t s3_key_format = NULL;
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
+    time_t t;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_FULL_TAG, t, TAG, TAG_DELIMITER, 0);
     TEST_CHECK(strcmp(s3_key_format, S3_OBJECT_KEY_FULL_TAG) == 0);
 
@@ -114,7 +173,9 @@ static void test_flb_get_s3_key_tag_special_characters()
 {
     flb_sds_t s3_key_format = NULL;
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
+    time_t t;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_SPECIAL_CHARCATERS_TAG, t, TAG,
                                    TAG_DELIMITER, 0);
     TEST_CHECK(strcmp(s3_key_format, S3_OBJECT_KEY_SPECIAL_CHARCATERS_TAG) == 0);
@@ -126,7 +187,9 @@ static void test_flb_get_s3_key_multi_tag_delimiter()
 {
     flb_sds_t s3_key_format = NULL;
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
+    time_t t;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_TAG_PART, t, MULTI_DELIMITER_TAG,
                                    TAG_DELIMITERS, 0);
     TEST_CHECK(strcmp(s3_key_format, S3_OBJECT_KEY_TAG_PART) == 0);
@@ -138,7 +201,9 @@ static void test_flb_get_s3_key_invalid_tag_delimiter()
 {
     flb_sds_t s3_key_format = NULL;
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
+    time_t t;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_TAG_PART, t, MULTI_DELIMITER_TAG,
                                    INVALID_TAG_DELIMITERS, 0);
     TEST_CHECK(strcmp(s3_key_format, S3_OBJECT_KEY_INVALID_DELIMITER)  == 0);
@@ -150,7 +215,9 @@ static void test_flb_get_s3_key_invalid_tag_index()
 {
     flb_sds_t s3_key_format = NULL;
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
+    time_t t;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_INVALID_TAG, t, TAG, TAG_DELIMITER, 0);
     TEST_CHECK(strcmp(s3_key_format, S3_OBJECY_KEY_INVALID_TAG) == 0);
 
@@ -169,7 +236,9 @@ static void test_flb_get_s3_key_invalid_key_length()
     }
     snprintf(buf, sizeof(buf), "%s%s", S3_KEY_FORMAT_SPECIAL_CHARCATERS_TAG, tmp);
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
+    time_t t;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(buf, t, TAG, TAG_DELIMITER, 0);
     TEST_CHECK(strlen(s3_key_format) <= 1024);
 
@@ -180,7 +249,9 @@ static void test_flb_get_s3_key_static_string()
 {
     flb_sds_t s3_key_format = NULL;
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
+    time_t t;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_STATIC_STRING, t, NO_TAG,
                                    TAG_DELIMITER, 0);
     TEST_CHECK(strcmp(s3_key_format, S3_KEY_FORMAT_STATIC_STRING) == 0);
@@ -192,7 +263,9 @@ static void test_flb_get_s3_key_valid_index()
 {
     flb_sds_t s3_key_format = NULL;
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
+    time_t t;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_VALID_INDEX, t, NO_TAG,
                                    TAG_DELIMITER, 12);
     TEST_CHECK(strcmp(s3_key_format, S3_OBJECT_KEY_VALID_INDEX) == 0);
@@ -203,9 +276,10 @@ static void test_flb_get_s3_key_valid_index()
 static void test_flb_get_s3_key_increment_index()
 {
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
-
+    time_t t;
     flb_sds_t s3_key_format = NULL;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_VALID_INDEX, t, NO_TAG,
                                     TAG_DELIMITER, 5);
 
@@ -225,9 +299,10 @@ static void test_flb_get_s3_key_index_overflow()
 {
     flb_sds_t s3_key_format = NULL;
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
-
+    time_t t;
     uint64_t index = 18446744073709551615U;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_VALID_INDEX, t, NO_TAG,
                                    TAG_DELIMITER, index);
     TEST_CHECK(strcmp(s3_key_format, S3_OBJECT_KEY_PRE_OVERFLOW_INDEX) == 0);
@@ -245,7 +320,9 @@ static void test_flb_get_s3_key_mixed_timestamp()
 {
     flb_sds_t s3_key_format = NULL;
     struct tm day = { 0, 0, 0, 15, 7, 120};
-    time_t t = mktime(&day);
+    time_t t;
+
+    mktime_utc(&day, &t);
     s3_key_format = flb_get_s3_key(S3_KEY_FORMAT_MIXED_TIMESTAMP, t, NO_TAG,
                                    TAG_DELIMITER, 12);
     TEST_CHECK(strcmp(s3_key_format, S3_OBJECT_KEY_MIXED_TIMESTAMP) == 0);