// SPDX-License-Identifier: GPL-2.0-only
/*
 * KUnit tests for channel helper functions
 *
 * Copyright (C) 2023-2024, 2026 Intel Corporation
 */
#include <net/cfg80211.h>
#include <kunit/test.h>

MODULE_IMPORT_NS("EXPORTED_FOR_KUNIT_TESTING");

static struct ieee80211_channel chan_2ghz_1 = {
	.band = NL80211_BAND_2GHZ,
	.center_freq = 2412,
};

static struct ieee80211_channel chan_6ghz_1 = {
	.band = NL80211_BAND_6GHZ,
	.center_freq = 5955,
};

static struct ieee80211_channel chan_6ghz_5 = {
	.band = NL80211_BAND_6GHZ,
	.center_freq = 5975,
};

static struct ieee80211_channel chan_6ghz_105 = {
	.band = NL80211_BAND_6GHZ,
	.center_freq = 6475,
};

static const struct chandef_compat_case {
	const char *desc;
	/* leave c1 empty for tests for identical */
	struct cfg80211_chan_def c1, c2;
	/* we test both ways around, so c2 should always be the compat one */
	bool compat;
} chandef_compat_cases[] = {
	{
		.desc = "identical non-HT",
		.c2 = {
			.width = NL80211_CHAN_WIDTH_20_NOHT,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955,
		},
		.compat = true,
	},
	{
		.desc = "identical 20 MHz",
		.c2 = {
			.width = NL80211_CHAN_WIDTH_20,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955,
		},
		.compat = true,
	},
	{
		.desc = "identical 40 MHz",
		.c2 = {
			.width = NL80211_CHAN_WIDTH_40,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955 + 10,
		},
		.compat = true,
	},
	{
		.desc = "identical 80 MHz",
		.c2 = {
			.width = NL80211_CHAN_WIDTH_80,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955 + 10 + 20,
		},
		.compat = true,
	},
	{
		.desc = "identical 160 MHz",
		.c2 = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955 + 10 + 20 + 40,
		},
		.compat = true,
	},
	{
		.desc = "identical 320 MHz",
		.c2 = {
			.width = NL80211_CHAN_WIDTH_320,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955 + 10 + 20 + 40 + 80,
		},
		.compat = true,
	},
	{
		.desc = "20 MHz in 320 MHz\n",
		.c1 = {
			.width = NL80211_CHAN_WIDTH_20,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955,
		},
		.c2 = {
			.width = NL80211_CHAN_WIDTH_320,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955 + 10 + 20 + 40 + 80,
		},
		.compat = true,
	},
	{
		.desc = "different 20 MHz",
		.c1 = {
			.width = NL80211_CHAN_WIDTH_20,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955,
		},
		.c2 = {
			.width = NL80211_CHAN_WIDTH_20,
			.chan = &chan_6ghz_5,
			.center_freq1 = 5975,
		},
	},
	{
		.desc = "different primary 320 MHz",
		.c1 = {
			.width = NL80211_CHAN_WIDTH_320,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 110,
		},
		.c2 = {
			.width = NL80211_CHAN_WIDTH_320,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 - 50,
		},
	},
	{
		/* similar to previous test but one has lower BW */
		.desc = "matching primary 160 MHz",
		.c1 = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
		},
		.c2 = {
			.width = NL80211_CHAN_WIDTH_320,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 - 50,
		},
		.compat = true,
	},
	{
		.desc = "matching primary 160 MHz & punctured secondary 160 Mhz",
		.c1 = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
		},
		.c2 = {
			.width = NL80211_CHAN_WIDTH_320,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 - 50,
			.punctured = 0xf,
		},
		.compat = true,
	},
	{
		.desc = "matching primary 160 MHz & punctured matching",
		.c1 = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
			.punctured = 0xc0,
		},
		.c2 = {
			.width = NL80211_CHAN_WIDTH_320,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 - 50,
			.punctured = 0xc000,
		},
		.compat = true,
	},
	{
		.desc = "matching primary 160 MHz & punctured not matching",
		.c1 = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
			.punctured = 0x80,
		},
		.c2 = {
			.width = NL80211_CHAN_WIDTH_320,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 - 50,
			.punctured = 0xc000,
		},
	},
};

KUNIT_ARRAY_PARAM_DESC(chandef_compat, chandef_compat_cases, desc)

static void test_chandef_compat(struct kunit *test)
{
	const struct chandef_compat_case *params = test->param_value;
	const struct cfg80211_chan_def *ret, *expect;
	struct cfg80211_chan_def c1 = params->c1;

	/* tests with identical ones */
	if (!params->c1.chan)
		c1 = params->c2;

	KUNIT_EXPECT_EQ(test, cfg80211_chandef_valid(&c1), true);
	KUNIT_EXPECT_EQ(test, cfg80211_chandef_valid(&params->c2), true);

	expect = params->compat ? &params->c2 : NULL;

	ret = cfg80211_chandef_compatible(&c1, &params->c2);
	KUNIT_EXPECT_PTR_EQ(test, ret, expect);

	if (!params->c1.chan)
		expect = &c1;

	ret = cfg80211_chandef_compatible(&params->c2, &c1);
	KUNIT_EXPECT_PTR_EQ(test, ret, expect);
}

