| // SPDX-License-Identifier: GPL-2.0-only | 
 | /* | 
 |  * KUnit tests for channel mode functions | 
 |  * | 
 |  * Copyright (C) 2024 Intel Corporation | 
 |  */ | 
 | #include <net/cfg80211.h> | 
 | #include <kunit/test.h> | 
 |  | 
 | #include "util.h" | 
 |  | 
 | MODULE_IMPORT_NS("EXPORTED_FOR_KUNIT_TESTING"); | 
 |  | 
 | static const struct determine_chan_mode_case { | 
 | 	const char *desc; | 
 | 	u8 extra_supp_rate; | 
 | 	enum ieee80211_conn_mode conn_mode; | 
 | 	enum ieee80211_conn_mode expected_mode; | 
 | 	bool strict; | 
 | 	u8 userspace_selector; | 
 | 	struct ieee80211_ht_cap ht_capa_mask; | 
 | 	struct ieee80211_vht_cap vht_capa; | 
 | 	struct ieee80211_vht_cap vht_capa_mask; | 
 | 	u8 vht_basic_mcs_1_4_set:1, | 
 | 	   vht_basic_mcs_5_8_set:1, | 
 | 	   he_basic_mcs_1_4_set:1, | 
 | 	   he_basic_mcs_5_8_set:1; | 
 | 	u8 vht_basic_mcs_1_4, vht_basic_mcs_5_8; | 
 | 	u8 he_basic_mcs_1_4, he_basic_mcs_5_8; | 
 | 	u8 eht_mcs7_min_nss; | 
 | 	int error; | 
 | } determine_chan_mode_cases[] = { | 
 | 	{ | 
 | 		.desc = "Normal case, EHT is working", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_EHT, | 
 | 	}, { | 
 | 		.desc = "Requiring EHT support is fine", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.extra_supp_rate = 0x80 | BSS_MEMBERSHIP_SELECTOR_EHT_PHY, | 
 | 	}, { | 
 | 		.desc = "Lowering the mode limits us", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_VHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_VHT, | 
 | 	}, { | 
 | 		.desc = "Requesting a basic rate/selector that we do not support", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.extra_supp_rate = 0x80 | (BSS_MEMBERSHIP_SELECTOR_MIN - 1), | 
 | 		.error = EINVAL, | 
 | 	}, { | 
 | 		.desc = "As before, but userspace says it is taking care of it", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.userspace_selector = BSS_MEMBERSHIP_SELECTOR_MIN - 1, | 
 | 		.extra_supp_rate = 0x80 | (BSS_MEMBERSHIP_SELECTOR_MIN - 1), | 
 | 		.expected_mode = IEEE80211_CONN_MODE_EHT, | 
 | 	}, { | 
 | 		.desc = "Masking out a supported rate in HT capabilities", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_LEGACY, | 
 | 		.ht_capa_mask = { | 
 | 			.mcs.rx_mask[0] = 0xf7, | 
 | 		}, | 
 | 	}, { | 
 | 		.desc = "Masking out a RX rate in VHT capabilities", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_HT, | 
 | 		/* Only one RX stream at MCS 0-7 */ | 
 | 		.vht_capa = { | 
 | 			.supp_mcs.rx_mcs_map = | 
 | 				cpu_to_le16(IEEE80211_VHT_MCS_SUPPORT_0_7), | 
 | 		}, | 
 | 		.vht_capa_mask = { | 
 | 			.supp_mcs.rx_mcs_map = cpu_to_le16(0xffff), | 
 | 		}, | 
 | 		.strict = true, | 
 | 	}, { | 
 | 		.desc = "Masking out a TX rate in VHT capabilities", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_HT, | 
 | 		/* Only one TX stream at MCS 0-7 */ | 
 | 		.vht_capa = { | 
 | 			.supp_mcs.tx_mcs_map = | 
 | 				cpu_to_le16(IEEE80211_VHT_MCS_SUPPORT_0_7), | 
 | 		}, | 
 | 		.vht_capa_mask = { | 
 | 			.supp_mcs.tx_mcs_map = cpu_to_le16(0xffff), | 
 | 		}, | 
 | 		.strict = true, | 
 | 	}, { | 
 | 		.desc = "AP has higher VHT requirement than client", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_HT, | 
 | 		.vht_basic_mcs_5_8_set = 1, | 
 | 		.vht_basic_mcs_5_8 = 0xFE, /* require 5th stream */ | 
 | 		.strict = true, | 
 | 	}, { | 
 | 		.desc = "all zero VHT basic rates are ignored (many APs broken)", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_VHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_VHT, | 
 | 		.vht_basic_mcs_1_4_set = 1, | 
 | 		.vht_basic_mcs_5_8_set = 1, | 
 | 	}, { | 
 | 		.desc = "AP requires 3 HE streams but client only has two", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_VHT, | 
 | 		.he_basic_mcs_1_4 = 0b11001010, | 
 | 		.he_basic_mcs_1_4_set = 1, | 
 | 	}, { | 
 | 		.desc = "all zero HE basic rates are ignored (iPhone workaround)", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_HE, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_HE, | 
 | 		.he_basic_mcs_1_4_set = 1, | 
 | 		.he_basic_mcs_5_8_set = 1, | 
 | 	}, { | 
 | 		.desc = "AP requires too many RX streams with EHT MCS 7", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_HE, | 
 | 		.eht_mcs7_min_nss = 0x15, | 
 | 	}, { | 
 | 		.desc = "AP requires too many TX streams with EHT MCS 7", | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.expected_mode = IEEE80211_CONN_MODE_HE, | 
 | 		.eht_mcs7_min_nss = 0x51, | 
 | 	}, { | 
 | 		.desc = "AP requires too many RX streams with EHT MCS 7 and EHT is required", | 
 | 		.extra_supp_rate = 0x80 | BSS_MEMBERSHIP_SELECTOR_EHT_PHY, | 
 | 		.conn_mode = IEEE80211_CONN_MODE_EHT, | 
 | 		.eht_mcs7_min_nss = 0x15, | 
 | 		.error = EINVAL, | 
 | 	} | 
 | }; | 
 | KUNIT_ARRAY_PARAM_DESC(determine_chan_mode, determine_chan_mode_cases, desc) | 
 |  | 
 | static void test_determine_chan_mode(struct kunit *test) | 
 | { | 
 | 	const struct determine_chan_mode_case *params = test->param_value; | 
 | 	struct t_sdata *t_sdata = T_SDATA(test); | 
 | 	struct ieee80211_conn_settings conn = { | 
 | 		.mode = params->conn_mode, | 
 | 		.bw_limit = IEEE80211_CONN_BW_LIMIT_20, | 
 | 	}; | 
 | 	struct cfg80211_bss cbss = { | 
 | 		.channel = &t_sdata->band_5ghz.channels[0], | 
 | 	}; | 
 | 	unsigned long userspace_selectors[BITS_TO_LONGS(128)] = {}; | 
 | 	u8 bss_ies[] = { | 
 | 		/* Supported Rates */ | 
 | 		WLAN_EID_SUPP_RATES, 0x08, | 
 | 		0x82, 0x84, 0x8b, 0x96, 0xc, 0x12, 0x18, 0x24, | 
 | 		/* Extended Supported Rates */ | 
 | 		WLAN_EID_EXT_SUPP_RATES, 0x05, | 
 | 		0x30, 0x48, 0x60, 0x6c, params->extra_supp_rate, | 
 | 		/* HT Capabilities */ | 
 | 		WLAN_EID_HT_CAPABILITY, 0x1a, | 
 | 		0x0c, 0x00, 0x1b, 0xff, 0xff, 0x00, 0x00, 0x00, | 
 | 		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, | 
 | 		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | 
 | 		0x00, 0x00, | 
 | 		/* HT Information (0xff for 1 stream) */ | 
 | 		WLAN_EID_HT_OPERATION, 0x16, | 
 | 		0x24, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff, 0x00, | 
 | 		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | 
 | 		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | 
 | 		/* VHT Capabilities */ | 
 | 		WLAN_EID_VHT_CAPABILITY, 0xc, | 
 | 		0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0x00, 0x00, | 
 | 		0xff, 0xff, 0x00, 0x00, | 
 | 		/* VHT Operation */ | 
 | 		WLAN_EID_VHT_OPERATION, 0x05, | 
 | 		0x00, 0x00, 0x00, | 
 | 		params->vht_basic_mcs_1_4_set ? | 
 | 			params->vht_basic_mcs_1_4 : | 
 | 			le16_get_bits(t_sdata->band_5ghz.vht_cap.vht_mcs.rx_mcs_map, 0xff), | 
 | 		params->vht_basic_mcs_5_8_set ? | 
 | 			params->vht_basic_mcs_5_8 : | 
 | 			le16_get_bits(t_sdata->band_5ghz.vht_cap.vht_mcs.rx_mcs_map, 0xff00), | 
 | 		/* HE Capabilities */ | 
 | 		WLAN_EID_EXTENSION, 0x16, WLAN_EID_EXT_HE_CAPABILITY, | 
 | 		0x01, 0x78, 0xc8, 0x1a, 0x40, 0x00, 0x00, 0xbf, | 
 | 		0xce, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | 
 | 		0x00, 0xfa, 0xff, 0xfa, 0xff, | 
 | 		/* HE Operation (permit overriding values) */ | 
 | 		WLAN_EID_EXTENSION, 0x07, WLAN_EID_EXT_HE_OPERATION, | 
 | 		0xf0, 0x3f, 0x00, 0xb0, | 
 | 		params->he_basic_mcs_1_4_set ? params->he_basic_mcs_1_4 : 0xfc, | 
 | 		params->he_basic_mcs_5_8_set ? params->he_basic_mcs_5_8 : 0xff, | 
 | 		/* EHT Capabilities */ | 
 | 		WLAN_EID_EXTENSION, 0x12, WLAN_EID_EXT_EHT_CAPABILITY, | 
 | 		0x07, 0x00, 0x1c, 0x00, 0x00, 0xfe, 0xff, 0xff, | 
 | 		0x7f, 0x01, 0x00, 0x88, 0x88, 0x88, 0x00, 0x00, | 
 | 		0x00, | 
 | 		/* EHT Operation */ | 
 | 		WLAN_EID_EXTENSION, 0x09, WLAN_EID_EXT_EHT_OPERATION, | 
 | 		0x01, params->eht_mcs7_min_nss ? params->eht_mcs7_min_nss : 0x11, | 
 | 		0x00, 0x00, 0x00, 0x00, 0x24, 0x00, | 
 | 	}; | 
 | 	struct ieee80211_chan_req chanreq = {}; | 
 | 	struct cfg80211_chan_def ap_chandef = {}; | 
 | 	struct ieee802_11_elems *elems; | 
 |  | 
 | 	if (params->strict) | 
 | 		set_bit(IEEE80211_HW_STRICT, t_sdata->local.hw.flags); | 
 | 	else | 
 | 		clear_bit(IEEE80211_HW_STRICT, t_sdata->local.hw.flags); | 
 |  | 
 | 	t_sdata->sdata->u.mgd.ht_capa_mask = params->ht_capa_mask; | 
 | 	t_sdata->sdata->u.mgd.vht_capa = params->vht_capa; | 
 | 	t_sdata->sdata->u.mgd.vht_capa_mask = params->vht_capa_mask; | 
 |  | 
 | 	if (params->userspace_selector) | 
 | 		set_bit(params->userspace_selector, userspace_selectors); | 
 |  | 
 | 	rcu_assign_pointer(cbss.ies, | 
 | 			   kunit_kzalloc(test, | 
 | 					 sizeof(cbss) + sizeof(bss_ies), | 
 | 					 GFP_KERNEL)); | 
 | 	KUNIT_ASSERT_NOT_NULL(test, rcu_access_pointer(cbss.ies)); | 
 | 	((struct cfg80211_bss_ies *)rcu_access_pointer(cbss.ies))->len = sizeof(bss_ies); | 
 |  | 
 | 	memcpy((void *)rcu_access_pointer(cbss.ies)->data, bss_ies, | 
 | 	       sizeof(bss_ies)); | 
 |  | 
 | 	rcu_read_lock(); | 
 | 	elems = ieee80211_determine_chan_mode(t_sdata->sdata, &conn, &cbss, | 
 | 					      0, &chanreq, &ap_chandef, | 
 | 					      userspace_selectors); | 
 | 	rcu_read_unlock(); | 
 |  | 
 | 	/* We do not need elems, free them if they are valid. */ | 
 | 	if (!IS_ERR_OR_NULL(elems)) | 
 | 		kfree(elems); | 
 |  | 
 | 	if (params->error) { | 
 | 		KUNIT_ASSERT_TRUE(test, IS_ERR(elems)); | 
 | 		KUNIT_ASSERT_EQ(test, PTR_ERR(elems), -params->error); | 
 | 	} else { | 
 | 		KUNIT_ASSERT_NOT_ERR_OR_NULL(test, elems); | 
 | 		KUNIT_ASSERT_EQ(test, conn.mode, params->expected_mode); | 
 | 	} | 
 | } | 
 |  | 
 | static struct kunit_case chan_mode_cases[] = { | 
 | 	KUNIT_CASE_PARAM(test_determine_chan_mode, | 
 | 			 determine_chan_mode_gen_params), | 
 | 	{} | 
 | }; | 
 |  | 
 | static struct kunit_suite chan_mode = { | 
 | 	.name = "mac80211-mlme-chan-mode", | 
 | 	.test_cases = chan_mode_cases, | 
 | }; | 
 |  | 
 | kunit_test_suite(chan_mode); |