Quelle  stats.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/*
 * Copyright (C) 2024-2025 Intel Corporation
 */

#include "mld.h"
#include "stats.h"
#include "sta.h"
#include "mlo.h"
#include "hcmd.h"
#include "iface.h"
#include "scan.h"
#include "phy.h"
#include "fw/api/stats.h"

static int iwl_mld_send_fw_stats_cmd(struct iwl_mld *mld, u32 cfg_mask,
         u32 cfg_time, u32 type_mask)
{
 u32 cmd_id = WIDE_ID(SYSTEM_GROUP, SYSTEM_STATISTICS_CMD);
 struct iwl_system_statistics_cmd stats_cmd = {
  .cfg_mask = cpu_to_le32(cfg_mask),
  .config_time_sec = cpu_to_le32(cfg_time),
  .type_id_mask = cpu_to_le32(type_mask),
 };

 return iwl_mld_send_cmd_pdu(mld, cmd_id, &stats_cmd);
}

int iwl_mld_clear_stats_in_fw(struct iwl_mld *mld)
{
 u32 cfg_mask = IWL_STATS_CFG_FLG_ON_DEMAND_NTFY_MSK;
 u32 type_mask = IWL_STATS_NTFY_TYPE_ID_OPER |
   IWL_STATS_NTFY_TYPE_ID_OPER_PART1;

 return iwl_mld_send_fw_stats_cmd(mld, cfg_mask, 0, type_mask);
}

static void
iwl_mld_fill_stats_from_oper_notif(struct iwl_mld *mld,
       struct iwl_rx_packet *pkt,
       u8 fw_sta_id, struct station_info *sinfo)
{
 const struct iwl_system_statistics_notif_oper *notif =
  (void *)&pkt->data;
 const struct iwl_stats_ntfy_per_sta *per_sta =
  ¬if->per_sta[fw_sta_id];
 struct ieee80211_link_sta *link_sta;
 struct iwl_mld_link_sta *mld_link_sta;

 /* 0 isn't a valid value, but FW might send 0.
 * In that case, set the latest non-zero value we stored
 */
 rcu_read_lock();

 link_sta = rcu_dereference(mld->fw_id_to_link_sta[fw_sta_id]);
 if (IS_ERR_OR_NULL(link_sta))
  goto unlock;

 mld_link_sta = iwl_mld_link_sta_from_mac80211(link_sta);
 if (WARN_ON(!mld_link_sta))
  goto unlock;

 if (per_sta->average_energy)
  mld_link_sta->signal_avg =
   -(s8)le32_to_cpu(per_sta->average_energy);

 sinfo->signal_avg = mld_link_sta->signal_avg;
 sinfo->filled |= BIT_ULL(NL80211_STA_INFO_SIGNAL_AVG);

unlock:
 rcu_read_unlock();
}

struct iwl_mld_stats_data {
 u8 fw_sta_id;
 struct station_info *sinfo;
 struct iwl_mld *mld;
};

static bool iwl_mld_wait_stats_handler(struct iwl_notif_wait_data *notif_data,
           struct iwl_rx_packet *pkt, void *data)
{
 struct iwl_mld_stats_data *stats_data = data;
 u16 cmd = WIDE_ID(pkt->hdr.group_id, pkt->hdr.cmd);

 switch (cmd) {
 case WIDE_ID(STATISTICS_GROUP, STATISTICS_OPER_NOTIF):
  iwl_mld_fill_stats_from_oper_notif(stats_data->mld, pkt,
         stats_data->fw_sta_id,
         stats_data->sinfo);
  break;
 case WIDE_ID(STATISTICS_GROUP, STATISTICS_OPER_PART1_NOTIF):
  break;
 case WIDE_ID(SYSTEM_GROUP, SYSTEM_STATISTICS_END_NOTIF):
  return true;
 }

 return false;
}

