// SPDX-License-Identifier: GPL-2.0+
/*
* Copyright (C) 2016 Red Hat, Inc.
*/
#include "nm-default.h"
#include "systemd/nm-sd.h"
#include "systemd/nm-sd-utils-shared.h"
#include "nm-test-utils-core.h"
/*****************************************************************************
* Stub implementations of libNetworkManagerBase symbols
*****************************************************************************/
gboolean
nm_utils_get_testing_initialized (void)
{
return TRUE;
}
void
_nm_utils_set_testing (NMUtilsTestFlags flags)
{
g_assert_not_reached ();
}
gint32
nm_utils_get_monotonic_timestamp_sec (void)
{
return 1;
}
NMLogDomain _nm_logging_enabled_state[_LOGL_N_REAL];
gboolean
_nm_log_enabled_impl (gboolean mt_require_locking,
NMLogLevel level,
NMLogDomain domain)
{
return FALSE;
}
void
_nm_log_impl (const char *file,
guint line,
const char *func,
gboolean mt_require_locking,
NMLogLevel level,
NMLogDomain domain,
int error,
const char *ifname,
const char *con_uuid,
const char *fmt,
...)
{
}
gboolean
nm_logging_setup (const char *level,
const char *domains,
char **bad_domains,
GError **error)
{
return TRUE;
}
const char *
nm_strerror_native (int errsv)
{
return g_strerror (errsv);
}
/*****************************************************************************/
static void
test_dhcp_create (void)
{
sd_dhcp_client *client4 = NULL;
int r;
r = sd_dhcp_client_new (&client4, FALSE);
g_assert (r == 0);
g_assert (client4);
if (/* never true */ client4 == (gpointer) &r) {
/* we don't want to call this, but ensure that the linker
* includes all these symbols. */
sd_dhcp_client_start (client4);
}
sd_dhcp_client_unref (client4);
}
/*****************************************************************************/
static void
test_lldp_create (void)
{
sd_lldp *lldp = NULL;
int r;
r = sd_lldp_new (&lldp);
g_assert (r == 0);
g_assert (lldp);
sd_lldp_unref (lldp);
}
/*****************************************************************************/
typedef struct {
GMainLoop *mainloop;
sd_event_source *event_source;
} TestSdEventData;
static int
_test_sd_event_timeout_cb (sd_event_source *s, uint64_t usec, void *userdata)
{
TestSdEventData *user_data = userdata;
g_assert (user_data);
g_assert (user_data->mainloop);
g_assert (user_data->event_source);
user_data->event_source = sd_event_source_unref (user_data->event_source);
g_main_loop_quit (user_data->mainloop);
return 0;
}
static void
test_sd_event (void)
{
int repeat;
for (repeat = 0; repeat < 2; repeat++) {
guint sd_id = 0;
int r;
int i, n;
sd_event *other_events[3] = { NULL }, *event = NULL;
TestSdEventData user_data = { 0 };
g_assert_cmpint (sd_event_default (NULL), ==, 0);
for (i = 0, n = (nmtst_get_rand_uint32 () % (G_N_ELEMENTS (other_events) + 1)); i < n; i++) {
r = sd_event_default (&other_events[i]);
g_assert (r >= 0 && other_events[i]);
}
sd_id = nm_sd_event_attach_default ();
r = sd_event_default (&event);
g_assert (r >= 0 && event);
r = sd_event_add_time (event, &user_data.event_source, CLOCK_MONOTONIC, 1, 0, _test_sd_event_timeout_cb, &user_data);
g_assert (r >= 0 && user_data.event_source);
user_data.mainloop = g_main_loop_new (NULL, FALSE);
g_main_loop_run (user_data.mainloop);
g_main_loop_unref (user_data.mainloop);
g_assert (!user_data.event_source);
event = sd_event_unref (event);
for (i = 0, n = (nmtst_get_rand_uint32 () % (G_N_ELEMENTS (other_events) + 1)); i < n; i++)
other_events[i] = sd_event_unref (other_events[i]);
nm_clear_g_source (&sd_id);
for (i = 0, n = G_N_ELEMENTS (other_events); i < n; i++)
other_events[i] = sd_event_unref (other_events[i]);
g_assert_cmpint (sd_event_default (NULL), ==, 0);
}
}
/*****************************************************************************/
static void
test_path_equal (void)
{
#define _path_equal_check1(path, kill_dots, expected) \
G_STMT_START { \
const gboolean _kill_dots = (kill_dots); \
const char *_path0 = (path); \
const char *_expected = (expected); \
gs_free char *_path = g_strdup (_path0); \
const char *_path_result; \
\
if ( !_kill_dots \
&& !nm_sd_utils_path_equal (_path0, _expected)) \
g_error ("Paths \"%s\" and \"%s\" don't compare equal", _path0, _expected); \
\
_path_result = nm_sd_utils_path_simplify (_path, _kill_dots); \
g_assert (_path_result == _path); \
g_assert_cmpstr (_path, ==, _expected); \
} G_STMT_END
#define _path_equal_check(path, expected_no_kill_dots, expected_kill_dots) \
G_STMT_START { \
_path_equal_check1 (path, FALSE, expected_no_kill_dots); \
_path_equal_check1 (path, TRUE, expected_kill_dots ?: expected_no_kill_dots); \
} G_STMT_END
_path_equal_check ("", "", NULL);
_path_equal_check (".", ".", NULL);
_path_equal_check ("..", "..", NULL);
_path_equal_check ("/..", "/..", NULL);
_path_equal_check ("//..", "/..", NULL);
_path_equal_check ("/.", "/.", "/");
_path_equal_check ("./", ".", ".");
_path_equal_check ("./.", "./.", ".");
_path_equal_check (".///.", "./.", ".");
_path_equal_check (".///./", "./.", ".");
_path_equal_check (".////", ".", ".");
_path_equal_check ("//..//foo/", "/../foo", NULL);
_path_equal_check ("///foo//./bar/.", "/foo/./bar/.", "/foo/bar");
_path_equal_check (".//./foo//./bar/.", "././foo/./bar/.", "foo/bar");
}
/*****************************************************************************/
static void
_test_unbase64char (char ch, gboolean maybe_invalid)
{
int r;
r = nm_sd_utils_unbase64char (ch, FALSE);
if (ch == '=') {
g_assert (!maybe_invalid);
g_assert_cmpint (r, <, 0);
g_assert_cmpint (nm_sd_utils_unbase64char (ch, TRUE), ==, G_MAXINT);
} else {
g_assert_cmpint (r, ==, nm_sd_utils_unbase64char (ch, TRUE));
if (r >= 0)
g_assert_cmpint (r, <=, 255);
if (!maybe_invalid)
g_assert_cmpint (r, >=, 0);
}
}
static void
_test_unbase64mem_mem (const char *base64, const guint8 *expected_arr, gsize expected_len)
{
gs_free char *expected_base64 = NULL;
int r;
gs_free guint8 *exp2_arr = NULL;
gs_free guint8 *exp3_arr = NULL;
gsize exp2_len;
gsize exp3_len;
gsize i;
expected_base64 = g_base64_encode (expected_arr, expected_len);
for (i = 0; expected_base64[i]; i++)
_test_unbase64char (expected_base64[i], FALSE);
r = nm_sd_utils_unbase64mem (expected_base64, strlen (expected_base64), TRUE, &exp2_arr, &exp2_len);
g_assert_cmpint (r, ==, 0);
g_assert_cmpmem (expected_arr, expected_len, exp2_arr, exp2_len);
if (!nm_streq (base64, expected_base64)) {
r = nm_sd_utils_unbase64mem (base64, strlen (base64), TRUE, &exp3_arr, &exp3_len);
g_assert_cmpint (r, ==, 0);
g_assert_cmpmem (expected_arr, expected_len, exp3_arr, exp3_len);
}
}
#define _test_unbase64mem(base64, expected_str) _test_unbase64mem_mem (base64, (const guint8 *) ""expected_str"", NM_STRLEN (expected_str))
static void
_test_unbase64mem_inval (const char *base64)
{
gs_free guint8 *exp_arr = NULL;
gsize exp_len = 0;
int r;
r = nm_sd_utils_unbase64mem (base64, strlen (base64), TRUE, &exp_arr, &exp_len);
g_assert_cmpint (r, <, 0);
g_assert (!exp_arr);
g_assert (exp_len == 0);
}
static void
test_nm_sd_utils_unbase64mem (void)
{
gs_free char *rnd_base64 = NULL;
guint8 rnd_buf[30];
guint i, rnd_len;
_test_unbase64mem ("", "");
_test_unbase64mem (" ", "");
_test_unbase64mem (" Y Q == ", "a");
_test_unbase64mem (" Y WJjZGV mZ 2g = ", "abcdefgh");
_test_unbase64mem_inval (" Y %WJjZGV mZ 2g = ");
_test_unbase64mem_inval (" Y %WJjZGV mZ 2g = a");
_test_unbase64mem ("YQ==", "a");
_test_unbase64mem_inval ("YQ==a");
rnd_len = nmtst_get_rand_uint32 () % sizeof (rnd_buf);
for (i = 0; i < rnd_len; i++)
rnd_buf[i] = nmtst_get_rand_uint32 () % 256;
rnd_base64 = g_base64_encode (rnd_buf, rnd_len);
_test_unbase64mem_mem (rnd_base64, rnd_buf, rnd_len);
_test_unbase64char ('=', FALSE);
for (i = 0; i < 10; i++) {
char ch = nmtst_get_rand_uint32 () % 256;
if (ch != '=')
_test_unbase64char (ch, TRUE);
}
}
/*****************************************************************************/
NMTST_DEFINE ();
int
main (int argc, char **argv)
{
nmtst_init_assert_logging (&argc, &argv, "INFO", "ALL");
g_test_add_func ("/systemd/dhcp/create", test_dhcp_create);
g_test_add_func ("/systemd/lldp/create", test_lldp_create);
g_test_add_func ("/systemd/sd-event", test_sd_event);
g_test_add_func ("/systemd/test_path_equal", test_path_equal);
g_test_add_func ("/systemd/test_nm_sd_utils_unbase64mem", test_nm_sd_utils_unbase64mem);
return g_test_run ();
}