/* SPDX-License-Identifier: GPL-2.0 */
#include <linux/slab.h>
#include <linux/pci.h>
#include <asm/apicdef.h>
#include <linux/io-64-nonatomic-lo-hi.h>

#include <linux/perf_event.h>
#include "../perf_event.h"

#define UNCORE_PMU_NAME_LEN             32
#define UNCORE_PMU_HRTIMER_INTERVAL     (60LL * NSEC_PER_SEC)
#define UNCORE_SNB_IMC_HRTIMER_INTERVAL (5ULL * NSEC_PER_SEC)

#define UNCORE_FIXED_EVENT              0xff
#define UNCORE_PMC_IDX_MAX_GENERIC      8
#define UNCORE_PMC_IDX_MAX_FIXED        1
#define UNCORE_PMC_IDX_MAX_FREERUNNING  1
#define UNCORE_PMC_IDX_FIXED            UNCORE_PMC_IDX_MAX_GENERIC
#define UNCORE_PMC_IDX_FREERUNNING      (UNCORE_PMC_IDX_FIXED + \
                                        UNCORE_PMC_IDX_MAX_FIXED)
#define UNCORE_PMC_IDX_MAX              (UNCORE_PMC_IDX_FREERUNNING + \
                                        UNCORE_PMC_IDX_MAX_FREERUNNING)

#define UNCORE_PCI_DEV_FULL_DATA(dev, func, type, idx)  \
                ((dev << 24) | (func << 16) | (type << 8) | idx)
#define UNCORE_PCI_DEV_DATA(type, idx)  ((type << 8) | idx)
#define UNCORE_PCI_DEV_DEV(data)        ((data >> 24) & 0xff)
#define UNCORE_PCI_DEV_FUNC(data)       ((data >> 16) & 0xff)
#define UNCORE_PCI_DEV_TYPE(data)       ((data >> 8) & 0xff)
#define UNCORE_PCI_DEV_IDX(data)        (data & 0xff)
#define UNCORE_EXTRA_PCI_DEV            0xff
#define UNCORE_EXTRA_PCI_DEV_MAX        4

#define UNCORE_EVENT_CONSTRAINT(c, n) EVENT_CONSTRAINT(c, n, 0xff)

struct pci_extra_dev {
        struct pci_dev *dev[UNCORE_EXTRA_PCI_DEV_MAX];
};

struct intel_uncore_ops;
struct intel_uncore_pmu;
struct intel_uncore_box;
struct uncore_event_desc;
struct freerunning_counters;
struct intel_uncore_topology;

struct intel_uncore_type {
        const char *name;
        int num_counters;
        int num_boxes;
        int perf_ctr_bits;
        int fixed_ctr_bits;
        int num_freerunning_types;
        int type_id;
        unsigned perf_ctr;
        unsigned event_ctl;
        unsigned event_mask;
        unsigned event_mask_ext;
        unsigned fixed_ctr;
        unsigned fixed_ctl;
        unsigned box_ctl;
        u64 *box_ctls;  /* Unit ctrl addr of the first box of each die */
        union {
                unsigned msr_offset;
                unsigned mmio_offset;
        };
        unsigned mmio_map_size;
        unsigned num_shared_regs:8;
        unsigned single_fixed:1;
        unsigned pair_ctr_ctl:1;
        union {
                unsigned *msr_offsets;
                unsigned *pci_offsets;
                unsigned *mmio_offsets;
        };
        unsigned *box_ids;
        struct event_constraint unconstrainted;
        struct event_constraint *constraints;
        struct intel_uncore_pmu *pmus;
        struct intel_uncore_ops *ops;
        struct uncore_event_desc *event_descs;
        struct freerunning_counters *freerunning;
        const struct attribute_group *attr_groups[4];
        const struct attribute_group **attr_update;
        struct pmu *pmu; /* for custom pmu ops */
        /*
         * Uncore PMU would store relevant platform topology configuration here
         * to identify which platform component each PMON block of that type is
         * supposed to monitor.
         */
        struct intel_uncore_topology *topology;
        /*
         * Optional callbacks for managing mapping of Uncore units to PMONs
         */
        int (*get_topology)(struct intel_uncore_type *type);
        int (*set_mapping)(struct intel_uncore_type *type);
        void (*cleanup_mapping)(struct intel_uncore_type *type);
};

