bus/dispatch.c

   1 /* -*- mode: C; c-file-style: "gnu" -*- */
   2 /* dispatch.c  Message dispatcher
   3  *
   4  * Copyright (C) 2003  CodeFactory AB
   5  *
   6  * Licensed under the Academic Free License version 1.2
   7  *
   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.
  12  *
  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.
  17  *
  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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  21  *
  22  */
  23
  24 #include "dispatch.h"
  25 #include "connection.h"
  26 #include "driver.h"
  27 #include "utils.h"
  28 #include <dbus/dbus-internals.h>
  29 #include <string.h>
  30
  31 static int message_handler_slot;
  32
  33 static void
  34 send_one_message (DBusConnection *connection, void *data)
  35 {
  36   BUS_HANDLE_OOM (dbus_connection_send_message (connection, data, NULL, NULL));
  37 }
  38
  39 void
  40 bus_dispatch_broadcast_message (DBusMessage *message)
  41 {
  42   _dbus_assert (dbus_message_get_sender (message) != NULL);
  43
  44   bus_connection_foreach (send_one_message, message);
  45
  46 }
  47
  48 static DBusHandlerResult
  49 bus_dispatch_message_handler (DBusMessageHandler *handler,
  50                               DBusConnection     *connection,
  51                               DBusMessage        *message,
  52                               void               *user_data)
  53 {
  54   const char *sender, *service_name;
  55
  56   /* Assign a sender to the message */
  57   sender = bus_connection_get_name (connection);
  58   BUS_HANDLE_OOM (dbus_message_set_sender (message, sender));
  59
  60   service_name = dbus_message_get_service (message);
  61
  62   /* See if the message is to the driver */
  63   if (strcmp (service_name, DBUS_SERVICE_DBUS) == 0)
  64     {
  65       bus_driver_handle_message (connection, message);
  66     }
  67   else if (sender == NULL)
  68     {
  69       _dbus_verbose ("Received message from non-registered client. Disconnecting.\n");
  70       dbus_connection_disconnect (connection);
  71     }
  72   else if (strcmp (service_name, DBUS_SERVICE_BROADCAST) == 0)
  73     {
  74       bus_dispatch_broadcast_message (message);
  75     }
  76   else
  77     {
  78       DBusString service_string;
  79       BusService *service;
  80
  81       _dbus_string_init_const (&service_string, service_name);
  82       service = bus_service_lookup (&service_string, FALSE);
  83
  84       _dbus_assert (bus_service_get_primary_owner (service) != NULL);
  85
  86       /* Dispatch the message */
  87       BUS_HANDLE_OOM (dbus_connection_send_message (bus_service_get_primary_owner (service),
  88                                                     message, NULL, NULL));
  89     }
  90
  91   return DBUS_HANDLER_RESULT_ALLOW_MORE_HANDLERS;
  92 }
  93
  94 dbus_bool_t
  95 bus_dispatch_add_connection (DBusConnection *connection)
  96 {
  97   DBusMessageHandler *handler;
  98
  99   message_handler_slot = dbus_connection_allocate_data_slot ();
 100
 101   if (message_handler_slot < 0)
 102     return FALSE;
 103
 104   handler = dbus_message_handler_new (bus_dispatch_message_handler, NULL, NULL);
 105
 106   if (!dbus_connection_add_filter (connection, handler))
 107     {
 108       dbus_message_handler_unref (handler);
 109
 110       return FALSE;
 111     }
 112
 113   if (!dbus_connection_set_data (connection,
 114                                  message_handler_slot,
 115                                  handler,
 116                                  (DBusFreeFunction)dbus_message_handler_unref))
 117     {
 118       dbus_connection_remove_filter (connection, handler);
 119       dbus_message_handler_unref (handler);
 120
 121       return FALSE;
 122     }
 123
 124   return TRUE;
 125 }
 126
 127 void
 128 bus_dispatch_remove_connection (DBusConnection *connection)
 129 {
 130   /* Here we tell the bus driver that we want to get off. */
 131   bus_driver_remove_connection (connection);
 132
 133   dbus_connection_set_data (connection,
 134                             message_handler_slot,
 135                             NULL, NULL);
 136 }
 137