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

/*

 * Copyright (C) 2015 Red Hat, Inc.

 */

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

#include "n-acd/src/n-acd.h"

#include <linux/if_ether.h>

#include "devices/nm-acd-manager.h"

#include "platform/tests/test-common.h"

#define IFACE_VETH0 "nm-test-veth0"

#define IFACE_VETH1 "nm-test-veth1"

#define ADDR1 0x01010101

#define ADDR2 0x02020202

#define ADDR3 0x03030303

#define ADDR4 0x04040404

/*****************************************************************************/

static gboolean

_skip_acd_test_check(void)

{

    NAcd *       acd;

    NAcdConfig * config;

    const guint8 hwaddr[ETH_ALEN] = {0x01, 0x02, 0x03, 0x04, 0x05, 0x06};

    int          r;

    static int   skip = -1;

    if (skip == -1) {

        r = n_acd_config_new(&config);

        g_assert(r == 0);

        n_acd_config_set_ifindex(config, 1);

        n_acd_config_set_transport(config, N_ACD_TRANSPORT_ETHERNET);

        n_acd_config_set_mac(config, hwaddr, sizeof(hwaddr));

        r = n_acd_new(&acd, config);

        n_acd_config_free(config);

        if (r == 0)

            n_acd_unref(acd);

        skip = (r != 0);

    }

    return skip;

}

#define _skip_acd_test()                                               \

    ({                                                                 \

        gboolean _skip = _skip_acd_test_check();                       \

                                                                       \

        if (_skip)                                                     \

            g_test_skip("Cannot create NAcd. Running under valgind?"); \

        _skip;                                                         \

    })

/*****************************************************************************/

typedef struct {

    int           ifindex0;

    int           ifindex1;

    const guint8 *hwaddr0;

    const guint8 *hwaddr1;

    size_t        hwaddr0_len;

    size_t        hwaddr1_len;

} test_fixture;

static void

fixture_setup(test_fixture *fixture, gconstpointer user_data)

{

    /* create veth pair. */

    fixture->ifindex0 =

        nmtstp_link_veth_add(NM_PLATFORM_GET, -1, IFACE_VETH0, IFACE_VETH1)->ifindex;

    fixture->ifindex1 =

        nmtstp_link_get_typed(NM_PLATFORM_GET, -1, IFACE_VETH1, NM_LINK_TYPE_VETH)->ifindex;

    g_assert(nm_platform_link_set_up(NM_PLATFORM_GET, fixture->ifindex0, NULL));

    g_assert(nm_platform_link_set_up(NM_PLATFORM_GET, fixture->ifindex1, NULL));

    fixture->hwaddr0 =

        nm_platform_link_get_address(NM_PLATFORM_GET, fixture->ifindex0, &fixture->hwaddr0_len);

    fixture->hwaddr1 =

        nm_platform_link_get_address(NM_PLATFORM_GET, fixture->ifindex1, &fixture->hwaddr1_len);

}

typedef struct {

    in_addr_t addresses[8];

    in_addr_t peer_addresses[8];

    gboolean  expected_result[8];

} TestInfo;

static void

acd_manager_probe_terminated(NMAcdManager *acd_manager, gpointer user_data)

{

    g_main_loop_quit(user_data);

}

static void

test_acd_common(test_fixture *fixture, TestInfo *info)