static int
iwl_mld_fw_stats_to_mac80211(struct iwl_mld *mld, struct iwl_mld_sta *mld_sta,
        struct station_info *sinfo)
{
 u32 cfg_mask = IWL_STATS_CFG_FLG_ON_DEMAND_NTFY_MSK |
         IWL_STATS_CFG_FLG_RESET_MSK;
 u32 type_mask = IWL_STATS_NTFY_TYPE_ID_OPER |
   IWL_STATS_NTFY_TYPE_ID_OPER_PART1;
 static const u16 notifications[] = {
  WIDE_ID(STATISTICS_GROUP, STATISTICS_OPER_NOTIF),
  WIDE_ID(STATISTICS_GROUP, STATISTICS_OPER_PART1_NOTIF),
  WIDE_ID(SYSTEM_GROUP, SYSTEM_STATISTICS_END_NOTIF),
 };
 struct iwl_mld_stats_data wait_stats_data = {
  /* We don't support drv_sta_statistics in EMLSR */
  .fw_sta_id = mld_sta->deflink.fw_id,
  .sinfo = sinfo,
  .mld = mld,
 };
 struct iwl_notification_wait stats_wait;
 int ret;

 iwl_init_notification_wait(&mld->notif_wait, &stats_wait,
       notifications, ARRAY_SIZE(notifications),
       iwl_mld_wait_stats_handler,
       &wait_stats_data);

 ret = iwl_mld_send_fw_stats_cmd(mld, cfg_mask, 0, type_mask);
 if (ret) {
  iwl_remove_notification(&mld->notif_wait, &stats_wait);
  return ret;
 }

 /* Wait 500ms for OPERATIONAL, PART1, and END notifications,
 * which should be sufficient for the firmware to gather data
 * from all LMACs and send notifications to the host.
 */
 ret = iwl_wait_notification(&mld->notif_wait, &stats_wait, HZ / 2);
 if (ret)
  return ret;

 /* When periodic statistics are sent, FW will clear its statistics DB.
 * If the statistics request here happens shortly afterwards,
 * the response will contain data collected over a short time
 * interval. The response we got here shouldn't be processed by
 * the general statistics processing because it's incomplete.
 * So, we delete it from the list so it won't be processed.
 */
 iwl_mld_delete_handlers(mld, notifications, ARRAY_SIZE(notifications));

 return 0;
}

#define PERIODIC_STATS_SECONDS 5

int iwl_mld_request_periodic_fw_stats(struct iwl_mld *mld, bool enable)
{
 u32 cfg_mask = enable ? 0 : IWL_STATS_CFG_FLG_DISABLE_NTFY_MSK;
 u32 type_mask = enable ? (IWL_STATS_NTFY_TYPE_ID_OPER |
      IWL_STATS_NTFY_TYPE_ID_OPER_PART1) : 0;
 u32 cfg_time = enable ? PERIODIC_STATS_SECONDS : 0;

 return iwl_mld_send_fw_stats_cmd(mld, cfg_mask, cfg_time, type_mask);
}

static void iwl_mld_sta_stats_fill_txrate(struct iwl_mld_sta *mld_sta,
       struct station_info *sinfo)
{
 struct rate_info *rinfo = &sinfo->txrate;
 u32 rate_n_flags = mld_sta->deflink.last_rate_n_flags;
 u32 format = rate_n_flags & RATE_MCS_MOD_TYPE_MSK;
 u32 gi_ltf;

 sinfo->filled |= BIT_ULL(NL80211_STA_INFO_TX_BITRATE);

 switch (rate_n_flags & RATE_MCS_CHAN_WIDTH_MSK) {
 case RATE_MCS_CHAN_WIDTH_20:
  rinfo->bw = RATE_INFO_BW_20;
  break;
 case RATE_MCS_CHAN_WIDTH_40:
  rinfo->bw = RATE_INFO_BW_40;
  break;
 case RATE_MCS_CHAN_WIDTH_80:
  rinfo->bw = RATE_INFO_BW_80;
  break;
 case RATE_MCS_CHAN_WIDTH_160:
  rinfo->bw = RATE_INFO_BW_160;
  break;
 case RATE_MCS_CHAN_WIDTH_320:
  rinfo->bw = RATE_INFO_BW_320;
  break;
 }

