test/test-service.c

   1
   2 #include "test-utils.h"
   3
   4 static DBusLoop *loop;
   5 static dbus_bool_t already_quit = FALSE;
   6
   7 static void
   8 quit (void)
   9 {
  10   if (!already_quit)
  11     {
  12       _dbus_loop_quit (loop);
  13       already_quit = TRUE;
  14     }
  15 }
  16
  17 static void
  18 die (const char *message)
  19 {
  20   fprintf (stderr, "*** %s", message);
  21   exit (1);
  22 }
  23
  24 static DBusHandlerResult
  25 handle_echo (DBusConnection     *connection,
  26              DBusMessage        *message)
  27 {
  28   DBusError error;
  29   DBusMessage *reply;
  30   DBusMessageIter iter;
  31   char *s;
  32
  33   dbus_error_init (&error);
  34
  35   if (!dbus_message_get_args (message,
  36                               &error,
  37                               DBUS_TYPE_STRING, &s,
  38                               DBUS_TYPE_INVALID))
  39     {
  40       reply = dbus_message_new_error (message,
  41                                       error.name,
  42                                       error.message);
  43
  44       if (reply == NULL)
  45         die ("No memory\n");
  46
  47       if (!dbus_connection_send (connection, reply, NULL))
  48         die ("No memory\n");
  49
  50       dbus_message_unref (reply);
  51
  52       return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
  53     }
  54
  55   reply = dbus_message_new_method_return (message);
  56   if (reply == NULL)
  57     die ("No memory\n");
  58
  59   dbus_message_append_iter_init (message, &iter);
  60
  61   if (!dbus_message_iter_append_string (&iter, s))
  62     die ("No memory");
  63
  64   if (!dbus_connection_send (connection, reply, NULL))
  65     die ("No memory\n");
  66
  67   dbus_free (s);
  68
  69   dbus_message_unref (reply);
  70
  71   return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
  72 }
  73
  74 static DBusHandlerResult
  75 filter_func (DBusMessageHandler *handler,
  76              DBusConnection     *connection,
  77              DBusMessage        *message,
  78              void               *user_data)
  79 {
  80   if (dbus_message_is_method_call (message,
  81                                    "org.freedesktop.TestSuite",
  82                                    "Echo"))
  83     return handle_echo (connection, message);
  84   else if (dbus_message_is_method_call (message,
  85                                         "org.freedesktop.TestSuite",
  86                                         "Exit") ||
  87            dbus_message_is_signal (message,
  88                                    DBUS_INTERFACE_ORG_FREEDESKTOP_LOCAL,
  89                                    "Disconnected"))
  90     {
  91       dbus_connection_disconnect (connection);
  92       quit ();
  93       return DBUS_HANDLER_RESULT_HANDLED;
  94     }
  95   else
  96     {
  97       return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
  98     }
  99 }
 100
 101 int
 102 main (int    argc,
 103       char **argv)
 104 {
 105   DBusConnection *connection;
 106   DBusError error;
 107   DBusMessageHandler *handler;
 108   int result;
 109
 110   dbus_error_init (&error);
 111   connection = dbus_bus_get (DBUS_BUS_ACTIVATION, &error);
 112   if (connection == NULL)
 113     {
 114       _dbus_verbose ("*** Failed to open connection to activating message bus: %s\n",
 115                      error.message);
 116       dbus_error_free (&error);
 117       return 1;
 118     }
 119
 120   loop = _dbus_loop_new ();
 121   if (loop == NULL)
 122     die ("No memory\n");
 123
 124   if (!test_connection_setup (loop, connection))
 125     die ("No memory\n");
 126
 127   handler = dbus_message_handler_new (filter_func, NULL, NULL);
 128   if (handler == NULL)
 129     die ("No memory");
 130
 131   if (!dbus_connection_add_filter (connection, handler))
 132     die ("No memory");
 133
 134   result = dbus_bus_acquire_service (connection, "org.freedesktop.DBus.TestSuiteEchoService",
 135                                      0, &error);
 136   if (dbus_error_is_set (&error))
 137     {
 138       _dbus_verbose ("*** Failed to acquire service: %s\n",
 139                      error.message);
 140       dbus_error_free (&error);
 141       exit (1);
 142     }
 143
 144   _dbus_verbose ("*** Test service entering main loop\n");
 145   _dbus_loop_run (loop);
 146
 147   test_connection_shutdown (loop, connection);
 148
 149   dbus_connection_unref (connection);
 150
 151   dbus_message_handler_unref (handler);
 152
 153   _dbus_loop_unref (loop);
 154   loop = NULL;
 155
 156   dbus_shutdown ();
 157
 158   _dbus_verbose ("*** Test service exiting\n");
 159
 160   return 0;
 161 }