qcacld-3.0: Add populate and parse APIs for MLO IE for STA

Convert mlo probe response frame to struct for partner links.
Also, populate mlo IE for unicast probe request frame.

Change-Id: Ie4526b2a10e6642f2094ac12508df71fb9b787b6
CRs-Fixed: 2973865

core/mac/inc/sir_api.h

@@ -49,6 +49,9 @@ struct mac_context;
 #include "wlan_tdls_public_structs.h"
 #include "qca_vendor.h"
 #include "wlan_cp_stats_mc_defs.h"
+#ifdef WLAN_FEATURE_11BE_MLO
+#include "wlan_mlo_mgr_public_structs.h"
+#endif
 
 /* The ini gDataInactivityTimeout is deprecated. So, definng a new macro
  * PS_DATA_INACTIVITY_TIMEOUT with the ini's default value.
@@ -990,6 +993,9 @@ struct join_req {
 	tSirRSNie rsnIE;
 	tSirAddie addIEScan;
 	tSirAddie addIEAssoc;
+#ifdef WLAN_FEATURE_11BE_MLO
+	struct mlo_partner_info partner_info;
+#endif
 	/* Warning:::::::::::: Do not add any new param in this struct */
 	/* Pls make this as last variable in struct */
 	struct bss_description bssDescription;

core/mac/src/include/parser_api.h