{

    nm_auto_free_acdmgr NMAcdManager *manager = NULL;

    nm_auto_unref_gmainloop GMainLoop *loop   = NULL;

    int                                i;

    const guint                        WAIT_TIME_OPTIMISTIC = 50;

    guint                              wait_time;

    static const NMAcdCallbacks        callbacks = {

        .probe_terminated_callback = acd_manager_probe_terminated,

        .user_data_destroy         = (GDestroyNotify) g_main_loop_unref,

    };

    int r;

    if (_skip_acd_test())

        return;

    /* first, try with a short waittime. We hope that this is long enough

     * to successfully complete the test. Only if that's not the case, we

     * assume the computer is currently busy (high load) and we retry with

     * a longer timeout. */

    wait_time = WAIT_TIME_OPTIMISTIC;

again:

    nm_clear_pointer(&loop, g_main_loop_unref);

    loop = g_main_loop_new(NULL, FALSE);

    nm_clear_pointer(&manager, nm_acd_manager_free);

    manager = nm_acd_manager_new(fixture->ifindex0,

                                 fixture->hwaddr0,

                                 fixture->hwaddr0_len,

                                 &callbacks,

                                 g_main_loop_ref(loop));

    g_assert(manager != NULL);

    for (i = 0; info->addresses[i]; i++)

        g_assert(nm_acd_manager_add_address(manager, info->addresses[i]));

    for (i = 0; info->peer_addresses[i]; i++) {

        nmtstp_ip4_address_add(NULL,

                               FALSE,

                               fixture->ifindex1,

                               info->peer_addresses[i],

                               24,

                               0,

                               3600,

                               1800,

                               0,

                               NULL);

    }

    r = nm_acd_manager_start_probe(manager, wait_time);

    g_assert_cmpint(r, ==, 0);

    g_assert(nmtst_main_loop_run(loop, 2000));

    for (i = 0; info->addresses[i]; i++) {

        gboolean val;

        char     sbuf[NM_UTILS_INET_ADDRSTRLEN];

        val = nm_acd_manager_check_address(manager, info->addresses[i]);

        if (val == info->expected_result[i])

            continue;

        if (wait_time == WAIT_TIME_OPTIMISTIC) {

            /* probably we just had a glitch and the system took longer than

             * expected. Re-verify with a large timeout this time. */

            wait_time = 1000;

            goto again;

        }

        g_error("expected check for address #%d (%s) to %s, but it didn't",

                i,

                _nm_utils_inet4_ntop(info->addresses[i], sbuf),

                info->expected_result[i] ? "detect no duplicated" : "detect a duplicate");

    }

}

static void

test_acd_probe_1(test_fixture *fixture, gconstpointer user_data)

{

    TestInfo info = {.addresses       = {ADDR1, ADDR2, ADDR3},

                     .peer_addresses  = {ADDR4},

                     .expected_result = {TRUE, TRUE, TRUE}};

    test_acd_common(fixture, &info;;

}

static void

test_acd_probe_2(test_fixture *fixture, gconstpointer user_data)

{

    TestInfo info = {.addresses       = {ADDR1, ADDR2, ADDR3, ADDR4},

                     .peer_addresses  = {ADDR3, ADDR2},

                     .expected_result = {TRUE, FALSE, FALSE, TRUE}};

    test_acd_common(fixture, &info;;

}

static void

test_acd_announce(test_fixture *fixture, gconstpointer user_data)

{

    nm_auto_free_acdmgr NMAcdManager *manager = NULL;

    nm_auto_unref_gmainloop GMainLoop *loop   = NULL;

    int                                r;

    if (_skip_acd_test())

        return;

    manager =

        nm_acd_manager_new(fixture->ifindex0, fixture->hwaddr0, fixture->hwaddr0_len, NULL, NULL);

    g_assert(manager != NULL);

    g_assert(nm_acd_manager_add_address(manager, ADDR1));

    g_assert(nm_acd_manager_add_address(manager, ADDR2));

    loop = g_main_loop_new(NULL, FALSE);

    r    = nm_acd_manager_announce_addresses(manager);

    g_assert_cmpint(r, ==, 0);

    g_assert(!nmtst_main_loop_run(loop, 200));

}

static void

fixture_teardown(test_fixture *fixture, gconstpointer user_data)

{

    nm_platform_link_delete(NM_PLATFORM_GET, fixture->ifindex0);

    nm_platform_link_delete(NM_PLATFORM_GET, fixture->ifindex1);

}

NMTstpSetupFunc const _nmtstp_setup_platform_func = nm_linux_platform_setup;

void

_nmtstp_init_tests(int *argc, char ***argv)

{

    nmtst_init_with_logging(argc, argv, NULL, "ALL");

}

void

_nmtstp_setup_tests(void)

{

    g_test_add("/acd/probe/1",

               test_fixture,

               NULL,

               fixture_setup,

               test_acd_probe_1,

               fixture_teardown);

    g_test_add("/acd/probe/2",

               test_fixture,

               NULL,

               fixture_setup,

               test_acd_probe_2,

               fixture_teardown);

    g_test_add("/acd/announce",

               test_fixture,

               NULL,

               fixture_setup,

               test_acd_announce,

               fixture_teardown);

}