 if (format == RATE_MCS_MOD_TYPE_CCK ||
     format == RATE_MCS_MOD_TYPE_LEGACY_OFDM) {
  int rate = u32_get_bits(rate_n_flags, RATE_LEGACY_RATE_MSK);

  /* add the offset needed to get to the legacy ofdm indices */
  if (format == RATE_MCS_MOD_TYPE_LEGACY_OFDM)
   rate += IWL_FIRST_OFDM_RATE;

  switch (rate) {
  case IWL_RATE_1M_INDEX:
   rinfo->legacy = 10;
   break;
  case IWL_RATE_2M_INDEX:
   rinfo->legacy = 20;
   break;
  case IWL_RATE_5M_INDEX:
   rinfo->legacy = 55;
   break;
  case IWL_RATE_11M_INDEX:
   rinfo->legacy = 110;
   break;
  case IWL_RATE_6M_INDEX:
   rinfo->legacy = 60;
   break;
  case IWL_RATE_9M_INDEX:
   rinfo->legacy = 90;
   break;
  case IWL_RATE_12M_INDEX:
   rinfo->legacy = 120;
   break;
  case IWL_RATE_18M_INDEX:
   rinfo->legacy = 180;
   break;
  case IWL_RATE_24M_INDEX:
   rinfo->legacy = 240;
   break;
  case IWL_RATE_36M_INDEX:
   rinfo->legacy = 360;
   break;
  case IWL_RATE_48M_INDEX:
   rinfo->legacy = 480;
   break;
  case IWL_RATE_54M_INDEX:
   rinfo->legacy = 540;
  }
  return;
 }

 rinfo->nss = u32_get_bits(rate_n_flags, RATE_MCS_NSS_MSK) + 1;

 if (format == RATE_MCS_MOD_TYPE_HT)
  rinfo->mcs = RATE_HT_MCS_INDEX(rate_n_flags);
 else
  rinfo->mcs = u32_get_bits(rate_n_flags, RATE_MCS_CODE_MSK);

 if (rate_n_flags & RATE_MCS_SGI_MSK)
  rinfo->flags |= RATE_INFO_FLAGS_SHORT_GI;

 switch (format) {
 case RATE_MCS_MOD_TYPE_EHT:
  rinfo->flags |= RATE_INFO_FLAGS_EHT_MCS;
  break;
 case RATE_MCS_MOD_TYPE_HE:
  gi_ltf = u32_get_bits(rate_n_flags, RATE_MCS_HE_GI_LTF_MSK);

  rinfo->flags |= RATE_INFO_FLAGS_HE_MCS;

  if (rate_n_flags & RATE_MCS_HE_106T_MSK) {
   rinfo->bw = RATE_INFO_BW_HE_RU;
   rinfo->he_ru_alloc = NL80211_RATE_INFO_HE_RU_ALLOC_106;
  }

  switch (rate_n_flags & RATE_MCS_HE_TYPE_MSK) {
  case RATE_MCS_HE_TYPE_SU:
  case RATE_MCS_HE_TYPE_EXT_SU:
   if (gi_ltf == 0 || gi_ltf == 1)
    rinfo->he_gi = NL80211_RATE_INFO_HE_GI_0_8;
   else if (gi_ltf == 2)
    rinfo->he_gi = NL80211_RATE_INFO_HE_GI_1_6;
   else if (gi_ltf == 3)
    rinfo->he_gi = NL80211_RATE_INFO_HE_GI_3_2;
   else
    rinfo->he_gi = NL80211_RATE_INFO_HE_GI_0_8;
   break;
  case RATE_MCS_HE_TYPE_MU:
   if (gi_ltf == 0 || gi_ltf == 1)
    rinfo->he_gi = NL80211_RATE_INFO_HE_GI_0_8;
   else if (gi_ltf == 2)
    rinfo->he_gi = NL80211_RATE_INFO_HE_GI_1_6;
   else
    rinfo->he_gi = NL80211_RATE_INFO_HE_GI_3_2;
   break;
  case RATE_MCS_HE_TYPE_TRIG:
   if (gi_ltf == 0 || gi_ltf == 1)
    rinfo->he_gi = NL80211_RATE_INFO_HE_GI_1_6;
   else
    rinfo->he_gi = NL80211_RATE_INFO_HE_GI_3_2;
   break;
  }

  if (rate_n_flags & RATE_HE_DUAL_CARRIER_MODE_MSK)
   rinfo->he_dcm = 1;
  break;
 case RATE_MCS_MOD_TYPE_HT:
  rinfo->flags |= RATE_INFO_FLAGS_MCS;
  break;
 case RATE_MCS_MOD_TYPE_VHT:
  rinfo->flags |= RATE_INFO_FLAGS_VHT_MCS;
  break;
 }
}

