xref: /linux/arch/riscv/kvm/vcpu.c (revision 5e3992fe72748ed3892be876f09d4d990548b7af)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (C) 2019 Western Digital Corporation or its affiliates.
4  *
5  * Authors:
6  *     Anup Patel <anup.patel@wdc.com>
7  */
8 
9 #include <linux/bitops.h>
10 #include <linux/entry-kvm.h>
11 #include <linux/errno.h>
12 #include <linux/err.h>
13 #include <linux/kdebug.h>
14 #include <linux/module.h>
15 #include <linux/percpu.h>
16 #include <linux/uaccess.h>
17 #include <linux/vmalloc.h>
18 #include <linux/sched/signal.h>
19 #include <linux/fs.h>
20 #include <linux/kvm_host.h>
21 #include <asm/csr.h>
22 #include <asm/cacheflush.h>
23 #include <asm/hwcap.h>
24 #include <asm/sbi.h>
25 
26 const struct _kvm_stats_desc kvm_vcpu_stats_desc[] = {
27 	KVM_GENERIC_VCPU_STATS(),
28 	STATS_DESC_COUNTER(VCPU, ecall_exit_stat),
29 	STATS_DESC_COUNTER(VCPU, wfi_exit_stat),
30 	STATS_DESC_COUNTER(VCPU, mmio_exit_user),
31 	STATS_DESC_COUNTER(VCPU, mmio_exit_kernel),
32 	STATS_DESC_COUNTER(VCPU, csr_exit_user),
33 	STATS_DESC_COUNTER(VCPU, csr_exit_kernel),
34 	STATS_DESC_COUNTER(VCPU, signal_exits),
35 	STATS_DESC_COUNTER(VCPU, exits)
36 };
37 
38 const struct kvm_stats_header kvm_vcpu_stats_header = {
39 	.name_size = KVM_STATS_NAME_SIZE,
40 	.num_desc = ARRAY_SIZE(kvm_vcpu_stats_desc),
41 	.id_offset = sizeof(struct kvm_stats_header),
42 	.desc_offset = sizeof(struct kvm_stats_header) + KVM_STATS_NAME_SIZE,
43 	.data_offset = sizeof(struct kvm_stats_header) + KVM_STATS_NAME_SIZE +
44 		       sizeof(kvm_vcpu_stats_desc),
45 };
46 
47 #define KVM_RISCV_BASE_ISA_MASK		GENMASK(25, 0)
48 
49 #define KVM_ISA_EXT_ARR(ext)		[KVM_RISCV_ISA_EXT_##ext] = RISCV_ISA_EXT_##ext
50 
51 /* Mapping between KVM ISA Extension ID & Host ISA extension ID */
52 static const unsigned long kvm_isa_ext_arr[] = {
53 	[KVM_RISCV_ISA_EXT_A] = RISCV_ISA_EXT_a,
54 	[KVM_RISCV_ISA_EXT_C] = RISCV_ISA_EXT_c,
55 	[KVM_RISCV_ISA_EXT_D] = RISCV_ISA_EXT_d,
56 	[KVM_RISCV_ISA_EXT_F] = RISCV_ISA_EXT_f,
57 	[KVM_RISCV_ISA_EXT_H] = RISCV_ISA_EXT_h,
58 	[KVM_RISCV_ISA_EXT_I] = RISCV_ISA_EXT_i,
59 	[KVM_RISCV_ISA_EXT_M] = RISCV_ISA_EXT_m,
60 
61 	KVM_ISA_EXT_ARR(SSAIA),
62 	KVM_ISA_EXT_ARR(SSTC),
63 	KVM_ISA_EXT_ARR(SVINVAL),
64 	KVM_ISA_EXT_ARR(SVPBMT),
65 	KVM_ISA_EXT_ARR(ZBB),
66 	KVM_ISA_EXT_ARR(ZIHINTPAUSE),
67 	KVM_ISA_EXT_ARR(ZICBOM),
68 	KVM_ISA_EXT_ARR(ZICBOZ),
69 };
70 
71 static unsigned long kvm_riscv_vcpu_base2isa_ext(unsigned long base_ext)
72 {
73 	unsigned long i;
74 
75 	for (i = 0; i < KVM_RISCV_ISA_EXT_MAX; i++) {
76 		if (kvm_isa_ext_arr[i] == base_ext)
77 			return i;
78 	}
79 
80 	return KVM_RISCV_ISA_EXT_MAX;
81 }
82 
83 static bool kvm_riscv_vcpu_isa_enable_allowed(unsigned long ext)
84 {
85 	switch (ext) {
86 	case KVM_RISCV_ISA_EXT_H:
87 		return false;
88 	default:
89 		break;
90 	}
91 
92 	return true;
93 }
94 
95 static bool kvm_riscv_vcpu_isa_disable_allowed(unsigned long ext)
96 {
97 	switch (ext) {
98 	case KVM_RISCV_ISA_EXT_A:
99 	case KVM_RISCV_ISA_EXT_C:
100 	case KVM_RISCV_ISA_EXT_I:
101 	case KVM_RISCV_ISA_EXT_M:
102 	case KVM_RISCV_ISA_EXT_SSAIA:
103 	case KVM_RISCV_ISA_EXT_SSTC:
104 	case KVM_RISCV_ISA_EXT_SVINVAL:
105 	case KVM_RISCV_ISA_EXT_ZIHINTPAUSE:
106 	case KVM_RISCV_ISA_EXT_ZBB:
107 		return false;
108 	default:
109 		break;
110 	}
111 
112 	return true;
113 }
114 
115 static void kvm_riscv_reset_vcpu(struct kvm_vcpu *vcpu)
116 {
117 	struct kvm_vcpu_csr *csr = &vcpu->arch.guest_csr;
118 	struct kvm_vcpu_csr *reset_csr = &vcpu->arch.guest_reset_csr;
119 	struct kvm_cpu_context *cntx = &vcpu->arch.guest_context;
120 	struct kvm_cpu_context *reset_cntx = &vcpu->arch.guest_reset_context;
121 	bool loaded;
122 
123 	/**
124 	 * The preemption should be disabled here because it races with
125 	 * kvm_sched_out/kvm_sched_in(called from preempt notifiers) which
126 	 * also calls vcpu_load/put.
127 	 */
128 	get_cpu();
129 	loaded = (vcpu->cpu != -1);
130 	if (loaded)
131 		kvm_arch_vcpu_put(vcpu);
132 
133 	vcpu->arch.last_exit_cpu = -1;
134 
135 	memcpy(csr, reset_csr, sizeof(*csr));
136 
137 	memcpy(cntx, reset_cntx, sizeof(*cntx));
138 
139 	kvm_riscv_vcpu_fp_reset(vcpu);
140 
141 	kvm_riscv_vcpu_timer_reset(vcpu);
142 
143 	kvm_riscv_vcpu_aia_reset(vcpu);
144 
145 	bitmap_zero(vcpu->arch.irqs_pending, KVM_RISCV_VCPU_NR_IRQS);
146 	bitmap_zero(vcpu->arch.irqs_pending_mask, KVM_RISCV_VCPU_NR_IRQS);
147 
148 	kvm_riscv_vcpu_pmu_reset(vcpu);
149 
150 	vcpu->arch.hfence_head = 0;
151 	vcpu->arch.hfence_tail = 0;
152 	memset(vcpu->arch.hfence_queue, 0, sizeof(vcpu->arch.hfence_queue));
153 
154 	/* Reset the guest CSRs for hotplug usecase */
155 	if (loaded)
156 		kvm_arch_vcpu_load(vcpu, smp_processor_id());
157 	put_cpu();
158 }
159 
160 int kvm_arch_vcpu_precreate(struct kvm *kvm, unsigned int id)
161 {
162 	return 0;
163 }
164 
165 int kvm_arch_vcpu_create(struct kvm_vcpu *vcpu)
166 {
167 	int rc;
168 	struct kvm_cpu_context *cntx;
169 	struct kvm_vcpu_csr *reset_csr = &vcpu->arch.guest_reset_csr;
170 	unsigned long host_isa, i;
171 
172 	/* Mark this VCPU never ran */
173 	vcpu->arch.ran_atleast_once = false;
174 	vcpu->arch.mmu_page_cache.gfp_zero = __GFP_ZERO;
175 	bitmap_zero(vcpu->arch.isa, RISCV_ISA_EXT_MAX);
176 
177 	/* Setup ISA features available to VCPU */
178 	for (i = 0; i < ARRAY_SIZE(kvm_isa_ext_arr); i++) {
179 		host_isa = kvm_isa_ext_arr[i];
180 		if (__riscv_isa_extension_available(NULL, host_isa) &&
181 		    kvm_riscv_vcpu_isa_enable_allowed(i))
182 			set_bit(host_isa, vcpu->arch.isa);
183 	}
184 
185 	/* Setup vendor, arch, and implementation details */
186 	vcpu->arch.mvendorid = sbi_get_mvendorid();
187 	vcpu->arch.marchid = sbi_get_marchid();
188 	vcpu->arch.mimpid = sbi_get_mimpid();
189 
190 	/* Setup VCPU hfence queue */
191 	spin_lock_init(&vcpu->arch.hfence_lock);
192 
193 	/* Setup reset state of shadow SSTATUS and HSTATUS CSRs */
194 	cntx = &vcpu->arch.guest_reset_context;
195 	cntx->sstatus = SR_SPP | SR_SPIE;
196 	cntx->hstatus = 0;
197 	cntx->hstatus |= HSTATUS_VTW;
198 	cntx->hstatus |= HSTATUS_SPVP;
199 	cntx->hstatus |= HSTATUS_SPV;
200 
201 	/* By default, make CY, TM, and IR counters accessible in VU mode */
202 	reset_csr->scounteren = 0x7;
203 
204 	/* Setup VCPU timer */
205 	kvm_riscv_vcpu_timer_init(vcpu);
206 
207 	/* setup performance monitoring */
208 	kvm_riscv_vcpu_pmu_init(vcpu);
209 
210 	/* Setup VCPU AIA */
211 	rc = kvm_riscv_vcpu_aia_init(vcpu);
212 	if (rc)
213 		return rc;
214 
215 	/* Reset VCPU */
216 	kvm_riscv_reset_vcpu(vcpu);
217 
218 	return 0;
219 }
220 
221 void kvm_arch_vcpu_postcreate(struct kvm_vcpu *vcpu)
222 {
223 	/**
224 	 * vcpu with id 0 is the designated boot cpu.
225 	 * Keep all vcpus with non-zero id in power-off state so that
226 	 * they can be brought up using SBI HSM extension.
227 	 */
228 	if (vcpu->vcpu_idx != 0)
229 		kvm_riscv_vcpu_power_off(vcpu);
230 }
231 
232 void kvm_arch_vcpu_destroy(struct kvm_vcpu *vcpu)
233 {
234 	/* Cleanup VCPU AIA context */
235 	kvm_riscv_vcpu_aia_deinit(vcpu);
236 
237 	/* Cleanup VCPU timer */
238 	kvm_riscv_vcpu_timer_deinit(vcpu);
239 
240 	kvm_riscv_vcpu_pmu_deinit(vcpu);
241 
242 	/* Free unused pages pre-allocated for G-stage page table mappings */
243 	kvm_mmu_free_memory_cache(&vcpu->arch.mmu_page_cache);
244 }
245 
246 int kvm_cpu_has_pending_timer(struct kvm_vcpu *vcpu)
247 {
248 	return kvm_riscv_vcpu_timer_pending(vcpu);
249 }
250 
251 void kvm_arch_vcpu_blocking(struct kvm_vcpu *vcpu)
252 {
253 }
254 
255 void kvm_arch_vcpu_unblocking(struct kvm_vcpu *vcpu)
256 {
257 }
258 
259 int kvm_arch_vcpu_runnable(struct kvm_vcpu *vcpu)
260 {
261 	return (kvm_riscv_vcpu_has_interrupts(vcpu, -1UL) &&
262 		!vcpu->arch.power_off && !vcpu->arch.pause);
263 }
264 
265 int kvm_arch_vcpu_should_kick(struct kvm_vcpu *vcpu)
266 {
267 	return kvm_vcpu_exiting_guest_mode(vcpu) == IN_GUEST_MODE;
268 }
269 
270 bool kvm_arch_vcpu_in_kernel(struct kvm_vcpu *vcpu)
271 {
272 	return (vcpu->arch.guest_context.sstatus & SR_SPP) ? true : false;
273 }
274 
275 vm_fault_t kvm_arch_vcpu_fault(struct kvm_vcpu *vcpu, struct vm_fault *vmf)
276 {
277 	return VM_FAULT_SIGBUS;
278 }
279 
280 static int kvm_riscv_vcpu_get_reg_config(struct kvm_vcpu *vcpu,
281 					 const struct kvm_one_reg *reg)
282 {
283 	unsigned long __user *uaddr =
284 			(unsigned long __user *)(unsigned long)reg->addr;
285 	unsigned long reg_num = reg->id & ~(KVM_REG_ARCH_MASK |
286 					    KVM_REG_SIZE_MASK |
287 					    KVM_REG_RISCV_CONFIG);
288 	unsigned long reg_val;
289 
290 	if (KVM_REG_SIZE(reg->id) != sizeof(unsigned long))
291 		return -EINVAL;
292 
293 	switch (reg_num) {
294 	case KVM_REG_RISCV_CONFIG_REG(isa):
295 		reg_val = vcpu->arch.isa[0] & KVM_RISCV_BASE_ISA_MASK;
296 		break;
297 	case KVM_REG_RISCV_CONFIG_REG(zicbom_block_size):
298 		if (!riscv_isa_extension_available(vcpu->arch.isa, ZICBOM))
299 			return -EINVAL;
300 		reg_val = riscv_cbom_block_size;
301 		break;
302 	case KVM_REG_RISCV_CONFIG_REG(zicboz_block_size):
303 		if (!riscv_isa_extension_available(vcpu->arch.isa, ZICBOZ))
304 			return -EINVAL;
305 		reg_val = riscv_cboz_block_size;
306 		break;
307 	case KVM_REG_RISCV_CONFIG_REG(mvendorid):
308 		reg_val = vcpu->arch.mvendorid;
309 		break;
310 	case KVM_REG_RISCV_CONFIG_REG(marchid):
311 		reg_val = vcpu->arch.marchid;
312 		break;
313 	case KVM_REG_RISCV_CONFIG_REG(mimpid):
314 		reg_val = vcpu->arch.mimpid;
315 		break;
316 	default:
317 		return -EINVAL;
318 	}
319 
320 	if (copy_to_user(uaddr, &reg_val, KVM_REG_SIZE(reg->id)))
321 		return -EFAULT;
322 
323 	return 0;
324 }
325 
326 static int kvm_riscv_vcpu_set_reg_config(struct kvm_vcpu *vcpu,
327 					 const struct kvm_one_reg *reg)
328 {
329 	unsigned long __user *uaddr =
330 			(unsigned long __user *)(unsigned long)reg->addr;
331 	unsigned long reg_num = reg->id & ~(KVM_REG_ARCH_MASK |
332 					    KVM_REG_SIZE_MASK |
333 					    KVM_REG_RISCV_CONFIG);
334 	unsigned long i, isa_ext, reg_val;
335 
336 	if (KVM_REG_SIZE(reg->id) != sizeof(unsigned long))
337 		return -EINVAL;
338 
339 	if (copy_from_user(&reg_val, uaddr, KVM_REG_SIZE(reg->id)))
340 		return -EFAULT;
341 
342 	switch (reg_num) {
343 	case KVM_REG_RISCV_CONFIG_REG(isa):
344 		/*
345 		 * This ONE REG interface is only defined for
346 		 * single letter extensions.
347 		 */
348 		if (fls(reg_val) >= RISCV_ISA_EXT_BASE)
349 			return -EINVAL;
350 
351 		if (!vcpu->arch.ran_atleast_once) {
352 			/* Ignore the enable/disable request for certain extensions */
353 			for (i = 0; i < RISCV_ISA_EXT_BASE; i++) {
354 				isa_ext = kvm_riscv_vcpu_base2isa_ext(i);
355 				if (isa_ext >= KVM_RISCV_ISA_EXT_MAX) {
356 					reg_val &= ~BIT(i);
357 					continue;
358 				}
359 				if (!kvm_riscv_vcpu_isa_enable_allowed(isa_ext))
360 					if (reg_val & BIT(i))
361 						reg_val &= ~BIT(i);
362 				if (!kvm_riscv_vcpu_isa_disable_allowed(isa_ext))
363 					if (!(reg_val & BIT(i)))
364 						reg_val |= BIT(i);
365 			}
366 			reg_val &= riscv_isa_extension_base(NULL);
367 			/* Do not modify anything beyond single letter extensions */
368 			reg_val = (vcpu->arch.isa[0] & ~KVM_RISCV_BASE_ISA_MASK) |
369 				  (reg_val & KVM_RISCV_BASE_ISA_MASK);
370 			vcpu->arch.isa[0] = reg_val;
371 			kvm_riscv_vcpu_fp_reset(vcpu);
372 		} else {
373 			return -EOPNOTSUPP;
374 		}
375 		break;
376 	case KVM_REG_RISCV_CONFIG_REG(zicbom_block_size):
377 		return -EOPNOTSUPP;
378 	case KVM_REG_RISCV_CONFIG_REG(zicboz_block_size):
379 		return -EOPNOTSUPP;
380 	case KVM_REG_RISCV_CONFIG_REG(mvendorid):
381 		if (!vcpu->arch.ran_atleast_once)
382 			vcpu->arch.mvendorid = reg_val;
383 		else
384 			return -EBUSY;
385 		break;
386 	case KVM_REG_RISCV_CONFIG_REG(marchid):
387 		if (!vcpu->arch.ran_atleast_once)
388 			vcpu->arch.marchid = reg_val;
389 		else
390 			return -EBUSY;
391 		break;
392 	case KVM_REG_RISCV_CONFIG_REG(mimpid):
393 		if (!vcpu->arch.ran_atleast_once)
394 			vcpu->arch.mimpid = reg_val;
395 		else
396 			return -EBUSY;
397 		break;
398 	default:
399 		return -EINVAL;
400 	}
401 
402 	return 0;
403 }
404 
405 static int kvm_riscv_vcpu_get_reg_core(struct kvm_vcpu *vcpu,
406 				       const struct kvm_one_reg *reg)
407 {
408 	struct kvm_cpu_context *cntx = &vcpu->arch.guest_context;
409 	unsigned long __user *uaddr =
410 			(unsigned long __user *)(unsigned long)reg->addr;
411 	unsigned long reg_num = reg->id & ~(KVM_REG_ARCH_MASK |
412 					    KVM_REG_SIZE_MASK |
413 					    KVM_REG_RISCV_CORE);
414 	unsigned long reg_val;
415 
416 	if (KVM_REG_SIZE(reg->id) != sizeof(unsigned long))
417 		return -EINVAL;
418 	if (reg_num >= sizeof(struct kvm_riscv_core) / sizeof(unsigned long))
419 		return -EINVAL;
420 
421 	if (reg_num == KVM_REG_RISCV_CORE_REG(regs.pc))
422 		reg_val = cntx->sepc;
423 	else if (KVM_REG_RISCV_CORE_REG(regs.pc) < reg_num &&
424 		 reg_num <= KVM_REG_RISCV_CORE_REG(regs.t6))
425 		reg_val = ((unsigned long *)cntx)[reg_num];
426 	else if (reg_num == KVM_REG_RISCV_CORE_REG(mode))
427 		reg_val = (cntx->sstatus & SR_SPP) ?
428 				KVM_RISCV_MODE_S : KVM_RISCV_MODE_U;
429 	else
430 		return -EINVAL;
431 
432 	if (copy_to_user(uaddr, &reg_val, KVM_REG_SIZE(reg->id)))
433 		return -EFAULT;
434 
435 	return 0;
436 }
437 
438 static int kvm_riscv_vcpu_set_reg_core(struct kvm_vcpu *vcpu,
439 				       const struct kvm_one_reg *reg)
440 {
441 	struct kvm_cpu_context *cntx = &vcpu->arch.guest_context;
442 	unsigned long __user *uaddr =
443 			(unsigned long __user *)(unsigned long)reg->addr;
444 	unsigned long reg_num = reg->id & ~(KVM_REG_ARCH_MASK |
445 					    KVM_REG_SIZE_MASK |
446 					    KVM_REG_RISCV_CORE);
447 	unsigned long reg_val;
448 
449 	if (KVM_REG_SIZE(reg->id) != sizeof(unsigned long))
450 		return -EINVAL;
451 	if (reg_num >= sizeof(struct kvm_riscv_core) / sizeof(unsigned long))
452 		return -EINVAL;
453 
454 	if (copy_from_user(&reg_val, uaddr, KVM_REG_SIZE(reg->id)))
455 		return -EFAULT;
456 
457 	if (reg_num == KVM_REG_RISCV_CORE_REG(regs.pc))
458 		cntx->sepc = reg_val;
459 	else if (KVM_REG_RISCV_CORE_REG(regs.pc) < reg_num &&
460 		 reg_num <= KVM_REG_RISCV_CORE_REG(regs.t6))
461 		((unsigned long *)cntx)[reg_num] = reg_val;
462 	else if (reg_num == KVM_REG_RISCV_CORE_REG(mode)) {
463 		if (reg_val == KVM_RISCV_MODE_S)
464 			cntx->sstatus |= SR_SPP;
465 		else
466 			cntx->sstatus &= ~SR_SPP;
467 	} else
468 		return -EINVAL;
469 
470 	return 0;
471 }
472 
473 static int kvm_riscv_vcpu_general_get_csr(struct kvm_vcpu *vcpu,
474 					  unsigned long reg_num,
475 					  unsigned long *out_val)
476 {
477 	struct kvm_vcpu_csr *csr = &vcpu->arch.guest_csr;
478 
479 	if (reg_num >= sizeof(struct kvm_riscv_csr) / sizeof(unsigned long))
480 		return -EINVAL;
481 
482 	if (reg_num == KVM_REG_RISCV_CSR_REG(sip)) {
483 		kvm_riscv_vcpu_flush_interrupts(vcpu);
484 		*out_val = (csr->hvip >> VSIP_TO_HVIP_SHIFT) & VSIP_VALID_MASK;
485 		*out_val |= csr->hvip & ~IRQ_LOCAL_MASK;
486 	} else
487 		*out_val = ((unsigned long *)csr)[reg_num];
488 
489 	return 0;
490 }
491 
492 static inline int kvm_riscv_vcpu_general_set_csr(struct kvm_vcpu *vcpu,
493 						 unsigned long reg_num,
494 						 unsigned long reg_val)
495 {
496 	struct kvm_vcpu_csr *csr = &vcpu->arch.guest_csr;
497 
498 	if (reg_num >= sizeof(struct kvm_riscv_csr) / sizeof(unsigned long))
499 		return -EINVAL;
500 
501 	if (reg_num == KVM_REG_RISCV_CSR_REG(sip)) {
502 		reg_val &= VSIP_VALID_MASK;
503 		reg_val <<= VSIP_TO_HVIP_SHIFT;
504 	}
505 
506 	((unsigned long *)csr)[reg_num] = reg_val;
507 
508 	if (reg_num == KVM_REG_RISCV_CSR_REG(sip))
509 		WRITE_ONCE(vcpu->arch.irqs_pending_mask[0], 0);
510 
511 	return 0;
512 }
513 
514 static int kvm_riscv_vcpu_get_reg_csr(struct kvm_vcpu *vcpu,
515 				      const struct kvm_one_reg *reg)
516 {
517 	int rc;
518 	unsigned long __user *uaddr =
519 			(unsigned long __user *)(unsigned long)reg->addr;
520 	unsigned long reg_num = reg->id & ~(KVM_REG_ARCH_MASK |
521 					    KVM_REG_SIZE_MASK |
522 					    KVM_REG_RISCV_CSR);
523 	unsigned long reg_val, reg_subtype;
524 
525 	if (KVM_REG_SIZE(reg->id) != sizeof(unsigned long))
526 		return -EINVAL;
527 
528 	reg_subtype = reg_num & KVM_REG_RISCV_SUBTYPE_MASK;
529 	reg_num &= ~KVM_REG_RISCV_SUBTYPE_MASK;
530 	switch (reg_subtype) {
531 	case KVM_REG_RISCV_CSR_GENERAL:
532 		rc = kvm_riscv_vcpu_general_get_csr(vcpu, reg_num, &reg_val);
533 		break;
534 	case KVM_REG_RISCV_CSR_AIA:
535 		rc = kvm_riscv_vcpu_aia_get_csr(vcpu, reg_num, &reg_val);
536 		break;
537 	default:
538 		rc = -EINVAL;
539 		break;
540 	}
541 	if (rc)
542 		return rc;
543 
544 	if (copy_to_user(uaddr, &reg_val, KVM_REG_SIZE(reg->id)))
545 		return -EFAULT;
546 
547 	return 0;
548 }
549 
550 static int kvm_riscv_vcpu_set_reg_csr(struct kvm_vcpu *vcpu,
551 				      const struct kvm_one_reg *reg)
552 {
553 	int rc;
554 	unsigned long __user *uaddr =
555 			(unsigned long __user *)(unsigned long)reg->addr;
556 	unsigned long reg_num = reg->id & ~(KVM_REG_ARCH_MASK |
557 					    KVM_REG_SIZE_MASK |
558 					    KVM_REG_RISCV_CSR);
559 	unsigned long reg_val, reg_subtype;
560 
561 	if (KVM_REG_SIZE(reg->id) != sizeof(unsigned long))
562 		return -EINVAL;
563 
564 	if (copy_from_user(&reg_val, uaddr, KVM_REG_SIZE(reg->id)))
565 		return -EFAULT;
566 
567 	reg_subtype = reg_num & KVM_REG_RISCV_SUBTYPE_MASK;
568 	reg_num &= ~KVM_REG_RISCV_SUBTYPE_MASK;
569 	switch (reg_subtype) {
570 	case KVM_REG_RISCV_CSR_GENERAL:
571 		rc = kvm_riscv_vcpu_general_set_csr(vcpu, reg_num, reg_val);
572 		break;
573 	case KVM_REG_RISCV_CSR_AIA:
574 		rc = kvm_riscv_vcpu_aia_set_csr(vcpu, reg_num, reg_val);
575 		break;
576 	default:
577 		rc = -EINVAL;
578 		break;
579 	}
580 	if (rc)
581 		return rc;
582 
583 	return 0;
584 }
585 
586 static int kvm_riscv_vcpu_get_reg_isa_ext(struct kvm_vcpu *vcpu,
587 					  const struct kvm_one_reg *reg)
588 {
589 	unsigned long __user *uaddr =
590 			(unsigned long __user *)(unsigned long)reg->addr;
591 	unsigned long reg_num = reg->id & ~(KVM_REG_ARCH_MASK |
592 					    KVM_REG_SIZE_MASK |
593 					    KVM_REG_RISCV_ISA_EXT);
594 	unsigned long reg_val = 0;
595 	unsigned long host_isa_ext;
596 
597 	if (KVM_REG_SIZE(reg->id) != sizeof(unsigned long))
598 		return -EINVAL;
599 
600 	if (reg_num >= KVM_RISCV_ISA_EXT_MAX ||
601 	    reg_num >= ARRAY_SIZE(kvm_isa_ext_arr))
602 		return -EINVAL;
603 
604 	host_isa_ext = kvm_isa_ext_arr[reg_num];
605 	if (__riscv_isa_extension_available(vcpu->arch.isa, host_isa_ext))
606 		reg_val = 1; /* Mark the given extension as available */
607 
608 	if (copy_to_user(uaddr, &reg_val, KVM_REG_SIZE(reg->id)))
609 		return -EFAULT;
610 
611 	return 0;
612 }
613 
614 static int kvm_riscv_vcpu_set_reg_isa_ext(struct kvm_vcpu *vcpu,
615 					  const struct kvm_one_reg *reg)
616 {
617 	unsigned long __user *uaddr =
618 			(unsigned long __user *)(unsigned long)reg->addr;
619 	unsigned long reg_num = reg->id & ~(KVM_REG_ARCH_MASK |
620 					    KVM_REG_SIZE_MASK |
621 					    KVM_REG_RISCV_ISA_EXT);
622 	unsigned long reg_val;
623 	unsigned long host_isa_ext;
624 
625 	if (KVM_REG_SIZE(reg->id) != sizeof(unsigned long))
626 		return -EINVAL;
627 
628 	if (reg_num >= KVM_RISCV_ISA_EXT_MAX ||
629 	    reg_num >= ARRAY_SIZE(kvm_isa_ext_arr))
630 		return -EINVAL;
631 
632 	if (copy_from_user(&reg_val, uaddr, KVM_REG_SIZE(reg->id)))
633 		return -EFAULT;
634 
635 	host_isa_ext = kvm_isa_ext_arr[reg_num];
636 	if (!__riscv_isa_extension_available(NULL, host_isa_ext))
637 		return	-EOPNOTSUPP;
638 
639 	if (!vcpu->arch.ran_atleast_once) {
640 		/*
641 		 * All multi-letter extension and a few single letter
642 		 * extension can be disabled
643 		 */
644 		if (reg_val == 1 &&
645 		    kvm_riscv_vcpu_isa_enable_allowed(reg_num))
646 			set_bit(host_isa_ext, vcpu->arch.isa);
647 		else if (!reg_val &&
648 			 kvm_riscv_vcpu_isa_disable_allowed(reg_num))
649 			clear_bit(host_isa_ext, vcpu->arch.isa);
650 		else
651 			return -EINVAL;
652 		kvm_riscv_vcpu_fp_reset(vcpu);
653 	} else {
654 		return -EOPNOTSUPP;
655 	}
656 
657 	return 0;
658 }
659 
660 static int kvm_riscv_vcpu_set_reg(struct kvm_vcpu *vcpu,
661 				  const struct kvm_one_reg *reg)
662 {
663 	switch (reg->id & KVM_REG_RISCV_TYPE_MASK) {
664 	case KVM_REG_RISCV_CONFIG:
665 		return kvm_riscv_vcpu_set_reg_config(vcpu, reg);
666 	case KVM_REG_RISCV_CORE:
667 		return kvm_riscv_vcpu_set_reg_core(vcpu, reg);
668 	case KVM_REG_RISCV_CSR:
669 		return kvm_riscv_vcpu_set_reg_csr(vcpu, reg);
670 	case KVM_REG_RISCV_TIMER:
671 		return kvm_riscv_vcpu_set_reg_timer(vcpu, reg);
672 	case KVM_REG_RISCV_FP_F:
673 		return kvm_riscv_vcpu_set_reg_fp(vcpu, reg,
674 						 KVM_REG_RISCV_FP_F);
675 	case KVM_REG_RISCV_FP_D:
676 		return kvm_riscv_vcpu_set_reg_fp(vcpu, reg,
677 						 KVM_REG_RISCV_FP_D);
678 	case KVM_REG_RISCV_ISA_EXT:
679 		return kvm_riscv_vcpu_set_reg_isa_ext(vcpu, reg);
680 	case KVM_REG_RISCV_SBI_EXT:
681 		return kvm_riscv_vcpu_set_reg_sbi_ext(vcpu, reg);
682 	default:
683 		break;
684 	}
685 
686 	return -EINVAL;
687 }
688 
689 static int kvm_riscv_vcpu_get_reg(struct kvm_vcpu *vcpu,
690 				  const struct kvm_one_reg *reg)
691 {
692 	switch (reg->id & KVM_REG_RISCV_TYPE_MASK) {
693 	case KVM_REG_RISCV_CONFIG:
694 		return kvm_riscv_vcpu_get_reg_config(vcpu, reg);
695 	case KVM_REG_RISCV_CORE:
696 		return kvm_riscv_vcpu_get_reg_core(vcpu, reg);
697 	case KVM_REG_RISCV_CSR:
698 		return kvm_riscv_vcpu_get_reg_csr(vcpu, reg);
699 	case KVM_REG_RISCV_TIMER:
700 		return kvm_riscv_vcpu_get_reg_timer(vcpu, reg);
701 	case KVM_REG_RISCV_FP_F:
702 		return kvm_riscv_vcpu_get_reg_fp(vcpu, reg,
703 						 KVM_REG_RISCV_FP_F);
704 	case KVM_REG_RISCV_FP_D:
705 		return kvm_riscv_vcpu_get_reg_fp(vcpu, reg,
706 						 KVM_REG_RISCV_FP_D);
707 	case KVM_REG_RISCV_ISA_EXT:
708 		return kvm_riscv_vcpu_get_reg_isa_ext(vcpu, reg);
709 	case KVM_REG_RISCV_SBI_EXT:
710 		return kvm_riscv_vcpu_get_reg_sbi_ext(vcpu, reg);
711 	default:
712 		break;
713 	}
714 
715 	return -EINVAL;
716 }
717 
718 long kvm_arch_vcpu_async_ioctl(struct file *filp,
719 			       unsigned int ioctl, unsigned long arg)
720 {
721 	struct kvm_vcpu *vcpu = filp->private_data;
722 	void __user *argp = (void __user *)arg;
723 
724 	if (ioctl == KVM_INTERRUPT) {
725 		struct kvm_interrupt irq;
726 
727 		if (copy_from_user(&irq, argp, sizeof(irq)))
728 			return -EFAULT;
729 
730 		if (irq.irq == KVM_INTERRUPT_SET)
731 			return kvm_riscv_vcpu_set_interrupt(vcpu, IRQ_VS_EXT);
732 		else
733 			return kvm_riscv_vcpu_unset_interrupt(vcpu, IRQ_VS_EXT);
734 	}
735 
736 	return -ENOIOCTLCMD;
737 }
738 
739 long kvm_arch_vcpu_ioctl(struct file *filp,
740 			 unsigned int ioctl, unsigned long arg)
741 {
742 	struct kvm_vcpu *vcpu = filp->private_data;
743 	void __user *argp = (void __user *)arg;
744 	long r = -EINVAL;
745 
746 	switch (ioctl) {
747 	case KVM_SET_ONE_REG:
748 	case KVM_GET_ONE_REG: {
749 		struct kvm_one_reg reg;
750 
751 		r = -EFAULT;
752 		if (copy_from_user(&reg, argp, sizeof(reg)))
753 			break;
754 
755 		if (ioctl == KVM_SET_ONE_REG)
756 			r = kvm_riscv_vcpu_set_reg(vcpu, &reg);
757 		else
758 			r = kvm_riscv_vcpu_get_reg(vcpu, &reg);
759 		break;
760 	}
761 	default:
762 		break;
763 	}
764 
765 	return r;
766 }
767 
768 int kvm_arch_vcpu_ioctl_get_sregs(struct kvm_vcpu *vcpu,
769 				  struct kvm_sregs *sregs)
770 {
771 	return -EINVAL;
772 }
773 
774 int kvm_arch_vcpu_ioctl_set_sregs(struct kvm_vcpu *vcpu,
775 				  struct kvm_sregs *sregs)
776 {
777 	return -EINVAL;
778 }
779 
780 int kvm_arch_vcpu_ioctl_get_fpu(struct kvm_vcpu *vcpu, struct kvm_fpu *fpu)
781 {
782 	return -EINVAL;
783 }
784 
785 int kvm_arch_vcpu_ioctl_set_fpu(struct kvm_vcpu *vcpu, struct kvm_fpu *fpu)
786 {
787 	return -EINVAL;
788 }
789 
790 int kvm_arch_vcpu_ioctl_translate(struct kvm_vcpu *vcpu,
791 				  struct kvm_translation *tr)
792 {
793 	return -EINVAL;
794 }
795 
796 int kvm_arch_vcpu_ioctl_get_regs(struct kvm_vcpu *vcpu, struct kvm_regs *regs)
797 {
798 	return -EINVAL;
799 }
800 
801 int kvm_arch_vcpu_ioctl_set_regs(struct kvm_vcpu *vcpu, struct kvm_regs *regs)
802 {
803 	return -EINVAL;
804 }
805 
806 void kvm_riscv_vcpu_flush_interrupts(struct kvm_vcpu *vcpu)
807 {
808 	struct kvm_vcpu_csr *csr = &vcpu->arch.guest_csr;
809 	unsigned long mask, val;
810 
811 	if (READ_ONCE(vcpu->arch.irqs_pending_mask[0])) {
812 		mask = xchg_acquire(&vcpu->arch.irqs_pending_mask[0], 0);
813 		val = READ_ONCE(vcpu->arch.irqs_pending[0]) & mask;
814 
815 		csr->hvip &= ~mask;
816 		csr->hvip |= val;
817 	}
818 
819 	/* Flush AIA high interrupts */
820 	kvm_riscv_vcpu_aia_flush_interrupts(vcpu);
821 }
822 
823 void kvm_riscv_vcpu_sync_interrupts(struct kvm_vcpu *vcpu)
824 {
825 	unsigned long hvip;
826 	struct kvm_vcpu_arch *v = &vcpu->arch;
827 	struct kvm_vcpu_csr *csr = &vcpu->arch.guest_csr;
828 
829 	/* Read current HVIP and VSIE CSRs */
830 	csr->vsie = csr_read(CSR_VSIE);
831 
832 	/* Sync-up HVIP.VSSIP bit changes does by Guest */
833 	hvip = csr_read(CSR_HVIP);
834 	if ((csr->hvip ^ hvip) & (1UL << IRQ_VS_SOFT)) {
835 		if (hvip & (1UL << IRQ_VS_SOFT)) {
836 			if (!test_and_set_bit(IRQ_VS_SOFT,
837 					      v->irqs_pending_mask))
838 				set_bit(IRQ_VS_SOFT, v->irqs_pending);
839 		} else {
840 			if (!test_and_set_bit(IRQ_VS_SOFT,
841 					      v->irqs_pending_mask))
842 				clear_bit(IRQ_VS_SOFT, v->irqs_pending);
843 		}
844 	}
845 
846 	/* Sync-up AIA high interrupts */
847 	kvm_riscv_vcpu_aia_sync_interrupts(vcpu);
848 
849 	/* Sync-up timer CSRs */
850 	kvm_riscv_vcpu_timer_sync(vcpu);
851 }
852 
853 int kvm_riscv_vcpu_set_interrupt(struct kvm_vcpu *vcpu, unsigned int irq)
854 {
855 	/*
856 	 * We only allow VS-mode software, timer, and external
857 	 * interrupts when irq is one of the local interrupts
858 	 * defined by RISC-V privilege specification.
859 	 */
860 	if (irq < IRQ_LOCAL_MAX &&
861 	    irq != IRQ_VS_SOFT &&
862 	    irq != IRQ_VS_TIMER &&
863 	    irq != IRQ_VS_EXT)
864 		return -EINVAL;
865 
866 	set_bit(irq, vcpu->arch.irqs_pending);
867 	smp_mb__before_atomic();
868 	set_bit(irq, vcpu->arch.irqs_pending_mask);
869 
870 	kvm_vcpu_kick(vcpu);
871 
872 	return 0;
873 }
874 
875 int kvm_riscv_vcpu_unset_interrupt(struct kvm_vcpu *vcpu, unsigned int irq)
876 {
877 	/*
878 	 * We only allow VS-mode software, timer, and external
879 	 * interrupts when irq is one of the local interrupts
880 	 * defined by RISC-V privilege specification.
881 	 */
882 	if (irq < IRQ_LOCAL_MAX &&
883 	    irq != IRQ_VS_SOFT &&
884 	    irq != IRQ_VS_TIMER &&
885 	    irq != IRQ_VS_EXT)
886 		return -EINVAL;
887 
888 	clear_bit(irq, vcpu->arch.irqs_pending);
889 	smp_mb__before_atomic();
890 	set_bit(irq, vcpu->arch.irqs_pending_mask);
891 
892 	return 0;
893 }
894 
895 bool kvm_riscv_vcpu_has_interrupts(struct kvm_vcpu *vcpu, u64 mask)
896 {
897 	unsigned long ie;
898 
899 	ie = ((vcpu->arch.guest_csr.vsie & VSIP_VALID_MASK)
900 		<< VSIP_TO_HVIP_SHIFT) & (unsigned long)mask;
901 	ie |= vcpu->arch.guest_csr.vsie & ~IRQ_LOCAL_MASK &
902 		(unsigned long)mask;
903 	if (READ_ONCE(vcpu->arch.irqs_pending[0]) & ie)
904 		return true;
905 
906 	/* Check AIA high interrupts */
907 	return kvm_riscv_vcpu_aia_has_interrupts(vcpu, mask);
908 }
909 
910 void kvm_riscv_vcpu_power_off(struct kvm_vcpu *vcpu)
911 {
912 	vcpu->arch.power_off = true;
913 	kvm_make_request(KVM_REQ_SLEEP, vcpu);
914 	kvm_vcpu_kick(vcpu);
915 }
916 
917 void kvm_riscv_vcpu_power_on(struct kvm_vcpu *vcpu)
918 {
919 	vcpu->arch.power_off = false;
920 	kvm_vcpu_wake_up(vcpu);
921 }
922 
923 int kvm_arch_vcpu_ioctl_get_mpstate(struct kvm_vcpu *vcpu,
924 				    struct kvm_mp_state *mp_state)
925 {
926 	if (vcpu->arch.power_off)
927 		mp_state->mp_state = KVM_MP_STATE_STOPPED;
928 	else
929 		mp_state->mp_state = KVM_MP_STATE_RUNNABLE;
930 
931 	return 0;
932 }
933 
934 int kvm_arch_vcpu_ioctl_set_mpstate(struct kvm_vcpu *vcpu,
935 				    struct kvm_mp_state *mp_state)
936 {
937 	int ret = 0;
938 
939 	switch (mp_state->mp_state) {
940 	case KVM_MP_STATE_RUNNABLE:
941 		vcpu->arch.power_off = false;
942 		break;
943 	case KVM_MP_STATE_STOPPED:
944 		kvm_riscv_vcpu_power_off(vcpu);
945 		break;
946 	default:
947 		ret = -EINVAL;
948 	}
949 
950 	return ret;
951 }
952 
953 int kvm_arch_vcpu_ioctl_set_guest_debug(struct kvm_vcpu *vcpu,
954 					struct kvm_guest_debug *dbg)
955 {
956 	/* TODO; To be implemented later. */
957 	return -EINVAL;
958 }
959 
960 static void kvm_riscv_vcpu_update_config(const unsigned long *isa)
961 {
962 	u64 henvcfg = 0;
963 
964 	if (riscv_isa_extension_available(isa, SVPBMT))
965 		henvcfg |= ENVCFG_PBMTE;
966 
967 	if (riscv_isa_extension_available(isa, SSTC))
968 		henvcfg |= ENVCFG_STCE;
969 
970 	if (riscv_isa_extension_available(isa, ZICBOM))
971 		henvcfg |= (ENVCFG_CBIE | ENVCFG_CBCFE);
972 
973 	if (riscv_isa_extension_available(isa, ZICBOZ))
974 		henvcfg |= ENVCFG_CBZE;
975 
976 	csr_write(CSR_HENVCFG, henvcfg);
977 #ifdef CONFIG_32BIT
978 	csr_write(CSR_HENVCFGH, henvcfg >> 32);
979 #endif
980 }
981 
982 void kvm_arch_vcpu_load(struct kvm_vcpu *vcpu, int cpu)
983 {
984 	struct kvm_vcpu_csr *csr = &vcpu->arch.guest_csr;
985 
986 	csr_write(CSR_VSSTATUS, csr->vsstatus);
987 	csr_write(CSR_VSIE, csr->vsie);
988 	csr_write(CSR_VSTVEC, csr->vstvec);
989 	csr_write(CSR_VSSCRATCH, csr->vsscratch);
990 	csr_write(CSR_VSEPC, csr->vsepc);
991 	csr_write(CSR_VSCAUSE, csr->vscause);
992 	csr_write(CSR_VSTVAL, csr->vstval);
993 	csr_write(CSR_HVIP, csr->hvip);
994 	csr_write(CSR_VSATP, csr->vsatp);
995 
996 	kvm_riscv_vcpu_update_config(vcpu->arch.isa);
997 
998 	kvm_riscv_gstage_update_hgatp(vcpu);
999 
1000 	kvm_riscv_vcpu_timer_restore(vcpu);
1001 
1002 	kvm_riscv_vcpu_host_fp_save(&vcpu->arch.host_context);
1003 	kvm_riscv_vcpu_guest_fp_restore(&vcpu->arch.guest_context,
1004 					vcpu->arch.isa);
1005 
1006 	kvm_riscv_vcpu_aia_load(vcpu, cpu);
1007 
1008 	vcpu->cpu = cpu;
1009 }
1010 
1011 void kvm_arch_vcpu_put(struct kvm_vcpu *vcpu)
1012 {
1013 	struct kvm_vcpu_csr *csr = &vcpu->arch.guest_csr;
1014 
1015 	vcpu->cpu = -1;
1016 
1017 	kvm_riscv_vcpu_aia_put(vcpu);
1018 
1019 	kvm_riscv_vcpu_guest_fp_save(&vcpu->arch.guest_context,
1020 				     vcpu->arch.isa);
1021 	kvm_riscv_vcpu_host_fp_restore(&vcpu->arch.host_context);
1022 
1023 	kvm_riscv_vcpu_timer_save(vcpu);
1024 
1025 	csr->vsstatus = csr_read(CSR_VSSTATUS);
1026 	csr->vsie = csr_read(CSR_VSIE);
1027 	csr->vstvec = csr_read(CSR_VSTVEC);
1028 	csr->vsscratch = csr_read(CSR_VSSCRATCH);
1029 	csr->vsepc = csr_read(CSR_VSEPC);
1030 	csr->vscause = csr_read(CSR_VSCAUSE);
1031 	csr->vstval = csr_read(CSR_VSTVAL);
1032 	csr->hvip = csr_read(CSR_HVIP);
1033 	csr->vsatp = csr_read(CSR_VSATP);
1034 }
1035 
1036 static void kvm_riscv_check_vcpu_requests(struct kvm_vcpu *vcpu)
1037 {
1038 	struct rcuwait *wait = kvm_arch_vcpu_get_wait(vcpu);
1039 
1040 	if (kvm_request_pending(vcpu)) {
1041 		if (kvm_check_request(KVM_REQ_SLEEP, vcpu)) {
1042 			kvm_vcpu_srcu_read_unlock(vcpu);
1043 			rcuwait_wait_event(wait,
1044 				(!vcpu->arch.power_off) && (!vcpu->arch.pause),
1045 				TASK_INTERRUPTIBLE);
1046 			kvm_vcpu_srcu_read_lock(vcpu);
1047 
1048 			if (vcpu->arch.power_off || vcpu->arch.pause) {
1049 				/*
1050 				 * Awaken to handle a signal, request to
1051 				 * sleep again later.
1052 				 */
1053 				kvm_make_request(KVM_REQ_SLEEP, vcpu);
1054 			}
1055 		}
1056 
1057 		if (kvm_check_request(KVM_REQ_VCPU_RESET, vcpu))
1058 			kvm_riscv_reset_vcpu(vcpu);
1059 
1060 		if (kvm_check_request(KVM_REQ_UPDATE_HGATP, vcpu))
1061 			kvm_riscv_gstage_update_hgatp(vcpu);
1062 
1063 		if (kvm_check_request(KVM_REQ_FENCE_I, vcpu))
1064 			kvm_riscv_fence_i_process(vcpu);
1065 
1066 		/*
1067 		 * The generic KVM_REQ_TLB_FLUSH is same as
1068 		 * KVM_REQ_HFENCE_GVMA_VMID_ALL
1069 		 */
1070 		if (kvm_check_request(KVM_REQ_HFENCE_GVMA_VMID_ALL, vcpu))
1071 			kvm_riscv_hfence_gvma_vmid_all_process(vcpu);
1072 
1073 		if (kvm_check_request(KVM_REQ_HFENCE_VVMA_ALL, vcpu))
1074 			kvm_riscv_hfence_vvma_all_process(vcpu);
1075 
1076 		if (kvm_check_request(KVM_REQ_HFENCE, vcpu))
1077 			kvm_riscv_hfence_process(vcpu);
1078 	}
1079 }
1080 
1081 static void kvm_riscv_update_hvip(struct kvm_vcpu *vcpu)
1082 {
1083 	struct kvm_vcpu_csr *csr = &vcpu->arch.guest_csr;
1084 
1085 	csr_write(CSR_HVIP, csr->hvip);
1086 	kvm_riscv_vcpu_aia_update_hvip(vcpu);
1087 }
1088 
1089 /*
1090  * Actually run the vCPU, entering an RCU extended quiescent state (EQS) while
1091  * the vCPU is running.
1092  *
1093  * This must be noinstr as instrumentation may make use of RCU, and this is not
1094  * safe during the EQS.
1095  */
1096 static void noinstr kvm_riscv_vcpu_enter_exit(struct kvm_vcpu *vcpu)
1097 {
1098 	guest_state_enter_irqoff();
1099 	__kvm_riscv_switch_to(&vcpu->arch);
1100 	vcpu->arch.last_exit_cpu = vcpu->cpu;
1101 	guest_state_exit_irqoff();
1102 }
1103 
1104 int kvm_arch_vcpu_ioctl_run(struct kvm_vcpu *vcpu)
1105 {
1106 	int ret;
1107 	struct kvm_cpu_trap trap;
1108 	struct kvm_run *run = vcpu->run;
1109 
1110 	/* Mark this VCPU ran at least once */
1111 	vcpu->arch.ran_atleast_once = true;
1112 
1113 	kvm_vcpu_srcu_read_lock(vcpu);
1114 
1115 	switch (run->exit_reason) {
1116 	case KVM_EXIT_MMIO:
1117 		/* Process MMIO value returned from user-space */
1118 		ret = kvm_riscv_vcpu_mmio_return(vcpu, vcpu->run);
1119 		break;
1120 	case KVM_EXIT_RISCV_SBI:
1121 		/* Process SBI value returned from user-space */
1122 		ret = kvm_riscv_vcpu_sbi_return(vcpu, vcpu->run);
1123 		break;
1124 	case KVM_EXIT_RISCV_CSR:
1125 		/* Process CSR value returned from user-space */
1126 		ret = kvm_riscv_vcpu_csr_return(vcpu, vcpu->run);
1127 		break;
1128 	default:
1129 		ret = 0;
1130 		break;
1131 	}
1132 	if (ret) {
1133 		kvm_vcpu_srcu_read_unlock(vcpu);
1134 		return ret;
1135 	}
1136 
1137 	if (run->immediate_exit) {
1138 		kvm_vcpu_srcu_read_unlock(vcpu);
1139 		return -EINTR;
1140 	}
1141 
1142 	vcpu_load(vcpu);
1143 
1144 	kvm_sigset_activate(vcpu);
1145 
1146 	ret = 1;
1147 	run->exit_reason = KVM_EXIT_UNKNOWN;
1148 	while (ret > 0) {
1149 		/* Check conditions before entering the guest */
1150 		ret = xfer_to_guest_mode_handle_work(vcpu);
1151 		if (ret)
1152 			continue;
1153 		ret = 1;
1154 
1155 		kvm_riscv_gstage_vmid_update(vcpu);
1156 
1157 		kvm_riscv_check_vcpu_requests(vcpu);
1158 
1159 		preempt_disable();
1160 
1161 		/* Update AIA HW state before entering guest */
1162 		ret = kvm_riscv_vcpu_aia_update(vcpu);
1163 		if (ret <= 0) {
1164 			preempt_enable();
1165 			continue;
1166 		}
1167 
1168 		local_irq_disable();
1169 
1170 		/*
1171 		 * Ensure we set mode to IN_GUEST_MODE after we disable
1172 		 * interrupts and before the final VCPU requests check.
1173 		 * See the comment in kvm_vcpu_exiting_guest_mode() and
1174 		 * Documentation/virt/kvm/vcpu-requests.rst
1175 		 */
1176 		vcpu->mode = IN_GUEST_MODE;
1177 
1178 		kvm_vcpu_srcu_read_unlock(vcpu);
1179 		smp_mb__after_srcu_read_unlock();
1180 
1181 		/*
1182 		 * We might have got VCPU interrupts updated asynchronously
1183 		 * so update it in HW.
1184 		 */
1185 		kvm_riscv_vcpu_flush_interrupts(vcpu);
1186 
1187 		/* Update HVIP CSR for current CPU */
1188 		kvm_riscv_update_hvip(vcpu);
1189 
1190 		if (ret <= 0 ||
1191 		    kvm_riscv_gstage_vmid_ver_changed(&vcpu->kvm->arch.vmid) ||
1192 		    kvm_request_pending(vcpu) ||
1193 		    xfer_to_guest_mode_work_pending()) {
1194 			vcpu->mode = OUTSIDE_GUEST_MODE;
1195 			local_irq_enable();
1196 			preempt_enable();
1197 			kvm_vcpu_srcu_read_lock(vcpu);
1198 			continue;
1199 		}
1200 
1201 		/*
1202 		 * Cleanup stale TLB enteries
1203 		 *
1204 		 * Note: This should be done after G-stage VMID has been
1205 		 * updated using kvm_riscv_gstage_vmid_ver_changed()
1206 		 */
1207 		kvm_riscv_local_tlb_sanitize(vcpu);
1208 
1209 		guest_timing_enter_irqoff();
1210 
1211 		kvm_riscv_vcpu_enter_exit(vcpu);
1212 
1213 		vcpu->mode = OUTSIDE_GUEST_MODE;
1214 		vcpu->stat.exits++;
1215 
1216 		/*
1217 		 * Save SCAUSE, STVAL, HTVAL, and HTINST because we might
1218 		 * get an interrupt between __kvm_riscv_switch_to() and
1219 		 * local_irq_enable() which can potentially change CSRs.
1220 		 */
1221 		trap.sepc = vcpu->arch.guest_context.sepc;
1222 		trap.scause = csr_read(CSR_SCAUSE);
1223 		trap.stval = csr_read(CSR_STVAL);
1224 		trap.htval = csr_read(CSR_HTVAL);
1225 		trap.htinst = csr_read(CSR_HTINST);
1226 
1227 		/* Syncup interrupts state with HW */
1228 		kvm_riscv_vcpu_sync_interrupts(vcpu);
1229 
1230 		/*
1231 		 * We must ensure that any pending interrupts are taken before
1232 		 * we exit guest timing so that timer ticks are accounted as
1233 		 * guest time. Transiently unmask interrupts so that any
1234 		 * pending interrupts are taken.
1235 		 *
1236 		 * There's no barrier which ensures that pending interrupts are
1237 		 * recognised, so we just hope that the CPU takes any pending
1238 		 * interrupts between the enable and disable.
1239 		 */
1240 		local_irq_enable();
1241 		local_irq_disable();
1242 
1243 		guest_timing_exit_irqoff();
1244 
1245 		local_irq_enable();
1246 
1247 		preempt_enable();
1248 
1249 		kvm_vcpu_srcu_read_lock(vcpu);
1250 
1251 		ret = kvm_riscv_vcpu_exit(vcpu, run, &trap);
1252 	}
1253 
1254 	kvm_sigset_deactivate(vcpu);
1255 
1256 	vcpu_put(vcpu);
1257 
1258 	kvm_vcpu_srcu_read_unlock(vcpu);
1259 
1260 	return ret;
1261 }
1262