// SPDX-License-Identifier: GPL-2.0-only
/*
 *  linux/arch/arm/kernel/traps.c
 *
 *  Copyright (C) 1995-2009 Russell King
 *  Fragments that appear the same as linux/arch/i386/kernel/traps.c (C) Linus Torvalds
 *
 *  'traps.c' handles hardware exceptions after we have saved some state in
 *  'linux/arch/arm/lib/traps.S'.  Mostly a debugging aid, but will probably
 *  kill the offending process.
 */
#include <linux/signal.h>
#include <linux/personality.h>
#include <linux/kallsyms.h>
#include <linux/spinlock.h>
#include <linux/uaccess.h>
#include <linux/hardirq.h>
#include <linux/kdebug.h>
#include <linux/kprobes.h>
#include <linux/module.h>
#include <linux/kexec.h>
#include <linux/bug.h>
#include <linux/delay.h>
#include <linux/init.h>
#include <linux/sched/signal.h>
#include <linux/sched/debug.h>
#include <linux/sched/task_stack.h>
#include <linux/irq.h>
#include <linux/vmalloc.h>

#include <linux/atomic.h>
#include <asm/cacheflush.h>
#include <asm/exception.h>
#include <asm/spectre.h>
#include <asm/unistd.h>
#include <asm/traps.h>
#include <asm/ptrace.h>
#include <asm/unwind.h>
#include <asm/tls.h>
#include <asm/stacktrace.h>
#include <asm/system_misc.h>
#include <asm/opcodes.h>


static const char *handler[]= {
 "prefetch abort",
 "data abort",
 "address exception",
 "interrupt",
 "undefined instruction",
};

void *vectors_page;

#ifdef CONFIG_DEBUG_USER
unsigned int user_debug;

static int __init user_debug_setup(char *str)
{
 get_option(&str, &user_debug);
 return 1;
}
__setup("user_debug=", user_debug_setup);
#endif

void dump_backtrace_entry(unsigned long where, unsigned long from,
     unsigned long frame, const char *loglvl)
{
 unsigned long end = frame + 4 + sizeof(struct pt_regs);

 if (IS_ENABLED(CONFIG_UNWINDER_FRAME_POINTER) &&
     IS_ENABLED(CONFIG_CC_IS_GCC) &&
     end > ALIGN(frame, THREAD_SIZE)) {
  /*
   * If we are walking past the end of the stack, it may be due
   * to the fact that we are on an IRQ or overflow stack. In this
   * case, we can load the address of the other stack from the
   * frame record.
 */
  frame = ((unsigned long *)frame)[-2] - 4;
  end = frame + 4 + sizeof(struct pt_regs);
 }

#ifndef CONFIG_KALLSYMS
 printk("%sFunction entered at [<%08lx>] from [<%08lx>]\n",
  loglvl, where, from);
#elif defined CONFIG_BACKTRACE_VERBOSE
 printk("%s[<%08lx>] (%ps) from [<%08lx>] (%pS)\n",
  loglvl, where, (void *)where, from, (void *)from);
#else
 printk("%s %ps from %pS\n", loglvl, (void *)where, (void *)from);
#endif

 if (in_entry_text(from) && end <= ALIGN(frame, THREAD_SIZE))
  dump_mem(loglvl, "Exception stack", frame + 4, end);
}

void dump_backtrace_stm(u32 *stack, u32 instruction, const char *loglvl)
{
 char str[80], *p;
 unsigned int x;
 int reg;

 for (reg = 10, x = 0, p = str; reg >= 0; reg--) {
  if (instruction & BIT(reg)) {
   p += sprintf(p, " r%d:%08x", reg, *stack--);
   if (++x == 6) {
    x = 0;
    p = str;
    printk("%s%s\n", loglvl, str);
   }
  }
 }
 if (p != str)
  printk("%s%s\n", loglvl, str);
}

