qcacld-3.0: Add MLO support in Auth frame for ML connection

Add MLO support in Authentication frames for ML STA and SAP,
as per Draft P802.11be_D1.3

Change-Id: Ic2c5814b7f8bb1938d9e8bb6681d92085f7f55ef
CRs-Fixed: 3136984

core/mac/inc/sir_mac_prot_def.h

@@ -1263,6 +1263,7 @@ typedef struct sSirMacAuthFrameBody {
 	uint8_t wrapped_data[SIR_FILS_WRAPPED_DATA_MAX_SIZE];
 	uint8_t nonce[SIR_FILS_NONCE_LENGTH];
 #endif
+	bool is_mlo_ie_present;
 } qdf_packed tSirMacAuthFrameBody, *tpSirMacAuthFrameBody;
 
 /* / Common header for all action frames */

core/mac/src/include/parser_api.h

@@ -1,6 +1,6 @@
 /*
  * Copyright (c) 2012-2021 The Linux Foundation. All rights reserved.
- * Copyright (c) 2021 Qualcomm Innovation Center, Inc. All rights reserved.
+ * Copyright (c) 2021-2022 Qualcomm Innovation Center, Inc. All rights reserved.
  *
  * Permission to use, copy, modify, and/or distribute this software for
  * any purpose with or without fee is hereby granted, provided that the
@@ -1476,6 +1476,20 @@ populate_dot11f_eht_operation(struct mac_context *mac_ctx,
 #endif
 
 #ifdef WLAN_FEATURE_11BE_MLO
+/**
+ * populate_dot11f_auth_mlo_ie() - populate MLO IE in Auth frame
+ * @mac_ctx: Global MAC context
+ * @pe_session: PE session
+ * @mlo_ie: pointer to MLO IE struct
+ *
+ * Return: Success if MLO IE is populated in Auth frame, else Failure
+ *
+ * Populate the MLO IE in Auth frame based on the session.
+ */
+QDF_STATUS populate_dot11f_auth_mlo_ie(struct mac_context *mac_ctx,
+				       struct pe_session *pe_session,
+				       tDot11fIEmlo_ie *mlo_ie);
+
 /**
  * populate_dot11f_assoc_req_mlo_ie() - populate MLO Operation IE
  in assoc req
@@ -1487,10 +1501,18 @@ populate_dot11f_eht_operation(struct mac_context *mac_ctx,
  */
 QDF_STATUS
 populate_dot11f_assoc_req_mlo_ie(struct mac_context *mac_ctx,
-					     struct pe_session *pe_session,
-					     tDot11fAssocRequest *frm);
+				 struct pe_session *pe_session,
+				 tDot11fAssocRequest *frm);
 
 #else
+static inline
+QDF_STATUS populate_dot11f_auth_mlo_ie(struct mac_context *mac_ctx,
+				       struct pe_session *pe_session,
+				       tDot11fIEmlo_ie *mlo_ie)
+{
+	return QDF_STATUS_E_NOSUPPORT;
+}
+
 static inline QDF_STATUS
 populate_dot11f_assoc_req_mlo_ie(struct mac_context *mac_ctx,
 					     struct pe_session *pe_session,

core/mac/src/pe/include/lim_global.h

@@ -1,5 +1,6 @@
 /*
  * Copyright (c) 2011-2021 The Linux Foundation. All rights reserved.
+ * Copyright (c) 2022 Qualcomm Innovation Center, Inc. All rights reserved.
  *
  * Permission to use, copy, modify, and/or distribute this software for
  * any purpose with or without fee is hereby granted, provided that the
@@ -217,6 +218,7 @@ typedef struct tLimPreAuthNode {
 	 * to defer the association request processing
 	 */
 	struct lim_assoc_data assoc_req;
+	bool is_mlo_ie_present;
 } tLimPreAuthNode, *tpLimPreAuthNode;
 
 /* Pre-authentication table definition */

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

