/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
* 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();
}