qcacld-3.0: Add api policy_mgr_get_connection_count_with_mlo()

Since for MLO there may be several link connections, for TDLS
using api policy_mgr_get_connection_count_with_mlo() to replace
with policy_mgr_get_connection_count() to follow the original
logic.

Change-Id: Ic13c89f2e834196c01ca6966329cbd0d1552f292
CRs-Fixed: 3436049

components/cmn_services/policy_mgr/inc/wlan_policy_mgr_api.h

@@ -626,6 +626,19 @@ void policy_mgr_clear_concurrency_mode(struct wlan_objmgr_psoc *psoc,
  */
 uint32_t policy_mgr_get_connection_count(struct wlan_objmgr_psoc *psoc);
 
+/**
+ * policy_mgr_get_connection_count_with_mlo() - provides the count of
+ * current connections
+ * @psoc: PSOC object information
+ *
+ * This function provides the count of current connections, MLD dev count
+ * 1 connection no matter how many links connection.
+ *
+ * Return: connection count
+ */
+uint32_t
+policy_mgr_get_connection_count_with_mlo(struct wlan_objmgr_psoc *psoc);
+
 /**
  * policy_mgr_get_concurrency_mode() - return concurrency mode
  * @psoc: PSOC object information

components/cmn_services/policy_mgr/src/wlan_policy_mgr_get_set_utils.c

@@ -3100,6 +3100,42 @@ uint32_t policy_mgr_get_connection_count(struct wlan_objmgr_psoc *psoc)
 	return count;
 }
 
+uint32_t
+policy_mgr_get_connection_count_with_mlo(struct wlan_objmgr_psoc *psoc)
+{
+	uint32_t conn_index, count = 0;
+	struct policy_mgr_psoc_priv_obj *pm_ctx;
+	enum policy_mgr_con_mode mode;
+	bool is_mlo = false, count_mlo = false;
+
+	pm_ctx = policy_mgr_get_context(psoc);
+	if (!pm_ctx) {
+		policy_mgr_err("Invalid Context");
+		return count;
+	}
+
+	qdf_mutex_acquire(&pm_ctx->qdf_conc_list_lock);
+	for (conn_index = 0; conn_index < MAX_NUMBER_OF_CONC_CONNECTIONS;
+		conn_index++) {
+		if (pm_conc_connection_list[conn_index].in_use) {
+			is_mlo = policy_mgr_is_ml_vdev_id(psoc,
+				   pm_conc_connection_list[conn_index].vdev_id);
+			mode = pm_conc_connection_list[conn_index].mode;
+			if (is_mlo && (mode == PM_STA_MODE)) {
+				if (!count_mlo) {
+					count_mlo = true;
+					count++;
+				}
+			} else {
+				count++;
+			}
+		}
+	}
+	qdf_mutex_release(&pm_ctx->qdf_conc_list_lock);
+
+	return count;
+}
+
 uint32_t policy_mgr_mode_specific_vdev_id(struct wlan_objmgr_psoc *psoc,
 					  enum policy_mgr_con_mode mode)
 {

components/tdls/core/src/wlan_tdls_main.c

@@ -1003,7 +1003,8 @@ bool tdls_check_is_tdls_allowed(struct wlan_objmgr_vdev *vdev)
 		goto exit;
 	}
 
-	connection_count = policy_mgr_get_connection_count(tdls_soc_obj->soc);
+	connection_count =
+		policy_mgr_get_connection_count_with_mlo(tdls_soc_obj->soc);
 	sta_count =
 		policy_mgr_mode_specific_connection_count(tdls_soc_obj->soc,
 							  PM_STA_MODE, NULL);
@@ -1040,7 +1041,7 @@ bool tdls_is_concurrency_allowed(struct wlan_objmgr_psoc *psoc)
 					   WLAN_TDLS_CONCURRENCIES_SUPPORT))
 		return false;
 
-	if (policy_mgr_get_connection_count(psoc) >
+	if (policy_mgr_get_connection_count_with_mlo(psoc) >
 	    WLAN_TDLS_MAX_CONCURRENT_VDEV_SUPPORTED)
 		return false;
 
@@ -1058,7 +1059,7 @@ bool tdls_is_concurrency_allowed(struct wlan_objmgr_psoc *psoc)
 	/*
 	 * Don't enable TDLS for P2P_CLI in concurrency cases
 	 */
-	if (policy_mgr_get_connection_count(psoc) > 1 &&
+	if (policy_mgr_get_connection_count_with_mlo(psoc) > 1 &&
 	    !policy_mgr_mode_specific_connection_count(psoc, PM_STA_MODE,
 						       NULL))
 		return false;
@@ -1101,7 +1102,8 @@ void tdls_set_ct_mode(struct wlan_objmgr_psoc *psoc,
 							  PM_P2P_CLIENT_MODE,
 							  NULL);
 	if (sta_count == 1 ||
-	    (policy_mgr_get_connection_count(psoc) == 1 && p2p_count == 1)) {
+	    (policy_mgr_get_connection_count_with_mlo(psoc) == 1 &&
+	     p2p_count == 1)) {
 		state = true;
 		/*
 		 * In case of TDLS external control, peer should be added
@@ -1270,7 +1272,7 @@ struct wlan_objmgr_vdev *tdls_get_vdev(struct wlan_objmgr_psoc *psoc,
 {
 	uint32_t vdev_id;
 
-	if (policy_mgr_get_connection_count(psoc) > 1 &&
+	if (policy_mgr_get_connection_count_with_mlo(psoc) > 1 &&
 	    !tdls_is_concurrency_allowed(psoc))
 		return NULL;
 
@@ -1285,7 +1287,7 @@ struct wlan_objmgr_vdev *tdls_get_vdev(struct wlan_objmgr_psoc *psoc,
 	 */
 	vdev_id = policy_mgr_mode_specific_vdev_id(psoc, PM_P2P_CLIENT_MODE);
 	if (WLAN_INVALID_VDEV_ID != vdev_id &&
-	    policy_mgr_get_connection_count(psoc) == 1)
+	    policy_mgr_get_connection_count_with_mlo(psoc) == 1)
 		return wlan_objmgr_get_vdev_by_id_from_psoc(psoc, vdev_id,
 							    dbg_id);