@@ -289,6 +289,14 @@ static void lim_process_auth_open_system_algo(struct mac_context *mac_ctx,
 	auth_node->seq_num = ((mac_hdr->seqControl.seqNumHi << 4) |
 				(mac_hdr->seqControl.seqNumLo));
 	auth_node->timestamp = qdf_mc_timer_get_system_ticks();
+
+	/* Check for MLO IE in Auth request in case of MLO connection */
+	if (wlan_vdev_mlme_is_mlo_ap(pe_session->vdev) &&
+	    rx_auth_frm_body->is_mlo_ie_present) {
+		auth_node->is_mlo_ie_present = true;
+		pe_debug("MLO IE is present in auth req");
+	}
+
 	lim_add_pre_auth_node(mac_ctx, auth_node);
 	/*
 	 * Send Authenticaton frame with Success
@@ -1612,6 +1620,17 @@ lim_process_auth_frame(struct mac_context *mac_ctx, uint8_t *rx_pkt_info,
 		goto free;
 	}
 
+	if (LIM_IS_STA_ROLE(pe_session) &&
+	    wlan_vdev_mlme_is_mlo_vdev(pe_session->vdev)) {
+		if (!rx_auth_frm_body->is_mlo_ie_present) {
+			pe_debug("MLO IE not present in auth frame from peer");
+			lim_send_deauth_mgmt_frame(
+				mac_ctx, REASON_UNSPEC_FAILURE,
+				pe_session->bssId, pe_session, false);
+			goto free;
+		}
+	}
+
 	/*
 	 * IOT Workaround: with invalid WEP key, some APs reply
 	 * AuthFrame 4 with invalid seqNumber. This AuthFrame

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

@@ -2924,6 +2924,59 @@ static QDF_STATUS lim_auth_tx_complete_cnf(void *context,
 	return QDF_STATUS_SUCCESS;
 }
 
+#ifdef WLAN_FEATURE_11BE_MLO
+static uint32_t lim_populate_auth_mlo_ie(struct mac_context *mac_ctx,
+					 struct pe_session *session,
+					 tSirMacAddr peer_addr,
+					 uint8_t *mlo_ie_buf)
+{
+	tDot11fIEmlo_ie mlo_ie;
+	uint32_t mlo_ie_len = 0;
+	struct tLimPreAuthNode *auth_node;
+
+	if (wlan_vdev_mlme_is_mlo_vdev(session->vdev)) {
+		qdf_mem_zero(&mlo_ie, sizeof(tDot11fIEmlo_ie));
+
+		pe_debug("Auth TX sys role: %d", GET_LIM_SYSTEM_ROLE(session));
+		if (LIM_IS_STA_ROLE(session)) {
+			populate_dot11f_auth_mlo_ie(mac_ctx, session, &mlo_ie);
+			dot11f_pack_ie_mlo_ie(mac_ctx, &mlo_ie, mlo_ie_buf,
+					      DOT11F_EID_MLO_IE, &mlo_ie_len);
+		} else if (LIM_IS_AP_ROLE(session)) {
+			auth_node = lim_search_pre_auth_list(mac_ctx, peer_addr);
+			if (!auth_node) {
+				pe_err("no pre-auth ctx with peer addr");
+				return 0;
+			}
+
+			/*
+			 * Populate MLO IE in Auth response frame if
+			 * is_mlo_ie_present flag is set. It is set when peer
+			 * is MLO AP
+			 */
+			if (auth_node && auth_node->is_mlo_ie_present) {
+				populate_dot11f_auth_mlo_ie(mac_ctx, session,
+							    &mlo_ie);
+				dot11f_pack_ie_mlo_ie(
+					mac_ctx, &mlo_ie, mlo_ie_buf,
+					DOT11F_EID_MLO_IE, &mlo_ie_len);
+			}
+		}
+	}
+
+	return mlo_ie_len;
+}
+#else
+static inline
+uint32_t lim_populate_auth_mlo_ie(struct mac_context *mac_ctx,
+				  struct pe_session *session,
+				  tSirMacAddr peer_addr,
+				  uint8_t *mlo_ie_buf)
+{
+	return 0;
+}
+#endif
+
 /**
  * lim_send_auth_mgmt_frame() - Send an Authentication frame
  *
@@ -2958,6 +3011,8 @@ lim_send_auth_mgmt_frame(struct mac_context *mac_ctx,
 	enum rateid min_rid = RATEID_DEFAULT;
 	uint16_t ch_freq_tx_frame = 0;
 	int8_t peer_rssi = 0;
+	uint8_t mlo_ie_buf[DOT11F_EID_MLO_IE];
+	uint32_t mlo_ie_len = 0;
 
 	if (!session) {
 		pe_err("Error: psession Entry is NULL");
@@ -3086,6 +3141,12 @@ lim_send_auth_mgmt_frame(struct mac_context *mac_ctx,
 		return;
 	} /* switch (auth_frame->authTransactionSeqNumber) */
 
