test/loopback.c

   1 /* Simple sanity-check for loopback through TCP and Unix sockets.
   2  *
   3  * Author: Simon McVittie <simon.mcvittie@collabora.co.uk>
   4  * Copyright © 2010-2012 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 #include <string.h>
  35
  36 typedef struct {
  37     DBusError e;
  38
  39     DBusServer *server;
  40     DBusConnection *server_conn;
  41     /* queue of DBusMessage */
  42     GQueue server_messages;
  43
  44     DBusConnection *client_conn;
  45 } Fixture;
  46
  47 static void
  48 assert_no_error (const DBusError *e)
  49 {
  50   if (G_UNLIKELY (dbus_error_is_set (e)))
  51     g_error ("expected success but got error: %s: %s", e->name, e->message);
  52 }
  53
  54 static DBusHandlerResult
  55 server_message_cb (DBusConnection *server_conn,
  56     DBusMessage *message,
  57     void *data)
  58 {
  59   Fixture *f = data;
  60
  61   g_assert (server_conn == f->server_conn);
  62   g_queue_push_tail (&f->server_messages, dbus_message_ref (message));
  63
  64   return DBUS_HANDLER_RESULT_HANDLED;
  65 }
  66
  67 static void
  68 new_conn_cb (DBusServer *server,
  69     DBusConnection *server_conn,
  70     void *data)
  71 {
  72   Fixture *f = data;
  73   dbus_bool_t have_mem;
  74
  75   g_assert (f->server_conn == NULL);
  76   f->server_conn = dbus_connection_ref (server_conn);
  77   dbus_connection_setup_with_g_main (server_conn, NULL);
  78
  79   have_mem = dbus_connection_add_filter (server_conn,
  80       server_message_cb, f, NULL);
  81   g_assert (have_mem);
  82 }
  83
  84 static void
  85 setup (Fixture *f,
  86     gconstpointer addr)
  87 {
  88   dbus_error_init (&f->e);
  89   g_queue_init (&f->server_messages);
  90
  91   f->server = dbus_server_listen (addr, &f->e);
  92   assert_no_error (&f->e);
  93   g_assert (f->server != NULL);
  94
  95   dbus_server_set_new_connection_function (f->server,
  96       new_conn_cb, f, NULL);
  97   dbus_server_setup_with_g_main (f->server, NULL);
  98 }
  99
 100 static void
 101 test_connect (Fixture *f,
 102     gconstpointer addr G_GNUC_UNUSED)
 103 {
 104   g_assert (f->server_conn == NULL);
 105
 106   f->client_conn = dbus_connection_open_private (
 107       dbus_server_get_address (f->server), &f->e);
 108   assert_no_error (&f->e);
 109   g_assert (f->client_conn != NULL);
 110   dbus_connection_setup_with_g_main (f->client_conn, NULL);
 111
 112   while (f->server_conn == NULL)
 113     {
 114       g_print (".");
 115       g_main_context_iteration (NULL, TRUE);
 116     }
 117 }
 118
 119 static void
 120 test_bad_guid (Fixture *f,
 121     gconstpointer addr G_GNUC_UNUSED)
 122 {
 123   DBusMessage *incoming;
 124   gchar *address = g_strdup (dbus_server_get_address (f->server));
 125   gchar *guid;
 126
 127   g_test_bug ("39720");
 128
 129   g_assert (f->server_conn == NULL);
 130
 131   g_assert (strstr (address, "guid=") != NULL);
 132   guid = strstr (address, "guid=");
 133   g_assert_cmpuint (strlen (guid), >=, 5 + 32);
 134
 135   /* Change the first char of the guid to something different */
 136   if (guid[5] == '0')
 137     guid[5] = 'f';
 138   else
 139     guid[5] = '0';
 140
 141   f->client_conn = dbus_connection_open_private (address, &f->e);
 142   assert_no_error (&f->e);
 143   g_assert (f->client_conn != NULL);
 144   dbus_connection_setup_with_g_main (f->client_conn, NULL);
 145
 146   while (f->server_conn == NULL)
 147     {
 148       g_print (".");
 149       g_main_context_iteration (NULL, TRUE);
 150     }
 151
 152   /* We get disconnected */
 153
 154   while (g_queue_is_empty (&f->server_messages))
 155     {
 156       g_print (".");
 157       g_main_context_iteration (NULL, TRUE);
 158     }
 159
 160   g_assert_cmpuint (g_queue_get_length (&f->server_messages), ==, 1);
 161
 162   incoming = g_queue_pop_head (&f->server_messages);
 163
 164   g_assert (!dbus_message_contains_unix_fds (incoming));
 165   g_assert_cmpstr (dbus_message_get_destination (incoming), ==, NULL);
 166   g_assert_cmpstr (dbus_message_get_error_name (incoming), ==, NULL);
 167   g_assert_cmpstr (dbus_message_get_interface (incoming), ==,
 168       DBUS_INTERFACE_LOCAL);
 169   g_assert_cmpstr (dbus_message_get_member (incoming), ==, "Disconnected");
 170   g_assert_cmpstr (dbus_message_get_sender (incoming), ==, NULL);
 171   g_assert_cmpstr (dbus_message_get_signature (incoming), ==, "");
 172   g_assert_cmpstr (dbus_message_get_path (incoming), ==, DBUS_PATH_LOCAL);
 173
 174   dbus_message_unref (incoming);
 175
 176   g_free (address);
 177 }
 178
 179 static void
 180 test_message (Fixture *f,
 181     gconstpointer addr)
 182 {
 183   dbus_bool_t have_mem;
 184   dbus_uint32_t serial;
 185   DBusMessage *outgoing, *incoming;
 186
 187   test_connect (f, addr);
 188
 189   outgoing = dbus_message_new_signal ("/com/example/Hello",
 190       "com.example.Hello", "Greeting");
 191   g_assert (outgoing != NULL);
 192
 193   have_mem = dbus_connection_send (f->client_conn, outgoing, &serial);
 194   g_assert (have_mem);
 195   g_assert (serial != 0);
 196
 197   while (g_queue_is_empty (&f->server_messages))
 198     {
 199       g_print (".");
 200       g_main_context_iteration (NULL, TRUE);
 201     }
 202
 203   g_assert_cmpuint (g_queue_get_length (&f->server_messages), ==, 1);
 204
 205   incoming = g_queue_pop_head (&f->server_messages);
 206
 207   g_assert (!dbus_message_contains_unix_fds (incoming));
 208   g_assert_cmpstr (dbus_message_get_destination (incoming), ==, NULL);
 209   g_assert_cmpstr (dbus_message_get_error_name (incoming), ==, NULL);
 210   g_assert_cmpstr (dbus_message_get_interface (incoming), ==,
 211       "com.example.Hello");
 212   g_assert_cmpstr (dbus_message_get_member (incoming), ==, "Greeting");
 213   g_assert_cmpstr (dbus_message_get_sender (incoming), ==, NULL);
 214   g_assert_cmpstr (dbus_message_get_signature (incoming), ==, "");
 215   g_assert_cmpstr (dbus_message_get_path (incoming), ==, "/com/example/Hello");
 216   g_assert_cmpuint (dbus_message_get_serial (incoming), ==, serial);
 217
 218   dbus_message_unref (incoming);
 219
 220   dbus_message_unref (outgoing);
 221 }
 222
 223 static void
 224 teardown (Fixture *f,
 225     gconstpointer addr G_GNUC_UNUSED)
 226 {
 227   if (f->client_conn != NULL)
 228     {
 229       dbus_connection_close (f->client_conn);
 230       dbus_connection_unref (f->client_conn);
 231       f->client_conn = NULL;
 232     }
 233
 234   if (f->server_conn != NULL)
 235     {
 236       dbus_connection_close (f->server_conn);
 237       dbus_connection_unref (f->server_conn);
 238       f->server_conn = NULL;
 239     }
 240
 241   if (f->server != NULL)
 242     {
 243       dbus_server_disconnect (f->server);
 244       dbus_server_unref (f->server);
 245       f->server = NULL;
 246     }
 247 }
 248
 249 int
 250 main (int argc,
 251     char **argv)
 252 {
 253   g_test_init (&argc, &argv, NULL);
 254   g_test_bug_base ("https://bugs.freedesktop.org/show_bug.cgi?id=");
 255
 256   g_test_add ("/connect/tcp", Fixture, "tcp:host=127.0.0.1", setup,
 257       test_connect, teardown);
 258   g_test_add ("/message/tcp", Fixture, "tcp:host=127.0.0.1", setup,
 259       test_message, teardown);
 260
 261   g_test_add ("/connect/nonce-tcp", Fixture, "nonce-tcp:host=127.0.0.1", setup,
 262       test_connect, teardown);
 263   g_test_add ("/message/nonce-tcp", Fixture, "nonce-tcp:host=127.0.0.1", setup,
 264       test_message, teardown);
 265
 266 #ifdef DBUS_UNIX
 267   g_test_add ("/connect/unix", Fixture, "unix:tmpdir=/tmp", setup,
 268       test_connect, teardown);
 269   g_test_add ("/message/unix", Fixture, "unix:tmpdir=/tmp", setup,
 270       test_message, teardown);
 271 #endif
 272
 273   g_test_add ("/message/bad-guid", Fixture, "tcp:host=127.0.0.1", setup,
 274       test_bad_guid, teardown);
 275
 276   return g_test_run ();
 277 }