| /* 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); |
| |
| 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; |
| } |
| |