static const struct chandef_dbe_case {
	const char *desc;
	struct cfg80211_chan_def c;
	u8 dbe[3];
	bool fails;
	u16 cf1;
} chandef_dbe_cases[] = {
	{
		.desc = "non-HT failure",
		.c = {
			.width = NL80211_CHAN_WIDTH_20_NOHT,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_40,
		.fails = true,
	},
	{
		.desc = "2.4 GHz fails",
		.c = {
			.width = NL80211_CHAN_WIDTH_20,
			.chan = &chan_2ghz_1,
			.center_freq1 = 2412,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_40,
		.fails = true,
	},
	{
		.desc = "DBE narrower",
		.c = {
			.width = NL80211_CHAN_WIDTH_320,
			.chan = &chan_6ghz_1,
			.center_freq1 = 5955 + 10 + 20 + 40 + 80,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_160,
		.fails = true,
	},
	{
		.desc = "DBE to 320-1",
		.c = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_320_1,
		.cf1 = 6425,
	},
	{
		.desc = "DBE to 320-2",
		.c = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_320_2,
		.cf1 = 6585,
	},
	{
		.desc = "bad disabled subchannel bitmap - not enough in BSS (1)",
		.c = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
			.punctured = 0x0001,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_320_1 |
			  IEEE80211_UHR_DBE_OPER_DIS_SUBCHANNEL_BITMAP_PRES,
		/* DBE disabled subchannel bitmap == 0 */
		.fails = true,
	},
	{
		.desc = "bad disabled subchannel bitmap - too much in BSS (1)",
		.c = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
			.punctured = 0x0001,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_320_1 |
			  IEEE80211_UHR_DBE_OPER_DIS_SUBCHANNEL_BITMAP_PRES,
		/* DBE disabled subchannel bitmap == 0x0300 */
		.dbe[2] = 0x03,
		.fails = true,
	},
	{
		.desc = "bad disabled subchannel bitmap - not enough in BSS (2)",
		.c = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
			.punctured = 0x0001,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_320_2 |
			  IEEE80211_UHR_DBE_OPER_DIS_SUBCHANNEL_BITMAP_PRES,
		/* DBE disabled subchannel bitmap == 0 */
		.fails = true,
	},
	{
		.desc = "bad disabled subchannel bitmap - too much in BSS (2)",
		.c = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
			.punctured = 0x0001,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_320_2 |
			  IEEE80211_UHR_DBE_OPER_DIS_SUBCHANNEL_BITMAP_PRES,
		/* DBE disabled subchannel bitmap == 0x03 */
		.dbe[1] = 0x03,
		.fails = true,
	},
	{
		.desc = "bad disabled subchannel bitmap - bad bitmap",
		.c = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
			.punctured = 0x0001,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_320_1 |
			  IEEE80211_UHR_DBE_OPER_DIS_SUBCHANNEL_BITMAP_PRES,
		/* DBE disabled subchannel bitmap == 0x1100 */
		.dbe[2] = 0x11,
		.fails = true,
	},
	{
		.desc = "good disabled subchannel bitmap (1)",
		.c = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
			.punctured = 0x0003,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_320_1 |
			  IEEE80211_UHR_DBE_OPER_DIS_SUBCHANNEL_BITMAP_PRES,
		/* DBE disabled subchannel bitmap == 0x0300 */
		.dbe[2] = 0x03,
		.cf1 = 6425,
	},
	{
		.desc = "good disabled subchannel bitmap (2)",
		.c = {
			.width = NL80211_CHAN_WIDTH_160,
			.chan = &chan_6ghz_105,
			.center_freq1 = 6475 + 30,
			.punctured = 0x0003,
		},
		.dbe[0] = IEEE80211_UHR_DBE_OPER_BW_320_2 |
			  IEEE80211_UHR_DBE_OPER_DIS_SUBCHANNEL_BITMAP_PRES,
		/* DBE disabled subchannel bitmap == 0x0003 */
		.dbe[1] = 0x03,
		.cf1 = 6585,
	},
};

KUNIT_ARRAY_PARAM_DESC(chandef_dbe, chandef_dbe_cases, desc)

static void test_chandef_dbe(struct kunit *test)
{
	const struct chandef_dbe_case *params = test->param_value;
	struct cfg80211_chan_def c = params->c;
	int ret;

	KUNIT_EXPECT_EQ(test, cfg80211_chandef_valid(&params->c), true);

	ret = cfg80211_chandef_add_dbe(&c, (void *)params->dbe);
	KUNIT_EXPECT_EQ(test, ret != 0, params->fails);

	if (params->fails)
		return;

	KUNIT_EXPECT_EQ(test, c.center_freq1, params->cf1);
}

static struct kunit_case chandef_test_cases[] = {
	KUNIT_CASE_PARAM(test_chandef_compat, chandef_compat_gen_params),
	KUNIT_CASE_PARAM(test_chandef_dbe, chandef_dbe_gen_params),
	{}
};

static struct kunit_suite chandef = {
	.name = "cfg80211-chandef",
	.test_cases = chandef_test_cases,
};

kunit_test_suite(chandef);