qcacld-3.0: Handle puncture bitmap in EHT operation IE dynamically

Puncture bitmap can be changed dynamically. 11be station should monitor
puncture bitmap in EHT operation IE and update it to firmware.

Change-Id: I9ec2ac592b7598d12b071f2bfad4fdbd449f04ee
CRs-Fixed: 3269516

core/mac/src/pe/lim/lim_process_beacon_frame.c

@@ -154,6 +154,137 @@ lim_validate_rsn_ie(const uint8_t *ie_ptr, uint16_t ie_len)
 	return QDF_STATUS_SUCCESS;
 }
 
+#ifdef WLAN_FEATURE_11BE
+/**
+ * lim_get_update_eht_bw_puncture_allow() - whether bw and puncture can be
+ *                                          sent to target directly
+ * @session: pe session
+ * @ori_bw: bandwdith from beacon
+ * @new_bw: bandwidth intersection between reference AP and STA
+ * @update_allow: return true if bw and puncture can be updated directly
+ *
+ * Return: QDF_STATUS
+ */
+static QDF_STATUS
+lim_get_update_eht_bw_puncture_allow(struct pe_session *session,
+				     enum phy_ch_width ori_bw,
+				     enum phy_ch_width *new_bw,
+				     bool *update_allow)
+{
+	enum phy_ch_width ch_width;
+	struct wlan_objmgr_psoc *psoc;
+	enum wlan_phymode phy_mode;
+	QDF_STATUS status = QDF_STATUS_SUCCESS;
+
+	*update_allow = false;
+
+	psoc = wlan_vdev_get_psoc(session->vdev);
+	if (!psoc) {
+		pe_err("psoc object invalid");
+		return QDF_STATUS_E_INVAL;
+	}
+	status = mlme_get_peer_phymode(psoc, session->bssId, &phy_mode);
+	if (QDF_IS_STATUS_ERROR(status)) {
+		pe_err("failed to get phy_mode %d mac: " QDF_MAC_ADDR_FMT,
+		       status, QDF_MAC_ADDR_REF(session->bssId));
+		return QDF_STATUS_E_INVAL;
+	}
+	ch_width = wlan_mlme_get_ch_width_from_phymode(phy_mode);
+
+	if (ori_bw <= ch_width) {
+		*new_bw = ori_bw;
+		*update_allow = true;
+		return QDF_STATUS_SUCCESS;
+	}
+
+	if ((ori_bw == CH_WIDTH_320MHZ) &&
+	    !session->eht_config.support_320mhz_6ghz) {
+		if (ch_width == CH_WIDTH_160MHZ) {
+			*new_bw = CH_WIDTH_160MHZ;
+			*update_allow = true;
+			return QDF_STATUS_SUCCESS;
+		}
+	}
+
+	return QDF_STATUS_SUCCESS;
+}
+
+void lim_process_beacon_eht(struct mac_context *mac_ctx,
+			    struct pe_session *session,
+			    tSchBeaconStruct *bcn_ptr)
+{
+	struct wlan_objmgr_vdev *vdev;
+	struct wlan_channel *des_chan;
+	uint16_t ori_punc = 0;
+	enum phy_ch_width ori_bw = CH_WIDTH_INVALID;
+	uint8_t cb_mode;
+	enum phy_ch_width new_bw;
+	bool update_allow;
+	QDF_STATUS status;
+
+	if (!session || !bcn_ptr || !mac_ctx) {
+		pe_err("invalid input parameters");
+		return;
+	}
+	vdev = session->vdev;
+	if (!vdev || wlan_vdev_mlme_get_opmode(vdev) != QDF_STA_MODE ||
+	    !qdf_is_macaddr_equal((struct qdf_mac_addr *)session->bssId,
+				  (struct qdf_mac_addr *)bcn_ptr->bssid))
+		return;
+	des_chan = wlan_vdev_mlme_get_des_chan(vdev);
+	if (!des_chan || !IS_WLAN_PHYMODE_EHT(des_chan->ch_phymode))
+		return;
+
+	if (mlo_is_mld_sta(vdev)) {
+		/* handle beacon IE for 11be mlo case */
+		lim_process_beacon_mlo(mac_ctx, session, bcn_ptr);
+		return;
+	}
+
+	if (wlan_reg_is_24ghz_ch_freq(session->curr_op_freq)) {
+		if (session->force_24ghz_in_ht20)
+			cb_mode = WNI_CFG_CHANNEL_BONDING_MODE_DISABLE;
+		else
+			cb_mode =
+			   mac_ctx->roam.configParam.channelBondingMode24GHz;
+	} else {
+		cb_mode = mac_ctx->roam.configParam.channelBondingMode5GHz;
+	}
+
+	if (cb_mode == WNI_CFG_CHANNEL_BONDING_MODE_DISABLE) {
+		/*
+		 * if channel bonding is disabled from INI do not
+		 * update the chan width
+		 */
+		pe_debug_rl("chan banding is disabled skip bw update");
+
+		return;
+	}
+	/* handle beacon IE for 11be non-mlo case */
+	if (bcn_ptr->eht_op.disabled_sub_chan_bitmap_present) {
+		ori_punc = QDF_GET_BITS(
+		    bcn_ptr->eht_op.disabled_sub_chan_bitmap[0][0], 0, 8);
+		ori_punc |= QDF_GET_BITS(
+		    bcn_ptr->eht_op.disabled_sub_chan_bitmap[0][1], 0, 8) << 8;
+	}
+	if (bcn_ptr->eht_op.eht_op_information_present) {
+		ori_bw = wlan_mlme_convert_eht_op_bw_to_phy_ch_width(
+						bcn_ptr->eht_op.channel_width);
+		status = lim_get_update_eht_bw_puncture_allow(session, ori_bw,
+							      &new_bw,
+							      &update_allow);
+		if (QDF_IS_STATUS_ERROR(status))
+			return;
+		if (update_allow)
+			wlan_cm_sta_update_bw_puncture(vdev, session->bssId,
+						       ori_punc, ori_bw,
+						       bcn_ptr->eht_op.ccfs0,
+						       bcn_ptr->eht_op.ccfs1,
+						       new_bw);
+	}
+}
+#endif
+
 /**
  * lim_process_beacon_frame() - to process beacon frames
  * @mac_ctx: Pointer to Global MAC structure
@@ -226,7 +357,7 @@ lim_process_beacon_frame(struct mac_context *mac_ctx, uint8_t *rx_pkt_info,
 	if (QDF_IS_STATUS_SUCCESS(lim_check_for_ml_probe_req(session)))
 		goto end;
 
-	lim_process_beacon_mlo(mac_ctx, session, bcn_ptr);
+	lim_process_beacon_eht(mac_ctx, session, bcn_ptr);
 
 	/*
 	 * during scanning, when any session is active, and

core/mac/src/pe/lim/lim_types.h

@@ -426,6 +426,27 @@ void lim_cleanup_mlm(struct mac_context *);
 
 /* Management frame handling functions */
 
