Merging upstream version 1.45.3+dfsg.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 143 insertions, 0 deletions

diff --git a/src/collectors/systemd-journal.plugin/systemd-main.c b/src/collectors/systemd-journal.plugin/systemd-main.c
new file mode 100644
index 000000000..89052b9f9
--- /dev/null
+++ b/src/collectors/systemd-journal.plugin/systemd-main.c
@@ -0,0 +1,143 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#include "systemd-internals.h"
+#include "libnetdata/required_dummies.h"
+
+#define SYSTEMD_JOURNAL_WORKER_THREADS          5
+
+netdata_mutex_t stdout_mutex = NETDATA_MUTEX_INITIALIZER;
+static bool plugin_should_exit = false;
+
+static bool journal_data_directories_exist() {
+    struct stat st;
+    for (unsigned i = 0; i < MAX_JOURNAL_DIRECTORIES && journal_directories[i].path; i++) {
+        if ((stat(string2str(journal_directories[i].path), &st) == 0) && S_ISDIR(st.st_mode))
+            return true;
+    }
+    return false;
+}
+
+int main(int argc __maybe_unused, char **argv __maybe_unused) {
+    clocks_init();
+    netdata_thread_set_tag("sd-jrnl.plugin");
+    nd_log_initialize_for_external_plugins("systemd-journal.plugin");
+
+    netdata_configured_host_prefix = getenv("NETDATA_HOST_PREFIX");
+    if(verify_netdata_host_prefix(true) == -1) exit(1);
+
+    // ------------------------------------------------------------------------
+    // initialization
+
+    netdata_systemd_journal_annotations_init();
+    journal_init_files_and_directories();
+
+    if (!journal_data_directories_exist()) {
+        nd_log_collector(NDLP_INFO, "unable to locate journal data directories. Exiting...");
+        fprintf(stdout, "DISABLE\n");
+        fflush(stdout);
+        exit(0);
+    }
+
+    // ------------------------------------------------------------------------
+    // debug
+
+    if(argc == 2 && strcmp(argv[1], "debug") == 0) {
+        journal_files_registry_update();
+
+        bool cancelled = false;
+        usec_t stop_monotonic_ut = now_monotonic_usec() + 600 * USEC_PER_SEC;
+        char buf[] = "systemd-journal after:-8640000 before:0 direction:backward last:200 data_only:false slice:true source:all";
+        // char buf[] = "systemd-journal after:1695332964 before:1695937764 direction:backward last:100 slice:true source:all DHKucpqUoe1:PtVoyIuX.MU";
+        // char buf[] = "systemd-journal after:1694511062 before:1694514662 anchor:1694514122024403";
+        function_systemd_journal("123", buf, &stop_monotonic_ut, &cancelled,
+                                 NULL, HTTP_ACCESS_ALL, NULL, NULL);
+//        function_systemd_units("123", "systemd-units", 600, &cancelled);
+        exit(1);
+    }
+#ifdef ENABLE_SYSTEMD_DBUS
+    if(argc == 2 && strcmp(argv[1], "debug-units") == 0) {
+        bool cancelled = false;
+        usec_t stop_monotonic_ut = now_monotonic_usec() + 600 * USEC_PER_SEC;
+        function_systemd_units("123", "systemd-units", &stop_monotonic_ut, &cancelled,
+                               NULL, HTTP_ACCESS_ALL, NULL, NULL);
+        exit(1);
+    }
+#endif
+
+    // ------------------------------------------------------------------------
+    // watcher thread
+
+    netdata_thread_t watcher_thread;
+    netdata_thread_create(&watcher_thread, "SDWATCH",
+                          NETDATA_THREAD_OPTION_DONT_LOG, journal_watcher_main, NULL);
+
+    // ------------------------------------------------------------------------
+    // the event loop for functions
+
+    struct functions_evloop_globals *wg =
+            functions_evloop_init(SYSTEMD_JOURNAL_WORKER_THREADS, "SDJ", &stdout_mutex, &plugin_should_exit);
+
+    functions_evloop_add_function(wg,
+                                  SYSTEMD_JOURNAL_FUNCTION_NAME,
+                                  function_systemd_journal,
+                                  SYSTEMD_JOURNAL_DEFAULT_TIMEOUT,
+                                  NULL);
+
+#ifdef ENABLE_SYSTEMD_DBUS
+    functions_evloop_add_function(wg,
+                                  SYSTEMD_UNITS_FUNCTION_NAME,
+                                  function_systemd_units,
+                                  SYSTEMD_UNITS_DEFAULT_TIMEOUT,
+                                  NULL);
+#endif
+
+    systemd_journal_dyncfg_init(wg);
+
+    // ------------------------------------------------------------------------
+    // register functions to netdata
+
+    netdata_mutex_lock(&stdout_mutex);
+
+    fprintf(stdout, PLUGINSD_KEYWORD_FUNCTION " GLOBAL \"%s\" %d \"%s\" \"logs\" "HTTP_ACCESS_FORMAT" %d\n",
+            SYSTEMD_JOURNAL_FUNCTION_NAME, SYSTEMD_JOURNAL_DEFAULT_TIMEOUT, SYSTEMD_JOURNAL_FUNCTION_DESCRIPTION,
+            (HTTP_ACCESS_FORMAT_CAST)(HTTP_ACCESS_SIGNED_ID | HTTP_ACCESS_SAME_SPACE | HTTP_ACCESS_SENSITIVE_DATA),
+            RRDFUNCTIONS_PRIORITY_DEFAULT);
+
+#ifdef ENABLE_SYSTEMD_DBUS
+    fprintf(stdout, PLUGINSD_KEYWORD_FUNCTION " GLOBAL \"%s\" %d \"%s\" \"top\" "HTTP_ACCESS_FORMAT" %d\n",
+            SYSTEMD_UNITS_FUNCTION_NAME, SYSTEMD_UNITS_DEFAULT_TIMEOUT, SYSTEMD_UNITS_FUNCTION_DESCRIPTION,
+            (HTTP_ACCESS_FORMAT_CAST)(HTTP_ACCESS_SIGNED_ID | HTTP_ACCESS_SAME_SPACE | HTTP_ACCESS_SENSITIVE_DATA),
+            RRDFUNCTIONS_PRIORITY_DEFAULT);
+#endif
+
+    fflush(stdout);
+    netdata_mutex_unlock(&stdout_mutex);
+
+    // ------------------------------------------------------------------------
+
+    usec_t step_ut = 100 * USEC_PER_MS;
+    usec_t send_newline_ut = 0;
+    usec_t since_last_scan_ut = SYSTEMD_JOURNAL_ALL_FILES_SCAN_EVERY_USEC * 2; // something big to trigger scanning at start
+    bool tty = isatty(fileno(stdout)) == 1;
+
+    heartbeat_t hb;
+    heartbeat_init(&hb);
+    while(!plugin_should_exit) {
+
+        if(since_last_scan_ut > SYSTEMD_JOURNAL_ALL_FILES_SCAN_EVERY_USEC) {
+            journal_files_registry_update();
+            since_last_scan_ut = 0;
+        }
+
+        usec_t dt_ut = heartbeat_next(&hb, step_ut);
+        since_last_scan_ut += dt_ut;
+        send_newline_ut += dt_ut;
+
+        if(!tty && send_newline_ut > USEC_PER_SEC) {
+            send_newline_and_flush(&stdout_mutex);
+            send_newline_ut = 0;
+        }
+    }
+
+    exit(0);
+}