dbus/test/internals/userdb.c
2023-08-18 18:51:08 +01:00

143 lines
3.2 KiB
C

/*
* Copyright © 2023 Collabora Ltd.
* SPDX-License-Identifier: MIT
*/
#include <config.h>
#include <glib.h>
#include <dbus/dbus.h>
#include "dbus/dbus-sysdeps.h"
#include "test-utils-glib.h"
#ifdef DBUS_UNIX
#include <errno.h>
#include <pwd.h>
#include <sys/types.h>
#include <unistd.h>
#include "dbus/dbus-sysdeps-unix.h"
#include "dbus/dbus-userdb.h"
#endif
typedef struct
{
int dummy;
} Fixture;
static void
setup (Fixture *f G_GNUC_UNUSED,
gconstpointer context G_GNUC_UNUSED)
{
}
static void
test_groups_from_uid (Fixture *f,
gconstpointer context G_GNUC_UNUSED)
{
DBusError error = DBUS_ERROR_INIT;
dbus_gid_t *gids = NULL;
int n_gids = -1;
dbus_bool_t ret;
#ifdef DBUS_UNIX
int i;
#endif
/* We assume that uid 0 (root) is available on all Unix systems,
* so this should succeed */
ret = _dbus_unix_groups_from_uid (0, &gids, &n_gids, &error);
#ifdef DBUS_UNIX
test_assert_no_error (&error);
g_assert_true (ret);
g_assert_cmpint (n_gids, >=, 0);
g_test_message ("Groups of uid 0:");
for (i = 0; i < n_gids; i++)
{
g_test_message ("[%d]: %ld", i, (long) gids[i]);
g_assert_cmpint (gids[i], >=, 0);
}
#else
g_assert_cmpstr (error.name, ==, DBUS_ERROR_NOT_SUPPORTED);
g_assert_false (ret);
g_test_message ("Getting Unix groups on Windows failed as expected: %s: %s",
error.name, error.message);
g_assert_null (gids);
g_assert_cmpint (n_gids, <=, 0);
#endif
dbus_free (gids);
dbus_error_free (&error);
#ifdef DBUS_UNIX
/* Assume that the current uid is something sensible */
ret = _dbus_unix_groups_from_uid (geteuid (), &gids, &n_gids, &error);
test_assert_no_error (&error);
g_assert_true (ret);
g_assert_cmpint (n_gids, >=, 0);
g_test_message ("Groups of uid %ld:", (long) geteuid ());
for (i = 0; i < n_gids; i++)
{
g_test_message ("[%d]: %ld", i, (long) gids[i]);
g_assert_cmpint (gids[i], >=, 0);
}
g_test_message ("Total: %i groups", n_gids);
dbus_free (gids);
dbus_error_free (&error);
errno = 0;
/* arbitrarily chosen, probably isn't a valid uid */
if (getpwuid (31337) == NULL)
{
g_test_message ("uid 31337 doesn't exist: %s",
errno == 0 ? "(no errno)" : g_strerror (errno));
ret = _dbus_unix_groups_from_uid (31337, &gids, &n_gids, &error);
g_assert_nonnull (error.name);
g_assert_nonnull (error.message);
g_assert_false (ret);
g_test_message ("Getting groups from non-uid failed as expected: %s: %s",
error.name, error.message);
/* The Unix implementation always clears gids/n_gids,
* even on failure, and even if they were uninitialized */
g_assert_null (gids);
g_assert_cmpint (n_gids, ==, 0);
dbus_free (gids);
dbus_error_free (&error);
}
else
{
g_test_skip ("against our expectations, uid 31337 exists on this system");
}
#endif
}
static void
teardown (Fixture *f G_GNUC_UNUSED,
gconstpointer context G_GNUC_UNUSED)
{
}
int
main (int argc,
char **argv)
{
int ret;
test_init (&argc, &argv);
g_test_add ("/userdb/groups_from_uid",
Fixture, NULL, setup, test_groups_from_uid, teardown);
ret = g_test_run ();
dbus_shutdown ();
return ret;
}