portal: Add some test coverage

This exercises Spawn() and reproduces #4286.

Signed-off-by: Simon McVittie <smcv@collabora.com>
This commit is contained in:
Simon McVittie
2021-05-21 15:56:31 +01:00
committed by Alexander Larsson
parent afd0cc4d18
commit 412c15772f
5 changed files with 743 additions and 2 deletions

View File

@@ -1,5 +1,6 @@
AM_TESTS_ENVIRONMENT = FLATPAK_TESTS_DEBUG=1 \
FLATPAK_CONFIG_DIR=/dev/null \
FLATPAK_PORTAL=$$(cd $(top_builddir) && pwd)/flatpak-portal \
FLATPAK_TRIGGERSDIR=$$(cd $(top_srcdir) && pwd)/triggers \
FLATPAK_VALIDATE_ICON=$$(cd $(top_builddir) && pwd)/flatpak-validate-icon \
FLATPAK_REVOKEFS_FUSE=$$(cd $(top_builddir) && pwd)/revokefs-fuse \
@@ -65,6 +66,7 @@ testcommon_CFLAGS = \
$(JSON_CFLAGS) \
$(APPSTREAM_GLIB_CFLAGS) \
-DFLATPAK_COMPILATION \
-DLIBEXECDIR=\"$(libexecdir)\" \
-I$(srcdir)/app \
-I$(builddir)/app \
$(NULL)
@@ -98,6 +100,14 @@ test_instance_SOURCES = \
tests/test-instance.c \
$(NULL)
test_portal_CFLAGS = $(testcommon_CFLAGS)
test_portal_LDADD = $(testcommon_LDADD) libtestlib.la
test_portal_SOURCES = \
portal/flatpak-portal-dbus.c \
portal/flatpak-portal-dbus.h \
tests/test-portal.c \
$(NULL)
tests_hold_lock_CFLAGS = $(AM_CFLAGS) $(BASE_CFLAGS)
tests_hold_lock_LDADD = $(AM_LDADD) $(BASE_LIBS) libglnx.la
tests_hold_lock_SOURCES = tests/hold-lock.c
@@ -108,6 +118,10 @@ tests_httpcache_CFLAGS = $(AM_CFLAGS) $(BASE_CFLAGS) $(OSTREE_CFLAGS) $(SOUP_CFL
tests_httpcache_LDADD = $(AM_LDADD) $(BASE_LIBS) $(OSTREE_LIBS) $(SOUP_LIBS) $(JSON_LIBS) $(APPSTREAM_GLIB_LIBS) \
libflatpak-common.la libflatpak-common-base.la libglnx.la
tests_mock_flatpak_CFLAGS = $(testcommon_CFLAGS)
tests_mock_flatpak_LDADD = $(testcommon_LDADD) libtestlib.la
tests_mock_flatpak_SOURCES = tests/mock-flatpak.c
tests_test_update_portal_CFLAGS = $(AM_CFLAGS) $(BASE_CFLAGS) \
-DFLATPAK_COMPILATION \
-DLOCALEDIR=\"$(localedir)\"
@@ -288,6 +302,7 @@ test_programs = \
test-context \
test-exports \
test-instance \
test-portal \
testcommon \
testlibrary \
$(NULL)
@@ -296,6 +311,7 @@ test_extra_programs = \
tests/hold-lock \
tests/httpcache \
tests/list-unused \
tests/mock-flatpak \
tests/test-authenticator \
tests/test-portal-impl \
tests/test-update-portal \

87
tests/mock-flatpak.c Normal file
View File

@@ -0,0 +1,87 @@
/*
* Copyright © 2018-2021 Collabora Ltd.
*
* This program 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.1 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, see <http://www.gnu.org/licenses/>.
*/
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include "libglnx.h"
#include <glib.h>
#include <gio/gio.h>
#include "flatpak-context-private.h"
int
main (int argc,
char **argv)
{
int i;
g_debug ("This is a mock implementation of `flatpak run` for the portal");
for (i = 0; i < argc; i++)
g_print ("argv[%d] = %s\n", i, argv[i]);
for (i = 0; i < argc; i++)
{
if (g_str_has_prefix (argv[i], "--env-fd="))
{
g_autoptr(FlatpakContext) context = flatpak_context_new ();
const char *value = argv[i] + strlen ("--env-fd=");
g_autoptr(GError) error = NULL;
guint64 fd;
gchar *endptr;
GHashTableIter iter;
gpointer k, v;
fd = g_ascii_strtoull (value, &endptr, 10);
if (endptr == NULL || *endptr != '\0' || fd > G_MAXINT)
g_error ("Not a valid file descriptor: %s", value);
flatpak_context_parse_env_fd (context, (int) fd, &error);
g_assert_no_error (error);
g_hash_table_iter_init (&iter, context->env_vars);
while (g_hash_table_iter_next (&iter, &k, &v))
g_print ("env[%s] = %s\n", (const char *) k, (const char *) v);
}
}
for (i = 0; i < 256; i++)
{
struct stat stat_buf;
if (fstat (i, &stat_buf) < 0)
{
int saved_errno = errno;
g_assert_cmpint (saved_errno, ==, EBADF);
}
else
{
g_print ("fd[%d] = (dev=%" G_GUINT64_FORMAT " ino=%" G_GUINT64_FORMAT ")\n",
i,
(guint64) stat_buf.st_dev,
(guint64) stat_buf.st_ino);
}
}
return 0;
}

483
tests/test-portal.c Normal file
View File

@@ -0,0 +1,483 @@
/*
* Copyright © 2018-2021 Collabora Ltd.
*
* This program 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.1 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, see <http://www.gnu.org/licenses/>.
*/
#include <errno.h>
#include <fcntl.h>
#include <string.h>
#include <sys/stat.h>
#include <unistd.h>
#include "libglnx.h"
#include <glib.h>
#include <gio/gio.h>
#include <gio/gunixfdlist.h>
#include "portal/flatpak-portal.h"
#include "portal/flatpak-portal-dbus.h"
#include "testlib.h"
typedef struct
{
TestsDBusDaemon dbus_daemon;
GSubprocess *portal;
gchar *portal_path;
gchar *mock_flatpak;
PortalFlatpak *proxy;
GDBusConnection *conn;
} Fixture;
typedef struct
{
int dummy;
} Config;
static void
setup (Fixture *f,
gconstpointer context G_GNUC_UNUSED)
{
g_autoptr(GError) error = NULL;
tests_dbus_daemon_setup (&f->dbus_daemon);
f->portal_path = g_strdup (g_getenv ("FLATPAK_PORTAL"));
if (f->portal_path == NULL)
f->portal_path = g_strdup (LIBEXECDIR "/flatpak-portal");
f->mock_flatpak = g_test_build_filename (G_TEST_BUILT, "mock-flatpak", NULL);
f->conn = g_dbus_connection_new_for_address_sync (f->dbus_daemon.dbus_address,
(G_DBUS_CONNECTION_FLAGS_AUTHENTICATION_CLIENT |
G_DBUS_CONNECTION_FLAGS_MESSAGE_BUS_CONNECTION),
NULL, NULL, &error);
g_assert_no_error (error);
g_assert_nonnull (f->conn);
}
/* Don't corrupt TAP output if portal outputs on stdout */
static void
launcher_stdout_to_our_stderr (GSubprocessLauncher *launcher)
{
glnx_autofd int stderr_copy = -1;
stderr_copy = dup (STDERR_FILENO);
g_assert_no_errno (stderr_copy);
g_assert_no_errno (fcntl (stderr_copy, F_SETFD, FD_CLOEXEC));
g_subprocess_launcher_take_stdout_fd (launcher, glnx_steal_fd (&stderr_copy));
}
static GSubprocessLauncher *
fixture_make_launcher (Fixture *f)
{
g_autoptr(GSubprocessLauncher) launcher = NULL;
launcher = g_subprocess_launcher_new (G_SUBPROCESS_FLAGS_NONE);
launcher_stdout_to_our_stderr (launcher);
g_subprocess_launcher_setenv (launcher,
"DBUS_SESSION_BUS_ADDRESS",
f->dbus_daemon.dbus_address,
TRUE);
g_subprocess_launcher_setenv (launcher,
"FLATPAK_PORTAL_MOCK_FLATPAK",
f->mock_flatpak,
TRUE);
return g_steal_pointer (&launcher);
}
static void
name_appeared_cb (GDBusConnection *conn,
const char *name,
const char *owner,
gpointer user_data)
{
gchar **name_owner_p = user_data;
g_clear_pointer (name_owner_p, g_free);
*name_owner_p = g_strdup (owner);
}
static void
name_vanished_cb (GDBusConnection *conn,
const char *name,
gpointer user_data)
{
gchar **name_owner_p = user_data;
g_clear_pointer (name_owner_p, g_free);
*name_owner_p = g_strdup ("");
}
static void
name_owner_watch_removed_cb (gpointer user_data)
{
gchar **name_owner_p = user_data;
g_clear_pointer (name_owner_p, g_free);
}
static void
fixture_wait_for_name_to_be_owned (Fixture *f,
const char *name)
{
g_autofree gchar *name_owner = NULL;
guint watch;
watch = g_bus_watch_name_on_connection (f->conn,
name,
G_BUS_NAME_WATCHER_FLAGS_NONE,
name_appeared_cb,
name_vanished_cb,
&name_owner,
name_owner_watch_removed_cb);
/* Wait for name to become owned */
while (name_owner == NULL || name_owner[0] == '\0')
g_main_context_iteration (NULL, TRUE);
g_bus_unwatch_name (watch);
/* Wait for watch to be cleaned up */
while (name_owner != NULL)
g_main_context_iteration (NULL, TRUE);
}
static void
fixture_start_portal (Fixture *f)
{
g_autoptr(GError) error = NULL;
g_autoptr(GSubprocessLauncher) launcher = NULL;
launcher = fixture_make_launcher (f);
f->portal = g_subprocess_launcher_spawn (launcher, &error,
f->portal_path,
NULL);
g_assert_no_error (error);
g_assert_nonnull (f->portal);
fixture_wait_for_name_to_be_owned (f, FLATPAK_PORTAL_BUS_NAME);
f->proxy = portal_flatpak_proxy_new_sync (f->conn,
G_DBUS_PROXY_FLAGS_DO_NOT_AUTO_START,
FLATPAK_PORTAL_BUS_NAME,
FLATPAK_PORTAL_PATH,
NULL,
&error);
g_assert_no_error (error);
g_assert_nonnull (f->proxy);
}
static void
test_help (Fixture *f,
gconstpointer context G_GNUC_UNUSED)
{
g_autoptr(GSubprocessLauncher) launcher = NULL;
g_autofree gchar *stdout_buf;
g_autofree gchar *stderr_buf;
g_autoptr(GError) error = NULL;
/* Don't use fixture_make_launcher() here because we want to
* capture stdout */
launcher = g_subprocess_launcher_new (G_SUBPROCESS_FLAGS_STDOUT_PIPE |
G_SUBPROCESS_FLAGS_STDERR_PIPE);
g_subprocess_launcher_setenv (launcher,
"DBUS_SESSION_BUS_ADDRESS",
f->dbus_daemon.dbus_address,
TRUE);
f->portal = g_subprocess_launcher_spawn (launcher, &error,
f->portal_path,
"--help",
NULL);
g_assert_no_error (error);
g_assert_nonnull (f->portal);
g_subprocess_communicate_utf8 (f->portal, NULL, NULL, &stdout_buf,
&stderr_buf, &error);
g_assert_nonnull (stdout_buf);
g_assert_nonnull (stderr_buf);
g_test_message ("flatpak-portal --help: %s", stdout_buf);
g_assert_true (strstr (stdout_buf, "--replace") != NULL);
g_subprocess_wait_check (f->portal, NULL, &error);
g_assert_no_error (error);
}
static void
count_successful_exit_cb (PortalFlatpak *proxy,
guint pid,
guint wait_status,
gpointer user_data)
{
gsize *times_exited_p = user_data;
g_debug ("Process %u exited with wait status %u", pid, wait_status);
g_assert_true (WIFEXITED (wait_status));
g_assert_cmpuint (WEXITSTATUS (wait_status), ==, 0);
(*times_exited_p) += 1;
}
static void
test_basic (Fixture *f,
gconstpointer context G_GNUC_UNUSED)
{
g_autoptr(GError) error = NULL;
g_autoptr(GUnixFDList) fds_out = NULL;
guint pid;
gboolean ok;
const char * const argv[] = { "hello", NULL };
gsize times_exited = 0;
gulong handler_id;
fixture_start_portal (f);
/* We can't easily tell whether EXPOSE_PIDS ought to be set or not */
g_assert_cmpuint ((portal_flatpak_get_supports (f->proxy) &
(~FLATPAK_SPAWN_SUPPORT_FLAGS_EXPOSE_PIDS)), ==, 0);
g_assert_cmpuint (portal_flatpak_get_version (f->proxy), ==, 6);
handler_id = g_signal_connect (f->proxy, "spawn-exited",
G_CALLBACK (count_successful_exit_cb),
&times_exited);
ok = portal_flatpak_call_spawn_sync (f->proxy,
"/", /* cwd */
argv, /* argv */
g_variant_new ("a{uh}", NULL),
g_variant_new ("a{ss}", NULL),
FLATPAK_SPAWN_FLAGS_NONE,
g_variant_new ("a{sv}", NULL),
NULL, /* fd list */
&pid,
&fds_out,
NULL,
&error);
g_assert_no_error (error);
g_assert_true (ok);
g_assert_cmpuint (pid, >, 1);
while (times_exited == 0)
g_main_context_iteration (NULL, TRUE);
g_signal_handler_disconnect (f->proxy, handler_id);
if (fds_out != NULL)
g_assert_cmpint (g_unix_fd_list_get_length (fds_out), ==, 0);
g_subprocess_send_signal (f->portal, SIGTERM);
g_subprocess_wait (f->portal, NULL, &error);
g_assert_no_error (error);
}
static void
test_fd_passing (Fixture *f,
gconstpointer context G_GNUC_UNUSED)
{
#define SOME_FDS 16
g_autoptr(GError) error = NULL;
char *tempfile_paths[SOME_FDS];
int tempfile_fds[SOME_FDS];
guint gap_size;
gsize times_exited = 0;
gulong handler_id;
gsize i;
fixture_start_portal (f);
handler_id = g_signal_connect (f->proxy, "spawn-exited",
G_CALLBACK (count_successful_exit_cb),
&times_exited);
for (i = 0; i < SOME_FDS; i++)
{
tempfile_paths[i] = g_strdup ("/tmp/flatpak-portal-test.XXXXXX");
tempfile_fds[i] = g_mkstemp (tempfile_paths[i]);
g_assert_no_errno (tempfile_fds[i]);
}
/* Using a non-contiguous block of fds can help to tickle bugs in the
* portal. */
for (gap_size = 0; gap_size < 128; gap_size += 16)
{
g_autoptr(GUnixFDList) fds_in = g_unix_fd_list_new ();
g_autoptr(GUnixFDList) fds_out = NULL;
g_auto(GVariantBuilder) fd_map_builder = {};
g_auto(GVariantBuilder) env_builder = {};
guint pid;
gboolean ok;
const char * const argv[] = { "hello", NULL };
g_autofree char *output = NULL;
g_variant_builder_init (&fd_map_builder, G_VARIANT_TYPE ("a{uh}"));
g_variant_builder_init (&env_builder, G_VARIANT_TYPE ("a{ss}"));
times_exited = 0;
g_variant_builder_add (&env_builder, "{ss}", "FOO", "bar");
for (i = 0; i < SOME_FDS; i++)
{
int handle = g_unix_fd_list_append (fds_in, tempfile_fds[i], &error);
guint32 desired_fd;
g_assert_no_error (error);
g_assert_cmpint (handle, >=, 0);
if (i <= STDERR_FILENO)
desired_fd = i;
else
desired_fd = i + gap_size;
g_variant_builder_add (&fd_map_builder, "{uh}",
desired_fd, (gint32) handle);
}
ok = portal_flatpak_call_spawn_sync (f->proxy,
"/", /* cwd */
argv, /* argv */
g_variant_builder_end (&fd_map_builder),
g_variant_builder_end (&env_builder),
FLATPAK_SPAWN_FLAGS_NONE,
g_variant_new ("a{sv}", NULL),
fds_in,
&pid,
&fds_out,
NULL,
&error);
g_assert_no_error (error);
g_assert_true (ok);
g_assert_cmpuint (pid, >, 1);
/* Wait for this one to exit */
while (times_exited == 0)
g_main_context_iteration (NULL, TRUE);
if (fds_out != NULL)
g_assert_cmpint (g_unix_fd_list_get_length (fds_out), ==, 0);
/* stdout from the portal should have ended up in temp file [1] */
g_assert_no_errno (lseek (tempfile_fds[1], 0, SEEK_SET));
output = glnx_fd_readall_utf8 (tempfile_fds[1], NULL, NULL, &error);
g_assert_no_error (error);
g_assert_nonnull (output);
g_assert_no_errno (lseek (tempfile_fds[1], 0, SEEK_SET));
g_assert_no_errno (ftruncate (tempfile_fds[1], 0));
g_test_message ("Output from mock Flatpak: %s", output);
if (strstr (output, "env[FOO] = bar") != NULL)
g_test_message ("Found env[FOO] = bar in output");
else
g_error ("env[FOO] = bar not found in \"%s\"", output);
for (i = 0; i < SOME_FDS; i++)
{
struct stat stat_buf;
g_autofree char *expected = NULL;
int desired_fd;
g_assert_no_errno (fstat (tempfile_fds[i], &stat_buf));
if (i <= STDERR_FILENO)
desired_fd = i;
else
desired_fd = i + gap_size;
expected = g_strdup_printf ("fd[%d] = (dev=%" G_GUINT64_FORMAT " ino=%" G_GUINT64_FORMAT ")",
desired_fd,
(guint64) stat_buf.st_dev,
(guint64) stat_buf.st_ino);
if (strstr (output, expected) != NULL)
g_test_message ("fd %d OK: \"%s\"", desired_fd, expected);
else
g_error ("\"%s\" not found in \"%s\"", expected, output);
}
}
g_signal_handler_disconnect (f->proxy, handler_id);
g_subprocess_send_signal (f->portal, SIGTERM);
g_subprocess_wait (f->portal, NULL, &error);
g_assert_no_error (error);
for (i = 0; i < SOME_FDS; i++)
{
g_assert_no_errno (unlink (tempfile_paths[i]));
glnx_close_fd (&tempfile_fds[i]);
g_free (tempfile_paths[i]);
}
}
static void
test_replace (Fixture *f,
gconstpointer context G_GNUC_UNUSED)
{
g_autoptr(GSubprocessLauncher) launcher = NULL;
g_autoptr(GError) error = NULL;
g_autoptr(GSubprocess) gets_replaced = NULL;
/* Not using fixture_start_portal() here because we want to --replace */
launcher = fixture_make_launcher (f);
gets_replaced = g_subprocess_launcher_spawn (launcher, &error,
f->portal_path,
"--replace",
NULL);
g_assert_no_error (error);
g_assert_nonnull (gets_replaced);
fixture_wait_for_name_to_be_owned (f, FLATPAK_PORTAL_BUS_NAME);
g_clear_object (&launcher);
launcher = fixture_make_launcher (f);
f->portal = g_subprocess_launcher_spawn (launcher, &error,
f->portal_path,
"--replace",
NULL);
g_assert_no_error (error);
g_assert_nonnull (f->portal);
/* f->portal replaces gets_replaced, which exits 0 */
g_subprocess_wait_check (gets_replaced, NULL, &error);
g_assert_no_error (error);
g_subprocess_send_signal (f->portal, SIGTERM);
g_subprocess_wait (f->portal, NULL, &error);
g_assert_no_error (error);
}
static void
teardown (Fixture *f,
gconstpointer context G_GNUC_UNUSED)
{
tests_dbus_daemon_teardown (&f->dbus_daemon);
g_clear_object (&f->portal);
g_free (f->portal_path);
}
int
main (int argc,
char **argv)
{
g_test_init (&argc, &argv, NULL);
g_test_add ("/help", Fixture, NULL, setup, test_help, teardown);
g_test_add ("/basic", Fixture, NULL, setup, test_basic, teardown);
g_test_add ("/fd-passing", Fixture, NULL, setup, test_fd_passing, teardown);
g_test_add ("/replace", Fixture, NULL, setup, test_replace, teardown);
return g_test_run ();
}

View File

@@ -1,5 +1,5 @@
/*
* Copyright © 2020-2021 Collabora Ltd.
* Copyright © 2018-2021 Collabora Ltd.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
@@ -95,3 +95,147 @@ isolated_test_dir_global_teardown (void)
g_free (isolated_test_dir);
isolated_test_dir = NULL;
}
static void
replace_tokens (const char *in_path,
const char *out_path)
{
g_autoptr(GError) error = NULL;
g_autoptr(GString) buffer = NULL;
g_autofree char *contents = NULL;
const char *iter;
g_file_get_contents (in_path, &contents, NULL, &error);
g_assert_no_error (error);
buffer = g_string_new ("");
iter = contents;
while (iter[0] != '\0')
{
const char *first_at = strchr (iter, '@');
const char *second_at;
if (first_at == NULL)
{
/* no more @token@s, append [iter..end] and stop */
g_string_append (buffer, iter);
break;
}
second_at = strchr (first_at + 1, '@');
if (second_at == NULL)
g_error ("Unterminated @token@ in %s: %s", in_path, first_at);
/* append the literal text [iter..first_at - 1], if non-empty */
if (first_at != iter)
g_string_append_len (buffer, iter, first_at - iter);
/* append the replacement for [first_at..second_at] if known */
if (g_str_has_prefix (first_at, "@testdir@"))
{
g_autofree char *testdir = g_test_build_filename (G_TEST_DIST, ".", NULL);
g_string_append (buffer, testdir);
}
else
{
g_error ("Unknown @token@ in %s: %.*s",
in_path, (int) (second_at - first_at) + 1, first_at);
}
/* continue to process [second_at + 1..end] */
iter = second_at + 1;
}
g_file_set_contents (out_path, buffer->str, -1, &error);
g_assert_no_error (error);
}
void
tests_dbus_daemon_setup (TestsDBusDaemon *self)
{
g_autoptr(GSubprocessLauncher) launcher = NULL;
g_autoptr(GError) error = NULL;
g_autofree char *config_arg = NULL;
g_autofree char *session_conf = NULL;
g_autofree char *session_conf_in = NULL;
GInputStream *address_pipe;
gchar address_buffer[4096] = { 0 };
char *newline;
g_return_if_fail (self != NULL);
g_return_if_fail (self->dbus_daemon == NULL);
g_return_if_fail (self->dbus_address == NULL);
g_return_if_fail (self->temp_dir == NULL);
self->temp_dir = g_dir_make_tmp ("flatpak-test.XXXXXX", &error);
g_assert_no_error (error);
g_assert_nonnull (self->temp_dir);
session_conf_in = g_test_build_filename (G_TEST_DIST, "session.conf.in", NULL);
session_conf = g_build_filename (self->temp_dir, "test-bus.conf", NULL);
replace_tokens (session_conf_in, session_conf);
config_arg = g_strdup_printf ("--config-file=%s", session_conf);
launcher = g_subprocess_launcher_new (G_SUBPROCESS_FLAGS_STDOUT_PIPE);
self->dbus_daemon = g_subprocess_launcher_spawn (launcher, &error,
"dbus-daemon",
config_arg,
"--print-address=1",
"--nofork",
NULL);
g_assert_no_error (error);
g_assert_nonnull (self->dbus_daemon);
address_pipe = g_subprocess_get_stdout_pipe (self->dbus_daemon);
g_assert_nonnull (address_pipe);
/* Crash if it takes too long to get the address */
alarm (30);
while (strchr (address_buffer, '\n') == NULL)
{
if (strlen (address_buffer) >= sizeof (address_buffer) - 1)
g_error ("Read %" G_GSIZE_FORMAT " bytes from dbus-daemon with "
"no newline",
sizeof (address_buffer) - 1);
g_input_stream_read (address_pipe,
address_buffer + strlen (address_buffer),
sizeof (address_buffer) - strlen (address_buffer),
NULL, &error);
g_assert_no_error (error);
}
/* Disable alarm */
alarm (0);
newline = strchr (address_buffer, '\n');
g_assert_nonnull (newline);
*newline = '\0';
self->dbus_address = g_strdup (address_buffer);
}
void
tests_dbus_daemon_teardown (TestsDBusDaemon *self)
{
g_autoptr(GError) error = NULL;
if (self->dbus_daemon != NULL)
{
g_subprocess_send_signal (self->dbus_daemon, SIGTERM);
g_subprocess_wait (self->dbus_daemon, NULL, &error);
g_assert_no_error (error);
}
if (self->temp_dir != NULL)
{
glnx_shutil_rm_rf_at (AT_FDCWD, self->temp_dir, NULL, &error);
g_assert_no_error (error);
}
g_clear_object (&self->dbus_daemon);
g_clear_pointer (&self->dbus_address, g_free);
g_clear_pointer (&self->temp_dir, g_free);
}

View File

@@ -1,5 +1,5 @@
/*
* Copyright © 2020-2021 Collabora Ltd.
* Copyright © 2018-2021 Collabora Ltd.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
@@ -19,6 +19,7 @@
#define TESTLIB_H
#include <glib.h>
#include <gio/gio.h>
#ifndef g_assert_no_errno
#define g_assert_no_errno(expr) \
@@ -31,4 +32,14 @@ extern char *isolated_test_dir;
void isolated_test_dir_global_setup (void);
void isolated_test_dir_global_teardown (void);
typedef struct
{
GSubprocess *dbus_daemon;
gchar *dbus_address;
gchar *temp_dir;
} TestsDBusDaemon;
void tests_dbus_daemon_setup (TestsDBusDaemon *self);
void tests_dbus_daemon_teardown (TestsDBusDaemon *self);
#endif