test/relay.c

   1 /* Regression test for passing unmodified messages between connections
   2  *
   3  * Author: Simon McVittie <simon.mcvittie@collabora.co.uk>
   4  * Copyright © 2010-2011 Nokia Corporation
   5  *
   6  * Permission is hereby granted, free of charge, to any person
   7  * obtaining a copy of this software and associated documentation files
   8  * (the "Software"), to deal in the Software without restriction,
   9  * including without limitation the rights to use, copy, modify, merge,
  10  * publish, distribute, sublicense, and/or sell copies of the Software,
  11  * and to permit persons to whom the Software is furnished to do so,
  12  * subject to the following conditions:
  13  *
  14  * The above copyright notice and this permission notice shall be
  15  * included in all copies or substantial portions of the Software.
  16  *
  17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  18  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  19  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  20  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  21  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  22  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  23  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  24  * SOFTWARE.
  25  */
  26
  27 #include <config.h>
  28
  29 #include <glib.h>
  30
  31 #include <dbus/dbus.h>
  32 #include <dbus/dbus-glib-lowlevel.h>
  33
  34 /* This is basically a miniature dbus-daemon. We relay messages from the client
  35  * on the left to the client on the right.
  36  *
  37  * left      socket     left      dispatch     right    socket     right
  38  * client ===========>  server --------------> server ===========> client
  39  * conn                 conn                   conn                conn
  40  *
  41  * In the real dbus-daemon, the client connections would be out-of-process,
  42  * but here we're cheating and doing everything in-process.
  43  */
  44
  45 typedef struct {
  46     DBusError e;
  47
  48     DBusServer *server;
  49
  50     DBusConnection *left_client_conn;
  51     DBusConnection *left_server_conn;
  52
  53     DBusConnection *right_server_conn;
  54     DBusConnection *right_client_conn;
  55     /* queue of DBusMessage received by right_client_conn */
  56     GQueue messages;
  57 } Fixture;
  58
  59 static void
  60 assert_no_error (const DBusError *e)
  61 {
  62   if (G_UNLIKELY (dbus_error_is_set (e)))
  63     g_error ("expected success but got error: %s: %s", e->name, e->message);
  64 }
  65
  66 static DBusHandlerResult
  67 server_message_cb (DBusConnection *server_conn,
  68     DBusMessage *message,
  69     void *data)
  70 {
  71   Fixture *f = data;
  72
  73   g_assert (server_conn == f->left_server_conn);
  74   g_assert (f->right_server_conn != NULL);
  75
  76   dbus_connection_send (f->right_server_conn, message, NULL);
  77
  78   return DBUS_HANDLER_RESULT_HANDLED;
  79 }
  80
  81 static DBusHandlerResult
  82 right_client_message_cb (DBusConnection *client_conn,
  83     DBusMessage *message,
  84     void *data)
  85 {
  86   Fixture *f = data;
  87
  88   g_assert (client_conn == f->right_client_conn);
  89   g_queue_push_tail (&f->messages, dbus_message_ref (message));
  90
  91   return DBUS_HANDLER_RESULT_HANDLED;
  92 }
  93
  94 static void
  95 new_conn_cb (DBusServer *server,
  96     DBusConnection *server_conn,
  97     void *data)
  98 {
  99   Fixture *f = data;
 100   dbus_bool_t have_mem;
 101
 102   if (f->left_server_conn == NULL)
 103     {
 104       f->left_server_conn = dbus_connection_ref (server_conn);
 105
 106       have_mem = dbus_connection_add_filter (server_conn,
 107           server_message_cb, f, NULL);
 108       g_assert (have_mem);
 109     }
 110   else
 111     {
 112       g_assert (f->right_server_conn == NULL);
 113       f->right_server_conn = dbus_connection_ref (server_conn);
 114     }
 115
 116   dbus_connection_setup_with_g_main (server_conn, NULL);
 117 }
 118
 119 static void
 120 setup (Fixture *f,
 121     gconstpointer data G_GNUC_UNUSED)
 122 {
 123   dbus_error_init (&f->e);
 124   g_queue_init (&f->messages);
 125
 126   f->server = dbus_server_listen ("tcp:host=127.0.0.1", &f->e);
 127   assert_no_error (&f->e);
 128   g_assert (f->server != NULL);
 129
 130   dbus_server_set_new_connection_function (f->server,
 131       new_conn_cb, f, NULL);
 132   dbus_server_setup_with_g_main (f->server, NULL);
 133 }
 134
 135 static void
 136 test_connect (Fixture *f,
 137     gconstpointer data G_GNUC_UNUSED)
 138 {
 139   dbus_bool_t have_mem;
 140   char *address;
 141
 142   g_assert (f->left_server_conn == NULL);
 143   g_assert (f->right_server_conn == NULL);
 144
 145   address = dbus_server_get_address (f->server);
 146   g_assert (address != NULL);
 147
 148   f->left_client_conn = dbus_connection_open_private (address, &f->e);
 149   assert_no_error (&f->e);
 150   g_assert (f->left_client_conn != NULL);
 151   dbus_connection_setup_with_g_main (f->left_client_conn, NULL);
 152
 153   while (f->left_server_conn == NULL)
 154     {
 155       g_print (".");
 156       g_main_context_iteration (NULL, TRUE);
 157     }
 158
 159   f->right_client_conn = dbus_connection_open_private (address, &f->e);
 160   assert_no_error (&f->e);
 161   g_assert (f->right_client_conn != NULL);
 162   dbus_connection_setup_with_g_main (f->right_client_conn, NULL);
 163
 164   dbus_free (address);
 165
 166   while (f->right_server_conn == NULL)
 167     {
 168       g_print (".");
 169       g_main_context_iteration (NULL, TRUE);
 170     }
 171
 172   have_mem = dbus_connection_add_filter (f->right_client_conn,
 173       right_client_message_cb, f, NULL);
 174   g_assert (have_mem);
 175 }
 176
 177 static dbus_uint32_t
 178 send_one (Fixture *f,
 179     const char *member)
 180 {
 181   dbus_bool_t have_mem;
 182   dbus_uint32_t serial;
 183   DBusMessage *outgoing;
 184
 185   outgoing = dbus_message_new_signal ("/com/example/Hello",
 186       "com.example.Hello", member);
 187   g_assert (outgoing != NULL);
 188
 189   have_mem = dbus_connection_send (f->left_client_conn, outgoing, &serial);
 190   g_assert (have_mem);
 191   g_assert (serial != 0);
 192
 193   dbus_message_unref (outgoing);
 194   return serial;
 195 }
 196
 197 static void
 198 test_relay (Fixture *f,
 199     gconstpointer data)
 200 {
 201   DBusMessage *incoming;
 202
 203   test_connect (f, data);
 204
 205   send_one (f, "First");
 206   send_one (f, "Second");
 207
 208   while (g_queue_get_length (&f->messages) < 2)
 209     {
 210       g_print (".");
 211       g_main_context_iteration (NULL, TRUE);
 212     }
 213
 214   g_assert_cmpuint (g_queue_get_length (&f->messages), ==, 2);
 215
 216   incoming = g_queue_pop_head (&f->messages);
 217   g_assert_cmpstr (dbus_message_get_member (incoming), ==, "First");
 218   dbus_message_unref (incoming);
 219
 220   incoming = g_queue_pop_head (&f->messages);
 221   g_assert_cmpstr (dbus_message_get_member (incoming), ==, "Second");
 222   dbus_message_unref (incoming);
 223 }
 224
 225 /* An arbitrary number of messages */
 226 #define MANY 8192
 227
 228 static void
 229 test_limit (Fixture *f,
 230     gconstpointer data)
 231 {
 232   DBusMessage *incoming;
 233   guint i;
 234
 235   test_connect (f, data);
 236
 237   /* This was an attempt to reproduce fd.o #34393. It didn't work. */
 238   g_test_bug ("34393");
 239   dbus_connection_set_max_received_size (f->left_server_conn, 1);
 240   g_main_context_iteration (NULL, TRUE);
 241
 242   for (i = 0; i < MANY; i++)
 243     {
 244       gchar *buf = g_strdup_printf ("Message%u", i);
 245
 246       send_one (f, buf);
 247       g_free (buf);
 248     }
 249
 250   i = 0;
 251
 252   while (i < MANY)
 253     {
 254       while (g_queue_is_empty (&f->messages))
 255         {
 256           g_main_context_iteration (NULL, TRUE);
 257         }
 258
 259       while ((incoming = g_queue_pop_head (&f->messages)) != NULL)
 260         {
 261           i++;
 262           dbus_message_unref (incoming);
 263         }
 264     }
 265 }
 266
 267 static void
 268 teardown (Fixture *f,
 269     gconstpointer data G_GNUC_UNUSED)
 270 {
 271   if (f->left_client_conn != NULL)
 272     {
 273       dbus_connection_close (f->left_client_conn);
 274       dbus_connection_unref (f->left_client_conn);
 275       f->left_client_conn = NULL;
 276     }
 277
 278   if (f->right_client_conn != NULL)
 279     {
 280       dbus_connection_close (f->right_client_conn);
 281       dbus_connection_unref (f->right_client_conn);
 282       f->right_client_conn = NULL;
 283     }
 284
 285   if (f->left_server_conn != NULL)
 286     {
 287       dbus_connection_close (f->left_server_conn);
 288       dbus_connection_unref (f->left_server_conn);
 289       f->left_server_conn = NULL;
 290     }
 291
 292   if (f->right_server_conn != NULL)
 293     {
 294       dbus_connection_close (f->right_server_conn);
 295       dbus_connection_unref (f->right_server_conn);
 296       f->right_server_conn = NULL;
 297     }
 298
 299   if (f->server != NULL)
 300     {
 301       dbus_server_disconnect (f->server);
 302       dbus_server_unref (f->server);
 303       f->server = NULL;
 304     }
 305 }
 306
 307 int
 308 main (int argc,
 309     char **argv)
 310 {
 311   g_test_init (&argc, &argv, NULL);
 312   g_test_bug_base ("https://bugs.freedesktop.org/show_bug.cgi?id=");
 313
 314   g_test_add ("/connect", Fixture, NULL, setup,
 315       test_connect, teardown);
 316   g_test_add ("/relay", Fixture, NULL, setup,
 317       test_relay, teardown);
 318   g_test_add ("/limit", Fixture, NULL, setup,
 319       test_limit, teardown);
 320
 321   return g_test_run ();
 322 }