Quelle  arm_brbe.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0-only
/*
 * Branch Record Buffer Extension Driver.
 *
 * Copyright (C) 2022-2025 ARM Limited
 *
 * Author: Anshuman Khandual <anshuman.khandual@arm.com>
 */
#include <linux/types.h>
#include <linux/bitmap.h>
#include <linux/perf/arm_pmu.h>
#include "arm_brbe.h"

#define BRBFCR_EL1_BRANCH_FILTERS (BRBFCR_EL1_DIRECT   | \
       BRBFCR_EL1_INDIRECT | \
       BRBFCR_EL1_RTN      | \
       BRBFCR_EL1_INDCALL  | \
       BRBFCR_EL1_DIRCALL  | \
       BRBFCR_EL1_CONDDIR)

/*
 * BRBTS_EL1 is currently not used for branch stack implementation
 * purpose but BRBCR_ELx.TS needs to have a valid value from all
 * available options. BRBCR_ELx_TS_VIRTUAL is selected for this.
 */
#define BRBCR_ELx_DEFAULT_TS      FIELD_PREP(BRBCR_ELx_TS_MASK, BRBCR_ELx_TS_VIRTUAL)

/*
 * BRBE Buffer Organization
 *
 * BRBE buffer is arranged as multiple banks of 32 branch record
 * entries each. An individual branch record in a given bank could
 * be accessed, after selecting the bank in BRBFCR_EL1.BANK and
 * accessing the registers i.e [BRBSRC, BRBTGT, BRBINF] set with
 * indices [0..31].
 *
 * Bank 0
 *
 * --------------------------------- ------
 * | 00 | BRBSRC | BRBTGT | BRBINF | | 00 |
 * --------------------------------- ------
 * | 01 | BRBSRC | BRBTGT | BRBINF | | 01 |
 * --------------------------------- ------
 * | .. | BRBSRC | BRBTGT | BRBINF | | .. |
 * --------------------------------- ------
 * | 31 | BRBSRC | BRBTGT | BRBINF | | 31 |
 * --------------------------------- ------
 *
 * Bank 1
 *
 * --------------------------------- ------
 * | 32 | BRBSRC | BRBTGT | BRBINF | | 00 |
 * --------------------------------- ------
 * | 33 | BRBSRC | BRBTGT | BRBINF | | 01 |
 * --------------------------------- ------
 * | .. | BRBSRC | BRBTGT | BRBINF | | .. |
 * --------------------------------- ------
 * | 63 | BRBSRC | BRBTGT | BRBINF | | 31 |
 * --------------------------------- ------
 */
#define BRBE_BANK_MAX_ENTRIES 32

struct brbe_regset {
 u64 brbsrc;
 u64 brbtgt;
 u64 brbinf;
};

#define PERF_BR_ARM64_MAX (PERF_BR_MAX + PERF_BR_NEW_MAX)

struct brbe_hw_attr {
 int brbe_version;
 int brbe_cc;
 int brbe_nr;
 int brbe_format;
};

#define BRBE_REGN_CASE(n, case_macro) \
 case n: case_macro(n); break

#define BRBE_REGN_SWITCH(x, case_macro)    \
 do {       \
  switch (x) {     \
  BRBE_REGN_CASE(0, case_macro);   \
  BRBE_REGN_CASE(1, case_macro);   \
  BRBE_REGN_CASE(2, case_macro);   \
  BRBE_REGN_CASE(3, case_macro);   \
  BRBE_REGN_CASE(4, case_macro);   \
  BRBE_REGN_CASE(5, case_macro);   \
  BRBE_REGN_CASE(6, case_macro);   \
  BRBE_REGN_CASE(7, case_macro);   \
  BRBE_REGN_CASE(8, case_macro);   \
  BRBE_REGN_CASE(9, case_macro);   \
  BRBE_REGN_CASE(10, case_macro);   \
  BRBE_REGN_CASE(11, case_macro);   \
  BRBE_REGN_CASE(12, case_macro);   \
  BRBE_REGN_CASE(13, case_macro);   \
  BRBE_REGN_CASE(14, case_macro);   \
  BRBE_REGN_CASE(15, case_macro);   \
  BRBE_REGN_CASE(16, case_macro);   \
  BRBE_REGN_CASE(17, case_macro);   \
  BRBE_REGN_CASE(18, case_macro);   \
  BRBE_REGN_CASE(19, case_macro);   \
  BRBE_REGN_CASE(20, case_macro);   \
  BRBE_REGN_CASE(21, case_macro);   \
  BRBE_REGN_CASE(22, case_macro);   \
  BRBE_REGN_CASE(23, case_macro);   \
  BRBE_REGN_CASE(24, case_macro);   \
  BRBE_REGN_CASE(25, case_macro);   \
  BRBE_REGN_CASE(26, case_macro);   \
  BRBE_REGN_CASE(27, case_macro);   \
  BRBE_REGN_CASE(28, case_macro);   \
  BRBE_REGN_CASE(29, case_macro);   \
  BRBE_REGN_CASE(30, case_macro);   \
  BRBE_REGN_CASE(31, case_macro);   \
  default: WARN(1, "Invalid BRB* index %d\n", x); \
  }      \
 } while (0)

