diff options
Diffstat (limited to 'src/resolve/resolved.c')
-rw-r--r-- | src/resolve/resolved.c | 81 |
1 files changed, 31 insertions, 50 deletions
diff --git a/src/resolve/resolved.c b/src/resolve/resolved.c index c01e53e9da..f4efddf8e5 100644 --- a/src/resolve/resolved.c +++ b/src/resolve/resolved.c @@ -4,6 +4,8 @@ #include "sd-event.h" #include "capability-util.h" +#include "daemon-util.h" +#include "main-func.h" #include "mkdir.h" #include "resolved-conf.h" #include "resolved-manager.h" @@ -12,45 +14,35 @@ #include "signal-util.h" #include "user-util.h" -int main(int argc, char *argv[]) { +static int run(int argc, char *argv[]) { + _cleanup_(notify_on_cleanup) const char *notify_stop = NULL; _cleanup_(manager_freep) Manager *m = NULL; const char *user = "systemd-resolve"; uid_t uid; gid_t gid; int r; - log_set_target(LOG_TARGET_AUTO); - log_parse_environment(); - log_open(); + log_setup_service(); - if (argc != 1) { - log_error("This program takes no arguments."); - r = -EINVAL; - goto finish; - } + if (argc != 1) + return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "This program takes no arguments."); umask(0022); r = mac_selinux_init(); - if (r < 0) { - log_error_errno(r, "SELinux setup failed: %m"); - goto finish; - } + if (r < 0) + return log_error_errno(r, "SELinux setup failed: %m"); - r = get_user_creds(&user, &uid, &gid, NULL, NULL); - if (r < 0) { - log_error_errno(r, "Cannot resolve user name %s: %m", user); - goto finish; - } + r = get_user_creds(&user, &uid, &gid, NULL, NULL, 0); + if (r < 0) + return log_error_errno(r, "Cannot resolve user name %s: %m", user); /* Always create the directory where resolv.conf will live */ r = mkdir_safe_label("/run/systemd/resolve", 0755, uid, gid, MKDIR_WARN_MODE); - if (r < 0) { - log_error_errno(r, "Could not create runtime directory: %m"); - goto finish; - } + if (r < 0) + return log_error_errno(r, "Could not create runtime directory: %m"); - /* Drop privileges, but only if we have been started as root. If we are not running as root we assume all + /* Drop privileges, but only if we have been started as root. If we are not running as root we assume most * privileges are already dropped. */ if (getuid() == 0) { @@ -60,49 +52,38 @@ int main(int argc, char *argv[]) { (UINT64_C(1) << CAP_NET_BIND_SERVICE)| /* needed to bind on port 53 */ (UINT64_C(1) << CAP_SETPCAP) /* needed in order to drop the caps later */); if (r < 0) - goto finish; + return log_error_errno(r, "Failed to drop privileges: %m"); } assert_se(sigprocmask_many(SIG_BLOCK, NULL, SIGTERM, SIGINT, SIGUSR1, SIGUSR2, SIGRTMIN+1, -1) >= 0); r = manager_new(&m); - if (r < 0) { - log_error_errno(r, "Could not create manager: %m"); - goto finish; - } + if (r < 0) + return log_error_errno(r, "Could not create manager: %m"); r = manager_start(m); - if (r < 0) { - log_error_errno(r, "Failed to start manager: %m"); - goto finish; - } + if (r < 0) + return log_error_errno(r, "Failed to start manager: %m"); /* Write finish default resolv.conf to avoid a dangling symlink */ (void) manager_write_resolv_conf(m); + (void) manager_check_resolv_conf(m); + /* Let's drop the remaining caps now */ r = capability_bounding_set_drop(0, true); - if (r < 0) { - log_error_errno(r, "Failed to drop remaining caps: %m"); - goto finish; - } + if (r < 0) + return log_error_errno(r, "Failed to drop remaining caps: %m"); - sd_notify(false, - "READY=1\n" - "STATUS=Processing requests..."); + notify_stop = notify_start(NOTIFY_READY, NOTIFY_STOPPING); r = sd_event_loop(m->event); - if (r < 0) { - log_error_errno(r, "Event loop failed: %m"); - goto finish; - } + if (r < 0) + return log_error_errno(r, "Event loop failed: %m"); - sd_event_get_exit_code(m->event, &r); + (void) sd_event_get_exit_code(m->event, &r); -finish: - sd_notify(false, - "STOPPING=1\n" - "STATUS=Shutting down..."); - - return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS; + return r; } + +DEFINE_MAIN_FUNCTION(run); |