Quelle  time_sync.c   Sprache: C

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

#include "mld.h"
#include "hcmd.h"
#include "ptp.h"
#include "time_sync.h"
#include <linux/ieee80211.h>

static int iwl_mld_init_time_sync(struct iwl_mld *mld, u32 protocols,
      const u8 *addr)
{
 struct iwl_mld_time_sync_data *time_sync = kzalloc(sizeof(*time_sync),
          GFP_KERNEL);

 if (!time_sync)
  return -ENOMEM;

 time_sync->active_protocols = protocols;
 ether_addr_copy(time_sync->peer_addr, addr);
 skb_queue_head_init(&time_sync->frame_list);
 rcu_assign_pointer(mld->time_sync, time_sync);

 return 0;
}

int iwl_mld_time_sync_fw_config(struct iwl_mld *mld)
{
 struct iwl_time_sync_cfg_cmd cmd = {};
 struct iwl_mld_time_sync_data *time_sync;
 int err;

 time_sync = wiphy_dereference(mld->wiphy, mld->time_sync);
 if (!time_sync)
  return -EINVAL;

 cmd.protocols = cpu_to_le32(time_sync->active_protocols);
 ether_addr_copy(cmd.peer_addr, time_sync->peer_addr);

 err = iwl_mld_send_cmd_pdu(mld,
       WIDE_ID(DATA_PATH_GROUP,
        WNM_80211V_TIMING_MEASUREMENT_CONFIG_CMD),
       &cmd);
 if (err)
  IWL_ERR(mld, "Failed to send time sync cfg cmd: %d\n", err);

 return err;
}

int iwl_mld_time_sync_config(struct iwl_mld *mld, const u8 *addr, u32 protocols)
{
 struct iwl_mld_time_sync_data *time_sync;
 int err;

 time_sync = wiphy_dereference(mld->wiphy, mld->time_sync);

 /* The fw only supports one peer. We do allow reconfiguration of the
 * same peer for cases of fw reset etc.
 */
 if (time_sync && time_sync->active_protocols &&
     !ether_addr_equal(addr, time_sync->peer_addr)) {
  IWL_DEBUG_INFO(mld, "Time sync: reject config for peer: %pM\n",
          addr);
  return -ENOBUFS;
 }

 if (protocols & ~(IWL_TIME_SYNC_PROTOCOL_TM |
     IWL_TIME_SYNC_PROTOCOL_FTM))
  return -EINVAL;

 IWL_DEBUG_INFO(mld, "Time sync: set peer addr=%pM\n", addr);

 iwl_mld_deinit_time_sync(mld);
 err = iwl_mld_init_time_sync(mld, protocols, addr);
 if (err)
  return err;

 err = iwl_mld_time_sync_fw_config(mld);
 return err;
}

void iwl_mld_deinit_time_sync(struct iwl_mld *mld)
{
 struct iwl_mld_time_sync_data *time_sync =
  wiphy_dereference(mld->wiphy, mld->time_sync);

 if (!time_sync)
  return;

 RCU_INIT_POINTER(mld->time_sync, NULL);
 skb_queue_purge(&time_sync->frame_list);
 kfree_rcu(time_sync, rcu_head);
}

bool iwl_mld_time_sync_frame(struct iwl_mld *mld, struct sk_buff *skb, u8 *addr)
{
 struct iwl_mld_time_sync_data *time_sync;

 rcu_read_lock();
 time_sync = rcu_dereference(mld->time_sync);
 if (time_sync && ether_addr_equal(time_sync->peer_addr, addr) &&
     (ieee80211_is_timing_measurement(skb) || ieee80211_is_ftm(skb))) {
  skb_queue_tail(&time_sync->frame_list, skb);
  rcu_read_unlock();
  return true;
 }
 rcu_read_unlock();

 return false;
}

static bool iwl_mld_is_skb_match(struct sk_buff *skb, u8 *addr, u8 dialog_token)
{
 struct ieee80211_mgmt *mgmt = (void *)skb->data;
 u8 skb_dialog_token;

 if (ieee80211_is_timing_measurement(skb))
  skb_dialog_token = mgmt->u.action.u.wnm_timing_msr.dialog_token;
 else
  skb_dialog_token = mgmt->u.action.u.ftm.dialog_token;

 if ((ether_addr_equal(mgmt->sa, addr) ||
      ether_addr_equal(mgmt->da, addr)) &&
     skb_dialog_token == dialog_token)
  return true;

 return false;
}