#define RETURN_READ_BRBSRCN(n) \
 return read_sysreg_s(SYS_BRBSRC_EL1(n))
static inline u64 get_brbsrc_reg(int idx)
{
 BRBE_REGN_SWITCH(idx, RETURN_READ_BRBSRCN);
 return 0;
}

#define RETURN_READ_BRBTGTN(n) \
 return read_sysreg_s(SYS_BRBTGT_EL1(n))
static u64 get_brbtgt_reg(int idx)
{
 BRBE_REGN_SWITCH(idx, RETURN_READ_BRBTGTN);
 return 0;
}

#define RETURN_READ_BRBINFN(n) \
 return read_sysreg_s(SYS_BRBINF_EL1(n))
static u64 get_brbinf_reg(int idx)
{
 BRBE_REGN_SWITCH(idx, RETURN_READ_BRBINFN);
 return 0;
}

static u64 brbe_record_valid(u64 brbinf)
{
 return FIELD_GET(BRBINFx_EL1_VALID_MASK, brbinf);
}

static bool brbe_invalid(u64 brbinf)
{
 return brbe_record_valid(brbinf) == BRBINFx_EL1_VALID_NONE;
}

static bool brbe_record_is_complete(u64 brbinf)
{
 return brbe_record_valid(brbinf) == BRBINFx_EL1_VALID_FULL;
}

static bool brbe_record_is_source_only(u64 brbinf)
{
 return brbe_record_valid(brbinf) == BRBINFx_EL1_VALID_SOURCE;
}

static bool brbe_record_is_target_only(u64 brbinf)
{
 return brbe_record_valid(brbinf) == BRBINFx_EL1_VALID_TARGET;
}

static int brbinf_get_in_tx(u64 brbinf)
{
 return FIELD_GET(BRBINFx_EL1_T_MASK, brbinf);
}

static int brbinf_get_mispredict(u64 brbinf)
{
 return FIELD_GET(BRBINFx_EL1_MPRED_MASK, brbinf);
}

static int brbinf_get_lastfailed(u64 brbinf)
{
 return FIELD_GET(BRBINFx_EL1_LASTFAILED_MASK, brbinf);
}

static u16 brbinf_get_cycles(u64 brbinf)
{
 u32 exp, mant, cycles;
 /*
 * Captured cycle count is unknown and hence
 * should not be passed on to userspace.
 */
 if (brbinf & BRBINFx_EL1_CCU)
  return 0;

 exp = FIELD_GET(BRBINFx_EL1_CC_EXP_MASK, brbinf);
 mant = FIELD_GET(BRBINFx_EL1_CC_MANT_MASK, brbinf);

 if (!exp)
  return mant;

 cycles = (mant | 0x100) << (exp - 1);

 return min(cycles, U16_MAX);
}

static int brbinf_get_type(u64 brbinf)
{
 return FIELD_GET(BRBINFx_EL1_TYPE_MASK, brbinf);
}

static int brbinf_get_el(u64 brbinf)
{
 return FIELD_GET(BRBINFx_EL1_EL_MASK, brbinf);
}

