tests: internal: upstream_tls: add test for tls handling

Signed-off-by: Eduardo Silva <[email protected]>

Author: edsiper

tests/internal/CMakeLists.txt

@@ -54,6 +54,14 @@ set(UNIT_TESTS_FILES
   opentelemetry.c
 )
 
+# TLS helpers
+if(FLB_TLS)
+  set(UNIT_TESTS_FILES
+    ${UNIT_TESTS_FILES}
+    upstream_tls.c
+    )
+endif()
+
 # Config format
 set(UNIT_TESTS_FILES
     ${UNIT_TESTS_FILES}

tests/internal/upstream_tls.c

@@ -0,0 +1,104 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+#include <fluent-bit/flb_info.h>
+#include <fluent-bit/flb_upstream.h>
+#include <fluent-bit/flb_upstream_conn.h>
+#include <fluent-bit/flb_connection.h>
+#include <fluent-bit/flb_pipe.h>
+#include <fluent-bit/flb_socket.h>
+#include <fluent-bit/tls/flb_tls.h>
+
+#include "flb_tests_internal.h"
+
+#ifdef FLB_HAVE_TLS
+
+#ifdef FLB_SYSTEM_WINDOWS
+#include <fluent-bit/flb_compat.h>
+#endif
+
+struct test_backend_ctx {
+    int invalidate_calls;
+};
+
+static void test_session_invalidate(void *session)
+{
+    struct test_backend_ctx *ctx = session;
+
+    if (ctx != NULL) {
+        ctx->invalidate_calls++;
+    }
+}
+
+void test_prepare_destroy_conn_marks_tls_session_stale(void)
+{
+    struct test_backend_ctx backend_session = {0};
+    struct flb_tls_backend backend_api = {0};
+    struct flb_tls tls_context = {0};
+    struct flb_tls_session tls_session = {0};
+    struct flb_connection conn = {0};
+    struct flb_upstream upstream = {0};
+    struct flb_config config = {0};
+    struct flb_upstream_queue *queue;
+    flb_pipefd_t socket_pair[2];
+    int ret;
+
+#ifdef FLB_SYSTEM_WINDOWS
+    WSADATA wsa_data;
+
+    WSAStartup(0x0201, &wsa_data);
+#endif
+
+    ret = flb_pipe_create(socket_pair);
+    TEST_CHECK(ret == 0);
+
+    backend_api.session_invalidate = test_session_invalidate;
+    tls_context.api = &backend_api;
+
+    tls_session.ptr = &backend_session;
+    tls_session.tls = &tls_context;
+    tls_session.connection = &conn;
+
+    config.is_shutting_down = FLB_FALSE;
+    upstream.base.config = &config;
+    upstream.base.net.keepalive = FLB_FALSE;
+    upstream.tcp_host = "example";
+    upstream.tcp_port = 443;
+
+    flb_upstream_queue_init(&upstream.queue);
+
+    conn.fd = socket_pair[0];
+    conn.event.fd = conn.fd;
+    conn.event.status = 0;
+    conn.stream = (struct flb_stream *) &upstream;
+    conn.net = &upstream.base.net;
+    conn.tls_session = &tls_session;
+    conn.net_error = 0;
+
+    mk_list_init(&conn._head);
+    queue = &upstream.queue;
+    mk_list_add(&conn._head, &queue->busy_queue);
+
+    ret = flb_upstream_conn_release(&conn);
+    TEST_CHECK(ret == 0);
+
+    TEST_CHECK(backend_session.invalidate_calls == 1);
+    TEST_CHECK(conn.fd == -1);
+    TEST_CHECK(conn.event.fd == -1);
+    TEST_CHECK(mk_list_size(&queue->destroy_queue) == 1);
+    TEST_CHECK(conn.shutdown_flag == FLB_TRUE);
+
+    flb_pipe_close(socket_pair[1]);
+
+#ifdef FLB_SYSTEM_WINDOWS
+    WSACleanup();
+#endif
+}
+
+#endif
+
+TEST_LIST = {
+#ifdef FLB_HAVE_TLS
+    {"prepare_destroy_conn_marks_tls_session_stale", test_prepare_destroy_conn_marks_tls_session_stale},
+#endif
+    {0}
+};