// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (C) 2015 Red Hat, Inc.
 */

#include "nm-default.h"

#include "n-acd/src/n-acd.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);
}