qcacld-3.0: Populate HE 160 MCS map for session ch bw 320

Populate HE 160 MCS rates even when session bandwidth is
320 MHz and peer supports 160 MHz BW.
Currently, HE 160 MCS map is populated only when session
bandwidth is 160 MHz, else it is reset. In the case of
11be SAP with 320 MHz support, HE 160 tx/rx MCS rates
must be updated with the intersection of self and peer
MCS rates, instead of HE_MCS_ALL_DISABLED.

Change-Id: I93d1d476c33aa98e96e76c2668f18456a9a874b9
CRs-Fixed: 3264361

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

@@ -5768,6 +5768,11 @@ static bool is_dot11mode_support_eht_cap(enum csr_cfgdot11mode dot11mode)
 
 	return false;
 }
+
+bool lim_is_session_chwidth_320mhz(struct pe_session *session)
+{
+	return session->ch_width == CH_WIDTH_320MHZ;
+}
 #else
 static bool is_dot11mode_support_eht_cap(enum csr_cfgdot11mode dot11mode)
 {
@@ -7922,6 +7927,7 @@ QDF_STATUS lim_populate_he_mcs_set(struct mac_context *mac_ctx,
 		return QDF_STATUS_E_FAILURE;
 	}
 
+	pe_debug("session chan width: %d", session_entry->ch_width);
 	pe_debug("PEER: lt 80: rx 0x%04x tx 0x%04x, 160: rx 0x%04x tx 0x%04x, 80+80: rx 0x%04x tx 0x%04x",
 		peer_he_caps->rx_he_mcs_map_lt_80,
 		peer_he_caps->tx_he_mcs_map_lt_80,
@@ -7960,7 +7966,9 @@ QDF_STATUS lim_populate_he_mcs_set(struct mac_context *mac_ctx,
 			mac_ctx->he_cap_5g.rx_he_mcs_map_lt_80,
 			mac_ctx->he_cap_5g.tx_he_mcs_map_lt_80);
 
-	if (session_entry->ch_width == CH_WIDTH_160MHZ) {
+	if ((session_entry->ch_width == CH_WIDTH_160MHZ ||
+	     lim_is_session_chwidth_320mhz(session_entry)) &&
+	     peer_he_caps->chan_width_2) {
 		lim_populate_he_mcs_per_bw(
 			mac_ctx, &rates->rx_he_mcs_map_160,
 			&rates->tx_he_mcs_map_160,

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

@@ -2050,6 +2050,15 @@ void lim_update_stads_eht_caps(struct mac_context *mac_ctx,
 void lim_update_stads_eht_bw_320mhz(struct pe_session *session,
 				    tpDphHashNode sta_ds);
 
+/**
+ * lim_is_session_chwidth_320mhz() - Check if session chan width is 320 MHz
+ * @session: pointer to PE session
+ *
+ * Check if session channel width is 320 MHz
+ *
+ * Return: bool
+ */
+bool lim_is_session_chwidth_320mhz(struct pe_session *session);
 #else
 static inline bool lim_is_session_eht_capable(struct pe_session *session)
 {
@@ -2220,6 +2229,12 @@ lim_update_stads_eht_bw_320mhz(struct pe_session *session,
 			       tpDphHashNode sta_ds)
 {
 }
+
+static inline bool
+lim_is_session_chwidth_320mhz(struct pe_session *session)
+{
+	return false;
+}
 #endif /* WLAN_FEATURE_11BE */
 
 #ifdef WLAN_FEATURE_11BE_MLO