MIPS: Loongson: Add NMI handler support

Signed-off-by: Huacai Chen <[e-mail]>
Cc: John Crispin <[e-mail]>
Cc: Steven J . Hill <[e-mail]>
Cc: Fuxin Zhang <[e-mail]>
Cc: Zhangjin Wu <[e-mail]>
Cc: [e-mail]
Patchwork: https://patchwork.linux-mips.org/patch/16587/
Signed-off-by: Ralf Baechle <[e-mail]>

arch/mips/loongson64/common/init.c

···
 
 #include <linux/bootmem.h>
 #include <asm/bootinfo.h>
+#include <asm/traps.h>
 #include <asm/smp-ops.h>
+#include <asm/cacheflush.h>
 
 #include <loongson.h>
 
 /* Loongson CPU address windows config space base address */
 unsigned long __maybe_unused _loongson_addrwincfg_base;
+
+static void __init mips_nmi_setup(void)
+{
+	void *base;
+	extern char except_vec_nmi;
+
+	base = (void *)(CAC_BASE + 0x380);
+	memcpy(base, &except_vec_nmi, 0x80);
+	flush_icache_range((unsigned long)base, (unsigned long)base + 0x80);
+}
 
 void __init prom_init(void)
 {
···
 	/*init the uart base address */
 	prom_init_uart_base();
 	register_smp_ops(&loongson3_smp_ops);
+	board_nmi_handler_setup = mips_nmi_setup;
 }
 
 void __init prom_free_prom_memory(void)