1 /* Integration tests for the dbus-daemon
3 * Author: Simon McVittie <simon.mcvittie@collabora.co.uk>
4 * Copyright © 2010-2011 Nokia Corporation
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:
14 * The above copyright notice and this permission notice shall be
15 * included in all copies or substantial portions of the Software.
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
31 #include <dbus/dbus.h>
32 #include <dbus/dbus-glib-lowlevel.h>
42 # include <sys/types.h>
53 DBusConnection *left_conn;
55 DBusConnection *right_conn;
56 gboolean right_conn_echo;
59 #define assert_no_error(e) _assert_no_error (e, __FILE__, __LINE__)
61 _assert_no_error (const DBusError *e,
65 if (G_UNLIKELY (dbus_error_is_set (e)))
66 g_error ("%s:%d: expected success but got error: %s: %s",
67 file, line, e->name, e->message);
71 spawn_dbus_daemon (gchar *binary,
82 "--print-address=1", /* stdout */
86 g_spawn_async_with_pipes (NULL, /* working directory */
89 G_SPAWN_DO_NOT_REAP_CHILD | G_SPAWN_SEARCH_PATH,
90 NULL, /* child_setup */
93 NULL, /* child's stdin = /dev/null */
95 NULL, /* child's stderr = our stderr */
97 g_assert_no_error (error);
99 address = g_string_new (NULL);
101 /* polling until the dbus-daemon writes out its address is a bit stupid,
102 * but at least it's simple, unlike dbus-launch... in principle we could
103 * use select() here, but life's too short */
110 bytes = read (address_fd, buf, sizeof (buf));
113 g_string_append_len (address, buf, bytes);
115 newline = strchr (address->str, '\n');
119 g_string_truncate (address, newline - address->str);
123 g_usleep (G_USEC_PER_SEC / 10);
126 return g_string_free (address, FALSE);
129 static DBusConnection *
130 connect_to_bus (const gchar *address)
132 DBusConnection *conn;
133 DBusError error = DBUS_ERROR_INIT;
136 conn = dbus_connection_open_private (address, &error);
137 assert_no_error (&error);
138 g_assert (conn != NULL);
140 ok = dbus_bus_register (conn, &error);
141 assert_no_error (&error);
143 g_assert (dbus_bus_get_unique_name (conn) != NULL);
145 dbus_connection_setup_with_g_main (conn, NULL);
149 static DBusHandlerResult
150 echo_filter (DBusConnection *connection,
151 DBusMessage *message,
156 if (dbus_message_get_type (message) != DBUS_MESSAGE_TYPE_METHOD_CALL)
157 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
159 reply = dbus_message_new_method_return (message);
164 if (!dbus_connection_send (connection, reply, NULL))
167 dbus_message_unref (reply);
169 return DBUS_HANDLER_RESULT_HANDLED;
175 const char *config_file;
180 gconstpointer context)
182 const Config *config = context;
188 dbus_error_init (&f->e);
190 if (config != NULL && config->config_file != NULL)
192 if (g_getenv ("DBUS_TEST_DATA") == NULL)
194 g_message ("SKIP: set DBUS_TEST_DATA to a directory containing %s",
195 config->config_file);
200 arg = g_strdup_printf (
201 "--config-file=%s/%s",
202 g_getenv ("DBUS_TEST_DATA"), config->config_file);
204 else if (g_getenv ("DBUS_TEST_SYSCONFDIR") != NULL)
206 arg = g_strdup_printf ("--config-file=%s/dbus-1/session.conf",
207 g_getenv ("DBUS_TEST_SYSCONFDIR"));
209 else if (g_getenv ("DBUS_TEST_DATA") != NULL)
211 arg = g_strdup_printf (
212 "--config-file=%s/valid-config-files/session.conf",
213 g_getenv ("DBUS_TEST_DATA"));
217 arg = g_strdup ("--session");
220 dbus_daemon = g_strdup (g_getenv ("DBUS_TEST_DAEMON"));
222 if (dbus_daemon == NULL)
223 dbus_daemon = g_strdup ("dbus-daemon");
225 address = spawn_dbus_daemon (dbus_daemon, arg, &f->daemon_pid);
227 g_free (dbus_daemon);
230 f->left_conn = connect_to_bus (address);
231 f->right_conn = connect_to_bus (address);
236 add_echo_filter (Fixture *f)
238 if (!dbus_connection_add_filter (f->right_conn, echo_filter, NULL, NULL))
241 f->right_conn_echo = TRUE;
245 pc_count (DBusPendingCall *pc,
248 guint *received_p = data;
254 test_echo (Fixture *f,
255 gconstpointer context)
257 const Config *config = context;
266 if (config != NULL && config->bug_ref != NULL)
267 g_test_bug (config->bug_ref);
273 count = MAX (config->min_messages, count);
277 g_test_timer_start ();
279 for (sent = 0; sent < count; sent++)
281 DBusMessage *m = dbus_message_new_method_call (
282 dbus_bus_get_unique_name (f->right_conn), "/",
283 "com.example", "Spam");
289 if (!dbus_connection_send_with_reply (f->left_conn, m, &pc,
290 DBUS_TIMEOUT_INFINITE) ||
294 if (dbus_pending_call_get_completed (pc))
295 pc_count (pc, &received);
296 else if (!dbus_pending_call_set_notify (pc, pc_count, &received,
300 dbus_pending_call_unref (pc);
301 dbus_message_unref (m);
304 while (received < count)
305 g_main_context_iteration (NULL, TRUE);
307 elapsed = g_test_timer_elapsed ();
309 g_test_maximized_result (count / elapsed, "%u messages / %f seconds",
314 pending_call_store_reply (DBusPendingCall *pc,
317 DBusMessage **message_p = data;
319 *message_p = dbus_pending_call_steal_reply (pc);
320 g_assert (*message_p != NULL);
324 test_creds (Fixture *f,
325 gconstpointer context)
327 const char *unique = dbus_bus_get_unique_name (f->left_conn);
328 DBusMessage *m = dbus_message_new_method_call (DBUS_SERVICE_DBUS,
329 DBUS_PATH_DBUS, DBUS_INTERFACE_DBUS, "GetConnectionCredentials");
331 DBusMessageIter args_iter;
332 DBusMessageIter arr_iter;
333 DBusMessageIter pair_iter;
334 DBusMessageIter var_iter;
344 if (!dbus_message_append_args (m,
345 DBUS_TYPE_STRING, &unique,
349 if (!dbus_connection_send_with_reply (f->left_conn, m, &pc,
350 DBUS_TIMEOUT_USE_DEFAULT) ||
354 dbus_message_unref (m);
357 if (dbus_pending_call_get_completed (pc))
358 pending_call_store_reply (pc, &m);
359 else if (!dbus_pending_call_set_notify (pc, pending_call_store_reply,
364 g_main_context_iteration (NULL, TRUE);
366 g_assert_cmpstr (dbus_message_get_signature (m), ==, "a{sv}");
368 dbus_message_iter_init (m, &args_iter);
369 g_assert_cmpuint (dbus_message_iter_get_arg_type (&args_iter), ==,
371 g_assert_cmpuint (dbus_message_iter_get_element_type (&args_iter), ==,
372 DBUS_TYPE_DICT_ENTRY);
373 dbus_message_iter_recurse (&args_iter, &arr_iter);
375 while (dbus_message_iter_get_arg_type (&arr_iter) != DBUS_TYPE_INVALID)
379 dbus_message_iter_recurse (&arr_iter, &pair_iter);
380 g_assert_cmpuint (dbus_message_iter_get_arg_type (&pair_iter), ==,
382 dbus_message_iter_get_basic (&pair_iter, &name);
383 dbus_message_iter_next (&pair_iter);
384 g_assert_cmpuint (dbus_message_iter_get_arg_type (&pair_iter), ==,
386 dbus_message_iter_recurse (&pair_iter, &var_iter);
388 if (g_strcmp0 (name, "UnixUserID") == 0)
393 g_assert (!(seen & SEEN_UNIX_USER));
394 g_assert_cmpuint (dbus_message_iter_get_arg_type (&var_iter), ==,
396 dbus_message_iter_get_basic (&var_iter, &u32);
397 g_message ("%s of this process is %u", name, u32);
398 g_assert_cmpuint (u32, ==, geteuid ());
399 seen |= SEEN_UNIX_USER;
401 g_assert_not_reached ();
404 else if (g_strcmp0 (name, "ProcessID") == 0)
408 g_assert (!(seen & SEEN_PID));
409 g_assert_cmpuint (dbus_message_iter_get_arg_type (&var_iter), ==,
411 dbus_message_iter_get_basic (&var_iter, &u32);
412 g_message ("%s of this process is %u", name, u32);
414 g_assert_cmpuint (u32, ==, getpid ());
415 #elif defined(G_OS_WIN32)
416 g_assert_cmpuint (u32, ==, GetCurrentProcessId ());
418 g_assert_not_reached ();
423 dbus_message_iter_next (&arr_iter);
427 g_assert (seen & SEEN_UNIX_USER);
428 g_assert (seen & SEEN_PID);
432 /* FIXME: when implemented:
433 g_assert (seen & SEEN_WINDOWS_SID);
439 teardown (Fixture *f,
440 gconstpointer context G_GNUC_UNUSED)
442 dbus_error_free (&f->e);
443 g_clear_error (&f->ge);
445 if (f->left_conn != NULL)
447 dbus_connection_close (f->left_conn);
448 dbus_connection_unref (f->left_conn);
452 if (f->right_conn != NULL)
454 if (f->right_conn_echo)
456 dbus_connection_remove_filter (f->right_conn, echo_filter, NULL);
457 f->right_conn_echo = FALSE;
460 dbus_connection_close (f->right_conn);
461 dbus_connection_unref (f->right_conn);
462 f->right_conn = NULL;
465 if (f->daemon_pid != 0)
468 TerminateProcess (f->daemon_pid, 1);
470 kill (f->daemon_pid, SIGTERM);
473 g_spawn_close_pid (f->daemon_pid);
478 static Config limited_config = {
479 "34393", 10000, "valid-config-files/incoming-limit.conf"
486 g_test_init (&argc, &argv, NULL);
487 g_test_bug_base ("https://bugs.freedesktop.org/show_bug.cgi?id=");
489 g_test_add ("/echo/session", Fixture, NULL, setup, test_echo, teardown);
490 g_test_add ("/echo/limited", Fixture, &limited_config,
491 setup, test_echo, teardown);
492 g_test_add ("/creds", Fixture, NULL, setup, test_creds, teardown);
494 return g_test_run ();