#ifndef CONFIG_ARM_UNWIND
/*
 * Stack pointers should always be within the kernels view of
 * physical memory.  If it is not there, then we can't dump
 * out any information relating to the stack.
 */
static int verify_stack(unsigned long sp)
{
 if (sp < PAGE_OFFSET ||
     (!IS_ENABLED(CONFIG_VMAP_STACK) &&
      sp > (unsigned long)high_memory && high_memory != NULL))
  return -EFAULT;

 return 0;
}
#endif

/*
 * Dump out the contents of some memory nicely...
 */
void dump_mem(const char *lvl, const char *str, unsigned long bottom,
       unsigned long top)
{
 unsigned long first;
 int i;

 printk("%s%s(0x%08lx to 0x%08lx)\n", lvl, str, bottom, top);

 for (first = bottom & ~31; first < top; first += 32) {
  unsigned long p;
  char str[sizeof(" 12345678") * 8 + 1];

  memset(str, ' ', sizeof(str));
  str[sizeof(str) - 1] = '\0';

  for (p = first, i = 0; i < 8 && p < top; i++, p += 4) {
   if (p >= bottom && p < top) {
    unsigned long val;
    if (!get_kernel_nofault(val, (unsigned long *)p))
     sprintf(str + i * 9, " %08lx", val);
    else
     sprintf(str + i * 9, " ????????");
   }
  }
  printk("%s%04lx:%s\n", lvl, first & 0xffff, str);
 }
}

static void dump_instr(const char *lvl, struct pt_regs *regs)
{
 unsigned long addr = instruction_pointer(regs);
 const int thumb = thumb_mode(regs);
 const int width = thumb ? 4 : 8;
 char str[sizeof("00000000 ") * 5 + 2 + 1], *p = str;
 int i;

 /*
  * Note that we now dump the code first, just in case the backtrace
  * kills us.
 */

 for (i = -4; i < 1 + !!thumb; i++) {
  unsigned int val, bad;

  if (thumb) {
   u16 tmp;

   if (user_mode(regs))
    bad = get_user(tmp, &((u16 __user *)addr)[i]);
   else
    bad = get_kernel_nofault(tmp, &((u16 *)addr)[i]);

   val = __mem_to_opcode_thumb16(tmp);
  } else {
   if (user_mode(regs))
    bad = get_user(val, &((u32 __user *)addr)[i]);
   else
    bad = get_kernel_nofault(val, &((u32 *)addr)[i]);

   val = __mem_to_opcode_arm(val);
  }

  if (!bad)
   p += sprintf(p, i == 0 ? "(%0*x) " : "%0*x ",
     width, val);
  else {
   p += sprintf(p, "bad PC value");
   break;
  }
 }
 printk("%sCode: %s\n", lvl, str);
}

#ifdef CONFIG_ARM_UNWIND
void dump_backtrace(struct pt_regs *regs, struct task_struct *tsk,
      const char *loglvl)
{
 unwind_backtrace(regs, tsk, loglvl);
}
#else
void dump_backtrace(struct pt_regs *regs, struct task_struct *tsk,
      const char *loglvl)
{
 unsigned int fp, mode;
 int ok = 1;

 printk("%sCall trace: ", loglvl);

 if (!tsk)
  tsk = current;

 if (regs) {
  fp = frame_pointer(regs);
  mode = processor_mode(regs);
 } else if (tsk != current) {
  fp = thread_saved_fp(tsk);
  mode = 0x10;
 } else {
  asm("mov %0, fp" : "=r" (fp) : : "cc");
  mode = 0x10;
 }

 if (!fp) {
  pr_cont("no frame pointer");
  ok = 0;
 } else if (verify_stack(fp)) {
  pr_cont("invalid frame pointer 0x%08x", fp);
  ok = 0;
 } else if (fp < (unsigned long)end_of_stack(tsk))
  pr_cont("frame pointer underflow");
 pr_cont("\n");

 if (ok)
  c_backtrace(fp, mode, loglvl);
}
#endif