void iwl_mld_mac80211_sta_statistics(struct ieee80211_hw *hw,
         struct ieee80211_vif *vif,
         struct ieee80211_sta *sta,
         struct station_info *sinfo)
{
 struct iwl_mld_sta *mld_sta = iwl_mld_sta_from_mac80211(sta);

 /* This API is not EMLSR ready, so we cannot provide complete
 * information if EMLSR is active
 */
 if (hweight16(vif->active_links) > 1)
  return;

 if (iwl_mld_fw_stats_to_mac80211(mld_sta->mld, mld_sta, sinfo))
  return;

 iwl_mld_sta_stats_fill_txrate(mld_sta, sinfo);

 /* TODO: NL80211_STA_INFO_BEACON_RX */

 /* TODO: NL80211_STA_INFO_BEACON_SIGNAL_AVG */
}

#define IWL_MLD_TRAFFIC_LOAD_MEDIUM_THRESH 10 /* percentage */
#define IWL_MLD_TRAFFIC_LOAD_HIGH_THRESH 50 /* percentage */
#define IWL_MLD_TRAFFIC_LOAD_MIN_WINDOW_USEC (500 * 1000)

static u8 iwl_mld_stats_load_percentage(u32 last_ts_usec, u32 curr_ts_usec,
     u32 total_airtime_usec)
{
 u32 elapsed_usec = curr_ts_usec - last_ts_usec;

 if (elapsed_usec < IWL_MLD_TRAFFIC_LOAD_MIN_WINDOW_USEC)
  return 0;

 return (100 * total_airtime_usec / elapsed_usec);
}

static void iwl_mld_stats_recalc_traffic_load(struct iwl_mld *mld,
           u32 total_airtime_usec,
           u32 curr_ts_usec)
{
 u32 last_ts_usec = mld->scan.traffic_load.last_stats_ts_usec;
 u8 load_prec;

 /* Skip the calculation as this is the first notification received */
 if (!last_ts_usec)
  goto out;

 load_prec = iwl_mld_stats_load_percentage(last_ts_usec, curr_ts_usec,
        total_airtime_usec);

 if (load_prec > IWL_MLD_TRAFFIC_LOAD_HIGH_THRESH)
  mld->scan.traffic_load.status = IWL_MLD_TRAFFIC_HIGH;
 else if (load_prec > IWL_MLD_TRAFFIC_LOAD_MEDIUM_THRESH)
  mld->scan.traffic_load.status = IWL_MLD_TRAFFIC_MEDIUM;
 else
  mld->scan.traffic_load.status = IWL_MLD_TRAFFIC_LOW;

out:
 mld->scan.traffic_load.last_stats_ts_usec = curr_ts_usec;
}

static void iwl_mld_update_link_sig(struct ieee80211_vif *vif, int sig,
        struct ieee80211_bss_conf *bss_conf)
{
 struct iwl_mld *mld = iwl_mld_vif_from_mac80211(vif)->mld;
 int exit_emlsr_thresh;

 if (sig == 0) {
  IWL_DEBUG_RX(mld, "RSSI is 0 - skip signal based decision\n");
  return;
 }

 /* TODO: task=statistics handle CQM notifications */

 if (!iwl_mld_emlsr_active(vif)) {
  /* We're not in EMLSR and our signal is bad,
 * try to switch link maybe. EMLSR will be handled below.
 */
  if (sig < IWL_MLD_LOW_RSSI_MLO_SCAN_THRESH)
   iwl_mld_int_mlo_scan(mld, vif);
  return;
 }

 /* We are in EMLSR, check if we need to exit */
 exit_emlsr_thresh =
  iwl_mld_get_emlsr_rssi_thresh(mld, &bss_conf->chanreq.oper,
           true);

 if (sig < exit_emlsr_thresh)
  iwl_mld_exit_emlsr(mld, vif, IWL_MLD_EMLSR_EXIT_LOW_RSSI,
       iwl_mld_get_other_link(vif,
         bss_conf->link_id));
}

