xref: /linux/arch/riscv/kvm/mmu.c (revision 49476d58f2171afc2e899da8040710d2c37760af)
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/errno.h>
10 #include <linux/hugetlb.h>
11 #include <linux/module.h>
12 #include <linux/uaccess.h>
13 #include <linux/vmalloc.h>
14 #include <linux/kvm_host.h>
15 #include <linux/sched/signal.h>
16 #include <asm/kvm_mmu.h>
17 #include <asm/kvm_nacl.h>
18 
19 static void mmu_wp_memory_region(struct kvm *kvm, int slot)
20 {
21 	struct kvm_memslots *slots = kvm_memslots(kvm);
22 	struct kvm_memory_slot *memslot = id_to_memslot(slots, slot);
23 	phys_addr_t start = memslot->base_gfn << PAGE_SHIFT;
24 	phys_addr_t end = (memslot->base_gfn + memslot->npages) << PAGE_SHIFT;
25 	struct kvm_gstage gstage;
26 	bool flush;
27 
28 	kvm_riscv_gstage_init(&gstage, kvm);
29 
30 	write_lock(&kvm->mmu_lock);
31 	flush = kvm_riscv_gstage_wp_range(&gstage, start, end);
32 	write_unlock(&kvm->mmu_lock);
33 	if (flush)
34 		kvm_flush_remote_tlbs_memslot(kvm, memslot);
35 }
36 
37 int kvm_riscv_mmu_ioremap(struct kvm *kvm, gpa_t gpa, phys_addr_t hpa,
38 			  unsigned long size, bool writable, bool in_atomic)
39 {
40 	int ret = 0;
41 	pgprot_t prot;
42 	unsigned long pfn;
43 	phys_addr_t addr, end;
44 	struct kvm_mmu_memory_cache pcache = {
45 		.gfp_custom = (in_atomic) ? GFP_ATOMIC | __GFP_ACCOUNT : 0,
46 		.gfp_zero = __GFP_ZERO,
47 	};
48 	struct kvm_gstage_mapping map;
49 	struct kvm_gstage gstage;
50 
51 	kvm_riscv_gstage_init(&gstage, kvm);
52 
53 	end = (gpa + size + PAGE_SIZE - 1) & PAGE_MASK;
54 	pfn = __phys_to_pfn(hpa);
55 	prot = pgprot_noncached(PAGE_WRITE);
56 
57 	for (addr = gpa; addr < end; addr += PAGE_SIZE) {
58 		map.addr = addr;
59 		map.pte = pfn_pte(pfn, prot);
60 		map.pte = pte_mkdirty(map.pte);
61 		map.level = 0;
62 
63 		if (!writable)
64 			map.pte = pte_wrprotect(map.pte);
65 
66 		ret = kvm_mmu_topup_memory_cache(&pcache, kvm->arch.pgd_levels);
67 		if (ret)
68 			goto out;
69 
70 		write_lock(&kvm->mmu_lock);
71 		ret = kvm_riscv_gstage_set_pte(&gstage, &pcache, &map);
72 		write_unlock(&kvm->mmu_lock);
73 		if (ret)
74 			goto out;
75 
76 		pfn++;
77 	}
78 
79 out:
80 	kvm_mmu_free_memory_cache(&pcache);
81 	return ret;
82 }
83 
84 void kvm_riscv_mmu_iounmap(struct kvm *kvm, gpa_t gpa, unsigned long size)
85 {
86 	struct kvm_gstage gstage;
87 	bool flush;
88 
89 	kvm_riscv_gstage_init(&gstage, kvm);
90 
91 	write_lock(&kvm->mmu_lock);
92 	flush = kvm_riscv_gstage_unmap_range(&gstage, gpa, size, false);
93 	write_unlock(&kvm->mmu_lock);
94 
95 	if (flush)
96 		kvm_flush_remote_tlbs_range(kvm, gpa >> PAGE_SHIFT,
97 					    size >> PAGE_SHIFT);
98 }
99 
100 void kvm_arch_mmu_enable_log_dirty_pt_masked(struct kvm *kvm,
101 					     struct kvm_memory_slot *slot,
102 					     gfn_t gfn_offset,
103 					     unsigned long mask)
104 {
105 	phys_addr_t base_gfn = slot->base_gfn + gfn_offset;
106 	phys_addr_t start = (base_gfn +  __ffs(mask)) << PAGE_SHIFT;
107 	phys_addr_t end = (base_gfn + __fls(mask) + 1) << PAGE_SHIFT;
108 	struct kvm_gstage gstage;
109 	bool flush;
110 
111 	kvm_riscv_gstage_init(&gstage, kvm);
112 
113 	flush = kvm_riscv_gstage_wp_range(&gstage, start, end);
114 	if (flush)
115 		kvm_flush_remote_tlbs_range(kvm, start >> PAGE_SHIFT,
116 					    (end - start) >> PAGE_SHIFT);
117 }
118 
119 void kvm_arch_sync_dirty_log(struct kvm *kvm, struct kvm_memory_slot *memslot)
120 {
121 }
122 
123 void kvm_arch_free_memslot(struct kvm *kvm, struct kvm_memory_slot *free)
124 {
125 }
126 
127 void kvm_arch_memslots_updated(struct kvm *kvm, u64 gen)
128 {
129 }
130 
131 void kvm_arch_flush_shadow_all(struct kvm *kvm)
132 {
133 	kvm_riscv_mmu_free_pgd(kvm);
134 }
135 
136 void kvm_arch_flush_shadow_memslot(struct kvm *kvm,
137 				   struct kvm_memory_slot *slot)
138 {
139 	gpa_t gpa = slot->base_gfn << PAGE_SHIFT;
140 	phys_addr_t size = slot->npages << PAGE_SHIFT;
141 	struct kvm_gstage gstage;
142 	bool flush;
143 
144 	kvm_riscv_gstage_init(&gstage, kvm);
145 
146 	write_lock(&kvm->mmu_lock);
147 	flush = kvm_riscv_gstage_unmap_range(&gstage, gpa, size, false);
148 	write_unlock(&kvm->mmu_lock);
149 	if (flush)
150 		kvm_flush_remote_tlbs_range(kvm, gpa >> PAGE_SHIFT,
151 					    size >> PAGE_SHIFT);
152 }
153 
154 void kvm_arch_commit_memory_region(struct kvm *kvm,
155 				struct kvm_memory_slot *old,
156 				const struct kvm_memory_slot *new,
157 				enum kvm_mr_change change)
158 {
159 	/*
160 	 * At this point memslot has been committed and dirty pages will be
161 	 * tracked while the memory slot is write protected.
162 	 */
163 	if (change != KVM_MR_DELETE && new->flags & KVM_MEM_LOG_DIRTY_PAGES) {
164 		if (kvm_dirty_log_manual_protect_and_init_set(kvm))
165 			return;
166 		mmu_wp_memory_region(kvm, new->id);
167 	}
168 }
169 
170 int kvm_arch_prepare_memory_region(struct kvm *kvm,
171 				const struct kvm_memory_slot *old,
172 				struct kvm_memory_slot *new,
173 				enum kvm_mr_change change)
174 {
175 	hva_t hva, reg_end, size;
176 	bool writable;
177 	int ret = 0;
178 
179 	if (change != KVM_MR_CREATE && change != KVM_MR_MOVE &&
180 			change != KVM_MR_FLAGS_ONLY)
181 		return 0;
182 
183 	/*
184 	 * Prevent userspace from creating a memory region outside of the GPA
185 	 * space addressable by the KVM guest GPA space.
186 	 */
187 	if ((new->base_gfn + new->npages) >=
188 	     kvm_riscv_gstage_gpa_size(kvm->arch.pgd_levels) >> PAGE_SHIFT)
189 		return -EFAULT;
190 
191 	hva = new->userspace_addr;
192 	size = new->npages << PAGE_SHIFT;
193 	reg_end = hva + size;
194 	writable = !(new->flags & KVM_MEM_READONLY);
195 
196 	mmap_read_lock(current->mm);
197 
198 	/*
199 	 * A memory region could potentially cover multiple VMAs, and
200 	 * any holes between them, so iterate over all of them.
201 	 *
202 	 *     +--------------------------------------------+
203 	 * +---------------+----------------+   +----------------+
204 	 * |   : VMA 1     |      VMA 2     |   |    VMA 3  :    |
205 	 * +---------------+----------------+   +----------------+
206 	 *     |               memory region                |
207 	 *     +--------------------------------------------+
208 	 */
209 	do {
210 		struct vm_area_struct *vma;
211 		hva_t vm_end;
212 
213 		vma = find_vma_intersection(current->mm, hva, reg_end);
214 		if (!vma)
215 			break;
216 
217 		/*
218 		 * Mapping a read-only VMA is only allowed if the
219 		 * memory region is configured as read-only.
220 		 */
221 		if (writable && !(vma->vm_flags & VM_WRITE)) {
222 			ret = -EPERM;
223 			break;
224 		}
225 
226 		/* Take the intersection of this VMA with the memory region */
227 		vm_end = min(reg_end, vma->vm_end);
228 
229 		if (vma->vm_flags & VM_PFNMAP) {
230 			/* IO region dirty page logging not allowed */
231 			if (new->flags & KVM_MEM_LOG_DIRTY_PAGES) {
232 				ret = -EINVAL;
233 				goto out;
234 			}
235 		}
236 		hva = vm_end;
237 	} while (hva < reg_end);
238 
239 out:
240 	mmap_read_unlock(current->mm);
241 	return ret;
242 }
243 
244 bool kvm_unmap_gfn_range(struct kvm *kvm, struct kvm_gfn_range *range)
245 {
246 	struct kvm_gstage gstage;
247 	bool flush;
248 
249 	if (!kvm->arch.pgd)
250 		return false;
251 
252 	lockdep_assert_held_write(&kvm->mmu_lock);
253 
254 	kvm_riscv_gstage_init(&gstage, kvm);
255 	flush = kvm_riscv_gstage_unmap_range(&gstage, range->start << PAGE_SHIFT,
256 					     (range->end - range->start) << PAGE_SHIFT,
257 					     range->may_block);
258 	if (flush)
259 		kvm_flush_remote_tlbs_range(kvm, range->start,
260 					    range->end - range->start);
261 	return false;
262 }
263 
264 bool kvm_age_gfn(struct kvm *kvm, struct kvm_gfn_range *range)
265 {
266 	pte_t *ptep;
267 	u32 ptep_level = 0;
268 	u64 size = (range->end - range->start) << PAGE_SHIFT;
269 	struct kvm_gstage gstage;
270 
271 	if (!kvm->arch.pgd)
272 		return false;
273 
274 	WARN_ON(size != PAGE_SIZE && size != PMD_SIZE && size != PUD_SIZE);
275 
276 	kvm_riscv_gstage_init(&gstage, kvm);
277 	if (!kvm_riscv_gstage_get_leaf(&gstage, range->start << PAGE_SHIFT,
278 				       &ptep, &ptep_level))
279 		return false;
280 
281 	return ptep_test_and_clear_young(NULL, 0, ptep);
282 }
283 
284 bool kvm_test_age_gfn(struct kvm *kvm, struct kvm_gfn_range *range)
285 {
286 	pte_t *ptep;
287 	u32 ptep_level = 0;
288 	u64 size = (range->end - range->start) << PAGE_SHIFT;
289 	struct kvm_gstage gstage;
290 
291 	if (!kvm->arch.pgd)
292 		return false;
293 
294 	WARN_ON(size != PAGE_SIZE && size != PMD_SIZE && size != PUD_SIZE);
295 
296 	kvm_riscv_gstage_init(&gstage, kvm);
297 	if (!kvm_riscv_gstage_get_leaf(&gstage, range->start << PAGE_SHIFT,
298 				       &ptep, &ptep_level))
299 		return false;
300 
301 	return pte_young(ptep_get(ptep));
302 }
303 
304 static bool fault_supports_gstage_huge_mapping(struct kvm_memory_slot *memslot,
305 					       unsigned long hva,
306 					       unsigned long map_size)
307 {
308 	hva_t uaddr_start, uaddr_end;
309 	gpa_t gpa_start;
310 	size_t size;
311 
312 	size = memslot->npages * PAGE_SIZE;
313 	uaddr_start = memslot->userspace_addr;
314 	uaddr_end = uaddr_start + size;
315 
316 	gpa_start = memslot->base_gfn << PAGE_SHIFT;
317 
318 	/*
319 	 * Pages belonging to memslots that don't have the same alignment
320 	 * within a huge page for userspace and GPA cannot be mapped with
321 	 * g-stage block entries, because we'll end up mapping the wrong pages.
322 	 *
323 	 * Consider a layout like the following:
324 	 *
325 	 *    memslot->userspace_addr:
326 	 *    +-----+--------------------+--------------------+---+
327 	 *    |abcde|fgh  vs-stage block  |    vs-stage block tv|xyz|
328 	 *    +-----+--------------------+--------------------+---+
329 	 *
330 	 *    memslot->base_gfn << PAGE_SHIFT:
331 	 *      +---+--------------------+--------------------+-----+
332 	 *      |abc|def  g-stage block  |    g-stage block   |tvxyz|
333 	 *      +---+--------------------+--------------------+-----+
334 	 *
335 	 * If we create those g-stage blocks, we'll end up with this incorrect
336 	 * mapping:
337 	 *   d -> f
338 	 *   e -> g
339 	 *   f -> h
340 	 */
341 	if ((gpa_start & (map_size - 1)) != (uaddr_start & (map_size - 1)))
342 		return false;
343 
344 	/*
345 	 * Next, let's make sure we're not trying to map anything not covered
346 	 * by the memslot. This means we have to prohibit block size mappings
347 	 * for the beginning and end of a non-block aligned and non-block sized
348 	 * memory slot (illustrated by the head and tail parts of the
349 	 * userspace view above containing pages 'abcde' and 'xyz',
350 	 * respectively).
351 	 *
352 	 * Note that it doesn't matter if we do the check using the
353 	 * userspace_addr or the base_gfn, as both are equally aligned (per
354 	 * the check above) and equally sized.
355 	 */
356 	return (hva >= ALIGN(uaddr_start, map_size)) &&
357 	       (hva < ALIGN_DOWN(uaddr_end, map_size));
358 }
359 
360 static int get_hva_mapping_size(struct kvm *kvm,
361 				unsigned long hva)
362 {
363 	int size = PAGE_SIZE;
364 	unsigned long flags;
365 	pgd_t pgd;
366 	p4d_t p4d;
367 	pud_t pud;
368 	pmd_t pmd;
369 
370 	/*
371 	 * Disable IRQs to prevent concurrent tear down of host page tables,
372 	 * e.g. if the primary MMU promotes a P*D to a huge page and then frees
373 	 * the original page table.
374 	 */
375 	local_irq_save(flags);
376 
377 	/*
378 	 * Read each entry once.  As above, a non-leaf entry can be promoted to
379 	 * a huge page _during_ this walk.  Re-reading the entry could send the
380 	 * walk into the weeks, e.g. p*d_leaf() returns false (sees the old
381 	 * value) and then p*d_offset() walks into the target huge page instead
382 	 * of the old page table (sees the new value).
383 	 */
384 	pgd = pgdp_get(pgd_offset(kvm->mm, hva));
385 	if (pgd_none(pgd))
386 		goto out;
387 
388 	p4d = p4dp_get(p4d_offset(&pgd, hva));
389 	if (p4d_none(p4d) || !p4d_present(p4d))
390 		goto out;
391 
392 	pud = pudp_get(pud_offset(&p4d, hva));
393 	if (pud_none(pud) || !pud_present(pud))
394 		goto out;
395 
396 	if (pud_leaf(pud)) {
397 		size = PUD_SIZE;
398 		goto out;
399 	}
400 
401 	pmd = pmdp_get(pmd_offset(&pud, hva));
402 	if (pmd_none(pmd) || !pmd_present(pmd))
403 		goto out;
404 
405 	if (pmd_leaf(pmd))
406 		size = PMD_SIZE;
407 
408 out:
409 	local_irq_restore(flags);
410 	return size;
411 }
412 
413 static unsigned long transparent_hugepage_adjust(struct kvm *kvm,
414 						 struct kvm_memory_slot *memslot,
415 						 unsigned long hva,
416 						 kvm_pfn_t *hfnp, gpa_t *gpa)
417 {
418 	kvm_pfn_t hfn = *hfnp;
419 
420 	/*
421 	 * Make sure the adjustment is done only for THP pages. Also make
422 	 * sure that the HVA and GPA are sufficiently aligned and that the
423 	 * block map is contained within the memslot.
424 	 */
425 	if (fault_supports_gstage_huge_mapping(memslot, hva, PMD_SIZE)) {
426 		int sz;
427 
428 		sz = get_hva_mapping_size(kvm, hva);
429 		if (sz < PMD_SIZE)
430 			return sz;
431 
432 		*gpa &= PMD_MASK;
433 		hfn &= ~(PTRS_PER_PMD - 1);
434 		*hfnp = hfn;
435 
436 		return PMD_SIZE;
437 	}
438 
439 	return PAGE_SIZE;
440 }
441 
442 static unsigned long hugetlb_mapping_size(struct kvm_memory_slot *memslot,
443 					  unsigned long hva,
444 					  unsigned long map_size)
445 {
446 	switch (map_size) {
447 #ifndef CONFIG_32BIT
448 	case PUD_SIZE:
449 		if (fault_supports_gstage_huge_mapping(memslot, hva, PUD_SIZE))
450 			return PUD_SIZE;
451 		fallthrough;
452 #endif
453 	case PMD_SIZE:
454 		if (fault_supports_gstage_huge_mapping(memslot, hva, PMD_SIZE))
455 			return PMD_SIZE;
456 		fallthrough;
457 	case PAGE_SIZE:
458 		return PAGE_SIZE;
459 	default:
460 		return map_size;
461 	}
462 }
463 
464 static bool kvm_riscv_mmu_dirty_log_write_fault_fast(struct kvm *kvm,
465 						     struct kvm_memory_slot *memslot,
466 						     gpa_t gpa,
467 						     struct kvm_gstage_mapping *out_map)
468 {
469 	struct kvm_gstage gstage;
470 	unsigned long mmu_seq;
471 	pte_t old_pte, new_pte;
472 	pte_t *ptep;
473 	gfn_t gfn = gpa >> PAGE_SHIFT;
474 	u32 ptep_level;
475 	bool dirty_marked = false;
476 	bool ret;
477 
478 	kvm_riscv_gstage_init(&gstage, kvm);
479 	mmu_seq = kvm->mmu_invalidate_seq;
480 
481 	read_lock(&kvm->mmu_lock);
482 
483 	if (mmu_invalidate_retry_gfn(kvm, mmu_seq, gfn)) {
484 		ret = false;
485 		goto out_unlock;
486 	}
487 
488 	if (!kvm_riscv_gstage_get_leaf(&gstage, gpa, &ptep, &ptep_level) ||
489 	    ptep_level) {
490 		ret = false;
491 		goto out_unlock;
492 	}
493 
494 	for (;;) {
495 		old_pte = ptep_get(ptep);
496 		if (!(pte_val(old_pte) & _PAGE_LEAF)) {
497 			ret = false;
498 			break;
499 		}
500 
501 		if (!dirty_marked) {
502 			mark_page_dirty_in_slot(kvm, memslot, gfn);
503 			dirty_marked = true;
504 		}
505 
506 		if ((pte_val(old_pte) & (_PAGE_WRITE | _PAGE_DIRTY)) ==
507 		    (_PAGE_WRITE | _PAGE_DIRTY)) {
508 			new_pte = old_pte;
509 			ret = true;
510 			break;
511 		}
512 
513 		new_pte = pte_mkdirty(pte_mkwrite_novma(old_pte));
514 
515 		if (kvm_riscv_gstage_try_update_pte(&gstage, ptep_level, gpa,
516 						    ptep, old_pte, new_pte)) {
517 			ret = true;
518 			break;
519 		}
520 		cpu_relax();
521 	}
522 
523 out_unlock:
524 	read_unlock(&kvm->mmu_lock);
525 
526 	if (ret) {
527 		out_map->addr = gpa & PAGE_MASK;
528 		out_map->level = 0;
529 		out_map->pte = new_pte;
530 	}
531 
532 	return ret;
533 }
534 
535 int kvm_riscv_mmu_map(struct kvm_vcpu *vcpu, struct kvm_memory_slot *memslot,
536 		      gpa_t gpa, unsigned long hva, bool is_write,
537 		      struct kvm_gstage_mapping *out_map)
538 {
539 	int ret;
540 	kvm_pfn_t hfn;
541 	bool is_hugetlb;
542 	bool writable;
543 	short vma_pageshift;
544 	gfn_t gfn = gpa >> PAGE_SHIFT;
545 	struct vm_area_struct *vma;
546 	struct kvm *kvm = vcpu->kvm;
547 	struct kvm_mmu_memory_cache *pcache = &vcpu->arch.mmu_page_cache;
548 	bool logging = kvm_slot_dirty_track_enabled(memslot) &&
549 		       !(memslot->flags & KVM_MEM_READONLY);
550 	unsigned long vma_pagesize, mmu_seq;
551 	struct kvm_gstage gstage;
552 	struct page *page;
553 
554 	kvm_riscv_gstage_init(&gstage, kvm);
555 
556 	/* Setup initial state of output mapping */
557 	memset(out_map, 0, sizeof(*out_map));
558 
559 	if (is_write && logging &&
560 	    kvm_riscv_mmu_dirty_log_write_fault_fast(kvm, memslot, gpa, out_map))
561 		return 0;
562 
563 	/* We need minimum second+third level pages */
564 	ret = kvm_mmu_topup_memory_cache(pcache, kvm->arch.pgd_levels);
565 	if (ret) {
566 		kvm_err("Failed to topup G-stage cache\n");
567 		return ret;
568 	}
569 
570 	mmap_read_lock(current->mm);
571 
572 	vma = vma_lookup(current->mm, hva);
573 	if (unlikely(!vma)) {
574 		kvm_err("Failed to find VMA for hva 0x%lx\n", hva);
575 		mmap_read_unlock(current->mm);
576 		return -EFAULT;
577 	}
578 
579 	is_hugetlb = is_vm_hugetlb_page(vma);
580 	if (is_hugetlb)
581 		vma_pageshift = huge_page_shift(hstate_vma(vma));
582 	else
583 		vma_pageshift = PAGE_SHIFT;
584 	vma_pagesize = 1ULL << vma_pageshift;
585 	if (logging || (vma->vm_flags & VM_PFNMAP))
586 		vma_pagesize = PAGE_SIZE;
587 	else if (is_hugetlb)
588 		vma_pagesize = hugetlb_mapping_size(memslot, hva, vma_pagesize);
589 
590 	/*
591 	 * For hugetlb mappings, vma_pagesize might have been reduced from the
592 	 * VMA size to a smaller safe mapping size.
593 	 */
594 	if (vma_pagesize == PMD_SIZE || vma_pagesize == PUD_SIZE)
595 		gfn = ALIGN_DOWN(gpa, vma_pagesize) >> PAGE_SHIFT;
596 
597 	/*
598 	 * Read mmu_invalidate_seq so that KVM can detect if the results of
599 	 * vma_lookup() or __kvm_faultin_pfn() become stale prior to acquiring
600 	 * kvm->mmu_lock.
601 	 *
602 	 * Rely on mmap_read_unlock() for an implicit smp_rmb(), which pairs
603 	 * with the smp_wmb() in kvm_mmu_invalidate_end().
604 	 */
605 	mmu_seq = kvm->mmu_invalidate_seq;
606 	mmap_read_unlock(current->mm);
607 
608 	if (vma_pagesize != PUD_SIZE &&
609 	    vma_pagesize != PMD_SIZE &&
610 	    vma_pagesize != PAGE_SIZE) {
611 		kvm_err("Invalid VMA page size 0x%lx\n", vma_pagesize);
612 		return -EFAULT;
613 	}
614 
615 	hfn = __kvm_faultin_pfn(memslot, gfn, is_write ? FOLL_WRITE : 0,
616 				&writable, &page);
617 	if (hfn == KVM_PFN_ERR_HWPOISON) {
618 		send_sig_mceerr(BUS_MCEERR_AR, (void __user *)hva,
619 				vma_pageshift, current);
620 		return 0;
621 	}
622 	if (is_error_noslot_pfn(hfn))
623 		return -EFAULT;
624 
625 	/*
626 	 * If logging is active then we allow writable pages only
627 	 * for write faults.
628 	 */
629 	if (logging && !is_write)
630 		writable = false;
631 
632 	write_lock(&kvm->mmu_lock);
633 
634 	if (mmu_invalidate_retry(kvm, mmu_seq))
635 		goto out_unlock;
636 
637 	/*
638 	 * Check if we are backed by a THP and thus use block mapping if
639 	 * possible. Hugetlb mappings already selected their target size above,
640 	 * so do not promote them through the THP helper.
641 	 */
642 	if (!logging && !is_hugetlb && vma_pagesize == PAGE_SIZE)
643 		vma_pagesize = transparent_hugepage_adjust(kvm, memslot, hva, &hfn, &gpa);
644 
645 	if (writable) {
646 		mark_page_dirty_in_slot(kvm, memslot, gfn);
647 		ret = kvm_riscv_gstage_map_page(&gstage, pcache, gpa, hfn << PAGE_SHIFT,
648 						vma_pagesize, false, true, out_map);
649 	} else {
650 		ret = kvm_riscv_gstage_map_page(&gstage, pcache, gpa, hfn << PAGE_SHIFT,
651 						vma_pagesize, true, true, out_map);
652 	}
653 
654 	if (ret)
655 		kvm_err("Failed to map in G-stage\n");
656 
657 out_unlock:
658 	kvm_release_faultin_page(kvm, page, ret && ret != -EEXIST, writable);
659 	write_unlock(&kvm->mmu_lock);
660 	return ret;
661 }
662 
663 int kvm_riscv_mmu_alloc_pgd(struct kvm *kvm)
664 {
665 	struct page *pgd_page;
666 
667 	if (kvm->arch.pgd != NULL) {
668 		kvm_err("kvm_arch already initialized?\n");
669 		return -EINVAL;
670 	}
671 
672 	pgd_page = alloc_pages(GFP_KERNEL | __GFP_ZERO,
673 				get_order(kvm_riscv_gstage_pgd_size));
674 	if (!pgd_page)
675 		return -ENOMEM;
676 	kvm->arch.pgd = page_to_virt(pgd_page);
677 	kvm->arch.pgd_phys = page_to_phys(pgd_page);
678 	kvm->arch.pgd_levels = kvm_riscv_gstage_max_pgd_levels;
679 
680 	return 0;
681 }
682 
683 void kvm_riscv_mmu_free_pgd(struct kvm *kvm)
684 {
685 	struct kvm_gstage gstage;
686 	void *pgd = NULL;
687 	bool flush = false;
688 
689 	write_lock(&kvm->mmu_lock);
690 	if (kvm->arch.pgd) {
691 		kvm_riscv_gstage_init(&gstage, kvm);
692 		flush = kvm_riscv_gstage_unmap_range(&gstage, 0UL,
693 			kvm_riscv_gstage_gpa_size(kvm->arch.pgd_levels), false);
694 		pgd = READ_ONCE(kvm->arch.pgd);
695 		kvm->arch.pgd = NULL;
696 		kvm->arch.pgd_phys = 0;
697 		kvm->arch.pgd_levels = 0;
698 	}
699 	write_unlock(&kvm->mmu_lock);
700 
701 	if (flush)
702 		kvm_flush_remote_tlbs(kvm);
703 
704 	if (pgd)
705 		free_pages((unsigned long)pgd, get_order(kvm_riscv_gstage_pgd_size));
706 }
707 
708 void kvm_riscv_mmu_update_hgatp(struct kvm_vcpu *vcpu)
709 {
710 	struct kvm_arch *ka = &vcpu->kvm->arch;
711 	unsigned long hgatp = kvm_riscv_gstage_mode(ka->pgd_levels)
712 			      << HGATP_MODE_SHIFT;
713 
714 	hgatp |= (READ_ONCE(ka->vmid.vmid) << HGATP_VMID_SHIFT) & HGATP_VMID;
715 	hgatp |= (ka->pgd_phys >> PAGE_SHIFT) & HGATP_PPN;
716 
717 	ncsr_write(CSR_HGATP, hgatp);
718 
719 	if (!kvm_riscv_gstage_vmid_bits())
720 		kvm_riscv_local_hfence_gvma_all();
721 }
722