return (ag->num_started == ag->num_devices);
 }
 
+static void ath12k_fw_stats_pdevs_free(struct list_head *head)
+{
+       struct ath12k_fw_stats_pdev *i, *tmp;
+
+       list_for_each_entry_safe(i, tmp, head, list) {
+               list_del(&i->list);
+               kfree(i);
+       }
+}
+
+void ath12k_fw_stats_bcn_free(struct list_head *head)
+{
+       struct ath12k_fw_stats_bcn *i, *tmp;
+
+       list_for_each_entry_safe(i, tmp, head, list) {
+               list_del(&i->list);
+               kfree(i);
+       }
+}
+
+static void ath12k_fw_stats_vdevs_free(struct list_head *head)
+{
+       struct ath12k_fw_stats_vdev *i, *tmp;
+
+       list_for_each_entry_safe(i, tmp, head, list) {
+               list_del(&i->list);
+               kfree(i);
+       }
+}
+
+void ath12k_fw_stats_init(struct ath12k *ar)
+{
+       INIT_LIST_HEAD(&ar->fw_stats.vdevs);
+       INIT_LIST_HEAD(&ar->fw_stats.pdevs);
+       INIT_LIST_HEAD(&ar->fw_stats.bcn);
+       init_completion(&ar->fw_stats_complete);
+}
+
+void ath12k_fw_stats_free(struct ath12k_fw_stats *stats)
+{
+       ath12k_fw_stats_pdevs_free(&stats->pdevs);
+       ath12k_fw_stats_vdevs_free(&stats->vdevs);
+       ath12k_fw_stats_bcn_free(&stats->bcn);
+}
+
+void ath12k_fw_stats_reset(struct ath12k *ar)
+{
+       spin_lock_bh(&ar->data_lock);
+       ar->fw_stats.fw_stats_done = false;
+       ath12k_fw_stats_free(&ar->fw_stats);
+       spin_unlock_bh(&ar->data_lock);
+}
+
 static void ath12k_core_trigger_partner(struct ath12k_base *ab)
 {
        struct ath12k_hw_group *ag = ab->ag;
 
 u32 ath12k_core_get_max_num_tids(struct ath12k_base *ab);
 
 void ath12k_core_hw_group_set_mlo_capable(struct ath12k_hw_group *ag);
+void ath12k_fw_stats_init(struct ath12k *ar);
+void ath12k_fw_stats_bcn_free(struct list_head *head);
+void ath12k_fw_stats_free(struct ath12k_fw_stats *stats);
+void ath12k_fw_stats_reset(struct ath12k *ar);
 
 static inline const char *ath12k_scan_state_str(enum ath12k_scan_state state)
 {
 
         */
 }
 
-static void ath12k_fw_stats_pdevs_free(struct list_head *head)
-{
-       struct ath12k_fw_stats_pdev *i, *tmp;
-
-       list_for_each_entry_safe(i, tmp, head, list) {
-               list_del(&i->list);
-               kfree(i);
-       }
-}
-
-static void ath12k_fw_stats_bcn_free(struct list_head *head)
-{
-       struct ath12k_fw_stats_bcn *i, *tmp;
-
-       list_for_each_entry_safe(i, tmp, head, list) {
-               list_del(&i->list);
-               kfree(i);
-       }
-}
-
-static void ath12k_fw_stats_vdevs_free(struct list_head *head)
-{
-       struct ath12k_fw_stats_vdev *i, *tmp;
-
-       list_for_each_entry_safe(i, tmp, head, list) {
-               list_del(&i->list);
-               kfree(i);
-       }
-}
-
-void ath12k_debugfs_fw_stats_reset(struct ath12k *ar)
-{
-       spin_lock_bh(&ar->data_lock);
-       ar->fw_stats.fw_stats_done = false;
-       ath12k_fw_stats_vdevs_free(&ar->fw_stats.vdevs);
-       ath12k_fw_stats_bcn_free(&ar->fw_stats.bcn);
-       ath12k_fw_stats_pdevs_free(&ar->fw_stats.pdevs);
-       spin_unlock_bh(&ar->data_lock);
-}
-
-static int ath12k_debugfs_fw_stats_request(struct ath12k *ar,
-                                          struct ath12k_fw_stats_req_params *param)
-{
-       struct ath12k_base *ab = ar->ab;
-       unsigned long timeout, time_left;
-       int ret;
-
-       lockdep_assert_wiphy(ath12k_ar_to_hw(ar)->wiphy);
-
-       /* FW stats can get split when exceeding the stats data buffer limit.
-        * In that case, since there is no end marking for the back-to-back
-        * received 'update stats' event, we keep a 3 seconds timeout in case,
-        * fw_stats_done is not marked yet
-        */
-       timeout = jiffies + msecs_to_jiffies(3 * 1000);
-
-       ath12k_debugfs_fw_stats_reset(ar);
-
-       reinit_completion(&ar->fw_stats_complete);
-
-       ret = ath12k_wmi_send_stats_request_cmd(ar, param->stats_id,
-                                               param->vdev_id, param->pdev_id);
-
-       if (ret) {
-               ath12k_warn(ab, "could not request fw stats (%d)\n",
-                           ret);
-               return ret;
-       }
-
-       time_left = wait_for_completion_timeout(&ar->fw_stats_complete,
-                                               1 * HZ);
-       /* If the wait timed out, return -ETIMEDOUT */
-       if (!time_left)
-               return -ETIMEDOUT;
-
-       /* Firmware sends WMI_UPDATE_STATS_EVENTID back-to-back
-        * when stats data buffer limit is reached. fw_stats_complete
-        * is completed once host receives first event from firmware, but
-        * still end might not be marked in the TLV.
-        * Below loop is to confirm that firmware completed sending all the event
-        * and fw_stats_done is marked true when end is marked in the TLV
-        */
-       for (;;) {
-               if (time_after(jiffies, timeout))
-                       break;
-
-               spin_lock_bh(&ar->data_lock);
-               if (ar->fw_stats.fw_stats_done) {
-                       spin_unlock_bh(&ar->data_lock);
-                       break;
-               }
-               spin_unlock_bh(&ar->data_lock);
-       }
-       return 0;
-}
-
 void
 ath12k_debugfs_fw_stats_process(struct ath12k *ar,
                                struct ath12k_fw_stats *stats)
                        num_bcn = 0;
                }
        }