void show_stack(struct task_struct *tsk, unsigned long *sp, const char *loglvl)
{
 dump_backtrace(NULL, tsk, loglvl);
 barrier();
}

#ifdef CONFIG_SMP
#define S_SMP " SMP"
#else
#define S_SMP ""
#endif
#ifdef CONFIG_THUMB2_KERNEL
#define S_ISA " THUMB2"
#else
#define S_ISA " ARM"
#endif

static int __die(const char *str, int err, struct pt_regs *regs)
{
 struct task_struct *tsk = current;
 static int die_counter;
 int ret;

 pr_emerg("Internal error: %s: %x [#%d]" S_SMP S_ISA "\n",
   str, err, ++die_counter);

 /* trap and error numbers are mostly meaningless on ARM */
 ret = notify_die(DIE_OOPS, str, regs, err, tsk->thread.trap_no, SIGSEGV);
 if (ret == NOTIFY_STOP)
  return 1;

 print_modules();
 __show_regs(regs);
 __show_regs_alloc_free(regs);
 pr_emerg("Process %.*s (pid: %d, stack limit = 0x%p)\n",
   TASK_COMM_LEN, tsk->comm, task_pid_nr(tsk), end_of_stack(tsk));

 if (!user_mode(regs) || in_interrupt()) {
  dump_mem(KERN_EMERG, "Stack: ", regs->ARM_sp,
    ALIGN(regs->ARM_sp - THREAD_SIZE, THREAD_ALIGN)
    + THREAD_SIZE);
  dump_backtrace(regs, tsk, KERN_EMERG);
  dump_instr(KERN_EMERG, regs);
 }

 return 0;
}

static arch_spinlock_t die_lock = __ARCH_SPIN_LOCK_UNLOCKED;
static int die_owner = -1;
static unsigned int die_nest_count;

static unsigned long oops_begin(void)
{
 int cpu;
 unsigned long flags;

 oops_enter();

 /* racy, but better than risking deadlock. */
 raw_local_irq_save(flags);
 cpu = smp_processor_id();
 if (!arch_spin_trylock(&die_lock)) {
  if (cpu == die_owner)
   /* nested oops. should stop eventually */;
  else
   arch_spin_lock(&die_lock);
 }
 die_nest_count++;
 die_owner = cpu;
 console_verbose();
 bust_spinlocks(1);
 return flags;
}

static void oops_end(unsigned long flags, struct pt_regs *regs, int signr)
{
 if (regs && kexec_should_crash(current))
  crash_kexec(regs);

 bust_spinlocks(0);
 die_owner = -1;
 add_taint(TAINT_DIE, LOCKDEP_NOW_UNRELIABLE);
 die_nest_count--;
 if (!die_nest_count)
  /* Nest count reaches zero, release the lock. */
  arch_spin_unlock(&die_lock);
 raw_local_irq_restore(flags);
 oops_exit();

 if (in_interrupt())
  panic("Fatal exception in interrupt");
 if (panic_on_oops)
  panic("Fatal exception");
 if (signr)
  make_task_dead(signr);
}

/*
 * This function is protected against re-entrancy.
 */
void die(const char *str, struct pt_regs *regs, int err)
{
 enum bug_trap_type bug_type = BUG_TRAP_TYPE_NONE;
 unsigned long flags = oops_begin();
 int sig = SIGSEGV;

 if (!user_mode(regs))
  bug_type = report_bug(regs->ARM_pc, regs);
 if (bug_type != BUG_TRAP_TYPE_NONE)
  str = "Oops - BUG";

 if (__die(str, err, regs))
  sig = 0;

 oops_end(flags, regs, sig);
}

void arm_notify_die(const char *str, struct pt_regs *regs,
  int signo, int si_code, void __user *addr,
  unsigned long err, unsigned long trap)
{
 if (user_mode(regs)) {
  current->thread.error_code = err;
  current->thread.trap_no = trap;

  force_sig_fault(signo, si_code, addr);
 } else {
  die(str, regs, err);
 }
}

