Logo Search packages:      
Sourcecode: obexd version File versions  Download package

mainloop.c

/*
 *
 *  D-Bus helper library
 *
 *  Copyright (C) 2004-2009  Marcel Holtmann <marcel@holtmann.org>
 *
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <stdint.h>

#include <glib.h>
#include <dbus/dbus.h>

#ifdef NEED_DBUS_WATCH_GET_UNIX_FD
#define dbus_watch_get_unix_fd dbus_watch_get_fd
#endif

#include "gdbus.h"

#define DISPATCH_TIMEOUT  0

#define info(fmt...)
#define error(fmt...)
#define debug(fmt...)

typedef struct {
      uint32_t id;
      DBusTimeout *timeout;
} timeout_handler_t;

struct watch_info {
      guint watch_id;
      GIOChannel *io;
      DBusConnection *conn;
};

struct server_info {
      guint watch_id;
      GIOChannel *io;
      DBusServer *server;
};

struct disconnect_data {
      GDBusWatchFunction disconnect_cb;
      void *user_data;
};

static DBusHandlerResult disconnect_filter(DBusConnection *conn,
                                    DBusMessage *msg, void *data)
{
      struct disconnect_data *dc_data = data;

      if (dbus_message_is_signal(msg,
                  DBUS_INTERFACE_LOCAL, "Disconnected") == TRUE) {
            error("Got disconnected from the system message bus");
            dc_data->disconnect_cb(conn, dc_data->user_data);
            dbus_connection_unref(conn);
      }

      return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}

static gboolean message_dispatch_cb(void *data)
{
      DBusConnection *connection = data;

      dbus_connection_ref(connection);

      /* Dispatch messages */
      while (dbus_connection_dispatch(connection) == DBUS_DISPATCH_DATA_REMAINS);

      dbus_connection_unref(connection);

      return FALSE;
}

static gboolean watch_func(GIOChannel *chan, GIOCondition cond, gpointer data)
{
      DBusWatch *watch = data;
      struct watch_info *info = dbus_watch_get_data(watch);
      int flags = 0;

      if (cond & G_IO_IN)  flags |= DBUS_WATCH_READABLE;
      if (cond & G_IO_OUT) flags |= DBUS_WATCH_WRITABLE;
      if (cond & G_IO_HUP) flags |= DBUS_WATCH_HANGUP;
      if (cond & G_IO_ERR) flags |= DBUS_WATCH_ERROR;

      dbus_watch_handle(watch, flags);

      if (dbus_connection_get_dispatch_status(info->conn) == DBUS_DISPATCH_DATA_REMAINS)
            g_timeout_add(DISPATCH_TIMEOUT, message_dispatch_cb, info->conn);

      return TRUE;
}

static dbus_bool_t add_watch(DBusWatch *watch, void *data)
{
      GIOCondition cond = G_IO_HUP | G_IO_ERR;
      DBusConnection *conn = data;
      struct watch_info *info;
      int fd, flags;

      if (!dbus_watch_get_enabled(watch))
            return TRUE;

      info = g_new(struct watch_info, 1);

      fd = dbus_watch_get_unix_fd(watch);
      info->io = g_io_channel_unix_new(fd);
      info->conn = dbus_connection_ref(conn);

      dbus_watch_set_data(watch, info, NULL);

      flags = dbus_watch_get_flags(watch);

      if (flags & DBUS_WATCH_READABLE) cond |= G_IO_IN;
      if (flags & DBUS_WATCH_WRITABLE) cond |= G_IO_OUT;

      info->watch_id = g_io_add_watch(info->io, cond, watch_func, watch);

      return TRUE;
}

static void remove_watch(DBusWatch *watch, void *data)
{
      struct watch_info *info = dbus_watch_get_data(watch);

      dbus_watch_set_data(watch, NULL, NULL);

      if (info) {
            g_source_remove(info->watch_id);
            g_io_channel_unref(info->io);
            dbus_connection_unref(info->conn);
            g_free(info);
      }
}

static void watch_toggled(DBusWatch *watch, void *data)
{
      /* Because we just exit on OOM, enable/disable is
       * no different from add/remove */
      if (dbus_watch_get_enabled(watch))
            add_watch(watch, data);
      else
            remove_watch(watch, data);
}

static gboolean timeout_handler_dispatch(gpointer data)
{
      timeout_handler_t *handler = data;

      /* if not enabled should not be polled by the main loop */
      if (dbus_timeout_get_enabled(handler->timeout) != TRUE)
            return FALSE;

      dbus_timeout_handle(handler->timeout);

      return FALSE;
}