static struct sk_buff *iwl_mld_time_sync_find_skb(struct iwl_mld *mld, u8 *addr,
        u8 dialog_token)
{
 struct iwl_mld_time_sync_data *time_sync;
 struct sk_buff *skb;

 rcu_read_lock();

 time_sync = rcu_dereference(mld->time_sync);
 if (IWL_FW_CHECK(mld, !time_sync,
    "Time sync notification but time sync is not initialized\n")) {
  rcu_read_unlock();
  return NULL;
 }

 /* The notifications are expected to arrive in the same order of the
 * frames. If the incoming notification doesn't match the first SKB
 * in the queue, it means there was no time sync notification for this
 * SKB and it can be dropped.
 */
 while ((skb = skb_dequeue(&time_sync->frame_list))) {
  if (iwl_mld_is_skb_match(skb, addr, dialog_token))
   break;

  kfree_skb(skb);
  skb = NULL;
  IWL_DEBUG_DROP(mld,
          "Time sync: drop SKB without matching notification\n");
 }
 rcu_read_unlock();

 return skb;
}

static u64 iwl_mld_get_64_bit(__le32 high, __le32 low)
{
 return ((u64)le32_to_cpu(high) << 32) | le32_to_cpu(low);
}

void iwl_mld_handle_time_msmt_notif(struct iwl_mld *mld,
        struct iwl_rx_packet *pkt)
{
 struct ptp_data *data = &mld->ptp_data;
 struct iwl_time_msmt_notify *notif = (void *)pkt->data;
 struct ieee80211_rx_status *rx_status;
 struct skb_shared_hwtstamps *shwt;
 u64 ts_10ns;
 struct sk_buff *skb =
  iwl_mld_time_sync_find_skb(mld, notif->peer_addr,
        le32_to_cpu(notif->dialog_token));
 u64 adj_time;

 if (IWL_FW_CHECK(mld, !skb, "Time sync event but no pending skb\n"))
  return;

 spin_lock_bh(&data->lock);
 ts_10ns = iwl_mld_get_64_bit(notif->t2_hi, notif->t2_lo);
 adj_time = iwl_mld_ptp_get_adj_time(mld, ts_10ns * 10);
 shwt = skb_hwtstamps(skb);
 shwt->hwtstamp = ktime_set(0, adj_time);

 ts_10ns = iwl_mld_get_64_bit(notif->t3_hi, notif->t3_lo);
 adj_time = iwl_mld_ptp_get_adj_time(mld, ts_10ns * 10);
 rx_status = IEEE80211_SKB_RXCB(skb);
 rx_status->ack_tx_hwtstamp = ktime_set(0, adj_time);
 spin_unlock_bh(&data->lock);

 IWL_DEBUG_INFO(mld,
         "Time sync: RX event - report frame t2=%llu t3=%llu\n",
         ktime_to_ns(shwt->hwtstamp),
         ktime_to_ns(rx_status->ack_tx_hwtstamp));
 ieee80211_rx_napi(mld->hw, NULL, skb, NULL);
}

void iwl_mld_handle_time_sync_confirm_notif(struct iwl_mld *mld,
         struct iwl_rx_packet *pkt)
{
 struct ptp_data *data = &mld->ptp_data;
 struct iwl_time_msmt_cfm_notify *notif = (void *)pkt->data;
 struct ieee80211_tx_status status = {};
 struct skb_shared_hwtstamps *shwt;
 u64 ts_10ns, adj_time;

 status.skb =
  iwl_mld_time_sync_find_skb(mld, notif->peer_addr,
        le32_to_cpu(notif->dialog_token));

 if (IWL_FW_CHECK(mld, !status.skb,
    "Time sync confirm but no pending skb\n"))
  return;

 spin_lock_bh(&data->lock);
 ts_10ns = iwl_mld_get_64_bit(notif->t1_hi, notif->t1_lo);
 adj_time = iwl_mld_ptp_get_adj_time(mld, ts_10ns * 10);
 shwt = skb_hwtstamps(status.skb);
 shwt->hwtstamp = ktime_set(0, adj_time);

 ts_10ns = iwl_mld_get_64_bit(notif->t4_hi, notif->t4_lo);
 adj_time = iwl_mld_ptp_get_adj_time(mld, ts_10ns * 10);
 status.info = IEEE80211_SKB_CB(status.skb);
 status.ack_hwtstamp = ktime_set(0, adj_time);
 spin_unlock_bh(&data->lock);

 IWL_DEBUG_INFO(mld,
         "Time sync: TX event - report frame t1=%llu t4=%llu\n",
         ktime_to_ns(shwt->hwtstamp),
         ktime_to_ns(status.ack_hwtstamp));
 ieee80211_tx_status_ext(mld->hw, &status);
}