#ifdef CONFIG_GENERIC_BUG

int is_valid_bugaddr(unsigned long pc)
{
#ifdef CONFIG_THUMB2_KERNEL
 u16 bkpt;
 u16 insn = __opcode_to_mem_thumb16(BUG_INSTR_VALUE);
#else
 u32 bkpt;
 u32 insn = __opcode_to_mem_arm(BUG_INSTR_VALUE);
#endif

 if (get_kernel_nofault(bkpt, (void *)pc))
  return 0;

 return bkpt == insn;
}

#endif

static LIST_HEAD(undef_hook);
static DEFINE_RAW_SPINLOCK(undef_lock);

void register_undef_hook(struct undef_hook *hook)
{
 unsigned long flags;

 raw_spin_lock_irqsave(&undef_lock, flags);
 list_add(&hook->node, &undef_hook);
 raw_spin_unlock_irqrestore(&undef_lock, flags);
}

void unregister_undef_hook(struct undef_hook *hook)
{
 unsigned long flags;

 raw_spin_lock_irqsave(&undef_lock, flags);
 list_del(&hook->node);
 raw_spin_unlock_irqrestore(&undef_lock, flags);
}

static nokprobe_inline
int call_undef_hook(struct pt_regs *regs, unsigned int instr)
{
 struct undef_hook *hook;
 unsigned long flags;
 int (*fn)(struct pt_regs *regs, unsigned int instr) = NULL;

 raw_spin_lock_irqsave(&undef_lock, flags);
 list_for_each_entry(hook, &undef_hook, node)
  if ((instr & hook->instr_mask) == hook->instr_val &&
      (regs->ARM_cpsr & hook->cpsr_mask) == hook->cpsr_val)
   fn = hook->fn;
 raw_spin_unlock_irqrestore(&undef_lock, flags);

 return fn ? fn(regs, instr) : 1;
}

asmlinkage void do_undefinstr(struct pt_regs *regs)
{
 unsigned int instr;
 void __user *pc;

 pc = (void __user *)instruction_pointer(regs);

 if (processor_mode(regs) == SVC_MODE) {
#ifdef CONFIG_THUMB2_KERNEL
  if (thumb_mode(regs)) {
   instr = __mem_to_opcode_thumb16(((u16 *)pc)[0]);
   if (is_wide_instruction(instr)) {
    u16 inst2;
    inst2 = __mem_to_opcode_thumb16(((u16 *)pc)[1]);
    instr = __opcode_thumb32_compose(instr, inst2);
   }
  } else
#endif
   instr = __mem_to_opcode_arm(*(u32 *) pc);
 } else if (thumb_mode(regs)) {
  if (get_user(instr, (u16 __user *)pc))
   goto die_sig;
  instr = __mem_to_opcode_thumb16(instr);
  if (is_wide_instruction(instr)) {
   unsigned int instr2;
   if (get_user(instr2, (u16 __user *)pc+1))
    goto die_sig;
   instr2 = __mem_to_opcode_thumb16(instr2);
   instr = __opcode_thumb32_compose(instr, instr2);
  }
 } else {
  if (get_user(instr, (u32 __user *)pc))
   goto die_sig;
  instr = __mem_to_opcode_arm(instr);
 }

 if (call_undef_hook(regs, instr) == 0)
  return;

die_sig:
#ifdef CONFIG_DEBUG_USER
 if (user_debug & UDBG_UNDEFINED) {
  pr_info("%s (%d): undefined instruction: pc=%px\n",
   current->comm, task_pid_nr(current), pc);
  __show_regs(regs);
  dump_instr(KERN_INFO, regs);
 }
#endif
 arm_notify_die("Oops - undefined instruction", regs,
         SIGILL, ILL_ILLOPC, pc, 0, 6);
}
NOKPROBE_SYMBOL(do_undefinstr)

