diff --git a/gio/tests/Makefile.am b/gio/tests/Makefile.am index 0d45e718d..cb48a3453 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 000000000..3b24b092e --- /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 + */ + +#include +#include + +#include +#include +#include + +#include "gdbus-tests.h" + +#ifdef G_OS_UNIX +#include +#include +#include +#include +#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; +} +