]>
Commit | Line | Data |
---|---|---|
ea430986 LP |
1 | /*-*- Mode: C; c-basic-offset: 8 -*-*/ |
2 | ||
a7334b09 LP |
3 | /*** |
4 | This file is part of systemd. | |
5 | ||
6 | Copyright 2010 Lennart Poettering | |
7 | ||
8 | systemd is free software; you can redistribute it and/or modify it | |
9 | under the terms of the GNU General Public License as published by | |
10 | the Free Software Foundation; either version 2 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 | General Public License for more details. | |
17 | ||
18 | You should have received a copy of the GNU General Public License | |
19 | along with systemd; If not, see <http://www.gnu.org/licenses/>. | |
20 | ***/ | |
21 | ||
ea430986 LP |
22 | #include <dbus/dbus.h> |
23 | ||
24 | #include <sys/epoll.h> | |
25 | #include <sys/timerfd.h> | |
26 | #include <errno.h> | |
27 | #include <unistd.h> | |
28 | ||
29 | #include "dbus.h" | |
30 | #include "log.h" | |
31 | #include "strv.h" | |
8e274523 | 32 | #include "cgroup.h" |
ea430986 | 33 | |
f278026d | 34 | static void api_bus_dispatch_status(DBusConnection *bus, DBusDispatchStatus status, void *data) { |
ea430986 LP |
35 | Manager *m = data; |
36 | ||
37 | assert(bus); | |
38 | assert(m); | |
2e317f52 LP |
39 | |
40 | if (!m->api_bus) | |
41 | return; | |
42 | ||
f278026d | 43 | assert(m->api_bus == bus); |
ea430986 | 44 | |
f278026d | 45 | m->request_api_bus_dispatch = status != DBUS_DISPATCH_COMPLETE; |
ea430986 LP |
46 | } |
47 | ||
8e274523 LP |
48 | static void system_bus_dispatch_status(DBusConnection *bus, DBusDispatchStatus status, void *data) { |
49 | Manager *m = data; | |
50 | ||
51 | assert(bus); | |
52 | assert(m); | |
2e317f52 LP |
53 | |
54 | if (!m->system_bus) | |
55 | return; | |
56 | ||
8e274523 LP |
57 | assert(m->system_bus == bus); |
58 | ||
59 | m->request_system_bus_dispatch = status != DBUS_DISPATCH_COMPLETE; | |
60 | } | |
61 | ||
ea430986 LP |
62 | static uint32_t bus_flags_to_events(DBusWatch *bus_watch) { |
63 | unsigned flags; | |
64 | uint32_t events = 0; | |
65 | ||
66 | assert(bus_watch); | |
67 | ||
68 | /* no watch flags for disabled watches */ | |
69 | if (!dbus_watch_get_enabled(bus_watch)) | |
70 | return 0; | |
71 | ||
72 | flags = dbus_watch_get_flags(bus_watch); | |
73 | ||
74 | if (flags & DBUS_WATCH_READABLE) | |
75 | events |= EPOLLIN; | |
76 | if (flags & DBUS_WATCH_WRITABLE) | |
77 | events |= EPOLLOUT; | |
78 | ||
79 | return events | EPOLLHUP | EPOLLERR; | |
80 | } | |
81 | ||
82 | static unsigned events_to_bus_flags(uint32_t events) { | |
83 | unsigned flags = 0; | |
84 | ||
85 | if (events & EPOLLIN) | |
86 | flags |= DBUS_WATCH_READABLE; | |
87 | if (events & EPOLLOUT) | |
88 | flags |= DBUS_WATCH_WRITABLE; | |
89 | if (events & EPOLLHUP) | |
90 | flags |= DBUS_WATCH_HANGUP; | |
91 | if (events & EPOLLERR) | |
92 | flags |= DBUS_WATCH_ERROR; | |
93 | ||
94 | return flags; | |
95 | } | |
96 | ||
97 | void bus_watch_event(Manager *m, Watch *w, int events) { | |
98 | assert(m); | |
99 | assert(w); | |
100 | ||
101 | /* This is called by the event loop whenever there is | |
102 | * something happening on D-Bus' file handles. */ | |
103 | ||
8e274523 | 104 | if (!dbus_watch_get_enabled(w->data.bus_watch)) |
ea430986 LP |
105 | return; |
106 | ||
107 | dbus_watch_handle(w->data.bus_watch, events_to_bus_flags(events)); | |
108 | } | |
109 | ||
110 | static dbus_bool_t bus_add_watch(DBusWatch *bus_watch, void *data) { | |
111 | Manager *m = data; | |
112 | Watch *w; | |
113 | struct epoll_event ev; | |
114 | ||
115 | assert(bus_watch); | |
116 | assert(m); | |
117 | ||
118 | if (!(w = new0(Watch, 1))) | |
119 | return FALSE; | |
120 | ||
121 | w->fd = dbus_watch_get_unix_fd(bus_watch); | |
122 | w->type = WATCH_DBUS_WATCH; | |
123 | w->data.bus_watch = bus_watch; | |
124 | ||
125 | zero(ev); | |
126 | ev.events = bus_flags_to_events(bus_watch); | |
127 | ev.data.ptr = w; | |
128 | ||
129 | if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, w->fd, &ev) < 0) { | |
130 | ||
131 | if (errno != EEXIST) { | |
132 | free(w); | |
133 | return FALSE; | |
134 | } | |
135 | ||
136 | /* Hmm, bloody D-Bus creates multiple watches on the | |
137 | * same fd. epoll() does not like that. As a dirty | |
138 | * hack we simply dup() the fd and hence get a second | |
139 | * one we can safely add to the epoll(). */ | |
140 | ||
141 | if ((w->fd = dup(w->fd)) < 0) { | |
142 | free(w); | |
143 | return FALSE; | |
144 | } | |
145 | ||
146 | if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, w->fd, &ev) < 0) { | |
147 | free(w); | |
148 | close_nointr_nofail(w->fd); | |
149 | return FALSE; | |
150 | } | |
151 | ||
152 | w->fd_is_dupped = true; | |
153 | } | |
154 | ||
155 | dbus_watch_set_data(bus_watch, w, NULL); | |
156 | ||
157 | return TRUE; | |
158 | } | |
159 | ||
160 | static void bus_remove_watch(DBusWatch *bus_watch, void *data) { | |
161 | Manager *m = data; | |
162 | Watch *w; | |
163 | ||
164 | assert(bus_watch); | |
165 | assert(m); | |
166 | ||
167 | if (!(w = dbus_watch_get_data(bus_watch))) | |
168 | return; | |
169 | ||
170 | assert(w->type == WATCH_DBUS_WATCH); | |
171 | assert_se(epoll_ctl(m->epoll_fd, EPOLL_CTL_DEL, w->fd, NULL) >= 0); | |
172 | ||
173 | if (w->fd_is_dupped) | |
174 | close_nointr_nofail(w->fd); | |
175 | ||
176 | free(w); | |
177 | } | |
178 | ||
179 | static void bus_toggle_watch(DBusWatch *bus_watch, void *data) { | |
180 | Manager *m = data; | |
181 | Watch *w; | |
182 | struct epoll_event ev; | |
183 | ||
184 | assert(bus_watch); | |
185 | assert(m); | |
186 | ||
187 | assert_se(w = dbus_watch_get_data(bus_watch)); | |
188 | assert(w->type == WATCH_DBUS_WATCH); | |
189 | ||
190 | zero(ev); | |
191 | ev.events = bus_flags_to_events(bus_watch); | |
192 | ev.data.ptr = w; | |
193 | ||
194 | assert_se(epoll_ctl(m->epoll_fd, EPOLL_CTL_MOD, w->fd, &ev) == 0); | |
195 | } | |
196 | ||
197 | static int bus_timeout_arm(Manager *m, Watch *w) { | |
198 | struct itimerspec its; | |
199 | ||
200 | assert(m); | |
201 | assert(w); | |
202 | ||
203 | zero(its); | |
204 | ||
205 | if (dbus_timeout_get_enabled(w->data.bus_timeout)) { | |
206 | timespec_store(&its.it_value, dbus_timeout_get_interval(w->data.bus_timeout) * USEC_PER_MSEC); | |
207 | its.it_interval = its.it_interval; | |
208 | } | |
209 | ||
210 | if (timerfd_settime(w->fd, 0, &its, NULL) < 0) | |
211 | return -errno; | |
212 | ||
213 | return 0; | |
214 | } | |
215 | ||
216 | void bus_timeout_event(Manager *m, Watch *w, int events) { | |
217 | assert(m); | |
218 | assert(w); | |
219 | ||
220 | /* This is called by the event loop whenever there is | |
221 | * something happening on D-Bus' file handles. */ | |
222 | ||
223 | if (!(dbus_timeout_get_enabled(w->data.bus_timeout))) | |
224 | return; | |
225 | ||
226 | dbus_timeout_handle(w->data.bus_timeout); | |
227 | } | |
228 | ||
229 | static dbus_bool_t bus_add_timeout(DBusTimeout *timeout, void *data) { | |
230 | Manager *m = data; | |
231 | Watch *w; | |
232 | struct epoll_event ev; | |
233 | ||
234 | assert(timeout); | |
235 | assert(m); | |
236 | ||
237 | if (!(w = new0(Watch, 1))) | |
238 | return FALSE; | |
239 | ||
240 | if (!(w->fd = timerfd_create(CLOCK_MONOTONIC, TFD_NONBLOCK|TFD_CLOEXEC)) < 0) | |
241 | goto fail; | |
242 | ||
243 | w->type = WATCH_DBUS_TIMEOUT; | |
244 | w->data.bus_timeout = timeout; | |
245 | ||
246 | if (bus_timeout_arm(m, w) < 0) | |
247 | goto fail; | |
248 | ||
249 | zero(ev); | |
250 | ev.events = EPOLLIN; | |
251 | ev.data.ptr = w; | |
252 | ||
253 | if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, w->fd, &ev) < 0) | |
254 | goto fail; | |
255 | ||
256 | dbus_timeout_set_data(timeout, w, NULL); | |
257 | ||
258 | return TRUE; | |
259 | ||
260 | fail: | |
261 | if (w->fd >= 0) | |
262 | close_nointr_nofail(w->fd); | |
263 | ||
264 | free(w); | |
265 | return FALSE; | |
266 | } | |
267 | ||
268 | static void bus_remove_timeout(DBusTimeout *timeout, void *data) { | |
269 | Manager *m = data; | |
270 | Watch *w; | |
271 | ||
272 | assert(timeout); | |
273 | assert(m); | |
274 | ||
275 | if (!(w = dbus_timeout_get_data(timeout))) | |
276 | return; | |
277 | ||
278 | assert(w->type == WATCH_DBUS_TIMEOUT); | |
279 | assert_se(epoll_ctl(m->epoll_fd, EPOLL_CTL_DEL, w->fd, NULL) >= 0); | |
280 | close_nointr_nofail(w->fd); | |
281 | free(w); | |
282 | } | |
283 | ||
284 | static void bus_toggle_timeout(DBusTimeout *timeout, void *data) { | |
285 | Manager *m = data; | |
286 | Watch *w; | |
287 | int r; | |
288 | ||
289 | assert(timeout); | |
290 | assert(m); | |
291 | ||
292 | assert_se(w = dbus_timeout_get_data(timeout)); | |
293 | assert(w->type == WATCH_DBUS_TIMEOUT); | |
294 | ||
295 | if ((r = bus_timeout_arm(m, w)) < 0) | |
296 | log_error("Failed to rearm timer: %s", strerror(-r)); | |
297 | } | |
298 | ||
f278026d | 299 | static DBusHandlerResult api_bus_message_filter(DBusConnection *connection, DBusMessage *message, void *data) { |
c1e1601e LP |
300 | Manager *m = data; |
301 | DBusError error; | |
302 | ||
303 | assert(connection); | |
304 | assert(message); | |
305 | assert(m); | |
306 | ||
307 | dbus_error_init(&error); | |
308 | ||
309 | /* log_debug("Got D-Bus request: %s.%s() on %s", */ | |
310 | /* dbus_message_get_interface(message), */ | |
311 | /* dbus_message_get_member(message), */ | |
312 | /* dbus_message_get_path(message)); */ | |
313 | ||
314 | if (dbus_message_is_signal(message, DBUS_INTERFACE_LOCAL, "Disconnected")) { | |
2e317f52 | 315 | log_error("Warning! API D-Bus connection terminated."); |
f278026d | 316 | bus_done_api(m); |
c1e1601e LP |
317 | |
318 | } else if (dbus_message_is_signal(message, DBUS_INTERFACE_DBUS, "NameOwnerChanged")) { | |
05e343b7 | 319 | const char *name, *old_owner, *new_owner; |
c1e1601e LP |
320 | |
321 | if (!dbus_message_get_args(message, &error, | |
322 | DBUS_TYPE_STRING, &name, | |
05e343b7 LP |
323 | DBUS_TYPE_STRING, &old_owner, |
324 | DBUS_TYPE_STRING, &new_owner, | |
c1e1601e LP |
325 | DBUS_TYPE_INVALID)) |
326 | log_error("Failed to parse NameOwnerChanged message: %s", error.message); | |
327 | else { | |
328 | if (set_remove(m->subscribed, (char*) name)) | |
329 | log_debug("Subscription client vanished: %s (left: %u)", name, set_size(m->subscribed)); | |
05e343b7 LP |
330 | |
331 | if (old_owner[0] == 0) | |
332 | old_owner = NULL; | |
333 | ||
334 | if (new_owner[0] == 0) | |
335 | new_owner = NULL; | |
336 | ||
337 | manager_dispatch_bus_name_owner_changed(m, name, old_owner, new_owner); | |
c1e1601e LP |
338 | } |
339 | } | |
340 | ||
341 | dbus_error_free(&error); | |
342 | return DBUS_HANDLER_RESULT_NOT_YET_HANDLED; | |
343 | } | |
344 | ||
8e274523 LP |
345 | static DBusHandlerResult system_bus_message_filter(DBusConnection *connection, DBusMessage *message, void *data) { |
346 | Manager *m = data; | |
347 | DBusError error; | |
348 | ||
349 | assert(connection); | |
350 | assert(message); | |
351 | assert(m); | |
352 | ||
353 | dbus_error_init(&error); | |
354 | ||
355 | /* log_debug("Got D-Bus request: %s.%s() on %s", */ | |
356 | /* dbus_message_get_interface(message), */ | |
357 | /* dbus_message_get_member(message), */ | |
358 | /* dbus_message_get_path(message)); */ | |
359 | ||
360 | if (dbus_message_is_signal(message, DBUS_INTERFACE_LOCAL, "Disconnected")) { | |
2e317f52 | 361 | log_error("Warning! System D-Bus connection terminated."); |
f278026d | 362 | bus_done_system(m); |
8e274523 LP |
363 | |
364 | } if (dbus_message_is_signal(message, "org.freedesktop.systemd1.Agent", "Released")) { | |
365 | const char *cgroup; | |
366 | ||
367 | if (!dbus_message_get_args(message, &error, | |
368 | DBUS_TYPE_STRING, &cgroup, | |
369 | DBUS_TYPE_INVALID)) | |
370 | log_error("Failed to parse Released message: %s", error.message); | |
371 | else | |
372 | cgroup_notify_empty(m, cgroup); | |
373 | } | |
374 | ||
375 | dbus_error_free(&error); | |
376 | return DBUS_HANDLER_RESULT_NOT_YET_HANDLED; | |
377 | } | |
378 | ||
c1e1601e | 379 | unsigned bus_dispatch(Manager *m) { |
ea430986 LP |
380 | assert(m); |
381 | ||
fd18e1f4 LP |
382 | if (m->request_api_bus_dispatch) { |
383 | if (dbus_connection_dispatch(m->api_bus) == DBUS_DISPATCH_COMPLETE) | |
f278026d | 384 | m->request_api_bus_dispatch = false; |
c1e1601e | 385 | |
fd18e1f4 LP |
386 | return 1; |
387 | } | |
388 | ||
389 | if (m->request_system_bus_dispatch) { | |
390 | if (dbus_connection_dispatch(m->system_bus) == DBUS_DISPATCH_COMPLETE) | |
8e274523 | 391 | m->request_system_bus_dispatch = false; |
fd18e1f4 LP |
392 | |
393 | return 1; | |
394 | } | |
c1e1601e | 395 | |
8e274523 | 396 | return 0; |
ea430986 LP |
397 | } |
398 | ||
05e343b7 | 399 | static void request_name_pending_cb(DBusPendingCall *pending, void *userdata) { |
61902ea3 LP |
400 | DBusMessage *reply; |
401 | DBusError error; | |
402 | ||
403 | dbus_error_init(&error); | |
404 | ||
405 | assert_se(reply = dbus_pending_call_steal_reply(pending)); | |
406 | ||
407 | switch (dbus_message_get_type(reply)) { | |
408 | ||
409 | case DBUS_MESSAGE_TYPE_ERROR: | |
410 | ||
411 | assert_se(dbus_set_error_from_message(&error, reply)); | |
412 | log_warning("RequestName() failed: %s", error.message); | |
413 | break; | |
414 | ||
415 | case DBUS_MESSAGE_TYPE_METHOD_RETURN: { | |
416 | uint32_t r; | |
417 | ||
418 | if (!dbus_message_get_args(reply, | |
419 | &error, | |
420 | DBUS_TYPE_UINT32, &r, | |
421 | DBUS_TYPE_INVALID)) { | |
422 | log_error("Failed to parse RequestName() reply: %s", error.message); | |
423 | break; | |
424 | } | |
425 | ||
426 | if (r == 1) | |
427 | log_debug("Successfully acquired name."); | |
428 | else | |
429 | log_error("Name already owned."); | |
430 | ||
431 | break; | |
432 | } | |
433 | ||
434 | default: | |
435 | assert_not_reached("Invalid reply message"); | |
436 | } | |
437 | ||
438 | dbus_message_unref(reply); | |
439 | dbus_error_free(&error); | |
440 | } | |
441 | ||
ea430986 | 442 | static int request_name(Manager *m) { |
ea430986 LP |
443 | const char *name = "org.freedesktop.systemd1"; |
444 | uint32_t flags = 0; | |
05e343b7 LP |
445 | DBusMessage *message = NULL; |
446 | DBusPendingCall *pending = NULL; | |
ea430986 LP |
447 | |
448 | if (!(message = dbus_message_new_method_call( | |
449 | DBUS_SERVICE_DBUS, | |
450 | DBUS_PATH_DBUS, | |
451 | DBUS_INTERFACE_DBUS, | |
452 | "RequestName"))) | |
05e343b7 | 453 | goto oom; |
ea430986 LP |
454 | |
455 | if (!dbus_message_append_args( | |
456 | message, | |
457 | DBUS_TYPE_STRING, &name, | |
458 | DBUS_TYPE_UINT32, &flags, | |
05e343b7 LP |
459 | DBUS_TYPE_INVALID)) |
460 | goto oom; | |
ea430986 | 461 | |
05e343b7 LP |
462 | if (!dbus_connection_send_with_reply(m->api_bus, message, &pending, -1)) |
463 | goto oom; | |
ea430986 | 464 | |
05e343b7 LP |
465 | if (!dbus_pending_call_set_notify(pending, request_name_pending_cb, m, NULL)) |
466 | goto oom; | |
ea430986 LP |
467 | |
468 | dbus_message_unref(message); | |
61902ea3 LP |
469 | dbus_pending_call_unref(pending); |
470 | ||
471 | /* We simple ask for the name and don't wait for it. Sooner or | |
472 | * later we'll have it. */ | |
473 | ||
ea430986 | 474 | return 0; |
05e343b7 LP |
475 | |
476 | oom: | |
477 | if (pending) { | |
478 | dbus_pending_call_cancel(pending); | |
479 | dbus_pending_call_unref(pending); | |
480 | } | |
481 | ||
482 | if (message) | |
483 | dbus_message_unref(message); | |
484 | ||
485 | return -ENOMEM; | |
ea430986 LP |
486 | } |
487 | ||
8e274523 LP |
488 | static int bus_setup_loop(Manager *m, DBusConnection *bus) { |
489 | assert(m); | |
490 | assert(bus); | |
491 | ||
492 | dbus_connection_set_exit_on_disconnect(bus, FALSE); | |
fd18e1f4 | 493 | |
8e274523 LP |
494 | if (!dbus_connection_set_watch_functions(bus, bus_add_watch, bus_remove_watch, bus_toggle_watch, m, NULL) || |
495 | !dbus_connection_set_timeout_functions(bus, bus_add_timeout, bus_remove_timeout, bus_toggle_timeout, m, NULL)) | |
496 | return -ENOMEM; | |
497 | ||
498 | return 0; | |
499 | } | |
500 | ||
f278026d | 501 | int bus_init_system(Manager *m) { |
ea430986 LP |
502 | DBusError error; |
503 | char *id; | |
504 | int r; | |
505 | ||
506 | assert(m); | |
507 | ||
ea430986 | 508 | dbus_error_init(&error); |
ea430986 | 509 | |
f278026d LP |
510 | if (m->system_bus) |
511 | return 0; | |
8e274523 | 512 | |
f278026d LP |
513 | if (m->running_as != MANAGER_SESSION && m->api_bus) |
514 | m->system_bus = m->api_bus; | |
515 | else { | |
8e274523 | 516 | if (!(m->system_bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, &error))) { |
266e525c | 517 | log_debug("Failed to get system D-Bus connection, retrying later: %s", error.message); |
8e274523 | 518 | dbus_error_free(&error); |
f278026d | 519 | return 0; |
8e274523 LP |
520 | } |
521 | ||
fd18e1f4 LP |
522 | dbus_connection_set_dispatch_status_function(m->system_bus, system_bus_dispatch_status, m, NULL); |
523 | m->request_system_bus_dispatch = true; | |
524 | ||
8e274523 | 525 | if ((r = bus_setup_loop(m, m->system_bus)) < 0) { |
f278026d | 526 | bus_done_system(m); |
8e274523 LP |
527 | return r; |
528 | } | |
f278026d LP |
529 | } |
530 | ||
531 | if (!dbus_connection_add_filter(m->system_bus, system_bus_message_filter, m, NULL)) { | |
532 | bus_done_system(m); | |
c1e1601e LP |
533 | return -ENOMEM; |
534 | } | |
535 | ||
f278026d | 536 | dbus_bus_add_match(m->system_bus, |
c1e1601e | 537 | "type='signal'," |
f278026d LP |
538 | "interface='org.freedesktop.systemd1.Agent'," |
539 | "path='/org/freedesktop/systemd1/agent'", | |
c1e1601e LP |
540 | &error); |
541 | ||
542 | if (dbus_error_is_set(&error)) { | |
543 | log_error("Failed to register match: %s", error.message); | |
544 | dbus_error_free(&error); | |
f278026d | 545 | bus_done_system(m); |
ea430986 LP |
546 | return -ENOMEM; |
547 | } | |
548 | ||
f278026d LP |
549 | log_debug("Successfully connected to system D-Bus bus %s as %s", |
550 | strnull((id = dbus_connection_get_server_id(m->system_bus))), | |
551 | strnull(dbus_bus_get_unique_name(m->system_bus))); | |
552 | dbus_free(id); | |
553 | ||
f278026d LP |
554 | return 0; |
555 | } | |
556 | ||
557 | int bus_init_api(Manager *m) { | |
558 | DBusError error; | |
559 | char *id; | |
560 | int r; | |
561 | ||
562 | assert(m); | |
563 | ||
564 | dbus_error_init(&error); | |
565 | ||
566 | if (m->api_bus) | |
567 | return 0; | |
568 | ||
05e343b7 LP |
569 | if (m->name_data_slot < 0) |
570 | if (!dbus_pending_call_allocate_data_slot(&m->name_data_slot)) | |
571 | return -ENOMEM; | |
572 | ||
f278026d LP |
573 | if (m->running_as != MANAGER_SESSION && m->system_bus) |
574 | m->api_bus = m->system_bus; | |
575 | else { | |
576 | if (!(m->api_bus = dbus_bus_get_private(m->running_as == MANAGER_SESSION ? DBUS_BUS_SESSION : DBUS_BUS_SYSTEM, &error))) { | |
266e525c | 577 | log_debug("Failed to get API D-Bus connection, retrying later: %s", error.message); |
f278026d LP |
578 | dbus_error_free(&error); |
579 | return 0; | |
580 | } | |
581 | ||
fd18e1f4 LP |
582 | dbus_connection_set_dispatch_status_function(m->api_bus, api_bus_dispatch_status, m, NULL); |
583 | m->request_api_bus_dispatch = true; | |
584 | ||
f278026d LP |
585 | if ((r = bus_setup_loop(m, m->api_bus)) < 0) { |
586 | bus_done_api(m); | |
587 | return r; | |
588 | } | |
ea430986 LP |
589 | } |
590 | ||
f278026d LP |
591 | if (!dbus_connection_register_object_path(m->api_bus, "/org/freedesktop/systemd1", &bus_manager_vtable, m) || |
592 | !dbus_connection_register_fallback(m->api_bus, "/org/freedesktop/systemd1/unit", &bus_unit_vtable, m) || | |
593 | !dbus_connection_register_fallback(m->api_bus, "/org/freedesktop/systemd1/job", &bus_job_vtable, m) || | |
594 | !dbus_connection_add_filter(m->api_bus, api_bus_message_filter, m, NULL)) { | |
595 | bus_done_api(m); | |
596 | return -ENOMEM; | |
597 | } | |
598 | ||
599 | dbus_bus_add_match(m->api_bus, | |
8e274523 | 600 | "type='signal'," |
f278026d LP |
601 | "sender='"DBUS_SERVICE_DBUS"'," |
602 | "interface='"DBUS_INTERFACE_DBUS"'," | |
603 | "path='"DBUS_PATH_DBUS"'", | |
8e274523 LP |
604 | &error); |
605 | ||
606 | if (dbus_error_is_set(&error)) { | |
607 | log_error("Failed to register match: %s", error.message); | |
608 | dbus_error_free(&error); | |
f278026d | 609 | bus_done_api(m); |
8e274523 LP |
610 | return -ENOMEM; |
611 | } | |
612 | ||
f278026d LP |
613 | if ((r = request_name(m)) < 0) { |
614 | bus_done_api(m); | |
615 | return r; | |
616 | } | |
ea430986 | 617 | |
f278026d LP |
618 | log_debug("Successfully connected to API D-Bus bus %s as %s", |
619 | strnull((id = dbus_connection_get_server_id(m->api_bus))), | |
620 | strnull(dbus_bus_get_unique_name(m->api_bus))); | |
8e274523 LP |
621 | dbus_free(id); |
622 | ||
f278026d LP |
623 | if (!m->subscribed) |
624 | if (!(m->subscribed = set_new(string_hash_func, string_compare_func))) | |
625 | return -ENOMEM; | |
626 | ||
ea430986 LP |
627 | return 0; |
628 | } | |
629 | ||
f278026d | 630 | void bus_done_api(Manager *m) { |
ea430986 LP |
631 | assert(m); |
632 | ||
f278026d LP |
633 | if (m->api_bus) { |
634 | if (m->system_bus == m->api_bus) | |
635 | m->system_bus = NULL; | |
636 | ||
90d3f30b | 637 | dbus_connection_set_dispatch_status_function(m->api_bus, NULL, NULL, NULL); |
2e317f52 | 638 | dbus_connection_close(m->api_bus); |
f278026d LP |
639 | dbus_connection_unref(m->api_bus); |
640 | m->api_bus = NULL; | |
ea430986 | 641 | } |
c1e1601e LP |
642 | |
643 | if (m->subscribed) { | |
644 | char *c; | |
645 | ||
646 | while ((c = set_steal_first(m->subscribed))) | |
647 | free(c); | |
648 | ||
649 | set_free(m->subscribed); | |
650 | m->subscribed = NULL; | |
651 | } | |
05e343b7 LP |
652 | |
653 | if (m->name_data_slot >= 0) | |
654 | dbus_pending_call_free_data_slot(&m->name_data_slot); | |
ea430986 LP |
655 | } |
656 | ||
f278026d LP |
657 | void bus_done_system(Manager *m) { |
658 | assert(m); | |
659 | ||
660 | if (m->system_bus == m->api_bus) | |
661 | bus_done_api(m); | |
662 | ||
663 | if (m->system_bus) { | |
90d3f30b | 664 | dbus_connection_set_dispatch_status_function(m->system_bus, NULL, NULL, NULL); |
2e317f52 | 665 | dbus_connection_close(m->system_bus); |
f278026d LP |
666 | dbus_connection_unref(m->system_bus); |
667 | m->system_bus = NULL; | |
668 | } | |
669 | } | |
670 | ||
05e343b7 LP |
671 | static void query_pid_pending_cb(DBusPendingCall *pending, void *userdata) { |
672 | Manager *m = userdata; | |
673 | DBusMessage *reply; | |
674 | DBusError error; | |
675 | const char *name; | |
676 | ||
677 | dbus_error_init(&error); | |
678 | ||
679 | assert_se(name = dbus_pending_call_get_data(pending, m->name_data_slot)); | |
680 | assert_se(reply = dbus_pending_call_steal_reply(pending)); | |
681 | ||
682 | switch (dbus_message_get_type(reply)) { | |
683 | ||
684 | case DBUS_MESSAGE_TYPE_ERROR: | |
685 | ||
686 | assert_se(dbus_set_error_from_message(&error, reply)); | |
687 | log_warning("GetConnectionUnixProcessID() failed: %s", error.message); | |
688 | break; | |
689 | ||
690 | case DBUS_MESSAGE_TYPE_METHOD_RETURN: { | |
691 | uint32_t r; | |
692 | ||
693 | if (!dbus_message_get_args(reply, | |
694 | &error, | |
695 | DBUS_TYPE_UINT32, &r, | |
696 | DBUS_TYPE_INVALID)) { | |
697 | log_error("Failed to parse GetConnectionUnixProcessID() reply: %s", error.message); | |
698 | break; | |
699 | } | |
700 | ||
701 | manager_dispatch_bus_query_pid_done(m, name, (pid_t) r); | |
702 | break; | |
703 | } | |
704 | ||
705 | default: | |
706 | assert_not_reached("Invalid reply message"); | |
707 | } | |
708 | ||
709 | dbus_message_unref(reply); | |
710 | dbus_error_free(&error); | |
711 | } | |
712 | ||
713 | int bus_query_pid(Manager *m, const char *name) { | |
714 | DBusMessage *message = NULL; | |
715 | DBusPendingCall *pending = NULL; | |
716 | char *n = NULL; | |
717 | ||
718 | assert(m); | |
719 | assert(name); | |
720 | ||
721 | if (!(message = dbus_message_new_method_call( | |
722 | DBUS_SERVICE_DBUS, | |
723 | DBUS_PATH_DBUS, | |
724 | DBUS_INTERFACE_DBUS, | |
725 | "GetConnectionUnixProcessID"))) | |
726 | goto oom; | |
727 | ||
728 | if (!(dbus_message_append_args( | |
729 | message, | |
730 | DBUS_TYPE_STRING, &name, | |
731 | DBUS_TYPE_INVALID))) | |
732 | goto oom; | |
733 | ||
734 | if (!dbus_connection_send_with_reply(m->api_bus, message, &pending, -1)) | |
735 | goto oom; | |
736 | ||
737 | if (!(n = strdup(name))) | |
738 | goto oom; | |
739 | ||
740 | if (!dbus_pending_call_set_data(pending, m->name_data_slot, n, free)) | |
741 | goto oom; | |
742 | ||
743 | n = NULL; | |
744 | ||
745 | if (!dbus_pending_call_set_notify(pending, query_pid_pending_cb, m, NULL)) | |
746 | goto oom; | |
747 | ||
748 | dbus_message_unref(message); | |
749 | dbus_pending_call_unref(pending); | |
750 | ||
751 | return 0; | |
752 | ||
753 | oom: | |
754 | free(n); | |
755 | ||
756 | if (pending) { | |
757 | dbus_pending_call_cancel(pending); | |
758 | dbus_pending_call_unref(pending); | |
759 | } | |
760 | ||
761 | if (message) | |
762 | dbus_message_unref(message); | |
763 | ||
764 | return -ENOMEM; | |
765 | } | |
766 | ||
ea430986 LP |
767 | DBusHandlerResult bus_default_message_handler(Manager *m, DBusMessage *message, const char*introspection, const BusProperty *properties) { |
768 | DBusError error; | |
769 | DBusMessage *reply = NULL; | |
770 | int r; | |
771 | ||
772 | assert(m); | |
773 | assert(message); | |
774 | ||
775 | dbus_error_init(&error); | |
776 | ||
777 | if (dbus_message_is_method_call(message, "org.freedesktop.DBus.Introspectable", "Introspect") && introspection) { | |
778 | ||
779 | if (!(reply = dbus_message_new_method_return(message))) | |
780 | goto oom; | |
781 | ||
782 | if (!dbus_message_append_args(reply, DBUS_TYPE_STRING, &introspection, DBUS_TYPE_INVALID)) | |
783 | goto oom; | |
784 | ||
785 | } else if (dbus_message_is_method_call(message, "org.freedesktop.DBus.Properties", "Get") && properties) { | |
786 | const char *interface, *property; | |
787 | const BusProperty *p; | |
788 | ||
789 | if (!dbus_message_get_args( | |
790 | message, | |
791 | &error, | |
792 | DBUS_TYPE_STRING, &interface, | |
793 | DBUS_TYPE_STRING, &property, | |
794 | DBUS_TYPE_INVALID)) | |
795 | return bus_send_error_reply(m, message, &error, -EINVAL); | |
796 | ||
797 | for (p = properties; p->property; p++) | |
798 | if (streq(p->interface, interface) && streq(p->property, property)) | |
799 | break; | |
800 | ||
801 | if (p->property) { | |
802 | DBusMessageIter iter, sub; | |
803 | ||
804 | if (!(reply = dbus_message_new_method_return(message))) | |
805 | goto oom; | |
806 | ||
807 | dbus_message_iter_init_append(reply, &iter); | |
808 | ||
809 | if (!dbus_message_iter_open_container(&iter, DBUS_TYPE_VARIANT, p->signature, &sub)) | |
810 | goto oom; | |
811 | ||
b9f49ee4 | 812 | if ((r = p->append(m, &sub, property, (void*) p->data)) < 0) { |
ea430986 LP |
813 | |
814 | if (r == -ENOMEM) | |
815 | goto oom; | |
816 | ||
817 | dbus_message_unref(reply); | |
818 | return bus_send_error_reply(m, message, NULL, r); | |
819 | } | |
820 | ||
821 | if (!dbus_message_iter_close_container(&iter, &sub)) | |
822 | goto oom; | |
823 | } | |
824 | } else if (dbus_message_is_method_call(message, "org.freedesktop.DBus.Properties", "GetAll") && properties) { | |
825 | const char *interface; | |
826 | const BusProperty *p; | |
827 | DBusMessageIter iter, sub, sub2, sub3; | |
828 | bool any = false; | |
829 | ||
830 | if (!dbus_message_get_args( | |
831 | message, | |
832 | &error, | |
833 | DBUS_TYPE_STRING, &interface, | |
834 | DBUS_TYPE_INVALID)) | |
835 | return bus_send_error_reply(m, message, &error, -EINVAL); | |
836 | ||
837 | if (!(reply = dbus_message_new_method_return(message))) | |
838 | goto oom; | |
839 | ||
840 | dbus_message_iter_init_append(reply, &iter); | |
841 | ||
842 | if (!dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "{sv}", &sub)) | |
843 | goto oom; | |
844 | ||
845 | for (p = properties; p->property; p++) { | |
846 | if (!streq(p->interface, interface)) | |
847 | continue; | |
848 | ||
c401a1e0 | 849 | if (!dbus_message_iter_open_container(&sub, DBUS_TYPE_DICT_ENTRY, NULL, &sub2) || |
ea430986 LP |
850 | !dbus_message_iter_append_basic(&sub2, DBUS_TYPE_STRING, &p->property) || |
851 | !dbus_message_iter_open_container(&sub2, DBUS_TYPE_VARIANT, p->signature, &sub3)) | |
852 | goto oom; | |
853 | ||
b9f49ee4 | 854 | if ((r = p->append(m, &sub3, p->property, (void*) p->data)) < 0) { |
ea430986 LP |
855 | |
856 | if (r == -ENOMEM) | |
857 | goto oom; | |
858 | ||
859 | dbus_message_unref(reply); | |
860 | return bus_send_error_reply(m, message, NULL, r); | |
861 | } | |
862 | ||
863 | if (!dbus_message_iter_close_container(&sub2, &sub3) || | |
864 | !dbus_message_iter_close_container(&sub, &sub2)) | |
865 | goto oom; | |
866 | ||
867 | any = true; | |
868 | } | |
869 | ||
870 | if (!dbus_message_iter_close_container(&iter, &sub)) | |
871 | goto oom; | |
872 | } | |
873 | ||
874 | if (reply) { | |
f278026d | 875 | if (!dbus_connection_send(m->api_bus, reply, NULL)) |
ea430986 LP |
876 | goto oom; |
877 | ||
878 | dbus_message_unref(reply); | |
879 | return DBUS_HANDLER_RESULT_HANDLED; | |
880 | } | |
881 | ||
882 | return DBUS_HANDLER_RESULT_NOT_YET_HANDLED; | |
883 | ||
884 | oom: | |
885 | if (reply) | |
886 | dbus_message_unref(reply); | |
887 | ||
888 | dbus_error_free(&error); | |
889 | ||
890 | return DBUS_HANDLER_RESULT_NEED_MEMORY; | |
891 | } | |
892 | ||
ea430986 LP |
893 | static const char *error_to_dbus(int error) { |
894 | ||
895 | switch(error) { | |
896 | ||
897 | case -EINVAL: | |
898 | return DBUS_ERROR_INVALID_ARGS; | |
899 | ||
900 | case -ENOMEM: | |
901 | return DBUS_ERROR_NO_MEMORY; | |
902 | ||
903 | case -EPERM: | |
904 | case -EACCES: | |
905 | return DBUS_ERROR_ACCESS_DENIED; | |
906 | ||
907 | case -ESRCH: | |
908 | return DBUS_ERROR_UNIX_PROCESS_ID_UNKNOWN; | |
909 | ||
910 | case -ENOENT: | |
911 | return DBUS_ERROR_FILE_NOT_FOUND; | |
912 | ||
913 | case -EEXIST: | |
914 | return DBUS_ERROR_FILE_EXISTS; | |
915 | ||
916 | case -ETIMEDOUT: | |
917 | return DBUS_ERROR_TIMEOUT; | |
918 | ||
919 | case -EIO: | |
920 | return DBUS_ERROR_IO_ERROR; | |
921 | ||
922 | case -ENETRESET: | |
923 | case -ECONNABORTED: | |
924 | case -ECONNRESET: | |
925 | return DBUS_ERROR_DISCONNECTED; | |
926 | } | |
927 | ||
928 | return DBUS_ERROR_FAILED; | |
929 | } | |
930 | ||
931 | DBusHandlerResult bus_send_error_reply(Manager *m, DBusMessage *message, DBusError *bus_error, int error) { | |
932 | DBusMessage *reply = NULL; | |
933 | const char *name, *text; | |
934 | ||
935 | if (bus_error && dbus_error_is_set(bus_error)) { | |
936 | name = bus_error->name; | |
937 | text = bus_error->message; | |
938 | } else { | |
939 | name = error_to_dbus(error); | |
940 | text = strerror(-error); | |
941 | } | |
942 | ||
943 | if (!(reply = dbus_message_new_error(message, name, text))) | |
944 | goto oom; | |
945 | ||
f278026d | 946 | if (!dbus_connection_send(m->api_bus, reply, NULL)) |
ea430986 LP |
947 | goto oom; |
948 | ||
949 | dbus_message_unref(reply); | |
950 | ||
951 | if (bus_error) | |
952 | dbus_error_free(bus_error); | |
953 | ||
954 | return DBUS_HANDLER_RESULT_HANDLED; | |
955 | ||
956 | oom: | |
957 | if (reply) | |
958 | dbus_message_unref(reply); | |
959 | ||
960 | if (bus_error) | |
961 | dbus_error_free(bus_error); | |
962 | ||
963 | return DBUS_HANDLER_RESULT_NEED_MEMORY; | |
964 | } | |
965 | ||
966 | int bus_property_append_string(Manager *m, DBusMessageIter *i, const char *property, void *data) { | |
967 | const char *t = data; | |
968 | ||
969 | assert(m); | |
970 | assert(i); | |
971 | assert(property); | |
38131695 LP |
972 | |
973 | if (!t) | |
974 | t = ""; | |
ea430986 LP |
975 | |
976 | if (!dbus_message_iter_append_basic(i, DBUS_TYPE_STRING, &t)) | |
977 | return -ENOMEM; | |
978 | ||
979 | return 0; | |
980 | } | |
981 | ||
982 | int bus_property_append_strv(Manager *m, DBusMessageIter *i, const char *property, void *data) { | |
983 | DBusMessageIter sub; | |
984 | char **t = data; | |
985 | ||
986 | assert(m); | |
987 | assert(i); | |
988 | assert(property); | |
989 | ||
990 | if (!dbus_message_iter_open_container(i, DBUS_TYPE_ARRAY, "s", &sub)) | |
991 | return -ENOMEM; | |
992 | ||
993 | STRV_FOREACH(t, t) | |
994 | if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, t)) | |
995 | return -ENOMEM; | |
996 | ||
997 | if (!dbus_message_iter_close_container(i, &sub)) | |
998 | return -ENOMEM; | |
999 | ||
1000 | return 0; | |
1001 | } | |
1002 | ||
1003 | int bus_property_append_bool(Manager *m, DBusMessageIter *i, const char *property, void *data) { | |
1004 | bool *b = data; | |
1005 | dbus_bool_t db; | |
1006 | ||
1007 | assert(m); | |
1008 | assert(i); | |
1009 | assert(property); | |
1010 | assert(b); | |
1011 | ||
1012 | db = *b; | |
1013 | ||
1014 | if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &db)) | |
1015 | return -ENOMEM; | |
1016 | ||
1017 | return 0; | |
1018 | } | |
38131695 LP |
1019 | |
1020 | int bus_property_append_uint64(Manager *m, DBusMessageIter *i, const char *property, void *data) { | |
1021 | assert(m); | |
1022 | assert(i); | |
1023 | assert(property); | |
1024 | assert(data); | |
1025 | ||
1026 | if (!dbus_message_iter_append_basic(i, DBUS_TYPE_UINT64, data)) | |
1027 | return -ENOMEM; | |
1028 | ||
1029 | return 0; | |
1030 | } | |
1031 | ||
1032 | int bus_property_append_uint32(Manager *m, DBusMessageIter *i, const char *property, void *data) { | |
1033 | assert(m); | |
1034 | assert(i); | |
1035 | assert(property); | |
1036 | assert(data); | |
1037 | ||
1038 | if (!dbus_message_iter_append_basic(i, DBUS_TYPE_UINT32, data)) | |
1039 | return -ENOMEM; | |
1040 | ||
1041 | return 0; | |
1042 | } |