/*
 * Handle FIQ similarly to NMI on x86 systems.
 *
 * The runtime environment for NMIs is extremely restrictive
 * (NMIs can pre-empt critical sections meaning almost all locking is
 * forbidden) meaning this default FIQ handling must only be used in
 * circumstances where non-maskability improves robustness, such as
 * watchdog or debug logic.
 *
 * This handler is not appropriate for general purpose use in drivers
 * platform code and can be overrideen using set_fiq_handler.
 */
asmlinkage void __exception_irq_entry handle_fiq_as_nmi(struct pt_regs *regs)
{
 struct pt_regs *old_regs = set_irq_regs(regs);

 nmi_enter();

 /* nop. FIQ handlers for special arch/arm features can be added here. */

 nmi_exit();

 set_irq_regs(old_regs);
}

/*
 * bad_mode handles the impossible case in the vectors.  If you see one of
 * these, then it's extremely serious, and could mean you have buggy hardware.
 * It never returns, and never tries to sync.  We hope that we can at least
 * dump out some state information...
 */
asmlinkage void bad_mode(struct pt_regs *regs, int reason)
{
 console_verbose();

 pr_crit("Bad mode in %s handler detected\n", handler[reason]);

 die("Oops - bad mode", regs, 0);
 local_irq_disable();
 panic("bad mode");
}

static int bad_syscall(int n, struct pt_regs *regs)
{
 if ((current->personality & PER_MASK) != PER_LINUX) {
  send_sig(SIGSEGV, current, 1);
  return regs->ARM_r0;
 }

#ifdef CONFIG_DEBUG_USER
 if (user_debug & UDBG_SYSCALL) {
  pr_err("[%d] %s: obsolete system call %08x.\n",
   task_pid_nr(current), current->comm, n);
  dump_instr(KERN_ERR, regs);
 }
#endif

 arm_notify_die("Oops - bad syscall", regs, SIGILL, ILL_ILLTRP,
         (void __user *)instruction_pointer(regs) -
    (thumb_mode(regs) ? 2 : 4),
         n, 0);

 return regs->ARM_r0;
}

static inline int
__do_cache_op(unsigned long start, unsigned long end)
{
 unsigned int ua_flags;
 int ret;

 do {
  unsigned long chunk = min(PAGE_SIZE, end - start);

  if (fatal_signal_pending(current))
   return 0;

  ua_flags = uaccess_save_and_enable();
  ret = flush_icache_user_range(start, start + chunk);
  uaccess_restore(ua_flags);
  if (ret)
   return ret;

  cond_resched();
  start += chunk;
 } while (start < end);

 return 0;
}

static inline int
do_cache_op(unsigned long start, unsigned long end, int flags)
{
 if (end < start || flags)
  return -EINVAL;

 if (!access_ok((void __user *)start, end - start))
  return -EFAULT;

 return __do_cache_op(start, end);
}

/*
 * Handle all unrecognised system calls.
 *  0x9f0000 - 0x9fffff are some more esoteric system calls
 */