void brbe_invalidate(void)
{
 /* Ensure all branches before this point are recorded */
 isb();
 asm volatile(BRB_IALL_INSN);
 /* Ensure all branch records are invalidated after this point */
 isb();
}

static bool valid_brbe_nr(int brbe_nr)
{
 return brbe_nr == BRBIDR0_EL1_NUMREC_8 ||
        brbe_nr == BRBIDR0_EL1_NUMREC_16 ||
        brbe_nr == BRBIDR0_EL1_NUMREC_32 ||
        brbe_nr == BRBIDR0_EL1_NUMREC_64;
}

static bool valid_brbe_cc(int brbe_cc)
{
 return brbe_cc == BRBIDR0_EL1_CC_20_BIT;
}

static bool valid_brbe_format(int brbe_format)
{
 return brbe_format == BRBIDR0_EL1_FORMAT_FORMAT_0;
}

static bool valid_brbidr(u64 brbidr)
{
 int brbe_format, brbe_cc, brbe_nr;

 brbe_format = FIELD_GET(BRBIDR0_EL1_FORMAT_MASK, brbidr);
 brbe_cc = FIELD_GET(BRBIDR0_EL1_CC_MASK, brbidr);
 brbe_nr = FIELD_GET(BRBIDR0_EL1_NUMREC_MASK, brbidr);

 return valid_brbe_format(brbe_format) && valid_brbe_cc(brbe_cc) && valid_brbe_nr(brbe_nr);
}

static bool valid_brbe_version(int brbe_version)
{
 return brbe_version == ID_AA64DFR0_EL1_BRBE_IMP ||
        brbe_version == ID_AA64DFR0_EL1_BRBE_BRBE_V1P1;
}

static void select_brbe_bank(int bank)
{
 u64 brbfcr;

 brbfcr = read_sysreg_s(SYS_BRBFCR_EL1);
 brbfcr &= ~BRBFCR_EL1_BANK_MASK;
 brbfcr |= SYS_FIELD_PREP(BRBFCR_EL1, BANK, bank);
 write_sysreg_s(brbfcr, SYS_BRBFCR_EL1);
 /*
 * Arm ARM (DDI 0487K.a) D.18.4 rule PPBZP requires explicit sync
 * between setting BANK and accessing branch records.
 */
 isb();
}

static bool __read_brbe_regset(struct brbe_regset *entry, int idx)
{
 entry->brbinf = get_brbinf_reg(idx);

 if (brbe_invalid(entry->brbinf))
  return false;

 entry->brbsrc = get_brbsrc_reg(idx);
 entry->brbtgt = get_brbtgt_reg(idx);
 return true;
}

/*
 * Generic perf branch filters supported on BRBE
 *
 * New branch filters need to be evaluated whether they could be supported on
 * BRBE. This ensures that such branch filters would not just be accepted, to
 * fail silently. PERF_SAMPLE_BRANCH_HV is a special case that is selectively
 * supported only on platforms where kernel is in hyp mode.
 */
#define BRBE_EXCLUDE_BRANCH_FILTERS (PERF_SAMPLE_BRANCH_ABORT_TX | \
         PERF_SAMPLE_BRANCH_IN_TX  | \
         PERF_SAMPLE_BRANCH_NO_TX  | \
         PERF_SAMPLE_BRANCH_CALL_STACK | \
         PERF_SAMPLE_BRANCH_COUNTERS)

#define BRBE_ALLOWED_BRANCH_TYPES   (PERF_SAMPLE_BRANCH_ANY  | \
         PERF_SAMPLE_BRANCH_ANY_CALL | \
         PERF_SAMPLE_BRANCH_ANY_RETURN | \
         PERF_SAMPLE_BRANCH_IND_CALL | \
         PERF_SAMPLE_BRANCH_COND  | \
         PERF_SAMPLE_BRANCH_IND_JUMP | \
         PERF_SAMPLE_BRANCH_CALL)


