qcacld-3.0: Configure power save on specific MLO links

Add support to configure Power save enable/disable on
specific MLO links based on the link ids.

Change-Id: I63bc332d2afb7e63417b8191dd463f9caa659270
CRs-Fixed: 3616677

core/hdd/inc/wlan_hdd_power.h

@@ -683,4 +683,17 @@ void hdd_enable_icmp_offload(struct hdd_adapter *adapter,
 {}
 #endif /* FEATURE_ICMP_OFFLOAD */
 
+#if defined(WLAN_FEATURE_11BE_MLO) && defined(CFG80211_11BE_BASIC)
+int wlan_hdd_set_mlo_ps(struct hdd_adapter *adapter,
+			bool allow_power_save, int timeout,
+			int link_id);
+#else
+static inline
+int wlan_hdd_set_mlo_ps(struct hdd_adapter *adapter,
+			bool allow_power_save, int timeout,
+			int link_id)
+{
+        return 0;
+}
+#endif
 #endif /* __WLAN_HDD_POWER_H */

core/hdd/src/wlan_hdd_cfg80211.c

@@ -8660,6 +8660,8 @@ wlan_hdd_wifi_test_config_policy[
 			.type = NLA_U8},
 		[QCA_WLAN_VENDOR_ATTR_WIFI_TEST_CONFIG_EHT_MLO_STR_TX] = {
 			.type = NLA_U8},
+		[QCA_WLAN_VENDOR_ATTR_WIFI_TEST_CONFIG_EHT_MLO_LINK_POWER_SAVE] = {
+			.type = NLA_NESTED},
 		[QCA_WLAN_VENDOR_ATTR_WIFI_TEST_CONFIG_MLD_ID_ML_PROBE_REQ] = {
 			.type = NLA_U8},
 };
@@ -14471,6 +14473,27 @@ __wlan_hdd_cfg80211_set_wifi_test_config(struct wiphy *wiphy,
 			hdd_err("Failed to send STR TX indication");
 	}
 
+	cmd_id = QCA_WLAN_VENDOR_ATTR_WIFI_TEST_CONFIG_EHT_MLO_LINK_POWER_SAVE;
+	if (tb[cmd_id]) {
+		struct nlattr *curr_attr;
+		int len;
+		uint8_t link_id, timeout = 0, num_links = 0;
+		bool allow_ps = true;
+
+		nla_for_each_nested(curr_attr, tb[cmd_id], len) {
+			if (nla_len(curr_attr) != sizeof(uint8_t)) {
+				hdd_err("len is not correct for idx %d",
+					num_links);
+				goto send_err;
+			}
+			link_id = nla_get_u8(curr_attr);
+			hdd_debug("link id[%d]: %d", num_links, link_id);
+			wlan_hdd_set_mlo_ps(adapter, allow_ps, timeout,
+					    link_id);
+			num_links++;
+		}
+	}
+
 	cmd_id = QCA_WLAN_VENDOR_ATTR_WIFI_TEST_CONFIG_MLD_ID_ML_PROBE_REQ;
 	if (tb[cmd_id]) {
 		cfg_val = nla_get_u8(tb[cmd_id]);

core/hdd/src/wlan_hdd_power.c

@@ -2872,13 +2872,18 @@ static int wlan_hdd_set_ps(struct wlan_hdd_link_info *link_info,
 
 #if defined(WLAN_FEATURE_11BE_MLO) && defined(CFG80211_11BE_BASIC)
 #ifdef WLAN_HDD_MULTI_VDEV_SINGLE_NDEV
-static int wlan_hdd_set_mlo_ps(struct hdd_adapter *adapter,
-			       bool allow_power_save, int timeout)
+int wlan_hdd_set_mlo_ps(struct hdd_adapter *adapter,
+			bool allow_power_save, int timeout,
+			int link_id)
 {
 	struct wlan_hdd_link_info *link_info;
 	int status = -EINVAL;
 
 	hdd_adapter_for_each_active_link_info(adapter, link_info) {
+		if (link_id >= 0 &&
+		    wlan_vdev_get_link_id(link_info->vdev) != link_id)
+			continue;
+
 		status = wlan_hdd_set_ps(link_info,
 					 link_info->link_addr.bytes,
 					 allow_power_save, timeout);
@@ -2889,8 +2894,9 @@ static int wlan_hdd_set_mlo_ps(struct hdd_adapter *adapter,
 	return status;
 }
 #else
-static int wlan_hdd_set_mlo_ps(struct hdd_adapter *adapter,
-			       bool allow_power_save, int timeout)
+int wlan_hdd_set_mlo_ps(struct hdd_adapter *adapter,
+			bool allow_power_save, int timeout,
+			int link_id)
 {
 	struct hdd_adapter *link_adapter;
 	struct hdd_mlo_adapter_info *mlo_adapter_info;
@@ -2901,6 +2907,12 @@ static int wlan_hdd_set_mlo_ps(struct hdd_adapter *adapter,
 		link_adapter = mlo_adapter_info->link_adapter[i];
 		if (!link_adapter)
 			continue;
+
+		if (link_id >= 0 &&
+		    wlan_vdev_get_link_id(link_adapter->deflink->vdev) !=
+		    link_id)
+			continue;
+
 		status = wlan_hdd_set_ps(link_adapter->deflink,
 					 link_adapter->mac_addr.bytes,
 					 allow_power_save, timeout);
@@ -2908,15 +2920,12 @@ static int wlan_hdd_set_mlo_ps(struct hdd_adapter *adapter,
 			break;
 	}
 
+	if (i == WLAN_MAX_MLD && link_id >= 0)
+		hdd_err("No link adapter found for link id: %d", link_id);
+
 	return status;
 }
 #endif
-#else
-static int wlan_hdd_set_mlo_ps(struct hdd_adapter *adapter,
-			       bool allow_power_save, int timeout)
-{
-	return 0;
-}
 #endif
 
 /**
@@ -2978,7 +2987,7 @@ static int __wlan_hdd_cfg80211_set_power_mgmt(struct wiphy *wiphy,
 
 	if (hdd_adapter_is_ml_adapter(adapter)) {
 		status = wlan_hdd_set_mlo_ps(adapter, allow_power_save,
-					     timeout);
+					     timeout, -1);
 		goto exit;
 	}