chiark / gitweb /
Make sure that keys are properly removed from hashmap
[elogind.git] / 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="USEC_FMT"us\n", 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                         &m,
213                         p,
214                         "org.freedesktop.systemd1.Scope",
215                         "RequestStop");
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 }