#define BRBE_ALLOWED_BRANCH_FILTERS (PERF_SAMPLE_BRANCH_USER  | \
         PERF_SAMPLE_BRANCH_KERNEL  | \
         PERF_SAMPLE_BRANCH_HV  | \
         BRBE_ALLOWED_BRANCH_TYPES  | \
         PERF_SAMPLE_BRANCH_NO_FLAGS | \
         PERF_SAMPLE_BRANCH_NO_CYCLES | \
         PERF_SAMPLE_BRANCH_TYPE_SAVE | \
         PERF_SAMPLE_BRANCH_HW_INDEX | \
         PERF_SAMPLE_BRANCH_PRIV_SAVE)

#define BRBE_PERF_BRANCH_FILTERS    (BRBE_ALLOWED_BRANCH_FILTERS | \
         BRBE_EXCLUDE_BRANCH_FILTERS)

/*
 * BRBE supports the following functional branch type filters while
 * generating branch records. These branch filters can be enabled,
 * either individually or as a group i.e ORing multiple filters
 * with each other.
 *
 * BRBFCR_EL1_CONDDIR  - Conditional direct branch
 * BRBFCR_EL1_DIRCALL  - Direct call
 * BRBFCR_EL1_INDCALL  - Indirect call
 * BRBFCR_EL1_INDIRECT - Indirect branch
 * BRBFCR_EL1_DIRECT   - Direct branch
 * BRBFCR_EL1_RTN      - Subroutine return
 */
static u64 branch_type_to_brbfcr(int branch_type)
{
 u64 brbfcr = 0;

 if (branch_type & PERF_SAMPLE_BRANCH_ANY) {
  brbfcr |= BRBFCR_EL1_BRANCH_FILTERS;
  return brbfcr;
 }

 if (branch_type & PERF_SAMPLE_BRANCH_ANY_CALL) {
  brbfcr |= BRBFCR_EL1_INDCALL;
  brbfcr |= BRBFCR_EL1_DIRCALL;
 }

 if (branch_type & PERF_SAMPLE_BRANCH_ANY_RETURN)
  brbfcr |= BRBFCR_EL1_RTN;

 if (branch_type & PERF_SAMPLE_BRANCH_IND_CALL)
  brbfcr |= BRBFCR_EL1_INDCALL;

 if (branch_type & PERF_SAMPLE_BRANCH_COND)
  brbfcr |= BRBFCR_EL1_CONDDIR;

 if (branch_type & PERF_SAMPLE_BRANCH_IND_JUMP)
  brbfcr |= BRBFCR_EL1_INDIRECT;

 if (branch_type & PERF_SAMPLE_BRANCH_CALL)
  brbfcr |= BRBFCR_EL1_DIRCALL;

 return brbfcr;
}

/*
 * BRBE supports the following privilege mode filters while generating
 * branch records.
 *
 * BRBCR_ELx_E0BRE - EL0 branch records
 * BRBCR_ELx_ExBRE - EL1/EL2 branch records
 *
 * BRBE also supports the following additional functional branch type
 * filters while generating branch records.
 *
 * BRBCR_ELx_EXCEPTION - Exception
 * BRBCR_ELx_ERTN     -  Exception return
 */
static u64 branch_type_to_brbcr(int branch_type)
{
 u64 brbcr = BRBCR_ELx_FZP | BRBCR_ELx_DEFAULT_TS;

 if (branch_type & PERF_SAMPLE_BRANCH_USER)
  brbcr |= BRBCR_ELx_E0BRE;

 /*
 * When running in the hyp mode, writing into BRBCR_EL1
 * actually writes into BRBCR_EL2 instead. Field E2BRE
 * is also at the same position as E1BRE.
 */
 if (branch_type & PERF_SAMPLE_BRANCH_KERNEL)
  brbcr |= BRBCR_ELx_ExBRE;

 if (branch_type & PERF_SAMPLE_BRANCH_HV) {
  if (is_kernel_in_hyp_mode())
   brbcr |= BRBCR_ELx_ExBRE;
 }

 if (!(branch_type & PERF_SAMPLE_BRANCH_NO_CYCLES))
  brbcr |= BRBCR_ELx_CC;

 if (!(branch_type & PERF_SAMPLE_BRANCH_NO_FLAGS))
  brbcr |= BRBCR_ELx_MPRED;

 /*
 * The exception and exception return branches could be
 * captured, irrespective of the perf event's privilege.
 * If the perf event does not have enough privilege for
 * a given exception level, then addresses which falls
 * under that exception level will be reported as zero
 * for the captured branch record, creating source only
 * or target only records.
 */
 if (branch_type & PERF_SAMPLE_BRANCH_KERNEL) {
  if (branch_type & PERF_SAMPLE_BRANCH_ANY) {
   brbcr |= BRBCR_ELx_EXCEPTION;
   brbcr |= BRBCR_ELx_ERTN;
  }

  if (branch_type & PERF_SAMPLE_BRANCH_ANY_CALL)
   brbcr |= BRBCR_ELx_EXCEPTION;

  if (branch_type & PERF_SAMPLE_BRANCH_ANY_RETURN)
   brbcr |= BRBCR_ELx_ERTN;
 }
 return brbcr;
}

