diff options
Diffstat (limited to 'src/main.c')
| -rw-r--r-- | src/main.c | 253 |
1 files changed, 24 insertions, 229 deletions
diff --git a/src/main.c b/src/main.c index f4f693d1e4..476d2a7711 100644 --- a/src/main.c +++ b/src/main.c @@ -42,6 +42,7 @@ #include "gsystem-local-alloc.h" #include "nm-dbus-interface.h" #include "NetworkManagerUtils.h" +#include "main-utils.h" #include "nm-manager.h" #include "nm-linux-platform.h" #include "nm-dns-manager.h" @@ -65,161 +66,7 @@ #define NM_DEFAULT_PID_FILE NMRUNDIR "/NetworkManager.pid" #define NM_DEFAULT_SYSTEM_STATE_FILE NMSTATEDIR "/NetworkManager.state" -/* - * Globals - */ static GMainLoop *main_loop = NULL; -static gboolean quit_early = FALSE; -static sigset_t signal_set; - -void *signal_handling_thread (void *arg); -/* - * Thread function waiting for signals and processing them. - * Wait for signals in signal set. The semantics of sigwait() require that all - * threads (including the thread calling sigwait()) have the signal masked, for - * reliable operation. Otherwise, a signal that arrives while this thread is - * not blocked in sigwait() might be delivered to another thread. - */ -void * -signal_handling_thread (void *arg) -{ - int signo; - - while (1) { - sigwait (&signal_set, &signo); - - switch (signo) { - case SIGINT: - case SIGTERM: - nm_log_info (LOGD_CORE, "caught signal %d, shutting down normally.", signo); - quit_early = TRUE; /* for quitting before entering the main loop */ - g_main_loop_quit (main_loop); - break; - case SIGHUP: - /* Reread config stuff like system config files, VPN service files, etc */ - nm_log_info (LOGD_CORE, "caught signal %d, not supported yet.", signo); - break; - case SIGPIPE: - /* silently ignore signal */ - break; - default: - nm_log_err (LOGD_CORE, "caught unexpected signal %d", signo); - break; - } - } - return NULL; -} - -/* - * Mask the signals we are interested in and create a signal handling thread. - * Because all threads inherit the signal mask from their creator, all threads - * in the process will have the signals masked. That's why setup_signals() has - * to be called before creating other threads. - */ -static gboolean -setup_signals (void) -{ - pthread_t signal_thread_id; - sigset_t old_sig_mask; - int status; - - sigemptyset (&signal_set); - sigaddset (&signal_set, SIGHUP); - sigaddset (&signal_set, SIGINT); - sigaddset (&signal_set, SIGTERM); - sigaddset (&signal_set, SIGPIPE); - - /* Block all signals of interest. */ - status = pthread_sigmask (SIG_BLOCK, &signal_set, &old_sig_mask); - if (status != 0) { - fprintf (stderr, _("Failed to set signal mask: %d"), status); - return FALSE; - } - /* Save original mask so that we could use it for child processes. */ - nm_save_original_signal_mask (old_sig_mask); - - /* Create the signal handling thread. */ - status = pthread_create (&signal_thread_id, NULL, signal_handling_thread, NULL); - if (status != 0) { - fprintf (stderr, _("Failed to create signal handling thread: %d"), status); - return FALSE; - } - - return TRUE; -} - -static gboolean -write_pidfile (const char *pidfile) -{ - char pid[16]; - int fd; - gboolean success = FALSE; - - if ((fd = open (pidfile, O_CREAT|O_WRONLY|O_TRUNC, 00644)) < 0) { - fprintf (stderr, _("Opening %s failed: %s\n"), pidfile, strerror (errno)); - return FALSE; - } - - g_snprintf (pid, sizeof (pid), "%d", getpid ()); - if (write (fd, pid, strlen (pid)) < 0) - fprintf (stderr, _("Writing to %s failed: %s\n"), pidfile, strerror (errno)); - else - success = TRUE; - - if (close (fd)) - fprintf (stderr, _("Closing %s failed: %s\n"), pidfile, strerror (errno)); - - return success; -} - -/* Check whether the pidfile already exists and contains PID of a running NetworkManager - * Returns: FALSE - specified pidfile doesn't exist or doesn't contain PID of a running NM process - * TRUE - specified pidfile already exists and contains PID of a running NM process - */ -static gboolean -check_pidfile (const char *pidfile) -{ - char *contents = NULL; - gsize len = 0; - glong pid; - char *proc_cmdline = NULL; - gboolean nm_running = FALSE; - const char *process_name; - - if (!g_file_get_contents (pidfile, &contents, &len, NULL)) - return FALSE; - - if (len <= 0) - goto done; - - errno = 0; - pid = strtol (contents, NULL, 10); - if (pid <= 0 || pid > 65536 || errno) - goto done; - - g_free (contents); - proc_cmdline = g_strdup_printf ("/proc/%ld/cmdline", pid); - if (!g_file_get_contents (proc_cmdline, &contents, &len, NULL)) - goto done; - - process_name = strrchr (contents, '/'); - if (process_name) - process_name++; - else - process_name = contents; - if (strcmp (process_name, "NetworkManager") == 0) { - /* Check that the process exists */ - if (kill (pid, 0) == 0) { - fprintf (stderr, _("NetworkManager is already running (pid %ld)\n"), pid); - nm_running = TRUE; - } - } - -done: - g_free (proc_cmdline); - g_free (contents); - return nm_running; -} static gboolean parse_state_file (const char *filename, @@ -330,6 +177,13 @@ _init_nm_debug (const char *debug) } } +static void +manager_configure_quit (NMManager *manager, gpointer user_data) +{ + nm_log_info (LOGD_CORE, "quitting now that startup is complete"); + g_main_loop_quit (main_loop); +} + /* * main * @@ -337,7 +191,6 @@ _init_nm_debug (const char *debug) int main (int argc, char *argv[]) { - GOptionContext *opt_ctx = NULL; char *opt_log_level = NULL; char *opt_log_domains = NULL; gboolean become_daemon = TRUE, run_from_build_dir = FALSE; @@ -347,7 +200,6 @@ main (int argc, char *argv[]) gs_free char *state_file = NULL; gboolean wifi_enabled = TRUE, net_enabled = TRUE, wwan_enabled = TRUE, wimax_enabled = TRUE; gboolean success, show_version = FALSE; - int i; NMManager *manager = NULL; gs_unref_object NMVpnManager *vpn_manager = NULL; gs_unref_object NMDnsManager *dns_mgr = NULL; @@ -361,6 +213,7 @@ main (int argc, char *argv[]) GError *error = NULL; gboolean wrote_pidfile = FALSE; char *bad_domains = NULL; + gboolean quit_early = FALSE; GOptionEntry options[] = { { "version", 'V', 0, G_OPTION_ARG_NONE, &show_version, N_("Print NetworkManager version and exit"), NULL }, @@ -377,66 +230,15 @@ main (int argc, char *argv[]) {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); - - /* Set locale to be able to use environment variables */ - setlocale (LC_ALL, ""); - - bindtextdomain (GETTEXT_PACKAGE, NMLOCALEDIR); - bind_textdomain_codeset (GETTEXT_PACKAGE, "UTF-8"); - textdomain (GETTEXT_PACKAGE); - - if (!g_module_supported ()) { - fprintf (stderr, _("GModules are not supported on your platform!\n")); - exit (1); - } - - if (getuid () != 0) { - fprintf (stderr, _("You must be root to run NetworkManager!\n")); - exit (1); - } - - for (i = 0; options[i].long_name; i++) { - if (!strcmp (options[i].long_name, "log-level")) { - options[i].description = g_strdup_printf (options[i].description, - nm_logging_all_levels_to_string ()); - } else if (!strcmp (options[i].long_name, "log-domains")) { - options[i].description = g_strdup_printf (options[i].description, - nm_logging_all_domains_to_string ()); - } - } - - /* 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_add_main_entries (opt_ctx, nm_config_get_options (), NULL); - - g_option_context_set_summary (opt_ctx, - _("NetworkManager monitors all network connections and automatically\nchooses the best connection to use. It also allows the user to\nspecify wireless access points which wireless cards in the computer\nshould associate with.")); - - success = g_option_context_parse (opt_ctx, &argc, &argv, &error); - g_option_context_free (opt_ctx); + main_loop = g_main_loop_new (NULL, FALSE); - if (!success) { - fprintf (stderr, _("%s. Please use --help to see a list of valid options.\n"), - error->message); - g_clear_error (&error); + if (!nm_main_utils_early_setup ("NetworkManager", + &argv, + &argc, + options, + nm_config_get_options (), + _("NetworkManager monitors all network connections and automatically\nchooses the best connection to use. It also allows the user to\nspecify wireless access points which wireless cards in the computer\nshould associate with."))) exit (1); - } if (show_version) { fprintf (stdout, NM_DIST_VERSION "\n"); @@ -482,12 +284,6 @@ main (int argc, char *argv[]) g_free (path); } - /* Setup runtime directory */ - if (g_mkdir_with_parents (NMRUNDIR, 0755) != 0) { - nm_log_err (LOGD_CORE, "Cannot create '%s': %s", NMRUNDIR, strerror (errno)); - exit (1); - } - /* Ensure state directory exists */ if (g_mkdir_with_parents (NMSTATEDIR, 0755) != 0) { nm_log_err (LOGD_CORE, "Cannot create '%s': %s", NMSTATEDIR, strerror (errno)); @@ -498,7 +294,7 @@ main (int argc, char *argv[]) state_file = state_file ? state_file : g_strdup (NM_DEFAULT_SYSTEM_STATE_FILE); /* check pid file */ - if (check_pidfile (pidfile)) + if (nm_main_utils_check_pidfile (pidfile, "NetworkManager")) exit (1); /* Read the config file and CLI overrides */ @@ -549,14 +345,13 @@ main (int argc, char *argv[]) saved_errno); exit (1); } - if (write_pidfile (pidfile)) - wrote_pidfile = TRUE; + wrote_pidfile = nm_main_utils_write_pidfile (pidfile); } _init_nm_debug (nm_config_get_debug (config)); /* Set up unix signal handling - before creating threads, but after daemonizing! */ - if (!setup_signals ()) + if (!nm_main_utils_setup_signals (main_loop, &quit_early)) exit (1); if (g_fatal_warnings) { @@ -592,8 +387,6 @@ main (int argc, char *argv[]) #endif ); - main_loop = g_main_loop_new (NULL, FALSE); - /* Set up platform interaction layer */ nm_linux_platform_setup (); @@ -662,6 +455,8 @@ main (int argc, char *argv[]) } } + g_signal_connect (manager, NM_MANAGER_CONFIGURE_QUIT, G_CALLBACK (manager_configure_quit), config); + nm_manager_start (manager); /* Make sure the loopback interface is up. If interface is down, we bring @@ -680,10 +475,10 @@ main (int argc, char *argv[]) success = TRUE; /* Told to quit before getting to the mainloop by the signal handler */ - if (quit_early == TRUE) - goto done; + if (!quit_early) + g_main_loop_run (main_loop); - g_main_loop_run (main_loop); + nm_manager_stop (manager); done: g_clear_object (&manager); |