src/machine/machined.c

	machined.c revision c34255bdb217c2a1d3ac6348252437ab8be9ca46
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster/***
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  This file is part of systemd.
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  Copyright 2013 Lennart Poettering
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  systemd is free software; you can redistribute it and/or modify it
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  under the terms of the GNU Lesser General Public License as published by
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  the Free Software Foundation; either version 2.1 of the License, or
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  (at your option) any later version.
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  systemd is distributed in the hope that it will be useful, but
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  WITHOUT ANY WARRANTY; without even the implied warranty of
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  Lesser General Public License for more details.
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  You should have received a copy of the GNU Lesser General Public License
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster  along with systemd; If not, see <http://www.gnu.org/licenses/>.
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster***/
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include <errno.h>
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include <pwd.h>
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include <fcntl.h>
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include <string.h>
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include <unistd.h>
4505caaaadb03cc04b4ecb08d728ddcbf690b9e4Mark de Reeper#include <sys/epoll.h>
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include "sd-daemon.h"
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include "strv.h"
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include "conf-parser.h"
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include "cgroup-util.h"
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include "mkdir.h"
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include "bus-util.h"
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include "bus-error.h"
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include "machined.h"
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster#include "label.h"
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan FosterManager *manager_new(void) {
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        Manager *m;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        int r;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        m = new0(Manager, 1);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        if (!m)
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                return NULL;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        m->machines = hashmap_new(&string_hash_ops);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        m->machine_units = hashmap_new(&string_hash_ops);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        m->machine_leaders = hashmap_new(NULL);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        if (!m->machines || !m->machine_units || !m->machine_leaders) {
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                manager_free(m);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                return NULL;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        }
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        r = sd_event_default(&m->event);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        if (r < 0) {
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                manager_free(m);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                return NULL;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        }
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        sd_event_set_watchdog(m->event, true);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        return m;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster}
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Fostervoid manager_free(Manager *m) {
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        Machine *machine;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        assert(m);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        while ((machine = hashmap_first(m->machines)))
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                machine_free(machine);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        hashmap_free(m->machines);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        hashmap_free(m->machine_units);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        hashmap_free(m->machine_leaders);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        sd_bus_unref(m->bus);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        sd_event_unref(m->event);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        free(m);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster}
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Fosterint manager_enumerate_machines(Manager *m) {
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        _cleanup_closedir_ DIR *d = NULL;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        struct dirent *de;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        int r = 0;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        assert(m);
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        /* Read in machine data stored on disk */
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        d = opendir("/run/systemd/machines");
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        if (!d) {
4505caaaadb03cc04b4ecb08d728ddcbf690b9e4Mark de Reeper                if (errno == ENOENT)
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                        return 0;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                log_error("Failed to open /run/systemd/machines: %m");
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                return -errno;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        }
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster        FOREACH_DIRENT(de, d, return -errno) {
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                struct Machine *machine;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                int k;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                if (!dirent_is_file(de))
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                        continue;
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster
4fe4e4f798a84a46e567f64ceadd3648eb0582d4Allan Foster                /* Ignore symlinks that map the unit name to the machine */
                if (startswith(de->d_name, "unit:"))
                        continue;

                k = manager_add_machine(m, de->d_name, &machine);
                if (k < 0) {
                        log_error("Failed to add machine by file name %s: %s", de->d_name, strerror(-k));

                        r = k;
                        continue;
                }

                machine_add_to_gc_queue(machine);

                k = machine_load(machine);
                if (k < 0)
                        r = k;
        }

        return r;
}