bool brbe_branch_attr_valid(struct perf_event *event)
{
 u64 branch_type = event->attr.branch_sample_type;

 /*
 * Ensure both perf branch filter allowed and exclude
 * masks are always in sync with the generic perf ABI.
 */
 BUILD_BUG_ON(BRBE_PERF_BRANCH_FILTERS != (PERF_SAMPLE_BRANCH_MAX - 1));

 if (branch_type & BRBE_EXCLUDE_BRANCH_FILTERS) {
  pr_debug("requested branch filter not supported 0x%llx\n", branch_type);
  return false;
 }

 /* Ensure at least 1 branch type is enabled */
 if (!(branch_type & BRBE_ALLOWED_BRANCH_TYPES)) {
  pr_debug("no branch type enabled 0x%llx\n", branch_type);
  return false;
 }

 /*
 * No branches are recorded in guests nor nVHE hypervisors, so
 * excluding the host or both kernel and user is invalid.
 *
 * Ideally we'd just require exclude_guest and exclude_hv, but setting
 * event filters with perf for kernel or user don't set exclude_guest.
 * So effectively, exclude_guest and exclude_hv are ignored.
 */
 if (event->attr.exclude_host || (event->attr.exclude_user && event->attr.exclude_kernel)) {
  pr_debug("branch filter in hypervisor or guest only not supported 0x%llx\n", branch_type);
  return false;
 }

 event->hw.branch_reg.config = branch_type_to_brbfcr(event->attr.branch_sample_type);
 event->hw.extra_reg.config = branch_type_to_brbcr(event->attr.branch_sample_type);

 return true;
}

unsigned int brbe_num_branch_records(const struct arm_pmu *armpmu)
{
 return FIELD_GET(BRBIDR0_EL1_NUMREC_MASK, armpmu->reg_brbidr);
}

void brbe_probe(struct arm_pmu *armpmu)
{
 u64 brbidr, aa64dfr0 = read_sysreg_s(SYS_ID_AA64DFR0_EL1);
 u32 brbe;

 brbe = cpuid_feature_extract_unsigned_field(aa64dfr0, ID_AA64DFR0_EL1_BRBE_SHIFT);
 if (!valid_brbe_version(brbe))
  return;

 brbidr = read_sysreg_s(SYS_BRBIDR0_EL1);
 if (!valid_brbidr(brbidr))
  return;

 armpmu->reg_brbidr = brbidr;
}

/*
 * BRBE is assumed to be disabled/paused on entry
 */
