5 * Copyright (C) 2004-2011 Marcel Holtmann <marcel@holtmann.org>
8 * This program is free software; you can redistribute it and/or modify
9 * it 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 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
29 #include <dbus/dbus.h>
33 #define DISPATCH_TIMEOUT 0
39 struct timeout_handler {
50 struct disconnect_data {
51 GDBusWatchFunction function;
55 static gboolean disconnected_signal(DBusConnection *conn,
56 DBusMessage *msg, void *data)
58 struct disconnect_data *dc_data = data;
60 error("Got disconnected from the system message bus");
62 dc_data->function(conn, dc_data->user_data);
64 dbus_connection_unref(conn);
69 static gboolean message_dispatch(void *data)
71 DBusConnection *conn = data;
73 dbus_connection_ref(conn);
75 /* Dispatch messages */
76 while (dbus_connection_dispatch(conn) == DBUS_DISPATCH_DATA_REMAINS);
78 dbus_connection_unref(conn);
83 static inline void queue_dispatch(DBusConnection *conn,
84 DBusDispatchStatus status)
86 if (status == DBUS_DISPATCH_DATA_REMAINS)
87 g_idle_add(message_dispatch, dbus_connection_ref(conn));
90 static gboolean watch_func(GIOChannel *chan, GIOCondition cond, gpointer data)
92 struct watch_info *info = data;
93 unsigned int flags = 0;
94 DBusDispatchStatus status;
97 if (cond & G_IO_IN) flags |= DBUS_WATCH_READABLE;
98 if (cond & G_IO_OUT) flags |= DBUS_WATCH_WRITABLE;
99 if (cond & G_IO_HUP) flags |= DBUS_WATCH_HANGUP;
100 if (cond & G_IO_ERR) flags |= DBUS_WATCH_ERROR;
102 /* Protect connection from being destroyed by dbus_watch_handle */
103 conn = dbus_connection_ref(info->conn);
105 dbus_watch_handle(info->watch, flags);
107 status = dbus_connection_get_dispatch_status(conn);
108 queue_dispatch(conn, status);
110 dbus_connection_unref(conn);
115 static void watch_info_free(void *data)
117 struct watch_info *info = data;
120 g_source_remove(info->id);
124 dbus_connection_unref(info->conn);
129 static dbus_bool_t add_watch(DBusWatch *watch, void *data)
131 DBusConnection *conn = data;
132 GIOCondition cond = G_IO_HUP | G_IO_ERR;
134 struct watch_info *info;
138 if (!dbus_watch_get_enabled(watch))
141 info = g_new0(struct watch_info, 1);
143 fd = dbus_watch_get_unix_fd(watch);
144 chan = g_io_channel_unix_new(fd);
147 info->conn = dbus_connection_ref(conn);
149 dbus_watch_set_data(watch, info, watch_info_free);
151 flags = dbus_watch_get_flags(watch);
153 if (flags & DBUS_WATCH_READABLE) cond |= G_IO_IN;
154 if (flags & DBUS_WATCH_WRITABLE) cond |= G_IO_OUT;
156 info->id = g_io_add_watch(chan, cond, watch_func, info);
158 g_io_channel_unref(chan);
163 static void remove_watch(DBusWatch *watch, void *data)
165 if (dbus_watch_get_enabled(watch))
168 /* will trigger watch_info_free() */
169 dbus_watch_set_data(watch, NULL, NULL);
172 static void watch_toggled(DBusWatch *watch, void *data)
174 /* Because we just exit on OOM, enable/disable is
175 * no different from add/remove */
176 if (dbus_watch_get_enabled(watch))
177 add_watch(watch, data);
179 remove_watch(watch, data);
182 static gboolean timeout_handler_dispatch(gpointer data)
184 struct timeout_handler *handler = data;
188 /* if not enabled should not be polled by the main loop */
189 if (!dbus_timeout_get_enabled(handler->timeout))
192 dbus_timeout_handle(handler->timeout);
197 static void timeout_handler_free(void *data)
199 struct timeout_handler *handler = data;
201 if (handler->id > 0) {
202 g_source_remove(handler->id);
209 static dbus_bool_t add_timeout(DBusTimeout *timeout, void *data)
211 int interval = dbus_timeout_get_interval(timeout);
212 struct timeout_handler *handler;
214 if (!dbus_timeout_get_enabled(timeout))
217 handler = g_new0(struct timeout_handler, 1);
219 handler->timeout = timeout;
221 dbus_timeout_set_data(timeout, handler, timeout_handler_free);
223 handler->id = g_timeout_add(interval, timeout_handler_dispatch,
229 static void remove_timeout(DBusTimeout *timeout, void *data)
231 /* will trigger timeout_handler_free() */
232 dbus_timeout_set_data(timeout, NULL, NULL);
235 static void timeout_toggled(DBusTimeout *timeout, void *data)
237 if (dbus_timeout_get_enabled(timeout))
238 add_timeout(timeout, data);
240 remove_timeout(timeout, data);
243 static void dispatch_status(DBusConnection *conn,
244 DBusDispatchStatus status, void *data)
246 if (!dbus_connection_get_is_connected(conn))
249 queue_dispatch(conn, status);
252 static inline void setup_dbus_with_main_loop(DBusConnection *conn)
254 dbus_connection_set_watch_functions(conn, add_watch, remove_watch,
255 watch_toggled, conn, NULL);
257 dbus_connection_set_timeout_functions(conn, add_timeout, remove_timeout,
258 timeout_toggled, NULL, NULL);
260 dbus_connection_set_dispatch_status_function(conn, dispatch_status,
264 static gboolean setup_bus(DBusConnection *conn, const char *name,
268 DBusDispatchStatus status;
271 result = g_dbus_request_name(conn, name, error);
274 if (dbus_error_is_set(error) == TRUE)
282 setup_dbus_with_main_loop(conn);
284 status = dbus_connection_get_dispatch_status(conn);
285 queue_dispatch(conn, status);
290 DBusConnection *g_dbus_setup_bus(DBusBusType type, const char *name,
293 DBusConnection *conn;
295 conn = dbus_bus_get(type, error);
298 if (dbus_error_is_set(error) == TRUE)
305 if (setup_bus(conn, name, error) == FALSE) {
306 dbus_connection_unref(conn);
313 DBusConnection *g_dbus_setup_private(DBusBusType type, const char *name,
316 DBusConnection *conn;
318 conn = dbus_bus_get_private(type, error);
321 if (dbus_error_is_set(error) == TRUE)
328 if (setup_bus(conn, name, error) == FALSE) {
329 dbus_connection_close(conn);
330 dbus_connection_unref(conn);
337 gboolean g_dbus_request_name(DBusConnection *connection, const char *name,
342 result = dbus_bus_request_name(connection, name,
343 DBUS_NAME_FLAG_DO_NOT_QUEUE, error);
346 if (dbus_error_is_set(error) == TRUE)
350 if (result != DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER) {
352 dbus_set_error(error, name, "Name already in use");
360 gboolean g_dbus_set_disconnect_function(DBusConnection *connection,
361 GDBusWatchFunction function,
362 void *user_data, DBusFreeFunction destroy)
364 struct disconnect_data *dc_data;
366 dc_data = g_new0(struct disconnect_data, 1);
368 dc_data->function = function;
369 dc_data->user_data = user_data;
371 dbus_connection_set_exit_on_disconnect(connection, FALSE);
373 if (g_dbus_add_signal_watch(connection, NULL, NULL,
374 DBUS_INTERFACE_LOCAL, "Disconnected",
375 disconnected_signal, dc_data, g_free) == 0) {
376 error("Failed to add watch for D-Bus Disconnected signal");