[glib] Check that auth methods work and interoperate with libdbus-1

[David Zeuthen <davidz src gnome org>]

commit b9d1fe7db23c14440c64a3878c9461162b948826
Author: David Zeuthen <davidz redhat com>
Date:   Wed Apr 11 23:34:08 2012 -0400

    Check that auth methods work and interoperate with libdbus-1
    
    See https://bugzilla.gnome.org/show_bug.cgi?id=673943
    
    Signed-off-by: David Zeuthen <davidz redhat com>

 gio/tests/Makefile.am  |    7 +
 gio/tests/gdbus-auth.c |  292 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 299 insertions(+), 0 deletions(-)
---
diff --git a/gio/tests/Makefile.am b/gio/tests/Makefile.am
index 0d45e71..cb48a34 100644
--- a/gio/tests/Makefile.am
+++ b/gio/tests/Makefile.am
@@ -265,6 +265,13 @@ gdbus_serialization_CFLAGS = $(DBUS1_CFLAGS)
 gdbus_serialization_LDADD = $(progs_ldadd) $(DBUS1_LIBS)
 endif
 
+if HAVE_DBUS1
+TEST_PROGS += gdbus-auth
+gdbus_auth_SOURCES = gdbus-auth.c gdbus-sessionbus.c gdbus-sessionbus.h gdbus-tests.h gdbus-tests.c
+gdbus_auth_CFLAGS = $(DBUS1_CFLAGS)
+gdbus_auth_LDADD = $(progs_ldadd) $(DBUS1_LIBS)
+endif
+
 gdbus_addresses_SOURCES = gdbus-addresses.c
 gdbus_addresses_LDADD = $(progs_ldadd)
 
