tpe.c

  1// SPDX-License-Identifier: GPL-2.0-only
  2/*
  3 * KUnit tests for TPE element handling
  4 *
  5 * Copyright (C) 2024 Intel Corporation
  6 */
  7#include <kunit/test.h>
  8#include "../ieee80211_i.h"
  9
 10MODULE_IMPORT_NS("EXPORTED_FOR_KUNIT_TESTING");
 11
 12static struct ieee80211_channel chan6g_1 = {
 13	.band = NL80211_BAND_6GHZ,
 14	.center_freq = 5955,
 15};
 16
 17static struct ieee80211_channel chan6g_33 = {
 18	.band = NL80211_BAND_6GHZ,
 19	.center_freq = 6115,
 20};
 21
 22static struct ieee80211_channel chan6g_61 = {
 23	.band = NL80211_BAND_6GHZ,
 24	.center_freq = 6255,
 25};
 26
 27static const struct subchan_test_case {
 28	const char *desc;
 29	struct cfg80211_chan_def c;
 30	u8 n;
 31	int expect;
 32} subchan_offset_cases[] = {
 33	{
 34		.desc = "identical 20 MHz",
 35		.c.width = NL80211_CHAN_WIDTH_20,
 36		.c.chan = &chan6g_1,
 37		.c.center_freq1 = 5955,
 38		.n = 1,
 39		.expect = 0,
 40	},
 41	{
 42		.desc = "identical 40 MHz",
 43		.c.width = NL80211_CHAN_WIDTH_40,
 44		.c.chan = &chan6g_1,
 45		.c.center_freq1 = 5965,
 46		.n = 2,
 47		.expect = 0,
 48	},
 49	{
 50		.desc = "identical 80+80 MHz",
 51		/* not really is valid? doesn't matter for the test */
 52		.c.width = NL80211_CHAN_WIDTH_80P80,
 53		.c.chan = &chan6g_1,
 54		.c.center_freq1 = 5985,
 55		.c.center_freq2 = 6225,
 56		.n = 16,
 57		.expect = 0,
 58	},
 59	{
 60		.desc = "identical 320 MHz",
 61		.c.width = NL80211_CHAN_WIDTH_320,
 62		.c.chan = &chan6g_1,
 63		.c.center_freq1 = 6105,
 64		.n = 16,
 65		.expect = 0,
 66	},
 67	{
 68		.desc = "lower 160 MHz of 320 MHz",
 69		.c.width = NL80211_CHAN_WIDTH_320,
 70		.c.chan = &chan6g_1,
 71		.c.center_freq1 = 6105,
 72		.n = 8,
 73		.expect = 0,
 74	},
 75	{
 76		.desc = "upper 160 MHz of 320 MHz",
 77		.c.width = NL80211_CHAN_WIDTH_320,
 78		.c.chan = &chan6g_61,
 79		.c.center_freq1 = 6105,
 80		.n = 8,
 81		.expect = 8,
 82	},
 83	{
 84		.desc = "upper 160 MHz of 320 MHz, go to 40",
 85		.c.width = NL80211_CHAN_WIDTH_320,
 86		.c.chan = &chan6g_61,
 87		.c.center_freq1 = 6105,
 88		.n = 2,
 89		.expect = 8 + 4 + 2,
 90	},
 91	{
 92		.desc = "secondary 80 above primary in 80+80 MHz",
 93		/* not really is valid? doesn't matter for the test */
 94		.c.width = NL80211_CHAN_WIDTH_80P80,
 95		.c.chan = &chan6g_1,
 96		.c.center_freq1 = 5985,
 97		.c.center_freq2 = 6225,
 98		.n = 4,
 99		.expect = 0,
100	},
101	{
102		.desc = "secondary 80 below primary in 80+80 MHz",
103		/* not really is valid? doesn't matter for the test */
104		.c.width = NL80211_CHAN_WIDTH_80P80,
105		.c.chan = &chan6g_61,
106		.c.center_freq1 = 6225,
107		.c.center_freq2 = 5985,
108		.n = 4,
109		.expect = 4,
110	},
111	{
112		.desc = "secondary 80 below primary in 80+80 MHz, go to 20",
113		/* not really is valid? doesn't matter for the test */
114		.c.width = NL80211_CHAN_WIDTH_80P80,
115		.c.chan = &chan6g_61,
116		.c.center_freq1 = 6225,
117		.c.center_freq2 = 5985,
118		.n = 1,
119		.expect = 7,
120	},
121};
122
123KUNIT_ARRAY_PARAM_DESC(subchan_offset, subchan_offset_cases, desc);
124
125static void subchan_offset(struct kunit *test)
126{
127	const struct subchan_test_case *params = test->param_value;
128	int offset;
129
130	KUNIT_ASSERT_EQ(test, cfg80211_chandef_valid(&params->c), true);
131
132	offset = ieee80211_calc_chandef_subchan_offset(&params->c, params->n);
133
134	KUNIT_EXPECT_EQ(test, params->expect, offset);
135}
136
137static const struct psd_reorder_test_case {
138	const char *desc;
139	struct cfg80211_chan_def ap, used;
140	struct ieee80211_parsed_tpe_psd psd, out;
141} psd_reorder_cases[] = {
142	{
143		.desc = "no changes, 320 MHz",
144
145		.ap.width = NL80211_CHAN_WIDTH_320,
146		.ap.chan = &chan6g_1,
147		.ap.center_freq1 = 6105,
148
149		.used.width = NL80211_CHAN_WIDTH_320,
150		.used.chan = &chan6g_1,
151		.used.center_freq1 = 6105,
152
153		.psd.valid = true,
154		.psd.count = 16,
155		.psd.n = 8,
156		.psd.power = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 },
157
158		.out.valid = true,
159		.out.count = 16,
160		.out.n = 8,
161		.out.power = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 },
162	},
163	{
164		.desc = "no changes, 320 MHz, 160 MHz used, n=0",
165
166		.ap.width = NL80211_CHAN_WIDTH_320,
167		.ap.chan = &chan6g_1,
168		.ap.center_freq1 = 6105,
169
170		.used.width = NL80211_CHAN_WIDTH_160,
171		.used.chan = &chan6g_1,
172		.used.center_freq1 = 6025,
173
174		.psd.valid = true,
175		.psd.count = 16,
176		.psd.n = 0,
177		.psd.power = { 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, },
178
179		.out.valid = true,
180		.out.count = 8,
181		.out.n = 0,
182		.out.power = { 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, },
183	},
184	{
185		.desc = "320 MHz, HE is 80, used 160, all lower",
186
187		.ap.width = NL80211_CHAN_WIDTH_320,
188		.ap.chan = &chan6g_1,
189		.ap.center_freq1 = 6105,
190
191		.used.width = NL80211_CHAN_WIDTH_160,
192		.used.chan = &chan6g_1,
193		.used.center_freq1 = 6025,
194
195		.psd.valid = true,
196		.psd.count = 16,
197		.psd.n = 4,
198		.psd.power = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 },
199
200		.out.valid = true,
201		.out.count = 8,
202		.out.n = 4,
203		.out.power = { 0, 1, 2, 3, 4, 5, 6, 7, 127, 127, 127, 127, 127, 127, 127, 127},
204	},
205	{
206		.desc = "320 MHz, HE is 80, used 160, all upper",
207		/*
208		 * EHT: | | | | | | | | | | | | | | | | |
209		 * HE:                          | | | | |
210		 * used:                | | | | | | | | |
211		 */
212
213		.ap.width = NL80211_CHAN_WIDTH_320,
214		.ap.chan = &chan6g_61,
215		.ap.center_freq1 = 6105,
216
217		.used.width = NL80211_CHAN_WIDTH_160,
218		.used.chan = &chan6g_61,
219		.used.center_freq1 = 6185,
220
221		.psd.valid = true,
222		.psd.count = 16,
223		.psd.n = 4,
224		.psd.power = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 },
225
226		.out.valid = true,
227		.out.count = 8,
228		.out.n = 4,
229		.out.power = { 12, 13, 14, 15, 0, 1, 2, 3, 127, 127, 127, 127, 127, 127, 127, 127},
230	},
231	{
232		.desc = "320 MHz, HE is 80, used 160, split",
233		/*
234		 * EHT: | | | | | | | | | | | | | | | | |
235		 * HE:                  | | | | |
236		 * used:                | | | | | | | | |
237		 */
238
239		.ap.width = NL80211_CHAN_WIDTH_320,
240		.ap.chan = &chan6g_33,
241		.ap.center_freq1 = 6105,
242
243		.used.width = NL80211_CHAN_WIDTH_160,
244		.used.chan = &chan6g_33,
245		.used.center_freq1 = 6185,
246
247		.psd.valid = true,
248		.psd.count = 16,
249		.psd.n = 4,
250		.psd.power = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 },
251
252		.out.valid = true,
253		.out.count = 8,
254		.out.n = 4,
255		.out.power = { 0, 1, 2, 3, 12, 13, 14, 15, 127, 127, 127, 127, 127, 127, 127, 127},
256	},
257};
258
259KUNIT_ARRAY_PARAM_DESC(psd_reorder, psd_reorder_cases, desc);
260
261static void psd_reorder(struct kunit *test)
262{
263	const struct psd_reorder_test_case *params = test->param_value;
264	struct ieee80211_parsed_tpe_psd tmp = params->psd;
265
266	KUNIT_ASSERT_EQ(test, cfg80211_chandef_valid(&params->ap), true);
267	KUNIT_ASSERT_EQ(test, cfg80211_chandef_valid(&params->used), true);
268
269	ieee80211_rearrange_tpe_psd(&tmp, &params->ap, &params->used);
270	KUNIT_EXPECT_MEMEQ(test, &tmp, &params->out, sizeof(tmp));
271}
272
273static struct kunit_case tpe_test_cases[] = {
274	KUNIT_CASE_PARAM(subchan_offset, subchan_offset_gen_params),
275	KUNIT_CASE_PARAM(psd_reorder, psd_reorder_gen_params),
276	{}
277};
278
279static struct kunit_suite tpe = {
280	.name = "mac80211-tpe",
281	.test_cases = tpe_test_cases,
282};
283
284kunit_test_suite(tpe);