+	qdf_mem_zero(&mlo_ie_buf, sizeof(mlo_ie_buf));
+
+	mlo_ie_len =  lim_populate_auth_mlo_ie(mac_ctx, session, peer_addr,
+					       mlo_ie_buf);
+	frame_len += mlo_ie_len;
+
 alloc_packet:
 	qdf_status = cds_packet_alloc((uint16_t) frame_len, (void **)&frame,
 				 (void **)&packet);
@@ -3200,6 +3261,9 @@ alloc_packet:
 		}
 	}
 
+	if (mlo_ie_len)
+		qdf_mem_copy(body, &mlo_ie_buf, mlo_ie_len);
+
 	pe_nofl_info("Auth TX: vdev %d seq %d seq num %d status %d WEP %d to " QDF_MAC_ADDR_FMT,
 		     vdev_id, auth_frame->authTransactionSeqNumber,
 		     mac_ctx->mgmtSeqNum, auth_frame->authStatusCode,

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

@@ -5135,6 +5135,10 @@ sir_convert_auth_frame2_struct(struct mac_context *mac,
 		qdf_mem_copy(pAuth->challengeText, auth.ChallengeText.text,
 			     auth.ChallengeText.num_text);
 	}
+
+	/* Copy MLO IE presence flag to pAuth in case of ML connection */
+	pAuth->is_mlo_ie_present = auth.mlo_ie.present;
+
 	sir_update_auth_frame2_struct_fils_conf(&auth, pAuth);
 
 	return QDF_STATUS_SUCCESS;
@@ -8567,9 +8571,35 @@ populate_dot11f_mlo_partner_sta_cap(struct mac_context *mac,
 	return QDF_STATUS_SUCCESS;
 }
 
+QDF_STATUS populate_dot11f_auth_mlo_ie(struct mac_context *mac_ctx,
+				       struct pe_session *pe_session,
+				       tDot11fIEmlo_ie *mlo_ie)
+{
+	struct qdf_mac_addr *mld_addr;
+
+	pe_debug("Populate Auth MLO IEs");
+
+	mlo_ie->present = 1;
+	mlo_ie->mld_mac_addr_present = 1;
+	mlo_ie->type = 0;
+
+	mld_addr = (struct qdf_mac_addr *)wlan_vdev_mlme_get_mldaddr(pe_session->vdev);
+	qdf_mem_copy(&mlo_ie->mld_mac_addr, mld_addr, QDF_MAC_ADDR_SIZE);
+
+	pe_debug("MLD mac addr: " QDF_MAC_ADDR_FMT, mld_addr);
+
+	mlo_ie->link_id_info_present = 0;
+	mlo_ie->bss_param_change_cnt_present = 0;
+	mlo_ie->medium_sync_delay_info_present = 0;
+	mlo_ie->eml_capab_present = 0;
+	mlo_ie->mld_capab_present = 0;
+
+	return QDF_STATUS_SUCCESS;
+}
+
 QDF_STATUS populate_dot11f_assoc_req_mlo_ie(struct mac_context *mac_ctx,
-					     struct pe_session *pe_session,
-					     tDot11fAssocRequest *frm)
+					    struct pe_session *pe_session,
+					    tDot11fAssocRequest *frm)
 {
 	uint8_t link;
 	uint8_t num_sta_prof = 0, total_sta_prof;