1c50d8ae3SPaolo Bonzini // SPDX-License-Identifier: GPL-2.0-only 2c50d8ae3SPaolo Bonzini /* 3c50d8ae3SPaolo Bonzini * Kernel-based Virtual Machine driver for Linux 4c50d8ae3SPaolo Bonzini * 5c50d8ae3SPaolo Bonzini * This module enables machines with Intel VT-x extensions to run virtual 6c50d8ae3SPaolo Bonzini * machines without emulation or binary translation. 7c50d8ae3SPaolo Bonzini * 8c50d8ae3SPaolo Bonzini * MMU support 9c50d8ae3SPaolo Bonzini * 10c50d8ae3SPaolo Bonzini * Copyright (C) 2006 Qumranet, Inc. 11c50d8ae3SPaolo Bonzini * Copyright 2010 Red Hat, Inc. and/or its affiliates. 12c50d8ae3SPaolo Bonzini * 13c50d8ae3SPaolo Bonzini * Authors: 14c50d8ae3SPaolo Bonzini * Yaniv Kamay <yaniv@qumranet.com> 15c50d8ae3SPaolo Bonzini * Avi Kivity <avi@qumranet.com> 16c50d8ae3SPaolo Bonzini */ 17c50d8ae3SPaolo Bonzini 18c50d8ae3SPaolo Bonzini #include "irq.h" 1988197e6aS彭浩(Richard) #include "ioapic.h" 20c50d8ae3SPaolo Bonzini #include "mmu.h" 216ca9a6f3SSean Christopherson #include "mmu_internal.h" 22fe5db27dSBen Gardon #include "tdp_mmu.h" 23c50d8ae3SPaolo Bonzini #include "x86.h" 24c50d8ae3SPaolo Bonzini #include "kvm_cache_regs.h" 25b0b42197SPaolo Bonzini #include "smm.h" 262f728d66SSean Christopherson #include "kvm_emulate.h" 27c50d8ae3SPaolo Bonzini #include "cpuid.h" 285a9624afSPaolo Bonzini #include "spte.h" 29c50d8ae3SPaolo Bonzini 30c50d8ae3SPaolo Bonzini #include <linux/kvm_host.h> 31c50d8ae3SPaolo Bonzini #include <linux/types.h> 32c50d8ae3SPaolo Bonzini #include <linux/string.h> 33c50d8ae3SPaolo Bonzini #include <linux/mm.h> 34c50d8ae3SPaolo Bonzini #include <linux/highmem.h> 35c50d8ae3SPaolo Bonzini #include <linux/moduleparam.h> 36c50d8ae3SPaolo Bonzini #include <linux/export.h> 37c50d8ae3SPaolo Bonzini #include <linux/swap.h> 38c50d8ae3SPaolo Bonzini #include <linux/hugetlb.h> 39c50d8ae3SPaolo Bonzini #include <linux/compiler.h> 40c50d8ae3SPaolo Bonzini #include <linux/srcu.h> 41c50d8ae3SPaolo Bonzini #include <linux/slab.h> 42c50d8ae3SPaolo Bonzini #include <linux/sched/signal.h> 43c50d8ae3SPaolo Bonzini #include <linux/uaccess.h> 44c50d8ae3SPaolo Bonzini #include <linux/hash.h> 45c50d8ae3SPaolo Bonzini #include <linux/kern_levels.h> 46c50d8ae3SPaolo Bonzini #include <linux/kthread.h> 47c50d8ae3SPaolo Bonzini 48c50d8ae3SPaolo Bonzini #include <asm/page.h> 49eb243d1dSIngo Molnar #include <asm/memtype.h> 50c50d8ae3SPaolo Bonzini #include <asm/cmpxchg.h> 51c50d8ae3SPaolo Bonzini #include <asm/io.h> 524a98623dSSean Christopherson #include <asm/set_memory.h> 53c50d8ae3SPaolo Bonzini #include <asm/vmx.h> 54c50d8ae3SPaolo Bonzini #include <asm/kvm_page_track.h> 55c50d8ae3SPaolo Bonzini #include "trace.h" 56c50d8ae3SPaolo Bonzini 57c50d8ae3SPaolo Bonzini extern bool itlb_multihit_kvm_mitigation; 58c50d8ae3SPaolo Bonzini 59a9d6496dSShaokun Zhang int __read_mostly nx_huge_pages = -1; 604dfe4f40SJunaid Shahid static uint __read_mostly nx_huge_pages_recovery_period_ms; 61c50d8ae3SPaolo Bonzini #ifdef CONFIG_PREEMPT_RT 62c50d8ae3SPaolo Bonzini /* Recovery can cause latency spikes, disable it for PREEMPT_RT. */ 63c50d8ae3SPaolo Bonzini static uint __read_mostly nx_huge_pages_recovery_ratio = 0; 64c50d8ae3SPaolo Bonzini #else 65c50d8ae3SPaolo Bonzini static uint __read_mostly nx_huge_pages_recovery_ratio = 60; 66c50d8ae3SPaolo Bonzini #endif 67c50d8ae3SPaolo Bonzini 68c50d8ae3SPaolo Bonzini static int set_nx_huge_pages(const char *val, const struct kernel_param *kp); 694dfe4f40SJunaid Shahid static int set_nx_huge_pages_recovery_param(const char *val, const struct kernel_param *kp); 70c50d8ae3SPaolo Bonzini 71d5d6c18dSJoe Perches static const struct kernel_param_ops nx_huge_pages_ops = { 72c50d8ae3SPaolo Bonzini .set = set_nx_huge_pages, 73c50d8ae3SPaolo Bonzini .get = param_get_bool, 74c50d8ae3SPaolo Bonzini }; 75c50d8ae3SPaolo Bonzini 764dfe4f40SJunaid Shahid static const struct kernel_param_ops nx_huge_pages_recovery_param_ops = { 774dfe4f40SJunaid Shahid .set = set_nx_huge_pages_recovery_param, 78c50d8ae3SPaolo Bonzini .get = param_get_uint, 79c50d8ae3SPaolo Bonzini }; 80c50d8ae3SPaolo Bonzini 81c50d8ae3SPaolo Bonzini module_param_cb(nx_huge_pages, &nx_huge_pages_ops, &nx_huge_pages, 0644); 82c50d8ae3SPaolo Bonzini __MODULE_PARM_TYPE(nx_huge_pages, "bool"); 834dfe4f40SJunaid Shahid module_param_cb(nx_huge_pages_recovery_ratio, &nx_huge_pages_recovery_param_ops, 84c50d8ae3SPaolo Bonzini &nx_huge_pages_recovery_ratio, 0644); 85c50d8ae3SPaolo Bonzini __MODULE_PARM_TYPE(nx_huge_pages_recovery_ratio, "uint"); 864dfe4f40SJunaid Shahid module_param_cb(nx_huge_pages_recovery_period_ms, &nx_huge_pages_recovery_param_ops, 874dfe4f40SJunaid Shahid &nx_huge_pages_recovery_period_ms, 0644); 884dfe4f40SJunaid Shahid __MODULE_PARM_TYPE(nx_huge_pages_recovery_period_ms, "uint"); 89c50d8ae3SPaolo Bonzini 9071fe7013SSean Christopherson static bool __read_mostly force_flush_and_sync_on_reuse; 9171fe7013SSean Christopherson module_param_named(flush_on_reuse, force_flush_and_sync_on_reuse, bool, 0644); 9271fe7013SSean Christopherson 93c50d8ae3SPaolo Bonzini /* 94c50d8ae3SPaolo Bonzini * When setting this variable to true it enables Two-Dimensional-Paging 95c50d8ae3SPaolo Bonzini * where the hardware walks 2 page tables: 96c50d8ae3SPaolo Bonzini * 1. the guest-virtual to guest-physical 97c50d8ae3SPaolo Bonzini * 2. while doing 1. it walks guest-physical to host-physical 98c50d8ae3SPaolo Bonzini * If the hardware supports that we don't need to do shadow paging. 99c50d8ae3SPaolo Bonzini */ 100c50d8ae3SPaolo Bonzini bool tdp_enabled = false; 101c50d8ae3SPaolo Bonzini 1021d92d2e8SSean Christopherson static int max_huge_page_level __read_mostly; 103746700d2SWei Huang static int tdp_root_level __read_mostly; 10483013059SSean Christopherson static int max_tdp_level __read_mostly; 105703c335dSSean Christopherson 106c50d8ae3SPaolo Bonzini #ifdef MMU_DEBUG 1075a9624afSPaolo Bonzini bool dbg = 0; 108c50d8ae3SPaolo Bonzini module_param(dbg, bool, 0644); 109c50d8ae3SPaolo Bonzini #endif 110c50d8ae3SPaolo Bonzini 111c50d8ae3SPaolo Bonzini #define PTE_PREFETCH_NUM 8 112c50d8ae3SPaolo Bonzini 113c50d8ae3SPaolo Bonzini #include <trace/events/kvm.h> 114c50d8ae3SPaolo Bonzini 115dc1cff96SPeter Xu /* make pte_list_desc fit well in cache lines */ 11613236e25SPeter Xu #define PTE_LIST_EXT 14 117c50d8ae3SPaolo Bonzini 11813236e25SPeter Xu /* 11913236e25SPeter Xu * Slight optimization of cacheline layout, by putting `more' and `spte_count' 12013236e25SPeter Xu * at the start; then accessing it will only use one single cacheline for 12113236e25SPeter Xu * either full (entries==PTE_LIST_EXT) case or entries<=6. 12213236e25SPeter Xu */ 123c50d8ae3SPaolo Bonzini struct pte_list_desc { 124c50d8ae3SPaolo Bonzini struct pte_list_desc *more; 12513236e25SPeter Xu /* 12613236e25SPeter Xu * Stores number of entries stored in the pte_list_desc. No need to be 12713236e25SPeter Xu * u64 but just for easier alignment. When PTE_LIST_EXT, means full. 12813236e25SPeter Xu */ 12913236e25SPeter Xu u64 spte_count; 13013236e25SPeter Xu u64 *sptes[PTE_LIST_EXT]; 131c50d8ae3SPaolo Bonzini }; 132c50d8ae3SPaolo Bonzini 133c50d8ae3SPaolo Bonzini struct kvm_shadow_walk_iterator { 134c50d8ae3SPaolo Bonzini u64 addr; 135c50d8ae3SPaolo Bonzini hpa_t shadow_addr; 136c50d8ae3SPaolo Bonzini u64 *sptep; 137c50d8ae3SPaolo Bonzini int level; 138c50d8ae3SPaolo Bonzini unsigned index; 139c50d8ae3SPaolo Bonzini }; 140c50d8ae3SPaolo Bonzini 141c50d8ae3SPaolo Bonzini #define for_each_shadow_entry_using_root(_vcpu, _root, _addr, _walker) \ 142c50d8ae3SPaolo Bonzini for (shadow_walk_init_using_root(&(_walker), (_vcpu), \ 143c50d8ae3SPaolo Bonzini (_root), (_addr)); \ 144c50d8ae3SPaolo Bonzini shadow_walk_okay(&(_walker)); \ 145c50d8ae3SPaolo Bonzini shadow_walk_next(&(_walker))) 146c50d8ae3SPaolo Bonzini 147c50d8ae3SPaolo Bonzini #define for_each_shadow_entry(_vcpu, _addr, _walker) \ 148c50d8ae3SPaolo Bonzini for (shadow_walk_init(&(_walker), _vcpu, _addr); \ 149c50d8ae3SPaolo Bonzini shadow_walk_okay(&(_walker)); \ 150c50d8ae3SPaolo Bonzini shadow_walk_next(&(_walker))) 151c50d8ae3SPaolo Bonzini 152c50d8ae3SPaolo Bonzini #define for_each_shadow_entry_lockless(_vcpu, _addr, _walker, spte) \ 153c50d8ae3SPaolo Bonzini for (shadow_walk_init(&(_walker), _vcpu, _addr); \ 154c50d8ae3SPaolo Bonzini shadow_walk_okay(&(_walker)) && \ 155c50d8ae3SPaolo Bonzini ({ spte = mmu_spte_get_lockless(_walker.sptep); 1; }); \ 156c50d8ae3SPaolo Bonzini __shadow_walk_next(&(_walker), spte)) 157c50d8ae3SPaolo Bonzini 158c50d8ae3SPaolo Bonzini static struct kmem_cache *pte_list_desc_cache; 15902c00b3aSBen Gardon struct kmem_cache *mmu_page_header_cache; 160c50d8ae3SPaolo Bonzini static struct percpu_counter kvm_total_used_mmu_pages; 161c50d8ae3SPaolo Bonzini 162c50d8ae3SPaolo Bonzini static void mmu_spte_set(u64 *sptep, u64 spte); 163c50d8ae3SPaolo Bonzini 164594e91a1SSean Christopherson struct kvm_mmu_role_regs { 165594e91a1SSean Christopherson const unsigned long cr0; 166594e91a1SSean Christopherson const unsigned long cr4; 167594e91a1SSean Christopherson const u64 efer; 168594e91a1SSean Christopherson }; 169594e91a1SSean Christopherson 170c50d8ae3SPaolo Bonzini #define CREATE_TRACE_POINTS 171c50d8ae3SPaolo Bonzini #include "mmutrace.h" 172c50d8ae3SPaolo Bonzini 173594e91a1SSean Christopherson /* 174594e91a1SSean Christopherson * Yes, lot's of underscores. They're a hint that you probably shouldn't be 1757a458f0eSPaolo Bonzini * reading from the role_regs. Once the root_role is constructed, it becomes 176594e91a1SSean Christopherson * the single source of truth for the MMU's state. 177594e91a1SSean Christopherson */ 178594e91a1SSean Christopherson #define BUILD_MMU_ROLE_REGS_ACCESSOR(reg, name, flag) \ 17982ffa13fSPaolo Bonzini static inline bool __maybe_unused \ 18082ffa13fSPaolo Bonzini ____is_##reg##_##name(const struct kvm_mmu_role_regs *regs) \ 181594e91a1SSean Christopherson { \ 182594e91a1SSean Christopherson return !!(regs->reg & flag); \ 183594e91a1SSean Christopherson } 184594e91a1SSean Christopherson BUILD_MMU_ROLE_REGS_ACCESSOR(cr0, pg, X86_CR0_PG); 185594e91a1SSean Christopherson BUILD_MMU_ROLE_REGS_ACCESSOR(cr0, wp, X86_CR0_WP); 186594e91a1SSean Christopherson BUILD_MMU_ROLE_REGS_ACCESSOR(cr4, pse, X86_CR4_PSE); 187594e91a1SSean Christopherson BUILD_MMU_ROLE_REGS_ACCESSOR(cr4, pae, X86_CR4_PAE); 188594e91a1SSean Christopherson BUILD_MMU_ROLE_REGS_ACCESSOR(cr4, smep, X86_CR4_SMEP); 189594e91a1SSean Christopherson BUILD_MMU_ROLE_REGS_ACCESSOR(cr4, smap, X86_CR4_SMAP); 190594e91a1SSean Christopherson BUILD_MMU_ROLE_REGS_ACCESSOR(cr4, pke, X86_CR4_PKE); 191594e91a1SSean Christopherson BUILD_MMU_ROLE_REGS_ACCESSOR(cr4, la57, X86_CR4_LA57); 192594e91a1SSean Christopherson BUILD_MMU_ROLE_REGS_ACCESSOR(efer, nx, EFER_NX); 193594e91a1SSean Christopherson BUILD_MMU_ROLE_REGS_ACCESSOR(efer, lma, EFER_LMA); 194594e91a1SSean Christopherson 19560667724SSean Christopherson /* 19660667724SSean Christopherson * The MMU itself (with a valid role) is the single source of truth for the 19760667724SSean Christopherson * MMU. Do not use the regs used to build the MMU/role, nor the vCPU. The 19860667724SSean Christopherson * regs don't account for dependencies, e.g. clearing CR4 bits if CR0.PG=1, 19960667724SSean Christopherson * and the vCPU may be incorrect/irrelevant. 20060667724SSean Christopherson */ 20160667724SSean Christopherson #define BUILD_MMU_ROLE_ACCESSOR(base_or_ext, reg, name) \ 2024ac21457SPaolo Bonzini static inline bool __maybe_unused is_##reg##_##name(struct kvm_mmu *mmu) \ 20360667724SSean Christopherson { \ 204e5ed0fb0SPaolo Bonzini return !!(mmu->cpu_role. base_or_ext . reg##_##name); \ 20560667724SSean Christopherson } 20660667724SSean Christopherson BUILD_MMU_ROLE_ACCESSOR(base, cr0, wp); 20760667724SSean Christopherson BUILD_MMU_ROLE_ACCESSOR(ext, cr4, pse); 20860667724SSean Christopherson BUILD_MMU_ROLE_ACCESSOR(ext, cr4, smep); 20960667724SSean Christopherson BUILD_MMU_ROLE_ACCESSOR(ext, cr4, smap); 21060667724SSean Christopherson BUILD_MMU_ROLE_ACCESSOR(ext, cr4, pke); 21160667724SSean Christopherson BUILD_MMU_ROLE_ACCESSOR(ext, cr4, la57); 21260667724SSean Christopherson BUILD_MMU_ROLE_ACCESSOR(base, efer, nx); 21356b321f9SPaolo Bonzini BUILD_MMU_ROLE_ACCESSOR(ext, efer, lma); 21460667724SSean Christopherson 215faf72962SPaolo Bonzini static inline bool is_cr0_pg(struct kvm_mmu *mmu) 216faf72962SPaolo Bonzini { 217faf72962SPaolo Bonzini return mmu->cpu_role.base.level > 0; 218faf72962SPaolo Bonzini } 219faf72962SPaolo Bonzini 220faf72962SPaolo Bonzini static inline bool is_cr4_pae(struct kvm_mmu *mmu) 221faf72962SPaolo Bonzini { 222faf72962SPaolo Bonzini return !mmu->cpu_role.base.has_4_byte_gpte; 223faf72962SPaolo Bonzini } 224faf72962SPaolo Bonzini 225594e91a1SSean Christopherson static struct kvm_mmu_role_regs vcpu_to_role_regs(struct kvm_vcpu *vcpu) 226594e91a1SSean Christopherson { 227594e91a1SSean Christopherson struct kvm_mmu_role_regs regs = { 228594e91a1SSean Christopherson .cr0 = kvm_read_cr0_bits(vcpu, KVM_MMU_CR0_ROLE_BITS), 229594e91a1SSean Christopherson .cr4 = kvm_read_cr4_bits(vcpu, KVM_MMU_CR4_ROLE_BITS), 230594e91a1SSean Christopherson .efer = vcpu->arch.efer, 231594e91a1SSean Christopherson }; 232594e91a1SSean Christopherson 233594e91a1SSean Christopherson return regs; 234594e91a1SSean Christopherson } 235c50d8ae3SPaolo Bonzini 236c50d8ae3SPaolo Bonzini static inline bool kvm_available_flush_tlb_with_range(void) 237c50d8ae3SPaolo Bonzini { 238afaf0b2fSSean Christopherson return kvm_x86_ops.tlb_remote_flush_with_range; 239c50d8ae3SPaolo Bonzini } 240c50d8ae3SPaolo Bonzini 241c50d8ae3SPaolo Bonzini static void kvm_flush_remote_tlbs_with_range(struct kvm *kvm, 242c50d8ae3SPaolo Bonzini struct kvm_tlb_range *range) 243c50d8ae3SPaolo Bonzini { 244c50d8ae3SPaolo Bonzini int ret = -ENOTSUPP; 245c50d8ae3SPaolo Bonzini 246afaf0b2fSSean Christopherson if (range && kvm_x86_ops.tlb_remote_flush_with_range) 247b3646477SJason Baron ret = static_call(kvm_x86_tlb_remote_flush_with_range)(kvm, range); 248c50d8ae3SPaolo Bonzini 249c50d8ae3SPaolo Bonzini if (ret) 250c50d8ae3SPaolo Bonzini kvm_flush_remote_tlbs(kvm); 251c50d8ae3SPaolo Bonzini } 252c50d8ae3SPaolo Bonzini 2532f2fad08SBen Gardon void kvm_flush_remote_tlbs_with_address(struct kvm *kvm, 254c50d8ae3SPaolo Bonzini u64 start_gfn, u64 pages) 255c50d8ae3SPaolo Bonzini { 256c50d8ae3SPaolo Bonzini struct kvm_tlb_range range; 257c50d8ae3SPaolo Bonzini 258c50d8ae3SPaolo Bonzini range.start_gfn = start_gfn; 259c50d8ae3SPaolo Bonzini range.pages = pages; 260c50d8ae3SPaolo Bonzini 261c50d8ae3SPaolo Bonzini kvm_flush_remote_tlbs_with_range(kvm, &range); 262c50d8ae3SPaolo Bonzini } 263c50d8ae3SPaolo Bonzini 2648f79b064SBen Gardon static void mark_mmio_spte(struct kvm_vcpu *vcpu, u64 *sptep, u64 gfn, 2658f79b064SBen Gardon unsigned int access) 2668f79b064SBen Gardon { 267c236d962SSean Christopherson u64 spte = make_mmio_spte(vcpu, gfn, access); 2688f79b064SBen Gardon 269c236d962SSean Christopherson trace_mark_mmio_spte(sptep, gfn, spte); 270c236d962SSean Christopherson mmu_spte_set(sptep, spte); 271c50d8ae3SPaolo Bonzini } 272c50d8ae3SPaolo Bonzini 273c50d8ae3SPaolo Bonzini static gfn_t get_mmio_spte_gfn(u64 spte) 274c50d8ae3SPaolo Bonzini { 275c50d8ae3SPaolo Bonzini u64 gpa = spte & shadow_nonpresent_or_rsvd_lower_gfn_mask; 276c50d8ae3SPaolo Bonzini 2778a967d65SPaolo Bonzini gpa |= (spte >> SHADOW_NONPRESENT_OR_RSVD_MASK_LEN) 278c50d8ae3SPaolo Bonzini & shadow_nonpresent_or_rsvd_mask; 279c50d8ae3SPaolo Bonzini 280c50d8ae3SPaolo Bonzini return gpa >> PAGE_SHIFT; 281c50d8ae3SPaolo Bonzini } 282c50d8ae3SPaolo Bonzini 283c50d8ae3SPaolo Bonzini static unsigned get_mmio_spte_access(u64 spte) 284c50d8ae3SPaolo Bonzini { 285c50d8ae3SPaolo Bonzini return spte & shadow_mmio_access_mask; 286c50d8ae3SPaolo Bonzini } 287c50d8ae3SPaolo Bonzini 288c50d8ae3SPaolo Bonzini static bool check_mmio_spte(struct kvm_vcpu *vcpu, u64 spte) 289c50d8ae3SPaolo Bonzini { 290c50d8ae3SPaolo Bonzini u64 kvm_gen, spte_gen, gen; 291c50d8ae3SPaolo Bonzini 292c50d8ae3SPaolo Bonzini gen = kvm_vcpu_memslots(vcpu)->generation; 293c50d8ae3SPaolo Bonzini if (unlikely(gen & KVM_MEMSLOT_GEN_UPDATE_IN_PROGRESS)) 294c50d8ae3SPaolo Bonzini return false; 295c50d8ae3SPaolo Bonzini 296c50d8ae3SPaolo Bonzini kvm_gen = gen & MMIO_SPTE_GEN_MASK; 297c50d8ae3SPaolo Bonzini spte_gen = get_mmio_spte_generation(spte); 298c50d8ae3SPaolo Bonzini 299c50d8ae3SPaolo Bonzini trace_check_mmio_spte(spte, kvm_gen, spte_gen); 300c50d8ae3SPaolo Bonzini return likely(kvm_gen == spte_gen); 301c50d8ae3SPaolo Bonzini } 302c50d8ae3SPaolo Bonzini 303c50d8ae3SPaolo Bonzini static int is_cpuid_PSE36(void) 304c50d8ae3SPaolo Bonzini { 305c50d8ae3SPaolo Bonzini return 1; 306c50d8ae3SPaolo Bonzini } 307c50d8ae3SPaolo Bonzini 308c50d8ae3SPaolo Bonzini #ifdef CONFIG_X86_64 309c50d8ae3SPaolo Bonzini static void __set_spte(u64 *sptep, u64 spte) 310c50d8ae3SPaolo Bonzini { 311c50d8ae3SPaolo Bonzini WRITE_ONCE(*sptep, spte); 312c50d8ae3SPaolo Bonzini } 313c50d8ae3SPaolo Bonzini 314c50d8ae3SPaolo Bonzini static void __update_clear_spte_fast(u64 *sptep, u64 spte) 315c50d8ae3SPaolo Bonzini { 316c50d8ae3SPaolo Bonzini WRITE_ONCE(*sptep, spte); 317c50d8ae3SPaolo Bonzini } 318c50d8ae3SPaolo Bonzini 319c50d8ae3SPaolo Bonzini static u64 __update_clear_spte_slow(u64 *sptep, u64 spte) 320c50d8ae3SPaolo Bonzini { 321c50d8ae3SPaolo Bonzini return xchg(sptep, spte); 322c50d8ae3SPaolo Bonzini } 323c50d8ae3SPaolo Bonzini 324c50d8ae3SPaolo Bonzini static u64 __get_spte_lockless(u64 *sptep) 325c50d8ae3SPaolo Bonzini { 326c50d8ae3SPaolo Bonzini return READ_ONCE(*sptep); 327c50d8ae3SPaolo Bonzini } 328c50d8ae3SPaolo Bonzini #else 329c50d8ae3SPaolo Bonzini union split_spte { 330c50d8ae3SPaolo Bonzini struct { 331c50d8ae3SPaolo Bonzini u32 spte_low; 332c50d8ae3SPaolo Bonzini u32 spte_high; 333c50d8ae3SPaolo Bonzini }; 334c50d8ae3SPaolo Bonzini u64 spte; 335c50d8ae3SPaolo Bonzini }; 336c50d8ae3SPaolo Bonzini 337c50d8ae3SPaolo Bonzini static void count_spte_clear(u64 *sptep, u64 spte) 338c50d8ae3SPaolo Bonzini { 33957354682SSean Christopherson struct kvm_mmu_page *sp = sptep_to_sp(sptep); 340c50d8ae3SPaolo Bonzini 341c50d8ae3SPaolo Bonzini if (is_shadow_present_pte(spte)) 342c50d8ae3SPaolo Bonzini return; 343c50d8ae3SPaolo Bonzini 344c50d8ae3SPaolo Bonzini /* Ensure the spte is completely set before we increase the count */ 345c50d8ae3SPaolo Bonzini smp_wmb(); 346c50d8ae3SPaolo Bonzini sp->clear_spte_count++; 347c50d8ae3SPaolo Bonzini } 348c50d8ae3SPaolo Bonzini 349c50d8ae3SPaolo Bonzini static void __set_spte(u64 *sptep, u64 spte) 350c50d8ae3SPaolo Bonzini { 351c50d8ae3SPaolo Bonzini union split_spte *ssptep, sspte; 352c50d8ae3SPaolo Bonzini 353c50d8ae3SPaolo Bonzini ssptep = (union split_spte *)sptep; 354c50d8ae3SPaolo Bonzini sspte = (union split_spte)spte; 355c50d8ae3SPaolo Bonzini 356c50d8ae3SPaolo Bonzini ssptep->spte_high = sspte.spte_high; 357c50d8ae3SPaolo Bonzini 358c50d8ae3SPaolo Bonzini /* 359c50d8ae3SPaolo Bonzini * If we map the spte from nonpresent to present, We should store 360c50d8ae3SPaolo Bonzini * the high bits firstly, then set present bit, so cpu can not 361c50d8ae3SPaolo Bonzini * fetch this spte while we are setting the spte. 362c50d8ae3SPaolo Bonzini */ 363c50d8ae3SPaolo Bonzini smp_wmb(); 364c50d8ae3SPaolo Bonzini 365c50d8ae3SPaolo Bonzini WRITE_ONCE(ssptep->spte_low, sspte.spte_low); 366c50d8ae3SPaolo Bonzini } 367c50d8ae3SPaolo Bonzini 368c50d8ae3SPaolo Bonzini static void __update_clear_spte_fast(u64 *sptep, u64 spte) 369c50d8ae3SPaolo Bonzini { 370c50d8ae3SPaolo Bonzini union split_spte *ssptep, sspte; 371c50d8ae3SPaolo Bonzini 372c50d8ae3SPaolo Bonzini ssptep = (union split_spte *)sptep; 373c50d8ae3SPaolo Bonzini sspte = (union split_spte)spte; 374c50d8ae3SPaolo Bonzini 375c50d8ae3SPaolo Bonzini WRITE_ONCE(ssptep->spte_low, sspte.spte_low); 376c50d8ae3SPaolo Bonzini 377c50d8ae3SPaolo Bonzini /* 378c50d8ae3SPaolo Bonzini * If we map the spte from present to nonpresent, we should clear 379c50d8ae3SPaolo Bonzini * present bit firstly to avoid vcpu fetch the old high bits. 380c50d8ae3SPaolo Bonzini */ 381c50d8ae3SPaolo Bonzini smp_wmb(); 382c50d8ae3SPaolo Bonzini 383c50d8ae3SPaolo Bonzini ssptep->spte_high = sspte.spte_high; 384c50d8ae3SPaolo Bonzini count_spte_clear(sptep, spte); 385c50d8ae3SPaolo Bonzini } 386c50d8ae3SPaolo Bonzini 387c50d8ae3SPaolo Bonzini static u64 __update_clear_spte_slow(u64 *sptep, u64 spte) 388c50d8ae3SPaolo Bonzini { 389c50d8ae3SPaolo Bonzini union split_spte *ssptep, sspte, orig; 390c50d8ae3SPaolo Bonzini 391c50d8ae3SPaolo Bonzini ssptep = (union split_spte *)sptep; 392c50d8ae3SPaolo Bonzini sspte = (union split_spte)spte; 393c50d8ae3SPaolo Bonzini 394c50d8ae3SPaolo Bonzini /* xchg acts as a barrier before the setting of the high bits */ 395c50d8ae3SPaolo Bonzini orig.spte_low = xchg(&ssptep->spte_low, sspte.spte_low); 396c50d8ae3SPaolo Bonzini orig.spte_high = ssptep->spte_high; 397c50d8ae3SPaolo Bonzini ssptep->spte_high = sspte.spte_high; 398c50d8ae3SPaolo Bonzini count_spte_clear(sptep, spte); 399c50d8ae3SPaolo Bonzini 400c50d8ae3SPaolo Bonzini return orig.spte; 401c50d8ae3SPaolo Bonzini } 402c50d8ae3SPaolo Bonzini 403c50d8ae3SPaolo Bonzini /* 404c50d8ae3SPaolo Bonzini * The idea using the light way get the spte on x86_32 guest is from 405c50d8ae3SPaolo Bonzini * gup_get_pte (mm/gup.c). 406c50d8ae3SPaolo Bonzini * 407aed02fe3SSean Christopherson * An spte tlb flush may be pending, because kvm_set_pte_rmap 408c50d8ae3SPaolo Bonzini * coalesces them and we are running out of the MMU lock. Therefore 409c50d8ae3SPaolo Bonzini * we need to protect against in-progress updates of the spte. 410c50d8ae3SPaolo Bonzini * 411c50d8ae3SPaolo Bonzini * Reading the spte while an update is in progress may get the old value 412c50d8ae3SPaolo Bonzini * for the high part of the spte. The race is fine for a present->non-present 413c50d8ae3SPaolo Bonzini * change (because the high part of the spte is ignored for non-present spte), 414c50d8ae3SPaolo Bonzini * but for a present->present change we must reread the spte. 415c50d8ae3SPaolo Bonzini * 416c50d8ae3SPaolo Bonzini * All such changes are done in two steps (present->non-present and 417c50d8ae3SPaolo Bonzini * non-present->present), hence it is enough to count the number of 418c50d8ae3SPaolo Bonzini * present->non-present updates: if it changed while reading the spte, 419c50d8ae3SPaolo Bonzini * we might have hit the race. This is done using clear_spte_count. 420c50d8ae3SPaolo Bonzini */ 421c50d8ae3SPaolo Bonzini static u64 __get_spte_lockless(u64 *sptep) 422c50d8ae3SPaolo Bonzini { 42357354682SSean Christopherson struct kvm_mmu_page *sp = sptep_to_sp(sptep); 424c50d8ae3SPaolo Bonzini union split_spte spte, *orig = (union split_spte *)sptep; 425c50d8ae3SPaolo Bonzini int count; 426c50d8ae3SPaolo Bonzini 427c50d8ae3SPaolo Bonzini retry: 428c50d8ae3SPaolo Bonzini count = sp->clear_spte_count; 429c50d8ae3SPaolo Bonzini smp_rmb(); 430c50d8ae3SPaolo Bonzini 431c50d8ae3SPaolo Bonzini spte.spte_low = orig->spte_low; 432c50d8ae3SPaolo Bonzini smp_rmb(); 433c50d8ae3SPaolo Bonzini 434c50d8ae3SPaolo Bonzini spte.spte_high = orig->spte_high; 435c50d8ae3SPaolo Bonzini smp_rmb(); 436c50d8ae3SPaolo Bonzini 437c50d8ae3SPaolo Bonzini if (unlikely(spte.spte_low != orig->spte_low || 438c50d8ae3SPaolo Bonzini count != sp->clear_spte_count)) 439c50d8ae3SPaolo Bonzini goto retry; 440c50d8ae3SPaolo Bonzini 441c50d8ae3SPaolo Bonzini return spte.spte; 442c50d8ae3SPaolo Bonzini } 443c50d8ae3SPaolo Bonzini #endif 444c50d8ae3SPaolo Bonzini 445c50d8ae3SPaolo Bonzini /* Rules for using mmu_spte_set: 446c50d8ae3SPaolo Bonzini * Set the sptep from nonpresent to present. 447c50d8ae3SPaolo Bonzini * Note: the sptep being assigned *must* be either not present 448c50d8ae3SPaolo Bonzini * or in a state where the hardware will not attempt to update 449c50d8ae3SPaolo Bonzini * the spte. 450c50d8ae3SPaolo Bonzini */ 451c50d8ae3SPaolo Bonzini static void mmu_spte_set(u64 *sptep, u64 new_spte) 452c50d8ae3SPaolo Bonzini { 453c50d8ae3SPaolo Bonzini WARN_ON(is_shadow_present_pte(*sptep)); 454c50d8ae3SPaolo Bonzini __set_spte(sptep, new_spte); 455c50d8ae3SPaolo Bonzini } 456c50d8ae3SPaolo Bonzini 457c50d8ae3SPaolo Bonzini /* 458c50d8ae3SPaolo Bonzini * Update the SPTE (excluding the PFN), but do not track changes in its 459c50d8ae3SPaolo Bonzini * accessed/dirty status. 460c50d8ae3SPaolo Bonzini */ 461c50d8ae3SPaolo Bonzini static u64 mmu_spte_update_no_track(u64 *sptep, u64 new_spte) 462c50d8ae3SPaolo Bonzini { 463c50d8ae3SPaolo Bonzini u64 old_spte = *sptep; 464c50d8ae3SPaolo Bonzini 465c50d8ae3SPaolo Bonzini WARN_ON(!is_shadow_present_pte(new_spte)); 466115111efSDavid Matlack check_spte_writable_invariants(new_spte); 467c50d8ae3SPaolo Bonzini 468c50d8ae3SPaolo Bonzini if (!is_shadow_present_pte(old_spte)) { 469c50d8ae3SPaolo Bonzini mmu_spte_set(sptep, new_spte); 470c50d8ae3SPaolo Bonzini return old_spte; 471c50d8ae3SPaolo Bonzini } 472c50d8ae3SPaolo Bonzini 473c50d8ae3SPaolo Bonzini if (!spte_has_volatile_bits(old_spte)) 474c50d8ae3SPaolo Bonzini __update_clear_spte_fast(sptep, new_spte); 475c50d8ae3SPaolo Bonzini else 476c50d8ae3SPaolo Bonzini old_spte = __update_clear_spte_slow(sptep, new_spte); 477c50d8ae3SPaolo Bonzini 478c50d8ae3SPaolo Bonzini WARN_ON(spte_to_pfn(old_spte) != spte_to_pfn(new_spte)); 479c50d8ae3SPaolo Bonzini 480c50d8ae3SPaolo Bonzini return old_spte; 481c50d8ae3SPaolo Bonzini } 482c50d8ae3SPaolo Bonzini 483c50d8ae3SPaolo Bonzini /* Rules for using mmu_spte_update: 484c50d8ae3SPaolo Bonzini * Update the state bits, it means the mapped pfn is not changed. 485c50d8ae3SPaolo Bonzini * 48602844ac1SDavid Matlack * Whenever an MMU-writable SPTE is overwritten with a read-only SPTE, remote 48702844ac1SDavid Matlack * TLBs must be flushed. Otherwise rmap_write_protect will find a read-only 48802844ac1SDavid Matlack * spte, even though the writable spte might be cached on a CPU's TLB. 489c50d8ae3SPaolo Bonzini * 490c50d8ae3SPaolo Bonzini * Returns true if the TLB needs to be flushed 491c50d8ae3SPaolo Bonzini */ 492c50d8ae3SPaolo Bonzini static bool mmu_spte_update(u64 *sptep, u64 new_spte) 493c50d8ae3SPaolo Bonzini { 494c50d8ae3SPaolo Bonzini bool flush = false; 495c50d8ae3SPaolo Bonzini u64 old_spte = mmu_spte_update_no_track(sptep, new_spte); 496c50d8ae3SPaolo Bonzini 497c50d8ae3SPaolo Bonzini if (!is_shadow_present_pte(old_spte)) 498c50d8ae3SPaolo Bonzini return false; 499c50d8ae3SPaolo Bonzini 500c50d8ae3SPaolo Bonzini /* 501c50d8ae3SPaolo Bonzini * For the spte updated out of mmu-lock is safe, since 502c50d8ae3SPaolo Bonzini * we always atomically update it, see the comments in 503c50d8ae3SPaolo Bonzini * spte_has_volatile_bits(). 504c50d8ae3SPaolo Bonzini */ 505706c9c55SSean Christopherson if (is_mmu_writable_spte(old_spte) && 506c50d8ae3SPaolo Bonzini !is_writable_pte(new_spte)) 507c50d8ae3SPaolo Bonzini flush = true; 508c50d8ae3SPaolo Bonzini 509c50d8ae3SPaolo Bonzini /* 510c50d8ae3SPaolo Bonzini * Flush TLB when accessed/dirty states are changed in the page tables, 511c50d8ae3SPaolo Bonzini * to guarantee consistency between TLB and page tables. 512c50d8ae3SPaolo Bonzini */ 513c50d8ae3SPaolo Bonzini 514c50d8ae3SPaolo Bonzini if (is_accessed_spte(old_spte) && !is_accessed_spte(new_spte)) { 515c50d8ae3SPaolo Bonzini flush = true; 516c50d8ae3SPaolo Bonzini kvm_set_pfn_accessed(spte_to_pfn(old_spte)); 517c50d8ae3SPaolo Bonzini } 518c50d8ae3SPaolo Bonzini 519c50d8ae3SPaolo Bonzini if (is_dirty_spte(old_spte) && !is_dirty_spte(new_spte)) { 520c50d8ae3SPaolo Bonzini flush = true; 521c50d8ae3SPaolo Bonzini kvm_set_pfn_dirty(spte_to_pfn(old_spte)); 522c50d8ae3SPaolo Bonzini } 523c50d8ae3SPaolo Bonzini 524c50d8ae3SPaolo Bonzini return flush; 525c50d8ae3SPaolo Bonzini } 526c50d8ae3SPaolo Bonzini 527c50d8ae3SPaolo Bonzini /* 528c50d8ae3SPaolo Bonzini * Rules for using mmu_spte_clear_track_bits: 529c50d8ae3SPaolo Bonzini * It sets the sptep from present to nonpresent, and track the 530c50d8ae3SPaolo Bonzini * state bits, it is used to clear the last level sptep. 5317fa2a347SSean Christopherson * Returns the old PTE. 532c50d8ae3SPaolo Bonzini */ 53335d539c3SSean Christopherson static u64 mmu_spte_clear_track_bits(struct kvm *kvm, u64 *sptep) 534c50d8ae3SPaolo Bonzini { 535c50d8ae3SPaolo Bonzini kvm_pfn_t pfn; 536c50d8ae3SPaolo Bonzini u64 old_spte = *sptep; 53771f51d2cSMingwei Zhang int level = sptep_to_sp(sptep)->role.level; 538b14b2690SSean Christopherson struct page *page; 539c50d8ae3SPaolo Bonzini 54054eb3ef5SSean Christopherson if (!is_shadow_present_pte(old_spte) || 54154eb3ef5SSean Christopherson !spte_has_volatile_bits(old_spte)) 542c50d8ae3SPaolo Bonzini __update_clear_spte_fast(sptep, 0ull); 543c50d8ae3SPaolo Bonzini else 544c50d8ae3SPaolo Bonzini old_spte = __update_clear_spte_slow(sptep, 0ull); 545c50d8ae3SPaolo Bonzini 546c50d8ae3SPaolo Bonzini if (!is_shadow_present_pte(old_spte)) 5477fa2a347SSean Christopherson return old_spte; 548c50d8ae3SPaolo Bonzini 54971f51d2cSMingwei Zhang kvm_update_page_stats(kvm, level, -1); 55071f51d2cSMingwei Zhang 551c50d8ae3SPaolo Bonzini pfn = spte_to_pfn(old_spte); 552c50d8ae3SPaolo Bonzini 553c50d8ae3SPaolo Bonzini /* 554b14b2690SSean Christopherson * KVM doesn't hold a reference to any pages mapped into the guest, and 555b14b2690SSean Christopherson * instead uses the mmu_notifier to ensure that KVM unmaps any pages 556b14b2690SSean Christopherson * before they are reclaimed. Sanity check that, if the pfn is backed 557b14b2690SSean Christopherson * by a refcounted page, the refcount is elevated. 558c50d8ae3SPaolo Bonzini */ 559b14b2690SSean Christopherson page = kvm_pfn_to_refcounted_page(pfn); 560b14b2690SSean Christopherson WARN_ON(page && !page_count(page)); 561c50d8ae3SPaolo Bonzini 562c50d8ae3SPaolo Bonzini if (is_accessed_spte(old_spte)) 563c50d8ae3SPaolo Bonzini kvm_set_pfn_accessed(pfn); 564c50d8ae3SPaolo Bonzini 565c50d8ae3SPaolo Bonzini if (is_dirty_spte(old_spte)) 566c50d8ae3SPaolo Bonzini kvm_set_pfn_dirty(pfn); 567c50d8ae3SPaolo Bonzini 5687fa2a347SSean Christopherson return old_spte; 569c50d8ae3SPaolo Bonzini } 570c50d8ae3SPaolo Bonzini 571c50d8ae3SPaolo Bonzini /* 572c50d8ae3SPaolo Bonzini * Rules for using mmu_spte_clear_no_track: 573c50d8ae3SPaolo Bonzini * Directly clear spte without caring the state bits of sptep, 574c50d8ae3SPaolo Bonzini * it is used to set the upper level spte. 575c50d8ae3SPaolo Bonzini */ 576c50d8ae3SPaolo Bonzini static void mmu_spte_clear_no_track(u64 *sptep) 577c50d8ae3SPaolo Bonzini { 578c50d8ae3SPaolo Bonzini __update_clear_spte_fast(sptep, 0ull); 579c50d8ae3SPaolo Bonzini } 580c50d8ae3SPaolo Bonzini 581c50d8ae3SPaolo Bonzini static u64 mmu_spte_get_lockless(u64 *sptep) 582c50d8ae3SPaolo Bonzini { 583c50d8ae3SPaolo Bonzini return __get_spte_lockless(sptep); 584c50d8ae3SPaolo Bonzini } 585c50d8ae3SPaolo Bonzini 586c50d8ae3SPaolo Bonzini /* Returns the Accessed status of the PTE and resets it at the same time. */ 587c50d8ae3SPaolo Bonzini static bool mmu_spte_age(u64 *sptep) 588c50d8ae3SPaolo Bonzini { 589c50d8ae3SPaolo Bonzini u64 spte = mmu_spte_get_lockless(sptep); 590c50d8ae3SPaolo Bonzini 591c50d8ae3SPaolo Bonzini if (!is_accessed_spte(spte)) 592c50d8ae3SPaolo Bonzini return false; 593c50d8ae3SPaolo Bonzini 594c50d8ae3SPaolo Bonzini if (spte_ad_enabled(spte)) { 595c50d8ae3SPaolo Bonzini clear_bit((ffs(shadow_accessed_mask) - 1), 596c50d8ae3SPaolo Bonzini (unsigned long *)sptep); 597c50d8ae3SPaolo Bonzini } else { 598c50d8ae3SPaolo Bonzini /* 599c50d8ae3SPaolo Bonzini * Capture the dirty status of the page, so that it doesn't get 600c50d8ae3SPaolo Bonzini * lost when the SPTE is marked for access tracking. 601c50d8ae3SPaolo Bonzini */ 602c50d8ae3SPaolo Bonzini if (is_writable_pte(spte)) 603c50d8ae3SPaolo Bonzini kvm_set_pfn_dirty(spte_to_pfn(spte)); 604c50d8ae3SPaolo Bonzini 605c50d8ae3SPaolo Bonzini spte = mark_spte_for_access_track(spte); 606c50d8ae3SPaolo Bonzini mmu_spte_update_no_track(sptep, spte); 607c50d8ae3SPaolo Bonzini } 608c50d8ae3SPaolo Bonzini 609c50d8ae3SPaolo Bonzini return true; 610c50d8ae3SPaolo Bonzini } 611c50d8ae3SPaolo Bonzini 612c50d8ae3SPaolo Bonzini static void walk_shadow_page_lockless_begin(struct kvm_vcpu *vcpu) 613c50d8ae3SPaolo Bonzini { 614c5c8c7c5SDavid Matlack if (is_tdp_mmu(vcpu->arch.mmu)) { 615c5c8c7c5SDavid Matlack kvm_tdp_mmu_walk_lockless_begin(); 616c5c8c7c5SDavid Matlack } else { 617c50d8ae3SPaolo Bonzini /* 618c50d8ae3SPaolo Bonzini * Prevent page table teardown by making any free-er wait during 619c50d8ae3SPaolo Bonzini * kvm_flush_remote_tlbs() IPI to all active vcpus. 620c50d8ae3SPaolo Bonzini */ 621c50d8ae3SPaolo Bonzini local_irq_disable(); 622c50d8ae3SPaolo Bonzini 623c50d8ae3SPaolo Bonzini /* 624c50d8ae3SPaolo Bonzini * Make sure a following spte read is not reordered ahead of the write 625c50d8ae3SPaolo Bonzini * to vcpu->mode. 626c50d8ae3SPaolo Bonzini */ 627c50d8ae3SPaolo Bonzini smp_store_mb(vcpu->mode, READING_SHADOW_PAGE_TABLES); 628c50d8ae3SPaolo Bonzini } 629c5c8c7c5SDavid Matlack } 630c50d8ae3SPaolo Bonzini 631c50d8ae3SPaolo Bonzini static void walk_shadow_page_lockless_end(struct kvm_vcpu *vcpu) 632c50d8ae3SPaolo Bonzini { 633c5c8c7c5SDavid Matlack if (is_tdp_mmu(vcpu->arch.mmu)) { 634c5c8c7c5SDavid Matlack kvm_tdp_mmu_walk_lockless_end(); 635c5c8c7c5SDavid Matlack } else { 636c50d8ae3SPaolo Bonzini /* 637c50d8ae3SPaolo Bonzini * Make sure the write to vcpu->mode is not reordered in front of 638c50d8ae3SPaolo Bonzini * reads to sptes. If it does, kvm_mmu_commit_zap_page() can see us 639c50d8ae3SPaolo Bonzini * OUTSIDE_GUEST_MODE and proceed to free the shadow page table. 640c50d8ae3SPaolo Bonzini */ 641c50d8ae3SPaolo Bonzini smp_store_release(&vcpu->mode, OUTSIDE_GUEST_MODE); 642c50d8ae3SPaolo Bonzini local_irq_enable(); 643c50d8ae3SPaolo Bonzini } 644c5c8c7c5SDavid Matlack } 645c50d8ae3SPaolo Bonzini 646378f5cd6SSean Christopherson static int mmu_topup_memory_caches(struct kvm_vcpu *vcpu, bool maybe_indirect) 647c50d8ae3SPaolo Bonzini { 648c50d8ae3SPaolo Bonzini int r; 649c50d8ae3SPaolo Bonzini 650531281adSSean Christopherson /* 1 rmap, 1 parent PTE per level, and the prefetched rmaps. */ 65194ce87efSSean Christopherson r = kvm_mmu_topup_memory_cache(&vcpu->arch.mmu_pte_list_desc_cache, 652531281adSSean Christopherson 1 + PT64_ROOT_MAX_LEVEL + PTE_PREFETCH_NUM); 653c50d8ae3SPaolo Bonzini if (r) 654c50d8ae3SPaolo Bonzini return r; 65594ce87efSSean Christopherson r = kvm_mmu_topup_memory_cache(&vcpu->arch.mmu_shadow_page_cache, 656171a90d7SSean Christopherson PT64_ROOT_MAX_LEVEL); 657171a90d7SSean Christopherson if (r) 658171a90d7SSean Christopherson return r; 659378f5cd6SSean Christopherson if (maybe_indirect) { 6606a97575dSDavid Matlack r = kvm_mmu_topup_memory_cache(&vcpu->arch.mmu_shadowed_info_cache, 661171a90d7SSean Christopherson PT64_ROOT_MAX_LEVEL); 662c50d8ae3SPaolo Bonzini if (r) 663c50d8ae3SPaolo Bonzini return r; 664378f5cd6SSean Christopherson } 66594ce87efSSean Christopherson return kvm_mmu_topup_memory_cache(&vcpu->arch.mmu_page_header_cache, 666531281adSSean Christopherson PT64_ROOT_MAX_LEVEL); 667c50d8ae3SPaolo Bonzini } 668c50d8ae3SPaolo Bonzini 669c50d8ae3SPaolo Bonzini static void mmu_free_memory_caches(struct kvm_vcpu *vcpu) 670c50d8ae3SPaolo Bonzini { 67194ce87efSSean Christopherson kvm_mmu_free_memory_cache(&vcpu->arch.mmu_pte_list_desc_cache); 67294ce87efSSean Christopherson kvm_mmu_free_memory_cache(&vcpu->arch.mmu_shadow_page_cache); 6736a97575dSDavid Matlack kvm_mmu_free_memory_cache(&vcpu->arch.mmu_shadowed_info_cache); 67494ce87efSSean Christopherson kvm_mmu_free_memory_cache(&vcpu->arch.mmu_page_header_cache); 675c50d8ae3SPaolo Bonzini } 676c50d8ae3SPaolo Bonzini 677c50d8ae3SPaolo Bonzini static void mmu_free_pte_list_desc(struct pte_list_desc *pte_list_desc) 678c50d8ae3SPaolo Bonzini { 679c50d8ae3SPaolo Bonzini kmem_cache_free(pte_list_desc_cache, pte_list_desc); 680c50d8ae3SPaolo Bonzini } 681c50d8ae3SPaolo Bonzini 6826a97575dSDavid Matlack static bool sp_has_gptes(struct kvm_mmu_page *sp); 6836a97575dSDavid Matlack 684c50d8ae3SPaolo Bonzini static gfn_t kvm_mmu_page_get_gfn(struct kvm_mmu_page *sp, int index) 685c50d8ae3SPaolo Bonzini { 68684e5ffd0SLai Jiangshan if (sp->role.passthrough) 68784e5ffd0SLai Jiangshan return sp->gfn; 68884e5ffd0SLai Jiangshan 689c50d8ae3SPaolo Bonzini if (!sp->role.direct) 6906a97575dSDavid Matlack return sp->shadowed_translation[index] >> PAGE_SHIFT; 691c50d8ae3SPaolo Bonzini 6922ca3129eSSean Christopherson return sp->gfn + (index << ((sp->role.level - 1) * SPTE_LEVEL_BITS)); 693c50d8ae3SPaolo Bonzini } 694c50d8ae3SPaolo Bonzini 6956a97575dSDavid Matlack /* 6966a97575dSDavid Matlack * For leaf SPTEs, fetch the *guest* access permissions being shadowed. Note 6976a97575dSDavid Matlack * that the SPTE itself may have a more constrained access permissions that 6986a97575dSDavid Matlack * what the guest enforces. For example, a guest may create an executable 6996a97575dSDavid Matlack * huge PTE but KVM may disallow execution to mitigate iTLB multihit. 7006a97575dSDavid Matlack */ 7016a97575dSDavid Matlack static u32 kvm_mmu_page_get_access(struct kvm_mmu_page *sp, int index) 702c50d8ae3SPaolo Bonzini { 7036a97575dSDavid Matlack if (sp_has_gptes(sp)) 7046a97575dSDavid Matlack return sp->shadowed_translation[index] & ACC_ALL; 7056a97575dSDavid Matlack 7066a97575dSDavid Matlack /* 7076a97575dSDavid Matlack * For direct MMUs (e.g. TDP or non-paging guests) or passthrough SPs, 7086a97575dSDavid Matlack * KVM is not shadowing any guest page tables, so the "guest access 7096a97575dSDavid Matlack * permissions" are just ACC_ALL. 7106a97575dSDavid Matlack * 7116a97575dSDavid Matlack * For direct SPs in indirect MMUs (shadow paging), i.e. when KVM 7126a97575dSDavid Matlack * is shadowing a guest huge page with small pages, the guest access 7136a97575dSDavid Matlack * permissions being shadowed are the access permissions of the huge 7146a97575dSDavid Matlack * page. 7156a97575dSDavid Matlack * 7166a97575dSDavid Matlack * In both cases, sp->role.access contains the correct access bits. 7176a97575dSDavid Matlack */ 7186a97575dSDavid Matlack return sp->role.access; 7196a97575dSDavid Matlack } 7206a97575dSDavid Matlack 72172ae5822SSean Christopherson static void kvm_mmu_page_set_translation(struct kvm_mmu_page *sp, int index, 72272ae5822SSean Christopherson gfn_t gfn, unsigned int access) 7236a97575dSDavid Matlack { 7246a97575dSDavid Matlack if (sp_has_gptes(sp)) { 7256a97575dSDavid Matlack sp->shadowed_translation[index] = (gfn << PAGE_SHIFT) | access; 72684e5ffd0SLai Jiangshan return; 72784e5ffd0SLai Jiangshan } 72884e5ffd0SLai Jiangshan 7296a97575dSDavid Matlack WARN_ONCE(access != kvm_mmu_page_get_access(sp, index), 7306a97575dSDavid Matlack "access mismatch under %s page %llx (expected %u, got %u)\n", 7316a97575dSDavid Matlack sp->role.passthrough ? "passthrough" : "direct", 7326a97575dSDavid Matlack sp->gfn, kvm_mmu_page_get_access(sp, index), access); 7336a97575dSDavid Matlack 7346a97575dSDavid Matlack WARN_ONCE(gfn != kvm_mmu_page_get_gfn(sp, index), 7356a97575dSDavid Matlack "gfn mismatch under %s page %llx (expected %llx, got %llx)\n", 7366a97575dSDavid Matlack sp->role.passthrough ? "passthrough" : "direct", 7376a97575dSDavid Matlack sp->gfn, kvm_mmu_page_get_gfn(sp, index), gfn); 738c50d8ae3SPaolo Bonzini } 739c50d8ae3SPaolo Bonzini 74072ae5822SSean Christopherson static void kvm_mmu_page_set_access(struct kvm_mmu_page *sp, int index, 74172ae5822SSean Christopherson unsigned int access) 7426a97575dSDavid Matlack { 7436a97575dSDavid Matlack gfn_t gfn = kvm_mmu_page_get_gfn(sp, index); 7446a97575dSDavid Matlack 7456a97575dSDavid Matlack kvm_mmu_page_set_translation(sp, index, gfn, access); 746c50d8ae3SPaolo Bonzini } 747c50d8ae3SPaolo Bonzini 748c50d8ae3SPaolo Bonzini /* 749c50d8ae3SPaolo Bonzini * Return the pointer to the large page information for a given gfn, 750c50d8ae3SPaolo Bonzini * handling slots that are not large page aligned. 751c50d8ae3SPaolo Bonzini */ 752c50d8ae3SPaolo Bonzini static struct kvm_lpage_info *lpage_info_slot(gfn_t gfn, 7538ca6f063SBen Gardon const struct kvm_memory_slot *slot, int level) 754c50d8ae3SPaolo Bonzini { 755c50d8ae3SPaolo Bonzini unsigned long idx; 756c50d8ae3SPaolo Bonzini 757c50d8ae3SPaolo Bonzini idx = gfn_to_index(gfn, slot->base_gfn, level); 758c50d8ae3SPaolo Bonzini return &slot->arch.lpage_info[level - 2][idx]; 759c50d8ae3SPaolo Bonzini } 760c50d8ae3SPaolo Bonzini 761269e9552SHamza Mahfooz static void update_gfn_disallow_lpage_count(const struct kvm_memory_slot *slot, 762c50d8ae3SPaolo Bonzini gfn_t gfn, int count) 763c50d8ae3SPaolo Bonzini { 764c50d8ae3SPaolo Bonzini struct kvm_lpage_info *linfo; 765c50d8ae3SPaolo Bonzini int i; 766c50d8ae3SPaolo Bonzini 7673bae0459SSean Christopherson for (i = PG_LEVEL_2M; i <= KVM_MAX_HUGEPAGE_LEVEL; ++i) { 768c50d8ae3SPaolo Bonzini linfo = lpage_info_slot(gfn, slot, i); 769c50d8ae3SPaolo Bonzini linfo->disallow_lpage += count; 770c50d8ae3SPaolo Bonzini WARN_ON(linfo->disallow_lpage < 0); 771c50d8ae3SPaolo Bonzini } 772c50d8ae3SPaolo Bonzini } 773c50d8ae3SPaolo Bonzini 774269e9552SHamza Mahfooz void kvm_mmu_gfn_disallow_lpage(const struct kvm_memory_slot *slot, gfn_t gfn) 775c50d8ae3SPaolo Bonzini { 776c50d8ae3SPaolo Bonzini update_gfn_disallow_lpage_count(slot, gfn, 1); 777c50d8ae3SPaolo Bonzini } 778c50d8ae3SPaolo Bonzini 779269e9552SHamza Mahfooz void kvm_mmu_gfn_allow_lpage(const struct kvm_memory_slot *slot, gfn_t gfn) 780c50d8ae3SPaolo Bonzini { 781c50d8ae3SPaolo Bonzini update_gfn_disallow_lpage_count(slot, gfn, -1); 782c50d8ae3SPaolo Bonzini } 783c50d8ae3SPaolo Bonzini 784c50d8ae3SPaolo Bonzini static void account_shadowed(struct kvm *kvm, struct kvm_mmu_page *sp) 785c50d8ae3SPaolo Bonzini { 786c50d8ae3SPaolo Bonzini struct kvm_memslots *slots; 787c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot; 788c50d8ae3SPaolo Bonzini gfn_t gfn; 789c50d8ae3SPaolo Bonzini 790c50d8ae3SPaolo Bonzini kvm->arch.indirect_shadow_pages++; 791c50d8ae3SPaolo Bonzini gfn = sp->gfn; 792c50d8ae3SPaolo Bonzini slots = kvm_memslots_for_spte_role(kvm, sp->role); 793c50d8ae3SPaolo Bonzini slot = __gfn_to_memslot(slots, gfn); 794c50d8ae3SPaolo Bonzini 795c50d8ae3SPaolo Bonzini /* the non-leaf shadow pages are keeping readonly. */ 7963bae0459SSean Christopherson if (sp->role.level > PG_LEVEL_4K) 797c50d8ae3SPaolo Bonzini return kvm_slot_page_track_add_page(kvm, slot, gfn, 798c50d8ae3SPaolo Bonzini KVM_PAGE_TRACK_WRITE); 799c50d8ae3SPaolo Bonzini 800c50d8ae3SPaolo Bonzini kvm_mmu_gfn_disallow_lpage(slot, gfn); 801be911771SDavid Matlack 802be911771SDavid Matlack if (kvm_mmu_slot_gfn_write_protect(kvm, slot, gfn, PG_LEVEL_4K)) 803be911771SDavid Matlack kvm_flush_remote_tlbs_with_address(kvm, gfn, 1); 804c50d8ae3SPaolo Bonzini } 805c50d8ae3SPaolo Bonzini 80661f94478SSean Christopherson void track_possible_nx_huge_page(struct kvm *kvm, struct kvm_mmu_page *sp) 807c50d8ae3SPaolo Bonzini { 808428e9216SSean Christopherson /* 809428e9216SSean Christopherson * If it's possible to replace the shadow page with an NX huge page, 810428e9216SSean Christopherson * i.e. if the shadow page is the only thing currently preventing KVM 811428e9216SSean Christopherson * from using a huge page, add the shadow page to the list of "to be 812428e9216SSean Christopherson * zapped for NX recovery" pages. Note, the shadow page can already be 813428e9216SSean Christopherson * on the list if KVM is reusing an existing shadow page, i.e. if KVM 814428e9216SSean Christopherson * links a shadow page at multiple points. 815428e9216SSean Christopherson */ 81661f94478SSean Christopherson if (!list_empty(&sp->possible_nx_huge_page_link)) 817c50d8ae3SPaolo Bonzini return; 818c50d8ae3SPaolo Bonzini 819c50d8ae3SPaolo Bonzini ++kvm->stat.nx_lpage_splits; 82055c510e2SSean Christopherson list_add_tail(&sp->possible_nx_huge_page_link, 82155c510e2SSean Christopherson &kvm->arch.possible_nx_huge_pages); 822c50d8ae3SPaolo Bonzini } 823c50d8ae3SPaolo Bonzini 82461f94478SSean Christopherson static void account_nx_huge_page(struct kvm *kvm, struct kvm_mmu_page *sp, 82561f94478SSean Christopherson bool nx_huge_page_possible) 82661f94478SSean Christopherson { 82761f94478SSean Christopherson sp->nx_huge_page_disallowed = true; 82861f94478SSean Christopherson 82961f94478SSean Christopherson if (nx_huge_page_possible) 83061f94478SSean Christopherson track_possible_nx_huge_page(kvm, sp); 83161f94478SSean Christopherson } 83261f94478SSean Christopherson 833c50d8ae3SPaolo Bonzini static void unaccount_shadowed(struct kvm *kvm, struct kvm_mmu_page *sp) 834c50d8ae3SPaolo Bonzini { 835c50d8ae3SPaolo Bonzini struct kvm_memslots *slots; 836c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot; 837c50d8ae3SPaolo Bonzini gfn_t gfn; 838c50d8ae3SPaolo Bonzini 839c50d8ae3SPaolo Bonzini kvm->arch.indirect_shadow_pages--; 840c50d8ae3SPaolo Bonzini gfn = sp->gfn; 841c50d8ae3SPaolo Bonzini slots = kvm_memslots_for_spte_role(kvm, sp->role); 842c50d8ae3SPaolo Bonzini slot = __gfn_to_memslot(slots, gfn); 8433bae0459SSean Christopherson if (sp->role.level > PG_LEVEL_4K) 844c50d8ae3SPaolo Bonzini return kvm_slot_page_track_remove_page(kvm, slot, gfn, 845c50d8ae3SPaolo Bonzini KVM_PAGE_TRACK_WRITE); 846c50d8ae3SPaolo Bonzini 847c50d8ae3SPaolo Bonzini kvm_mmu_gfn_allow_lpage(slot, gfn); 848c50d8ae3SPaolo Bonzini } 849c50d8ae3SPaolo Bonzini 85061f94478SSean Christopherson void untrack_possible_nx_huge_page(struct kvm *kvm, struct kvm_mmu_page *sp) 851c50d8ae3SPaolo Bonzini { 85255c510e2SSean Christopherson if (list_empty(&sp->possible_nx_huge_page_link)) 853428e9216SSean Christopherson return; 854428e9216SSean Christopherson 855428e9216SSean Christopherson --kvm->stat.nx_lpage_splits; 85655c510e2SSean Christopherson list_del_init(&sp->possible_nx_huge_page_link); 857c50d8ae3SPaolo Bonzini } 858c50d8ae3SPaolo Bonzini 85961f94478SSean Christopherson static void unaccount_nx_huge_page(struct kvm *kvm, struct kvm_mmu_page *sp) 86061f94478SSean Christopherson { 86161f94478SSean Christopherson sp->nx_huge_page_disallowed = false; 86261f94478SSean Christopherson 86361f94478SSean Christopherson untrack_possible_nx_huge_page(kvm, sp); 86461f94478SSean Christopherson } 86561f94478SSean Christopherson 866c50d8ae3SPaolo Bonzini static struct kvm_memory_slot * 867c50d8ae3SPaolo Bonzini gfn_to_memslot_dirty_bitmap(struct kvm_vcpu *vcpu, gfn_t gfn, 868c50d8ae3SPaolo Bonzini bool no_dirty_log) 869c50d8ae3SPaolo Bonzini { 870c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot; 871c50d8ae3SPaolo Bonzini 872c50d8ae3SPaolo Bonzini slot = kvm_vcpu_gfn_to_memslot(vcpu, gfn); 87391b0d268SPaolo Bonzini if (!slot || slot->flags & KVM_MEMSLOT_INVALID) 87491b0d268SPaolo Bonzini return NULL; 875044c59c4SPeter Xu if (no_dirty_log && kvm_slot_dirty_track_enabled(slot)) 87691b0d268SPaolo Bonzini return NULL; 877c50d8ae3SPaolo Bonzini 878c50d8ae3SPaolo Bonzini return slot; 879c50d8ae3SPaolo Bonzini } 880c50d8ae3SPaolo Bonzini 881c50d8ae3SPaolo Bonzini /* 882c50d8ae3SPaolo Bonzini * About rmap_head encoding: 883c50d8ae3SPaolo Bonzini * 884c50d8ae3SPaolo Bonzini * If the bit zero of rmap_head->val is clear, then it points to the only spte 885c50d8ae3SPaolo Bonzini * in this rmap chain. Otherwise, (rmap_head->val & ~1) points to a struct 886c50d8ae3SPaolo Bonzini * pte_list_desc containing more mappings. 887c50d8ae3SPaolo Bonzini */ 888c50d8ae3SPaolo Bonzini 889c50d8ae3SPaolo Bonzini /* 890c50d8ae3SPaolo Bonzini * Returns the number of pointers in the rmap chain, not counting the new one. 891c50d8ae3SPaolo Bonzini */ 8922ff9039aSDavid Matlack static int pte_list_add(struct kvm_mmu_memory_cache *cache, u64 *spte, 893c50d8ae3SPaolo Bonzini struct kvm_rmap_head *rmap_head) 894c50d8ae3SPaolo Bonzini { 895c50d8ae3SPaolo Bonzini struct pte_list_desc *desc; 89613236e25SPeter Xu int count = 0; 897c50d8ae3SPaolo Bonzini 898c50d8ae3SPaolo Bonzini if (!rmap_head->val) { 899805a0f83SStephen Zhang rmap_printk("%p %llx 0->1\n", spte, *spte); 900c50d8ae3SPaolo Bonzini rmap_head->val = (unsigned long)spte; 901c50d8ae3SPaolo Bonzini } else if (!(rmap_head->val & 1)) { 902805a0f83SStephen Zhang rmap_printk("%p %llx 1->many\n", spte, *spte); 9032ff9039aSDavid Matlack desc = kvm_mmu_memory_cache_alloc(cache); 904c50d8ae3SPaolo Bonzini desc->sptes[0] = (u64 *)rmap_head->val; 905c50d8ae3SPaolo Bonzini desc->sptes[1] = spte; 90613236e25SPeter Xu desc->spte_count = 2; 907c50d8ae3SPaolo Bonzini rmap_head->val = (unsigned long)desc | 1; 908c50d8ae3SPaolo Bonzini ++count; 909c50d8ae3SPaolo Bonzini } else { 910805a0f83SStephen Zhang rmap_printk("%p %llx many->many\n", spte, *spte); 911c50d8ae3SPaolo Bonzini desc = (struct pte_list_desc *)(rmap_head->val & ~1ul); 91213236e25SPeter Xu while (desc->spte_count == PTE_LIST_EXT) { 913c50d8ae3SPaolo Bonzini count += PTE_LIST_EXT; 914c6c4f961SLi RongQing if (!desc->more) { 9152ff9039aSDavid Matlack desc->more = kvm_mmu_memory_cache_alloc(cache); 916c50d8ae3SPaolo Bonzini desc = desc->more; 91713236e25SPeter Xu desc->spte_count = 0; 918c6c4f961SLi RongQing break; 919c6c4f961SLi RongQing } 920c6c4f961SLi RongQing desc = desc->more; 921c50d8ae3SPaolo Bonzini } 92213236e25SPeter Xu count += desc->spte_count; 92313236e25SPeter Xu desc->sptes[desc->spte_count++] = spte; 924c50d8ae3SPaolo Bonzini } 925c50d8ae3SPaolo Bonzini return count; 926c50d8ae3SPaolo Bonzini } 927c50d8ae3SPaolo Bonzini 928c50d8ae3SPaolo Bonzini static void 929c50d8ae3SPaolo Bonzini pte_list_desc_remove_entry(struct kvm_rmap_head *rmap_head, 930c50d8ae3SPaolo Bonzini struct pte_list_desc *desc, int i, 931c50d8ae3SPaolo Bonzini struct pte_list_desc *prev_desc) 932c50d8ae3SPaolo Bonzini { 93313236e25SPeter Xu int j = desc->spte_count - 1; 934c50d8ae3SPaolo Bonzini 935c50d8ae3SPaolo Bonzini desc->sptes[i] = desc->sptes[j]; 936c50d8ae3SPaolo Bonzini desc->sptes[j] = NULL; 93713236e25SPeter Xu desc->spte_count--; 93813236e25SPeter Xu if (desc->spte_count) 939c50d8ae3SPaolo Bonzini return; 940c50d8ae3SPaolo Bonzini if (!prev_desc && !desc->more) 941fe3c2b4cSMiaohe Lin rmap_head->val = 0; 942c50d8ae3SPaolo Bonzini else 943c50d8ae3SPaolo Bonzini if (prev_desc) 944c50d8ae3SPaolo Bonzini prev_desc->more = desc->more; 945c50d8ae3SPaolo Bonzini else 946c50d8ae3SPaolo Bonzini rmap_head->val = (unsigned long)desc->more | 1; 947c50d8ae3SPaolo Bonzini mmu_free_pte_list_desc(desc); 948c50d8ae3SPaolo Bonzini } 949c50d8ae3SPaolo Bonzini 9503c2e1037SSean Christopherson static void pte_list_remove(u64 *spte, struct kvm_rmap_head *rmap_head) 951c50d8ae3SPaolo Bonzini { 952c50d8ae3SPaolo Bonzini struct pte_list_desc *desc; 953c50d8ae3SPaolo Bonzini struct pte_list_desc *prev_desc; 954c50d8ae3SPaolo Bonzini int i; 955c50d8ae3SPaolo Bonzini 956c50d8ae3SPaolo Bonzini if (!rmap_head->val) { 957c50d8ae3SPaolo Bonzini pr_err("%s: %p 0->BUG\n", __func__, spte); 958c50d8ae3SPaolo Bonzini BUG(); 959c50d8ae3SPaolo Bonzini } else if (!(rmap_head->val & 1)) { 960805a0f83SStephen Zhang rmap_printk("%p 1->0\n", spte); 961c50d8ae3SPaolo Bonzini if ((u64 *)rmap_head->val != spte) { 962c50d8ae3SPaolo Bonzini pr_err("%s: %p 1->BUG\n", __func__, spte); 963c50d8ae3SPaolo Bonzini BUG(); 964c50d8ae3SPaolo Bonzini } 965c50d8ae3SPaolo Bonzini rmap_head->val = 0; 966c50d8ae3SPaolo Bonzini } else { 967805a0f83SStephen Zhang rmap_printk("%p many->many\n", spte); 968c50d8ae3SPaolo Bonzini desc = (struct pte_list_desc *)(rmap_head->val & ~1ul); 969c50d8ae3SPaolo Bonzini prev_desc = NULL; 970c50d8ae3SPaolo Bonzini while (desc) { 97113236e25SPeter Xu for (i = 0; i < desc->spte_count; ++i) { 972c50d8ae3SPaolo Bonzini if (desc->sptes[i] == spte) { 973c50d8ae3SPaolo Bonzini pte_list_desc_remove_entry(rmap_head, 974c50d8ae3SPaolo Bonzini desc, i, prev_desc); 975c50d8ae3SPaolo Bonzini return; 976c50d8ae3SPaolo Bonzini } 977c50d8ae3SPaolo Bonzini } 978c50d8ae3SPaolo Bonzini prev_desc = desc; 979c50d8ae3SPaolo Bonzini desc = desc->more; 980c50d8ae3SPaolo Bonzini } 981c50d8ae3SPaolo Bonzini pr_err("%s: %p many->many\n", __func__, spte); 982c50d8ae3SPaolo Bonzini BUG(); 983c50d8ae3SPaolo Bonzini } 984c50d8ae3SPaolo Bonzini } 985c50d8ae3SPaolo Bonzini 9869202aee8SSean Christopherson static void kvm_zap_one_rmap_spte(struct kvm *kvm, 9879202aee8SSean Christopherson struct kvm_rmap_head *rmap_head, u64 *sptep) 988c50d8ae3SPaolo Bonzini { 98971f51d2cSMingwei Zhang mmu_spte_clear_track_bits(kvm, sptep); 9903c2e1037SSean Christopherson pte_list_remove(sptep, rmap_head); 991c50d8ae3SPaolo Bonzini } 992c50d8ae3SPaolo Bonzini 9939202aee8SSean Christopherson /* Return true if at least one SPTE was zapped, false otherwise */ 9949202aee8SSean Christopherson static bool kvm_zap_all_rmap_sptes(struct kvm *kvm, 9959202aee8SSean Christopherson struct kvm_rmap_head *rmap_head) 996a75b5404SPeter Xu { 997a75b5404SPeter Xu struct pte_list_desc *desc, *next; 998a75b5404SPeter Xu int i; 999a75b5404SPeter Xu 1000a75b5404SPeter Xu if (!rmap_head->val) 1001a75b5404SPeter Xu return false; 1002a75b5404SPeter Xu 1003a75b5404SPeter Xu if (!(rmap_head->val & 1)) { 100471f51d2cSMingwei Zhang mmu_spte_clear_track_bits(kvm, (u64 *)rmap_head->val); 1005a75b5404SPeter Xu goto out; 1006a75b5404SPeter Xu } 1007a75b5404SPeter Xu 1008a75b5404SPeter Xu desc = (struct pte_list_desc *)(rmap_head->val & ~1ul); 1009a75b5404SPeter Xu 1010a75b5404SPeter Xu for (; desc; desc = next) { 1011a75b5404SPeter Xu for (i = 0; i < desc->spte_count; i++) 101271f51d2cSMingwei Zhang mmu_spte_clear_track_bits(kvm, desc->sptes[i]); 1013a75b5404SPeter Xu next = desc->more; 1014a75b5404SPeter Xu mmu_free_pte_list_desc(desc); 1015a75b5404SPeter Xu } 1016a75b5404SPeter Xu out: 1017a75b5404SPeter Xu /* rmap_head is meaningless now, remember to reset it */ 1018a75b5404SPeter Xu rmap_head->val = 0; 1019a75b5404SPeter Xu return true; 1020a75b5404SPeter Xu } 1021a75b5404SPeter Xu 10223bcd0662SPeter Xu unsigned int pte_list_count(struct kvm_rmap_head *rmap_head) 10233bcd0662SPeter Xu { 10243bcd0662SPeter Xu struct pte_list_desc *desc; 10253bcd0662SPeter Xu unsigned int count = 0; 10263bcd0662SPeter Xu 10273bcd0662SPeter Xu if (!rmap_head->val) 10283bcd0662SPeter Xu return 0; 10293bcd0662SPeter Xu else if (!(rmap_head->val & 1)) 10303bcd0662SPeter Xu return 1; 10313bcd0662SPeter Xu 10323bcd0662SPeter Xu desc = (struct pte_list_desc *)(rmap_head->val & ~1ul); 10333bcd0662SPeter Xu 10343bcd0662SPeter Xu while (desc) { 10353bcd0662SPeter Xu count += desc->spte_count; 10363bcd0662SPeter Xu desc = desc->more; 10373bcd0662SPeter Xu } 10383bcd0662SPeter Xu 10393bcd0662SPeter Xu return count; 10403bcd0662SPeter Xu } 10413bcd0662SPeter Xu 104293e083d4SDavid Matlack static struct kvm_rmap_head *gfn_to_rmap(gfn_t gfn, int level, 1043269e9552SHamza Mahfooz const struct kvm_memory_slot *slot) 1044c50d8ae3SPaolo Bonzini { 1045c50d8ae3SPaolo Bonzini unsigned long idx; 1046c50d8ae3SPaolo Bonzini 1047c50d8ae3SPaolo Bonzini idx = gfn_to_index(gfn, slot->base_gfn, level); 10483bae0459SSean Christopherson return &slot->arch.rmap[level - PG_LEVEL_4K][idx]; 1049c50d8ae3SPaolo Bonzini } 1050c50d8ae3SPaolo Bonzini 1051c50d8ae3SPaolo Bonzini static bool rmap_can_add(struct kvm_vcpu *vcpu) 1052c50d8ae3SPaolo Bonzini { 1053356ec69aSSean Christopherson struct kvm_mmu_memory_cache *mc; 1054c50d8ae3SPaolo Bonzini 1055356ec69aSSean Christopherson mc = &vcpu->arch.mmu_pte_list_desc_cache; 105694ce87efSSean Christopherson return kvm_mmu_memory_cache_nr_free_objects(mc); 1057c50d8ae3SPaolo Bonzini } 1058c50d8ae3SPaolo Bonzini 1059c50d8ae3SPaolo Bonzini static void rmap_remove(struct kvm *kvm, u64 *spte) 1060c50d8ae3SPaolo Bonzini { 1061601f8af0SDavid Matlack struct kvm_memslots *slots; 1062601f8af0SDavid Matlack struct kvm_memory_slot *slot; 1063c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 1064c50d8ae3SPaolo Bonzini gfn_t gfn; 1065c50d8ae3SPaolo Bonzini struct kvm_rmap_head *rmap_head; 1066c50d8ae3SPaolo Bonzini 106757354682SSean Christopherson sp = sptep_to_sp(spte); 106879e48cecSSean Christopherson gfn = kvm_mmu_page_get_gfn(sp, spte_index(spte)); 1069601f8af0SDavid Matlack 1070601f8af0SDavid Matlack /* 107168be1306SDavid Matlack * Unlike rmap_add, rmap_remove does not run in the context of a vCPU 107268be1306SDavid Matlack * so we have to determine which memslots to use based on context 107368be1306SDavid Matlack * information in sp->role. 1074601f8af0SDavid Matlack */ 1075601f8af0SDavid Matlack slots = kvm_memslots_for_spte_role(kvm, sp->role); 1076601f8af0SDavid Matlack 1077601f8af0SDavid Matlack slot = __gfn_to_memslot(slots, gfn); 107893e083d4SDavid Matlack rmap_head = gfn_to_rmap(gfn, sp->role.level, slot); 1079601f8af0SDavid Matlack 10803c2e1037SSean Christopherson pte_list_remove(spte, rmap_head); 1081c50d8ae3SPaolo Bonzini } 1082c50d8ae3SPaolo Bonzini 1083c50d8ae3SPaolo Bonzini /* 1084c50d8ae3SPaolo Bonzini * Used by the following functions to iterate through the sptes linked by a 1085c50d8ae3SPaolo Bonzini * rmap. All fields are private and not assumed to be used outside. 1086c50d8ae3SPaolo Bonzini */ 1087c50d8ae3SPaolo Bonzini struct rmap_iterator { 1088c50d8ae3SPaolo Bonzini /* private fields */ 1089c50d8ae3SPaolo Bonzini struct pte_list_desc *desc; /* holds the sptep if not NULL */ 1090c50d8ae3SPaolo Bonzini int pos; /* index of the sptep */ 1091c50d8ae3SPaolo Bonzini }; 1092c50d8ae3SPaolo Bonzini 1093c50d8ae3SPaolo Bonzini /* 1094c50d8ae3SPaolo Bonzini * Iteration must be started by this function. This should also be used after 1095c50d8ae3SPaolo Bonzini * removing/dropping sptes from the rmap link because in such cases the 10960a03cbdaSMiaohe Lin * information in the iterator may not be valid. 1097c50d8ae3SPaolo Bonzini * 1098c50d8ae3SPaolo Bonzini * Returns sptep if found, NULL otherwise. 1099c50d8ae3SPaolo Bonzini */ 1100c50d8ae3SPaolo Bonzini static u64 *rmap_get_first(struct kvm_rmap_head *rmap_head, 1101c50d8ae3SPaolo Bonzini struct rmap_iterator *iter) 1102c50d8ae3SPaolo Bonzini { 1103c50d8ae3SPaolo Bonzini u64 *sptep; 1104c50d8ae3SPaolo Bonzini 1105c50d8ae3SPaolo Bonzini if (!rmap_head->val) 1106c50d8ae3SPaolo Bonzini return NULL; 1107c50d8ae3SPaolo Bonzini 1108c50d8ae3SPaolo Bonzini if (!(rmap_head->val & 1)) { 1109c50d8ae3SPaolo Bonzini iter->desc = NULL; 1110c50d8ae3SPaolo Bonzini sptep = (u64 *)rmap_head->val; 1111c50d8ae3SPaolo Bonzini goto out; 1112c50d8ae3SPaolo Bonzini } 1113c50d8ae3SPaolo Bonzini 1114c50d8ae3SPaolo Bonzini iter->desc = (struct pte_list_desc *)(rmap_head->val & ~1ul); 1115c50d8ae3SPaolo Bonzini iter->pos = 0; 1116c50d8ae3SPaolo Bonzini sptep = iter->desc->sptes[iter->pos]; 1117c50d8ae3SPaolo Bonzini out: 1118c50d8ae3SPaolo Bonzini BUG_ON(!is_shadow_present_pte(*sptep)); 1119c50d8ae3SPaolo Bonzini return sptep; 1120c50d8ae3SPaolo Bonzini } 1121c50d8ae3SPaolo Bonzini 1122c50d8ae3SPaolo Bonzini /* 1123c50d8ae3SPaolo Bonzini * Must be used with a valid iterator: e.g. after rmap_get_first(). 1124c50d8ae3SPaolo Bonzini * 1125c50d8ae3SPaolo Bonzini * Returns sptep if found, NULL otherwise. 1126c50d8ae3SPaolo Bonzini */ 1127c50d8ae3SPaolo Bonzini static u64 *rmap_get_next(struct rmap_iterator *iter) 1128c50d8ae3SPaolo Bonzini { 1129c50d8ae3SPaolo Bonzini u64 *sptep; 1130c50d8ae3SPaolo Bonzini 1131c50d8ae3SPaolo Bonzini if (iter->desc) { 1132c50d8ae3SPaolo Bonzini if (iter->pos < PTE_LIST_EXT - 1) { 1133c50d8ae3SPaolo Bonzini ++iter->pos; 1134c50d8ae3SPaolo Bonzini sptep = iter->desc->sptes[iter->pos]; 1135c50d8ae3SPaolo Bonzini if (sptep) 1136c50d8ae3SPaolo Bonzini goto out; 1137c50d8ae3SPaolo Bonzini } 1138c50d8ae3SPaolo Bonzini 1139c50d8ae3SPaolo Bonzini iter->desc = iter->desc->more; 1140c50d8ae3SPaolo Bonzini 1141c50d8ae3SPaolo Bonzini if (iter->desc) { 1142c50d8ae3SPaolo Bonzini iter->pos = 0; 1143c50d8ae3SPaolo Bonzini /* desc->sptes[0] cannot be NULL */ 1144c50d8ae3SPaolo Bonzini sptep = iter->desc->sptes[iter->pos]; 1145c50d8ae3SPaolo Bonzini goto out; 1146c50d8ae3SPaolo Bonzini } 1147c50d8ae3SPaolo Bonzini } 1148c50d8ae3SPaolo Bonzini 1149c50d8ae3SPaolo Bonzini return NULL; 1150c50d8ae3SPaolo Bonzini out: 1151c50d8ae3SPaolo Bonzini BUG_ON(!is_shadow_present_pte(*sptep)); 1152c50d8ae3SPaolo Bonzini return sptep; 1153c50d8ae3SPaolo Bonzini } 1154c50d8ae3SPaolo Bonzini 1155c50d8ae3SPaolo Bonzini #define for_each_rmap_spte(_rmap_head_, _iter_, _spte_) \ 1156c50d8ae3SPaolo Bonzini for (_spte_ = rmap_get_first(_rmap_head_, _iter_); \ 1157c50d8ae3SPaolo Bonzini _spte_; _spte_ = rmap_get_next(_iter_)) 1158c50d8ae3SPaolo Bonzini 1159c50d8ae3SPaolo Bonzini static void drop_spte(struct kvm *kvm, u64 *sptep) 1160c50d8ae3SPaolo Bonzini { 116171f51d2cSMingwei Zhang u64 old_spte = mmu_spte_clear_track_bits(kvm, sptep); 11627fa2a347SSean Christopherson 11637fa2a347SSean Christopherson if (is_shadow_present_pte(old_spte)) 1164c50d8ae3SPaolo Bonzini rmap_remove(kvm, sptep); 1165c50d8ae3SPaolo Bonzini } 1166c50d8ae3SPaolo Bonzini 116703787394SPaolo Bonzini static void drop_large_spte(struct kvm *kvm, u64 *sptep, bool flush) 1168c50d8ae3SPaolo Bonzini { 11690cd8dc73SPaolo Bonzini struct kvm_mmu_page *sp; 11700cd8dc73SPaolo Bonzini 11710cd8dc73SPaolo Bonzini sp = sptep_to_sp(sptep); 11720cd8dc73SPaolo Bonzini WARN_ON(sp->role.level == PG_LEVEL_4K); 11730cd8dc73SPaolo Bonzini 1174c50d8ae3SPaolo Bonzini drop_spte(kvm, sptep); 117503787394SPaolo Bonzini 117603787394SPaolo Bonzini if (flush) 11770cd8dc73SPaolo Bonzini kvm_flush_remote_tlbs_with_address(kvm, sp->gfn, 1178c50d8ae3SPaolo Bonzini KVM_PAGES_PER_HPAGE(sp->role.level)); 1179c50d8ae3SPaolo Bonzini } 1180c50d8ae3SPaolo Bonzini 1181c50d8ae3SPaolo Bonzini /* 1182c50d8ae3SPaolo Bonzini * Write-protect on the specified @sptep, @pt_protect indicates whether 1183c50d8ae3SPaolo Bonzini * spte write-protection is caused by protecting shadow page table. 1184c50d8ae3SPaolo Bonzini * 1185c50d8ae3SPaolo Bonzini * Note: write protection is difference between dirty logging and spte 1186c50d8ae3SPaolo Bonzini * protection: 1187c50d8ae3SPaolo Bonzini * - for dirty logging, the spte can be set to writable at anytime if 1188c50d8ae3SPaolo Bonzini * its dirty bitmap is properly set. 1189c50d8ae3SPaolo Bonzini * - for spte protection, the spte can be writable only after unsync-ing 1190c50d8ae3SPaolo Bonzini * shadow page. 1191c50d8ae3SPaolo Bonzini * 1192c50d8ae3SPaolo Bonzini * Return true if tlb need be flushed. 1193c50d8ae3SPaolo Bonzini */ 1194c50d8ae3SPaolo Bonzini static bool spte_write_protect(u64 *sptep, bool pt_protect) 1195c50d8ae3SPaolo Bonzini { 1196c50d8ae3SPaolo Bonzini u64 spte = *sptep; 1197c50d8ae3SPaolo Bonzini 1198c50d8ae3SPaolo Bonzini if (!is_writable_pte(spte) && 1199706c9c55SSean Christopherson !(pt_protect && is_mmu_writable_spte(spte))) 1200c50d8ae3SPaolo Bonzini return false; 1201c50d8ae3SPaolo Bonzini 1202805a0f83SStephen Zhang rmap_printk("spte %p %llx\n", sptep, *sptep); 1203c50d8ae3SPaolo Bonzini 1204c50d8ae3SPaolo Bonzini if (pt_protect) 12055fc3424fSSean Christopherson spte &= ~shadow_mmu_writable_mask; 1206c50d8ae3SPaolo Bonzini spte = spte & ~PT_WRITABLE_MASK; 1207c50d8ae3SPaolo Bonzini 1208c50d8ae3SPaolo Bonzini return mmu_spte_update(sptep, spte); 1209c50d8ae3SPaolo Bonzini } 1210c50d8ae3SPaolo Bonzini 12111346bbb6SDavid Matlack static bool rmap_write_protect(struct kvm_rmap_head *rmap_head, 1212c50d8ae3SPaolo Bonzini bool pt_protect) 1213c50d8ae3SPaolo Bonzini { 1214c50d8ae3SPaolo Bonzini u64 *sptep; 1215c50d8ae3SPaolo Bonzini struct rmap_iterator iter; 1216c50d8ae3SPaolo Bonzini bool flush = false; 1217c50d8ae3SPaolo Bonzini 1218c50d8ae3SPaolo Bonzini for_each_rmap_spte(rmap_head, &iter, sptep) 1219c50d8ae3SPaolo Bonzini flush |= spte_write_protect(sptep, pt_protect); 1220c50d8ae3SPaolo Bonzini 1221c50d8ae3SPaolo Bonzini return flush; 1222c50d8ae3SPaolo Bonzini } 1223c50d8ae3SPaolo Bonzini 1224c50d8ae3SPaolo Bonzini static bool spte_clear_dirty(u64 *sptep) 1225c50d8ae3SPaolo Bonzini { 1226c50d8ae3SPaolo Bonzini u64 spte = *sptep; 1227c50d8ae3SPaolo Bonzini 1228805a0f83SStephen Zhang rmap_printk("spte %p %llx\n", sptep, *sptep); 1229c50d8ae3SPaolo Bonzini 1230c50d8ae3SPaolo Bonzini MMU_WARN_ON(!spte_ad_enabled(spte)); 1231c50d8ae3SPaolo Bonzini spte &= ~shadow_dirty_mask; 1232c50d8ae3SPaolo Bonzini return mmu_spte_update(sptep, spte); 1233c50d8ae3SPaolo Bonzini } 1234c50d8ae3SPaolo Bonzini 1235c50d8ae3SPaolo Bonzini static bool spte_wrprot_for_clear_dirty(u64 *sptep) 1236c50d8ae3SPaolo Bonzini { 1237c50d8ae3SPaolo Bonzini bool was_writable = test_and_clear_bit(PT_WRITABLE_SHIFT, 1238c50d8ae3SPaolo Bonzini (unsigned long *)sptep); 1239c50d8ae3SPaolo Bonzini if (was_writable && !spte_ad_enabled(*sptep)) 1240c50d8ae3SPaolo Bonzini kvm_set_pfn_dirty(spte_to_pfn(*sptep)); 1241c50d8ae3SPaolo Bonzini 1242c50d8ae3SPaolo Bonzini return was_writable; 1243c50d8ae3SPaolo Bonzini } 1244c50d8ae3SPaolo Bonzini 1245c50d8ae3SPaolo Bonzini /* 1246c50d8ae3SPaolo Bonzini * Gets the GFN ready for another round of dirty logging by clearing the 1247c50d8ae3SPaolo Bonzini * - D bit on ad-enabled SPTEs, and 1248c50d8ae3SPaolo Bonzini * - W bit on ad-disabled SPTEs. 1249c50d8ae3SPaolo Bonzini * Returns true iff any D or W bits were cleared. 1250c50d8ae3SPaolo Bonzini */ 12510a234f5dSSean Christopherson static bool __rmap_clear_dirty(struct kvm *kvm, struct kvm_rmap_head *rmap_head, 1252269e9552SHamza Mahfooz const struct kvm_memory_slot *slot) 1253c50d8ae3SPaolo Bonzini { 1254c50d8ae3SPaolo Bonzini u64 *sptep; 1255c50d8ae3SPaolo Bonzini struct rmap_iterator iter; 1256c50d8ae3SPaolo Bonzini bool flush = false; 1257c50d8ae3SPaolo Bonzini 1258c50d8ae3SPaolo Bonzini for_each_rmap_spte(rmap_head, &iter, sptep) 1259c50d8ae3SPaolo Bonzini if (spte_ad_need_write_protect(*sptep)) 1260c50d8ae3SPaolo Bonzini flush |= spte_wrprot_for_clear_dirty(sptep); 1261c50d8ae3SPaolo Bonzini else 1262c50d8ae3SPaolo Bonzini flush |= spte_clear_dirty(sptep); 1263c50d8ae3SPaolo Bonzini 1264c50d8ae3SPaolo Bonzini return flush; 1265c50d8ae3SPaolo Bonzini } 1266c50d8ae3SPaolo Bonzini 1267c50d8ae3SPaolo Bonzini /** 1268c50d8ae3SPaolo Bonzini * kvm_mmu_write_protect_pt_masked - write protect selected PT level pages 1269c50d8ae3SPaolo Bonzini * @kvm: kvm instance 1270c50d8ae3SPaolo Bonzini * @slot: slot to protect 1271c50d8ae3SPaolo Bonzini * @gfn_offset: start of the BITS_PER_LONG pages we care about 1272c50d8ae3SPaolo Bonzini * @mask: indicates which pages we should protect 1273c50d8ae3SPaolo Bonzini * 127489212919SKeqian Zhu * Used when we do not need to care about huge page mappings. 1275c50d8ae3SPaolo Bonzini */ 1276c50d8ae3SPaolo Bonzini static void kvm_mmu_write_protect_pt_masked(struct kvm *kvm, 1277c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot, 1278c50d8ae3SPaolo Bonzini gfn_t gfn_offset, unsigned long mask) 1279c50d8ae3SPaolo Bonzini { 1280c50d8ae3SPaolo Bonzini struct kvm_rmap_head *rmap_head; 1281c50d8ae3SPaolo Bonzini 1282897218ffSPaolo Bonzini if (is_tdp_mmu_enabled(kvm)) 1283a6a0b05dSBen Gardon kvm_tdp_mmu_clear_dirty_pt_masked(kvm, slot, 1284a6a0b05dSBen Gardon slot->base_gfn + gfn_offset, mask, true); 1285e2209710SBen Gardon 1286e2209710SBen Gardon if (!kvm_memslots_have_rmaps(kvm)) 1287e2209710SBen Gardon return; 1288e2209710SBen Gardon 1289c50d8ae3SPaolo Bonzini while (mask) { 129093e083d4SDavid Matlack rmap_head = gfn_to_rmap(slot->base_gfn + gfn_offset + __ffs(mask), 12913bae0459SSean Christopherson PG_LEVEL_4K, slot); 12921346bbb6SDavid Matlack rmap_write_protect(rmap_head, false); 1293c50d8ae3SPaolo Bonzini 1294c50d8ae3SPaolo Bonzini /* clear the first set bit */ 1295c50d8ae3SPaolo Bonzini mask &= mask - 1; 1296c50d8ae3SPaolo Bonzini } 1297c50d8ae3SPaolo Bonzini } 1298c50d8ae3SPaolo Bonzini 1299c50d8ae3SPaolo Bonzini /** 1300c50d8ae3SPaolo Bonzini * kvm_mmu_clear_dirty_pt_masked - clear MMU D-bit for PT level pages, or write 1301c50d8ae3SPaolo Bonzini * protect the page if the D-bit isn't supported. 1302c50d8ae3SPaolo Bonzini * @kvm: kvm instance 1303c50d8ae3SPaolo Bonzini * @slot: slot to clear D-bit 1304c50d8ae3SPaolo Bonzini * @gfn_offset: start of the BITS_PER_LONG pages we care about 1305c50d8ae3SPaolo Bonzini * @mask: indicates which pages we should clear D-bit 1306c50d8ae3SPaolo Bonzini * 1307c50d8ae3SPaolo Bonzini * Used for PML to re-log the dirty GPAs after userspace querying dirty_bitmap. 1308c50d8ae3SPaolo Bonzini */ 1309a018eba5SSean Christopherson static void kvm_mmu_clear_dirty_pt_masked(struct kvm *kvm, 1310c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot, 1311c50d8ae3SPaolo Bonzini gfn_t gfn_offset, unsigned long mask) 1312c50d8ae3SPaolo Bonzini { 1313c50d8ae3SPaolo Bonzini struct kvm_rmap_head *rmap_head; 1314c50d8ae3SPaolo Bonzini 1315897218ffSPaolo Bonzini if (is_tdp_mmu_enabled(kvm)) 1316a6a0b05dSBen Gardon kvm_tdp_mmu_clear_dirty_pt_masked(kvm, slot, 1317a6a0b05dSBen Gardon slot->base_gfn + gfn_offset, mask, false); 1318e2209710SBen Gardon 1319e2209710SBen Gardon if (!kvm_memslots_have_rmaps(kvm)) 1320e2209710SBen Gardon return; 1321e2209710SBen Gardon 1322c50d8ae3SPaolo Bonzini while (mask) { 132393e083d4SDavid Matlack rmap_head = gfn_to_rmap(slot->base_gfn + gfn_offset + __ffs(mask), 13243bae0459SSean Christopherson PG_LEVEL_4K, slot); 13250a234f5dSSean Christopherson __rmap_clear_dirty(kvm, rmap_head, slot); 1326c50d8ae3SPaolo Bonzini 1327c50d8ae3SPaolo Bonzini /* clear the first set bit */ 1328c50d8ae3SPaolo Bonzini mask &= mask - 1; 1329c50d8ae3SPaolo Bonzini } 1330c50d8ae3SPaolo Bonzini } 1331c50d8ae3SPaolo Bonzini 1332c50d8ae3SPaolo Bonzini /** 1333c50d8ae3SPaolo Bonzini * kvm_arch_mmu_enable_log_dirty_pt_masked - enable dirty logging for selected 1334c50d8ae3SPaolo Bonzini * PT level pages. 1335c50d8ae3SPaolo Bonzini * 1336c50d8ae3SPaolo Bonzini * It calls kvm_mmu_write_protect_pt_masked to write protect selected pages to 1337c50d8ae3SPaolo Bonzini * enable dirty logging for them. 1338c50d8ae3SPaolo Bonzini * 133989212919SKeqian Zhu * We need to care about huge page mappings: e.g. during dirty logging we may 134089212919SKeqian Zhu * have such mappings. 1341c50d8ae3SPaolo Bonzini */ 1342c50d8ae3SPaolo Bonzini void kvm_arch_mmu_enable_log_dirty_pt_masked(struct kvm *kvm, 1343c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot, 1344c50d8ae3SPaolo Bonzini gfn_t gfn_offset, unsigned long mask) 1345c50d8ae3SPaolo Bonzini { 134689212919SKeqian Zhu /* 134789212919SKeqian Zhu * Huge pages are NOT write protected when we start dirty logging in 134889212919SKeqian Zhu * initially-all-set mode; must write protect them here so that they 134989212919SKeqian Zhu * are split to 4K on the first write. 135089212919SKeqian Zhu * 135189212919SKeqian Zhu * The gfn_offset is guaranteed to be aligned to 64, but the base_gfn 135289212919SKeqian Zhu * of memslot has no such restriction, so the range can cross two large 135389212919SKeqian Zhu * pages. 135489212919SKeqian Zhu */ 135589212919SKeqian Zhu if (kvm_dirty_log_manual_protect_and_init_set(kvm)) { 135689212919SKeqian Zhu gfn_t start = slot->base_gfn + gfn_offset + __ffs(mask); 135789212919SKeqian Zhu gfn_t end = slot->base_gfn + gfn_offset + __fls(mask); 135889212919SKeqian Zhu 1359cb00a70bSDavid Matlack if (READ_ONCE(eager_page_split)) 1360cb00a70bSDavid Matlack kvm_mmu_try_split_huge_pages(kvm, slot, start, end, PG_LEVEL_4K); 1361cb00a70bSDavid Matlack 136289212919SKeqian Zhu kvm_mmu_slot_gfn_write_protect(kvm, slot, start, PG_LEVEL_2M); 136389212919SKeqian Zhu 136489212919SKeqian Zhu /* Cross two large pages? */ 136589212919SKeqian Zhu if (ALIGN(start << PAGE_SHIFT, PMD_SIZE) != 136689212919SKeqian Zhu ALIGN(end << PAGE_SHIFT, PMD_SIZE)) 136789212919SKeqian Zhu kvm_mmu_slot_gfn_write_protect(kvm, slot, end, 136889212919SKeqian Zhu PG_LEVEL_2M); 136989212919SKeqian Zhu } 137089212919SKeqian Zhu 137189212919SKeqian Zhu /* Now handle 4K PTEs. */ 1372a018eba5SSean Christopherson if (kvm_x86_ops.cpu_dirty_log_size) 1373a018eba5SSean Christopherson kvm_mmu_clear_dirty_pt_masked(kvm, slot, gfn_offset, mask); 1374c50d8ae3SPaolo Bonzini else 1375c50d8ae3SPaolo Bonzini kvm_mmu_write_protect_pt_masked(kvm, slot, gfn_offset, mask); 1376c50d8ae3SPaolo Bonzini } 1377c50d8ae3SPaolo Bonzini 1378fb04a1edSPeter Xu int kvm_cpu_dirty_log_size(void) 1379fb04a1edSPeter Xu { 13806dd03800SSean Christopherson return kvm_x86_ops.cpu_dirty_log_size; 1381fb04a1edSPeter Xu } 1382fb04a1edSPeter Xu 1383c50d8ae3SPaolo Bonzini bool kvm_mmu_slot_gfn_write_protect(struct kvm *kvm, 13843ad93562SKeqian Zhu struct kvm_memory_slot *slot, u64 gfn, 13853ad93562SKeqian Zhu int min_level) 1386c50d8ae3SPaolo Bonzini { 1387c50d8ae3SPaolo Bonzini struct kvm_rmap_head *rmap_head; 1388c50d8ae3SPaolo Bonzini int i; 1389c50d8ae3SPaolo Bonzini bool write_protected = false; 1390c50d8ae3SPaolo Bonzini 1391e2209710SBen Gardon if (kvm_memslots_have_rmaps(kvm)) { 13923ad93562SKeqian Zhu for (i = min_level; i <= KVM_MAX_HUGEPAGE_LEVEL; ++i) { 139393e083d4SDavid Matlack rmap_head = gfn_to_rmap(gfn, i, slot); 13941346bbb6SDavid Matlack write_protected |= rmap_write_protect(rmap_head, true); 1395c50d8ae3SPaolo Bonzini } 1396e2209710SBen Gardon } 1397c50d8ae3SPaolo Bonzini 1398897218ffSPaolo Bonzini if (is_tdp_mmu_enabled(kvm)) 139946044f72SBen Gardon write_protected |= 14003ad93562SKeqian Zhu kvm_tdp_mmu_write_protect_gfn(kvm, slot, gfn, min_level); 140146044f72SBen Gardon 1402c50d8ae3SPaolo Bonzini return write_protected; 1403c50d8ae3SPaolo Bonzini } 1404c50d8ae3SPaolo Bonzini 1405cf48f9e2SDavid Matlack static bool kvm_vcpu_write_protect_gfn(struct kvm_vcpu *vcpu, u64 gfn) 1406c50d8ae3SPaolo Bonzini { 1407c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot; 1408c50d8ae3SPaolo Bonzini 1409c50d8ae3SPaolo Bonzini slot = kvm_vcpu_gfn_to_memslot(vcpu, gfn); 14103ad93562SKeqian Zhu return kvm_mmu_slot_gfn_write_protect(vcpu->kvm, slot, gfn, PG_LEVEL_4K); 1411c50d8ae3SPaolo Bonzini } 1412c50d8ae3SPaolo Bonzini 1413f8480721SSean Christopherson static bool __kvm_zap_rmap(struct kvm *kvm, struct kvm_rmap_head *rmap_head, 1414269e9552SHamza Mahfooz const struct kvm_memory_slot *slot) 1415c50d8ae3SPaolo Bonzini { 14169202aee8SSean Christopherson return kvm_zap_all_rmap_sptes(kvm, rmap_head); 1417c50d8ae3SPaolo Bonzini } 1418c50d8ae3SPaolo Bonzini 1419f8480721SSean Christopherson static bool kvm_zap_rmap(struct kvm *kvm, struct kvm_rmap_head *rmap_head, 1420c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot, gfn_t gfn, int level, 14213039bcc7SSean Christopherson pte_t unused) 1422c50d8ae3SPaolo Bonzini { 1423f8480721SSean Christopherson return __kvm_zap_rmap(kvm, rmap_head, slot); 1424c50d8ae3SPaolo Bonzini } 1425c50d8ae3SPaolo Bonzini 1426aed02fe3SSean Christopherson static bool kvm_set_pte_rmap(struct kvm *kvm, struct kvm_rmap_head *rmap_head, 1427c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot, gfn_t gfn, int level, 14283039bcc7SSean Christopherson pte_t pte) 1429c50d8ae3SPaolo Bonzini { 1430c50d8ae3SPaolo Bonzini u64 *sptep; 1431c50d8ae3SPaolo Bonzini struct rmap_iterator iter; 143298a26b69SVihas Mak bool need_flush = false; 1433c50d8ae3SPaolo Bonzini u64 new_spte; 1434c50d8ae3SPaolo Bonzini kvm_pfn_t new_pfn; 1435c50d8ae3SPaolo Bonzini 14363039bcc7SSean Christopherson WARN_ON(pte_huge(pte)); 14373039bcc7SSean Christopherson new_pfn = pte_pfn(pte); 1438c50d8ae3SPaolo Bonzini 1439c50d8ae3SPaolo Bonzini restart: 1440c50d8ae3SPaolo Bonzini for_each_rmap_spte(rmap_head, &iter, sptep) { 1441805a0f83SStephen Zhang rmap_printk("spte %p %llx gfn %llx (%d)\n", 1442c50d8ae3SPaolo Bonzini sptep, *sptep, gfn, level); 1443c50d8ae3SPaolo Bonzini 144498a26b69SVihas Mak need_flush = true; 1445c50d8ae3SPaolo Bonzini 14463039bcc7SSean Christopherson if (pte_write(pte)) { 14479202aee8SSean Christopherson kvm_zap_one_rmap_spte(kvm, rmap_head, sptep); 1448c50d8ae3SPaolo Bonzini goto restart; 1449c50d8ae3SPaolo Bonzini } else { 1450cb3eedabSPaolo Bonzini new_spte = kvm_mmu_changed_pte_notifier_make_spte( 1451cb3eedabSPaolo Bonzini *sptep, new_pfn); 1452c50d8ae3SPaolo Bonzini 145371f51d2cSMingwei Zhang mmu_spte_clear_track_bits(kvm, sptep); 1454c50d8ae3SPaolo Bonzini mmu_spte_set(sptep, new_spte); 1455c50d8ae3SPaolo Bonzini } 1456c50d8ae3SPaolo Bonzini } 1457c50d8ae3SPaolo Bonzini 1458c50d8ae3SPaolo Bonzini if (need_flush && kvm_available_flush_tlb_with_range()) { 1459c50d8ae3SPaolo Bonzini kvm_flush_remote_tlbs_with_address(kvm, gfn, 1); 146098a26b69SVihas Mak return false; 1461c50d8ae3SPaolo Bonzini } 1462c50d8ae3SPaolo Bonzini 1463c50d8ae3SPaolo Bonzini return need_flush; 1464c50d8ae3SPaolo Bonzini } 1465c50d8ae3SPaolo Bonzini 1466c50d8ae3SPaolo Bonzini struct slot_rmap_walk_iterator { 1467c50d8ae3SPaolo Bonzini /* input fields. */ 1468269e9552SHamza Mahfooz const struct kvm_memory_slot *slot; 1469c50d8ae3SPaolo Bonzini gfn_t start_gfn; 1470c50d8ae3SPaolo Bonzini gfn_t end_gfn; 1471c50d8ae3SPaolo Bonzini int start_level; 1472c50d8ae3SPaolo Bonzini int end_level; 1473c50d8ae3SPaolo Bonzini 1474c50d8ae3SPaolo Bonzini /* output fields. */ 1475c50d8ae3SPaolo Bonzini gfn_t gfn; 1476c50d8ae3SPaolo Bonzini struct kvm_rmap_head *rmap; 1477c50d8ae3SPaolo Bonzini int level; 1478c50d8ae3SPaolo Bonzini 1479c50d8ae3SPaolo Bonzini /* private field. */ 1480c50d8ae3SPaolo Bonzini struct kvm_rmap_head *end_rmap; 1481c50d8ae3SPaolo Bonzini }; 1482c50d8ae3SPaolo Bonzini 1483c50d8ae3SPaolo Bonzini static void 1484c50d8ae3SPaolo Bonzini rmap_walk_init_level(struct slot_rmap_walk_iterator *iterator, int level) 1485c50d8ae3SPaolo Bonzini { 1486c50d8ae3SPaolo Bonzini iterator->level = level; 1487c50d8ae3SPaolo Bonzini iterator->gfn = iterator->start_gfn; 148893e083d4SDavid Matlack iterator->rmap = gfn_to_rmap(iterator->gfn, level, iterator->slot); 148993e083d4SDavid Matlack iterator->end_rmap = gfn_to_rmap(iterator->end_gfn, level, iterator->slot); 1490c50d8ae3SPaolo Bonzini } 1491c50d8ae3SPaolo Bonzini 1492c50d8ae3SPaolo Bonzini static void 1493c50d8ae3SPaolo Bonzini slot_rmap_walk_init(struct slot_rmap_walk_iterator *iterator, 1494269e9552SHamza Mahfooz const struct kvm_memory_slot *slot, int start_level, 1495c50d8ae3SPaolo Bonzini int end_level, gfn_t start_gfn, gfn_t end_gfn) 1496c50d8ae3SPaolo Bonzini { 1497c50d8ae3SPaolo Bonzini iterator->slot = slot; 1498c50d8ae3SPaolo Bonzini iterator->start_level = start_level; 1499c50d8ae3SPaolo Bonzini iterator->end_level = end_level; 1500c50d8ae3SPaolo Bonzini iterator->start_gfn = start_gfn; 1501c50d8ae3SPaolo Bonzini iterator->end_gfn = end_gfn; 1502c50d8ae3SPaolo Bonzini 1503c50d8ae3SPaolo Bonzini rmap_walk_init_level(iterator, iterator->start_level); 1504c50d8ae3SPaolo Bonzini } 1505c50d8ae3SPaolo Bonzini 1506c50d8ae3SPaolo Bonzini static bool slot_rmap_walk_okay(struct slot_rmap_walk_iterator *iterator) 1507c50d8ae3SPaolo Bonzini { 1508c50d8ae3SPaolo Bonzini return !!iterator->rmap; 1509c50d8ae3SPaolo Bonzini } 1510c50d8ae3SPaolo Bonzini 1511c50d8ae3SPaolo Bonzini static void slot_rmap_walk_next(struct slot_rmap_walk_iterator *iterator) 1512c50d8ae3SPaolo Bonzini { 15136ba1e04fSVipin Sharma while (++iterator->rmap <= iterator->end_rmap) { 1514c50d8ae3SPaolo Bonzini iterator->gfn += (1UL << KVM_HPAGE_GFN_SHIFT(iterator->level)); 15156ba1e04fSVipin Sharma 15166ba1e04fSVipin Sharma if (iterator->rmap->val) 1517c50d8ae3SPaolo Bonzini return; 1518c50d8ae3SPaolo Bonzini } 1519c50d8ae3SPaolo Bonzini 1520c50d8ae3SPaolo Bonzini if (++iterator->level > iterator->end_level) { 1521c50d8ae3SPaolo Bonzini iterator->rmap = NULL; 1522c50d8ae3SPaolo Bonzini return; 1523c50d8ae3SPaolo Bonzini } 1524c50d8ae3SPaolo Bonzini 1525c50d8ae3SPaolo Bonzini rmap_walk_init_level(iterator, iterator->level); 1526c50d8ae3SPaolo Bonzini } 1527c50d8ae3SPaolo Bonzini 1528c50d8ae3SPaolo Bonzini #define for_each_slot_rmap_range(_slot_, _start_level_, _end_level_, \ 1529c50d8ae3SPaolo Bonzini _start_gfn, _end_gfn, _iter_) \ 1530c50d8ae3SPaolo Bonzini for (slot_rmap_walk_init(_iter_, _slot_, _start_level_, \ 1531c50d8ae3SPaolo Bonzini _end_level_, _start_gfn, _end_gfn); \ 1532c50d8ae3SPaolo Bonzini slot_rmap_walk_okay(_iter_); \ 1533c50d8ae3SPaolo Bonzini slot_rmap_walk_next(_iter_)) 1534c50d8ae3SPaolo Bonzini 15353039bcc7SSean Christopherson typedef bool (*rmap_handler_t)(struct kvm *kvm, struct kvm_rmap_head *rmap_head, 1536c1b91493SSean Christopherson struct kvm_memory_slot *slot, gfn_t gfn, 15373039bcc7SSean Christopherson int level, pte_t pte); 1538c1b91493SSean Christopherson 15393039bcc7SSean Christopherson static __always_inline bool kvm_handle_gfn_range(struct kvm *kvm, 15403039bcc7SSean Christopherson struct kvm_gfn_range *range, 1541c1b91493SSean Christopherson rmap_handler_t handler) 1542c50d8ae3SPaolo Bonzini { 1543c50d8ae3SPaolo Bonzini struct slot_rmap_walk_iterator iterator; 15443039bcc7SSean Christopherson bool ret = false; 1545c50d8ae3SPaolo Bonzini 15463039bcc7SSean Christopherson for_each_slot_rmap_range(range->slot, PG_LEVEL_4K, KVM_MAX_HUGEPAGE_LEVEL, 15473039bcc7SSean Christopherson range->start, range->end - 1, &iterator) 15483039bcc7SSean Christopherson ret |= handler(kvm, iterator.rmap, range->slot, iterator.gfn, 15493039bcc7SSean Christopherson iterator.level, range->pte); 1550c50d8ae3SPaolo Bonzini 1551c50d8ae3SPaolo Bonzini return ret; 1552c50d8ae3SPaolo Bonzini } 1553c50d8ae3SPaolo Bonzini 15543039bcc7SSean Christopherson bool kvm_unmap_gfn_range(struct kvm *kvm, struct kvm_gfn_range *range) 1555c50d8ae3SPaolo Bonzini { 1556e2209710SBen Gardon bool flush = false; 1557c50d8ae3SPaolo Bonzini 1558e2209710SBen Gardon if (kvm_memslots_have_rmaps(kvm)) 1559f8480721SSean Christopherson flush = kvm_handle_gfn_range(kvm, range, kvm_zap_rmap); 1560063afacdSBen Gardon 1561897218ffSPaolo Bonzini if (is_tdp_mmu_enabled(kvm)) 1562c7785d85SHou Wenlong flush = kvm_tdp_mmu_unmap_gfn_range(kvm, range, flush); 1563063afacdSBen Gardon 15643039bcc7SSean Christopherson return flush; 1565c50d8ae3SPaolo Bonzini } 1566c50d8ae3SPaolo Bonzini 15673039bcc7SSean Christopherson bool kvm_set_spte_gfn(struct kvm *kvm, struct kvm_gfn_range *range) 1568c50d8ae3SPaolo Bonzini { 1569e2209710SBen Gardon bool flush = false; 15701d8dd6b3SBen Gardon 1571e2209710SBen Gardon if (kvm_memslots_have_rmaps(kvm)) 1572aed02fe3SSean Christopherson flush = kvm_handle_gfn_range(kvm, range, kvm_set_pte_rmap); 15731d8dd6b3SBen Gardon 1574897218ffSPaolo Bonzini if (is_tdp_mmu_enabled(kvm)) 15753039bcc7SSean Christopherson flush |= kvm_tdp_mmu_set_spte_gfn(kvm, range); 15761d8dd6b3SBen Gardon 15773039bcc7SSean Christopherson return flush; 1578c50d8ae3SPaolo Bonzini } 1579c50d8ae3SPaolo Bonzini 1580aed02fe3SSean Christopherson static bool kvm_age_rmap(struct kvm *kvm, struct kvm_rmap_head *rmap_head, 1581c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot, gfn_t gfn, int level, 15823039bcc7SSean Christopherson pte_t unused) 1583c50d8ae3SPaolo Bonzini { 1584c50d8ae3SPaolo Bonzini u64 *sptep; 15853f649ab7SKees Cook struct rmap_iterator iter; 1586c50d8ae3SPaolo Bonzini int young = 0; 1587c50d8ae3SPaolo Bonzini 1588c50d8ae3SPaolo Bonzini for_each_rmap_spte(rmap_head, &iter, sptep) 1589c50d8ae3SPaolo Bonzini young |= mmu_spte_age(sptep); 1590c50d8ae3SPaolo Bonzini 1591c50d8ae3SPaolo Bonzini return young; 1592c50d8ae3SPaolo Bonzini } 1593c50d8ae3SPaolo Bonzini 1594aed02fe3SSean Christopherson static bool kvm_test_age_rmap(struct kvm *kvm, struct kvm_rmap_head *rmap_head, 1595c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot, gfn_t gfn, 15963039bcc7SSean Christopherson int level, pte_t unused) 1597c50d8ae3SPaolo Bonzini { 1598c50d8ae3SPaolo Bonzini u64 *sptep; 1599c50d8ae3SPaolo Bonzini struct rmap_iterator iter; 1600c50d8ae3SPaolo Bonzini 1601c50d8ae3SPaolo Bonzini for_each_rmap_spte(rmap_head, &iter, sptep) 1602c50d8ae3SPaolo Bonzini if (is_accessed_spte(*sptep)) 160398a26b69SVihas Mak return true; 160498a26b69SVihas Mak return false; 1605c50d8ae3SPaolo Bonzini } 1606c50d8ae3SPaolo Bonzini 1607c50d8ae3SPaolo Bonzini #define RMAP_RECYCLE_THRESHOLD 1000 1608c50d8ae3SPaolo Bonzini 16092ff9039aSDavid Matlack static void __rmap_add(struct kvm *kvm, 16102ff9039aSDavid Matlack struct kvm_mmu_memory_cache *cache, 16112ff9039aSDavid Matlack const struct kvm_memory_slot *slot, 161272ae5822SSean Christopherson u64 *spte, gfn_t gfn, unsigned int access) 1613c50d8ae3SPaolo Bonzini { 1614c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 161568be1306SDavid Matlack struct kvm_rmap_head *rmap_head; 161668be1306SDavid Matlack int rmap_count; 1617c50d8ae3SPaolo Bonzini 161857354682SSean Christopherson sp = sptep_to_sp(spte); 161979e48cecSSean Christopherson kvm_mmu_page_set_translation(sp, spte_index(spte), gfn, access); 162081cb4657SDavid Matlack kvm_update_page_stats(kvm, sp->role.level, 1); 162181cb4657SDavid Matlack 162293e083d4SDavid Matlack rmap_head = gfn_to_rmap(gfn, sp->role.level, slot); 16232ff9039aSDavid Matlack rmap_count = pte_list_add(cache, spte, rmap_head); 1624c50d8ae3SPaolo Bonzini 1625604f5332SMiaohe Lin if (rmap_count > kvm->stat.max_mmu_rmap_size) 1626604f5332SMiaohe Lin kvm->stat.max_mmu_rmap_size = rmap_count; 162768be1306SDavid Matlack if (rmap_count > RMAP_RECYCLE_THRESHOLD) { 16289202aee8SSean Christopherson kvm_zap_all_rmap_sptes(kvm, rmap_head); 162968be1306SDavid Matlack kvm_flush_remote_tlbs_with_address( 16302ff9039aSDavid Matlack kvm, sp->gfn, KVM_PAGES_PER_HPAGE(sp->role.level)); 163168be1306SDavid Matlack } 1632c50d8ae3SPaolo Bonzini } 1633c50d8ae3SPaolo Bonzini 16342ff9039aSDavid Matlack static void rmap_add(struct kvm_vcpu *vcpu, const struct kvm_memory_slot *slot, 163572ae5822SSean Christopherson u64 *spte, gfn_t gfn, unsigned int access) 16362ff9039aSDavid Matlack { 16372ff9039aSDavid Matlack struct kvm_mmu_memory_cache *cache = &vcpu->arch.mmu_pte_list_desc_cache; 16382ff9039aSDavid Matlack 16396a97575dSDavid Matlack __rmap_add(vcpu->kvm, cache, slot, spte, gfn, access); 16402ff9039aSDavid Matlack } 16412ff9039aSDavid Matlack 16423039bcc7SSean Christopherson bool kvm_age_gfn(struct kvm *kvm, struct kvm_gfn_range *range) 1643c50d8ae3SPaolo Bonzini { 1644e2209710SBen Gardon bool young = false; 1645f8e14497SBen Gardon 1646e2209710SBen Gardon if (kvm_memslots_have_rmaps(kvm)) 1647aed02fe3SSean Christopherson young = kvm_handle_gfn_range(kvm, range, kvm_age_rmap); 16483039bcc7SSean Christopherson 1649897218ffSPaolo Bonzini if (is_tdp_mmu_enabled(kvm)) 16503039bcc7SSean Christopherson young |= kvm_tdp_mmu_age_gfn_range(kvm, range); 1651f8e14497SBen Gardon 1652f8e14497SBen Gardon return young; 1653c50d8ae3SPaolo Bonzini } 1654c50d8ae3SPaolo Bonzini 16553039bcc7SSean Christopherson bool kvm_test_age_gfn(struct kvm *kvm, struct kvm_gfn_range *range) 1656c50d8ae3SPaolo Bonzini { 1657e2209710SBen Gardon bool young = false; 1658f8e14497SBen Gardon 1659e2209710SBen Gardon if (kvm_memslots_have_rmaps(kvm)) 1660aed02fe3SSean Christopherson young = kvm_handle_gfn_range(kvm, range, kvm_test_age_rmap); 16613039bcc7SSean Christopherson 1662897218ffSPaolo Bonzini if (is_tdp_mmu_enabled(kvm)) 16633039bcc7SSean Christopherson young |= kvm_tdp_mmu_test_age_gfn(kvm, range); 1664f8e14497SBen Gardon 1665f8e14497SBen Gardon return young; 1666c50d8ae3SPaolo Bonzini } 1667c50d8ae3SPaolo Bonzini 1668c50d8ae3SPaolo Bonzini #ifdef MMU_DEBUG 1669c50d8ae3SPaolo Bonzini static int is_empty_shadow_page(u64 *spt) 1670c50d8ae3SPaolo Bonzini { 1671c50d8ae3SPaolo Bonzini u64 *pos; 1672c50d8ae3SPaolo Bonzini u64 *end; 1673c50d8ae3SPaolo Bonzini 16743adbdf81SMiaohe Lin for (pos = spt, end = pos + SPTE_ENT_PER_PAGE; pos != end; pos++) 1675c50d8ae3SPaolo Bonzini if (is_shadow_present_pte(*pos)) { 1676c50d8ae3SPaolo Bonzini printk(KERN_ERR "%s: %p %llx\n", __func__, 1677c50d8ae3SPaolo Bonzini pos, *pos); 1678c50d8ae3SPaolo Bonzini return 0; 1679c50d8ae3SPaolo Bonzini } 1680c50d8ae3SPaolo Bonzini return 1; 1681c50d8ae3SPaolo Bonzini } 1682c50d8ae3SPaolo Bonzini #endif 1683c50d8ae3SPaolo Bonzini 1684c50d8ae3SPaolo Bonzini /* 1685c50d8ae3SPaolo Bonzini * This value is the sum of all of the kvm instances's 1686c50d8ae3SPaolo Bonzini * kvm->arch.n_used_mmu_pages values. We need a global, 1687c50d8ae3SPaolo Bonzini * aggregate version in order to make the slab shrinker 1688c50d8ae3SPaolo Bonzini * faster 1689c50d8ae3SPaolo Bonzini */ 1690d5aaad6fSSean Christopherson static inline void kvm_mod_used_mmu_pages(struct kvm *kvm, long nr) 1691c50d8ae3SPaolo Bonzini { 1692c50d8ae3SPaolo Bonzini kvm->arch.n_used_mmu_pages += nr; 1693c50d8ae3SPaolo Bonzini percpu_counter_add(&kvm_total_used_mmu_pages, nr); 1694c50d8ae3SPaolo Bonzini } 1695c50d8ae3SPaolo Bonzini 169643a063caSYosry Ahmed static void kvm_account_mmu_page(struct kvm *kvm, struct kvm_mmu_page *sp) 169743a063caSYosry Ahmed { 169843a063caSYosry Ahmed kvm_mod_used_mmu_pages(kvm, +1); 169943a063caSYosry Ahmed kvm_account_pgtable_pages((void *)sp->spt, +1); 170043a063caSYosry Ahmed } 170143a063caSYosry Ahmed 170243a063caSYosry Ahmed static void kvm_unaccount_mmu_page(struct kvm *kvm, struct kvm_mmu_page *sp) 170343a063caSYosry Ahmed { 170443a063caSYosry Ahmed kvm_mod_used_mmu_pages(kvm, -1); 170543a063caSYosry Ahmed kvm_account_pgtable_pages((void *)sp->spt, -1); 170643a063caSYosry Ahmed } 170743a063caSYosry Ahmed 170887654643SDavid Matlack static void kvm_mmu_free_shadow_page(struct kvm_mmu_page *sp) 1709c50d8ae3SPaolo Bonzini { 1710c50d8ae3SPaolo Bonzini MMU_WARN_ON(!is_empty_shadow_page(sp->spt)); 1711c50d8ae3SPaolo Bonzini hlist_del(&sp->hash_link); 1712c50d8ae3SPaolo Bonzini list_del(&sp->link); 1713c50d8ae3SPaolo Bonzini free_page((unsigned long)sp->spt); 1714c50d8ae3SPaolo Bonzini if (!sp->role.direct) 17156a97575dSDavid Matlack free_page((unsigned long)sp->shadowed_translation); 1716c50d8ae3SPaolo Bonzini kmem_cache_free(mmu_page_header_cache, sp); 1717c50d8ae3SPaolo Bonzini } 1718c50d8ae3SPaolo Bonzini 1719c50d8ae3SPaolo Bonzini static unsigned kvm_page_table_hashfn(gfn_t gfn) 1720c50d8ae3SPaolo Bonzini { 1721c50d8ae3SPaolo Bonzini return hash_64(gfn, KVM_MMU_HASH_SHIFT); 1722c50d8ae3SPaolo Bonzini } 1723c50d8ae3SPaolo Bonzini 17242ff9039aSDavid Matlack static void mmu_page_add_parent_pte(struct kvm_mmu_memory_cache *cache, 1725c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp, u64 *parent_pte) 1726c50d8ae3SPaolo Bonzini { 1727c50d8ae3SPaolo Bonzini if (!parent_pte) 1728c50d8ae3SPaolo Bonzini return; 1729c50d8ae3SPaolo Bonzini 17302ff9039aSDavid Matlack pte_list_add(cache, parent_pte, &sp->parent_ptes); 1731c50d8ae3SPaolo Bonzini } 1732c50d8ae3SPaolo Bonzini 1733c50d8ae3SPaolo Bonzini static void mmu_page_remove_parent_pte(struct kvm_mmu_page *sp, 1734c50d8ae3SPaolo Bonzini u64 *parent_pte) 1735c50d8ae3SPaolo Bonzini { 17363c2e1037SSean Christopherson pte_list_remove(parent_pte, &sp->parent_ptes); 1737c50d8ae3SPaolo Bonzini } 1738c50d8ae3SPaolo Bonzini 1739c50d8ae3SPaolo Bonzini static void drop_parent_pte(struct kvm_mmu_page *sp, 1740c50d8ae3SPaolo Bonzini u64 *parent_pte) 1741c50d8ae3SPaolo Bonzini { 1742c50d8ae3SPaolo Bonzini mmu_page_remove_parent_pte(sp, parent_pte); 1743c50d8ae3SPaolo Bonzini mmu_spte_clear_no_track(parent_pte); 1744c50d8ae3SPaolo Bonzini } 1745c50d8ae3SPaolo Bonzini 1746c50d8ae3SPaolo Bonzini static void mark_unsync(u64 *spte); 1747c50d8ae3SPaolo Bonzini static void kvm_mmu_mark_parents_unsync(struct kvm_mmu_page *sp) 1748c50d8ae3SPaolo Bonzini { 1749c50d8ae3SPaolo Bonzini u64 *sptep; 1750c50d8ae3SPaolo Bonzini struct rmap_iterator iter; 1751c50d8ae3SPaolo Bonzini 1752c50d8ae3SPaolo Bonzini for_each_rmap_spte(&sp->parent_ptes, &iter, sptep) { 1753c50d8ae3SPaolo Bonzini mark_unsync(sptep); 1754c50d8ae3SPaolo Bonzini } 1755c50d8ae3SPaolo Bonzini } 1756c50d8ae3SPaolo Bonzini 1757c50d8ae3SPaolo Bonzini static void mark_unsync(u64 *spte) 1758c50d8ae3SPaolo Bonzini { 1759c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 1760c50d8ae3SPaolo Bonzini 176157354682SSean Christopherson sp = sptep_to_sp(spte); 176279e48cecSSean Christopherson if (__test_and_set_bit(spte_index(spte), sp->unsync_child_bitmap)) 1763c50d8ae3SPaolo Bonzini return; 1764c50d8ae3SPaolo Bonzini if (sp->unsync_children++) 1765c50d8ae3SPaolo Bonzini return; 1766c50d8ae3SPaolo Bonzini kvm_mmu_mark_parents_unsync(sp); 1767c50d8ae3SPaolo Bonzini } 1768c50d8ae3SPaolo Bonzini 1769c50d8ae3SPaolo Bonzini static int nonpaging_sync_page(struct kvm_vcpu *vcpu, 1770c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp) 1771c50d8ae3SPaolo Bonzini { 1772c3e5e415SLai Jiangshan return -1; 1773c50d8ae3SPaolo Bonzini } 1774c50d8ae3SPaolo Bonzini 1775c50d8ae3SPaolo Bonzini #define KVM_PAGE_ARRAY_NR 16 1776c50d8ae3SPaolo Bonzini 1777c50d8ae3SPaolo Bonzini struct kvm_mmu_pages { 1778c50d8ae3SPaolo Bonzini struct mmu_page_and_offset { 1779c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 1780c50d8ae3SPaolo Bonzini unsigned int idx; 1781c50d8ae3SPaolo Bonzini } page[KVM_PAGE_ARRAY_NR]; 1782c50d8ae3SPaolo Bonzini unsigned int nr; 1783c50d8ae3SPaolo Bonzini }; 1784c50d8ae3SPaolo Bonzini 1785c50d8ae3SPaolo Bonzini static int mmu_pages_add(struct kvm_mmu_pages *pvec, struct kvm_mmu_page *sp, 1786c50d8ae3SPaolo Bonzini int idx) 1787c50d8ae3SPaolo Bonzini { 1788c50d8ae3SPaolo Bonzini int i; 1789c50d8ae3SPaolo Bonzini 1790c50d8ae3SPaolo Bonzini if (sp->unsync) 1791c50d8ae3SPaolo Bonzini for (i=0; i < pvec->nr; i++) 1792c50d8ae3SPaolo Bonzini if (pvec->page[i].sp == sp) 1793c50d8ae3SPaolo Bonzini return 0; 1794c50d8ae3SPaolo Bonzini 1795c50d8ae3SPaolo Bonzini pvec->page[pvec->nr].sp = sp; 1796c50d8ae3SPaolo Bonzini pvec->page[pvec->nr].idx = idx; 1797c50d8ae3SPaolo Bonzini pvec->nr++; 1798c50d8ae3SPaolo Bonzini return (pvec->nr == KVM_PAGE_ARRAY_NR); 1799c50d8ae3SPaolo Bonzini } 1800c50d8ae3SPaolo Bonzini 1801c50d8ae3SPaolo Bonzini static inline void clear_unsync_child_bit(struct kvm_mmu_page *sp, int idx) 1802c50d8ae3SPaolo Bonzini { 1803c50d8ae3SPaolo Bonzini --sp->unsync_children; 1804c50d8ae3SPaolo Bonzini WARN_ON((int)sp->unsync_children < 0); 1805c50d8ae3SPaolo Bonzini __clear_bit(idx, sp->unsync_child_bitmap); 1806c50d8ae3SPaolo Bonzini } 1807c50d8ae3SPaolo Bonzini 1808c50d8ae3SPaolo Bonzini static int __mmu_unsync_walk(struct kvm_mmu_page *sp, 1809c50d8ae3SPaolo Bonzini struct kvm_mmu_pages *pvec) 1810c50d8ae3SPaolo Bonzini { 1811c50d8ae3SPaolo Bonzini int i, ret, nr_unsync_leaf = 0; 1812c50d8ae3SPaolo Bonzini 1813c50d8ae3SPaolo Bonzini for_each_set_bit(i, sp->unsync_child_bitmap, 512) { 1814c50d8ae3SPaolo Bonzini struct kvm_mmu_page *child; 1815c50d8ae3SPaolo Bonzini u64 ent = sp->spt[i]; 1816c50d8ae3SPaolo Bonzini 1817c50d8ae3SPaolo Bonzini if (!is_shadow_present_pte(ent) || is_large_pte(ent)) { 1818c50d8ae3SPaolo Bonzini clear_unsync_child_bit(sp, i); 1819c50d8ae3SPaolo Bonzini continue; 1820c50d8ae3SPaolo Bonzini } 1821c50d8ae3SPaolo Bonzini 1822*5e3edd7eSSean Christopherson child = spte_to_child_sp(ent); 1823c50d8ae3SPaolo Bonzini 1824c50d8ae3SPaolo Bonzini if (child->unsync_children) { 1825c50d8ae3SPaolo Bonzini if (mmu_pages_add(pvec, child, i)) 1826c50d8ae3SPaolo Bonzini return -ENOSPC; 1827c50d8ae3SPaolo Bonzini 1828c50d8ae3SPaolo Bonzini ret = __mmu_unsync_walk(child, pvec); 1829c50d8ae3SPaolo Bonzini if (!ret) { 1830c50d8ae3SPaolo Bonzini clear_unsync_child_bit(sp, i); 1831c50d8ae3SPaolo Bonzini continue; 1832c50d8ae3SPaolo Bonzini } else if (ret > 0) { 1833c50d8ae3SPaolo Bonzini nr_unsync_leaf += ret; 1834c50d8ae3SPaolo Bonzini } else 1835c50d8ae3SPaolo Bonzini return ret; 1836c50d8ae3SPaolo Bonzini } else if (child->unsync) { 1837c50d8ae3SPaolo Bonzini nr_unsync_leaf++; 1838c50d8ae3SPaolo Bonzini if (mmu_pages_add(pvec, child, i)) 1839c50d8ae3SPaolo Bonzini return -ENOSPC; 1840c50d8ae3SPaolo Bonzini } else 1841c50d8ae3SPaolo Bonzini clear_unsync_child_bit(sp, i); 1842c50d8ae3SPaolo Bonzini } 1843c50d8ae3SPaolo Bonzini 1844c50d8ae3SPaolo Bonzini return nr_unsync_leaf; 1845c50d8ae3SPaolo Bonzini } 1846c50d8ae3SPaolo Bonzini 1847c50d8ae3SPaolo Bonzini #define INVALID_INDEX (-1) 1848c50d8ae3SPaolo Bonzini 1849c50d8ae3SPaolo Bonzini static int mmu_unsync_walk(struct kvm_mmu_page *sp, 1850c50d8ae3SPaolo Bonzini struct kvm_mmu_pages *pvec) 1851c50d8ae3SPaolo Bonzini { 1852c50d8ae3SPaolo Bonzini pvec->nr = 0; 1853c50d8ae3SPaolo Bonzini if (!sp->unsync_children) 1854c50d8ae3SPaolo Bonzini return 0; 1855c50d8ae3SPaolo Bonzini 1856c50d8ae3SPaolo Bonzini mmu_pages_add(pvec, sp, INVALID_INDEX); 1857c50d8ae3SPaolo Bonzini return __mmu_unsync_walk(sp, pvec); 1858c50d8ae3SPaolo Bonzini } 1859c50d8ae3SPaolo Bonzini 1860c50d8ae3SPaolo Bonzini static void kvm_unlink_unsync_page(struct kvm *kvm, struct kvm_mmu_page *sp) 1861c50d8ae3SPaolo Bonzini { 1862c50d8ae3SPaolo Bonzini WARN_ON(!sp->unsync); 1863c50d8ae3SPaolo Bonzini trace_kvm_mmu_sync_page(sp); 1864c50d8ae3SPaolo Bonzini sp->unsync = 0; 1865c50d8ae3SPaolo Bonzini --kvm->stat.mmu_unsync; 1866c50d8ae3SPaolo Bonzini } 1867c50d8ae3SPaolo Bonzini 1868c50d8ae3SPaolo Bonzini static bool kvm_mmu_prepare_zap_page(struct kvm *kvm, struct kvm_mmu_page *sp, 1869c50d8ae3SPaolo Bonzini struct list_head *invalid_list); 1870c50d8ae3SPaolo Bonzini static void kvm_mmu_commit_zap_page(struct kvm *kvm, 1871c50d8ae3SPaolo Bonzini struct list_head *invalid_list); 1872c50d8ae3SPaolo Bonzini 1873767d8d8dSLai Jiangshan static bool sp_has_gptes(struct kvm_mmu_page *sp) 1874767d8d8dSLai Jiangshan { 1875767d8d8dSLai Jiangshan if (sp->role.direct) 1876767d8d8dSLai Jiangshan return false; 1877767d8d8dSLai Jiangshan 187884e5ffd0SLai Jiangshan if (sp->role.passthrough) 187984e5ffd0SLai Jiangshan return false; 188084e5ffd0SLai Jiangshan 1881767d8d8dSLai Jiangshan return true; 1882767d8d8dSLai Jiangshan } 1883767d8d8dSLai Jiangshan 1884ac101b7cSSean Christopherson #define for_each_valid_sp(_kvm, _sp, _list) \ 1885ac101b7cSSean Christopherson hlist_for_each_entry(_sp, _list, hash_link) \ 1886c50d8ae3SPaolo Bonzini if (is_obsolete_sp((_kvm), (_sp))) { \ 1887c50d8ae3SPaolo Bonzini } else 1888c50d8ae3SPaolo Bonzini 1889767d8d8dSLai Jiangshan #define for_each_gfn_valid_sp_with_gptes(_kvm, _sp, _gfn) \ 1890ac101b7cSSean Christopherson for_each_valid_sp(_kvm, _sp, \ 1891ac101b7cSSean Christopherson &(_kvm)->arch.mmu_page_hash[kvm_page_table_hashfn(_gfn)]) \ 1892767d8d8dSLai Jiangshan if ((_sp)->gfn != (_gfn) || !sp_has_gptes(_sp)) {} else 1893c50d8ae3SPaolo Bonzini 18948d5678a7SHou Wenlong static int kvm_sync_page(struct kvm_vcpu *vcpu, struct kvm_mmu_page *sp, 1895c50d8ae3SPaolo Bonzini struct list_head *invalid_list) 1896c50d8ae3SPaolo Bonzini { 1897c3e5e415SLai Jiangshan int ret = vcpu->arch.mmu->sync_page(vcpu, sp); 1898c3e5e415SLai Jiangshan 18998d5678a7SHou Wenlong if (ret < 0) 1900c50d8ae3SPaolo Bonzini kvm_mmu_prepare_zap_page(vcpu->kvm, sp, invalid_list); 19018d5678a7SHou Wenlong return ret; 1902c50d8ae3SPaolo Bonzini } 1903c50d8ae3SPaolo Bonzini 1904c50d8ae3SPaolo Bonzini static bool kvm_mmu_remote_flush_or_zap(struct kvm *kvm, 1905c50d8ae3SPaolo Bonzini struct list_head *invalid_list, 1906c50d8ae3SPaolo Bonzini bool remote_flush) 1907c50d8ae3SPaolo Bonzini { 1908c50d8ae3SPaolo Bonzini if (!remote_flush && list_empty(invalid_list)) 1909c50d8ae3SPaolo Bonzini return false; 1910c50d8ae3SPaolo Bonzini 1911c50d8ae3SPaolo Bonzini if (!list_empty(invalid_list)) 1912c50d8ae3SPaolo Bonzini kvm_mmu_commit_zap_page(kvm, invalid_list); 1913c50d8ae3SPaolo Bonzini else 1914c50d8ae3SPaolo Bonzini kvm_flush_remote_tlbs(kvm); 1915c50d8ae3SPaolo Bonzini return true; 1916c50d8ae3SPaolo Bonzini } 1917c50d8ae3SPaolo Bonzini 1918c50d8ae3SPaolo Bonzini static bool is_obsolete_sp(struct kvm *kvm, struct kvm_mmu_page *sp) 1919c50d8ae3SPaolo Bonzini { 1920a955cad8SSean Christopherson if (sp->role.invalid) 1921a955cad8SSean Christopherson return true; 1922a955cad8SSean Christopherson 1923fa3e4203SMiaohe Lin /* TDP MMU pages do not use the MMU generation. */ 1924a955cad8SSean Christopherson return !sp->tdp_mmu_page && 1925c50d8ae3SPaolo Bonzini unlikely(sp->mmu_valid_gen != kvm->arch.mmu_valid_gen); 1926c50d8ae3SPaolo Bonzini } 1927c50d8ae3SPaolo Bonzini 1928c50d8ae3SPaolo Bonzini struct mmu_page_path { 1929c50d8ae3SPaolo Bonzini struct kvm_mmu_page *parent[PT64_ROOT_MAX_LEVEL]; 1930c50d8ae3SPaolo Bonzini unsigned int idx[PT64_ROOT_MAX_LEVEL]; 1931c50d8ae3SPaolo Bonzini }; 1932c50d8ae3SPaolo Bonzini 1933c50d8ae3SPaolo Bonzini #define for_each_sp(pvec, sp, parents, i) \ 1934c50d8ae3SPaolo Bonzini for (i = mmu_pages_first(&pvec, &parents); \ 1935c50d8ae3SPaolo Bonzini i < pvec.nr && ({ sp = pvec.page[i].sp; 1;}); \ 1936c50d8ae3SPaolo Bonzini i = mmu_pages_next(&pvec, &parents, i)) 1937c50d8ae3SPaolo Bonzini 1938c50d8ae3SPaolo Bonzini static int mmu_pages_next(struct kvm_mmu_pages *pvec, 1939c50d8ae3SPaolo Bonzini struct mmu_page_path *parents, 1940c50d8ae3SPaolo Bonzini int i) 1941c50d8ae3SPaolo Bonzini { 1942c50d8ae3SPaolo Bonzini int n; 1943c50d8ae3SPaolo Bonzini 1944c50d8ae3SPaolo Bonzini for (n = i+1; n < pvec->nr; n++) { 1945c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp = pvec->page[n].sp; 1946c50d8ae3SPaolo Bonzini unsigned idx = pvec->page[n].idx; 1947c50d8ae3SPaolo Bonzini int level = sp->role.level; 1948c50d8ae3SPaolo Bonzini 1949c50d8ae3SPaolo Bonzini parents->idx[level-1] = idx; 19503bae0459SSean Christopherson if (level == PG_LEVEL_4K) 1951c50d8ae3SPaolo Bonzini break; 1952c50d8ae3SPaolo Bonzini 1953c50d8ae3SPaolo Bonzini parents->parent[level-2] = sp; 1954c50d8ae3SPaolo Bonzini } 1955c50d8ae3SPaolo Bonzini 1956c50d8ae3SPaolo Bonzini return n; 1957c50d8ae3SPaolo Bonzini } 1958c50d8ae3SPaolo Bonzini 1959c50d8ae3SPaolo Bonzini static int mmu_pages_first(struct kvm_mmu_pages *pvec, 1960c50d8ae3SPaolo Bonzini struct mmu_page_path *parents) 1961c50d8ae3SPaolo Bonzini { 1962c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 1963c50d8ae3SPaolo Bonzini int level; 1964c50d8ae3SPaolo Bonzini 1965c50d8ae3SPaolo Bonzini if (pvec->nr == 0) 1966c50d8ae3SPaolo Bonzini return 0; 1967c50d8ae3SPaolo Bonzini 1968c50d8ae3SPaolo Bonzini WARN_ON(pvec->page[0].idx != INVALID_INDEX); 1969c50d8ae3SPaolo Bonzini 1970c50d8ae3SPaolo Bonzini sp = pvec->page[0].sp; 1971c50d8ae3SPaolo Bonzini level = sp->role.level; 19723bae0459SSean Christopherson WARN_ON(level == PG_LEVEL_4K); 1973c50d8ae3SPaolo Bonzini 1974c50d8ae3SPaolo Bonzini parents->parent[level-2] = sp; 1975c50d8ae3SPaolo Bonzini 1976c50d8ae3SPaolo Bonzini /* Also set up a sentinel. Further entries in pvec are all 1977c50d8ae3SPaolo Bonzini * children of sp, so this element is never overwritten. 1978c50d8ae3SPaolo Bonzini */ 1979c50d8ae3SPaolo Bonzini parents->parent[level-1] = NULL; 1980c50d8ae3SPaolo Bonzini return mmu_pages_next(pvec, parents, 0); 1981c50d8ae3SPaolo Bonzini } 1982c50d8ae3SPaolo Bonzini 1983c50d8ae3SPaolo Bonzini static void mmu_pages_clear_parents(struct mmu_page_path *parents) 1984c50d8ae3SPaolo Bonzini { 1985c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 1986c50d8ae3SPaolo Bonzini unsigned int level = 0; 1987c50d8ae3SPaolo Bonzini 1988c50d8ae3SPaolo Bonzini do { 1989c50d8ae3SPaolo Bonzini unsigned int idx = parents->idx[level]; 1990c50d8ae3SPaolo Bonzini sp = parents->parent[level]; 1991c50d8ae3SPaolo Bonzini if (!sp) 1992c50d8ae3SPaolo Bonzini return; 1993c50d8ae3SPaolo Bonzini 1994c50d8ae3SPaolo Bonzini WARN_ON(idx == INVALID_INDEX); 1995c50d8ae3SPaolo Bonzini clear_unsync_child_bit(sp, idx); 1996c50d8ae3SPaolo Bonzini level++; 1997c50d8ae3SPaolo Bonzini } while (!sp->unsync_children); 1998c50d8ae3SPaolo Bonzini } 1999c50d8ae3SPaolo Bonzini 200065855ed8SLai Jiangshan static int mmu_sync_children(struct kvm_vcpu *vcpu, 200165855ed8SLai Jiangshan struct kvm_mmu_page *parent, bool can_yield) 2002c50d8ae3SPaolo Bonzini { 2003c50d8ae3SPaolo Bonzini int i; 2004c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 2005c50d8ae3SPaolo Bonzini struct mmu_page_path parents; 2006c50d8ae3SPaolo Bonzini struct kvm_mmu_pages pages; 2007c50d8ae3SPaolo Bonzini LIST_HEAD(invalid_list); 2008c50d8ae3SPaolo Bonzini bool flush = false; 2009c50d8ae3SPaolo Bonzini 2010c50d8ae3SPaolo Bonzini while (mmu_unsync_walk(parent, &pages)) { 2011c50d8ae3SPaolo Bonzini bool protected = false; 2012c50d8ae3SPaolo Bonzini 2013c50d8ae3SPaolo Bonzini for_each_sp(pages, sp, parents, i) 2014cf48f9e2SDavid Matlack protected |= kvm_vcpu_write_protect_gfn(vcpu, sp->gfn); 2015c50d8ae3SPaolo Bonzini 2016c50d8ae3SPaolo Bonzini if (protected) { 20175591c069SLai Jiangshan kvm_mmu_remote_flush_or_zap(vcpu->kvm, &invalid_list, true); 2018c50d8ae3SPaolo Bonzini flush = false; 2019c50d8ae3SPaolo Bonzini } 2020c50d8ae3SPaolo Bonzini 2021c50d8ae3SPaolo Bonzini for_each_sp(pages, sp, parents, i) { 2022479a1efcSSean Christopherson kvm_unlink_unsync_page(vcpu->kvm, sp); 20238d5678a7SHou Wenlong flush |= kvm_sync_page(vcpu, sp, &invalid_list) > 0; 2024c50d8ae3SPaolo Bonzini mmu_pages_clear_parents(&parents); 2025c50d8ae3SPaolo Bonzini } 2026531810caSBen Gardon if (need_resched() || rwlock_needbreak(&vcpu->kvm->mmu_lock)) { 2027c3e5e415SLai Jiangshan kvm_mmu_remote_flush_or_zap(vcpu->kvm, &invalid_list, flush); 202865855ed8SLai Jiangshan if (!can_yield) { 202965855ed8SLai Jiangshan kvm_make_request(KVM_REQ_MMU_SYNC, vcpu); 203065855ed8SLai Jiangshan return -EINTR; 203165855ed8SLai Jiangshan } 203265855ed8SLai Jiangshan 2033531810caSBen Gardon cond_resched_rwlock_write(&vcpu->kvm->mmu_lock); 2034c50d8ae3SPaolo Bonzini flush = false; 2035c50d8ae3SPaolo Bonzini } 2036c50d8ae3SPaolo Bonzini } 2037c50d8ae3SPaolo Bonzini 2038c3e5e415SLai Jiangshan kvm_mmu_remote_flush_or_zap(vcpu->kvm, &invalid_list, flush); 203965855ed8SLai Jiangshan return 0; 2040c50d8ae3SPaolo Bonzini } 2041c50d8ae3SPaolo Bonzini 2042c50d8ae3SPaolo Bonzini static void __clear_sp_write_flooding_count(struct kvm_mmu_page *sp) 2043c50d8ae3SPaolo Bonzini { 2044c50d8ae3SPaolo Bonzini atomic_set(&sp->write_flooding_count, 0); 2045c50d8ae3SPaolo Bonzini } 2046c50d8ae3SPaolo Bonzini 2047c50d8ae3SPaolo Bonzini static void clear_sp_write_flooding_count(u64 *spte) 2048c50d8ae3SPaolo Bonzini { 204957354682SSean Christopherson __clear_sp_write_flooding_count(sptep_to_sp(spte)); 2050c50d8ae3SPaolo Bonzini } 2051c50d8ae3SPaolo Bonzini 2052cbd858b1SDavid Matlack /* 2053cbd858b1SDavid Matlack * The vCPU is required when finding indirect shadow pages; the shadow 2054cbd858b1SDavid Matlack * page may already exist and syncing it needs the vCPU pointer in 2055cbd858b1SDavid Matlack * order to read guest page tables. Direct shadow pages are never 2056cbd858b1SDavid Matlack * unsync, thus @vcpu can be NULL if @role.direct is true. 2057cbd858b1SDavid Matlack */ 20583cc736b3SDavid Matlack static struct kvm_mmu_page *kvm_mmu_find_shadow_page(struct kvm *kvm, 20593cc736b3SDavid Matlack struct kvm_vcpu *vcpu, 206094c81364SDavid Matlack gfn_t gfn, 206194c81364SDavid Matlack struct hlist_head *sp_list, 20622e65e842SDavid Matlack union kvm_mmu_page_role role) 2063c50d8ae3SPaolo Bonzini { 2064c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 20658d5678a7SHou Wenlong int ret; 2066c50d8ae3SPaolo Bonzini int collisions = 0; 2067c50d8ae3SPaolo Bonzini LIST_HEAD(invalid_list); 2068c50d8ae3SPaolo Bonzini 20693cc736b3SDavid Matlack for_each_valid_sp(kvm, sp, sp_list) { 2070c50d8ae3SPaolo Bonzini if (sp->gfn != gfn) { 2071c50d8ae3SPaolo Bonzini collisions++; 2072c50d8ae3SPaolo Bonzini continue; 2073c50d8ae3SPaolo Bonzini } 2074c50d8ae3SPaolo Bonzini 2075ddc16abbSSean Christopherson if (sp->role.word != role.word) { 2076ddc16abbSSean Christopherson /* 2077ddc16abbSSean Christopherson * If the guest is creating an upper-level page, zap 2078ddc16abbSSean Christopherson * unsync pages for the same gfn. While it's possible 2079ddc16abbSSean Christopherson * the guest is using recursive page tables, in all 2080ddc16abbSSean Christopherson * likelihood the guest has stopped using the unsync 2081ddc16abbSSean Christopherson * page and is installing a completely unrelated page. 2082ddc16abbSSean Christopherson * Unsync pages must not be left as is, because the new 2083ddc16abbSSean Christopherson * upper-level page will be write-protected. 2084ddc16abbSSean Christopherson */ 20852e65e842SDavid Matlack if (role.level > PG_LEVEL_4K && sp->unsync) 20863cc736b3SDavid Matlack kvm_mmu_prepare_zap_page(kvm, sp, 2087ddc16abbSSean Christopherson &invalid_list); 2088c50d8ae3SPaolo Bonzini continue; 2089ddc16abbSSean Christopherson } 2090c50d8ae3SPaolo Bonzini 2091bb924ca6SDavid Matlack /* unsync and write-flooding only apply to indirect SPs. */ 2092bb924ca6SDavid Matlack if (sp->role.direct) 209394c81364SDavid Matlack goto out; 2094fb58a9c3SSean Christopherson 2095c50d8ae3SPaolo Bonzini if (sp->unsync) { 2096cbd858b1SDavid Matlack if (KVM_BUG_ON(!vcpu, kvm)) 2097cbd858b1SDavid Matlack break; 2098cbd858b1SDavid Matlack 209907dc4f35SSean Christopherson /* 2100479a1efcSSean Christopherson * The page is good, but is stale. kvm_sync_page does 210107dc4f35SSean Christopherson * get the latest guest state, but (unlike mmu_unsync_children) 210207dc4f35SSean Christopherson * it doesn't write-protect the page or mark it synchronized! 210307dc4f35SSean Christopherson * This way the validity of the mapping is ensured, but the 210407dc4f35SSean Christopherson * overhead of write protection is not incurred until the 210507dc4f35SSean Christopherson * guest invalidates the TLB mapping. This allows multiple 210607dc4f35SSean Christopherson * SPs for a single gfn to be unsync. 210707dc4f35SSean Christopherson * 210807dc4f35SSean Christopherson * If the sync fails, the page is zapped. If so, break 210907dc4f35SSean Christopherson * in order to rebuild it. 2110c50d8ae3SPaolo Bonzini */ 21118d5678a7SHou Wenlong ret = kvm_sync_page(vcpu, sp, &invalid_list); 21128d5678a7SHou Wenlong if (ret < 0) 2113c50d8ae3SPaolo Bonzini break; 2114c50d8ae3SPaolo Bonzini 2115c50d8ae3SPaolo Bonzini WARN_ON(!list_empty(&invalid_list)); 21168d5678a7SHou Wenlong if (ret > 0) 21173cc736b3SDavid Matlack kvm_flush_remote_tlbs(kvm); 2118c50d8ae3SPaolo Bonzini } 2119c50d8ae3SPaolo Bonzini 2120c50d8ae3SPaolo Bonzini __clear_sp_write_flooding_count(sp); 2121fb58a9c3SSean Christopherson 2122c50d8ae3SPaolo Bonzini goto out; 2123c50d8ae3SPaolo Bonzini } 2124c50d8ae3SPaolo Bonzini 212594c81364SDavid Matlack sp = NULL; 21263cc736b3SDavid Matlack ++kvm->stat.mmu_cache_miss; 2127c50d8ae3SPaolo Bonzini 212894c81364SDavid Matlack out: 21293cc736b3SDavid Matlack kvm_mmu_commit_zap_page(kvm, &invalid_list); 213094c81364SDavid Matlack 21313cc736b3SDavid Matlack if (collisions > kvm->stat.max_mmu_page_hash_collisions) 21323cc736b3SDavid Matlack kvm->stat.max_mmu_page_hash_collisions = collisions; 213394c81364SDavid Matlack return sp; 213494c81364SDavid Matlack } 213594c81364SDavid Matlack 21362f8b1b53SDavid Matlack /* Caches used when allocating a new shadow page. */ 21372f8b1b53SDavid Matlack struct shadow_page_caches { 21382f8b1b53SDavid Matlack struct kvm_mmu_memory_cache *page_header_cache; 21392f8b1b53SDavid Matlack struct kvm_mmu_memory_cache *shadow_page_cache; 21406a97575dSDavid Matlack struct kvm_mmu_memory_cache *shadowed_info_cache; 21412f8b1b53SDavid Matlack }; 21422f8b1b53SDavid Matlack 2143336081fbSDavid Matlack static struct kvm_mmu_page *kvm_mmu_alloc_shadow_page(struct kvm *kvm, 21442f8b1b53SDavid Matlack struct shadow_page_caches *caches, 214594c81364SDavid Matlack gfn_t gfn, 214694c81364SDavid Matlack struct hlist_head *sp_list, 214794c81364SDavid Matlack union kvm_mmu_page_role role) 214894c81364SDavid Matlack { 2149c306aec8SDavid Matlack struct kvm_mmu_page *sp; 2150c306aec8SDavid Matlack 21512f8b1b53SDavid Matlack sp = kvm_mmu_memory_cache_alloc(caches->page_header_cache); 21522f8b1b53SDavid Matlack sp->spt = kvm_mmu_memory_cache_alloc(caches->shadow_page_cache); 2153c306aec8SDavid Matlack if (!role.direct) 21546a97575dSDavid Matlack sp->shadowed_translation = kvm_mmu_memory_cache_alloc(caches->shadowed_info_cache); 2155c306aec8SDavid Matlack 2156c306aec8SDavid Matlack set_page_private(virt_to_page(sp->spt), (unsigned long)sp); 2157c306aec8SDavid Matlack 215855c510e2SSean Christopherson INIT_LIST_HEAD(&sp->possible_nx_huge_page_link); 2159428e9216SSean Christopherson 2160c306aec8SDavid Matlack /* 2161c306aec8SDavid Matlack * active_mmu_pages must be a FIFO list, as kvm_zap_obsolete_pages() 2162c306aec8SDavid Matlack * depends on valid pages being added to the head of the list. See 2163c306aec8SDavid Matlack * comments in kvm_zap_obsolete_pages(). 2164c306aec8SDavid Matlack */ 2165336081fbSDavid Matlack sp->mmu_valid_gen = kvm->arch.mmu_valid_gen; 2166336081fbSDavid Matlack list_add(&sp->link, &kvm->arch.active_mmu_pages); 216743a063caSYosry Ahmed kvm_account_mmu_page(kvm, sp); 2168c50d8ae3SPaolo Bonzini 2169c50d8ae3SPaolo Bonzini sp->gfn = gfn; 2170c50d8ae3SPaolo Bonzini sp->role = role; 2171ac101b7cSSean Christopherson hlist_add_head(&sp->hash_link, sp_list); 2172be911771SDavid Matlack if (sp_has_gptes(sp)) 2173336081fbSDavid Matlack account_shadowed(kvm, sp); 2174ddc16abbSSean Christopherson 217594c81364SDavid Matlack return sp; 217694c81364SDavid Matlack } 217794c81364SDavid Matlack 2178cbd858b1SDavid Matlack /* Note, @vcpu may be NULL if @role.direct is true; see kvm_mmu_find_shadow_page. */ 21793cc736b3SDavid Matlack static struct kvm_mmu_page *__kvm_mmu_get_shadow_page(struct kvm *kvm, 21803cc736b3SDavid Matlack struct kvm_vcpu *vcpu, 21812f8b1b53SDavid Matlack struct shadow_page_caches *caches, 218287654643SDavid Matlack gfn_t gfn, 218394c81364SDavid Matlack union kvm_mmu_page_role role) 218494c81364SDavid Matlack { 218594c81364SDavid Matlack struct hlist_head *sp_list; 218694c81364SDavid Matlack struct kvm_mmu_page *sp; 218794c81364SDavid Matlack bool created = false; 218894c81364SDavid Matlack 21893cc736b3SDavid Matlack sp_list = &kvm->arch.mmu_page_hash[kvm_page_table_hashfn(gfn)]; 219094c81364SDavid Matlack 21913cc736b3SDavid Matlack sp = kvm_mmu_find_shadow_page(kvm, vcpu, gfn, sp_list, role); 219294c81364SDavid Matlack if (!sp) { 219394c81364SDavid Matlack created = true; 21943cc736b3SDavid Matlack sp = kvm_mmu_alloc_shadow_page(kvm, caches, gfn, sp_list, role); 219594c81364SDavid Matlack } 219694c81364SDavid Matlack 219794c81364SDavid Matlack trace_kvm_mmu_get_page(sp, created); 2198c50d8ae3SPaolo Bonzini return sp; 2199c50d8ae3SPaolo Bonzini } 2200c50d8ae3SPaolo Bonzini 22012f8b1b53SDavid Matlack static struct kvm_mmu_page *kvm_mmu_get_shadow_page(struct kvm_vcpu *vcpu, 22022f8b1b53SDavid Matlack gfn_t gfn, 22032f8b1b53SDavid Matlack union kvm_mmu_page_role role) 22042f8b1b53SDavid Matlack { 22052f8b1b53SDavid Matlack struct shadow_page_caches caches = { 22062f8b1b53SDavid Matlack .page_header_cache = &vcpu->arch.mmu_page_header_cache, 22072f8b1b53SDavid Matlack .shadow_page_cache = &vcpu->arch.mmu_shadow_page_cache, 22086a97575dSDavid Matlack .shadowed_info_cache = &vcpu->arch.mmu_shadowed_info_cache, 22092f8b1b53SDavid Matlack }; 22102f8b1b53SDavid Matlack 22113cc736b3SDavid Matlack return __kvm_mmu_get_shadow_page(vcpu->kvm, vcpu, &caches, gfn, role); 22122f8b1b53SDavid Matlack } 22132f8b1b53SDavid Matlack 221439944ab9SSean Christopherson static union kvm_mmu_page_role kvm_mmu_child_role(u64 *sptep, bool direct, 221539944ab9SSean Christopherson unsigned int access) 22162e65e842SDavid Matlack { 22172e65e842SDavid Matlack struct kvm_mmu_page *parent_sp = sptep_to_sp(sptep); 22182e65e842SDavid Matlack union kvm_mmu_page_role role; 22192e65e842SDavid Matlack 22202e65e842SDavid Matlack role = parent_sp->role; 22212e65e842SDavid Matlack role.level--; 22222e65e842SDavid Matlack role.access = access; 22232e65e842SDavid Matlack role.direct = direct; 22242e65e842SDavid Matlack role.passthrough = 0; 22252e65e842SDavid Matlack 22262e65e842SDavid Matlack /* 22272e65e842SDavid Matlack * If the guest has 4-byte PTEs then that means it's using 32-bit, 22282e65e842SDavid Matlack * 2-level, non-PAE paging. KVM shadows such guests with PAE paging 22292e65e842SDavid Matlack * (i.e. 8-byte PTEs). The difference in PTE size means that KVM must 22302e65e842SDavid Matlack * shadow each guest page table with multiple shadow page tables, which 22312e65e842SDavid Matlack * requires extra bookkeeping in the role. 22322e65e842SDavid Matlack * 22332e65e842SDavid Matlack * Specifically, to shadow the guest's page directory (which covers a 22342e65e842SDavid Matlack * 4GiB address space), KVM uses 4 PAE page directories, each mapping 22352e65e842SDavid Matlack * 1GiB of the address space. @role.quadrant encodes which quarter of 22362e65e842SDavid Matlack * the address space each maps. 22372e65e842SDavid Matlack * 22382e65e842SDavid Matlack * To shadow the guest's page tables (which each map a 4MiB region), KVM 22392e65e842SDavid Matlack * uses 2 PAE page tables, each mapping a 2MiB region. For these, 22402e65e842SDavid Matlack * @role.quadrant encodes which half of the region they map. 22412e65e842SDavid Matlack * 224239944ab9SSean Christopherson * Concretely, a 4-byte PDE consumes bits 31:22, while an 8-byte PDE 224339944ab9SSean Christopherson * consumes bits 29:21. To consume bits 31:30, KVM's uses 4 shadow 224439944ab9SSean Christopherson * PDPTEs; those 4 PAE page directories are pre-allocated and their 224539944ab9SSean Christopherson * quadrant is assigned in mmu_alloc_root(). A 4-byte PTE consumes 224639944ab9SSean Christopherson * bits 21:12, while an 8-byte PTE consumes bits 20:12. To consume 224739944ab9SSean Christopherson * bit 21 in the PTE (the child here), KVM propagates that bit to the 224839944ab9SSean Christopherson * quadrant, i.e. sets quadrant to '0' or '1'. The parent 8-byte PDE 224939944ab9SSean Christopherson * covers bit 21 (see above), thus the quadrant is calculated from the 225039944ab9SSean Christopherson * _least_ significant bit of the PDE index. 22512e65e842SDavid Matlack */ 22522e65e842SDavid Matlack if (role.has_4_byte_gpte) { 22532e65e842SDavid Matlack WARN_ON_ONCE(role.level != PG_LEVEL_4K); 225479e48cecSSean Christopherson role.quadrant = spte_index(sptep) & 1; 22552e65e842SDavid Matlack } 22562e65e842SDavid Matlack 22572e65e842SDavid Matlack return role; 22582e65e842SDavid Matlack } 22592e65e842SDavid Matlack 22602e65e842SDavid Matlack static struct kvm_mmu_page *kvm_mmu_get_child_sp(struct kvm_vcpu *vcpu, 22612e65e842SDavid Matlack u64 *sptep, gfn_t gfn, 22622e65e842SDavid Matlack bool direct, unsigned int access) 22632e65e842SDavid Matlack { 22642e65e842SDavid Matlack union kvm_mmu_page_role role; 22652e65e842SDavid Matlack 22660cd8dc73SPaolo Bonzini if (is_shadow_present_pte(*sptep) && !is_large_pte(*sptep)) 22670cd8dc73SPaolo Bonzini return ERR_PTR(-EEXIST); 22680cd8dc73SPaolo Bonzini 22692e65e842SDavid Matlack role = kvm_mmu_child_role(sptep, direct, access); 227087654643SDavid Matlack return kvm_mmu_get_shadow_page(vcpu, gfn, role); 22712e65e842SDavid Matlack } 22722e65e842SDavid Matlack 2273c50d8ae3SPaolo Bonzini static void shadow_walk_init_using_root(struct kvm_shadow_walk_iterator *iterator, 2274c50d8ae3SPaolo Bonzini struct kvm_vcpu *vcpu, hpa_t root, 2275c50d8ae3SPaolo Bonzini u64 addr) 2276c50d8ae3SPaolo Bonzini { 2277c50d8ae3SPaolo Bonzini iterator->addr = addr; 2278c50d8ae3SPaolo Bonzini iterator->shadow_addr = root; 2279a972e29cSPaolo Bonzini iterator->level = vcpu->arch.mmu->root_role.level; 2280c50d8ae3SPaolo Bonzini 228112ec33a7SLai Jiangshan if (iterator->level >= PT64_ROOT_4LEVEL && 22824d25502aSPaolo Bonzini vcpu->arch.mmu->cpu_role.base.level < PT64_ROOT_4LEVEL && 2283347a0d0dSPaolo Bonzini !vcpu->arch.mmu->root_role.direct) 228412ec33a7SLai Jiangshan iterator->level = PT32E_ROOT_LEVEL; 2285c50d8ae3SPaolo Bonzini 2286c50d8ae3SPaolo Bonzini if (iterator->level == PT32E_ROOT_LEVEL) { 2287c50d8ae3SPaolo Bonzini /* 2288c50d8ae3SPaolo Bonzini * prev_root is currently only used for 64-bit hosts. So only 2289c50d8ae3SPaolo Bonzini * the active root_hpa is valid here. 2290c50d8ae3SPaolo Bonzini */ 2291b9e5603cSPaolo Bonzini BUG_ON(root != vcpu->arch.mmu->root.hpa); 2292c50d8ae3SPaolo Bonzini 2293c50d8ae3SPaolo Bonzini iterator->shadow_addr 2294c50d8ae3SPaolo Bonzini = vcpu->arch.mmu->pae_root[(addr >> 30) & 3]; 22952ca3129eSSean Christopherson iterator->shadow_addr &= SPTE_BASE_ADDR_MASK; 2296c50d8ae3SPaolo Bonzini --iterator->level; 2297c50d8ae3SPaolo Bonzini if (!iterator->shadow_addr) 2298c50d8ae3SPaolo Bonzini iterator->level = 0; 2299c50d8ae3SPaolo Bonzini } 2300c50d8ae3SPaolo Bonzini } 2301c50d8ae3SPaolo Bonzini 2302c50d8ae3SPaolo Bonzini static void shadow_walk_init(struct kvm_shadow_walk_iterator *iterator, 2303c50d8ae3SPaolo Bonzini struct kvm_vcpu *vcpu, u64 addr) 2304c50d8ae3SPaolo Bonzini { 2305b9e5603cSPaolo Bonzini shadow_walk_init_using_root(iterator, vcpu, vcpu->arch.mmu->root.hpa, 2306c50d8ae3SPaolo Bonzini addr); 2307c50d8ae3SPaolo Bonzini } 2308c50d8ae3SPaolo Bonzini 2309c50d8ae3SPaolo Bonzini static bool shadow_walk_okay(struct kvm_shadow_walk_iterator *iterator) 2310c50d8ae3SPaolo Bonzini { 23113bae0459SSean Christopherson if (iterator->level < PG_LEVEL_4K) 2312c50d8ae3SPaolo Bonzini return false; 2313c50d8ae3SPaolo Bonzini 23142ca3129eSSean Christopherson iterator->index = SPTE_INDEX(iterator->addr, iterator->level); 2315c50d8ae3SPaolo Bonzini iterator->sptep = ((u64 *)__va(iterator->shadow_addr)) + iterator->index; 2316c50d8ae3SPaolo Bonzini return true; 2317c50d8ae3SPaolo Bonzini } 2318c50d8ae3SPaolo Bonzini 2319c50d8ae3SPaolo Bonzini static void __shadow_walk_next(struct kvm_shadow_walk_iterator *iterator, 2320c50d8ae3SPaolo Bonzini u64 spte) 2321c50d8ae3SPaolo Bonzini { 23223e44dce4SLai Jiangshan if (!is_shadow_present_pte(spte) || is_last_spte(spte, iterator->level)) { 2323c50d8ae3SPaolo Bonzini iterator->level = 0; 2324c50d8ae3SPaolo Bonzini return; 2325c50d8ae3SPaolo Bonzini } 2326c50d8ae3SPaolo Bonzini 23272ca3129eSSean Christopherson iterator->shadow_addr = spte & SPTE_BASE_ADDR_MASK; 2328c50d8ae3SPaolo Bonzini --iterator->level; 2329c50d8ae3SPaolo Bonzini } 2330c50d8ae3SPaolo Bonzini 2331c50d8ae3SPaolo Bonzini static void shadow_walk_next(struct kvm_shadow_walk_iterator *iterator) 2332c50d8ae3SPaolo Bonzini { 2333c50d8ae3SPaolo Bonzini __shadow_walk_next(iterator, *iterator->sptep); 2334c50d8ae3SPaolo Bonzini } 2335c50d8ae3SPaolo Bonzini 23360cd8dc73SPaolo Bonzini static void __link_shadow_page(struct kvm *kvm, 23370cd8dc73SPaolo Bonzini struct kvm_mmu_memory_cache *cache, u64 *sptep, 233803787394SPaolo Bonzini struct kvm_mmu_page *sp, bool flush) 2339c50d8ae3SPaolo Bonzini { 2340c50d8ae3SPaolo Bonzini u64 spte; 2341c50d8ae3SPaolo Bonzini 2342c50d8ae3SPaolo Bonzini BUILD_BUG_ON(VMX_EPT_WRITABLE_MASK != PT_WRITABLE_MASK); 2343c50d8ae3SPaolo Bonzini 23440cd8dc73SPaolo Bonzini /* 23450cd8dc73SPaolo Bonzini * If an SPTE is present already, it must be a leaf and therefore 234603787394SPaolo Bonzini * a large one. Drop it, and flush the TLB if needed, before 234703787394SPaolo Bonzini * installing sp. 23480cd8dc73SPaolo Bonzini */ 23490cd8dc73SPaolo Bonzini if (is_shadow_present_pte(*sptep)) 235003787394SPaolo Bonzini drop_large_spte(kvm, sptep, flush); 23510cd8dc73SPaolo Bonzini 2352cc4674d0SBen Gardon spte = make_nonleaf_spte(sp->spt, sp_ad_disabled(sp)); 2353c50d8ae3SPaolo Bonzini 2354c50d8ae3SPaolo Bonzini mmu_spte_set(sptep, spte); 2355c50d8ae3SPaolo Bonzini 23562ff9039aSDavid Matlack mmu_page_add_parent_pte(cache, sp, sptep); 2357c50d8ae3SPaolo Bonzini 2358c50d8ae3SPaolo Bonzini if (sp->unsync_children || sp->unsync) 2359c50d8ae3SPaolo Bonzini mark_unsync(sptep); 2360c50d8ae3SPaolo Bonzini } 2361c50d8ae3SPaolo Bonzini 23622ff9039aSDavid Matlack static void link_shadow_page(struct kvm_vcpu *vcpu, u64 *sptep, 23632ff9039aSDavid Matlack struct kvm_mmu_page *sp) 23642ff9039aSDavid Matlack { 236503787394SPaolo Bonzini __link_shadow_page(vcpu->kvm, &vcpu->arch.mmu_pte_list_desc_cache, sptep, sp, true); 23662ff9039aSDavid Matlack } 23672ff9039aSDavid Matlack 2368c50d8ae3SPaolo Bonzini static void validate_direct_spte(struct kvm_vcpu *vcpu, u64 *sptep, 2369c50d8ae3SPaolo Bonzini unsigned direct_access) 2370c50d8ae3SPaolo Bonzini { 2371c50d8ae3SPaolo Bonzini if (is_shadow_present_pte(*sptep) && !is_large_pte(*sptep)) { 2372c50d8ae3SPaolo Bonzini struct kvm_mmu_page *child; 2373c50d8ae3SPaolo Bonzini 2374c50d8ae3SPaolo Bonzini /* 2375c50d8ae3SPaolo Bonzini * For the direct sp, if the guest pte's dirty bit 2376c50d8ae3SPaolo Bonzini * changed form clean to dirty, it will corrupt the 2377c50d8ae3SPaolo Bonzini * sp's access: allow writable in the read-only sp, 2378c50d8ae3SPaolo Bonzini * so we should update the spte at this point to get 2379c50d8ae3SPaolo Bonzini * a new sp with the correct access. 2380c50d8ae3SPaolo Bonzini */ 2381*5e3edd7eSSean Christopherson child = spte_to_child_sp(*sptep); 2382c50d8ae3SPaolo Bonzini if (child->role.access == direct_access) 2383c50d8ae3SPaolo Bonzini return; 2384c50d8ae3SPaolo Bonzini 2385c50d8ae3SPaolo Bonzini drop_parent_pte(child, sptep); 2386c50d8ae3SPaolo Bonzini kvm_flush_remote_tlbs_with_address(vcpu->kvm, child->gfn, 1); 2387c50d8ae3SPaolo Bonzini } 2388c50d8ae3SPaolo Bonzini } 2389c50d8ae3SPaolo Bonzini 23902de4085cSBen Gardon /* Returns the number of zapped non-leaf child shadow pages. */ 23912de4085cSBen Gardon static int mmu_page_zap_pte(struct kvm *kvm, struct kvm_mmu_page *sp, 23922de4085cSBen Gardon u64 *spte, struct list_head *invalid_list) 2393c50d8ae3SPaolo Bonzini { 2394c50d8ae3SPaolo Bonzini u64 pte; 2395c50d8ae3SPaolo Bonzini struct kvm_mmu_page *child; 2396c50d8ae3SPaolo Bonzini 2397c50d8ae3SPaolo Bonzini pte = *spte; 2398c50d8ae3SPaolo Bonzini if (is_shadow_present_pte(pte)) { 2399c50d8ae3SPaolo Bonzini if (is_last_spte(pte, sp->role.level)) { 2400c50d8ae3SPaolo Bonzini drop_spte(kvm, spte); 2401c50d8ae3SPaolo Bonzini } else { 2402*5e3edd7eSSean Christopherson child = spte_to_child_sp(pte); 2403c50d8ae3SPaolo Bonzini drop_parent_pte(child, spte); 24042de4085cSBen Gardon 24052de4085cSBen Gardon /* 24062de4085cSBen Gardon * Recursively zap nested TDP SPs, parentless SPs are 24072de4085cSBen Gardon * unlikely to be used again in the near future. This 24082de4085cSBen Gardon * avoids retaining a large number of stale nested SPs. 24092de4085cSBen Gardon */ 24102de4085cSBen Gardon if (tdp_enabled && invalid_list && 24112de4085cSBen Gardon child->role.guest_mode && !child->parent_ptes.val) 24122de4085cSBen Gardon return kvm_mmu_prepare_zap_page(kvm, child, 24132de4085cSBen Gardon invalid_list); 2414c50d8ae3SPaolo Bonzini } 2415ace569e0SSean Christopherson } else if (is_mmio_spte(pte)) { 2416c50d8ae3SPaolo Bonzini mmu_spte_clear_no_track(spte); 2417ace569e0SSean Christopherson } 24182de4085cSBen Gardon return 0; 2419c50d8ae3SPaolo Bonzini } 2420c50d8ae3SPaolo Bonzini 24212de4085cSBen Gardon static int kvm_mmu_page_unlink_children(struct kvm *kvm, 24222de4085cSBen Gardon struct kvm_mmu_page *sp, 24232de4085cSBen Gardon struct list_head *invalid_list) 2424c50d8ae3SPaolo Bonzini { 24252de4085cSBen Gardon int zapped = 0; 2426c50d8ae3SPaolo Bonzini unsigned i; 2427c50d8ae3SPaolo Bonzini 24282ca3129eSSean Christopherson for (i = 0; i < SPTE_ENT_PER_PAGE; ++i) 24292de4085cSBen Gardon zapped += mmu_page_zap_pte(kvm, sp, sp->spt + i, invalid_list); 24302de4085cSBen Gardon 24312de4085cSBen Gardon return zapped; 2432c50d8ae3SPaolo Bonzini } 2433c50d8ae3SPaolo Bonzini 243461827671SJinrong Liang static void kvm_mmu_unlink_parents(struct kvm_mmu_page *sp) 2435c50d8ae3SPaolo Bonzini { 2436c50d8ae3SPaolo Bonzini u64 *sptep; 2437c50d8ae3SPaolo Bonzini struct rmap_iterator iter; 2438c50d8ae3SPaolo Bonzini 2439c50d8ae3SPaolo Bonzini while ((sptep = rmap_get_first(&sp->parent_ptes, &iter))) 2440c50d8ae3SPaolo Bonzini drop_parent_pte(sp, sptep); 2441c50d8ae3SPaolo Bonzini } 2442c50d8ae3SPaolo Bonzini 2443c50d8ae3SPaolo Bonzini static int mmu_zap_unsync_children(struct kvm *kvm, 2444c50d8ae3SPaolo Bonzini struct kvm_mmu_page *parent, 2445c50d8ae3SPaolo Bonzini struct list_head *invalid_list) 2446c50d8ae3SPaolo Bonzini { 2447c50d8ae3SPaolo Bonzini int i, zapped = 0; 2448c50d8ae3SPaolo Bonzini struct mmu_page_path parents; 2449c50d8ae3SPaolo Bonzini struct kvm_mmu_pages pages; 2450c50d8ae3SPaolo Bonzini 24513bae0459SSean Christopherson if (parent->role.level == PG_LEVEL_4K) 2452c50d8ae3SPaolo Bonzini return 0; 2453c50d8ae3SPaolo Bonzini 2454c50d8ae3SPaolo Bonzini while (mmu_unsync_walk(parent, &pages)) { 2455c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 2456c50d8ae3SPaolo Bonzini 2457c50d8ae3SPaolo Bonzini for_each_sp(pages, sp, parents, i) { 2458c50d8ae3SPaolo Bonzini kvm_mmu_prepare_zap_page(kvm, sp, invalid_list); 2459c50d8ae3SPaolo Bonzini mmu_pages_clear_parents(&parents); 2460c50d8ae3SPaolo Bonzini zapped++; 2461c50d8ae3SPaolo Bonzini } 2462c50d8ae3SPaolo Bonzini } 2463c50d8ae3SPaolo Bonzini 2464c50d8ae3SPaolo Bonzini return zapped; 2465c50d8ae3SPaolo Bonzini } 2466c50d8ae3SPaolo Bonzini 2467c50d8ae3SPaolo Bonzini static bool __kvm_mmu_prepare_zap_page(struct kvm *kvm, 2468c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp, 2469c50d8ae3SPaolo Bonzini struct list_head *invalid_list, 2470c50d8ae3SPaolo Bonzini int *nr_zapped) 2471c50d8ae3SPaolo Bonzini { 2472527d5cd7SSean Christopherson bool list_unstable, zapped_root = false; 2473c50d8ae3SPaolo Bonzini 2474c50d8ae3SPaolo Bonzini trace_kvm_mmu_prepare_zap_page(sp); 2475c50d8ae3SPaolo Bonzini ++kvm->stat.mmu_shadow_zapped; 2476c50d8ae3SPaolo Bonzini *nr_zapped = mmu_zap_unsync_children(kvm, sp, invalid_list); 24772de4085cSBen Gardon *nr_zapped += kvm_mmu_page_unlink_children(kvm, sp, invalid_list); 247861827671SJinrong Liang kvm_mmu_unlink_parents(sp); 2479c50d8ae3SPaolo Bonzini 2480c50d8ae3SPaolo Bonzini /* Zapping children means active_mmu_pages has become unstable. */ 2481c50d8ae3SPaolo Bonzini list_unstable = *nr_zapped; 2482c50d8ae3SPaolo Bonzini 2483767d8d8dSLai Jiangshan if (!sp->role.invalid && sp_has_gptes(sp)) 2484c50d8ae3SPaolo Bonzini unaccount_shadowed(kvm, sp); 2485c50d8ae3SPaolo Bonzini 2486c50d8ae3SPaolo Bonzini if (sp->unsync) 2487c50d8ae3SPaolo Bonzini kvm_unlink_unsync_page(kvm, sp); 2488c50d8ae3SPaolo Bonzini if (!sp->root_count) { 2489c50d8ae3SPaolo Bonzini /* Count self */ 2490c50d8ae3SPaolo Bonzini (*nr_zapped)++; 2491f95eec9bSSean Christopherson 2492f95eec9bSSean Christopherson /* 2493f95eec9bSSean Christopherson * Already invalid pages (previously active roots) are not on 2494f95eec9bSSean Christopherson * the active page list. See list_del() in the "else" case of 2495f95eec9bSSean Christopherson * !sp->root_count. 2496f95eec9bSSean Christopherson */ 2497f95eec9bSSean Christopherson if (sp->role.invalid) 2498f95eec9bSSean Christopherson list_add(&sp->link, invalid_list); 2499f95eec9bSSean Christopherson else 2500c50d8ae3SPaolo Bonzini list_move(&sp->link, invalid_list); 250143a063caSYosry Ahmed kvm_unaccount_mmu_page(kvm, sp); 2502c50d8ae3SPaolo Bonzini } else { 2503f95eec9bSSean Christopherson /* 2504f95eec9bSSean Christopherson * Remove the active root from the active page list, the root 2505f95eec9bSSean Christopherson * will be explicitly freed when the root_count hits zero. 2506f95eec9bSSean Christopherson */ 2507f95eec9bSSean Christopherson list_del(&sp->link); 2508c50d8ae3SPaolo Bonzini 2509c50d8ae3SPaolo Bonzini /* 2510c50d8ae3SPaolo Bonzini * Obsolete pages cannot be used on any vCPUs, see the comment 2511c50d8ae3SPaolo Bonzini * in kvm_mmu_zap_all_fast(). Note, is_obsolete_sp() also 2512c50d8ae3SPaolo Bonzini * treats invalid shadow pages as being obsolete. 2513c50d8ae3SPaolo Bonzini */ 2514527d5cd7SSean Christopherson zapped_root = !is_obsolete_sp(kvm, sp); 2515c50d8ae3SPaolo Bonzini } 2516c50d8ae3SPaolo Bonzini 251755c510e2SSean Christopherson if (sp->nx_huge_page_disallowed) 251855c510e2SSean Christopherson unaccount_nx_huge_page(kvm, sp); 2519c50d8ae3SPaolo Bonzini 2520c50d8ae3SPaolo Bonzini sp->role.invalid = 1; 2521527d5cd7SSean Christopherson 2522527d5cd7SSean Christopherson /* 2523527d5cd7SSean Christopherson * Make the request to free obsolete roots after marking the root 2524527d5cd7SSean Christopherson * invalid, otherwise other vCPUs may not see it as invalid. 2525527d5cd7SSean Christopherson */ 2526527d5cd7SSean Christopherson if (zapped_root) 2527527d5cd7SSean Christopherson kvm_make_all_cpus_request(kvm, KVM_REQ_MMU_FREE_OBSOLETE_ROOTS); 2528c50d8ae3SPaolo Bonzini return list_unstable; 2529c50d8ae3SPaolo Bonzini } 2530c50d8ae3SPaolo Bonzini 2531c50d8ae3SPaolo Bonzini static bool kvm_mmu_prepare_zap_page(struct kvm *kvm, struct kvm_mmu_page *sp, 2532c50d8ae3SPaolo Bonzini struct list_head *invalid_list) 2533c50d8ae3SPaolo Bonzini { 2534c50d8ae3SPaolo Bonzini int nr_zapped; 2535c50d8ae3SPaolo Bonzini 2536c50d8ae3SPaolo Bonzini __kvm_mmu_prepare_zap_page(kvm, sp, invalid_list, &nr_zapped); 2537c50d8ae3SPaolo Bonzini return nr_zapped; 2538c50d8ae3SPaolo Bonzini } 2539c50d8ae3SPaolo Bonzini 2540c50d8ae3SPaolo Bonzini static void kvm_mmu_commit_zap_page(struct kvm *kvm, 2541c50d8ae3SPaolo Bonzini struct list_head *invalid_list) 2542c50d8ae3SPaolo Bonzini { 2543c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp, *nsp; 2544c50d8ae3SPaolo Bonzini 2545c50d8ae3SPaolo Bonzini if (list_empty(invalid_list)) 2546c50d8ae3SPaolo Bonzini return; 2547c50d8ae3SPaolo Bonzini 2548c50d8ae3SPaolo Bonzini /* 2549c50d8ae3SPaolo Bonzini * We need to make sure everyone sees our modifications to 2550c50d8ae3SPaolo Bonzini * the page tables and see changes to vcpu->mode here. The barrier 2551c50d8ae3SPaolo Bonzini * in the kvm_flush_remote_tlbs() achieves this. This pairs 2552c50d8ae3SPaolo Bonzini * with vcpu_enter_guest and walk_shadow_page_lockless_begin/end. 2553c50d8ae3SPaolo Bonzini * 2554c50d8ae3SPaolo Bonzini * In addition, kvm_flush_remote_tlbs waits for all vcpus to exit 2555c50d8ae3SPaolo Bonzini * guest mode and/or lockless shadow page table walks. 2556c50d8ae3SPaolo Bonzini */ 2557c50d8ae3SPaolo Bonzini kvm_flush_remote_tlbs(kvm); 2558c50d8ae3SPaolo Bonzini 2559c50d8ae3SPaolo Bonzini list_for_each_entry_safe(sp, nsp, invalid_list, link) { 2560c50d8ae3SPaolo Bonzini WARN_ON(!sp->role.invalid || sp->root_count); 256187654643SDavid Matlack kvm_mmu_free_shadow_page(sp); 2562c50d8ae3SPaolo Bonzini } 2563c50d8ae3SPaolo Bonzini } 2564c50d8ae3SPaolo Bonzini 25656b82ef2cSSean Christopherson static unsigned long kvm_mmu_zap_oldest_mmu_pages(struct kvm *kvm, 25666b82ef2cSSean Christopherson unsigned long nr_to_zap) 2567c50d8ae3SPaolo Bonzini { 25686b82ef2cSSean Christopherson unsigned long total_zapped = 0; 25696b82ef2cSSean Christopherson struct kvm_mmu_page *sp, *tmp; 2570ba7888ddSSean Christopherson LIST_HEAD(invalid_list); 25716b82ef2cSSean Christopherson bool unstable; 25726b82ef2cSSean Christopherson int nr_zapped; 2573c50d8ae3SPaolo Bonzini 2574c50d8ae3SPaolo Bonzini if (list_empty(&kvm->arch.active_mmu_pages)) 2575ba7888ddSSean Christopherson return 0; 2576c50d8ae3SPaolo Bonzini 25776b82ef2cSSean Christopherson restart: 25788fc51726SSean Christopherson list_for_each_entry_safe_reverse(sp, tmp, &kvm->arch.active_mmu_pages, link) { 25796b82ef2cSSean Christopherson /* 25806b82ef2cSSean Christopherson * Don't zap active root pages, the page itself can't be freed 25816b82ef2cSSean Christopherson * and zapping it will just force vCPUs to realloc and reload. 25826b82ef2cSSean Christopherson */ 25836b82ef2cSSean Christopherson if (sp->root_count) 25846b82ef2cSSean Christopherson continue; 25856b82ef2cSSean Christopherson 25866b82ef2cSSean Christopherson unstable = __kvm_mmu_prepare_zap_page(kvm, sp, &invalid_list, 25876b82ef2cSSean Christopherson &nr_zapped); 25886b82ef2cSSean Christopherson total_zapped += nr_zapped; 25896b82ef2cSSean Christopherson if (total_zapped >= nr_to_zap) 2590ba7888ddSSean Christopherson break; 2591ba7888ddSSean Christopherson 25926b82ef2cSSean Christopherson if (unstable) 25936b82ef2cSSean Christopherson goto restart; 2594ba7888ddSSean Christopherson } 25956b82ef2cSSean Christopherson 25966b82ef2cSSean Christopherson kvm_mmu_commit_zap_page(kvm, &invalid_list); 25976b82ef2cSSean Christopherson 25986b82ef2cSSean Christopherson kvm->stat.mmu_recycled += total_zapped; 25996b82ef2cSSean Christopherson return total_zapped; 26006b82ef2cSSean Christopherson } 26016b82ef2cSSean Christopherson 2602afe8d7e6SSean Christopherson static inline unsigned long kvm_mmu_available_pages(struct kvm *kvm) 2603afe8d7e6SSean Christopherson { 2604afe8d7e6SSean Christopherson if (kvm->arch.n_max_mmu_pages > kvm->arch.n_used_mmu_pages) 2605afe8d7e6SSean Christopherson return kvm->arch.n_max_mmu_pages - 2606afe8d7e6SSean Christopherson kvm->arch.n_used_mmu_pages; 2607afe8d7e6SSean Christopherson 2608afe8d7e6SSean Christopherson return 0; 2609c50d8ae3SPaolo Bonzini } 2610c50d8ae3SPaolo Bonzini 2611ba7888ddSSean Christopherson static int make_mmu_pages_available(struct kvm_vcpu *vcpu) 2612ba7888ddSSean Christopherson { 26136b82ef2cSSean Christopherson unsigned long avail = kvm_mmu_available_pages(vcpu->kvm); 2614ba7888ddSSean Christopherson 26156b82ef2cSSean Christopherson if (likely(avail >= KVM_MIN_FREE_MMU_PAGES)) 2616ba7888ddSSean Christopherson return 0; 2617ba7888ddSSean Christopherson 26186b82ef2cSSean Christopherson kvm_mmu_zap_oldest_mmu_pages(vcpu->kvm, KVM_REFILL_PAGES - avail); 2619ba7888ddSSean Christopherson 26206e6ec584SSean Christopherson /* 26216e6ec584SSean Christopherson * Note, this check is intentionally soft, it only guarantees that one 26226e6ec584SSean Christopherson * page is available, while the caller may end up allocating as many as 26236e6ec584SSean Christopherson * four pages, e.g. for PAE roots or for 5-level paging. Temporarily 26246e6ec584SSean Christopherson * exceeding the (arbitrary by default) limit will not harm the host, 2625c4342633SIngo Molnar * being too aggressive may unnecessarily kill the guest, and getting an 26266e6ec584SSean Christopherson * exact count is far more trouble than it's worth, especially in the 26276e6ec584SSean Christopherson * page fault paths. 26286e6ec584SSean Christopherson */ 2629ba7888ddSSean Christopherson if (!kvm_mmu_available_pages(vcpu->kvm)) 2630ba7888ddSSean Christopherson return -ENOSPC; 2631ba7888ddSSean Christopherson return 0; 2632ba7888ddSSean Christopherson } 2633ba7888ddSSean Christopherson 2634c50d8ae3SPaolo Bonzini /* 2635c50d8ae3SPaolo Bonzini * Changing the number of mmu pages allocated to the vm 2636c50d8ae3SPaolo Bonzini * Note: if goal_nr_mmu_pages is too small, you will get dead lock 2637c50d8ae3SPaolo Bonzini */ 2638c50d8ae3SPaolo Bonzini void kvm_mmu_change_mmu_pages(struct kvm *kvm, unsigned long goal_nr_mmu_pages) 2639c50d8ae3SPaolo Bonzini { 2640531810caSBen Gardon write_lock(&kvm->mmu_lock); 2641c50d8ae3SPaolo Bonzini 2642c50d8ae3SPaolo Bonzini if (kvm->arch.n_used_mmu_pages > goal_nr_mmu_pages) { 26436b82ef2cSSean Christopherson kvm_mmu_zap_oldest_mmu_pages(kvm, kvm->arch.n_used_mmu_pages - 26446b82ef2cSSean Christopherson goal_nr_mmu_pages); 2645c50d8ae3SPaolo Bonzini 2646c50d8ae3SPaolo Bonzini goal_nr_mmu_pages = kvm->arch.n_used_mmu_pages; 2647c50d8ae3SPaolo Bonzini } 2648c50d8ae3SPaolo Bonzini 2649c50d8ae3SPaolo Bonzini kvm->arch.n_max_mmu_pages = goal_nr_mmu_pages; 2650c50d8ae3SPaolo Bonzini 2651531810caSBen Gardon write_unlock(&kvm->mmu_lock); 2652c50d8ae3SPaolo Bonzini } 2653c50d8ae3SPaolo Bonzini 2654c50d8ae3SPaolo Bonzini int kvm_mmu_unprotect_page(struct kvm *kvm, gfn_t gfn) 2655c50d8ae3SPaolo Bonzini { 2656c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 2657c50d8ae3SPaolo Bonzini LIST_HEAD(invalid_list); 2658c50d8ae3SPaolo Bonzini int r; 2659c50d8ae3SPaolo Bonzini 2660c50d8ae3SPaolo Bonzini pgprintk("%s: looking for gfn %llx\n", __func__, gfn); 2661c50d8ae3SPaolo Bonzini r = 0; 2662531810caSBen Gardon write_lock(&kvm->mmu_lock); 2663767d8d8dSLai Jiangshan for_each_gfn_valid_sp_with_gptes(kvm, sp, gfn) { 2664c50d8ae3SPaolo Bonzini pgprintk("%s: gfn %llx role %x\n", __func__, gfn, 2665c50d8ae3SPaolo Bonzini sp->role.word); 2666c50d8ae3SPaolo Bonzini r = 1; 2667c50d8ae3SPaolo Bonzini kvm_mmu_prepare_zap_page(kvm, sp, &invalid_list); 2668c50d8ae3SPaolo Bonzini } 2669c50d8ae3SPaolo Bonzini kvm_mmu_commit_zap_page(kvm, &invalid_list); 2670531810caSBen Gardon write_unlock(&kvm->mmu_lock); 2671c50d8ae3SPaolo Bonzini 2672c50d8ae3SPaolo Bonzini return r; 2673c50d8ae3SPaolo Bonzini } 267496ad91aeSSean Christopherson 267596ad91aeSSean Christopherson static int kvm_mmu_unprotect_page_virt(struct kvm_vcpu *vcpu, gva_t gva) 267696ad91aeSSean Christopherson { 267796ad91aeSSean Christopherson gpa_t gpa; 267896ad91aeSSean Christopherson int r; 267996ad91aeSSean Christopherson 2680347a0d0dSPaolo Bonzini if (vcpu->arch.mmu->root_role.direct) 268196ad91aeSSean Christopherson return 0; 268296ad91aeSSean Christopherson 268396ad91aeSSean Christopherson gpa = kvm_mmu_gva_to_gpa_read(vcpu, gva, NULL); 268496ad91aeSSean Christopherson 268596ad91aeSSean Christopherson r = kvm_mmu_unprotect_page(vcpu->kvm, gpa >> PAGE_SHIFT); 268696ad91aeSSean Christopherson 268796ad91aeSSean Christopherson return r; 268896ad91aeSSean Christopherson } 2689c50d8ae3SPaolo Bonzini 26904d78d0b3SBen Gardon static void kvm_unsync_page(struct kvm *kvm, struct kvm_mmu_page *sp) 2691c50d8ae3SPaolo Bonzini { 2692c50d8ae3SPaolo Bonzini trace_kvm_mmu_unsync_page(sp); 26934d78d0b3SBen Gardon ++kvm->stat.mmu_unsync; 2694c50d8ae3SPaolo Bonzini sp->unsync = 1; 2695c50d8ae3SPaolo Bonzini 2696c50d8ae3SPaolo Bonzini kvm_mmu_mark_parents_unsync(sp); 2697c50d8ae3SPaolo Bonzini } 2698c50d8ae3SPaolo Bonzini 26990337f585SSean Christopherson /* 27000337f585SSean Christopherson * Attempt to unsync any shadow pages that can be reached by the specified gfn, 27010337f585SSean Christopherson * KVM is creating a writable mapping for said gfn. Returns 0 if all pages 27020337f585SSean Christopherson * were marked unsync (or if there is no shadow page), -EPERM if the SPTE must 27030337f585SSean Christopherson * be write-protected. 27040337f585SSean Christopherson */ 27058283e36aSBen Gardon int mmu_try_to_unsync_pages(struct kvm *kvm, const struct kvm_memory_slot *slot, 27062839180cSPaolo Bonzini gfn_t gfn, bool can_unsync, bool prefetch) 2707c50d8ae3SPaolo Bonzini { 2708c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 2709ce25681dSSean Christopherson bool locked = false; 2710c50d8ae3SPaolo Bonzini 27110337f585SSean Christopherson /* 27120337f585SSean Christopherson * Force write-protection if the page is being tracked. Note, the page 27130337f585SSean Christopherson * track machinery is used to write-protect upper-level shadow pages, 27140337f585SSean Christopherson * i.e. this guards the role.level == 4K assertion below! 27150337f585SSean Christopherson */ 27164d78d0b3SBen Gardon if (kvm_slot_page_track_is_active(kvm, slot, gfn, KVM_PAGE_TRACK_WRITE)) 27170337f585SSean Christopherson return -EPERM; 2718c50d8ae3SPaolo Bonzini 27190337f585SSean Christopherson /* 27200337f585SSean Christopherson * The page is not write-tracked, mark existing shadow pages unsync 27210337f585SSean Christopherson * unless KVM is synchronizing an unsync SP (can_unsync = false). In 27220337f585SSean Christopherson * that case, KVM must complete emulation of the guest TLB flush before 27230337f585SSean Christopherson * allowing shadow pages to become unsync (writable by the guest). 27240337f585SSean Christopherson */ 2725767d8d8dSLai Jiangshan for_each_gfn_valid_sp_with_gptes(kvm, sp, gfn) { 2726c50d8ae3SPaolo Bonzini if (!can_unsync) 27270337f585SSean Christopherson return -EPERM; 2728c50d8ae3SPaolo Bonzini 2729c50d8ae3SPaolo Bonzini if (sp->unsync) 2730c50d8ae3SPaolo Bonzini continue; 2731c50d8ae3SPaolo Bonzini 27322839180cSPaolo Bonzini if (prefetch) 2733f1c4a88cSLai Jiangshan return -EEXIST; 2734f1c4a88cSLai Jiangshan 2735ce25681dSSean Christopherson /* 2736ce25681dSSean Christopherson * TDP MMU page faults require an additional spinlock as they 2737ce25681dSSean Christopherson * run with mmu_lock held for read, not write, and the unsync 2738ce25681dSSean Christopherson * logic is not thread safe. Take the spinklock regardless of 2739ce25681dSSean Christopherson * the MMU type to avoid extra conditionals/parameters, there's 2740ce25681dSSean Christopherson * no meaningful penalty if mmu_lock is held for write. 2741ce25681dSSean Christopherson */ 2742ce25681dSSean Christopherson if (!locked) { 2743ce25681dSSean Christopherson locked = true; 27444d78d0b3SBen Gardon spin_lock(&kvm->arch.mmu_unsync_pages_lock); 2745ce25681dSSean Christopherson 2746ce25681dSSean Christopherson /* 2747ce25681dSSean Christopherson * Recheck after taking the spinlock, a different vCPU 2748ce25681dSSean Christopherson * may have since marked the page unsync. A false 2749ce25681dSSean Christopherson * positive on the unprotected check above is not 2750ce25681dSSean Christopherson * possible as clearing sp->unsync _must_ hold mmu_lock 2751ce25681dSSean Christopherson * for write, i.e. unsync cannot transition from 0->1 2752ce25681dSSean Christopherson * while this CPU holds mmu_lock for read (or write). 2753ce25681dSSean Christopherson */ 2754ce25681dSSean Christopherson if (READ_ONCE(sp->unsync)) 2755ce25681dSSean Christopherson continue; 2756ce25681dSSean Christopherson } 2757ce25681dSSean Christopherson 27583bae0459SSean Christopherson WARN_ON(sp->role.level != PG_LEVEL_4K); 27594d78d0b3SBen Gardon kvm_unsync_page(kvm, sp); 2760c50d8ae3SPaolo Bonzini } 2761ce25681dSSean Christopherson if (locked) 27624d78d0b3SBen Gardon spin_unlock(&kvm->arch.mmu_unsync_pages_lock); 2763c50d8ae3SPaolo Bonzini 2764c50d8ae3SPaolo Bonzini /* 2765c50d8ae3SPaolo Bonzini * We need to ensure that the marking of unsync pages is visible 2766c50d8ae3SPaolo Bonzini * before the SPTE is updated to allow writes because 2767c50d8ae3SPaolo Bonzini * kvm_mmu_sync_roots() checks the unsync flags without holding 2768c50d8ae3SPaolo Bonzini * the MMU lock and so can race with this. If the SPTE was updated 2769c50d8ae3SPaolo Bonzini * before the page had been marked as unsync-ed, something like the 2770c50d8ae3SPaolo Bonzini * following could happen: 2771c50d8ae3SPaolo Bonzini * 2772c50d8ae3SPaolo Bonzini * CPU 1 CPU 2 2773c50d8ae3SPaolo Bonzini * --------------------------------------------------------------------- 2774c50d8ae3SPaolo Bonzini * 1.2 Host updates SPTE 2775c50d8ae3SPaolo Bonzini * to be writable 2776c50d8ae3SPaolo Bonzini * 2.1 Guest writes a GPTE for GVA X. 2777c50d8ae3SPaolo Bonzini * (GPTE being in the guest page table shadowed 2778c50d8ae3SPaolo Bonzini * by the SP from CPU 1.) 2779c50d8ae3SPaolo Bonzini * This reads SPTE during the page table walk. 2780c50d8ae3SPaolo Bonzini * Since SPTE.W is read as 1, there is no 2781c50d8ae3SPaolo Bonzini * fault. 2782c50d8ae3SPaolo Bonzini * 2783c50d8ae3SPaolo Bonzini * 2.2 Guest issues TLB flush. 2784c50d8ae3SPaolo Bonzini * That causes a VM Exit. 2785c50d8ae3SPaolo Bonzini * 27860337f585SSean Christopherson * 2.3 Walking of unsync pages sees sp->unsync is 27870337f585SSean Christopherson * false and skips the page. 2788c50d8ae3SPaolo Bonzini * 2789c50d8ae3SPaolo Bonzini * 2.4 Guest accesses GVA X. 2790c50d8ae3SPaolo Bonzini * Since the mapping in the SP was not updated, 2791c50d8ae3SPaolo Bonzini * so the old mapping for GVA X incorrectly 2792c50d8ae3SPaolo Bonzini * gets used. 2793c50d8ae3SPaolo Bonzini * 1.1 Host marks SP 2794c50d8ae3SPaolo Bonzini * as unsync 2795c50d8ae3SPaolo Bonzini * (sp->unsync = true) 2796c50d8ae3SPaolo Bonzini * 2797c50d8ae3SPaolo Bonzini * The write barrier below ensures that 1.1 happens before 1.2 and thus 2798264d3dc1SLai Jiangshan * the situation in 2.4 does not arise. It pairs with the read barrier 2799264d3dc1SLai Jiangshan * in is_unsync_root(), placed between 2.1's load of SPTE.W and 2.3. 2800c50d8ae3SPaolo Bonzini */ 2801c50d8ae3SPaolo Bonzini smp_wmb(); 2802c50d8ae3SPaolo Bonzini 28030337f585SSean Christopherson return 0; 2804c50d8ae3SPaolo Bonzini } 2805c50d8ae3SPaolo Bonzini 28068a9f566aSDavid Matlack static int mmu_set_spte(struct kvm_vcpu *vcpu, struct kvm_memory_slot *slot, 28078a9f566aSDavid Matlack u64 *sptep, unsigned int pte_access, gfn_t gfn, 2808a12f4381SPaolo Bonzini kvm_pfn_t pfn, struct kvm_page_fault *fault) 2809799a4190SBen Gardon { 2810d786c778SPaolo Bonzini struct kvm_mmu_page *sp = sptep_to_sp(sptep); 2811eb5cd7ffSPaolo Bonzini int level = sp->role.level; 2812c50d8ae3SPaolo Bonzini int was_rmapped = 0; 2813c4371c2aSSean Christopherson int ret = RET_PF_FIXED; 2814c50d8ae3SPaolo Bonzini bool flush = false; 2815ad67e480SPaolo Bonzini bool wrprot; 2816d786c778SPaolo Bonzini u64 spte; 2817c50d8ae3SPaolo Bonzini 2818a12f4381SPaolo Bonzini /* Prefetching always gets a writable pfn. */ 2819a12f4381SPaolo Bonzini bool host_writable = !fault || fault->map_writable; 28202839180cSPaolo Bonzini bool prefetch = !fault || fault->prefetch; 2821a12f4381SPaolo Bonzini bool write_fault = fault && fault->write; 2822c50d8ae3SPaolo Bonzini 2823c50d8ae3SPaolo Bonzini pgprintk("%s: spte %llx write_fault %d gfn %llx\n", __func__, 2824c50d8ae3SPaolo Bonzini *sptep, write_fault, gfn); 2825c50d8ae3SPaolo Bonzini 2826a54aa15cSSean Christopherson if (unlikely(is_noslot_pfn(pfn))) { 28271075d41eSSean Christopherson vcpu->stat.pf_mmio_spte_created++; 2828a54aa15cSSean Christopherson mark_mmio_spte(vcpu, sptep, gfn, pte_access); 2829a54aa15cSSean Christopherson return RET_PF_EMULATE; 2830a54aa15cSSean Christopherson } 2831a54aa15cSSean Christopherson 2832c50d8ae3SPaolo Bonzini if (is_shadow_present_pte(*sptep)) { 2833c50d8ae3SPaolo Bonzini /* 2834c50d8ae3SPaolo Bonzini * If we overwrite a PTE page pointer with a 2MB PMD, unlink 2835c50d8ae3SPaolo Bonzini * the parent of the now unreachable PTE. 2836c50d8ae3SPaolo Bonzini */ 28373bae0459SSean Christopherson if (level > PG_LEVEL_4K && !is_large_pte(*sptep)) { 2838c50d8ae3SPaolo Bonzini struct kvm_mmu_page *child; 2839c50d8ae3SPaolo Bonzini u64 pte = *sptep; 2840c50d8ae3SPaolo Bonzini 2841*5e3edd7eSSean Christopherson child = spte_to_child_sp(pte); 2842c50d8ae3SPaolo Bonzini drop_parent_pte(child, sptep); 2843c50d8ae3SPaolo Bonzini flush = true; 2844c50d8ae3SPaolo Bonzini } else if (pfn != spte_to_pfn(*sptep)) { 2845c50d8ae3SPaolo Bonzini pgprintk("hfn old %llx new %llx\n", 2846c50d8ae3SPaolo Bonzini spte_to_pfn(*sptep), pfn); 2847c50d8ae3SPaolo Bonzini drop_spte(vcpu->kvm, sptep); 2848c50d8ae3SPaolo Bonzini flush = true; 2849c50d8ae3SPaolo Bonzini } else 2850c50d8ae3SPaolo Bonzini was_rmapped = 1; 2851c50d8ae3SPaolo Bonzini } 2852c50d8ae3SPaolo Bonzini 28532839180cSPaolo Bonzini wrprot = make_spte(vcpu, sp, slot, pte_access, gfn, pfn, *sptep, prefetch, 28547158bee4SPaolo Bonzini true, host_writable, &spte); 2855d786c778SPaolo Bonzini 2856d786c778SPaolo Bonzini if (*sptep == spte) { 2857d786c778SPaolo Bonzini ret = RET_PF_SPURIOUS; 2858d786c778SPaolo Bonzini } else { 2859d786c778SPaolo Bonzini flush |= mmu_spte_update(sptep, spte); 28605959ff4aSMaxim Levitsky trace_kvm_mmu_set_spte(level, gfn, sptep); 2861c50d8ae3SPaolo Bonzini } 2862c50d8ae3SPaolo Bonzini 2863ad67e480SPaolo Bonzini if (wrprot) { 2864c50d8ae3SPaolo Bonzini if (write_fault) 2865c50d8ae3SPaolo Bonzini ret = RET_PF_EMULATE; 2866c50d8ae3SPaolo Bonzini } 2867c50d8ae3SPaolo Bonzini 2868d786c778SPaolo Bonzini if (flush) 2869c50d8ae3SPaolo Bonzini kvm_flush_remote_tlbs_with_address(vcpu->kvm, gfn, 2870c50d8ae3SPaolo Bonzini KVM_PAGES_PER_HPAGE(level)); 2871c50d8ae3SPaolo Bonzini 2872c50d8ae3SPaolo Bonzini pgprintk("%s: setting spte %llx\n", __func__, *sptep); 2873c50d8ae3SPaolo Bonzini 2874c50d8ae3SPaolo Bonzini if (!was_rmapped) { 2875d786c778SPaolo Bonzini WARN_ON_ONCE(ret == RET_PF_SPURIOUS); 28766a97575dSDavid Matlack rmap_add(vcpu, slot, sptep, gfn, pte_access); 28776a97575dSDavid Matlack } else { 28786a97575dSDavid Matlack /* Already rmapped but the pte_access bits may have changed. */ 287979e48cecSSean Christopherson kvm_mmu_page_set_access(sp, spte_index(sptep), pte_access); 2880c50d8ae3SPaolo Bonzini } 2881c50d8ae3SPaolo Bonzini 2882c50d8ae3SPaolo Bonzini return ret; 2883c50d8ae3SPaolo Bonzini } 2884c50d8ae3SPaolo Bonzini 2885c50d8ae3SPaolo Bonzini static int direct_pte_prefetch_many(struct kvm_vcpu *vcpu, 2886c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp, 2887c50d8ae3SPaolo Bonzini u64 *start, u64 *end) 2888c50d8ae3SPaolo Bonzini { 2889c50d8ae3SPaolo Bonzini struct page *pages[PTE_PREFETCH_NUM]; 2890c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot; 28910a2b64c5SBen Gardon unsigned int access = sp->role.access; 2892c50d8ae3SPaolo Bonzini int i, ret; 2893c50d8ae3SPaolo Bonzini gfn_t gfn; 2894c50d8ae3SPaolo Bonzini 289579e48cecSSean Christopherson gfn = kvm_mmu_page_get_gfn(sp, spte_index(start)); 2896c50d8ae3SPaolo Bonzini slot = gfn_to_memslot_dirty_bitmap(vcpu, gfn, access & ACC_WRITE_MASK); 2897c50d8ae3SPaolo Bonzini if (!slot) 2898c50d8ae3SPaolo Bonzini return -1; 2899c50d8ae3SPaolo Bonzini 2900c50d8ae3SPaolo Bonzini ret = gfn_to_page_many_atomic(slot, gfn, pages, end - start); 2901c50d8ae3SPaolo Bonzini if (ret <= 0) 2902c50d8ae3SPaolo Bonzini return -1; 2903c50d8ae3SPaolo Bonzini 2904c50d8ae3SPaolo Bonzini for (i = 0; i < ret; i++, gfn++, start++) { 29058a9f566aSDavid Matlack mmu_set_spte(vcpu, slot, start, access, gfn, 2906a12f4381SPaolo Bonzini page_to_pfn(pages[i]), NULL); 2907c50d8ae3SPaolo Bonzini put_page(pages[i]); 2908c50d8ae3SPaolo Bonzini } 2909c50d8ae3SPaolo Bonzini 2910c50d8ae3SPaolo Bonzini return 0; 2911c50d8ae3SPaolo Bonzini } 2912c50d8ae3SPaolo Bonzini 2913c50d8ae3SPaolo Bonzini static void __direct_pte_prefetch(struct kvm_vcpu *vcpu, 2914c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp, u64 *sptep) 2915c50d8ae3SPaolo Bonzini { 2916c50d8ae3SPaolo Bonzini u64 *spte, *start = NULL; 2917c50d8ae3SPaolo Bonzini int i; 2918c50d8ae3SPaolo Bonzini 2919c50d8ae3SPaolo Bonzini WARN_ON(!sp->role.direct); 2920c50d8ae3SPaolo Bonzini 292179e48cecSSean Christopherson i = spte_index(sptep) & ~(PTE_PREFETCH_NUM - 1); 2922c50d8ae3SPaolo Bonzini spte = sp->spt + i; 2923c50d8ae3SPaolo Bonzini 2924c50d8ae3SPaolo Bonzini for (i = 0; i < PTE_PREFETCH_NUM; i++, spte++) { 2925c50d8ae3SPaolo Bonzini if (is_shadow_present_pte(*spte) || spte == sptep) { 2926c50d8ae3SPaolo Bonzini if (!start) 2927c50d8ae3SPaolo Bonzini continue; 2928c50d8ae3SPaolo Bonzini if (direct_pte_prefetch_many(vcpu, sp, start, spte) < 0) 2929c6cecc4bSSean Christopherson return; 2930c50d8ae3SPaolo Bonzini start = NULL; 2931c50d8ae3SPaolo Bonzini } else if (!start) 2932c50d8ae3SPaolo Bonzini start = spte; 2933c50d8ae3SPaolo Bonzini } 2934c6cecc4bSSean Christopherson if (start) 2935c6cecc4bSSean Christopherson direct_pte_prefetch_many(vcpu, sp, start, spte); 2936c50d8ae3SPaolo Bonzini } 2937c50d8ae3SPaolo Bonzini 2938c50d8ae3SPaolo Bonzini static void direct_pte_prefetch(struct kvm_vcpu *vcpu, u64 *sptep) 2939c50d8ae3SPaolo Bonzini { 2940c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 2941c50d8ae3SPaolo Bonzini 294257354682SSean Christopherson sp = sptep_to_sp(sptep); 2943c50d8ae3SPaolo Bonzini 2944c50d8ae3SPaolo Bonzini /* 2945c50d8ae3SPaolo Bonzini * Without accessed bits, there's no way to distinguish between 2946c50d8ae3SPaolo Bonzini * actually accessed translations and prefetched, so disable pte 2947c50d8ae3SPaolo Bonzini * prefetch if accessed bits aren't available. 2948c50d8ae3SPaolo Bonzini */ 2949c50d8ae3SPaolo Bonzini if (sp_ad_disabled(sp)) 2950c50d8ae3SPaolo Bonzini return; 2951c50d8ae3SPaolo Bonzini 29523bae0459SSean Christopherson if (sp->role.level > PG_LEVEL_4K) 2953c50d8ae3SPaolo Bonzini return; 2954c50d8ae3SPaolo Bonzini 29554a42d848SDavid Stevens /* 29564a42d848SDavid Stevens * If addresses are being invalidated, skip prefetching to avoid 29574a42d848SDavid Stevens * accidentally prefetching those addresses. 29584a42d848SDavid Stevens */ 295920ec3ebdSChao Peng if (unlikely(vcpu->kvm->mmu_invalidate_in_progress)) 29604a42d848SDavid Stevens return; 29614a42d848SDavid Stevens 2962c50d8ae3SPaolo Bonzini __direct_pte_prefetch(vcpu, sp, sptep); 2963c50d8ae3SPaolo Bonzini } 2964c50d8ae3SPaolo Bonzini 296565e3b446SSean Christopherson /* 296665e3b446SSean Christopherson * Lookup the mapping level for @gfn in the current mm. 296765e3b446SSean Christopherson * 296865e3b446SSean Christopherson * WARNING! Use of host_pfn_mapping_level() requires the caller and the end 296965e3b446SSean Christopherson * consumer to be tied into KVM's handlers for MMU notifier events! 297065e3b446SSean Christopherson * 297165e3b446SSean Christopherson * There are several ways to safely use this helper: 297265e3b446SSean Christopherson * 297320ec3ebdSChao Peng * - Check mmu_invalidate_retry_hva() after grabbing the mapping level, before 297465e3b446SSean Christopherson * consuming it. In this case, mmu_lock doesn't need to be held during the 297565e3b446SSean Christopherson * lookup, but it does need to be held while checking the MMU notifier. 297665e3b446SSean Christopherson * 297765e3b446SSean Christopherson * - Hold mmu_lock AND ensure there is no in-progress MMU notifier invalidation 297865e3b446SSean Christopherson * event for the hva. This can be done by explicit checking the MMU notifier 297965e3b446SSean Christopherson * or by ensuring that KVM already has a valid mapping that covers the hva. 298065e3b446SSean Christopherson * 298165e3b446SSean Christopherson * - Do not use the result to install new mappings, e.g. use the host mapping 298265e3b446SSean Christopherson * level only to decide whether or not to zap an entry. In this case, it's 298365e3b446SSean Christopherson * not required to hold mmu_lock (though it's highly likely the caller will 298465e3b446SSean Christopherson * want to hold mmu_lock anyways, e.g. to modify SPTEs). 298565e3b446SSean Christopherson * 298665e3b446SSean Christopherson * Note! The lookup can still race with modifications to host page tables, but 298765e3b446SSean Christopherson * the above "rules" ensure KVM will not _consume_ the result of the walk if a 298865e3b446SSean Christopherson * race with the primary MMU occurs. 298965e3b446SSean Christopherson */ 2990a8ac499bSSean Christopherson static int host_pfn_mapping_level(struct kvm *kvm, gfn_t gfn, 29918ca6f063SBen Gardon const struct kvm_memory_slot *slot) 2992db543216SSean Christopherson { 2993284dc493SSean Christopherson int level = PG_LEVEL_4K; 2994db543216SSean Christopherson unsigned long hva; 299544187235SMingwei Zhang unsigned long flags; 299644187235SMingwei Zhang pgd_t pgd; 299744187235SMingwei Zhang p4d_t p4d; 299844187235SMingwei Zhang pud_t pud; 299944187235SMingwei Zhang pmd_t pmd; 3000db543216SSean Christopherson 30015d49f08cSSean Christopherson /* 3002293e306eSSean Christopherson * Note, using the already-retrieved memslot and __gfn_to_hva_memslot() 3003293e306eSSean Christopherson * is not solely for performance, it's also necessary to avoid the 3004293e306eSSean Christopherson * "writable" check in __gfn_to_hva_many(), which will always fail on 3005293e306eSSean Christopherson * read-only memslots due to gfn_to_hva() assuming writes. Earlier 3006293e306eSSean Christopherson * page fault steps have already verified the guest isn't writing a 3007293e306eSSean Christopherson * read-only memslot. 3008293e306eSSean Christopherson */ 3009db543216SSean Christopherson hva = __gfn_to_hva_memslot(slot, gfn); 3010db543216SSean Christopherson 301144187235SMingwei Zhang /* 301265e3b446SSean Christopherson * Disable IRQs to prevent concurrent tear down of host page tables, 301365e3b446SSean Christopherson * e.g. if the primary MMU promotes a P*D to a huge page and then frees 301465e3b446SSean Christopherson * the original page table. 301544187235SMingwei Zhang */ 301644187235SMingwei Zhang local_irq_save(flags); 3017db543216SSean Christopherson 301865e3b446SSean Christopherson /* 301965e3b446SSean Christopherson * Read each entry once. As above, a non-leaf entry can be promoted to 302065e3b446SSean Christopherson * a huge page _during_ this walk. Re-reading the entry could send the 302165e3b446SSean Christopherson * walk into the weeks, e.g. p*d_large() returns false (sees the old 302265e3b446SSean Christopherson * value) and then p*d_offset() walks into the target huge page instead 302365e3b446SSean Christopherson * of the old page table (sees the new value). 302465e3b446SSean Christopherson */ 302544187235SMingwei Zhang pgd = READ_ONCE(*pgd_offset(kvm->mm, hva)); 302644187235SMingwei Zhang if (pgd_none(pgd)) 302744187235SMingwei Zhang goto out; 302844187235SMingwei Zhang 302944187235SMingwei Zhang p4d = READ_ONCE(*p4d_offset(&pgd, hva)); 303044187235SMingwei Zhang if (p4d_none(p4d) || !p4d_present(p4d)) 303144187235SMingwei Zhang goto out; 303244187235SMingwei Zhang 303344187235SMingwei Zhang pud = READ_ONCE(*pud_offset(&p4d, hva)); 303444187235SMingwei Zhang if (pud_none(pud) || !pud_present(pud)) 303544187235SMingwei Zhang goto out; 303644187235SMingwei Zhang 303744187235SMingwei Zhang if (pud_large(pud)) { 303844187235SMingwei Zhang level = PG_LEVEL_1G; 303944187235SMingwei Zhang goto out; 304044187235SMingwei Zhang } 304144187235SMingwei Zhang 304244187235SMingwei Zhang pmd = READ_ONCE(*pmd_offset(&pud, hva)); 304344187235SMingwei Zhang if (pmd_none(pmd) || !pmd_present(pmd)) 304444187235SMingwei Zhang goto out; 304544187235SMingwei Zhang 304644187235SMingwei Zhang if (pmd_large(pmd)) 304744187235SMingwei Zhang level = PG_LEVEL_2M; 304844187235SMingwei Zhang 304944187235SMingwei Zhang out: 305044187235SMingwei Zhang local_irq_restore(flags); 3051db543216SSean Christopherson return level; 3052db543216SSean Christopherson } 3053db543216SSean Christopherson 30548ca6f063SBen Gardon int kvm_mmu_max_mapping_level(struct kvm *kvm, 30558ca6f063SBen Gardon const struct kvm_memory_slot *slot, gfn_t gfn, 3056a8ac499bSSean Christopherson int max_level) 30571b6d9d9eSSean Christopherson { 30581b6d9d9eSSean Christopherson struct kvm_lpage_info *linfo; 3059ec607a56SPaolo Bonzini int host_level; 30601b6d9d9eSSean Christopherson 30611b6d9d9eSSean Christopherson max_level = min(max_level, max_huge_page_level); 30621b6d9d9eSSean Christopherson for ( ; max_level > PG_LEVEL_4K; max_level--) { 30631b6d9d9eSSean Christopherson linfo = lpage_info_slot(gfn, slot, max_level); 30641b6d9d9eSSean Christopherson if (!linfo->disallow_lpage) 30651b6d9d9eSSean Christopherson break; 30661b6d9d9eSSean Christopherson } 30671b6d9d9eSSean Christopherson 30681b6d9d9eSSean Christopherson if (max_level == PG_LEVEL_4K) 30691b6d9d9eSSean Christopherson return PG_LEVEL_4K; 30701b6d9d9eSSean Christopherson 3071a8ac499bSSean Christopherson host_level = host_pfn_mapping_level(kvm, gfn, slot); 3072ec607a56SPaolo Bonzini return min(host_level, max_level); 30731b6d9d9eSSean Christopherson } 30741b6d9d9eSSean Christopherson 307573a3c659SPaolo Bonzini void kvm_mmu_hugepage_adjust(struct kvm_vcpu *vcpu, struct kvm_page_fault *fault) 30760885904dSSean Christopherson { 3077e710c5f6SDavid Matlack struct kvm_memory_slot *slot = fault->slot; 307817eff019SSean Christopherson kvm_pfn_t mask; 30790885904dSSean Christopherson 308073a3c659SPaolo Bonzini fault->huge_page_disallowed = fault->exec && fault->nx_huge_page_workaround_enabled; 30813cf06612SSean Christopherson 308273a3c659SPaolo Bonzini if (unlikely(fault->max_level == PG_LEVEL_4K)) 308373a3c659SPaolo Bonzini return; 308417eff019SSean Christopherson 30855d49f08cSSean Christopherson if (is_error_noslot_pfn(fault->pfn)) 308673a3c659SPaolo Bonzini return; 308717eff019SSean Christopherson 3088e710c5f6SDavid Matlack if (kvm_slot_dirty_track_enabled(slot)) 308973a3c659SPaolo Bonzini return; 3090293e306eSSean Christopherson 30913cf06612SSean Christopherson /* 30923cf06612SSean Christopherson * Enforce the iTLB multihit workaround after capturing the requested 30933cf06612SSean Christopherson * level, which will be used to do precise, accurate accounting. 30943cf06612SSean Christopherson */ 309573a3c659SPaolo Bonzini fault->req_level = kvm_mmu_max_mapping_level(vcpu->kvm, slot, 3096a8ac499bSSean Christopherson fault->gfn, fault->max_level); 309773a3c659SPaolo Bonzini if (fault->req_level == PG_LEVEL_4K || fault->huge_page_disallowed) 309873a3c659SPaolo Bonzini return; 30994cd071d1SSean Christopherson 31000885904dSSean Christopherson /* 310120ec3ebdSChao Peng * mmu_invalidate_retry() was successful and mmu_lock is held, so 31024cd071d1SSean Christopherson * the pmd can't be split from under us. 31030885904dSSean Christopherson */ 310473a3c659SPaolo Bonzini fault->goal_level = fault->req_level; 310573a3c659SPaolo Bonzini mask = KVM_PAGES_PER_HPAGE(fault->goal_level) - 1; 310673a3c659SPaolo Bonzini VM_BUG_ON((fault->gfn & mask) != (fault->pfn & mask)); 310773a3c659SPaolo Bonzini fault->pfn &= ~mask; 31080885904dSSean Christopherson } 31090885904dSSean Christopherson 3110536f0e6aSPaolo Bonzini void disallowed_hugepage_adjust(struct kvm_page_fault *fault, u64 spte, int cur_level) 3111c50d8ae3SPaolo Bonzini { 3112536f0e6aSPaolo Bonzini if (cur_level > PG_LEVEL_4K && 3113536f0e6aSPaolo Bonzini cur_level == fault->goal_level && 3114c50d8ae3SPaolo Bonzini is_shadow_present_pte(spte) && 3115c50d8ae3SPaolo Bonzini !is_large_pte(spte)) { 3116c50d8ae3SPaolo Bonzini /* 3117c50d8ae3SPaolo Bonzini * A small SPTE exists for this pfn, but FNAME(fetch) 3118c50d8ae3SPaolo Bonzini * and __direct_map would like to create a large PTE 3119c50d8ae3SPaolo Bonzini * instead: just force them to go down another level, 3120c50d8ae3SPaolo Bonzini * patching back for them into pfn the next 9 bits of 3121c50d8ae3SPaolo Bonzini * the address. 3122c50d8ae3SPaolo Bonzini */ 3123536f0e6aSPaolo Bonzini u64 page_mask = KVM_PAGES_PER_HPAGE(cur_level) - 3124536f0e6aSPaolo Bonzini KVM_PAGES_PER_HPAGE(cur_level - 1); 3125536f0e6aSPaolo Bonzini fault->pfn |= fault->gfn & page_mask; 3126536f0e6aSPaolo Bonzini fault->goal_level--; 3127c50d8ae3SPaolo Bonzini } 3128c50d8ae3SPaolo Bonzini } 3129c50d8ae3SPaolo Bonzini 313043b74355SPaolo Bonzini static int __direct_map(struct kvm_vcpu *vcpu, struct kvm_page_fault *fault) 3131c50d8ae3SPaolo Bonzini { 3132c50d8ae3SPaolo Bonzini struct kvm_shadow_walk_iterator it; 3133c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 313473a3c659SPaolo Bonzini int ret; 313543b74355SPaolo Bonzini gfn_t base_gfn = fault->gfn; 3136c50d8ae3SPaolo Bonzini 313773a3c659SPaolo Bonzini kvm_mmu_hugepage_adjust(vcpu, fault); 31384cd071d1SSean Christopherson 3139f0066d94SPaolo Bonzini trace_kvm_mmu_spte_requested(fault); 314043b74355SPaolo Bonzini for_each_shadow_entry(vcpu, fault->addr, it) { 3141c50d8ae3SPaolo Bonzini /* 3142c50d8ae3SPaolo Bonzini * We cannot overwrite existing page tables with an NX 3143c50d8ae3SPaolo Bonzini * large page, as the leaf could be executable. 3144c50d8ae3SPaolo Bonzini */ 314573a3c659SPaolo Bonzini if (fault->nx_huge_page_workaround_enabled) 3146536f0e6aSPaolo Bonzini disallowed_hugepage_adjust(fault, *it.sptep, it.level); 3147c50d8ae3SPaolo Bonzini 314843b74355SPaolo Bonzini base_gfn = fault->gfn & ~(KVM_PAGES_PER_HPAGE(it.level) - 1); 314973a3c659SPaolo Bonzini if (it.level == fault->goal_level) 3150c50d8ae3SPaolo Bonzini break; 3151c50d8ae3SPaolo Bonzini 31522e65e842SDavid Matlack sp = kvm_mmu_get_child_sp(vcpu, it.sptep, base_gfn, true, ACC_ALL); 31530cd8dc73SPaolo Bonzini if (sp == ERR_PTR(-EEXIST)) 31540cd8dc73SPaolo Bonzini continue; 3155c50d8ae3SPaolo Bonzini 3156c50d8ae3SPaolo Bonzini link_shadow_page(vcpu, it.sptep, sp); 3157b5b0977fSSean Christopherson if (fault->huge_page_disallowed) 315855c510e2SSean Christopherson account_nx_huge_page(vcpu->kvm, sp, 3159428e9216SSean Christopherson fault->req_level >= it.level); 3160c50d8ae3SPaolo Bonzini } 3161c50d8ae3SPaolo Bonzini 3162b1a429fbSSean Christopherson if (WARN_ON_ONCE(it.level != fault->goal_level)) 3163b1a429fbSSean Christopherson return -EFAULT; 3164b1a429fbSSean Christopherson 31658a9f566aSDavid Matlack ret = mmu_set_spte(vcpu, fault->slot, it.sptep, ACC_ALL, 3166a12f4381SPaolo Bonzini base_gfn, fault->pfn, fault); 316712703759SSean Christopherson if (ret == RET_PF_SPURIOUS) 316812703759SSean Christopherson return ret; 316912703759SSean Christopherson 3170c50d8ae3SPaolo Bonzini direct_pte_prefetch(vcpu, it.sptep); 3171c50d8ae3SPaolo Bonzini return ret; 3172c50d8ae3SPaolo Bonzini } 3173c50d8ae3SPaolo Bonzini 3174c50d8ae3SPaolo Bonzini static void kvm_send_hwpoison_signal(unsigned long address, struct task_struct *tsk) 3175c50d8ae3SPaolo Bonzini { 3176c50d8ae3SPaolo Bonzini send_sig_mceerr(BUS_MCEERR_AR, (void __user *)address, PAGE_SHIFT, tsk); 3177c50d8ae3SPaolo Bonzini } 3178c50d8ae3SPaolo Bonzini 317976657687SPeter Xu static int kvm_handle_error_pfn(struct kvm_vcpu *vcpu, gfn_t gfn, kvm_pfn_t pfn) 3180c50d8ae3SPaolo Bonzini { 318176657687SPeter Xu if (is_sigpending_pfn(pfn)) { 318276657687SPeter Xu kvm_handle_signal_exit(vcpu); 318376657687SPeter Xu return -EINTR; 318476657687SPeter Xu } 318576657687SPeter Xu 3186c50d8ae3SPaolo Bonzini /* 3187c50d8ae3SPaolo Bonzini * Do not cache the mmio info caused by writing the readonly gfn 3188c50d8ae3SPaolo Bonzini * into the spte otherwise read access on readonly gfn also can 3189c50d8ae3SPaolo Bonzini * caused mmio page fault and treat it as mmio access. 3190c50d8ae3SPaolo Bonzini */ 3191c50d8ae3SPaolo Bonzini if (pfn == KVM_PFN_ERR_RO_FAULT) 3192c50d8ae3SPaolo Bonzini return RET_PF_EMULATE; 3193c50d8ae3SPaolo Bonzini 3194c50d8ae3SPaolo Bonzini if (pfn == KVM_PFN_ERR_HWPOISON) { 3195c50d8ae3SPaolo Bonzini kvm_send_hwpoison_signal(kvm_vcpu_gfn_to_hva(vcpu, gfn), current); 3196c50d8ae3SPaolo Bonzini return RET_PF_RETRY; 3197c50d8ae3SPaolo Bonzini } 3198c50d8ae3SPaolo Bonzini 3199c50d8ae3SPaolo Bonzini return -EFAULT; 3200c50d8ae3SPaolo Bonzini } 3201c50d8ae3SPaolo Bonzini 32025276c616SSean Christopherson static int handle_abnormal_pfn(struct kvm_vcpu *vcpu, struct kvm_page_fault *fault, 32035276c616SSean Christopherson unsigned int access) 3204c50d8ae3SPaolo Bonzini { 3205c50d8ae3SPaolo Bonzini /* The pfn is invalid, report the error! */ 32065276c616SSean Christopherson if (unlikely(is_error_pfn(fault->pfn))) 320776657687SPeter Xu return kvm_handle_error_pfn(vcpu, fault->gfn, fault->pfn); 3208c50d8ae3SPaolo Bonzini 3209e710c5f6SDavid Matlack if (unlikely(!fault->slot)) { 32103a13f4feSPaolo Bonzini gva_t gva = fault->is_tdp ? 0 : fault->addr; 32113a13f4feSPaolo Bonzini 32123a13f4feSPaolo Bonzini vcpu_cache_mmio_info(vcpu, gva, fault->gfn, 3213c50d8ae3SPaolo Bonzini access & shadow_mmio_access_mask); 321430ab5901SSean Christopherson /* 321530ab5901SSean Christopherson * If MMIO caching is disabled, emulate immediately without 321630ab5901SSean Christopherson * touching the shadow page tables as attempting to install an 321786931ff7SSean Christopherson * MMIO SPTE will just be an expensive nop. Do not cache MMIO 321886931ff7SSean Christopherson * whose gfn is greater than host.MAXPHYADDR, any guest that 321986931ff7SSean Christopherson * generates such gfns is running nested and is being tricked 322086931ff7SSean Christopherson * by L0 userspace (you can observe gfn > L1.MAXPHYADDR if 322186931ff7SSean Christopherson * and only if L1's MAXPHYADDR is inaccurate with respect to 322286931ff7SSean Christopherson * the hardware's). 322330ab5901SSean Christopherson */ 32248b9e74bfSSean Christopherson if (unlikely(!enable_mmio_caching) || 32255276c616SSean Christopherson unlikely(fault->gfn > kvm_mmu_max_gfn())) 32265276c616SSean Christopherson return RET_PF_EMULATE; 322730ab5901SSean Christopherson } 3228c50d8ae3SPaolo Bonzini 32295276c616SSean Christopherson return RET_PF_CONTINUE; 3230c50d8ae3SPaolo Bonzini } 3231c50d8ae3SPaolo Bonzini 32323c8ad5a6SPaolo Bonzini static bool page_fault_can_be_fast(struct kvm_page_fault *fault) 3233c50d8ae3SPaolo Bonzini { 3234c50d8ae3SPaolo Bonzini /* 32355c64aba5SSean Christopherson * Page faults with reserved bits set, i.e. faults on MMIO SPTEs, only 32365c64aba5SSean Christopherson * reach the common page fault handler if the SPTE has an invalid MMIO 32375c64aba5SSean Christopherson * generation number. Refreshing the MMIO generation needs to go down 32385c64aba5SSean Christopherson * the slow path. Note, EPT Misconfigs do NOT set the PRESENT flag! 3239c50d8ae3SPaolo Bonzini */ 32403c8ad5a6SPaolo Bonzini if (fault->rsvd) 3241c50d8ae3SPaolo Bonzini return false; 3242c50d8ae3SPaolo Bonzini 3243c50d8ae3SPaolo Bonzini /* 3244c50d8ae3SPaolo Bonzini * #PF can be fast if: 3245c50d8ae3SPaolo Bonzini * 324654275f74SSean Christopherson * 1. The shadow page table entry is not present and A/D bits are 324754275f74SSean Christopherson * disabled _by KVM_, which could mean that the fault is potentially 324854275f74SSean Christopherson * caused by access tracking (if enabled). If A/D bits are enabled 324954275f74SSean Christopherson * by KVM, but disabled by L1 for L2, KVM is forced to disable A/D 325054275f74SSean Christopherson * bits for L2 and employ access tracking, but the fast page fault 325154275f74SSean Christopherson * mechanism only supports direct MMUs. 325254275f74SSean Christopherson * 2. The shadow page table entry is present, the access is a write, 325354275f74SSean Christopherson * and no reserved bits are set (MMIO SPTEs cannot be "fixed"), i.e. 325454275f74SSean Christopherson * the fault was caused by a write-protection violation. If the 325554275f74SSean Christopherson * SPTE is MMU-writable (determined later), the fault can be fixed 325654275f74SSean Christopherson * by setting the Writable bit, which can be done out of mmu_lock. 3257c50d8ae3SPaolo Bonzini */ 32585c64aba5SSean Christopherson if (!fault->present) 32595c64aba5SSean Christopherson return !kvm_ad_enabled(); 32605c64aba5SSean Christopherson 32615c64aba5SSean Christopherson /* 32625c64aba5SSean Christopherson * Note, instruction fetches and writes are mutually exclusive, ignore 32635c64aba5SSean Christopherson * the "exec" flag. 32645c64aba5SSean Christopherson */ 32655c64aba5SSean Christopherson return fault->write; 3266c50d8ae3SPaolo Bonzini } 3267c50d8ae3SPaolo Bonzini 3268c50d8ae3SPaolo Bonzini /* 3269c50d8ae3SPaolo Bonzini * Returns true if the SPTE was fixed successfully. Otherwise, 3270c50d8ae3SPaolo Bonzini * someone else modified the SPTE from its original value. 3271c50d8ae3SPaolo Bonzini */ 3272c50d8ae3SPaolo Bonzini static bool 3273e710c5f6SDavid Matlack fast_pf_fix_direct_spte(struct kvm_vcpu *vcpu, struct kvm_page_fault *fault, 3274c50d8ae3SPaolo Bonzini u64 *sptep, u64 old_spte, u64 new_spte) 3275c50d8ae3SPaolo Bonzini { 3276c50d8ae3SPaolo Bonzini /* 3277c50d8ae3SPaolo Bonzini * Theoretically we could also set dirty bit (and flush TLB) here in 3278c50d8ae3SPaolo Bonzini * order to eliminate unnecessary PML logging. See comments in 3279c50d8ae3SPaolo Bonzini * set_spte. But fast_page_fault is very unlikely to happen with PML 3280c50d8ae3SPaolo Bonzini * enabled, so we do not do this. This might result in the same GPA 3281c50d8ae3SPaolo Bonzini * to be logged in PML buffer again when the write really happens, and 3282c50d8ae3SPaolo Bonzini * eventually to be called by mark_page_dirty twice. But it's also no 3283c50d8ae3SPaolo Bonzini * harm. This also avoids the TLB flush needed after setting dirty bit 3284c50d8ae3SPaolo Bonzini * so non-PML cases won't be impacted. 3285c50d8ae3SPaolo Bonzini * 3286c50d8ae3SPaolo Bonzini * Compare with set_spte where instead shadow_dirty_mask is set. 3287c50d8ae3SPaolo Bonzini */ 32882db2f46fSUros Bizjak if (!try_cmpxchg64(sptep, &old_spte, new_spte)) 3289c50d8ae3SPaolo Bonzini return false; 3290c50d8ae3SPaolo Bonzini 3291e710c5f6SDavid Matlack if (is_writable_pte(new_spte) && !is_writable_pte(old_spte)) 3292e710c5f6SDavid Matlack mark_page_dirty_in_slot(vcpu->kvm, fault->slot, fault->gfn); 3293c50d8ae3SPaolo Bonzini 3294c50d8ae3SPaolo Bonzini return true; 3295c50d8ae3SPaolo Bonzini } 3296c50d8ae3SPaolo Bonzini 32973c8ad5a6SPaolo Bonzini static bool is_access_allowed(struct kvm_page_fault *fault, u64 spte) 3298c50d8ae3SPaolo Bonzini { 32993c8ad5a6SPaolo Bonzini if (fault->exec) 3300c50d8ae3SPaolo Bonzini return is_executable_pte(spte); 3301c50d8ae3SPaolo Bonzini 33023c8ad5a6SPaolo Bonzini if (fault->write) 3303c50d8ae3SPaolo Bonzini return is_writable_pte(spte); 3304c50d8ae3SPaolo Bonzini 3305c50d8ae3SPaolo Bonzini /* Fault was on Read access */ 3306c50d8ae3SPaolo Bonzini return spte & PT_PRESENT_MASK; 3307c50d8ae3SPaolo Bonzini } 3308c50d8ae3SPaolo Bonzini 3309c50d8ae3SPaolo Bonzini /* 33106e8eb206SDavid Matlack * Returns the last level spte pointer of the shadow page walk for the given 33116e8eb206SDavid Matlack * gpa, and sets *spte to the spte value. This spte may be non-preset. If no 33126e8eb206SDavid Matlack * walk could be performed, returns NULL and *spte does not contain valid data. 33136e8eb206SDavid Matlack * 33146e8eb206SDavid Matlack * Contract: 33156e8eb206SDavid Matlack * - Must be called between walk_shadow_page_lockless_{begin,end}. 33166e8eb206SDavid Matlack * - The returned sptep must not be used after walk_shadow_page_lockless_end. 33176e8eb206SDavid Matlack */ 33186e8eb206SDavid Matlack static u64 *fast_pf_get_last_sptep(struct kvm_vcpu *vcpu, gpa_t gpa, u64 *spte) 33196e8eb206SDavid Matlack { 33206e8eb206SDavid Matlack struct kvm_shadow_walk_iterator iterator; 33216e8eb206SDavid Matlack u64 old_spte; 33226e8eb206SDavid Matlack u64 *sptep = NULL; 33236e8eb206SDavid Matlack 33246e8eb206SDavid Matlack for_each_shadow_entry_lockless(vcpu, gpa, iterator, old_spte) { 33256e8eb206SDavid Matlack sptep = iterator.sptep; 33266e8eb206SDavid Matlack *spte = old_spte; 33276e8eb206SDavid Matlack } 33286e8eb206SDavid Matlack 33296e8eb206SDavid Matlack return sptep; 33306e8eb206SDavid Matlack } 33316e8eb206SDavid Matlack 33326e8eb206SDavid Matlack /* 3333c4371c2aSSean Christopherson * Returns one of RET_PF_INVALID, RET_PF_FIXED or RET_PF_SPURIOUS. 3334c50d8ae3SPaolo Bonzini */ 33353c8ad5a6SPaolo Bonzini static int fast_page_fault(struct kvm_vcpu *vcpu, struct kvm_page_fault *fault) 3336c50d8ae3SPaolo Bonzini { 3337c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 3338c4371c2aSSean Christopherson int ret = RET_PF_INVALID; 3339c50d8ae3SPaolo Bonzini u64 spte = 0ull; 33406e8eb206SDavid Matlack u64 *sptep = NULL; 3341c50d8ae3SPaolo Bonzini uint retry_count = 0; 3342c50d8ae3SPaolo Bonzini 33433c8ad5a6SPaolo Bonzini if (!page_fault_can_be_fast(fault)) 3344c4371c2aSSean Christopherson return ret; 3345c50d8ae3SPaolo Bonzini 3346c50d8ae3SPaolo Bonzini walk_shadow_page_lockless_begin(vcpu); 3347c50d8ae3SPaolo Bonzini 3348c50d8ae3SPaolo Bonzini do { 3349c50d8ae3SPaolo Bonzini u64 new_spte; 3350c50d8ae3SPaolo Bonzini 33516e8eb206SDavid Matlack if (is_tdp_mmu(vcpu->arch.mmu)) 33523c8ad5a6SPaolo Bonzini sptep = kvm_tdp_mmu_fast_pf_get_last_sptep(vcpu, fault->addr, &spte); 33536e8eb206SDavid Matlack else 33543c8ad5a6SPaolo Bonzini sptep = fast_pf_get_last_sptep(vcpu, fault->addr, &spte); 3355c50d8ae3SPaolo Bonzini 3356ec89e643SSean Christopherson if (!is_shadow_present_pte(spte)) 3357ec89e643SSean Christopherson break; 3358ec89e643SSean Christopherson 33596e8eb206SDavid Matlack sp = sptep_to_sp(sptep); 3360c50d8ae3SPaolo Bonzini if (!is_last_spte(spte, sp->role.level)) 3361c50d8ae3SPaolo Bonzini break; 3362c50d8ae3SPaolo Bonzini 3363c50d8ae3SPaolo Bonzini /* 3364c50d8ae3SPaolo Bonzini * Check whether the memory access that caused the fault would 3365c50d8ae3SPaolo Bonzini * still cause it if it were to be performed right now. If not, 3366c50d8ae3SPaolo Bonzini * then this is a spurious fault caused by TLB lazily flushed, 3367c50d8ae3SPaolo Bonzini * or some other CPU has already fixed the PTE after the 3368c50d8ae3SPaolo Bonzini * current CPU took the fault. 3369c50d8ae3SPaolo Bonzini * 3370c50d8ae3SPaolo Bonzini * Need not check the access of upper level table entries since 3371c50d8ae3SPaolo Bonzini * they are always ACC_ALL. 3372c50d8ae3SPaolo Bonzini */ 33733c8ad5a6SPaolo Bonzini if (is_access_allowed(fault, spte)) { 3374c4371c2aSSean Christopherson ret = RET_PF_SPURIOUS; 3375c50d8ae3SPaolo Bonzini break; 3376c50d8ae3SPaolo Bonzini } 3377c50d8ae3SPaolo Bonzini 3378c50d8ae3SPaolo Bonzini new_spte = spte; 3379c50d8ae3SPaolo Bonzini 338054275f74SSean Christopherson /* 338154275f74SSean Christopherson * KVM only supports fixing page faults outside of MMU lock for 338254275f74SSean Christopherson * direct MMUs, nested MMUs are always indirect, and KVM always 338354275f74SSean Christopherson * uses A/D bits for non-nested MMUs. Thus, if A/D bits are 338454275f74SSean Christopherson * enabled, the SPTE can't be an access-tracked SPTE. 338554275f74SSean Christopherson */ 338654275f74SSean Christopherson if (unlikely(!kvm_ad_enabled()) && is_access_track_spte(spte)) 3387c50d8ae3SPaolo Bonzini new_spte = restore_acc_track_spte(new_spte); 3388c50d8ae3SPaolo Bonzini 3389c50d8ae3SPaolo Bonzini /* 339054275f74SSean Christopherson * To keep things simple, only SPTEs that are MMU-writable can 339154275f74SSean Christopherson * be made fully writable outside of mmu_lock, e.g. only SPTEs 339254275f74SSean Christopherson * that were write-protected for dirty-logging or access 339354275f74SSean Christopherson * tracking are handled here. Don't bother checking if the 339454275f74SSean Christopherson * SPTE is writable to prioritize running with A/D bits enabled. 339554275f74SSean Christopherson * The is_access_allowed() check above handles the common case 339654275f74SSean Christopherson * of the fault being spurious, and the SPTE is known to be 339754275f74SSean Christopherson * shadow-present, i.e. except for access tracking restoration 339854275f74SSean Christopherson * making the new SPTE writable, the check is wasteful. 3399c50d8ae3SPaolo Bonzini */ 3400706c9c55SSean Christopherson if (fault->write && is_mmu_writable_spte(spte)) { 3401c50d8ae3SPaolo Bonzini new_spte |= PT_WRITABLE_MASK; 3402c50d8ae3SPaolo Bonzini 3403c50d8ae3SPaolo Bonzini /* 340410c30de0SJunaid Shahid * Do not fix write-permission on the large spte when 340510c30de0SJunaid Shahid * dirty logging is enabled. Since we only dirty the 340610c30de0SJunaid Shahid * first page into the dirty-bitmap in 3407c50d8ae3SPaolo Bonzini * fast_pf_fix_direct_spte(), other pages are missed 3408c50d8ae3SPaolo Bonzini * if its slot has dirty logging enabled. 3409c50d8ae3SPaolo Bonzini * 3410c50d8ae3SPaolo Bonzini * Instead, we let the slow page fault path create a 3411c50d8ae3SPaolo Bonzini * normal spte to fix the access. 3412c50d8ae3SPaolo Bonzini */ 341310c30de0SJunaid Shahid if (sp->role.level > PG_LEVEL_4K && 341410c30de0SJunaid Shahid kvm_slot_dirty_track_enabled(fault->slot)) 3415c50d8ae3SPaolo Bonzini break; 3416c50d8ae3SPaolo Bonzini } 3417c50d8ae3SPaolo Bonzini 3418c50d8ae3SPaolo Bonzini /* Verify that the fault can be handled in the fast path */ 3419c50d8ae3SPaolo Bonzini if (new_spte == spte || 34203c8ad5a6SPaolo Bonzini !is_access_allowed(fault, new_spte)) 3421c50d8ae3SPaolo Bonzini break; 3422c50d8ae3SPaolo Bonzini 3423c50d8ae3SPaolo Bonzini /* 3424c50d8ae3SPaolo Bonzini * Currently, fast page fault only works for direct mapping 3425c50d8ae3SPaolo Bonzini * since the gfn is not stable for indirect shadow page. See 34263ecad8c2SMauro Carvalho Chehab * Documentation/virt/kvm/locking.rst to get more detail. 3427c50d8ae3SPaolo Bonzini */ 3428e710c5f6SDavid Matlack if (fast_pf_fix_direct_spte(vcpu, fault, sptep, spte, new_spte)) { 3429c4371c2aSSean Christopherson ret = RET_PF_FIXED; 3430c50d8ae3SPaolo Bonzini break; 3431c4371c2aSSean Christopherson } 3432c50d8ae3SPaolo Bonzini 3433c50d8ae3SPaolo Bonzini if (++retry_count > 4) { 3434c50d8ae3SPaolo Bonzini printk_once(KERN_WARNING 3435c50d8ae3SPaolo Bonzini "kvm: Fast #PF retrying more than 4 times.\n"); 3436c50d8ae3SPaolo Bonzini break; 3437c50d8ae3SPaolo Bonzini } 3438c50d8ae3SPaolo Bonzini 3439c50d8ae3SPaolo Bonzini } while (true); 3440c50d8ae3SPaolo Bonzini 3441f0066d94SPaolo Bonzini trace_fast_page_fault(vcpu, fault, sptep, spte, ret); 3442c50d8ae3SPaolo Bonzini walk_shadow_page_lockless_end(vcpu); 3443c50d8ae3SPaolo Bonzini 34441075d41eSSean Christopherson if (ret != RET_PF_INVALID) 34451075d41eSSean Christopherson vcpu->stat.pf_fast++; 34461075d41eSSean Christopherson 3447c4371c2aSSean Christopherson return ret; 3448c50d8ae3SPaolo Bonzini } 3449c50d8ae3SPaolo Bonzini 3450c50d8ae3SPaolo Bonzini static void mmu_free_root_page(struct kvm *kvm, hpa_t *root_hpa, 3451c50d8ae3SPaolo Bonzini struct list_head *invalid_list) 3452c50d8ae3SPaolo Bonzini { 3453c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 3454c50d8ae3SPaolo Bonzini 3455c50d8ae3SPaolo Bonzini if (!VALID_PAGE(*root_hpa)) 3456c50d8ae3SPaolo Bonzini return; 3457c50d8ae3SPaolo Bonzini 3458*5e3edd7eSSean Christopherson /* 3459*5e3edd7eSSean Christopherson * The "root" may be a special root, e.g. a PAE entry, treat it as a 3460*5e3edd7eSSean Christopherson * SPTE to ensure any non-PA bits are dropped. 3461*5e3edd7eSSean Christopherson */ 3462*5e3edd7eSSean Christopherson sp = spte_to_child_sp(*root_hpa); 34639191b8f0SPaolo Bonzini if (WARN_ON(!sp)) 34649191b8f0SPaolo Bonzini return; 346502c00b3aSBen Gardon 3466897218ffSPaolo Bonzini if (is_tdp_mmu_page(sp)) 34676103bc07SBen Gardon kvm_tdp_mmu_put_root(kvm, sp, false); 346876eb54e7SBen Gardon else if (!--sp->root_count && sp->role.invalid) 3469c50d8ae3SPaolo Bonzini kvm_mmu_prepare_zap_page(kvm, sp, invalid_list); 3470c50d8ae3SPaolo Bonzini 3471c50d8ae3SPaolo Bonzini *root_hpa = INVALID_PAGE; 3472c50d8ae3SPaolo Bonzini } 3473c50d8ae3SPaolo Bonzini 3474c50d8ae3SPaolo Bonzini /* roots_to_free must be some combination of the KVM_MMU_ROOT_* flags */ 34750c1c92f1SPaolo Bonzini void kvm_mmu_free_roots(struct kvm *kvm, struct kvm_mmu *mmu, 3476c50d8ae3SPaolo Bonzini ulong roots_to_free) 3477c50d8ae3SPaolo Bonzini { 3478c50d8ae3SPaolo Bonzini int i; 3479c50d8ae3SPaolo Bonzini LIST_HEAD(invalid_list); 3480594bef79SPaolo Bonzini bool free_active_root; 3481c50d8ae3SPaolo Bonzini 3482c50d8ae3SPaolo Bonzini BUILD_BUG_ON(KVM_MMU_NUM_PREV_ROOTS >= BITS_PER_LONG); 3483c50d8ae3SPaolo Bonzini 3484c50d8ae3SPaolo Bonzini /* Before acquiring the MMU lock, see if we need to do any real work. */ 3485594bef79SPaolo Bonzini free_active_root = (roots_to_free & KVM_MMU_ROOT_CURRENT) 3486594bef79SPaolo Bonzini && VALID_PAGE(mmu->root.hpa); 3487594bef79SPaolo Bonzini 3488594bef79SPaolo Bonzini if (!free_active_root) { 3489c50d8ae3SPaolo Bonzini for (i = 0; i < KVM_MMU_NUM_PREV_ROOTS; i++) 3490c50d8ae3SPaolo Bonzini if ((roots_to_free & KVM_MMU_ROOT_PREVIOUS(i)) && 3491c50d8ae3SPaolo Bonzini VALID_PAGE(mmu->prev_roots[i].hpa)) 3492c50d8ae3SPaolo Bonzini break; 3493c50d8ae3SPaolo Bonzini 3494c50d8ae3SPaolo Bonzini if (i == KVM_MMU_NUM_PREV_ROOTS) 3495c50d8ae3SPaolo Bonzini return; 3496c50d8ae3SPaolo Bonzini } 3497c50d8ae3SPaolo Bonzini 3498531810caSBen Gardon write_lock(&kvm->mmu_lock); 3499c50d8ae3SPaolo Bonzini 3500c50d8ae3SPaolo Bonzini for (i = 0; i < KVM_MMU_NUM_PREV_ROOTS; i++) 3501c50d8ae3SPaolo Bonzini if (roots_to_free & KVM_MMU_ROOT_PREVIOUS(i)) 35024d710de9SSean Christopherson mmu_free_root_page(kvm, &mmu->prev_roots[i].hpa, 3503c50d8ae3SPaolo Bonzini &invalid_list); 3504c50d8ae3SPaolo Bonzini 3505c50d8ae3SPaolo Bonzini if (free_active_root) { 3506594bef79SPaolo Bonzini if (to_shadow_page(mmu->root.hpa)) { 3507b9e5603cSPaolo Bonzini mmu_free_root_page(kvm, &mmu->root.hpa, &invalid_list); 350804d45551SSean Christopherson } else if (mmu->pae_root) { 3509c834e5e4SSean Christopherson for (i = 0; i < 4; ++i) { 3510c834e5e4SSean Christopherson if (!IS_VALID_PAE_ROOT(mmu->pae_root[i])) 3511c834e5e4SSean Christopherson continue; 3512c834e5e4SSean Christopherson 3513c834e5e4SSean Christopherson mmu_free_root_page(kvm, &mmu->pae_root[i], 3514c50d8ae3SPaolo Bonzini &invalid_list); 3515c834e5e4SSean Christopherson mmu->pae_root[i] = INVALID_PAE_ROOT; 3516c50d8ae3SPaolo Bonzini } 3517c50d8ae3SPaolo Bonzini } 3518b9e5603cSPaolo Bonzini mmu->root.hpa = INVALID_PAGE; 3519b9e5603cSPaolo Bonzini mmu->root.pgd = 0; 3520c50d8ae3SPaolo Bonzini } 3521c50d8ae3SPaolo Bonzini 35224d710de9SSean Christopherson kvm_mmu_commit_zap_page(kvm, &invalid_list); 3523531810caSBen Gardon write_unlock(&kvm->mmu_lock); 3524c50d8ae3SPaolo Bonzini } 3525c50d8ae3SPaolo Bonzini EXPORT_SYMBOL_GPL(kvm_mmu_free_roots); 3526c50d8ae3SPaolo Bonzini 35270c1c92f1SPaolo Bonzini void kvm_mmu_free_guest_mode_roots(struct kvm *kvm, struct kvm_mmu *mmu) 352825b62c62SSean Christopherson { 352925b62c62SSean Christopherson unsigned long roots_to_free = 0; 353025b62c62SSean Christopherson hpa_t root_hpa; 353125b62c62SSean Christopherson int i; 353225b62c62SSean Christopherson 353325b62c62SSean Christopherson /* 353425b62c62SSean Christopherson * This should not be called while L2 is active, L2 can't invalidate 353525b62c62SSean Christopherson * _only_ its own roots, e.g. INVVPID unconditionally exits. 353625b62c62SSean Christopherson */ 35377a458f0eSPaolo Bonzini WARN_ON_ONCE(mmu->root_role.guest_mode); 353825b62c62SSean Christopherson 353925b62c62SSean Christopherson for (i = 0; i < KVM_MMU_NUM_PREV_ROOTS; i++) { 354025b62c62SSean Christopherson root_hpa = mmu->prev_roots[i].hpa; 354125b62c62SSean Christopherson if (!VALID_PAGE(root_hpa)) 354225b62c62SSean Christopherson continue; 354325b62c62SSean Christopherson 354425b62c62SSean Christopherson if (!to_shadow_page(root_hpa) || 354525b62c62SSean Christopherson to_shadow_page(root_hpa)->role.guest_mode) 354625b62c62SSean Christopherson roots_to_free |= KVM_MMU_ROOT_PREVIOUS(i); 354725b62c62SSean Christopherson } 354825b62c62SSean Christopherson 35490c1c92f1SPaolo Bonzini kvm_mmu_free_roots(kvm, mmu, roots_to_free); 355025b62c62SSean Christopherson } 355125b62c62SSean Christopherson EXPORT_SYMBOL_GPL(kvm_mmu_free_guest_mode_roots); 355225b62c62SSean Christopherson 355325b62c62SSean Christopherson 3554c50d8ae3SPaolo Bonzini static int mmu_check_root(struct kvm_vcpu *vcpu, gfn_t root_gfn) 3555c50d8ae3SPaolo Bonzini { 3556c50d8ae3SPaolo Bonzini int ret = 0; 3557c50d8ae3SPaolo Bonzini 3558995decb6SVitaly Kuznetsov if (!kvm_vcpu_is_visible_gfn(vcpu, root_gfn)) { 3559c50d8ae3SPaolo Bonzini kvm_make_request(KVM_REQ_TRIPLE_FAULT, vcpu); 3560c50d8ae3SPaolo Bonzini ret = 1; 3561c50d8ae3SPaolo Bonzini } 3562c50d8ae3SPaolo Bonzini 3563c50d8ae3SPaolo Bonzini return ret; 3564c50d8ae3SPaolo Bonzini } 3565c50d8ae3SPaolo Bonzini 35662e65e842SDavid Matlack static hpa_t mmu_alloc_root(struct kvm_vcpu *vcpu, gfn_t gfn, int quadrant, 356786938ab6SDavid Matlack u8 level) 3568c50d8ae3SPaolo Bonzini { 35692e65e842SDavid Matlack union kvm_mmu_page_role role = vcpu->arch.mmu->root_role; 3570c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 35718123f265SSean Christopherson 35722e65e842SDavid Matlack role.level = level; 35732e65e842SDavid Matlack role.quadrant = quadrant; 35742e65e842SDavid Matlack 35757f497775SDavid Matlack WARN_ON_ONCE(quadrant && !role.has_4_byte_gpte); 35767f497775SDavid Matlack WARN_ON_ONCE(role.direct && role.has_4_byte_gpte); 35777f497775SDavid Matlack 357887654643SDavid Matlack sp = kvm_mmu_get_shadow_page(vcpu, gfn, role); 35798123f265SSean Christopherson ++sp->root_count; 35808123f265SSean Christopherson 35818123f265SSean Christopherson return __pa(sp->spt); 35828123f265SSean Christopherson } 35838123f265SSean Christopherson 35848123f265SSean Christopherson static int mmu_alloc_direct_roots(struct kvm_vcpu *vcpu) 35858123f265SSean Christopherson { 3586b37233c9SSean Christopherson struct kvm_mmu *mmu = vcpu->arch.mmu; 3587a972e29cSPaolo Bonzini u8 shadow_root_level = mmu->root_role.level; 35888123f265SSean Christopherson hpa_t root; 3589c50d8ae3SPaolo Bonzini unsigned i; 35904a38162eSPaolo Bonzini int r; 35914a38162eSPaolo Bonzini 35924a38162eSPaolo Bonzini write_lock(&vcpu->kvm->mmu_lock); 35934a38162eSPaolo Bonzini r = make_mmu_pages_available(vcpu); 35944a38162eSPaolo Bonzini if (r < 0) 35954a38162eSPaolo Bonzini goto out_unlock; 3596c50d8ae3SPaolo Bonzini 3597897218ffSPaolo Bonzini if (is_tdp_mmu_enabled(vcpu->kvm)) { 359802c00b3aSBen Gardon root = kvm_tdp_mmu_get_vcpu_root_hpa(vcpu); 3599b9e5603cSPaolo Bonzini mmu->root.hpa = root; 360002c00b3aSBen Gardon } else if (shadow_root_level >= PT64_ROOT_4LEVEL) { 360186938ab6SDavid Matlack root = mmu_alloc_root(vcpu, 0, 0, shadow_root_level); 3602b9e5603cSPaolo Bonzini mmu->root.hpa = root; 36038123f265SSean Christopherson } else if (shadow_root_level == PT32E_ROOT_LEVEL) { 36044a38162eSPaolo Bonzini if (WARN_ON_ONCE(!mmu->pae_root)) { 36054a38162eSPaolo Bonzini r = -EIO; 36064a38162eSPaolo Bonzini goto out_unlock; 36074a38162eSPaolo Bonzini } 360873ad1606SSean Christopherson 3609c50d8ae3SPaolo Bonzini for (i = 0; i < 4; ++i) { 3610c834e5e4SSean Christopherson WARN_ON_ONCE(IS_VALID_PAE_ROOT(mmu->pae_root[i])); 3611c50d8ae3SPaolo Bonzini 36127f497775SDavid Matlack root = mmu_alloc_root(vcpu, i << (30 - PAGE_SHIFT), 0, 36132e65e842SDavid Matlack PT32_ROOT_LEVEL); 361417e368d9SSean Christopherson mmu->pae_root[i] = root | PT_PRESENT_MASK | 3615d2263de1SYuan Yao shadow_me_value; 3616c50d8ae3SPaolo Bonzini } 3617b9e5603cSPaolo Bonzini mmu->root.hpa = __pa(mmu->pae_root); 361873ad1606SSean Christopherson } else { 361973ad1606SSean Christopherson WARN_ONCE(1, "Bad TDP root level = %d\n", shadow_root_level); 36204a38162eSPaolo Bonzini r = -EIO; 36214a38162eSPaolo Bonzini goto out_unlock; 362273ad1606SSean Christopherson } 36233651c7fcSSean Christopherson 3624b9e5603cSPaolo Bonzini /* root.pgd is ignored for direct MMUs. */ 3625b9e5603cSPaolo Bonzini mmu->root.pgd = 0; 36264a38162eSPaolo Bonzini out_unlock: 36274a38162eSPaolo Bonzini write_unlock(&vcpu->kvm->mmu_lock); 36284a38162eSPaolo Bonzini return r; 3629c50d8ae3SPaolo Bonzini } 3630c50d8ae3SPaolo Bonzini 36311e76a3ceSDavid Stevens static int mmu_first_shadow_root_alloc(struct kvm *kvm) 36321e76a3ceSDavid Stevens { 36331e76a3ceSDavid Stevens struct kvm_memslots *slots; 36341e76a3ceSDavid Stevens struct kvm_memory_slot *slot; 3635a54d8066SMaciej S. Szmigiero int r = 0, i, bkt; 36361e76a3ceSDavid Stevens 36371e76a3ceSDavid Stevens /* 36381e76a3ceSDavid Stevens * Check if this is the first shadow root being allocated before 36391e76a3ceSDavid Stevens * taking the lock. 36401e76a3ceSDavid Stevens */ 36411e76a3ceSDavid Stevens if (kvm_shadow_root_allocated(kvm)) 36421e76a3ceSDavid Stevens return 0; 36431e76a3ceSDavid Stevens 36441e76a3ceSDavid Stevens mutex_lock(&kvm->slots_arch_lock); 36451e76a3ceSDavid Stevens 36461e76a3ceSDavid Stevens /* Recheck, under the lock, whether this is the first shadow root. */ 36471e76a3ceSDavid Stevens if (kvm_shadow_root_allocated(kvm)) 36481e76a3ceSDavid Stevens goto out_unlock; 36491e76a3ceSDavid Stevens 36501e76a3ceSDavid Stevens /* 36511e76a3ceSDavid Stevens * Check if anything actually needs to be allocated, e.g. all metadata 36521e76a3ceSDavid Stevens * will be allocated upfront if TDP is disabled. 36531e76a3ceSDavid Stevens */ 36541e76a3ceSDavid Stevens if (kvm_memslots_have_rmaps(kvm) && 36551e76a3ceSDavid Stevens kvm_page_track_write_tracking_enabled(kvm)) 36561e76a3ceSDavid Stevens goto out_success; 36571e76a3ceSDavid Stevens 36581e76a3ceSDavid Stevens for (i = 0; i < KVM_ADDRESS_SPACE_NUM; i++) { 36591e76a3ceSDavid Stevens slots = __kvm_memslots(kvm, i); 3660a54d8066SMaciej S. Szmigiero kvm_for_each_memslot(slot, bkt, slots) { 36611e76a3ceSDavid Stevens /* 36621e76a3ceSDavid Stevens * Both of these functions are no-ops if the target is 36631e76a3ceSDavid Stevens * already allocated, so unconditionally calling both 36641e76a3ceSDavid Stevens * is safe. Intentionally do NOT free allocations on 36651e76a3ceSDavid Stevens * failure to avoid having to track which allocations 36661e76a3ceSDavid Stevens * were made now versus when the memslot was created. 36671e76a3ceSDavid Stevens * The metadata is guaranteed to be freed when the slot 36681e76a3ceSDavid Stevens * is freed, and will be kept/used if userspace retries 36691e76a3ceSDavid Stevens * KVM_RUN instead of killing the VM. 36701e76a3ceSDavid Stevens */ 36711e76a3ceSDavid Stevens r = memslot_rmap_alloc(slot, slot->npages); 36721e76a3ceSDavid Stevens if (r) 36731e76a3ceSDavid Stevens goto out_unlock; 36741e76a3ceSDavid Stevens r = kvm_page_track_write_tracking_alloc(slot); 36751e76a3ceSDavid Stevens if (r) 36761e76a3ceSDavid Stevens goto out_unlock; 36771e76a3ceSDavid Stevens } 36781e76a3ceSDavid Stevens } 36791e76a3ceSDavid Stevens 36801e76a3ceSDavid Stevens /* 36811e76a3ceSDavid Stevens * Ensure that shadow_root_allocated becomes true strictly after 36821e76a3ceSDavid Stevens * all the related pointers are set. 36831e76a3ceSDavid Stevens */ 36841e76a3ceSDavid Stevens out_success: 36851e76a3ceSDavid Stevens smp_store_release(&kvm->arch.shadow_root_allocated, true); 36861e76a3ceSDavid Stevens 36871e76a3ceSDavid Stevens out_unlock: 36881e76a3ceSDavid Stevens mutex_unlock(&kvm->slots_arch_lock); 36891e76a3ceSDavid Stevens return r; 36901e76a3ceSDavid Stevens } 36911e76a3ceSDavid Stevens 3692c50d8ae3SPaolo Bonzini static int mmu_alloc_shadow_roots(struct kvm_vcpu *vcpu) 3693c50d8ae3SPaolo Bonzini { 3694b37233c9SSean Christopherson struct kvm_mmu *mmu = vcpu->arch.mmu; 36956e0918aeSSean Christopherson u64 pdptrs[4], pm_mask; 3696be01e8e2SSean Christopherson gfn_t root_gfn, root_pgd; 36977f497775SDavid Matlack int quadrant, i, r; 36988123f265SSean Christopherson hpa_t root; 3699c50d8ae3SPaolo Bonzini 3700b37233c9SSean Christopherson root_pgd = mmu->get_guest_pgd(vcpu); 3701be01e8e2SSean Christopherson root_gfn = root_pgd >> PAGE_SHIFT; 3702c50d8ae3SPaolo Bonzini 3703c50d8ae3SPaolo Bonzini if (mmu_check_root(vcpu, root_gfn)) 3704c50d8ae3SPaolo Bonzini return 1; 3705c50d8ae3SPaolo Bonzini 3706c50d8ae3SPaolo Bonzini /* 37074a38162eSPaolo Bonzini * On SVM, reading PDPTRs might access guest memory, which might fault 37084a38162eSPaolo Bonzini * and thus might sleep. Grab the PDPTRs before acquiring mmu_lock. 37094a38162eSPaolo Bonzini */ 37104d25502aSPaolo Bonzini if (mmu->cpu_role.base.level == PT32E_ROOT_LEVEL) { 37116e0918aeSSean Christopherson for (i = 0; i < 4; ++i) { 37126e0918aeSSean Christopherson pdptrs[i] = mmu->get_pdptr(vcpu, i); 37136e0918aeSSean Christopherson if (!(pdptrs[i] & PT_PRESENT_MASK)) 37146e0918aeSSean Christopherson continue; 37156e0918aeSSean Christopherson 37166e0918aeSSean Christopherson if (mmu_check_root(vcpu, pdptrs[i] >> PAGE_SHIFT)) 37176e0918aeSSean Christopherson return 1; 37186e0918aeSSean Christopherson } 37196e0918aeSSean Christopherson } 37206e0918aeSSean Christopherson 37211e76a3ceSDavid Stevens r = mmu_first_shadow_root_alloc(vcpu->kvm); 3722d501f747SBen Gardon if (r) 3723d501f747SBen Gardon return r; 3724d501f747SBen Gardon 37254a38162eSPaolo Bonzini write_lock(&vcpu->kvm->mmu_lock); 37264a38162eSPaolo Bonzini r = make_mmu_pages_available(vcpu); 37274a38162eSPaolo Bonzini if (r < 0) 37284a38162eSPaolo Bonzini goto out_unlock; 37294a38162eSPaolo Bonzini 3730c50d8ae3SPaolo Bonzini /* 3731c50d8ae3SPaolo Bonzini * Do we shadow a long mode page table? If so we need to 3732c50d8ae3SPaolo Bonzini * write-protect the guests page table root. 3733c50d8ae3SPaolo Bonzini */ 37344d25502aSPaolo Bonzini if (mmu->cpu_role.base.level >= PT64_ROOT_4LEVEL) { 37358123f265SSean Christopherson root = mmu_alloc_root(vcpu, root_gfn, 0, 373686938ab6SDavid Matlack mmu->root_role.level); 3737b9e5603cSPaolo Bonzini mmu->root.hpa = root; 3738be01e8e2SSean Christopherson goto set_root_pgd; 3739c50d8ae3SPaolo Bonzini } 3740c50d8ae3SPaolo Bonzini 37414a38162eSPaolo Bonzini if (WARN_ON_ONCE(!mmu->pae_root)) { 37424a38162eSPaolo Bonzini r = -EIO; 37434a38162eSPaolo Bonzini goto out_unlock; 37444a38162eSPaolo Bonzini } 374573ad1606SSean Christopherson 3746c50d8ae3SPaolo Bonzini /* 3747c50d8ae3SPaolo Bonzini * We shadow a 32 bit page table. This may be a legacy 2-level 3748c50d8ae3SPaolo Bonzini * or a PAE 3-level page table. In either case we need to be aware that 3749c50d8ae3SPaolo Bonzini * the shadow page table may be a PAE or a long mode page table. 3750c50d8ae3SPaolo Bonzini */ 3751e54f1ff2SKai Huang pm_mask = PT_PRESENT_MASK | shadow_me_value; 3752a972e29cSPaolo Bonzini if (mmu->root_role.level >= PT64_ROOT_4LEVEL) { 3753c50d8ae3SPaolo Bonzini pm_mask |= PT_ACCESSED_MASK | PT_WRITABLE_MASK | PT_USER_MASK; 3754c50d8ae3SPaolo Bonzini 375503ca4589SSean Christopherson if (WARN_ON_ONCE(!mmu->pml4_root)) { 37564a38162eSPaolo Bonzini r = -EIO; 37574a38162eSPaolo Bonzini goto out_unlock; 37584a38162eSPaolo Bonzini } 375903ca4589SSean Christopherson mmu->pml4_root[0] = __pa(mmu->pae_root) | pm_mask; 3760cb0f722aSWei Huang 3761a972e29cSPaolo Bonzini if (mmu->root_role.level == PT64_ROOT_5LEVEL) { 3762cb0f722aSWei Huang if (WARN_ON_ONCE(!mmu->pml5_root)) { 3763cb0f722aSWei Huang r = -EIO; 3764cb0f722aSWei Huang goto out_unlock; 3765cb0f722aSWei Huang } 3766cb0f722aSWei Huang mmu->pml5_root[0] = __pa(mmu->pml4_root) | pm_mask; 3767cb0f722aSWei Huang } 376804d45551SSean Christopherson } 376904d45551SSean Christopherson 3770c50d8ae3SPaolo Bonzini for (i = 0; i < 4; ++i) { 3771c834e5e4SSean Christopherson WARN_ON_ONCE(IS_VALID_PAE_ROOT(mmu->pae_root[i])); 37726e6ec584SSean Christopherson 37734d25502aSPaolo Bonzini if (mmu->cpu_role.base.level == PT32E_ROOT_LEVEL) { 37746e0918aeSSean Christopherson if (!(pdptrs[i] & PT_PRESENT_MASK)) { 3775c834e5e4SSean Christopherson mmu->pae_root[i] = INVALID_PAE_ROOT; 3776c50d8ae3SPaolo Bonzini continue; 3777c50d8ae3SPaolo Bonzini } 37786e0918aeSSean Christopherson root_gfn = pdptrs[i] >> PAGE_SHIFT; 3779c50d8ae3SPaolo Bonzini } 3780c50d8ae3SPaolo Bonzini 37817f497775SDavid Matlack /* 37827f497775SDavid Matlack * If shadowing 32-bit non-PAE page tables, each PAE page 37837f497775SDavid Matlack * directory maps one quarter of the guest's non-PAE page 37847f497775SDavid Matlack * directory. Othwerise each PAE page direct shadows one guest 37857f497775SDavid Matlack * PAE page directory so that quadrant should be 0. 37867f497775SDavid Matlack */ 37877f497775SDavid Matlack quadrant = (mmu->cpu_role.base.level == PT32_ROOT_LEVEL) ? i : 0; 37887f497775SDavid Matlack 37897f497775SDavid Matlack root = mmu_alloc_root(vcpu, root_gfn, quadrant, PT32_ROOT_LEVEL); 3790b37233c9SSean Christopherson mmu->pae_root[i] = root | pm_mask; 3791c50d8ae3SPaolo Bonzini } 3792c50d8ae3SPaolo Bonzini 3793a972e29cSPaolo Bonzini if (mmu->root_role.level == PT64_ROOT_5LEVEL) 3794b9e5603cSPaolo Bonzini mmu->root.hpa = __pa(mmu->pml5_root); 3795a972e29cSPaolo Bonzini else if (mmu->root_role.level == PT64_ROOT_4LEVEL) 3796b9e5603cSPaolo Bonzini mmu->root.hpa = __pa(mmu->pml4_root); 3797ba0a194fSSean Christopherson else 3798b9e5603cSPaolo Bonzini mmu->root.hpa = __pa(mmu->pae_root); 3799c50d8ae3SPaolo Bonzini 3800be01e8e2SSean Christopherson set_root_pgd: 3801b9e5603cSPaolo Bonzini mmu->root.pgd = root_pgd; 38024a38162eSPaolo Bonzini out_unlock: 38034a38162eSPaolo Bonzini write_unlock(&vcpu->kvm->mmu_lock); 3804c50d8ae3SPaolo Bonzini 3805c6c937d6SLike Xu return r; 3806c50d8ae3SPaolo Bonzini } 3807c50d8ae3SPaolo Bonzini 3808748e52b9SSean Christopherson static int mmu_alloc_special_roots(struct kvm_vcpu *vcpu) 3809c50d8ae3SPaolo Bonzini { 3810748e52b9SSean Christopherson struct kvm_mmu *mmu = vcpu->arch.mmu; 3811a972e29cSPaolo Bonzini bool need_pml5 = mmu->root_role.level > PT64_ROOT_4LEVEL; 3812cb0f722aSWei Huang u64 *pml5_root = NULL; 3813cb0f722aSWei Huang u64 *pml4_root = NULL; 3814cb0f722aSWei Huang u64 *pae_root; 3815748e52b9SSean Christopherson 3816748e52b9SSean Christopherson /* 3817748e52b9SSean Christopherson * When shadowing 32-bit or PAE NPT with 64-bit NPT, the PML4 and PDP 3818748e52b9SSean Christopherson * tables are allocated and initialized at root creation as there is no 3819748e52b9SSean Christopherson * equivalent level in the guest's NPT to shadow. Allocate the tables 3820748e52b9SSean Christopherson * on demand, as running a 32-bit L1 VMM on 64-bit KVM is very rare. 3821748e52b9SSean Christopherson */ 3822347a0d0dSPaolo Bonzini if (mmu->root_role.direct || 3823347a0d0dSPaolo Bonzini mmu->cpu_role.base.level >= PT64_ROOT_4LEVEL || 3824a972e29cSPaolo Bonzini mmu->root_role.level < PT64_ROOT_4LEVEL) 3825748e52b9SSean Christopherson return 0; 3826748e52b9SSean Christopherson 3827a717a780SSean Christopherson /* 3828a717a780SSean Christopherson * NPT, the only paging mode that uses this horror, uses a fixed number 3829a717a780SSean Christopherson * of levels for the shadow page tables, e.g. all MMUs are 4-level or 3830a717a780SSean Christopherson * all MMus are 5-level. Thus, this can safely require that pml5_root 3831a717a780SSean Christopherson * is allocated if the other roots are valid and pml5 is needed, as any 3832a717a780SSean Christopherson * prior MMU would also have required pml5. 3833a717a780SSean Christopherson */ 3834a717a780SSean Christopherson if (mmu->pae_root && mmu->pml4_root && (!need_pml5 || mmu->pml5_root)) 3835748e52b9SSean Christopherson return 0; 3836748e52b9SSean Christopherson 3837748e52b9SSean Christopherson /* 3838748e52b9SSean Christopherson * The special roots should always be allocated in concert. Yell and 3839748e52b9SSean Christopherson * bail if KVM ends up in a state where only one of the roots is valid. 3840748e52b9SSean Christopherson */ 3841cb0f722aSWei Huang if (WARN_ON_ONCE(!tdp_enabled || mmu->pae_root || mmu->pml4_root || 3842a717a780SSean Christopherson (need_pml5 && mmu->pml5_root))) 3843748e52b9SSean Christopherson return -EIO; 3844748e52b9SSean Christopherson 38454a98623dSSean Christopherson /* 38464a98623dSSean Christopherson * Unlike 32-bit NPT, the PDP table doesn't need to be in low mem, and 38474a98623dSSean Christopherson * doesn't need to be decrypted. 38484a98623dSSean Christopherson */ 3849748e52b9SSean Christopherson pae_root = (void *)get_zeroed_page(GFP_KERNEL_ACCOUNT); 3850748e52b9SSean Christopherson if (!pae_root) 3851748e52b9SSean Christopherson return -ENOMEM; 3852748e52b9SSean Christopherson 3853cb0f722aSWei Huang #ifdef CONFIG_X86_64 385403ca4589SSean Christopherson pml4_root = (void *)get_zeroed_page(GFP_KERNEL_ACCOUNT); 3855cb0f722aSWei Huang if (!pml4_root) 3856cb0f722aSWei Huang goto err_pml4; 3857cb0f722aSWei Huang 3858a717a780SSean Christopherson if (need_pml5) { 3859cb0f722aSWei Huang pml5_root = (void *)get_zeroed_page(GFP_KERNEL_ACCOUNT); 3860cb0f722aSWei Huang if (!pml5_root) 3861cb0f722aSWei Huang goto err_pml5; 3862748e52b9SSean Christopherson } 3863cb0f722aSWei Huang #endif 3864748e52b9SSean Christopherson 3865748e52b9SSean Christopherson mmu->pae_root = pae_root; 386603ca4589SSean Christopherson mmu->pml4_root = pml4_root; 3867cb0f722aSWei Huang mmu->pml5_root = pml5_root; 3868748e52b9SSean Christopherson 3869748e52b9SSean Christopherson return 0; 3870cb0f722aSWei Huang 3871cb0f722aSWei Huang #ifdef CONFIG_X86_64 3872cb0f722aSWei Huang err_pml5: 3873cb0f722aSWei Huang free_page((unsigned long)pml4_root); 3874cb0f722aSWei Huang err_pml4: 3875cb0f722aSWei Huang free_page((unsigned long)pae_root); 3876cb0f722aSWei Huang return -ENOMEM; 3877cb0f722aSWei Huang #endif 3878c50d8ae3SPaolo Bonzini } 3879c50d8ae3SPaolo Bonzini 3880264d3dc1SLai Jiangshan static bool is_unsync_root(hpa_t root) 3881264d3dc1SLai Jiangshan { 3882264d3dc1SLai Jiangshan struct kvm_mmu_page *sp; 3883264d3dc1SLai Jiangshan 388461b05a9fSLai Jiangshan if (!VALID_PAGE(root)) 388561b05a9fSLai Jiangshan return false; 388661b05a9fSLai Jiangshan 3887264d3dc1SLai Jiangshan /* 3888264d3dc1SLai Jiangshan * The read barrier orders the CPU's read of SPTE.W during the page table 3889264d3dc1SLai Jiangshan * walk before the reads of sp->unsync/sp->unsync_children here. 3890264d3dc1SLai Jiangshan * 3891264d3dc1SLai Jiangshan * Even if another CPU was marking the SP as unsync-ed simultaneously, 3892264d3dc1SLai Jiangshan * any guest page table changes are not guaranteed to be visible anyway 3893264d3dc1SLai Jiangshan * until this VCPU issues a TLB flush strictly after those changes are 3894264d3dc1SLai Jiangshan * made. We only need to ensure that the other CPU sets these flags 3895264d3dc1SLai Jiangshan * before any actual changes to the page tables are made. The comments 3896264d3dc1SLai Jiangshan * in mmu_try_to_unsync_pages() describe what could go wrong if this 3897264d3dc1SLai Jiangshan * requirement isn't satisfied. 3898264d3dc1SLai Jiangshan */ 3899264d3dc1SLai Jiangshan smp_rmb(); 3900264d3dc1SLai Jiangshan sp = to_shadow_page(root); 39015d6a3221SSean Christopherson 39025d6a3221SSean Christopherson /* 39035d6a3221SSean Christopherson * PAE roots (somewhat arbitrarily) aren't backed by shadow pages, the 39045d6a3221SSean Christopherson * PDPTEs for a given PAE root need to be synchronized individually. 39055d6a3221SSean Christopherson */ 39065d6a3221SSean Christopherson if (WARN_ON_ONCE(!sp)) 39075d6a3221SSean Christopherson return false; 39085d6a3221SSean Christopherson 3909264d3dc1SLai Jiangshan if (sp->unsync || sp->unsync_children) 3910264d3dc1SLai Jiangshan return true; 3911264d3dc1SLai Jiangshan 3912264d3dc1SLai Jiangshan return false; 3913264d3dc1SLai Jiangshan } 3914264d3dc1SLai Jiangshan 3915c50d8ae3SPaolo Bonzini void kvm_mmu_sync_roots(struct kvm_vcpu *vcpu) 3916c50d8ae3SPaolo Bonzini { 3917c50d8ae3SPaolo Bonzini int i; 3918c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 3919c50d8ae3SPaolo Bonzini 3920347a0d0dSPaolo Bonzini if (vcpu->arch.mmu->root_role.direct) 3921c50d8ae3SPaolo Bonzini return; 3922c50d8ae3SPaolo Bonzini 3923b9e5603cSPaolo Bonzini if (!VALID_PAGE(vcpu->arch.mmu->root.hpa)) 3924c50d8ae3SPaolo Bonzini return; 3925c50d8ae3SPaolo Bonzini 3926c50d8ae3SPaolo Bonzini vcpu_clear_mmio_info(vcpu, MMIO_GVA_ANY); 3927c50d8ae3SPaolo Bonzini 39284d25502aSPaolo Bonzini if (vcpu->arch.mmu->cpu_role.base.level >= PT64_ROOT_4LEVEL) { 3929b9e5603cSPaolo Bonzini hpa_t root = vcpu->arch.mmu->root.hpa; 3930e47c4aeeSSean Christopherson sp = to_shadow_page(root); 3931c50d8ae3SPaolo Bonzini 3932264d3dc1SLai Jiangshan if (!is_unsync_root(root)) 3933c50d8ae3SPaolo Bonzini return; 3934c50d8ae3SPaolo Bonzini 3935531810caSBen Gardon write_lock(&vcpu->kvm->mmu_lock); 393665855ed8SLai Jiangshan mmu_sync_children(vcpu, sp, true); 3937531810caSBen Gardon write_unlock(&vcpu->kvm->mmu_lock); 3938c50d8ae3SPaolo Bonzini return; 3939c50d8ae3SPaolo Bonzini } 3940c50d8ae3SPaolo Bonzini 3941531810caSBen Gardon write_lock(&vcpu->kvm->mmu_lock); 3942c50d8ae3SPaolo Bonzini 3943c50d8ae3SPaolo Bonzini for (i = 0; i < 4; ++i) { 3944c50d8ae3SPaolo Bonzini hpa_t root = vcpu->arch.mmu->pae_root[i]; 3945c50d8ae3SPaolo Bonzini 3946c834e5e4SSean Christopherson if (IS_VALID_PAE_ROOT(root)) { 3947*5e3edd7eSSean Christopherson sp = spte_to_child_sp(root); 394865855ed8SLai Jiangshan mmu_sync_children(vcpu, sp, true); 3949c50d8ae3SPaolo Bonzini } 3950c50d8ae3SPaolo Bonzini } 3951c50d8ae3SPaolo Bonzini 3952531810caSBen Gardon write_unlock(&vcpu->kvm->mmu_lock); 3953c50d8ae3SPaolo Bonzini } 3954c50d8ae3SPaolo Bonzini 395561b05a9fSLai Jiangshan void kvm_mmu_sync_prev_roots(struct kvm_vcpu *vcpu) 395661b05a9fSLai Jiangshan { 395761b05a9fSLai Jiangshan unsigned long roots_to_free = 0; 395861b05a9fSLai Jiangshan int i; 395961b05a9fSLai Jiangshan 396061b05a9fSLai Jiangshan for (i = 0; i < KVM_MMU_NUM_PREV_ROOTS; i++) 396161b05a9fSLai Jiangshan if (is_unsync_root(vcpu->arch.mmu->prev_roots[i].hpa)) 396261b05a9fSLai Jiangshan roots_to_free |= KVM_MMU_ROOT_PREVIOUS(i); 396361b05a9fSLai Jiangshan 396461b05a9fSLai Jiangshan /* sync prev_roots by simply freeing them */ 39650c1c92f1SPaolo Bonzini kvm_mmu_free_roots(vcpu->kvm, vcpu->arch.mmu, roots_to_free); 396661b05a9fSLai Jiangshan } 396761b05a9fSLai Jiangshan 39681f5a21eeSLai Jiangshan static gpa_t nonpaging_gva_to_gpa(struct kvm_vcpu *vcpu, struct kvm_mmu *mmu, 39695b22bbe7SLai Jiangshan gpa_t vaddr, u64 access, 3970c50d8ae3SPaolo Bonzini struct x86_exception *exception) 3971c50d8ae3SPaolo Bonzini { 3972c50d8ae3SPaolo Bonzini if (exception) 3973c50d8ae3SPaolo Bonzini exception->error_code = 0; 3974c59a0f57SLai Jiangshan return kvm_translate_gpa(vcpu, mmu, vaddr, access, exception); 3975c50d8ae3SPaolo Bonzini } 3976c50d8ae3SPaolo Bonzini 3977c50d8ae3SPaolo Bonzini static bool mmio_info_in_cache(struct kvm_vcpu *vcpu, u64 addr, bool direct) 3978c50d8ae3SPaolo Bonzini { 3979c50d8ae3SPaolo Bonzini /* 3980c50d8ae3SPaolo Bonzini * A nested guest cannot use the MMIO cache if it is using nested 3981c50d8ae3SPaolo Bonzini * page tables, because cr2 is a nGPA while the cache stores GPAs. 3982c50d8ae3SPaolo Bonzini */ 3983c50d8ae3SPaolo Bonzini if (mmu_is_nested(vcpu)) 3984c50d8ae3SPaolo Bonzini return false; 3985c50d8ae3SPaolo Bonzini 3986c50d8ae3SPaolo Bonzini if (direct) 3987c50d8ae3SPaolo Bonzini return vcpu_match_mmio_gpa(vcpu, addr); 3988c50d8ae3SPaolo Bonzini 3989c50d8ae3SPaolo Bonzini return vcpu_match_mmio_gva(vcpu, addr); 3990c50d8ae3SPaolo Bonzini } 3991c50d8ae3SPaolo Bonzini 399295fb5b02SBen Gardon /* 399395fb5b02SBen Gardon * Return the level of the lowest level SPTE added to sptes. 399495fb5b02SBen Gardon * That SPTE may be non-present. 3995c5c8c7c5SDavid Matlack * 3996c5c8c7c5SDavid Matlack * Must be called between walk_shadow_page_lockless_{begin,end}. 399795fb5b02SBen Gardon */ 399839b4d43eSSean Christopherson static int get_walk(struct kvm_vcpu *vcpu, u64 addr, u64 *sptes, int *root_level) 3999c50d8ae3SPaolo Bonzini { 4000c50d8ae3SPaolo Bonzini struct kvm_shadow_walk_iterator iterator; 40012aa07893SSean Christopherson int leaf = -1; 400295fb5b02SBen Gardon u64 spte; 4003c50d8ae3SPaolo Bonzini 400439b4d43eSSean Christopherson for (shadow_walk_init(&iterator, vcpu, addr), 400539b4d43eSSean Christopherson *root_level = iterator.level; 4006c50d8ae3SPaolo Bonzini shadow_walk_okay(&iterator); 4007c50d8ae3SPaolo Bonzini __shadow_walk_next(&iterator, spte)) { 400895fb5b02SBen Gardon leaf = iterator.level; 4009c50d8ae3SPaolo Bonzini spte = mmu_spte_get_lockless(iterator.sptep); 4010c50d8ae3SPaolo Bonzini 4011dde81f94SSean Christopherson sptes[leaf] = spte; 401295fb5b02SBen Gardon } 401395fb5b02SBen Gardon 401495fb5b02SBen Gardon return leaf; 401595fb5b02SBen Gardon } 401695fb5b02SBen Gardon 40179aa41879SSean Christopherson /* return true if reserved bit(s) are detected on a valid, non-MMIO SPTE. */ 401895fb5b02SBen Gardon static bool get_mmio_spte(struct kvm_vcpu *vcpu, u64 addr, u64 *sptep) 401995fb5b02SBen Gardon { 4020dde81f94SSean Christopherson u64 sptes[PT64_ROOT_MAX_LEVEL + 1]; 402195fb5b02SBen Gardon struct rsvd_bits_validate *rsvd_check; 402239b4d43eSSean Christopherson int root, leaf, level; 402395fb5b02SBen Gardon bool reserved = false; 402495fb5b02SBen Gardon 4025c5c8c7c5SDavid Matlack walk_shadow_page_lockless_begin(vcpu); 4026c5c8c7c5SDavid Matlack 402763c0cac9SDavid Matlack if (is_tdp_mmu(vcpu->arch.mmu)) 402839b4d43eSSean Christopherson leaf = kvm_tdp_mmu_get_walk(vcpu, addr, sptes, &root); 402995fb5b02SBen Gardon else 403039b4d43eSSean Christopherson leaf = get_walk(vcpu, addr, sptes, &root); 403195fb5b02SBen Gardon 4032c5c8c7c5SDavid Matlack walk_shadow_page_lockless_end(vcpu); 4033c5c8c7c5SDavid Matlack 40342aa07893SSean Christopherson if (unlikely(leaf < 0)) { 40352aa07893SSean Christopherson *sptep = 0ull; 40362aa07893SSean Christopherson return reserved; 40372aa07893SSean Christopherson } 40382aa07893SSean Christopherson 40399aa41879SSean Christopherson *sptep = sptes[leaf]; 40409aa41879SSean Christopherson 40419aa41879SSean Christopherson /* 40429aa41879SSean Christopherson * Skip reserved bits checks on the terminal leaf if it's not a valid 40439aa41879SSean Christopherson * SPTE. Note, this also (intentionally) skips MMIO SPTEs, which, by 40449aa41879SSean Christopherson * design, always have reserved bits set. The purpose of the checks is 40459aa41879SSean Christopherson * to detect reserved bits on non-MMIO SPTEs. i.e. buggy SPTEs. 40469aa41879SSean Christopherson */ 40479aa41879SSean Christopherson if (!is_shadow_present_pte(sptes[leaf])) 40489aa41879SSean Christopherson leaf++; 404995fb5b02SBen Gardon 405095fb5b02SBen Gardon rsvd_check = &vcpu->arch.mmu->shadow_zero_check; 405195fb5b02SBen Gardon 40529aa41879SSean Christopherson for (level = root; level >= leaf; level--) 4053961f8445SSean Christopherson reserved |= is_rsvd_spte(rsvd_check, sptes[level], level); 4054c50d8ae3SPaolo Bonzini 4055c50d8ae3SPaolo Bonzini if (reserved) { 4056bb4cdf3aSSean Christopherson pr_err("%s: reserved bits set on MMU-present spte, addr 0x%llx, hierarchy:\n", 4057c50d8ae3SPaolo Bonzini __func__, addr); 405895fb5b02SBen Gardon for (level = root; level >= leaf; level--) 4059bb4cdf3aSSean Christopherson pr_err("------ spte = 0x%llx level = %d, rsvd bits = 0x%llx", 4060bb4cdf3aSSean Christopherson sptes[level], level, 4061961f8445SSean Christopherson get_rsvd_bits(rsvd_check, sptes[level], level)); 4062c50d8ae3SPaolo Bonzini } 4063ddce6208SSean Christopherson 4064c50d8ae3SPaolo Bonzini return reserved; 4065c50d8ae3SPaolo Bonzini } 4066c50d8ae3SPaolo Bonzini 4067c50d8ae3SPaolo Bonzini static int handle_mmio_page_fault(struct kvm_vcpu *vcpu, u64 addr, bool direct) 4068c50d8ae3SPaolo Bonzini { 4069c50d8ae3SPaolo Bonzini u64 spte; 4070c50d8ae3SPaolo Bonzini bool reserved; 4071c50d8ae3SPaolo Bonzini 4072c50d8ae3SPaolo Bonzini if (mmio_info_in_cache(vcpu, addr, direct)) 4073c50d8ae3SPaolo Bonzini return RET_PF_EMULATE; 4074c50d8ae3SPaolo Bonzini 407595fb5b02SBen Gardon reserved = get_mmio_spte(vcpu, addr, &spte); 4076c50d8ae3SPaolo Bonzini if (WARN_ON(reserved)) 4077c50d8ae3SPaolo Bonzini return -EINVAL; 4078c50d8ae3SPaolo Bonzini 4079c50d8ae3SPaolo Bonzini if (is_mmio_spte(spte)) { 4080c50d8ae3SPaolo Bonzini gfn_t gfn = get_mmio_spte_gfn(spte); 40810a2b64c5SBen Gardon unsigned int access = get_mmio_spte_access(spte); 4082c50d8ae3SPaolo Bonzini 4083c50d8ae3SPaolo Bonzini if (!check_mmio_spte(vcpu, spte)) 4084c50d8ae3SPaolo Bonzini return RET_PF_INVALID; 4085c50d8ae3SPaolo Bonzini 4086c50d8ae3SPaolo Bonzini if (direct) 4087c50d8ae3SPaolo Bonzini addr = 0; 4088c50d8ae3SPaolo Bonzini 4089c50d8ae3SPaolo Bonzini trace_handle_mmio_page_fault(addr, gfn, access); 4090c50d8ae3SPaolo Bonzini vcpu_cache_mmio_info(vcpu, addr, gfn, access); 4091c50d8ae3SPaolo Bonzini return RET_PF_EMULATE; 4092c50d8ae3SPaolo Bonzini } 4093c50d8ae3SPaolo Bonzini 4094c50d8ae3SPaolo Bonzini /* 4095c50d8ae3SPaolo Bonzini * If the page table is zapped by other cpus, let CPU fault again on 4096c50d8ae3SPaolo Bonzini * the address. 4097c50d8ae3SPaolo Bonzini */ 4098c50d8ae3SPaolo Bonzini return RET_PF_RETRY; 4099c50d8ae3SPaolo Bonzini } 4100c50d8ae3SPaolo Bonzini 4101c50d8ae3SPaolo Bonzini static bool page_fault_handle_page_track(struct kvm_vcpu *vcpu, 4102b8a5d551SPaolo Bonzini struct kvm_page_fault *fault) 4103c50d8ae3SPaolo Bonzini { 4104b8a5d551SPaolo Bonzini if (unlikely(fault->rsvd)) 4105c50d8ae3SPaolo Bonzini return false; 4106c50d8ae3SPaolo Bonzini 4107b8a5d551SPaolo Bonzini if (!fault->present || !fault->write) 4108c50d8ae3SPaolo Bonzini return false; 4109c50d8ae3SPaolo Bonzini 4110c50d8ae3SPaolo Bonzini /* 4111c50d8ae3SPaolo Bonzini * guest is writing the page which is write tracked which can 4112c50d8ae3SPaolo Bonzini * not be fixed by page fault handler. 4113c50d8ae3SPaolo Bonzini */ 41149d395a0aSBen Gardon if (kvm_slot_page_track_is_active(vcpu->kvm, fault->slot, fault->gfn, KVM_PAGE_TRACK_WRITE)) 4115c50d8ae3SPaolo Bonzini return true; 4116c50d8ae3SPaolo Bonzini 4117c50d8ae3SPaolo Bonzini return false; 4118c50d8ae3SPaolo Bonzini } 4119c50d8ae3SPaolo Bonzini 4120c50d8ae3SPaolo Bonzini static void shadow_page_table_clear_flood(struct kvm_vcpu *vcpu, gva_t addr) 4121c50d8ae3SPaolo Bonzini { 4122c50d8ae3SPaolo Bonzini struct kvm_shadow_walk_iterator iterator; 4123c50d8ae3SPaolo Bonzini u64 spte; 4124c50d8ae3SPaolo Bonzini 4125c50d8ae3SPaolo Bonzini walk_shadow_page_lockless_begin(vcpu); 41263e44dce4SLai Jiangshan for_each_shadow_entry_lockless(vcpu, addr, iterator, spte) 4127c50d8ae3SPaolo Bonzini clear_sp_write_flooding_count(iterator.sptep); 4128c50d8ae3SPaolo Bonzini walk_shadow_page_lockless_end(vcpu); 4129c50d8ae3SPaolo Bonzini } 4130c50d8ae3SPaolo Bonzini 41316f3c1fc5SLiang Zhang static u32 alloc_apf_token(struct kvm_vcpu *vcpu) 41326f3c1fc5SLiang Zhang { 41336f3c1fc5SLiang Zhang /* make sure the token value is not 0 */ 41346f3c1fc5SLiang Zhang u32 id = vcpu->arch.apf.id; 41356f3c1fc5SLiang Zhang 41366f3c1fc5SLiang Zhang if (id << 12 == 0) 41376f3c1fc5SLiang Zhang vcpu->arch.apf.id = 1; 41386f3c1fc5SLiang Zhang 41396f3c1fc5SLiang Zhang return (vcpu->arch.apf.id++ << 12) | vcpu->vcpu_id; 41406f3c1fc5SLiang Zhang } 41416f3c1fc5SLiang Zhang 4142e8c22266SVitaly Kuznetsov static bool kvm_arch_setup_async_pf(struct kvm_vcpu *vcpu, gpa_t cr2_or_gpa, 41439f1a8526SSean Christopherson gfn_t gfn) 4144c50d8ae3SPaolo Bonzini { 4145c50d8ae3SPaolo Bonzini struct kvm_arch_async_pf arch; 4146c50d8ae3SPaolo Bonzini 41476f3c1fc5SLiang Zhang arch.token = alloc_apf_token(vcpu); 4148c50d8ae3SPaolo Bonzini arch.gfn = gfn; 4149347a0d0dSPaolo Bonzini arch.direct_map = vcpu->arch.mmu->root_role.direct; 4150d8dd54e0SSean Christopherson arch.cr3 = vcpu->arch.mmu->get_guest_pgd(vcpu); 4151c50d8ae3SPaolo Bonzini 41529f1a8526SSean Christopherson return kvm_setup_async_pf(vcpu, cr2_or_gpa, 41539f1a8526SSean Christopherson kvm_vcpu_gfn_to_hva(vcpu, gfn), &arch); 4154c50d8ae3SPaolo Bonzini } 4155c50d8ae3SPaolo Bonzini 41568a009d5bSSean Christopherson void kvm_arch_async_page_ready(struct kvm_vcpu *vcpu, struct kvm_async_pf *work) 41578a009d5bSSean Christopherson { 41588a009d5bSSean Christopherson int r; 41598a009d5bSSean Christopherson 41608a009d5bSSean Christopherson if ((vcpu->arch.mmu->root_role.direct != work->arch.direct_map) || 41618a009d5bSSean Christopherson work->wakeup_all) 41628a009d5bSSean Christopherson return; 41638a009d5bSSean Christopherson 41648a009d5bSSean Christopherson r = kvm_mmu_reload(vcpu); 41658a009d5bSSean Christopherson if (unlikely(r)) 41668a009d5bSSean Christopherson return; 41678a009d5bSSean Christopherson 41688a009d5bSSean Christopherson if (!vcpu->arch.mmu->root_role.direct && 41698a009d5bSSean Christopherson work->arch.cr3 != vcpu->arch.mmu->get_guest_pgd(vcpu)) 41708a009d5bSSean Christopherson return; 41718a009d5bSSean Christopherson 41728a009d5bSSean Christopherson kvm_mmu_do_page_fault(vcpu, work->cr2_or_gpa, 0, true); 41738a009d5bSSean Christopherson } 41748a009d5bSSean Christopherson 41755276c616SSean Christopherson static int kvm_faultin_pfn(struct kvm_vcpu *vcpu, struct kvm_page_fault *fault) 4176c50d8ae3SPaolo Bonzini { 4177e710c5f6SDavid Matlack struct kvm_memory_slot *slot = fault->slot; 4178c50d8ae3SPaolo Bonzini bool async; 4179c50d8ae3SPaolo Bonzini 4180e0c37868SSean Christopherson /* 4181e0c37868SSean Christopherson * Retry the page fault if the gfn hit a memslot that is being deleted 4182e0c37868SSean Christopherson * or moved. This ensures any existing SPTEs for the old memslot will 4183e0c37868SSean Christopherson * be zapped before KVM inserts a new MMIO SPTE for the gfn. 4184e0c37868SSean Christopherson */ 4185e0c37868SSean Christopherson if (slot && (slot->flags & KVM_MEMSLOT_INVALID)) 41865276c616SSean Christopherson return RET_PF_RETRY; 4187e0c37868SSean Christopherson 41889cc13d60SMaxim Levitsky if (!kvm_is_visible_memslot(slot)) { 4189c36b7150SPaolo Bonzini /* Don't expose private memslots to L2. */ 41909cc13d60SMaxim Levitsky if (is_guest_mode(vcpu)) { 4191e710c5f6SDavid Matlack fault->slot = NULL; 41923647cd04SPaolo Bonzini fault->pfn = KVM_PFN_NOSLOT; 41933647cd04SPaolo Bonzini fault->map_writable = false; 41945276c616SSean Christopherson return RET_PF_CONTINUE; 4195c50d8ae3SPaolo Bonzini } 41969cc13d60SMaxim Levitsky /* 41979cc13d60SMaxim Levitsky * If the APIC access page exists but is disabled, go directly 41989cc13d60SMaxim Levitsky * to emulation without caching the MMIO access or creating a 41999cc13d60SMaxim Levitsky * MMIO SPTE. That way the cache doesn't need to be purged 42009cc13d60SMaxim Levitsky * when the AVIC is re-enabled. 42019cc13d60SMaxim Levitsky */ 42029cc13d60SMaxim Levitsky if (slot && slot->id == APIC_ACCESS_PAGE_PRIVATE_MEMSLOT && 42035276c616SSean Christopherson !kvm_apicv_activated(vcpu->kvm)) 42045276c616SSean Christopherson return RET_PF_EMULATE; 42059cc13d60SMaxim Levitsky } 4206c50d8ae3SPaolo Bonzini 4207c50d8ae3SPaolo Bonzini async = false; 4208c8b88b33SPeter Xu fault->pfn = __gfn_to_pfn_memslot(slot, fault->gfn, false, false, &async, 42093647cd04SPaolo Bonzini fault->write, &fault->map_writable, 42103647cd04SPaolo Bonzini &fault->hva); 4211c50d8ae3SPaolo Bonzini if (!async) 42125276c616SSean Christopherson return RET_PF_CONTINUE; /* *pfn has correct page already */ 4213c50d8ae3SPaolo Bonzini 42142839180cSPaolo Bonzini if (!fault->prefetch && kvm_can_do_async_pf(vcpu)) { 42153647cd04SPaolo Bonzini trace_kvm_try_async_get_page(fault->addr, fault->gfn); 42163647cd04SPaolo Bonzini if (kvm_find_async_pf_gfn(vcpu, fault->gfn)) { 42171685c0f3SMingwei Zhang trace_kvm_async_pf_repeated_fault(fault->addr, fault->gfn); 4218c50d8ae3SPaolo Bonzini kvm_make_request(KVM_REQ_APF_HALT, vcpu); 42195276c616SSean Christopherson return RET_PF_RETRY; 42205276c616SSean Christopherson } else if (kvm_arch_setup_async_pf(vcpu, fault->addr, fault->gfn)) { 42215276c616SSean Christopherson return RET_PF_RETRY; 42225276c616SSean Christopherson } 4223c50d8ae3SPaolo Bonzini } 4224c50d8ae3SPaolo Bonzini 422576657687SPeter Xu /* 422676657687SPeter Xu * Allow gup to bail on pending non-fatal signals when it's also allowed 422776657687SPeter Xu * to wait for IO. Note, gup always bails if it is unable to quickly 422876657687SPeter Xu * get a page and a fatal signal, i.e. SIGKILL, is pending. 422976657687SPeter Xu */ 423076657687SPeter Xu fault->pfn = __gfn_to_pfn_memslot(slot, fault->gfn, false, true, NULL, 42313647cd04SPaolo Bonzini fault->write, &fault->map_writable, 42323647cd04SPaolo Bonzini &fault->hva); 42335276c616SSean Christopherson return RET_PF_CONTINUE; 4234c50d8ae3SPaolo Bonzini } 4235c50d8ae3SPaolo Bonzini 4236a955cad8SSean Christopherson /* 4237a955cad8SSean Christopherson * Returns true if the page fault is stale and needs to be retried, i.e. if the 4238a955cad8SSean Christopherson * root was invalidated by a memslot update or a relevant mmu_notifier fired. 4239a955cad8SSean Christopherson */ 4240a955cad8SSean Christopherson static bool is_page_fault_stale(struct kvm_vcpu *vcpu, 4241a955cad8SSean Christopherson struct kvm_page_fault *fault, int mmu_seq) 4242a955cad8SSean Christopherson { 4243b9e5603cSPaolo Bonzini struct kvm_mmu_page *sp = to_shadow_page(vcpu->arch.mmu->root.hpa); 424418c841e1SSean Christopherson 424518c841e1SSean Christopherson /* Special roots, e.g. pae_root, are not backed by shadow pages. */ 424618c841e1SSean Christopherson if (sp && is_obsolete_sp(vcpu->kvm, sp)) 424718c841e1SSean Christopherson return true; 424818c841e1SSean Christopherson 424918c841e1SSean Christopherson /* 425018c841e1SSean Christopherson * Roots without an associated shadow page are considered invalid if 425118c841e1SSean Christopherson * there is a pending request to free obsolete roots. The request is 425218c841e1SSean Christopherson * only a hint that the current root _may_ be obsolete and needs to be 425318c841e1SSean Christopherson * reloaded, e.g. if the guest frees a PGD that KVM is tracking as a 425418c841e1SSean Christopherson * previous root, then __kvm_mmu_prepare_zap_page() signals all vCPUs 425518c841e1SSean Christopherson * to reload even if no vCPU is actively using the root. 425618c841e1SSean Christopherson */ 4257527d5cd7SSean Christopherson if (!sp && kvm_test_request(KVM_REQ_MMU_FREE_OBSOLETE_ROOTS, vcpu)) 4258a955cad8SSean Christopherson return true; 4259a955cad8SSean Christopherson 4260a955cad8SSean Christopherson return fault->slot && 426120ec3ebdSChao Peng mmu_invalidate_retry_hva(vcpu->kvm, mmu_seq, fault->hva); 4262a955cad8SSean Christopherson } 4263a955cad8SSean Christopherson 42644326e57eSPaolo Bonzini static int direct_page_fault(struct kvm_vcpu *vcpu, struct kvm_page_fault *fault) 4265c50d8ae3SPaolo Bonzini { 426663c0cac9SDavid Matlack bool is_tdp_mmu_fault = is_tdp_mmu(vcpu->arch.mmu); 4267c50d8ae3SPaolo Bonzini 42680f90e1c1SSean Christopherson unsigned long mmu_seq; 426983f06fa7SSean Christopherson int r; 4270c50d8ae3SPaolo Bonzini 42713c8ad5a6SPaolo Bonzini fault->gfn = fault->addr >> PAGE_SHIFT; 4272e710c5f6SDavid Matlack fault->slot = kvm_vcpu_gfn_to_memslot(vcpu, fault->gfn); 4273e710c5f6SDavid Matlack 4274b8a5d551SPaolo Bonzini if (page_fault_handle_page_track(vcpu, fault)) 4275c50d8ae3SPaolo Bonzini return RET_PF_EMULATE; 4276c50d8ae3SPaolo Bonzini 42773c8ad5a6SPaolo Bonzini r = fast_page_fault(vcpu, fault); 4278c4371c2aSSean Christopherson if (r != RET_PF_INVALID) 4279c4371c2aSSean Christopherson return r; 428083291445SSean Christopherson 4281378f5cd6SSean Christopherson r = mmu_topup_memory_caches(vcpu, false); 4282c50d8ae3SPaolo Bonzini if (r) 4283c50d8ae3SPaolo Bonzini return r; 4284c50d8ae3SPaolo Bonzini 428520ec3ebdSChao Peng mmu_seq = vcpu->kvm->mmu_invalidate_seq; 4286367fd790SSean Christopherson smp_rmb(); 4287367fd790SSean Christopherson 42885276c616SSean Christopherson r = kvm_faultin_pfn(vcpu, fault); 42895276c616SSean Christopherson if (r != RET_PF_CONTINUE) 42908f32d5e5SMaxim Levitsky return r; 4291367fd790SSean Christopherson 42925276c616SSean Christopherson r = handle_abnormal_pfn(vcpu, fault, ACC_ALL); 42935276c616SSean Christopherson if (r != RET_PF_CONTINUE) 4294367fd790SSean Christopherson return r; 4295367fd790SSean Christopherson 4296367fd790SSean Christopherson r = RET_PF_RETRY; 4297a2855afcSBen Gardon 42980b873fd7SDavid Matlack if (is_tdp_mmu_fault) 4299a2855afcSBen Gardon read_lock(&vcpu->kvm->mmu_lock); 4300a2855afcSBen Gardon else 4301531810caSBen Gardon write_lock(&vcpu->kvm->mmu_lock); 4302a2855afcSBen Gardon 4303a955cad8SSean Christopherson if (is_page_fault_stale(vcpu, fault, mmu_seq)) 4304367fd790SSean Christopherson goto out_unlock; 4305a955cad8SSean Christopherson 43067bd7ded6SSean Christopherson r = make_mmu_pages_available(vcpu); 43077bd7ded6SSean Christopherson if (r) 4308367fd790SSean Christopherson goto out_unlock; 4309bb18842eSBen Gardon 43100b873fd7SDavid Matlack if (is_tdp_mmu_fault) 43112f6305ddSPaolo Bonzini r = kvm_tdp_mmu_map(vcpu, fault); 4312bb18842eSBen Gardon else 431343b74355SPaolo Bonzini r = __direct_map(vcpu, fault); 43140f90e1c1SSean Christopherson 4315367fd790SSean Christopherson out_unlock: 43160b873fd7SDavid Matlack if (is_tdp_mmu_fault) 4317a2855afcSBen Gardon read_unlock(&vcpu->kvm->mmu_lock); 4318a2855afcSBen Gardon else 4319531810caSBen Gardon write_unlock(&vcpu->kvm->mmu_lock); 43203647cd04SPaolo Bonzini kvm_release_pfn_clean(fault->pfn); 4321367fd790SSean Christopherson return r; 4322c50d8ae3SPaolo Bonzini } 4323c50d8ae3SPaolo Bonzini 4324c501040aSPaolo Bonzini static int nonpaging_page_fault(struct kvm_vcpu *vcpu, 4325c501040aSPaolo Bonzini struct kvm_page_fault *fault) 43260f90e1c1SSean Christopherson { 43274326e57eSPaolo Bonzini pgprintk("%s: gva %lx error %x\n", __func__, fault->addr, fault->error_code); 43280f90e1c1SSean Christopherson 43290f90e1c1SSean Christopherson /* This path builds a PAE pagetable, we can map 2mb pages at maximum. */ 43304326e57eSPaolo Bonzini fault->max_level = PG_LEVEL_2M; 43314326e57eSPaolo Bonzini return direct_page_fault(vcpu, fault); 43320f90e1c1SSean Christopherson } 43330f90e1c1SSean Christopherson 4334c50d8ae3SPaolo Bonzini int kvm_handle_page_fault(struct kvm_vcpu *vcpu, u64 error_code, 4335c50d8ae3SPaolo Bonzini u64 fault_address, char *insn, int insn_len) 4336c50d8ae3SPaolo Bonzini { 4337c50d8ae3SPaolo Bonzini int r = 1; 43389ce372b3SVitaly Kuznetsov u32 flags = vcpu->arch.apf.host_apf_flags; 4339c50d8ae3SPaolo Bonzini 4340736c291cSSean Christopherson #ifndef CONFIG_X86_64 4341736c291cSSean Christopherson /* A 64-bit CR2 should be impossible on 32-bit KVM. */ 4342736c291cSSean Christopherson if (WARN_ON_ONCE(fault_address >> 32)) 4343736c291cSSean Christopherson return -EFAULT; 4344736c291cSSean Christopherson #endif 4345736c291cSSean Christopherson 4346c50d8ae3SPaolo Bonzini vcpu->arch.l1tf_flush_l1d = true; 43479ce372b3SVitaly Kuznetsov if (!flags) { 4348faa03b39SWonhyuk Yang trace_kvm_page_fault(vcpu, fault_address, error_code); 4349c50d8ae3SPaolo Bonzini 4350c50d8ae3SPaolo Bonzini if (kvm_event_needs_reinjection(vcpu)) 4351c50d8ae3SPaolo Bonzini kvm_mmu_unprotect_page_virt(vcpu, fault_address); 4352c50d8ae3SPaolo Bonzini r = kvm_mmu_page_fault(vcpu, fault_address, error_code, insn, 4353c50d8ae3SPaolo Bonzini insn_len); 43549ce372b3SVitaly Kuznetsov } else if (flags & KVM_PV_REASON_PAGE_NOT_PRESENT) { 435568fd66f1SVitaly Kuznetsov vcpu->arch.apf.host_apf_flags = 0; 4356c50d8ae3SPaolo Bonzini local_irq_disable(); 43576bca69adSThomas Gleixner kvm_async_pf_task_wait_schedule(fault_address); 4358c50d8ae3SPaolo Bonzini local_irq_enable(); 43599ce372b3SVitaly Kuznetsov } else { 43609ce372b3SVitaly Kuznetsov WARN_ONCE(1, "Unexpected host async PF flags: %x\n", flags); 4361c50d8ae3SPaolo Bonzini } 43629ce372b3SVitaly Kuznetsov 4363c50d8ae3SPaolo Bonzini return r; 4364c50d8ae3SPaolo Bonzini } 4365c50d8ae3SPaolo Bonzini EXPORT_SYMBOL_GPL(kvm_handle_page_fault); 4366c50d8ae3SPaolo Bonzini 4367c501040aSPaolo Bonzini int kvm_tdp_page_fault(struct kvm_vcpu *vcpu, struct kvm_page_fault *fault) 4368c50d8ae3SPaolo Bonzini { 4369d5e90a69SSean Christopherson /* 4370d5e90a69SSean Christopherson * If the guest's MTRRs may be used to compute the "real" memtype, 4371d5e90a69SSean Christopherson * restrict the mapping level to ensure KVM uses a consistent memtype 4372d5e90a69SSean Christopherson * across the entire mapping. If the host MTRRs are ignored by TDP 4373d5e90a69SSean Christopherson * (shadow_memtype_mask is non-zero), and the VM has non-coherent DMA 4374d5e90a69SSean Christopherson * (DMA doesn't snoop CPU caches), KVM's ABI is to honor the memtype 4375d5e90a69SSean Christopherson * from the guest's MTRRs so that guest accesses to memory that is 4376d5e90a69SSean Christopherson * DMA'd aren't cached against the guest's wishes. 4377d5e90a69SSean Christopherson * 4378d5e90a69SSean Christopherson * Note, KVM may still ultimately ignore guest MTRRs for certain PFNs, 4379d5e90a69SSean Christopherson * e.g. KVM will force UC memtype for host MMIO. 4380d5e90a69SSean Christopherson */ 4381d5e90a69SSean Christopherson if (shadow_memtype_mask && kvm_arch_has_noncoherent_dma(vcpu->kvm)) { 4382d5e90a69SSean Christopherson for ( ; fault->max_level > PG_LEVEL_4K; --fault->max_level) { 43834326e57eSPaolo Bonzini int page_num = KVM_PAGES_PER_HPAGE(fault->max_level); 43844326e57eSPaolo Bonzini gfn_t base = (fault->addr >> PAGE_SHIFT) & ~(page_num - 1); 4385c50d8ae3SPaolo Bonzini 4386cb9b88c6SSean Christopherson if (kvm_mtrr_check_gfn_range_consistency(vcpu, base, page_num)) 4387cb9b88c6SSean Christopherson break; 4388d5e90a69SSean Christopherson } 4389c50d8ae3SPaolo Bonzini } 4390c50d8ae3SPaolo Bonzini 43914326e57eSPaolo Bonzini return direct_page_fault(vcpu, fault); 4392c50d8ae3SPaolo Bonzini } 4393c50d8ae3SPaolo Bonzini 439484a16226SSean Christopherson static void nonpaging_init_context(struct kvm_mmu *context) 4395c50d8ae3SPaolo Bonzini { 4396c50d8ae3SPaolo Bonzini context->page_fault = nonpaging_page_fault; 4397c50d8ae3SPaolo Bonzini context->gva_to_gpa = nonpaging_gva_to_gpa; 4398c50d8ae3SPaolo Bonzini context->sync_page = nonpaging_sync_page; 43995efac074SPaolo Bonzini context->invlpg = NULL; 4400c50d8ae3SPaolo Bonzini } 4401c50d8ae3SPaolo Bonzini 4402be01e8e2SSean Christopherson static inline bool is_root_usable(struct kvm_mmu_root_info *root, gpa_t pgd, 44030be44352SSean Christopherson union kvm_mmu_page_role role) 44040be44352SSean Christopherson { 4405be01e8e2SSean Christopherson return (role.direct || pgd == root->pgd) && 44065499ea73SPaolo Bonzini VALID_PAGE(root->hpa) && 4407e47c4aeeSSean Christopherson role.word == to_shadow_page(root->hpa)->role.word; 44080be44352SSean Christopherson } 44090be44352SSean Christopherson 4410c50d8ae3SPaolo Bonzini /* 44115499ea73SPaolo Bonzini * Find out if a previously cached root matching the new pgd/role is available, 44125499ea73SPaolo Bonzini * and insert the current root as the MRU in the cache. 44135499ea73SPaolo Bonzini * If a matching root is found, it is assigned to kvm_mmu->root and 44145499ea73SPaolo Bonzini * true is returned. 44155499ea73SPaolo Bonzini * If no match is found, kvm_mmu->root is left invalid, the LRU root is 44165499ea73SPaolo Bonzini * evicted to make room for the current root, and false is returned. 4417c50d8ae3SPaolo Bonzini */ 44185499ea73SPaolo Bonzini static bool cached_root_find_and_keep_current(struct kvm *kvm, struct kvm_mmu *mmu, 44195499ea73SPaolo Bonzini gpa_t new_pgd, 4420c50d8ae3SPaolo Bonzini union kvm_mmu_page_role new_role) 4421c50d8ae3SPaolo Bonzini { 4422c50d8ae3SPaolo Bonzini uint i; 4423c50d8ae3SPaolo Bonzini 4424b9e5603cSPaolo Bonzini if (is_root_usable(&mmu->root, new_pgd, new_role)) 44250be44352SSean Christopherson return true; 44260be44352SSean Christopherson 4427c50d8ae3SPaolo Bonzini for (i = 0; i < KVM_MMU_NUM_PREV_ROOTS; i++) { 44285499ea73SPaolo Bonzini /* 44295499ea73SPaolo Bonzini * The swaps end up rotating the cache like this: 44305499ea73SPaolo Bonzini * C 0 1 2 3 (on entry to the function) 44315499ea73SPaolo Bonzini * 0 C 1 2 3 44325499ea73SPaolo Bonzini * 1 C 0 2 3 44335499ea73SPaolo Bonzini * 2 C 0 1 3 44345499ea73SPaolo Bonzini * 3 C 0 1 2 (on exit from the loop) 44355499ea73SPaolo Bonzini */ 4436b9e5603cSPaolo Bonzini swap(mmu->root, mmu->prev_roots[i]); 4437b9e5603cSPaolo Bonzini if (is_root_usable(&mmu->root, new_pgd, new_role)) 44385499ea73SPaolo Bonzini return true; 4439c50d8ae3SPaolo Bonzini } 4440c50d8ae3SPaolo Bonzini 44415499ea73SPaolo Bonzini kvm_mmu_free_roots(kvm, mmu, KVM_MMU_ROOT_CURRENT); 44425499ea73SPaolo Bonzini return false; 4443c50d8ae3SPaolo Bonzini } 4444c50d8ae3SPaolo Bonzini 4445c50d8ae3SPaolo Bonzini /* 44465499ea73SPaolo Bonzini * Find out if a previously cached root matching the new pgd/role is available. 44475499ea73SPaolo Bonzini * On entry, mmu->root is invalid. 44485499ea73SPaolo Bonzini * If a matching root is found, it is assigned to kvm_mmu->root, the LRU entry 44495499ea73SPaolo Bonzini * of the cache becomes invalid, and true is returned. 44505499ea73SPaolo Bonzini * If no match is found, kvm_mmu->root is left invalid and false is returned. 44515499ea73SPaolo Bonzini */ 44525499ea73SPaolo Bonzini static bool cached_root_find_without_current(struct kvm *kvm, struct kvm_mmu *mmu, 44535499ea73SPaolo Bonzini gpa_t new_pgd, 44545499ea73SPaolo Bonzini union kvm_mmu_page_role new_role) 44555499ea73SPaolo Bonzini { 44565499ea73SPaolo Bonzini uint i; 44575499ea73SPaolo Bonzini 44585499ea73SPaolo Bonzini for (i = 0; i < KVM_MMU_NUM_PREV_ROOTS; i++) 44595499ea73SPaolo Bonzini if (is_root_usable(&mmu->prev_roots[i], new_pgd, new_role)) 44605499ea73SPaolo Bonzini goto hit; 44615499ea73SPaolo Bonzini 44625499ea73SPaolo Bonzini return false; 44635499ea73SPaolo Bonzini 44645499ea73SPaolo Bonzini hit: 44655499ea73SPaolo Bonzini swap(mmu->root, mmu->prev_roots[i]); 44665499ea73SPaolo Bonzini /* Bubble up the remaining roots. */ 44675499ea73SPaolo Bonzini for (; i < KVM_MMU_NUM_PREV_ROOTS - 1; i++) 44685499ea73SPaolo Bonzini mmu->prev_roots[i] = mmu->prev_roots[i + 1]; 44695499ea73SPaolo Bonzini mmu->prev_roots[i].hpa = INVALID_PAGE; 44705499ea73SPaolo Bonzini return true; 44715499ea73SPaolo Bonzini } 44725499ea73SPaolo Bonzini 44735499ea73SPaolo Bonzini static bool fast_pgd_switch(struct kvm *kvm, struct kvm_mmu *mmu, 44745499ea73SPaolo Bonzini gpa_t new_pgd, union kvm_mmu_page_role new_role) 44755499ea73SPaolo Bonzini { 44765499ea73SPaolo Bonzini /* 44775499ea73SPaolo Bonzini * For now, limit the caching to 64-bit hosts+VMs in order to avoid 4478c50d8ae3SPaolo Bonzini * having to deal with PDPTEs. We may add support for 32-bit hosts/VMs 4479c50d8ae3SPaolo Bonzini * later if necessary. 4480c50d8ae3SPaolo Bonzini */ 44815499ea73SPaolo Bonzini if (VALID_PAGE(mmu->root.hpa) && !to_shadow_page(mmu->root.hpa)) 44825499ea73SPaolo Bonzini kvm_mmu_free_roots(kvm, mmu, KVM_MMU_ROOT_CURRENT); 4483c50d8ae3SPaolo Bonzini 44845499ea73SPaolo Bonzini if (VALID_PAGE(mmu->root.hpa)) 44855499ea73SPaolo Bonzini return cached_root_find_and_keep_current(kvm, mmu, new_pgd, new_role); 44865499ea73SPaolo Bonzini else 44875499ea73SPaolo Bonzini return cached_root_find_without_current(kvm, mmu, new_pgd, new_role); 4488c50d8ae3SPaolo Bonzini } 4489c50d8ae3SPaolo Bonzini 4490d2e5f333SPaolo Bonzini void kvm_mmu_new_pgd(struct kvm_vcpu *vcpu, gpa_t new_pgd) 4491c50d8ae3SPaolo Bonzini { 44920c1c92f1SPaolo Bonzini struct kvm_mmu *mmu = vcpu->arch.mmu; 44937a458f0eSPaolo Bonzini union kvm_mmu_page_role new_role = mmu->root_role; 44940c1c92f1SPaolo Bonzini 44955499ea73SPaolo Bonzini if (!fast_pgd_switch(vcpu->kvm, mmu, new_pgd, new_role)) { 44965499ea73SPaolo Bonzini /* kvm_mmu_ensure_valid_pgd will set up a new root. */ 4497b869855bSSean Christopherson return; 4498c50d8ae3SPaolo Bonzini } 4499c50d8ae3SPaolo Bonzini 4500c50d8ae3SPaolo Bonzini /* 4501b869855bSSean Christopherson * It's possible that the cached previous root page is obsolete because 4502b869855bSSean Christopherson * of a change in the MMU generation number. However, changing the 4503527d5cd7SSean Christopherson * generation number is accompanied by KVM_REQ_MMU_FREE_OBSOLETE_ROOTS, 4504527d5cd7SSean Christopherson * which will free the root set here and allocate a new one. 4505b869855bSSean Christopherson */ 4506b869855bSSean Christopherson kvm_make_request(KVM_REQ_LOAD_MMU_PGD, vcpu); 4507b869855bSSean Christopherson 4508b5129100SSean Christopherson if (force_flush_and_sync_on_reuse) { 4509b869855bSSean Christopherson kvm_make_request(KVM_REQ_MMU_SYNC, vcpu); 4510b869855bSSean Christopherson kvm_make_request(KVM_REQ_TLB_FLUSH_CURRENT, vcpu); 4511b5129100SSean Christopherson } 4512b869855bSSean Christopherson 4513b869855bSSean Christopherson /* 4514b869855bSSean Christopherson * The last MMIO access's GVA and GPA are cached in the VCPU. When 4515b869855bSSean Christopherson * switching to a new CR3, that GVA->GPA mapping may no longer be 4516b869855bSSean Christopherson * valid. So clear any cached MMIO info even when we don't need to sync 4517b869855bSSean Christopherson * the shadow page tables. 4518c50d8ae3SPaolo Bonzini */ 4519c50d8ae3SPaolo Bonzini vcpu_clear_mmio_info(vcpu, MMIO_GVA_ANY); 4520c50d8ae3SPaolo Bonzini 4521daa5b6c1SBen Gardon /* 4522daa5b6c1SBen Gardon * If this is a direct root page, it doesn't have a write flooding 4523daa5b6c1SBen Gardon * count. Otherwise, clear the write flooding count. 4524daa5b6c1SBen Gardon */ 4525daa5b6c1SBen Gardon if (!new_role.direct) 4526daa5b6c1SBen Gardon __clear_sp_write_flooding_count( 4527b9e5603cSPaolo Bonzini to_shadow_page(vcpu->arch.mmu->root.hpa)); 4528c50d8ae3SPaolo Bonzini } 4529be01e8e2SSean Christopherson EXPORT_SYMBOL_GPL(kvm_mmu_new_pgd); 4530c50d8ae3SPaolo Bonzini 4531c50d8ae3SPaolo Bonzini static unsigned long get_cr3(struct kvm_vcpu *vcpu) 4532c50d8ae3SPaolo Bonzini { 4533c50d8ae3SPaolo Bonzini return kvm_read_cr3(vcpu); 4534c50d8ae3SPaolo Bonzini } 4535c50d8ae3SPaolo Bonzini 4536c50d8ae3SPaolo Bonzini static bool sync_mmio_spte(struct kvm_vcpu *vcpu, u64 *sptep, gfn_t gfn, 4537c3e5e415SLai Jiangshan unsigned int access) 4538c50d8ae3SPaolo Bonzini { 4539c50d8ae3SPaolo Bonzini if (unlikely(is_mmio_spte(*sptep))) { 4540c50d8ae3SPaolo Bonzini if (gfn != get_mmio_spte_gfn(*sptep)) { 4541c50d8ae3SPaolo Bonzini mmu_spte_clear_no_track(sptep); 4542c50d8ae3SPaolo Bonzini return true; 4543c50d8ae3SPaolo Bonzini } 4544c50d8ae3SPaolo Bonzini 4545c50d8ae3SPaolo Bonzini mark_mmio_spte(vcpu, sptep, gfn, access); 4546c50d8ae3SPaolo Bonzini return true; 4547c50d8ae3SPaolo Bonzini } 4548c50d8ae3SPaolo Bonzini 4549c50d8ae3SPaolo Bonzini return false; 4550c50d8ae3SPaolo Bonzini } 4551c50d8ae3SPaolo Bonzini 4552c50d8ae3SPaolo Bonzini #define PTTYPE_EPT 18 /* arbitrary */ 4553c50d8ae3SPaolo Bonzini #define PTTYPE PTTYPE_EPT 4554c50d8ae3SPaolo Bonzini #include "paging_tmpl.h" 4555c50d8ae3SPaolo Bonzini #undef PTTYPE 4556c50d8ae3SPaolo Bonzini 4557c50d8ae3SPaolo Bonzini #define PTTYPE 64 4558c50d8ae3SPaolo Bonzini #include "paging_tmpl.h" 4559c50d8ae3SPaolo Bonzini #undef PTTYPE 4560c50d8ae3SPaolo Bonzini 4561c50d8ae3SPaolo Bonzini #define PTTYPE 32 4562c50d8ae3SPaolo Bonzini #include "paging_tmpl.h" 4563c50d8ae3SPaolo Bonzini #undef PTTYPE 4564c50d8ae3SPaolo Bonzini 4565c50d8ae3SPaolo Bonzini static void 4566b705a277SSean Christopherson __reset_rsvds_bits_mask(struct rsvd_bits_validate *rsvd_check, 45675b7f575cSSean Christopherson u64 pa_bits_rsvd, int level, bool nx, bool gbpages, 4568c50d8ae3SPaolo Bonzini bool pse, bool amd) 4569c50d8ae3SPaolo Bonzini { 4570c50d8ae3SPaolo Bonzini u64 gbpages_bit_rsvd = 0; 4571c50d8ae3SPaolo Bonzini u64 nonleaf_bit8_rsvd = 0; 45725b7f575cSSean Christopherson u64 high_bits_rsvd; 4573c50d8ae3SPaolo Bonzini 4574c50d8ae3SPaolo Bonzini rsvd_check->bad_mt_xwr = 0; 4575c50d8ae3SPaolo Bonzini 4576c50d8ae3SPaolo Bonzini if (!gbpages) 4577c50d8ae3SPaolo Bonzini gbpages_bit_rsvd = rsvd_bits(7, 7); 4578c50d8ae3SPaolo Bonzini 45795b7f575cSSean Christopherson if (level == PT32E_ROOT_LEVEL) 45805b7f575cSSean Christopherson high_bits_rsvd = pa_bits_rsvd & rsvd_bits(0, 62); 45815b7f575cSSean Christopherson else 45825b7f575cSSean Christopherson high_bits_rsvd = pa_bits_rsvd & rsvd_bits(0, 51); 45835b7f575cSSean Christopherson 45845b7f575cSSean Christopherson /* Note, NX doesn't exist in PDPTEs, this is handled below. */ 45855b7f575cSSean Christopherson if (!nx) 45865b7f575cSSean Christopherson high_bits_rsvd |= rsvd_bits(63, 63); 45875b7f575cSSean Christopherson 4588c50d8ae3SPaolo Bonzini /* 4589c50d8ae3SPaolo Bonzini * Non-leaf PML4Es and PDPEs reserve bit 8 (which would be the G bit for 4590c50d8ae3SPaolo Bonzini * leaf entries) on AMD CPUs only. 4591c50d8ae3SPaolo Bonzini */ 4592c50d8ae3SPaolo Bonzini if (amd) 4593c50d8ae3SPaolo Bonzini nonleaf_bit8_rsvd = rsvd_bits(8, 8); 4594c50d8ae3SPaolo Bonzini 4595c50d8ae3SPaolo Bonzini switch (level) { 4596c50d8ae3SPaolo Bonzini case PT32_ROOT_LEVEL: 4597c50d8ae3SPaolo Bonzini /* no rsvd bits for 2 level 4K page table entries */ 4598c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[0][1] = 0; 4599c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[0][0] = 0; 4600c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][0] = 4601c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[0][0]; 4602c50d8ae3SPaolo Bonzini 4603c50d8ae3SPaolo Bonzini if (!pse) { 4604c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][1] = 0; 4605c50d8ae3SPaolo Bonzini break; 4606c50d8ae3SPaolo Bonzini } 4607c50d8ae3SPaolo Bonzini 4608c50d8ae3SPaolo Bonzini if (is_cpuid_PSE36()) 4609c50d8ae3SPaolo Bonzini /* 36bits PSE 4MB page */ 4610c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][1] = rsvd_bits(17, 21); 4611c50d8ae3SPaolo Bonzini else 4612c50d8ae3SPaolo Bonzini /* 32 bits PSE 4MB page */ 4613c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][1] = rsvd_bits(13, 21); 4614c50d8ae3SPaolo Bonzini break; 4615c50d8ae3SPaolo Bonzini case PT32E_ROOT_LEVEL: 46165b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][2] = rsvd_bits(63, 63) | 46175b7f575cSSean Christopherson high_bits_rsvd | 46185b7f575cSSean Christopherson rsvd_bits(5, 8) | 46195b7f575cSSean Christopherson rsvd_bits(1, 2); /* PDPTE */ 46205b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][1] = high_bits_rsvd; /* PDE */ 46215b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][0] = high_bits_rsvd; /* PTE */ 46225b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[1][1] = high_bits_rsvd | 4623c50d8ae3SPaolo Bonzini rsvd_bits(13, 20); /* large page */ 4624c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][0] = 4625c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[0][0]; 4626c50d8ae3SPaolo Bonzini break; 4627c50d8ae3SPaolo Bonzini case PT64_ROOT_5LEVEL: 46285b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][4] = high_bits_rsvd | 46295b7f575cSSean Christopherson nonleaf_bit8_rsvd | 46305b7f575cSSean Christopherson rsvd_bits(7, 7); 4631c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][4] = 4632c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[0][4]; 4633df561f66SGustavo A. R. Silva fallthrough; 4634c50d8ae3SPaolo Bonzini case PT64_ROOT_4LEVEL: 46355b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][3] = high_bits_rsvd | 46365b7f575cSSean Christopherson nonleaf_bit8_rsvd | 46375b7f575cSSean Christopherson rsvd_bits(7, 7); 46385b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][2] = high_bits_rsvd | 46395b7f575cSSean Christopherson gbpages_bit_rsvd; 46405b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][1] = high_bits_rsvd; 46415b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][0] = high_bits_rsvd; 4642c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][3] = 4643c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[0][3]; 46445b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[1][2] = high_bits_rsvd | 46455b7f575cSSean Christopherson gbpages_bit_rsvd | 4646c50d8ae3SPaolo Bonzini rsvd_bits(13, 29); 46475b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[1][1] = high_bits_rsvd | 4648c50d8ae3SPaolo Bonzini rsvd_bits(13, 20); /* large page */ 4649c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][0] = 4650c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[0][0]; 4651c50d8ae3SPaolo Bonzini break; 4652c50d8ae3SPaolo Bonzini } 4653c50d8ae3SPaolo Bonzini } 4654c50d8ae3SPaolo Bonzini 465527de9250SSean Christopherson static bool guest_can_use_gbpages(struct kvm_vcpu *vcpu) 465627de9250SSean Christopherson { 465727de9250SSean Christopherson /* 465827de9250SSean Christopherson * If TDP is enabled, let the guest use GBPAGES if they're supported in 465927de9250SSean Christopherson * hardware. The hardware page walker doesn't let KVM disable GBPAGES, 466027de9250SSean Christopherson * i.e. won't treat them as reserved, and KVM doesn't redo the GVA->GPA 466127de9250SSean Christopherson * walk for performance and complexity reasons. Not to mention KVM 466227de9250SSean Christopherson * _can't_ solve the problem because GVA->GPA walks aren't visible to 466327de9250SSean Christopherson * KVM once a TDP translation is installed. Mimic hardware behavior so 466427de9250SSean Christopherson * that KVM's is at least consistent, i.e. doesn't randomly inject #PF. 466527de9250SSean Christopherson */ 466627de9250SSean Christopherson return tdp_enabled ? boot_cpu_has(X86_FEATURE_GBPAGES) : 466727de9250SSean Christopherson guest_cpuid_has(vcpu, X86_FEATURE_GBPAGES); 466827de9250SSean Christopherson } 466927de9250SSean Christopherson 4670c919e881SKai Huang static void reset_guest_rsvds_bits_mask(struct kvm_vcpu *vcpu, 4671c50d8ae3SPaolo Bonzini struct kvm_mmu *context) 4672c50d8ae3SPaolo Bonzini { 4673b705a277SSean Christopherson __reset_rsvds_bits_mask(&context->guest_rsvd_check, 46745b7f575cSSean Christopherson vcpu->arch.reserved_gpa_bits, 46754d25502aSPaolo Bonzini context->cpu_role.base.level, is_efer_nx(context), 467627de9250SSean Christopherson guest_can_use_gbpages(vcpu), 46774e9c0d80SSean Christopherson is_cr4_pse(context), 467823493d0aSSean Christopherson guest_cpuid_is_amd_or_hygon(vcpu)); 4679c50d8ae3SPaolo Bonzini } 4680c50d8ae3SPaolo Bonzini 4681c50d8ae3SPaolo Bonzini static void 4682c50d8ae3SPaolo Bonzini __reset_rsvds_bits_mask_ept(struct rsvd_bits_validate *rsvd_check, 468384ea5c09SLai Jiangshan u64 pa_bits_rsvd, bool execonly, int huge_page_level) 4684c50d8ae3SPaolo Bonzini { 46855b7f575cSSean Christopherson u64 high_bits_rsvd = pa_bits_rsvd & rsvd_bits(0, 51); 468684ea5c09SLai Jiangshan u64 large_1g_rsvd = 0, large_2m_rsvd = 0; 4687c50d8ae3SPaolo Bonzini u64 bad_mt_xwr; 4688c50d8ae3SPaolo Bonzini 468984ea5c09SLai Jiangshan if (huge_page_level < PG_LEVEL_1G) 469084ea5c09SLai Jiangshan large_1g_rsvd = rsvd_bits(7, 7); 469184ea5c09SLai Jiangshan if (huge_page_level < PG_LEVEL_2M) 469284ea5c09SLai Jiangshan large_2m_rsvd = rsvd_bits(7, 7); 469384ea5c09SLai Jiangshan 46945b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][4] = high_bits_rsvd | rsvd_bits(3, 7); 46955b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][3] = high_bits_rsvd | rsvd_bits(3, 7); 469684ea5c09SLai Jiangshan rsvd_check->rsvd_bits_mask[0][2] = high_bits_rsvd | rsvd_bits(3, 6) | large_1g_rsvd; 469784ea5c09SLai Jiangshan rsvd_check->rsvd_bits_mask[0][1] = high_bits_rsvd | rsvd_bits(3, 6) | large_2m_rsvd; 46985b7f575cSSean Christopherson rsvd_check->rsvd_bits_mask[0][0] = high_bits_rsvd; 4699c50d8ae3SPaolo Bonzini 4700c50d8ae3SPaolo Bonzini /* large page */ 4701c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][4] = rsvd_check->rsvd_bits_mask[0][4]; 4702c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][3] = rsvd_check->rsvd_bits_mask[0][3]; 470384ea5c09SLai Jiangshan rsvd_check->rsvd_bits_mask[1][2] = high_bits_rsvd | rsvd_bits(12, 29) | large_1g_rsvd; 470484ea5c09SLai Jiangshan rsvd_check->rsvd_bits_mask[1][1] = high_bits_rsvd | rsvd_bits(12, 20) | large_2m_rsvd; 4705c50d8ae3SPaolo Bonzini rsvd_check->rsvd_bits_mask[1][0] = rsvd_check->rsvd_bits_mask[0][0]; 4706c50d8ae3SPaolo Bonzini 4707c50d8ae3SPaolo Bonzini bad_mt_xwr = 0xFFull << (2 * 8); /* bits 3..5 must not be 2 */ 4708c50d8ae3SPaolo Bonzini bad_mt_xwr |= 0xFFull << (3 * 8); /* bits 3..5 must not be 3 */ 4709c50d8ae3SPaolo Bonzini bad_mt_xwr |= 0xFFull << (7 * 8); /* bits 3..5 must not be 7 */ 4710c50d8ae3SPaolo Bonzini bad_mt_xwr |= REPEAT_BYTE(1ull << 2); /* bits 0..2 must not be 010 */ 4711c50d8ae3SPaolo Bonzini bad_mt_xwr |= REPEAT_BYTE(1ull << 6); /* bits 0..2 must not be 110 */ 4712c50d8ae3SPaolo Bonzini if (!execonly) { 4713c50d8ae3SPaolo Bonzini /* bits 0..2 must not be 100 unless VMX capabilities allow it */ 4714c50d8ae3SPaolo Bonzini bad_mt_xwr |= REPEAT_BYTE(1ull << 4); 4715c50d8ae3SPaolo Bonzini } 4716c50d8ae3SPaolo Bonzini rsvd_check->bad_mt_xwr = bad_mt_xwr; 4717c50d8ae3SPaolo Bonzini } 4718c50d8ae3SPaolo Bonzini 4719c50d8ae3SPaolo Bonzini static void reset_rsvds_bits_mask_ept(struct kvm_vcpu *vcpu, 472084ea5c09SLai Jiangshan struct kvm_mmu *context, bool execonly, int huge_page_level) 4721c50d8ae3SPaolo Bonzini { 4722c50d8ae3SPaolo Bonzini __reset_rsvds_bits_mask_ept(&context->guest_rsvd_check, 472384ea5c09SLai Jiangshan vcpu->arch.reserved_gpa_bits, execonly, 472484ea5c09SLai Jiangshan huge_page_level); 4725c50d8ae3SPaolo Bonzini } 4726c50d8ae3SPaolo Bonzini 47276f8e65a6SSean Christopherson static inline u64 reserved_hpa_bits(void) 47286f8e65a6SSean Christopherson { 47296f8e65a6SSean Christopherson return rsvd_bits(shadow_phys_bits, 63); 47306f8e65a6SSean Christopherson } 47316f8e65a6SSean Christopherson 4732c50d8ae3SPaolo Bonzini /* 4733c50d8ae3SPaolo Bonzini * the page table on host is the shadow page table for the page 4734c50d8ae3SPaolo Bonzini * table in guest or amd nested guest, its mmu features completely 4735c50d8ae3SPaolo Bonzini * follow the features in guest. 4736c50d8ae3SPaolo Bonzini */ 473716be1d12SSean Christopherson static void reset_shadow_zero_bits_mask(struct kvm_vcpu *vcpu, 473816be1d12SSean Christopherson struct kvm_mmu *context) 4739c50d8ae3SPaolo Bonzini { 47408c985b2dSSean Christopherson /* @amd adds a check on bit of SPTEs, which KVM shouldn't use anyways. */ 47418c985b2dSSean Christopherson bool is_amd = true; 47428c985b2dSSean Christopherson /* KVM doesn't use 2-level page tables for the shadow MMU. */ 47438c985b2dSSean Christopherson bool is_pse = false; 4744c50d8ae3SPaolo Bonzini struct rsvd_bits_validate *shadow_zero_check; 4745c50d8ae3SPaolo Bonzini int i; 4746c50d8ae3SPaolo Bonzini 4747a972e29cSPaolo Bonzini WARN_ON_ONCE(context->root_role.level < PT32E_ROOT_LEVEL); 47488c985b2dSSean Christopherson 4749c50d8ae3SPaolo Bonzini shadow_zero_check = &context->shadow_zero_check; 4750b705a277SSean Christopherson __reset_rsvds_bits_mask(shadow_zero_check, reserved_hpa_bits(), 4751a972e29cSPaolo Bonzini context->root_role.level, 47527a458f0eSPaolo Bonzini context->root_role.efer_nx, 475327de9250SSean Christopherson guest_can_use_gbpages(vcpu), is_pse, is_amd); 4754c50d8ae3SPaolo Bonzini 4755c50d8ae3SPaolo Bonzini if (!shadow_me_mask) 4756c50d8ae3SPaolo Bonzini return; 4757c50d8ae3SPaolo Bonzini 4758a972e29cSPaolo Bonzini for (i = context->root_role.level; --i >= 0;) { 4759e54f1ff2SKai Huang /* 4760e54f1ff2SKai Huang * So far shadow_me_value is a constant during KVM's life 4761e54f1ff2SKai Huang * time. Bits in shadow_me_value are allowed to be set. 4762e54f1ff2SKai Huang * Bits in shadow_me_mask but not in shadow_me_value are 4763e54f1ff2SKai Huang * not allowed to be set. 4764e54f1ff2SKai Huang */ 4765e54f1ff2SKai Huang shadow_zero_check->rsvd_bits_mask[0][i] |= shadow_me_mask; 4766e54f1ff2SKai Huang shadow_zero_check->rsvd_bits_mask[1][i] |= shadow_me_mask; 4767e54f1ff2SKai Huang shadow_zero_check->rsvd_bits_mask[0][i] &= ~shadow_me_value; 4768e54f1ff2SKai Huang shadow_zero_check->rsvd_bits_mask[1][i] &= ~shadow_me_value; 4769c50d8ae3SPaolo Bonzini } 4770c50d8ae3SPaolo Bonzini 4771c50d8ae3SPaolo Bonzini } 4772c50d8ae3SPaolo Bonzini 4773c50d8ae3SPaolo Bonzini static inline bool boot_cpu_is_amd(void) 4774c50d8ae3SPaolo Bonzini { 4775c50d8ae3SPaolo Bonzini WARN_ON_ONCE(!tdp_enabled); 4776c50d8ae3SPaolo Bonzini return shadow_x_mask == 0; 4777c50d8ae3SPaolo Bonzini } 4778c50d8ae3SPaolo Bonzini 4779c50d8ae3SPaolo Bonzini /* 4780c50d8ae3SPaolo Bonzini * the direct page table on host, use as much mmu features as 4781c50d8ae3SPaolo Bonzini * possible, however, kvm currently does not do execution-protection. 4782c50d8ae3SPaolo Bonzini */ 4783c50d8ae3SPaolo Bonzini static void 4784e8f6e738SJinrong Liang reset_tdp_shadow_zero_bits_mask(struct kvm_mmu *context) 4785c50d8ae3SPaolo Bonzini { 4786c50d8ae3SPaolo Bonzini struct rsvd_bits_validate *shadow_zero_check; 4787c50d8ae3SPaolo Bonzini int i; 4788c50d8ae3SPaolo Bonzini 4789c50d8ae3SPaolo Bonzini shadow_zero_check = &context->shadow_zero_check; 4790c50d8ae3SPaolo Bonzini 4791c50d8ae3SPaolo Bonzini if (boot_cpu_is_amd()) 4792b705a277SSean Christopherson __reset_rsvds_bits_mask(shadow_zero_check, reserved_hpa_bits(), 47936c6ab524SSean Christopherson context->root_role.level, true, 4794c50d8ae3SPaolo Bonzini boot_cpu_has(X86_FEATURE_GBPAGES), 47958c985b2dSSean Christopherson false, true); 4796c50d8ae3SPaolo Bonzini else 4797c50d8ae3SPaolo Bonzini __reset_rsvds_bits_mask_ept(shadow_zero_check, 479884ea5c09SLai Jiangshan reserved_hpa_bits(), false, 479984ea5c09SLai Jiangshan max_huge_page_level); 4800c50d8ae3SPaolo Bonzini 4801c50d8ae3SPaolo Bonzini if (!shadow_me_mask) 4802c50d8ae3SPaolo Bonzini return; 4803c50d8ae3SPaolo Bonzini 4804a972e29cSPaolo Bonzini for (i = context->root_role.level; --i >= 0;) { 4805c50d8ae3SPaolo Bonzini shadow_zero_check->rsvd_bits_mask[0][i] &= ~shadow_me_mask; 4806c50d8ae3SPaolo Bonzini shadow_zero_check->rsvd_bits_mask[1][i] &= ~shadow_me_mask; 4807c50d8ae3SPaolo Bonzini } 4808c50d8ae3SPaolo Bonzini } 4809c50d8ae3SPaolo Bonzini 4810c50d8ae3SPaolo Bonzini /* 4811c50d8ae3SPaolo Bonzini * as the comments in reset_shadow_zero_bits_mask() except it 4812c50d8ae3SPaolo Bonzini * is the shadow page table for intel nested guest. 4813c50d8ae3SPaolo Bonzini */ 4814c50d8ae3SPaolo Bonzini static void 4815e8f6e738SJinrong Liang reset_ept_shadow_zero_bits_mask(struct kvm_mmu *context, bool execonly) 4816c50d8ae3SPaolo Bonzini { 4817c50d8ae3SPaolo Bonzini __reset_rsvds_bits_mask_ept(&context->shadow_zero_check, 481884ea5c09SLai Jiangshan reserved_hpa_bits(), execonly, 481984ea5c09SLai Jiangshan max_huge_page_level); 4820c50d8ae3SPaolo Bonzini } 4821c50d8ae3SPaolo Bonzini 4822c50d8ae3SPaolo Bonzini #define BYTE_MASK(access) \ 4823c50d8ae3SPaolo Bonzini ((1 & (access) ? 2 : 0) | \ 4824c50d8ae3SPaolo Bonzini (2 & (access) ? 4 : 0) | \ 4825c50d8ae3SPaolo Bonzini (3 & (access) ? 8 : 0) | \ 4826c50d8ae3SPaolo Bonzini (4 & (access) ? 16 : 0) | \ 4827c50d8ae3SPaolo Bonzini (5 & (access) ? 32 : 0) | \ 4828c50d8ae3SPaolo Bonzini (6 & (access) ? 64 : 0) | \ 4829c50d8ae3SPaolo Bonzini (7 & (access) ? 128 : 0)) 4830c50d8ae3SPaolo Bonzini 4831c50d8ae3SPaolo Bonzini 4832c596f147SSean Christopherson static void update_permission_bitmask(struct kvm_mmu *mmu, bool ept) 4833c50d8ae3SPaolo Bonzini { 4834c50d8ae3SPaolo Bonzini unsigned byte; 4835c50d8ae3SPaolo Bonzini 4836c50d8ae3SPaolo Bonzini const u8 x = BYTE_MASK(ACC_EXEC_MASK); 4837c50d8ae3SPaolo Bonzini const u8 w = BYTE_MASK(ACC_WRITE_MASK); 4838c50d8ae3SPaolo Bonzini const u8 u = BYTE_MASK(ACC_USER_MASK); 4839c50d8ae3SPaolo Bonzini 4840c596f147SSean Christopherson bool cr4_smep = is_cr4_smep(mmu); 4841c596f147SSean Christopherson bool cr4_smap = is_cr4_smap(mmu); 4842c596f147SSean Christopherson bool cr0_wp = is_cr0_wp(mmu); 484390599c28SSean Christopherson bool efer_nx = is_efer_nx(mmu); 4844c50d8ae3SPaolo Bonzini 4845c50d8ae3SPaolo Bonzini for (byte = 0; byte < ARRAY_SIZE(mmu->permissions); ++byte) { 4846c50d8ae3SPaolo Bonzini unsigned pfec = byte << 1; 4847c50d8ae3SPaolo Bonzini 4848c50d8ae3SPaolo Bonzini /* 4849c50d8ae3SPaolo Bonzini * Each "*f" variable has a 1 bit for each UWX value 4850c50d8ae3SPaolo Bonzini * that causes a fault with the given PFEC. 4851c50d8ae3SPaolo Bonzini */ 4852c50d8ae3SPaolo Bonzini 4853c50d8ae3SPaolo Bonzini /* Faults from writes to non-writable pages */ 4854c50d8ae3SPaolo Bonzini u8 wf = (pfec & PFERR_WRITE_MASK) ? (u8)~w : 0; 4855c50d8ae3SPaolo Bonzini /* Faults from user mode accesses to supervisor pages */ 4856c50d8ae3SPaolo Bonzini u8 uf = (pfec & PFERR_USER_MASK) ? (u8)~u : 0; 4857c50d8ae3SPaolo Bonzini /* Faults from fetches of non-executable pages*/ 4858c50d8ae3SPaolo Bonzini u8 ff = (pfec & PFERR_FETCH_MASK) ? (u8)~x : 0; 4859c50d8ae3SPaolo Bonzini /* Faults from kernel mode fetches of user pages */ 4860c50d8ae3SPaolo Bonzini u8 smepf = 0; 4861c50d8ae3SPaolo Bonzini /* Faults from kernel mode accesses of user pages */ 4862c50d8ae3SPaolo Bonzini u8 smapf = 0; 4863c50d8ae3SPaolo Bonzini 4864c50d8ae3SPaolo Bonzini if (!ept) { 4865c50d8ae3SPaolo Bonzini /* Faults from kernel mode accesses to user pages */ 4866c50d8ae3SPaolo Bonzini u8 kf = (pfec & PFERR_USER_MASK) ? 0 : u; 4867c50d8ae3SPaolo Bonzini 4868c50d8ae3SPaolo Bonzini /* Not really needed: !nx will cause pte.nx to fault */ 486990599c28SSean Christopherson if (!efer_nx) 4870c50d8ae3SPaolo Bonzini ff = 0; 4871c50d8ae3SPaolo Bonzini 4872c50d8ae3SPaolo Bonzini /* Allow supervisor writes if !cr0.wp */ 4873c50d8ae3SPaolo Bonzini if (!cr0_wp) 4874c50d8ae3SPaolo Bonzini wf = (pfec & PFERR_USER_MASK) ? wf : 0; 4875c50d8ae3SPaolo Bonzini 4876c50d8ae3SPaolo Bonzini /* Disallow supervisor fetches of user code if cr4.smep */ 4877c50d8ae3SPaolo Bonzini if (cr4_smep) 4878c50d8ae3SPaolo Bonzini smepf = (pfec & PFERR_FETCH_MASK) ? kf : 0; 4879c50d8ae3SPaolo Bonzini 4880c50d8ae3SPaolo Bonzini /* 4881c50d8ae3SPaolo Bonzini * SMAP:kernel-mode data accesses from user-mode 4882c50d8ae3SPaolo Bonzini * mappings should fault. A fault is considered 4883c50d8ae3SPaolo Bonzini * as a SMAP violation if all of the following 4884c50d8ae3SPaolo Bonzini * conditions are true: 4885c50d8ae3SPaolo Bonzini * - X86_CR4_SMAP is set in CR4 4886c50d8ae3SPaolo Bonzini * - A user page is accessed 4887c50d8ae3SPaolo Bonzini * - The access is not a fetch 48884f4aa80eSLai Jiangshan * - The access is supervisor mode 48894f4aa80eSLai Jiangshan * - If implicit supervisor access or X86_EFLAGS_AC is clear 4890c50d8ae3SPaolo Bonzini * 489194b4a2f1SLai Jiangshan * Here, we cover the first four conditions. 489294b4a2f1SLai Jiangshan * The fifth is computed dynamically in permission_fault(); 4893c50d8ae3SPaolo Bonzini * PFERR_RSVD_MASK bit will be set in PFEC if the access is 4894c50d8ae3SPaolo Bonzini * *not* subject to SMAP restrictions. 4895c50d8ae3SPaolo Bonzini */ 4896c50d8ae3SPaolo Bonzini if (cr4_smap) 4897c50d8ae3SPaolo Bonzini smapf = (pfec & (PFERR_RSVD_MASK|PFERR_FETCH_MASK)) ? 0 : kf; 4898c50d8ae3SPaolo Bonzini } 4899c50d8ae3SPaolo Bonzini 4900c50d8ae3SPaolo Bonzini mmu->permissions[byte] = ff | uf | wf | smepf | smapf; 4901c50d8ae3SPaolo Bonzini } 4902c50d8ae3SPaolo Bonzini } 4903c50d8ae3SPaolo Bonzini 4904c50d8ae3SPaolo Bonzini /* 4905c50d8ae3SPaolo Bonzini * PKU is an additional mechanism by which the paging controls access to 4906c50d8ae3SPaolo Bonzini * user-mode addresses based on the value in the PKRU register. Protection 4907c50d8ae3SPaolo Bonzini * key violations are reported through a bit in the page fault error code. 4908c50d8ae3SPaolo Bonzini * Unlike other bits of the error code, the PK bit is not known at the 4909c50d8ae3SPaolo Bonzini * call site of e.g. gva_to_gpa; it must be computed directly in 4910c50d8ae3SPaolo Bonzini * permission_fault based on two bits of PKRU, on some machine state (CR4, 4911c50d8ae3SPaolo Bonzini * CR0, EFER, CPL), and on other bits of the error code and the page tables. 4912c50d8ae3SPaolo Bonzini * 4913c50d8ae3SPaolo Bonzini * In particular the following conditions come from the error code, the 4914c50d8ae3SPaolo Bonzini * page tables and the machine state: 4915c50d8ae3SPaolo Bonzini * - PK is always zero unless CR4.PKE=1 and EFER.LMA=1 4916c50d8ae3SPaolo Bonzini * - PK is always zero if RSVD=1 (reserved bit set) or F=1 (instruction fetch) 4917c50d8ae3SPaolo Bonzini * - PK is always zero if U=0 in the page tables 4918c50d8ae3SPaolo Bonzini * - PKRU.WD is ignored if CR0.WP=0 and the access is a supervisor access. 4919c50d8ae3SPaolo Bonzini * 4920c50d8ae3SPaolo Bonzini * The PKRU bitmask caches the result of these four conditions. The error 4921c50d8ae3SPaolo Bonzini * code (minus the P bit) and the page table's U bit form an index into the 4922c50d8ae3SPaolo Bonzini * PKRU bitmask. Two bits of the PKRU bitmask are then extracted and ANDed 4923c50d8ae3SPaolo Bonzini * with the two bits of the PKRU register corresponding to the protection key. 4924c50d8ae3SPaolo Bonzini * For the first three conditions above the bits will be 00, thus masking 4925c50d8ae3SPaolo Bonzini * away both AD and WD. For all reads or if the last condition holds, WD 4926c50d8ae3SPaolo Bonzini * only will be masked away. 4927c50d8ae3SPaolo Bonzini */ 49282e4c0661SSean Christopherson static void update_pkru_bitmask(struct kvm_mmu *mmu) 4929c50d8ae3SPaolo Bonzini { 4930c50d8ae3SPaolo Bonzini unsigned bit; 4931c50d8ae3SPaolo Bonzini bool wp; 4932c50d8ae3SPaolo Bonzini 4933c50d8ae3SPaolo Bonzini mmu->pkru_mask = 0; 4934a3ca5281SChenyi Qiang 4935a3ca5281SChenyi Qiang if (!is_cr4_pke(mmu)) 4936c50d8ae3SPaolo Bonzini return; 4937c50d8ae3SPaolo Bonzini 49382e4c0661SSean Christopherson wp = is_cr0_wp(mmu); 4939c50d8ae3SPaolo Bonzini 4940c50d8ae3SPaolo Bonzini for (bit = 0; bit < ARRAY_SIZE(mmu->permissions); ++bit) { 4941c50d8ae3SPaolo Bonzini unsigned pfec, pkey_bits; 4942c50d8ae3SPaolo Bonzini bool check_pkey, check_write, ff, uf, wf, pte_user; 4943c50d8ae3SPaolo Bonzini 4944c50d8ae3SPaolo Bonzini pfec = bit << 1; 4945c50d8ae3SPaolo Bonzini ff = pfec & PFERR_FETCH_MASK; 4946c50d8ae3SPaolo Bonzini uf = pfec & PFERR_USER_MASK; 4947c50d8ae3SPaolo Bonzini wf = pfec & PFERR_WRITE_MASK; 4948c50d8ae3SPaolo Bonzini 4949c50d8ae3SPaolo Bonzini /* PFEC.RSVD is replaced by ACC_USER_MASK. */ 4950c50d8ae3SPaolo Bonzini pte_user = pfec & PFERR_RSVD_MASK; 4951c50d8ae3SPaolo Bonzini 4952c50d8ae3SPaolo Bonzini /* 4953c50d8ae3SPaolo Bonzini * Only need to check the access which is not an 4954c50d8ae3SPaolo Bonzini * instruction fetch and is to a user page. 4955c50d8ae3SPaolo Bonzini */ 4956c50d8ae3SPaolo Bonzini check_pkey = (!ff && pte_user); 4957c50d8ae3SPaolo Bonzini /* 4958c50d8ae3SPaolo Bonzini * write access is controlled by PKRU if it is a 4959c50d8ae3SPaolo Bonzini * user access or CR0.WP = 1. 4960c50d8ae3SPaolo Bonzini */ 4961c50d8ae3SPaolo Bonzini check_write = check_pkey && wf && (uf || wp); 4962c50d8ae3SPaolo Bonzini 4963c50d8ae3SPaolo Bonzini /* PKRU.AD stops both read and write access. */ 4964c50d8ae3SPaolo Bonzini pkey_bits = !!check_pkey; 4965c50d8ae3SPaolo Bonzini /* PKRU.WD stops write access. */ 4966c50d8ae3SPaolo Bonzini pkey_bits |= (!!check_write) << 1; 4967c50d8ae3SPaolo Bonzini 4968c50d8ae3SPaolo Bonzini mmu->pkru_mask |= (pkey_bits & 3) << pfec; 4969c50d8ae3SPaolo Bonzini } 4970c50d8ae3SPaolo Bonzini } 4971c50d8ae3SPaolo Bonzini 4972533f9a4bSSean Christopherson static void reset_guest_paging_metadata(struct kvm_vcpu *vcpu, 4973533f9a4bSSean Christopherson struct kvm_mmu *mmu) 4974c50d8ae3SPaolo Bonzini { 4975533f9a4bSSean Christopherson if (!is_cr0_pg(mmu)) 4976533f9a4bSSean Christopherson return; 4977c50d8ae3SPaolo Bonzini 4978c919e881SKai Huang reset_guest_rsvds_bits_mask(vcpu, mmu); 4979533f9a4bSSean Christopherson update_permission_bitmask(mmu, false); 4980533f9a4bSSean Christopherson update_pkru_bitmask(mmu); 4981c50d8ae3SPaolo Bonzini } 4982c50d8ae3SPaolo Bonzini 4983fe660f72SSean Christopherson static void paging64_init_context(struct kvm_mmu *context) 4984c50d8ae3SPaolo Bonzini { 4985c50d8ae3SPaolo Bonzini context->page_fault = paging64_page_fault; 4986c50d8ae3SPaolo Bonzini context->gva_to_gpa = paging64_gva_to_gpa; 4987c50d8ae3SPaolo Bonzini context->sync_page = paging64_sync_page; 4988c50d8ae3SPaolo Bonzini context->invlpg = paging64_invlpg; 4989c50d8ae3SPaolo Bonzini } 4990c50d8ae3SPaolo Bonzini 499184a16226SSean Christopherson static void paging32_init_context(struct kvm_mmu *context) 4992c50d8ae3SPaolo Bonzini { 4993c50d8ae3SPaolo Bonzini context->page_fault = paging32_page_fault; 4994c50d8ae3SPaolo Bonzini context->gva_to_gpa = paging32_gva_to_gpa; 4995c50d8ae3SPaolo Bonzini context->sync_page = paging32_sync_page; 4996c50d8ae3SPaolo Bonzini context->invlpg = paging32_invlpg; 4997c50d8ae3SPaolo Bonzini } 4998c50d8ae3SPaolo Bonzini 49997a7ae829SPaolo Bonzini static union kvm_cpu_role 5000e5ed0fb0SPaolo Bonzini kvm_calc_cpu_role(struct kvm_vcpu *vcpu, const struct kvm_mmu_role_regs *regs) 5001e5ed0fb0SPaolo Bonzini { 50027a7ae829SPaolo Bonzini union kvm_cpu_role role = {0}; 5003e5ed0fb0SPaolo Bonzini 5004e5ed0fb0SPaolo Bonzini role.base.access = ACC_ALL; 5005e5ed0fb0SPaolo Bonzini role.base.smm = is_smm(vcpu); 5006e5ed0fb0SPaolo Bonzini role.base.guest_mode = is_guest_mode(vcpu); 5007e5ed0fb0SPaolo Bonzini role.ext.valid = 1; 5008e5ed0fb0SPaolo Bonzini 5009e5ed0fb0SPaolo Bonzini if (!____is_cr0_pg(regs)) { 5010e5ed0fb0SPaolo Bonzini role.base.direct = 1; 5011e5ed0fb0SPaolo Bonzini return role; 5012e5ed0fb0SPaolo Bonzini } 5013e5ed0fb0SPaolo Bonzini 5014e5ed0fb0SPaolo Bonzini role.base.efer_nx = ____is_efer_nx(regs); 5015e5ed0fb0SPaolo Bonzini role.base.cr0_wp = ____is_cr0_wp(regs); 5016e5ed0fb0SPaolo Bonzini role.base.smep_andnot_wp = ____is_cr4_smep(regs) && !____is_cr0_wp(regs); 5017e5ed0fb0SPaolo Bonzini role.base.smap_andnot_wp = ____is_cr4_smap(regs) && !____is_cr0_wp(regs); 5018e5ed0fb0SPaolo Bonzini role.base.has_4_byte_gpte = !____is_cr4_pae(regs); 501960f3cb60SPaolo Bonzini 502060f3cb60SPaolo Bonzini if (____is_efer_lma(regs)) 502160f3cb60SPaolo Bonzini role.base.level = ____is_cr4_la57(regs) ? PT64_ROOT_5LEVEL 502260f3cb60SPaolo Bonzini : PT64_ROOT_4LEVEL; 502360f3cb60SPaolo Bonzini else if (____is_cr4_pae(regs)) 502460f3cb60SPaolo Bonzini role.base.level = PT32E_ROOT_LEVEL; 502560f3cb60SPaolo Bonzini else 502660f3cb60SPaolo Bonzini role.base.level = PT32_ROOT_LEVEL; 5027e5ed0fb0SPaolo Bonzini 5028e5ed0fb0SPaolo Bonzini role.ext.cr4_smep = ____is_cr4_smep(regs); 5029e5ed0fb0SPaolo Bonzini role.ext.cr4_smap = ____is_cr4_smap(regs); 5030e5ed0fb0SPaolo Bonzini role.ext.cr4_pse = ____is_cr4_pse(regs); 5031e5ed0fb0SPaolo Bonzini 5032e5ed0fb0SPaolo Bonzini /* PKEY and LA57 are active iff long mode is active. */ 5033e5ed0fb0SPaolo Bonzini role.ext.cr4_pke = ____is_efer_lma(regs) && ____is_cr4_pke(regs); 5034e5ed0fb0SPaolo Bonzini role.ext.cr4_la57 = ____is_efer_lma(regs) && ____is_cr4_la57(regs); 5035e5ed0fb0SPaolo Bonzini role.ext.efer_lma = ____is_efer_lma(regs); 5036e5ed0fb0SPaolo Bonzini return role; 5037e5ed0fb0SPaolo Bonzini } 5038e5ed0fb0SPaolo Bonzini 5039d468d94bSSean Christopherson static inline int kvm_mmu_get_tdp_level(struct kvm_vcpu *vcpu) 5040d468d94bSSean Christopherson { 5041746700d2SWei Huang /* tdp_root_level is architecture forced level, use it if nonzero */ 5042746700d2SWei Huang if (tdp_root_level) 5043746700d2SWei Huang return tdp_root_level; 5044746700d2SWei Huang 5045d468d94bSSean Christopherson /* Use 5-level TDP if and only if it's useful/necessary. */ 504683013059SSean Christopherson if (max_tdp_level == 5 && cpuid_maxphyaddr(vcpu) <= 48) 5047d468d94bSSean Christopherson return 4; 5048d468d94bSSean Christopherson 504983013059SSean Christopherson return max_tdp_level; 5050d468d94bSSean Christopherson } 5051d468d94bSSean Christopherson 50527a458f0eSPaolo Bonzini static union kvm_mmu_page_role 50538626c120SSean Christopherson kvm_calc_tdp_mmu_root_page_role(struct kvm_vcpu *vcpu, 50547a7ae829SPaolo Bonzini union kvm_cpu_role cpu_role) 5055c50d8ae3SPaolo Bonzini { 50567a458f0eSPaolo Bonzini union kvm_mmu_page_role role = {0}; 5057c50d8ae3SPaolo Bonzini 50587a458f0eSPaolo Bonzini role.access = ACC_ALL; 50597a458f0eSPaolo Bonzini role.cr0_wp = true; 50607a458f0eSPaolo Bonzini role.efer_nx = true; 50617a458f0eSPaolo Bonzini role.smm = cpu_role.base.smm; 50627a458f0eSPaolo Bonzini role.guest_mode = cpu_role.base.guest_mode; 506354275f74SSean Christopherson role.ad_disabled = !kvm_ad_enabled(); 50647a458f0eSPaolo Bonzini role.level = kvm_mmu_get_tdp_level(vcpu); 50657a458f0eSPaolo Bonzini role.direct = true; 50667a458f0eSPaolo Bonzini role.has_4_byte_gpte = false; 5067c50d8ae3SPaolo Bonzini 5068c50d8ae3SPaolo Bonzini return role; 5069c50d8ae3SPaolo Bonzini } 5070c50d8ae3SPaolo Bonzini 507139e7e2bfSPaolo Bonzini static void init_kvm_tdp_mmu(struct kvm_vcpu *vcpu, 5072a7f1de9bSPaolo Bonzini union kvm_cpu_role cpu_role) 5073c50d8ae3SPaolo Bonzini { 50748c008659SPaolo Bonzini struct kvm_mmu *context = &vcpu->arch.root_mmu; 50757a458f0eSPaolo Bonzini union kvm_mmu_page_role root_role = kvm_calc_tdp_mmu_root_page_role(vcpu, cpu_role); 5076c50d8ae3SPaolo Bonzini 5077e5ed0fb0SPaolo Bonzini if (cpu_role.as_u64 == context->cpu_role.as_u64 && 50787a458f0eSPaolo Bonzini root_role.word == context->root_role.word) 5079c50d8ae3SPaolo Bonzini return; 5080c50d8ae3SPaolo Bonzini 5081e5ed0fb0SPaolo Bonzini context->cpu_role.as_u64 = cpu_role.as_u64; 50827a458f0eSPaolo Bonzini context->root_role.word = root_role.word; 50837a02674dSSean Christopherson context->page_fault = kvm_tdp_page_fault; 5084c50d8ae3SPaolo Bonzini context->sync_page = nonpaging_sync_page; 50855efac074SPaolo Bonzini context->invlpg = NULL; 5086d8dd54e0SSean Christopherson context->get_guest_pgd = get_cr3; 5087c50d8ae3SPaolo Bonzini context->get_pdptr = kvm_pdptr_read; 5088c50d8ae3SPaolo Bonzini context->inject_page_fault = kvm_inject_page_fault; 5089c50d8ae3SPaolo Bonzini 509036f26787SSean Christopherson if (!is_cr0_pg(context)) 5091c50d8ae3SPaolo Bonzini context->gva_to_gpa = nonpaging_gva_to_gpa; 509236f26787SSean Christopherson else if (is_cr4_pae(context)) 5093c50d8ae3SPaolo Bonzini context->gva_to_gpa = paging64_gva_to_gpa; 5094f4bd6f73SSean Christopherson else 5095c50d8ae3SPaolo Bonzini context->gva_to_gpa = paging32_gva_to_gpa; 5096c50d8ae3SPaolo Bonzini 5097533f9a4bSSean Christopherson reset_guest_paging_metadata(vcpu, context); 5098e8f6e738SJinrong Liang reset_tdp_shadow_zero_bits_mask(context); 5099c50d8ae3SPaolo Bonzini } 5100c50d8ae3SPaolo Bonzini 51018c008659SPaolo Bonzini static void shadow_mmu_init_context(struct kvm_vcpu *vcpu, struct kvm_mmu *context, 51027a7ae829SPaolo Bonzini union kvm_cpu_role cpu_role, 51037a458f0eSPaolo Bonzini union kvm_mmu_page_role root_role) 5104c50d8ae3SPaolo Bonzini { 5105e5ed0fb0SPaolo Bonzini if (cpu_role.as_u64 == context->cpu_role.as_u64 && 51067a458f0eSPaolo Bonzini root_role.word == context->root_role.word) 510718db1b17SSean Christopherson return; 5108c50d8ae3SPaolo Bonzini 5109e5ed0fb0SPaolo Bonzini context->cpu_role.as_u64 = cpu_role.as_u64; 51107a458f0eSPaolo Bonzini context->root_role.word = root_role.word; 511118db1b17SSean Christopherson 511236f26787SSean Christopherson if (!is_cr0_pg(context)) 511384a16226SSean Christopherson nonpaging_init_context(context); 511436f26787SSean Christopherson else if (is_cr4_pae(context)) 5115fe660f72SSean Christopherson paging64_init_context(context); 5116c50d8ae3SPaolo Bonzini else 511784a16226SSean Christopherson paging32_init_context(context); 5118c50d8ae3SPaolo Bonzini 5119533f9a4bSSean Christopherson reset_guest_paging_metadata(vcpu, context); 5120c50d8ae3SPaolo Bonzini reset_shadow_zero_bits_mask(vcpu, context); 5121c50d8ae3SPaolo Bonzini } 51220f04a2acSVitaly Kuznetsov 5123594e91a1SSean Christopherson static void kvm_init_shadow_mmu(struct kvm_vcpu *vcpu, 5124a7f1de9bSPaolo Bonzini union kvm_cpu_role cpu_role) 51250f04a2acSVitaly Kuznetsov { 51268c008659SPaolo Bonzini struct kvm_mmu *context = &vcpu->arch.root_mmu; 512756b321f9SPaolo Bonzini union kvm_mmu_page_role root_role; 512856b321f9SPaolo Bonzini 512956b321f9SPaolo Bonzini root_role = cpu_role.base; 513056b321f9SPaolo Bonzini 513156b321f9SPaolo Bonzini /* KVM uses PAE paging whenever the guest isn't using 64-bit paging. */ 513256b321f9SPaolo Bonzini root_role.level = max_t(u32, root_role.level, PT32E_ROOT_LEVEL); 513356b321f9SPaolo Bonzini 513456b321f9SPaolo Bonzini /* 513556b321f9SPaolo Bonzini * KVM forces EFER.NX=1 when TDP is disabled, reflect it in the MMU role. 513656b321f9SPaolo Bonzini * KVM uses NX when TDP is disabled to handle a variety of scenarios, 513756b321f9SPaolo Bonzini * notably for huge SPTEs if iTLB multi-hit mitigation is enabled and 513856b321f9SPaolo Bonzini * to generate correct permissions for CR0.WP=0/CR4.SMEP=1/EFER.NX=0. 513956b321f9SPaolo Bonzini * The iTLB multi-hit workaround can be toggled at any time, so assume 514056b321f9SPaolo Bonzini * NX can be used by any non-nested shadow MMU to avoid having to reset 514156b321f9SPaolo Bonzini * MMU contexts. 514256b321f9SPaolo Bonzini */ 514356b321f9SPaolo Bonzini root_role.efer_nx = true; 51440f04a2acSVitaly Kuznetsov 51457a458f0eSPaolo Bonzini shadow_mmu_init_context(vcpu, context, cpu_role, root_role); 51460f04a2acSVitaly Kuznetsov } 51470f04a2acSVitaly Kuznetsov 5148dbc4739bSSean Christopherson void kvm_init_shadow_npt_mmu(struct kvm_vcpu *vcpu, unsigned long cr0, 5149dbc4739bSSean Christopherson unsigned long cr4, u64 efer, gpa_t nested_cr3) 51500f04a2acSVitaly Kuznetsov { 51518c008659SPaolo Bonzini struct kvm_mmu *context = &vcpu->arch.guest_mmu; 5152594e91a1SSean Christopherson struct kvm_mmu_role_regs regs = { 5153594e91a1SSean Christopherson .cr0 = cr0, 515428f091bcSPaolo Bonzini .cr4 = cr4 & ~X86_CR4_PKE, 5155594e91a1SSean Christopherson .efer = efer, 5156594e91a1SSean Christopherson }; 51577a7ae829SPaolo Bonzini union kvm_cpu_role cpu_role = kvm_calc_cpu_role(vcpu, ®s); 515856b321f9SPaolo Bonzini union kvm_mmu_page_role root_role; 515956b321f9SPaolo Bonzini 516056b321f9SPaolo Bonzini /* NPT requires CR0.PG=1. */ 516156b321f9SPaolo Bonzini WARN_ON_ONCE(cpu_role.base.direct); 516256b321f9SPaolo Bonzini 516356b321f9SPaolo Bonzini root_role = cpu_role.base; 516456b321f9SPaolo Bonzini root_role.level = kvm_mmu_get_tdp_level(vcpu); 516584e5ffd0SLai Jiangshan if (root_role.level == PT64_ROOT_5LEVEL && 516684e5ffd0SLai Jiangshan cpu_role.base.level == PT64_ROOT_4LEVEL) 516784e5ffd0SLai Jiangshan root_role.passthrough = 1; 51680f04a2acSVitaly Kuznetsov 51697a458f0eSPaolo Bonzini shadow_mmu_init_context(vcpu, context, cpu_role, root_role); 5170d2e5f333SPaolo Bonzini kvm_mmu_new_pgd(vcpu, nested_cr3); 51710f04a2acSVitaly Kuznetsov } 51720f04a2acSVitaly Kuznetsov EXPORT_SYMBOL_GPL(kvm_init_shadow_npt_mmu); 5173c50d8ae3SPaolo Bonzini 51747a7ae829SPaolo Bonzini static union kvm_cpu_role 5175c50d8ae3SPaolo Bonzini kvm_calc_shadow_ept_root_page_role(struct kvm_vcpu *vcpu, bool accessed_dirty, 5176bb1fcc70SSean Christopherson bool execonly, u8 level) 5177c50d8ae3SPaolo Bonzini { 51787a7ae829SPaolo Bonzini union kvm_cpu_role role = {0}; 5179c50d8ae3SPaolo Bonzini 5180daed87b8SPaolo Bonzini /* 5181daed87b8SPaolo Bonzini * KVM does not support SMM transfer monitors, and consequently does not 5182daed87b8SPaolo Bonzini * support the "entry to SMM" control either. role.base.smm is always 0. 5183daed87b8SPaolo Bonzini */ 5184daed87b8SPaolo Bonzini WARN_ON_ONCE(is_smm(vcpu)); 5185bb1fcc70SSean Christopherson role.base.level = level; 5186bb3b394dSLai Jiangshan role.base.has_4_byte_gpte = false; 5187c50d8ae3SPaolo Bonzini role.base.direct = false; 5188c50d8ae3SPaolo Bonzini role.base.ad_disabled = !accessed_dirty; 5189c50d8ae3SPaolo Bonzini role.base.guest_mode = true; 5190c50d8ae3SPaolo Bonzini role.base.access = ACC_ALL; 5191c50d8ae3SPaolo Bonzini 5192cd6767c3SSean Christopherson role.ext.word = 0; 5193c50d8ae3SPaolo Bonzini role.ext.execonly = execonly; 5194cd6767c3SSean Christopherson role.ext.valid = 1; 5195c50d8ae3SPaolo Bonzini 5196c50d8ae3SPaolo Bonzini return role; 5197c50d8ae3SPaolo Bonzini } 5198c50d8ae3SPaolo Bonzini 5199c50d8ae3SPaolo Bonzini void kvm_init_shadow_ept_mmu(struct kvm_vcpu *vcpu, bool execonly, 5200cc022ae1SLai Jiangshan int huge_page_level, bool accessed_dirty, 5201cc022ae1SLai Jiangshan gpa_t new_eptp) 5202c50d8ae3SPaolo Bonzini { 52038c008659SPaolo Bonzini struct kvm_mmu *context = &vcpu->arch.guest_mmu; 5204bb1fcc70SSean Christopherson u8 level = vmx_eptp_page_walk_level(new_eptp); 52057a7ae829SPaolo Bonzini union kvm_cpu_role new_mode = 5206c50d8ae3SPaolo Bonzini kvm_calc_shadow_ept_root_page_role(vcpu, accessed_dirty, 5207bb1fcc70SSean Christopherson execonly, level); 5208c50d8ae3SPaolo Bonzini 5209e5ed0fb0SPaolo Bonzini if (new_mode.as_u64 != context->cpu_role.as_u64) { 5210e5ed0fb0SPaolo Bonzini /* EPT, and thus nested EPT, does not consume CR0, CR4, nor EFER. */ 5211e5ed0fb0SPaolo Bonzini context->cpu_role.as_u64 = new_mode.as_u64; 52127a458f0eSPaolo Bonzini context->root_role.word = new_mode.base.word; 521318db1b17SSean Christopherson 5214c50d8ae3SPaolo Bonzini context->page_fault = ept_page_fault; 5215c50d8ae3SPaolo Bonzini context->gva_to_gpa = ept_gva_to_gpa; 5216c50d8ae3SPaolo Bonzini context->sync_page = ept_sync_page; 5217c50d8ae3SPaolo Bonzini context->invlpg = ept_invlpg; 5218347a0d0dSPaolo Bonzini 5219c596f147SSean Christopherson update_permission_bitmask(context, true); 522028f091bcSPaolo Bonzini context->pkru_mask = 0; 5221cc022ae1SLai Jiangshan reset_rsvds_bits_mask_ept(vcpu, context, execonly, huge_page_level); 5222e8f6e738SJinrong Liang reset_ept_shadow_zero_bits_mask(context, execonly); 5223c50d8ae3SPaolo Bonzini } 52243cffc89dSPaolo Bonzini 5225d2e5f333SPaolo Bonzini kvm_mmu_new_pgd(vcpu, new_eptp); 52263cffc89dSPaolo Bonzini } 5227c50d8ae3SPaolo Bonzini EXPORT_SYMBOL_GPL(kvm_init_shadow_ept_mmu); 5228c50d8ae3SPaolo Bonzini 522939e7e2bfSPaolo Bonzini static void init_kvm_softmmu(struct kvm_vcpu *vcpu, 5230a7f1de9bSPaolo Bonzini union kvm_cpu_role cpu_role) 5231c50d8ae3SPaolo Bonzini { 52328c008659SPaolo Bonzini struct kvm_mmu *context = &vcpu->arch.root_mmu; 5233c50d8ae3SPaolo Bonzini 5234a7f1de9bSPaolo Bonzini kvm_init_shadow_mmu(vcpu, cpu_role); 5235929d1cfaSPaolo Bonzini 5236d8dd54e0SSean Christopherson context->get_guest_pgd = get_cr3; 5237c50d8ae3SPaolo Bonzini context->get_pdptr = kvm_pdptr_read; 5238c50d8ae3SPaolo Bonzini context->inject_page_fault = kvm_inject_page_fault; 5239c50d8ae3SPaolo Bonzini } 5240c50d8ae3SPaolo Bonzini 524139e7e2bfSPaolo Bonzini static void init_kvm_nested_mmu(struct kvm_vcpu *vcpu, 5242a7f1de9bSPaolo Bonzini union kvm_cpu_role new_mode) 5243c50d8ae3SPaolo Bonzini { 5244c50d8ae3SPaolo Bonzini struct kvm_mmu *g_context = &vcpu->arch.nested_mmu; 5245c50d8ae3SPaolo Bonzini 5246e5ed0fb0SPaolo Bonzini if (new_mode.as_u64 == g_context->cpu_role.as_u64) 5247c50d8ae3SPaolo Bonzini return; 5248c50d8ae3SPaolo Bonzini 5249e5ed0fb0SPaolo Bonzini g_context->cpu_role.as_u64 = new_mode.as_u64; 5250d8dd54e0SSean Christopherson g_context->get_guest_pgd = get_cr3; 5251c50d8ae3SPaolo Bonzini g_context->get_pdptr = kvm_pdptr_read; 5252c50d8ae3SPaolo Bonzini g_context->inject_page_fault = kvm_inject_page_fault; 5253c50d8ae3SPaolo Bonzini 5254c50d8ae3SPaolo Bonzini /* 52555efac074SPaolo Bonzini * L2 page tables are never shadowed, so there is no need to sync 52565efac074SPaolo Bonzini * SPTEs. 52575efac074SPaolo Bonzini */ 52585efac074SPaolo Bonzini g_context->invlpg = NULL; 52595efac074SPaolo Bonzini 52605efac074SPaolo Bonzini /* 5261c50d8ae3SPaolo Bonzini * Note that arch.mmu->gva_to_gpa translates l2_gpa to l1_gpa using 5262c50d8ae3SPaolo Bonzini * L1's nested page tables (e.g. EPT12). The nested translation 5263c50d8ae3SPaolo Bonzini * of l2_gva to l1_gpa is done by arch.nested_mmu.gva_to_gpa using 5264c50d8ae3SPaolo Bonzini * L2's page tables as the first level of translation and L1's 5265c50d8ae3SPaolo Bonzini * nested page tables as the second level of translation. Basically 5266c50d8ae3SPaolo Bonzini * the gva_to_gpa functions between mmu and nested_mmu are swapped. 5267c50d8ae3SPaolo Bonzini */ 5268fa4b5588SSean Christopherson if (!is_paging(vcpu)) 52691f5a21eeSLai Jiangshan g_context->gva_to_gpa = nonpaging_gva_to_gpa; 5270fa4b5588SSean Christopherson else if (is_long_mode(vcpu)) 52711f5a21eeSLai Jiangshan g_context->gva_to_gpa = paging64_gva_to_gpa; 5272fa4b5588SSean Christopherson else if (is_pae(vcpu)) 52731f5a21eeSLai Jiangshan g_context->gva_to_gpa = paging64_gva_to_gpa; 5274fa4b5588SSean Christopherson else 52751f5a21eeSLai Jiangshan g_context->gva_to_gpa = paging32_gva_to_gpa; 5276fa4b5588SSean Christopherson 5277533f9a4bSSean Christopherson reset_guest_paging_metadata(vcpu, g_context); 5278c50d8ae3SPaolo Bonzini } 5279c50d8ae3SPaolo Bonzini 5280c9060662SSean Christopherson void kvm_init_mmu(struct kvm_vcpu *vcpu) 5281c50d8ae3SPaolo Bonzini { 528239e7e2bfSPaolo Bonzini struct kvm_mmu_role_regs regs = vcpu_to_role_regs(vcpu); 5283a7f1de9bSPaolo Bonzini union kvm_cpu_role cpu_role = kvm_calc_cpu_role(vcpu, ®s); 528439e7e2bfSPaolo Bonzini 5285c50d8ae3SPaolo Bonzini if (mmu_is_nested(vcpu)) 5286a7f1de9bSPaolo Bonzini init_kvm_nested_mmu(vcpu, cpu_role); 5287c50d8ae3SPaolo Bonzini else if (tdp_enabled) 5288a7f1de9bSPaolo Bonzini init_kvm_tdp_mmu(vcpu, cpu_role); 5289c50d8ae3SPaolo Bonzini else 5290a7f1de9bSPaolo Bonzini init_kvm_softmmu(vcpu, cpu_role); 5291c50d8ae3SPaolo Bonzini } 5292c50d8ae3SPaolo Bonzini EXPORT_SYMBOL_GPL(kvm_init_mmu); 5293c50d8ae3SPaolo Bonzini 529449c6f875SSean Christopherson void kvm_mmu_after_set_cpuid(struct kvm_vcpu *vcpu) 529549c6f875SSean Christopherson { 529649c6f875SSean Christopherson /* 529749c6f875SSean Christopherson * Invalidate all MMU roles to force them to reinitialize as CPUID 529849c6f875SSean Christopherson * information is factored into reserved bit calculations. 5299feb627e8SVitaly Kuznetsov * 5300feb627e8SVitaly Kuznetsov * Correctly handling multiple vCPU models with respect to paging and 5301feb627e8SVitaly Kuznetsov * physical address properties) in a single VM would require tracking 5302feb627e8SVitaly Kuznetsov * all relevant CPUID information in kvm_mmu_page_role. That is very 5303feb627e8SVitaly Kuznetsov * undesirable as it would increase the memory requirements for 5304feb627e8SVitaly Kuznetsov * gfn_track (see struct kvm_mmu_page_role comments). For now that 5305feb627e8SVitaly Kuznetsov * problem is swept under the rug; KVM's CPUID API is horrific and 5306feb627e8SVitaly Kuznetsov * it's all but impossible to solve it without introducing a new API. 530749c6f875SSean Christopherson */ 53087a458f0eSPaolo Bonzini vcpu->arch.root_mmu.root_role.word = 0; 53097a458f0eSPaolo Bonzini vcpu->arch.guest_mmu.root_role.word = 0; 53107a458f0eSPaolo Bonzini vcpu->arch.nested_mmu.root_role.word = 0; 5311e5ed0fb0SPaolo Bonzini vcpu->arch.root_mmu.cpu_role.ext.valid = 0; 5312e5ed0fb0SPaolo Bonzini vcpu->arch.guest_mmu.cpu_role.ext.valid = 0; 5313e5ed0fb0SPaolo Bonzini vcpu->arch.nested_mmu.cpu_role.ext.valid = 0; 531449c6f875SSean Christopherson kvm_mmu_reset_context(vcpu); 531563f5a190SSean Christopherson 531663f5a190SSean Christopherson /* 5317feb627e8SVitaly Kuznetsov * Changing guest CPUID after KVM_RUN is forbidden, see the comment in 5318feb627e8SVitaly Kuznetsov * kvm_arch_vcpu_ioctl(). 531963f5a190SSean Christopherson */ 5320feb627e8SVitaly Kuznetsov KVM_BUG_ON(vcpu->arch.last_vmentry_cpu != -1, vcpu->kvm); 532149c6f875SSean Christopherson } 532249c6f875SSean Christopherson 5323c50d8ae3SPaolo Bonzini void kvm_mmu_reset_context(struct kvm_vcpu *vcpu) 5324c50d8ae3SPaolo Bonzini { 5325c50d8ae3SPaolo Bonzini kvm_mmu_unload(vcpu); 5326c9060662SSean Christopherson kvm_init_mmu(vcpu); 5327c50d8ae3SPaolo Bonzini } 5328c50d8ae3SPaolo Bonzini EXPORT_SYMBOL_GPL(kvm_mmu_reset_context); 5329c50d8ae3SPaolo Bonzini 5330c50d8ae3SPaolo Bonzini int kvm_mmu_load(struct kvm_vcpu *vcpu) 5331c50d8ae3SPaolo Bonzini { 5332c50d8ae3SPaolo Bonzini int r; 5333c50d8ae3SPaolo Bonzini 5334347a0d0dSPaolo Bonzini r = mmu_topup_memory_caches(vcpu, !vcpu->arch.mmu->root_role.direct); 5335c50d8ae3SPaolo Bonzini if (r) 5336c50d8ae3SPaolo Bonzini goto out; 5337748e52b9SSean Christopherson r = mmu_alloc_special_roots(vcpu); 5338c50d8ae3SPaolo Bonzini if (r) 5339c50d8ae3SPaolo Bonzini goto out; 5340347a0d0dSPaolo Bonzini if (vcpu->arch.mmu->root_role.direct) 53416e6ec584SSean Christopherson r = mmu_alloc_direct_roots(vcpu); 53426e6ec584SSean Christopherson else 53436e6ec584SSean Christopherson r = mmu_alloc_shadow_roots(vcpu); 5344c50d8ae3SPaolo Bonzini if (r) 5345c50d8ae3SPaolo Bonzini goto out; 5346a91f387bSSean Christopherson 5347a91f387bSSean Christopherson kvm_mmu_sync_roots(vcpu); 5348a91f387bSSean Christopherson 5349727a7e27SPaolo Bonzini kvm_mmu_load_pgd(vcpu); 5350db01416bSSean Christopherson 5351db01416bSSean Christopherson /* 5352db01416bSSean Christopherson * Flush any TLB entries for the new root, the provenance of the root 5353db01416bSSean Christopherson * is unknown. Even if KVM ensures there are no stale TLB entries 5354db01416bSSean Christopherson * for a freed root, in theory another hypervisor could have left 5355db01416bSSean Christopherson * stale entries. Flushing on alloc also allows KVM to skip the TLB 5356db01416bSSean Christopherson * flush when freeing a root (see kvm_tdp_mmu_put_root()). 5357db01416bSSean Christopherson */ 5358e27bc044SSean Christopherson static_call(kvm_x86_flush_tlb_current)(vcpu); 5359c50d8ae3SPaolo Bonzini out: 5360c50d8ae3SPaolo Bonzini return r; 5361c50d8ae3SPaolo Bonzini } 5362c50d8ae3SPaolo Bonzini 5363c50d8ae3SPaolo Bonzini void kvm_mmu_unload(struct kvm_vcpu *vcpu) 5364c50d8ae3SPaolo Bonzini { 53650c1c92f1SPaolo Bonzini struct kvm *kvm = vcpu->kvm; 53660c1c92f1SPaolo Bonzini 53670c1c92f1SPaolo Bonzini kvm_mmu_free_roots(kvm, &vcpu->arch.root_mmu, KVM_MMU_ROOTS_ALL); 5368b9e5603cSPaolo Bonzini WARN_ON(VALID_PAGE(vcpu->arch.root_mmu.root.hpa)); 53690c1c92f1SPaolo Bonzini kvm_mmu_free_roots(kvm, &vcpu->arch.guest_mmu, KVM_MMU_ROOTS_ALL); 5370b9e5603cSPaolo Bonzini WARN_ON(VALID_PAGE(vcpu->arch.guest_mmu.root.hpa)); 53716d58f275SPaolo Bonzini vcpu_clear_mmio_info(vcpu, MMIO_GVA_ANY); 5372c50d8ae3SPaolo Bonzini } 5373c50d8ae3SPaolo Bonzini 5374527d5cd7SSean Christopherson static bool is_obsolete_root(struct kvm *kvm, hpa_t root_hpa) 5375527d5cd7SSean Christopherson { 5376527d5cd7SSean Christopherson struct kvm_mmu_page *sp; 5377527d5cd7SSean Christopherson 5378527d5cd7SSean Christopherson if (!VALID_PAGE(root_hpa)) 5379527d5cd7SSean Christopherson return false; 5380527d5cd7SSean Christopherson 5381527d5cd7SSean Christopherson /* 5382527d5cd7SSean Christopherson * When freeing obsolete roots, treat roots as obsolete if they don't 5383527d5cd7SSean Christopherson * have an associated shadow page. This does mean KVM will get false 5384527d5cd7SSean Christopherson * positives and free roots that don't strictly need to be freed, but 5385527d5cd7SSean Christopherson * such false positives are relatively rare: 5386527d5cd7SSean Christopherson * 5387527d5cd7SSean Christopherson * (a) only PAE paging and nested NPT has roots without shadow pages 5388527d5cd7SSean Christopherson * (b) remote reloads due to a memslot update obsoletes _all_ roots 5389527d5cd7SSean Christopherson * (c) KVM doesn't track previous roots for PAE paging, and the guest 5390527d5cd7SSean Christopherson * is unlikely to zap an in-use PGD. 5391527d5cd7SSean Christopherson */ 5392527d5cd7SSean Christopherson sp = to_shadow_page(root_hpa); 5393527d5cd7SSean Christopherson return !sp || is_obsolete_sp(kvm, sp); 5394527d5cd7SSean Christopherson } 5395527d5cd7SSean Christopherson 5396527d5cd7SSean Christopherson static void __kvm_mmu_free_obsolete_roots(struct kvm *kvm, struct kvm_mmu *mmu) 5397527d5cd7SSean Christopherson { 5398527d5cd7SSean Christopherson unsigned long roots_to_free = 0; 5399527d5cd7SSean Christopherson int i; 5400527d5cd7SSean Christopherson 5401527d5cd7SSean Christopherson if (is_obsolete_root(kvm, mmu->root.hpa)) 5402527d5cd7SSean Christopherson roots_to_free |= KVM_MMU_ROOT_CURRENT; 5403527d5cd7SSean Christopherson 5404527d5cd7SSean Christopherson for (i = 0; i < KVM_MMU_NUM_PREV_ROOTS; i++) { 5405cf4a8693SShaoqin Huang if (is_obsolete_root(kvm, mmu->prev_roots[i].hpa)) 5406527d5cd7SSean Christopherson roots_to_free |= KVM_MMU_ROOT_PREVIOUS(i); 5407527d5cd7SSean Christopherson } 5408527d5cd7SSean Christopherson 5409527d5cd7SSean Christopherson if (roots_to_free) 5410527d5cd7SSean Christopherson kvm_mmu_free_roots(kvm, mmu, roots_to_free); 5411527d5cd7SSean Christopherson } 5412527d5cd7SSean Christopherson 5413527d5cd7SSean Christopherson void kvm_mmu_free_obsolete_roots(struct kvm_vcpu *vcpu) 5414527d5cd7SSean Christopherson { 5415527d5cd7SSean Christopherson __kvm_mmu_free_obsolete_roots(vcpu->kvm, &vcpu->arch.root_mmu); 5416527d5cd7SSean Christopherson __kvm_mmu_free_obsolete_roots(vcpu->kvm, &vcpu->arch.guest_mmu); 5417527d5cd7SSean Christopherson } 5418527d5cd7SSean Christopherson 5419c50d8ae3SPaolo Bonzini static u64 mmu_pte_write_fetch_gpte(struct kvm_vcpu *vcpu, gpa_t *gpa, 5420c50d8ae3SPaolo Bonzini int *bytes) 5421c50d8ae3SPaolo Bonzini { 5422c50d8ae3SPaolo Bonzini u64 gentry = 0; 5423c50d8ae3SPaolo Bonzini int r; 5424c50d8ae3SPaolo Bonzini 5425c50d8ae3SPaolo Bonzini /* 5426c50d8ae3SPaolo Bonzini * Assume that the pte write on a page table of the same type 5427c50d8ae3SPaolo Bonzini * as the current vcpu paging mode since we update the sptes only 5428c50d8ae3SPaolo Bonzini * when they have the same mode. 5429c50d8ae3SPaolo Bonzini */ 5430c50d8ae3SPaolo Bonzini if (is_pae(vcpu) && *bytes == 4) { 5431c50d8ae3SPaolo Bonzini /* Handle a 32-bit guest writing two halves of a 64-bit gpte */ 5432c50d8ae3SPaolo Bonzini *gpa &= ~(gpa_t)7; 5433c50d8ae3SPaolo Bonzini *bytes = 8; 5434c50d8ae3SPaolo Bonzini } 5435c50d8ae3SPaolo Bonzini 5436c50d8ae3SPaolo Bonzini if (*bytes == 4 || *bytes == 8) { 5437c50d8ae3SPaolo Bonzini r = kvm_vcpu_read_guest_atomic(vcpu, *gpa, &gentry, *bytes); 5438c50d8ae3SPaolo Bonzini if (r) 5439c50d8ae3SPaolo Bonzini gentry = 0; 5440c50d8ae3SPaolo Bonzini } 5441c50d8ae3SPaolo Bonzini 5442c50d8ae3SPaolo Bonzini return gentry; 5443c50d8ae3SPaolo Bonzini } 5444c50d8ae3SPaolo Bonzini 5445c50d8ae3SPaolo Bonzini /* 5446c50d8ae3SPaolo Bonzini * If we're seeing too many writes to a page, it may no longer be a page table, 5447c50d8ae3SPaolo Bonzini * or we may be forking, in which case it is better to unmap the page. 5448c50d8ae3SPaolo Bonzini */ 5449c50d8ae3SPaolo Bonzini static bool detect_write_flooding(struct kvm_mmu_page *sp) 5450c50d8ae3SPaolo Bonzini { 5451c50d8ae3SPaolo Bonzini /* 5452c50d8ae3SPaolo Bonzini * Skip write-flooding detected for the sp whose level is 1, because 5453c50d8ae3SPaolo Bonzini * it can become unsync, then the guest page is not write-protected. 5454c50d8ae3SPaolo Bonzini */ 54553bae0459SSean Christopherson if (sp->role.level == PG_LEVEL_4K) 5456c50d8ae3SPaolo Bonzini return false; 5457c50d8ae3SPaolo Bonzini 5458c50d8ae3SPaolo Bonzini atomic_inc(&sp->write_flooding_count); 5459c50d8ae3SPaolo Bonzini return atomic_read(&sp->write_flooding_count) >= 3; 5460c50d8ae3SPaolo Bonzini } 5461c50d8ae3SPaolo Bonzini 5462c50d8ae3SPaolo Bonzini /* 5463c50d8ae3SPaolo Bonzini * Misaligned accesses are too much trouble to fix up; also, they usually 5464c50d8ae3SPaolo Bonzini * indicate a page is not used as a page table. 5465c50d8ae3SPaolo Bonzini */ 5466c50d8ae3SPaolo Bonzini static bool detect_write_misaligned(struct kvm_mmu_page *sp, gpa_t gpa, 5467c50d8ae3SPaolo Bonzini int bytes) 5468c50d8ae3SPaolo Bonzini { 5469c50d8ae3SPaolo Bonzini unsigned offset, pte_size, misaligned; 5470c50d8ae3SPaolo Bonzini 5471c50d8ae3SPaolo Bonzini pgprintk("misaligned: gpa %llx bytes %d role %x\n", 5472c50d8ae3SPaolo Bonzini gpa, bytes, sp->role.word); 5473c50d8ae3SPaolo Bonzini 5474c50d8ae3SPaolo Bonzini offset = offset_in_page(gpa); 5475bb3b394dSLai Jiangshan pte_size = sp->role.has_4_byte_gpte ? 4 : 8; 5476c50d8ae3SPaolo Bonzini 5477c50d8ae3SPaolo Bonzini /* 5478c50d8ae3SPaolo Bonzini * Sometimes, the OS only writes the last one bytes to update status 5479c50d8ae3SPaolo Bonzini * bits, for example, in linux, andb instruction is used in clear_bit(). 5480c50d8ae3SPaolo Bonzini */ 5481c50d8ae3SPaolo Bonzini if (!(offset & (pte_size - 1)) && bytes == 1) 5482c50d8ae3SPaolo Bonzini return false; 5483c50d8ae3SPaolo Bonzini 5484c50d8ae3SPaolo Bonzini misaligned = (offset ^ (offset + bytes - 1)) & ~(pte_size - 1); 5485c50d8ae3SPaolo Bonzini misaligned |= bytes < 4; 5486c50d8ae3SPaolo Bonzini 5487c50d8ae3SPaolo Bonzini return misaligned; 5488c50d8ae3SPaolo Bonzini } 5489c50d8ae3SPaolo Bonzini 5490c50d8ae3SPaolo Bonzini static u64 *get_written_sptes(struct kvm_mmu_page *sp, gpa_t gpa, int *nspte) 5491c50d8ae3SPaolo Bonzini { 5492c50d8ae3SPaolo Bonzini unsigned page_offset, quadrant; 5493c50d8ae3SPaolo Bonzini u64 *spte; 5494c50d8ae3SPaolo Bonzini int level; 5495c50d8ae3SPaolo Bonzini 5496c50d8ae3SPaolo Bonzini page_offset = offset_in_page(gpa); 5497c50d8ae3SPaolo Bonzini level = sp->role.level; 5498c50d8ae3SPaolo Bonzini *nspte = 1; 5499bb3b394dSLai Jiangshan if (sp->role.has_4_byte_gpte) { 5500c50d8ae3SPaolo Bonzini page_offset <<= 1; /* 32->64 */ 5501c50d8ae3SPaolo Bonzini /* 5502c50d8ae3SPaolo Bonzini * A 32-bit pde maps 4MB while the shadow pdes map 5503c50d8ae3SPaolo Bonzini * only 2MB. So we need to double the offset again 5504c50d8ae3SPaolo Bonzini * and zap two pdes instead of one. 5505c50d8ae3SPaolo Bonzini */ 5506c50d8ae3SPaolo Bonzini if (level == PT32_ROOT_LEVEL) { 5507c50d8ae3SPaolo Bonzini page_offset &= ~7; /* kill rounding error */ 5508c50d8ae3SPaolo Bonzini page_offset <<= 1; 5509c50d8ae3SPaolo Bonzini *nspte = 2; 5510c50d8ae3SPaolo Bonzini } 5511c50d8ae3SPaolo Bonzini quadrant = page_offset >> PAGE_SHIFT; 5512c50d8ae3SPaolo Bonzini page_offset &= ~PAGE_MASK; 5513c50d8ae3SPaolo Bonzini if (quadrant != sp->role.quadrant) 5514c50d8ae3SPaolo Bonzini return NULL; 5515c50d8ae3SPaolo Bonzini } 5516c50d8ae3SPaolo Bonzini 5517c50d8ae3SPaolo Bonzini spte = &sp->spt[page_offset / sizeof(*spte)]; 5518c50d8ae3SPaolo Bonzini return spte; 5519c50d8ae3SPaolo Bonzini } 5520c50d8ae3SPaolo Bonzini 5521c50d8ae3SPaolo Bonzini static void kvm_mmu_pte_write(struct kvm_vcpu *vcpu, gpa_t gpa, 5522c50d8ae3SPaolo Bonzini const u8 *new, int bytes, 5523c50d8ae3SPaolo Bonzini struct kvm_page_track_notifier_node *node) 5524c50d8ae3SPaolo Bonzini { 5525c50d8ae3SPaolo Bonzini gfn_t gfn = gpa >> PAGE_SHIFT; 5526c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 5527c50d8ae3SPaolo Bonzini LIST_HEAD(invalid_list); 5528c50d8ae3SPaolo Bonzini u64 entry, gentry, *spte; 5529c50d8ae3SPaolo Bonzini int npte; 553006152b2dSLai Jiangshan bool flush = false; 5531c50d8ae3SPaolo Bonzini 5532c50d8ae3SPaolo Bonzini /* 5533c50d8ae3SPaolo Bonzini * If we don't have indirect shadow pages, it means no page is 5534c50d8ae3SPaolo Bonzini * write-protected, so we can exit simply. 5535c50d8ae3SPaolo Bonzini */ 5536c50d8ae3SPaolo Bonzini if (!READ_ONCE(vcpu->kvm->arch.indirect_shadow_pages)) 5537c50d8ae3SPaolo Bonzini return; 5538c50d8ae3SPaolo Bonzini 5539c50d8ae3SPaolo Bonzini pgprintk("%s: gpa %llx bytes %d\n", __func__, gpa, bytes); 5540c50d8ae3SPaolo Bonzini 5541531810caSBen Gardon write_lock(&vcpu->kvm->mmu_lock); 5542c50d8ae3SPaolo Bonzini 5543c50d8ae3SPaolo Bonzini gentry = mmu_pte_write_fetch_gpte(vcpu, &gpa, &bytes); 5544c50d8ae3SPaolo Bonzini 5545c50d8ae3SPaolo Bonzini ++vcpu->kvm->stat.mmu_pte_write; 5546c50d8ae3SPaolo Bonzini 5547767d8d8dSLai Jiangshan for_each_gfn_valid_sp_with_gptes(vcpu->kvm, sp, gfn) { 5548c50d8ae3SPaolo Bonzini if (detect_write_misaligned(sp, gpa, bytes) || 5549c50d8ae3SPaolo Bonzini detect_write_flooding(sp)) { 5550c50d8ae3SPaolo Bonzini kvm_mmu_prepare_zap_page(vcpu->kvm, sp, &invalid_list); 5551c50d8ae3SPaolo Bonzini ++vcpu->kvm->stat.mmu_flooded; 5552c50d8ae3SPaolo Bonzini continue; 5553c50d8ae3SPaolo Bonzini } 5554c50d8ae3SPaolo Bonzini 5555c50d8ae3SPaolo Bonzini spte = get_written_sptes(sp, gpa, &npte); 5556c50d8ae3SPaolo Bonzini if (!spte) 5557c50d8ae3SPaolo Bonzini continue; 5558c50d8ae3SPaolo Bonzini 5559c50d8ae3SPaolo Bonzini while (npte--) { 5560c50d8ae3SPaolo Bonzini entry = *spte; 55612de4085cSBen Gardon mmu_page_zap_pte(vcpu->kvm, sp, spte, NULL); 5562c5e2184dSSean Christopherson if (gentry && sp->role.level != PG_LEVEL_4K) 5563c5e2184dSSean Christopherson ++vcpu->kvm->stat.mmu_pde_zapped; 55641441ca14SJunaid Shahid if (is_shadow_present_pte(entry)) 556506152b2dSLai Jiangshan flush = true; 5566c50d8ae3SPaolo Bonzini ++spte; 5567c50d8ae3SPaolo Bonzini } 5568c50d8ae3SPaolo Bonzini } 556906152b2dSLai Jiangshan kvm_mmu_remote_flush_or_zap(vcpu->kvm, &invalid_list, flush); 5570531810caSBen Gardon write_unlock(&vcpu->kvm->mmu_lock); 5571c50d8ae3SPaolo Bonzini } 5572c50d8ae3SPaolo Bonzini 55731075d41eSSean Christopherson int noinline kvm_mmu_page_fault(struct kvm_vcpu *vcpu, gpa_t cr2_or_gpa, u64 error_code, 5574c50d8ae3SPaolo Bonzini void *insn, int insn_len) 5575c50d8ae3SPaolo Bonzini { 557692daa48bSSean Christopherson int r, emulation_type = EMULTYPE_PF; 5577347a0d0dSPaolo Bonzini bool direct = vcpu->arch.mmu->root_role.direct; 5578c50d8ae3SPaolo Bonzini 5579b9e5603cSPaolo Bonzini if (WARN_ON(!VALID_PAGE(vcpu->arch.mmu->root.hpa))) 5580ddce6208SSean Christopherson return RET_PF_RETRY; 5581ddce6208SSean Christopherson 5582c50d8ae3SPaolo Bonzini r = RET_PF_INVALID; 5583c50d8ae3SPaolo Bonzini if (unlikely(error_code & PFERR_RSVD_MASK)) { 5584736c291cSSean Christopherson r = handle_mmio_page_fault(vcpu, cr2_or_gpa, direct); 5585c50d8ae3SPaolo Bonzini if (r == RET_PF_EMULATE) 5586c50d8ae3SPaolo Bonzini goto emulate; 5587c50d8ae3SPaolo Bonzini } 5588c50d8ae3SPaolo Bonzini 5589c50d8ae3SPaolo Bonzini if (r == RET_PF_INVALID) { 55907a02674dSSean Christopherson r = kvm_mmu_do_page_fault(vcpu, cr2_or_gpa, 55917a02674dSSean Christopherson lower_32_bits(error_code), false); 559219025e7bSSean Christopherson if (KVM_BUG_ON(r == RET_PF_INVALID, vcpu->kvm)) 55937b367bc9SSean Christopherson return -EIO; 5594c50d8ae3SPaolo Bonzini } 5595c50d8ae3SPaolo Bonzini 5596c50d8ae3SPaolo Bonzini if (r < 0) 5597c50d8ae3SPaolo Bonzini return r; 559883a2ba4cSSean Christopherson if (r != RET_PF_EMULATE) 559983a2ba4cSSean Christopherson return 1; 5600c50d8ae3SPaolo Bonzini 5601c50d8ae3SPaolo Bonzini /* 5602c50d8ae3SPaolo Bonzini * Before emulating the instruction, check if the error code 5603c50d8ae3SPaolo Bonzini * was due to a RO violation while translating the guest page. 5604c50d8ae3SPaolo Bonzini * This can occur when using nested virtualization with nested 5605c50d8ae3SPaolo Bonzini * paging in both guests. If true, we simply unprotect the page 5606c50d8ae3SPaolo Bonzini * and resume the guest. 5607c50d8ae3SPaolo Bonzini */ 5608347a0d0dSPaolo Bonzini if (vcpu->arch.mmu->root_role.direct && 5609c50d8ae3SPaolo Bonzini (error_code & PFERR_NESTED_GUEST_PAGE) == PFERR_NESTED_GUEST_PAGE) { 5610736c291cSSean Christopherson kvm_mmu_unprotect_page(vcpu->kvm, gpa_to_gfn(cr2_or_gpa)); 5611c50d8ae3SPaolo Bonzini return 1; 5612c50d8ae3SPaolo Bonzini } 5613c50d8ae3SPaolo Bonzini 5614c50d8ae3SPaolo Bonzini /* 5615c50d8ae3SPaolo Bonzini * vcpu->arch.mmu.page_fault returned RET_PF_EMULATE, but we can still 5616c50d8ae3SPaolo Bonzini * optimistically try to just unprotect the page and let the processor 5617c50d8ae3SPaolo Bonzini * re-execute the instruction that caused the page fault. Do not allow 5618c50d8ae3SPaolo Bonzini * retrying MMIO emulation, as it's not only pointless but could also 5619c50d8ae3SPaolo Bonzini * cause us to enter an infinite loop because the processor will keep 5620c50d8ae3SPaolo Bonzini * faulting on the non-existent MMIO address. Retrying an instruction 5621c50d8ae3SPaolo Bonzini * from a nested guest is also pointless and dangerous as we are only 5622c50d8ae3SPaolo Bonzini * explicitly shadowing L1's page tables, i.e. unprotecting something 5623c50d8ae3SPaolo Bonzini * for L1 isn't going to magically fix whatever issue cause L2 to fail. 5624c50d8ae3SPaolo Bonzini */ 5625736c291cSSean Christopherson if (!mmio_info_in_cache(vcpu, cr2_or_gpa, direct) && !is_guest_mode(vcpu)) 562692daa48bSSean Christopherson emulation_type |= EMULTYPE_ALLOW_RETRY_PF; 5627c50d8ae3SPaolo Bonzini emulate: 5628736c291cSSean Christopherson return x86_emulate_instruction(vcpu, cr2_or_gpa, emulation_type, insn, 5629c50d8ae3SPaolo Bonzini insn_len); 5630c50d8ae3SPaolo Bonzini } 5631c50d8ae3SPaolo Bonzini EXPORT_SYMBOL_GPL(kvm_mmu_page_fault); 5632c50d8ae3SPaolo Bonzini 56335efac074SPaolo Bonzini void kvm_mmu_invalidate_gva(struct kvm_vcpu *vcpu, struct kvm_mmu *mmu, 56345efac074SPaolo Bonzini gva_t gva, hpa_t root_hpa) 5635c50d8ae3SPaolo Bonzini { 5636c50d8ae3SPaolo Bonzini int i; 5637c50d8ae3SPaolo Bonzini 56385efac074SPaolo Bonzini /* It's actually a GPA for vcpu->arch.guest_mmu. */ 56395efac074SPaolo Bonzini if (mmu != &vcpu->arch.guest_mmu) { 56405efac074SPaolo Bonzini /* INVLPG on a non-canonical address is a NOP according to the SDM. */ 5641c50d8ae3SPaolo Bonzini if (is_noncanonical_address(gva, vcpu)) 5642c50d8ae3SPaolo Bonzini return; 5643c50d8ae3SPaolo Bonzini 5644e27bc044SSean Christopherson static_call(kvm_x86_flush_tlb_gva)(vcpu, gva); 56455efac074SPaolo Bonzini } 56465efac074SPaolo Bonzini 56475efac074SPaolo Bonzini if (!mmu->invlpg) 56485efac074SPaolo Bonzini return; 56495efac074SPaolo Bonzini 56505efac074SPaolo Bonzini if (root_hpa == INVALID_PAGE) { 5651b9e5603cSPaolo Bonzini mmu->invlpg(vcpu, gva, mmu->root.hpa); 5652c50d8ae3SPaolo Bonzini 5653c50d8ae3SPaolo Bonzini /* 5654c50d8ae3SPaolo Bonzini * INVLPG is required to invalidate any global mappings for the VA, 5655c50d8ae3SPaolo Bonzini * irrespective of PCID. Since it would take us roughly similar amount 5656c50d8ae3SPaolo Bonzini * of work to determine whether any of the prev_root mappings of the VA 5657c50d8ae3SPaolo Bonzini * is marked global, or to just sync it blindly, so we might as well 5658c50d8ae3SPaolo Bonzini * just always sync it. 5659c50d8ae3SPaolo Bonzini * 5660c50d8ae3SPaolo Bonzini * Mappings not reachable via the current cr3 or the prev_roots will be 5661c50d8ae3SPaolo Bonzini * synced when switching to that cr3, so nothing needs to be done here 5662c50d8ae3SPaolo Bonzini * for them. 5663c50d8ae3SPaolo Bonzini */ 5664c50d8ae3SPaolo Bonzini for (i = 0; i < KVM_MMU_NUM_PREV_ROOTS; i++) 5665c50d8ae3SPaolo Bonzini if (VALID_PAGE(mmu->prev_roots[i].hpa)) 5666c50d8ae3SPaolo Bonzini mmu->invlpg(vcpu, gva, mmu->prev_roots[i].hpa); 56675efac074SPaolo Bonzini } else { 56685efac074SPaolo Bonzini mmu->invlpg(vcpu, gva, root_hpa); 56695efac074SPaolo Bonzini } 56705efac074SPaolo Bonzini } 5671c50d8ae3SPaolo Bonzini 56725efac074SPaolo Bonzini void kvm_mmu_invlpg(struct kvm_vcpu *vcpu, gva_t gva) 56735efac074SPaolo Bonzini { 567405b29633SLai Jiangshan kvm_mmu_invalidate_gva(vcpu, vcpu->arch.walk_mmu, gva, INVALID_PAGE); 5675c50d8ae3SPaolo Bonzini ++vcpu->stat.invlpg; 5676c50d8ae3SPaolo Bonzini } 5677c50d8ae3SPaolo Bonzini EXPORT_SYMBOL_GPL(kvm_mmu_invlpg); 5678c50d8ae3SPaolo Bonzini 56795efac074SPaolo Bonzini 5680c50d8ae3SPaolo Bonzini void kvm_mmu_invpcid_gva(struct kvm_vcpu *vcpu, gva_t gva, unsigned long pcid) 5681c50d8ae3SPaolo Bonzini { 5682c50d8ae3SPaolo Bonzini struct kvm_mmu *mmu = vcpu->arch.mmu; 5683c50d8ae3SPaolo Bonzini bool tlb_flush = false; 5684c50d8ae3SPaolo Bonzini uint i; 5685c50d8ae3SPaolo Bonzini 5686c50d8ae3SPaolo Bonzini if (pcid == kvm_get_active_pcid(vcpu)) { 56879f46c187SPaolo Bonzini if (mmu->invlpg) 5688b9e5603cSPaolo Bonzini mmu->invlpg(vcpu, gva, mmu->root.hpa); 5689c50d8ae3SPaolo Bonzini tlb_flush = true; 5690c50d8ae3SPaolo Bonzini } 5691c50d8ae3SPaolo Bonzini 5692c50d8ae3SPaolo Bonzini for (i = 0; i < KVM_MMU_NUM_PREV_ROOTS; i++) { 5693c50d8ae3SPaolo Bonzini if (VALID_PAGE(mmu->prev_roots[i].hpa) && 5694be01e8e2SSean Christopherson pcid == kvm_get_pcid(vcpu, mmu->prev_roots[i].pgd)) { 56959f46c187SPaolo Bonzini if (mmu->invlpg) 5696c50d8ae3SPaolo Bonzini mmu->invlpg(vcpu, gva, mmu->prev_roots[i].hpa); 5697c50d8ae3SPaolo Bonzini tlb_flush = true; 5698c50d8ae3SPaolo Bonzini } 5699c50d8ae3SPaolo Bonzini } 5700c50d8ae3SPaolo Bonzini 5701c50d8ae3SPaolo Bonzini if (tlb_flush) 5702e27bc044SSean Christopherson static_call(kvm_x86_flush_tlb_gva)(vcpu, gva); 5703c50d8ae3SPaolo Bonzini 5704c50d8ae3SPaolo Bonzini ++vcpu->stat.invlpg; 5705c50d8ae3SPaolo Bonzini 5706c50d8ae3SPaolo Bonzini /* 5707c50d8ae3SPaolo Bonzini * Mappings not reachable via the current cr3 or the prev_roots will be 5708c50d8ae3SPaolo Bonzini * synced when switching to that cr3, so nothing needs to be done here 5709c50d8ae3SPaolo Bonzini * for them. 5710c50d8ae3SPaolo Bonzini */ 5711c50d8ae3SPaolo Bonzini } 5712c50d8ae3SPaolo Bonzini 5713746700d2SWei Huang void kvm_configure_mmu(bool enable_tdp, int tdp_forced_root_level, 5714746700d2SWei Huang int tdp_max_root_level, int tdp_huge_page_level) 5715c50d8ae3SPaolo Bonzini { 5716bde77235SSean Christopherson tdp_enabled = enable_tdp; 5717746700d2SWei Huang tdp_root_level = tdp_forced_root_level; 571883013059SSean Christopherson max_tdp_level = tdp_max_root_level; 5719703c335dSSean Christopherson 5720703c335dSSean Christopherson /* 57211d92d2e8SSean Christopherson * max_huge_page_level reflects KVM's MMU capabilities irrespective 5722703c335dSSean Christopherson * of kernel support, e.g. KVM may be capable of using 1GB pages when 5723703c335dSSean Christopherson * the kernel is not. But, KVM never creates a page size greater than 5724703c335dSSean Christopherson * what is used by the kernel for any given HVA, i.e. the kernel's 5725703c335dSSean Christopherson * capabilities are ultimately consulted by kvm_mmu_hugepage_adjust(). 5726703c335dSSean Christopherson */ 5727703c335dSSean Christopherson if (tdp_enabled) 57281d92d2e8SSean Christopherson max_huge_page_level = tdp_huge_page_level; 5729703c335dSSean Christopherson else if (boot_cpu_has(X86_FEATURE_GBPAGES)) 57301d92d2e8SSean Christopherson max_huge_page_level = PG_LEVEL_1G; 5731703c335dSSean Christopherson else 57321d92d2e8SSean Christopherson max_huge_page_level = PG_LEVEL_2M; 5733c50d8ae3SPaolo Bonzini } 5734bde77235SSean Christopherson EXPORT_SYMBOL_GPL(kvm_configure_mmu); 5735c50d8ae3SPaolo Bonzini 5736c50d8ae3SPaolo Bonzini /* The return value indicates if tlb flush on all vcpus is needed. */ 5737269e9552SHamza Mahfooz typedef bool (*slot_level_handler) (struct kvm *kvm, 5738269e9552SHamza Mahfooz struct kvm_rmap_head *rmap_head, 5739269e9552SHamza Mahfooz const struct kvm_memory_slot *slot); 5740c50d8ae3SPaolo Bonzini 5741c50d8ae3SPaolo Bonzini /* The caller should hold mmu-lock before calling this function. */ 5742c50d8ae3SPaolo Bonzini static __always_inline bool 5743269e9552SHamza Mahfooz slot_handle_level_range(struct kvm *kvm, const struct kvm_memory_slot *memslot, 5744c50d8ae3SPaolo Bonzini slot_level_handler fn, int start_level, int end_level, 57451a61b7dbSSean Christopherson gfn_t start_gfn, gfn_t end_gfn, bool flush_on_yield, 57461a61b7dbSSean Christopherson bool flush) 5747c50d8ae3SPaolo Bonzini { 5748c50d8ae3SPaolo Bonzini struct slot_rmap_walk_iterator iterator; 5749c50d8ae3SPaolo Bonzini 5750c50d8ae3SPaolo Bonzini for_each_slot_rmap_range(memslot, start_level, end_level, start_gfn, 5751c50d8ae3SPaolo Bonzini end_gfn, &iterator) { 5752c50d8ae3SPaolo Bonzini if (iterator.rmap) 57530a234f5dSSean Christopherson flush |= fn(kvm, iterator.rmap, memslot); 5754c50d8ae3SPaolo Bonzini 5755531810caSBen Gardon if (need_resched() || rwlock_needbreak(&kvm->mmu_lock)) { 5756302695a5SSean Christopherson if (flush && flush_on_yield) { 5757c50d8ae3SPaolo Bonzini kvm_flush_remote_tlbs_with_address(kvm, 5758c50d8ae3SPaolo Bonzini start_gfn, 5759c50d8ae3SPaolo Bonzini iterator.gfn - start_gfn + 1); 5760c50d8ae3SPaolo Bonzini flush = false; 5761c50d8ae3SPaolo Bonzini } 5762531810caSBen Gardon cond_resched_rwlock_write(&kvm->mmu_lock); 5763c50d8ae3SPaolo Bonzini } 5764c50d8ae3SPaolo Bonzini } 5765c50d8ae3SPaolo Bonzini 5766c50d8ae3SPaolo Bonzini return flush; 5767c50d8ae3SPaolo Bonzini } 5768c50d8ae3SPaolo Bonzini 5769c50d8ae3SPaolo Bonzini static __always_inline bool 5770269e9552SHamza Mahfooz slot_handle_level(struct kvm *kvm, const struct kvm_memory_slot *memslot, 5771c50d8ae3SPaolo Bonzini slot_level_handler fn, int start_level, int end_level, 5772302695a5SSean Christopherson bool flush_on_yield) 5773c50d8ae3SPaolo Bonzini { 5774c50d8ae3SPaolo Bonzini return slot_handle_level_range(kvm, memslot, fn, start_level, 5775c50d8ae3SPaolo Bonzini end_level, memslot->base_gfn, 5776c50d8ae3SPaolo Bonzini memslot->base_gfn + memslot->npages - 1, 57771a61b7dbSSean Christopherson flush_on_yield, false); 5778c50d8ae3SPaolo Bonzini } 5779c50d8ae3SPaolo Bonzini 5780c50d8ae3SPaolo Bonzini static __always_inline bool 5781610265eaSDavid Matlack slot_handle_level_4k(struct kvm *kvm, const struct kvm_memory_slot *memslot, 5782302695a5SSean Christopherson slot_level_handler fn, bool flush_on_yield) 5783c50d8ae3SPaolo Bonzini { 57843bae0459SSean Christopherson return slot_handle_level(kvm, memslot, fn, PG_LEVEL_4K, 5785302695a5SSean Christopherson PG_LEVEL_4K, flush_on_yield); 5786c50d8ae3SPaolo Bonzini } 5787c50d8ae3SPaolo Bonzini 5788c50d8ae3SPaolo Bonzini static void free_mmu_pages(struct kvm_mmu *mmu) 5789c50d8ae3SPaolo Bonzini { 57904a98623dSSean Christopherson if (!tdp_enabled && mmu->pae_root) 57914a98623dSSean Christopherson set_memory_encrypted((unsigned long)mmu->pae_root, 1); 5792c50d8ae3SPaolo Bonzini free_page((unsigned long)mmu->pae_root); 579303ca4589SSean Christopherson free_page((unsigned long)mmu->pml4_root); 5794cb0f722aSWei Huang free_page((unsigned long)mmu->pml5_root); 5795c50d8ae3SPaolo Bonzini } 5796c50d8ae3SPaolo Bonzini 579704d28e37SSean Christopherson static int __kvm_mmu_create(struct kvm_vcpu *vcpu, struct kvm_mmu *mmu) 5798c50d8ae3SPaolo Bonzini { 5799c50d8ae3SPaolo Bonzini struct page *page; 5800c50d8ae3SPaolo Bonzini int i; 5801c50d8ae3SPaolo Bonzini 5802b9e5603cSPaolo Bonzini mmu->root.hpa = INVALID_PAGE; 5803b9e5603cSPaolo Bonzini mmu->root.pgd = 0; 580404d28e37SSean Christopherson for (i = 0; i < KVM_MMU_NUM_PREV_ROOTS; i++) 580504d28e37SSean Christopherson mmu->prev_roots[i] = KVM_MMU_ROOT_INFO_INVALID; 580604d28e37SSean Christopherson 580727f4fca2SLai Jiangshan /* vcpu->arch.guest_mmu isn't used when !tdp_enabled. */ 580827f4fca2SLai Jiangshan if (!tdp_enabled && mmu == &vcpu->arch.guest_mmu) 580927f4fca2SLai Jiangshan return 0; 581027f4fca2SLai Jiangshan 5811c50d8ae3SPaolo Bonzini /* 5812c50d8ae3SPaolo Bonzini * When using PAE paging, the four PDPTEs are treated as 'root' pages, 5813c50d8ae3SPaolo Bonzini * while the PDP table is a per-vCPU construct that's allocated at MMU 5814c50d8ae3SPaolo Bonzini * creation. When emulating 32-bit mode, cr3 is only 32 bits even on 5815c50d8ae3SPaolo Bonzini * x86_64. Therefore we need to allocate the PDP table in the first 581604d45551SSean Christopherson * 4GB of memory, which happens to fit the DMA32 zone. TDP paging 581704d45551SSean Christopherson * generally doesn't use PAE paging and can skip allocating the PDP 581804d45551SSean Christopherson * table. The main exception, handled here, is SVM's 32-bit NPT. The 581904d45551SSean Christopherson * other exception is for shadowing L1's 32-bit or PAE NPT on 64-bit 582084432316SLai Jiangshan * KVM; that horror is handled on-demand by mmu_alloc_special_roots(). 5821c50d8ae3SPaolo Bonzini */ 5822d468d94bSSean Christopherson if (tdp_enabled && kvm_mmu_get_tdp_level(vcpu) > PT32E_ROOT_LEVEL) 5823c50d8ae3SPaolo Bonzini return 0; 5824c50d8ae3SPaolo Bonzini 5825c50d8ae3SPaolo Bonzini page = alloc_page(GFP_KERNEL_ACCOUNT | __GFP_DMA32); 5826c50d8ae3SPaolo Bonzini if (!page) 5827c50d8ae3SPaolo Bonzini return -ENOMEM; 5828c50d8ae3SPaolo Bonzini 5829c50d8ae3SPaolo Bonzini mmu->pae_root = page_address(page); 58304a98623dSSean Christopherson 58314a98623dSSean Christopherson /* 58324a98623dSSean Christopherson * CR3 is only 32 bits when PAE paging is used, thus it's impossible to 58334a98623dSSean Christopherson * get the CPU to treat the PDPTEs as encrypted. Decrypt the page so 58344a98623dSSean Christopherson * that KVM's writes and the CPU's reads get along. Note, this is 58354a98623dSSean Christopherson * only necessary when using shadow paging, as 64-bit NPT can get at 58364a98623dSSean Christopherson * the C-bit even when shadowing 32-bit NPT, and SME isn't supported 58374a98623dSSean Christopherson * by 32-bit kernels (when KVM itself uses 32-bit NPT). 58384a98623dSSean Christopherson */ 58394a98623dSSean Christopherson if (!tdp_enabled) 58404a98623dSSean Christopherson set_memory_decrypted((unsigned long)mmu->pae_root, 1); 58414a98623dSSean Christopherson else 5842e54f1ff2SKai Huang WARN_ON_ONCE(shadow_me_value); 58434a98623dSSean Christopherson 5844c50d8ae3SPaolo Bonzini for (i = 0; i < 4; ++i) 5845c834e5e4SSean Christopherson mmu->pae_root[i] = INVALID_PAE_ROOT; 5846c50d8ae3SPaolo Bonzini 5847c50d8ae3SPaolo Bonzini return 0; 5848c50d8ae3SPaolo Bonzini } 5849c50d8ae3SPaolo Bonzini 5850c50d8ae3SPaolo Bonzini int kvm_mmu_create(struct kvm_vcpu *vcpu) 5851c50d8ae3SPaolo Bonzini { 5852c50d8ae3SPaolo Bonzini int ret; 5853c50d8ae3SPaolo Bonzini 58545962bfb7SSean Christopherson vcpu->arch.mmu_pte_list_desc_cache.kmem_cache = pte_list_desc_cache; 58555f6078f9SSean Christopherson vcpu->arch.mmu_pte_list_desc_cache.gfp_zero = __GFP_ZERO; 58565f6078f9SSean Christopherson 58575962bfb7SSean Christopherson vcpu->arch.mmu_page_header_cache.kmem_cache = mmu_page_header_cache; 58585f6078f9SSean Christopherson vcpu->arch.mmu_page_header_cache.gfp_zero = __GFP_ZERO; 58595962bfb7SSean Christopherson 586096880883SSean Christopherson vcpu->arch.mmu_shadow_page_cache.gfp_zero = __GFP_ZERO; 586196880883SSean Christopherson 5862c50d8ae3SPaolo Bonzini vcpu->arch.mmu = &vcpu->arch.root_mmu; 5863c50d8ae3SPaolo Bonzini vcpu->arch.walk_mmu = &vcpu->arch.root_mmu; 5864c50d8ae3SPaolo Bonzini 586504d28e37SSean Christopherson ret = __kvm_mmu_create(vcpu, &vcpu->arch.guest_mmu); 5866c50d8ae3SPaolo Bonzini if (ret) 5867c50d8ae3SPaolo Bonzini return ret; 5868c50d8ae3SPaolo Bonzini 586904d28e37SSean Christopherson ret = __kvm_mmu_create(vcpu, &vcpu->arch.root_mmu); 5870c50d8ae3SPaolo Bonzini if (ret) 5871c50d8ae3SPaolo Bonzini goto fail_allocate_root; 5872c50d8ae3SPaolo Bonzini 5873c50d8ae3SPaolo Bonzini return ret; 5874c50d8ae3SPaolo Bonzini fail_allocate_root: 5875c50d8ae3SPaolo Bonzini free_mmu_pages(&vcpu->arch.guest_mmu); 5876c50d8ae3SPaolo Bonzini return ret; 5877c50d8ae3SPaolo Bonzini } 5878c50d8ae3SPaolo Bonzini 5879c50d8ae3SPaolo Bonzini #define BATCH_ZAP_PAGES 10 5880c50d8ae3SPaolo Bonzini static void kvm_zap_obsolete_pages(struct kvm *kvm) 5881c50d8ae3SPaolo Bonzini { 5882c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp, *node; 5883c50d8ae3SPaolo Bonzini int nr_zapped, batch = 0; 5884b28cb0cdSSean Christopherson bool unstable; 5885c50d8ae3SPaolo Bonzini 5886c50d8ae3SPaolo Bonzini restart: 5887c50d8ae3SPaolo Bonzini list_for_each_entry_safe_reverse(sp, node, 5888c50d8ae3SPaolo Bonzini &kvm->arch.active_mmu_pages, link) { 5889c50d8ae3SPaolo Bonzini /* 5890c50d8ae3SPaolo Bonzini * No obsolete valid page exists before a newly created page 5891c50d8ae3SPaolo Bonzini * since active_mmu_pages is a FIFO list. 5892c50d8ae3SPaolo Bonzini */ 5893c50d8ae3SPaolo Bonzini if (!is_obsolete_sp(kvm, sp)) 5894c50d8ae3SPaolo Bonzini break; 5895c50d8ae3SPaolo Bonzini 5896c50d8ae3SPaolo Bonzini /* 5897f95eec9bSSean Christopherson * Invalid pages should never land back on the list of active 5898f95eec9bSSean Christopherson * pages. Skip the bogus page, otherwise we'll get stuck in an 5899f95eec9bSSean Christopherson * infinite loop if the page gets put back on the list (again). 5900c50d8ae3SPaolo Bonzini */ 5901f95eec9bSSean Christopherson if (WARN_ON(sp->role.invalid)) 5902c50d8ae3SPaolo Bonzini continue; 5903c50d8ae3SPaolo Bonzini 5904c50d8ae3SPaolo Bonzini /* 5905c50d8ae3SPaolo Bonzini * No need to flush the TLB since we're only zapping shadow 5906c50d8ae3SPaolo Bonzini * pages with an obsolete generation number and all vCPUS have 5907c50d8ae3SPaolo Bonzini * loaded a new root, i.e. the shadow pages being zapped cannot 5908c50d8ae3SPaolo Bonzini * be in active use by the guest. 5909c50d8ae3SPaolo Bonzini */ 5910c50d8ae3SPaolo Bonzini if (batch >= BATCH_ZAP_PAGES && 5911531810caSBen Gardon cond_resched_rwlock_write(&kvm->mmu_lock)) { 5912c50d8ae3SPaolo Bonzini batch = 0; 5913c50d8ae3SPaolo Bonzini goto restart; 5914c50d8ae3SPaolo Bonzini } 5915c50d8ae3SPaolo Bonzini 5916b28cb0cdSSean Christopherson unstable = __kvm_mmu_prepare_zap_page(kvm, sp, 5917b28cb0cdSSean Christopherson &kvm->arch.zapped_obsolete_pages, &nr_zapped); 5918c50d8ae3SPaolo Bonzini batch += nr_zapped; 5919b28cb0cdSSean Christopherson 5920b28cb0cdSSean Christopherson if (unstable) 5921c50d8ae3SPaolo Bonzini goto restart; 5922c50d8ae3SPaolo Bonzini } 5923c50d8ae3SPaolo Bonzini 5924c50d8ae3SPaolo Bonzini /* 59257ae5840eSSean Christopherson * Kick all vCPUs (via remote TLB flush) before freeing the page tables 59267ae5840eSSean Christopherson * to ensure KVM is not in the middle of a lockless shadow page table 59277ae5840eSSean Christopherson * walk, which may reference the pages. The remote TLB flush itself is 59287ae5840eSSean Christopherson * not required and is simply a convenient way to kick vCPUs as needed. 59297ae5840eSSean Christopherson * KVM performs a local TLB flush when allocating a new root (see 59307ae5840eSSean Christopherson * kvm_mmu_load()), and the reload in the caller ensure no vCPUs are 59317ae5840eSSean Christopherson * running with an obsolete MMU. 5932c50d8ae3SPaolo Bonzini */ 5933c50d8ae3SPaolo Bonzini kvm_mmu_commit_zap_page(kvm, &kvm->arch.zapped_obsolete_pages); 5934c50d8ae3SPaolo Bonzini } 5935c50d8ae3SPaolo Bonzini 5936c50d8ae3SPaolo Bonzini /* 5937c50d8ae3SPaolo Bonzini * Fast invalidate all shadow pages and use lock-break technique 5938c50d8ae3SPaolo Bonzini * to zap obsolete pages. 5939c50d8ae3SPaolo Bonzini * 5940c50d8ae3SPaolo Bonzini * It's required when memslot is being deleted or VM is being 5941c50d8ae3SPaolo Bonzini * destroyed, in these cases, we should ensure that KVM MMU does 5942c50d8ae3SPaolo Bonzini * not use any resource of the being-deleted slot or all slots 5943c50d8ae3SPaolo Bonzini * after calling the function. 5944c50d8ae3SPaolo Bonzini */ 5945c50d8ae3SPaolo Bonzini static void kvm_mmu_zap_all_fast(struct kvm *kvm) 5946c50d8ae3SPaolo Bonzini { 5947c50d8ae3SPaolo Bonzini lockdep_assert_held(&kvm->slots_lock); 5948c50d8ae3SPaolo Bonzini 5949531810caSBen Gardon write_lock(&kvm->mmu_lock); 5950c50d8ae3SPaolo Bonzini trace_kvm_mmu_zap_all_fast(kvm); 5951c50d8ae3SPaolo Bonzini 5952c50d8ae3SPaolo Bonzini /* 5953c50d8ae3SPaolo Bonzini * Toggle mmu_valid_gen between '0' and '1'. Because slots_lock is 5954c50d8ae3SPaolo Bonzini * held for the entire duration of zapping obsolete pages, it's 5955c50d8ae3SPaolo Bonzini * impossible for there to be multiple invalid generations associated 5956c50d8ae3SPaolo Bonzini * with *valid* shadow pages at any given time, i.e. there is exactly 5957c50d8ae3SPaolo Bonzini * one valid generation and (at most) one invalid generation. 5958c50d8ae3SPaolo Bonzini */ 5959c50d8ae3SPaolo Bonzini kvm->arch.mmu_valid_gen = kvm->arch.mmu_valid_gen ? 0 : 1; 5960c50d8ae3SPaolo Bonzini 59612f6f66ccSSean Christopherson /* 59622f6f66ccSSean Christopherson * In order to ensure all vCPUs drop their soon-to-be invalid roots, 59632f6f66ccSSean Christopherson * invalidating TDP MMU roots must be done while holding mmu_lock for 59642f6f66ccSSean Christopherson * write and in the same critical section as making the reload request, 59652f6f66ccSSean Christopherson * e.g. before kvm_zap_obsolete_pages() could drop mmu_lock and yield. 5966b7cccd39SBen Gardon */ 5967b7cccd39SBen Gardon if (is_tdp_mmu_enabled(kvm)) 5968b7cccd39SBen Gardon kvm_tdp_mmu_invalidate_all_roots(kvm); 5969b7cccd39SBen Gardon 5970c50d8ae3SPaolo Bonzini /* 5971c50d8ae3SPaolo Bonzini * Notify all vcpus to reload its shadow page table and flush TLB. 5972c50d8ae3SPaolo Bonzini * Then all vcpus will switch to new shadow page table with the new 5973c50d8ae3SPaolo Bonzini * mmu_valid_gen. 5974c50d8ae3SPaolo Bonzini * 5975c50d8ae3SPaolo Bonzini * Note: we need to do this under the protection of mmu_lock, 5976c50d8ae3SPaolo Bonzini * otherwise, vcpu would purge shadow page but miss tlb flush. 5977c50d8ae3SPaolo Bonzini */ 5978527d5cd7SSean Christopherson kvm_make_all_cpus_request(kvm, KVM_REQ_MMU_FREE_OBSOLETE_ROOTS); 5979c50d8ae3SPaolo Bonzini 5980c50d8ae3SPaolo Bonzini kvm_zap_obsolete_pages(kvm); 5981faaf05b0SBen Gardon 5982531810caSBen Gardon write_unlock(&kvm->mmu_lock); 59834c6654bdSBen Gardon 5984f28e9c7fSSean Christopherson /* 5985f28e9c7fSSean Christopherson * Zap the invalidated TDP MMU roots, all SPTEs must be dropped before 5986f28e9c7fSSean Christopherson * returning to the caller, e.g. if the zap is in response to a memslot 5987f28e9c7fSSean Christopherson * deletion, mmu_notifier callbacks will be unable to reach the SPTEs 5988f28e9c7fSSean Christopherson * associated with the deleted memslot once the update completes, and 5989f28e9c7fSSean Christopherson * Deferring the zap until the final reference to the root is put would 5990f28e9c7fSSean Christopherson * lead to use-after-free. 5991f28e9c7fSSean Christopherson */ 599222b94c4bSPaolo Bonzini if (is_tdp_mmu_enabled(kvm)) 59934c6654bdSBen Gardon kvm_tdp_mmu_zap_invalidated_roots(kvm); 5994c50d8ae3SPaolo Bonzini } 5995c50d8ae3SPaolo Bonzini 5996c50d8ae3SPaolo Bonzini static bool kvm_has_zapped_obsolete_pages(struct kvm *kvm) 5997c50d8ae3SPaolo Bonzini { 5998c50d8ae3SPaolo Bonzini return unlikely(!list_empty_careful(&kvm->arch.zapped_obsolete_pages)); 5999c50d8ae3SPaolo Bonzini } 6000c50d8ae3SPaolo Bonzini 6001c50d8ae3SPaolo Bonzini static void kvm_mmu_invalidate_zap_pages_in_memslot(struct kvm *kvm, 6002c50d8ae3SPaolo Bonzini struct kvm_memory_slot *slot, 6003c50d8ae3SPaolo Bonzini struct kvm_page_track_notifier_node *node) 6004c50d8ae3SPaolo Bonzini { 6005c50d8ae3SPaolo Bonzini kvm_mmu_zap_all_fast(kvm); 6006c50d8ae3SPaolo Bonzini } 6007c50d8ae3SPaolo Bonzini 6008a1a39128SPaolo Bonzini int kvm_mmu_init_vm(struct kvm *kvm) 6009c50d8ae3SPaolo Bonzini { 6010c50d8ae3SPaolo Bonzini struct kvm_page_track_notifier_node *node = &kvm->arch.mmu_sp_tracker; 6011a1a39128SPaolo Bonzini int r; 6012c50d8ae3SPaolo Bonzini 6013a1a39128SPaolo Bonzini INIT_LIST_HEAD(&kvm->arch.active_mmu_pages); 6014a1a39128SPaolo Bonzini INIT_LIST_HEAD(&kvm->arch.zapped_obsolete_pages); 601555c510e2SSean Christopherson INIT_LIST_HEAD(&kvm->arch.possible_nx_huge_pages); 6016ce25681dSSean Christopherson spin_lock_init(&kvm->arch.mmu_unsync_pages_lock); 6017ce25681dSSean Christopherson 6018a1a39128SPaolo Bonzini r = kvm_mmu_init_tdp_mmu(kvm); 6019a1a39128SPaolo Bonzini if (r < 0) 6020a1a39128SPaolo Bonzini return r; 6021fe5db27dSBen Gardon 6022c50d8ae3SPaolo Bonzini node->track_write = kvm_mmu_pte_write; 6023c50d8ae3SPaolo Bonzini node->track_flush_slot = kvm_mmu_invalidate_zap_pages_in_memslot; 6024c50d8ae3SPaolo Bonzini kvm_page_track_register_notifier(kvm, node); 6025ada51a9dSDavid Matlack 6026ada51a9dSDavid Matlack kvm->arch.split_page_header_cache.kmem_cache = mmu_page_header_cache; 6027ada51a9dSDavid Matlack kvm->arch.split_page_header_cache.gfp_zero = __GFP_ZERO; 6028ada51a9dSDavid Matlack 6029ada51a9dSDavid Matlack kvm->arch.split_shadow_page_cache.gfp_zero = __GFP_ZERO; 6030ada51a9dSDavid Matlack 6031ada51a9dSDavid Matlack kvm->arch.split_desc_cache.kmem_cache = pte_list_desc_cache; 6032ada51a9dSDavid Matlack kvm->arch.split_desc_cache.gfp_zero = __GFP_ZERO; 6033ada51a9dSDavid Matlack 6034a1a39128SPaolo Bonzini return 0; 6035c50d8ae3SPaolo Bonzini } 6036c50d8ae3SPaolo Bonzini 6037ada51a9dSDavid Matlack static void mmu_free_vm_memory_caches(struct kvm *kvm) 6038ada51a9dSDavid Matlack { 6039ada51a9dSDavid Matlack kvm_mmu_free_memory_cache(&kvm->arch.split_desc_cache); 6040ada51a9dSDavid Matlack kvm_mmu_free_memory_cache(&kvm->arch.split_page_header_cache); 6041ada51a9dSDavid Matlack kvm_mmu_free_memory_cache(&kvm->arch.split_shadow_page_cache); 6042ada51a9dSDavid Matlack } 6043ada51a9dSDavid Matlack 6044c50d8ae3SPaolo Bonzini void kvm_mmu_uninit_vm(struct kvm *kvm) 6045c50d8ae3SPaolo Bonzini { 6046c50d8ae3SPaolo Bonzini struct kvm_page_track_notifier_node *node = &kvm->arch.mmu_sp_tracker; 6047c50d8ae3SPaolo Bonzini 6048c50d8ae3SPaolo Bonzini kvm_page_track_unregister_notifier(kvm, node); 6049fe5db27dSBen Gardon 6050fe5db27dSBen Gardon kvm_mmu_uninit_tdp_mmu(kvm); 6051ada51a9dSDavid Matlack 6052ada51a9dSDavid Matlack mmu_free_vm_memory_caches(kvm); 6053c50d8ae3SPaolo Bonzini } 6054c50d8ae3SPaolo Bonzini 60552833eda0SSean Christopherson static bool kvm_rmap_zap_gfn_range(struct kvm *kvm, gfn_t gfn_start, gfn_t gfn_end) 605621fa3246SSean Christopherson { 605721fa3246SSean Christopherson const struct kvm_memory_slot *memslot; 605821fa3246SSean Christopherson struct kvm_memslots *slots; 6059f4209439SMaciej S. Szmigiero struct kvm_memslot_iter iter; 606021fa3246SSean Christopherson bool flush = false; 606121fa3246SSean Christopherson gfn_t start, end; 6062f4209439SMaciej S. Szmigiero int i; 606321fa3246SSean Christopherson 606421fa3246SSean Christopherson if (!kvm_memslots_have_rmaps(kvm)) 606521fa3246SSean Christopherson return flush; 606621fa3246SSean Christopherson 606721fa3246SSean Christopherson for (i = 0; i < KVM_ADDRESS_SPACE_NUM; i++) { 606821fa3246SSean Christopherson slots = __kvm_memslots(kvm, i); 6069f4209439SMaciej S. Szmigiero 6070f4209439SMaciej S. Szmigiero kvm_for_each_memslot_in_gfn_range(&iter, slots, gfn_start, gfn_end) { 6071f4209439SMaciej S. Szmigiero memslot = iter.slot; 607221fa3246SSean Christopherson start = max(gfn_start, memslot->base_gfn); 607321fa3246SSean Christopherson end = min(gfn_end, memslot->base_gfn + memslot->npages); 6074f4209439SMaciej S. Szmigiero if (WARN_ON_ONCE(start >= end)) 607521fa3246SSean Christopherson continue; 607621fa3246SSean Christopherson 6077f8480721SSean Christopherson flush = slot_handle_level_range(kvm, memslot, __kvm_zap_rmap, 607821fa3246SSean Christopherson PG_LEVEL_4K, KVM_MAX_HUGEPAGE_LEVEL, 607921fa3246SSean Christopherson start, end - 1, true, flush); 608021fa3246SSean Christopherson } 608121fa3246SSean Christopherson } 608221fa3246SSean Christopherson 608321fa3246SSean Christopherson return flush; 608421fa3246SSean Christopherson } 608521fa3246SSean Christopherson 608688f58535SMaxim Levitsky /* 608788f58535SMaxim Levitsky * Invalidate (zap) SPTEs that cover GFNs from gfn_start and up to gfn_end 608888f58535SMaxim Levitsky * (not including it) 608988f58535SMaxim Levitsky */ 6090c50d8ae3SPaolo Bonzini void kvm_zap_gfn_range(struct kvm *kvm, gfn_t gfn_start, gfn_t gfn_end) 6091c50d8ae3SPaolo Bonzini { 609221fa3246SSean Christopherson bool flush; 6093c50d8ae3SPaolo Bonzini int i; 6094c50d8ae3SPaolo Bonzini 6095f4209439SMaciej S. Szmigiero if (WARN_ON_ONCE(gfn_end <= gfn_start)) 6096f4209439SMaciej S. Szmigiero return; 6097f4209439SMaciej S. Szmigiero 6098531810caSBen Gardon write_lock(&kvm->mmu_lock); 60995a324c24SSean Christopherson 610020ec3ebdSChao Peng kvm_mmu_invalidate_begin(kvm, gfn_start, gfn_end); 6101edb298c6SMaxim Levitsky 61022833eda0SSean Christopherson flush = kvm_rmap_zap_gfn_range(kvm, gfn_start, gfn_end); 61036103bc07SBen Gardon 61046103bc07SBen Gardon if (is_tdp_mmu_enabled(kvm)) { 61056103bc07SBen Gardon for (i = 0; i < KVM_ADDRESS_SPACE_NUM; i++) 6106f47e5bbbSSean Christopherson flush = kvm_tdp_mmu_zap_leafs(kvm, i, gfn_start, 6107f47e5bbbSSean Christopherson gfn_end, true, flush); 61086103bc07SBen Gardon } 61095a324c24SSean Christopherson 61105a324c24SSean Christopherson if (flush) 6111bc3b3c10SSean Christopherson kvm_flush_remote_tlbs_with_address(kvm, gfn_start, 6112bc3b3c10SSean Christopherson gfn_end - gfn_start); 61135a324c24SSean Christopherson 611420ec3ebdSChao Peng kvm_mmu_invalidate_end(kvm, gfn_start, gfn_end); 6115edb298c6SMaxim Levitsky 61165a324c24SSean Christopherson write_unlock(&kvm->mmu_lock); 6117c50d8ae3SPaolo Bonzini } 6118c50d8ae3SPaolo Bonzini 6119c50d8ae3SPaolo Bonzini static bool slot_rmap_write_protect(struct kvm *kvm, 61200a234f5dSSean Christopherson struct kvm_rmap_head *rmap_head, 6121269e9552SHamza Mahfooz const struct kvm_memory_slot *slot) 6122c50d8ae3SPaolo Bonzini { 61231346bbb6SDavid Matlack return rmap_write_protect(rmap_head, false); 6124c50d8ae3SPaolo Bonzini } 6125c50d8ae3SPaolo Bonzini 6126c50d8ae3SPaolo Bonzini void kvm_mmu_slot_remove_write_access(struct kvm *kvm, 6127269e9552SHamza Mahfooz const struct kvm_memory_slot *memslot, 61283c9bd400SJay Zhou int start_level) 6129c50d8ae3SPaolo Bonzini { 6130e2209710SBen Gardon if (kvm_memslots_have_rmaps(kvm)) { 6131531810caSBen Gardon write_lock(&kvm->mmu_lock); 6132b64d740eSJunaid Shahid slot_handle_level(kvm, memslot, slot_rmap_write_protect, 6133b64d740eSJunaid Shahid start_level, KVM_MAX_HUGEPAGE_LEVEL, false); 6134531810caSBen Gardon write_unlock(&kvm->mmu_lock); 6135e2209710SBen Gardon } 6136c50d8ae3SPaolo Bonzini 613724ae4cfaSBen Gardon if (is_tdp_mmu_enabled(kvm)) { 613824ae4cfaSBen Gardon read_lock(&kvm->mmu_lock); 6139b64d740eSJunaid Shahid kvm_tdp_mmu_wrprot_slot(kvm, memslot, start_level); 614024ae4cfaSBen Gardon read_unlock(&kvm->mmu_lock); 614124ae4cfaSBen Gardon } 6142c50d8ae3SPaolo Bonzini } 6143c50d8ae3SPaolo Bonzini 6144ada51a9dSDavid Matlack static inline bool need_topup(struct kvm_mmu_memory_cache *cache, int min) 6145ada51a9dSDavid Matlack { 6146ada51a9dSDavid Matlack return kvm_mmu_memory_cache_nr_free_objects(cache) < min; 6147ada51a9dSDavid Matlack } 6148ada51a9dSDavid Matlack 6149ada51a9dSDavid Matlack static bool need_topup_split_caches_or_resched(struct kvm *kvm) 6150ada51a9dSDavid Matlack { 6151ada51a9dSDavid Matlack if (need_resched() || rwlock_needbreak(&kvm->mmu_lock)) 6152ada51a9dSDavid Matlack return true; 6153ada51a9dSDavid Matlack 6154ada51a9dSDavid Matlack /* 6155ada51a9dSDavid Matlack * In the worst case, SPLIT_DESC_CACHE_MIN_NR_OBJECTS descriptors are needed 6156ada51a9dSDavid Matlack * to split a single huge page. Calculating how many are actually needed 6157ada51a9dSDavid Matlack * is possible but not worth the complexity. 6158ada51a9dSDavid Matlack */ 6159ada51a9dSDavid Matlack return need_topup(&kvm->arch.split_desc_cache, SPLIT_DESC_CACHE_MIN_NR_OBJECTS) || 6160ada51a9dSDavid Matlack need_topup(&kvm->arch.split_page_header_cache, 1) || 6161ada51a9dSDavid Matlack need_topup(&kvm->arch.split_shadow_page_cache, 1); 6162ada51a9dSDavid Matlack } 6163ada51a9dSDavid Matlack 6164ada51a9dSDavid Matlack static int topup_split_caches(struct kvm *kvm) 6165ada51a9dSDavid Matlack { 6166b9b71f43SSean Christopherson /* 6167b9b71f43SSean Christopherson * Allocating rmap list entries when splitting huge pages for nested 6168dfd4eb44SSean Christopherson * MMUs is uncommon as KVM needs to use a list if and only if there is 6169b9b71f43SSean Christopherson * more than one rmap entry for a gfn, i.e. requires an L1 gfn to be 6170dfd4eb44SSean Christopherson * aliased by multiple L2 gfns and/or from multiple nested roots with 6171dfd4eb44SSean Christopherson * different roles. Aliasing gfns when using TDP is atypical for VMMs; 6172dfd4eb44SSean Christopherson * a few gfns are often aliased during boot, e.g. when remapping BIOS, 6173dfd4eb44SSean Christopherson * but aliasing rarely occurs post-boot or for many gfns. If there is 6174dfd4eb44SSean Christopherson * only one rmap entry, rmap->val points directly at that one entry and 6175dfd4eb44SSean Christopherson * doesn't need to allocate a list. Buffer the cache by the default 6176dfd4eb44SSean Christopherson * capacity so that KVM doesn't have to drop mmu_lock to topup if KVM 6177b9b71f43SSean Christopherson * encounters an aliased gfn or two. 6178b9b71f43SSean Christopherson */ 6179b9b71f43SSean Christopherson const int capacity = SPLIT_DESC_CACHE_MIN_NR_OBJECTS + 6180b9b71f43SSean Christopherson KVM_ARCH_NR_OBJS_PER_MEMORY_CACHE; 6181ada51a9dSDavid Matlack int r; 6182ada51a9dSDavid Matlack 6183ada51a9dSDavid Matlack lockdep_assert_held(&kvm->slots_lock); 6184ada51a9dSDavid Matlack 6185b9b71f43SSean Christopherson r = __kvm_mmu_topup_memory_cache(&kvm->arch.split_desc_cache, capacity, 6186ada51a9dSDavid Matlack SPLIT_DESC_CACHE_MIN_NR_OBJECTS); 6187ada51a9dSDavid Matlack if (r) 6188ada51a9dSDavid Matlack return r; 6189ada51a9dSDavid Matlack 6190ada51a9dSDavid Matlack r = kvm_mmu_topup_memory_cache(&kvm->arch.split_page_header_cache, 1); 6191ada51a9dSDavid Matlack if (r) 6192ada51a9dSDavid Matlack return r; 6193ada51a9dSDavid Matlack 6194ada51a9dSDavid Matlack return kvm_mmu_topup_memory_cache(&kvm->arch.split_shadow_page_cache, 1); 6195ada51a9dSDavid Matlack } 6196ada51a9dSDavid Matlack 6197ada51a9dSDavid Matlack static struct kvm_mmu_page *shadow_mmu_get_sp_for_split(struct kvm *kvm, u64 *huge_sptep) 6198ada51a9dSDavid Matlack { 6199ada51a9dSDavid Matlack struct kvm_mmu_page *huge_sp = sptep_to_sp(huge_sptep); 6200ada51a9dSDavid Matlack struct shadow_page_caches caches = {}; 6201ada51a9dSDavid Matlack union kvm_mmu_page_role role; 6202ada51a9dSDavid Matlack unsigned int access; 6203ada51a9dSDavid Matlack gfn_t gfn; 6204ada51a9dSDavid Matlack 620579e48cecSSean Christopherson gfn = kvm_mmu_page_get_gfn(huge_sp, spte_index(huge_sptep)); 620679e48cecSSean Christopherson access = kvm_mmu_page_get_access(huge_sp, spte_index(huge_sptep)); 6207ada51a9dSDavid Matlack 6208ada51a9dSDavid Matlack /* 6209ada51a9dSDavid Matlack * Note, huge page splitting always uses direct shadow pages, regardless 6210ada51a9dSDavid Matlack * of whether the huge page itself is mapped by a direct or indirect 6211ada51a9dSDavid Matlack * shadow page, since the huge page region itself is being directly 6212ada51a9dSDavid Matlack * mapped with smaller pages. 6213ada51a9dSDavid Matlack */ 6214ada51a9dSDavid Matlack role = kvm_mmu_child_role(huge_sptep, /*direct=*/true, access); 6215ada51a9dSDavid Matlack 6216ada51a9dSDavid Matlack /* Direct SPs do not require a shadowed_info_cache. */ 6217ada51a9dSDavid Matlack caches.page_header_cache = &kvm->arch.split_page_header_cache; 6218ada51a9dSDavid Matlack caches.shadow_page_cache = &kvm->arch.split_shadow_page_cache; 6219ada51a9dSDavid Matlack 6220ada51a9dSDavid Matlack /* Safe to pass NULL for vCPU since requesting a direct SP. */ 6221ada51a9dSDavid Matlack return __kvm_mmu_get_shadow_page(kvm, NULL, &caches, gfn, role); 6222ada51a9dSDavid Matlack } 6223ada51a9dSDavid Matlack 6224ada51a9dSDavid Matlack static void shadow_mmu_split_huge_page(struct kvm *kvm, 6225ada51a9dSDavid Matlack const struct kvm_memory_slot *slot, 6226ada51a9dSDavid Matlack u64 *huge_sptep) 6227ada51a9dSDavid Matlack 6228ada51a9dSDavid Matlack { 6229ada51a9dSDavid Matlack struct kvm_mmu_memory_cache *cache = &kvm->arch.split_desc_cache; 6230ada51a9dSDavid Matlack u64 huge_spte = READ_ONCE(*huge_sptep); 6231ada51a9dSDavid Matlack struct kvm_mmu_page *sp; 623203787394SPaolo Bonzini bool flush = false; 6233ada51a9dSDavid Matlack u64 *sptep, spte; 6234ada51a9dSDavid Matlack gfn_t gfn; 6235ada51a9dSDavid Matlack int index; 6236ada51a9dSDavid Matlack 6237ada51a9dSDavid Matlack sp = shadow_mmu_get_sp_for_split(kvm, huge_sptep); 6238ada51a9dSDavid Matlack 6239ada51a9dSDavid Matlack for (index = 0; index < SPTE_ENT_PER_PAGE; index++) { 6240ada51a9dSDavid Matlack sptep = &sp->spt[index]; 6241ada51a9dSDavid Matlack gfn = kvm_mmu_page_get_gfn(sp, index); 6242ada51a9dSDavid Matlack 6243ada51a9dSDavid Matlack /* 6244ada51a9dSDavid Matlack * The SP may already have populated SPTEs, e.g. if this huge 6245ada51a9dSDavid Matlack * page is aliased by multiple sptes with the same access 6246ada51a9dSDavid Matlack * permissions. These entries are guaranteed to map the same 6247ada51a9dSDavid Matlack * gfn-to-pfn translation since the SP is direct, so no need to 6248ada51a9dSDavid Matlack * modify them. 6249ada51a9dSDavid Matlack * 625003787394SPaolo Bonzini * However, if a given SPTE points to a lower level page table, 625103787394SPaolo Bonzini * that lower level page table may only be partially populated. 625203787394SPaolo Bonzini * Installing such SPTEs would effectively unmap a potion of the 625303787394SPaolo Bonzini * huge page. Unmapping guest memory always requires a TLB flush 625403787394SPaolo Bonzini * since a subsequent operation on the unmapped regions would 625503787394SPaolo Bonzini * fail to detect the need to flush. 6256ada51a9dSDavid Matlack */ 625703787394SPaolo Bonzini if (is_shadow_present_pte(*sptep)) { 625803787394SPaolo Bonzini flush |= !is_last_spte(*sptep, sp->role.level); 6259ada51a9dSDavid Matlack continue; 626003787394SPaolo Bonzini } 6261ada51a9dSDavid Matlack 6262ada51a9dSDavid Matlack spte = make_huge_page_split_spte(kvm, huge_spte, sp->role, index); 6263ada51a9dSDavid Matlack mmu_spte_set(sptep, spte); 6264ada51a9dSDavid Matlack __rmap_add(kvm, cache, slot, sptep, gfn, sp->role.access); 6265ada51a9dSDavid Matlack } 6266ada51a9dSDavid Matlack 626703787394SPaolo Bonzini __link_shadow_page(kvm, cache, huge_sptep, sp, flush); 6268ada51a9dSDavid Matlack } 6269ada51a9dSDavid Matlack 6270ada51a9dSDavid Matlack static int shadow_mmu_try_split_huge_page(struct kvm *kvm, 6271ada51a9dSDavid Matlack const struct kvm_memory_slot *slot, 6272ada51a9dSDavid Matlack u64 *huge_sptep) 6273ada51a9dSDavid Matlack { 6274ada51a9dSDavid Matlack struct kvm_mmu_page *huge_sp = sptep_to_sp(huge_sptep); 6275ada51a9dSDavid Matlack int level, r = 0; 6276ada51a9dSDavid Matlack gfn_t gfn; 6277ada51a9dSDavid Matlack u64 spte; 6278ada51a9dSDavid Matlack 6279ada51a9dSDavid Matlack /* Grab information for the tracepoint before dropping the MMU lock. */ 628079e48cecSSean Christopherson gfn = kvm_mmu_page_get_gfn(huge_sp, spte_index(huge_sptep)); 6281ada51a9dSDavid Matlack level = huge_sp->role.level; 6282ada51a9dSDavid Matlack spte = *huge_sptep; 6283ada51a9dSDavid Matlack 6284ada51a9dSDavid Matlack if (kvm_mmu_available_pages(kvm) <= KVM_MIN_FREE_MMU_PAGES) { 6285ada51a9dSDavid Matlack r = -ENOSPC; 6286ada51a9dSDavid Matlack goto out; 6287ada51a9dSDavid Matlack } 6288ada51a9dSDavid Matlack 6289ada51a9dSDavid Matlack if (need_topup_split_caches_or_resched(kvm)) { 6290ada51a9dSDavid Matlack write_unlock(&kvm->mmu_lock); 6291ada51a9dSDavid Matlack cond_resched(); 6292ada51a9dSDavid Matlack /* 6293ada51a9dSDavid Matlack * If the topup succeeds, return -EAGAIN to indicate that the 6294ada51a9dSDavid Matlack * rmap iterator should be restarted because the MMU lock was 6295ada51a9dSDavid Matlack * dropped. 6296ada51a9dSDavid Matlack */ 6297ada51a9dSDavid Matlack r = topup_split_caches(kvm) ?: -EAGAIN; 6298ada51a9dSDavid Matlack write_lock(&kvm->mmu_lock); 6299ada51a9dSDavid Matlack goto out; 6300ada51a9dSDavid Matlack } 6301ada51a9dSDavid Matlack 6302ada51a9dSDavid Matlack shadow_mmu_split_huge_page(kvm, slot, huge_sptep); 6303ada51a9dSDavid Matlack 6304ada51a9dSDavid Matlack out: 6305ada51a9dSDavid Matlack trace_kvm_mmu_split_huge_page(gfn, spte, level, r); 6306ada51a9dSDavid Matlack return r; 6307ada51a9dSDavid Matlack } 6308ada51a9dSDavid Matlack 6309ada51a9dSDavid Matlack static bool shadow_mmu_try_split_huge_pages(struct kvm *kvm, 6310ada51a9dSDavid Matlack struct kvm_rmap_head *rmap_head, 6311ada51a9dSDavid Matlack const struct kvm_memory_slot *slot) 6312ada51a9dSDavid Matlack { 6313ada51a9dSDavid Matlack struct rmap_iterator iter; 6314ada51a9dSDavid Matlack struct kvm_mmu_page *sp; 6315ada51a9dSDavid Matlack u64 *huge_sptep; 6316ada51a9dSDavid Matlack int r; 6317ada51a9dSDavid Matlack 6318ada51a9dSDavid Matlack restart: 6319ada51a9dSDavid Matlack for_each_rmap_spte(rmap_head, &iter, huge_sptep) { 6320ada51a9dSDavid Matlack sp = sptep_to_sp(huge_sptep); 6321ada51a9dSDavid Matlack 6322ada51a9dSDavid Matlack /* TDP MMU is enabled, so rmap only contains nested MMU SPs. */ 6323ada51a9dSDavid Matlack if (WARN_ON_ONCE(!sp->role.guest_mode)) 6324ada51a9dSDavid Matlack continue; 6325ada51a9dSDavid Matlack 6326ada51a9dSDavid Matlack /* The rmaps should never contain non-leaf SPTEs. */ 6327ada51a9dSDavid Matlack if (WARN_ON_ONCE(!is_large_pte(*huge_sptep))) 6328ada51a9dSDavid Matlack continue; 6329ada51a9dSDavid Matlack 6330ada51a9dSDavid Matlack /* SPs with level >PG_LEVEL_4K should never by unsync. */ 6331ada51a9dSDavid Matlack if (WARN_ON_ONCE(sp->unsync)) 6332ada51a9dSDavid Matlack continue; 6333ada51a9dSDavid Matlack 6334ada51a9dSDavid Matlack /* Don't bother splitting huge pages on invalid SPs. */ 6335ada51a9dSDavid Matlack if (sp->role.invalid) 6336ada51a9dSDavid Matlack continue; 6337ada51a9dSDavid Matlack 6338ada51a9dSDavid Matlack r = shadow_mmu_try_split_huge_page(kvm, slot, huge_sptep); 6339ada51a9dSDavid Matlack 6340ada51a9dSDavid Matlack /* 6341ada51a9dSDavid Matlack * The split succeeded or needs to be retried because the MMU 6342ada51a9dSDavid Matlack * lock was dropped. Either way, restart the iterator to get it 6343ada51a9dSDavid Matlack * back into a consistent state. 6344ada51a9dSDavid Matlack */ 6345ada51a9dSDavid Matlack if (!r || r == -EAGAIN) 6346ada51a9dSDavid Matlack goto restart; 6347ada51a9dSDavid Matlack 6348ada51a9dSDavid Matlack /* The split failed and shouldn't be retried (e.g. -ENOMEM). */ 6349ada51a9dSDavid Matlack break; 6350ada51a9dSDavid Matlack } 6351ada51a9dSDavid Matlack 6352ada51a9dSDavid Matlack return false; 6353ada51a9dSDavid Matlack } 6354ada51a9dSDavid Matlack 6355ada51a9dSDavid Matlack static void kvm_shadow_mmu_try_split_huge_pages(struct kvm *kvm, 6356ada51a9dSDavid Matlack const struct kvm_memory_slot *slot, 6357ada51a9dSDavid Matlack gfn_t start, gfn_t end, 6358ada51a9dSDavid Matlack int target_level) 6359ada51a9dSDavid Matlack { 6360ada51a9dSDavid Matlack int level; 6361ada51a9dSDavid Matlack 6362ada51a9dSDavid Matlack /* 6363ada51a9dSDavid Matlack * Split huge pages starting with KVM_MAX_HUGEPAGE_LEVEL and working 6364ada51a9dSDavid Matlack * down to the target level. This ensures pages are recursively split 6365ada51a9dSDavid Matlack * all the way to the target level. There's no need to split pages 6366ada51a9dSDavid Matlack * already at the target level. 6367ada51a9dSDavid Matlack */ 6368ada51a9dSDavid Matlack for (level = KVM_MAX_HUGEPAGE_LEVEL; level > target_level; level--) { 6369ada51a9dSDavid Matlack slot_handle_level_range(kvm, slot, shadow_mmu_try_split_huge_pages, 6370ada51a9dSDavid Matlack level, level, start, end - 1, true, false); 6371ada51a9dSDavid Matlack } 6372ada51a9dSDavid Matlack } 6373ada51a9dSDavid Matlack 6374cb00a70bSDavid Matlack /* Must be called with the mmu_lock held in write-mode. */ 6375cb00a70bSDavid Matlack void kvm_mmu_try_split_huge_pages(struct kvm *kvm, 6376cb00a70bSDavid Matlack const struct kvm_memory_slot *memslot, 6377cb00a70bSDavid Matlack u64 start, u64 end, 6378cb00a70bSDavid Matlack int target_level) 6379cb00a70bSDavid Matlack { 6380ada51a9dSDavid Matlack if (!is_tdp_mmu_enabled(kvm)) 6381ada51a9dSDavid Matlack return; 6382ada51a9dSDavid Matlack 6383ada51a9dSDavid Matlack if (kvm_memslots_have_rmaps(kvm)) 6384ada51a9dSDavid Matlack kvm_shadow_mmu_try_split_huge_pages(kvm, memslot, start, end, target_level); 6385ada51a9dSDavid Matlack 6386ada51a9dSDavid Matlack kvm_tdp_mmu_try_split_huge_pages(kvm, memslot, start, end, target_level, false); 6387cb00a70bSDavid Matlack 6388cb00a70bSDavid Matlack /* 6389cb00a70bSDavid Matlack * A TLB flush is unnecessary at this point for the same resons as in 6390cb00a70bSDavid Matlack * kvm_mmu_slot_try_split_huge_pages(). 6391cb00a70bSDavid Matlack */ 6392cb00a70bSDavid Matlack } 6393cb00a70bSDavid Matlack 6394a3fe5dbdSDavid Matlack void kvm_mmu_slot_try_split_huge_pages(struct kvm *kvm, 6395a3fe5dbdSDavid Matlack const struct kvm_memory_slot *memslot, 6396a3fe5dbdSDavid Matlack int target_level) 6397a3fe5dbdSDavid Matlack { 6398a3fe5dbdSDavid Matlack u64 start = memslot->base_gfn; 6399a3fe5dbdSDavid Matlack u64 end = start + memslot->npages; 6400a3fe5dbdSDavid Matlack 6401ada51a9dSDavid Matlack if (!is_tdp_mmu_enabled(kvm)) 6402ada51a9dSDavid Matlack return; 6403ada51a9dSDavid Matlack 6404ada51a9dSDavid Matlack if (kvm_memslots_have_rmaps(kvm)) { 6405ada51a9dSDavid Matlack write_lock(&kvm->mmu_lock); 6406ada51a9dSDavid Matlack kvm_shadow_mmu_try_split_huge_pages(kvm, memslot, start, end, target_level); 6407ada51a9dSDavid Matlack write_unlock(&kvm->mmu_lock); 6408ada51a9dSDavid Matlack } 6409ada51a9dSDavid Matlack 6410a3fe5dbdSDavid Matlack read_lock(&kvm->mmu_lock); 6411cb00a70bSDavid Matlack kvm_tdp_mmu_try_split_huge_pages(kvm, memslot, start, end, target_level, true); 6412a3fe5dbdSDavid Matlack read_unlock(&kvm->mmu_lock); 6413a3fe5dbdSDavid Matlack 6414a3fe5dbdSDavid Matlack /* 6415a3fe5dbdSDavid Matlack * No TLB flush is necessary here. KVM will flush TLBs after 6416a3fe5dbdSDavid Matlack * write-protecting and/or clearing dirty on the newly split SPTEs to 6417a3fe5dbdSDavid Matlack * ensure that guest writes are reflected in the dirty log before the 6418a3fe5dbdSDavid Matlack * ioctl to enable dirty logging on this memslot completes. Since the 6419a3fe5dbdSDavid Matlack * split SPTEs retain the write and dirty bits of the huge SPTE, it is 6420a3fe5dbdSDavid Matlack * safe for KVM to decide if a TLB flush is necessary based on the split 6421a3fe5dbdSDavid Matlack * SPTEs. 6422a3fe5dbdSDavid Matlack */ 6423a3fe5dbdSDavid Matlack } 6424a3fe5dbdSDavid Matlack 6425c50d8ae3SPaolo Bonzini static bool kvm_mmu_zap_collapsible_spte(struct kvm *kvm, 64260a234f5dSSean Christopherson struct kvm_rmap_head *rmap_head, 6427269e9552SHamza Mahfooz const struct kvm_memory_slot *slot) 6428c50d8ae3SPaolo Bonzini { 6429c50d8ae3SPaolo Bonzini u64 *sptep; 6430c50d8ae3SPaolo Bonzini struct rmap_iterator iter; 6431c50d8ae3SPaolo Bonzini int need_tlb_flush = 0; 6432c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 6433c50d8ae3SPaolo Bonzini 6434c50d8ae3SPaolo Bonzini restart: 6435c50d8ae3SPaolo Bonzini for_each_rmap_spte(rmap_head, &iter, sptep) { 643657354682SSean Christopherson sp = sptep_to_sp(sptep); 6437c50d8ae3SPaolo Bonzini 6438c50d8ae3SPaolo Bonzini /* 6439c50d8ae3SPaolo Bonzini * We cannot do huge page mapping for indirect shadow pages, 6440c50d8ae3SPaolo Bonzini * which are found on the last rmap (level = 1) when not using 6441c50d8ae3SPaolo Bonzini * tdp; such shadow pages are synced with the page table in 6442c50d8ae3SPaolo Bonzini * the guest, and the guest page table is using 4K page size 6443c50d8ae3SPaolo Bonzini * mapping if the indirect sp has level = 1. 6444c50d8ae3SPaolo Bonzini */ 64455d49f08cSSean Christopherson if (sp->role.direct && 64469eba50f8SSean Christopherson sp->role.level < kvm_mmu_max_mapping_level(kvm, slot, sp->gfn, 6447a8ac499bSSean Christopherson PG_LEVEL_NUM)) { 64489202aee8SSean Christopherson kvm_zap_one_rmap_spte(kvm, rmap_head, sptep); 6449c50d8ae3SPaolo Bonzini 6450c50d8ae3SPaolo Bonzini if (kvm_available_flush_tlb_with_range()) 6451c50d8ae3SPaolo Bonzini kvm_flush_remote_tlbs_with_address(kvm, sp->gfn, 6452c50d8ae3SPaolo Bonzini KVM_PAGES_PER_HPAGE(sp->role.level)); 6453c50d8ae3SPaolo Bonzini else 6454c50d8ae3SPaolo Bonzini need_tlb_flush = 1; 6455c50d8ae3SPaolo Bonzini 6456c50d8ae3SPaolo Bonzini goto restart; 6457c50d8ae3SPaolo Bonzini } 6458c50d8ae3SPaolo Bonzini } 6459c50d8ae3SPaolo Bonzini 6460c50d8ae3SPaolo Bonzini return need_tlb_flush; 6461c50d8ae3SPaolo Bonzini } 6462c50d8ae3SPaolo Bonzini 646320d49186SDavid Matlack static void kvm_rmap_zap_collapsible_sptes(struct kvm *kvm, 646420d49186SDavid Matlack const struct kvm_memory_slot *slot) 646520d49186SDavid Matlack { 646620d49186SDavid Matlack /* 646720d49186SDavid Matlack * Note, use KVM_MAX_HUGEPAGE_LEVEL - 1 since there's no need to zap 646820d49186SDavid Matlack * pages that are already mapped at the maximum hugepage level. 646920d49186SDavid Matlack */ 647020d49186SDavid Matlack if (slot_handle_level(kvm, slot, kvm_mmu_zap_collapsible_spte, 647120d49186SDavid Matlack PG_LEVEL_4K, KVM_MAX_HUGEPAGE_LEVEL - 1, true)) 647220d49186SDavid Matlack kvm_arch_flush_remote_tlbs_memslot(kvm, slot); 647320d49186SDavid Matlack } 647420d49186SDavid Matlack 6475c50d8ae3SPaolo Bonzini void kvm_mmu_zap_collapsible_sptes(struct kvm *kvm, 6476269e9552SHamza Mahfooz const struct kvm_memory_slot *slot) 6477c50d8ae3SPaolo Bonzini { 6478e2209710SBen Gardon if (kvm_memslots_have_rmaps(kvm)) { 6479531810caSBen Gardon write_lock(&kvm->mmu_lock); 648020d49186SDavid Matlack kvm_rmap_zap_collapsible_sptes(kvm, slot); 6481531810caSBen Gardon write_unlock(&kvm->mmu_lock); 6482e2209710SBen Gardon } 64832db6f772SBen Gardon 64842db6f772SBen Gardon if (is_tdp_mmu_enabled(kvm)) { 64852db6f772SBen Gardon read_lock(&kvm->mmu_lock); 64864b85c921SSean Christopherson kvm_tdp_mmu_zap_collapsible_sptes(kvm, slot); 64872db6f772SBen Gardon read_unlock(&kvm->mmu_lock); 64882db6f772SBen Gardon } 6489c50d8ae3SPaolo Bonzini } 6490c50d8ae3SPaolo Bonzini 6491b3594ffbSSean Christopherson void kvm_arch_flush_remote_tlbs_memslot(struct kvm *kvm, 64926c9dd6d2SPaolo Bonzini const struct kvm_memory_slot *memslot) 6493b3594ffbSSean Christopherson { 6494b3594ffbSSean Christopherson /* 64957f42aa76SSean Christopherson * All current use cases for flushing the TLBs for a specific memslot 6496302695a5SSean Christopherson * related to dirty logging, and many do the TLB flush out of mmu_lock. 64977f42aa76SSean Christopherson * The interaction between the various operations on memslot must be 64987f42aa76SSean Christopherson * serialized by slots_locks to ensure the TLB flush from one operation 64997f42aa76SSean Christopherson * is observed by any other operation on the same memslot. 6500b3594ffbSSean Christopherson */ 6501b3594ffbSSean Christopherson lockdep_assert_held(&kvm->slots_lock); 6502cec37648SSean Christopherson kvm_flush_remote_tlbs_with_address(kvm, memslot->base_gfn, 6503cec37648SSean Christopherson memslot->npages); 6504b3594ffbSSean Christopherson } 6505b3594ffbSSean Christopherson 6506c50d8ae3SPaolo Bonzini void kvm_mmu_slot_leaf_clear_dirty(struct kvm *kvm, 6507269e9552SHamza Mahfooz const struct kvm_memory_slot *memslot) 6508c50d8ae3SPaolo Bonzini { 6509e2209710SBen Gardon if (kvm_memslots_have_rmaps(kvm)) { 6510531810caSBen Gardon write_lock(&kvm->mmu_lock); 6511610265eaSDavid Matlack /* 6512610265eaSDavid Matlack * Clear dirty bits only on 4k SPTEs since the legacy MMU only 6513610265eaSDavid Matlack * support dirty logging at a 4k granularity. 6514610265eaSDavid Matlack */ 6515b64d740eSJunaid Shahid slot_handle_level_4k(kvm, memslot, __rmap_clear_dirty, false); 6516531810caSBen Gardon write_unlock(&kvm->mmu_lock); 6517e2209710SBen Gardon } 6518c50d8ae3SPaolo Bonzini 651924ae4cfaSBen Gardon if (is_tdp_mmu_enabled(kvm)) { 652024ae4cfaSBen Gardon read_lock(&kvm->mmu_lock); 6521b64d740eSJunaid Shahid kvm_tdp_mmu_clear_dirty_slot(kvm, memslot); 652224ae4cfaSBen Gardon read_unlock(&kvm->mmu_lock); 652324ae4cfaSBen Gardon } 652424ae4cfaSBen Gardon 6525c50d8ae3SPaolo Bonzini /* 6526b64d740eSJunaid Shahid * The caller will flush the TLBs after this function returns. 6527b64d740eSJunaid Shahid * 6528c50d8ae3SPaolo Bonzini * It's also safe to flush TLBs out of mmu lock here as currently this 6529c50d8ae3SPaolo Bonzini * function is only used for dirty logging, in which case flushing TLB 6530c50d8ae3SPaolo Bonzini * out of mmu lock also guarantees no dirty pages will be lost in 6531c50d8ae3SPaolo Bonzini * dirty_bitmap. 6532c50d8ae3SPaolo Bonzini */ 6533c50d8ae3SPaolo Bonzini } 6534c50d8ae3SPaolo Bonzini 6535c50d8ae3SPaolo Bonzini void kvm_mmu_zap_all(struct kvm *kvm) 6536c50d8ae3SPaolo Bonzini { 6537c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp, *node; 6538c50d8ae3SPaolo Bonzini LIST_HEAD(invalid_list); 6539c50d8ae3SPaolo Bonzini int ign; 6540c50d8ae3SPaolo Bonzini 6541531810caSBen Gardon write_lock(&kvm->mmu_lock); 6542c50d8ae3SPaolo Bonzini restart: 6543c50d8ae3SPaolo Bonzini list_for_each_entry_safe(sp, node, &kvm->arch.active_mmu_pages, link) { 6544f95eec9bSSean Christopherson if (WARN_ON(sp->role.invalid)) 6545c50d8ae3SPaolo Bonzini continue; 6546c50d8ae3SPaolo Bonzini if (__kvm_mmu_prepare_zap_page(kvm, sp, &invalid_list, &ign)) 6547c50d8ae3SPaolo Bonzini goto restart; 6548531810caSBen Gardon if (cond_resched_rwlock_write(&kvm->mmu_lock)) 6549c50d8ae3SPaolo Bonzini goto restart; 6550c50d8ae3SPaolo Bonzini } 6551c50d8ae3SPaolo Bonzini 6552c50d8ae3SPaolo Bonzini kvm_mmu_commit_zap_page(kvm, &invalid_list); 6553faaf05b0SBen Gardon 6554897218ffSPaolo Bonzini if (is_tdp_mmu_enabled(kvm)) 6555faaf05b0SBen Gardon kvm_tdp_mmu_zap_all(kvm); 6556faaf05b0SBen Gardon 6557531810caSBen Gardon write_unlock(&kvm->mmu_lock); 6558c50d8ae3SPaolo Bonzini } 6559c50d8ae3SPaolo Bonzini 6560c50d8ae3SPaolo Bonzini void kvm_mmu_invalidate_mmio_sptes(struct kvm *kvm, u64 gen) 6561c50d8ae3SPaolo Bonzini { 6562c50d8ae3SPaolo Bonzini WARN_ON(gen & KVM_MEMSLOT_GEN_UPDATE_IN_PROGRESS); 6563c50d8ae3SPaolo Bonzini 6564c50d8ae3SPaolo Bonzini gen &= MMIO_SPTE_GEN_MASK; 6565c50d8ae3SPaolo Bonzini 6566c50d8ae3SPaolo Bonzini /* 6567c50d8ae3SPaolo Bonzini * Generation numbers are incremented in multiples of the number of 6568c50d8ae3SPaolo Bonzini * address spaces in order to provide unique generations across all 6569c50d8ae3SPaolo Bonzini * address spaces. Strip what is effectively the address space 6570c50d8ae3SPaolo Bonzini * modifier prior to checking for a wrap of the MMIO generation so 6571c50d8ae3SPaolo Bonzini * that a wrap in any address space is detected. 6572c50d8ae3SPaolo Bonzini */ 6573c50d8ae3SPaolo Bonzini gen &= ~((u64)KVM_ADDRESS_SPACE_NUM - 1); 6574c50d8ae3SPaolo Bonzini 6575c50d8ae3SPaolo Bonzini /* 6576c50d8ae3SPaolo Bonzini * The very rare case: if the MMIO generation number has wrapped, 6577c50d8ae3SPaolo Bonzini * zap all shadow pages. 6578c50d8ae3SPaolo Bonzini */ 6579c50d8ae3SPaolo Bonzini if (unlikely(gen == 0)) { 6580c50d8ae3SPaolo Bonzini kvm_debug_ratelimited("kvm: zapping shadow pages for mmio generation wraparound\n"); 6581c50d8ae3SPaolo Bonzini kvm_mmu_zap_all_fast(kvm); 6582c50d8ae3SPaolo Bonzini } 6583c50d8ae3SPaolo Bonzini } 6584c50d8ae3SPaolo Bonzini 6585c50d8ae3SPaolo Bonzini static unsigned long 6586c50d8ae3SPaolo Bonzini mmu_shrink_scan(struct shrinker *shrink, struct shrink_control *sc) 6587c50d8ae3SPaolo Bonzini { 6588c50d8ae3SPaolo Bonzini struct kvm *kvm; 6589c50d8ae3SPaolo Bonzini int nr_to_scan = sc->nr_to_scan; 6590c50d8ae3SPaolo Bonzini unsigned long freed = 0; 6591c50d8ae3SPaolo Bonzini 6592c50d8ae3SPaolo Bonzini mutex_lock(&kvm_lock); 6593c50d8ae3SPaolo Bonzini 6594c50d8ae3SPaolo Bonzini list_for_each_entry(kvm, &vm_list, vm_list) { 6595c50d8ae3SPaolo Bonzini int idx; 6596c50d8ae3SPaolo Bonzini LIST_HEAD(invalid_list); 6597c50d8ae3SPaolo Bonzini 6598c50d8ae3SPaolo Bonzini /* 6599c50d8ae3SPaolo Bonzini * Never scan more than sc->nr_to_scan VM instances. 6600c50d8ae3SPaolo Bonzini * Will not hit this condition practically since we do not try 6601c50d8ae3SPaolo Bonzini * to shrink more than one VM and it is very unlikely to see 6602c50d8ae3SPaolo Bonzini * !n_used_mmu_pages so many times. 6603c50d8ae3SPaolo Bonzini */ 6604c50d8ae3SPaolo Bonzini if (!nr_to_scan--) 6605c50d8ae3SPaolo Bonzini break; 6606c50d8ae3SPaolo Bonzini /* 6607c50d8ae3SPaolo Bonzini * n_used_mmu_pages is accessed without holding kvm->mmu_lock 6608c50d8ae3SPaolo Bonzini * here. We may skip a VM instance errorneosly, but we do not 6609c50d8ae3SPaolo Bonzini * want to shrink a VM that only started to populate its MMU 6610c50d8ae3SPaolo Bonzini * anyway. 6611c50d8ae3SPaolo Bonzini */ 6612c50d8ae3SPaolo Bonzini if (!kvm->arch.n_used_mmu_pages && 6613c50d8ae3SPaolo Bonzini !kvm_has_zapped_obsolete_pages(kvm)) 6614c50d8ae3SPaolo Bonzini continue; 6615c50d8ae3SPaolo Bonzini 6616c50d8ae3SPaolo Bonzini idx = srcu_read_lock(&kvm->srcu); 6617531810caSBen Gardon write_lock(&kvm->mmu_lock); 6618c50d8ae3SPaolo Bonzini 6619c50d8ae3SPaolo Bonzini if (kvm_has_zapped_obsolete_pages(kvm)) { 6620c50d8ae3SPaolo Bonzini kvm_mmu_commit_zap_page(kvm, 6621c50d8ae3SPaolo Bonzini &kvm->arch.zapped_obsolete_pages); 6622c50d8ae3SPaolo Bonzini goto unlock; 6623c50d8ae3SPaolo Bonzini } 6624c50d8ae3SPaolo Bonzini 6625ebdb292dSSean Christopherson freed = kvm_mmu_zap_oldest_mmu_pages(kvm, sc->nr_to_scan); 6626c50d8ae3SPaolo Bonzini 6627c50d8ae3SPaolo Bonzini unlock: 6628531810caSBen Gardon write_unlock(&kvm->mmu_lock); 6629c50d8ae3SPaolo Bonzini srcu_read_unlock(&kvm->srcu, idx); 6630c50d8ae3SPaolo Bonzini 6631c50d8ae3SPaolo Bonzini /* 6632c50d8ae3SPaolo Bonzini * unfair on small ones 6633c50d8ae3SPaolo Bonzini * per-vm shrinkers cry out 6634c50d8ae3SPaolo Bonzini * sadness comes quickly 6635c50d8ae3SPaolo Bonzini */ 6636c50d8ae3SPaolo Bonzini list_move_tail(&kvm->vm_list, &vm_list); 6637c50d8ae3SPaolo Bonzini break; 6638c50d8ae3SPaolo Bonzini } 6639c50d8ae3SPaolo Bonzini 6640c50d8ae3SPaolo Bonzini mutex_unlock(&kvm_lock); 6641c50d8ae3SPaolo Bonzini return freed; 6642c50d8ae3SPaolo Bonzini } 6643c50d8ae3SPaolo Bonzini 6644c50d8ae3SPaolo Bonzini static unsigned long 6645c50d8ae3SPaolo Bonzini mmu_shrink_count(struct shrinker *shrink, struct shrink_control *sc) 6646c50d8ae3SPaolo Bonzini { 6647c50d8ae3SPaolo Bonzini return percpu_counter_read_positive(&kvm_total_used_mmu_pages); 6648c50d8ae3SPaolo Bonzini } 6649c50d8ae3SPaolo Bonzini 6650c50d8ae3SPaolo Bonzini static struct shrinker mmu_shrinker = { 6651c50d8ae3SPaolo Bonzini .count_objects = mmu_shrink_count, 6652c50d8ae3SPaolo Bonzini .scan_objects = mmu_shrink_scan, 6653c50d8ae3SPaolo Bonzini .seeks = DEFAULT_SEEKS * 10, 6654c50d8ae3SPaolo Bonzini }; 6655c50d8ae3SPaolo Bonzini 6656c50d8ae3SPaolo Bonzini static void mmu_destroy_caches(void) 6657c50d8ae3SPaolo Bonzini { 6658c50d8ae3SPaolo Bonzini kmem_cache_destroy(pte_list_desc_cache); 6659c50d8ae3SPaolo Bonzini kmem_cache_destroy(mmu_page_header_cache); 6660c50d8ae3SPaolo Bonzini } 6661c50d8ae3SPaolo Bonzini 6662c50d8ae3SPaolo Bonzini static bool get_nx_auto_mode(void) 6663c50d8ae3SPaolo Bonzini { 6664c50d8ae3SPaolo Bonzini /* Return true when CPU has the bug, and mitigations are ON */ 6665c50d8ae3SPaolo Bonzini return boot_cpu_has_bug(X86_BUG_ITLB_MULTIHIT) && !cpu_mitigations_off(); 6666c50d8ae3SPaolo Bonzini } 6667c50d8ae3SPaolo Bonzini 6668c50d8ae3SPaolo Bonzini static void __set_nx_huge_pages(bool val) 6669c50d8ae3SPaolo Bonzini { 6670c50d8ae3SPaolo Bonzini nx_huge_pages = itlb_multihit_kvm_mitigation = val; 6671c50d8ae3SPaolo Bonzini } 6672c50d8ae3SPaolo Bonzini 6673c50d8ae3SPaolo Bonzini static int set_nx_huge_pages(const char *val, const struct kernel_param *kp) 6674c50d8ae3SPaolo Bonzini { 6675c50d8ae3SPaolo Bonzini bool old_val = nx_huge_pages; 6676c50d8ae3SPaolo Bonzini bool new_val; 6677c50d8ae3SPaolo Bonzini 6678c50d8ae3SPaolo Bonzini /* In "auto" mode deploy workaround only if CPU has the bug. */ 6679c50d8ae3SPaolo Bonzini if (sysfs_streq(val, "off")) 6680c50d8ae3SPaolo Bonzini new_val = 0; 6681c50d8ae3SPaolo Bonzini else if (sysfs_streq(val, "force")) 6682c50d8ae3SPaolo Bonzini new_val = 1; 6683c50d8ae3SPaolo Bonzini else if (sysfs_streq(val, "auto")) 6684c50d8ae3SPaolo Bonzini new_val = get_nx_auto_mode(); 6685c50d8ae3SPaolo Bonzini else if (strtobool(val, &new_val) < 0) 6686c50d8ae3SPaolo Bonzini return -EINVAL; 6687c50d8ae3SPaolo Bonzini 6688c50d8ae3SPaolo Bonzini __set_nx_huge_pages(new_val); 6689c50d8ae3SPaolo Bonzini 6690c50d8ae3SPaolo Bonzini if (new_val != old_val) { 6691c50d8ae3SPaolo Bonzini struct kvm *kvm; 6692c50d8ae3SPaolo Bonzini 6693c50d8ae3SPaolo Bonzini mutex_lock(&kvm_lock); 6694c50d8ae3SPaolo Bonzini 6695c50d8ae3SPaolo Bonzini list_for_each_entry(kvm, &vm_list, vm_list) { 6696c50d8ae3SPaolo Bonzini mutex_lock(&kvm->slots_lock); 6697c50d8ae3SPaolo Bonzini kvm_mmu_zap_all_fast(kvm); 6698c50d8ae3SPaolo Bonzini mutex_unlock(&kvm->slots_lock); 6699c50d8ae3SPaolo Bonzini 670055c510e2SSean Christopherson wake_up_process(kvm->arch.nx_huge_page_recovery_thread); 6701c50d8ae3SPaolo Bonzini } 6702c50d8ae3SPaolo Bonzini mutex_unlock(&kvm_lock); 6703c50d8ae3SPaolo Bonzini } 6704c50d8ae3SPaolo Bonzini 6705c50d8ae3SPaolo Bonzini return 0; 6706c50d8ae3SPaolo Bonzini } 6707c50d8ae3SPaolo Bonzini 67081d0e8480SSean Christopherson /* 67091d0e8480SSean Christopherson * nx_huge_pages needs to be resolved to true/false when kvm.ko is loaded, as 67101d0e8480SSean Christopherson * its default value of -1 is technically undefined behavior for a boolean. 6711c3e0c8c2SSean Christopherson * Forward the module init call to SPTE code so that it too can handle module 6712c3e0c8c2SSean Christopherson * params that need to be resolved/snapshot. 67131d0e8480SSean Christopherson */ 6714982bae43SSean Christopherson void __init kvm_mmu_x86_module_init(void) 6715c50d8ae3SPaolo Bonzini { 6716c50d8ae3SPaolo Bonzini if (nx_huge_pages == -1) 6717c50d8ae3SPaolo Bonzini __set_nx_huge_pages(get_nx_auto_mode()); 6718c3e0c8c2SSean Christopherson 6719c3e0c8c2SSean Christopherson kvm_mmu_spte_module_init(); 67201d0e8480SSean Christopherson } 67211d0e8480SSean Christopherson 67221d0e8480SSean Christopherson /* 67231d0e8480SSean Christopherson * The bulk of the MMU initialization is deferred until the vendor module is 67241d0e8480SSean Christopherson * loaded as many of the masks/values may be modified by VMX or SVM, i.e. need 67251d0e8480SSean Christopherson * to be reset when a potentially different vendor module is loaded. 67261d0e8480SSean Christopherson */ 67271d0e8480SSean Christopherson int kvm_mmu_vendor_module_init(void) 67281d0e8480SSean Christopherson { 67291d0e8480SSean Christopherson int ret = -ENOMEM; 6730c50d8ae3SPaolo Bonzini 6731c50d8ae3SPaolo Bonzini /* 6732c50d8ae3SPaolo Bonzini * MMU roles use union aliasing which is, generally speaking, an 6733c50d8ae3SPaolo Bonzini * undefined behavior. However, we supposedly know how compilers behave 6734c50d8ae3SPaolo Bonzini * and the current status quo is unlikely to change. Guardians below are 6735c50d8ae3SPaolo Bonzini * supposed to let us know if the assumption becomes false. 6736c50d8ae3SPaolo Bonzini */ 6737c50d8ae3SPaolo Bonzini BUILD_BUG_ON(sizeof(union kvm_mmu_page_role) != sizeof(u32)); 6738c50d8ae3SPaolo Bonzini BUILD_BUG_ON(sizeof(union kvm_mmu_extended_role) != sizeof(u32)); 67397a7ae829SPaolo Bonzini BUILD_BUG_ON(sizeof(union kvm_cpu_role) != sizeof(u64)); 6740c50d8ae3SPaolo Bonzini 6741c50d8ae3SPaolo Bonzini kvm_mmu_reset_all_pte_masks(); 6742c50d8ae3SPaolo Bonzini 6743c50d8ae3SPaolo Bonzini pte_list_desc_cache = kmem_cache_create("pte_list_desc", 6744c50d8ae3SPaolo Bonzini sizeof(struct pte_list_desc), 6745c50d8ae3SPaolo Bonzini 0, SLAB_ACCOUNT, NULL); 6746c50d8ae3SPaolo Bonzini if (!pte_list_desc_cache) 6747c50d8ae3SPaolo Bonzini goto out; 6748c50d8ae3SPaolo Bonzini 6749c50d8ae3SPaolo Bonzini mmu_page_header_cache = kmem_cache_create("kvm_mmu_page_header", 6750c50d8ae3SPaolo Bonzini sizeof(struct kvm_mmu_page), 6751c50d8ae3SPaolo Bonzini 0, SLAB_ACCOUNT, NULL); 6752c50d8ae3SPaolo Bonzini if (!mmu_page_header_cache) 6753c50d8ae3SPaolo Bonzini goto out; 6754c50d8ae3SPaolo Bonzini 6755c50d8ae3SPaolo Bonzini if (percpu_counter_init(&kvm_total_used_mmu_pages, 0, GFP_KERNEL)) 6756c50d8ae3SPaolo Bonzini goto out; 6757c50d8ae3SPaolo Bonzini 6758e33c267aSRoman Gushchin ret = register_shrinker(&mmu_shrinker, "x86-mmu"); 6759c50d8ae3SPaolo Bonzini if (ret) 6760d7c9bfb9SMiaohe Lin goto out_shrinker; 6761c50d8ae3SPaolo Bonzini 6762c50d8ae3SPaolo Bonzini return 0; 6763c50d8ae3SPaolo Bonzini 6764d7c9bfb9SMiaohe Lin out_shrinker: 6765d7c9bfb9SMiaohe Lin percpu_counter_destroy(&kvm_total_used_mmu_pages); 6766c50d8ae3SPaolo Bonzini out: 6767c50d8ae3SPaolo Bonzini mmu_destroy_caches(); 6768c50d8ae3SPaolo Bonzini return ret; 6769c50d8ae3SPaolo Bonzini } 6770c50d8ae3SPaolo Bonzini 6771c50d8ae3SPaolo Bonzini void kvm_mmu_destroy(struct kvm_vcpu *vcpu) 6772c50d8ae3SPaolo Bonzini { 6773c50d8ae3SPaolo Bonzini kvm_mmu_unload(vcpu); 6774c50d8ae3SPaolo Bonzini free_mmu_pages(&vcpu->arch.root_mmu); 6775c50d8ae3SPaolo Bonzini free_mmu_pages(&vcpu->arch.guest_mmu); 6776c50d8ae3SPaolo Bonzini mmu_free_memory_caches(vcpu); 6777c50d8ae3SPaolo Bonzini } 6778c50d8ae3SPaolo Bonzini 67791d0e8480SSean Christopherson void kvm_mmu_vendor_module_exit(void) 6780c50d8ae3SPaolo Bonzini { 6781c50d8ae3SPaolo Bonzini mmu_destroy_caches(); 6782c50d8ae3SPaolo Bonzini percpu_counter_destroy(&kvm_total_used_mmu_pages); 6783c50d8ae3SPaolo Bonzini unregister_shrinker(&mmu_shrinker); 6784c50d8ae3SPaolo Bonzini } 6785c50d8ae3SPaolo Bonzini 6786f47491d7SSean Christopherson /* 6787f47491d7SSean Christopherson * Calculate the effective recovery period, accounting for '0' meaning "let KVM 6788f47491d7SSean Christopherson * select a halving time of 1 hour". Returns true if recovery is enabled. 6789f47491d7SSean Christopherson */ 6790f47491d7SSean Christopherson static bool calc_nx_huge_pages_recovery_period(uint *period) 6791f47491d7SSean Christopherson { 6792f47491d7SSean Christopherson /* 6793f47491d7SSean Christopherson * Use READ_ONCE to get the params, this may be called outside of the 6794f47491d7SSean Christopherson * param setters, e.g. by the kthread to compute its next timeout. 6795f47491d7SSean Christopherson */ 6796f47491d7SSean Christopherson bool enabled = READ_ONCE(nx_huge_pages); 6797f47491d7SSean Christopherson uint ratio = READ_ONCE(nx_huge_pages_recovery_ratio); 6798f47491d7SSean Christopherson 6799f47491d7SSean Christopherson if (!enabled || !ratio) 6800f47491d7SSean Christopherson return false; 6801f47491d7SSean Christopherson 6802f47491d7SSean Christopherson *period = READ_ONCE(nx_huge_pages_recovery_period_ms); 6803f47491d7SSean Christopherson if (!*period) { 6804f47491d7SSean Christopherson /* Make sure the period is not less than one second. */ 6805f47491d7SSean Christopherson ratio = min(ratio, 3600u); 6806f47491d7SSean Christopherson *period = 60 * 60 * 1000 / ratio; 6807f47491d7SSean Christopherson } 6808f47491d7SSean Christopherson return true; 6809f47491d7SSean Christopherson } 6810f47491d7SSean Christopherson 68114dfe4f40SJunaid Shahid static int set_nx_huge_pages_recovery_param(const char *val, const struct kernel_param *kp) 6812c50d8ae3SPaolo Bonzini { 68134dfe4f40SJunaid Shahid bool was_recovery_enabled, is_recovery_enabled; 68144dfe4f40SJunaid Shahid uint old_period, new_period; 6815c50d8ae3SPaolo Bonzini int err; 6816c50d8ae3SPaolo Bonzini 6817f47491d7SSean Christopherson was_recovery_enabled = calc_nx_huge_pages_recovery_period(&old_period); 68184dfe4f40SJunaid Shahid 6819c50d8ae3SPaolo Bonzini err = param_set_uint(val, kp); 6820c50d8ae3SPaolo Bonzini if (err) 6821c50d8ae3SPaolo Bonzini return err; 6822c50d8ae3SPaolo Bonzini 6823f47491d7SSean Christopherson is_recovery_enabled = calc_nx_huge_pages_recovery_period(&new_period); 68244dfe4f40SJunaid Shahid 6825f47491d7SSean Christopherson if (is_recovery_enabled && 68264dfe4f40SJunaid Shahid (!was_recovery_enabled || old_period > new_period)) { 6827c50d8ae3SPaolo Bonzini struct kvm *kvm; 6828c50d8ae3SPaolo Bonzini 6829c50d8ae3SPaolo Bonzini mutex_lock(&kvm_lock); 6830c50d8ae3SPaolo Bonzini 6831c50d8ae3SPaolo Bonzini list_for_each_entry(kvm, &vm_list, vm_list) 683255c510e2SSean Christopherson wake_up_process(kvm->arch.nx_huge_page_recovery_thread); 6833c50d8ae3SPaolo Bonzini 6834c50d8ae3SPaolo Bonzini mutex_unlock(&kvm_lock); 6835c50d8ae3SPaolo Bonzini } 6836c50d8ae3SPaolo Bonzini 6837c50d8ae3SPaolo Bonzini return err; 6838c50d8ae3SPaolo Bonzini } 6839c50d8ae3SPaolo Bonzini 684055c510e2SSean Christopherson static void kvm_recover_nx_huge_pages(struct kvm *kvm) 6841c50d8ae3SPaolo Bonzini { 6842ade74e14SSean Christopherson unsigned long nx_lpage_splits = kvm->stat.nx_lpage_splits; 6843c50d8ae3SPaolo Bonzini int rcu_idx; 6844c50d8ae3SPaolo Bonzini struct kvm_mmu_page *sp; 6845c50d8ae3SPaolo Bonzini unsigned int ratio; 6846c50d8ae3SPaolo Bonzini LIST_HEAD(invalid_list); 6847048f4980SSean Christopherson bool flush = false; 6848c50d8ae3SPaolo Bonzini ulong to_zap; 6849c50d8ae3SPaolo Bonzini 6850c50d8ae3SPaolo Bonzini rcu_idx = srcu_read_lock(&kvm->srcu); 6851531810caSBen Gardon write_lock(&kvm->mmu_lock); 6852c50d8ae3SPaolo Bonzini 6853bb95dfb9SSean Christopherson /* 6854bb95dfb9SSean Christopherson * Zapping TDP MMU shadow pages, including the remote TLB flush, must 6855bb95dfb9SSean Christopherson * be done under RCU protection, because the pages are freed via RCU 6856bb95dfb9SSean Christopherson * callback. 6857bb95dfb9SSean Christopherson */ 6858bb95dfb9SSean Christopherson rcu_read_lock(); 6859bb95dfb9SSean Christopherson 6860c50d8ae3SPaolo Bonzini ratio = READ_ONCE(nx_huge_pages_recovery_ratio); 6861ade74e14SSean Christopherson to_zap = ratio ? DIV_ROUND_UP(nx_lpage_splits, ratio) : 0; 68627d919c7aSSean Christopherson for ( ; to_zap; --to_zap) { 686355c510e2SSean Christopherson if (list_empty(&kvm->arch.possible_nx_huge_pages)) 68647d919c7aSSean Christopherson break; 68657d919c7aSSean Christopherson 6866c50d8ae3SPaolo Bonzini /* 6867c50d8ae3SPaolo Bonzini * We use a separate list instead of just using active_mmu_pages 686855c510e2SSean Christopherson * because the number of shadow pages that be replaced with an 686955c510e2SSean Christopherson * NX huge page is expected to be relatively small compared to 687055c510e2SSean Christopherson * the total number of shadow pages. And because the TDP MMU 687155c510e2SSean Christopherson * doesn't use active_mmu_pages. 6872c50d8ae3SPaolo Bonzini */ 687355c510e2SSean Christopherson sp = list_first_entry(&kvm->arch.possible_nx_huge_pages, 6874c50d8ae3SPaolo Bonzini struct kvm_mmu_page, 687555c510e2SSean Christopherson possible_nx_huge_page_link); 687655c510e2SSean Christopherson WARN_ON_ONCE(!sp->nx_huge_page_disallowed); 6877897218ffSPaolo Bonzini if (is_tdp_mmu_page(sp)) { 6878315f02c6SPaolo Bonzini flush |= kvm_tdp_mmu_zap_sp(kvm, sp); 68798d1a182eSBen Gardon } else { 6880c50d8ae3SPaolo Bonzini kvm_mmu_prepare_zap_page(kvm, sp, &invalid_list); 688155c510e2SSean Christopherson WARN_ON_ONCE(sp->nx_huge_page_disallowed); 688229cf0f50SBen Gardon } 6883c50d8ae3SPaolo Bonzini 6884531810caSBen Gardon if (need_resched() || rwlock_needbreak(&kvm->mmu_lock)) { 6885048f4980SSean Christopherson kvm_mmu_remote_flush_or_zap(kvm, &invalid_list, flush); 6886bb95dfb9SSean Christopherson rcu_read_unlock(); 6887bb95dfb9SSean Christopherson 6888531810caSBen Gardon cond_resched_rwlock_write(&kvm->mmu_lock); 6889048f4980SSean Christopherson flush = false; 6890bb95dfb9SSean Christopherson 6891bb95dfb9SSean Christopherson rcu_read_lock(); 6892c50d8ae3SPaolo Bonzini } 6893c50d8ae3SPaolo Bonzini } 6894048f4980SSean Christopherson kvm_mmu_remote_flush_or_zap(kvm, &invalid_list, flush); 6895c50d8ae3SPaolo Bonzini 6896bb95dfb9SSean Christopherson rcu_read_unlock(); 6897bb95dfb9SSean Christopherson 6898531810caSBen Gardon write_unlock(&kvm->mmu_lock); 6899c50d8ae3SPaolo Bonzini srcu_read_unlock(&kvm->srcu, rcu_idx); 6900c50d8ae3SPaolo Bonzini } 6901c50d8ae3SPaolo Bonzini 690255c510e2SSean Christopherson static long get_nx_huge_page_recovery_timeout(u64 start_time) 6903c50d8ae3SPaolo Bonzini { 6904f47491d7SSean Christopherson bool enabled; 6905f47491d7SSean Christopherson uint period; 69064dfe4f40SJunaid Shahid 6907f47491d7SSean Christopherson enabled = calc_nx_huge_pages_recovery_period(&period); 69084dfe4f40SJunaid Shahid 6909f47491d7SSean Christopherson return enabled ? start_time + msecs_to_jiffies(period) - get_jiffies_64() 6910c50d8ae3SPaolo Bonzini : MAX_SCHEDULE_TIMEOUT; 6911c50d8ae3SPaolo Bonzini } 6912c50d8ae3SPaolo Bonzini 691355c510e2SSean Christopherson static int kvm_nx_huge_page_recovery_worker(struct kvm *kvm, uintptr_t data) 6914c50d8ae3SPaolo Bonzini { 6915c50d8ae3SPaolo Bonzini u64 start_time; 6916c50d8ae3SPaolo Bonzini long remaining_time; 6917c50d8ae3SPaolo Bonzini 6918c50d8ae3SPaolo Bonzini while (true) { 6919c50d8ae3SPaolo Bonzini start_time = get_jiffies_64(); 692055c510e2SSean Christopherson remaining_time = get_nx_huge_page_recovery_timeout(start_time); 6921c50d8ae3SPaolo Bonzini 6922c50d8ae3SPaolo Bonzini set_current_state(TASK_INTERRUPTIBLE); 6923c50d8ae3SPaolo Bonzini while (!kthread_should_stop() && remaining_time > 0) { 6924c50d8ae3SPaolo Bonzini schedule_timeout(remaining_time); 692555c510e2SSean Christopherson remaining_time = get_nx_huge_page_recovery_timeout(start_time); 6926c50d8ae3SPaolo Bonzini set_current_state(TASK_INTERRUPTIBLE); 6927c50d8ae3SPaolo Bonzini } 6928c50d8ae3SPaolo Bonzini 6929c50d8ae3SPaolo Bonzini set_current_state(TASK_RUNNING); 6930c50d8ae3SPaolo Bonzini 6931c50d8ae3SPaolo Bonzini if (kthread_should_stop()) 6932c50d8ae3SPaolo Bonzini return 0; 6933c50d8ae3SPaolo Bonzini 693455c510e2SSean Christopherson kvm_recover_nx_huge_pages(kvm); 6935c50d8ae3SPaolo Bonzini } 6936c50d8ae3SPaolo Bonzini } 6937c50d8ae3SPaolo Bonzini 6938c50d8ae3SPaolo Bonzini int kvm_mmu_post_init_vm(struct kvm *kvm) 6939c50d8ae3SPaolo Bonzini { 6940c50d8ae3SPaolo Bonzini int err; 6941c50d8ae3SPaolo Bonzini 694255c510e2SSean Christopherson err = kvm_vm_create_worker_thread(kvm, kvm_nx_huge_page_recovery_worker, 0, 6943c50d8ae3SPaolo Bonzini "kvm-nx-lpage-recovery", 694455c510e2SSean Christopherson &kvm->arch.nx_huge_page_recovery_thread); 6945c50d8ae3SPaolo Bonzini if (!err) 694655c510e2SSean Christopherson kthread_unpark(kvm->arch.nx_huge_page_recovery_thread); 6947c50d8ae3SPaolo Bonzini 6948c50d8ae3SPaolo Bonzini return err; 6949c50d8ae3SPaolo Bonzini } 6950c50d8ae3SPaolo Bonzini 6951c50d8ae3SPaolo Bonzini void kvm_mmu_pre_destroy_vm(struct kvm *kvm) 6952c50d8ae3SPaolo Bonzini { 695355c510e2SSean Christopherson if (kvm->arch.nx_huge_page_recovery_thread) 695455c510e2SSean Christopherson kthread_stop(kvm->arch.nx_huge_page_recovery_thread); 6955c50d8ae3SPaolo Bonzini } 6956