src/core/dbus-scope.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2013 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU Lesser General Public License as published by
  10   the Free Software Foundation; either version 2.1 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   Lesser General Public License for more details.
  17
  18   You should have received a copy of the GNU Lesser General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include "unit.h"
  23 #include "scope.h"
  24 #include "dbus-unit.h"
  25 #include "dbus-cgroup.h"
  26 #include "dbus-kill.h"
  27 #include "dbus-scope.h"
  28 #include "bus-util.h"
  29 #include "bus-internal.h"
  30 #include "bus-errors.h"
  31
  32 static int bus_scope_abandon(sd_bus *bus, sd_bus_message *message, void *userdata, sd_bus_error *error) {
  33         Scope *s = userdata;
  34         int r;
  35
  36         assert(bus);
  37         assert(message);
  38         assert(s);
  39
  40         r = scope_abandon(s);
  41         if (sd_bus_error_is_set(error))
  42                 return r;
  43
  44         if (r == -ESTALE)
  45                 return sd_bus_error_setf(error, BUS_ERROR_SCOPE_NOT_RUNNING, "Scope %s is not running, cannot abandon.", UNIT(s)->id);
  46
  47         return sd_bus_reply_method_return(message, NULL);
  48 }
  49
  50 static BUS_DEFINE_PROPERTY_GET_ENUM(property_get_result, scope_result, ScopeResult);
  51
  52 const sd_bus_vtable bus_scope_vtable[] = {
  53         SD_BUS_VTABLE_START(0),
  54         SD_BUS_PROPERTY("Controller", "s", NULL, offsetof(Scope, controller), SD_BUS_VTABLE_PROPERTY_CONST),
  55         SD_BUS_PROPERTY("TimeoutStopUSec", "t", bus_property_get_usec, offsetof(Scope, timeout_stop_usec), SD_BUS_VTABLE_PROPERTY_CONST),
  56         SD_BUS_PROPERTY("Result", "s", property_get_result, offsetof(Scope, result), SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
  57         SD_BUS_SIGNAL("RequestStop", NULL, 0),
  58         SD_BUS_METHOD("Abandon", NULL, NULL, bus_scope_abandon, 0),
  59         SD_BUS_VTABLE_END
  60 };
  61
  62 static int bus_scope_set_transient_property(
  63                 Scope *s,
  64                 const char *name,
  65                 sd_bus_message *message,
  66                 UnitSetPropertiesMode mode,
  67                 sd_bus_error *error) {
  68
  69         int r;
  70
  71         assert(s);
  72         assert(name);
  73         assert(message);
  74
  75         if (streq(name, "PIDs")) {
  76                 unsigned n = 0;
  77                 uint32_t pid;
  78
  79                 r = sd_bus_message_enter_container(message, 'a', "u");
  80                 if (r < 0)
  81                         return r;
  82
  83                 while ((r = sd_bus_message_read(message, "u", &pid)) > 0) {
  84
  85                         if (pid <= 1)
  86                                 return -EINVAL;
  87
  88                         if (mode != UNIT_CHECK) {
  89                                 r = unit_watch_pid(UNIT(s), pid);
  90                                 if (r < 0 && r != -EEXIST)
  91                                         return r;
  92                         }
  93
  94                         n++;
  95                 }
  96                 if (r < 0)
  97                         return r;
  98
  99                 r = sd_bus_message_exit_container(message);
 100                 if (r < 0)
 101                         return r;
 102
 103                 if (n <= 0)
 104                         return -EINVAL;
 105
 106                 return 1;
 107
 108         } else if (streq(name, "Controller")) {
 109                 const char *controller;
 110                 char *c;
 111
 112                 r = sd_bus_message_read(message, "s", &controller);
 113                 if (r < 0)
 114                         return r;
 115
 116                 if (!isempty(controller) && !service_name_is_valid(controller))
 117                         return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Controller '%s' is not a valid bus name.", controller);
 118
 119                 if (mode != UNIT_CHECK) {
 120                         if (isempty(controller))
 121                                 c = NULL;
 122                         else {
 123                                 c = strdup(controller);
 124                                 if (!c)
 125                                         return -ENOMEM;
 126                         }
 127
 128                         free(s->controller);
 129                         s->controller = c;
 130                 }
 131
 132                 return 1;
 133
 134         } else if (streq(name, "TimeoutStopUSec")) {
 135
 136                 if (mode != UNIT_CHECK) {
 137                         r = sd_bus_message_read(message, "t", &s->timeout_stop_usec);
 138                         if (r < 0)
 139                                 return r;
 140
 141                         unit_write_drop_in_format(UNIT(s), mode, name, "[Scope]\nTimeoutStopSec=%lluus\n", (unsigned long long) s->timeout_stop_usec);
 142                 } else {
 143                         r = sd_bus_message_skip(message, "t");
 144                         if (r < 0)
 145                                 return r;
 146                 }
 147
 148                 return 1;
 149         }
 150
 151         return 0;
 152 }
 153
 154 int bus_scope_set_property(
 155                 Unit *u,
 156                 const char *name,
 157                 sd_bus_message *message,
 158                 UnitSetPropertiesMode mode,
 159                 sd_bus_error *error) {
 160
 161         Scope *s = SCOPE(u);
 162         int r;
 163
 164         assert(s);
 165         assert(name);
 166         assert(message);
 167
 168         r = bus_cgroup_set_property(u, &s->cgroup_context, name, message, mode, error);
 169         if (r != 0)
 170                 return r;
 171
 172         if (u->load_state == UNIT_STUB) {
 173                 /* While we are created we still accept PIDs */
 174
 175                 r = bus_scope_set_transient_property(s, name, message, mode, error);
 176                 if (r != 0)
 177                         return r;
 178
 179                 r = bus_kill_context_set_transient_property(u, &s->kill_context, name, message, mode, error);
 180                 if (r != 0)
 181                         return r;
 182         }
 183
 184         return 0;
 185 }
 186
 187 int bus_scope_commit_properties(Unit *u) {
 188         assert(u);
 189
 190         unit_update_cgroup_members_masks(u);
 191         unit_realize_cgroup(u);
 192
 193         return 0;
 194 }
 195
 196 int bus_scope_send_request_stop(Scope *s) {
 197         _cleanup_bus_message_unref_ sd_bus_message *m = NULL;
 198         _cleanup_free_ char *p = NULL;
 199         int r;
 200
 201         assert(s);
 202
 203         if (!s->controller)
 204                 return 0;
 205
 206         p = unit_dbus_path(UNIT(s));
 207         if (!p)
 208                 return -ENOMEM;
 209
 210         r = sd_bus_message_new_signal(
 211                         UNIT(s)->manager->api_bus,
 212                         p,
 213                         "org.freedesktop.systemd1.Scope",
 214                         "RequestStop",
 215                         &m);
 216         if (r < 0)
 217                 return r;
 218
 219         return sd_bus_send_to(UNIT(s)->manager->api_bus, m, /* s->controller */ NULL, NULL);
 220 }