test/name-test/test-privserver-client.c

   1 #include "../test-utils.h"
   2
   3 static void
   4 die (const char *message, ...)
   5 {
   6   va_list args;
   7   va_start (args, message);
   8   vfprintf (stderr, message, args);
   9   va_end (args);
  10   exit (1);
  11 }
  12
  13 static DBusHandlerResult
  14 filter_private_message (DBusConnection     *connection,
  15                         DBusMessage        *message,
  16                         void               *user_data)
  17 {
  18   if (dbus_message_is_signal (message,
  19                               DBUS_INTERFACE_LOCAL,
  20                               "Disconnected"))
  21     {
  22        DBusLoop *loop = user_data;
  23        _dbus_loop_quit (loop);
  24        return DBUS_HANDLER_RESULT_HANDLED;
  25     }
  26   return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
  27 }
  28
  29 static void
  30 open_shutdown_private_connection ()
  31 {
  32   DBusError error;
  33   DBusLoop *loop;
  34   DBusConnection *session;
  35   DBusMessage *msg;
  36   DBusMessage *reply;
  37   DBusConnection *privconn;
  38   const char *addr;
  39
  40   dbus_error_init (&error);
  41
  42   loop = _dbus_loop_new ();
  43
  44   session = dbus_bus_get (DBUS_BUS_SESSION, &error);
  45   if (!session)
  46     die ("couldn't access session bus\n");
  47   dbus_connection_set_exit_on_disconnect (session, FALSE);
  48   msg = dbus_message_new_method_call ("org.freedesktop.DBus.TestSuite.TestServer",
  49                                       "/",
  50                                       "org.freedesktop.DBus.TestSuite.TestServer",
  51                                       "GetPrivateAddress");
  52   if (!(reply = dbus_connection_send_with_reply_and_block (session, msg, -1, &error)))
  53     die ("couldn't send message: %s\n", error.message);
  54   dbus_message_unref (msg);
  55   if (!dbus_message_get_args (reply, &error, DBUS_TYPE_STRING, &addr, DBUS_TYPE_INVALID))
  56     die ("couldn't parse message replym\n");
  57   printf ("got private temp address %s\n", addr);
  58
  59   privconn = dbus_connection_open (addr, &error);
  60   if (!privconn)
  61     die ("couldn't connect to server direct connection: %s\n", error.message);
  62   dbus_message_unref (reply);
  63
  64   dbus_connection_set_exit_on_disconnect (privconn, FALSE);
  65   dbus_connection_add_filter (privconn, filter_private_message, loop, NULL);
  66   test_connection_setup (loop, privconn);
  67
  68   msg = dbus_message_new_method_call ("org.freedesktop.DBus.TestSuite.TestServer",
  69                                       "/",
  70                                       "org.freedesktop.DBus.TestSuite.TestServer",
  71                                       "Quit");
  72   if (!dbus_connection_send (session, msg, NULL))
  73     die ("couldn't send Quit message\n");
  74   dbus_message_unref (msg);
  75
  76   _dbus_loop_run (loop);
  77
  78   test_connection_shutdown (loop, session);
  79   dbus_connection_unref (session);
  80
  81   test_connection_shutdown (loop, privconn);
  82   dbus_connection_remove_filter (privconn, filter_private_message, loop);
  83   dbus_connection_unref (privconn);
  84
  85   _dbus_loop_unref (loop);
  86 }
  87
  88 int
  89 main (int argc, char *argv[])
  90 {
  91   open_shutdown_private_connection ();
  92
  93   dbus_shutdown ();
  94
  95   open_shutdown_private_connection ();
  96
  97   dbus_shutdown ();
  98
  99   return 0;
 100 }