-       if (stats->stats_id == WMI_REQUEST_PDEV_STAT) {
-               list_splice_tail_init(&stats->pdevs, &ar->fw_stats.pdevs);
-               ar->fw_stats.fw_stats_done = true;
-       }
 }
 
 static int ath12k_open_vdev_stats(struct inode *inode, struct file *file)
        param.vdev_id = 0;
        param.stats_id = WMI_REQUEST_VDEV_STAT;
 
-       ret = ath12k_debugfs_fw_stats_request(ar, ¶m);
+       ret = ath12k_mac_get_fw_stats(ar, ¶m);
        if (ret) {
                ath12k_warn(ar->ab, "failed to request fw vdev stats: %d\n", ret);
                return ret;
                        continue;
 
                param.vdev_id = arvif->vdev_id;
-               ret = ath12k_debugfs_fw_stats_request(ar, ¶m);
+               ret = ath12k_mac_get_fw_stats(ar, ¶m);
                if (ret) {
                        ath12k_warn(ar->ab, "failed to request fw bcn stats: %d\n", ret);
                        return ret;
        param.vdev_id = 0;
        param.stats_id = WMI_REQUEST_PDEV_STAT;
 
-       ret = ath12k_debugfs_fw_stats_request(ar, ¶m);
+       ret = ath12k_mac_get_fw_stats(ar, ¶m);
        if (ret) {
                ath12k_warn(ab, "failed to request fw pdev stats: %d\n", ret);
                return ret;
        debugfs_create_file("pdev_stats", 0600, fwstats_dir, ar,
                            &fops_pdev_stats);
 
-       INIT_LIST_HEAD(&ar->fw_stats.vdevs);
-       INIT_LIST_HEAD(&ar->fw_stats.bcn);
-       INIT_LIST_HEAD(&ar->fw_stats.pdevs);
-
-       init_completion(&ar->fw_stats_complete);
+       ath12k_fw_stats_init(ar);
 }
 
 void ath12k_debugfs_register(struct ath12k *ar)
 
 void ath12k_debugfs_unregister(struct ath12k *ar);
 void ath12k_debugfs_fw_stats_process(struct ath12k *ar,
                                     struct ath12k_fw_stats *stats);
-void ath12k_debugfs_fw_stats_reset(struct ath12k *ar);
 
 static inline bool ath12k_debugfs_is_extd_rx_stats_enabled(struct ath12k *ar)
 {
 {
 }
 
-static inline void ath12k_debugfs_fw_stats_reset(struct ath12k *ar)
-{
-}
-
 static inline bool ath12k_debugfs_is_extd_rx_stats_enabled(struct ath12k *ar)
 {
        return false;
 
        return 0;
 }
 
-static int ath12k_mac_get_fw_stats(struct ath12k *ar, u32 pdev_id,
-                                  u32 vdev_id, u32 stats_id)
+int ath12k_mac_get_fw_stats(struct ath12k *ar,
+                           struct ath12k_fw_stats_req_params *param)
 {
        struct ath12k_base *ab = ar->ab;
        struct ath12k_hw *ah = ath12k_ar_to_ah(ar);
-       unsigned long time_left;
+       unsigned long timeout, time_left;
        int ret;
 
        guard(mutex)(&ah->hw_mutex);
        if (ah->state != ATH12K_HW_STATE_ON)
                return -ENETDOWN;
 
+       /* FW stats can get split when exceeding the stats data buffer limit.
+        * In that case, since there is no end marking for the back-to-back
+        * received 'update stats' event, we keep a 3 seconds timeout in case,
+        * fw_stats_done is not marked yet
+        */
+       timeout = jiffies + msecs_to_jiffies(3 * 1000);
+       ath12k_fw_stats_reset(ar);
+
        reinit_completion(&ar->fw_stats_complete);
 
-       ret = ath12k_wmi_send_stats_request_cmd(ar, stats_id, vdev_id, pdev_id);
+       ret = ath12k_wmi_send_stats_request_cmd(ar, param->stats_id,
+                                               param->vdev_id, param->pdev_id);
 
        if (ret) {
                ath12k_warn(ab, "failed to request fw stats: %d\n", ret);
 
        ath12k_dbg(ab, ATH12K_DBG_WMI,
                   "get fw stat pdev id %d vdev id %d stats id 0x%x\n",
-                  pdev_id, vdev_id, stats_id);
+                  param->pdev_id, param->vdev_id, param->stats_id);
 
        time_left = wait_for_completion_timeout(&ar->fw_stats_complete, 1 * HZ);
 
-       if (!time_left)
+       if (!time_left) {
                ath12k_warn(ab, "time out while waiting for get fw stats\n");
+               return -ETIMEDOUT;
+       }
 
-       return ret;
+       /* Firmware sends WMI_UPDATE_STATS_EVENTID back-to-back
+        * when stats data buffer limit is reached. fw_stats_complete
+        * is completed once host receives first event from firmware, but
+        * still end might not be marked in the TLV.
+        * Below loop is to confirm that firmware completed sending all the event
+        * and fw_stats_done is marked true when end is marked in the TLV.
+        */
+       for (;;) {
+               if (time_after(jiffies, timeout))
+                       break;
+               spin_lock_bh(&ar->data_lock);
+               if (ar->fw_stats.fw_stats_done) {
+                       spin_unlock_bh(&ar->data_lock);
+                       break;
+               }
+               spin_unlock_bh(&ar->data_lock);
+       }
+       return 0;
 }
 
 static void ath12k_mac_op_sta_statistics(struct ieee80211_hw *hw,
                                         struct station_info *sinfo)
 {
        struct ath12k_sta *ahsta = ath12k_sta_to_ahsta(sta);
+       struct ath12k_fw_stats_req_params params = {};
        struct ath12k_link_sta *arsta;
        struct ath12k *ar;
        s8 signal;
        /* TODO: Use real NF instead of default one. */
        signal = arsta->rssi_comb;
 
+       params.pdev_id = ar->pdev->pdev_id;
+       params.vdev_id = 0;
+       params.stats_id = WMI_REQUEST_VDEV_STAT;
+
        if (!signal &&
            ahsta->ahvif->vdev_type == WMI_VDEV_TYPE_STA &&
-           !(ath12k_mac_get_fw_stats(ar, ar->pdev->pdev_id, 0,
-                                     WMI_REQUEST_VDEV_STAT)))
+           !(ath12k_mac_get_fw_stats(ar, ¶ms)))
                signal = arsta->rssi_beacon;
 
        if (signal) {
 
 struct ath12k *ath12k_get_ar_by_vif(struct ieee80211_hw *hw,
                                    struct ieee80211_vif *vif,
                                    u8 link_id);
+int ath12k_mac_get_fw_stats(struct ath12k *ar, struct ath12k_fw_stats_req_params *param);
 #endif
 
 
 struct wmi_tlv_fw_stats_parse {
        const struct wmi_stats_event *ev;
+       struct ath12k_fw_stats *stats;
 };
 
 struct ath12k_wmi_dma_ring_caps_parse {
        else
                buf[len] = 0;
 
-       ath12k_debugfs_fw_stats_reset(ar);
+       ath12k_fw_stats_reset(ar);
 }
 
 static void
                                              u16 len)
 {
        const struct wmi_stats_event *ev = parse->ev;
-       struct ath12k_fw_stats stats = {0};
+       struct ath12k_fw_stats *stats = parse->stats;
        struct ath12k *ar;
        struct ath12k_link_vif *arvif;
        struct ieee80211_sta *sta;
        int i, ret = 0;
        const void *data = ptr;
 
-       INIT_LIST_HEAD(&stats.vdevs);
-       INIT_LIST_HEAD(&stats.bcn);
-       INIT_LIST_HEAD(&stats.pdevs);
-
        if (!ev) {
                ath12k_warn(ab, "failed to fetch update stats ev");
                return -EPROTO;
        }
 
+       if (!stats)
+               return -EINVAL;
+
        rcu_read_lock();
 
-       ar = ath12k_mac_get_ar_by_pdev_id(ab, le32_to_cpu(ev->pdev_id));
+       stats->pdev_id = le32_to_cpu(ev->pdev_id);
+       ar = ath12k_mac_get_ar_by_pdev_id(ab, stats->pdev_id);
        if (!ar) {
                ath12k_warn(ab, "invalid pdev id %d in update stats event\n",
                            le32_to_cpu(ev->pdev_id));
                if (!dst)
                        continue;
                ath12k_wmi_pull_vdev_stats(src, dst);
-               stats.stats_id = WMI_REQUEST_VDEV_STAT;
-               list_add_tail(&dst->list, &stats.vdevs);
+               stats->stats_id = WMI_REQUEST_VDEV_STAT;
+               list_add_tail(&dst->list, &stats->vdevs);
        }
        for (i = 0; i < le32_to_cpu(ev->num_bcn_stats); i++) {
                const struct ath12k_wmi_bcn_stats_params *src;
                if (!dst)
                        continue;
                ath12k_wmi_pull_bcn_stats(src, dst);
-               stats.stats_id = WMI_REQUEST_BCN_STAT;
-               list_add_tail(&dst->list, &stats.bcn);
+               stats->stats_id = WMI_REQUEST_BCN_STAT;
+               list_add_tail(&dst->list, &stats->bcn);
        }
        for (i = 0; i < le32_to_cpu(ev->num_pdev_stats); i++) {
                const struct ath12k_wmi_pdev_stats_params *src;
                        goto exit;
                }
 
-               stats.stats_id = WMI_REQUEST_PDEV_STAT;
+               stats->stats_id = WMI_REQUEST_PDEV_STAT;
 
                data += sizeof(*src);
                len -= sizeof(*src);
                ath12k_wmi_pull_pdev_stats_base(&src->base, dst);
                ath12k_wmi_pull_pdev_stats_tx(&src->tx, dst);
                ath12k_wmi_pull_pdev_stats_rx(&src->rx, dst);
-               list_add_tail(&dst->list, &stats.pdevs);
+               list_add_tail(&dst->list, &stats->pdevs);
        }
 
-       complete(&ar->fw_stats_complete);
-       ath12k_debugfs_fw_stats_process(ar, &stats);
 exit:
        rcu_read_unlock();
        return ret;
        return ret;
 }
 
+static int ath12k_wmi_pull_fw_stats(struct ath12k_base *ab, struct sk_buff *skb,
+                                   struct ath12k_fw_stats *stats)
+{
+       struct wmi_tlv_fw_stats_parse parse = {};
+
+       stats->stats_id = 0;
+       parse.stats = stats;
+
+       return ath12k_wmi_tlv_iter(ab, skb->data, skb->len,
+                                  ath12k_wmi_tlv_fw_stats_parse,
+                                  &parse);
+}
+
 static void ath12k_update_stats_event(struct ath12k_base *ab, struct sk_buff *skb)
 {
+       struct ath12k_fw_stats stats = {};
+       struct ath12k *ar;
        int ret;
-       struct wmi_tlv_fw_stats_parse parse = {};
 
-       ret = ath12k_wmi_tlv_iter(ab, skb->data, skb->len,
-                                 ath12k_wmi_tlv_fw_stats_parse,
-                                 &parse);
-       if (ret)
-               ath12k_warn(ab, "failed to parse fw stats %d\n", ret);
+       INIT_LIST_HEAD(&stats.pdevs);
+       INIT_LIST_HEAD(&stats.vdevs);
+       INIT_LIST_HEAD(&stats.bcn);
+
+       ret = ath12k_wmi_pull_fw_stats(ab, skb, &stats);
+       if (ret) {
+               ath12k_warn(ab, "failed to pull fw stats: %d\n", ret);
+               goto free;
+       }
+
+       ath12k_dbg(ab, ATH12K_DBG_WMI, "event update stats");
+
+       rcu_read_lock();
+       ar = ath12k_mac_get_ar_by_pdev_id(ab, stats.pdev_id);
+       if (!ar) {
+               rcu_read_unlock();
+               ath12k_warn(ab, "failed to get ar for pdev_id %d: %d\n",
+                           stats.pdev_id, ret);
+               goto free;
+       }
+
+       spin_lock_bh(&ar->data_lock);
+
+       /* WMI_REQUEST_PDEV_STAT can be requested via .get_txpower mac ops or via
+        * debugfs fw stats. Therefore, processing it separately.
+        */
+       if (stats.stats_id == WMI_REQUEST_PDEV_STAT) {
+               list_splice_tail_init(&stats.pdevs, &ar->fw_stats.pdevs);
+               ar->fw_stats.fw_stats_done = true;
+               goto complete;
+       }
+
+       /* WMI_REQUEST_VDEV_STAT and WMI_REQUEST_BCN_STAT are currently requested only
+        * via debugfs fw stats. Hence, processing these in debugfs context.
+        */
+       ath12k_debugfs_fw_stats_process(ar, &stats);
+
+complete:
+       complete(&ar->fw_stats_complete);
+       spin_unlock_bh(&ar->data_lock);
+       rcu_read_unlock();
+
+       /* Since the stats's pdev, vdev and beacon list are spliced and reinitialised
+        * at this point, no need to free the individual list.
+        */
+       return;
+
+free:
+       ath12k_fw_stats_free(&stats);
 }
 
 /* PDEV_CTL_FAILSAFE_CHECK_EVENT is received from FW when the frequency scanned