| 1 | // SPDX-License-Identifier: GPL-2.0-only |
| 2 | /* |
| 3 | * Utilities for mac80211 unit testing |
| 4 | * |
| 5 | * Copyright (C) 2024 Intel Corporation |
| 6 | */ |
| 7 | #include <linux/ieee80211.h> |
| 8 | #include <net/mac80211.h> |
| 9 | #include <kunit/test.h> |
| 10 | #include <kunit/test-bug.h> |
| 11 | #include "util.h" |
| 12 | |
| 13 | #define CHAN2G(_freq) { \ |
| 14 | .band = NL80211_BAND_2GHZ, \ |
| 15 | .center_freq = (_freq), \ |
| 16 | .hw_value = (_freq), \ |
| 17 | } |
| 18 | |
| 19 | static const struct ieee80211_channel channels_2ghz[] = { |
| 20 | CHAN2G(2412), /* Channel 1 */ |
| 21 | CHAN2G(2417), /* Channel 2 */ |
| 22 | CHAN2G(2422), /* Channel 3 */ |
| 23 | CHAN2G(2427), /* Channel 4 */ |
| 24 | CHAN2G(2432), /* Channel 5 */ |
| 25 | CHAN2G(2437), /* Channel 6 */ |
| 26 | CHAN2G(2442), /* Channel 7 */ |
| 27 | CHAN2G(2447), /* Channel 8 */ |
| 28 | CHAN2G(2452), /* Channel 9 */ |
| 29 | CHAN2G(2457), /* Channel 10 */ |
| 30 | CHAN2G(2462), /* Channel 11 */ |
| 31 | CHAN2G(2467), /* Channel 12 */ |
| 32 | CHAN2G(2472), /* Channel 13 */ |
| 33 | CHAN2G(2484), /* Channel 14 */ |
| 34 | }; |
| 35 | |
| 36 | #define CHAN5G(_freq) { \ |
| 37 | .band = NL80211_BAND_5GHZ, \ |
| 38 | .center_freq = (_freq), \ |
| 39 | .hw_value = (_freq), \ |
| 40 | } |
| 41 | |
| 42 | static const struct ieee80211_channel channels_5ghz[] = { |
| 43 | CHAN5G(5180), /* Channel 36 */ |
| 44 | CHAN5G(5200), /* Channel 40 */ |
| 45 | CHAN5G(5220), /* Channel 44 */ |
| 46 | CHAN5G(5240), /* Channel 48 */ |
| 47 | }; |
| 48 | |
| 49 | static const struct ieee80211_rate bitrates[] = { |
| 50 | { .bitrate = 10 }, |
| 51 | { .bitrate = 20, .flags = IEEE80211_RATE_SHORT_PREAMBLE }, |
| 52 | { .bitrate = 55, .flags = IEEE80211_RATE_SHORT_PREAMBLE }, |
| 53 | { .bitrate = 110, .flags = IEEE80211_RATE_SHORT_PREAMBLE }, |
| 54 | { .bitrate = 60 }, |
| 55 | { .bitrate = 90 }, |
| 56 | { .bitrate = 120 }, |
| 57 | { .bitrate = 180 }, |
| 58 | { .bitrate = 240 }, |
| 59 | { .bitrate = 360 }, |
| 60 | { .bitrate = 480 }, |
| 61 | { .bitrate = 540 } |
| 62 | }; |
| 63 | |
| 64 | /* Copied from hwsim except that it only supports 4 EHT streams and STA/P2P mode */ |
| 65 | static const struct ieee80211_sband_iftype_data sband_capa_5ghz[] = { |
| 66 | { |
| 67 | .types_mask = BIT(NL80211_IFTYPE_STATION) | |
| 68 | BIT(NL80211_IFTYPE_P2P_CLIENT), |
| 69 | .he_cap = { |
| 70 | .has_he = true, |
| 71 | .he_cap_elem = { |
| 72 | .mac_cap_info[0] = |
| 73 | IEEE80211_HE_MAC_CAP0_HTC_HE, |
| 74 | .mac_cap_info[1] = |
| 75 | IEEE80211_HE_MAC_CAP1_TF_MAC_PAD_DUR_16US | |
| 76 | IEEE80211_HE_MAC_CAP1_MULTI_TID_AGG_RX_QOS_8, |
| 77 | .mac_cap_info[2] = |
| 78 | IEEE80211_HE_MAC_CAP2_BSR | |
| 79 | IEEE80211_HE_MAC_CAP2_MU_CASCADING | |
| 80 | IEEE80211_HE_MAC_CAP2_ACK_EN, |
| 81 | .mac_cap_info[3] = |
| 82 | IEEE80211_HE_MAC_CAP3_OMI_CONTROL | |
| 83 | IEEE80211_HE_MAC_CAP3_MAX_AMPDU_LEN_EXP_EXT_3, |
| 84 | .mac_cap_info[4] = IEEE80211_HE_MAC_CAP4_AMSDU_IN_AMPDU, |
| 85 | .phy_cap_info[0] = |
| 86 | IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_40MHZ_80MHZ_IN_5G | |
| 87 | IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G | |
| 88 | IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G, |
| 89 | .phy_cap_info[1] = |
| 90 | IEEE80211_HE_PHY_CAP1_PREAMBLE_PUNC_RX_MASK | |
| 91 | IEEE80211_HE_PHY_CAP1_DEVICE_CLASS_A | |
| 92 | IEEE80211_HE_PHY_CAP1_LDPC_CODING_IN_PAYLOAD | |
| 93 | IEEE80211_HE_PHY_CAP1_MIDAMBLE_RX_TX_MAX_NSTS, |
| 94 | .phy_cap_info[2] = |
| 95 | IEEE80211_HE_PHY_CAP2_NDP_4x_LTF_AND_3_2US | |
| 96 | IEEE80211_HE_PHY_CAP2_STBC_TX_UNDER_80MHZ | |
| 97 | IEEE80211_HE_PHY_CAP2_STBC_RX_UNDER_80MHZ | |
| 98 | IEEE80211_HE_PHY_CAP2_UL_MU_FULL_MU_MIMO | |
| 99 | IEEE80211_HE_PHY_CAP2_UL_MU_PARTIAL_MU_MIMO, |
| 100 | |
| 101 | /* Leave all the other PHY capability bytes |
| 102 | * unset, as DCM, beam forming, RU and PPE |
| 103 | * threshold information are not supported |
| 104 | */ |
| 105 | }, |
| 106 | .he_mcs_nss_supp = { |
| 107 | .rx_mcs_80 = cpu_to_le16(0xfffa), |
| 108 | .tx_mcs_80 = cpu_to_le16(0xfffa), |
| 109 | .rx_mcs_160 = cpu_to_le16(0xfffa), |
| 110 | .tx_mcs_160 = cpu_to_le16(0xfffa), |
| 111 | .rx_mcs_80p80 = cpu_to_le16(0xfffa), |
| 112 | .tx_mcs_80p80 = cpu_to_le16(0xfffa), |
| 113 | }, |
| 114 | }, |
| 115 | .eht_cap = { |
| 116 | .has_eht = true, |
| 117 | .eht_cap_elem = { |
| 118 | .mac_cap_info[0] = |
| 119 | IEEE80211_EHT_MAC_CAP0_EPCS_PRIO_ACCESS | |
| 120 | IEEE80211_EHT_MAC_CAP0_OM_CONTROL | |
| 121 | IEEE80211_EHT_MAC_CAP0_TRIG_TXOP_SHARING_MODE1, |
| 122 | .phy_cap_info[0] = |
| 123 | IEEE80211_EHT_PHY_CAP0_242_TONE_RU_GT20MHZ | |
| 124 | IEEE80211_EHT_PHY_CAP0_NDP_4_EHT_LFT_32_GI | |
| 125 | IEEE80211_EHT_PHY_CAP0_PARTIAL_BW_UL_MU_MIMO | |
| 126 | IEEE80211_EHT_PHY_CAP0_SU_BEAMFORMER | |
| 127 | IEEE80211_EHT_PHY_CAP0_SU_BEAMFORMEE | |
| 128 | IEEE80211_EHT_PHY_CAP0_BEAMFORMEE_SS_80MHZ_MASK, |
| 129 | .phy_cap_info[1] = |
| 130 | IEEE80211_EHT_PHY_CAP1_BEAMFORMEE_SS_80MHZ_MASK | |
| 131 | IEEE80211_EHT_PHY_CAP1_BEAMFORMEE_SS_160MHZ_MASK, |
| 132 | .phy_cap_info[2] = |
| 133 | IEEE80211_EHT_PHY_CAP2_SOUNDING_DIM_80MHZ_MASK | |
| 134 | IEEE80211_EHT_PHY_CAP2_SOUNDING_DIM_160MHZ_MASK, |
| 135 | .phy_cap_info[3] = |
| 136 | IEEE80211_EHT_PHY_CAP3_NG_16_SU_FEEDBACK | |
| 137 | IEEE80211_EHT_PHY_CAP3_NG_16_MU_FEEDBACK | |
| 138 | IEEE80211_EHT_PHY_CAP3_CODEBOOK_4_2_SU_FDBK | |
| 139 | IEEE80211_EHT_PHY_CAP3_CODEBOOK_7_5_MU_FDBK | |
| 140 | IEEE80211_EHT_PHY_CAP3_TRIG_SU_BF_FDBK | |
| 141 | IEEE80211_EHT_PHY_CAP3_TRIG_MU_BF_PART_BW_FDBK | |
| 142 | IEEE80211_EHT_PHY_CAP3_TRIG_CQI_FDBK, |
| 143 | .phy_cap_info[4] = |
| 144 | IEEE80211_EHT_PHY_CAP4_PART_BW_DL_MU_MIMO | |
| 145 | IEEE80211_EHT_PHY_CAP4_PSR_SR_SUPP | |
| 146 | IEEE80211_EHT_PHY_CAP4_POWER_BOOST_FACT_SUPP | |
| 147 | IEEE80211_EHT_PHY_CAP4_EHT_MU_PPDU_4_EHT_LTF_08_GI | |
| 148 | IEEE80211_EHT_PHY_CAP4_MAX_NC_MASK, |
| 149 | .phy_cap_info[5] = |
| 150 | IEEE80211_EHT_PHY_CAP5_NON_TRIG_CQI_FEEDBACK | |
| 151 | IEEE80211_EHT_PHY_CAP5_TX_LESS_242_TONE_RU_SUPP | |
| 152 | IEEE80211_EHT_PHY_CAP5_RX_LESS_242_TONE_RU_SUPP | |
| 153 | IEEE80211_EHT_PHY_CAP5_PPE_THRESHOLD_PRESENT | |
| 154 | IEEE80211_EHT_PHY_CAP5_COMMON_NOMINAL_PKT_PAD_MASK | |
| 155 | IEEE80211_EHT_PHY_CAP5_MAX_NUM_SUPP_EHT_LTF_MASK, |
| 156 | .phy_cap_info[6] = |
| 157 | IEEE80211_EHT_PHY_CAP6_MAX_NUM_SUPP_EHT_LTF_MASK | |
| 158 | IEEE80211_EHT_PHY_CAP6_MCS15_SUPP_MASK, |
| 159 | .phy_cap_info[7] = |
| 160 | IEEE80211_EHT_PHY_CAP7_20MHZ_STA_RX_NDP_WIDER_BW | |
| 161 | IEEE80211_EHT_PHY_CAP7_NON_OFDMA_UL_MU_MIMO_80MHZ | |
| 162 | IEEE80211_EHT_PHY_CAP7_NON_OFDMA_UL_MU_MIMO_160MHZ | |
| 163 | IEEE80211_EHT_PHY_CAP7_MU_BEAMFORMER_80MHZ | |
| 164 | IEEE80211_EHT_PHY_CAP7_MU_BEAMFORMER_160MHZ, |
| 165 | }, |
| 166 | |
| 167 | /* For all MCS and bandwidth, set 4 NSS for both Tx and |
| 168 | * Rx |
| 169 | */ |
| 170 | .eht_mcs_nss_supp = { |
| 171 | /* |
| 172 | * As B1 and B2 are set in the supported |
| 173 | * channel width set field in the HE PHY |
| 174 | * capabilities information field include all |
| 175 | * the following MCS/NSS. |
| 176 | */ |
| 177 | .bw._80 = { |
| 178 | .rx_tx_mcs9_max_nss = 0x44, |
| 179 | .rx_tx_mcs11_max_nss = 0x44, |
| 180 | .rx_tx_mcs13_max_nss = 0x44, |
| 181 | }, |
| 182 | .bw._160 = { |
| 183 | .rx_tx_mcs9_max_nss = 0x44, |
| 184 | .rx_tx_mcs11_max_nss = 0x44, |
| 185 | .rx_tx_mcs13_max_nss = 0x44, |
| 186 | }, |
| 187 | }, |
| 188 | /* PPE threshold information is not supported */ |
| 189 | }, |
| 190 | }, |
| 191 | }; |
| 192 | |
| 193 | int t_sdata_init(struct kunit_resource *resource, void *ctx) |
| 194 | { |
| 195 | struct kunit *test = kunit_get_current_test(); |
| 196 | struct t_sdata *t_sdata; |
| 197 | |
| 198 | t_sdata = kzalloc(sizeof(*t_sdata), GFP_KERNEL); |
| 199 | KUNIT_ASSERT_NOT_NULL(test, t_sdata); |
| 200 | |
| 201 | resource->data = t_sdata; |
| 202 | resource->name = "sdata" ; |
| 203 | |
| 204 | t_sdata->sdata = kzalloc(sizeof(*t_sdata->sdata), GFP_KERNEL); |
| 205 | KUNIT_ASSERT_NOT_NULL(test, t_sdata->sdata); |
| 206 | |
| 207 | t_sdata->wiphy = kzalloc(sizeof(*t_sdata->wiphy), GFP_KERNEL); |
| 208 | KUNIT_ASSERT_NOT_NULL(test, t_sdata->wiphy); |
| 209 | |
| 210 | strscpy(t_sdata->sdata->name, "kunit" ); |
| 211 | |
| 212 | t_sdata->sdata->local = &t_sdata->local; |
| 213 | t_sdata->sdata->local->hw.wiphy = t_sdata->wiphy; |
| 214 | t_sdata->sdata->wdev.wiphy = t_sdata->wiphy; |
| 215 | t_sdata->sdata->vif.type = NL80211_IFTYPE_STATION; |
| 216 | |
| 217 | t_sdata->sdata->deflink.sdata = t_sdata->sdata; |
| 218 | t_sdata->sdata->deflink.link_id = 0; |
| 219 | |
| 220 | t_sdata->wiphy->bands[NL80211_BAND_2GHZ] = &t_sdata->band_2ghz; |
| 221 | t_sdata->wiphy->bands[NL80211_BAND_5GHZ] = &t_sdata->band_5ghz; |
| 222 | |
| 223 | for (int band = NL80211_BAND_2GHZ; band <= NL80211_BAND_5GHZ; band++) { |
| 224 | struct ieee80211_supported_band *sband; |
| 225 | |
| 226 | sband = t_sdata->wiphy->bands[band]; |
| 227 | sband->band = band; |
| 228 | |
| 229 | sband->bitrates = |
| 230 | kmemdup(bitrates, sizeof(bitrates), GFP_KERNEL); |
| 231 | sband->n_bitrates = ARRAY_SIZE(bitrates); |
| 232 | |
| 233 | /* Initialize channels, feel free to add more channels/bands */ |
| 234 | switch (band) { |
| 235 | case NL80211_BAND_2GHZ: |
| 236 | sband->channels = kmemdup(channels_2ghz, |
| 237 | sizeof(channels_2ghz), |
| 238 | GFP_KERNEL); |
| 239 | sband->n_channels = ARRAY_SIZE(channels_2ghz); |
| 240 | sband->bitrates = kmemdup(bitrates, |
| 241 | sizeof(bitrates), |
| 242 | GFP_KERNEL); |
| 243 | sband->n_bitrates = ARRAY_SIZE(bitrates); |
| 244 | break; |
| 245 | case NL80211_BAND_5GHZ: |
| 246 | sband->channels = kmemdup(channels_5ghz, |
| 247 | sizeof(channels_5ghz), |
| 248 | GFP_KERNEL); |
| 249 | sband->n_channels = ARRAY_SIZE(channels_5ghz); |
| 250 | sband->bitrates = kmemdup(bitrates, |
| 251 | sizeof(bitrates), |
| 252 | GFP_KERNEL); |
| 253 | sband->n_bitrates = ARRAY_SIZE(bitrates); |
| 254 | |
| 255 | sband->vht_cap.vht_supported = true; |
| 256 | sband->vht_cap.cap = |
| 257 | IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454 | |
| 258 | IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ | |
| 259 | IEEE80211_VHT_CAP_RXLDPC | |
| 260 | IEEE80211_VHT_CAP_SHORT_GI_80 | |
| 261 | IEEE80211_VHT_CAP_SHORT_GI_160 | |
| 262 | IEEE80211_VHT_CAP_TXSTBC | |
| 263 | IEEE80211_VHT_CAP_RXSTBC_4 | |
| 264 | IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK; |
| 265 | sband->vht_cap.vht_mcs.rx_mcs_map = |
| 266 | cpu_to_le16(IEEE80211_VHT_MCS_SUPPORT_0_9 << 0 | |
| 267 | IEEE80211_VHT_MCS_SUPPORT_0_9 << 2 | |
| 268 | IEEE80211_VHT_MCS_SUPPORT_0_9 << 4 | |
| 269 | IEEE80211_VHT_MCS_SUPPORT_0_9 << 6); |
| 270 | sband->vht_cap.vht_mcs.tx_mcs_map = |
| 271 | sband->vht_cap.vht_mcs.rx_mcs_map; |
| 272 | break; |
| 273 | default: |
| 274 | continue; |
| 275 | } |
| 276 | |
| 277 | sband->ht_cap.ht_supported = band != NL80211_BAND_6GHZ; |
| 278 | sband->ht_cap.cap = IEEE80211_HT_CAP_SUP_WIDTH_20_40 | |
| 279 | IEEE80211_HT_CAP_GRN_FLD | |
| 280 | IEEE80211_HT_CAP_SGI_20 | |
| 281 | IEEE80211_HT_CAP_SGI_40 | |
| 282 | IEEE80211_HT_CAP_DSSSCCK40; |
| 283 | sband->ht_cap.ampdu_factor = 0x3; |
| 284 | sband->ht_cap.ampdu_density = 0x6; |
| 285 | memset(&sband->ht_cap.mcs, 0, sizeof(sband->ht_cap.mcs)); |
| 286 | sband->ht_cap.mcs.rx_mask[0] = 0xff; |
| 287 | sband->ht_cap.mcs.rx_mask[1] = 0xff; |
| 288 | sband->ht_cap.mcs.tx_params = IEEE80211_HT_MCS_TX_DEFINED; |
| 289 | } |
| 290 | |
| 291 | ieee80211_set_sband_iftype_data(&t_sdata->band_5ghz, sband_capa_5ghz); |
| 292 | |
| 293 | return 0; |
| 294 | } |
| 295 | |
| 296 | void t_sdata_exit(struct kunit_resource *resource) |
| 297 | { |
| 298 | struct t_sdata *t_sdata = resource->data; |
| 299 | |
| 300 | kfree(objp: t_sdata->band_2ghz.channels); |
| 301 | kfree(objp: t_sdata->band_2ghz.bitrates); |
| 302 | kfree(objp: t_sdata->band_5ghz.channels); |
| 303 | kfree(objp: t_sdata->band_5ghz.bitrates); |
| 304 | |
| 305 | kfree(objp: t_sdata->sdata); |
| 306 | kfree(objp: t_sdata->wiphy); |
| 307 | |
| 308 | kfree(objp: t_sdata); |
| 309 | } |
| 310 | |