// SPDX-License-Identifier: GPL-2.0
/*
 * PCIe TLP Log handling
 *
 * Copyright (C) 2024 Intel Corporation
 */

#include <linux/aer.h>
#include <linux/array_size.h>
#include <linux/bitfield.h>
#include <linux/pci.h>
#include <linux/string.h>

#include "../pci.h"

/**
 * aer_tlp_log_len - Calculate AER Capability TLP Header/Prefix Log length
 * @dev: PCIe device
 * @aercc: AER Capabilities and Control register value
 *
 * Return: TLP Header/Prefix Log length
 */
unsigned int aer_tlp_log_len(struct pci_dev *dev, u32 aercc)
{
 if (aercc & PCI_ERR_CAP_TLP_LOG_FLIT)
  return FIELD_GET(PCI_ERR_CAP_TLP_LOG_SIZE, aercc);

 return PCIE_STD_NUM_TLP_HEADERLOG +
        ((aercc & PCI_ERR_CAP_PREFIX_LOG_PRESENT) ?
  dev->eetlp_prefix_max : 0);
}

#ifdef CONFIG_PCIE_DPC
/**
 * dpc_tlp_log_len - Calculate DPC RP PIO TLP Header/Prefix Log length
 * @dev: PCIe device
 *
 * Return: TLP Header/Prefix Log length
 */
unsigned int dpc_tlp_log_len(struct pci_dev *dev)
{
 /* Remove ImpSpec Log register from the count */
 if (dev->dpc_rp_log_size >= PCIE_STD_NUM_TLP_HEADERLOG + 1)
  return dev->dpc_rp_log_size - 1;

 return dev->dpc_rp_log_size;
}
#endif

/**
 * pcie_read_tlp_log - read TLP Header Log
 * @dev: PCIe device
 * @where: PCI Config offset of TLP Header Log
 * @where2: PCI Config offset of TLP Prefix Log
 * @tlp_len: TLP Log length (Header Log + TLP Prefix Log in DWORDs)
 * @flit: TLP Logged in Flit mode
 * @log: TLP Log structure to fill
 *
 * Fill @log from TLP Header Log registers, e.g., AER or DPC.
 *
 * Return: 0 on success and filled TLP Log structure, <0 on error.
 */
int pcie_read_tlp_log(struct pci_dev *dev, int where, int where2,
        unsigned int tlp_len, bool flit, struct pcie_tlp_log *log)
{
 unsigned int i;
 int off, ret;

 if (tlp_len > ARRAY_SIZE(log->dw))
  tlp_len = ARRAY_SIZE(log->dw);

 memset(log, 0, sizeof(*log));

 for (i = 0; i < tlp_len; i++) {
  if (i < PCIE_STD_NUM_TLP_HEADERLOG)
   off = where + i * 4;
  else
   off = where2 + (i - PCIE_STD_NUM_TLP_HEADERLOG) * 4;

  ret = pci_read_config_dword(dev, off, &log->dw[i]);
  if (ret)
   return pcibios_err_to_errno(ret);
 }

 /*
  * Hard-code non-Flit mode to 4 DWORDs, for now. The exact length
  * can only be known if the TLP is parsed.
 */
 log->header_len = flit ? tlp_len : 4;
 log->flit = flit;

 return 0;
}

#define EE_PREFIX_STR " E-E Prefixes:"

/**
 * pcie_print_tlp_log - Print TLP Header / Prefix Log contents
 * @dev: PCIe device
 * @log: TLP Log structure
 * @level: Printk log level
 * @pfx: String prefix
 *
 * Prints TLP Header and Prefix Log information held by @log.
 */
void pcie_print_tlp_log(const struct pci_dev *dev,
   const struct pcie_tlp_log *log, const char *level,
   const char *pfx)
{
 /* EE_PREFIX_STR fits the extended DW space needed for the Flit mode */
 char buf[11 * PCIE_STD_MAX_TLP_HEADERLOG + 1];
 unsigned int i;
 int len;

 len = scnprintf(buf, sizeof(buf), "%#010x %#010x %#010x %#010x",
   log->dw[0], log->dw[1], log->dw[2], log->dw[3]);

 if (log->flit) {
  for (i = PCIE_STD_NUM_TLP_HEADERLOG; i < log->header_len; i++) {
   len += scnprintf(buf + len, sizeof(buf) - len,
      " %#010x", log->dw[i]);
  }
 } else {
  if (log->prefix[0])
   len += scnprintf(buf + len, sizeof(buf) - len,
      EE_PREFIX_STR);
  for (i = 0; i < ARRAY_SIZE(log->prefix); i++) {
   if (!log->prefix[i])
    break;
   len += scnprintf(buf + len, sizeof(buf) - len,
      " %#010x", log->prefix[i]);
  }
 }

 dev_printk(level, &dev->dev, "%sTLP Header%s: %s\n", pfx,
  log->flit ? " (Flit)" : "", buf);
}