// SPDX-License-Identifier: MIT
/*
 * Copyright © 2023-2024 Intel Corporation
 */

#include "abi/guc_actions_sriov_abi.h"
#include "abi/guc_messages_abi.h"

#include "xe_gt_sriov_pf_config.h"
#include "xe_gt_sriov_pf_helpers.h"
#include "xe_gt_sriov_pf_monitor.h"
#include "xe_gt_sriov_printk.h"
#include "xe_guc_klv_helpers.h"
#include "xe_guc_klv_thresholds_set.h"

/**
 * xe_gt_sriov_pf_monitor_flr - Cleanup VF data after VF FLR.
 * @gt: the &xe_gt
 * @vfid: the VF identifier
 *
 * On FLR this function will reset all event data related to the VF.
 * This function is for PF only.
 */
void xe_gt_sriov_pf_monitor_flr(struct xe_gt *gt, u32 vfid)
{
 int e;

 xe_gt_assert(gt, IS_SRIOV_PF(gt_to_xe(gt)));
 xe_gt_sriov_pf_assert_vfid(gt, vfid);

 for (e = 0; e < XE_GUC_KLV_NUM_THRESHOLDS; e++)
  gt->sriov.pf.vfs[vfid].monitor.guc.events[e] = 0;
}

static void pf_update_event_counter(struct xe_gt *gt, u32 vfid,
        enum xe_guc_klv_threshold_index e)
{
 xe_gt_assert(gt, IS_SRIOV_PF(gt_to_xe(gt)));
 xe_gt_assert(gt, e < XE_GUC_KLV_NUM_THRESHOLDS);

 gt->sriov.pf.vfs[vfid].monitor.guc.events[e]++;
}

static int pf_handle_vf_threshold_event(struct xe_gt *gt, u32 vfid, u32 threshold)
{
 char origin[8];
 int e;

 e = xe_guc_klv_threshold_key_to_index(threshold);
 xe_sriov_function_name(vfid, origin, sizeof(origin));

 /* was there a new KEY added that we missed? */
 if (unlikely(e < 0)) {
  xe_gt_sriov_notice(gt, "unknown threshold key %#x reported for %s\n",
       threshold, origin);
  return -ENOTCONN;
 }

 xe_gt_sriov_dbg(gt, "%s exceeded threshold %u %s\n",
   origin, xe_gt_sriov_pf_config_get_threshold(gt, vfid, e),
   xe_guc_klv_key_to_string(threshold));

 pf_update_event_counter(gt, vfid, e);

 return 0;
}

/**
 * xe_gt_sriov_pf_monitor_process_guc2pf - Handle adverse event notification from the GuC.
 * @gt: the &xe_gt
 * @msg: G2H event message
 * @len: length of the message
 *
 * This function is intended for PF only.
 *
 * Return: 0 on success or a negative error code on failure.
 */
int xe_gt_sriov_pf_monitor_process_guc2pf(struct xe_gt *gt, const u32 *msg, u32 len)
{
 struct xe_device *xe = gt_to_xe(gt);
 u32 vfid;
 u32 threshold;

 xe_gt_assert(gt, len >= GUC_HXG_MSG_MIN_LEN);
 xe_gt_assert(gt, FIELD_GET(GUC_HXG_MSG_0_ORIGIN, msg[0]) == GUC_HXG_ORIGIN_GUC);
 xe_gt_assert(gt, FIELD_GET(GUC_HXG_MSG_0_TYPE, msg[0]) == GUC_HXG_TYPE_EVENT);
 xe_gt_assert(gt, FIELD_GET(GUC_HXG_EVENT_MSG_0_ACTION, msg[0]) ==
       GUC_ACTION_GUC2PF_ADVERSE_EVENT);

 if (unlikely(!IS_SRIOV_PF(xe)))
  return -EPROTO;

 if (unlikely(FIELD_GET(GUC2PF_ADVERSE_EVENT_EVENT_MSG_0_MBZ, msg[0])))
  return -EPFNOSUPPORT;

 if (unlikely(len < GUC2PF_ADVERSE_EVENT_EVENT_MSG_LEN))
  return -EPROTO;

 vfid = FIELD_GET(GUC2PF_ADVERSE_EVENT_EVENT_MSG_1_VFID, msg[1]);
 threshold = FIELD_GET(GUC2PF_ADVERSE_EVENT_EVENT_MSG_2_THRESHOLD, msg[2]);

 if (unlikely(vfid > xe_gt_sriov_pf_get_totalvfs(gt)))
  return -EINVAL;

 return pf_handle_vf_threshold_event(gt, vfid, threshold);
}

/**
 * xe_gt_sriov_pf_monitor_print_events - Print adverse events counters.
 * @gt: the &xe_gt to print events from
 * @p: the &drm_printer
 *
 * Print adverse events counters for all VFs.
 * VFs with no events are not printed.
 *
 * This function can only be called on PF.
 */
void xe_gt_sriov_pf_monitor_print_events(struct xe_gt *gt, struct drm_printer *p)
{
 unsigned int n, total_vfs = xe_gt_sriov_pf_get_totalvfs(gt);
 const struct xe_gt_sriov_monitor *data;
 int e;

 xe_gt_assert(gt, IS_SRIOV_PF(gt_to_xe(gt)));

 for (n = 1; n <= total_vfs; n++) {
  data = >->sriov.pf.vfs[n].monitor;

  for (e = 0; e < XE_GUC_KLV_NUM_THRESHOLDS; e++)
   if (data->guc.events[e])
    break;

  /* skip empty unless in debug mode */
  if (e >= XE_GUC_KLV_NUM_THRESHOLDS &&
      !IS_ENABLED(CONFIG_DRM_XE_DEBUG_SRIOV))
   continue;

#define __format(...) "%s:%u "
#define __value(TAG, NAME, ...) , #NAME, data->guc.events[MAKE_XE_GUC_KLV_THRESHOLD_INDEX(TAG)]

  drm_printf(p, "VF%u:\t" MAKE_XE_GUC_KLV_THRESHOLDS_SET(__format) "\n",
      n MAKE_XE_GUC_KLV_THRESHOLDS_SET(__value));

#undef __format
#undef __value
 }
}