static void
iwl_mld_process_per_link_stats(struct iwl_mld *mld,
          const struct iwl_stats_ntfy_per_link *per_link,
          u32 curr_ts_usec)
{
 u32 total_airtime_usec = 0;

 for (u32 fw_id = 0;
      fw_id < ARRAY_SIZE(mld->fw_id_to_bss_conf);
      fw_id++) {
  const struct iwl_stats_ntfy_per_link *link_stats;
  struct ieee80211_bss_conf *bss_conf;
  int sig;

  bss_conf = wiphy_dereference(mld->wiphy,
          mld->fw_id_to_bss_conf[fw_id]);
  if (!bss_conf || bss_conf->vif->type != NL80211_IFTYPE_STATION)
   continue;

  link_stats = &per_link[fw_id];

  total_airtime_usec += le32_to_cpu(link_stats->air_time);

  sig = -le32_to_cpu(link_stats->beacon_filter_average_energy);
  iwl_mld_update_link_sig(bss_conf->vif, sig, bss_conf);

  /* TODO: parse more fields here (task=statistics)*/
 }

 iwl_mld_stats_recalc_traffic_load(mld, total_airtime_usec,
       curr_ts_usec);
}

static void
iwl_mld_process_per_sta_stats(struct iwl_mld *mld,
         const struct iwl_stats_ntfy_per_sta *per_sta)
{
 for (int i = 0; i < mld->fw->ucode_capa.num_stations; i++) {
  struct ieee80211_link_sta *link_sta =
   wiphy_dereference(mld->wiphy,
       mld->fw_id_to_link_sta[i]);
  struct iwl_mld_link_sta *mld_link_sta;
  s8 avg_energy =
   -(s8)le32_to_cpu(per_sta[i].average_energy);

  if (IS_ERR_OR_NULL(link_sta) || !avg_energy)
   continue;

  mld_link_sta = iwl_mld_link_sta_from_mac80211(link_sta);
  if (WARN_ON(!mld_link_sta))
   continue;

  mld_link_sta->signal_avg = avg_energy;
 }
}

static void iwl_mld_fill_chanctx_stats(struct ieee80211_hw *hw,
           struct ieee80211_chanctx_conf *ctx,
           void *data)
{
 struct iwl_mld_phy *phy = iwl_mld_phy_from_mac80211(ctx);
 const struct iwl_stats_ntfy_per_phy *per_phy = data;
 u32 new_load, old_load;

 if (WARN_ON(phy->fw_id >= IWL_STATS_MAX_PHY_OPERATIONAL))
  return;

 phy->channel_load_by_us =
  le32_to_cpu(per_phy[phy->fw_id].channel_load_by_us);

 old_load = phy->avg_channel_load_not_by_us;
 new_load = le32_to_cpu(per_phy[phy->fw_id].channel_load_not_by_us);

 if (IWL_FW_CHECK(phy->mld,
    new_load != IWL_STATS_UNKNOWN_CHANNEL_LOAD &&
    new_load > 100,
    "Invalid channel load %u\n", new_load))
  return;

 if (new_load != IWL_STATS_UNKNOWN_CHANNEL_LOAD) {
  /* update giving a weight of 0.5 for the old value */
  phy->avg_channel_load_not_by_us = (new_load >> 1) +
        (old_load >> 1);
 }

 iwl_mld_emlsr_check_chan_load(hw, phy, old_load);
}

static void
iwl_mld_process_per_phy_stats(struct iwl_mld *mld,
         const struct iwl_stats_ntfy_per_phy *per_phy)
{
 ieee80211_iter_chan_contexts_mtx(mld->hw,
      iwl_mld_fill_chanctx_stats,
      (void *)(uintptr_t)per_phy);

}

void iwl_mld_handle_stats_oper_notif(struct iwl_mld *mld,
         struct iwl_rx_packet *pkt)
{
 const struct iwl_system_statistics_notif_oper *stats =
  (void *)&pkt->data;
 u32 curr_ts_usec = le32_to_cpu(stats->time_stamp);

 BUILD_BUG_ON(ARRAY_SIZE(stats->per_sta) != IWL_STATION_COUNT_MAX);
 BUILD_BUG_ON(ARRAY_SIZE(stats->per_link) <
       ARRAY_SIZE(mld->fw_id_to_bss_conf));

 iwl_mld_process_per_link_stats(mld, stats->per_link, curr_ts_usec);
 iwl_mld_process_per_sta_stats(mld, stats->per_sta);
 iwl_mld_process_per_phy_stats(mld, stats->per_phy);
}

void iwl_mld_handle_stats_oper_part1_notif(struct iwl_mld *mld,
        struct iwl_rx_packet *pkt)
{
 /* TODO */
}