#define NR(x) ((__ARM_NR_##x) - __ARM_NR_BASE)
asmlinkage int arm_syscall(int no, struct pt_regs *regs)
{
 if ((no >> 16) != (__ARM_NR_BASE>> 16))
  return bad_syscall(no, regs);

 switch (no & 0xffff) {
 case 0: /* branch through 0 */
  arm_notify_die("branch through zero", regs,
          SIGSEGV, SEGV_MAPERR, NULL, 0, 0);
  return 0;

 case NR(breakpoint): /* SWI BREAK_POINT */
  regs->ARM_pc -= thumb_mode(regs) ? 2 : 4;
  ptrace_break(regs);
  return regs->ARM_r0;

 /*
  * Flush a region from virtual address 'r0' to virtual address 'r1'
  * _exclusive_.  There is no alignment requirement on either address;
  * user space does not need to know the hardware cache layout.
  *
  * r2 contains flags.  It should ALWAYS be passed as ZERO until it
  * is defined to be something else.  For now we ignore it, but may
  * the fires of hell burn in your belly if you break this rule. ;)
  *
  * (at a later date, we may want to allow this call to not flush
  * various aspects of the cache.  Passing '0' will guarantee that
  * everything necessary gets flushed to maintain consistency in
  * the specified region).
 */
 case NR(cacheflush):
  return do_cache_op(regs->ARM_r0, regs->ARM_r1, regs->ARM_r2);

 case NR(usr26):
  if (!(elf_hwcap & HWCAP_26BIT))
   break;
  regs->ARM_cpsr &= ~MODE32_BIT;
  return regs->ARM_r0;

 case NR(usr32):
  if (!(elf_hwcap & HWCAP_26BIT))
   break;
  regs->ARM_cpsr |= MODE32_BIT;
  return regs->ARM_r0;

 case NR(set_tls):
  set_tls(regs->ARM_r0);
  return 0;

 case NR(get_tls):
  return current_thread_info()->tp_value[0];

 default:
  /* Calls 9f00xx..9f07ff are defined to return -ENOSYS
     if not implemented, rather than raising SIGILL.  This
     way the calling program can gracefully determine whether
   a feature is supported.  */
  if ((no & 0xffff) <= 0x7ff)
   return -ENOSYS;
  break;
 }
#ifdef CONFIG_DEBUG_USER
 /*
  * experience shows that these seem to indicate that
  * something catastrophic has happened
 */
 if (user_debug & UDBG_SYSCALL) {
  pr_err("[%d] %s: arm syscall %d\n",
         task_pid_nr(current), current->comm, no);
  dump_instr(KERN_ERR, regs);
  if (user_mode(regs)) {
   __show_regs(regs);
   c_backtrace(frame_pointer(regs), processor_mode(regs), KERN_ERR);
  }
 }
#endif
 arm_notify_die("Oops - bad syscall(2)", regs, SIGILL, ILL_ILLTRP,
         (void __user *)instruction_pointer(regs) -
    (thumb_mode(regs) ? 2 : 4),
         no, 0);
 return 0;
}

#ifdef CONFIG_TLS_REG_EMUL

/*
 * We might be running on an ARMv6+ processor which should have the TLS
 * register but for some reason we can't use it, or maybe an SMP system
 * using a pre-ARMv6 processor (there are apparently a few prototypes like
 * that in existence) and therefore access to that register must be
 * emulated.
 */

static int get_tp_trap(struct pt_regs *regs, unsigned int instr)
{
 int reg = (instr >> 12) & 15;
 if (reg == 15)
  return 1;
 regs->uregs[reg] = current_thread_info()->tp_value[0];
 regs->ARM_pc += 4;
 return 0;
}

static struct undef_hook arm_mrc_hook = {
 .instr_mask = 0x0fff0fff,
 .instr_val = 0x0e1d0f70,
 .cpsr_mask = PSR_T_BIT,
 .cpsr_val = 0,
 .fn  = get_tp_trap,
};

static int __init arm_mrc_hook_init(void)
{
 register_undef_hook(&arm_mrc_hook);
 return 0;
}

late_initcall(arm_mrc_hook_init);

#endif

/*
 * A data abort trap was taken, but we did not handle the instruction.
 * Try to abort the user program, or panic if it was the kernel.
 */
asmlinkage void
baddataabort(int code, unsigned long instr, struct pt_regs *regs)
{
 unsigned long addr = instruction_pointer(regs);

#ifdef CONFIG_DEBUG_USER
 if (user_debug & UDBG_BADABORT) {
  pr_err("8<--- cut here ---\n");
  pr_err("[%d] %s: bad data abort: code %d instr 0x%08lx\n",
         task_pid_nr(current), current->comm, code, instr);
  dump_instr(KERN_ERR, regs);
  show_pte(KERN_ERR, current->mm, addr);
 }
#endif

 arm_notify_die("unknown data abort code", regs,
         SIGILL, ILL_ILLOPC, (void __user *)addr, instr, 0);
}

