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