test/test-shell-service.c

   1 #include <config.h>
   2
   3 #include "test-utils.h"
   4
   5 static DBusLoop *loop;
   6 static dbus_bool_t already_quit = FALSE;
   7 static const char* echo_path = "/org/freedesktop/TestSuite";
   8
   9 typedef struct
  10 {
  11   int argc;
  12   char **argv;
  13 } EchoData;
  14
  15 static void
  16 quit (void)
  17 {
  18   if (!already_quit)
  19     {
  20       _dbus_loop_quit (loop);
  21       already_quit = TRUE;
  22     }
  23 }
  24
  25 static void
  26 die (const char *message)
  27 {
  28   fprintf (stderr, "*** test-service: %s", message);
  29   exit (1);
  30 }
  31
  32 static DBusHandlerResult
  33 handle_echo (DBusConnection     *connection,
  34              DBusMessage        *message)
  35 {
  36   DBusError error;
  37   DBusMessage *reply;
  38   DBusMessageIter iter;
  39   int i;
  40   EchoData *d;
  41
  42   _dbus_verbose ("sending reply to Echo method\n");
  43
  44   if (!dbus_connection_get_object_path_data (connection, echo_path, (void **)&d))
  45       die ("No memory");
  46
  47
  48   dbus_error_init (&error);
  49
  50   reply = dbus_message_new_method_return (message);
  51   if (reply == NULL)
  52     die ("No memory\n");
  53
  54   dbus_message_iter_init_append (reply, &iter);
  55   for (i = 0; i < d->argc; ++i)
  56     if (!dbus_message_iter_append_basic (&iter, DBUS_TYPE_STRING, &(d->argv[i])))
  57       die ("No memory\n");
  58
  59   if (!dbus_connection_send (connection, reply, NULL))
  60     die ("No memory\n");
  61
  62   fprintf (stderr, "Shell echo service echoed the command line\n");
  63
  64   dbus_message_unref (reply);
  65
  66   return DBUS_HANDLER_RESULT_HANDLED;
  67 }
  68
  69 static void
  70 path_unregistered_func (DBusConnection  *connection,
  71                         void            *user_data)
  72 {
  73   /* connection was finalized */
  74 }
  75
  76 static DBusHandlerResult
  77 path_message_func (DBusConnection  *connection,
  78                    DBusMessage     *message,
  79                    void            *user_data)
  80 {
  81   if (dbus_message_is_method_call (message,
  82                                    "org.freedesktop.TestSuite",
  83                                    "Echo"))
  84     return handle_echo (connection, message);
  85   else if (dbus_message_is_method_call (message,
  86                                         "org.freedesktop.TestSuite",
  87                                         "Exit"))
  88     {
  89       quit ();
  90       return DBUS_HANDLER_RESULT_HANDLED;
  91     }
  92   else
  93     return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
  94 }
  95
  96 static DBusObjectPathVTable
  97 echo_vtable = {
  98   path_unregistered_func,
  99   path_message_func,
 100   NULL,
 101 };
 102
 103 static DBusHandlerResult
 104 filter_func (DBusConnection     *connection,
 105              DBusMessage        *message,
 106              void               *user_data)
 107 {
 108   if (dbus_message_is_signal (message,
 109                               DBUS_INTERFACE_LOCAL,
 110                               "Disconnected"))
 111     {
 112       quit ();
 113       return DBUS_HANDLER_RESULT_HANDLED;
 114     }
 115   else
 116     {
 117       return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 118     }
 119 }
 120
 121 int
 122 main (int    argc,
 123       char **argv)
 124 {
 125   DBusConnection *connection;
 126   DBusError error;
 127   EchoData echo_data;
 128   int result;
 129
 130   echo_data.argc = argc;
 131   echo_data.argv = argv;
 132
 133   dbus_error_init (&error);
 134   connection = dbus_bus_get (DBUS_BUS_STARTER, &error);
 135   if (connection == NULL)
 136     {
 137       fprintf (stderr, "*** Failed to open connection to activating message bus: %s\n",
 138                error.message);
 139       dbus_error_free (&error);
 140       return 1;
 141     }
 142
 143   loop = _dbus_loop_new ();
 144   if (loop == NULL)
 145     die ("No memory\n");
 146
 147   if (!test_connection_setup (loop, connection))
 148     die ("No memory\n");
 149
 150   if (!dbus_connection_add_filter (connection,
 151                                    filter_func, NULL, NULL))
 152     die ("No memory");
 153
 154   if (!dbus_connection_register_object_path (connection,
 155                                              echo_path,
 156                                              &echo_vtable,
 157                                              (void*) &echo_data))
 158     die ("No memory");
 159
 160   {
 161     void *d;
 162     if (!dbus_connection_get_object_path_data (connection, echo_path, &d))
 163       die ("No memory");
 164     if (d != (void*) &echo_data)
 165       die ("dbus_connection_get_object_path_data() doesn't seem to work right\n");
 166   }
 167
 168   result = dbus_bus_request_name (connection, "org.freedesktop.DBus.TestSuiteShellEchoServiceSuccess",
 169                                   0, &error);
 170   if (dbus_error_is_set (&error))
 171     {
 172       fprintf (stderr, "Error %s\n", error.message);
 173       _dbus_verbose ("*** Failed to acquire service: %s\n",
 174                      error.message);
 175       dbus_error_free (&error);
 176       exit (1);
 177     }
 178
 179   if (result != DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER)
 180     {
 181       fprintf (stderr, "Unable to acquire service: code %d\n", result);
 182       _dbus_verbose ("*** Failed to acquire service: %d\n", result);
 183       exit (1);
 184     }
 185
 186   _dbus_verbose ("*** Test service entering main loop\n");
 187   _dbus_loop_run (loop);
 188
 189   test_connection_shutdown (loop, connection);
 190
 191   dbus_connection_remove_filter (connection, filter_func, NULL);
 192
 193   dbus_connection_unref (connection);
 194
 195   _dbus_loop_unref (loop);
 196   loop = NULL;
 197
 198   dbus_shutdown ();
 199
 200   _dbus_verbose ("*** Test service exiting\n");
 201
 202   return 0;
 203 }