#define pmu_group attr_groups[0]
#define format_group attr_groups[1]
#define events_group attr_groups[2]

struct intel_uncore_ops {
        void (*init_box)(struct intel_uncore_box *);
        void (*exit_box)(struct intel_uncore_box *);
        void (*disable_box)(struct intel_uncore_box *);
        void (*enable_box)(struct intel_uncore_box *);
        void (*disable_event)(struct intel_uncore_box *, struct perf_event *);
        void (*enable_event)(struct intel_uncore_box *, struct perf_event *);
        u64 (*read_counter)(struct intel_uncore_box *, struct perf_event *);
        int (*hw_config)(struct intel_uncore_box *, struct perf_event *);
        struct event_constraint *(*get_constraint)(struct intel_uncore_box *,
                                                   struct perf_event *);
        void (*put_constraint)(struct intel_uncore_box *, struct perf_event *);
};

struct intel_uncore_pmu {
        struct pmu                      pmu;
        char                            name[UNCORE_PMU_NAME_LEN];
        int                             pmu_idx;
        int                             func_id;
        bool                            registered;
        atomic_t                        activeboxes;
        struct intel_uncore_type        *type;
        struct intel_uncore_box         **boxes;
};

struct intel_uncore_extra_reg {
        raw_spinlock_t lock;
        u64 config, config1, config2;
        atomic_t ref;
};

struct intel_uncore_box {
        int dieid;      /* Logical die ID */
        int n_active;   /* number of active events */
        int n_events;
        int cpu;        /* cpu to collect events */
        unsigned long flags;
        atomic_t refcnt;
        struct perf_event *events[UNCORE_PMC_IDX_MAX];
        struct perf_event *event_list[UNCORE_PMC_IDX_MAX];
        struct event_constraint *event_constraint[UNCORE_PMC_IDX_MAX];
        unsigned long active_mask[BITS_TO_LONGS(UNCORE_PMC_IDX_MAX)];
        u64 tags[UNCORE_PMC_IDX_MAX];
        struct pci_dev *pci_dev;
        struct intel_uncore_pmu *pmu;
        u64 hrtimer_duration; /* hrtimer timeout for this box */
        struct hrtimer hrtimer;
        struct list_head list;
        struct list_head active_list;
        void __iomem *io_addr;
        struct intel_uncore_extra_reg shared_regs[];
};

/* CFL uncore 8th cbox MSRs */
#define CFL_UNC_CBO_7_PERFEVTSEL0               0xf70
#define CFL_UNC_CBO_7_PER_CTR0                  0xf76

#define UNCORE_BOX_FLAG_INITIATED               0
/* event config registers are 8-byte apart */
#define UNCORE_BOX_FLAG_CTL_OFFS8               1
/* CFL 8th CBOX has different MSR space */
#define UNCORE_BOX_FLAG_CFL8_CBOX_MSR_OFFS      2

struct uncore_event_desc {
        struct device_attribute attr;
        const char *config;
};

struct freerunning_counters {
        unsigned int counter_base;
        unsigned int counter_offset;
        unsigned int box_offset;
        unsigned int num_counters;
        unsigned int bits;
        unsigned *box_offsets;
};

struct intel_uncore_topology {
        u64 configuration;
        int segment;
};

struct pci2phy_map {
        struct list_head list;
        int segment;
        int pbus_to_dieid[256];
};

struct pci2phy_map *__find_pci2phy_map(int segment);
int uncore_pcibus_to_dieid(struct pci_bus *bus);
int uncore_die_to_segment(int die);