void __readwrite_bug(const char *fn)
{
 pr_err("%s called, but not implemented\n", fn);
 BUG();
}
EXPORT_SYMBOL(__readwrite_bug);

#ifdef CONFIG_MMU
void __pte_error(const char *file, int line, pte_t pte)
{
 pr_err("%s:%d: bad pte %08llx.\n", file, line, (long long)pte_val(pte));
}

void __pmd_error(const char *file, int line, pmd_t pmd)
{
 pr_err("%s:%d: bad pmd %08llx.\n", file, line, (long long)pmd_val(pmd));
}

void __pgd_error(const char *file, int line, pgd_t pgd)
{
 pr_err("%s:%d: bad pgd %08llx.\n", file, line, (long long)pgd_val(pgd));
}
#endif

asmlinkage void __div0(void)
{
 pr_err("Division by zero in kernel.\n");
 dump_stack();
}
EXPORT_SYMBOL(__div0);

void abort(void)
{
 BUG();

 /* if that doesn't kill us, halt */
 panic("Oops failed to kill thread");
}

#ifdef CONFIG_KUSER_HELPERS
static void __init kuser_init(void *vectors)
{
 extern char __kuser_helper_start[], __kuser_helper_end[];
 int kuser_sz = __kuser_helper_end - __kuser_helper_start;

 memcpy(vectors + 0x1000 - kuser_sz, __kuser_helper_start, kuser_sz);

 /*
  * vectors + 0xfe0 = __kuser_get_tls
  * vectors + 0xfe8 = hardware TLS instruction at 0xffff0fe8
 */
 if (tls_emu || has_tls_reg)
  memcpy(vectors + 0xfe0, vectors + 0xfe8, 4);
}
#else
static inline void __init kuser_init(void *vectors)
{
}
#endif

#ifndef CONFIG_CPU_V7M
static void copy_from_lma(void *vma, void *lma_start, void *lma_end)
{
 memcpy(vma, lma_start, lma_end - lma_start);
}

static void flush_vectors(void *vma, size_t offset, size_t size)
{
 unsigned long start = (unsigned long)vma + offset;
 unsigned long end = start + size;

 flush_icache_range(start, end);
}

#ifdef CONFIG_HARDEN_BRANCH_HISTORY
int spectre_bhb_update_vectors(unsigned int method)
{
 extern char __vectors_bhb_bpiall_start[], __vectors_bhb_bpiall_end[];
 extern char __vectors_bhb_loop8_start[], __vectors_bhb_loop8_end[];
 void *vec_start, *vec_end;

 if (system_state >= SYSTEM_FREEING_INITMEM) {
  pr_err("CPU%u: Spectre BHB workaround too late - system vulnerable\n",
         smp_processor_id());
  return SPECTRE_VULNERABLE;
 }

 switch (method) {
 case SPECTRE_V2_METHOD_LOOP8:
  vec_start = __vectors_bhb_loop8_start;
  vec_end = __vectors_bhb_loop8_end;
  break;

 case SPECTRE_V2_METHOD_BPIALL:
  vec_start = __vectors_bhb_bpiall_start;
  vec_end = __vectors_bhb_bpiall_end;
  break;

 default:
  pr_err("CPU%u: unknown Spectre BHB state %d\n",
         smp_processor_id(), method);
  return SPECTRE_VULNERABLE;
 }

 copy_from_lma(vectors_page, vec_start, vec_end);
 flush_vectors(vectors_page, 0, vec_end - vec_start);

 return SPECTRE_MITIGATED;
}
#endif