+#ifdef WLAN_FEATURE_11BE
+/**
+ * lim_process_beacon_eht() - process beacon 11be IE
+ * @mac_ctx: global mac context
+ * @session: pe session
+ * @bcn_ptr: pointer to tSchBeaconStruct
+ *
+ * Return none
+ */
+void lim_process_beacon_eht(struct mac_context *mac_ctx,
+			    struct pe_session *session,
+			    tSchBeaconStruct *bcn_ptr);
+#else
+static inline
+void lim_process_beacon_eht(struct mac_context *mac_ctx,
+			    struct pe_session *session,
+			    tSchBeaconStruct *bcn_ptr)
+{
+}
+#endif
+
 #ifdef WLAN_FEATURE_11BE_MLO
 /**
  * lim_process_beacon_mlo() - process beacon mlo IE

core/mac/src/pe/sch/sch_beacon_process.c

@@ -450,6 +450,10 @@ sch_bcn_update_opmode_change(struct mac_context *mac_ctx, tpDphHashNode sta_ds,
 		pe_debug("Ignore opmode change as channel switch is in progress");
 		return;
 	}
+	if (bcn->eht_op.eht_op_information_present) {
+		pe_debug("Ignore opmode change as there is EHT operation information");
+		return;
+	}
 
 	if (session->vhtCapability && bcn->OperatingMode.present) {
 		pe_debug("OMN IE is present in the beacon, update NSS/Ch width");
@@ -768,6 +772,7 @@ static void __sch_beacon_process_for_session(struct mac_context *mac_ctx,
 							      session);
 		session->send_p2p_conf_frame = false;
 	}
+	lim_process_beacon_eht(mac_ctx, session, bcn);
 }
 
 #ifdef WLAN_FEATURE_11AX_BSS_COLOR
@@ -996,7 +1001,6 @@ sch_beacon_process(struct mac_context *mac_ctx, uint8_t *rx_pkt_info,
 
 	sch_send_beacon_report(mac_ctx, &bcn, session);
 	__sch_beacon_process_for_session(mac_ctx, &bcn, rx_pkt_info, session);
-	lim_process_beacon_mlo(mac_ctx, session, &bcn);
 }
 
 /**