ssize_t uncore_event_show(struct device *dev,
                          struct device_attribute *attr, char *buf);

static inline struct intel_uncore_pmu *dev_to_uncore_pmu(struct device *dev)
{
        return container_of(dev_get_drvdata(dev), struct intel_uncore_pmu, pmu);
}

#define to_device_attribute(n)  container_of(n, struct device_attribute, attr)
#define to_dev_ext_attribute(n) container_of(n, struct dev_ext_attribute, attr)
#define attr_to_ext_attr(n)     to_dev_ext_attribute(to_device_attribute(n))

extern int __uncore_max_dies;
#define uncore_max_dies()       (__uncore_max_dies)

#define INTEL_UNCORE_EVENT_DESC(_name, _config)                 \
{                                                               \
        .attr   = __ATTR(_name, 0444, uncore_event_show, NULL), \
        .config = _config,                                      \
}

#define DEFINE_UNCORE_FORMAT_ATTR(_var, _name, _format)                 \
static ssize_t __uncore_##_var##_show(struct device *dev,               \
                                struct device_attribute *attr,          \
                                char *page)                             \
{                                                                       \
        BUILD_BUG_ON(sizeof(_format) >= PAGE_SIZE);                     \
        return sprintf(page, _format "\n");                             \
}                                                                       \
static struct device_attribute format_attr_##_var =                     \
        __ATTR(_name, 0444, __uncore_##_var##_show, NULL)

static inline bool uncore_pmc_fixed(int idx)
{
        return idx == UNCORE_PMC_IDX_FIXED;
}

static inline bool uncore_pmc_freerunning(int idx)
{
        return idx == UNCORE_PMC_IDX_FREERUNNING;
}

static inline bool uncore_mmio_is_valid_offset(struct intel_uncore_box *box,
                                               unsigned long offset)
{
        if (offset < box->pmu->type->mmio_map_size)
                return true;

        pr_warn_once("perf uncore: Invalid offset 0x%lx exceeds mapped area of %s.\n",
                     offset, box->pmu->type->name);

        return false;
}

static inline
unsigned int uncore_mmio_box_ctl(struct intel_uncore_box *box)
{
        return box->pmu->type->box_ctl +
               box->pmu->type->mmio_offset * box->pmu->pmu_idx;
}

static inline unsigned uncore_pci_box_ctl(struct intel_uncore_box *box)
{
        return box->pmu->type->box_ctl;
}

static inline unsigned uncore_pci_fixed_ctl(struct intel_uncore_box *box)
{
        return box->pmu->type->fixed_ctl;
}

static inline unsigned uncore_pci_fixed_ctr(struct intel_uncore_box *box)
{
        return box->pmu->type->fixed_ctr;
}

static inline
unsigned uncore_pci_event_ctl(struct intel_uncore_box *box, int idx)
{
        if (test_bit(UNCORE_BOX_FLAG_CTL_OFFS8, &box->flags))
                return idx * 8 + box->pmu->type->event_ctl;

        return idx * 4 + box->pmu->type->event_ctl;
}

static inline
unsigned uncore_pci_perf_ctr(struct intel_uncore_box *box, int idx)
{
        return idx * 8 + box->pmu->type->perf_ctr;
}

static inline unsigned uncore_msr_box_offset(struct intel_uncore_box *box)
{
        struct intel_uncore_pmu *pmu = box->pmu;
        return pmu->type->msr_offsets ?
                pmu->type->msr_offsets[pmu->pmu_idx] :
                pmu->type->msr_offset * pmu->pmu_idx;
}

static inline unsigned uncore_msr_box_ctl(struct intel_uncore_box *box)
{
        if (!box->pmu->type->box_ctl)
                return 0;
        return box->pmu->type->box_ctl + uncore_msr_box_offset(box);
}

static inline unsigned uncore_msr_fixed_ctl(struct intel_uncore_box *box)
{
        if (!box->pmu->type->fixed_ctl)
                return 0;
        return box->pmu->type->fixed_ctl + uncore_msr_box_offset(box);
}

static inline unsigned uncore_msr_fixed_ctr(struct intel_uncore_box *box)
{
        return box->pmu->type->fixed_ctr + uncore_msr_box_offset(box);
}


/*
 * In the uncore document, there is no event-code assigned to free running
 * counters. Some events need to be defined to indicate the free running
 * counters. The events are encoded as event-code + umask-code.
 *
 * The event-code for all free running counters is 0xff, which is the same as
 * the fixed counters.
 *
 * The umask-code is used to distinguish a fixed counter and a free running
 * counter, and different types of free running counters.
 * - For fixed counters, the umask-code is 0x0X.
 *   X indicates the index of the fixed counter, which starts from 0.
 * - For free running counters, the umask-code uses the rest of the space.
 *   It would bare the format of 0xXY.
 *   X stands for the type of free running counters, which starts from 1.
 *   Y stands for the index of free running counters of same type, which
 *   starts from 0.
 *
 * For example, there are three types of IIO free running counters on Skylake
 * server, IO CLOCKS counters, BANDWIDTH counters and UTILIZATION counters.
 * The event-code for all the free running counters is 0xff.
 * 'ioclk' is the first counter of IO CLOCKS. IO CLOCKS is the first type,
 * which umask-code starts from 0x10.
 * So 'ioclk' is encoded as event=0xff,umask=0x10
 * 'bw_in_port2' is the third counter of BANDWIDTH counters. BANDWIDTH is
 * the second type, which umask-code starts from 0x20.
 * So 'bw_in_port2' is encoded as event=0xff,umask=0x22
 */
static inline unsigned int uncore_freerunning_idx(u64 config)
{
        return ((config >> 8) & 0xf);
}

#define UNCORE_FREERUNNING_UMASK_START          0x10

static inline unsigned int uncore_freerunning_type(u64 config)
{
        return ((((config >> 8) - UNCORE_FREERUNNING_UMASK_START) >> 4) & 0xf);
}

static inline
unsigned int uncore_freerunning_counter(struct intel_uncore_box *box,
                                        struct perf_event *event)
{
        unsigned int type = uncore_freerunning_type(event->hw.config);
        unsigned int idx = uncore_freerunning_idx(event->hw.config);
        struct intel_uncore_pmu *pmu = box->pmu;

        return pmu->type->freerunning[type].counter_base +
               pmu->type->freerunning[type].counter_offset * idx +
               (pmu->type->freerunning[type].box_offsets ?
                pmu->type->freerunning[type].box_offsets[pmu->pmu_idx] :
                pmu->type->freerunning[type].box_offset * pmu->pmu_idx);
}

static inline
unsigned uncore_msr_event_ctl(struct intel_uncore_box *box, int idx)
{
        if (test_bit(UNCORE_BOX_FLAG_CFL8_CBOX_MSR_OFFS, &box->flags)) {
                return CFL_UNC_CBO_7_PERFEVTSEL0 +
                       (box->pmu->type->pair_ctr_ctl ? 2 * idx : idx);
        } else {
                return box->pmu->type->event_ctl +
                       (box->pmu->type->pair_ctr_ctl ? 2 * idx : idx) +
                       uncore_msr_box_offset(box);
        }
}

static inline
unsigned uncore_msr_perf_ctr(struct intel_uncore_box *box, int idx)
{
        if (test_bit(UNCORE_BOX_FLAG_CFL8_CBOX_MSR_OFFS, &box->flags)) {
                return CFL_UNC_CBO_7_PER_CTR0 +
                       (box->pmu->type->pair_ctr_ctl ? 2 * idx : idx);
        } else {
                return box->pmu->type->perf_ctr +
                       (box->pmu->type->pair_ctr_ctl ? 2 * idx : idx) +
                       uncore_msr_box_offset(box);
        }
}

static inline
unsigned uncore_fixed_ctl(struct intel_uncore_box *box)
{
        if (box->pci_dev || box->io_addr)
                return uncore_pci_fixed_ctl(box);
        else
                return uncore_msr_fixed_ctl(box);
}

static inline
unsigned uncore_fixed_ctr(struct intel_uncore_box *box)
{
        if (box->pci_dev || box->io_addr)
                return uncore_pci_fixed_ctr(box);
        else
                return uncore_msr_fixed_ctr(box);
}

static inline
unsigned uncore_event_ctl(struct intel_uncore_box *box, int idx)
{
        if (box->pci_dev || box->io_addr)
                return uncore_pci_event_ctl(box, idx);
        else
                return uncore_msr_event_ctl(box, idx);
}

static inline
unsigned uncore_perf_ctr(struct intel_uncore_box *box, int idx)
{
        if (box->pci_dev || box->io_addr)
                return uncore_pci_perf_ctr(box, idx);
        else
                return uncore_msr_perf_ctr(box, idx);
}

static inline int uncore_perf_ctr_bits(struct intel_uncore_box *box)
{
        return box->pmu->type->perf_ctr_bits;
}

static inline int uncore_fixed_ctr_bits(struct intel_uncore_box *box)
{
        return box->pmu->type->fixed_ctr_bits;
}

static inline
unsigned int uncore_freerunning_bits(struct intel_uncore_box *box,
                                     struct perf_event *event)
{
        unsigned int type = uncore_freerunning_type(event->hw.config);

        return box->pmu->type->freerunning[type].bits;
}

static inline int uncore_num_freerunning(struct intel_uncore_box *box,
                                         struct perf_event *event)
{
        unsigned int type = uncore_freerunning_type(event->hw.config);

        return box->pmu->type->freerunning[type].num_counters;
}

static inline int uncore_num_freerunning_types(struct intel_uncore_box *box,
                                               struct perf_event *event)
{
        return box->pmu->type->num_freerunning_types;
}

static inline bool check_valid_freerunning_event(struct intel_uncore_box *box,
                                                 struct perf_event *event)
{
        unsigned int type = uncore_freerunning_type(event->hw.config);
        unsigned int idx = uncore_freerunning_idx(event->hw.config);

        return (type < uncore_num_freerunning_types(box, event)) &&
               (idx < uncore_num_freerunning(box, event));
}

static inline int uncore_num_counters(struct intel_uncore_box *box)
{
        return box->pmu->type->num_counters;
}

static inline bool is_freerunning_event(struct perf_event *event)
{
        u64 cfg = event->attr.config;

        return ((cfg & UNCORE_FIXED_EVENT) == UNCORE_FIXED_EVENT) &&
               (((cfg >> 8) & 0xff) >= UNCORE_FREERUNNING_UMASK_START);
}

/* Check and reject invalid config */
static inline int uncore_freerunning_hw_config(struct intel_uncore_box *box,
                                               struct perf_event *event)
{
        if (is_freerunning_event(event))
                return 0;

        return -EINVAL;
}

static inline void uncore_disable_event(struct intel_uncore_box *box,
                                struct perf_event *event)
{
        box->pmu->type->ops->disable_event(box, event);
}

static inline void uncore_enable_event(struct intel_uncore_box *box,
                                struct perf_event *event)
{
        box->pmu->type->ops->enable_event(box, event);
}

static inline u64 uncore_read_counter(struct intel_uncore_box *box,
                                struct perf_event *event)
{
        return box->pmu->type->ops->read_counter(box, event);
}

static inline void uncore_box_init(struct intel_uncore_box *box)
{
        if (!test_and_set_bit(UNCORE_BOX_FLAG_INITIATED, &box->flags)) {
                if (box->pmu->type->ops->init_box)
                        box->pmu->type->ops->init_box(box);
        }
}

static inline void uncore_box_exit(struct intel_uncore_box *box)
{
        if (test_and_clear_bit(UNCORE_BOX_FLAG_INITIATED, &box->flags)) {
                if (box->pmu->type->ops->exit_box)
                        box->pmu->type->ops->exit_box(box);
        }
}

static inline bool uncore_box_is_fake(struct intel_uncore_box *box)
{
        return (box->dieid < 0);
}

static inline struct intel_uncore_pmu *uncore_event_to_pmu(struct perf_event *event)
{
        return container_of(event->pmu, struct intel_uncore_pmu, pmu);
}

static inline struct intel_uncore_box *uncore_event_to_box(struct perf_event *event)
{
        return event->pmu_private;
}

struct intel_uncore_box *uncore_pmu_to_box(struct intel_uncore_pmu *pmu, int cpu);
u64 uncore_msr_read_counter(struct intel_uncore_box *box, struct perf_event *event);
void uncore_mmio_exit_box(struct intel_uncore_box *box);
u64 uncore_mmio_read_counter(struct intel_uncore_box *box,
                             struct perf_event *event);
void uncore_pmu_start_hrtimer(struct intel_uncore_box *box);
void uncore_pmu_cancel_hrtimer(struct intel_uncore_box *box);
void uncore_pmu_event_start(struct perf_event *event, int flags);
void uncore_pmu_event_stop(struct perf_event *event, int flags);
int uncore_pmu_event_add(struct perf_event *event, int flags);
void uncore_pmu_event_del(struct perf_event *event, int flags);
void uncore_pmu_event_read(struct perf_event *event);
void uncore_perf_event_update(struct intel_uncore_box *box, struct perf_event *event);
struct event_constraint *
uncore_get_constraint(struct intel_uncore_box *box, struct perf_event *event);
void uncore_put_constraint(struct intel_uncore_box *box, struct perf_event *event);
u64 uncore_shared_reg_config(struct intel_uncore_box *box, int idx);

extern struct intel_uncore_type *empty_uncore[];
extern struct intel_uncore_type **uncore_msr_uncores;
extern struct intel_uncore_type **uncore_pci_uncores;
extern struct intel_uncore_type **uncore_mmio_uncores;
extern struct pci_driver *uncore_pci_driver;
extern struct pci_driver *uncore_pci_sub_driver;
extern raw_spinlock_t pci2phy_map_lock;
extern struct list_head pci2phy_map_head;
extern struct pci_extra_dev *uncore_extra_pci_dev;
extern struct event_constraint uncore_constraint_empty;

/* uncore_snb.c */
int snb_uncore_pci_init(void);
int ivb_uncore_pci_init(void);
int hsw_uncore_pci_init(void);
int bdw_uncore_pci_init(void);
int skl_uncore_pci_init(void);
void snb_uncore_cpu_init(void);
void nhm_uncore_cpu_init(void);
void skl_uncore_cpu_init(void);
void icl_uncore_cpu_init(void);
void adl_uncore_cpu_init(void);
void tgl_uncore_cpu_init(void);
void tgl_uncore_mmio_init(void);
void tgl_l_uncore_mmio_init(void);
int snb_pci2phy_map_init(int devid);

/* uncore_snbep.c */
int snbep_uncore_pci_init(void);
void snbep_uncore_cpu_init(void);
int ivbep_uncore_pci_init(void);
void ivbep_uncore_cpu_init(void);
int hswep_uncore_pci_init(void);
void hswep_uncore_cpu_init(void);
int bdx_uncore_pci_init(void);
void bdx_uncore_cpu_init(void);
int knl_uncore_pci_init(void);
void knl_uncore_cpu_init(void);
int skx_uncore_pci_init(void);
void skx_uncore_cpu_init(void);
int snr_uncore_pci_init(void);
void snr_uncore_cpu_init(void);
void snr_uncore_mmio_init(void);
int icx_uncore_pci_init(void);
void icx_uncore_cpu_init(void);
void icx_uncore_mmio_init(void);

/* uncore_nhmex.c */
void nhmex_uncore_cpu_init(void);