1 /*-*- Mode: C; c-basic-offset: 8 -*-*/
4 This file is part of systemd.
6 Copyright 2010 Lennart Poettering
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.
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.
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/>.
22 #include <dbus/dbus.h>
24 #include <sys/epoll.h>
25 #include <sys/timerfd.h>
34 static void bus_dispatch_status(DBusConnection
*bus
, DBusDispatchStatus status
, void *data
) {
39 assert(m
->bus
== bus
);
41 m
->request_bus_dispatch
= status
!= DBUS_DISPATCH_COMPLETE
;
44 static void system_bus_dispatch_status(DBusConnection
*bus
, DBusDispatchStatus status
, void *data
) {
49 assert(m
->system_bus
== bus
);
51 m
->request_system_bus_dispatch
= status
!= DBUS_DISPATCH_COMPLETE
;
54 static uint32_t bus_flags_to_events(DBusWatch
*bus_watch
) {
60 /* no watch flags for disabled watches */
61 if (!dbus_watch_get_enabled(bus_watch
))
64 flags
= dbus_watch_get_flags(bus_watch
);
66 if (flags
& DBUS_WATCH_READABLE
)
68 if (flags
& DBUS_WATCH_WRITABLE
)
71 return events
| EPOLLHUP
| EPOLLERR
;
74 static unsigned events_to_bus_flags(uint32_t events
) {
78 flags
|= DBUS_WATCH_READABLE
;
79 if (events
& EPOLLOUT
)
80 flags
|= DBUS_WATCH_WRITABLE
;
81 if (events
& EPOLLHUP
)
82 flags
|= DBUS_WATCH_HANGUP
;
83 if (events
& EPOLLERR
)
84 flags
|= DBUS_WATCH_ERROR
;
89 void bus_watch_event(Manager
*m
, Watch
*w
, int events
) {
93 /* This is called by the event loop whenever there is
94 * something happening on D-Bus' file handles. */
96 if (!dbus_watch_get_enabled(w
->data
.bus_watch
))
99 dbus_watch_handle(w
->data
.bus_watch
, events_to_bus_flags(events
));
102 static dbus_bool_t
bus_add_watch(DBusWatch
*bus_watch
, void *data
) {
105 struct epoll_event ev
;
110 if (!(w
= new0(Watch
, 1)))
113 w
->fd
= dbus_watch_get_unix_fd(bus_watch
);
114 w
->type
= WATCH_DBUS_WATCH
;
115 w
->data
.bus_watch
= bus_watch
;
118 ev
.events
= bus_flags_to_events(bus_watch
);
121 if (epoll_ctl(m
->epoll_fd
, EPOLL_CTL_ADD
, w
->fd
, &ev
) < 0) {
123 if (errno
!= EEXIST
) {
128 /* Hmm, bloody D-Bus creates multiple watches on the
129 * same fd. epoll() does not like that. As a dirty
130 * hack we simply dup() the fd and hence get a second
131 * one we can safely add to the epoll(). */
133 if ((w
->fd
= dup(w
->fd
)) < 0) {
138 if (epoll_ctl(m
->epoll_fd
, EPOLL_CTL_ADD
, w
->fd
, &ev
) < 0) {
140 close_nointr_nofail(w
->fd
);
144 w
->fd_is_dupped
= true;
147 dbus_watch_set_data(bus_watch
, w
, NULL
);
152 static void bus_remove_watch(DBusWatch
*bus_watch
, void *data
) {
159 if (!(w
= dbus_watch_get_data(bus_watch
)))
162 assert(w
->type
== WATCH_DBUS_WATCH
);
163 assert_se(epoll_ctl(m
->epoll_fd
, EPOLL_CTL_DEL
, w
->fd
, NULL
) >= 0);
166 close_nointr_nofail(w
->fd
);
171 static void bus_toggle_watch(DBusWatch
*bus_watch
, void *data
) {
174 struct epoll_event ev
;
179 assert_se(w
= dbus_watch_get_data(bus_watch
));
180 assert(w
->type
== WATCH_DBUS_WATCH
);
183 ev
.events
= bus_flags_to_events(bus_watch
);
186 assert_se(epoll_ctl(m
->epoll_fd
, EPOLL_CTL_MOD
, w
->fd
, &ev
) == 0);
189 static int bus_timeout_arm(Manager
*m
, Watch
*w
) {
190 struct itimerspec its
;
197 if (dbus_timeout_get_enabled(w
->data
.bus_timeout
)) {
198 timespec_store(&its
.it_value
, dbus_timeout_get_interval(w
->data
.bus_timeout
) * USEC_PER_MSEC
);
199 its
.it_interval
= its
.it_interval
;
202 if (timerfd_settime(w
->fd
, 0, &its
, NULL
) < 0)
208 void bus_timeout_event(Manager
*m
, Watch
*w
, int events
) {
212 /* This is called by the event loop whenever there is
213 * something happening on D-Bus' file handles. */
215 if (!(dbus_timeout_get_enabled(w
->data
.bus_timeout
)))
218 dbus_timeout_handle(w
->data
.bus_timeout
);
221 static dbus_bool_t
bus_add_timeout(DBusTimeout
*timeout
, void *data
) {
224 struct epoll_event ev
;
229 if (!(w
= new0(Watch
, 1)))
232 if (!(w
->fd
= timerfd_create(CLOCK_MONOTONIC
, TFD_NONBLOCK
|TFD_CLOEXEC
)) < 0)
235 w
->type
= WATCH_DBUS_TIMEOUT
;
236 w
->data
.bus_timeout
= timeout
;
238 if (bus_timeout_arm(m
, w
) < 0)
245 if (epoll_ctl(m
->epoll_fd
, EPOLL_CTL_ADD
, w
->fd
, &ev
) < 0)
248 dbus_timeout_set_data(timeout
, w
, NULL
);
254 close_nointr_nofail(w
->fd
);
260 static void bus_remove_timeout(DBusTimeout
*timeout
, void *data
) {
267 if (!(w
= dbus_timeout_get_data(timeout
)))
270 assert(w
->type
== WATCH_DBUS_TIMEOUT
);
271 assert_se(epoll_ctl(m
->epoll_fd
, EPOLL_CTL_DEL
, w
->fd
, NULL
) >= 0);
272 close_nointr_nofail(w
->fd
);
276 static void bus_toggle_timeout(DBusTimeout
*timeout
, void *data
) {
284 assert_se(w
= dbus_timeout_get_data(timeout
));
285 assert(w
->type
== WATCH_DBUS_TIMEOUT
);
287 if ((r
= bus_timeout_arm(m
, w
)) < 0)
288 log_error("Failed to rearm timer: %s", strerror(-r
));
291 static DBusHandlerResult
bus_message_filter(DBusConnection
*connection
, DBusMessage
*message
, void *data
) {
299 dbus_error_init(&error
);
301 /* log_debug("Got D-Bus request: %s.%s() on %s", */
302 /* dbus_message_get_interface(message), */
303 /* dbus_message_get_member(message), */
304 /* dbus_message_get_path(message)); */
306 if (dbus_message_is_signal(message
, DBUS_INTERFACE_LOCAL
, "Disconnected")) {
307 log_error("Warning! D-Bus connection terminated.");
309 /* FIXME: we probably should restart D-Bus here */
311 } else if (dbus_message_is_signal(message
, DBUS_INTERFACE_DBUS
, "NameOwnerChanged")) {
312 const char *name
, *old
, *new;
314 if (!dbus_message_get_args(message
, &error
,
315 DBUS_TYPE_STRING
, &name
,
316 DBUS_TYPE_STRING
, &old
,
317 DBUS_TYPE_STRING
, &new,
319 log_error("Failed to parse NameOwnerChanged message: %s", error
.message
);
321 if (set_remove(m
->subscribed
, (char*) name
))
322 log_debug("Subscription client vanished: %s (left: %u)", name
, set_size(m
->subscribed
));
326 dbus_error_free(&error
);
327 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED
;
330 static DBusHandlerResult
system_bus_message_filter(DBusConnection
*connection
, DBusMessage
*message
, void *data
) {
338 dbus_error_init(&error
);
340 /* log_debug("Got D-Bus request: %s.%s() on %s", */
341 /* dbus_message_get_interface(message), */
342 /* dbus_message_get_member(message), */
343 /* dbus_message_get_path(message)); */
345 if (dbus_message_is_signal(message
, DBUS_INTERFACE_LOCAL
, "Disconnected")) {
346 log_error("Warning! D-Bus connection terminated.");
348 /* FIXME: we probably should restart D-Bus here */
350 } if (dbus_message_is_signal(message
, "org.freedesktop.systemd1.Agent", "Released")) {
353 if (!dbus_message_get_args(message
, &error
,
354 DBUS_TYPE_STRING
, &cgroup
,
356 log_error("Failed to parse Released message: %s", error
.message
);
358 cgroup_notify_empty(m
, cgroup
);
361 dbus_error_free(&error
);
362 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED
;
365 unsigned bus_dispatch(Manager
*m
) {
368 if (m
->request_bus_dispatch
)
369 if (dbus_connection_dispatch(m
->bus
) == DBUS_DISPATCH_COMPLETE
) {
370 m
->request_bus_dispatch
= false;
374 if (m
->request_system_bus_dispatch
)
375 if (dbus_connection_dispatch(m
->system_bus
) == DBUS_DISPATCH_COMPLETE
) {
376 m
->request_system_bus_dispatch
= false;
383 static int request_name(Manager
*m
) {
384 DBusMessage
*message
;
385 const char *name
= "org.freedesktop.systemd1";
388 if (!(message
= dbus_message_new_method_call(
395 if (!dbus_message_append_args(
397 DBUS_TYPE_STRING
, &name
,
398 DBUS_TYPE_UINT32
, &flags
,
399 DBUS_TYPE_INVALID
)) {
400 dbus_message_unref(message
);
404 if (!dbus_connection_send(m
->bus
, message
, NULL
)) {
405 dbus_message_unref(message
);
409 /* We simple ask for the name and don't wait for it. Sooner or
410 * later we'll have it, and we wouldn't know what to do on
413 dbus_message_unref(message
);
418 static int bus_setup_loop(Manager
*m
, DBusConnection
*bus
) {
422 dbus_connection_set_exit_on_disconnect(bus
, FALSE
);
423 if (!dbus_connection_set_watch_functions(bus
, bus_add_watch
, bus_remove_watch
, bus_toggle_watch
, m
, NULL
) ||
424 !dbus_connection_set_timeout_functions(bus
, bus_add_timeout
, bus_remove_timeout
, bus_toggle_timeout
, m
, NULL
))
430 int bus_init(Manager
*m
) {
440 if (!(m
->subscribed
= set_new(string_hash_func
, string_compare_func
)))
443 dbus_connection_set_change_sigpipe(FALSE
);
445 dbus_error_init(&error
);
446 if (!(m
->bus
= dbus_bus_get_private(m
->running_as
== MANAGER_SESSION
? DBUS_BUS_SESSION
: DBUS_BUS_SYSTEM
, &error
))) {
447 log_error("Failed to get D-Bus connection: %s", error
.message
);
448 dbus_error_free(&error
);
450 return -ECONNREFUSED
;
453 if ((r
= bus_setup_loop(m
, m
->bus
)) < 0) {
458 dbus_connection_set_dispatch_status_function(m
->bus
, bus_dispatch_status
, m
, NULL
);
460 if (m
->running_as
== MANAGER_SESSION
) {
461 if (!(m
->system_bus
= dbus_bus_get_private(DBUS_BUS_SYSTEM
, &error
))) {
462 log_error("Failed to get D-Bus connection: %s", error
.message
);
463 dbus_error_free(&error
);
465 return -ECONNREFUSED
;
468 if ((r
= bus_setup_loop(m
, m
->system_bus
)) < 0) {
473 dbus_connection_set_dispatch_status_function(m
->system_bus
, system_bus_dispatch_status
, m
, NULL
);
475 m
->system_bus
= m
->bus
;
477 if (!dbus_connection_register_object_path(m
->bus
, "/org/freedesktop/systemd1", &bus_manager_vtable
, m
) ||
478 !dbus_connection_register_fallback(m
->bus
, "/org/freedesktop/systemd1/unit", &bus_unit_vtable
, m
) ||
479 !dbus_connection_register_fallback(m
->bus
, "/org/freedesktop/systemd1/job", &bus_job_vtable
, m
) ||
480 !dbus_connection_add_filter(m
->bus
, bus_message_filter
, m
, NULL
) ||
481 !dbus_connection_add_filter(m
->system_bus
, system_bus_message_filter
, m
, NULL
)) {
486 dbus_bus_add_match(m
->bus
,
488 "sender='"DBUS_SERVICE_DBUS
"',"
489 "interface='"DBUS_INTERFACE_DBUS
"',"
490 "path='"DBUS_PATH_DBUS
"'",
493 if (dbus_error_is_set(&error
)) {
494 log_error("Failed to register match: %s", error
.message
);
495 dbus_error_free(&error
);
499 if ((r
= request_name(m
)) < 0) {
504 dbus_bus_add_match(m
->system_bus
,
506 "interface='org.freedesktop.systemd1.Agent',"
507 "path='/org/freedesktop/systemd1/agent'",
510 if (dbus_error_is_set(&error
)) {
511 log_error("Failed to register match: %s", error
.message
);
512 dbus_error_free(&error
);
517 log_debug("Successfully connected to D-Bus bus %s as %s",
518 strnull((id
= dbus_connection_get_server_id(m
->bus
))),
519 strnull(dbus_bus_get_unique_name(m
->bus
)));
522 log_debug("Successfully connected to system D-Bus bus %s as %s",
523 strnull((id
= dbus_connection_get_server_id(m
->system_bus
))),
524 strnull(dbus_bus_get_unique_name(m
->system_bus
)));
527 m
->request_bus_dispatch
= true;
528 m
->request_system_bus_dispatch
= true;
533 void bus_done(Manager
*m
) {
536 if (m
->system_bus
&& m
->system_bus
!= m
->bus
) {
537 dbus_connection_close(m
->system_bus
);
538 dbus_connection_unref(m
->system_bus
);
539 m
->system_bus
= NULL
;
543 dbus_connection_close(m
->bus
);
544 dbus_connection_unref(m
->bus
);
551 while ((c
= set_steal_first(m
->subscribed
)))
554 set_free(m
->subscribed
);
555 m
->subscribed
= NULL
;
559 DBusHandlerResult
bus_default_message_handler(Manager
*m
, DBusMessage
*message
, const char*introspection
, const BusProperty
*properties
) {
561 DBusMessage
*reply
= NULL
;
567 dbus_error_init(&error
);
569 if (dbus_message_is_method_call(message
, "org.freedesktop.DBus.Introspectable", "Introspect") && introspection
) {
571 if (!(reply
= dbus_message_new_method_return(message
)))
574 if (!dbus_message_append_args(reply
, DBUS_TYPE_STRING
, &introspection
, DBUS_TYPE_INVALID
))
577 } else if (dbus_message_is_method_call(message
, "org.freedesktop.DBus.Properties", "Get") && properties
) {
578 const char *interface
, *property
;
579 const BusProperty
*p
;
581 if (!dbus_message_get_args(
584 DBUS_TYPE_STRING
, &interface
,
585 DBUS_TYPE_STRING
, &property
,
587 return bus_send_error_reply(m
, message
, &error
, -EINVAL
);
589 for (p
= properties
; p
->property
; p
++)
590 if (streq(p
->interface
, interface
) && streq(p
->property
, property
))
594 DBusMessageIter iter
, sub
;
596 if (!(reply
= dbus_message_new_method_return(message
)))
599 dbus_message_iter_init_append(reply
, &iter
);
601 if (!dbus_message_iter_open_container(&iter
, DBUS_TYPE_VARIANT
, p
->signature
, &sub
))
604 if ((r
= p
->append(m
, &sub
, property
, p
->data
)) < 0) {
609 dbus_message_unref(reply
);
610 return bus_send_error_reply(m
, message
, NULL
, r
);
613 if (!dbus_message_iter_close_container(&iter
, &sub
))
616 } else if (dbus_message_is_method_call(message
, "org.freedesktop.DBus.Properties", "GetAll") && properties
) {
617 const char *interface
;
618 const BusProperty
*p
;
619 DBusMessageIter iter
, sub
, sub2
, sub3
;
622 if (!dbus_message_get_args(
625 DBUS_TYPE_STRING
, &interface
,
627 return bus_send_error_reply(m
, message
, &error
, -EINVAL
);
629 if (!(reply
= dbus_message_new_method_return(message
)))
632 dbus_message_iter_init_append(reply
, &iter
);
634 if (!dbus_message_iter_open_container(&iter
, DBUS_TYPE_ARRAY
, "{sv}", &sub
))
637 for (p
= properties
; p
->property
; p
++) {
638 if (!streq(p
->interface
, interface
))
641 if (!dbus_message_iter_open_container(&sub
, DBUS_TYPE_DICT_ENTRY
, NULL
, &sub2
) ||
642 !dbus_message_iter_append_basic(&sub2
, DBUS_TYPE_STRING
, &p
->property
) ||
643 !dbus_message_iter_open_container(&sub2
, DBUS_TYPE_VARIANT
, p
->signature
, &sub3
))
646 if ((r
= p
->append(m
, &sub3
, p
->property
, p
->data
)) < 0) {
651 dbus_message_unref(reply
);
652 return bus_send_error_reply(m
, message
, NULL
, r
);
655 if (!dbus_message_iter_close_container(&sub2
, &sub3
) ||
656 !dbus_message_iter_close_container(&sub
, &sub2
))
662 if (!dbus_message_iter_close_container(&iter
, &sub
))
667 if (!dbus_connection_send(m
->bus
, reply
, NULL
))
670 dbus_message_unref(reply
);
671 return DBUS_HANDLER_RESULT_HANDLED
;
674 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED
;
678 dbus_message_unref(reply
);
680 dbus_error_free(&error
);
682 return DBUS_HANDLER_RESULT_NEED_MEMORY
;
685 static const char *error_to_dbus(int error
) {
690 return DBUS_ERROR_INVALID_ARGS
;
693 return DBUS_ERROR_NO_MEMORY
;
697 return DBUS_ERROR_ACCESS_DENIED
;
700 return DBUS_ERROR_UNIX_PROCESS_ID_UNKNOWN
;
703 return DBUS_ERROR_FILE_NOT_FOUND
;
706 return DBUS_ERROR_FILE_EXISTS
;
709 return DBUS_ERROR_TIMEOUT
;
712 return DBUS_ERROR_IO_ERROR
;
717 return DBUS_ERROR_DISCONNECTED
;
720 return DBUS_ERROR_FAILED
;
723 DBusHandlerResult
bus_send_error_reply(Manager
*m
, DBusMessage
*message
, DBusError
*bus_error
, int error
) {
724 DBusMessage
*reply
= NULL
;
725 const char *name
, *text
;
727 if (bus_error
&& dbus_error_is_set(bus_error
)) {
728 name
= bus_error
->name
;
729 text
= bus_error
->message
;
731 name
= error_to_dbus(error
);
732 text
= strerror(-error
);
735 if (!(reply
= dbus_message_new_error(message
, name
, text
)))
738 if (!dbus_connection_send(m
->bus
, reply
, NULL
))
741 dbus_message_unref(reply
);
744 dbus_error_free(bus_error
);
746 return DBUS_HANDLER_RESULT_HANDLED
;
750 dbus_message_unref(reply
);
753 dbus_error_free(bus_error
);
755 return DBUS_HANDLER_RESULT_NEED_MEMORY
;
758 int bus_property_append_string(Manager
*m
, DBusMessageIter
*i
, const char *property
, void *data
) {
759 const char *t
= data
;
768 if (!dbus_message_iter_append_basic(i
, DBUS_TYPE_STRING
, &t
))
774 int bus_property_append_strv(Manager
*m
, DBusMessageIter
*i
, const char *property
, void *data
) {
782 if (!dbus_message_iter_open_container(i
, DBUS_TYPE_ARRAY
, "s", &sub
))
786 if (!dbus_message_iter_append_basic(&sub
, DBUS_TYPE_STRING
, t
))
789 if (!dbus_message_iter_close_container(i
, &sub
))
795 int bus_property_append_bool(Manager
*m
, DBusMessageIter
*i
, const char *property
, void *data
) {
806 if (!dbus_message_iter_append_basic(i
, DBUS_TYPE_BOOLEAN
, &db
))
812 int bus_property_append_uint64(Manager
*m
, DBusMessageIter
*i
, const char *property
, void *data
) {
818 if (!dbus_message_iter_append_basic(i
, DBUS_TYPE_UINT64
, data
))
824 int bus_property_append_uint32(Manager
*m
, DBusMessageIter
*i
, const char *property
, void *data
) {
830 if (!dbus_message_iter_append_basic(i
, DBUS_TYPE_UINT32
, data
))