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

MODULE_IMPORT_NS("EXPORTED_FOR_KUNIT_TESTING");

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 struct kunit_case chandef_compat_test_cases[] = {
        KUNIT_CASE_PARAM(test_chandef_compat, chandef_compat_gen_params),
        {}
};

static struct kunit_suite chandef_compat = {
        .name = "cfg80211-chandef-compat",
        .test_cases = chandef_compat_test_cases,
};

kunit_test_suite(chandef_compat);