New upstream version 1.29.90 upstream/1.29.90

1 files changed, 300 insertions, 0 deletions

diff --git a/src/core/main-utils.c b/src/core/main-utils.c
new file mode 100644
index 00000000..9bc497ca
--- /dev/null
+++ b/src/core/main-utils.c
@@ -0,0 +1,300 @@
+/* SPDX-License-Identifier: GPL-2.0-or-later */
+/*
+ * Copyright (C) 2004 - 2012 Red Hat, Inc.
+ * Copyright (C) 2005 - 2008 Novell, Inc.
+ */
+
+#include "src/core/nm-default-daemon.h"
+
+#include <stdio.h>
+#include <fcntl.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <sys/stat.h>
+#include <locale.h>
+
+#include <glib/gstdio.h>
+#include <glib-unix.h>
+
+#include "main-utils.h"
+#include "NetworkManagerUtils.h"
+#include "nm-config.h"
+
+static gboolean
+sighup_handler(gpointer user_data)
+{
+    nm_main_config_reload(GPOINTER_TO_INT(user_data));
+    return G_SOURCE_CONTINUE;
+}
+
+static gboolean
+sigint_handler(gpointer user_data)
+{
+    GMainLoop *main_loop = user_data;
+
+    nm_log_info(LOGD_CORE, "caught SIGINT, shutting down normally.");
+    g_main_loop_quit(main_loop);
+
+    return G_SOURCE_REMOVE;
+}
+
+static gboolean
+sigterm_handler(gpointer user_data)
+{
+    GMainLoop *main_loop = user_data;
+
+    nm_log_info(LOGD_CORE, "caught SIGTERM, shutting down normally.");
+    g_main_loop_quit(main_loop);
+
+    return G_SOURCE_REMOVE;
+}
+
+/**
+ * nm_main_utils_setup_signals:
+ * @main_loop: the #GMainLoop to quit when SIGINT or SIGTERM is received
+ *
+ * Sets up signal handling for NetworkManager.
+ */
+void
+nm_main_utils_setup_signals(GMainLoop *main_loop)
+{
+    g_return_if_fail(main_loop != NULL);
+
+    signal(SIGPIPE, SIG_IGN);
+
+    g_unix_signal_add(SIGHUP, sighup_handler, GINT_TO_POINTER(SIGHUP));
+    if (nm_glib_check_version(2, 36, 0)) {
+        g_unix_signal_add(SIGUSR1, sighup_handler, GINT_TO_POINTER(SIGUSR1));
+        g_unix_signal_add(SIGUSR2, sighup_handler, GINT_TO_POINTER(SIGUSR2));
+    } else
+        nm_log_warn(LOGD_CORE,
+                    "glib-version: cannot handle SIGUSR1 and SIGUSR2 signals. Consider upgrading "
+                    "glib to 2.36.0 or newer");
+    g_unix_signal_add(SIGINT, sigint_handler, main_loop);
+    g_unix_signal_add(SIGTERM, sigterm_handler, main_loop);
+}
+
+gboolean
+nm_main_utils_write_pidfile(const char *pidfile)
+{
+    char     pid[16];
+    int      fd;
+    int      errsv;
+    gboolean success = FALSE;
+
+    if ((fd = open(pidfile, O_CREAT | O_WRONLY | O_TRUNC | O_CLOEXEC, 00644)) < 0) {
+        errsv = errno;
+        fprintf(stderr, _("Opening %s failed: %s\n"), pidfile, nm_strerror_native(errsv));
+        return FALSE;
+    }
+
+    g_snprintf(pid, sizeof(pid), "%d", getpid());
+    if (write(fd, pid, strlen(pid)) < 0) {
+        errsv = errno;
+        fprintf(stderr, _("Writing to %s failed: %s\n"), pidfile, nm_strerror_native(errsv));
+    } else
+        success = TRUE;
+
+    if (nm_close(fd)) {
+        errsv = errno;
+        fprintf(stderr, _("Closing %s failed: %s\n"), pidfile, nm_strerror_native(errsv));
+    }
+
+    return success;
+}
+
+void
+nm_main_utils_ensure_statedir()
+{
+    gs_free char *parent = NULL;
+    int           errsv;
+
+    parent = g_path_get_dirname(NMSTATEDIR);
+
+    /* Ensure parent state directories exists */
+    if (parent && parent[0] == '/' && parent[1] != '\0'
+        && g_mkdir_with_parents(parent, 0755) != 0) {
+        errsv = errno;
+        fprintf(stderr,
+                "Cannot create parents for '%s': %s",
+                NMSTATEDIR,
+                nm_strerror_native(errsv));
+        exit(1);
+    }
+    /* Ensure state directory exists */
+    if (g_mkdir_with_parents(NMSTATEDIR, 0700) != 0) {
+        errsv = errno;
+        fprintf(stderr, "Cannot create '%s': %s", NMSTATEDIR, nm_strerror_native(errsv));
+        exit(1);
+    }
+}
+
+void
+nm_main_utils_ensure_rundir()
+{
+    int errsv;
+
+    /* Setup runtime directory */
+    if (g_mkdir_with_parents(NMRUNDIR, 0755) != 0) {
+        errsv = errno;
+        fprintf(stderr, _("Cannot create '%s': %s"), NMRUNDIR, nm_strerror_native(errsv));
+        exit(1);
+    }
+
+    /* NM_CONFIG_DEVICE_STATE_DIR is used to determine whether NM is restarted or not.
+     * It is important to set NMConfigCmdLineOptions.first_start before creating
+     * the directory. */
+    nm_assert(g_str_has_prefix(NM_CONFIG_DEVICE_STATE_DIR, NMRUNDIR "/"));
+    if (g_mkdir(NM_CONFIG_DEVICE_STATE_DIR, 0755) != 0) {
+        errsv = errno;
+        if (errsv != EEXIST) {
+            fprintf(stderr,
+                    _("Cannot create '%s': %s"),
+                    NM_CONFIG_DEVICE_STATE_DIR,
+                    nm_strerror_native(errsv));
+            exit(1);
+        }
+    }
+}
+
+/**
+ * nm_main_utils_ensure_not_running_pidfile:
+ * @pidfile: the pid file
+ *
+ * Checks whether the pidfile already exists and contains PID of a running
+ * process.
+ *
+ * Exits with code 1 if a conflicting process is running.
+ */
+void
+nm_main_utils_ensure_not_running_pidfile(const char *pidfile)
+{
+    gs_free char *contents     = NULL;
+    gs_free char *proc_cmdline = NULL;
+    gsize         len          = 0;
+    long          pid;
+    const char *  process_name;
+    const char *  prgname = g_get_prgname();
+
+    g_return_if_fail(prgname);
+
+    if (!pidfile || !*pidfile)
+        return;
+
+    if (!g_file_get_contents(pidfile, &contents, &len, NULL))
+        return;
+    if (len <= 0)
+        return;
+
+    errno = 0;
+    pid   = strtol(contents, NULL, 10);
+    if (pid <= 0 || pid > 65536 || errno)
+        return;
+
+    nm_clear_g_free(&contents);
+    proc_cmdline = g_strdup_printf("/proc/%ld/cmdline", pid);
+    if (!g_file_get_contents(proc_cmdline, &contents, &len, NULL))
+        return;
+
+    process_name = strrchr(contents, '/');
+    if (process_name)
+        process_name++;
+    else
+        process_name = contents;
+    if (strcmp(process_name, prgname) == 0) {
+        /* Check that the process exists */
+        if (kill(pid, 0) == 0) {
+            fprintf(stderr, _("%s is already running (pid %ld)\n"), prgname, pid);
+            exit(1);
+        }
+    }
+}
+
+void
+nm_main_utils_ensure_root()
+{
+    if (getuid() != 0) {
+        fprintf(stderr, _("You must be root to run %s!\n"), g_get_prgname() ?: "");
+        exit(1);
+    }
+}
+
+gboolean
+nm_main_utils_early_setup(const char *  progname,
+                          int *         argc,
+                          char **       argv[],
+                          GOptionEntry *options,
+                          void (*option_context_hook)(gpointer user_data, GOptionContext *opt_ctx),
+                          gpointer    option_context_hook_data,
+                          const char *summary)
+{
+    GOptionContext *opt_ctx = NULL;
+    GError *        error   = NULL;
+    gboolean        success = FALSE;
+    int             i;
+    const char *    opt_fmt_log_level = NULL, *opt_fmt_log_domains = NULL;
+    const char **   opt_loc_log_level = NULL, **opt_loc_log_domains = NULL;
+
+    /* Make GIO ignore the remote VFS service; otherwise it tries to use the
+     * session bus to contact the remote service, and NM shouldn't ever be
+     * talking on the session bus.  See rh #588745
+     */
+    setenv("GIO_USE_VFS", "local", 1);
+
+    /*
+     * Set the umask to 0022, which results in 0666 & ~0022 = 0644.
+     * Otherwise, if root (or an su'ing user) has a wacky umask, we could
+     * write out an unreadable resolv.conf.
+     */
+    umask(022);
+
+    /* Ensure gettext() gets the right environment (bgo #666516) */
+    setlocale(LC_ALL, "");
+    textdomain(GETTEXT_PACKAGE);
+
+    for (i = 0; options[i].long_name; i++) {
+        NM_PRAGMA_WARNING_DISABLE("-Wformat-nonliteral")
+        if (!strcmp(options[i].long_name, "log-level")) {
+            opt_fmt_log_level = options[i].description;
+            opt_loc_log_level = &options[i].description;
+            options[i].description =
+                g_strdup_printf(options[i].description, nm_logging_all_levels_to_string());
+        } else if (!strcmp(options[i].long_name, "log-domains")) {
+            opt_fmt_log_domains = options[i].description;
+            opt_loc_log_domains = &options[i].description;
+            options[i].description =
+                g_strdup_printf(options[i].description, nm_logging_all_domains_to_string());
+        }
+        NM_PRAGMA_WARNING_REENABLE
+    }
+
+    /* Parse options */
+    opt_ctx = g_option_context_new(NULL);
+    g_option_context_set_translation_domain(opt_ctx, GETTEXT_PACKAGE);
+    g_option_context_set_ignore_unknown_options(opt_ctx, FALSE);
+    g_option_context_set_help_enabled(opt_ctx, TRUE);
+    g_option_context_add_main_entries(opt_ctx, options, NULL);
+    g_option_context_set_summary(opt_ctx, summary);
+    if (option_context_hook)
+        option_context_hook(option_context_hook_data, opt_ctx);
+
+    success = g_option_context_parse(opt_ctx, argc, argv, &error);
+    if (!success) {
+        fprintf(stderr,
+                _("%s.  Please use --help to see a list of valid options.\n"),
+                error->message);
+        g_clear_error(&error);
+    }
+    g_option_context_free(opt_ctx);
+
+    if (opt_loc_log_level) {
+        g_free((char *) *opt_loc_log_level);
+        *opt_loc_log_level = opt_fmt_log_level;
+    }
+    if (opt_loc_log_domains) {
+        g_free((char *) *opt_loc_log_domains);
+        *opt_loc_log_domains = opt_fmt_log_domains;
+    }
+
+    return success;
+}