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