qcacld-3.0: Support AP mode to get peer station info

AP mode, function __wlan_hdd_cfg80211_dump_station() only get AP's
statistics. Add code to support get peer station statistics.

Change-Id: Ie821484e8a21116f4a4b6402ea2fcd85453e4cdf
CRs-Fixed: 3088695

core/hdd/src/wlan_hdd_sta_info.c

@@ -1,5 +1,6 @@
 /*
  * Copyright (c) 2019-2021 The Linux Foundation. All rights reserved.
+ * Copyright (c) 2021 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
@@ -42,7 +43,7 @@ char *sta_info_string_from_dbgid(wlan_sta_info_dbgid id)
 				"STA_INFO_SOFTAP_REGISTER_STA",
 				"STA_INFO_GET_CACHED_STATION_REMOTE",
 				"STA_INFO_HDD_GET_STATION_REMOTE",
-				"STA_INFO_WLAN_HDD_GET_STATION_REMOTE",
+				"STA_INFO_WLAN_HDD_CFG80211_GET_STATION",
 				"STA_INFO_SOFTAP_DEAUTH_CURRENT_STA",
 				"STA_INFO_SOFTAP_DEAUTH_ALL_STA",
 				"STA_INFO_CFG80211_DEL_STATION",
@@ -61,6 +62,8 @@ char *sta_info_string_from_dbgid(wlan_sta_info_dbgid id)
 				"STA_INFO_CLEAR_CACHED_STA_INFO",
 				"STA_INFO_ATTACH_DETACH",
 				"STA_INFO_SHOW",
+				"STA_INFO_SOFTAP_IPA_RX_PKT_CALLBACK",
+				"STA_INFO_WLAN_HDD_CFG80211_DUMP_STATION",
 				"STA_INFO_ID_MAX"};
 	int32_t num_dbg_strings = QDF_ARRAY_SIZE(strings);
 
@@ -147,6 +150,38 @@ void hdd_sta_info_detach(struct hdd_sta_info_obj *sta_info_container,
 	qdf_spin_unlock_bh(&sta_info_container->sta_obj_lock);
 }
 
+struct hdd_station_info *hdd_get_sta_info_by_id(
+				struct hdd_sta_info_obj *sta_info_container,
+				const int idx,
+				wlan_sta_info_dbgid sta_info_dbgid)
+{
+	struct hdd_station_info *sta_info = NULL;
+	int i = 0;
+
+	if (!sta_info_container) {
+		hdd_err("Parameter(s) null");
+		return NULL;
+	}
+
+	qdf_spin_lock_bh(&sta_info_container->sta_obj_lock);
+
+	qdf_list_for_each(&sta_info_container->sta_obj, sta_info, sta_node) {
+		if (qdf_is_macaddr_broadcast(&sta_info->sta_mac))
+			continue;
+		if (i == idx) {
+			hdd_take_sta_info_ref(sta_info_container,
+					      sta_info, false, sta_info_dbgid);
+			qdf_spin_unlock_bh(&sta_info_container->sta_obj_lock);
+			return sta_info;
+		}
+		i++;
+	}
+
+	qdf_spin_unlock_bh(&sta_info_container->sta_obj_lock);
+
+	return NULL;
+}
+
 struct hdd_station_info *hdd_get_sta_info_by_mac(
 				struct hdd_sta_info_obj *sta_info_container,
 				const uint8_t *mac_addr,

core/hdd/src/wlan_hdd_sta_info.h

@@ -86,8 +86,8 @@ enum dhcp_nego_status {
  * @STA_INFO_SOFTAP_REGISTER_STA:       Register a SoftAP STA
  * @STA_INFO_GET_CACHED_STATION_REMOTE: Get cached peer's info
  * @STA_INFO_HDD_GET_STATION_REMOTE:    Get remote peer's info
- * @STA_INFO_WLAN_HDD_GET_STATION_REMOTE: NL80211_CMD_GET_STATION handler for
- *                                        SoftAP
+ * @STA_INFO_WLAN_HDD_CFG80211_GET_STATION: NL80211_CMD_GET_STATION handler for
+ *                                          SoftAP
  * @STA_INFO_SOFTAP_DEAUTH_CURRENT_STA: Deauth current sta
  * @STA_INFO_SOFTAP_DEAUTH_ALL_STA:     Deauth all sta in the sta list
  * @STA_INFO_CFG80211_DEL_STATION:      CFG80211 del station handler
@@ -109,6 +109,8 @@ enum dhcp_nego_status {
  * @STA_INFO_ATTACH_DETACH:             Station info attach/detach
  * @STA_INFO_SHOW:     Station info show
  * @STA_INFO_SOFTAP_IPA_RX_PKT_CALLBACK: Update rx mcbc stats for IPA case
+ * @STA_INFO_WLAN_HDD_CFG80211_DUMP_STATION: NL80211_CMD_GET_STATION dumpit
+ *                                           handler for SoftAP
  *
  */
 /*
@@ -127,7 +129,7 @@ typedef enum {
 	STA_INFO_SOFTAP_REGISTER_STA = 8,
 	STA_INFO_GET_CACHED_STATION_REMOTE = 9,
 	STA_INFO_HDD_GET_STATION_REMOTE = 10,
-	STA_INFO_WLAN_HDD_GET_STATION_REMOTE = 11,
+	STA_INFO_WLAN_HDD_CFG80211_GET_STATION = 11,
 	STA_INFO_SOFTAP_DEAUTH_CURRENT_STA = 12,
 	STA_INFO_SOFTAP_DEAUTH_ALL_STA = 13,
 	STA_INFO_CFG80211_DEL_STATION = 14,
@@ -147,6 +149,7 @@ typedef enum {
 	STA_INFO_ATTACH_DETACH = 28,
 	STA_INFO_SHOW = 29,
 	STA_INFO_SOFTAP_IPA_RX_PKT_CALLBACK = 30,
+	STA_INFO_WLAN_HDD_CFG80211_DUMP_STATION = 31,
 
 	STA_INFO_ID_MAX,
 } wlan_sta_info_dbgid;
@@ -565,6 +568,21 @@ void hdd_sta_info_detach(struct hdd_sta_info_obj *sta_info_container,
 QDF_STATUS hdd_sta_info_attach(struct hdd_sta_info_obj *sta_info_container,
 			       struct hdd_station_info *sta_info);
 
+/**
+ * hdd_get_sta_info_by_id() - Find the sta_info structure by index
+ * @sta_info_container: The station info container obj that stores and maintains
+ *                      the sta_info obj.
+ * @idx: The index which the sta_info has to be fetched.
+ * @sta_info_dbgid: Debug ID of the caller API
+ *
+ * Return: Reference-counted Pointer to the hdd_station_info structure which
+ *         contains the mac address passed
+ */
+struct hdd_station_info *hdd_get_sta_info_by_id(
+				struct hdd_sta_info_obj *sta_info_container,
+				const int idx,
+				wlan_sta_info_dbgid sta_info_dbgid);
+
 /**
  * hdd_get_sta_info_by_mac() - Find the sta_info structure by mac addr
  * @sta_info_container: The station info container obj that stores and maintains

core/hdd/src/wlan_hdd_stats.c

@@ -5191,7 +5191,7 @@ static void wlan_hdd_fill_rate_info(struct hdd_fw_txrx_stats *txrx_stats,
  * wlan_hdd_get_station_remote() - NL80211_CMD_GET_STATION handler for SoftAP
  * @wiphy: pointer to wiphy
  * @dev: pointer to net_device structure
- * @mac: request peer mac address
+ * @stainfo: request peer station info
  * @sinfo: pointer to station_info struct
  *
  * This function will get remote peer info from fw and fill sinfo struct
@@ -5200,36 +5200,19 @@ static void wlan_hdd_fill_rate_info(struct hdd_fw_txrx_stats *txrx_stats,
  */
 static int wlan_hdd_get_station_remote(struct wiphy *wiphy,
 				       struct net_device *dev,
-				       const u8 *mac,
+				       struct hdd_station_info *stainfo,
 				       struct station_info *sinfo)
 {
 	struct hdd_adapter *adapter = WLAN_HDD_GET_PRIV_PTR(dev);
 	struct hdd_context *hddctx = wiphy_priv(wiphy);
-	struct hdd_station_info *stainfo = NULL;
 	struct stats_event *stats;
 	struct hdd_fw_txrx_stats txrx_stats;
 	int i, status;
 
-	status = wlan_hdd_validate_context(hddctx);
-	if (status != 0)
-		return status;
-
-	hdd_debug("Peer "QDF_MAC_ADDR_FMT, QDF_MAC_ADDR_REF(mac));
-
-	stainfo = hdd_get_sta_info_by_mac(&adapter->sta_info_list, mac,
-					  STA_INFO_WLAN_HDD_GET_STATION_REMOTE);
-	if (!stainfo) {
-		hdd_err("peer "QDF_MAC_ADDR_FMT" not found",
-			QDF_MAC_ADDR_REF(mac));
-		return -EINVAL;
-	}
-
 	stats = wlan_cfg80211_mc_cp_stats_get_peer_stats(adapter->vdev,
-							 mac, &status);
+					stainfo->sta_mac.bytes, &status);
 	if (status || !stats) {
 		wlan_cfg80211_mc_cp_stats_free_stats_event(stats);
-		hdd_put_sta_info_ref(&adapter->sta_info_list, &stainfo, true,
-				     STA_INFO_WLAN_HDD_GET_STATION_REMOTE);
 		hdd_err("fail to get peer info from fw");
 		return -EPERM;
 	}