static void timeout_handler_free(void *data)
{
      timeout_handler_t *handler = data;
      if (!handler)
            return;

      g_source_remove(handler->id);
      g_free(handler);
}

static dbus_bool_t add_timeout(DBusTimeout *timeout, void *data)
{
      timeout_handler_t *handler;

      if (!dbus_timeout_get_enabled(timeout))
            return TRUE;

      handler = g_new0(timeout_handler_t, 1);

      handler->timeout = timeout;
      handler->id = g_timeout_add(dbus_timeout_get_interval(timeout),
                              timeout_handler_dispatch, handler);

      dbus_timeout_set_data(timeout, handler, timeout_handler_free);

      return TRUE;
}

static void remove_timeout(DBusTimeout *timeout, void *data)
{
}

static void timeout_toggled(DBusTimeout *timeout, void *data)
{
      if (dbus_timeout_get_enabled(timeout))
            add_timeout(timeout, data);
      else
            remove_timeout(timeout, data);
}

static void dispatch_status_cb(DBusConnection *conn,
                        DBusDispatchStatus new_status, void *data)
{
      if (!dbus_connection_get_is_connected(conn))
            return;

      if (new_status == DBUS_DISPATCH_DATA_REMAINS)
            g_timeout_add(DISPATCH_TIMEOUT, message_dispatch_cb, data);
}

static void setup_dbus_with_main_loop(DBusConnection *conn)
{
      dbus_connection_set_watch_functions(conn, add_watch, remove_watch,
                                    watch_toggled, conn, NULL);

      dbus_connection_set_timeout_functions(conn, add_timeout, remove_timeout,
                                    timeout_toggled, conn, NULL);

      dbus_connection_set_dispatch_status_function(conn, dispatch_status_cb,
                                                conn, NULL);
}

DBusConnection *g_dbus_setup_bus(DBusBusType type, const char *name,
                                          DBusError *error)
{
      DBusConnection *conn;

      conn = dbus_bus_get(type, error);

      if (error != NULL) {
            if (dbus_error_is_set(error) == TRUE)
                  return NULL;
      }

      if (conn == NULL)
            return NULL;

      if (name != NULL) {
            if (dbus_bus_request_name(conn, name,
                        DBUS_NAME_FLAG_DO_NOT_QUEUE, error) !=
                        DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER ) {
                  dbus_connection_unref(conn);
                  return NULL;
            }

            if (error != NULL) {
                  if (dbus_error_is_set(error) == TRUE) {
                        dbus_connection_unref(conn);
                        return NULL;
                  }
            }
      }

      setup_dbus_with_main_loop(conn);

      return conn;
}

gboolean g_dbus_request_name(DBusConnection *connection, const char *name,
                                          DBusError *error)
{
      return TRUE;
}

gboolean g_dbus_check_service(DBusConnection *connection, const char *name)
{
      DBusMessage *message, *reply;
      const char **names;
      int i, count;
      gboolean result = FALSE;

      message = dbus_message_new_method_call(DBUS_SERVICE_DBUS,
                  DBUS_PATH_DBUS, DBUS_INTERFACE_DBUS, "ListNames");
      if (message == NULL) {
            error("Can't allocate new message");
            return FALSE;
      }

      reply = dbus_connection_send_with_reply_and_block(connection,
                                          message, -1, NULL);

      dbus_message_unref(message);

      if (reply == NULL) {
            error("Failed to execute method call");
            return FALSE;
      }

      if (dbus_message_get_args(reply, NULL,
                        DBUS_TYPE_ARRAY, DBUS_TYPE_STRING,
                        &names, &count, DBUS_TYPE_INVALID) == FALSE) {
            error("Failed to read name list");
            goto done;
      }

      for (i = 0; i < count; i++)
            if (g_str_equal(names[i], name) == TRUE) {
                  result = TRUE;
                  break;
            }

done:
      dbus_message_unref(reply);

      return result;
}

gboolean g_dbus_set_disconnect_function(DBusConnection *connection,
                        GDBusWatchFunction function,
                        void *user_data, DBusFreeFunction destroy)
{
      struct disconnect_data *dc_data;

      dc_data = g_new(struct disconnect_data, 1);

      dc_data->disconnect_cb = function;
      dc_data->user_data = user_data;

      dbus_connection_set_exit_on_disconnect(connection, FALSE);

      if (dbus_connection_add_filter(connection, disconnect_filter,
                                    dc_data, g_free) == FALSE) {
            error("Can't add D-Bus disconnect filter");
            g_free(dc_data);
            return FALSE;
      }

      return TRUE;
}

Generated by  Doxygen 1.6.0   Back to index