xref: /linux/arch/arm64/include/asm/kvm_host.h (revision 80476f22b8b7e193b26f285a7c9f9e4b63abca16)
1 /* SPDX-License-Identifier: GPL-2.0-only */
2 /*
3  * Copyright (C) 2012,2013 - ARM Ltd
4  * Author: Marc Zyngier <marc.zyngier@arm.com>
5  *
6  * Derived from arch/arm/include/asm/kvm_host.h:
7  * Copyright (C) 2012 - Virtual Open Systems and Columbia University
8  * Author: Christoffer Dall <c.dall@virtualopensystems.com>
9  */
10 
11 #ifndef __ARM64_KVM_HOST_H__
12 #define __ARM64_KVM_HOST_H__
13 
14 #include <linux/arm-smccc.h>
15 #include <linux/bitmap.h>
16 #include <linux/types.h>
17 #include <linux/jump_label.h>
18 #include <linux/kvm_types.h>
19 #include <linux/maple_tree.h>
20 #include <linux/percpu.h>
21 #include <linux/psci.h>
22 #include <asm/arch_gicv3.h>
23 #include <asm/barrier.h>
24 #include <asm/cpufeature.h>
25 #include <asm/cputype.h>
26 #include <asm/daifflags.h>
27 #include <asm/fpsimd.h>
28 #include <asm/kvm.h>
29 #include <asm/kvm_asm.h>
30 #include <asm/vncr_mapping.h>
31 
32 #define __KVM_HAVE_ARCH_INTC_INITIALIZED
33 
34 #define KVM_HALT_POLL_NS_DEFAULT 500000
35 
36 #include <kvm/arm_vgic.h>
37 #include <kvm/arm_arch_timer.h>
38 #include <kvm/arm_pmu.h>
39 
40 #define KVM_MAX_VCPUS VGIC_V3_MAX_CPUS
41 
42 #define KVM_VCPU_MAX_FEATURES 9
43 #define KVM_VCPU_VALID_FEATURES	(BIT(KVM_VCPU_MAX_FEATURES) - 1)
44 
45 #define KVM_REQ_SLEEP \
46 	KVM_ARCH_REQ_FLAGS(0, KVM_REQUEST_WAIT | KVM_REQUEST_NO_WAKEUP)
47 #define KVM_REQ_IRQ_PENDING		KVM_ARCH_REQ(1)
48 #define KVM_REQ_VCPU_RESET		KVM_ARCH_REQ(2)
49 #define KVM_REQ_RECORD_STEAL		KVM_ARCH_REQ(3)
50 #define KVM_REQ_RELOAD_GICv4		KVM_ARCH_REQ(4)
51 #define KVM_REQ_RELOAD_PMU		KVM_ARCH_REQ(5)
52 #define KVM_REQ_SUSPEND			KVM_ARCH_REQ(6)
53 #define KVM_REQ_RESYNC_PMU_EL0		KVM_ARCH_REQ(7)
54 #define KVM_REQ_NESTED_S2_UNMAP		KVM_ARCH_REQ(8)
55 #define KVM_REQ_GUEST_HYP_IRQ_PENDING	KVM_ARCH_REQ(9)
56 #define KVM_REQ_MAP_L1_VNCR_EL2		KVM_ARCH_REQ(10)
57 #define KVM_REQ_VGIC_PROCESS_UPDATE	KVM_ARCH_REQ(11)
58 
59 #define KVM_DIRTY_LOG_MANUAL_CAPS   (KVM_DIRTY_LOG_MANUAL_PROTECT_ENABLE | \
60 				     KVM_DIRTY_LOG_INITIALLY_SET)
61 
62 #define KVM_HAVE_MMU_RWLOCK
63 
64 /*
65  * Mode of operation configurable with kvm-arm.mode early param.
66  * See Documentation/admin-guide/kernel-parameters.txt for more information.
67  */
68 enum kvm_mode {
69 	KVM_MODE_DEFAULT,
70 	KVM_MODE_PROTECTED,
71 	KVM_MODE_NV,
72 	KVM_MODE_NONE,
73 };
74 #ifdef CONFIG_KVM
75 enum kvm_mode kvm_get_mode(void);
76 #else
77 static inline enum kvm_mode kvm_get_mode(void) { return KVM_MODE_NONE; };
78 #endif
79 
80 extern unsigned int __ro_after_init kvm_sve_max_vl;
81 extern unsigned int __ro_after_init kvm_host_sve_max_vl;
82 int __init kvm_arm_init_sve(void);
83 
84 u32 __attribute_const__ kvm_target_cpu(void);
85 void kvm_reset_vcpu(struct kvm_vcpu *vcpu);
86 void kvm_arm_vcpu_destroy(struct kvm_vcpu *vcpu);
87 
88 struct kvm_hyp_memcache {
89 	phys_addr_t head;
90 	unsigned long nr_pages;
91 	struct pkvm_mapping *mapping; /* only used from EL1 */
92 
93 #define	HYP_MEMCACHE_ACCOUNT_STAGE2	BIT(1)
94 	unsigned long flags;
95 };
96 
97 static inline void push_hyp_memcache(struct kvm_hyp_memcache *mc,
98 				     phys_addr_t *p,
99 				     phys_addr_t (*to_pa)(void *virt))
100 {
101 	*p = mc->head;
102 	mc->head = to_pa(p);
103 	mc->nr_pages++;
104 }
105 
106 static inline void *pop_hyp_memcache(struct kvm_hyp_memcache *mc,
107 				     void *(*to_va)(phys_addr_t phys))
108 {
109 	phys_addr_t *p = to_va(mc->head & PAGE_MASK);
110 
111 	if (!mc->nr_pages)
112 		return NULL;
113 
114 	mc->head = *p;
115 	mc->nr_pages--;
116 
117 	return p;
118 }
119 
120 static inline int __topup_hyp_memcache(struct kvm_hyp_memcache *mc,
121 				       unsigned long min_pages,
122 				       void *(*alloc_fn)(void *arg),
123 				       phys_addr_t (*to_pa)(void *virt),
124 				       void *arg)
125 {
126 	while (mc->nr_pages < min_pages) {
127 		phys_addr_t *p = alloc_fn(arg);
128 
129 		if (!p)
130 			return -ENOMEM;
131 		push_hyp_memcache(mc, p, to_pa);
132 	}
133 
134 	return 0;
135 }
136 
137 static inline void __free_hyp_memcache(struct kvm_hyp_memcache *mc,
138 				       void (*free_fn)(void *virt, void *arg),
139 				       void *(*to_va)(phys_addr_t phys),
140 				       void *arg)
141 {
142 	while (mc->nr_pages)
143 		free_fn(pop_hyp_memcache(mc, to_va), arg);
144 }
145 
146 void free_hyp_memcache(struct kvm_hyp_memcache *mc);
147 int topup_hyp_memcache(struct kvm_hyp_memcache *mc, unsigned long min_pages);
148 
149 struct kvm_vmid {
150 	atomic64_t id;
151 };
152 
153 struct kvm_s2_mmu {
154 	struct kvm_vmid vmid;
155 
156 	/*
157 	 * stage2 entry level table
158 	 *
159 	 * Two kvm_s2_mmu structures in the same VM can point to the same
160 	 * pgd here.  This happens when running a guest using a
161 	 * translation regime that isn't affected by its own stage-2
162 	 * translation, such as a non-VHE hypervisor running at vEL2, or
163 	 * for vEL1/EL0 with vHCR_EL2.VM == 0.  In that case, we use the
164 	 * canonical stage-2 page tables.
165 	 */
166 	phys_addr_t	pgd_phys;
167 	struct kvm_pgtable *pgt;
168 
169 	/*
170 	 * VTCR value used on the host. For a non-NV guest (or a NV
171 	 * guest that runs in a context where its own S2 doesn't
172 	 * apply), its T0SZ value reflects that of the IPA size.
173 	 *
174 	 * For a shadow S2 MMU, T0SZ reflects the PARange exposed to
175 	 * the guest.
176 	 */
177 	u64	vtcr;
178 
179 	/* The last vcpu id that ran on each physical CPU */
180 	int __percpu *last_vcpu_ran;
181 
182 #define KVM_ARM_EAGER_SPLIT_CHUNK_SIZE_DEFAULT 0
183 	/*
184 	 * Memory cache used to split
185 	 * KVM_CAP_ARM_EAGER_SPLIT_CHUNK_SIZE worth of huge pages. It
186 	 * is used to allocate stage2 page tables while splitting huge
187 	 * pages. The choice of KVM_CAP_ARM_EAGER_SPLIT_CHUNK_SIZE
188 	 * influences both the capacity of the split page cache, and
189 	 * how often KVM reschedules. Be wary of raising CHUNK_SIZE
190 	 * too high.
191 	 *
192 	 * Protected by kvm->slots_lock.
193 	 */
194 	struct kvm_mmu_memory_cache split_page_cache;
195 	uint64_t split_page_chunk_size;
196 
197 	struct kvm_arch *arch;
198 
199 	/*
200 	 * For a shadow stage-2 MMU, the virtual vttbr used by the
201 	 * host to parse the guest S2.
202 	 * This either contains:
203 	 * - the virtual VTTBR programmed by the guest hypervisor with
204 	 *   CnP cleared
205 	 * - The value 1 (VMID=0, BADDR=0, CnP=1) if invalid
206 	 *
207 	 * We also cache the full VTCR which gets used for TLB invalidation,
208 	 * taking the ARM ARM's "Any of the bits in VTCR_EL2 are permitted
209 	 * to be cached in a TLB" to the letter.
210 	 */
211 	u64	tlb_vttbr;
212 	u64	tlb_vtcr;
213 
214 	/*
215 	 * true when this represents a nested context where virtual
216 	 * HCR_EL2.VM == 1
217 	 */
218 	bool	nested_stage2_enabled;
219 
220 #ifdef CONFIG_PTDUMP_STAGE2_DEBUGFS
221 	struct dentry *shadow_pt_debugfs_dentry;
222 #endif
223 
224 	/*
225 	 * true when this MMU needs to be unmapped before being used for a new
226 	 * purpose.
227 	 */
228 	bool	pending_unmap;
229 
230 	/*
231 	 *  0: Nobody is currently using this, check vttbr for validity
232 	 * >0: Somebody is actively using this.
233 	 */
234 	atomic_t refcnt;
235 };
236 
237 struct kvm_arch_memory_slot {
238 };
239 
240 /**
241  * struct kvm_smccc_features: Descriptor of the hypercall services exposed to the guests
242  *
243  * @std_bmap: Bitmap of standard secure service calls
244  * @std_hyp_bmap: Bitmap of standard hypervisor service calls
245  * @vendor_hyp_bmap: Bitmap of vendor specific hypervisor service calls
246  */
247 struct kvm_smccc_features {
248 	unsigned long std_bmap;
249 	unsigned long std_hyp_bmap;
250 	unsigned long vendor_hyp_bmap; /* Function numbers 0-63 */
251 	unsigned long vendor_hyp_bmap_2; /* Function numbers 64-127 */
252 };
253 
254 typedef u16 pkvm_handle_t;
255 
256 struct kvm_protected_vm {
257 	pkvm_handle_t handle;
258 	struct kvm_hyp_memcache teardown_mc;
259 	struct kvm_hyp_memcache stage2_teardown_mc;
260 	bool is_protected;
261 	bool is_created;
262 
263 	/*
264 	 * True when the guest is being torn down. When in this state, the
265 	 * guest's vCPUs can't be loaded anymore, but its pages can be
266 	 * reclaimed by the host.
267 	 */
268 	bool is_dying;
269 };
270 
271 struct kvm_mpidr_data {
272 	u64			mpidr_mask;
273 	DECLARE_FLEX_ARRAY(u16, cmpidr_to_idx);
274 };
275 
276 static inline u16 kvm_mpidr_index(struct kvm_mpidr_data *data, u64 mpidr)
277 {
278 	unsigned long index = 0, mask = data->mpidr_mask;
279 	unsigned long aff = mpidr & MPIDR_HWID_BITMASK;
280 
281 	bitmap_gather(&index, &aff, &mask, fls(mask));
282 
283 	return index;
284 }
285 
286 struct kvm_sysreg_masks;
287 
288 enum fgt_group_id {
289 	__NO_FGT_GROUP__,
290 	HFGRTR_GROUP,
291 	HFGWTR_GROUP = HFGRTR_GROUP,
292 	HDFGRTR_GROUP,
293 	HDFGWTR_GROUP = HDFGRTR_GROUP,
294 	HFGITR_GROUP,
295 	HAFGRTR_GROUP,
296 	HFGRTR2_GROUP,
297 	HFGWTR2_GROUP = HFGRTR2_GROUP,
298 	HDFGRTR2_GROUP,
299 	HDFGWTR2_GROUP = HDFGRTR2_GROUP,
300 	HFGITR2_GROUP,
301 	ICH_HFGRTR_GROUP,
302 	ICH_HFGWTR_GROUP = ICH_HFGRTR_GROUP,
303 	ICH_HFGITR_GROUP,
304 
305 	/* Must be last */
306 	__NR_FGT_GROUP_IDS__
307 };
308 
309 struct kvm_arch {
310 	struct kvm_s2_mmu mmu;
311 
312 	/*
313 	 * Fine-Grained UNDEF, mimicking the FGT layout defined by the
314 	 * architecture. We track them globally, as we present the
315 	 * same feature-set to all vcpus.
316 	 *
317 	 * Index 0 is currently spare.
318 	 */
319 	u64 fgu[__NR_FGT_GROUP_IDS__];
320 
321 	/*
322 	 * Stage 2 paging state for VMs with nested S2 using a virtual
323 	 * VMID.
324 	 */
325 	struct kvm_s2_mmu *nested_mmus;
326 	size_t nested_mmus_size;
327 	int nested_mmus_next;
328 
329 	/* Interrupt controller */
330 	struct vgic_dist	vgic;
331 
332 	/* Timers */
333 	struct arch_timer_vm_data timer_data;
334 
335 	/* Mandated version of PSCI */
336 	u32 psci_version;
337 
338 	/* Protects VM-scoped configuration data */
339 	struct mutex config_lock;
340 
341 	/*
342 	 * If we encounter a data abort without valid instruction syndrome
343 	 * information, report this to user space.  User space can (and
344 	 * should) opt in to this feature if KVM_CAP_ARM_NISV_TO_USER is
345 	 * supported.
346 	 */
347 #define KVM_ARCH_FLAG_RETURN_NISV_IO_ABORT_TO_USER	0
348 	/* Memory Tagging Extension enabled for the guest */
349 #define KVM_ARCH_FLAG_MTE_ENABLED			1
350 	/* At least one vCPU has ran in the VM */
351 #define KVM_ARCH_FLAG_HAS_RAN_ONCE			2
352 	/* The vCPU feature set for the VM is configured */
353 #define KVM_ARCH_FLAG_VCPU_FEATURES_CONFIGURED		3
354 	/* PSCI SYSTEM_SUSPEND enabled for the guest */
355 #define KVM_ARCH_FLAG_SYSTEM_SUSPEND_ENABLED		4
356 	/* VM counter offset */
357 #define KVM_ARCH_FLAG_VM_COUNTER_OFFSET			5
358 	/* Timer PPIs made immutable */
359 #define KVM_ARCH_FLAG_TIMER_PPIS_IMMUTABLE		6
360 	/* Initial ID reg values loaded */
361 #define KVM_ARCH_FLAG_ID_REGS_INITIALIZED		7
362 	/* Fine-Grained UNDEF initialised */
363 #define KVM_ARCH_FLAG_FGU_INITIALIZED			8
364 	/* SVE exposed to guest */
365 #define KVM_ARCH_FLAG_GUEST_HAS_SVE			9
366 	/* MIDR_EL1, REVIDR_EL1, and AIDR_EL1 are writable from userspace */
367 #define KVM_ARCH_FLAG_WRITABLE_IMP_ID_REGS		10
368 	/* Unhandled SEAs are taken to userspace */
369 #define KVM_ARCH_FLAG_EXIT_SEA				11
370 	unsigned long flags;
371 
372 	/* VM-wide vCPU feature set */
373 	DECLARE_BITMAP(vcpu_features, KVM_VCPU_MAX_FEATURES);
374 
375 	/* MPIDR to vcpu index mapping, optional */
376 	struct kvm_mpidr_data *mpidr_data;
377 
378 	/*
379 	 * VM-wide PMU filter, implemented as a bitmap and big enough for
380 	 * up to 2^10 events (ARMv8.0) or 2^16 events (ARMv8.1+).
381 	 */
382 	unsigned long *pmu_filter;
383 	struct arm_pmu *arm_pmu;
384 
385 	cpumask_var_t supported_cpus;
386 
387 	/* Maximum number of counters for the guest */
388 	u8 nr_pmu_counters;
389 
390 	/* Hypercall features firmware registers' descriptor */
391 	struct kvm_smccc_features smccc_feat;
392 	struct maple_tree smccc_filter;
393 
394 	/*
395 	 * Emulated CPU ID registers per VM
396 	 * (Op0, Op1, CRn, CRm, Op2) of the ID registers to be saved in it
397 	 * is (3, 0, 0, crm, op2), where 1<=crm<8, 0<=op2<8.
398 	 *
399 	 * These emulated idregs are VM-wide, but accessed from the context of a vCPU.
400 	 * Atomic access to multiple idregs are guarded by kvm_arch.config_lock.
401 	 */
402 #define IDREG_IDX(id)		(((sys_reg_CRm(id) - 1) << 3) | sys_reg_Op2(id))
403 #define KVM_ARM_ID_REG_NUM	(IDREG_IDX(sys_reg(3, 0, 0, 7, 7)) + 1)
404 	u64 id_regs[KVM_ARM_ID_REG_NUM];
405 
406 	u64 midr_el1;
407 	u64 revidr_el1;
408 	u64 aidr_el1;
409 	u64 ctr_el0;
410 
411 	/* Masks for VNCR-backed and general EL2 sysregs */
412 	struct kvm_sysreg_masks	*sysreg_masks;
413 
414 	/* Count the number of VNCR_EL2 currently mapped */
415 	atomic_t vncr_map_count;
416 
417 	/*
418 	 * For an untrusted host VM, 'pkvm.handle' is used to lookup
419 	 * the associated pKVM instance in the hypervisor.
420 	 */
421 	struct kvm_protected_vm pkvm;
422 
423 #ifdef CONFIG_PTDUMP_STAGE2_DEBUGFS
424 	/* Nested virtualization info */
425 	struct dentry *debugfs_nv_dentry;
426 #endif
427 };
428 
429 struct kvm_vcpu_fault_info {
430 	u64 esr_el2;		/* Hyp Syndrom Register */
431 	u64 far_el2;		/* Hyp Fault Address Register */
432 	u64 hpfar_el2;		/* Hyp IPA Fault Address Register */
433 	u64 disr_el1;		/* Deferred [SError] Status Register */
434 };
435 
436 /*
437  * VNCR() just places the VNCR_capable registers in the enum after
438  * __VNCR_START__, and the value (after correction) to be an 8-byte offset
439  * from the VNCR base. As we don't require the enum to be otherwise ordered,
440  * we need the terrible hack below to ensure that we correctly size the
441  * sys_regs array, no matter what.
442  *
443  * The __MAX__ macro has been lifted from Sean Eron Anderson's wonderful
444  * treasure trove of bit hacks:
445  * https://graphics.stanford.edu/~seander/bithacks.html#IntegerMinOrMax
446  */
447 #define __MAX__(x,y)	((x) ^ (((x) ^ (y)) & -((x) < (y))))
448 #define VNCR(r)						\
449 	__before_##r,					\
450 	r = __VNCR_START__ + ((VNCR_ ## r) / 8),	\
451 	__after_##r = __MAX__(__before_##r - 1, r)
452 
453 enum vcpu_sysreg {
454 	__INVALID_SYSREG__,   /* 0 is reserved as an invalid value */
455 	MPIDR_EL1,	/* MultiProcessor Affinity Register */
456 	CLIDR_EL1,	/* Cache Level ID Register */
457 	CSSELR_EL1,	/* Cache Size Selection Register */
458 	TPIDR_EL0,	/* Thread ID, User R/W */
459 	TPIDRRO_EL0,	/* Thread ID, User R/O */
460 	TPIDR_EL1,	/* Thread ID, Privileged */
461 	CNTKCTL_EL1,	/* Timer Control Register (EL1) */
462 	PAR_EL1,	/* Physical Address Register */
463 	MDCCINT_EL1,	/* Monitor Debug Comms Channel Interrupt Enable Reg */
464 	OSLSR_EL1,	/* OS Lock Status Register */
465 	DISR_EL1,	/* Deferred Interrupt Status Register */
466 
467 	/* Performance Monitors Registers */
468 	PMCR_EL0,	/* Control Register */
469 	PMSELR_EL0,	/* Event Counter Selection Register */
470 	PMEVCNTR0_EL0,	/* Event Counter Register (0-30) */
471 	PMEVCNTR30_EL0 = PMEVCNTR0_EL0 + 30,
472 	PMCCNTR_EL0,	/* Cycle Counter Register */
473 	PMEVTYPER0_EL0,	/* Event Type Register (0-30) */
474 	PMEVTYPER30_EL0 = PMEVTYPER0_EL0 + 30,
475 	PMCCFILTR_EL0,	/* Cycle Count Filter Register */
476 	PMCNTENSET_EL0,	/* Count Enable Set Register */
477 	PMINTENSET_EL1,	/* Interrupt Enable Set Register */
478 	PMOVSSET_EL0,	/* Overflow Flag Status Set Register */
479 	PMUSERENR_EL0,	/* User Enable Register */
480 
481 	/* Pointer Authentication Registers in a strict increasing order. */
482 	APIAKEYLO_EL1,
483 	APIAKEYHI_EL1,
484 	APIBKEYLO_EL1,
485 	APIBKEYHI_EL1,
486 	APDAKEYLO_EL1,
487 	APDAKEYHI_EL1,
488 	APDBKEYLO_EL1,
489 	APDBKEYHI_EL1,
490 	APGAKEYLO_EL1,
491 	APGAKEYHI_EL1,
492 
493 	/* Memory Tagging Extension registers */
494 	RGSR_EL1,	/* Random Allocation Tag Seed Register */
495 	GCR_EL1,	/* Tag Control Register */
496 	TFSRE0_EL1,	/* Tag Fault Status Register (EL0) */
497 
498 	POR_EL0,	/* Permission Overlay Register 0 (EL0) */
499 
500 	/* FP/SIMD/SVE */
501 	SVCR,
502 	FPMR,
503 
504 	/* 32bit specific registers. */
505 	DACR32_EL2,	/* Domain Access Control Register */
506 	IFSR32_EL2,	/* Instruction Fault Status Register */
507 	FPEXC32_EL2,	/* Floating-Point Exception Control Register */
508 	DBGVCR32_EL2,	/* Debug Vector Catch Register */
509 
510 	/* EL2 registers */
511 	ACTLR_EL2,	/* Auxiliary Control Register (EL2) */
512 	CPTR_EL2,	/* Architectural Feature Trap Register (EL2) */
513 	HACR_EL2,	/* Hypervisor Auxiliary Control Register */
514 	TTBR0_EL2,	/* Translation Table Base Register 0 (EL2) */
515 	TTBR1_EL2,	/* Translation Table Base Register 1 (EL2) */
516 	TCR_EL2,	/* Translation Control Register (EL2) */
517 	PIRE0_EL2,	/* Permission Indirection Register 0 (EL2) */
518 	PIR_EL2,	/* Permission Indirection Register 1 (EL2) */
519 	POR_EL2,	/* Permission Overlay Register 2 (EL2) */
520 	SPSR_EL2,	/* EL2 saved program status register */
521 	ELR_EL2,	/* EL2 exception link register */
522 	AFSR0_EL2,	/* Auxiliary Fault Status Register 0 (EL2) */
523 	AFSR1_EL2,	/* Auxiliary Fault Status Register 1 (EL2) */
524 	ESR_EL2,	/* Exception Syndrome Register (EL2) */
525 	FAR_EL2,	/* Fault Address Register (EL2) */
526 	HPFAR_EL2,	/* Hypervisor IPA Fault Address Register */
527 	MAIR_EL2,	/* Memory Attribute Indirection Register (EL2) */
528 	AMAIR_EL2,	/* Auxiliary Memory Attribute Indirection Register (EL2) */
529 	VBAR_EL2,	/* Vector Base Address Register (EL2) */
530 	RVBAR_EL2,	/* Reset Vector Base Address Register */
531 	CONTEXTIDR_EL2,	/* Context ID Register (EL2) */
532 	SP_EL2,		/* EL2 Stack Pointer */
533 	CNTHP_CTL_EL2,
534 	CNTHP_CVAL_EL2,
535 	CNTHV_CTL_EL2,
536 	CNTHV_CVAL_EL2,
537 
538 	/* Anything from this can be RES0/RES1 sanitised */
539 	MARKER(__SANITISED_REG_START__),
540 	SCTLR_EL2,	/* System Control Register (EL2) */
541 	TCR2_EL2,	/* Extended Translation Control Register (EL2) */
542 	SCTLR2_EL2,	/* System Control Register 2 (EL2) */
543 	MDCR_EL2,	/* Monitor Debug Configuration Register (EL2) */
544 	CNTHCTL_EL2,	/* Counter-timer Hypervisor Control register */
545 	ZCR_EL2,	/* SVE Control Register (EL2) */
546 
547 	/* Any VNCR-capable reg goes after this point */
548 	MARKER(__VNCR_START__),
549 
550 	VNCR(SCTLR_EL1),/* System Control Register */
551 	VNCR(ACTLR_EL1),/* Auxiliary Control Register */
552 	VNCR(CPACR_EL1),/* Coprocessor Access Control */
553 	VNCR(ZCR_EL1),	/* SVE Control */
554 	VNCR(TTBR0_EL1),/* Translation Table Base Register 0 */
555 	VNCR(TTBR1_EL1),/* Translation Table Base Register 1 */
556 	VNCR(TCR_EL1),	/* Translation Control Register */
557 	VNCR(TCR2_EL1),	/* Extended Translation Control Register */
558 	VNCR(SCTLR2_EL1), /* System Control Register 2 */
559 	VNCR(ESR_EL1),	/* Exception Syndrome Register */
560 	VNCR(AFSR0_EL1),/* Auxiliary Fault Status Register 0 */
561 	VNCR(AFSR1_EL1),/* Auxiliary Fault Status Register 1 */
562 	VNCR(FAR_EL1),	/* Fault Address Register */
563 	VNCR(MAIR_EL1),	/* Memory Attribute Indirection Register */
564 	VNCR(VBAR_EL1),	/* Vector Base Address Register */
565 	VNCR(CONTEXTIDR_EL1),	/* Context ID Register */
566 	VNCR(AMAIR_EL1),/* Aux Memory Attribute Indirection Register */
567 	VNCR(MDSCR_EL1),/* Monitor Debug System Control Register */
568 	VNCR(ELR_EL1),
569 	VNCR(SP_EL1),
570 	VNCR(SPSR_EL1),
571 	VNCR(TFSR_EL1),	/* Tag Fault Status Register (EL1) */
572 	VNCR(VPIDR_EL2),/* Virtualization Processor ID Register */
573 	VNCR(VMPIDR_EL2),/* Virtualization Multiprocessor ID Register */
574 	VNCR(HCR_EL2),	/* Hypervisor Configuration Register */
575 	VNCR(HSTR_EL2),	/* Hypervisor System Trap Register */
576 	VNCR(VTTBR_EL2),/* Virtualization Translation Table Base Register */
577 	VNCR(VTCR_EL2),	/* Virtualization Translation Control Register */
578 	VNCR(TPIDR_EL2),/* EL2 Software Thread ID Register */
579 	VNCR(HCRX_EL2),	/* Extended Hypervisor Configuration Register */
580 
581 	/* Permission Indirection Extension registers */
582 	VNCR(PIR_EL1),	 /* Permission Indirection Register 1 (EL1) */
583 	VNCR(PIRE0_EL1), /*  Permission Indirection Register 0 (EL1) */
584 
585 	VNCR(POR_EL1),	/* Permission Overlay Register 1 (EL1) */
586 
587 	/* FEAT_RAS registers */
588 	VNCR(VDISR_EL2),
589 	VNCR(VSESR_EL2),
590 
591 	VNCR(HFGRTR_EL2),
592 	VNCR(HFGWTR_EL2),
593 	VNCR(HFGITR_EL2),
594 	VNCR(HDFGRTR_EL2),
595 	VNCR(HDFGWTR_EL2),
596 	VNCR(HAFGRTR_EL2),
597 	VNCR(HFGRTR2_EL2),
598 	VNCR(HFGWTR2_EL2),
599 	VNCR(HFGITR2_EL2),
600 	VNCR(HDFGRTR2_EL2),
601 	VNCR(HDFGWTR2_EL2),
602 
603 	VNCR(VNCR_EL2),
604 
605 	VNCR(CNTVOFF_EL2),
606 	VNCR(CNTV_CVAL_EL0),
607 	VNCR(CNTV_CTL_EL0),
608 	VNCR(CNTP_CVAL_EL0),
609 	VNCR(CNTP_CTL_EL0),
610 
611 	VNCR(ICH_LR0_EL2),
612 	VNCR(ICH_LR1_EL2),
613 	VNCR(ICH_LR2_EL2),
614 	VNCR(ICH_LR3_EL2),
615 	VNCR(ICH_LR4_EL2),
616 	VNCR(ICH_LR5_EL2),
617 	VNCR(ICH_LR6_EL2),
618 	VNCR(ICH_LR7_EL2),
619 	VNCR(ICH_LR8_EL2),
620 	VNCR(ICH_LR9_EL2),
621 	VNCR(ICH_LR10_EL2),
622 	VNCR(ICH_LR11_EL2),
623 	VNCR(ICH_LR12_EL2),
624 	VNCR(ICH_LR13_EL2),
625 	VNCR(ICH_LR14_EL2),
626 	VNCR(ICH_LR15_EL2),
627 
628 	VNCR(ICH_AP0R0_EL2),
629 	VNCR(ICH_AP0R1_EL2),
630 	VNCR(ICH_AP0R2_EL2),
631 	VNCR(ICH_AP0R3_EL2),
632 	VNCR(ICH_AP1R0_EL2),
633 	VNCR(ICH_AP1R1_EL2),
634 	VNCR(ICH_AP1R2_EL2),
635 	VNCR(ICH_AP1R3_EL2),
636 	VNCR(ICH_HCR_EL2),
637 	VNCR(ICH_VMCR_EL2),
638 
639 	VNCR(ICH_HFGRTR_EL2),
640 	VNCR(ICH_HFGWTR_EL2),
641 	VNCR(ICH_HFGITR_EL2),
642 
643 	NR_SYS_REGS	/* Nothing after this line! */
644 };
645 
646 struct resx {
647 	u64	res0;
648 	u64	res1;
649 };
650 
651 struct kvm_sysreg_masks {
652 	struct resx mask[NR_SYS_REGS - __SANITISED_REG_START__];
653 };
654 
655 static inline struct resx __kvm_get_sysreg_resx(struct kvm_arch *arch,
656 						enum vcpu_sysreg sr)
657 {
658 	struct kvm_sysreg_masks *masks;
659 
660 	masks = arch->sysreg_masks;
661 	if (likely(masks &&
662 		   sr >= __SANITISED_REG_START__ && sr < NR_SYS_REGS))
663 		return masks->mask[sr - __SANITISED_REG_START__];
664 
665 	return (struct resx){};
666 }
667 
668 #define kvm_get_sysreg_resx(k, sr) __kvm_get_sysreg_resx(&(k)->arch, (sr))
669 
670 static inline void __kvm_set_sysreg_resx(struct kvm_arch *arch,
671 					 enum vcpu_sysreg sr, struct resx resx)
672 {
673 	arch->sysreg_masks->mask[sr - __SANITISED_REG_START__] = resx;
674 }
675 
676 #define kvm_set_sysreg_resx(k, sr, resx)		\
677 	__kvm_set_sysreg_resx(&(k)->arch, (sr), (resx))
678 
679 struct fgt_masks {
680 	const char	*str;
681 	u64		mask;
682 	u64		nmask;
683 	u64		res0;
684 	u64		res1;
685 };
686 
687 extern struct fgt_masks hfgrtr_masks;
688 extern struct fgt_masks hfgwtr_masks;
689 extern struct fgt_masks hfgitr_masks;
690 extern struct fgt_masks hdfgrtr_masks;
691 extern struct fgt_masks hdfgwtr_masks;
692 extern struct fgt_masks hafgrtr_masks;
693 extern struct fgt_masks hfgrtr2_masks;
694 extern struct fgt_masks hfgwtr2_masks;
695 extern struct fgt_masks hfgitr2_masks;
696 extern struct fgt_masks hdfgrtr2_masks;
697 extern struct fgt_masks hdfgwtr2_masks;
698 extern struct fgt_masks ich_hfgrtr_masks;
699 extern struct fgt_masks ich_hfgwtr_masks;
700 extern struct fgt_masks ich_hfgitr_masks;
701 
702 extern struct fgt_masks kvm_nvhe_sym(hfgrtr_masks);
703 extern struct fgt_masks kvm_nvhe_sym(hfgwtr_masks);
704 extern struct fgt_masks kvm_nvhe_sym(hfgitr_masks);
705 extern struct fgt_masks kvm_nvhe_sym(hdfgrtr_masks);
706 extern struct fgt_masks kvm_nvhe_sym(hdfgwtr_masks);
707 extern struct fgt_masks kvm_nvhe_sym(hafgrtr_masks);
708 extern struct fgt_masks kvm_nvhe_sym(hfgrtr2_masks);
709 extern struct fgt_masks kvm_nvhe_sym(hfgwtr2_masks);
710 extern struct fgt_masks kvm_nvhe_sym(hfgitr2_masks);
711 extern struct fgt_masks kvm_nvhe_sym(hdfgrtr2_masks);
712 extern struct fgt_masks kvm_nvhe_sym(hdfgwtr2_masks);
713 extern struct fgt_masks kvm_nvhe_sym(ich_hfgrtr_masks);
714 extern struct fgt_masks kvm_nvhe_sym(ich_hfgwtr_masks);
715 extern struct fgt_masks kvm_nvhe_sym(ich_hfgitr_masks);
716 
717 struct kvm_cpu_context {
718 	struct user_pt_regs regs;	/* sp = sp_el0 */
719 
720 	u64	spsr_abt;
721 	u64	spsr_und;
722 	u64	spsr_irq;
723 	u64	spsr_fiq;
724 
725 	struct user_fpsimd_state fp_regs;
726 
727 	u64 sys_regs[NR_SYS_REGS];
728 
729 	struct kvm_vcpu *__hyp_running_vcpu;
730 
731 	/* This pointer has to be 4kB aligned. */
732 	u64 *vncr_array;
733 };
734 
735 /*
736  * This structure is instantiated on a per-CPU basis, and contains
737  * data that is:
738  *
739  * - tied to a single physical CPU, and
740  * - either have a lifetime that does not extend past vcpu_put()
741  * - or is an invariant for the lifetime of the system
742  *
743  * Use host_data_ptr(field) as a way to access a pointer to such a
744  * field.
745  */
746 struct kvm_host_data {
747 #define KVM_HOST_DATA_FLAG_HAS_SPE			0
748 #define KVM_HOST_DATA_FLAG_HAS_TRBE			1
749 #define KVM_HOST_DATA_FLAG_TRBE_ENABLED			2
750 #define KVM_HOST_DATA_FLAG_EL1_TRACING_CONFIGURED	3
751 #define KVM_HOST_DATA_FLAG_VCPU_IN_HYP_CONTEXT		4
752 #define KVM_HOST_DATA_FLAG_L1_VNCR_MAPPED		5
753 #define KVM_HOST_DATA_FLAG_HAS_BRBE			6
754 	unsigned long flags;
755 
756 	struct kvm_cpu_context host_ctxt;
757 
758 	/*
759 	 * Hyp VA.
760 	 * sve_regs is only used in pKVM and if system_supports_sve().
761 	 */
762 	struct arm64_sve_state *sve_regs;
763 
764 	/* Ownership of the FP regs */
765 	enum {
766 		FP_STATE_FREE,
767 		FP_STATE_HOST_OWNED,
768 		FP_STATE_GUEST_OWNED,
769 	} fp_owner;
770 
771 	/*
772 	 * host_debug_state contains the host registers which are
773 	 * saved and restored during world switches.
774 	 */
775 	struct {
776 		/* {Break,watch}point registers */
777 		struct kvm_guest_debug_arch regs;
778 		/* Statistical profiling extension */
779 		u64 pmscr_el1;
780 		u64 pmblimitr_el1;
781 		/* Self-hosted trace */
782 		u64 trfcr_el1;
783 		u64 trblimitr_el1;
784 		/* Values of trap registers for the host before guest entry. */
785 		u64 mdcr_el2;
786 		u64 brbcr_el1;
787 	} host_debug_state;
788 
789 	/* Guest trace filter value */
790 	u64 trfcr_while_in_guest;
791 
792 	/* Number of programmable event counters (PMCR_EL0.N) for this CPU */
793 	unsigned int nr_event_counters;
794 
795 	/* Number of debug breakpoints/watchpoints for this CPU (minus 1) */
796 	unsigned int debug_brps;
797 	unsigned int debug_wrps;
798 
799 	/* Last vgic_irq part of the AP list recorded in an LR */
800 	struct vgic_irq *last_lr_irq;
801 
802 	/* PPI state tracking for GICv5-based guests */
803 	struct {
804 		DECLARE_BITMAP(pendr, VGIC_V5_NR_PRIVATE_IRQS);
805 
806 		/* The saved state of the regs when leaving the guest */
807 		DECLARE_BITMAP(activer_exit, VGIC_V5_NR_PRIVATE_IRQS);
808 	} vgic_v5_ppi_state;
809 };
810 
811 struct kvm_host_psci_config {
812 	/* PSCI version used by host. */
813 	u32 version;
814 	u32 smccc_version;
815 
816 	/* Function IDs used by host if version is v0.1. */
817 	struct psci_0_1_function_ids function_ids_0_1;
818 
819 	bool psci_0_1_cpu_suspend_implemented;
820 	bool psci_0_1_cpu_on_implemented;
821 	bool psci_0_1_cpu_off_implemented;
822 	bool psci_0_1_migrate_implemented;
823 };
824 
825 extern struct kvm_host_psci_config kvm_nvhe_sym(kvm_host_psci_config);
826 #define kvm_host_psci_config CHOOSE_NVHE_SYM(kvm_host_psci_config)
827 
828 extern s64 kvm_nvhe_sym(hyp_physvirt_offset);
829 #define hyp_physvirt_offset CHOOSE_NVHE_SYM(hyp_physvirt_offset)
830 
831 extern u64 kvm_nvhe_sym(hyp_cpu_logical_map)[NR_CPUS];
832 #define hyp_cpu_logical_map CHOOSE_NVHE_SYM(hyp_cpu_logical_map)
833 
834 struct vcpu_reset_state {
835 	unsigned long	pc;
836 	unsigned long	r0;
837 	bool		be;
838 	bool		reset;
839 };
840 
841 struct vncr_tlb;
842 
843 struct kvm_vcpu_arch {
844 	struct kvm_cpu_context ctxt;
845 
846 	/*
847 	 * Guest floating point state
848 	 *
849 	 * The architecture has two main floating point extensions,
850 	 * the original FPSIMD and SVE.  These have overlapping
851 	 * register views, with the FPSIMD V registers occupying the
852 	 * low 128 bits of the SVE Z registers.  When the core
853 	 * floating point code saves the register state of a task it
854 	 * records which view it saved in fp_type.
855 	 */
856 	struct arm64_sve_state *sve_state;
857 	enum fp_type fp_type;
858 	unsigned int sve_max_vl;
859 
860 	/* Stage 2 paging state used by the hardware on next switch */
861 	struct kvm_s2_mmu *hw_mmu;
862 
863 	/* Values of trap registers for the guest. */
864 	u64 hcr_el2;
865 	u64 hcrx_el2;
866 	u64 mdcr_el2;
867 
868 	struct {
869 		u64 r;
870 		u64 w;
871 	} fgt[__NR_FGT_GROUP_IDS__];
872 
873 	/* Exception Information */
874 	struct kvm_vcpu_fault_info fault;
875 
876 	/* Configuration flags, set once and for all before the vcpu can run */
877 	u8 cflags;
878 
879 	/* Input flags to the hypervisor code, potentially cleared after use */
880 	u8 iflags;
881 
882 	/* State flags for kernel bookkeeping, unused by the hypervisor code */
883 	u16 sflags;
884 
885 	/*
886 	 * Don't run the guest (internal implementation need).
887 	 *
888 	 * Contrary to the flags above, this is set/cleared outside of
889 	 * a vcpu context, and thus cannot be mixed with the flags
890 	 * themselves (or the flag accesses need to be made atomic).
891 	 */
892 	bool pause;
893 
894 	/*
895 	 * We maintain more than a single set of debug registers to support
896 	 * debugging the guest from the host and to maintain separate host and
897 	 * guest state during world switches. vcpu_debug_state are the debug
898 	 * registers of the vcpu as the guest sees them.
899 	 *
900 	 * external_debug_state contains the debug values we want to debug the
901 	 * guest. This is set via the KVM_SET_GUEST_DEBUG ioctl.
902 	 */
903 	struct kvm_guest_debug_arch vcpu_debug_state;
904 	struct kvm_guest_debug_arch external_debug_state;
905 	u64 external_mdscr_el1;
906 
907 	enum {
908 		VCPU_DEBUG_FREE,
909 		VCPU_DEBUG_HOST_OWNED,
910 		VCPU_DEBUG_GUEST_OWNED,
911 	} debug_owner;
912 
913 	/* VGIC state */
914 	struct vgic_cpu vgic_cpu;
915 	struct arch_timer_cpu timer_cpu;
916 	struct kvm_pmu pmu;
917 
918 	/* vcpu power state */
919 	struct kvm_mp_state mp_state;
920 	spinlock_t mp_state_lock;
921 
922 	/* Cache some mmu pages needed inside spinlock regions */
923 	struct kvm_mmu_memory_cache mmu_page_cache;
924 
925 	/* Pages to top-up the pKVM/EL2 guest pool */
926 	struct kvm_hyp_memcache pkvm_memcache;
927 
928 	/* Virtual SError ESR to restore when HCR_EL2.VSE is set */
929 	u64 vsesr_el2;
930 
931 	/* Additional reset state */
932 	struct vcpu_reset_state	reset_state;
933 
934 	/* Guest PV state */
935 	struct {
936 		u64 last_steal;
937 		gpa_t base;
938 	} steal;
939 
940 	/* Per-vcpu CCSIDR override or NULL */
941 	u32 *ccsidr;
942 
943 	/* Per-vcpu TLB for VNCR_EL2 -- NULL when !NV */
944 	struct vncr_tlb	*vncr_tlb;
945 
946 	/* Hyp-readable copy of kvm_vcpu::pid */
947 	pid_t pid;
948 };
949 
950 /*
951  * Each 'flag' is composed of a comma-separated triplet:
952  *
953  * - the flag-set it belongs to in the vcpu->arch structure
954  * - the value for that flag
955  * - the mask for that flag
956  *
957  *  __vcpu_single_flag() builds such a triplet for a single-bit flag.
958  * unpack_vcpu_flag() extract the flag value from the triplet for
959  * direct use outside of the flag accessors.
960  */
961 #define __vcpu_single_flag(_set, _f)	_set, (_f), (_f)
962 
963 #define __unpack_flag(_set, _f, _m)	_f
964 #define unpack_vcpu_flag(...)		__unpack_flag(__VA_ARGS__)
965 
966 #define __build_check_flag(v, flagset, f, m)			\
967 	do {							\
968 		typeof(v->arch.flagset) *_fset;			\
969 								\
970 		/* Check that the flags fit in the mask */	\
971 		BUILD_BUG_ON(HWEIGHT(m) != HWEIGHT((f) | (m)));	\
972 		/* Check that the flags fit in the type */	\
973 		BUILD_BUG_ON((sizeof(*_fset) * 8) <= __fls(m));	\
974 	} while (0)
975 
976 #define __vcpu_get_flag(v, flagset, f, m)			\
977 	({							\
978 		__build_check_flag(v, flagset, f, m);		\
979 								\
980 		READ_ONCE(v->arch.flagset) & (m);		\
981 	})
982 
983 /*
984  * Note that the set/clear accessors must be preempt-safe in order to
985  * avoid nesting them with load/put which also manipulate flags...
986  */
987 #ifdef __KVM_NVHE_HYPERVISOR__
988 /* the nVHE hypervisor is always non-preemptible */
989 #define __vcpu_flags_preempt_disable()
990 #define __vcpu_flags_preempt_enable()
991 #else
992 #define __vcpu_flags_preempt_disable()	preempt_disable()
993 #define __vcpu_flags_preempt_enable()	preempt_enable()
994 #endif
995 
996 #define __vcpu_set_flag(v, flagset, f, m)			\
997 	do {							\
998 		typeof(v->arch.flagset) *fset;			\
999 								\
1000 		__build_check_flag(v, flagset, f, m);		\
1001 								\
1002 		fset = &v->arch.flagset;			\
1003 		__vcpu_flags_preempt_disable();			\
1004 		if (HWEIGHT(m) > 1)				\
1005 			*fset &= ~(m);				\
1006 		*fset |= (f);					\
1007 		__vcpu_flags_preempt_enable();			\
1008 	} while (0)
1009 
1010 #define __vcpu_clear_flag(v, flagset, f, m)			\
1011 	do {							\
1012 		typeof(v->arch.flagset) *fset;			\
1013 								\
1014 		__build_check_flag(v, flagset, f, m);		\
1015 								\
1016 		fset = &v->arch.flagset;			\
1017 		__vcpu_flags_preempt_disable();			\
1018 		*fset &= ~(m);					\
1019 		__vcpu_flags_preempt_enable();			\
1020 	} while (0)
1021 
1022 #define __vcpu_test_and_clear_flag(v, flagset, f, m)		\
1023 	({							\
1024 		typeof(v->arch.flagset) set;			\
1025 								\
1026 		set = __vcpu_get_flag(v, flagset, f, m);	\
1027 		__vcpu_clear_flag(v, flagset, f, m);		\
1028 								\
1029 		set;						\
1030 	})
1031 
1032 #define vcpu_get_flag(v, ...)	__vcpu_get_flag((v), __VA_ARGS__)
1033 #define vcpu_set_flag(v, ...)	__vcpu_set_flag((v), __VA_ARGS__)
1034 #define vcpu_clear_flag(v, ...)	__vcpu_clear_flag((v), __VA_ARGS__)
1035 #define vcpu_test_and_clear_flag(v, ...)			\
1036 	__vcpu_test_and_clear_flag((v), __VA_ARGS__)
1037 
1038 /* KVM_ARM_VCPU_INIT completed */
1039 #define VCPU_INITIALIZED	__vcpu_single_flag(cflags, BIT(0))
1040 /* SVE config completed */
1041 #define VCPU_SVE_FINALIZED	__vcpu_single_flag(cflags, BIT(1))
1042 /* pKVM VCPU setup completed */
1043 #define VCPU_PKVM_FINALIZED	__vcpu_single_flag(cflags, BIT(2))
1044 
1045 /* Exception pending */
1046 #define PENDING_EXCEPTION	__vcpu_single_flag(iflags, BIT(0))
1047 /*
1048  * PC increment. Overlaps with EXCEPT_MASK on purpose so that it can't
1049  * be set together with an exception...
1050  */
1051 #define INCREMENT_PC		__vcpu_single_flag(iflags, BIT(1))
1052 /* Target EL/MODE (not a single flag, but let's abuse the macro) */
1053 #define EXCEPT_MASK		__vcpu_single_flag(iflags, GENMASK(3, 1))
1054 
1055 /* Helpers to encode exceptions with minimum fuss */
1056 #define __EXCEPT_MASK_VAL	unpack_vcpu_flag(EXCEPT_MASK)
1057 #define __EXCEPT_SHIFT		__builtin_ctzl(__EXCEPT_MASK_VAL)
1058 #define __vcpu_except_flags(_f)	iflags, (_f << __EXCEPT_SHIFT), __EXCEPT_MASK_VAL
1059 
1060 /*
1061  * When PENDING_EXCEPTION is set, EXCEPT_MASK can take the following
1062  * values:
1063  *
1064  * For AArch32 EL1:
1065  */
1066 #define EXCEPT_AA32_UND		__vcpu_except_flags(0)
1067 #define EXCEPT_AA32_IABT	__vcpu_except_flags(1)
1068 #define EXCEPT_AA32_DABT	__vcpu_except_flags(2)
1069 /* For AArch64: */
1070 #define EXCEPT_AA64_EL1_SYNC	__vcpu_except_flags(0)
1071 #define EXCEPT_AA64_EL1_IRQ	__vcpu_except_flags(1)
1072 #define EXCEPT_AA64_EL1_FIQ	__vcpu_except_flags(2)
1073 #define EXCEPT_AA64_EL1_SERR	__vcpu_except_flags(3)
1074 /* For AArch64 with NV: */
1075 #define EXCEPT_AA64_EL2_SYNC	__vcpu_except_flags(4)
1076 #define EXCEPT_AA64_EL2_IRQ	__vcpu_except_flags(5)
1077 #define EXCEPT_AA64_EL2_FIQ	__vcpu_except_flags(6)
1078 #define EXCEPT_AA64_EL2_SERR	__vcpu_except_flags(7)
1079 
1080 /* Physical CPU not in supported_cpus */
1081 #define ON_UNSUPPORTED_CPU	__vcpu_single_flag(sflags, BIT(0))
1082 /* WFIT instruction trapped */
1083 #define IN_WFIT			__vcpu_single_flag(sflags, BIT(1))
1084 /* vcpu system registers loaded on physical CPU */
1085 #define SYSREGS_ON_CPU		__vcpu_single_flag(sflags, BIT(2))
1086 /* Software step state is Active-pending for external debug */
1087 #define HOST_SS_ACTIVE_PENDING	__vcpu_single_flag(sflags, BIT(3))
1088 /* Software step state is Active pending for guest debug */
1089 #define GUEST_SS_ACTIVE_PENDING __vcpu_single_flag(sflags, BIT(4))
1090 /* PMUSERENR for the guest EL0 is on physical CPU */
1091 #define PMUSERENR_ON_CPU	__vcpu_single_flag(sflags, BIT(5))
1092 /* WFI instruction trapped */
1093 #define IN_WFI			__vcpu_single_flag(sflags, BIT(6))
1094 /* KVM is currently emulating a nested ERET */
1095 #define IN_NESTED_ERET		__vcpu_single_flag(sflags, BIT(7))
1096 /* SError pending for nested guest */
1097 #define NESTED_SERROR_PENDING	__vcpu_single_flag(sflags, BIT(8))
1098 
1099 
1100 #define vcpu_sve_max_vq(vcpu)	sve_vq_from_vl((vcpu)->arch.sve_max_vl)
1101 
1102 #define vcpu_sve_zcr_elx(vcpu)						\
1103 	(unlikely(is_hyp_ctxt(vcpu)) ? ZCR_EL2 : ZCR_EL1)
1104 
1105 #define sve_state_size_from_vl(sve_max_vl) ({				\
1106 	size_t __size_ret;						\
1107 	unsigned int __vq;						\
1108 									\
1109 	if (WARN_ON(!sve_vl_valid(sve_max_vl))) {			\
1110 		__size_ret = 0;						\
1111 	} else {							\
1112 		__vq = sve_vq_from_vl(sve_max_vl);			\
1113 		__size_ret = SVE_SIG_REGS_SIZE(__vq);			\
1114 	}								\
1115 									\
1116 	__size_ret;							\
1117 })
1118 
1119 #define vcpu_sve_state_size(vcpu) sve_state_size_from_vl((vcpu)->arch.sve_max_vl)
1120 
1121 #define KVM_GUESTDBG_VALID_MASK (KVM_GUESTDBG_ENABLE | \
1122 				 KVM_GUESTDBG_USE_SW_BP | \
1123 				 KVM_GUESTDBG_USE_HW | \
1124 				 KVM_GUESTDBG_SINGLESTEP)
1125 
1126 #define kvm_has_sve(kvm)	(system_supports_sve() &&		\
1127 				 test_bit(KVM_ARCH_FLAG_GUEST_HAS_SVE, &(kvm)->arch.flags))
1128 
1129 #ifdef __KVM_NVHE_HYPERVISOR__
1130 #define vcpu_has_sve(vcpu)	kvm_has_sve(kern_hyp_va((vcpu)->kvm))
1131 #else
1132 #define vcpu_has_sve(vcpu)	kvm_has_sve((vcpu)->kvm)
1133 #endif
1134 
1135 #ifdef CONFIG_ARM64_PTR_AUTH
1136 #define vcpu_has_ptrauth(vcpu)						\
1137 	((cpus_have_final_cap(ARM64_HAS_ADDRESS_AUTH) ||		\
1138 	  cpus_have_final_cap(ARM64_HAS_GENERIC_AUTH)) &&		\
1139 	 (vcpu_has_feature(vcpu, KVM_ARM_VCPU_PTRAUTH_ADDRESS) ||       \
1140 	  vcpu_has_feature(vcpu, KVM_ARM_VCPU_PTRAUTH_GENERIC)))
1141 #else
1142 #define vcpu_has_ptrauth(vcpu)		false
1143 #endif
1144 
1145 #define vcpu_on_unsupported_cpu(vcpu)					\
1146 	vcpu_get_flag(vcpu, ON_UNSUPPORTED_CPU)
1147 
1148 #define vcpu_set_on_unsupported_cpu(vcpu)				\
1149 	vcpu_set_flag(vcpu, ON_UNSUPPORTED_CPU)
1150 
1151 #define vcpu_clear_on_unsupported_cpu(vcpu)				\
1152 	vcpu_clear_flag(vcpu, ON_UNSUPPORTED_CPU)
1153 
1154 #define vcpu_gp_regs(v)		(&(v)->arch.ctxt.regs)
1155 
1156 /*
1157  * Only use __vcpu_sys_reg/ctxt_sys_reg if you know you want the
1158  * memory backed version of a register, and not the one most recently
1159  * accessed by a running VCPU.  For example, for userspace access or
1160  * for system registers that are never context switched, but only
1161  * emulated.
1162  *
1163  * Don't bother with VNCR-based accesses in the nVHE code, it has no
1164  * business dealing with NV.
1165  */
1166 static inline u64 *___ctxt_sys_reg(const struct kvm_cpu_context *ctxt, int r)
1167 {
1168 #if !defined (__KVM_NVHE_HYPERVISOR__)
1169 	if (unlikely(cpus_have_final_cap(ARM64_HAS_NESTED_VIRT) &&
1170 		     r >= __VNCR_START__ && ctxt->vncr_array))
1171 		return &ctxt->vncr_array[r - __VNCR_START__];
1172 #endif
1173 	return (u64 *)&ctxt->sys_regs[r];
1174 }
1175 
1176 #define __ctxt_sys_reg(c,r)						\
1177 	({								\
1178 		BUILD_BUG_ON(__builtin_constant_p(r) &&			\
1179 			     (r) >= NR_SYS_REGS);			\
1180 		___ctxt_sys_reg(c, r);					\
1181 	})
1182 
1183 #define ctxt_sys_reg(c,r)	(*__ctxt_sys_reg(c,r))
1184 
1185 u64 kvm_vcpu_apply_reg_masks(const struct kvm_vcpu *, enum vcpu_sysreg, u64);
1186 
1187 #define __vcpu_assign_sys_reg(v, r, val)				\
1188 	do {								\
1189 		const struct kvm_cpu_context *ctxt = &(v)->arch.ctxt;	\
1190 		u64 __v = (val);					\
1191 		if (vcpu_has_nv((v)) && (r) >= __SANITISED_REG_START__)	\
1192 			__v = kvm_vcpu_apply_reg_masks((v), (r), __v);	\
1193 									\
1194 		ctxt_sys_reg(ctxt, (r)) = __v;				\
1195 	} while (0)
1196 
1197 #define __vcpu_rmw_sys_reg(v, r, op, val)				\
1198 	do {								\
1199 		const struct kvm_cpu_context *ctxt = &(v)->arch.ctxt;	\
1200 		u64 __v = ctxt_sys_reg(ctxt, (r));			\
1201 		__v op (val);						\
1202 		if (vcpu_has_nv((v)) && (r) >= __SANITISED_REG_START__)	\
1203 			__v = kvm_vcpu_apply_reg_masks((v), (r), __v);	\
1204 									\
1205 		ctxt_sys_reg(ctxt, (r)) = __v;				\
1206 	} while (0)
1207 
1208 #define __vcpu_sys_reg(v,r)						\
1209 	({								\
1210 		const struct kvm_cpu_context *ctxt = &(v)->arch.ctxt;	\
1211 		u64 __v = ctxt_sys_reg(ctxt, (r));			\
1212 		if (vcpu_has_nv((v)) && (r) >= __SANITISED_REG_START__)	\
1213 			__v = kvm_vcpu_apply_reg_masks((v), (r), __v);	\
1214 		__v;							\
1215 	})
1216 
1217 u64 vcpu_read_sys_reg(const struct kvm_vcpu *, enum vcpu_sysreg);
1218 void vcpu_write_sys_reg(struct kvm_vcpu *, u64, enum vcpu_sysreg);
1219 
1220 struct kvm_vm_stat {
1221 	struct kvm_vm_stat_generic generic;
1222 };
1223 
1224 struct kvm_vcpu_stat {
1225 	struct kvm_vcpu_stat_generic generic;
1226 	u64 hvc_exit_stat;
1227 	u64 wfe_exit_stat;
1228 	u64 wfi_exit_stat;
1229 	u64 mmio_exit_user;
1230 	u64 mmio_exit_kernel;
1231 	u64 signal_exits;
1232 	u64 exits;
1233 };
1234 
1235 unsigned long kvm_arm_num_regs(struct kvm_vcpu *vcpu);
1236 int kvm_arm_copy_reg_indices(struct kvm_vcpu *vcpu, u64 __user *indices);
1237 int kvm_arm_get_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg);
1238 int kvm_arm_set_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg);
1239 
1240 unsigned long kvm_arm_num_sys_reg_descs(struct kvm_vcpu *vcpu);
1241 int kvm_arm_copy_sys_reg_indices(struct kvm_vcpu *vcpu, u64 __user *uindices);
1242 
1243 int __kvm_arm_vcpu_get_events(struct kvm_vcpu *vcpu,
1244 			      struct kvm_vcpu_events *events);
1245 
1246 int __kvm_arm_vcpu_set_events(struct kvm_vcpu *vcpu,
1247 			      struct kvm_vcpu_events *events);
1248 
1249 void kvm_arm_halt_guest(struct kvm *kvm);
1250 void kvm_arm_resume_guest(struct kvm *kvm);
1251 
1252 #define vcpu_has_run_once(vcpu)	(!!READ_ONCE((vcpu)->pid))
1253 
1254 #ifndef __KVM_NVHE_HYPERVISOR__
1255 #define kvm_call_hyp_nvhe(f, ...)						\
1256 	({								\
1257 		struct arm_smccc_res res;				\
1258 									\
1259 		arm_smccc_1_1_hvc(KVM_HOST_SMCCC_FUNC(f),		\
1260 				  ##__VA_ARGS__, &res);			\
1261 		WARN_ON(res.a0 != SMCCC_RET_SUCCESS);			\
1262 									\
1263 		res.a1;							\
1264 	})
1265 
1266 /*
1267  * The isb() below is there to guarantee the same behaviour on VHE as on !VHE,
1268  * where the eret to EL1 acts as a context synchronization event.
1269  */
1270 #define kvm_call_hyp(f, ...)						\
1271 	do {								\
1272 		if (has_vhe()) {					\
1273 			f(__VA_ARGS__);					\
1274 			isb();						\
1275 		} else {						\
1276 			kvm_call_hyp_nvhe(f, ##__VA_ARGS__);		\
1277 		}							\
1278 	} while(0)
1279 
1280 #define kvm_call_hyp_ret(f, ...)					\
1281 	({								\
1282 		typeof(f(__VA_ARGS__)) ret;				\
1283 									\
1284 		if (has_vhe()) {					\
1285 			ret = f(__VA_ARGS__);				\
1286 		} else {						\
1287 			ret = kvm_call_hyp_nvhe(f, ##__VA_ARGS__);	\
1288 		}							\
1289 									\
1290 		ret;							\
1291 	})
1292 #else /* __KVM_NVHE_HYPERVISOR__ */
1293 #define kvm_call_hyp(f, ...) f(__VA_ARGS__)
1294 #define kvm_call_hyp_ret(f, ...) f(__VA_ARGS__)
1295 #define kvm_call_hyp_nvhe(f, ...) f(__VA_ARGS__)
1296 #endif /* __KVM_NVHE_HYPERVISOR__ */
1297 
1298 int handle_exit(struct kvm_vcpu *vcpu, int exception_index);
1299 void handle_exit_early(struct kvm_vcpu *vcpu, int exception_index);
1300 
1301 int kvm_handle_cp14_load_store(struct kvm_vcpu *vcpu);
1302 int kvm_handle_cp14_32(struct kvm_vcpu *vcpu);
1303 int kvm_handle_cp14_64(struct kvm_vcpu *vcpu);
1304 int kvm_handle_cp15_32(struct kvm_vcpu *vcpu);
1305 int kvm_handle_cp15_64(struct kvm_vcpu *vcpu);
1306 int kvm_handle_sys_reg(struct kvm_vcpu *vcpu);
1307 int kvm_handle_cp10_id(struct kvm_vcpu *vcpu);
1308 
1309 void kvm_sys_regs_create_debugfs(struct kvm *kvm);
1310 void kvm_reset_sys_regs(struct kvm_vcpu *vcpu);
1311 
1312 int __init kvm_sys_reg_table_init(void);
1313 struct sys_reg_desc;
1314 int __init populate_sysreg_config(const struct sys_reg_desc *sr,
1315 				  unsigned int idx);
1316 int __init populate_nv_trap_config(void);
1317 
1318 void kvm_calculate_traps(struct kvm_vcpu *vcpu);
1319 
1320 /* MMIO helpers */
1321 void kvm_mmio_write_buf(void *buf, unsigned int len, unsigned long data);
1322 unsigned long kvm_mmio_read_buf(const void *buf, unsigned int len);
1323 
1324 int kvm_handle_mmio_return(struct kvm_vcpu *vcpu);
1325 int io_mem_abort(struct kvm_vcpu *vcpu, phys_addr_t fault_ipa);
1326 
1327 /*
1328  * Returns true if a Performance Monitoring Interrupt (PMI), a.k.a. perf event,
1329  * arrived in guest context.  For arm64, any event that arrives while a vCPU is
1330  * loaded is considered to be "in guest".
1331  */
1332 static inline bool kvm_arch_pmi_in_guest(struct kvm_vcpu *vcpu)
1333 {
1334 	return IS_ENABLED(CONFIG_GUEST_PERF_EVENTS) && !!vcpu;
1335 }
1336 
1337 long kvm_hypercall_pv_features(struct kvm_vcpu *vcpu);
1338 gpa_t kvm_init_stolen_time(struct kvm_vcpu *vcpu);
1339 void kvm_update_stolen_time(struct kvm_vcpu *vcpu);
1340 
1341 bool kvm_arm_pvtime_supported(void);
1342 int kvm_arm_pvtime_set_attr(struct kvm_vcpu *vcpu,
1343 			    struct kvm_device_attr *attr);
1344 int kvm_arm_pvtime_get_attr(struct kvm_vcpu *vcpu,
1345 			    struct kvm_device_attr *attr);
1346 int kvm_arm_pvtime_has_attr(struct kvm_vcpu *vcpu,
1347 			    struct kvm_device_attr *attr);
1348 
1349 extern unsigned int __ro_after_init kvm_arm_vmid_bits;
1350 int __init kvm_arm_vmid_alloc_init(void);
1351 void __init kvm_arm_vmid_alloc_free(void);
1352 void kvm_arm_vmid_update(struct kvm_vmid *kvm_vmid);
1353 void kvm_arm_vmid_clear_active(void);
1354 
1355 static inline void kvm_arm_pvtime_vcpu_init(struct kvm_vcpu_arch *vcpu_arch)
1356 {
1357 	vcpu_arch->steal.base = INVALID_GPA;
1358 }
1359 
1360 static inline bool kvm_arm_is_pvtime_enabled(struct kvm_vcpu_arch *vcpu_arch)
1361 {
1362 	return (vcpu_arch->steal.base != INVALID_GPA);
1363 }
1364 
1365 struct kvm_vcpu *kvm_mpidr_to_vcpu(struct kvm *kvm, unsigned long mpidr);
1366 
1367 DECLARE_KVM_HYP_PER_CPU(struct kvm_host_data, kvm_host_data);
1368 
1369 /*
1370  * How we access per-CPU host data depends on the where we access it from,
1371  * and the mode we're in:
1372  *
1373  * - VHE and nVHE hypervisor bits use their locally defined instance
1374  *
1375  * - the rest of the kernel use either the VHE or nVHE one, depending on
1376  *   the mode we're running in.
1377  *
1378  *   Unless we're in protected mode, fully deprivileged, and the nVHE
1379  *   per-CPU stuff is exclusively accessible to the protected EL2 code.
1380  *   In this case, the EL1 code uses the *VHE* data as its private state
1381  *   (which makes sense in a way as there shouldn't be any shared state
1382  *   between the host and the hypervisor).
1383  *
1384  * Yes, this is all totally trivial. Shoot me now.
1385  */
1386 #if defined(__KVM_NVHE_HYPERVISOR__) || defined(__KVM_VHE_HYPERVISOR__)
1387 #define host_data_ptr(f)	(&this_cpu_ptr(&kvm_host_data)->f)
1388 #else
1389 #define host_data_ptr(f)						\
1390 	(static_branch_unlikely(&kvm_protected_mode_initialized) ?	\
1391 	 &this_cpu_ptr(&kvm_host_data)->f :				\
1392 	 &this_cpu_ptr_hyp_sym(kvm_host_data)->f)
1393 #endif
1394 
1395 #define host_data_test_flag(flag)					\
1396 	(test_bit(KVM_HOST_DATA_FLAG_##flag, host_data_ptr(flags)))
1397 #define host_data_set_flag(flag)					\
1398 	set_bit(KVM_HOST_DATA_FLAG_##flag, host_data_ptr(flags))
1399 #define host_data_clear_flag(flag)					\
1400 	clear_bit(KVM_HOST_DATA_FLAG_##flag, host_data_ptr(flags))
1401 
1402 /* Check whether the FP regs are owned by the guest */
1403 static inline bool guest_owns_fp_regs(void)
1404 {
1405 	return *host_data_ptr(fp_owner) == FP_STATE_GUEST_OWNED;
1406 }
1407 
1408 /* Check whether the FP regs are owned by the host */
1409 static inline bool host_owns_fp_regs(void)
1410 {
1411 	return *host_data_ptr(fp_owner) == FP_STATE_HOST_OWNED;
1412 }
1413 
1414 static inline void kvm_init_host_cpu_context(struct kvm_cpu_context *cpu_ctxt)
1415 {
1416 	/* The host's MPIDR is immutable, so let's set it up at boot time */
1417 	ctxt_sys_reg(cpu_ctxt, MPIDR_EL1) = read_cpuid_mpidr();
1418 }
1419 
1420 static inline bool kvm_system_needs_idmapped_vectors(void)
1421 {
1422 	return cpus_have_final_cap(ARM64_SPECTRE_V3A);
1423 }
1424 
1425 void kvm_init_host_debug_data(void);
1426 void kvm_debug_init_vhe(void);
1427 void kvm_vcpu_load_debug(struct kvm_vcpu *vcpu);
1428 void kvm_vcpu_put_debug(struct kvm_vcpu *vcpu);
1429 void kvm_debug_set_guest_ownership(struct kvm_vcpu *vcpu);
1430 void kvm_debug_handle_oslar(struct kvm_vcpu *vcpu, u64 val);
1431 
1432 #define kvm_vcpu_os_lock_enabled(vcpu)		\
1433 	(!!(__vcpu_sys_reg(vcpu, OSLSR_EL1) & OSLSR_EL1_OSLK))
1434 
1435 #define kvm_debug_regs_in_use(vcpu)		\
1436 	((vcpu)->arch.debug_owner != VCPU_DEBUG_FREE)
1437 #define kvm_host_owns_debug_regs(vcpu)		\
1438 	((vcpu)->arch.debug_owner == VCPU_DEBUG_HOST_OWNED)
1439 #define kvm_guest_owns_debug_regs(vcpu)		\
1440 	((vcpu)->arch.debug_owner == VCPU_DEBUG_GUEST_OWNED)
1441 
1442 int kvm_arm_vcpu_arch_set_attr(struct kvm_vcpu *vcpu,
1443 			       struct kvm_device_attr *attr);
1444 int kvm_arm_vcpu_arch_get_attr(struct kvm_vcpu *vcpu,
1445 			       struct kvm_device_attr *attr);
1446 int kvm_arm_vcpu_arch_has_attr(struct kvm_vcpu *vcpu,
1447 			       struct kvm_device_attr *attr);
1448 
1449 int kvm_vm_ioctl_mte_copy_tags(struct kvm *kvm,
1450 			       struct kvm_arm_copy_mte_tags *copy_tags);
1451 int kvm_vm_ioctl_set_counter_offset(struct kvm *kvm,
1452 				    struct kvm_arm_counter_offset *offset);
1453 int kvm_vm_ioctl_get_reg_writable_masks(struct kvm *kvm,
1454 					struct reg_mask_range *range);
1455 
1456 /* Guest/host FPSIMD coordination helpers */
1457 void kvm_arch_vcpu_load_fp(struct kvm_vcpu *vcpu);
1458 void kvm_arch_vcpu_ctxflush_fp(struct kvm_vcpu *vcpu);
1459 void kvm_arch_vcpu_ctxsync_fp(struct kvm_vcpu *vcpu);
1460 void kvm_arch_vcpu_put_fp(struct kvm_vcpu *vcpu);
1461 
1462 static inline bool kvm_pmu_counter_deferred(struct perf_event_attr *attr)
1463 {
1464 	return (!has_vhe() && attr->exclude_host);
1465 }
1466 
1467 #ifdef CONFIG_KVM
1468 void kvm_set_pmu_events(u64 set, struct perf_event_attr *attr);
1469 void kvm_clr_pmu_events(u64 clr);
1470 bool kvm_set_pmuserenr(u64 val);
1471 void kvm_enable_trbe(void);
1472 void kvm_disable_trbe(void);
1473 void kvm_tracing_set_el1_configuration(u64 trfcr_while_in_guest);
1474 #else
1475 static inline void kvm_set_pmu_events(u64 set, struct perf_event_attr *attr) {}
1476 static inline void kvm_clr_pmu_events(u64 clr) {}
1477 static inline bool kvm_set_pmuserenr(u64 val)
1478 {
1479 	return false;
1480 }
1481 static inline void kvm_enable_trbe(void) {}
1482 static inline void kvm_disable_trbe(void) {}
1483 static inline void kvm_tracing_set_el1_configuration(u64 trfcr_while_in_guest) {}
1484 #endif
1485 
1486 void kvm_vcpu_load_vhe(struct kvm_vcpu *vcpu);
1487 void kvm_vcpu_put_vhe(struct kvm_vcpu *vcpu);
1488 
1489 int __init kvm_set_ipa_limit(void);
1490 u32 kvm_get_pa_bits(struct kvm *kvm);
1491 
1492 #define __KVM_HAVE_ARCH_VM_ALLOC
1493 struct kvm *kvm_arch_alloc_vm(void);
1494 
1495 #define __KVM_HAVE_ARCH_FLUSH_REMOTE_TLBS
1496 
1497 #define __KVM_HAVE_ARCH_FLUSH_REMOTE_TLBS_RANGE
1498 
1499 #define kvm_vm_is_protected(kvm)	(is_protected_kvm_enabled() && (kvm)->arch.pkvm.is_protected)
1500 
1501 #define vcpu_is_protected(vcpu)		kvm_vm_is_protected((vcpu)->kvm)
1502 
1503 int kvm_arm_vcpu_finalize(struct kvm_vcpu *vcpu, int feature);
1504 bool kvm_arm_vcpu_is_finalized(struct kvm_vcpu *vcpu);
1505 
1506 #define kvm_arm_vcpu_sve_finalized(vcpu) vcpu_get_flag(vcpu, VCPU_SVE_FINALIZED)
1507 
1508 #define kvm_has_mte(kvm)					\
1509 	(system_supports_mte() &&				\
1510 	 test_bit(KVM_ARCH_FLAG_MTE_ENABLED, &(kvm)->arch.flags))
1511 
1512 #define kvm_supports_32bit_el0()				\
1513 	(system_supports_32bit_el0() &&				\
1514 	 !static_branch_unlikely(&arm64_mismatched_32bit_el0))
1515 
1516 #define kvm_vm_has_ran_once(kvm)					\
1517 	(test_bit(KVM_ARCH_FLAG_HAS_RAN_ONCE, &(kvm)->arch.flags))
1518 
1519 static inline bool __vcpu_has_feature(const struct kvm_arch *ka, int feature)
1520 {
1521 	return test_bit(feature, ka->vcpu_features);
1522 }
1523 
1524 #define kvm_vcpu_has_feature(k, f)	__vcpu_has_feature(&(k)->arch, (f))
1525 #define vcpu_has_feature(v, f)	__vcpu_has_feature(&(v)->kvm->arch, (f))
1526 
1527 #define kvm_vcpu_initialized(v) vcpu_get_flag(v, VCPU_INITIALIZED)
1528 
1529 int kvm_trng_call(struct kvm_vcpu *vcpu);
1530 #ifdef CONFIG_KVM
1531 extern phys_addr_t hyp_mem_base;
1532 extern phys_addr_t hyp_mem_size;
1533 void __init kvm_hyp_reserve(void);
1534 #else
1535 static inline void kvm_hyp_reserve(void) { }
1536 #endif
1537 
1538 void kvm_arm_vcpu_power_off(struct kvm_vcpu *vcpu);
1539 bool kvm_arm_vcpu_stopped(struct kvm_vcpu *vcpu);
1540 
1541 static inline u64 *__vm_id_reg(struct kvm_arch *ka, u32 reg)
1542 {
1543 	switch (reg) {
1544 	case sys_reg(3, 0, 0, 1, 0) ... sys_reg(3, 0, 0, 7, 7):
1545 		return &ka->id_regs[IDREG_IDX(reg)];
1546 	case SYS_CTR_EL0:
1547 		return &ka->ctr_el0;
1548 	case SYS_MIDR_EL1:
1549 		return &ka->midr_el1;
1550 	case SYS_REVIDR_EL1:
1551 		return &ka->revidr_el1;
1552 	case SYS_AIDR_EL1:
1553 		return &ka->aidr_el1;
1554 	default:
1555 		WARN_ON_ONCE(1);
1556 		return NULL;
1557 	}
1558 }
1559 
1560 #define kvm_read_vm_id_reg(kvm, reg)					\
1561 	({ u64 __val = *__vm_id_reg(&(kvm)->arch, reg); __val; })
1562 
1563 void kvm_set_vm_id_reg(struct kvm *kvm, u32 reg, u64 val);
1564 
1565 #define __expand_field_sign_unsigned(id, fld, val)			\
1566 	((u64)SYS_FIELD_VALUE(id, fld, val))
1567 
1568 #define __expand_field_sign_signed(id, fld, val)			\
1569 	({								\
1570 		u64 __val = SYS_FIELD_VALUE(id, fld, val);		\
1571 		sign_extend64(__val, id##_##fld##_WIDTH - 1);		\
1572 	})
1573 
1574 #define get_idreg_field_unsigned(kvm, id, fld)				\
1575 	({								\
1576 		u64 __val = kvm_read_vm_id_reg((kvm), SYS_##id);	\
1577 		FIELD_GET(id##_##fld##_MASK, __val);			\
1578 	})
1579 
1580 #define get_idreg_field_signed(kvm, id, fld)				\
1581 	({								\
1582 		u64 __val = get_idreg_field_unsigned(kvm, id, fld);	\
1583 		sign_extend64(__val, id##_##fld##_WIDTH - 1);		\
1584 	})
1585 
1586 #define get_idreg_field_enum(kvm, id, fld)				\
1587 	get_idreg_field_unsigned(kvm, id, fld)
1588 
1589 #define kvm_cmp_feat_signed(kvm, id, fld, op, limit)			\
1590 	(get_idreg_field_signed((kvm), id, fld) op __expand_field_sign_signed(id, fld, limit))
1591 
1592 #define kvm_cmp_feat_unsigned(kvm, id, fld, op, limit)			\
1593 	(get_idreg_field_unsigned((kvm), id, fld) op __expand_field_sign_unsigned(id, fld, limit))
1594 
1595 #define kvm_cmp_feat(kvm, id, fld, op, limit)				\
1596 	(id##_##fld##_SIGNED ?						\
1597 	 kvm_cmp_feat_signed(kvm, id, fld, op, limit) :			\
1598 	 kvm_cmp_feat_unsigned(kvm, id, fld, op, limit))
1599 
1600 #define __kvm_has_feat(kvm, id, fld, limit)				\
1601 	kvm_cmp_feat(kvm, id, fld, >=, limit)
1602 
1603 #define kvm_has_feat(kvm, ...) __kvm_has_feat(kvm, __VA_ARGS__)
1604 
1605 #define __kvm_has_feat_enum(kvm, id, fld, val)				\
1606 	kvm_cmp_feat_unsigned(kvm, id, fld, ==, val)
1607 
1608 #define kvm_has_feat_enum(kvm, ...) __kvm_has_feat_enum(kvm, __VA_ARGS__)
1609 
1610 #define kvm_has_feat_range(kvm, id, fld, min, max)			\
1611 	(kvm_cmp_feat(kvm, id, fld, >=, min) &&				\
1612 	kvm_cmp_feat(kvm, id, fld, <=, max))
1613 
1614 /* Check for a given level of PAuth support */
1615 #define kvm_has_pauth(k, l)						\
1616 	({								\
1617 		bool pa, pi, pa3;					\
1618 									\
1619 		pa  = kvm_has_feat((k), ID_AA64ISAR1_EL1, APA, l);	\
1620 		pa &= kvm_has_feat((k), ID_AA64ISAR1_EL1, GPA, IMP);	\
1621 		pi  = kvm_has_feat((k), ID_AA64ISAR1_EL1, API, l);	\
1622 		pi &= kvm_has_feat((k), ID_AA64ISAR1_EL1, GPI, IMP);	\
1623 		pa3  = kvm_has_feat((k), ID_AA64ISAR2_EL1, APA3, l);	\
1624 		pa3 &= kvm_has_feat((k), ID_AA64ISAR2_EL1, GPA3, IMP);	\
1625 									\
1626 		(pa + pi + pa3) == 1;					\
1627 	})
1628 
1629 #define kvm_has_fpmr(k)					\
1630 	(system_supports_fpmr() &&			\
1631 	 kvm_has_feat((k), ID_AA64PFR2_EL1, FPMR, IMP))
1632 
1633 #define kvm_has_tcr2(k)				\
1634 	(kvm_has_feat((k), ID_AA64MMFR3_EL1, TCRX, IMP))
1635 
1636 #define kvm_has_s1pie(k)				\
1637 	(kvm_has_feat((k), ID_AA64MMFR3_EL1, S1PIE, IMP))
1638 
1639 #define kvm_has_s1poe(k)				\
1640 	(system_supports_poe() &&			\
1641 	 kvm_has_feat((k), ID_AA64MMFR3_EL1, S1POE, IMP))
1642 
1643 #define kvm_has_ras(k)					\
1644 	(kvm_has_feat((k), ID_AA64PFR0_EL1, RAS, IMP))
1645 
1646 #define kvm_has_sctlr2(k)				\
1647 	(kvm_has_feat((k), ID_AA64MMFR3_EL1, SCTLRX, IMP))
1648 
1649 static inline bool kvm_arch_has_irq_bypass(void)
1650 {
1651 	return true;
1652 }
1653 
1654 void compute_fgu(struct kvm *kvm, enum fgt_group_id fgt);
1655 struct resx get_reg_fixed_bits(struct kvm *kvm, enum vcpu_sysreg reg);
1656 void check_feature_map(void);
1657 void kvm_vcpu_load_fgt(struct kvm_vcpu *vcpu);
1658 
1659 static __always_inline enum fgt_group_id __fgt_reg_to_group_id(enum vcpu_sysreg reg)
1660 {
1661 	switch (reg) {
1662 	case HFGRTR_EL2:
1663 	case HFGWTR_EL2:
1664 		return HFGRTR_GROUP;
1665 	case HFGITR_EL2:
1666 		return HFGITR_GROUP;
1667 	case HDFGRTR_EL2:
1668 	case HDFGWTR_EL2:
1669 		return HDFGRTR_GROUP;
1670 	case HAFGRTR_EL2:
1671 		return HAFGRTR_GROUP;
1672 	case HFGRTR2_EL2:
1673 	case HFGWTR2_EL2:
1674 		return HFGRTR2_GROUP;
1675 	case HFGITR2_EL2:
1676 		return HFGITR2_GROUP;
1677 	case HDFGRTR2_EL2:
1678 	case HDFGWTR2_EL2:
1679 		return HDFGRTR2_GROUP;
1680 	case ICH_HFGRTR_EL2:
1681 	case ICH_HFGWTR_EL2:
1682 		return ICH_HFGRTR_GROUP;
1683 	case ICH_HFGITR_EL2:
1684 		return ICH_HFGITR_GROUP;
1685 	default:
1686 		BUILD_BUG_ON(1);
1687 	}
1688 }
1689 
1690 #define vcpu_fgt(vcpu, reg)						\
1691 	({								\
1692 		enum fgt_group_id id = __fgt_reg_to_group_id(reg);	\
1693 		u64 *p;							\
1694 		switch (reg) {						\
1695 		case HFGWTR_EL2:					\
1696 		case HDFGWTR_EL2:					\
1697 		case HFGWTR2_EL2:					\
1698 		case HDFGWTR2_EL2:					\
1699 		case ICH_HFGWTR_EL2:					\
1700 			p = &(vcpu)->arch.fgt[id].w;			\
1701 			break;						\
1702 		default:						\
1703 			p = &(vcpu)->arch.fgt[id].r;			\
1704 			break;						\
1705 		}							\
1706 									\
1707 		p;							\
1708 	})
1709 
1710 long kvm_get_cap_for_kvm_ioctl(unsigned int ioctl, long *ext);
1711 
1712 #endif /* __ARM64_KVM_HOST_H__ */
1713