| /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ |
| |
| /*** |
| This file is part of systemd. |
| |
| Copyright 2010 Lennart Poettering |
| |
| systemd is free software; you can redistribute it and/or modify it |
| under the terms of the GNU General Public License as published by |
| the Free Software Foundation; either version 2 of the License, or |
| (at your option) any later version. |
| |
| systemd is distributed in the hope that it will be useful, but |
| WITHOUT ANY WARRANTY; without even the implied warranty of |
| MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| General Public License for more details. |
| |
| You should have received a copy of the GNU General Public License |
| along with systemd; If not, see <http://www.gnu.org/licenses/>. |
| ***/ |
| |
| #include <errno.h> |
| |
| #include "dbus.h" |
| #include "log.h" |
| #include "dbus-unit.h" |
| #include "bus-errors.h" |
| #include "dbus-common.h" |
| |
| const char bus_unit_interface[] _introspect_("Unit") = BUS_UNIT_INTERFACE; |
| |
| #define INVALIDATING_PROPERTIES \ |
| "LoadState\0" \ |
| "ActiveState\0" \ |
| "SubState\0" \ |
| "InactiveExitTimestamp\0" \ |
| "ActiveEnterTimestamp\0" \ |
| "ActiveExitTimestamp\0" \ |
| "InactiveEnterTimestamp\0" \ |
| "Job\0" \ |
| "NeedDaemonReload\0" |
| |
| int bus_unit_append_names(DBusMessageIter *i, const char *property, void *data) { |
| char *t; |
| Iterator j; |
| DBusMessageIter sub; |
| Unit *u = data; |
| |
| if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "s", &sub)) |
| return -ENOMEM; |
| |
| SET_FOREACH(t, u->meta.names, j) |
| if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &t)) |
| return -ENOMEM; |
| |
| if (!dbus_message_iter_close_container(i, &sub)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_following(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data, *f; |
| const char *d; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| f = unit_following(u); |
| d = f ? f->meta.id : ""; |
| |
| if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &d)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_dependencies(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u; |
| Iterator j; |
| DBusMessageIter sub; |
| Set *s = data; |
| |
| if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "s", &sub)) |
| return -ENOMEM; |
| |
| SET_FOREACH(u, s, j) |
| if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &u->meta.id)) |
| return -ENOMEM; |
| |
| if (!dbus_message_iter_close_container(i, &sub)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_description(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| const char *d; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| d = unit_description(u); |
| |
| if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &d)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_unit_append_load_state, unit_load_state, UnitLoadState); |
| |
| int bus_unit_append_active_state(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| const char *state; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| state = unit_active_state_to_string(unit_active_state(u)); |
| |
| if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &state)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_sub_state(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| const char *state; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| state = unit_sub_state_to_string(u); |
| |
| if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &state)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_can_start(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| dbus_bool_t b; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| b = unit_can_start(u) && |
| !u->meta.refuse_manual_start; |
| |
| if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_can_stop(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| dbus_bool_t b; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| /* On the lower levels we assume that every unit we can start |
| * we can also stop */ |
| |
| b = unit_can_start(u) && |
| !u->meta.refuse_manual_stop; |
| |
| if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_can_reload(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| dbus_bool_t b; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| b = unit_can_reload(u); |
| |
| if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_can_isolate(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| dbus_bool_t b; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| b = unit_can_isolate(u) && |
| !u->meta.refuse_manual_start; |
| |
| if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_job(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| DBusMessageIter sub; |
| char *p; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| if (!dbus_message_iter_open_container(i, DBUS_TYPE_STRUCT, NULL, &sub)) |
| return -ENOMEM; |
| |
| if (u->meta.job) { |
| |
| if (!(p = job_dbus_path(u->meta.job))) |
| return -ENOMEM; |
| |
| if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_UINT32, &u->meta.job->id) || |
| !dbus_message_iter_append_basic(&sub, DBUS_TYPE_OBJECT_PATH, &p)) { |
| free(p); |
| return -ENOMEM; |
| } |
| } else { |
| uint32_t id = 0; |
| |
| /* No job, so let's fill in some placeholder |
| * data. Since we need to fill in a valid path we |
| * simple point to ourselves. */ |
| |
| if (!(p = unit_dbus_path(u))) |
| return -ENOMEM; |
| |
| if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_UINT32, &id) || |
| !dbus_message_iter_append_basic(&sub, DBUS_TYPE_OBJECT_PATH, &p)) { |
| free(p); |
| return -ENOMEM; |
| } |
| } |
| |
| free(p); |
| |
| if (!dbus_message_iter_close_container(i, &sub)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_default_cgroup(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| char *t; |
| CGroupBonding *cgb; |
| bool success; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| if ((cgb = unit_get_default_cgroup(u))) { |
| if (!(t = cgroup_bonding_to_string(cgb))) |
| return -ENOMEM; |
| } else |
| t = (char*) ""; |
| |
| success = dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &t); |
| |
| if (cgb) |
| free(t); |
| |
| return success ? 0 : -ENOMEM; |
| } |
| |
| int bus_unit_append_cgroups(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| CGroupBonding *cgb; |
| DBusMessageIter sub; |
| |
| if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "s", &sub)) |
| return -ENOMEM; |
| |
| LIST_FOREACH(by_unit, cgb, u->meta.cgroup_bondings) { |
| char *t; |
| bool success; |
| |
| if (!(t = cgroup_bonding_to_string(cgb))) |
| return -ENOMEM; |
| |
| success = dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &t); |
| free(t); |
| |
| if (!success) |
| return -ENOMEM; |
| } |
| |
| if (!dbus_message_iter_close_container(i, &sub)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| int bus_unit_append_need_daemon_reload(DBusMessageIter *i, const char *property, void *data) { |
| Unit *u = data; |
| dbus_bool_t b; |
| |
| assert(i); |
| assert(property); |
| assert(u); |
| |
| b = unit_need_daemon_reload(u); |
| |
| if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &b)) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| static DBusHandlerResult bus_unit_message_dispatch(Unit *u, DBusConnection *connection, DBusMessage *message) { |
| DBusMessage *reply = NULL; |
| Manager *m = u->meta.manager; |
| DBusError error; |
| JobType job_type = _JOB_TYPE_INVALID; |
| char *path = NULL; |
| bool reload_if_possible = false; |
| |
| dbus_error_init(&error); |
| |
| if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Start")) |
| job_type = JOB_START; |
| else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Stop")) |
| job_type = JOB_STOP; |
| else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Reload")) |
| job_type = JOB_RELOAD; |
| else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Restart")) |
| job_type = JOB_RESTART; |
| else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "TryRestart")) |
| job_type = JOB_TRY_RESTART; |
| else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "ReloadOrRestart")) { |
| reload_if_possible = true; |
| job_type = JOB_RESTART; |
| } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "ReloadOrTryRestart")) { |
| reload_if_possible = true; |
| job_type = JOB_TRY_RESTART; |
| } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Kill")) { |
| const char *swho, *smode; |
| int32_t signo; |
| KillMode mode; |
| KillWho who; |
| int r; |
| |
| if (!dbus_message_get_args( |
| message, |
| &error, |
| DBUS_TYPE_STRING, &swho, |
| DBUS_TYPE_STRING, &smode, |
| DBUS_TYPE_INT32, &signo, |
| DBUS_TYPE_INVALID)) |
| return bus_send_error_reply(connection, message, &error, -EINVAL); |
| |
| if (isempty(swho)) |
| who = KILL_ALL; |
| else { |
| who = kill_who_from_string(swho); |
| if (who < 0) |
| return bus_send_error_reply(connection, message, &error, -EINVAL); |
| } |
| |
| if (isempty(smode)) |
| mode = KILL_CONTROL_GROUP; |
| else { |
| mode = kill_mode_from_string(smode); |
| if (mode < 0) |
| return bus_send_error_reply(connection, message, &error, -EINVAL); |
| } |
| |
| if (signo <= 0 || signo >= _NSIG) |
| return bus_send_error_reply(connection, message, &error, -EINVAL); |
| |
| if ((r = unit_kill(u, who, mode, signo, &error)) < 0) |
| return bus_send_error_reply(connection, message, &error, r); |
| |
| if (!(reply = dbus_message_new_method_return(message))) |
| goto oom; |
| |
| } else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "ResetFailed")) { |
| |
| unit_reset_failed(u); |
| |
| if (!(reply = dbus_message_new_method_return(message))) |
| goto oom; |
| |
| } else if (UNIT_VTABLE(u)->bus_message_handler) |
| return UNIT_VTABLE(u)->bus_message_handler(u, connection, message); |
| else |
| return DBUS_HANDLER_RESULT_NOT_YET_HANDLED; |
| |
| if (job_type != _JOB_TYPE_INVALID) { |
| const char *smode; |
| JobMode mode; |
| Job *j; |
| int r; |
| |
| if ((job_type == JOB_START && u->meta.refuse_manual_start) || |
| (job_type == JOB_STOP && u->meta.refuse_manual_stop) || |
| ((job_type == JOB_RESTART || job_type == JOB_TRY_RESTART) && |
| (u->meta.refuse_manual_start || u->meta.refuse_manual_stop))) { |
| dbus_set_error(&error, BUS_ERROR_ONLY_BY_DEPENDENCY, "Operation refused, may be requested by dependency only."); |
| return bus_send_error_reply(connection, message, &error, -EPERM); |
| } |
| |
| if (!dbus_message_get_args( |
| message, |
| &error, |
| DBUS_TYPE_STRING, &smode, |
| DBUS_TYPE_INVALID)) |
| return bus_send_error_reply(connection, message, &error, -EINVAL); |
| |
| if (reload_if_possible && unit_can_reload(u)) { |
| if (job_type == JOB_RESTART) |
| job_type = JOB_RELOAD_OR_START; |
| else if (job_type == JOB_TRY_RESTART) |
| job_type = JOB_RELOAD; |
| } |
| |
| if ((mode = job_mode_from_string(smode)) == _JOB_MODE_INVALID) { |
| dbus_set_error(&error, BUS_ERROR_INVALID_JOB_MODE, "Job mode %s is invalid.", smode); |
| return bus_send_error_reply(connection, message, &error, -EINVAL); |
| } |
| |
| if ((r = manager_add_job(m, job_type, u, mode, true, &error, &j)) < 0) |
| return bus_send_error_reply(connection, message, &error, r); |
| |
| if (!(reply = dbus_message_new_method_return(message))) |
| goto oom; |
| |
| if (!(path = job_dbus_path(j))) |
| goto oom; |
| |
| if (!dbus_message_append_args( |
| reply, |
| DBUS_TYPE_OBJECT_PATH, &path, |
| DBUS_TYPE_INVALID)) |
| goto oom; |
| } |
| |
| if (reply) { |
| if (!dbus_connection_send(connection, reply, NULL)) |
| goto oom; |
| |
| dbus_message_unref(reply); |
| } |
| |
| free(path); |
| |
| return DBUS_HANDLER_RESULT_HANDLED; |
| |
| oom: |
| free(path); |
| |
| if (reply) |
| dbus_message_unref(reply); |
| |
| dbus_error_free(&error); |
| |
| return DBUS_HANDLER_RESULT_NEED_MEMORY; |
| } |
| |
| static DBusHandlerResult bus_unit_message_handler(DBusConnection *connection, DBusMessage *message, void *data) { |
| Manager *m = data; |
| Unit *u; |
| int r; |
| DBusMessage *reply; |
| |
| assert(connection); |
| assert(message); |
| assert(m); |
| |
| if (streq(dbus_message_get_path(message), "/org/freedesktop/systemd1/unit")) { |
| /* Be nice to gdbus and return introspection data for our mid-level paths */ |
| |
| if (dbus_message_is_method_call(message, "org.freedesktop.DBus.Introspectable", "Introspect")) { |
| char *introspection = NULL; |
| FILE *f; |
| Iterator i; |
| const char *k; |
| size_t size; |
| |
| if (!(reply = dbus_message_new_method_return(message))) |
| goto oom; |
| |
| /* We roll our own introspection code here, instead of |
| * relying on bus_default_message_handler() because we |
| * need to generate our introspection string |
| * dynamically. */ |
| |
| if (!(f = open_memstream(&introspection, &size))) |
| goto oom; |
| |
| fputs(DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE |
| "<node>\n", f); |
| |
| fputs(BUS_INTROSPECTABLE_INTERFACE, f); |
| fputs(BUS_PEER_INTERFACE, f); |
| |
| HASHMAP_FOREACH_KEY(u, k, m->units, i) { |
| char *p; |
| |
| if (k != u->meta.id) |
| continue; |
| |
| if (!(p = bus_path_escape(k))) { |
| fclose(f); |
| free(introspection); |
| goto oom; |
| } |
| |
| fprintf(f, "<node name=\"%s\"/>", p); |
| free(p); |
| } |
| |
| fputs("</node>\n", f); |
| |
| if (ferror(f)) { |
| fclose(f); |
| free(introspection); |
| goto oom; |
| } |
| |
| fclose(f); |
| |
| if (!introspection) |
| goto oom; |
| |
| if (!dbus_message_append_args(reply, DBUS_TYPE_STRING, &introspection, DBUS_TYPE_INVALID)) { |
| free(introspection); |
| goto oom; |
| } |
| |
| free(introspection); |
| |
| if (!dbus_connection_send(connection, reply, NULL)) |
| goto oom; |
| |
| dbus_message_unref(reply); |
| |
| return DBUS_HANDLER_RESULT_HANDLED; |
| } |
| |
| return DBUS_HANDLER_RESULT_NOT_YET_HANDLED; |
| } |
| |
| if ((r = manager_get_unit_from_dbus_path(m, dbus_message_get_path(message), &u)) < 0) { |
| |
| if (r == -ENOMEM) |
| return DBUS_HANDLER_RESULT_NEED_MEMORY; |
| |
| if (r == -ENOENT) { |
| DBusError e; |
| |
| dbus_error_init(&e); |
| dbus_set_error_const(&e, DBUS_ERROR_UNKNOWN_OBJECT, "Unknown unit"); |
| return bus_send_error_reply(connection, message, &e, r); |
| } |
| |
| return bus_send_error_reply(connection, message, NULL, r); |
| } |
| |
| return bus_unit_message_dispatch(u, connection, message); |
| |
| oom: |
| if (reply) |
| dbus_message_unref(reply); |
| |
| return DBUS_HANDLER_RESULT_NEED_MEMORY; |
| } |
| |
| const DBusObjectPathVTable bus_unit_vtable = { |
| .message_function = bus_unit_message_handler |
| }; |
| |
| void bus_unit_send_change_signal(Unit *u) { |
| char *p = NULL; |
| DBusMessage *m = NULL; |
| |
| assert(u); |
| |
| if (u->meta.in_dbus_queue) { |
| LIST_REMOVE(Meta, dbus_queue, u->meta.manager->dbus_unit_queue, &u->meta); |
| u->meta.in_dbus_queue = false; |
| } |
| |
| if (!u->meta.id) |
| return; |
| |
| if (!bus_has_subscriber(u->meta.manager)) { |
| u->meta.sent_dbus_new_signal = true; |
| return; |
| } |
| |
| if (!(p = unit_dbus_path(u))) |
| goto oom; |
| |
| if (u->meta.sent_dbus_new_signal) { |
| /* Send a properties changed signal. First for the |
| * specific type, then for the generic unit. The |
| * clients may rely on this order to get atomic |
| * behaviour if needed. */ |
| |
| if (UNIT_VTABLE(u)->bus_invalidating_properties) { |
| |
| if (!(m = bus_properties_changed_new(p, |
| UNIT_VTABLE(u)->bus_interface, |
| UNIT_VTABLE(u)->bus_invalidating_properties))) |
| goto oom; |
| |
| if (bus_broadcast(u->meta.manager, m) < 0) |
| goto oom; |
| |
| dbus_message_unref(m); |
| } |
| |
| if (!(m = bus_properties_changed_new(p, "org.freedesktop.systemd1.Unit", INVALIDATING_PROPERTIES))) |
| goto oom; |
| |
| } else { |
| /* Send a new signal */ |
| |
| if (!(m = dbus_message_new_signal("/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "UnitNew"))) |
| goto oom; |
| |
| if (!dbus_message_append_args(m, |
| DBUS_TYPE_STRING, &u->meta.id, |
| DBUS_TYPE_OBJECT_PATH, &p, |
| DBUS_TYPE_INVALID)) |
| goto oom; |
| } |
| |
| if (bus_broadcast(u->meta.manager, m) < 0) |
| goto oom; |
| |
| free(p); |
| dbus_message_unref(m); |
| |
| u->meta.sent_dbus_new_signal = true; |
| |
| return; |
| |
| oom: |
| free(p); |
| |
| if (m) |
| dbus_message_unref(m); |
| |
| log_error("Failed to allocate unit change/new signal."); |
| } |
| |
| void bus_unit_send_removed_signal(Unit *u) { |
| char *p = NULL; |
| DBusMessage *m = NULL; |
| |
| assert(u); |
| |
| if (!bus_has_subscriber(u->meta.manager)) |
| return; |
| |
| if (!u->meta.sent_dbus_new_signal) |
| bus_unit_send_change_signal(u); |
| |
| if (!u->meta.id) |
| return; |
| |
| if (!(p = unit_dbus_path(u))) |
| goto oom; |
| |
| if (!(m = dbus_message_new_signal("/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "UnitRemoved"))) |
| goto oom; |
| |
| if (!dbus_message_append_args(m, |
| DBUS_TYPE_STRING, &u->meta.id, |
| DBUS_TYPE_OBJECT_PATH, &p, |
| DBUS_TYPE_INVALID)) |
| goto oom; |
| |
| if (bus_broadcast(u->meta.manager, m) < 0) |
| goto oom; |
| |
| free(p); |
| dbus_message_unref(m); |
| |
| return; |
| |
| oom: |
| free(p); |
| |
| if (m) |
| dbus_message_unref(m); |
| |
| log_error("Failed to allocate unit remove signal."); |
| } |