Linux Audio

Check our new training course

Loading...
Note: File does not exist in v5.14.15.
   1// SPDX-License-Identifier: GPL-2.0-only
   2/*
   3 * Copyright 2002-2005, Instant802 Networks, Inc.
   4 * Copyright 2005-2006, Devicescape Software, Inc.
   5 * Copyright 2006-2007	Jiri Benc <jbenc@suse.cz>
   6 * Copyright 2007	Johannes Berg <johannes@sipsolutions.net>
   7 * Copyright 2013-2014  Intel Mobile Communications GmbH
   8 * Copyright (C) 2015-2017	Intel Deutschland GmbH
   9 * Copyright (C) 2018-2024 Intel Corporation
  10 *
  11 * element parsing for mac80211
  12 */
  13
  14#include <net/mac80211.h>
  15#include <linux/netdevice.h>
  16#include <linux/export.h>
  17#include <linux/types.h>
  18#include <linux/slab.h>
  19#include <linux/skbuff.h>
  20#include <linux/etherdevice.h>
  21#include <linux/if_arp.h>
  22#include <linux/bitmap.h>
  23#include <linux/crc32.h>
  24#include <net/net_namespace.h>
  25#include <net/cfg80211.h>
  26#include <net/rtnetlink.h>
  27#include <kunit/visibility.h>
  28
  29#include "ieee80211_i.h"
  30#include "driver-ops.h"
  31#include "rate.h"
  32#include "mesh.h"
  33#include "wme.h"
  34#include "led.h"
  35#include "wep.h"
  36
  37struct ieee80211_elems_parse {
  38	/* must be first for kfree to work */
  39	struct ieee802_11_elems elems;
  40
  41	/* The basic Multi-Link element in the original elements */
  42	const struct element *ml_basic_elem;
  43
  44	/* The reconfiguration Multi-Link element in the original elements */
  45	const struct element *ml_reconf_elem;
  46
  47	/* The EPCS Multi-Link element in the original elements */
  48	const struct element *ml_epcs_elem;
  49
  50	bool multi_link_inner;
  51	bool skip_vendor;
  52
  53	/*
  54	 * scratch buffer that can be used for various element parsing related
  55	 * tasks, e.g., element de-fragmentation etc.
  56	 */
  57	size_t scratch_len;
  58	u8 *scratch_pos;
  59	u8 scratch[] __counted_by(scratch_len);
  60};
  61
  62static void
  63ieee80211_parse_extension_element(u32 *crc,
  64				  const struct element *elem,
  65				  struct ieee80211_elems_parse *elems_parse,
  66				  struct ieee80211_elems_parse_params *params)
  67{
  68	struct ieee802_11_elems *elems = &elems_parse->elems;
  69	const void *data = elem->data + 1;
  70	bool calc_crc = false;
  71	u8 len;
  72
  73	if (!elem->datalen)
  74		return;
  75
  76	len = elem->datalen - 1;
  77
  78	switch (elem->data[0]) {
  79	case WLAN_EID_EXT_HE_MU_EDCA:
  80		if (params->mode < IEEE80211_CONN_MODE_HE)
  81			break;
  82		calc_crc = true;
  83		if (len >= sizeof(*elems->mu_edca_param_set))
  84			elems->mu_edca_param_set = data;
  85		break;
  86	case WLAN_EID_EXT_HE_CAPABILITY:
  87		if (params->mode < IEEE80211_CONN_MODE_HE)
  88			break;
  89		if (ieee80211_he_capa_size_ok(data, len)) {
  90			elems->he_cap = data;
  91			elems->he_cap_len = len;
  92		}
  93		break;
  94	case WLAN_EID_EXT_HE_OPERATION:
  95		if (params->mode < IEEE80211_CONN_MODE_HE)
  96			break;
  97		calc_crc = true;
  98		if (len >= sizeof(*elems->he_operation) &&
  99		    len >= ieee80211_he_oper_size(data) - 1)
 100			elems->he_operation = data;
 101		break;
 102	case WLAN_EID_EXT_UORA:
 103		if (params->mode < IEEE80211_CONN_MODE_HE)
 104			break;
 105		if (len >= 1)
 106			elems->uora_element = data;
 107		break;
 108	case WLAN_EID_EXT_MAX_CHANNEL_SWITCH_TIME:
 109		if (len == 3)
 110			elems->max_channel_switch_time = data;
 111		break;
 112	case WLAN_EID_EXT_MULTIPLE_BSSID_CONFIGURATION:
 113		if (len >= sizeof(*elems->mbssid_config_ie))
 114			elems->mbssid_config_ie = data;
 115		break;
 116	case WLAN_EID_EXT_HE_SPR:
 117		if (params->mode < IEEE80211_CONN_MODE_HE)
 118			break;
 119		if (len >= sizeof(*elems->he_spr) &&
 120		    len >= ieee80211_he_spr_size(data) - 1)
 121			elems->he_spr = data;
 122		break;
 123	case WLAN_EID_EXT_HE_6GHZ_CAPA:
 124		if (params->mode < IEEE80211_CONN_MODE_HE)
 125			break;
 126		if (len >= sizeof(*elems->he_6ghz_capa))
 127			elems->he_6ghz_capa = data;
 128		break;
 129	case WLAN_EID_EXT_EHT_CAPABILITY:
 130		if (params->mode < IEEE80211_CONN_MODE_EHT)
 131			break;
 132		if (ieee80211_eht_capa_size_ok(elems->he_cap,
 133					       data, len,
 134					       params->from_ap)) {
 135			elems->eht_cap = data;
 136			elems->eht_cap_len = len;
 137		}
 138		break;
 139	case WLAN_EID_EXT_EHT_OPERATION:
 140		if (params->mode < IEEE80211_CONN_MODE_EHT)
 141			break;
 142		if (ieee80211_eht_oper_size_ok(data, len))
 143			elems->eht_operation = data;
 144		calc_crc = true;
 145		break;
 146	case WLAN_EID_EXT_EHT_MULTI_LINK:
 147		if (params->mode < IEEE80211_CONN_MODE_EHT)
 148			break;
 149		calc_crc = true;
 150
 151		if (ieee80211_mle_size_ok(data, len)) {
 152			const struct ieee80211_multi_link_elem *mle =
 153				(void *)data;
 154
 155			switch (le16_get_bits(mle->control,
 156					      IEEE80211_ML_CONTROL_TYPE)) {
 157			case IEEE80211_ML_CONTROL_TYPE_BASIC:
 158				if (elems_parse->multi_link_inner) {
 159					elems->parse_error |=
 160						IEEE80211_PARSE_ERR_DUP_NEST_ML_BASIC;
 161					break;
 162				}
 163				break;
 164			case IEEE80211_ML_CONTROL_TYPE_RECONF:
 165				elems_parse->ml_reconf_elem = elem;
 166				break;
 167			case IEEE80211_ML_CONTROL_TYPE_PRIO_ACCESS:
 168				elems_parse->ml_epcs_elem = elem;
 169				break;
 170			default:
 171				break;
 172			}
 173		}
 174		break;
 175	case WLAN_EID_EXT_BANDWIDTH_INDICATION:
 176		if (params->mode < IEEE80211_CONN_MODE_EHT)
 177			break;
 178		if (ieee80211_bandwidth_indication_size_ok(data, len))
 179			elems->bandwidth_indication = data;
 180		calc_crc = true;
 181		break;
 182	case WLAN_EID_EXT_TID_TO_LINK_MAPPING:
 183		if (params->mode < IEEE80211_CONN_MODE_EHT)
 184			break;
 185		calc_crc = true;
 186		if (ieee80211_tid_to_link_map_size_ok(data, len) &&
 187		    elems->ttlm_num < ARRAY_SIZE(elems->ttlm)) {
 188			elems->ttlm[elems->ttlm_num] = (void *)data;
 189			elems->ttlm_num++;
 190		}
 191		break;
 192	}
 193
 194	if (crc && calc_crc)
 195		*crc = crc32_be(*crc, (void *)elem, elem->datalen + 2);
 196}
 197
 198static void ieee80211_parse_tpe(struct ieee80211_parsed_tpe *tpe,
 199				const u8 *data, u8 len)
 200{
 201	const struct ieee80211_tx_pwr_env *env = (const void *)data;
 202	u8 count, interpret, category;
 203	u8 *out, N, *cnt_out = NULL, *N_out = NULL;
 204
 205	if (!ieee80211_valid_tpe_element(data, len))
 206		return;
 207
 208	count = u8_get_bits(env->info, IEEE80211_TX_PWR_ENV_INFO_COUNT);
 209	interpret = u8_get_bits(env->info, IEEE80211_TX_PWR_ENV_INFO_INTERPRET);
 210	category = u8_get_bits(env->info, IEEE80211_TX_PWR_ENV_INFO_CATEGORY);
 211
 212	switch (interpret) {
 213	case IEEE80211_TPE_LOCAL_EIRP:
 214		out = tpe->max_local[category].power;
 215		cnt_out = &tpe->max_local[category].count;
 216		tpe->max_local[category].valid = true;
 217		break;
 218	case IEEE80211_TPE_REG_CLIENT_EIRP:
 219		out = tpe->max_reg_client[category].power;
 220		cnt_out = &tpe->max_reg_client[category].count;
 221		tpe->max_reg_client[category].valid = true;
 222		break;
 223	case IEEE80211_TPE_LOCAL_EIRP_PSD:
 224		out = tpe->psd_local[category].power;
 225		cnt_out = &tpe->psd_local[category].count;
 226		N_out = &tpe->psd_local[category].n;
 227		tpe->psd_local[category].valid = true;
 228		break;
 229	case IEEE80211_TPE_REG_CLIENT_EIRP_PSD:
 230		out = tpe->psd_reg_client[category].power;
 231		cnt_out = &tpe->psd_reg_client[category].count;
 232		N_out = &tpe->psd_reg_client[category].n;
 233		tpe->psd_reg_client[category].valid = true;
 234		break;
 235	}
 236
 237	switch (interpret) {
 238	case IEEE80211_TPE_LOCAL_EIRP:
 239	case IEEE80211_TPE_REG_CLIENT_EIRP:
 240		/* count was validated <= 3, plus 320 MHz */
 241		BUILD_BUG_ON(IEEE80211_TPE_EIRP_ENTRIES_320MHZ < 5);
 242		memcpy(out, env->variable, count + 1);
 243		*cnt_out = count + 1;
 244		/* separately take 320 MHz if present */
 245		if (count == 3 && len > sizeof(*env) + count + 1) {
 246			out[4] = env->variable[4];
 247			*cnt_out = 5;
 248		}
 249		break;
 250	case IEEE80211_TPE_LOCAL_EIRP_PSD:
 251	case IEEE80211_TPE_REG_CLIENT_EIRP_PSD:
 252		if (!count) {
 253			memset(out, env->variable[0],
 254			       IEEE80211_TPE_PSD_ENTRIES_320MHZ);
 255			*cnt_out = IEEE80211_TPE_PSD_ENTRIES_320MHZ;
 256			break;
 257		}
 258
 259		N = 1 << (count - 1);
 260		memcpy(out, env->variable, N);
 261		*cnt_out = N;
 262		*N_out = N;
 263
 264		if (len > sizeof(*env) + N) {
 265			int K = u8_get_bits(env->variable[N],
 266					    IEEE80211_TX_PWR_ENV_EXT_COUNT);
 267
 268			K = min(K, IEEE80211_TPE_PSD_ENTRIES_320MHZ - N);
 269			memcpy(out + N, env->variable + N + 1, K);
 270			(*cnt_out) += K;
 271		}
 272		break;
 273	}
 274}
 275
 276static u32
 277_ieee802_11_parse_elems_full(struct ieee80211_elems_parse_params *params,
 278			     struct ieee80211_elems_parse *elems_parse,
 279			     const struct element *check_inherit)
 280{
 281	struct ieee802_11_elems *elems = &elems_parse->elems;
 282	const struct element *elem;
 283	bool calc_crc = params->filter != 0;
 284	DECLARE_BITMAP(seen_elems, 256);
 285	u32 crc = params->crc;
 286
 287	bitmap_zero(seen_elems, 256);
 288
 289	for_each_element(elem, params->start, params->len) {
 290		const struct element *subelem;
 291		u8 elem_parse_failed;
 292		u8 id = elem->id;
 293		u8 elen = elem->datalen;
 294		const u8 *pos = elem->data;
 295
 296		if (check_inherit &&
 297		    !cfg80211_is_element_inherited(elem,
 298						   check_inherit))
 299			continue;
 300
 301		switch (id) {
 302		case WLAN_EID_SSID:
 303		case WLAN_EID_SUPP_RATES:
 304		case WLAN_EID_FH_PARAMS:
 305		case WLAN_EID_DS_PARAMS:
 306		case WLAN_EID_CF_PARAMS:
 307		case WLAN_EID_TIM:
 308		case WLAN_EID_IBSS_PARAMS:
 309		case WLAN_EID_CHALLENGE:
 310		case WLAN_EID_RSN:
 311		case WLAN_EID_ERP_INFO:
 312		case WLAN_EID_EXT_SUPP_RATES:
 313		case WLAN_EID_HT_CAPABILITY:
 314		case WLAN_EID_HT_OPERATION:
 315		case WLAN_EID_VHT_CAPABILITY:
 316		case WLAN_EID_VHT_OPERATION:
 317		case WLAN_EID_MESH_ID:
 318		case WLAN_EID_MESH_CONFIG:
 319		case WLAN_EID_PEER_MGMT:
 320		case WLAN_EID_PREQ:
 321		case WLAN_EID_PREP:
 322		case WLAN_EID_PERR:
 323		case WLAN_EID_RANN:
 324		case WLAN_EID_CHANNEL_SWITCH:
 325		case WLAN_EID_EXT_CHANSWITCH_ANN:
 326		case WLAN_EID_COUNTRY:
 327		case WLAN_EID_PWR_CONSTRAINT:
 328		case WLAN_EID_TIMEOUT_INTERVAL:
 329		case WLAN_EID_SECONDARY_CHANNEL_OFFSET:
 330		case WLAN_EID_WIDE_BW_CHANNEL_SWITCH:
 331		case WLAN_EID_CHAN_SWITCH_PARAM:
 332		case WLAN_EID_EXT_CAPABILITY:
 333		case WLAN_EID_CHAN_SWITCH_TIMING:
 334		case WLAN_EID_LINK_ID:
 335		case WLAN_EID_BSS_MAX_IDLE_PERIOD:
 336		case WLAN_EID_RSNX:
 337		case WLAN_EID_S1G_BCN_COMPAT:
 338		case WLAN_EID_S1G_CAPABILITIES:
 339		case WLAN_EID_S1G_OPERATION:
 340		case WLAN_EID_AID_RESPONSE:
 341		case WLAN_EID_S1G_SHORT_BCN_INTERVAL:
 342		/*
 343		 * not listing WLAN_EID_CHANNEL_SWITCH_WRAPPER -- it seems possible
 344		 * that if the content gets bigger it might be needed more than once
 345		 */
 346			if (test_bit(id, seen_elems)) {
 347				elems->parse_error |=
 348					IEEE80211_PARSE_ERR_DUP_ELEM;
 349				continue;
 350			}
 351			break;
 352		}
 353
 354		if (calc_crc && id < 64 && (params->filter & (1ULL << id)))
 355			crc = crc32_be(crc, pos - 2, elen + 2);
 356
 357		elem_parse_failed = 0;
 358
 359		switch (id) {
 360		case WLAN_EID_LINK_ID:
 361			if (elen + 2 < sizeof(struct ieee80211_tdls_lnkie)) {
 362				elem_parse_failed =
 363					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 364				break;
 365			}
 366			elems->lnk_id = (void *)(pos - 2);
 367			break;
 368		case WLAN_EID_CHAN_SWITCH_TIMING:
 369			if (elen < sizeof(struct ieee80211_ch_switch_timing)) {
 370				elem_parse_failed =
 371					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 372				break;
 373			}
 374			elems->ch_sw_timing = (void *)pos;
 375			break;
 376		case WLAN_EID_EXT_CAPABILITY:
 377			elems->ext_capab = pos;
 378			elems->ext_capab_len = elen;
 379			break;
 380		case WLAN_EID_SSID:
 381			elems->ssid = pos;
 382			elems->ssid_len = elen;
 383			break;
 384		case WLAN_EID_SUPP_RATES:
 385			elems->supp_rates = pos;
 386			elems->supp_rates_len = elen;
 387			break;
 388		case WLAN_EID_DS_PARAMS:
 389			if (elen >= 1)
 390				elems->ds_params = pos;
 391			else
 392				elem_parse_failed =
 393					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 394			break;
 395		case WLAN_EID_TIM:
 396			if (elen >= sizeof(struct ieee80211_tim_ie)) {
 397				elems->tim = (void *)pos;
 398				elems->tim_len = elen;
 399			} else
 400				elem_parse_failed =
 401					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 402			break;
 403		case WLAN_EID_VENDOR_SPECIFIC:
 404			if (elems_parse->skip_vendor)
 405				break;
 406
 407			if (elen >= 4 && pos[0] == 0x00 && pos[1] == 0x50 &&
 408			    pos[2] == 0xf2) {
 409				/* Microsoft OUI (00:50:F2) */
 410
 411				if (calc_crc)
 412					crc = crc32_be(crc, pos - 2, elen + 2);
 413
 414				if (elen >= 5 && pos[3] == 2) {
 415					/* OUI Type 2 - WMM IE */
 416					if (pos[4] == 0) {
 417						elems->wmm_info = pos;
 418						elems->wmm_info_len = elen;
 419					} else if (pos[4] == 1) {
 420						elems->wmm_param = pos;
 421						elems->wmm_param_len = elen;
 422					}
 423				}
 424			}
 425			break;
 426		case WLAN_EID_RSN:
 427			elems->rsn = pos;
 428			elems->rsn_len = elen;
 429			break;
 430		case WLAN_EID_ERP_INFO:
 431			if (elen >= 1)
 432				elems->erp_info = pos;
 433			else
 434				elem_parse_failed =
 435					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 436			break;
 437		case WLAN_EID_EXT_SUPP_RATES:
 438			elems->ext_supp_rates = pos;
 439			elems->ext_supp_rates_len = elen;
 440			break;
 441		case WLAN_EID_HT_CAPABILITY:
 442			if (params->mode < IEEE80211_CONN_MODE_HT)
 443				break;
 444			if (elen >= sizeof(struct ieee80211_ht_cap))
 445				elems->ht_cap_elem = (void *)pos;
 446			else
 447				elem_parse_failed =
 448					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 449			break;
 450		case WLAN_EID_HT_OPERATION:
 451			if (params->mode < IEEE80211_CONN_MODE_HT)
 452				break;
 453			if (elen >= sizeof(struct ieee80211_ht_operation))
 454				elems->ht_operation = (void *)pos;
 455			else
 456				elem_parse_failed =
 457					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 458			break;
 459		case WLAN_EID_VHT_CAPABILITY:
 460			if (params->mode < IEEE80211_CONN_MODE_VHT)
 461				break;
 462			if (elen >= sizeof(struct ieee80211_vht_cap))
 463				elems->vht_cap_elem = (void *)pos;
 464			else
 465				elem_parse_failed =
 466					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 467			break;
 468		case WLAN_EID_VHT_OPERATION:
 469			if (params->mode < IEEE80211_CONN_MODE_VHT)
 470				break;
 471			if (elen >= sizeof(struct ieee80211_vht_operation)) {
 472				elems->vht_operation = (void *)pos;
 473				if (calc_crc)
 474					crc = crc32_be(crc, pos - 2, elen + 2);
 475				break;
 476			}
 477			elem_parse_failed =
 478				IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 479			break;
 480		case WLAN_EID_OPMODE_NOTIF:
 481			if (params->mode < IEEE80211_CONN_MODE_VHT)
 482				break;
 483			if (elen > 0) {
 484				elems->opmode_notif = pos;
 485				if (calc_crc)
 486					crc = crc32_be(crc, pos - 2, elen + 2);
 487				break;
 488			}
 489			elem_parse_failed =
 490				IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 491			break;
 492		case WLAN_EID_MESH_ID:
 493			elems->mesh_id = pos;
 494			elems->mesh_id_len = elen;
 495			break;
 496		case WLAN_EID_MESH_CONFIG:
 497			if (elen >= sizeof(struct ieee80211_meshconf_ie))
 498				elems->mesh_config = (void *)pos;
 499			else
 500				elem_parse_failed =
 501					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 502			break;
 503		case WLAN_EID_PEER_MGMT:
 504			elems->peering = pos;
 505			elems->peering_len = elen;
 506			break;
 507		case WLAN_EID_MESH_AWAKE_WINDOW:
 508			if (elen >= 2)
 509				elems->awake_window = (void *)pos;
 510			break;
 511		case WLAN_EID_PREQ:
 512			elems->preq = pos;
 513			elems->preq_len = elen;
 514			break;
 515		case WLAN_EID_PREP:
 516			elems->prep = pos;
 517			elems->prep_len = elen;
 518			break;
 519		case WLAN_EID_PERR:
 520			elems->perr = pos;
 521			elems->perr_len = elen;
 522			break;
 523		case WLAN_EID_RANN:
 524			if (elen >= sizeof(struct ieee80211_rann_ie))
 525				elems->rann = (void *)pos;
 526			else
 527				elem_parse_failed =
 528					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 529			break;
 530		case WLAN_EID_CHANNEL_SWITCH:
 531			if (elen != sizeof(struct ieee80211_channel_sw_ie)) {
 532				elem_parse_failed =
 533					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 534				break;
 535			}
 536			elems->ch_switch_ie = (void *)pos;
 537			break;
 538		case WLAN_EID_EXT_CHANSWITCH_ANN:
 539			if (elen != sizeof(struct ieee80211_ext_chansw_ie)) {
 540				elem_parse_failed =
 541					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 542				break;
 543			}
 544			elems->ext_chansw_ie = (void *)pos;
 545			break;
 546		case WLAN_EID_SECONDARY_CHANNEL_OFFSET:
 547			if (params->mode < IEEE80211_CONN_MODE_HT)
 548				break;
 549			if (elen != sizeof(struct ieee80211_sec_chan_offs_ie)) {
 550				elem_parse_failed =
 551					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 552				break;
 553			}
 554			elems->sec_chan_offs = (void *)pos;
 555			break;
 556		case WLAN_EID_CHAN_SWITCH_PARAM:
 557			if (elen <
 558			    sizeof(*elems->mesh_chansw_params_ie)) {
 559				elem_parse_failed =
 560					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 561				break;
 562			}
 563			elems->mesh_chansw_params_ie = (void *)pos;
 564			break;
 565		case WLAN_EID_WIDE_BW_CHANNEL_SWITCH:
 566			if (params->mode < IEEE80211_CONN_MODE_VHT)
 567				break;
 568
 569			if (!params->action) {
 570				elem_parse_failed =
 571					IEEE80211_PARSE_ERR_UNEXPECTED_ELEM;
 572				break;
 573			}
 574
 575			if (elen < sizeof(*elems->wide_bw_chansw_ie)) {
 576				elem_parse_failed =
 577					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 578				break;
 579			}
 580			elems->wide_bw_chansw_ie = (void *)pos;
 581			break;
 582		case WLAN_EID_CHANNEL_SWITCH_WRAPPER:
 583			if (params->mode < IEEE80211_CONN_MODE_VHT)
 584				break;
 585			if (params->action) {
 586				elem_parse_failed =
 587					IEEE80211_PARSE_ERR_UNEXPECTED_ELEM;
 588				break;
 589			}
 590			/*
 591			 * This is a bit tricky, but as we only care about
 592			 * a few elements, parse them out manually.
 593			 */
 594			subelem = cfg80211_find_elem(WLAN_EID_WIDE_BW_CHANNEL_SWITCH,
 595						     pos, elen);
 596			if (subelem) {
 597				if (subelem->datalen >= sizeof(*elems->wide_bw_chansw_ie))
 598					elems->wide_bw_chansw_ie =
 599						(void *)subelem->data;
 600				else
 601					elem_parse_failed =
 602						IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 603			}
 604
 605			if (params->mode < IEEE80211_CONN_MODE_EHT)
 606				break;
 607
 608			subelem = cfg80211_find_ext_elem(WLAN_EID_EXT_BANDWIDTH_INDICATION,
 609							 pos, elen);
 610			if (subelem) {
 611				const void *edata = subelem->data + 1;
 612				u8 edatalen = subelem->datalen - 1;
 613
 614				if (ieee80211_bandwidth_indication_size_ok(edata,
 615									   edatalen))
 616					elems->bandwidth_indication = edata;
 617				else
 618					elem_parse_failed =
 619						IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 620			}
 621
 622			subelem = cfg80211_find_ext_elem(WLAN_EID_TX_POWER_ENVELOPE,
 623							 pos, elen);
 624			if (subelem)
 625				ieee80211_parse_tpe(&elems->csa_tpe,
 626						    subelem->data + 1,
 627						    subelem->datalen - 1);
 628			break;
 629		case WLAN_EID_COUNTRY:
 630			elems->country_elem = pos;
 631			elems->country_elem_len = elen;
 632			break;
 633		case WLAN_EID_PWR_CONSTRAINT:
 634			if (elen != 1) {
 635				elem_parse_failed =
 636					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 637				break;
 638			}
 639			elems->pwr_constr_elem = pos;
 640			break;
 641		case WLAN_EID_CISCO_VENDOR_SPECIFIC:
 642			/* Lots of different options exist, but we only care
 643			 * about the Dynamic Transmit Power Control element.
 644			 * First check for the Cisco OUI, then for the DTPC
 645			 * tag (0x00).
 646			 */
 647			if (elen < 4) {
 648				elem_parse_failed =
 649					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 650				break;
 651			}
 652
 653			if (pos[0] != 0x00 || pos[1] != 0x40 ||
 654			    pos[2] != 0x96 || pos[3] != 0x00)
 655				break;
 656
 657			if (elen != 6) {
 658				elem_parse_failed =
 659					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 660				break;
 661			}
 662
 663			if (calc_crc)
 664				crc = crc32_be(crc, pos - 2, elen + 2);
 665
 666			elems->cisco_dtpc_elem = pos;
 667			break;
 668		case WLAN_EID_ADDBA_EXT:
 669			if (elen < sizeof(struct ieee80211_addba_ext_ie)) {
 670				elem_parse_failed =
 671					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 672				break;
 673			}
 674			elems->addba_ext_ie = (void *)pos;
 675			break;
 676		case WLAN_EID_TIMEOUT_INTERVAL:
 677			if (elen >= sizeof(struct ieee80211_timeout_interval_ie))
 678				elems->timeout_int = (void *)pos;
 679			else
 680				elem_parse_failed =
 681					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 682			break;
 683		case WLAN_EID_BSS_MAX_IDLE_PERIOD:
 684			if (elen >= sizeof(*elems->max_idle_period_ie))
 685				elems->max_idle_period_ie = (void *)pos;
 686			break;
 687		case WLAN_EID_RSNX:
 688			elems->rsnx = pos;
 689			elems->rsnx_len = elen;
 690			break;
 691		case WLAN_EID_TX_POWER_ENVELOPE:
 692			if (params->mode < IEEE80211_CONN_MODE_HE)
 693				break;
 694			ieee80211_parse_tpe(&elems->tpe, pos, elen);
 695			break;
 696		case WLAN_EID_EXTENSION:
 697			ieee80211_parse_extension_element(calc_crc ?
 698								&crc : NULL,
 699							  elem, elems_parse,
 700							  params);
 701			break;
 702		case WLAN_EID_S1G_CAPABILITIES:
 703			if (params->mode != IEEE80211_CONN_MODE_S1G)
 704				break;
 705			if (elen >= sizeof(*elems->s1g_capab))
 706				elems->s1g_capab = (void *)pos;
 707			else
 708				elem_parse_failed =
 709					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 710			break;
 711		case WLAN_EID_S1G_OPERATION:
 712			if (params->mode != IEEE80211_CONN_MODE_S1G)
 713				break;
 714			if (elen == sizeof(*elems->s1g_oper))
 715				elems->s1g_oper = (void *)pos;
 716			else
 717				elem_parse_failed =
 718					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 719			break;
 720		case WLAN_EID_S1G_BCN_COMPAT:
 721			if (params->mode != IEEE80211_CONN_MODE_S1G)
 722				break;
 723			if (elen == sizeof(*elems->s1g_bcn_compat))
 724				elems->s1g_bcn_compat = (void *)pos;
 725			else
 726				elem_parse_failed =
 727					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 728			break;
 729		case WLAN_EID_AID_RESPONSE:
 730			if (params->mode != IEEE80211_CONN_MODE_S1G)
 731				break;
 732			if (elen == sizeof(struct ieee80211_aid_response_ie))
 733				elems->aid_resp = (void *)pos;
 734			else
 735				elem_parse_failed =
 736					IEEE80211_PARSE_ERR_BAD_ELEM_SIZE;
 737			break;
 738		default:
 739			break;
 740		}
 741
 742		if (elem_parse_failed)
 743			elems->parse_error |= elem_parse_failed;
 744		else
 745			__set_bit(id, seen_elems);
 746	}
 747
 748	if (!for_each_element_completed(elem, params->start, params->len))
 749		elems->parse_error |= IEEE80211_PARSE_ERR_INVALID_END;
 750
 751	return crc;
 752}
 753
 754static size_t ieee802_11_find_bssid_profile(const u8 *start, size_t len,
 755					    struct ieee802_11_elems *elems,
 756					    struct cfg80211_bss *bss,
 757					    u8 *nontransmitted_profile)
 758{
 759	const struct element *elem, *sub;
 760	size_t profile_len = 0;
 761	bool found = false;
 762
 763	if (!bss || !bss->transmitted_bss)
 764		return profile_len;
 765
 766	for_each_element_id(elem, WLAN_EID_MULTIPLE_BSSID, start, len) {
 767		if (elem->datalen < 2)
 768			continue;
 769		if (elem->data[0] < 1 || elem->data[0] > 8)
 770			continue;
 771
 772		for_each_element(sub, elem->data + 1, elem->datalen - 1) {
 773			u8 new_bssid[ETH_ALEN];
 774			const u8 *index;
 775
 776			if (sub->id != 0 || sub->datalen < 4) {
 777				/* not a valid BSS profile */
 778				continue;
 779			}
 780
 781			if (sub->data[0] != WLAN_EID_NON_TX_BSSID_CAP ||
 782			    sub->data[1] != 2) {
 783				/* The first element of the
 784				 * Nontransmitted BSSID Profile is not
 785				 * the Nontransmitted BSSID Capability
 786				 * element.
 787				 */
 788				continue;
 789			}
 790
 791			memset(nontransmitted_profile, 0, len);
 792			profile_len = cfg80211_merge_profile(start, len,
 793							     elem,
 794							     sub,
 795							     nontransmitted_profile,
 796							     len);
 797
 798			/* found a Nontransmitted BSSID Profile */
 799			index = cfg80211_find_ie(WLAN_EID_MULTI_BSSID_IDX,
 800						 nontransmitted_profile,
 801						 profile_len);
 802			if (!index || index[1] < 1 || index[2] == 0) {
 803				/* Invalid MBSSID Index element */
 804				continue;
 805			}
 806
 807			cfg80211_gen_new_bssid(bss->transmitted_bss->bssid,
 808					       elem->data[0],
 809					       index[2],
 810					       new_bssid);
 811			if (ether_addr_equal(new_bssid, bss->bssid)) {
 812				found = true;
 813				elems->bssid_index_len = index[1];
 814				elems->bssid_index = (void *)&index[2];
 815				break;
 816			}
 817		}
 818	}
 819
 820	return found ? profile_len : 0;
 821}
 822
 823static void
 824ieee80211_mle_get_sta_prof(struct ieee80211_elems_parse *elems_parse,
 825			   u8 link_id)
 826{
 827	struct ieee802_11_elems *elems = &elems_parse->elems;
 828	const struct ieee80211_multi_link_elem *ml = elems->ml_basic;
 829	ssize_t ml_len = elems->ml_basic_len;
 830	const struct element *sub;
 831
 832	for_each_mle_subelement(sub, (u8 *)ml, ml_len) {
 833		struct ieee80211_mle_per_sta_profile *prof = (void *)sub->data;
 834		ssize_t sta_prof_len;
 835		u16 control;
 836
 837		if (sub->id != IEEE80211_MLE_SUBELEM_PER_STA_PROFILE)
 838			continue;
 839
 840		if (!ieee80211_mle_basic_sta_prof_size_ok(sub->data,
 841							  sub->datalen))
 842			return;
 843
 844		control = le16_to_cpu(prof->control);
 845
 846		if (link_id != u16_get_bits(control,
 847					    IEEE80211_MLE_STA_CONTROL_LINK_ID))
 848			continue;
 849
 850		if (!(control & IEEE80211_MLE_STA_CONTROL_COMPLETE_PROFILE))
 851			return;
 852
 853		/* the sub element can be fragmented */
 854		sta_prof_len =
 855			cfg80211_defragment_element(sub,
 856						    (u8 *)ml, ml_len,
 857						    elems_parse->scratch_pos,
 858						    elems_parse->scratch +
 859							elems_parse->scratch_len -
 860							elems_parse->scratch_pos,
 861						    IEEE80211_MLE_SUBELEM_FRAGMENT);
 862
 863		if (sta_prof_len < 0)
 864			return;
 865
 866		elems->prof = (void *)elems_parse->scratch_pos;
 867		elems->sta_prof_len = sta_prof_len;
 868		elems_parse->scratch_pos += sta_prof_len;
 869
 870		return;
 871	}
 872}
 873
 874static const struct element *
 875ieee80211_prep_mle_link_parse(struct ieee80211_elems_parse *elems_parse,
 876			      struct ieee80211_elems_parse_params *params,
 877			      struct ieee80211_elems_parse_params *sub)
 878{
 879	struct ieee802_11_elems *elems = &elems_parse->elems;
 880	struct ieee80211_mle_per_sta_profile *prof;
 881	const struct element *tmp;
 882	ssize_t ml_len;
 883	const u8 *end;
 884
 885	if (params->mode < IEEE80211_CONN_MODE_EHT)
 886		return NULL;
 887
 888	for_each_element_extid(tmp, WLAN_EID_EXT_EHT_MULTI_LINK,
 889			       elems->ie_start, elems->total_len) {
 890		const struct ieee80211_multi_link_elem *mle =
 891			(void *)tmp->data + 1;
 892
 893		if (!ieee80211_mle_size_ok(tmp->data + 1, tmp->datalen - 1))
 894			continue;
 895
 896		if (le16_get_bits(mle->control, IEEE80211_ML_CONTROL_TYPE) !=
 897		    IEEE80211_ML_CONTROL_TYPE_BASIC)
 898			continue;
 899
 900		elems_parse->ml_basic_elem = tmp;
 901		break;
 902	}
 903
 904	ml_len = cfg80211_defragment_element(elems_parse->ml_basic_elem,
 905					     elems->ie_start,
 906					     elems->total_len,
 907					     elems_parse->scratch_pos,
 908					     elems_parse->scratch +
 909						elems_parse->scratch_len -
 910						elems_parse->scratch_pos,
 911					     WLAN_EID_FRAGMENT);
 912
 913	if (ml_len < 0)
 914		return NULL;
 915
 916	elems->ml_basic = (const void *)elems_parse->scratch_pos;
 917	elems->ml_basic_len = ml_len;
 918	elems_parse->scratch_pos += ml_len;
 919
 920	if (params->link_id == -1)
 921		return NULL;
 922
 923	ieee80211_mle_get_sta_prof(elems_parse, params->link_id);
 924	prof = elems->prof;
 925
 926	if (!prof)
 927		return NULL;
 928
 929	/* check if we have the 4 bytes for the fixed part in assoc response */
 930	if (elems->sta_prof_len < sizeof(*prof) + prof->sta_info_len - 1 + 4) {
 931		elems->prof = NULL;
 932		elems->sta_prof_len = 0;
 933		return NULL;
 934	}
 935
 936	/*
 937	 * Skip the capability information and the status code that are expected
 938	 * as part of the station profile in association response frames. Note
 939	 * the -1 is because the 'sta_info_len' is accounted to as part of the
 940	 * per-STA profile, but not part of the 'u8 variable[]' portion.
 941	 */
 942	sub->start = prof->variable + prof->sta_info_len - 1 + 4;
 943	end = (const u8 *)prof + elems->sta_prof_len;
 944	sub->len = end - sub->start;
 945
 946	sub->mode = params->mode;
 947	sub->action = params->action;
 948	sub->from_ap = params->from_ap;
 949	sub->link_id = -1;
 950
 951	return cfg80211_find_ext_elem(WLAN_EID_EXT_NON_INHERITANCE,
 952				      sub->start, sub->len);
 953}
 954
 955static void
 956ieee80211_mle_defrag_reconf(struct ieee80211_elems_parse *elems_parse)
 957{
 958	struct ieee802_11_elems *elems = &elems_parse->elems;
 959	ssize_t ml_len;
 960
 961	ml_len = cfg80211_defragment_element(elems_parse->ml_reconf_elem,
 962					     elems->ie_start,
 963					     elems->total_len,
 964					     elems_parse->scratch_pos,
 965					     elems_parse->scratch +
 966						elems_parse->scratch_len -
 967						elems_parse->scratch_pos,
 968					     WLAN_EID_FRAGMENT);
 969	if (ml_len < 0)
 970		return;
 971	elems->ml_reconf = (void *)elems_parse->scratch_pos;
 972	elems->ml_reconf_len = ml_len;
 973	elems_parse->scratch_pos += ml_len;
 974}
 975
 976static void
 977ieee80211_mle_defrag_epcs(struct ieee80211_elems_parse *elems_parse)
 978{
 979	struct ieee802_11_elems *elems = &elems_parse->elems;
 980	ssize_t ml_len;
 981
 982	ml_len = cfg80211_defragment_element(elems_parse->ml_epcs_elem,
 983					     elems->ie_start,
 984					     elems->total_len,
 985					     elems_parse->scratch_pos,
 986					     elems_parse->scratch +
 987						elems_parse->scratch_len -
 988						elems_parse->scratch_pos,
 989					     WLAN_EID_FRAGMENT);
 990	if (ml_len < 0)
 991		return;
 992	elems->ml_epcs = (void *)elems_parse->scratch_pos;
 993	elems->ml_epcs_len = ml_len;
 994	elems_parse->scratch_pos += ml_len;
 995}
 996
 997struct ieee802_11_elems *
 998ieee802_11_parse_elems_full(struct ieee80211_elems_parse_params *params)
 999{
1000	struct ieee80211_elems_parse_params sub = {};
1001	struct ieee80211_elems_parse *elems_parse;
1002	const struct element *non_inherit = NULL;
1003	struct ieee802_11_elems *elems;
1004	size_t scratch_len = 3 * params->len;
1005	bool multi_link_inner = false;
1006
1007	BUILD_BUG_ON(offsetof(typeof(*elems_parse), elems) != 0);
1008
1009	/* cannot parse for both a specific link and non-transmitted BSS */
1010	if (WARN_ON(params->link_id >= 0 && params->bss))
1011		return NULL;
1012
1013	elems_parse = kzalloc(struct_size(elems_parse, scratch, scratch_len),
1014			      GFP_ATOMIC);
1015	if (!elems_parse)
1016		return NULL;
1017
1018	elems_parse->scratch_len = scratch_len;
1019	elems_parse->scratch_pos = elems_parse->scratch;
1020
1021	elems = &elems_parse->elems;
1022	elems->ie_start = params->start;
1023	elems->total_len = params->len;
1024
1025	/* set all TPE entries to unlimited (but invalid) */
1026	ieee80211_clear_tpe(&elems->tpe);
1027	ieee80211_clear_tpe(&elems->csa_tpe);
1028
1029	/*
1030	 * If we're looking for a non-transmitted BSS then we cannot at
1031	 * the same time be looking for a second link as the two can only
1032	 * appear in the same frame carrying info for different BSSes.
1033	 *
1034	 * In any case, we only look for one at a time, as encoded by
1035	 * the WARN_ON above.
1036	 */
1037	if (params->bss) {
1038		int nontx_len =
1039			ieee802_11_find_bssid_profile(params->start,
1040						      params->len,
1041						      elems, params->bss,
1042						      elems_parse->scratch_pos);
1043		sub.start = elems_parse->scratch_pos;
1044		sub.mode = params->mode;
1045		sub.len = nontx_len;
1046		sub.action = params->action;
1047		sub.link_id = params->link_id;
1048
1049		/* consume the space used for non-transmitted profile */
1050		elems_parse->scratch_pos += nontx_len;
1051
1052		non_inherit = cfg80211_find_ext_elem(WLAN_EID_EXT_NON_INHERITANCE,
1053						     sub.start, nontx_len);
1054	} else {
1055		/* must always parse to get elems_parse->ml_basic_elem */
1056		non_inherit = ieee80211_prep_mle_link_parse(elems_parse, params,
1057							    &sub);
1058		multi_link_inner = true;
1059	}
1060
1061	elems_parse->skip_vendor =
1062		cfg80211_find_elem(WLAN_EID_VENDOR_SPECIFIC,
1063				   sub.start, sub.len);
1064	elems->crc = _ieee802_11_parse_elems_full(params, elems_parse,
1065						  non_inherit);
1066
1067	/* Override with nontransmitted/per-STA profile if found */
1068	if (sub.len) {
1069		elems_parse->multi_link_inner = multi_link_inner;
1070		elems_parse->skip_vendor = false;
1071		_ieee802_11_parse_elems_full(&sub, elems_parse, NULL);
1072	}
1073
1074	ieee80211_mle_defrag_reconf(elems_parse);
1075
1076	ieee80211_mle_defrag_epcs(elems_parse);
1077
1078	if (elems->tim && !elems->parse_error) {
1079		const struct ieee80211_tim_ie *tim_ie = elems->tim;
1080
1081		elems->dtim_period = tim_ie->dtim_period;
1082		elems->dtim_count = tim_ie->dtim_count;
1083	}
1084
1085	/* Override DTIM period and count if needed */
1086	if (elems->bssid_index &&
1087	    elems->bssid_index_len >=
1088	    offsetofend(struct ieee80211_bssid_index, dtim_period))
1089		elems->dtim_period = elems->bssid_index->dtim_period;
1090
1091	if (elems->bssid_index &&
1092	    elems->bssid_index_len >=
1093	    offsetofend(struct ieee80211_bssid_index, dtim_count))
1094		elems->dtim_count = elems->bssid_index->dtim_count;
1095
1096	return elems;
1097}
1098EXPORT_SYMBOL_IF_KUNIT(ieee802_11_parse_elems_full);
1099
1100int ieee80211_parse_bitrates(enum nl80211_chan_width width,
1101			     const struct ieee80211_supported_band *sband,
1102			     const u8 *srates, int srates_len, u32 *rates)
1103{
1104	u32 rate_flags = ieee80211_chanwidth_rate_flags(width);
1105	struct ieee80211_rate *br;
1106	int brate, rate, i, j, count = 0;
1107
1108	*rates = 0;
1109
1110	for (i = 0; i < srates_len; i++) {
1111		rate = srates[i] & 0x7f;
1112
1113		for (j = 0; j < sband->n_bitrates; j++) {
1114			br = &sband->bitrates[j];
1115			if ((rate_flags & br->flags) != rate_flags)
1116				continue;
1117
1118			brate = DIV_ROUND_UP(br->bitrate, 5);
1119			if (brate == rate) {
1120				*rates |= BIT(j);
1121				count++;
1122				break;
1123			}
1124		}
1125	}
1126	return count;
1127}