@@ -684,7 +684,16 @@ sir_convert_qos_map_configure_frame2_struct(struct mac_context *mac,
 #ifdef WLAN_FEATURE_11BE_MLO
 QDF_STATUS
 mlo_ie_convert_assoc_rsp_frame2_struct(tDot11fAssocResponse *ar,
-				       tpSirMultiLink_IE pMloIe);
+				     tpSirMultiLink_IE pMloIe);
+
+QDF_STATUS
+populate_dot11f_probe_req_mlo_ie(struct mac_context *mac_ctx,
+				 struct pe_session *session,
+				 tDot11fIEmlo_ie *mlo_ie);
+
+QDF_STATUS
+sir_convert_mlo_probe_rsp_frame2_struct(tDot11fProbeResponse *pr,
+					tpSirMultiLink_IE mlo_ie_ptr);
 #else
 static inline QDF_STATUS
 mlo_ie_convert_assoc_rsp_frame2_struct(tDot11fAssocResponse *ar,
@@ -692,6 +701,21 @@ mlo_ie_convert_assoc_rsp_frame2_struct(tDot11fAssocResponse *ar,
 {
 	return QDF_STATUS_E_NOSUPPORT;
 }
+
+static inline QDF_STATUS
+populate_dot11f_probe_req_mlo_ie(struct mac_context *mac_ctx,
+				 struct pe_session *session,
+				 tDot11fIEmlo_ie *mlo_ie)
+{
+	return QDF_STATUS_E_NOSUPPORT;
+}
+
+static inline QDF_STATUS
+sir_convert_mlo_probe_rsp_frame2_struct(tDot11fProbeResponse *pr,
+					tpSirMultiLink_IE mlo_ie_ptr)
+{
+	return QDF_STATUS_E_NOSUPPORT;
+}
 #endif
 
 #ifdef ANI_SUPPORT_11H

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

@@ -3528,6 +3528,35 @@ static void lim_fill_crypto_params(struct mac_context *mac_ctx,
 	lim_update_fils_config(mac_ctx, session, req);
 }
 
+#ifdef WLAN_FEATURE_11BE_MLO
+static void lim_fill_ml_partner_info(struct cm_vdev_join_req *req,
+				     struct join_req *pe_join_req)
+{
+	uint8_t idx, num_links = 0;
+	struct mlo_partner_info *partner_info = NULL;
+
+	partner_info = &pe_join_req->partner_info;
+	if (!partner_info) {
+		pe_err("Partner link info not present");
+		return;
+	}
+	num_links = req->partner_info.num_partner_links;
+
+	for (idx = 0; idx < num_links; idx++) {
+		partner_info->partner_link_info[idx].link_id =
+			req->partner_info.partner_link_info[idx].link_id;
+		qdf_copy_macaddr(
+			&partner_info->partner_link_info[idx].link_addr,
+			&req->partner_info.partner_link_info[idx].link_addr);
+	}
+}
+#else
+static void lim_fill_ml_partner_info(struct cm_vdev_join_req *req,
+				     struct join_req *pe_join_req)
+{
+}
+#endif
+
 static QDF_STATUS
 lim_fill_session_params(struct mac_context *mac_ctx,
 			struct pe_session *session,
@@ -3608,6 +3637,8 @@ lim_fill_session_params(struct mac_context *mac_ctx,
 		}
 	}
 
+	lim_fill_ml_partner_info(req, pe_join_req);
+
 	return QDF_STATUS_SUCCESS;
 }
 

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

@@ -329,9 +329,11 @@ lim_send_probe_req_mgmt_frame(struct mac_context *mac_ctx,
 	populate_dot11f_he_6ghz_cap(mac_ctx, pesession,
 				    &pr->he_6ghz_band_cap);
 
-	if (IS_DOT11_MODE_EHT(dot11mode) && pesession)
+	if (IS_DOT11_MODE_EHT(dot11mode) && pesession) {
 		lim_update_session_eht_capable(mac_ctx, pesession);
-
+		populate_dot11f_probe_req_mlo_ie(mac_ctx, pesession,
+						 &pr->mlo_ie);
+	}
 	populate_dot11f_eht_caps(mac_ctx, pesession, &pr->eht_cap);
 
 	if (addn_ielen && additional_ie) {

core/mac/src/sys/legacy/src/utils/src/parser_api.c

@@ -2943,6 +2943,7 @@ QDF_STATUS sir_convert_probe_frame2_struct(struct mac_context *mac,
 	}
 
 	update_bss_color_change_ie_from_probe_rsp(pr, pProbeResp);
+	sir_convert_mlo_probe_rsp_frame2_struct(pr, pProbeResp->mlo_ie);
 
 	qdf_mem_free(pr);
 	return QDF_STATUS_SUCCESS;
@@ -7640,6 +7641,150 @@ void populate_dot11f_rnr_tbtt_info_10(struct mac_context *mac_ctx,
 	dot11f->tbtt_info.tbtt_info_10.bss_param_change_cnt =
 		rnr_session->mlo_link_info.link_ie.bss_param_change_cnt;
 }
+
+QDF_STATUS
+populate_dot11f_probe_req_mlo_ie(struct mac_context *mac_ctx,
+				 struct pe_session *session,
+				 tDot11fIEmlo_ie *mlo_ie)
+{
+	int link = 0, num_sta_pro = 0;
+	tDot11fIEsta_profile *sta_pro;
+	struct mlo_partner_info *link_info;
+	uint16_t vdev_count;
+	struct wlan_objmgr_vdev *wlan_vdev_list[WLAN_UMAC_MLO_MAX_VDEVS];
+	struct pe_session *link_session;
+	uint8_t *mld_addr;
+
+	mlo_ie->present = 1;
+	mlo_ie->type = 1;
+	mlo_ie->mld_mac_addr_present = 1;
+	mld_addr = wlan_vdev_mlme_get_mldaddr(session->vdev);
+	qdf_mem_copy(&mlo_ie->mld_mac_addr.info.mld_mac_addr, mld_addr,
+		     sizeof(mlo_ie->mld_mac_addr.info.mld_mac_addr));
+	mlo_ie->link_id_info_present = 1;
+
+
+	lim_get_mlo_vdev_list(session, &vdev_count, wlan_vdev_list);
+	link_session = pe_find_session_by_vdev_id(
+			mac_ctx, wlan_vdev_list[link]->vdev_objmgr.vdev_id);
+	sta_pro = &mlo_ie->sta_profile[num_sta_pro];
+	link_info = &link_session->lim_join_req->partner_info;
+	sta_pro->present = 1;
+	sta_pro->complete_profile = 1;
+	sta_pro->sta_mac_addr_present = 1;
+	qdf_mem_copy(&sta_pro->sta_mac_addr.info.sta_mac_addr,
+		     &link_info->partner_link_info[0].link_addr.bytes,
+		     QDF_MAC_ADDR_SIZE);
+	mlo_ie->link_id_info.info.link_id =
+			link_info->partner_link_info[0].link_id;
+
+	mlo_ie->num_sta_profile = num_sta_pro;
+
+	return QDF_STATUS_SUCCESS;
+}
+
+QDF_STATUS
+sir_convert_mlo_probe_rsp_frame2_struct(tDot11fProbeResponse *pr,
+					tpSirMultiLink_IE mlo_ie_ptr)
+{
+	tDot11fIEmlo_ie *mlo_ie;
+	tDot11fIEsta_profile *sta_prof, *pStaProf;
+
+	if (!pr)
+		return QDF_STATUS_E_NULL_VALUE;
+
+	if (!pr->mlo_ie.present) {
+		pe_err("MLO IE not present");
+		return QDF_STATUS_E_NULL_VALUE;
+	}
+
+	mlo_ie = qdf_mem_malloc(sizeof(*mlo_ie));
+	if (!mlo_ie)
+		return QDF_STATUS_E_FAILURE;
+
+	qdf_mem_zero((uint8_t *)mlo_ie_ptr, sizeof(tSirMultiLink_IE));
+
+	mlo_ie_ptr->mlo_ie.present = pr->mlo_ie.present;
+	mlo_ie_ptr->mlo_ie.mld_mac_addr_present =
+			pr->mlo_ie.mld_mac_addr_present;
+	qdf_mem_copy(&mlo_ie_ptr->mlo_ie.mld_mac_addr.info.mld_mac_addr,
+		     &pr->mlo_ie.mld_mac_addr.info.mld_mac_addr,
+		     QDF_MAC_ADDR_SIZE);
+
+	mlo_ie_ptr->mlo_ie.link_id_info_present =
+				pr->mlo_ie.link_id_info_present;
+	mlo_ie_ptr->mlo_ie.link_id_info.info.link_id =
+				pr->mlo_ie.link_id_info.info.link_id;
+
+	mlo_ie_ptr->mlo_ie.bss_param_change_cnt_present =
+			pr->mlo_ie.bss_param_change_cnt_present;
+	mlo_ie_ptr->mlo_ie.bss_param_change_cnt.info.bss_param_change_count =
+		pr->mlo_ie.bss_param_change_cnt.info.bss_param_change_count;
+
+	sta_prof = &pr->mlo_ie.sta_profile[0];
+	pStaProf = &mlo_ie_ptr->mlo_ie.sta_profile[0];
+
+	if (!sta_prof || !pStaProf)
+		return QDF_STATUS_E_NULL_VALUE;
+
+	if (sta_prof->sta_mac_addr_present)
+		qdf_mem_copy(&pStaProf->sta_mac_addr.info.sta_mac_addr,
+			     &sta_prof->sta_mac_addr.info.sta_mac_addr,
+			     QDF_MAC_ADDR_SIZE);
+
+	if (sta_prof->HTCaps.present)
+		qdf_mem_copy(&pStaProf->HTCaps, &sta_prof->HTCaps,
+			     sizeof(tDot11fIEHTCaps));
+
+	if (sta_prof->HTInfo.present)
+		qdf_mem_copy(&pStaProf->HTInfo, &sta_prof->HTInfo,
+			     sizeof(tDot11fIEHTInfo));
+
+	if (sta_prof->VHTCaps.present)
+		qdf_mem_copy(&pStaProf->VHTCaps, &sta_prof->VHTCaps,
+			     sizeof(tDot11fIEVHTCaps));
+
+	if (sta_prof->VHTOperation.present)
+		qdf_mem_copy(&pStaProf->VHTOperation, &sta_prof->VHTOperation,
+			     sizeof(tDot11fIEVHTOperation));
+
+	if (sta_prof->he_cap.present)
+		qdf_mem_copy(&pStaProf->he_cap, &sta_prof->he_cap,
+			     sizeof(tDot11fIEhe_cap));
+
+	if (sta_prof->he_op.present)
+		qdf_mem_copy(&pStaProf->he_op, &sta_prof->he_op,
+			     sizeof(tDot11fIEhe_op));
+
+	if (sta_prof->eht_cap.present)
+		qdf_mem_copy(&pStaProf->eht_cap, &sta_prof->eht_cap,
+			     sizeof(tDot11fIEeht_cap));
+
+	if (sta_prof->eht_op.present)
+		qdf_mem_copy(&pStaProf->eht_op, &sta_prof->eht_op,
+			     sizeof(tDot11fIEeht_op));
+
+	if (sta_prof->ChanSwitchAnn.present)
+		qdf_mem_copy(&pStaProf->ChanSwitchAnn,
+			     &sta_prof->ChanSwitchAnn,
+			     sizeof(tDot11fIEChanSwitchAnn));
+
+	if (sta_prof->ext_chan_switch_ann.present)
+		qdf_mem_copy(&pStaProf->ext_chan_switch_ann,
+			     &sta_prof->ext_chan_switch_ann,
+			     sizeof(tDot11fIEext_chan_switch_ann));
+
+	if (sta_prof->Quiet.present)
+		qdf_mem_copy(&pStaProf->Quiet, &sta_prof->Quiet,
+			     sizeof(tDot11fIEQuiet));
+
+	if (sta_prof->max_chan_switch_time.present)
+		qdf_mem_copy(&pStaProf->max_chan_switch_time,
+			     &sta_prof->max_chan_switch_time,
+			     sizeof(tDot11fIEmax_chan_switch_time));
+
+	return QDF_STATUS_SUCCESS;
+}
 #endif /* WLAN_FEATURE_11BE_MLO */
 
 #if defined(WLAN_FEATURE_11AX) && defined(WLAN_SUPPORT_TWT)