/* SPDX-License-Identifier: GPL-2.0-or-later */

/*

 * Copyright (C) 2016 Red Hat, Inc.

 */

#include "src/core/systemd/nm-default-systemd.h"

#include "systemd/nm-sd.h"

#include "libnm-systemd-shared/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();

}