void __init early_trap_init(void *vectors_base)
{
 extern char __stubs_start[], __stubs_end[];
 extern char __vectors_start[], __vectors_end[];
 unsigned i;

 vectors_page = vectors_base;

 /*
  * Poison the vectors page with an undefined instruction.  This
  * instruction is chosen to be undefined for both ARM and Thumb
  * ISAs.  The Thumb version is an undefined instruction with a
  * branch back to the undefined instruction.
 */
 for (i = 0; i < PAGE_SIZE / sizeof(u32); i++)
  ((u32 *)vectors_base)[i] = 0xe7fddef1;

 /*
  * Copy the vectors, stubs and kuser helpers (in entry-armv.S)
  * into the vector page, mapped at 0xffff0000, and ensure these
  * are visible to the instruction stream.
 */
 copy_from_lma(vectors_base, __vectors_start, __vectors_end);
 copy_from_lma(vectors_base + 0x1000, __stubs_start, __stubs_end);

 kuser_init(vectors_base);

 flush_vectors(vectors_base, 0, PAGE_SIZE * 2);
}
#else /* ifndef CONFIG_CPU_V7M */
void __init early_trap_init(void *vectors_base)
{
 /*
  * on V7-M there is no need to copy the vector table to a dedicated
  * memory area. The address is configurable and so a table in the kernel
  * image can be used.
 */
}
#endif

#ifdef CONFIG_VMAP_STACK

DECLARE_PER_CPU(u8 *, irq_stack_ptr);

asmlinkage DEFINE_PER_CPU(u8 *, overflow_stack_ptr);

static int __init allocate_overflow_stacks(void)
{
 u8 *stack;
 int cpu;

 for_each_possible_cpu(cpu) {
  stack = (u8 *)__get_free_page(GFP_KERNEL);
  if (WARN_ON(!stack))
   return -ENOMEM;
  per_cpu(overflow_stack_ptr, cpu) = &stack[OVERFLOW_STACK_SIZE];
 }
 return 0;
}
early_initcall(allocate_overflow_stacks);

asmlinkage void handle_bad_stack(struct pt_regs *regs)
{
 unsigned long tsk_stk = (unsigned long)current->stack;
#ifdef CONFIG_IRQSTACKS
 unsigned long irq_stk = (unsigned long)raw_cpu_read(irq_stack_ptr);
#endif
 unsigned long ovf_stk = (unsigned long)raw_cpu_read(overflow_stack_ptr);

 console_verbose();
 pr_emerg("Insufficient stack space to handle exception!");

 pr_emerg("Task stack:     [0x%08lx..0x%08lx]\n",
   tsk_stk, tsk_stk + THREAD_SIZE);
#ifdef CONFIG_IRQSTACKS
 pr_emerg("IRQ stack:      [0x%08lx..0x%08lx]\n",
   irq_stk - THREAD_SIZE, irq_stk);
#endif
 pr_emerg("Overflow stack: [0x%08lx..0x%08lx]\n",
   ovf_stk - OVERFLOW_STACK_SIZE, ovf_stk);

 die("kernel stack overflow", regs, 0);
}

#ifndef CONFIG_ARM_LPAE
/*
 * Normally, we rely on the logic in do_translation_fault() to update stale PMD
 * entries covering the vmalloc space in a task's page tables when it first
 * accesses the region in question. Unfortunately, this is not sufficient when
 * the task stack resides in the vmalloc region, as do_translation_fault() is a
 * C function that needs a stack to run.
 *
 * So we need to ensure that these PMD entries are up to date *before* the MM
 * switch. As we already have some logic in the MM switch path that takes care
 * of this, let's trigger it by bumping the counter every time the core vmalloc
 * code modifies a PMD entry in the vmalloc region. Use release semantics on
 * the store so that other CPUs observing the counter's new value are
 * guaranteed to see the updated page table entries as well.
 */
void arch_sync_kernel_mappings(unsigned long start, unsigned long end)
{
 if (start < VMALLOC_END && end > VMALLOC_START)
  atomic_inc_return_release(&init_mm.context.vmalloc_seq);
}
#endif
#endif