void brbe_enable(const struct arm_pmu *arm_pmu)
{
 struct pmu_hw_events *cpuc = this_cpu_ptr(arm_pmu->hw_events);
 u64 brbfcr = 0, brbcr = 0;

 /*
 * Discard existing records to avoid a discontinuity, e.g. records
 * missed during handling an overflow.
 */
 brbe_invalidate();

 /*
 * Merge the permitted branch filters of all events.
 */
 for (int i = 0; i < ARMPMU_MAX_HWEVENTS; i++) {
  struct perf_event *event = cpuc->events[i];

  if (event && has_branch_stack(event)) {
   brbfcr |= event->hw.branch_reg.config;
   brbcr |= event->hw.extra_reg.config;
  }
 }

 /*
 * In VHE mode with MDCR_EL2.HPMN equal to PMCR_EL0.N, BRBCR_EL1.FZP
 * controls freezing the branch records on counter overflow rather than
 * BRBCR_EL2.FZP (which writes to BRBCR_EL1 are redirected to).
 * The exception levels are enabled/disabled in BRBCR_EL2, so keep EL1
 * and EL0 recording disabled for guests.
 *
 * As BRBCR_EL1 CC and MPRED bits also need to match, use the same
 * value for both registers just masking the exception levels.
 */
 if (is_kernel_in_hyp_mode())
  write_sysreg_s(brbcr & ~(BRBCR_ELx_ExBRE | BRBCR_ELx_E0BRE), SYS_BRBCR_EL12);
 write_sysreg_s(brbcr, SYS_BRBCR_EL1);
 /* Ensure BRBCR_ELx settings take effect before unpausing */
 isb();

 /* Finally write SYS_BRBFCR_EL to unpause BRBE */
 write_sysreg_s(brbfcr, SYS_BRBFCR_EL1);
 /* Synchronization in PMCR write ensures ordering WRT PMU enabling */
}

void brbe_disable(void)
{
 /*
 * No need for synchronization here as synchronization in PMCR write
 * ensures ordering and in the interrupt handler this is a NOP as
 * we're already paused.
 */
 write_sysreg_s(BRBFCR_EL1_PAUSED, SYS_BRBFCR_EL1);
 write_sysreg_s(0, SYS_BRBCR_EL1);
}

static const int brbe_type_to_perf_type_map[BRBINFx_EL1_TYPE_DEBUG_EXIT + 1][2] = {
 [BRBINFx_EL1_TYPE_DIRECT_UNCOND] = { PERF_BR_UNCOND, 0 },
 [BRBINFx_EL1_TYPE_INDIRECT] = { PERF_BR_IND, 0 },
 [BRBINFx_EL1_TYPE_DIRECT_LINK] = { PERF_BR_CALL, 0 },
 [BRBINFx_EL1_TYPE_INDIRECT_LINK] = { PERF_BR_IND_CALL, 0 },
 [BRBINFx_EL1_TYPE_RET] = { PERF_BR_RET, 0 },
 [BRBINFx_EL1_TYPE_DIRECT_COND] = { PERF_BR_COND, 0 },
 [BRBINFx_EL1_TYPE_CALL] = { PERF_BR_SYSCALL, 0 },
 [BRBINFx_EL1_TYPE_ERET] = { PERF_BR_ERET, 0 },
 [BRBINFx_EL1_TYPE_IRQ] = { PERF_BR_IRQ, 0 },
 [BRBINFx_EL1_TYPE_TRAP] = { PERF_BR_IRQ, 0 },
 [BRBINFx_EL1_TYPE_SERROR] = { PERF_BR_SERROR, 0 },
 [BRBINFx_EL1_TYPE_ALIGN_FAULT] = { PERF_BR_EXTEND_ABI, PERF_BR_NEW_FAULT_ALGN },
 [BRBINFx_EL1_TYPE_INSN_FAULT] = { PERF_BR_EXTEND_ABI, PERF_BR_NEW_FAULT_INST },
 [BRBINFx_EL1_TYPE_DATA_FAULT] = { PERF_BR_EXTEND_ABI, PERF_BR_NEW_FAULT_DATA },
};

static void brbe_set_perf_entry_type(struct perf_branch_entry *entry, u64 brbinf)
{
 int brbe_type = brbinf_get_type(brbinf);

 if (brbe_type <= BRBINFx_EL1_TYPE_DEBUG_EXIT) {
  const int *br_type = brbe_type_to_perf_type_map[brbe_type];

  entry->type = br_type[0];
  entry->new_type = br_type[1];
 }
}

static int brbinf_get_perf_priv(u64 brbinf)
{
 int brbe_el = brbinf_get_el(brbinf);

 switch (brbe_el) {
 case BRBINFx_EL1_EL_EL0:
  return PERF_BR_PRIV_USER;
 case BRBINFx_EL1_EL_EL1:
  return PERF_BR_PRIV_KERNEL;
 case BRBINFx_EL1_EL_EL2:
  if (is_kernel_in_hyp_mode())
   return PERF_BR_PRIV_KERNEL;
  return PERF_BR_PRIV_HV;
 default:
  pr_warn_once("%d - unknown branch privilege captured\n", brbe_el);
  return PERF_BR_PRIV_UNKNOWN;
 }
}