diff --git a/gio/tests/gdbus-auth.c b/gio/tests/gdbus-auth.c
new file mode 100644
index 0000000..3b24b09
--- /dev/null
+++ b/gio/tests/gdbus-auth.c
@@ -0,0 +1,292 @@
+/* GLib testing framework examples and tests
+ *
+ * Copyright (C) 2008-2012 Red Hat, Inc.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General
+ * Public License along with this library; if not, write to the
+ * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
+ * Boston, MA 02111-1307, USA.
+ *
+ * Author: David Zeuthen <davidz redhat com>
+ */
+
+#include <locale.h>
+#include <gio/gio.h>
+
+#include <string.h>
+#include <unistd.h>
+#include <dbus/dbus.h>
+
+#include "gdbus-tests.h"
+
+#ifdef G_OS_UNIX
+#include <gio/gunixconnection.h>
+#include <gio/gnetworkingprivate.h>
+#include <gio/gunixsocketaddress.h>
+#include <gio/gunixfdlist.h>
+#endif
+
+/* ---------------------------------------------------------------------------------------------------- */
+
+static gboolean
+on_allow_mechanism (GDBusAuthObserver *observer,
+                    const gchar       *mechanism,
+                    gpointer           user_data)
+{
+  const gchar *mechanism_to_allow = user_data;
+  if (g_strcmp0 (mechanism, mechanism_to_allow) == 0)
+    return TRUE;
+  else
+    return FALSE;
+}
+
+static void
+auth_client_mechanism (const gchar *mechanism)
+{
+  gchar *address = NULL;
+  GDBusConnection *c = NULL;
+  GError *error = NULL;
+  GDBusAuthObserver *auth_observer;
+
+  address = g_dbus_address_get_for_bus_sync (G_BUS_TYPE_SESSION, NULL, &error);
+  g_assert_no_error (error);
+  g_assert (address != NULL);
+
+  auth_observer = g_dbus_auth_observer_new ();
+
+  g_signal_connect (auth_observer,
+                    "allow-mechanism",
+                    G_CALLBACK (on_allow_mechanism),
+                    (gpointer) mechanism);
+
+  c = g_dbus_connection_new_for_address_sync (address,
+                                              G_DBUS_CONNECTION_FLAGS_AUTHENTICATION_CLIENT |
+                                              G_DBUS_CONNECTION_FLAGS_MESSAGE_BUS_CONNECTION,
+                                              auth_observer,
+                                              NULL,
+                                              &error);
+  g_assert_no_error (error);
+  g_assert (c != NULL);
+
+  g_free (address);
+  g_object_unref (c);
+  g_object_unref (auth_observer);
+}
+
+static void
+auth_client_external (void)
+{
+  auth_client_mechanism ("EXTERNAL");
+}
+
+static void
+auth_client_dbus_cookie_sha1 (void)
+{
+  auth_client_mechanism ("DBUS_COOKIE_SHA1");
+}
+
+/* ---------------------------------------------------------------------------------------------------- */
+
+static gboolean
+on_new_connection (GDBusServer     *server,
+                   GDBusConnection *connection,
+                   gpointer         user_data)
+{
+  GMainLoop *loop = user_data;
+  g_main_loop_quit (loop);
+  return FALSE;
+}
+
+static gboolean
+on_timeout (gpointer user_data)
+{
+  g_error ("Timeout waiting for client");
+  g_assert_not_reached ();
+  return FALSE;
+}
+
+static gpointer
+dbus_1_client_thread_func (gpointer user_data)
+{
+  const gchar *address = user_data;
+  gchar *str_stdout;
+  gchar *str_stderr;
+  gint exit_status;
+  gboolean rc;
+  gchar *command_line;
+  GError *error;
+
+  // For the dbus part, just use dbus-monitor to connect(1)
+  command_line = g_strdup_printf ("dbus-monitor --address %s", address);
+
+  error = NULL;
+  rc = g_spawn_command_line_sync (command_line,
+                                  &str_stdout,
+                                  &str_stderr,
+                                  &exit_status,
+                                  &error);
+  g_assert_no_error (error);
+  g_assert (rc);
+  g_free (str_stdout);
+  g_free (str_stderr);
+  g_free (command_line);
+  return NULL;
+}
+
+static void
+auth_server_mechanism (const gchar *mechanism)
+{
+  gchar *addr;
+  gchar *guid;
+  GDBusServer *server;
+  GDBusAuthObserver *auth_observer;
+  GMainLoop *loop;
+  GError *error;
+  GThread *client_thread;
+  GDBusServerFlags flags;
+
+  /* Skip DBUS_COOKIE_SHA1 test unless we have a sufficiently new version of dbus-1 */
+  if (g_strcmp0 (mechanism, "DBUS_COOKIE_SHA1") == 0)
+    {
+      int dbus_major, dbus_minor, dbus_micro;
+      dbus_get_version (&dbus_major, &dbus_minor, &dbus_micro);
+      if (!((dbus_major == 1 && dbus_minor == 4 && dbus_micro >= 21) ||
+            (dbus_major == 1 && dbus_minor == 5 && dbus_micro >= 13) ||
+            (dbus_major == 1 && dbus_minor > 5) ||
+            (dbus_major > 1)))
+        {
+          g_print ("Your libdbus-1 library is too old (version %d.%d.%d) so skipping DBUS_COOKIE_SHA1 test. See https://bugs.freedesktop.org/show_bug.cgi?id=48580 for more details.\n",
+                   dbus_major, dbus_minor, dbus_micro);
+          return;
+        }
+    }
+
+  guid = g_dbus_generate_guid ();
+
+#ifdef G_OS_UNIX
+  if (g_unix_socket_address_abstract_names_supported ())
+    {
+      addr = g_strdup ("unix:tmpdir=/tmp/gdbus-test-");
+    }
+  else
+    {
+      gchar *tmpdir;
+      tmpdir = g_dir_make_tmp ("gdbus-test-XXXXXX", NULL);
+      addr = g_strdup_printf ("unix:tmpdir=%s", tmpdir);
+      g_free (tmpdir);
+    }
+#else
+  addr = g_strdup ("nonce-tcp:");
+#endif
+
+  loop = g_main_loop_new (NULL, FALSE);
+
+  auth_observer = g_dbus_auth_observer_new ();
+
+  flags = G_DBUS_SERVER_FLAGS_NONE;
+  if (g_strcmp0 (mechanism, "ANONYMOUS") == 0)
+    flags |= G_DBUS_SERVER_FLAGS_AUTHENTICATION_ALLOW_ANONYMOUS;
+
+  error = NULL;
+  server = g_dbus_server_new_sync (addr,
+                                   flags,
+                                   guid,
+                                   auth_observer,
+                                   NULL, /* cancellable */
+                                   &error);
+  g_assert_no_error (error);
+  g_assert (server != NULL);
+
+  g_signal_connect (auth_observer,
+                    "allow-mechanism",
+                    G_CALLBACK (on_allow_mechanism),
+                    (gpointer) mechanism);
+
+  g_signal_connect (server,
+                    "new-connection",
+                    G_CALLBACK (on_new_connection),
+                    loop);
+
+  g_dbus_server_start (server);
+
+  g_timeout_add_seconds (5, on_timeout, NULL);
+
+  /* run the libdbus-1 client in a thread */
+  client_thread = g_thread_new ("dbus-1-client-thread",
+                                dbus_1_client_thread_func,
+                                (gpointer) g_dbus_server_get_client_address (server));
+
+  g_main_loop_run (loop);
+
+  g_dbus_server_stop (server);
+
+  g_thread_join (client_thread);
+
+  g_free (addr);
+  g_free (guid);
+  g_object_unref (auth_observer);
+  g_object_unref (server);
+}
+
+static void
+auth_server_anonymous (void)
+{
+  auth_server_mechanism ("ANONYMOUS");
+}
+
+static void
+auth_server_external (void)
+{
+  auth_server_mechanism ("EXTERNAL");
+}
+
+static void
+auth_server_dbus_cookie_sha1 (void)
+{
+  auth_server_mechanism ("DBUS_COOKIE_SHA1");
+}
+
+/* ---------------------------------------------------------------------------------------------------- */
+
+int
+main (int   argc,
+      char *argv[])
+{
+  gint ret;
+
+  setlocale (LC_ALL, "C");
+
+  g_type_init ();
+  g_test_init (&argc, &argv, NULL);
+
+  /* all the tests use a session bus with a well-known address that we can bring up and down
+   * using session_bus_up() and session_bus_down().
+   */
+  g_unsetenv ("DISPLAY");
+  g_setenv ("DBUS_SESSION_BUS_ADDRESS", session_bus_get_temporary_address (), TRUE);
+  session_bus_up ();
+
+  g_test_add_func ("/gdbus/auth/client/EXTERNAL",         auth_client_external);
+  g_test_add_func ("/gdbus/auth/client/DBUS_COOKIE_SHA1", auth_client_dbus_cookie_sha1);
+  g_test_add_func ("/gdbus/auth/server/ANONYMOUS",        auth_server_anonymous);
+  g_test_add_func ("/gdbus/auth/server/EXTERNAL",         auth_server_external);
+  g_test_add_func ("/gdbus/auth/server/DBUS_COOKIE_SHA1", auth_server_dbus_cookie_sha1);
+
+  ret = g_test_run();
+
+  /* tear down bus */
+  session_bus_down ();
+
+  return ret;
+}
+