@@ -5253,8 +5236,6 @@ static int wlan_hdd_get_station_remote(struct wiphy *wiphy,
 	wlan_hdd_fill_station_info(hddctx->psoc, adapter,
 				   sinfo, stainfo, &txrx_stats);
 	wlan_cfg80211_mc_cp_stats_free_stats_event(stats);
-	hdd_put_sta_info_ref(&adapter->sta_info_list, &stainfo, true,
-			     STA_INFO_WLAN_HDD_GET_STATION_REMOTE);
 
 	return status;
 }
@@ -5995,9 +5976,10 @@ static int __wlan_hdd_cfg80211_get_station(struct wiphy *wiphy,
 					   const uint8_t *mac,
 					   struct station_info *sinfo)
 {
-	int status;
+	int errno;
 	struct hdd_adapter *adapter = WLAN_HDD_GET_PRIV_PTR(dev);
 	struct hdd_context *hdd_ctx = wiphy_priv(wiphy);
+	struct hdd_station_info *stainfo;
 	bool get_peer_info_enable;
 	QDF_STATUS qdf_status;
 
@@ -6008,9 +5990,8 @@ static int __wlan_hdd_cfg80211_get_station(struct wiphy *wiphy,
 		return -EINVAL;
 	}
 
-	status = wlan_hdd_validate_context(hdd_ctx);
-	if (status)
-		return status;
+	if (wlan_hdd_validate_context(hdd_ctx))
+		return -EINVAL;
 
 	if (wlan_hdd_validate_vdev_id(adapter->vdev_id))
 		return -EINVAL;
@@ -6020,9 +6001,22 @@ static int __wlan_hdd_cfg80211_get_station(struct wiphy *wiphy,
 		qdf_status = ucfg_mlme_get_sap_get_peer_info(
 				hdd_ctx->psoc, &get_peer_info_enable);
 		if (qdf_status == QDF_STATUS_SUCCESS && get_peer_info_enable) {
-			status = wlan_hdd_get_station_remote(wiphy, dev,
-							     mac, sinfo);
-			if (!status)
+			stainfo = hdd_get_sta_info_by_mac(
+					&adapter->sta_info_list, mac,
+					STA_INFO_WLAN_HDD_CFG80211_GET_STATION);
+			if (!stainfo) {
+				hdd_debug("Peer " QDF_MAC_ADDR_FMT " not found",
+					  QDF_MAC_ADDR_REF(mac));
+				return -EINVAL;
+			}
+
+			errno = wlan_hdd_get_station_remote(wiphy, dev,
+							    stainfo, sinfo);
+			hdd_put_sta_info_ref(&adapter->sta_info_list, &stainfo,
+					     true,
+					STA_INFO_WLAN_HDD_CFG80211_GET_STATION
+					);
+			if (!errno)
 				return 0;
 		}
 		return wlan_hdd_get_sap_stats(adapter, sinfo);
@@ -6115,7 +6109,7 @@ int wlan_hdd_cfg80211_get_station(struct wiphy *wiphy,
  * __wlan_hdd_cfg80211_dump_station() - dump station statistics
  * @wiphy: Pointer to wiphy
  * @dev: Pointer to network device
- * @idx: variable to determine whether to get stats or not
+ * @idx: variable to station index, kernel iterate all stations over idx
  * @mac: Pointer to mac
  * @sinfo: Pointer to station info
  *
@@ -6126,11 +6120,59 @@ static int __wlan_hdd_cfg80211_dump_station(struct wiphy *wiphy,
 				int idx, u8 *mac,
 				struct station_info *sinfo)
 {
+	int errno;
+	struct hdd_adapter *adapter = WLAN_HDD_GET_PRIV_PTR(dev);
+	struct hdd_context *hdd_ctx = wiphy_priv(wiphy);
+	struct hdd_station_info *stainfo;
+	bool get_peer_info_enable;
+	QDF_STATUS qdf_status;
+
 	hdd_debug("idx: %d", idx);
-	if (idx != 0)
-		return -ENOENT;
-	qdf_mem_copy(mac, dev->dev_addr, QDF_MAC_ADDR_SIZE);
-	return __wlan_hdd_cfg80211_get_station(wiphy, dev, mac, sinfo);
+
+	if (QDF_GLOBAL_FTM_MODE == hdd_get_conparam()) {
+		hdd_err("Command not allowed in FTM mode");
+		return -EINVAL;
+	}
+
+	if (wlan_hdd_validate_context(hdd_ctx))
+		return -EINVAL;
+
+	if (wlan_hdd_validate_vdev_id(adapter->vdev_id))
+		return -EINVAL;
+
+	if (adapter->device_mode == QDF_SAP_MODE ||
+	    adapter->device_mode == QDF_P2P_GO_MODE) {
+		qdf_status = ucfg_mlme_get_sap_get_peer_info(
+				hdd_ctx->psoc, &get_peer_info_enable);
+		if (qdf_status == QDF_STATUS_SUCCESS && get_peer_info_enable) {
+			stainfo = hdd_get_sta_info_by_id(
+					&adapter->sta_info_list,
+					idx,
+					STA_INFO_WLAN_HDD_CFG80211_DUMP_STATION
+					);
+			if (!stainfo) {
+				hdd_err("peer idx %d NOT FOUND", idx);
+				return -ENOENT;
+			}
+
+			qdf_mem_copy(mac, &stainfo->sta_mac.bytes,
+				     QDF_MAC_ADDR_SIZE);
+			errno = wlan_hdd_get_station_remote(wiphy, dev,
+							    stainfo, sinfo);
+			hdd_put_sta_info_ref(&adapter->sta_info_list, &stainfo,
+					     true,
+					STA_INFO_WLAN_HDD_CFG80211_DUMP_STATION
+					);
+		} else {
+			errno = -EINVAL;
+			hdd_err("sap get peer info disabled!");
+		}
+	} else {
+		qdf_mem_copy(mac, dev->dev_addr, QDF_MAC_ADDR_SIZE);
+		errno = wlan_hdd_get_sta_stats(wiphy, adapter, mac, sinfo);
+	}
+
+	return errno;
 }
 
 /**