qcacld-3.0: Fill TDLS peer cap for add sta correctly

Fill TDLS 6Ghz peer capability correctly in Add STA
so that carrect Peer capability should be sent in
Peer Assoc.

Change-Id: I3eb6c387286b86e07c33be4751501b356b24ff68
CRs-Fixed: 3008407

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

@@ -2121,11 +2121,26 @@ static void lim_add_tdls_sta_he_config(tpAddStaParams add_sta_params,
 	qdf_mem_copy(&add_sta_params->he_config, &sta_ds->he_config,
 		     sizeof(add_sta_params->he_config));
 }
+
+static void lim_add_tdls_sta_6ghz_he_cap(struct mac_context *mac_ctx,
+					 tpAddStaParams add_sta_params,
+					 tpDphHashNode sta_ds)
+{
+	lim_update_he_6ghz_band_caps(mac_ctx, &sta_ds->he_6g_band_cap,
+				     add_sta_params);
+}
+
 #else
 static void lim_add_tdls_sta_he_config(tpAddStaParams add_sta_params,
 				       tpDphHashNode sta_ds)
 {
 }
+
+static void lim_add_tdls_sta_6ghz_he_cap(struct mac_context *mac_ctx,
+					 tpAddStaParams add_sta_params,
+					 tpDphHashNode sta_ds)
+{
+}
 #endif /* WLAN_FEATURE_11AX */
 #endif /* FEATURE_WLAN_TDLS */
 
@@ -2426,6 +2441,10 @@ lim_add_sta(struct mac_context *mac_ctx,
 			  add_sta_params->ht_caps,
 			  add_sta_params->vht_caps);
 		lim_add_tdls_sta_he_config(add_sta_params, sta_ds);
+
+		if (lim_is_he_6ghz_band(session_entry))
+			lim_add_tdls_sta_6ghz_he_cap(mac_ctx, add_sta_params,
+						     sta_ds);
 	}
 #endif