32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <dirent.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <errno.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <fcntl.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <linux/kd.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <signal.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <string.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <sys/epoll.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <sys/inotify.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <sys/ioctl.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <sys/reboot.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <sys/timerfd.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <sys/wait.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <unistd.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#ifdef HAVE_AUDIT

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include <libaudit.h>

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#endif

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "sd-daemon.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "sd-messages.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "alloc-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "audit-fd.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "boot-timestamps.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "bus-common-errors.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "bus-error.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "bus-kernel.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "bus-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "dbus-job.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "dbus-manager.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "dbus-unit.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "dbus.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "env-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "escape.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "exit-status.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "fd-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "fileio.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "fs-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "hashmap.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "io-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "locale-setup.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "log.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "macro.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "manager.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "missing.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "mkdir.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "parse-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "path-lookup.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "path-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "process-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "ratelimit.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "rm-rf.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "signal-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "special.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "stat-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "string-table.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "string-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "strv.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "terminal-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "time-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "transaction.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "umask-util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "unit-name.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "util.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "virt.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#include "watchdog.h"

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#define NOTIFY_RCVBUF_SIZE (8*1024*1024)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#define JOBS_IN_PROGRESS_WAIT_USEC (5*USEC_PER_SEC)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#define JOBS_IN_PROGRESS_PERIOD_USEC (USEC_PER_SEC / 3)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#define JOBS_IN_PROGRESS_PERIOD_DIVISOR 3

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_dispatch_notify_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_dispatch_signal_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_dispatch_time_change_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_dispatch_idle_pipe_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_dispatch_jobs_in_progress(sd_event_source *source, usec_t usec, void *userdata);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_dispatch_run_queue(sd_event_source *source, void *userdata);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_run_generators(Manager *m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic void manager_undo_generators(Manager *m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic void manager_watch_jobs_in_progress(Manager *m) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt usec_t next;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt int r;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->jobs_in_progress_event_source)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt next = now(CLOCK_MONOTONIC) + JOBS_IN_PROGRESS_WAIT_USEC;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = sd_event_add_time(

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->event,

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt &m->jobs_in_progress_event_source,

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt CLOCK_MONOTONIC,

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt next, 0,

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt manager_dispatch_jobs_in_progress, m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt (void) sd_event_source_set_description(m->jobs_in_progress_event_source, "manager-jobs-in-progress");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#define CYLON_BUFFER_EXTRA (2*(sizeof(ANSI_RED)-1) + sizeof(ANSI_HIGHLIGHT_RED)-1 + 2*(sizeof(ANSI_NORMAL)-1))

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic void draw_cylon(char buffer[], size_t buflen, unsigned width, unsigned pos) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt char *p = buffer;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(buflen >= CYLON_BUFFER_EXTRA + width + 1);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(pos <= width+1); /* 0 or width+1 mean that the center light is behind the corner */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (pos > 1) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (pos > 2)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt p = mempset(p, ' ', pos-2);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt p = stpcpy(p, ANSI_RED);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt *p++ = '*';

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt }

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (pos > 0 && pos <= width) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt p = stpcpy(p, ANSI_HIGHLIGHT_RED);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt *p++ = '*';

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt }

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt p = stpcpy(p, ANSI_NORMAL);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (pos < width) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt p = stpcpy(p, ANSI_RED);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt *p++ = '*';

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (pos < width-1)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt p = mempset(p, ' ', width-1-pos);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt strcpy(p, ANSI_NORMAL);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt }

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidtvoid manager_flip_auto_status(Manager *m, bool enable) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (enable) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->show_status == SHOW_STATUS_AUTO)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt manager_set_show_status(m, SHOW_STATUS_TEMPORARY);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt } else {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->show_status == SHOW_STATUS_TEMPORARY)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt manager_set_show_status(m, SHOW_STATUS_AUTO);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt }

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidtstatic void manager_print_jobs_in_progress(Manager *m) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt _cleanup_free_ char *job_of_n = NULL;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt Iterator i;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt Job *j;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt unsigned counter = 0, print_nr;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt char cylon[6 + CYLON_BUFFER_EXTRA + 1];

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt unsigned cylon_pos;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt char time[FORMAT_TIMESPAN_MAX], limit[FORMAT_TIMESPAN_MAX] = "no limit";

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt uint64_t x;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(m->n_running_jobs > 0);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt manager_flip_auto_status(m, true);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt print_nr = (m->jobs_in_progress_iteration / JOBS_IN_PROGRESS_PERIOD_DIVISOR) % m->n_running_jobs;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt HASHMAP_FOREACH(j, m->jobs, i)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (j->state == JOB_RUNNING && counter++ == print_nr)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt break;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt /* m->n_running_jobs must be consistent with the contents of m->jobs,

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(counter == print_nr + 1);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(j);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt cylon_pos = m->jobs_in_progress_iteration % 14;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (cylon_pos >= 8)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt cylon_pos = 14 - cylon_pos;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt draw_cylon(cylon, sizeof(cylon), 6, cylon_pos);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->jobs_in_progress_iteration++;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (m->n_running_jobs > 1) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (asprintf(&job_of_n, "(%u of %u) ", counter, m->n_running_jobs) < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt job_of_n = NULL;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt }

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt format_timespan(time, sizeof(time), now(CLOCK_MONOTONIC) - j->begin_usec, 1*USEC_PER_SEC);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (job_get_timeout(j, &x) > 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt format_timespan(limit, sizeof(limit), x - j->begin_usec, 1*USEC_PER_SEC);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt manager_status_printf(m, STATUS_TYPE_EPHEMERAL, cylon,

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt "%sA %s job is running for %s (%s / %s)",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt strempty(job_of_n),

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt job_type_to_string(j->type),

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt unit_description(j->unit),

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt time, limit);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int have_ask_password(void) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt _cleanup_closedir_ DIR *dir;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt dir = opendir("/run/systemd/ask-password");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (!dir) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (errno == ENOENT)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return false;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt else

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return -errno;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt }

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt for (;;) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt struct dirent *de;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt errno = 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt de = readdir(dir);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (!de && errno != 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return -errno;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (!de)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return false;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (startswith(de->d_name, "ask."))

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return true;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt }

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_dispatch_ask_password_fd(sd_event_source *source,

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt int fd, uint32_t revents, void *userdata) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt Manager *m = userdata;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt flush_fd(fd);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->have_ask_password = have_ask_password();

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (m->have_ask_password < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* Log error but continue. Negative have_ask_password

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt log_error_errno(m->have_ask_password, "Failed to list /run/systemd/ask-password: %m");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidtstatic void manager_close_ask_password(Manager *m) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->ask_password_event_source = sd_event_source_unref(m->ask_password_event_source);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->ask_password_inotify_fd = safe_close(m->ask_password_inotify_fd);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->have_ask_password = -EINVAL;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidtstatic int manager_check_ask_password(Manager *m) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt int r;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (!m->ask_password_event_source) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(m->ask_password_inotify_fd < 0);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt mkdir_p_label("/run/systemd/ask-password", 0755);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->ask_password_inotify_fd = inotify_init1(IN_NONBLOCK|IN_CLOEXEC);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (m->ask_password_inotify_fd < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return log_error_errno(errno, "inotify_init1() failed: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (inotify_add_watch(m->ask_password_inotify_fd, "/run/systemd/ask-password", IN_CREATE|IN_DELETE|IN_MOVE) < 0) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt log_error_errno(errno, "Failed to add watch on /run/systemd/ask-password: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt manager_close_ask_password(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return -errno;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt }

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = sd_event_add_io(m->event, &m->ask_password_event_source,

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->ask_password_inotify_fd, EPOLLIN,

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt manager_dispatch_ask_password_fd, m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt log_error_errno(errno, "Failed to add event source for /run/systemd/ask-password: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt manager_close_ask_password(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return -errno;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt }

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt (void) sd_event_source_set_description(m->ask_password_event_source, "manager-ask-password");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt /* Queries might have been added meanwhile... */

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt manager_dispatch_ask_password_fd(m->ask_password_event_source,

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->ask_password_inotify_fd, EPOLLIN, m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt }

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return m->have_ask_password;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidtstatic int manager_watch_idle_pipe(Manager *m) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt int r;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (m->idle_pipe_event_source)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (m->idle_pipe[2] < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = sd_event_add_io(m->event, &m->idle_pipe_event_source, m->idle_pipe[2], EPOLLIN, manager_dispatch_idle_pipe_fd, m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return log_error_errno(r, "Failed to watch idle pipe: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt (void) sd_event_source_set_description(m->idle_pipe_event_source, "manager-idle-pipe");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidtstatic void manager_close_idle_pipe(Manager *m) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->idle_pipe_event_source = sd_event_source_unref(m->idle_pipe_event_source);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt safe_close_pair(m->idle_pipe);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt safe_close_pair(m->idle_pipe + 2);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidtstatic int manager_setup_time_change(Manager *m) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt int r;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt /* We only care for the cancellation event, hence we set the

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt struct itimerspec its = {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt .it_value.tv_sec = TIME_T_MAX,

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt };

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert_cc(sizeof(time_t) == sizeof(TIME_T_MAX));

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->test_run)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return 0;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt /* Uses TFD_TIMER_CANCEL_ON_SET to get notifications whenever

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->time_change_fd = timerfd_create(CLOCK_REALTIME, TFD_NONBLOCK|TFD_CLOEXEC);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->time_change_fd < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return log_error_errno(errno, "Failed to create timerfd: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (timerfd_settime(m->time_change_fd, TFD_TIMER_ABSTIME|TFD_TIMER_CANCEL_ON_SET, &its, NULL) < 0) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt log_debug_errno(errno, "Failed to set up TFD_TIMER_CANCEL_ON_SET, ignoring: %m");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->time_change_fd = safe_close(m->time_change_fd);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return 0;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt }

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = sd_event_add_io(m->event, &m->time_change_event_source, m->time_change_fd, EPOLLIN, manager_dispatch_time_change_fd, m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return log_error_errno(r, "Failed to create time change event source: %m");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt (void) sd_event_source_set_description(m->time_change_event_source, "manager-time-change");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt log_debug("Set up TFD_TIMER_CANCEL_ON_SET timerfd.");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidtstatic int enable_special_signals(Manager *m) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt _cleanup_close_ int fd = -1;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (m->test_run)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt /* Enable that we get SIGINT on control-alt-del. In containers

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (reboot(RB_DISABLE_CAD) < 0 && errno != EPERM && errno != EINVAL)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt log_warning_errno(errno, "Failed to enable ctrl-alt-del handling: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (fd < 0) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt /* Support systems without virtual console */

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (fd != -ENOENT)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt log_warning_errno(errno, "Failed to open /dev/tty0: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt } else {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt /* Enable that we get SIGWINCH on kbrequest */

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (ioctl(fd, KDSIGACCEPT, SIGWINCH) < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt log_warning_errno(errno, "Failed to enable kbrequest handling: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt }

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidtstatic int manager_setup_signals(Manager *m) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt struct sigaction sa = {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt .sa_handler = SIG_DFL,

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt .sa_flags = SA_NOCLDSTOP|SA_RESTART,

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt };

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt sigset_t mask;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt int r;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert_se(sigaction(SIGCHLD, &sa, NULL) == 0);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* We make liberal use of realtime signals here. On

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert_se(sigemptyset(&mask) == 0);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt sigset_add_many(&mask,

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGCHLD, /* Child died */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGTERM, /* Reexecute daemon */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGHUP, /* Reload configuration */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGUSR1, /* systemd/upstart: reconnect to D-Bus */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGUSR2, /* systemd: dump status */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGINT, /* Kernel sends us this on control-alt-del */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGWINCH, /* Kernel sends us this on kbrequest (alt-arrowup) */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGPWR, /* Some kernel drivers and upsd send us this on power failure */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+0, /* systemd: start default.target */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+1, /* systemd: isolate rescue.target */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+2, /* systemd: isolate emergency.target */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+3, /* systemd: start halt.target */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+4, /* systemd: start poweroff.target */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+5, /* systemd: start reboot.target */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+6, /* systemd: start kexec.target */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* ... space for more special targets ... */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+13, /* systemd: Immediate halt */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+14, /* systemd: Immediate poweroff */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+15, /* systemd: Immediate reboot */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+16, /* systemd: Immediate kexec */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* ... space for more immediate system state changes ... */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+20, /* systemd: enable status messages */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+21, /* systemd: disable status messages */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+22, /* systemd: set log level to LOG_DEBUG */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+23, /* systemd: set log level to LOG_INFO */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+24, /* systemd: Immediate exit (--user only) */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* .. one free signal here ... */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#if !defined(__hppa64__) && !defined(__hppa__)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* Apparently Linux on hppa has fewer RT

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+26, /* systemd: set log target to journal-or-kmsg */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+27, /* systemd: set log target to console */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+28, /* systemd: set log target to kmsg */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SIGRTMIN+29, /* systemd: set log target to syslog-or-kmsg (obsolete) */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* ... one free signal here SIGRTMIN+30 ... */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#endif

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt -1);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt assert_se(sigprocmask(SIG_SETMASK, &mask, NULL) == 0);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->signal_fd = signalfd(-1, &mask, SFD_NONBLOCK|SFD_CLOEXEC);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (m->signal_fd < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return -errno;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = sd_event_add_io(m->event, &m->signal_event_source, m->signal_fd, EPOLLIN, manager_dispatch_signal_fd, m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return r;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt (void) sd_event_source_set_description(m->signal_event_source, "manager-signal");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt /* Process signals a bit earlier than the rest of things, but

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = sd_event_source_set_priority(m->signal_event_source, SD_EVENT_PRIORITY_NORMAL-5);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return r;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->running_as == MANAGER_SYSTEM)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return enable_special_signals(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return 0;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic void manager_clean_environment(Manager *m) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* Let's remove some environment variables that we

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt strv_env_unset_many(

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->environment,

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt "NOTIFY_SOCKET",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt "MAINPID",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt "MANAGERPID",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt "LISTEN_PID",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt "LISTEN_FDS",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt "LISTEN_FDNAMES",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt "WATCHDOG_PID",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt "WATCHDOG_USEC",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt NULL);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_default_environment(Manager *m) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->running_as == MANAGER_SYSTEM) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* The system manager always starts with a clean

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->environment = strv_new("PATH=" DEFAULT_PATH,

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt NULL);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* Import locale variables LC_*= from configuration */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt locale_setup(&m->environment);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt } else {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* The user manager passes its own environment

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->environment = strv_copy(environ);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt }

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (!m->environment)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return -ENOMEM;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt manager_clean_environment(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt strv_sort(m->environment);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return 0;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtint manager_new(ManagerRunningAs running_as, bool test_run, Manager **_m) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt static const char * const unit_log_fields[_MANAGER_RUNNING_AS_MAX] = {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt [MANAGER_SYSTEM] = "UNIT=",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt [MANAGER_USER] = "USER_UNIT=",

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt };

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt static const char * const unit_log_format_strings[_MANAGER_RUNNING_AS_MAX] = {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt [MANAGER_SYSTEM] = "UNIT=%s",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt [MANAGER_USER] = "USER_UNIT=%s",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt };

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt Manager *m;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt int r;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(_m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(running_as >= 0);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(running_as < _MANAGER_RUNNING_AS_MAX);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m = new0(Manager, 1);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (!m)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return -ENOMEM;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#ifdef ENABLE_EFI

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (running_as == MANAGER_SYSTEM && detect_container() <= 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt boot_timestamps(&m->userspace_timestamp, &m->firmware_timestamp, &m->loader_timestamp);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt#endif

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->running_as = running_as;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->exit_code = _MANAGER_EXIT_CODE_INVALID;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->default_timer_accuracy_usec = USEC_PER_MINUTE;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->default_tasks_accounting = true;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->default_tasks_max = UINT64_C(512);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* Prepare log fields we can use for structured logging */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->unit_log_field = unit_log_fields[running_as];

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->unit_log_format_string = unit_log_format_strings[running_as];

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->idle_pipe[0] = m->idle_pipe[1] = m->idle_pipe[2] = m->idle_pipe[3] = -1;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->pin_cgroupfs_fd = m->notify_fd = m->signal_fd = m->time_change_fd =

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->dev_autofs_fd = m->private_listen_fd = m->kdbus_fd = m->cgroup_inotify_fd = -1;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->current_job_id = 1; /* start as id #1, so that we can leave #0 around as "null-like" value */

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->ask_password_inotify_fd = -1;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->have_ask_password = -EINVAL; /* we don't know */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->first_boot = -1;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->cgroup_netclass_registry_last = CGROUP_NETCLASS_FIXED_MAX;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->test_run = test_run;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* Reboot immediately if the user hits C-A-D more often than 7x per 2s */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt RATELIMIT_INIT(m->ctrl_alt_del_ratelimit, 2 * USEC_PER_SEC, 7);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = manager_default_environment(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt goto fail;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = hashmap_ensure_allocated(&m->units, &string_hash_ops);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt goto fail;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = hashmap_ensure_allocated(&m->jobs, NULL);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt goto fail;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = hashmap_ensure_allocated(&m->cgroup_unit, &string_hash_ops);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt goto fail;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = hashmap_ensure_allocated(&m->watch_bus, &string_hash_ops);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt goto fail;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = sd_event_default(&m->event);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt goto fail;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = sd_event_add_defer(m->event, &m->run_queue_event_source, manager_dispatch_run_queue, m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt goto fail;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = sd_event_source_set_priority(m->run_queue_event_source, SD_EVENT_PRIORITY_IDLE);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt goto fail;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = sd_event_source_set_enabled(m->run_queue_event_source, SD_EVENT_OFF);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt goto fail;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt (void) sd_event_source_set_description(m->run_queue_event_source, "manager-run-queue");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = manager_setup_signals(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt goto fail;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = manager_setup_cgroup(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt goto fail;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = manager_setup_time_change(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt goto fail;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->udev = udev_new();

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (!m->udev) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = -ENOMEM;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt goto fail;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt }

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt /* Note that we set up neither kdbus, nor the notify fd

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->taint_usr = dir_is_empty("/usr") > 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt *_m = m;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidtfail:

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt manager_free(m);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return r;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt}

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_setup_notify(Manager *m) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt int r;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (m->test_run)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return 0;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (m->notify_fd < 0) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt _cleanup_close_ int fd = -1;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt union sockaddr_union sa = {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt .sa.sa_family = AF_UNIX,

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt };

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt static const int one = 1;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt /* First free all secondary fields */

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->notify_socket = mfree(m->notify_socket);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->notify_event_source = sd_event_source_unref(m->notify_event_source);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt fd = socket(AF_UNIX, SOCK_DGRAM|SOCK_CLOEXEC|SOCK_NONBLOCK, 0);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (fd < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return log_error_errno(errno, "Failed to allocate notification socket: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt fd_inc_rcvbuf(fd, NOTIFY_RCVBUF_SIZE);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->running_as == MANAGER_SYSTEM)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->notify_socket = strdup("/run/systemd/notify");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt else {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt const char *e;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt e = getenv("XDG_RUNTIME_DIR");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (!e) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt log_error_errno(errno, "XDG_RUNTIME_DIR is not set: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return -EINVAL;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt }

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->notify_socket = strappend(e, "/systemd/notify");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt }

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (!m->notify_socket)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return log_oom();

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt (void) mkdir_parents_label(m->notify_socket, 0755);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt (void) unlink(m->notify_socket);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt strncpy(sa.un.sun_path, m->notify_socket, sizeof(sa.un.sun_path)-1);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = bind(fd, &sa.sa, offsetof(struct sockaddr_un, sun_path) + strlen(sa.un.sun_path));

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return log_error_errno(errno, "bind(%s) failed: %m", sa.un.sun_path);

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt r = setsockopt(fd, SOL_SOCKET, SO_PASSCRED, &one, sizeof(one));

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (r < 0)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return log_error_errno(errno, "SO_PASSCRED failed: %m");

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt m->notify_fd = fd;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt fd = -1;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt log_debug("Using notification socket %s", m->notify_socket);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt }

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (!m->notify_event_source) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = sd_event_add_io(m->event, &m->notify_event_source, m->notify_fd, EPOLLIN, manager_dispatch_notify_fd, m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return log_error_errno(r, "Failed to allocate notify event source: %m");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* Process signals a bit earlier than SIGCHLD, so that we can

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt r = sd_event_source_set_priority(m->notify_event_source, SD_EVENT_PRIORITY_NORMAL-7);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (r < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return log_error_errno(r, "Failed to set priority of notify event source: %m");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt (void) sd_event_source_set_description(m->notify_event_source, "manager-notify");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt }

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return 0;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_setup_kdbus(Manager *m) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt _cleanup_free_ char *p = NULL;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->test_run || m->kdbus_fd >= 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return 0;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (!is_kdbus_available())

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return -ESOCKTNOSUPPORT;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->kdbus_fd = bus_kernel_create_bus(

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->running_as == MANAGER_SYSTEM ? "system" : "user",

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->running_as == MANAGER_SYSTEM, &p);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->kdbus_fd < 0)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return log_debug_errno(m->kdbus_fd, "Failed to set up kdbus: %m");

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt log_debug("Successfully set up kdbus on %s", p);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return 0;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic int manager_connect_bus(Manager *m, bool reexecuting) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt bool try_bus_connect;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (m->test_run)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return 0;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt try_bus_connect =

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt m->kdbus_fd >= 0 ||

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt reexecuting ||

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt (m->running_as == MANAGER_USER && getenv("DBUS_SESSION_BUS_ADDRESS"));

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt /* Try to connect to the buses, if possible. */

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt return bus_init(m, try_bus_connect);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt}

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic unsigned manager_dispatch_cleanup_queue(Manager *m) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt Unit *u;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt unsigned n = 0;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(m);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt while ((u = m->cleanup_queue)) {

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt assert(u->in_cleanup_queue);

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt unit_free(u);

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt n++;

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt }

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt return n;

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt}

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidtenum {

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt GC_OFFSET_IN_PATH, /* This one is on the path we were traveling */

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt GC_OFFSET_UNSURE, /* No clue */

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt GC_OFFSET_GOOD, /* We still need this unit */

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt GC_OFFSET_BAD, /* We don't need this unit anymore */

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt _GC_OFFSET_MAX

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt};

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidtstatic void unit_gc_sweep(Unit *u, unsigned gc_marker) {

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt Iterator i;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt Unit *other;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt bool is_bad;

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt assert(u);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (u->gc_marker == gc_marker + GC_OFFSET_GOOD ||

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt u->gc_marker == gc_marker + GC_OFFSET_BAD ||

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt u->gc_marker == gc_marker + GC_OFFSET_IN_PATH)

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt return;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt if (u->in_cleanup_queue)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt goto bad;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (unit_check_gc(u))

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt goto good;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt u->gc_marker = gc_marker + GC_OFFSET_IN_PATH;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt is_bad = true;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt SET_FOREACH(other, u->dependencies[UNIT_REFERENCED_BY], i) {

9ba81d5a61b7c992a1d2e5e02f334b8e2a0b0c22Michal Schmidt unit_gc_sweep(other, gc_marker);

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt if (other->gc_marker == gc_marker + GC_OFFSET_GOOD)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt goto good;

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt

8f88aed740ded77af443bb1b7c79bb229b50f8f8Michal Schmidt if (other->gc_marker != gc_marker + GC_OFFSET_BAD)

32a4456cc252689f51f383d150d34ed636bfec4dMichal Schmidt is_bad = false;

}

if (is_bad)

goto bad;

/* We were unable to find anything out about this entry, so

u->gc_marker = gc_marker + GC_OFFSET_UNSURE;

unit_add_to_gc_queue(u);

return;

bad:

/* We definitely know that this one is not useful anymore, so

u->gc_marker = gc_marker + GC_OFFSET_BAD;

unit_add_to_cleanup_queue(u);

return;

good:

u->gc_marker = gc_marker + GC_OFFSET_GOOD;

}

static unsigned manager_dispatch_gc_queue(Manager *m) {

Unit *u;

unsigned n = 0;

unsigned gc_marker;

assert(m);

/* log_debug("Running GC..."); */

m->gc_marker += _GC_OFFSET_MAX;

if (m->gc_marker + _GC_OFFSET_MAX <= _GC_OFFSET_MAX)

m->gc_marker = 1;

gc_marker = m->gc_marker;

while ((u = m->gc_queue)) {

assert(u->in_gc_queue);

unit_gc_sweep(u, gc_marker);

LIST_REMOVE(gc_queue, m->gc_queue, u);

u->in_gc_queue = false;

n++;

if (u->gc_marker == gc_marker + GC_OFFSET_BAD ||

u->gc_marker == gc_marker + GC_OFFSET_UNSURE) {

if (u->id)

log_unit_debug(u, "Collecting.");

u->gc_marker = gc_marker + GC_OFFSET_BAD;

unit_add_to_cleanup_queue(u);

}

}

m->n_in_gc_queue = 0;

return n;

}

static void manager_clear_jobs_and_units(Manager *m) {

Unit *u;

assert(m);

while ((u = hashmap_first(m->units)))

unit_free(u);

manager_dispatch_cleanup_queue(m);

assert(!m->load_queue);

assert(!m->run_queue);

assert(!m->dbus_unit_queue);

assert(!m->dbus_job_queue);

assert(!m->cleanup_queue);

assert(!m->gc_queue);

assert(hashmap_isempty(m->jobs));

assert(hashmap_isempty(m->units));

m->n_on_console = 0;

m->n_running_jobs = 0;

}

Manager* manager_free(Manager *m) {

UnitType c;

int i;

if (!m)

return NULL;

manager_clear_jobs_and_units(m);

for (c = 0; c < _UNIT_TYPE_MAX; c++)

if (unit_vtable[c]->shutdown)

unit_vtable[c]->shutdown(m);

/* If we reexecute ourselves, we keep the root cgroup

manager_shutdown_cgroup(m, m->exit_code != MANAGER_REEXECUTE);

manager_undo_generators(m);

bus_done(m);

hashmap_free(m->units);

hashmap_free(m->jobs);

hashmap_free(m->watch_pids1);

hashmap_free(m->watch_pids2);

hashmap_free(m->watch_bus);

set_free(m->startup_units);

set_free(m->failed_units);

sd_event_source_unref(m->signal_event_source);

sd_event_source_unref(m->notify_event_source);

sd_event_source_unref(m->time_change_event_source);

sd_event_source_unref(m->jobs_in_progress_event_source);

sd_event_source_unref(m->run_queue_event_source);

safe_close(m->signal_fd);

safe_close(m->notify_fd);

safe_close(m->time_change_fd);

safe_close(m->kdbus_fd);

manager_close_ask_password(m);

manager_close_idle_pipe(m);

udev_unref(m->udev);

sd_event_unref(m->event);

free(m->notify_socket);

lookup_paths_free(&m->lookup_paths);

strv_free(m->environment);

hashmap_free(m->cgroup_unit);

set_free_free(m->unit_path_cache);

hashmap_free(m->cgroup_netclass_registry);

free(m->switch_root);

free(m->switch_root_init);

for (i = 0; i < _RLIMIT_MAX; i++)

m->rlimit[i] = mfree(m->rlimit[i]);

assert(hashmap_isempty(m->units_requiring_mounts_for));

hashmap_free(m->units_requiring_mounts_for);

free(m);

return NULL;

}

void manager_enumerate(Manager *m) {

UnitType c;

assert(m);

/* Let's ask every type to load all units from disk/kernel

for (c = 0; c < _UNIT_TYPE_MAX; c++) {

if (!unit_type_supported(c)) {

log_debug("Unit type .%s is not supported on this system.", unit_type_to_string(c));

continue;

}

if (!unit_vtable[c]->enumerate)

continue;

unit_vtable[c]->enumerate(m);

}

manager_dispatch_load_queue(m);

}

static void manager_coldplug(Manager *m) {

Iterator i;

Unit *u;

char *k;

int r;

assert(m);

/* Then, let's set up their initial state. */

HASHMAP_FOREACH_KEY(u, k, m->units, i) {

/* ignore aliases */

if (u->id != k)

continue;

r = unit_coldplug(u);

if (r < 0)

log_warning_errno(r, "We couldn't coldplug %s, proceeding anyway: %m", u->id);

}

}

static void manager_build_unit_path_cache(Manager *m) {

char **i;

_cleanup_closedir_ DIR *d = NULL;

int r;

assert(m);

set_free_free(m->unit_path_cache);

m->unit_path_cache = set_new(&string_hash_ops);

if (!m->unit_path_cache) {

log_error("Failed to allocate unit path cache.");

return;

}

/* This simply builds a list of files we know exist, so that

STRV_FOREACH(i, m->lookup_paths.unit_path) {

struct dirent *de;

d = opendir(*i);

if (!d) {

if (errno != ENOENT)

log_error_errno(errno, "Failed to open directory %s: %m", *i);

continue;

}

while ((de = readdir(d))) {

char *p;

if (hidden_file(de->d_name))

continue;

p = strjoin(streq(*i, "/") ? "" : *i, "/", de->d_name, NULL);

if (!p) {

r = -ENOMEM;

goto fail;

}

r = set_consume(m->unit_path_cache, p);

if (r < 0)

goto fail;

}

d = safe_closedir(d);

}

return;

fail:

log_error_errno(r, "Failed to build unit path cache: %m");

set_free_free(m->unit_path_cache);

m->unit_path_cache = NULL;

}

static void manager_distribute_fds(Manager *m, FDSet *fds) {

Iterator i;

Unit *u;

assert(m);

HASHMAP_FOREACH(u, m->units, i) {

if (fdset_size(fds) <= 0)

break;

if (!UNIT_VTABLE(u)->distribute_fds)

continue;

UNIT_VTABLE(u)->distribute_fds(u, fds);

}

}

int manager_startup(Manager *m, FILE *serialization, FDSet *fds) {

int r, q;

assert(m);

dual_timestamp_get(&m->generators_start_timestamp);

r = manager_run_generators(m);

dual_timestamp_get(&m->generators_finish_timestamp);

if (r < 0)

return r;

r = lookup_paths_init(

&m->lookup_paths, m->running_as, true,

NULL,

m->generator_unit_path,

m->generator_unit_path_early,

m->generator_unit_path_late);

if (r < 0)

return r;

manager_build_unit_path_cache(m);

/* If we will deserialize make sure that during enumeration

if (serialization)

m->n_reloading ++;

/* First, enumerate what we can from all config files */

dual_timestamp_get(&m->units_load_start_timestamp);

manager_enumerate(m);

dual_timestamp_get(&m->units_load_finish_timestamp);

/* Second, deserialize if there is something to deserialize */

if (serialization)

r = manager_deserialize(m, serialization, fds);

/* Any fds left? Find some unit which wants them. This is

manager_distribute_fds(m, fds);

/* We might have deserialized the notify fd, but if we didn't

q = manager_setup_notify(m);

if (q < 0 && r == 0)

r = q;

/* We might have deserialized the kdbus control fd, but if we

manager_setup_kdbus(m);

manager_connect_bus(m, !!serialization);

bus_track_coldplug(m, &m->subscribed, &m->deserialized_subscribed);

/* Third, fire things up! */

manager_coldplug(m);

if (serialization) {

assert(m->n_reloading > 0);

m->n_reloading --;

/* Let's wait for the UnitNew/JobNew messages being

m->send_reloading_done = true;

}

return r;

}

int manager_add_job(Manager *m, JobType type, Unit *unit, JobMode mode, sd_bus_error *e, Job **_ret) {

int r;

Transaction *tr;

assert(m);

assert(type < _JOB_TYPE_MAX);

assert(unit);

assert(mode < _JOB_MODE_MAX);

if (mode == JOB_ISOLATE && type != JOB_START)

return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Isolate is only valid for start.");

if (mode == JOB_ISOLATE && !unit->allow_isolate)

return sd_bus_error_setf(e, BUS_ERROR_NO_ISOLATION, "Operation refused, unit may not be isolated.");

log_unit_debug(unit, "Trying to enqueue job %s/%s/%s", unit->id, job_type_to_string(type), job_mode_to_string(mode));

type = job_type_collapse(type, unit);

tr = transaction_new(mode == JOB_REPLACE_IRREVERSIBLY);

if (!tr)

return -ENOMEM;

r = transaction_add_job_and_dependencies(tr, type, unit, NULL, true, false,

mode == JOB_IGNORE_DEPENDENCIES || mode == JOB_IGNORE_REQUIREMENTS,

mode == JOB_IGNORE_DEPENDENCIES, e);

if (r < 0)

goto tr_abort;

if (mode == JOB_ISOLATE) {

r = transaction_add_isolate_jobs(tr, m);

if (r < 0)

goto tr_abort;

}

r = transaction_activate(tr, m, mode, e);

if (r < 0)

goto tr_abort;

log_unit_debug(unit,

"Enqueued job %s/%s as %u", unit->id,

job_type_to_string(type), (unsigned) tr->anchor_job->id);

if (_ret)

*_ret = tr->anchor_job;

transaction_free(tr);

return 0;

tr_abort:

transaction_abort(tr);

transaction_free(tr);

return r;

}

int manager_add_job_by_name(Manager *m, JobType type, const char *name, JobMode mode, sd_bus_error *e, Job **ret) {

Unit *unit;

int r;

assert(m);

assert(type < _JOB_TYPE_MAX);

assert(name);

assert(mode < _JOB_MODE_MAX);

r = manager_load_unit(m, name, NULL, NULL, &unit);

if (r < 0)

return r;

return manager_add_job(m, type, unit, mode, e, ret);

}

int manager_add_job_by_name_and_warn(Manager *m, JobType type, const char *name, JobMode mode, Job **ret) {

_cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;

int r;

assert(m);

assert(type < _JOB_TYPE_MAX);

assert(name);

assert(mode < _JOB_MODE_MAX);

r = manager_add_job_by_name(m, type, name, mode, &error, ret);

if (r < 0)

return log_warning_errno(r, "Failed to enqueue %s job for %s: %s", job_mode_to_string(mode), name, bus_error_message(&error, r));

return r;

}

Job *manager_get_job(Manager *m, uint32_t id) {

assert(m);

return hashmap_get(m->jobs, UINT32_TO_PTR(id));

}

Unit *manager_get_unit(Manager *m, const char *name) {

assert(m);

assert(name);

return hashmap_get(m->units, name);

}

unsigned manager_dispatch_load_queue(Manager *m) {

Unit *u;

unsigned n = 0;

assert(m);

/* Make sure we are not run recursively */

if (m->dispatching_load_queue)

return 0;

m->dispatching_load_queue = true;

/* Dispatches the load queue. Takes a unit from the queue and

while ((u = m->load_queue)) {

assert(u->in_load_queue);

unit_load(u);

n++;

}

m->dispatching_load_queue = false;

return n;

}

int manager_load_unit_prepare(

Manager *m,

const char *name,

const char *path,

sd_bus_error *e,

Unit **_ret) {

Unit *ret;

UnitType t;

int r;

assert(m);

assert(name || path);

/* This will prepare the unit for loading, but not actually

if (path && !is_path(path))

return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Path %s is not absolute.", path);

if (!name)

name = basename(path);

t = unit_name_to_type(name);

if (t == _UNIT_TYPE_INVALID || !unit_name_is_valid(name, UNIT_NAME_PLAIN|UNIT_NAME_INSTANCE))

return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Unit name %s is not valid.", name);

ret = manager_get_unit(m, name);

if (ret) {

*_ret = ret;

return 1;

}

ret = unit_new(m, unit_vtable[t]->object_size);

if (!ret)

return -ENOMEM;

if (path) {

ret->fragment_path = strdup(path);

if (!ret->fragment_path) {

unit_free(ret);

return -ENOMEM;

}

}

r = unit_add_name(ret, name);

if (r < 0) {

unit_free(ret);

return r;

}

unit_add_to_load_queue(ret);

unit_add_to_dbus_queue(ret);

unit_add_to_gc_queue(ret);

if (_ret)

*_ret = ret;

return 0;

}

int manager_load_unit(

Manager *m,

const char *name,

const char *path,

sd_bus_error *e,

Unit **_ret) {

int r;

assert(m);

/* This will load the service information files, but not actually

r = manager_load_unit_prepare(m, name, path, e, _ret);

if (r != 0)

return r;

manager_dispatch_load_queue(m);

if (_ret)

*_ret = unit_follow_merge(*_ret);

return 0;