static bool perf_entry_from_brbe_regset(int index, struct perf_branch_entry *entry,
     const struct perf_event *event)
{
 struct brbe_regset bregs;
 u64 brbinf;

 if (!__read_brbe_regset(&bregs, index))
  return false;

 brbinf = bregs.brbinf;
 perf_clear_branch_entry_bitfields(entry);
 if (brbe_record_is_complete(brbinf)) {
  entry->from = bregs.brbsrc;
  entry->to = bregs.brbtgt;
 } else if (brbe_record_is_source_only(brbinf)) {
  entry->from = bregs.brbsrc;
  entry->to = 0;
 } else if (brbe_record_is_target_only(brbinf)) {
  entry->from = 0;
  entry->to = bregs.brbtgt;
 }

 brbe_set_perf_entry_type(entry, brbinf);

 if (!branch_sample_no_cycles(event))
  entry->cycles = brbinf_get_cycles(brbinf);

 if (!branch_sample_no_flags(event)) {
  /* Mispredict info is available for source only and complete branch records. */
  if (!brbe_record_is_target_only(brbinf)) {
   entry->mispred = brbinf_get_mispredict(brbinf);
   entry->predicted = !entry->mispred;
  }

  /*
 * Currently TME feature is neither implemented in any hardware
 * nor it is being supported in the kernel. Just warn here once
 * if TME related information shows up rather unexpectedly.
 */
  if (brbinf_get_lastfailed(brbinf) || brbinf_get_in_tx(brbinf))
   pr_warn_once("Unknown transaction states\n");
 }

 /*
 * Branch privilege level is available for target only and complete
 * branch records.
 */
 if (!brbe_record_is_source_only(brbinf))
  entry->priv = brbinf_get_perf_priv(brbinf);

 return true;
}

#define PERF_BR_ARM64_ALL (    \
 BIT(PERF_BR_COND) |    \
 BIT(PERF_BR_UNCOND) |    \
 BIT(PERF_BR_IND) |    \
 BIT(PERF_BR_CALL) |    \
 BIT(PERF_BR_IND_CALL) |    \
 BIT(PERF_BR_RET))

#define PERF_BR_ARM64_ALL_KERNEL (   \
 BIT(PERF_BR_SYSCALL) |    \
 BIT(PERF_BR_IRQ) |    \
 BIT(PERF_BR_SERROR) |    \
 BIT(PERF_BR_MAX + PERF_BR_NEW_FAULT_ALGN) | \
 BIT(PERF_BR_MAX + PERF_BR_NEW_FAULT_DATA) | \
 BIT(PERF_BR_MAX + PERF_BR_NEW_FAULT_INST))

static void prepare_event_branch_type_mask(u64 branch_sample,
        unsigned long *event_type_mask)
{
 if (branch_sample & PERF_SAMPLE_BRANCH_ANY) {
  if (branch_sample & PERF_SAMPLE_BRANCH_KERNEL)
   bitmap_from_u64(event_type_mask,
    BIT(PERF_BR_ERET) | PERF_BR_ARM64_ALL |
    PERF_BR_ARM64_ALL_KERNEL);
  else
   bitmap_from_u64(event_type_mask, PERF_BR_ARM64_ALL);
  return;
 }

 bitmap_zero(event_type_mask, PERF_BR_ARM64_MAX);

 if (branch_sample & PERF_SAMPLE_BRANCH_ANY_CALL) {
  if (branch_sample & PERF_SAMPLE_BRANCH_KERNEL)
   bitmap_from_u64(event_type_mask, PERF_BR_ARM64_ALL_KERNEL);

  set_bit(PERF_BR_CALL, event_type_mask);
  set_bit(PERF_BR_IND_CALL, event_type_mask);
 }

 if (branch_sample & PERF_SAMPLE_BRANCH_IND_JUMP)
  set_bit(PERF_BR_IND, event_type_mask);

 if (branch_sample & PERF_SAMPLE_BRANCH_COND)
  set_bit(PERF_BR_COND, event_type_mask);

 if (branch_sample & PERF_SAMPLE_BRANCH_CALL)
  set_bit(PERF_BR_CALL, event_type_mask);

 if (branch_sample & PERF_SAMPLE_BRANCH_IND_CALL)
  set_bit(PERF_BR_IND_CALL, event_type_mask);

 if (branch_sample & PERF_SAMPLE_BRANCH_ANY_RETURN) {
  set_bit(PERF_BR_RET, event_type_mask);

  if (branch_sample & PERF_SAMPLE_BRANCH_KERNEL)
   set_bit(PERF_BR_ERET, event_type_mask);
 }
}