static int manager_connect_bus(Manager *m) {
        _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
        int r;

        assert(m);
        assert(!m->bus);

        r = sd_bus_default_system(&m->bus);
        if (r < 0) {
                log_error("Failed to connect to system bus: %s", strerror(-r));
                return r;
        }

        r = sd_bus_add_object_vtable(m->bus, NULL, "/org/freedesktop/machine1", "org.freedesktop.machine1.Manager", manager_vtable, m);
        if (r < 0) {
                log_error("Failed to add manager object vtable: %s", strerror(-r));
                return r;
        }

        r = sd_bus_add_fallback_vtable(m->bus, NULL, "/org/freedesktop/machine1/machine", "org.freedesktop.machine1.Machine", machine_vtable, machine_object_find, m);
        if (r < 0) {
                log_error("Failed to add machine object vtable: %s", strerror(-r));
                return r;
        }

        r = sd_bus_add_node_enumerator(m->bus, NULL, "/org/freedesktop/machine1/machine", machine_node_enumerator, m);
        if (r < 0) {
                log_error("Failed to add machine enumerator: %s", strerror(-r));
                return r;
        }

        r = sd_bus_add_match(m->bus,
                             NULL,
                             "type='signal',"
                             "sender='org.freedesktop.systemd1',"
                             "interface='org.freedesktop.systemd1.Manager',"
                             "member='JobRemoved',"
                             "path='/org/freedesktop/systemd1'",
                             match_job_removed,
                             m);
        if (r < 0) {
                log_error("Failed to add match for JobRemoved: %s", strerror(-r));
                return r;
        }

        r = sd_bus_add_match(m->bus,
                             NULL,
                             "type='signal',"
                             "sender='org.freedesktop.systemd1',"
                             "interface='org.freedesktop.systemd1.Manager',"
                             "member='UnitRemoved',"
                             "path='/org/freedesktop/systemd1'",
                             match_unit_removed,
                             m);
        if (r < 0) {
                log_error("Failed to add match for UnitRemoved: %s", strerror(-r));
                return r;
        }

        r = sd_bus_add_match(m->bus,
                             NULL,
                             "type='signal',"
                             "sender='org.freedesktop.systemd1',"
                             "interface='org.freedesktop.DBus.Properties',"
                             "member='PropertiesChanged'",
                             match_properties_changed,
                             m);
        if (r < 0) {
                log_error("Failed to add match for PropertiesChanged: %s", strerror(-r));
                return r;
        }

        r = sd_bus_add_match(m->bus,
                             NULL,
                             "type='signal',"
                             "sender='org.freedesktop.systemd1',"
                             "interface='org.freedesktop.systemd1.Manager',"
                             "member='Reloading',"
                             "path='/org/freedesktop/systemd1'",
                             match_reloading,
                             m);
        if (r < 0) {
                log_error("Failed to add match for Reloading: %s", strerror(-r));
                return r;
        }

        r = sd_bus_call_method(
                        m->bus,
                        "org.freedesktop.systemd1",
                        "/org/freedesktop/systemd1",
                        "org.freedesktop.systemd1.Manager",
                        "Subscribe",
                        &error,
                        NULL, NULL);
        if (r < 0) {
                log_error("Failed to enable subscription: %s", bus_error_message(&error, r));
                return r;
        }

        r = sd_bus_request_name(m->bus, "org.freedesktop.machine1", 0);
        if (r < 0) {
                log_error("Failed to register name: %s", strerror(-r));
                return r;
        }

        r = sd_bus_attach_event(m->bus, m->event, 0);
        if (r < 0) {
                log_error("Failed to attach bus to event loop: %s", strerror(-r));
                return r;
        }

        return 0;
}

void manager_gc(Manager *m, bool drop_not_started) {
        Machine *machine;

        assert(m);

        while ((machine = m->machine_gc_queue)) {
                LIST_REMOVE(gc_queue, m->machine_gc_queue, machine);
                machine->in_gc_queue = false;

                if (!machine_check_gc(machine, drop_not_started)) {
                        machine_stop(machine);
                        machine_free(machine);
                }
        }
}

int manager_startup(Manager *m) {
        Machine *machine;
        Iterator i;
        int r;

        assert(m);

        /* Connect to the bus */
        r = manager_connect_bus(m);
        if (r < 0)
                return r;

        /* Deserialize state */
        manager_enumerate_machines(m);

        /* Remove stale objects before we start them */
        manager_gc(m, false);

        /* And start everything */
        HASHMAP_FOREACH(machine, m->machines, i)
                machine_start(machine, NULL, NULL);

        return 0;
}

static bool check_idle(void *userdata) {
        Manager *m = userdata;

        manager_gc(m, true);

        return hashmap_isempty(m->machines);
}

int manager_run(Manager *m) {
        assert(m);

        return bus_event_loop_with_idle(
                        m->event,
                        m->bus,
                        "org.freedesktop.machine1",
                        DEFAULT_EXIT_USEC,
                        check_idle, m);
}

int main(int argc, char *argv[]) {
        Manager *m = NULL;
        int r;

        log_set_target(LOG_TARGET_AUTO);
        log_set_facility(LOG_AUTH);
        log_parse_environment();
        log_open();

        umask(0022);

        if (argc != 1) {
                log_error("This program takes no arguments.");
                r = -EINVAL;
                goto finish;
        }

        /* Always create the directories people can create inotify
         * watches in. Note that some applications might check for the
         * existence of /run/systemd/machines/ to determine whether
         * machined is available, so please always make sure this
         * check stays in. */
        mkdir_label("/run/systemd/machines", 0755);

        m = manager_new();
        if (!m) {
                r = log_oom();
                goto finish;
        }

        r = manager_startup(m);
        if (r < 0) {
                log_error("Failed to fully start up daemon: %s", strerror(-r));
                goto finish;
        }

        log_debug("systemd-machined running as pid "PID_FMT, getpid());

        sd_notify(false,
                  "READY=1\n"
                  "STATUS=Processing requests...");

        r = manager_run(m);

        log_debug("systemd-machined stopped as pid "PID_FMT, getpid());

finish:
        if (m)
                manager_free(m);

        return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
}