gio/tests/gdbus-unix-addresses.c

   1 /* GLib testing framework examples and tests
   2  *
   3  * Copyright © 2015 Collabora Ltd.
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General
  16  * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19 #include <glib.h>
  20
  21 #ifndef G_OS_UNIX
  22 #error This is a Unix-specific test
  23 #endif
  24
  25 #include <errno.h>
  26
  27 #include <glib/gstdio.h>
  28 #include <gio/gio.h>
  29 #include <gio/gunixsocketaddress.h>
  30
  31 static void
  32 print_address (void)
  33 {
  34   GError *error = NULL;
  35   gchar *addr;
  36
  37   addr = g_dbus_address_get_for_bus_sync (G_BUS_TYPE_SESSION, NULL,
  38       &error);
  39
  40   g_assert_no_error (error);
  41   g_assert_nonnull (addr);
  42   g_print ("%s\n", addr);
  43   g_free (addr);
  44 }
  45
  46 static GSocket *mock_bus = NULL;
  47 static gchar *mock_bus_path = NULL;
  48 /* this is deliberately something that needs escaping */
  49 static gchar tmpdir[] = "/tmp/gdbus,unix,test.XXXXXX";
  50
  51 static void
  52 set_up_mock_xdg_runtime_dir (void)
  53 {
  54   GError *error = NULL;
  55   GSocketAddress *addr;
  56
  57   mock_bus = g_socket_new (G_SOCKET_FAMILY_UNIX, G_SOCKET_TYPE_STREAM, 0,
  58       &error);
  59   g_assert_no_error (error);
  60   g_assert_true (G_IS_SOCKET (mock_bus));
  61
  62   /* alters tmpdir in-place */
  63   if (g_mkdtemp_full (tmpdir, 0700) == NULL)
  64     {
  65       int errsv = errno;
  66       g_error ("g_mkdtemp_full: %s", g_strerror (errsv));
  67     }
  68
  69   mock_bus_path = g_strconcat (tmpdir, "/bus", NULL);
  70   addr = g_unix_socket_address_new (mock_bus_path);
  71   g_socket_bind (mock_bus, addr, FALSE, &error);
  72   g_assert_no_error (error);
  73   g_object_unref (addr);
  74
  75   g_setenv ("XDG_RUNTIME_DIR", tmpdir, TRUE);
  76 }
  77
  78 static void
  79 tear_down_mock_xdg_runtime_dir (void)
  80 {
  81   GError *error = NULL;
  82
  83   g_socket_close (mock_bus, &error);
  84   g_assert_no_error (error);
  85
  86   if (g_unlink (mock_bus_path) < 0)
  87     {
  88       int errsv = errno;
  89       g_error ("g_unlink(\"%s\"): %s", mock_bus_path, g_strerror (errsv));
  90     }
  91
  92   if (g_rmdir (tmpdir) < 0)
  93     {
  94       int errsv = errno;
  95       g_error ("g_rmdir(\"%s\"): %s", tmpdir, g_strerror (errsv));
  96     }
  97
  98   g_clear_object (&mock_bus);
  99   g_clear_pointer (&mock_bus_path, g_free);
 100 }
 101
 102 static gchar *path = NULL;
 103
 104 static void
 105 set_up_mock_dbus_launch (void)
 106 {
 107   path = g_strconcat (g_test_get_dir (G_TEST_BUILT), ":",
 108       g_getenv ("PATH"), NULL);
 109   g_setenv ("PATH", path, TRUE);
 110
 111   /* libdbus won't even try X11 autolaunch if DISPLAY is unset; GDBus
 112    * does the same in Debian derivatives (proposed upstream in
 113    * GNOME#723506) */
 114   g_setenv ("DISPLAY", "an unrealistic mock X11 display", TRUE);
 115 }
 116
 117 static void
 118 tear_down_mock_dbus_launch (void)
 119 {
 120   g_clear_pointer (&path, g_free);
 121 }
 122
 123 static void
 124 test_x11_autolaunch (void)
 125 {
 126   if (g_test_subprocess ())
 127     {
 128       g_unsetenv ("DISPLAY");
 129       g_unsetenv ("DBUS_SESSION_BUS_ADDRESS");
 130       g_unsetenv ("XDG_RUNTIME_DIR");
 131       g_unsetenv ("G_MESSAGES_DEBUG");
 132       set_up_mock_dbus_launch ();
 133
 134       print_address ();
 135
 136       tear_down_mock_dbus_launch ();
 137       return;
 138     }
 139
 140   g_test_trap_subprocess (NULL, 0, 0);
 141   g_test_trap_assert_stderr_unmatched ("?*");
 142   g_test_trap_assert_stdout ("hello:this=address-is-from-the,mock=dbus-launch\n");
 143   g_test_trap_assert_passed ();
 144 }
 145
 146 static void
 147 test_xdg_runtime (void)
 148 {
 149   if (g_test_subprocess ())
 150     {
 151       g_unsetenv ("DISPLAY");
 152       g_unsetenv ("DBUS_SESSION_BUS_ADDRESS");
 153       set_up_mock_xdg_runtime_dir ();
 154       set_up_mock_dbus_launch ();
 155
 156       print_address ();
 157
 158       tear_down_mock_dbus_launch ();
 159       tear_down_mock_xdg_runtime_dir ();
 160       return;
 161     }
 162
 163   g_test_trap_subprocess (NULL, 0, 0);
 164   g_test_trap_assert_stderr_unmatched ("?*");
 165   g_test_trap_assert_stdout ("unix:path=/tmp/gdbus%2Cunix%2Ctest.*/bus\n");
 166   g_test_trap_assert_passed ();
 167 }
 168
 169 int
 170 main (int   argc,
 171       char *argv[])
 172 {
 173   g_test_init (&argc, &argv, NULL);
 174
 175   g_test_add_func ("/gdbus/x11-autolaunch", test_x11_autolaunch);
 176   g_test_add_func ("/gdbus/xdg-runtime", test_xdg_runtime);
 177
 178   return g_test_run();
 179 }