1/* SPDX-License-Identifier: GPL-2.0-or-later */
  2/*
  3 * Copyright (c) 2012 Linaro Limited.
  4 */
  5
  6#include <linux/init.h>
  7#include <linux/irqchip/arm-gic-v3.h>
  8#include <linux/linkage.h>
  9#include <asm/assembler.h>
 10#include <asm/virt.h>
 11
 12#ifndef ZIMAGE
 13/*
 14 * For the kernel proper, we need to find out the CPU boot mode long after
 15 * boot, so we need to store it in a writable variable.
 16 *
 17 * This is not in .bss, because we set it sufficiently early that the boot-time
 18 * zeroing of .bss would clobber it.
 19 */
 20.data
 21	.align	2
 22ENTRY(__boot_cpu_mode)
 23	.long	0
 24.text
 25
 26	/*
 27	 * Save the primary CPU boot mode. Requires 3 scratch registers.
 28	 */
 29	.macro	store_primary_cpu_mode	reg1, reg2, reg3
 30	mrs	\reg1, cpsr
 31	and	\reg1, \reg1, #MODE_MASK
 32	adr	\reg2, .L__boot_cpu_mode_offset
 33	ldr	\reg3, [\reg2]
 34	str	\reg1, [\reg2, \reg3]
 35	.endm
 36
 37	/*
 38	 * Compare the current mode with the one saved on the primary CPU.
 39	 * If they don't match, record that fact. The Z bit indicates
 40	 * if there's a match or not.
 41	 * Requires 3 additionnal scratch registers.
 42	 */
 43	.macro	compare_cpu_mode_with_primary mode, reg1, reg2, reg3
 44	adr	\reg2, .L__boot_cpu_mode_offset
 45	ldr	\reg3, [\reg2]
 46	ldr	\reg1, [\reg2, \reg3]
 47	cmp	\mode, \reg1		@ matches primary CPU boot mode?
 48	orrne	\reg1, \reg1, #BOOT_CPU_MODE_MISMATCH
 49	strne	\reg1, [\reg2, \reg3]	@ record what happened and give up
 50	.endm
 51
 52#else	/* ZIMAGE */
 53
 54	.macro	store_primary_cpu_mode	reg1:req, reg2:req, reg3:req
 55	.endm
 56
 57/*
 58 * The zImage loader only runs on one CPU, so we don't bother with mult-CPU
 59 * consistency checking:
 60 */
 61	.macro	compare_cpu_mode_with_primary mode, reg1, reg2, reg3
 62	cmp	\mode, \mode
 63	.endm
 64
 65#endif /* ZIMAGE */
 66
 67/*
 68 * Hypervisor stub installation functions.
 69 *
 70 * These must be called with the MMU and D-cache off.
 71 * They are not ABI compliant and are only intended to be called from the kernel
 72 * entry points in head.S.
 73 */
 74@ Call this from the primary CPU
 75ENTRY(__hyp_stub_install)
 76	store_primary_cpu_mode	r4, r5, r6
 77ENDPROC(__hyp_stub_install)
 78
 79	@ fall through...
 80
 81@ Secondary CPUs should call here
 82ENTRY(__hyp_stub_install_secondary)
 83	mrs	r4, cpsr
 84	and	r4, r4, #MODE_MASK
 85
 86	/*
 87	 * If the secondary has booted with a different mode, give up
 88	 * immediately.
 89	 */
 90	compare_cpu_mode_with_primary	r4, r5, r6, r7
 91	retne	lr
 92
 93	/*
 94	 * Once we have given up on one CPU, we do not try to install the
 95	 * stub hypervisor on the remaining ones: because the saved boot mode
 96	 * is modified, it can't compare equal to the CPSR mode field any
 97	 * more.
 98	 *
 99	 * Otherwise...
100	 */
101
102	cmp	r4, #HYP_MODE
103	retne	lr			@ give up if the CPU is not in HYP mode
104
105/*
106 * Configure HSCTLR to set correct exception endianness/instruction set
107 * state etc.
108 * Turn off all traps
109 * Eventually, CPU-specific code might be needed -- assume not for now
110 *
111 * This code relies on the "eret" instruction to synchronize the
112 * various coprocessor accesses. This is done when we switch to SVC
113 * (see safe_svcmode_maskall).
114 */
115	@ Now install the hypervisor stub:
116	W(adr)	r7, __hyp_stub_vectors
117	mcr	p15, 4, r7, c12, c0, 0	@ set hypervisor vector base (HVBAR)
118
119	@ Disable all traps, so we don't get any nasty surprise
120	mov	r7, #0
121	mcr	p15, 4, r7, c1, c1, 0	@ HCR
122	mcr	p15, 4, r7, c1, c1, 2	@ HCPTR
123	mcr	p15, 4, r7, c1, c1, 3	@ HSTR
124
125THUMB(	orr	r7, #(1 << 30)	)	@ HSCTLR.TE
126ARM_BE8(orr	r7, r7, #(1 << 25))     @ HSCTLR.EE
127	mcr	p15, 4, r7, c1, c0, 0	@ HSCTLR
128
129	mrc	p15, 4, r7, c1, c1, 1	@ HDCR
130	and	r7, #0x1f		@ Preserve HPMN
131	mcr	p15, 4, r7, c1, c1, 1	@ HDCR
132
133	@ Make sure NS-SVC is initialised appropriately
134	mrc	p15, 0, r7, c1, c0, 0	@ SCTLR
135	orr	r7, #(1 << 5)		@ CP15 barriers enabled
136	bic	r7, #(3 << 7)		@ Clear SED/ITD for v8 (RES0 for v7)
137	bic	r7, #(3 << 19)		@ WXN and UWXN disabled
138	mcr	p15, 0, r7, c1, c0, 0	@ SCTLR
139
140	mrc	p15, 0, r7, c0, c0, 0	@ MIDR
141	mcr	p15, 4, r7, c0, c0, 0	@ VPIDR
142
143	mrc	p15, 0, r7, c0, c0, 5	@ MPIDR
144	mcr	p15, 4, r7, c0, c0, 5	@ VMPIDR
145
146#if !defined(ZIMAGE) && defined(CONFIG_ARM_ARCH_TIMER)
147	@ make CNTP_* and CNTPCT accessible from PL1
148	mrc	p15, 0, r7, c0, c1, 1	@ ID_PFR1
149	ubfx	r7, r7, #16, #4
150	teq	r7, #0
151	beq	1f
152	mrc	p15, 4, r7, c14, c1, 0	@ CNTHCTL
153	orr	r7, r7, #3		@ PL1PCEN | PL1PCTEN
154	mcr	p15, 4, r7, c14, c1, 0	@ CNTHCTL
155	mov	r7, #0
156	mcrr	p15, 4, r7, r7, c14	@ CNTVOFF
157
158	@ Disable virtual timer in case it was counting
159	mrc	p15, 0, r7, c14, c3, 1	@ CNTV_CTL
160	bic	r7, #1			@ Clear ENABLE
161	mcr	p15, 0, r7, c14, c3, 1	@ CNTV_CTL
1621:
163#endif
164
165#ifdef CONFIG_ARM_GIC_V3
166	@ Check whether GICv3 system registers are available
167	mrc	p15, 0, r7, c0, c1, 1	@ ID_PFR1
168	ubfx	r7, r7, #28, #4
169	teq	r7, #0
170	beq	2f
171
172	@ Enable system register accesses
173	mrc	p15, 4, r7, c12, c9, 5	@ ICC_HSRE
174	orr	r7, r7, #(ICC_SRE_EL2_ENABLE | ICC_SRE_EL2_SRE)
175	mcr	p15, 4, r7, c12, c9, 5	@ ICC_HSRE
176	isb
177
178	@ SRE bit could be forced to 0 by firmware.
179	@ Check whether it sticks before accessing any other sysreg
180	mrc	p15, 4, r7, c12, c9, 5	@ ICC_HSRE
181	tst	r7, #ICC_SRE_EL2_SRE
182	beq	2f
183	mov	r7, #0
184	mcr	p15, 4, r7, c12, c11, 0	@ ICH_HCR
1852:
186#endif
187
188	bx	lr			@ The boot CPU mode is left in r4.
189ENDPROC(__hyp_stub_install_secondary)
190
191__hyp_stub_do_trap:
192#ifdef ZIMAGE
193	teq	r0, #HVC_SET_VECTORS
194	bne	1f
195	/* Only the ZIMAGE stubs can change the HYP vectors */
196	mcr	p15, 4, r1, c12, c0, 0	@ set HVBAR
197	b	__hyp_stub_exit
198#endif
199
2001:	teq	r0, #HVC_SOFT_RESTART
201	bne	2f
202	bx	r1
203
2042:	ldr	r0, =HVC_STUB_ERR
205	__ERET
206
207__hyp_stub_exit:
208	mov	r0, #0
209	__ERET
210ENDPROC(__hyp_stub_do_trap)
211
212/*
213 * __hyp_set_vectors is only used when ZIMAGE must bounce between HYP
214 * and SVC. For the kernel itself, the vectors are set once and for
215 * all by the stubs.
216 */
217ENTRY(__hyp_set_vectors)
218	mov	r1, r0
219	mov	r0, #HVC_SET_VECTORS
220	__HVC(0)
221	ret	lr
222ENDPROC(__hyp_set_vectors)
223
224ENTRY(__hyp_soft_restart)
225	mov	r1, r0
226	mov	r0, #HVC_SOFT_RESTART
227	__HVC(0)
228	ret	lr
229ENDPROC(__hyp_soft_restart)
230
231#ifndef ZIMAGE
232.align 2
233.L__boot_cpu_mode_offset:
234	.long	__boot_cpu_mode - .
235#endif
236
237.align 5
238ENTRY(__hyp_stub_vectors)
239__hyp_stub_reset:	W(b)	.
240__hyp_stub_und:		W(b)	.
241__hyp_stub_svc:		W(b)	.
242__hyp_stub_pabort:	W(b)	.
243__hyp_stub_dabort:	W(b)	.
244__hyp_stub_trap:	W(b)	__hyp_stub_do_trap
245__hyp_stub_irq:		W(b)	.
246__hyp_stub_fiq:		W(b)	.
247ENDPROC(__hyp_stub_vectors)
248