| /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ |
| |
| /*** |
| This file is part of systemd. |
| |
| Copyright 2011 Lennart Poettering |
| |
| systemd is free software; you can redistribute it and/or modify it |
| under the terms of the GNU Lesser General Public License as published by |
| the Free Software Foundation; either version 2.1 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 |
| Lesser General Public License for more details. |
| |
| You should have received a copy of the GNU Lesser General Public License |
| along with systemd; If not, see <http://www.gnu.org/licenses/>. |
| ***/ |
| |
| #include <errno.h> |
| #include <string.h> |
| #include <sys/capability.h> |
| |
| #include "bus-util.h" |
| #include "strv.h" |
| #include "machine.h" |
| |
| static int property_get_id( |
| sd_bus *bus, |
| const char *path, |
| const char *interface, |
| const char *property, |
| sd_bus_message *reply, |
| void *userdata, |
| sd_bus_error *error) { |
| |
| Machine *m = userdata; |
| int r; |
| |
| assert(bus); |
| assert(reply); |
| assert(m); |
| |
| r = sd_bus_message_append_array(reply, 'y', &m->id, 16); |
| if (r < 0) |
| return r; |
| |
| return 1; |
| } |
| |
| static int property_get_state( |
| sd_bus *bus, |
| const char *path, |
| const char *interface, |
| const char *property, |
| sd_bus_message *reply, |
| void *userdata, |
| sd_bus_error *error) { |
| |
| Machine *m = userdata; |
| const char *state; |
| int r; |
| |
| assert(bus); |
| assert(reply); |
| assert(m); |
| |
| state = machine_state_to_string(machine_get_state(m)); |
| |
| r = sd_bus_message_append_basic(reply, 's', state); |
| if (r < 0) |
| return r; |
| |
| return 1; |
| } |
| |
| static BUS_DEFINE_PROPERTY_GET_ENUM(property_get_class, machine_class, MachineClass); |
| |
| static int method_terminate(sd_bus *bus, sd_bus_message *message, void *userdata, sd_bus_error *error) { |
| Machine *m = userdata; |
| int r; |
| |
| assert(bus); |
| assert(message); |
| assert(m); |
| |
| r = machine_stop(m); |
| if (r < 0) |
| return r; |
| |
| return sd_bus_reply_method_return(message, NULL); |
| } |
| |
| static int method_kill(sd_bus *bus, sd_bus_message *message, void *userdata, sd_bus_error *error) { |
| Machine *m = userdata; |
| const char *swho; |
| int32_t signo; |
| KillWho who; |
| int r; |
| |
| assert(bus); |
| assert(message); |
| assert(m); |
| |
| r = sd_bus_message_read(message, "si", &swho, &signo); |
| if (r < 0) |
| return r; |
| |
| if (isempty(swho)) |
| who = KILL_ALL; |
| else { |
| who = kill_who_from_string(swho); |
| if (who < 0) |
| return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid kill parameter '%s'", swho); |
| } |
| |
| if (signo <= 0 || signo >= _NSIG) |
| return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid signal %i", signo); |
| |
| r = machine_kill(m, who, signo); |
| if (r < 0) |
| return r; |
| |
| return sd_bus_reply_method_return(message, NULL); |
| } |
| |
| const sd_bus_vtable machine_vtable[] = { |
| SD_BUS_VTABLE_START(0), |
| SD_BUS_PROPERTY("Name", "s", NULL, offsetof(Machine, name), SD_BUS_VTABLE_PROPERTY_CONST), |
| SD_BUS_PROPERTY("Id", "ay", property_get_id, 0, SD_BUS_VTABLE_PROPERTY_CONST), |
| BUS_PROPERTY_DUAL_TIMESTAMP("Timestamp", offsetof(Machine, timestamp), SD_BUS_VTABLE_PROPERTY_CONST), |
| SD_BUS_PROPERTY("Service", "s", NULL, offsetof(Machine, service), SD_BUS_VTABLE_PROPERTY_CONST), |
| SD_BUS_PROPERTY("Unit", "s", NULL, offsetof(Machine, unit), SD_BUS_VTABLE_PROPERTY_CONST), |
| SD_BUS_PROPERTY("Scope", "s", NULL, offsetof(Machine, unit), SD_BUS_VTABLE_PROPERTY_CONST|SD_BUS_VTABLE_HIDDEN), |
| SD_BUS_PROPERTY("Leader", "u", NULL, offsetof(Machine, leader), SD_BUS_VTABLE_PROPERTY_CONST), |
| SD_BUS_PROPERTY("Class", "s", property_get_class, offsetof(Machine, class), SD_BUS_VTABLE_PROPERTY_CONST), |
| SD_BUS_PROPERTY("RootDirectory", "s", NULL, offsetof(Machine, root_directory), SD_BUS_VTABLE_PROPERTY_CONST), |
| SD_BUS_PROPERTY("State", "s", property_get_state, 0, 0), |
| SD_BUS_METHOD("Terminate", NULL, NULL, method_terminate, SD_BUS_VTABLE_CAPABILITY(CAP_KILL)), |
| SD_BUS_METHOD("Kill", "si", NULL, method_kill, SD_BUS_VTABLE_CAPABILITY(CAP_KILL)), |
| SD_BUS_VTABLE_END |
| }; |
| |
| int machine_object_find(sd_bus *bus, const char *path, const char *interface, void *userdata, void **found, sd_bus_error *error) { |
| Manager *m = userdata; |
| Machine *machine; |
| int r; |
| |
| assert(bus); |
| assert(path); |
| assert(interface); |
| assert(found); |
| assert(m); |
| |
| if (streq(path, "/org/freedesktop/machine1/machine/self")) { |
| _cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL; |
| sd_bus_message *message; |
| pid_t pid; |
| |
| message = sd_bus_get_current(bus); |
| if (!message) |
| return 0; |
| |
| r = sd_bus_query_sender_creds(message, SD_BUS_CREDS_PID, &creds); |
| if (r < 0) |
| return r; |
| |
| r = sd_bus_creds_get_pid(creds, &pid); |
| if (r < 0) |
| return r; |
| |
| r = manager_get_machine_by_pid(m, pid, &machine); |
| if (r <= 0) |
| return 0; |
| } else { |
| _cleanup_free_ char *e = NULL; |
| const char *p; |
| |
| p = startswith(path, "/org/freedesktop/machine1/machine/"); |
| if (!p) |
| return 0; |
| |
| e = sd_bus_label_unescape(p); |
| if (!e) |
| return -ENOMEM; |
| |
| machine = hashmap_get(m->machines, e); |
| if (!machine) |
| return 0; |
| } |
| |
| *found = machine; |
| return 1; |
| } |
| |
| char *machine_bus_path(Machine *m) { |
| _cleanup_free_ char *e = NULL; |
| |
| assert(m); |
| |
| e = sd_bus_label_escape(m->name); |
| if (!e) |
| return NULL; |
| |
| return strappend("/org/freedesktop/machine1/machine/", e); |
| } |
| |
| int machine_node_enumerator(sd_bus *bus, const char *path, void *userdata, char ***nodes, sd_bus_error *error) { |
| _cleanup_strv_free_ char **l = NULL; |
| Machine *machine = NULL; |
| Manager *m = userdata; |
| Iterator i; |
| int r; |
| |
| assert(bus); |
| assert(path); |
| assert(nodes); |
| |
| HASHMAP_FOREACH(machine, m->machines, i) { |
| char *p; |
| |
| p = machine_bus_path(machine); |
| if (!p) |
| return -ENOMEM; |
| |
| r = strv_push(&l, p); |
| if (r < 0) { |
| free(p); |
| return r; |
| } |
| } |
| |
| *nodes = l; |
| l = NULL; |
| |
| return 1; |
| } |
| |
| int machine_send_signal(Machine *m, bool new_machine) { |
| _cleanup_free_ char *p = NULL; |
| |
| assert(m); |
| |
| p = machine_bus_path(m); |
| if (!p) |
| return -ENOMEM; |
| |
| return sd_bus_emit_signal( |
| m->manager->bus, |
| "/org/freedesktop/machine1", |
| "org.freedesktop.machine1.Manager", |
| new_machine ? "MachineNew" : "MachineRemoved", |
| "so", m->name, p); |
| } |
| |
| int machine_send_create_reply(Machine *m, sd_bus_error *error) { |
| _cleanup_bus_message_unref_ sd_bus_message *c = NULL; |
| _cleanup_free_ char *p = NULL; |
| |
| assert(m); |
| |
| if (!m->create_message) |
| return 0; |
| |
| c = m->create_message; |
| m->create_message = NULL; |
| |
| if (error) |
| return sd_bus_reply_method_error(c, error); |
| |
| /* Update the machine state file before we notify the client |
| * about the result. */ |
| machine_save(m); |
| |
| p = machine_bus_path(m); |
| if (!p) |
| return -ENOMEM; |
| |
| return sd_bus_reply_method_return(c, "o", p); |
| } |