/*
 * BRBE is configured with an OR of permissions from all events, so there may
 * be events which have to be dropped or events where just the source or target
 * address has to be zeroed.
 */
static bool filter_branch_privilege(struct perf_branch_entry *entry, u64 branch_sample_type)
{
 bool from_user = access_ok((void __user *)(unsigned long)entry->from, 4);
 bool to_user = access_ok((void __user *)(unsigned long)entry->to, 4);
 bool exclude_kernel = !((branch_sample_type & PERF_SAMPLE_BRANCH_KERNEL) ||
  (is_kernel_in_hyp_mode() && (branch_sample_type & PERF_SAMPLE_BRANCH_HV)));

 /* We can only have a half record if permissions have not been expanded */
 if (!entry->from || !entry->to)
  return true;

 /*
 * If record is within a single exception level, just need to either
 * drop or keep the entire record.
 */
 if (from_user == to_user)
  return ((entry->priv == PERF_BR_PRIV_KERNEL) && !exclude_kernel) ||
   ((entry->priv == PERF_BR_PRIV_USER) &&
    (branch_sample_type & PERF_SAMPLE_BRANCH_USER));

 /*
 * Record is across exception levels, mask addresses for the exception
 * level we're not capturing.
 */
 if (!(branch_sample_type & PERF_SAMPLE_BRANCH_USER)) {
  if (from_user)
   entry->from = 0;
  if (to_user)
   entry->to = 0;
 }

 if (exclude_kernel) {
  if (!from_user)
   entry->from = 0;
  if (!to_user)
   entry->to = 0;
 }

 return true;
}

static bool filter_branch_type(struct perf_branch_entry *entry,
          const unsigned long *event_type_mask)
{
 if (entry->type == PERF_BR_EXTEND_ABI)
  return test_bit(PERF_BR_MAX + entry->new_type, event_type_mask);
 else
  return test_bit(entry->type, event_type_mask);
}

static bool filter_branch_record(struct perf_branch_entry *entry,
     u64 branch_sample,
     const unsigned long *event_type_mask)
{
 return filter_branch_type(entry, event_type_mask) &&
  filter_branch_privilege(entry, branch_sample);
}

void brbe_read_filtered_entries(struct perf_branch_stack *branch_stack,
    const struct perf_event *event)
{
 struct arm_pmu *cpu_pmu = to_arm_pmu(event->pmu);
 int nr_hw = brbe_num_branch_records(cpu_pmu);
 int nr_banks = DIV_ROUND_UP(nr_hw, BRBE_BANK_MAX_ENTRIES);
 int nr_filtered = 0;
 u64 branch_sample_type = event->attr.branch_sample_type;
 DECLARE_BITMAP(event_type_mask, PERF_BR_ARM64_MAX);

 prepare_event_branch_type_mask(branch_sample_type, event_type_mask);

 for (int bank = 0; bank < nr_banks; bank++) {
  int nr_remaining = nr_hw - (bank * BRBE_BANK_MAX_ENTRIES);
  int nr_this_bank = min(nr_remaining, BRBE_BANK_MAX_ENTRIES);

  select_brbe_bank(bank);

  for (int i = 0; i < nr_this_bank; i++) {
   struct perf_branch_entry *pbe = &branch_stack->entries[nr_filtered];

   if (!perf_entry_from_brbe_regset(i, pbe, event))
    goto done;

   if (!filter_branch_record(pbe, branch_sample_type, event_type_mask))
    continue;

   nr_filtered++;
  }
 }

done:
 branch_stack->nr = nr_filtered;
}