]> git.proxmox.com Git - mirror_ubuntu-jammy-kernel.git/blame - arch/x86/kvm/mmu.c
KVM: Remove SMAP bit from CR4_RESERVED_BITS
[mirror_ubuntu-jammy-kernel.git] / arch / x86 / kvm / mmu.c
CommitLineData
6aa8b732
AK
1/*
2 * Kernel-based Virtual Machine driver for Linux
3 *
4 * This module enables machines with Intel VT-x extensions to run virtual
5 * machines without emulation or binary translation.
6 *
7 * MMU support
8 *
9 * Copyright (C) 2006 Qumranet, Inc.
9611c187 10 * Copyright 2010 Red Hat, Inc. and/or its affiliates.
6aa8b732
AK
11 *
12 * Authors:
13 * Yaniv Kamay <yaniv@qumranet.com>
14 * Avi Kivity <avi@qumranet.com>
15 *
16 * This work is licensed under the terms of the GNU GPL, version 2. See
17 * the COPYING file in the top-level directory.
18 *
19 */
e495606d 20
af585b92 21#include "irq.h"
1d737c8a 22#include "mmu.h"
836a1b3c 23#include "x86.h"
6de4f3ad 24#include "kvm_cache_regs.h"
e495606d 25
edf88417 26#include <linux/kvm_host.h>
6aa8b732
AK
27#include <linux/types.h>
28#include <linux/string.h>
6aa8b732
AK
29#include <linux/mm.h>
30#include <linux/highmem.h>
31#include <linux/module.h>
448353ca 32#include <linux/swap.h>
05da4558 33#include <linux/hugetlb.h>
2f333bcb 34#include <linux/compiler.h>
bc6678a3 35#include <linux/srcu.h>
5a0e3ad6 36#include <linux/slab.h>
bf998156 37#include <linux/uaccess.h>
6aa8b732 38
e495606d
AK
39#include <asm/page.h>
40#include <asm/cmpxchg.h>
4e542370 41#include <asm/io.h>
13673a90 42#include <asm/vmx.h>
6aa8b732 43
18552672
JR
44/*
45 * When setting this variable to true it enables Two-Dimensional-Paging
46 * where the hardware walks 2 page tables:
47 * 1. the guest-virtual to guest-physical
48 * 2. while doing 1. it walks guest-physical to host-physical
49 * If the hardware supports that we don't need to do shadow paging.
50 */
2f333bcb 51bool tdp_enabled = false;
18552672 52
8b1fe17c
XG
53enum {
54 AUDIT_PRE_PAGE_FAULT,
55 AUDIT_POST_PAGE_FAULT,
56 AUDIT_PRE_PTE_WRITE,
6903074c
XG
57 AUDIT_POST_PTE_WRITE,
58 AUDIT_PRE_SYNC,
59 AUDIT_POST_SYNC
8b1fe17c 60};
37a7d8b0 61
8b1fe17c 62#undef MMU_DEBUG
37a7d8b0
AK
63
64#ifdef MMU_DEBUG
65
66#define pgprintk(x...) do { if (dbg) printk(x); } while (0)
67#define rmap_printk(x...) do { if (dbg) printk(x); } while (0)
68
69#else
70
71#define pgprintk(x...) do { } while (0)
72#define rmap_printk(x...) do { } while (0)
73
74#endif
75
8b1fe17c 76#ifdef MMU_DEBUG
476bc001 77static bool dbg = 0;
6ada8cca 78module_param(dbg, bool, 0644);
37a7d8b0 79#endif
6aa8b732 80
d6c69ee9
YD
81#ifndef MMU_DEBUG
82#define ASSERT(x) do { } while (0)
83#else
6aa8b732
AK
84#define ASSERT(x) \
85 if (!(x)) { \
86 printk(KERN_WARNING "assertion failed %s:%d: %s\n", \
87 __FILE__, __LINE__, #x); \
88 }
d6c69ee9 89#endif
6aa8b732 90
957ed9ef
XG
91#define PTE_PREFETCH_NUM 8
92
00763e41 93#define PT_FIRST_AVAIL_BITS_SHIFT 10
6aa8b732
AK
94#define PT64_SECOND_AVAIL_BITS_SHIFT 52
95
6aa8b732
AK
96#define PT64_LEVEL_BITS 9
97
98#define PT64_LEVEL_SHIFT(level) \
d77c26fc 99 (PAGE_SHIFT + (level - 1) * PT64_LEVEL_BITS)
6aa8b732 100
6aa8b732
AK
101#define PT64_INDEX(address, level)\
102 (((address) >> PT64_LEVEL_SHIFT(level)) & ((1 << PT64_LEVEL_BITS) - 1))
103
104
105#define PT32_LEVEL_BITS 10
106
107#define PT32_LEVEL_SHIFT(level) \
d77c26fc 108 (PAGE_SHIFT + (level - 1) * PT32_LEVEL_BITS)
6aa8b732 109
e04da980
JR
110#define PT32_LVL_OFFSET_MASK(level) \
111 (PT32_BASE_ADDR_MASK & ((1ULL << (PAGE_SHIFT + (((level) - 1) \
112 * PT32_LEVEL_BITS))) - 1))
6aa8b732
AK
113
114#define PT32_INDEX(address, level)\
115 (((address) >> PT32_LEVEL_SHIFT(level)) & ((1 << PT32_LEVEL_BITS) - 1))
116
117
27aba766 118#define PT64_BASE_ADDR_MASK (((1ULL << 52) - 1) & ~(u64)(PAGE_SIZE-1))
6aa8b732
AK
119#define PT64_DIR_BASE_ADDR_MASK \
120 (PT64_BASE_ADDR_MASK & ~((1ULL << (PAGE_SHIFT + PT64_LEVEL_BITS)) - 1))
e04da980
JR
121#define PT64_LVL_ADDR_MASK(level) \
122 (PT64_BASE_ADDR_MASK & ~((1ULL << (PAGE_SHIFT + (((level) - 1) \
123 * PT64_LEVEL_BITS))) - 1))
124#define PT64_LVL_OFFSET_MASK(level) \
125 (PT64_BASE_ADDR_MASK & ((1ULL << (PAGE_SHIFT + (((level) - 1) \
126 * PT64_LEVEL_BITS))) - 1))
6aa8b732
AK
127
128#define PT32_BASE_ADDR_MASK PAGE_MASK
129#define PT32_DIR_BASE_ADDR_MASK \
130 (PAGE_MASK & ~((1ULL << (PAGE_SHIFT + PT32_LEVEL_BITS)) - 1))
e04da980
JR
131#define PT32_LVL_ADDR_MASK(level) \
132 (PAGE_MASK & ~((1ULL << (PAGE_SHIFT + (((level) - 1) \
133 * PT32_LEVEL_BITS))) - 1))
6aa8b732 134
53166229
GN
135#define PT64_PERM_MASK (PT_PRESENT_MASK | PT_WRITABLE_MASK | shadow_user_mask \
136 | shadow_x_mask | shadow_nx_mask)
6aa8b732 137
fe135d2c
AK
138#define ACC_EXEC_MASK 1
139#define ACC_WRITE_MASK PT_WRITABLE_MASK
140#define ACC_USER_MASK PT_USER_MASK
141#define ACC_ALL (ACC_EXEC_MASK | ACC_WRITE_MASK | ACC_USER_MASK)
142
90bb6fc5
AK
143#include <trace/events/kvm.h>
144
07420171
AK
145#define CREATE_TRACE_POINTS
146#include "mmutrace.h"
147
49fde340
XG
148#define SPTE_HOST_WRITEABLE (1ULL << PT_FIRST_AVAIL_BITS_SHIFT)
149#define SPTE_MMU_WRITEABLE (1ULL << (PT_FIRST_AVAIL_BITS_SHIFT + 1))
1403283a 150
135f8c2b
AK
151#define SHADOW_PT_INDEX(addr, level) PT64_INDEX(addr, level)
152
220f773a
TY
153/* make pte_list_desc fit well in cache line */
154#define PTE_LIST_EXT 3
155
53c07b18
XG
156struct pte_list_desc {
157 u64 *sptes[PTE_LIST_EXT];
158 struct pte_list_desc *more;
cd4a4e53
AK
159};
160
2d11123a
AK
161struct kvm_shadow_walk_iterator {
162 u64 addr;
163 hpa_t shadow_addr;
2d11123a 164 u64 *sptep;
dd3bfd59 165 int level;
2d11123a
AK
166 unsigned index;
167};
168
169#define for_each_shadow_entry(_vcpu, _addr, _walker) \
170 for (shadow_walk_init(&(_walker), _vcpu, _addr); \
171 shadow_walk_okay(&(_walker)); \
172 shadow_walk_next(&(_walker)))
173
c2a2ac2b
XG
174#define for_each_shadow_entry_lockless(_vcpu, _addr, _walker, spte) \
175 for (shadow_walk_init(&(_walker), _vcpu, _addr); \
176 shadow_walk_okay(&(_walker)) && \
177 ({ spte = mmu_spte_get_lockless(_walker.sptep); 1; }); \
178 __shadow_walk_next(&(_walker), spte))
179
53c07b18 180static struct kmem_cache *pte_list_desc_cache;
d3d25b04 181static struct kmem_cache *mmu_page_header_cache;
45221ab6 182static struct percpu_counter kvm_total_used_mmu_pages;
b5a33a75 183
7b52345e
SY
184static u64 __read_mostly shadow_nx_mask;
185static u64 __read_mostly shadow_x_mask; /* mutual exclusive with nx_mask */
186static u64 __read_mostly shadow_user_mask;
187static u64 __read_mostly shadow_accessed_mask;
188static u64 __read_mostly shadow_dirty_mask;
ce88decf
XG
189static u64 __read_mostly shadow_mmio_mask;
190
191static void mmu_spte_set(u64 *sptep, u64 spte);
e676505a 192static void mmu_free_roots(struct kvm_vcpu *vcpu);
ce88decf
XG
193
194void kvm_mmu_set_mmio_spte_mask(u64 mmio_mask)
195{
196 shadow_mmio_mask = mmio_mask;
197}
198EXPORT_SYMBOL_GPL(kvm_mmu_set_mmio_spte_mask);
199
f2fd125d
XG
200/*
201 * spte bits of bit 3 ~ bit 11 are used as low 9 bits of generation number,
202 * the bits of bits 52 ~ bit 61 are used as high 10 bits of generation
203 * number.
204 */
205#define MMIO_SPTE_GEN_LOW_SHIFT 3
206#define MMIO_SPTE_GEN_HIGH_SHIFT 52
207
f8f55942 208#define MMIO_GEN_SHIFT 19
f2fd125d
XG
209#define MMIO_GEN_LOW_SHIFT 9
210#define MMIO_GEN_LOW_MASK ((1 << MMIO_GEN_LOW_SHIFT) - 1)
f8f55942
XG
211#define MMIO_GEN_MASK ((1 << MMIO_GEN_SHIFT) - 1)
212#define MMIO_MAX_GEN ((1 << MMIO_GEN_SHIFT) - 1)
f2fd125d
XG
213
214static u64 generation_mmio_spte_mask(unsigned int gen)
215{
216 u64 mask;
217
218 WARN_ON(gen > MMIO_MAX_GEN);
219
220 mask = (gen & MMIO_GEN_LOW_MASK) << MMIO_SPTE_GEN_LOW_SHIFT;
221 mask |= ((u64)gen >> MMIO_GEN_LOW_SHIFT) << MMIO_SPTE_GEN_HIGH_SHIFT;
222 return mask;
223}
224
225static unsigned int get_mmio_spte_generation(u64 spte)
226{
227 unsigned int gen;
228
229 spte &= ~shadow_mmio_mask;
230
231 gen = (spte >> MMIO_SPTE_GEN_LOW_SHIFT) & MMIO_GEN_LOW_MASK;
232 gen |= (spte >> MMIO_SPTE_GEN_HIGH_SHIFT) << MMIO_GEN_LOW_SHIFT;
233 return gen;
234}
235
f8f55942
XG
236static unsigned int kvm_current_mmio_generation(struct kvm *kvm)
237{
69c9ea93
XG
238 /*
239 * Init kvm generation close to MMIO_MAX_GEN to easily test the
240 * code of handling generation number wrap-around.
241 */
242 return (kvm_memslots(kvm)->generation +
243 MMIO_MAX_GEN - 150) & MMIO_GEN_MASK;
f8f55942
XG
244}
245
f2fd125d
XG
246static void mark_mmio_spte(struct kvm *kvm, u64 *sptep, u64 gfn,
247 unsigned access)
ce88decf 248{
f8f55942
XG
249 unsigned int gen = kvm_current_mmio_generation(kvm);
250 u64 mask = generation_mmio_spte_mask(gen);
95b0430d 251
ce88decf 252 access &= ACC_WRITE_MASK | ACC_USER_MASK;
f2fd125d 253 mask |= shadow_mmio_mask | access | gfn << PAGE_SHIFT;
f2fd125d 254
f8f55942 255 trace_mark_mmio_spte(sptep, gfn, access, gen);
f2fd125d 256 mmu_spte_set(sptep, mask);
ce88decf
XG
257}
258
259static bool is_mmio_spte(u64 spte)
260{
261 return (spte & shadow_mmio_mask) == shadow_mmio_mask;
262}
263
264static gfn_t get_mmio_spte_gfn(u64 spte)
265{
f2fd125d
XG
266 u64 mask = generation_mmio_spte_mask(MMIO_MAX_GEN) | shadow_mmio_mask;
267 return (spte & ~mask) >> PAGE_SHIFT;
ce88decf
XG
268}
269
270static unsigned get_mmio_spte_access(u64 spte)
271{
f2fd125d
XG
272 u64 mask = generation_mmio_spte_mask(MMIO_MAX_GEN) | shadow_mmio_mask;
273 return (spte & ~mask) & ~PAGE_MASK;
ce88decf
XG
274}
275
f2fd125d
XG
276static bool set_mmio_spte(struct kvm *kvm, u64 *sptep, gfn_t gfn,
277 pfn_t pfn, unsigned access)
ce88decf
XG
278{
279 if (unlikely(is_noslot_pfn(pfn))) {
f2fd125d 280 mark_mmio_spte(kvm, sptep, gfn, access);
ce88decf
XG
281 return true;
282 }
283
284 return false;
285}
c7addb90 286
f8f55942
XG
287static bool check_mmio_spte(struct kvm *kvm, u64 spte)
288{
089504c0
XG
289 unsigned int kvm_gen, spte_gen;
290
291 kvm_gen = kvm_current_mmio_generation(kvm);
292 spte_gen = get_mmio_spte_generation(spte);
293
294 trace_check_mmio_spte(spte, kvm_gen, spte_gen);
295 return likely(kvm_gen == spte_gen);
f8f55942
XG
296}
297
82725b20
DE
298static inline u64 rsvd_bits(int s, int e)
299{
300 return ((1ULL << (e - s + 1)) - 1) << s;
301}
302
7b52345e 303void kvm_mmu_set_mask_ptes(u64 user_mask, u64 accessed_mask,
4b12f0de 304 u64 dirty_mask, u64 nx_mask, u64 x_mask)
7b52345e
SY
305{
306 shadow_user_mask = user_mask;
307 shadow_accessed_mask = accessed_mask;
308 shadow_dirty_mask = dirty_mask;
309 shadow_nx_mask = nx_mask;
310 shadow_x_mask = x_mask;
311}
312EXPORT_SYMBOL_GPL(kvm_mmu_set_mask_ptes);
313
6aa8b732
AK
314static int is_cpuid_PSE36(void)
315{
316 return 1;
317}
318
73b1087e
AK
319static int is_nx(struct kvm_vcpu *vcpu)
320{
f6801dff 321 return vcpu->arch.efer & EFER_NX;
73b1087e
AK
322}
323
c7addb90
AK
324static int is_shadow_present_pte(u64 pte)
325{
ce88decf 326 return pte & PT_PRESENT_MASK && !is_mmio_spte(pte);
c7addb90
AK
327}
328
05da4558
MT
329static int is_large_pte(u64 pte)
330{
331 return pte & PT_PAGE_SIZE_MASK;
332}
333
43a3795a 334static int is_rmap_spte(u64 pte)
cd4a4e53 335{
4b1a80fa 336 return is_shadow_present_pte(pte);
cd4a4e53
AK
337}
338
776e6633
MT
339static int is_last_spte(u64 pte, int level)
340{
341 if (level == PT_PAGE_TABLE_LEVEL)
342 return 1;
852e3c19 343 if (is_large_pte(pte))
776e6633
MT
344 return 1;
345 return 0;
346}
347
35149e21 348static pfn_t spte_to_pfn(u64 pte)
0b49ea86 349{
35149e21 350 return (pte & PT64_BASE_ADDR_MASK) >> PAGE_SHIFT;
0b49ea86
AK
351}
352
da928521
AK
353static gfn_t pse36_gfn_delta(u32 gpte)
354{
355 int shift = 32 - PT32_DIR_PSE36_SHIFT - PAGE_SHIFT;
356
357 return (gpte & PT32_DIR_PSE36_MASK) << shift;
358}
359
603e0651 360#ifdef CONFIG_X86_64
d555c333 361static void __set_spte(u64 *sptep, u64 spte)
e663ee64 362{
603e0651 363 *sptep = spte;
e663ee64
AK
364}
365
603e0651 366static void __update_clear_spte_fast(u64 *sptep, u64 spte)
a9221dd5 367{
603e0651
XG
368 *sptep = spte;
369}
370
371static u64 __update_clear_spte_slow(u64 *sptep, u64 spte)
372{
373 return xchg(sptep, spte);
374}
c2a2ac2b
XG
375
376static u64 __get_spte_lockless(u64 *sptep)
377{
378 return ACCESS_ONCE(*sptep);
379}
ce88decf
XG
380
381static bool __check_direct_spte_mmio_pf(u64 spte)
382{
383 /* It is valid if the spte is zapped. */
384 return spte == 0ull;
385}
a9221dd5 386#else
603e0651
XG
387union split_spte {
388 struct {
389 u32 spte_low;
390 u32 spte_high;
391 };
392 u64 spte;
393};
a9221dd5 394
c2a2ac2b
XG
395static void count_spte_clear(u64 *sptep, u64 spte)
396{
397 struct kvm_mmu_page *sp = page_header(__pa(sptep));
398
399 if (is_shadow_present_pte(spte))
400 return;
401
402 /* Ensure the spte is completely set before we increase the count */
403 smp_wmb();
404 sp->clear_spte_count++;
405}
406
603e0651
XG
407static void __set_spte(u64 *sptep, u64 spte)
408{
409 union split_spte *ssptep, sspte;
a9221dd5 410
603e0651
XG
411 ssptep = (union split_spte *)sptep;
412 sspte = (union split_spte)spte;
413
414 ssptep->spte_high = sspte.spte_high;
415
416 /*
417 * If we map the spte from nonpresent to present, We should store
418 * the high bits firstly, then set present bit, so cpu can not
419 * fetch this spte while we are setting the spte.
420 */
421 smp_wmb();
422
423 ssptep->spte_low = sspte.spte_low;
a9221dd5
AK
424}
425
603e0651
XG
426static void __update_clear_spte_fast(u64 *sptep, u64 spte)
427{
428 union split_spte *ssptep, sspte;
429
430 ssptep = (union split_spte *)sptep;
431 sspte = (union split_spte)spte;
432
433 ssptep->spte_low = sspte.spte_low;
434
435 /*
436 * If we map the spte from present to nonpresent, we should clear
437 * present bit firstly to avoid vcpu fetch the old high bits.
438 */
439 smp_wmb();
440
441 ssptep->spte_high = sspte.spte_high;
c2a2ac2b 442 count_spte_clear(sptep, spte);
603e0651
XG
443}
444
445static u64 __update_clear_spte_slow(u64 *sptep, u64 spte)
446{
447 union split_spte *ssptep, sspte, orig;
448
449 ssptep = (union split_spte *)sptep;
450 sspte = (union split_spte)spte;
451
452 /* xchg acts as a barrier before the setting of the high bits */
453 orig.spte_low = xchg(&ssptep->spte_low, sspte.spte_low);
41bc3186
ZJ
454 orig.spte_high = ssptep->spte_high;
455 ssptep->spte_high = sspte.spte_high;
c2a2ac2b 456 count_spte_clear(sptep, spte);
603e0651
XG
457
458 return orig.spte;
459}
c2a2ac2b
XG
460
461/*
462 * The idea using the light way get the spte on x86_32 guest is from
463 * gup_get_pte(arch/x86/mm/gup.c).
accaefe0
XG
464 *
465 * An spte tlb flush may be pending, because kvm_set_pte_rmapp
466 * coalesces them and we are running out of the MMU lock. Therefore
467 * we need to protect against in-progress updates of the spte.
468 *
469 * Reading the spte while an update is in progress may get the old value
470 * for the high part of the spte. The race is fine for a present->non-present
471 * change (because the high part of the spte is ignored for non-present spte),
472 * but for a present->present change we must reread the spte.
473 *
474 * All such changes are done in two steps (present->non-present and
475 * non-present->present), hence it is enough to count the number of
476 * present->non-present updates: if it changed while reading the spte,
477 * we might have hit the race. This is done using clear_spte_count.
c2a2ac2b
XG
478 */
479static u64 __get_spte_lockless(u64 *sptep)
480{
481 struct kvm_mmu_page *sp = page_header(__pa(sptep));
482 union split_spte spte, *orig = (union split_spte *)sptep;
483 int count;
484
485retry:
486 count = sp->clear_spte_count;
487 smp_rmb();
488
489 spte.spte_low = orig->spte_low;
490 smp_rmb();
491
492 spte.spte_high = orig->spte_high;
493 smp_rmb();
494
495 if (unlikely(spte.spte_low != orig->spte_low ||
496 count != sp->clear_spte_count))
497 goto retry;
498
499 return spte.spte;
500}
ce88decf
XG
501
502static bool __check_direct_spte_mmio_pf(u64 spte)
503{
504 union split_spte sspte = (union split_spte)spte;
505 u32 high_mmio_mask = shadow_mmio_mask >> 32;
506
507 /* It is valid if the spte is zapped. */
508 if (spte == 0ull)
509 return true;
510
511 /* It is valid if the spte is being zapped. */
512 if (sspte.spte_low == 0ull &&
513 (sspte.spte_high & high_mmio_mask) == high_mmio_mask)
514 return true;
515
516 return false;
517}
603e0651
XG
518#endif
519
c7ba5b48
XG
520static bool spte_is_locklessly_modifiable(u64 spte)
521{
feb3eb70
GN
522 return (spte & (SPTE_HOST_WRITEABLE | SPTE_MMU_WRITEABLE)) ==
523 (SPTE_HOST_WRITEABLE | SPTE_MMU_WRITEABLE);
c7ba5b48
XG
524}
525
8672b721
XG
526static bool spte_has_volatile_bits(u64 spte)
527{
c7ba5b48
XG
528 /*
529 * Always atomicly update spte if it can be updated
530 * out of mmu-lock, it can ensure dirty bit is not lost,
531 * also, it can help us to get a stable is_writable_pte()
532 * to ensure tlb flush is not missed.
533 */
534 if (spte_is_locklessly_modifiable(spte))
535 return true;
536
8672b721
XG
537 if (!shadow_accessed_mask)
538 return false;
539
540 if (!is_shadow_present_pte(spte))
541 return false;
542
4132779b
XG
543 if ((spte & shadow_accessed_mask) &&
544 (!is_writable_pte(spte) || (spte & shadow_dirty_mask)))
8672b721
XG
545 return false;
546
547 return true;
548}
549
4132779b
XG
550static bool spte_is_bit_cleared(u64 old_spte, u64 new_spte, u64 bit_mask)
551{
552 return (old_spte & bit_mask) && !(new_spte & bit_mask);
553}
554
1df9f2dc
XG
555/* Rules for using mmu_spte_set:
556 * Set the sptep from nonpresent to present.
557 * Note: the sptep being assigned *must* be either not present
558 * or in a state where the hardware will not attempt to update
559 * the spte.
560 */
561static void mmu_spte_set(u64 *sptep, u64 new_spte)
562{
563 WARN_ON(is_shadow_present_pte(*sptep));
564 __set_spte(sptep, new_spte);
565}
566
567/* Rules for using mmu_spte_update:
568 * Update the state bits, it means the mapped pfn is not changged.
6e7d0354
XG
569 *
570 * Whenever we overwrite a writable spte with a read-only one we
571 * should flush remote TLBs. Otherwise rmap_write_protect
572 * will find a read-only spte, even though the writable spte
573 * might be cached on a CPU's TLB, the return value indicates this
574 * case.
1df9f2dc 575 */
6e7d0354 576static bool mmu_spte_update(u64 *sptep, u64 new_spte)
b79b93f9 577{
c7ba5b48 578 u64 old_spte = *sptep;
6e7d0354 579 bool ret = false;
4132779b
XG
580
581 WARN_ON(!is_rmap_spte(new_spte));
b79b93f9 582
6e7d0354
XG
583 if (!is_shadow_present_pte(old_spte)) {
584 mmu_spte_set(sptep, new_spte);
585 return ret;
586 }
4132779b 587
c7ba5b48 588 if (!spte_has_volatile_bits(old_spte))
603e0651 589 __update_clear_spte_fast(sptep, new_spte);
4132779b 590 else
603e0651 591 old_spte = __update_clear_spte_slow(sptep, new_spte);
4132779b 592
c7ba5b48
XG
593 /*
594 * For the spte updated out of mmu-lock is safe, since
595 * we always atomicly update it, see the comments in
596 * spte_has_volatile_bits().
597 */
6e7d0354
XG
598 if (is_writable_pte(old_spte) && !is_writable_pte(new_spte))
599 ret = true;
600
4132779b 601 if (!shadow_accessed_mask)
6e7d0354 602 return ret;
4132779b
XG
603
604 if (spte_is_bit_cleared(old_spte, new_spte, shadow_accessed_mask))
605 kvm_set_pfn_accessed(spte_to_pfn(old_spte));
606 if (spte_is_bit_cleared(old_spte, new_spte, shadow_dirty_mask))
607 kvm_set_pfn_dirty(spte_to_pfn(old_spte));
6e7d0354
XG
608
609 return ret;
b79b93f9
AK
610}
611
1df9f2dc
XG
612/*
613 * Rules for using mmu_spte_clear_track_bits:
614 * It sets the sptep from present to nonpresent, and track the
615 * state bits, it is used to clear the last level sptep.
616 */
617static int mmu_spte_clear_track_bits(u64 *sptep)
618{
619 pfn_t pfn;
620 u64 old_spte = *sptep;
621
622 if (!spte_has_volatile_bits(old_spte))
603e0651 623 __update_clear_spte_fast(sptep, 0ull);
1df9f2dc 624 else
603e0651 625 old_spte = __update_clear_spte_slow(sptep, 0ull);
1df9f2dc
XG
626
627 if (!is_rmap_spte(old_spte))
628 return 0;
629
630 pfn = spte_to_pfn(old_spte);
86fde74c
XG
631
632 /*
633 * KVM does not hold the refcount of the page used by
634 * kvm mmu, before reclaiming the page, we should
635 * unmap it from mmu first.
636 */
637 WARN_ON(!kvm_is_mmio_pfn(pfn) && !page_count(pfn_to_page(pfn)));
638
1df9f2dc
XG
639 if (!shadow_accessed_mask || old_spte & shadow_accessed_mask)
640 kvm_set_pfn_accessed(pfn);
641 if (!shadow_dirty_mask || (old_spte & shadow_dirty_mask))
642 kvm_set_pfn_dirty(pfn);
643 return 1;
644}
645
646/*
647 * Rules for using mmu_spte_clear_no_track:
648 * Directly clear spte without caring the state bits of sptep,
649 * it is used to set the upper level spte.
650 */
651static void mmu_spte_clear_no_track(u64 *sptep)
652{
603e0651 653 __update_clear_spte_fast(sptep, 0ull);
1df9f2dc
XG
654}
655
c2a2ac2b
XG
656static u64 mmu_spte_get_lockless(u64 *sptep)
657{
658 return __get_spte_lockless(sptep);
659}
660
661static void walk_shadow_page_lockless_begin(struct kvm_vcpu *vcpu)
662{
c142786c
AK
663 /*
664 * Prevent page table teardown by making any free-er wait during
665 * kvm_flush_remote_tlbs() IPI to all active vcpus.
666 */
667 local_irq_disable();
668 vcpu->mode = READING_SHADOW_PAGE_TABLES;
669 /*
670 * Make sure a following spte read is not reordered ahead of the write
671 * to vcpu->mode.
672 */
673 smp_mb();
c2a2ac2b
XG
674}
675
676static void walk_shadow_page_lockless_end(struct kvm_vcpu *vcpu)
677{
c142786c
AK
678 /*
679 * Make sure the write to vcpu->mode is not reordered in front of
680 * reads to sptes. If it does, kvm_commit_zap_page() can see us
681 * OUTSIDE_GUEST_MODE and proceed to free the shadow page table.
682 */
683 smp_mb();
684 vcpu->mode = OUTSIDE_GUEST_MODE;
685 local_irq_enable();
c2a2ac2b
XG
686}
687
e2dec939 688static int mmu_topup_memory_cache(struct kvm_mmu_memory_cache *cache,
2e3e5882 689 struct kmem_cache *base_cache, int min)
714b93da
AK
690{
691 void *obj;
692
693 if (cache->nobjs >= min)
e2dec939 694 return 0;
714b93da 695 while (cache->nobjs < ARRAY_SIZE(cache->objects)) {
2e3e5882 696 obj = kmem_cache_zalloc(base_cache, GFP_KERNEL);
714b93da 697 if (!obj)
e2dec939 698 return -ENOMEM;
714b93da
AK
699 cache->objects[cache->nobjs++] = obj;
700 }
e2dec939 701 return 0;
714b93da
AK
702}
703
f759e2b4
XG
704static int mmu_memory_cache_free_objects(struct kvm_mmu_memory_cache *cache)
705{
706 return cache->nobjs;
707}
708
e8ad9a70
XG
709static void mmu_free_memory_cache(struct kvm_mmu_memory_cache *mc,
710 struct kmem_cache *cache)
714b93da
AK
711{
712 while (mc->nobjs)
e8ad9a70 713 kmem_cache_free(cache, mc->objects[--mc->nobjs]);
714b93da
AK
714}
715
c1158e63 716static int mmu_topup_memory_cache_page(struct kvm_mmu_memory_cache *cache,
2e3e5882 717 int min)
c1158e63 718{
842f22ed 719 void *page;
c1158e63
AK
720
721 if (cache->nobjs >= min)
722 return 0;
723 while (cache->nobjs < ARRAY_SIZE(cache->objects)) {
842f22ed 724 page = (void *)__get_free_page(GFP_KERNEL);
c1158e63
AK
725 if (!page)
726 return -ENOMEM;
842f22ed 727 cache->objects[cache->nobjs++] = page;
c1158e63
AK
728 }
729 return 0;
730}
731
732static void mmu_free_memory_cache_page(struct kvm_mmu_memory_cache *mc)
733{
734 while (mc->nobjs)
c4d198d5 735 free_page((unsigned long)mc->objects[--mc->nobjs]);
c1158e63
AK
736}
737
2e3e5882 738static int mmu_topup_memory_caches(struct kvm_vcpu *vcpu)
714b93da 739{
e2dec939
AK
740 int r;
741
53c07b18 742 r = mmu_topup_memory_cache(&vcpu->arch.mmu_pte_list_desc_cache,
67052b35 743 pte_list_desc_cache, 8 + PTE_PREFETCH_NUM);
d3d25b04
AK
744 if (r)
745 goto out;
ad312c7c 746 r = mmu_topup_memory_cache_page(&vcpu->arch.mmu_page_cache, 8);
d3d25b04
AK
747 if (r)
748 goto out;
ad312c7c 749 r = mmu_topup_memory_cache(&vcpu->arch.mmu_page_header_cache,
2e3e5882 750 mmu_page_header_cache, 4);
e2dec939
AK
751out:
752 return r;
714b93da
AK
753}
754
755static void mmu_free_memory_caches(struct kvm_vcpu *vcpu)
756{
53c07b18
XG
757 mmu_free_memory_cache(&vcpu->arch.mmu_pte_list_desc_cache,
758 pte_list_desc_cache);
ad312c7c 759 mmu_free_memory_cache_page(&vcpu->arch.mmu_page_cache);
e8ad9a70
XG
760 mmu_free_memory_cache(&vcpu->arch.mmu_page_header_cache,
761 mmu_page_header_cache);
714b93da
AK
762}
763
80feb89a 764static void *mmu_memory_cache_alloc(struct kvm_mmu_memory_cache *mc)
714b93da
AK
765{
766 void *p;
767
768 BUG_ON(!mc->nobjs);
769 p = mc->objects[--mc->nobjs];
714b93da
AK
770 return p;
771}
772
53c07b18 773static struct pte_list_desc *mmu_alloc_pte_list_desc(struct kvm_vcpu *vcpu)
714b93da 774{
80feb89a 775 return mmu_memory_cache_alloc(&vcpu->arch.mmu_pte_list_desc_cache);
714b93da
AK
776}
777
53c07b18 778static void mmu_free_pte_list_desc(struct pte_list_desc *pte_list_desc)
714b93da 779{
53c07b18 780 kmem_cache_free(pte_list_desc_cache, pte_list_desc);
714b93da
AK
781}
782
2032a93d
LJ
783static gfn_t kvm_mmu_page_get_gfn(struct kvm_mmu_page *sp, int index)
784{
785 if (!sp->role.direct)
786 return sp->gfns[index];
787
788 return sp->gfn + (index << ((sp->role.level - 1) * PT64_LEVEL_BITS));
789}
790
791static void kvm_mmu_page_set_gfn(struct kvm_mmu_page *sp, int index, gfn_t gfn)
792{
793 if (sp->role.direct)
794 BUG_ON(gfn != kvm_mmu_page_get_gfn(sp, index));
795 else
796 sp->gfns[index] = gfn;
797}
798
05da4558 799/*
d4dbf470
TY
800 * Return the pointer to the large page information for a given gfn,
801 * handling slots that are not large page aligned.
05da4558 802 */
d4dbf470
TY
803static struct kvm_lpage_info *lpage_info_slot(gfn_t gfn,
804 struct kvm_memory_slot *slot,
805 int level)
05da4558
MT
806{
807 unsigned long idx;
808
fb03cb6f 809 idx = gfn_to_index(gfn, slot->base_gfn, level);
db3fe4eb 810 return &slot->arch.lpage_info[level - 2][idx];
05da4558
MT
811}
812
813static void account_shadowed(struct kvm *kvm, gfn_t gfn)
814{
d25797b2 815 struct kvm_memory_slot *slot;
d4dbf470 816 struct kvm_lpage_info *linfo;
d25797b2 817 int i;
05da4558 818
a1f4d395 819 slot = gfn_to_memslot(kvm, gfn);
d25797b2
JR
820 for (i = PT_DIRECTORY_LEVEL;
821 i < PT_PAGE_TABLE_LEVEL + KVM_NR_PAGE_SIZES; ++i) {
d4dbf470
TY
822 linfo = lpage_info_slot(gfn, slot, i);
823 linfo->write_count += 1;
d25797b2 824 }
332b207d 825 kvm->arch.indirect_shadow_pages++;
05da4558
MT
826}
827
828static void unaccount_shadowed(struct kvm *kvm, gfn_t gfn)
829{
d25797b2 830 struct kvm_memory_slot *slot;
d4dbf470 831 struct kvm_lpage_info *linfo;
d25797b2 832 int i;
05da4558 833
a1f4d395 834 slot = gfn_to_memslot(kvm, gfn);
d25797b2
JR
835 for (i = PT_DIRECTORY_LEVEL;
836 i < PT_PAGE_TABLE_LEVEL + KVM_NR_PAGE_SIZES; ++i) {
d4dbf470
TY
837 linfo = lpage_info_slot(gfn, slot, i);
838 linfo->write_count -= 1;
839 WARN_ON(linfo->write_count < 0);
d25797b2 840 }
332b207d 841 kvm->arch.indirect_shadow_pages--;
05da4558
MT
842}
843
d25797b2
JR
844static int has_wrprotected_page(struct kvm *kvm,
845 gfn_t gfn,
846 int level)
05da4558 847{
2843099f 848 struct kvm_memory_slot *slot;
d4dbf470 849 struct kvm_lpage_info *linfo;
05da4558 850
a1f4d395 851 slot = gfn_to_memslot(kvm, gfn);
05da4558 852 if (slot) {
d4dbf470
TY
853 linfo = lpage_info_slot(gfn, slot, level);
854 return linfo->write_count;
05da4558
MT
855 }
856
857 return 1;
858}
859
d25797b2 860static int host_mapping_level(struct kvm *kvm, gfn_t gfn)
05da4558 861{
8f0b1ab6 862 unsigned long page_size;
d25797b2 863 int i, ret = 0;
05da4558 864
8f0b1ab6 865 page_size = kvm_host_page_size(kvm, gfn);
05da4558 866
d25797b2
JR
867 for (i = PT_PAGE_TABLE_LEVEL;
868 i < (PT_PAGE_TABLE_LEVEL + KVM_NR_PAGE_SIZES); ++i) {
869 if (page_size >= KVM_HPAGE_SIZE(i))
870 ret = i;
871 else
872 break;
873 }
874
4c2155ce 875 return ret;
05da4558
MT
876}
877
5d163b1c
XG
878static struct kvm_memory_slot *
879gfn_to_memslot_dirty_bitmap(struct kvm_vcpu *vcpu, gfn_t gfn,
880 bool no_dirty_log)
05da4558
MT
881{
882 struct kvm_memory_slot *slot;
5d163b1c
XG
883
884 slot = gfn_to_memslot(vcpu->kvm, gfn);
885 if (!slot || slot->flags & KVM_MEMSLOT_INVALID ||
886 (no_dirty_log && slot->dirty_bitmap))
887 slot = NULL;
888
889 return slot;
890}
891
892static bool mapping_level_dirty_bitmap(struct kvm_vcpu *vcpu, gfn_t large_gfn)
893{
a0a8eaba 894 return !gfn_to_memslot_dirty_bitmap(vcpu, large_gfn, true);
936a5fe6
AA
895}
896
897static int mapping_level(struct kvm_vcpu *vcpu, gfn_t large_gfn)
898{
899 int host_level, level, max_level;
05da4558 900
d25797b2
JR
901 host_level = host_mapping_level(vcpu->kvm, large_gfn);
902
903 if (host_level == PT_PAGE_TABLE_LEVEL)
904 return host_level;
905
55dd98c3 906 max_level = min(kvm_x86_ops->get_lpage_level(), host_level);
878403b7
SY
907
908 for (level = PT_DIRECTORY_LEVEL; level <= max_level; ++level)
d25797b2
JR
909 if (has_wrprotected_page(vcpu->kvm, large_gfn, level))
910 break;
d25797b2
JR
911
912 return level - 1;
05da4558
MT
913}
914
290fc38d 915/*
53c07b18 916 * Pte mapping structures:
cd4a4e53 917 *
53c07b18 918 * If pte_list bit zero is zero, then pte_list point to the spte.
cd4a4e53 919 *
53c07b18
XG
920 * If pte_list bit zero is one, (then pte_list & ~1) points to a struct
921 * pte_list_desc containing more mappings.
53a27b39 922 *
53c07b18 923 * Returns the number of pte entries before the spte was added or zero if
53a27b39
MT
924 * the spte was not added.
925 *
cd4a4e53 926 */
53c07b18
XG
927static int pte_list_add(struct kvm_vcpu *vcpu, u64 *spte,
928 unsigned long *pte_list)
cd4a4e53 929{
53c07b18 930 struct pte_list_desc *desc;
53a27b39 931 int i, count = 0;
cd4a4e53 932
53c07b18
XG
933 if (!*pte_list) {
934 rmap_printk("pte_list_add: %p %llx 0->1\n", spte, *spte);
935 *pte_list = (unsigned long)spte;
936 } else if (!(*pte_list & 1)) {
937 rmap_printk("pte_list_add: %p %llx 1->many\n", spte, *spte);
938 desc = mmu_alloc_pte_list_desc(vcpu);
939 desc->sptes[0] = (u64 *)*pte_list;
d555c333 940 desc->sptes[1] = spte;
53c07b18 941 *pte_list = (unsigned long)desc | 1;
cb16a7b3 942 ++count;
cd4a4e53 943 } else {
53c07b18
XG
944 rmap_printk("pte_list_add: %p %llx many->many\n", spte, *spte);
945 desc = (struct pte_list_desc *)(*pte_list & ~1ul);
946 while (desc->sptes[PTE_LIST_EXT-1] && desc->more) {
cd4a4e53 947 desc = desc->more;
53c07b18 948 count += PTE_LIST_EXT;
53a27b39 949 }
53c07b18
XG
950 if (desc->sptes[PTE_LIST_EXT-1]) {
951 desc->more = mmu_alloc_pte_list_desc(vcpu);
cd4a4e53
AK
952 desc = desc->more;
953 }
d555c333 954 for (i = 0; desc->sptes[i]; ++i)
cb16a7b3 955 ++count;
d555c333 956 desc->sptes[i] = spte;
cd4a4e53 957 }
53a27b39 958 return count;
cd4a4e53
AK
959}
960
53c07b18
XG
961static void
962pte_list_desc_remove_entry(unsigned long *pte_list, struct pte_list_desc *desc,
963 int i, struct pte_list_desc *prev_desc)
cd4a4e53
AK
964{
965 int j;
966
53c07b18 967 for (j = PTE_LIST_EXT - 1; !desc->sptes[j] && j > i; --j)
cd4a4e53 968 ;
d555c333
AK
969 desc->sptes[i] = desc->sptes[j];
970 desc->sptes[j] = NULL;
cd4a4e53
AK
971 if (j != 0)
972 return;
973 if (!prev_desc && !desc->more)
53c07b18 974 *pte_list = (unsigned long)desc->sptes[0];
cd4a4e53
AK
975 else
976 if (prev_desc)
977 prev_desc->more = desc->more;
978 else
53c07b18
XG
979 *pte_list = (unsigned long)desc->more | 1;
980 mmu_free_pte_list_desc(desc);
cd4a4e53
AK
981}
982
53c07b18 983static void pte_list_remove(u64 *spte, unsigned long *pte_list)
cd4a4e53 984{
53c07b18
XG
985 struct pte_list_desc *desc;
986 struct pte_list_desc *prev_desc;
cd4a4e53
AK
987 int i;
988
53c07b18
XG
989 if (!*pte_list) {
990 printk(KERN_ERR "pte_list_remove: %p 0->BUG\n", spte);
cd4a4e53 991 BUG();
53c07b18
XG
992 } else if (!(*pte_list & 1)) {
993 rmap_printk("pte_list_remove: %p 1->0\n", spte);
994 if ((u64 *)*pte_list != spte) {
995 printk(KERN_ERR "pte_list_remove: %p 1->BUG\n", spte);
cd4a4e53
AK
996 BUG();
997 }
53c07b18 998 *pte_list = 0;
cd4a4e53 999 } else {
53c07b18
XG
1000 rmap_printk("pte_list_remove: %p many->many\n", spte);
1001 desc = (struct pte_list_desc *)(*pte_list & ~1ul);
cd4a4e53
AK
1002 prev_desc = NULL;
1003 while (desc) {
53c07b18 1004 for (i = 0; i < PTE_LIST_EXT && desc->sptes[i]; ++i)
d555c333 1005 if (desc->sptes[i] == spte) {
53c07b18 1006 pte_list_desc_remove_entry(pte_list,
714b93da 1007 desc, i,
cd4a4e53
AK
1008 prev_desc);
1009 return;
1010 }
1011 prev_desc = desc;
1012 desc = desc->more;
1013 }
53c07b18 1014 pr_err("pte_list_remove: %p many->many\n", spte);
cd4a4e53
AK
1015 BUG();
1016 }
1017}
1018
67052b35
XG
1019typedef void (*pte_list_walk_fn) (u64 *spte);
1020static void pte_list_walk(unsigned long *pte_list, pte_list_walk_fn fn)
1021{
1022 struct pte_list_desc *desc;
1023 int i;
1024
1025 if (!*pte_list)
1026 return;
1027
1028 if (!(*pte_list & 1))
1029 return fn((u64 *)*pte_list);
1030
1031 desc = (struct pte_list_desc *)(*pte_list & ~1ul);
1032 while (desc) {
1033 for (i = 0; i < PTE_LIST_EXT && desc->sptes[i]; ++i)
1034 fn(desc->sptes[i]);
1035 desc = desc->more;
1036 }
1037}
1038
9373e2c0 1039static unsigned long *__gfn_to_rmap(gfn_t gfn, int level,
9b9b1492 1040 struct kvm_memory_slot *slot)
53c07b18 1041{
77d11309 1042 unsigned long idx;
53c07b18 1043
77d11309 1044 idx = gfn_to_index(gfn, slot->base_gfn, level);
d89cc617 1045 return &slot->arch.rmap[level - PT_PAGE_TABLE_LEVEL][idx];
53c07b18
XG
1046}
1047
9b9b1492
TY
1048/*
1049 * Take gfn and return the reverse mapping to it.
1050 */
1051static unsigned long *gfn_to_rmap(struct kvm *kvm, gfn_t gfn, int level)
1052{
1053 struct kvm_memory_slot *slot;
1054
1055 slot = gfn_to_memslot(kvm, gfn);
9373e2c0 1056 return __gfn_to_rmap(gfn, level, slot);
9b9b1492
TY
1057}
1058
f759e2b4
XG
1059static bool rmap_can_add(struct kvm_vcpu *vcpu)
1060{
1061 struct kvm_mmu_memory_cache *cache;
1062
1063 cache = &vcpu->arch.mmu_pte_list_desc_cache;
1064 return mmu_memory_cache_free_objects(cache);
1065}
1066
53c07b18
XG
1067static int rmap_add(struct kvm_vcpu *vcpu, u64 *spte, gfn_t gfn)
1068{
1069 struct kvm_mmu_page *sp;
1070 unsigned long *rmapp;
1071
53c07b18
XG
1072 sp = page_header(__pa(spte));
1073 kvm_mmu_page_set_gfn(sp, spte - sp->spt, gfn);
1074 rmapp = gfn_to_rmap(vcpu->kvm, gfn, sp->role.level);
1075 return pte_list_add(vcpu, spte, rmapp);
1076}
1077
53c07b18
XG
1078static void rmap_remove(struct kvm *kvm, u64 *spte)
1079{
1080 struct kvm_mmu_page *sp;
1081 gfn_t gfn;
1082 unsigned long *rmapp;
1083
1084 sp = page_header(__pa(spte));
1085 gfn = kvm_mmu_page_get_gfn(sp, spte - sp->spt);
1086 rmapp = gfn_to_rmap(kvm, gfn, sp->role.level);
1087 pte_list_remove(spte, rmapp);
1088}
1089
1e3f42f0
TY
1090/*
1091 * Used by the following functions to iterate through the sptes linked by a
1092 * rmap. All fields are private and not assumed to be used outside.
1093 */
1094struct rmap_iterator {
1095 /* private fields */
1096 struct pte_list_desc *desc; /* holds the sptep if not NULL */
1097 int pos; /* index of the sptep */
1098};
1099
1100/*
1101 * Iteration must be started by this function. This should also be used after
1102 * removing/dropping sptes from the rmap link because in such cases the
1103 * information in the itererator may not be valid.
1104 *
1105 * Returns sptep if found, NULL otherwise.
1106 */
1107static u64 *rmap_get_first(unsigned long rmap, struct rmap_iterator *iter)
1108{
1109 if (!rmap)
1110 return NULL;
1111
1112 if (!(rmap & 1)) {
1113 iter->desc = NULL;
1114 return (u64 *)rmap;
1115 }
1116
1117 iter->desc = (struct pte_list_desc *)(rmap & ~1ul);
1118 iter->pos = 0;
1119 return iter->desc->sptes[iter->pos];
1120}
1121
1122/*
1123 * Must be used with a valid iterator: e.g. after rmap_get_first().
1124 *
1125 * Returns sptep if found, NULL otherwise.
1126 */
1127static u64 *rmap_get_next(struct rmap_iterator *iter)
1128{
1129 if (iter->desc) {
1130 if (iter->pos < PTE_LIST_EXT - 1) {
1131 u64 *sptep;
1132
1133 ++iter->pos;
1134 sptep = iter->desc->sptes[iter->pos];
1135 if (sptep)
1136 return sptep;
1137 }
1138
1139 iter->desc = iter->desc->more;
1140
1141 if (iter->desc) {
1142 iter->pos = 0;
1143 /* desc->sptes[0] cannot be NULL */
1144 return iter->desc->sptes[iter->pos];
1145 }
1146 }
1147
1148 return NULL;
1149}
1150
c3707958 1151static void drop_spte(struct kvm *kvm, u64 *sptep)
e4b502ea 1152{
1df9f2dc 1153 if (mmu_spte_clear_track_bits(sptep))
eb45fda4 1154 rmap_remove(kvm, sptep);
be38d276
AK
1155}
1156
8e22f955
XG
1157
1158static bool __drop_large_spte(struct kvm *kvm, u64 *sptep)
1159{
1160 if (is_large_pte(*sptep)) {
1161 WARN_ON(page_header(__pa(sptep))->role.level ==
1162 PT_PAGE_TABLE_LEVEL);
1163 drop_spte(kvm, sptep);
1164 --kvm->stat.lpages;
1165 return true;
1166 }
1167
1168 return false;
1169}
1170
1171static void drop_large_spte(struct kvm_vcpu *vcpu, u64 *sptep)
1172{
1173 if (__drop_large_spte(vcpu->kvm, sptep))
1174 kvm_flush_remote_tlbs(vcpu->kvm);
1175}
1176
1177/*
49fde340 1178 * Write-protect on the specified @sptep, @pt_protect indicates whether
6b73a960
MT
1179 * spte writ-protection is caused by protecting shadow page table.
1180 * @flush indicates whether tlb need be flushed.
49fde340
XG
1181 *
1182 * Note: write protection is difference between drity logging and spte
1183 * protection:
1184 * - for dirty logging, the spte can be set to writable at anytime if
1185 * its dirty bitmap is properly set.
1186 * - for spte protection, the spte can be writable only after unsync-ing
1187 * shadow page.
8e22f955 1188 *
6b73a960 1189 * Return true if the spte is dropped.
8e22f955 1190 */
6b73a960
MT
1191static bool
1192spte_write_protect(struct kvm *kvm, u64 *sptep, bool *flush, bool pt_protect)
d13bc5b5
XG
1193{
1194 u64 spte = *sptep;
1195
49fde340
XG
1196 if (!is_writable_pte(spte) &&
1197 !(pt_protect && spte_is_locklessly_modifiable(spte)))
d13bc5b5
XG
1198 return false;
1199
1200 rmap_printk("rmap_write_protect: spte %p %llx\n", sptep, *sptep);
1201
6b73a960
MT
1202 if (__drop_large_spte(kvm, sptep)) {
1203 *flush |= true;
1204 return true;
1205 }
1206
49fde340
XG
1207 if (pt_protect)
1208 spte &= ~SPTE_MMU_WRITEABLE;
d13bc5b5 1209 spte = spte & ~PT_WRITABLE_MASK;
49fde340 1210
6b73a960
MT
1211 *flush |= mmu_spte_update(sptep, spte);
1212 return false;
d13bc5b5
XG
1213}
1214
49fde340 1215static bool __rmap_write_protect(struct kvm *kvm, unsigned long *rmapp,
245c3912 1216 bool pt_protect)
98348e95 1217{
1e3f42f0
TY
1218 u64 *sptep;
1219 struct rmap_iterator iter;
d13bc5b5 1220 bool flush = false;
374cbac0 1221
1e3f42f0
TY
1222 for (sptep = rmap_get_first(*rmapp, &iter); sptep;) {
1223 BUG_ON(!(*sptep & PT_PRESENT_MASK));
6b73a960
MT
1224 if (spte_write_protect(kvm, sptep, &flush, pt_protect)) {
1225 sptep = rmap_get_first(*rmapp, &iter);
1226 continue;
1227 }
a0ed4607 1228
d13bc5b5 1229 sptep = rmap_get_next(&iter);
374cbac0 1230 }
855149aa 1231
d13bc5b5 1232 return flush;
a0ed4607
TY
1233}
1234
5dc99b23
TY
1235/**
1236 * kvm_mmu_write_protect_pt_masked - write protect selected PT level pages
1237 * @kvm: kvm instance
1238 * @slot: slot to protect
1239 * @gfn_offset: start of the BITS_PER_LONG pages we care about
1240 * @mask: indicates which pages we should protect
1241 *
1242 * Used when we do not need to care about huge page mappings: e.g. during dirty
1243 * logging we do not have any such mappings.
1244 */
1245void kvm_mmu_write_protect_pt_masked(struct kvm *kvm,
1246 struct kvm_memory_slot *slot,
1247 gfn_t gfn_offset, unsigned long mask)
a0ed4607
TY
1248{
1249 unsigned long *rmapp;
a0ed4607 1250
5dc99b23 1251 while (mask) {
65fbe37c
TY
1252 rmapp = __gfn_to_rmap(slot->base_gfn + gfn_offset + __ffs(mask),
1253 PT_PAGE_TABLE_LEVEL, slot);
245c3912 1254 __rmap_write_protect(kvm, rmapp, false);
05da4558 1255
5dc99b23
TY
1256 /* clear the first set bit */
1257 mask &= mask - 1;
1258 }
374cbac0
AK
1259}
1260
2f84569f 1261static bool rmap_write_protect(struct kvm *kvm, u64 gfn)
95d4c16c
TY
1262{
1263 struct kvm_memory_slot *slot;
5dc99b23
TY
1264 unsigned long *rmapp;
1265 int i;
2f84569f 1266 bool write_protected = false;
95d4c16c
TY
1267
1268 slot = gfn_to_memslot(kvm, gfn);
5dc99b23
TY
1269
1270 for (i = PT_PAGE_TABLE_LEVEL;
1271 i < PT_PAGE_TABLE_LEVEL + KVM_NR_PAGE_SIZES; ++i) {
1272 rmapp = __gfn_to_rmap(gfn, i, slot);
245c3912 1273 write_protected |= __rmap_write_protect(kvm, rmapp, true);
5dc99b23
TY
1274 }
1275
1276 return write_protected;
95d4c16c
TY
1277}
1278
8a8365c5 1279static int kvm_unmap_rmapp(struct kvm *kvm, unsigned long *rmapp,
048212d0 1280 struct kvm_memory_slot *slot, unsigned long data)
e930bffe 1281{
1e3f42f0
TY
1282 u64 *sptep;
1283 struct rmap_iterator iter;
e930bffe
AA
1284 int need_tlb_flush = 0;
1285
1e3f42f0
TY
1286 while ((sptep = rmap_get_first(*rmapp, &iter))) {
1287 BUG_ON(!(*sptep & PT_PRESENT_MASK));
1288 rmap_printk("kvm_rmap_unmap_hva: spte %p %llx\n", sptep, *sptep);
1289
1290 drop_spte(kvm, sptep);
e930bffe
AA
1291 need_tlb_flush = 1;
1292 }
1e3f42f0 1293
e930bffe
AA
1294 return need_tlb_flush;
1295}
1296
8a8365c5 1297static int kvm_set_pte_rmapp(struct kvm *kvm, unsigned long *rmapp,
048212d0 1298 struct kvm_memory_slot *slot, unsigned long data)
3da0dd43 1299{
1e3f42f0
TY
1300 u64 *sptep;
1301 struct rmap_iterator iter;
3da0dd43 1302 int need_flush = 0;
1e3f42f0 1303 u64 new_spte;
3da0dd43
IE
1304 pte_t *ptep = (pte_t *)data;
1305 pfn_t new_pfn;
1306
1307 WARN_ON(pte_huge(*ptep));
1308 new_pfn = pte_pfn(*ptep);
1e3f42f0
TY
1309
1310 for (sptep = rmap_get_first(*rmapp, &iter); sptep;) {
1311 BUG_ON(!is_shadow_present_pte(*sptep));
1312 rmap_printk("kvm_set_pte_rmapp: spte %p %llx\n", sptep, *sptep);
1313
3da0dd43 1314 need_flush = 1;
1e3f42f0 1315
3da0dd43 1316 if (pte_write(*ptep)) {
1e3f42f0
TY
1317 drop_spte(kvm, sptep);
1318 sptep = rmap_get_first(*rmapp, &iter);
3da0dd43 1319 } else {
1e3f42f0 1320 new_spte = *sptep & ~PT64_BASE_ADDR_MASK;
3da0dd43
IE
1321 new_spte |= (u64)new_pfn << PAGE_SHIFT;
1322
1323 new_spte &= ~PT_WRITABLE_MASK;
1324 new_spte &= ~SPTE_HOST_WRITEABLE;
b79b93f9 1325 new_spte &= ~shadow_accessed_mask;
1e3f42f0
TY
1326
1327 mmu_spte_clear_track_bits(sptep);
1328 mmu_spte_set(sptep, new_spte);
1329 sptep = rmap_get_next(&iter);
3da0dd43
IE
1330 }
1331 }
1e3f42f0 1332
3da0dd43
IE
1333 if (need_flush)
1334 kvm_flush_remote_tlbs(kvm);
1335
1336 return 0;
1337}
1338
84504ef3
TY
1339static int kvm_handle_hva_range(struct kvm *kvm,
1340 unsigned long start,
1341 unsigned long end,
1342 unsigned long data,
1343 int (*handler)(struct kvm *kvm,
1344 unsigned long *rmapp,
048212d0 1345 struct kvm_memory_slot *slot,
84504ef3 1346 unsigned long data))
e930bffe 1347{
be6ba0f0 1348 int j;
f395302e 1349 int ret = 0;
bc6678a3 1350 struct kvm_memslots *slots;
be6ba0f0 1351 struct kvm_memory_slot *memslot;
bc6678a3 1352
90d83dc3 1353 slots = kvm_memslots(kvm);
e930bffe 1354
be6ba0f0 1355 kvm_for_each_memslot(memslot, slots) {
84504ef3 1356 unsigned long hva_start, hva_end;
bcd3ef58 1357 gfn_t gfn_start, gfn_end;
e930bffe 1358
84504ef3
TY
1359 hva_start = max(start, memslot->userspace_addr);
1360 hva_end = min(end, memslot->userspace_addr +
1361 (memslot->npages << PAGE_SHIFT));
1362 if (hva_start >= hva_end)
1363 continue;
1364 /*
1365 * {gfn(page) | page intersects with [hva_start, hva_end)} =
bcd3ef58 1366 * {gfn_start, gfn_start+1, ..., gfn_end-1}.
84504ef3 1367 */
bcd3ef58 1368 gfn_start = hva_to_gfn_memslot(hva_start, memslot);
84504ef3 1369 gfn_end = hva_to_gfn_memslot(hva_end + PAGE_SIZE - 1, memslot);
852e3c19 1370
bcd3ef58
TY
1371 for (j = PT_PAGE_TABLE_LEVEL;
1372 j < PT_PAGE_TABLE_LEVEL + KVM_NR_PAGE_SIZES; ++j) {
1373 unsigned long idx, idx_end;
1374 unsigned long *rmapp;
d4dbf470 1375
bcd3ef58
TY
1376 /*
1377 * {idx(page_j) | page_j intersects with
1378 * [hva_start, hva_end)} = {idx, idx+1, ..., idx_end}.
1379 */
1380 idx = gfn_to_index(gfn_start, memslot->base_gfn, j);
1381 idx_end = gfn_to_index(gfn_end - 1, memslot->base_gfn, j);
852e3c19 1382
bcd3ef58 1383 rmapp = __gfn_to_rmap(gfn_start, j, memslot);
d4dbf470 1384
bcd3ef58
TY
1385 for (; idx <= idx_end; ++idx)
1386 ret |= handler(kvm, rmapp++, memslot, data);
e930bffe
AA
1387 }
1388 }
1389
f395302e 1390 return ret;
e930bffe
AA
1391}
1392
84504ef3
TY
1393static int kvm_handle_hva(struct kvm *kvm, unsigned long hva,
1394 unsigned long data,
1395 int (*handler)(struct kvm *kvm, unsigned long *rmapp,
048212d0 1396 struct kvm_memory_slot *slot,
84504ef3
TY
1397 unsigned long data))
1398{
1399 return kvm_handle_hva_range(kvm, hva, hva + 1, data, handler);
e930bffe
AA
1400}
1401
1402int kvm_unmap_hva(struct kvm *kvm, unsigned long hva)
1403{
3da0dd43
IE
1404 return kvm_handle_hva(kvm, hva, 0, kvm_unmap_rmapp);
1405}
1406
b3ae2096
TY
1407int kvm_unmap_hva_range(struct kvm *kvm, unsigned long start, unsigned long end)
1408{
1409 return kvm_handle_hva_range(kvm, start, end, 0, kvm_unmap_rmapp);
1410}
1411
3da0dd43
IE
1412void kvm_set_spte_hva(struct kvm *kvm, unsigned long hva, pte_t pte)
1413{
8a8365c5 1414 kvm_handle_hva(kvm, hva, (unsigned long)&pte, kvm_set_pte_rmapp);
e930bffe
AA
1415}
1416
8a8365c5 1417static int kvm_age_rmapp(struct kvm *kvm, unsigned long *rmapp,
048212d0 1418 struct kvm_memory_slot *slot, unsigned long data)
e930bffe 1419{
1e3f42f0 1420 u64 *sptep;
79f702a6 1421 struct rmap_iterator uninitialized_var(iter);
e930bffe
AA
1422 int young = 0;
1423
6316e1c8 1424 /*
3f6d8c8a
XH
1425 * In case of absence of EPT Access and Dirty Bits supports,
1426 * emulate the accessed bit for EPT, by checking if this page has
6316e1c8
RR
1427 * an EPT mapping, and clearing it if it does. On the next access,
1428 * a new EPT mapping will be established.
1429 * This has some overhead, but not as much as the cost of swapping
1430 * out actively used pages or breaking up actively used hugepages.
1431 */
f395302e
TY
1432 if (!shadow_accessed_mask) {
1433 young = kvm_unmap_rmapp(kvm, rmapp, slot, data);
1434 goto out;
1435 }
534e38b4 1436
1e3f42f0
TY
1437 for (sptep = rmap_get_first(*rmapp, &iter); sptep;
1438 sptep = rmap_get_next(&iter)) {
3f6d8c8a 1439 BUG_ON(!is_shadow_present_pte(*sptep));
1e3f42f0 1440
3f6d8c8a 1441 if (*sptep & shadow_accessed_mask) {
e930bffe 1442 young = 1;
3f6d8c8a
XH
1443 clear_bit((ffs(shadow_accessed_mask) - 1),
1444 (unsigned long *)sptep);
e930bffe 1445 }
e930bffe 1446 }
f395302e
TY
1447out:
1448 /* @data has hva passed to kvm_age_hva(). */
1449 trace_kvm_age_page(data, slot, young);
e930bffe
AA
1450 return young;
1451}
1452
8ee53820 1453static int kvm_test_age_rmapp(struct kvm *kvm, unsigned long *rmapp,
048212d0 1454 struct kvm_memory_slot *slot, unsigned long data)
8ee53820 1455{
1e3f42f0
TY
1456 u64 *sptep;
1457 struct rmap_iterator iter;
8ee53820
AA
1458 int young = 0;
1459
1460 /*
1461 * If there's no access bit in the secondary pte set by the
1462 * hardware it's up to gup-fast/gup to set the access bit in
1463 * the primary pte or in the page structure.
1464 */
1465 if (!shadow_accessed_mask)
1466 goto out;
1467
1e3f42f0
TY
1468 for (sptep = rmap_get_first(*rmapp, &iter); sptep;
1469 sptep = rmap_get_next(&iter)) {
3f6d8c8a 1470 BUG_ON(!is_shadow_present_pte(*sptep));
1e3f42f0 1471
3f6d8c8a 1472 if (*sptep & shadow_accessed_mask) {
8ee53820
AA
1473 young = 1;
1474 break;
1475 }
8ee53820
AA
1476 }
1477out:
1478 return young;
1479}
1480
53a27b39
MT
1481#define RMAP_RECYCLE_THRESHOLD 1000
1482
852e3c19 1483static void rmap_recycle(struct kvm_vcpu *vcpu, u64 *spte, gfn_t gfn)
53a27b39
MT
1484{
1485 unsigned long *rmapp;
852e3c19
JR
1486 struct kvm_mmu_page *sp;
1487
1488 sp = page_header(__pa(spte));
53a27b39 1489
852e3c19 1490 rmapp = gfn_to_rmap(vcpu->kvm, gfn, sp->role.level);
53a27b39 1491
048212d0 1492 kvm_unmap_rmapp(vcpu->kvm, rmapp, NULL, 0);
53a27b39
MT
1493 kvm_flush_remote_tlbs(vcpu->kvm);
1494}
1495
e930bffe
AA
1496int kvm_age_hva(struct kvm *kvm, unsigned long hva)
1497{
f395302e 1498 return kvm_handle_hva(kvm, hva, hva, kvm_age_rmapp);
e930bffe
AA
1499}
1500
8ee53820
AA
1501int kvm_test_age_hva(struct kvm *kvm, unsigned long hva)
1502{
1503 return kvm_handle_hva(kvm, hva, 0, kvm_test_age_rmapp);
1504}
1505
d6c69ee9 1506#ifdef MMU_DEBUG
47ad8e68 1507static int is_empty_shadow_page(u64 *spt)
6aa8b732 1508{
139bdb2d
AK
1509 u64 *pos;
1510 u64 *end;
1511
47ad8e68 1512 for (pos = spt, end = pos + PAGE_SIZE / sizeof(u64); pos != end; pos++)
3c915510 1513 if (is_shadow_present_pte(*pos)) {
b8688d51 1514 printk(KERN_ERR "%s: %p %llx\n", __func__,
139bdb2d 1515 pos, *pos);
6aa8b732 1516 return 0;
139bdb2d 1517 }
6aa8b732
AK
1518 return 1;
1519}
d6c69ee9 1520#endif
6aa8b732 1521
45221ab6
DH
1522/*
1523 * This value is the sum of all of the kvm instances's
1524 * kvm->arch.n_used_mmu_pages values. We need a global,
1525 * aggregate version in order to make the slab shrinker
1526 * faster
1527 */
1528static inline void kvm_mod_used_mmu_pages(struct kvm *kvm, int nr)
1529{
1530 kvm->arch.n_used_mmu_pages += nr;
1531 percpu_counter_add(&kvm_total_used_mmu_pages, nr);
1532}
1533
834be0d8 1534static void kvm_mmu_free_page(struct kvm_mmu_page *sp)
260746c0 1535{
4db35314 1536 ASSERT(is_empty_shadow_page(sp->spt));
7775834a 1537 hlist_del(&sp->hash_link);
bd4c86ea
XG
1538 list_del(&sp->link);
1539 free_page((unsigned long)sp->spt);
834be0d8
GN
1540 if (!sp->role.direct)
1541 free_page((unsigned long)sp->gfns);
e8ad9a70 1542 kmem_cache_free(mmu_page_header_cache, sp);
260746c0
AK
1543}
1544
cea0f0e7
AK
1545static unsigned kvm_page_table_hashfn(gfn_t gfn)
1546{
1ae0a13d 1547 return gfn & ((1 << KVM_MMU_HASH_SHIFT) - 1);
cea0f0e7
AK
1548}
1549
714b93da 1550static void mmu_page_add_parent_pte(struct kvm_vcpu *vcpu,
4db35314 1551 struct kvm_mmu_page *sp, u64 *parent_pte)
cea0f0e7 1552{
cea0f0e7
AK
1553 if (!parent_pte)
1554 return;
cea0f0e7 1555
67052b35 1556 pte_list_add(vcpu, parent_pte, &sp->parent_ptes);
cea0f0e7
AK
1557}
1558
4db35314 1559static void mmu_page_remove_parent_pte(struct kvm_mmu_page *sp,
cea0f0e7
AK
1560 u64 *parent_pte)
1561{
67052b35 1562 pte_list_remove(parent_pte, &sp->parent_ptes);
cea0f0e7
AK
1563}
1564
bcdd9a93
XG
1565static void drop_parent_pte(struct kvm_mmu_page *sp,
1566 u64 *parent_pte)
1567{
1568 mmu_page_remove_parent_pte(sp, parent_pte);
1df9f2dc 1569 mmu_spte_clear_no_track(parent_pte);
bcdd9a93
XG
1570}
1571
67052b35
XG
1572static struct kvm_mmu_page *kvm_mmu_alloc_page(struct kvm_vcpu *vcpu,
1573 u64 *parent_pte, int direct)
ad8cfbe3 1574{
67052b35 1575 struct kvm_mmu_page *sp;
7ddca7e4 1576
80feb89a
TY
1577 sp = mmu_memory_cache_alloc(&vcpu->arch.mmu_page_header_cache);
1578 sp->spt = mmu_memory_cache_alloc(&vcpu->arch.mmu_page_cache);
67052b35 1579 if (!direct)
80feb89a 1580 sp->gfns = mmu_memory_cache_alloc(&vcpu->arch.mmu_page_cache);
67052b35 1581 set_page_private(virt_to_page(sp->spt), (unsigned long)sp);
5304b8d3
XG
1582
1583 /*
1584 * The active_mmu_pages list is the FIFO list, do not move the
1585 * page until it is zapped. kvm_zap_obsolete_pages depends on
1586 * this feature. See the comments in kvm_zap_obsolete_pages().
1587 */
67052b35 1588 list_add(&sp->link, &vcpu->kvm->arch.active_mmu_pages);
67052b35
XG
1589 sp->parent_ptes = 0;
1590 mmu_page_add_parent_pte(vcpu, sp, parent_pte);
1591 kvm_mod_used_mmu_pages(vcpu->kvm, +1);
1592 return sp;
ad8cfbe3
MT
1593}
1594
67052b35 1595static void mark_unsync(u64 *spte);
1047df1f 1596static void kvm_mmu_mark_parents_unsync(struct kvm_mmu_page *sp)
0074ff63 1597{
67052b35 1598 pte_list_walk(&sp->parent_ptes, mark_unsync);
0074ff63
MT
1599}
1600
67052b35 1601static void mark_unsync(u64 *spte)
0074ff63 1602{
67052b35 1603 struct kvm_mmu_page *sp;
1047df1f 1604 unsigned int index;
0074ff63 1605
67052b35 1606 sp = page_header(__pa(spte));
1047df1f
XG
1607 index = spte - sp->spt;
1608 if (__test_and_set_bit(index, sp->unsync_child_bitmap))
0074ff63 1609 return;
1047df1f 1610 if (sp->unsync_children++)
0074ff63 1611 return;
1047df1f 1612 kvm_mmu_mark_parents_unsync(sp);
0074ff63
MT
1613}
1614
e8bc217a 1615static int nonpaging_sync_page(struct kvm_vcpu *vcpu,
a4a8e6f7 1616 struct kvm_mmu_page *sp)
e8bc217a
MT
1617{
1618 return 1;
1619}
1620
a7052897
MT
1621static void nonpaging_invlpg(struct kvm_vcpu *vcpu, gva_t gva)
1622{
1623}
1624
0f53b5b1
XG
1625static void nonpaging_update_pte(struct kvm_vcpu *vcpu,
1626 struct kvm_mmu_page *sp, u64 *spte,
7c562522 1627 const void *pte)
0f53b5b1
XG
1628{
1629 WARN_ON(1);
1630}
1631
60c8aec6
MT
1632#define KVM_PAGE_ARRAY_NR 16
1633
1634struct kvm_mmu_pages {
1635 struct mmu_page_and_offset {
1636 struct kvm_mmu_page *sp;
1637 unsigned int idx;
1638 } page[KVM_PAGE_ARRAY_NR];
1639 unsigned int nr;
1640};
1641
cded19f3
HE
1642static int mmu_pages_add(struct kvm_mmu_pages *pvec, struct kvm_mmu_page *sp,
1643 int idx)
4731d4c7 1644{
60c8aec6 1645 int i;
4731d4c7 1646
60c8aec6
MT
1647 if (sp->unsync)
1648 for (i=0; i < pvec->nr; i++)
1649 if (pvec->page[i].sp == sp)
1650 return 0;
1651
1652 pvec->page[pvec->nr].sp = sp;
1653 pvec->page[pvec->nr].idx = idx;
1654 pvec->nr++;
1655 return (pvec->nr == KVM_PAGE_ARRAY_NR);
1656}
1657
1658static int __mmu_unsync_walk(struct kvm_mmu_page *sp,
1659 struct kvm_mmu_pages *pvec)
1660{
1661 int i, ret, nr_unsync_leaf = 0;
4731d4c7 1662
37178b8b 1663 for_each_set_bit(i, sp->unsync_child_bitmap, 512) {
7a8f1a74 1664 struct kvm_mmu_page *child;
4731d4c7
MT
1665 u64 ent = sp->spt[i];
1666
7a8f1a74
XG
1667 if (!is_shadow_present_pte(ent) || is_large_pte(ent))
1668 goto clear_child_bitmap;
1669
1670 child = page_header(ent & PT64_BASE_ADDR_MASK);
1671
1672 if (child->unsync_children) {
1673 if (mmu_pages_add(pvec, child, i))
1674 return -ENOSPC;
1675
1676 ret = __mmu_unsync_walk(child, pvec);
1677 if (!ret)
1678 goto clear_child_bitmap;
1679 else if (ret > 0)
1680 nr_unsync_leaf += ret;
1681 else
1682 return ret;
1683 } else if (child->unsync) {
1684 nr_unsync_leaf++;
1685 if (mmu_pages_add(pvec, child, i))
1686 return -ENOSPC;
1687 } else
1688 goto clear_child_bitmap;
1689
1690 continue;
1691
1692clear_child_bitmap:
1693 __clear_bit(i, sp->unsync_child_bitmap);
1694 sp->unsync_children--;
1695 WARN_ON((int)sp->unsync_children < 0);
4731d4c7
MT
1696 }
1697
4731d4c7 1698
60c8aec6
MT
1699 return nr_unsync_leaf;
1700}
1701
1702static int mmu_unsync_walk(struct kvm_mmu_page *sp,
1703 struct kvm_mmu_pages *pvec)
1704{
1705 if (!sp->unsync_children)
1706 return 0;
1707
1708 mmu_pages_add(pvec, sp, 0);
1709 return __mmu_unsync_walk(sp, pvec);
4731d4c7
MT
1710}
1711
4731d4c7
MT
1712static void kvm_unlink_unsync_page(struct kvm *kvm, struct kvm_mmu_page *sp)
1713{
1714 WARN_ON(!sp->unsync);
5e1b3ddb 1715 trace_kvm_mmu_sync_page(sp);
4731d4c7
MT
1716 sp->unsync = 0;
1717 --kvm->stat.mmu_unsync;
1718}
1719
7775834a
XG
1720static int kvm_mmu_prepare_zap_page(struct kvm *kvm, struct kvm_mmu_page *sp,
1721 struct list_head *invalid_list);
1722static void kvm_mmu_commit_zap_page(struct kvm *kvm,
1723 struct list_head *invalid_list);
4731d4c7 1724
f34d251d
XG
1725/*
1726 * NOTE: we should pay more attention on the zapped-obsolete page
1727 * (is_obsolete_sp(sp) && sp->role.invalid) when you do hash list walk
1728 * since it has been deleted from active_mmu_pages but still can be found
1729 * at hast list.
1730 *
1731 * for_each_gfn_indirect_valid_sp has skipped that kind of page and
1732 * kvm_mmu_get_page(), the only user of for_each_gfn_sp(), has skipped
1733 * all the obsolete pages.
1734 */
1044b030
TY
1735#define for_each_gfn_sp(_kvm, _sp, _gfn) \
1736 hlist_for_each_entry(_sp, \
1737 &(_kvm)->arch.mmu_page_hash[kvm_page_table_hashfn(_gfn)], hash_link) \
1738 if ((_sp)->gfn != (_gfn)) {} else
1739
1740#define for_each_gfn_indirect_valid_sp(_kvm, _sp, _gfn) \
1741 for_each_gfn_sp(_kvm, _sp, _gfn) \
1742 if ((_sp)->role.direct || (_sp)->role.invalid) {} else
7ae680eb 1743
f918b443 1744/* @sp->gfn should be write-protected at the call site */
1d9dc7e0 1745static int __kvm_sync_page(struct kvm_vcpu *vcpu, struct kvm_mmu_page *sp,
d98ba053 1746 struct list_head *invalid_list, bool clear_unsync)
4731d4c7 1747{
5b7e0102 1748 if (sp->role.cr4_pae != !!is_pae(vcpu)) {
d98ba053 1749 kvm_mmu_prepare_zap_page(vcpu->kvm, sp, invalid_list);
4731d4c7
MT
1750 return 1;
1751 }
1752
f918b443 1753 if (clear_unsync)
1d9dc7e0 1754 kvm_unlink_unsync_page(vcpu->kvm, sp);
1d9dc7e0 1755
a4a8e6f7 1756 if (vcpu->arch.mmu.sync_page(vcpu, sp)) {
d98ba053 1757 kvm_mmu_prepare_zap_page(vcpu->kvm, sp, invalid_list);
4731d4c7
MT
1758 return 1;
1759 }
1760
1761 kvm_mmu_flush_tlb(vcpu);
4731d4c7
MT
1762 return 0;
1763}
1764
1d9dc7e0
XG
1765static int kvm_sync_page_transient(struct kvm_vcpu *vcpu,
1766 struct kvm_mmu_page *sp)
1767{
d98ba053 1768 LIST_HEAD(invalid_list);
1d9dc7e0
XG
1769 int ret;
1770
d98ba053 1771 ret = __kvm_sync_page(vcpu, sp, &invalid_list, false);
be71e061 1772 if (ret)
d98ba053
XG
1773 kvm_mmu_commit_zap_page(vcpu->kvm, &invalid_list);
1774
1d9dc7e0
XG
1775 return ret;
1776}
1777
e37fa785
XG
1778#ifdef CONFIG_KVM_MMU_AUDIT
1779#include "mmu_audit.c"
1780#else
1781static void kvm_mmu_audit(struct kvm_vcpu *vcpu, int point) { }
1782static void mmu_audit_disable(void) { }
1783#endif
1784
d98ba053
XG
1785static int kvm_sync_page(struct kvm_vcpu *vcpu, struct kvm_mmu_page *sp,
1786 struct list_head *invalid_list)
1d9dc7e0 1787{
d98ba053 1788 return __kvm_sync_page(vcpu, sp, invalid_list, true);
1d9dc7e0
XG
1789}
1790
9f1a122f
XG
1791/* @gfn should be write-protected at the call site */
1792static void kvm_sync_pages(struct kvm_vcpu *vcpu, gfn_t gfn)
1793{
9f1a122f 1794 struct kvm_mmu_page *s;
d98ba053 1795 LIST_HEAD(invalid_list);
9f1a122f
XG
1796 bool flush = false;
1797
b67bfe0d 1798 for_each_gfn_indirect_valid_sp(vcpu->kvm, s, gfn) {
7ae680eb 1799 if (!s->unsync)
9f1a122f
XG
1800 continue;
1801
1802 WARN_ON(s->role.level != PT_PAGE_TABLE_LEVEL);
a4a8e6f7 1803 kvm_unlink_unsync_page(vcpu->kvm, s);
9f1a122f 1804 if ((s->role.cr4_pae != !!is_pae(vcpu)) ||
a4a8e6f7 1805 (vcpu->arch.mmu.sync_page(vcpu, s))) {
d98ba053 1806 kvm_mmu_prepare_zap_page(vcpu->kvm, s, &invalid_list);
9f1a122f
XG
1807 continue;
1808 }
9f1a122f
XG
1809 flush = true;
1810 }
1811
d98ba053 1812 kvm_mmu_commit_zap_page(vcpu->kvm, &invalid_list);
9f1a122f
XG
1813 if (flush)
1814 kvm_mmu_flush_tlb(vcpu);
1815}
1816
60c8aec6
MT
1817struct mmu_page_path {
1818 struct kvm_mmu_page *parent[PT64_ROOT_LEVEL-1];
1819 unsigned int idx[PT64_ROOT_LEVEL-1];
4731d4c7
MT
1820};
1821
60c8aec6
MT
1822#define for_each_sp(pvec, sp, parents, i) \
1823 for (i = mmu_pages_next(&pvec, &parents, -1), \
1824 sp = pvec.page[i].sp; \
1825 i < pvec.nr && ({ sp = pvec.page[i].sp; 1;}); \
1826 i = mmu_pages_next(&pvec, &parents, i))
1827
cded19f3
HE
1828static int mmu_pages_next(struct kvm_mmu_pages *pvec,
1829 struct mmu_page_path *parents,
1830 int i)
60c8aec6
MT
1831{
1832 int n;
1833
1834 for (n = i+1; n < pvec->nr; n++) {
1835 struct kvm_mmu_page *sp = pvec->page[n].sp;
1836
1837 if (sp->role.level == PT_PAGE_TABLE_LEVEL) {
1838 parents->idx[0] = pvec->page[n].idx;
1839 return n;
1840 }
1841
1842 parents->parent[sp->role.level-2] = sp;
1843 parents->idx[sp->role.level-1] = pvec->page[n].idx;
1844 }
1845
1846 return n;
1847}
1848
cded19f3 1849static void mmu_pages_clear_parents(struct mmu_page_path *parents)
4731d4c7 1850{
60c8aec6
MT
1851 struct kvm_mmu_page *sp;
1852 unsigned int level = 0;
1853
1854 do {
1855 unsigned int idx = parents->idx[level];
4731d4c7 1856
60c8aec6
MT
1857 sp = parents->parent[level];
1858 if (!sp)
1859 return;
1860
1861 --sp->unsync_children;
1862 WARN_ON((int)sp->unsync_children < 0);
1863 __clear_bit(idx, sp->unsync_child_bitmap);
1864 level++;
1865 } while (level < PT64_ROOT_LEVEL-1 && !sp->unsync_children);
4731d4c7
MT
1866}
1867
60c8aec6
MT
1868static void kvm_mmu_pages_init(struct kvm_mmu_page *parent,
1869 struct mmu_page_path *parents,
1870 struct kvm_mmu_pages *pvec)
4731d4c7 1871{
60c8aec6
MT
1872 parents->parent[parent->role.level-1] = NULL;
1873 pvec->nr = 0;
1874}
4731d4c7 1875
60c8aec6
MT
1876static void mmu_sync_children(struct kvm_vcpu *vcpu,
1877 struct kvm_mmu_page *parent)
1878{
1879 int i;
1880 struct kvm_mmu_page *sp;
1881 struct mmu_page_path parents;
1882 struct kvm_mmu_pages pages;
d98ba053 1883 LIST_HEAD(invalid_list);
60c8aec6
MT
1884
1885 kvm_mmu_pages_init(parent, &parents, &pages);
1886 while (mmu_unsync_walk(parent, &pages)) {
2f84569f 1887 bool protected = false;
b1a36821
MT
1888
1889 for_each_sp(pages, sp, parents, i)
1890 protected |= rmap_write_protect(vcpu->kvm, sp->gfn);
1891
1892 if (protected)
1893 kvm_flush_remote_tlbs(vcpu->kvm);
1894
60c8aec6 1895 for_each_sp(pages, sp, parents, i) {
d98ba053 1896 kvm_sync_page(vcpu, sp, &invalid_list);
60c8aec6
MT
1897 mmu_pages_clear_parents(&parents);
1898 }
d98ba053 1899 kvm_mmu_commit_zap_page(vcpu->kvm, &invalid_list);
4731d4c7 1900 cond_resched_lock(&vcpu->kvm->mmu_lock);
60c8aec6
MT
1901 kvm_mmu_pages_init(parent, &parents, &pages);
1902 }
4731d4c7
MT
1903}
1904
c3707958
XG
1905static void init_shadow_page_table(struct kvm_mmu_page *sp)
1906{
1907 int i;
1908
1909 for (i = 0; i < PT64_ENT_PER_PAGE; ++i)
1910 sp->spt[i] = 0ull;
1911}
1912
a30f47cb
XG
1913static void __clear_sp_write_flooding_count(struct kvm_mmu_page *sp)
1914{
1915 sp->write_flooding_count = 0;
1916}
1917
1918static void clear_sp_write_flooding_count(u64 *spte)
1919{
1920 struct kvm_mmu_page *sp = page_header(__pa(spte));
1921
1922 __clear_sp_write_flooding_count(sp);
1923}
1924
5304b8d3
XG
1925static bool is_obsolete_sp(struct kvm *kvm, struct kvm_mmu_page *sp)
1926{
1927 return unlikely(sp->mmu_valid_gen != kvm->arch.mmu_valid_gen);
1928}
1929
cea0f0e7
AK
1930static struct kvm_mmu_page *kvm_mmu_get_page(struct kvm_vcpu *vcpu,
1931 gfn_t gfn,
1932 gva_t gaddr,
1933 unsigned level,
f6e2c02b 1934 int direct,
41074d07 1935 unsigned access,
f7d9c7b7 1936 u64 *parent_pte)
cea0f0e7
AK
1937{
1938 union kvm_mmu_page_role role;
cea0f0e7 1939 unsigned quadrant;
9f1a122f 1940 struct kvm_mmu_page *sp;
9f1a122f 1941 bool need_sync = false;
cea0f0e7 1942
a770f6f2 1943 role = vcpu->arch.mmu.base_role;
cea0f0e7 1944 role.level = level;
f6e2c02b 1945 role.direct = direct;
84b0c8c6 1946 if (role.direct)
5b7e0102 1947 role.cr4_pae = 0;
41074d07 1948 role.access = access;
c5a78f2b
JR
1949 if (!vcpu->arch.mmu.direct_map
1950 && vcpu->arch.mmu.root_level <= PT32_ROOT_LEVEL) {
cea0f0e7
AK
1951 quadrant = gaddr >> (PAGE_SHIFT + (PT64_PT_BITS * level));
1952 quadrant &= (1 << ((PT32_PT_BITS - PT64_PT_BITS) * level)) - 1;
1953 role.quadrant = quadrant;
1954 }
b67bfe0d 1955 for_each_gfn_sp(vcpu->kvm, sp, gfn) {
7f52af74
XG
1956 if (is_obsolete_sp(vcpu->kvm, sp))
1957 continue;
1958
7ae680eb
XG
1959 if (!need_sync && sp->unsync)
1960 need_sync = true;
4731d4c7 1961
7ae680eb
XG
1962 if (sp->role.word != role.word)
1963 continue;
4731d4c7 1964
7ae680eb
XG
1965 if (sp->unsync && kvm_sync_page_transient(vcpu, sp))
1966 break;
e02aa901 1967
7ae680eb
XG
1968 mmu_page_add_parent_pte(vcpu, sp, parent_pte);
1969 if (sp->unsync_children) {
a8eeb04a 1970 kvm_make_request(KVM_REQ_MMU_SYNC, vcpu);
7ae680eb
XG
1971 kvm_mmu_mark_parents_unsync(sp);
1972 } else if (sp->unsync)
1973 kvm_mmu_mark_parents_unsync(sp);
e02aa901 1974
a30f47cb 1975 __clear_sp_write_flooding_count(sp);
7ae680eb
XG
1976 trace_kvm_mmu_get_page(sp, false);
1977 return sp;
1978 }
dfc5aa00 1979 ++vcpu->kvm->stat.mmu_cache_miss;
2032a93d 1980 sp = kvm_mmu_alloc_page(vcpu, parent_pte, direct);
4db35314
AK
1981 if (!sp)
1982 return sp;
4db35314
AK
1983 sp->gfn = gfn;
1984 sp->role = role;
7ae680eb
XG
1985 hlist_add_head(&sp->hash_link,
1986 &vcpu->kvm->arch.mmu_page_hash[kvm_page_table_hashfn(gfn)]);
f6e2c02b 1987 if (!direct) {
b1a36821
MT
1988 if (rmap_write_protect(vcpu->kvm, gfn))
1989 kvm_flush_remote_tlbs(vcpu->kvm);
9f1a122f
XG
1990 if (level > PT_PAGE_TABLE_LEVEL && need_sync)
1991 kvm_sync_pages(vcpu, gfn);
1992
4731d4c7
MT
1993 account_shadowed(vcpu->kvm, gfn);
1994 }
5304b8d3 1995 sp->mmu_valid_gen = vcpu->kvm->arch.mmu_valid_gen;
c3707958 1996 init_shadow_page_table(sp);
f691fe1d 1997 trace_kvm_mmu_get_page(sp, true);
4db35314 1998 return sp;
cea0f0e7
AK
1999}
2000
2d11123a
AK
2001static void shadow_walk_init(struct kvm_shadow_walk_iterator *iterator,
2002 struct kvm_vcpu *vcpu, u64 addr)
2003{
2004 iterator->addr = addr;
2005 iterator->shadow_addr = vcpu->arch.mmu.root_hpa;
2006 iterator->level = vcpu->arch.mmu.shadow_root_level;
81407ca5
JR
2007
2008 if (iterator->level == PT64_ROOT_LEVEL &&
2009 vcpu->arch.mmu.root_level < PT64_ROOT_LEVEL &&
2010 !vcpu->arch.mmu.direct_map)
2011 --iterator->level;
2012
2d11123a
AK
2013 if (iterator->level == PT32E_ROOT_LEVEL) {
2014 iterator->shadow_addr
2015 = vcpu->arch.mmu.pae_root[(addr >> 30) & 3];
2016 iterator->shadow_addr &= PT64_BASE_ADDR_MASK;
2017 --iterator->level;
2018 if (!iterator->shadow_addr)
2019 iterator->level = 0;
2020 }
2021}
2022
2023static bool shadow_walk_okay(struct kvm_shadow_walk_iterator *iterator)
2024{
2025 if (iterator->level < PT_PAGE_TABLE_LEVEL)
2026 return false;
4d88954d 2027
2d11123a
AK
2028 iterator->index = SHADOW_PT_INDEX(iterator->addr, iterator->level);
2029 iterator->sptep = ((u64 *)__va(iterator->shadow_addr)) + iterator->index;
2030 return true;
2031}
2032
c2a2ac2b
XG
2033static void __shadow_walk_next(struct kvm_shadow_walk_iterator *iterator,
2034 u64 spte)
2d11123a 2035{
c2a2ac2b 2036 if (is_last_spte(spte, iterator->level)) {
052331be
XG
2037 iterator->level = 0;
2038 return;
2039 }
2040
c2a2ac2b 2041 iterator->shadow_addr = spte & PT64_BASE_ADDR_MASK;
2d11123a
AK
2042 --iterator->level;
2043}
2044
c2a2ac2b
XG
2045static void shadow_walk_next(struct kvm_shadow_walk_iterator *iterator)
2046{
2047 return __shadow_walk_next(iterator, *iterator->sptep);
2048}
2049
7a1638ce 2050static void link_shadow_page(u64 *sptep, struct kvm_mmu_page *sp, bool accessed)
32ef26a3
AK
2051{
2052 u64 spte;
2053
7a1638ce
YZ
2054 BUILD_BUG_ON(VMX_EPT_READABLE_MASK != PT_PRESENT_MASK ||
2055 VMX_EPT_WRITABLE_MASK != PT_WRITABLE_MASK);
2056
24db2734 2057 spte = __pa(sp->spt) | PT_PRESENT_MASK | PT_WRITABLE_MASK |
7a1638ce
YZ
2058 shadow_user_mask | shadow_x_mask;
2059
2060 if (accessed)
2061 spte |= shadow_accessed_mask;
24db2734 2062
1df9f2dc 2063 mmu_spte_set(sptep, spte);
32ef26a3
AK
2064}
2065
a357bd22
AK
2066static void validate_direct_spte(struct kvm_vcpu *vcpu, u64 *sptep,
2067 unsigned direct_access)
2068{
2069 if (is_shadow_present_pte(*sptep) && !is_large_pte(*sptep)) {
2070 struct kvm_mmu_page *child;
2071
2072 /*
2073 * For the direct sp, if the guest pte's dirty bit
2074 * changed form clean to dirty, it will corrupt the
2075 * sp's access: allow writable in the read-only sp,
2076 * so we should update the spte at this point to get
2077 * a new sp with the correct access.
2078 */
2079 child = page_header(*sptep & PT64_BASE_ADDR_MASK);
2080 if (child->role.access == direct_access)
2081 return;
2082
bcdd9a93 2083 drop_parent_pte(child, sptep);
a357bd22
AK
2084 kvm_flush_remote_tlbs(vcpu->kvm);
2085 }
2086}
2087
505aef8f 2088static bool mmu_page_zap_pte(struct kvm *kvm, struct kvm_mmu_page *sp,
38e3b2b2
XG
2089 u64 *spte)
2090{
2091 u64 pte;
2092 struct kvm_mmu_page *child;
2093
2094 pte = *spte;
2095 if (is_shadow_present_pte(pte)) {
505aef8f 2096 if (is_last_spte(pte, sp->role.level)) {
c3707958 2097 drop_spte(kvm, spte);
505aef8f
XG
2098 if (is_large_pte(pte))
2099 --kvm->stat.lpages;
2100 } else {
38e3b2b2 2101 child = page_header(pte & PT64_BASE_ADDR_MASK);
bcdd9a93 2102 drop_parent_pte(child, spte);
38e3b2b2 2103 }
505aef8f
XG
2104 return true;
2105 }
2106
2107 if (is_mmio_spte(pte))
ce88decf 2108 mmu_spte_clear_no_track(spte);
c3707958 2109
505aef8f 2110 return false;
38e3b2b2
XG
2111}
2112
90cb0529 2113static void kvm_mmu_page_unlink_children(struct kvm *kvm,
4db35314 2114 struct kvm_mmu_page *sp)
a436036b 2115{
697fe2e2 2116 unsigned i;
697fe2e2 2117
38e3b2b2
XG
2118 for (i = 0; i < PT64_ENT_PER_PAGE; ++i)
2119 mmu_page_zap_pte(kvm, sp, sp->spt + i);
a436036b
AK
2120}
2121
4db35314 2122static void kvm_mmu_put_page(struct kvm_mmu_page *sp, u64 *parent_pte)
cea0f0e7 2123{
4db35314 2124 mmu_page_remove_parent_pte(sp, parent_pte);
a436036b
AK
2125}
2126
31aa2b44 2127static void kvm_mmu_unlink_parents(struct kvm *kvm, struct kvm_mmu_page *sp)
a436036b 2128{
1e3f42f0
TY
2129 u64 *sptep;
2130 struct rmap_iterator iter;
a436036b 2131
1e3f42f0
TY
2132 while ((sptep = rmap_get_first(sp->parent_ptes, &iter)))
2133 drop_parent_pte(sp, sptep);
31aa2b44
AK
2134}
2135
60c8aec6 2136static int mmu_zap_unsync_children(struct kvm *kvm,
7775834a
XG
2137 struct kvm_mmu_page *parent,
2138 struct list_head *invalid_list)
4731d4c7 2139{
60c8aec6
MT
2140 int i, zapped = 0;
2141 struct mmu_page_path parents;
2142 struct kvm_mmu_pages pages;
4731d4c7 2143
60c8aec6 2144 if (parent->role.level == PT_PAGE_TABLE_LEVEL)
4731d4c7 2145 return 0;
60c8aec6
MT
2146
2147 kvm_mmu_pages_init(parent, &parents, &pages);
2148 while (mmu_unsync_walk(parent, &pages)) {
2149 struct kvm_mmu_page *sp;
2150
2151 for_each_sp(pages, sp, parents, i) {
7775834a 2152 kvm_mmu_prepare_zap_page(kvm, sp, invalid_list);
60c8aec6 2153 mmu_pages_clear_parents(&parents);
77662e00 2154 zapped++;
60c8aec6 2155 }
60c8aec6
MT
2156 kvm_mmu_pages_init(parent, &parents, &pages);
2157 }
2158
2159 return zapped;
4731d4c7
MT
2160}
2161
7775834a
XG
2162static int kvm_mmu_prepare_zap_page(struct kvm *kvm, struct kvm_mmu_page *sp,
2163 struct list_head *invalid_list)
31aa2b44 2164{
4731d4c7 2165 int ret;
f691fe1d 2166
7775834a 2167 trace_kvm_mmu_prepare_zap_page(sp);
31aa2b44 2168 ++kvm->stat.mmu_shadow_zapped;
7775834a 2169 ret = mmu_zap_unsync_children(kvm, sp, invalid_list);
4db35314 2170 kvm_mmu_page_unlink_children(kvm, sp);
31aa2b44 2171 kvm_mmu_unlink_parents(kvm, sp);
5304b8d3 2172
f6e2c02b 2173 if (!sp->role.invalid && !sp->role.direct)
5b5c6a5a 2174 unaccount_shadowed(kvm, sp->gfn);
5304b8d3 2175
4731d4c7
MT
2176 if (sp->unsync)
2177 kvm_unlink_unsync_page(kvm, sp);
4db35314 2178 if (!sp->root_count) {
54a4f023
GJ
2179 /* Count self */
2180 ret++;
7775834a 2181 list_move(&sp->link, invalid_list);
aa6bd187 2182 kvm_mod_used_mmu_pages(kvm, -1);
2e53d63a 2183 } else {
5b5c6a5a 2184 list_move(&sp->link, &kvm->arch.active_mmu_pages);
05988d72
GN
2185
2186 /*
2187 * The obsolete pages can not be used on any vcpus.
2188 * See the comments in kvm_mmu_invalidate_zap_all_pages().
2189 */
2190 if (!sp->role.invalid && !is_obsolete_sp(kvm, sp))
2191 kvm_reload_remote_mmus(kvm);
2e53d63a 2192 }
7775834a
XG
2193
2194 sp->role.invalid = 1;
4731d4c7 2195 return ret;
a436036b
AK
2196}
2197
7775834a
XG
2198static void kvm_mmu_commit_zap_page(struct kvm *kvm,
2199 struct list_head *invalid_list)
2200{
945315b9 2201 struct kvm_mmu_page *sp, *nsp;
7775834a
XG
2202
2203 if (list_empty(invalid_list))
2204 return;
2205
c142786c
AK
2206 /*
2207 * wmb: make sure everyone sees our modifications to the page tables
2208 * rmb: make sure we see changes to vcpu->mode
2209 */
2210 smp_mb();
4f022648 2211
c142786c
AK
2212 /*
2213 * Wait for all vcpus to exit guest mode and/or lockless shadow
2214 * page table walks.
2215 */
2216 kvm_flush_remote_tlbs(kvm);
c2a2ac2b 2217
945315b9 2218 list_for_each_entry_safe(sp, nsp, invalid_list, link) {
7775834a 2219 WARN_ON(!sp->role.invalid || sp->root_count);
aa6bd187 2220 kvm_mmu_free_page(sp);
945315b9 2221 }
7775834a
XG
2222}
2223
5da59607
TY
2224static bool prepare_zap_oldest_mmu_page(struct kvm *kvm,
2225 struct list_head *invalid_list)
2226{
2227 struct kvm_mmu_page *sp;
2228
2229 if (list_empty(&kvm->arch.active_mmu_pages))
2230 return false;
2231
2232 sp = list_entry(kvm->arch.active_mmu_pages.prev,
2233 struct kvm_mmu_page, link);
2234 kvm_mmu_prepare_zap_page(kvm, sp, invalid_list);
2235
2236 return true;
2237}
2238
82ce2c96
IE
2239/*
2240 * Changing the number of mmu pages allocated to the vm
49d5ca26 2241 * Note: if goal_nr_mmu_pages is too small, you will get dead lock
82ce2c96 2242 */
49d5ca26 2243void kvm_mmu_change_mmu_pages(struct kvm *kvm, unsigned int goal_nr_mmu_pages)
82ce2c96 2244{
d98ba053 2245 LIST_HEAD(invalid_list);
82ce2c96 2246
b34cb590
TY
2247 spin_lock(&kvm->mmu_lock);
2248
49d5ca26 2249 if (kvm->arch.n_used_mmu_pages > goal_nr_mmu_pages) {
5da59607
TY
2250 /* Need to free some mmu pages to achieve the goal. */
2251 while (kvm->arch.n_used_mmu_pages > goal_nr_mmu_pages)
2252 if (!prepare_zap_oldest_mmu_page(kvm, &invalid_list))
2253 break;
82ce2c96 2254
aa6bd187 2255 kvm_mmu_commit_zap_page(kvm, &invalid_list);
49d5ca26 2256 goal_nr_mmu_pages = kvm->arch.n_used_mmu_pages;
82ce2c96 2257 }
82ce2c96 2258
49d5ca26 2259 kvm->arch.n_max_mmu_pages = goal_nr_mmu_pages;
b34cb590
TY
2260
2261 spin_unlock(&kvm->mmu_lock);
82ce2c96
IE
2262}
2263
1cb3f3ae 2264int kvm_mmu_unprotect_page(struct kvm *kvm, gfn_t gfn)
a436036b 2265{
4db35314 2266 struct kvm_mmu_page *sp;
d98ba053 2267 LIST_HEAD(invalid_list);
a436036b
AK
2268 int r;
2269
9ad17b10 2270 pgprintk("%s: looking for gfn %llx\n", __func__, gfn);
a436036b 2271 r = 0;
1cb3f3ae 2272 spin_lock(&kvm->mmu_lock);
b67bfe0d 2273 for_each_gfn_indirect_valid_sp(kvm, sp, gfn) {
9ad17b10 2274 pgprintk("%s: gfn %llx role %x\n", __func__, gfn,
7ae680eb
XG
2275 sp->role.word);
2276 r = 1;
f41d335a 2277 kvm_mmu_prepare_zap_page(kvm, sp, &invalid_list);
7ae680eb 2278 }
d98ba053 2279 kvm_mmu_commit_zap_page(kvm, &invalid_list);
1cb3f3ae
XG
2280 spin_unlock(&kvm->mmu_lock);
2281
a436036b 2282 return r;
cea0f0e7 2283}
1cb3f3ae 2284EXPORT_SYMBOL_GPL(kvm_mmu_unprotect_page);
cea0f0e7 2285
74be52e3
SY
2286/*
2287 * The function is based on mtrr_type_lookup() in
2288 * arch/x86/kernel/cpu/mtrr/generic.c
2289 */
2290static int get_mtrr_type(struct mtrr_state_type *mtrr_state,
2291 u64 start, u64 end)
2292{
2293 int i;
2294 u64 base, mask;
2295 u8 prev_match, curr_match;
2296 int num_var_ranges = KVM_NR_VAR_MTRR;
2297
2298 if (!mtrr_state->enabled)
2299 return 0xFF;
2300
2301 /* Make end inclusive end, instead of exclusive */
2302 end--;
2303
2304 /* Look in fixed ranges. Just return the type as per start */
2305 if (mtrr_state->have_fixed && (start < 0x100000)) {
2306 int idx;
2307
2308 if (start < 0x80000) {
2309 idx = 0;
2310 idx += (start >> 16);
2311 return mtrr_state->fixed_ranges[idx];
2312 } else if (start < 0xC0000) {
2313 idx = 1 * 8;
2314 idx += ((start - 0x80000) >> 14);
2315 return mtrr_state->fixed_ranges[idx];
2316 } else if (start < 0x1000000) {
2317 idx = 3 * 8;
2318 idx += ((start - 0xC0000) >> 12);
2319 return mtrr_state->fixed_ranges[idx];
2320 }
2321 }
2322
2323 /*
2324 * Look in variable ranges
2325 * Look of multiple ranges matching this address and pick type
2326 * as per MTRR precedence
2327 */
2328 if (!(mtrr_state->enabled & 2))
2329 return mtrr_state->def_type;
2330
2331 prev_match = 0xFF;
2332 for (i = 0; i < num_var_ranges; ++i) {
2333 unsigned short start_state, end_state;
2334
2335 if (!(mtrr_state->var_ranges[i].mask_lo & (1 << 11)))
2336 continue;
2337
2338 base = (((u64)mtrr_state->var_ranges[i].base_hi) << 32) +
2339 (mtrr_state->var_ranges[i].base_lo & PAGE_MASK);
2340 mask = (((u64)mtrr_state->var_ranges[i].mask_hi) << 32) +
2341 (mtrr_state->var_ranges[i].mask_lo & PAGE_MASK);
2342
2343 start_state = ((start & mask) == (base & mask));
2344 end_state = ((end & mask) == (base & mask));
2345 if (start_state != end_state)
2346 return 0xFE;
2347
2348 if ((start & mask) != (base & mask))
2349 continue;
2350
2351 curr_match = mtrr_state->var_ranges[i].base_lo & 0xff;
2352 if (prev_match == 0xFF) {
2353 prev_match = curr_match;
2354 continue;
2355 }
2356
2357 if (prev_match == MTRR_TYPE_UNCACHABLE ||
2358 curr_match == MTRR_TYPE_UNCACHABLE)
2359 return MTRR_TYPE_UNCACHABLE;
2360
2361 if ((prev_match == MTRR_TYPE_WRBACK &&
2362 curr_match == MTRR_TYPE_WRTHROUGH) ||
2363 (prev_match == MTRR_TYPE_WRTHROUGH &&
2364 curr_match == MTRR_TYPE_WRBACK)) {
2365 prev_match = MTRR_TYPE_WRTHROUGH;
2366 curr_match = MTRR_TYPE_WRTHROUGH;
2367 }
2368
2369 if (prev_match != curr_match)
2370 return MTRR_TYPE_UNCACHABLE;
2371 }
2372
2373 if (prev_match != 0xFF)
2374 return prev_match;
2375
2376 return mtrr_state->def_type;
2377}
2378
4b12f0de 2379u8 kvm_get_guest_memory_type(struct kvm_vcpu *vcpu, gfn_t gfn)
74be52e3
SY
2380{
2381 u8 mtrr;
2382
2383 mtrr = get_mtrr_type(&vcpu->arch.mtrr_state, gfn << PAGE_SHIFT,
2384 (gfn << PAGE_SHIFT) + PAGE_SIZE);
2385 if (mtrr == 0xfe || mtrr == 0xff)
2386 mtrr = MTRR_TYPE_WRBACK;
2387 return mtrr;
2388}
4b12f0de 2389EXPORT_SYMBOL_GPL(kvm_get_guest_memory_type);
74be52e3 2390
9cf5cf5a
XG
2391static void __kvm_unsync_page(struct kvm_vcpu *vcpu, struct kvm_mmu_page *sp)
2392{
2393 trace_kvm_mmu_unsync_page(sp);
2394 ++vcpu->kvm->stat.mmu_unsync;
2395 sp->unsync = 1;
2396
2397 kvm_mmu_mark_parents_unsync(sp);
9cf5cf5a
XG
2398}
2399
2400static void kvm_unsync_pages(struct kvm_vcpu *vcpu, gfn_t gfn)
4731d4c7 2401{
4731d4c7 2402 struct kvm_mmu_page *s;
9cf5cf5a 2403
b67bfe0d 2404 for_each_gfn_indirect_valid_sp(vcpu->kvm, s, gfn) {
7ae680eb 2405 if (s->unsync)
4731d4c7 2406 continue;
9cf5cf5a
XG
2407 WARN_ON(s->role.level != PT_PAGE_TABLE_LEVEL);
2408 __kvm_unsync_page(vcpu, s);
4731d4c7 2409 }
4731d4c7
MT
2410}
2411
2412static int mmu_need_write_protect(struct kvm_vcpu *vcpu, gfn_t gfn,
2413 bool can_unsync)
2414{
9cf5cf5a 2415 struct kvm_mmu_page *s;
9cf5cf5a
XG
2416 bool need_unsync = false;
2417
b67bfe0d 2418 for_each_gfn_indirect_valid_sp(vcpu->kvm, s, gfn) {
36a2e677
XG
2419 if (!can_unsync)
2420 return 1;
2421
9cf5cf5a 2422 if (s->role.level != PT_PAGE_TABLE_LEVEL)
4731d4c7 2423 return 1;
9cf5cf5a 2424
9bb4f6b1 2425 if (!s->unsync)
9cf5cf5a 2426 need_unsync = true;
4731d4c7 2427 }
9cf5cf5a
XG
2428 if (need_unsync)
2429 kvm_unsync_pages(vcpu, gfn);
4731d4c7
MT
2430 return 0;
2431}
2432
d555c333 2433static int set_spte(struct kvm_vcpu *vcpu, u64 *sptep,
c2288505 2434 unsigned pte_access, int level,
c2d0ee46 2435 gfn_t gfn, pfn_t pfn, bool speculative,
9bdbba13 2436 bool can_unsync, bool host_writable)
1c4f1fd6 2437{
6e7d0354 2438 u64 spte;
1e73f9dd 2439 int ret = 0;
64d4d521 2440
f2fd125d 2441 if (set_mmio_spte(vcpu->kvm, sptep, gfn, pfn, pte_access))
ce88decf
XG
2442 return 0;
2443
982c2565 2444 spte = PT_PRESENT_MASK;
947da538 2445 if (!speculative)
3201b5d9 2446 spte |= shadow_accessed_mask;
640d9b0d 2447
7b52345e
SY
2448 if (pte_access & ACC_EXEC_MASK)
2449 spte |= shadow_x_mask;
2450 else
2451 spte |= shadow_nx_mask;
49fde340 2452
1c4f1fd6 2453 if (pte_access & ACC_USER_MASK)
7b52345e 2454 spte |= shadow_user_mask;
49fde340 2455
852e3c19 2456 if (level > PT_PAGE_TABLE_LEVEL)
05da4558 2457 spte |= PT_PAGE_SIZE_MASK;
b0bc3ee2 2458 if (tdp_enabled)
4b12f0de
SY
2459 spte |= kvm_x86_ops->get_mt_mask(vcpu, gfn,
2460 kvm_is_mmio_pfn(pfn));
1c4f1fd6 2461
9bdbba13 2462 if (host_writable)
1403283a 2463 spte |= SPTE_HOST_WRITEABLE;
f8e453b0
XG
2464 else
2465 pte_access &= ~ACC_WRITE_MASK;
1403283a 2466
35149e21 2467 spte |= (u64)pfn << PAGE_SHIFT;
1c4f1fd6 2468
c2288505 2469 if (pte_access & ACC_WRITE_MASK) {
1c4f1fd6 2470
c2193463 2471 /*
7751babd
XG
2472 * Other vcpu creates new sp in the window between
2473 * mapping_level() and acquiring mmu-lock. We can
2474 * allow guest to retry the access, the mapping can
2475 * be fixed if guest refault.
c2193463 2476 */
852e3c19 2477 if (level > PT_PAGE_TABLE_LEVEL &&
c2193463 2478 has_wrprotected_page(vcpu->kvm, gfn, level))
be38d276 2479 goto done;
38187c83 2480
49fde340 2481 spte |= PT_WRITABLE_MASK | SPTE_MMU_WRITEABLE;
1c4f1fd6 2482
ecc5589f
MT
2483 /*
2484 * Optimization: for pte sync, if spte was writable the hash
2485 * lookup is unnecessary (and expensive). Write protection
2486 * is responsibility of mmu_get_page / kvm_sync_page.
2487 * Same reasoning can be applied to dirty page accounting.
2488 */
8dae4445 2489 if (!can_unsync && is_writable_pte(*sptep))
ecc5589f
MT
2490 goto set_pte;
2491
4731d4c7 2492 if (mmu_need_write_protect(vcpu, gfn, can_unsync)) {
9ad17b10 2493 pgprintk("%s: found shadow page for %llx, marking ro\n",
b8688d51 2494 __func__, gfn);
1e73f9dd 2495 ret = 1;
1c4f1fd6 2496 pte_access &= ~ACC_WRITE_MASK;
49fde340 2497 spte &= ~(PT_WRITABLE_MASK | SPTE_MMU_WRITEABLE);
1c4f1fd6
AK
2498 }
2499 }
2500
1c4f1fd6
AK
2501 if (pte_access & ACC_WRITE_MASK)
2502 mark_page_dirty(vcpu->kvm, gfn);
2503
38187c83 2504set_pte:
6e7d0354 2505 if (mmu_spte_update(sptep, spte))
b330aa0c 2506 kvm_flush_remote_tlbs(vcpu->kvm);
be38d276 2507done:
1e73f9dd
MT
2508 return ret;
2509}
2510
d555c333 2511static void mmu_set_spte(struct kvm_vcpu *vcpu, u64 *sptep,
f7616203
XG
2512 unsigned pte_access, int write_fault, int *emulate,
2513 int level, gfn_t gfn, pfn_t pfn, bool speculative,
2514 bool host_writable)
1e73f9dd
MT
2515{
2516 int was_rmapped = 0;
53a27b39 2517 int rmap_count;
1e73f9dd 2518
f7616203
XG
2519 pgprintk("%s: spte %llx write_fault %d gfn %llx\n", __func__,
2520 *sptep, write_fault, gfn);
1e73f9dd 2521
d555c333 2522 if (is_rmap_spte(*sptep)) {
1e73f9dd
MT
2523 /*
2524 * If we overwrite a PTE page pointer with a 2MB PMD, unlink
2525 * the parent of the now unreachable PTE.
2526 */
852e3c19
JR
2527 if (level > PT_PAGE_TABLE_LEVEL &&
2528 !is_large_pte(*sptep)) {
1e73f9dd 2529 struct kvm_mmu_page *child;
d555c333 2530 u64 pte = *sptep;
1e73f9dd
MT
2531
2532 child = page_header(pte & PT64_BASE_ADDR_MASK);
bcdd9a93 2533 drop_parent_pte(child, sptep);
3be2264b 2534 kvm_flush_remote_tlbs(vcpu->kvm);
d555c333 2535 } else if (pfn != spte_to_pfn(*sptep)) {
9ad17b10 2536 pgprintk("hfn old %llx new %llx\n",
d555c333 2537 spte_to_pfn(*sptep), pfn);
c3707958 2538 drop_spte(vcpu->kvm, sptep);
91546356 2539 kvm_flush_remote_tlbs(vcpu->kvm);
6bed6b9e
JR
2540 } else
2541 was_rmapped = 1;
1e73f9dd 2542 }
852e3c19 2543
c2288505
XG
2544 if (set_spte(vcpu, sptep, pte_access, level, gfn, pfn, speculative,
2545 true, host_writable)) {
1e73f9dd 2546 if (write_fault)
b90a0e6c 2547 *emulate = 1;
5304efde 2548 kvm_mmu_flush_tlb(vcpu);
a378b4e6 2549 }
1e73f9dd 2550
ce88decf
XG
2551 if (unlikely(is_mmio_spte(*sptep) && emulate))
2552 *emulate = 1;
2553
d555c333 2554 pgprintk("%s: setting spte %llx\n", __func__, *sptep);
9ad17b10 2555 pgprintk("instantiating %s PTE (%s) at %llx (%llx) addr %p\n",
d555c333 2556 is_large_pte(*sptep)? "2MB" : "4kB",
a205bc19
JR
2557 *sptep & PT_PRESENT_MASK ?"RW":"R", gfn,
2558 *sptep, sptep);
d555c333 2559 if (!was_rmapped && is_large_pte(*sptep))
05da4558
MT
2560 ++vcpu->kvm->stat.lpages;
2561
ffb61bb3 2562 if (is_shadow_present_pte(*sptep)) {
ffb61bb3
XG
2563 if (!was_rmapped) {
2564 rmap_count = rmap_add(vcpu, sptep, gfn);
2565 if (rmap_count > RMAP_RECYCLE_THRESHOLD)
2566 rmap_recycle(vcpu, sptep, gfn);
2567 }
1c4f1fd6 2568 }
cb9aaa30 2569
f3ac1a4b 2570 kvm_release_pfn_clean(pfn);
1c4f1fd6
AK
2571}
2572
957ed9ef
XG
2573static pfn_t pte_prefetch_gfn_to_pfn(struct kvm_vcpu *vcpu, gfn_t gfn,
2574 bool no_dirty_log)
2575{
2576 struct kvm_memory_slot *slot;
957ed9ef 2577
5d163b1c 2578 slot = gfn_to_memslot_dirty_bitmap(vcpu, gfn, no_dirty_log);
903816fa 2579 if (!slot)
6c8ee57b 2580 return KVM_PFN_ERR_FAULT;
957ed9ef 2581
037d92dc 2582 return gfn_to_pfn_memslot_atomic(slot, gfn);
957ed9ef
XG
2583}
2584
2585static int direct_pte_prefetch_many(struct kvm_vcpu *vcpu,
2586 struct kvm_mmu_page *sp,
2587 u64 *start, u64 *end)
2588{
2589 struct page *pages[PTE_PREFETCH_NUM];
2590 unsigned access = sp->role.access;
2591 int i, ret;
2592 gfn_t gfn;
2593
2594 gfn = kvm_mmu_page_get_gfn(sp, start - sp->spt);
5d163b1c 2595 if (!gfn_to_memslot_dirty_bitmap(vcpu, gfn, access & ACC_WRITE_MASK))
957ed9ef
XG
2596 return -1;
2597
2598 ret = gfn_to_page_many_atomic(vcpu->kvm, gfn, pages, end - start);
2599 if (ret <= 0)
2600 return -1;
2601
2602 for (i = 0; i < ret; i++, gfn++, start++)
f7616203 2603 mmu_set_spte(vcpu, start, access, 0, NULL,
c2288505
XG
2604 sp->role.level, gfn, page_to_pfn(pages[i]),
2605 true, true);
957ed9ef
XG
2606
2607 return 0;
2608}
2609
2610static void __direct_pte_prefetch(struct kvm_vcpu *vcpu,
2611 struct kvm_mmu_page *sp, u64 *sptep)
2612{
2613 u64 *spte, *start = NULL;
2614 int i;
2615
2616 WARN_ON(!sp->role.direct);
2617
2618 i = (sptep - sp->spt) & ~(PTE_PREFETCH_NUM - 1);
2619 spte = sp->spt + i;
2620
2621 for (i = 0; i < PTE_PREFETCH_NUM; i++, spte++) {
c3707958 2622 if (is_shadow_present_pte(*spte) || spte == sptep) {
957ed9ef
XG
2623 if (!start)
2624 continue;
2625 if (direct_pte_prefetch_many(vcpu, sp, start, spte) < 0)
2626 break;
2627 start = NULL;
2628 } else if (!start)
2629 start = spte;
2630 }
2631}
2632
2633static void direct_pte_prefetch(struct kvm_vcpu *vcpu, u64 *sptep)
2634{
2635 struct kvm_mmu_page *sp;
2636
2637 /*
2638 * Since it's no accessed bit on EPT, it's no way to
2639 * distinguish between actually accessed translations
2640 * and prefetched, so disable pte prefetch if EPT is
2641 * enabled.
2642 */
2643 if (!shadow_accessed_mask)
2644 return;
2645
2646 sp = page_header(__pa(sptep));
2647 if (sp->role.level > PT_PAGE_TABLE_LEVEL)
2648 return;
2649
2650 __direct_pte_prefetch(vcpu, sp, sptep);
2651}
2652
9f652d21 2653static int __direct_map(struct kvm_vcpu *vcpu, gpa_t v, int write,
2ec4739d
XG
2654 int map_writable, int level, gfn_t gfn, pfn_t pfn,
2655 bool prefault)
140754bc 2656{
9f652d21 2657 struct kvm_shadow_walk_iterator iterator;
140754bc 2658 struct kvm_mmu_page *sp;
b90a0e6c 2659 int emulate = 0;
140754bc 2660 gfn_t pseudo_gfn;
6aa8b732 2661
989c6b34
MT
2662 if (!VALID_PAGE(vcpu->arch.mmu.root_hpa))
2663 return 0;
2664
9f652d21 2665 for_each_shadow_entry(vcpu, (u64)gfn << PAGE_SHIFT, iterator) {
852e3c19 2666 if (iterator.level == level) {
f7616203 2667 mmu_set_spte(vcpu, iterator.sptep, ACC_ALL,
c2288505
XG
2668 write, &emulate, level, gfn, pfn,
2669 prefault, map_writable);
957ed9ef 2670 direct_pte_prefetch(vcpu, iterator.sptep);
9f652d21
AK
2671 ++vcpu->stat.pf_fixed;
2672 break;
6aa8b732
AK
2673 }
2674
404381c5 2675 drop_large_spte(vcpu, iterator.sptep);
c3707958 2676 if (!is_shadow_present_pte(*iterator.sptep)) {
c9fa0b3b
LJ
2677 u64 base_addr = iterator.addr;
2678
2679 base_addr &= PT64_LVL_ADDR_MASK(iterator.level);
2680 pseudo_gfn = base_addr >> PAGE_SHIFT;
9f652d21
AK
2681 sp = kvm_mmu_get_page(vcpu, pseudo_gfn, iterator.addr,
2682 iterator.level - 1,
2683 1, ACC_ALL, iterator.sptep);
140754bc 2684
7a1638ce 2685 link_shadow_page(iterator.sptep, sp, true);
9f652d21
AK
2686 }
2687 }
b90a0e6c 2688 return emulate;
6aa8b732
AK
2689}
2690
77db5cbd 2691static void kvm_send_hwpoison_signal(unsigned long address, struct task_struct *tsk)
bf998156 2692{
77db5cbd
HY
2693 siginfo_t info;
2694
2695 info.si_signo = SIGBUS;
2696 info.si_errno = 0;
2697 info.si_code = BUS_MCEERR_AR;
2698 info.si_addr = (void __user *)address;
2699 info.si_addr_lsb = PAGE_SHIFT;
bf998156 2700
77db5cbd 2701 send_sig_info(SIGBUS, &info, tsk);
bf998156
HY
2702}
2703
d7c55201 2704static int kvm_handle_bad_page(struct kvm_vcpu *vcpu, gfn_t gfn, pfn_t pfn)
bf998156 2705{
4d8b81ab
XG
2706 /*
2707 * Do not cache the mmio info caused by writing the readonly gfn
2708 * into the spte otherwise read access on readonly gfn also can
2709 * caused mmio page fault and treat it as mmio access.
2710 * Return 1 to tell kvm to emulate it.
2711 */
2712 if (pfn == KVM_PFN_ERR_RO_FAULT)
2713 return 1;
2714
e6c1502b 2715 if (pfn == KVM_PFN_ERR_HWPOISON) {
bebb106a 2716 kvm_send_hwpoison_signal(gfn_to_hva(vcpu->kvm, gfn), current);
bf998156 2717 return 0;
d7c55201 2718 }
edba23e5 2719
d7c55201 2720 return -EFAULT;
bf998156
HY
2721}
2722
936a5fe6
AA
2723static void transparent_hugepage_adjust(struct kvm_vcpu *vcpu,
2724 gfn_t *gfnp, pfn_t *pfnp, int *levelp)
2725{
2726 pfn_t pfn = *pfnp;
2727 gfn_t gfn = *gfnp;
2728 int level = *levelp;
2729
2730 /*
2731 * Check if it's a transparent hugepage. If this would be an
2732 * hugetlbfs page, level wouldn't be set to
2733 * PT_PAGE_TABLE_LEVEL and there would be no adjustment done
2734 * here.
2735 */
81c52c56 2736 if (!is_error_noslot_pfn(pfn) && !kvm_is_mmio_pfn(pfn) &&
936a5fe6
AA
2737 level == PT_PAGE_TABLE_LEVEL &&
2738 PageTransCompound(pfn_to_page(pfn)) &&
2739 !has_wrprotected_page(vcpu->kvm, gfn, PT_DIRECTORY_LEVEL)) {
2740 unsigned long mask;
2741 /*
2742 * mmu_notifier_retry was successful and we hold the
2743 * mmu_lock here, so the pmd can't become splitting
2744 * from under us, and in turn
2745 * __split_huge_page_refcount() can't run from under
2746 * us and we can safely transfer the refcount from
2747 * PG_tail to PG_head as we switch the pfn to tail to
2748 * head.
2749 */
2750 *levelp = level = PT_DIRECTORY_LEVEL;
2751 mask = KVM_PAGES_PER_HPAGE(level) - 1;
2752 VM_BUG_ON((gfn & mask) != (pfn & mask));
2753 if (pfn & mask) {
2754 gfn &= ~mask;
2755 *gfnp = gfn;
2756 kvm_release_pfn_clean(pfn);
2757 pfn &= ~mask;
c3586667 2758 kvm_get_pfn(pfn);
936a5fe6
AA
2759 *pfnp = pfn;
2760 }
2761 }
2762}
2763
d7c55201
XG
2764static bool handle_abnormal_pfn(struct kvm_vcpu *vcpu, gva_t gva, gfn_t gfn,
2765 pfn_t pfn, unsigned access, int *ret_val)
2766{
2767 bool ret = true;
2768
2769 /* The pfn is invalid, report the error! */
81c52c56 2770 if (unlikely(is_error_pfn(pfn))) {
d7c55201
XG
2771 *ret_val = kvm_handle_bad_page(vcpu, gfn, pfn);
2772 goto exit;
2773 }
2774
ce88decf 2775 if (unlikely(is_noslot_pfn(pfn)))
d7c55201 2776 vcpu_cache_mmio_info(vcpu, gva, gfn, access);
d7c55201
XG
2777
2778 ret = false;
2779exit:
2780 return ret;
2781}
2782
e5552fd2 2783static bool page_fault_can_be_fast(u32 error_code)
c7ba5b48 2784{
1c118b82
XG
2785 /*
2786 * Do not fix the mmio spte with invalid generation number which
2787 * need to be updated by slow page fault path.
2788 */
2789 if (unlikely(error_code & PFERR_RSVD_MASK))
2790 return false;
2791
c7ba5b48
XG
2792 /*
2793 * #PF can be fast only if the shadow page table is present and it
2794 * is caused by write-protect, that means we just need change the
2795 * W bit of the spte which can be done out of mmu-lock.
2796 */
2797 if (!(error_code & PFERR_PRESENT_MASK) ||
2798 !(error_code & PFERR_WRITE_MASK))
2799 return false;
2800
2801 return true;
2802}
2803
2804static bool
2805fast_pf_fix_direct_spte(struct kvm_vcpu *vcpu, u64 *sptep, u64 spte)
2806{
2807 struct kvm_mmu_page *sp = page_header(__pa(sptep));
2808 gfn_t gfn;
2809
2810 WARN_ON(!sp->role.direct);
2811
2812 /*
2813 * The gfn of direct spte is stable since it is calculated
2814 * by sp->gfn.
2815 */
2816 gfn = kvm_mmu_page_get_gfn(sp, sptep - sp->spt);
2817
2818 if (cmpxchg64(sptep, spte, spte | PT_WRITABLE_MASK) == spte)
2819 mark_page_dirty(vcpu->kvm, gfn);
2820
2821 return true;
2822}
2823
2824/*
2825 * Return value:
2826 * - true: let the vcpu to access on the same address again.
2827 * - false: let the real page fault path to fix it.
2828 */
2829static bool fast_page_fault(struct kvm_vcpu *vcpu, gva_t gva, int level,
2830 u32 error_code)
2831{
2832 struct kvm_shadow_walk_iterator iterator;
2833 bool ret = false;
2834 u64 spte = 0ull;
2835
37f6a4e2
MT
2836 if (!VALID_PAGE(vcpu->arch.mmu.root_hpa))
2837 return false;
2838
e5552fd2 2839 if (!page_fault_can_be_fast(error_code))
c7ba5b48
XG
2840 return false;
2841
2842 walk_shadow_page_lockless_begin(vcpu);
2843 for_each_shadow_entry_lockless(vcpu, gva, iterator, spte)
2844 if (!is_shadow_present_pte(spte) || iterator.level < level)
2845 break;
2846
2847 /*
2848 * If the mapping has been changed, let the vcpu fault on the
2849 * same address again.
2850 */
2851 if (!is_rmap_spte(spte)) {
2852 ret = true;
2853 goto exit;
2854 }
2855
2856 if (!is_last_spte(spte, level))
2857 goto exit;
2858
2859 /*
2860 * Check if it is a spurious fault caused by TLB lazily flushed.
2861 *
2862 * Need not check the access of upper level table entries since
2863 * they are always ACC_ALL.
2864 */
2865 if (is_writable_pte(spte)) {
2866 ret = true;
2867 goto exit;
2868 }
2869
2870 /*
2871 * Currently, to simplify the code, only the spte write-protected
2872 * by dirty-log can be fast fixed.
2873 */
2874 if (!spte_is_locklessly_modifiable(spte))
2875 goto exit;
2876
2877 /*
2878 * Currently, fast page fault only works for direct mapping since
2879 * the gfn is not stable for indirect shadow page.
2880 * See Documentation/virtual/kvm/locking.txt to get more detail.
2881 */
2882 ret = fast_pf_fix_direct_spte(vcpu, iterator.sptep, spte);
2883exit:
a72faf25
XG
2884 trace_fast_page_fault(vcpu, gva, error_code, iterator.sptep,
2885 spte, ret);
c7ba5b48
XG
2886 walk_shadow_page_lockless_end(vcpu);
2887
2888 return ret;
2889}
2890
78b2c54a 2891static bool try_async_pf(struct kvm_vcpu *vcpu, bool prefault, gfn_t gfn,
060c2abe 2892 gva_t gva, pfn_t *pfn, bool write, bool *writable);
450e0b41 2893static void make_mmu_pages_available(struct kvm_vcpu *vcpu);
060c2abe 2894
c7ba5b48
XG
2895static int nonpaging_map(struct kvm_vcpu *vcpu, gva_t v, u32 error_code,
2896 gfn_t gfn, bool prefault)
10589a46
MT
2897{
2898 int r;
852e3c19 2899 int level;
936a5fe6 2900 int force_pt_level;
35149e21 2901 pfn_t pfn;
e930bffe 2902 unsigned long mmu_seq;
c7ba5b48 2903 bool map_writable, write = error_code & PFERR_WRITE_MASK;
aaee2c94 2904
936a5fe6
AA
2905 force_pt_level = mapping_level_dirty_bitmap(vcpu, gfn);
2906 if (likely(!force_pt_level)) {
2907 level = mapping_level(vcpu, gfn);
2908 /*
2909 * This path builds a PAE pagetable - so we can map
2910 * 2mb pages at maximum. Therefore check if the level
2911 * is larger than that.
2912 */
2913 if (level > PT_DIRECTORY_LEVEL)
2914 level = PT_DIRECTORY_LEVEL;
852e3c19 2915
936a5fe6
AA
2916 gfn &= ~(KVM_PAGES_PER_HPAGE(level) - 1);
2917 } else
2918 level = PT_PAGE_TABLE_LEVEL;
05da4558 2919
c7ba5b48
XG
2920 if (fast_page_fault(vcpu, v, level, error_code))
2921 return 0;
2922
e930bffe 2923 mmu_seq = vcpu->kvm->mmu_notifier_seq;
4c2155ce 2924 smp_rmb();
060c2abe 2925
78b2c54a 2926 if (try_async_pf(vcpu, prefault, gfn, v, &pfn, write, &map_writable))
060c2abe 2927 return 0;
aaee2c94 2928
d7c55201
XG
2929 if (handle_abnormal_pfn(vcpu, v, gfn, pfn, ACC_ALL, &r))
2930 return r;
d196e343 2931
aaee2c94 2932 spin_lock(&vcpu->kvm->mmu_lock);
8ca40a70 2933 if (mmu_notifier_retry(vcpu->kvm, mmu_seq))
e930bffe 2934 goto out_unlock;
450e0b41 2935 make_mmu_pages_available(vcpu);
936a5fe6
AA
2936 if (likely(!force_pt_level))
2937 transparent_hugepage_adjust(vcpu, &gfn, &pfn, &level);
2ec4739d
XG
2938 r = __direct_map(vcpu, v, write, map_writable, level, gfn, pfn,
2939 prefault);
aaee2c94
MT
2940 spin_unlock(&vcpu->kvm->mmu_lock);
2941
aaee2c94 2942
10589a46 2943 return r;
e930bffe
AA
2944
2945out_unlock:
2946 spin_unlock(&vcpu->kvm->mmu_lock);
2947 kvm_release_pfn_clean(pfn);
2948 return 0;
10589a46
MT
2949}
2950
2951
17ac10ad
AK
2952static void mmu_free_roots(struct kvm_vcpu *vcpu)
2953{
2954 int i;
4db35314 2955 struct kvm_mmu_page *sp;
d98ba053 2956 LIST_HEAD(invalid_list);
17ac10ad 2957
ad312c7c 2958 if (!VALID_PAGE(vcpu->arch.mmu.root_hpa))
7b53aa56 2959 return;
35af577a 2960
81407ca5
JR
2961 if (vcpu->arch.mmu.shadow_root_level == PT64_ROOT_LEVEL &&
2962 (vcpu->arch.mmu.root_level == PT64_ROOT_LEVEL ||
2963 vcpu->arch.mmu.direct_map)) {
ad312c7c 2964 hpa_t root = vcpu->arch.mmu.root_hpa;
17ac10ad 2965
35af577a 2966 spin_lock(&vcpu->kvm->mmu_lock);
4db35314
AK
2967 sp = page_header(root);
2968 --sp->root_count;
d98ba053
XG
2969 if (!sp->root_count && sp->role.invalid) {
2970 kvm_mmu_prepare_zap_page(vcpu->kvm, sp, &invalid_list);
2971 kvm_mmu_commit_zap_page(vcpu->kvm, &invalid_list);
2972 }
aaee2c94 2973 spin_unlock(&vcpu->kvm->mmu_lock);
35af577a 2974 vcpu->arch.mmu.root_hpa = INVALID_PAGE;
17ac10ad
AK
2975 return;
2976 }
35af577a
GN
2977
2978 spin_lock(&vcpu->kvm->mmu_lock);
17ac10ad 2979 for (i = 0; i < 4; ++i) {
ad312c7c 2980 hpa_t root = vcpu->arch.mmu.pae_root[i];
17ac10ad 2981
417726a3 2982 if (root) {
417726a3 2983 root &= PT64_BASE_ADDR_MASK;
4db35314
AK
2984 sp = page_header(root);
2985 --sp->root_count;
2e53d63a 2986 if (!sp->root_count && sp->role.invalid)
d98ba053
XG
2987 kvm_mmu_prepare_zap_page(vcpu->kvm, sp,
2988 &invalid_list);
417726a3 2989 }
ad312c7c 2990 vcpu->arch.mmu.pae_root[i] = INVALID_PAGE;
17ac10ad 2991 }
d98ba053 2992 kvm_mmu_commit_zap_page(vcpu->kvm, &invalid_list);
aaee2c94 2993 spin_unlock(&vcpu->kvm->mmu_lock);
ad312c7c 2994 vcpu->arch.mmu.root_hpa = INVALID_PAGE;
17ac10ad
AK
2995}
2996
8986ecc0
MT
2997static int mmu_check_root(struct kvm_vcpu *vcpu, gfn_t root_gfn)
2998{
2999 int ret = 0;
3000
3001 if (!kvm_is_visible_gfn(vcpu->kvm, root_gfn)) {
a8eeb04a 3002 kvm_make_request(KVM_REQ_TRIPLE_FAULT, vcpu);
8986ecc0
MT
3003 ret = 1;
3004 }
3005
3006 return ret;
3007}
3008
651dd37a
JR
3009static int mmu_alloc_direct_roots(struct kvm_vcpu *vcpu)
3010{
3011 struct kvm_mmu_page *sp;
7ebaf15e 3012 unsigned i;
651dd37a
JR
3013
3014 if (vcpu->arch.mmu.shadow_root_level == PT64_ROOT_LEVEL) {
3015 spin_lock(&vcpu->kvm->mmu_lock);
450e0b41 3016 make_mmu_pages_available(vcpu);
651dd37a
JR
3017 sp = kvm_mmu_get_page(vcpu, 0, 0, PT64_ROOT_LEVEL,
3018 1, ACC_ALL, NULL);
3019 ++sp->root_count;
3020 spin_unlock(&vcpu->kvm->mmu_lock);
3021 vcpu->arch.mmu.root_hpa = __pa(sp->spt);
3022 } else if (vcpu->arch.mmu.shadow_root_level == PT32E_ROOT_LEVEL) {
3023 for (i = 0; i < 4; ++i) {
3024 hpa_t root = vcpu->arch.mmu.pae_root[i];
3025
3026 ASSERT(!VALID_PAGE(root));
3027 spin_lock(&vcpu->kvm->mmu_lock);
450e0b41 3028 make_mmu_pages_available(vcpu);
649497d1
AK
3029 sp = kvm_mmu_get_page(vcpu, i << (30 - PAGE_SHIFT),
3030 i << 30,
651dd37a
JR
3031 PT32_ROOT_LEVEL, 1, ACC_ALL,
3032 NULL);
3033 root = __pa(sp->spt);
3034 ++sp->root_count;
3035 spin_unlock(&vcpu->kvm->mmu_lock);
3036 vcpu->arch.mmu.pae_root[i] = root | PT_PRESENT_MASK;
651dd37a 3037 }
6292757f 3038 vcpu->arch.mmu.root_hpa = __pa(vcpu->arch.mmu.pae_root);
651dd37a
JR
3039 } else
3040 BUG();
3041
3042 return 0;
3043}
3044
3045static int mmu_alloc_shadow_roots(struct kvm_vcpu *vcpu)
17ac10ad 3046{
4db35314 3047 struct kvm_mmu_page *sp;
81407ca5
JR
3048 u64 pdptr, pm_mask;
3049 gfn_t root_gfn;
3050 int i;
3bb65a22 3051
5777ed34 3052 root_gfn = vcpu->arch.mmu.get_cr3(vcpu) >> PAGE_SHIFT;
17ac10ad 3053
651dd37a
JR
3054 if (mmu_check_root(vcpu, root_gfn))
3055 return 1;
3056
3057 /*
3058 * Do we shadow a long mode page table? If so we need to
3059 * write-protect the guests page table root.
3060 */
3061 if (vcpu->arch.mmu.root_level == PT64_ROOT_LEVEL) {
ad312c7c 3062 hpa_t root = vcpu->arch.mmu.root_hpa;
17ac10ad
AK
3063
3064 ASSERT(!VALID_PAGE(root));
651dd37a 3065
8facbbff 3066 spin_lock(&vcpu->kvm->mmu_lock);
450e0b41 3067 make_mmu_pages_available(vcpu);
651dd37a
JR
3068 sp = kvm_mmu_get_page(vcpu, root_gfn, 0, PT64_ROOT_LEVEL,
3069 0, ACC_ALL, NULL);
4db35314
AK
3070 root = __pa(sp->spt);
3071 ++sp->root_count;
8facbbff 3072 spin_unlock(&vcpu->kvm->mmu_lock);
ad312c7c 3073 vcpu->arch.mmu.root_hpa = root;
8986ecc0 3074 return 0;
17ac10ad 3075 }
f87f9288 3076
651dd37a
JR
3077 /*
3078 * We shadow a 32 bit page table. This may be a legacy 2-level
81407ca5
JR
3079 * or a PAE 3-level page table. In either case we need to be aware that
3080 * the shadow page table may be a PAE or a long mode page table.
651dd37a 3081 */
81407ca5
JR
3082 pm_mask = PT_PRESENT_MASK;
3083 if (vcpu->arch.mmu.shadow_root_level == PT64_ROOT_LEVEL)
3084 pm_mask |= PT_ACCESSED_MASK | PT_WRITABLE_MASK | PT_USER_MASK;
3085
17ac10ad 3086 for (i = 0; i < 4; ++i) {
ad312c7c 3087 hpa_t root = vcpu->arch.mmu.pae_root[i];
17ac10ad
AK
3088
3089 ASSERT(!VALID_PAGE(root));
ad312c7c 3090 if (vcpu->arch.mmu.root_level == PT32E_ROOT_LEVEL) {
e4e517b4 3091 pdptr = vcpu->arch.mmu.get_pdptr(vcpu, i);
43a3795a 3092 if (!is_present_gpte(pdptr)) {
ad312c7c 3093 vcpu->arch.mmu.pae_root[i] = 0;
417726a3
AK
3094 continue;
3095 }
6de4f3ad 3096 root_gfn = pdptr >> PAGE_SHIFT;
f87f9288
JR
3097 if (mmu_check_root(vcpu, root_gfn))
3098 return 1;
5a7388c2 3099 }
8facbbff 3100 spin_lock(&vcpu->kvm->mmu_lock);
450e0b41 3101 make_mmu_pages_available(vcpu);
4db35314 3102 sp = kvm_mmu_get_page(vcpu, root_gfn, i << 30,
651dd37a 3103 PT32_ROOT_LEVEL, 0,
f7d9c7b7 3104 ACC_ALL, NULL);
4db35314
AK
3105 root = __pa(sp->spt);
3106 ++sp->root_count;
8facbbff
AK
3107 spin_unlock(&vcpu->kvm->mmu_lock);
3108
81407ca5 3109 vcpu->arch.mmu.pae_root[i] = root | pm_mask;
17ac10ad 3110 }
6292757f 3111 vcpu->arch.mmu.root_hpa = __pa(vcpu->arch.mmu.pae_root);
81407ca5
JR
3112
3113 /*
3114 * If we shadow a 32 bit page table with a long mode page
3115 * table we enter this path.
3116 */
3117 if (vcpu->arch.mmu.shadow_root_level == PT64_ROOT_LEVEL) {
3118 if (vcpu->arch.mmu.lm_root == NULL) {
3119 /*
3120 * The additional page necessary for this is only
3121 * allocated on demand.
3122 */
3123
3124 u64 *lm_root;
3125
3126 lm_root = (void*)get_zeroed_page(GFP_KERNEL);
3127 if (lm_root == NULL)
3128 return 1;
3129
3130 lm_root[0] = __pa(vcpu->arch.mmu.pae_root) | pm_mask;
3131
3132 vcpu->arch.mmu.lm_root = lm_root;
3133 }
3134
3135 vcpu->arch.mmu.root_hpa = __pa(vcpu->arch.mmu.lm_root);
3136 }
3137
8986ecc0 3138 return 0;
17ac10ad
AK
3139}
3140
651dd37a
JR
3141static int mmu_alloc_roots(struct kvm_vcpu *vcpu)
3142{
3143 if (vcpu->arch.mmu.direct_map)
3144 return mmu_alloc_direct_roots(vcpu);
3145 else
3146 return mmu_alloc_shadow_roots(vcpu);
3147}
3148
0ba73cda
MT
3149static void mmu_sync_roots(struct kvm_vcpu *vcpu)
3150{
3151 int i;
3152 struct kvm_mmu_page *sp;
3153
81407ca5
JR
3154 if (vcpu->arch.mmu.direct_map)
3155 return;
3156
0ba73cda
MT
3157 if (!VALID_PAGE(vcpu->arch.mmu.root_hpa))
3158 return;
6903074c 3159
bebb106a 3160 vcpu_clear_mmio_info(vcpu, ~0ul);
0375f7fa 3161 kvm_mmu_audit(vcpu, AUDIT_PRE_SYNC);
81407ca5 3162 if (vcpu->arch.mmu.root_level == PT64_ROOT_LEVEL) {
0ba73cda
MT
3163 hpa_t root = vcpu->arch.mmu.root_hpa;
3164 sp = page_header(root);
3165 mmu_sync_children(vcpu, sp);
0375f7fa 3166 kvm_mmu_audit(vcpu, AUDIT_POST_SYNC);
0ba73cda
MT
3167 return;
3168 }
3169 for (i = 0; i < 4; ++i) {
3170 hpa_t root = vcpu->arch.mmu.pae_root[i];
3171
8986ecc0 3172 if (root && VALID_PAGE(root)) {
0ba73cda
MT
3173 root &= PT64_BASE_ADDR_MASK;
3174 sp = page_header(root);
3175 mmu_sync_children(vcpu, sp);
3176 }
3177 }
0375f7fa 3178 kvm_mmu_audit(vcpu, AUDIT_POST_SYNC);
0ba73cda
MT
3179}
3180
3181void kvm_mmu_sync_roots(struct kvm_vcpu *vcpu)
3182{
3183 spin_lock(&vcpu->kvm->mmu_lock);
3184 mmu_sync_roots(vcpu);
6cffe8ca 3185 spin_unlock(&vcpu->kvm->mmu_lock);
0ba73cda 3186}
bfd0a56b 3187EXPORT_SYMBOL_GPL(kvm_mmu_sync_roots);
0ba73cda 3188
1871c602 3189static gpa_t nonpaging_gva_to_gpa(struct kvm_vcpu *vcpu, gva_t vaddr,
ab9ae313 3190 u32 access, struct x86_exception *exception)
6aa8b732 3191{
ab9ae313
AK
3192 if (exception)
3193 exception->error_code = 0;
6aa8b732
AK
3194 return vaddr;
3195}
3196
6539e738 3197static gpa_t nonpaging_gva_to_gpa_nested(struct kvm_vcpu *vcpu, gva_t vaddr,
ab9ae313
AK
3198 u32 access,
3199 struct x86_exception *exception)
6539e738 3200{
ab9ae313
AK
3201 if (exception)
3202 exception->error_code = 0;
6539e738
JR
3203 return vcpu->arch.nested_mmu.translate_gpa(vcpu, vaddr, access);
3204}
3205
ce88decf
XG
3206static bool quickly_check_mmio_pf(struct kvm_vcpu *vcpu, u64 addr, bool direct)
3207{
3208 if (direct)
3209 return vcpu_match_mmio_gpa(vcpu, addr);
3210
3211 return vcpu_match_mmio_gva(vcpu, addr);
3212}
3213
3214
3215/*
3216 * On direct hosts, the last spte is only allows two states
3217 * for mmio page fault:
3218 * - It is the mmio spte
3219 * - It is zapped or it is being zapped.
3220 *
3221 * This function completely checks the spte when the last spte
3222 * is not the mmio spte.
3223 */
3224static bool check_direct_spte_mmio_pf(u64 spte)
3225{
3226 return __check_direct_spte_mmio_pf(spte);
3227}
3228
3229static u64 walk_shadow_page_get_mmio_spte(struct kvm_vcpu *vcpu, u64 addr)
3230{
3231 struct kvm_shadow_walk_iterator iterator;
3232 u64 spte = 0ull;
3233
37f6a4e2
MT
3234 if (!VALID_PAGE(vcpu->arch.mmu.root_hpa))
3235 return spte;
3236
ce88decf
XG
3237 walk_shadow_page_lockless_begin(vcpu);
3238 for_each_shadow_entry_lockless(vcpu, addr, iterator, spte)
3239 if (!is_shadow_present_pte(spte))
3240 break;
3241 walk_shadow_page_lockless_end(vcpu);
3242
3243 return spte;
3244}
3245
ce88decf
XG
3246int handle_mmio_page_fault_common(struct kvm_vcpu *vcpu, u64 addr, bool direct)
3247{
3248 u64 spte;
3249
3250 if (quickly_check_mmio_pf(vcpu, addr, direct))
b37fbea6 3251 return RET_MMIO_PF_EMULATE;
ce88decf
XG
3252
3253 spte = walk_shadow_page_get_mmio_spte(vcpu, addr);
3254
3255 if (is_mmio_spte(spte)) {
3256 gfn_t gfn = get_mmio_spte_gfn(spte);
3257 unsigned access = get_mmio_spte_access(spte);
3258
f8f55942
XG
3259 if (!check_mmio_spte(vcpu->kvm, spte))
3260 return RET_MMIO_PF_INVALID;
3261
ce88decf
XG
3262 if (direct)
3263 addr = 0;
4f022648
XG
3264
3265 trace_handle_mmio_page_fault(addr, gfn, access);
ce88decf 3266 vcpu_cache_mmio_info(vcpu, addr, gfn, access);
b37fbea6 3267 return RET_MMIO_PF_EMULATE;
ce88decf
XG
3268 }
3269
3270 /*
3271 * It's ok if the gva is remapped by other cpus on shadow guest,
3272 * it's a BUG if the gfn is not a mmio page.
3273 */
3274 if (direct && !check_direct_spte_mmio_pf(spte))
b37fbea6 3275 return RET_MMIO_PF_BUG;
ce88decf
XG
3276
3277 /*
3278 * If the page table is zapped by other cpus, let CPU fault again on
3279 * the address.
3280 */
b37fbea6 3281 return RET_MMIO_PF_RETRY;
ce88decf
XG
3282}
3283EXPORT_SYMBOL_GPL(handle_mmio_page_fault_common);
3284
3285static int handle_mmio_page_fault(struct kvm_vcpu *vcpu, u64 addr,
3286 u32 error_code, bool direct)
3287{
3288 int ret;
3289
3290 ret = handle_mmio_page_fault_common(vcpu, addr, direct);
b37fbea6 3291 WARN_ON(ret == RET_MMIO_PF_BUG);
ce88decf
XG
3292 return ret;
3293}
3294
6aa8b732 3295static int nonpaging_page_fault(struct kvm_vcpu *vcpu, gva_t gva,
78b2c54a 3296 u32 error_code, bool prefault)
6aa8b732 3297{
e833240f 3298 gfn_t gfn;
e2dec939 3299 int r;
6aa8b732 3300
b8688d51 3301 pgprintk("%s: gva %lx error %x\n", __func__, gva, error_code);
ce88decf 3302
f8f55942
XG
3303 if (unlikely(error_code & PFERR_RSVD_MASK)) {
3304 r = handle_mmio_page_fault(vcpu, gva, error_code, true);
3305
3306 if (likely(r != RET_MMIO_PF_INVALID))
3307 return r;
3308 }
ce88decf 3309
e2dec939
AK
3310 r = mmu_topup_memory_caches(vcpu);
3311 if (r)
3312 return r;
714b93da 3313
6aa8b732 3314 ASSERT(vcpu);
ad312c7c 3315 ASSERT(VALID_PAGE(vcpu->arch.mmu.root_hpa));
6aa8b732 3316
e833240f 3317 gfn = gva >> PAGE_SHIFT;
6aa8b732 3318
e833240f 3319 return nonpaging_map(vcpu, gva & PAGE_MASK,
c7ba5b48 3320 error_code, gfn, prefault);
6aa8b732
AK
3321}
3322
7e1fbeac 3323static int kvm_arch_setup_async_pf(struct kvm_vcpu *vcpu, gva_t gva, gfn_t gfn)
af585b92
GN
3324{
3325 struct kvm_arch_async_pf arch;
fb67e14f 3326
7c90705b 3327 arch.token = (vcpu->arch.apf.id++ << 12) | vcpu->vcpu_id;
af585b92 3328 arch.gfn = gfn;
c4806acd 3329 arch.direct_map = vcpu->arch.mmu.direct_map;
fb67e14f 3330 arch.cr3 = vcpu->arch.mmu.get_cr3(vcpu);
af585b92 3331
e0ead41a 3332 return kvm_setup_async_pf(vcpu, gva, gfn_to_hva(vcpu->kvm, gfn), &arch);
af585b92
GN
3333}
3334
3335static bool can_do_async_pf(struct kvm_vcpu *vcpu)
3336{
3337 if (unlikely(!irqchip_in_kernel(vcpu->kvm) ||
3338 kvm_event_needs_reinjection(vcpu)))
3339 return false;
3340
3341 return kvm_x86_ops->interrupt_allowed(vcpu);
3342}
3343
78b2c54a 3344static bool try_async_pf(struct kvm_vcpu *vcpu, bool prefault, gfn_t gfn,
612819c3 3345 gva_t gva, pfn_t *pfn, bool write, bool *writable)
af585b92
GN
3346{
3347 bool async;
3348
612819c3 3349 *pfn = gfn_to_pfn_async(vcpu->kvm, gfn, &async, write, writable);
af585b92
GN
3350
3351 if (!async)
3352 return false; /* *pfn has correct page already */
3353
78b2c54a 3354 if (!prefault && can_do_async_pf(vcpu)) {
c9b263d2 3355 trace_kvm_try_async_get_page(gva, gfn);
af585b92
GN
3356 if (kvm_find_async_pf_gfn(vcpu, gfn)) {
3357 trace_kvm_async_pf_doublefault(gva, gfn);
3358 kvm_make_request(KVM_REQ_APF_HALT, vcpu);
3359 return true;
3360 } else if (kvm_arch_setup_async_pf(vcpu, gva, gfn))
3361 return true;
3362 }
3363
612819c3 3364 *pfn = gfn_to_pfn_prot(vcpu->kvm, gfn, write, writable);
af585b92
GN
3365
3366 return false;
3367}
3368
56028d08 3369static int tdp_page_fault(struct kvm_vcpu *vcpu, gva_t gpa, u32 error_code,
78b2c54a 3370 bool prefault)
fb72d167 3371{
35149e21 3372 pfn_t pfn;
fb72d167 3373 int r;
852e3c19 3374 int level;
936a5fe6 3375 int force_pt_level;
05da4558 3376 gfn_t gfn = gpa >> PAGE_SHIFT;
e930bffe 3377 unsigned long mmu_seq;
612819c3
MT
3378 int write = error_code & PFERR_WRITE_MASK;
3379 bool map_writable;
fb72d167
JR
3380
3381 ASSERT(vcpu);
3382 ASSERT(VALID_PAGE(vcpu->arch.mmu.root_hpa));
3383
f8f55942
XG
3384 if (unlikely(error_code & PFERR_RSVD_MASK)) {
3385 r = handle_mmio_page_fault(vcpu, gpa, error_code, true);
3386
3387 if (likely(r != RET_MMIO_PF_INVALID))
3388 return r;
3389 }
ce88decf 3390
fb72d167
JR
3391 r = mmu_topup_memory_caches(vcpu);
3392 if (r)
3393 return r;
3394
936a5fe6
AA
3395 force_pt_level = mapping_level_dirty_bitmap(vcpu, gfn);
3396 if (likely(!force_pt_level)) {
3397 level = mapping_level(vcpu, gfn);
3398 gfn &= ~(KVM_PAGES_PER_HPAGE(level) - 1);
3399 } else
3400 level = PT_PAGE_TABLE_LEVEL;
852e3c19 3401
c7ba5b48
XG
3402 if (fast_page_fault(vcpu, gpa, level, error_code))
3403 return 0;
3404
e930bffe 3405 mmu_seq = vcpu->kvm->mmu_notifier_seq;
4c2155ce 3406 smp_rmb();
af585b92 3407
78b2c54a 3408 if (try_async_pf(vcpu, prefault, gfn, gpa, &pfn, write, &map_writable))
af585b92
GN
3409 return 0;
3410
d7c55201
XG
3411 if (handle_abnormal_pfn(vcpu, 0, gfn, pfn, ACC_ALL, &r))
3412 return r;
3413
fb72d167 3414 spin_lock(&vcpu->kvm->mmu_lock);
8ca40a70 3415 if (mmu_notifier_retry(vcpu->kvm, mmu_seq))
e930bffe 3416 goto out_unlock;
450e0b41 3417 make_mmu_pages_available(vcpu);
936a5fe6
AA
3418 if (likely(!force_pt_level))
3419 transparent_hugepage_adjust(vcpu, &gfn, &pfn, &level);
612819c3 3420 r = __direct_map(vcpu, gpa, write, map_writable,
2ec4739d 3421 level, gfn, pfn, prefault);
fb72d167 3422 spin_unlock(&vcpu->kvm->mmu_lock);
fb72d167
JR
3423
3424 return r;
e930bffe
AA
3425
3426out_unlock:
3427 spin_unlock(&vcpu->kvm->mmu_lock);
3428 kvm_release_pfn_clean(pfn);
3429 return 0;
fb72d167
JR
3430}
3431
8a3c1a33
PB
3432static void nonpaging_init_context(struct kvm_vcpu *vcpu,
3433 struct kvm_mmu *context)
6aa8b732 3434{
6aa8b732 3435 context->page_fault = nonpaging_page_fault;
6aa8b732 3436 context->gva_to_gpa = nonpaging_gva_to_gpa;
e8bc217a 3437 context->sync_page = nonpaging_sync_page;
a7052897 3438 context->invlpg = nonpaging_invlpg;
0f53b5b1 3439 context->update_pte = nonpaging_update_pte;
cea0f0e7 3440 context->root_level = 0;
6aa8b732 3441 context->shadow_root_level = PT32E_ROOT_LEVEL;
17c3ba9d 3442 context->root_hpa = INVALID_PAGE;
c5a78f2b 3443 context->direct_map = true;
2d48a985 3444 context->nx = false;
6aa8b732
AK
3445}
3446
d835dfec 3447void kvm_mmu_flush_tlb(struct kvm_vcpu *vcpu)
6aa8b732 3448{
1165f5fe 3449 ++vcpu->stat.tlb_flush;
a8eeb04a 3450 kvm_make_request(KVM_REQ_TLB_FLUSH, vcpu);
6aa8b732 3451}
bfd0a56b 3452EXPORT_SYMBOL_GPL(kvm_mmu_flush_tlb);
6aa8b732 3453
d8d173da 3454void kvm_mmu_new_cr3(struct kvm_vcpu *vcpu)
6aa8b732 3455{
cea0f0e7 3456 mmu_free_roots(vcpu);
6aa8b732
AK
3457}
3458
5777ed34
JR
3459static unsigned long get_cr3(struct kvm_vcpu *vcpu)
3460{
9f8fe504 3461 return kvm_read_cr3(vcpu);
5777ed34
JR
3462}
3463
6389ee94
AK
3464static void inject_page_fault(struct kvm_vcpu *vcpu,
3465 struct x86_exception *fault)
6aa8b732 3466{
6389ee94 3467 vcpu->arch.mmu.inject_page_fault(vcpu, fault);
6aa8b732
AK
3468}
3469
f2fd125d
XG
3470static bool sync_mmio_spte(struct kvm *kvm, u64 *sptep, gfn_t gfn,
3471 unsigned access, int *nr_present)
ce88decf
XG
3472{
3473 if (unlikely(is_mmio_spte(*sptep))) {
3474 if (gfn != get_mmio_spte_gfn(*sptep)) {
3475 mmu_spte_clear_no_track(sptep);
3476 return true;
3477 }
3478
3479 (*nr_present)++;
f2fd125d 3480 mark_mmio_spte(kvm, sptep, gfn, access);
ce88decf
XG
3481 return true;
3482 }
3483
3484 return false;
3485}
3486
6fd01b71
AK
3487static inline bool is_last_gpte(struct kvm_mmu *mmu, unsigned level, unsigned gpte)
3488{
3489 unsigned index;
3490
3491 index = level - 1;
3492 index |= (gpte & PT_PAGE_SIZE_MASK) >> (PT_PAGE_SIZE_SHIFT - 2);
3493 return mmu->last_pte_bitmap & (1 << index);
3494}
3495
37406aaa
NHE
3496#define PTTYPE_EPT 18 /* arbitrary */
3497#define PTTYPE PTTYPE_EPT
3498#include "paging_tmpl.h"
3499#undef PTTYPE
3500
6aa8b732
AK
3501#define PTTYPE 64
3502#include "paging_tmpl.h"
3503#undef PTTYPE
3504
3505#define PTTYPE 32
3506#include "paging_tmpl.h"
3507#undef PTTYPE
3508
52fde8df 3509static void reset_rsvds_bits_mask(struct kvm_vcpu *vcpu,
4d6931c3 3510 struct kvm_mmu *context)
82725b20 3511{
82725b20
DE
3512 int maxphyaddr = cpuid_maxphyaddr(vcpu);
3513 u64 exb_bit_rsvd = 0;
3514
25d92081
YZ
3515 context->bad_mt_xwr = 0;
3516
2d48a985 3517 if (!context->nx)
82725b20 3518 exb_bit_rsvd = rsvd_bits(63, 63);
4d6931c3 3519 switch (context->root_level) {
82725b20
DE
3520 case PT32_ROOT_LEVEL:
3521 /* no rsvd bits for 2 level 4K page table entries */
3522 context->rsvd_bits_mask[0][1] = 0;
3523 context->rsvd_bits_mask[0][0] = 0;
f815bce8
XG
3524 context->rsvd_bits_mask[1][0] = context->rsvd_bits_mask[0][0];
3525
3526 if (!is_pse(vcpu)) {
3527 context->rsvd_bits_mask[1][1] = 0;
3528 break;
3529 }
3530
82725b20
DE
3531 if (is_cpuid_PSE36())
3532 /* 36bits PSE 4MB page */
3533 context->rsvd_bits_mask[1][1] = rsvd_bits(17, 21);
3534 else
3535 /* 32 bits PSE 4MB page */
3536 context->rsvd_bits_mask[1][1] = rsvd_bits(13, 21);
82725b20
DE
3537 break;
3538 case PT32E_ROOT_LEVEL:
20c466b5
DE
3539 context->rsvd_bits_mask[0][2] =
3540 rsvd_bits(maxphyaddr, 63) |
3541 rsvd_bits(7, 8) | rsvd_bits(1, 2); /* PDPTE */
82725b20 3542 context->rsvd_bits_mask[0][1] = exb_bit_rsvd |
4c26b4cd 3543 rsvd_bits(maxphyaddr, 62); /* PDE */
82725b20
DE
3544 context->rsvd_bits_mask[0][0] = exb_bit_rsvd |
3545 rsvd_bits(maxphyaddr, 62); /* PTE */
3546 context->rsvd_bits_mask[1][1] = exb_bit_rsvd |
3547 rsvd_bits(maxphyaddr, 62) |
3548 rsvd_bits(13, 20); /* large page */
f815bce8 3549 context->rsvd_bits_mask[1][0] = context->rsvd_bits_mask[0][0];
82725b20
DE
3550 break;
3551 case PT64_ROOT_LEVEL:
3552 context->rsvd_bits_mask[0][3] = exb_bit_rsvd |
3553 rsvd_bits(maxphyaddr, 51) | rsvd_bits(7, 8);
3554 context->rsvd_bits_mask[0][2] = exb_bit_rsvd |
3555 rsvd_bits(maxphyaddr, 51) | rsvd_bits(7, 8);
3556 context->rsvd_bits_mask[0][1] = exb_bit_rsvd |
4c26b4cd 3557 rsvd_bits(maxphyaddr, 51);
82725b20
DE
3558 context->rsvd_bits_mask[0][0] = exb_bit_rsvd |
3559 rsvd_bits(maxphyaddr, 51);
3560 context->rsvd_bits_mask[1][3] = context->rsvd_bits_mask[0][3];
e04da980
JR
3561 context->rsvd_bits_mask[1][2] = exb_bit_rsvd |
3562 rsvd_bits(maxphyaddr, 51) |
3563 rsvd_bits(13, 29);
82725b20 3564 context->rsvd_bits_mask[1][1] = exb_bit_rsvd |
4c26b4cd
SY
3565 rsvd_bits(maxphyaddr, 51) |
3566 rsvd_bits(13, 20); /* large page */
f815bce8 3567 context->rsvd_bits_mask[1][0] = context->rsvd_bits_mask[0][0];
82725b20
DE
3568 break;
3569 }
3570}
3571
25d92081
YZ
3572static void reset_rsvds_bits_mask_ept(struct kvm_vcpu *vcpu,
3573 struct kvm_mmu *context, bool execonly)
3574{
3575 int maxphyaddr = cpuid_maxphyaddr(vcpu);
3576 int pte;
3577
3578 context->rsvd_bits_mask[0][3] =
3579 rsvd_bits(maxphyaddr, 51) | rsvd_bits(3, 7);
3580 context->rsvd_bits_mask[0][2] =
3581 rsvd_bits(maxphyaddr, 51) | rsvd_bits(3, 6);
3582 context->rsvd_bits_mask[0][1] =
3583 rsvd_bits(maxphyaddr, 51) | rsvd_bits(3, 6);
3584 context->rsvd_bits_mask[0][0] = rsvd_bits(maxphyaddr, 51);
3585
3586 /* large page */
3587 context->rsvd_bits_mask[1][3] = context->rsvd_bits_mask[0][3];
3588 context->rsvd_bits_mask[1][2] =
3589 rsvd_bits(maxphyaddr, 51) | rsvd_bits(12, 29);
3590 context->rsvd_bits_mask[1][1] =
3591 rsvd_bits(maxphyaddr, 51) | rsvd_bits(12, 20);
3592 context->rsvd_bits_mask[1][0] = context->rsvd_bits_mask[0][0];
3593
3594 for (pte = 0; pte < 64; pte++) {
3595 int rwx_bits = pte & 7;
3596 int mt = pte >> 3;
3597 if (mt == 0x2 || mt == 0x3 || mt == 0x7 ||
3598 rwx_bits == 0x2 || rwx_bits == 0x6 ||
3599 (rwx_bits == 0x4 && !execonly))
3600 context->bad_mt_xwr |= (1ull << pte);
3601 }
3602}
3603
3604static void update_permission_bitmask(struct kvm_vcpu *vcpu,
3605 struct kvm_mmu *mmu, bool ept)
97d64b78
AK
3606{
3607 unsigned bit, byte, pfec;
3608 u8 map;
3609 bool fault, x, w, u, wf, uf, ff, smep;
3610
3611 smep = kvm_read_cr4_bits(vcpu, X86_CR4_SMEP);
3612 for (byte = 0; byte < ARRAY_SIZE(mmu->permissions); ++byte) {
3613 pfec = byte << 1;
3614 map = 0;
3615 wf = pfec & PFERR_WRITE_MASK;
3616 uf = pfec & PFERR_USER_MASK;
3617 ff = pfec & PFERR_FETCH_MASK;
3618 for (bit = 0; bit < 8; ++bit) {
3619 x = bit & ACC_EXEC_MASK;
3620 w = bit & ACC_WRITE_MASK;
3621 u = bit & ACC_USER_MASK;
3622
25d92081
YZ
3623 if (!ept) {
3624 /* Not really needed: !nx will cause pte.nx to fault */
3625 x |= !mmu->nx;
3626 /* Allow supervisor writes if !cr0.wp */
3627 w |= !is_write_protection(vcpu) && !uf;
3628 /* Disallow supervisor fetches of user code if cr4.smep */
3629 x &= !(smep && u && !uf);
3630 } else
3631 /* Not really needed: no U/S accesses on ept */
3632 u = 1;
97d64b78
AK
3633
3634 fault = (ff && !x) || (uf && !u) || (wf && !w);
3635 map |= fault << bit;
3636 }
3637 mmu->permissions[byte] = map;
3638 }
3639}
3640
6fd01b71
AK
3641static void update_last_pte_bitmap(struct kvm_vcpu *vcpu, struct kvm_mmu *mmu)
3642{
3643 u8 map;
3644 unsigned level, root_level = mmu->root_level;
3645 const unsigned ps_set_index = 1 << 2; /* bit 2 of index: ps */
3646
3647 if (root_level == PT32E_ROOT_LEVEL)
3648 --root_level;
3649 /* PT_PAGE_TABLE_LEVEL always terminates */
3650 map = 1 | (1 << ps_set_index);
3651 for (level = PT_DIRECTORY_LEVEL; level <= root_level; ++level) {
3652 if (level <= PT_PDPE_LEVEL
3653 && (mmu->root_level >= PT32E_ROOT_LEVEL || is_pse(vcpu)))
3654 map |= 1 << (ps_set_index | (level - 1));
3655 }
3656 mmu->last_pte_bitmap = map;
3657}
3658
8a3c1a33
PB
3659static void paging64_init_context_common(struct kvm_vcpu *vcpu,
3660 struct kvm_mmu *context,
3661 int level)
6aa8b732 3662{
2d48a985 3663 context->nx = is_nx(vcpu);
4d6931c3 3664 context->root_level = level;
2d48a985 3665
4d6931c3 3666 reset_rsvds_bits_mask(vcpu, context);
25d92081 3667 update_permission_bitmask(vcpu, context, false);
6fd01b71 3668 update_last_pte_bitmap(vcpu, context);
6aa8b732
AK
3669
3670 ASSERT(is_pae(vcpu));
6aa8b732 3671 context->page_fault = paging64_page_fault;
6aa8b732 3672 context->gva_to_gpa = paging64_gva_to_gpa;
e8bc217a 3673 context->sync_page = paging64_sync_page;
a7052897 3674 context->invlpg = paging64_invlpg;
0f53b5b1 3675 context->update_pte = paging64_update_pte;
17ac10ad 3676 context->shadow_root_level = level;
17c3ba9d 3677 context->root_hpa = INVALID_PAGE;
c5a78f2b 3678 context->direct_map = false;
6aa8b732
AK
3679}
3680
8a3c1a33
PB
3681static void paging64_init_context(struct kvm_vcpu *vcpu,
3682 struct kvm_mmu *context)
17ac10ad 3683{
8a3c1a33 3684 paging64_init_context_common(vcpu, context, PT64_ROOT_LEVEL);
17ac10ad
AK
3685}
3686
8a3c1a33
PB
3687static void paging32_init_context(struct kvm_vcpu *vcpu,
3688 struct kvm_mmu *context)
6aa8b732 3689{
2d48a985 3690 context->nx = false;
4d6931c3 3691 context->root_level = PT32_ROOT_LEVEL;
2d48a985 3692
4d6931c3 3693 reset_rsvds_bits_mask(vcpu, context);
25d92081 3694 update_permission_bitmask(vcpu, context, false);
6fd01b71 3695 update_last_pte_bitmap(vcpu, context);
6aa8b732 3696
6aa8b732 3697 context->page_fault = paging32_page_fault;
6aa8b732 3698 context->gva_to_gpa = paging32_gva_to_gpa;
e8bc217a 3699 context->sync_page = paging32_sync_page;
a7052897 3700 context->invlpg = paging32_invlpg;
0f53b5b1 3701 context->update_pte = paging32_update_pte;
6aa8b732 3702 context->shadow_root_level = PT32E_ROOT_LEVEL;
17c3ba9d 3703 context->root_hpa = INVALID_PAGE;
c5a78f2b 3704 context->direct_map = false;
6aa8b732
AK
3705}
3706
8a3c1a33
PB
3707static void paging32E_init_context(struct kvm_vcpu *vcpu,
3708 struct kvm_mmu *context)
6aa8b732 3709{
8a3c1a33 3710 paging64_init_context_common(vcpu, context, PT32E_ROOT_LEVEL);
6aa8b732
AK
3711}
3712
8a3c1a33 3713static void init_kvm_tdp_mmu(struct kvm_vcpu *vcpu)
fb72d167 3714{
14dfe855 3715 struct kvm_mmu *context = vcpu->arch.walk_mmu;
fb72d167 3716
c445f8ef 3717 context->base_role.word = 0;
fb72d167 3718 context->page_fault = tdp_page_fault;
e8bc217a 3719 context->sync_page = nonpaging_sync_page;
a7052897 3720 context->invlpg = nonpaging_invlpg;
0f53b5b1 3721 context->update_pte = nonpaging_update_pte;
67253af5 3722 context->shadow_root_level = kvm_x86_ops->get_tdp_level();
fb72d167 3723 context->root_hpa = INVALID_PAGE;
c5a78f2b 3724 context->direct_map = true;
1c97f0a0 3725 context->set_cr3 = kvm_x86_ops->set_tdp_cr3;
5777ed34 3726 context->get_cr3 = get_cr3;
e4e517b4 3727 context->get_pdptr = kvm_pdptr_read;
cb659db8 3728 context->inject_page_fault = kvm_inject_page_fault;
fb72d167
JR
3729
3730 if (!is_paging(vcpu)) {
2d48a985 3731 context->nx = false;
fb72d167
JR
3732 context->gva_to_gpa = nonpaging_gva_to_gpa;
3733 context->root_level = 0;
3734 } else if (is_long_mode(vcpu)) {
2d48a985 3735 context->nx = is_nx(vcpu);
fb72d167 3736 context->root_level = PT64_ROOT_LEVEL;
4d6931c3
DB
3737 reset_rsvds_bits_mask(vcpu, context);
3738 context->gva_to_gpa = paging64_gva_to_gpa;
fb72d167 3739 } else if (is_pae(vcpu)) {
2d48a985 3740 context->nx = is_nx(vcpu);
fb72d167 3741 context->root_level = PT32E_ROOT_LEVEL;
4d6931c3
DB
3742 reset_rsvds_bits_mask(vcpu, context);
3743 context->gva_to_gpa = paging64_gva_to_gpa;
fb72d167 3744 } else {
2d48a985 3745 context->nx = false;
fb72d167 3746 context->root_level = PT32_ROOT_LEVEL;
4d6931c3
DB
3747 reset_rsvds_bits_mask(vcpu, context);
3748 context->gva_to_gpa = paging32_gva_to_gpa;
fb72d167
JR
3749 }
3750
25d92081 3751 update_permission_bitmask(vcpu, context, false);
6fd01b71 3752 update_last_pte_bitmap(vcpu, context);
fb72d167
JR
3753}
3754
8a3c1a33 3755void kvm_init_shadow_mmu(struct kvm_vcpu *vcpu, struct kvm_mmu *context)
6aa8b732 3756{
411c588d 3757 bool smep = kvm_read_cr4_bits(vcpu, X86_CR4_SMEP);
6aa8b732 3758 ASSERT(vcpu);
ad312c7c 3759 ASSERT(!VALID_PAGE(vcpu->arch.mmu.root_hpa));
6aa8b732
AK
3760
3761 if (!is_paging(vcpu))
8a3c1a33 3762 nonpaging_init_context(vcpu, context);
a9058ecd 3763 else if (is_long_mode(vcpu))
8a3c1a33 3764 paging64_init_context(vcpu, context);
6aa8b732 3765 else if (is_pae(vcpu))
8a3c1a33 3766 paging32E_init_context(vcpu, context);
6aa8b732 3767 else
8a3c1a33 3768 paging32_init_context(vcpu, context);
a770f6f2 3769
2c9afa52 3770 vcpu->arch.mmu.base_role.nxe = is_nx(vcpu);
5b7e0102 3771 vcpu->arch.mmu.base_role.cr4_pae = !!is_pae(vcpu);
f43addd4 3772 vcpu->arch.mmu.base_role.cr0_wp = is_write_protection(vcpu);
411c588d
AK
3773 vcpu->arch.mmu.base_role.smep_andnot_wp
3774 = smep && !is_write_protection(vcpu);
52fde8df
JR
3775}
3776EXPORT_SYMBOL_GPL(kvm_init_shadow_mmu);
3777
8a3c1a33 3778void kvm_init_shadow_ept_mmu(struct kvm_vcpu *vcpu, struct kvm_mmu *context,
155a97a3
NHE
3779 bool execonly)
3780{
3781 ASSERT(vcpu);
3782 ASSERT(!VALID_PAGE(vcpu->arch.mmu.root_hpa));
3783
3784 context->shadow_root_level = kvm_x86_ops->get_tdp_level();
3785
3786 context->nx = true;
155a97a3
NHE
3787 context->page_fault = ept_page_fault;
3788 context->gva_to_gpa = ept_gva_to_gpa;
3789 context->sync_page = ept_sync_page;
3790 context->invlpg = ept_invlpg;
3791 context->update_pte = ept_update_pte;
155a97a3
NHE
3792 context->root_level = context->shadow_root_level;
3793 context->root_hpa = INVALID_PAGE;
3794 context->direct_map = false;
3795
3796 update_permission_bitmask(vcpu, context, true);
3797 reset_rsvds_bits_mask_ept(vcpu, context, execonly);
155a97a3
NHE
3798}
3799EXPORT_SYMBOL_GPL(kvm_init_shadow_ept_mmu);
3800
8a3c1a33 3801static void init_kvm_softmmu(struct kvm_vcpu *vcpu)
52fde8df 3802{
8a3c1a33 3803 kvm_init_shadow_mmu(vcpu, vcpu->arch.walk_mmu);
14dfe855
JR
3804 vcpu->arch.walk_mmu->set_cr3 = kvm_x86_ops->set_cr3;
3805 vcpu->arch.walk_mmu->get_cr3 = get_cr3;
e4e517b4 3806 vcpu->arch.walk_mmu->get_pdptr = kvm_pdptr_read;
14dfe855 3807 vcpu->arch.walk_mmu->inject_page_fault = kvm_inject_page_fault;
6aa8b732
AK
3808}
3809
8a3c1a33 3810static void init_kvm_nested_mmu(struct kvm_vcpu *vcpu)
02f59dc9
JR
3811{
3812 struct kvm_mmu *g_context = &vcpu->arch.nested_mmu;
3813
3814 g_context->get_cr3 = get_cr3;
e4e517b4 3815 g_context->get_pdptr = kvm_pdptr_read;
02f59dc9
JR
3816 g_context->inject_page_fault = kvm_inject_page_fault;
3817
3818 /*
3819 * Note that arch.mmu.gva_to_gpa translates l2_gva to l1_gpa. The
3820 * translation of l2_gpa to l1_gpa addresses is done using the
3821 * arch.nested_mmu.gva_to_gpa function. Basically the gva_to_gpa
3822 * functions between mmu and nested_mmu are swapped.
3823 */
3824 if (!is_paging(vcpu)) {
2d48a985 3825 g_context->nx = false;
02f59dc9
JR
3826 g_context->root_level = 0;
3827 g_context->gva_to_gpa = nonpaging_gva_to_gpa_nested;
3828 } else if (is_long_mode(vcpu)) {
2d48a985 3829 g_context->nx = is_nx(vcpu);
02f59dc9 3830 g_context->root_level = PT64_ROOT_LEVEL;
4d6931c3 3831 reset_rsvds_bits_mask(vcpu, g_context);
02f59dc9
JR
3832 g_context->gva_to_gpa = paging64_gva_to_gpa_nested;
3833 } else if (is_pae(vcpu)) {
2d48a985 3834 g_context->nx = is_nx(vcpu);
02f59dc9 3835 g_context->root_level = PT32E_ROOT_LEVEL;
4d6931c3 3836 reset_rsvds_bits_mask(vcpu, g_context);
02f59dc9
JR
3837 g_context->gva_to_gpa = paging64_gva_to_gpa_nested;
3838 } else {
2d48a985 3839 g_context->nx = false;
02f59dc9 3840 g_context->root_level = PT32_ROOT_LEVEL;
4d6931c3 3841 reset_rsvds_bits_mask(vcpu, g_context);
02f59dc9
JR
3842 g_context->gva_to_gpa = paging32_gva_to_gpa_nested;
3843 }
3844
25d92081 3845 update_permission_bitmask(vcpu, g_context, false);
6fd01b71 3846 update_last_pte_bitmap(vcpu, g_context);
02f59dc9
JR
3847}
3848
8a3c1a33 3849static void init_kvm_mmu(struct kvm_vcpu *vcpu)
fb72d167 3850{
02f59dc9
JR
3851 if (mmu_is_nested(vcpu))
3852 return init_kvm_nested_mmu(vcpu);
3853 else if (tdp_enabled)
fb72d167
JR
3854 return init_kvm_tdp_mmu(vcpu);
3855 else
3856 return init_kvm_softmmu(vcpu);
3857}
3858
8a3c1a33 3859void kvm_mmu_reset_context(struct kvm_vcpu *vcpu)
6aa8b732
AK
3860{
3861 ASSERT(vcpu);
6aa8b732 3862
95f93af4 3863 kvm_mmu_unload(vcpu);
8a3c1a33 3864 init_kvm_mmu(vcpu);
17c3ba9d 3865}
8668a3c4 3866EXPORT_SYMBOL_GPL(kvm_mmu_reset_context);
17c3ba9d
AK
3867
3868int kvm_mmu_load(struct kvm_vcpu *vcpu)
6aa8b732 3869{
714b93da
AK
3870 int r;
3871
e2dec939 3872 r = mmu_topup_memory_caches(vcpu);
17c3ba9d
AK
3873 if (r)
3874 goto out;
8986ecc0 3875 r = mmu_alloc_roots(vcpu);
e2858b4a 3876 kvm_mmu_sync_roots(vcpu);
8986ecc0
MT
3877 if (r)
3878 goto out;
3662cb1c 3879 /* set_cr3() should ensure TLB has been flushed */
f43addd4 3880 vcpu->arch.mmu.set_cr3(vcpu, vcpu->arch.mmu.root_hpa);
714b93da
AK
3881out:
3882 return r;
6aa8b732 3883}
17c3ba9d
AK
3884EXPORT_SYMBOL_GPL(kvm_mmu_load);
3885
3886void kvm_mmu_unload(struct kvm_vcpu *vcpu)
3887{
3888 mmu_free_roots(vcpu);
95f93af4 3889 WARN_ON(VALID_PAGE(vcpu->arch.mmu.root_hpa));
17c3ba9d 3890}
4b16184c 3891EXPORT_SYMBOL_GPL(kvm_mmu_unload);
6aa8b732 3892
0028425f 3893static void mmu_pte_write_new_pte(struct kvm_vcpu *vcpu,
7c562522
XG
3894 struct kvm_mmu_page *sp, u64 *spte,
3895 const void *new)
0028425f 3896{
30945387 3897 if (sp->role.level != PT_PAGE_TABLE_LEVEL) {
7e4e4056
JR
3898 ++vcpu->kvm->stat.mmu_pde_zapped;
3899 return;
30945387 3900 }
0028425f 3901
4cee5764 3902 ++vcpu->kvm->stat.mmu_pte_updated;
7c562522 3903 vcpu->arch.mmu.update_pte(vcpu, sp, spte, new);
0028425f
AK
3904}
3905
79539cec
AK
3906static bool need_remote_flush(u64 old, u64 new)
3907{
3908 if (!is_shadow_present_pte(old))
3909 return false;
3910 if (!is_shadow_present_pte(new))
3911 return true;
3912 if ((old ^ new) & PT64_BASE_ADDR_MASK)
3913 return true;
53166229
GN
3914 old ^= shadow_nx_mask;
3915 new ^= shadow_nx_mask;
79539cec
AK
3916 return (old & ~new & PT64_PERM_MASK) != 0;
3917}
3918
0671a8e7
XG
3919static void mmu_pte_write_flush_tlb(struct kvm_vcpu *vcpu, bool zap_page,
3920 bool remote_flush, bool local_flush)
79539cec 3921{
0671a8e7
XG
3922 if (zap_page)
3923 return;
3924
3925 if (remote_flush)
79539cec 3926 kvm_flush_remote_tlbs(vcpu->kvm);
0671a8e7 3927 else if (local_flush)
79539cec
AK
3928 kvm_mmu_flush_tlb(vcpu);
3929}
3930
889e5cbc
XG
3931static u64 mmu_pte_write_fetch_gpte(struct kvm_vcpu *vcpu, gpa_t *gpa,
3932 const u8 *new, int *bytes)
da4a00f0 3933{
889e5cbc
XG
3934 u64 gentry;
3935 int r;
72016f3a 3936
72016f3a
AK
3937 /*
3938 * Assume that the pte write on a page table of the same type
49b26e26
XG
3939 * as the current vcpu paging mode since we update the sptes only
3940 * when they have the same mode.
72016f3a 3941 */
889e5cbc 3942 if (is_pae(vcpu) && *bytes == 4) {
72016f3a 3943 /* Handle a 32-bit guest writing two halves of a 64-bit gpte */
889e5cbc
XG
3944 *gpa &= ~(gpa_t)7;
3945 *bytes = 8;
116eb3d3 3946 r = kvm_read_guest(vcpu->kvm, *gpa, &gentry, 8);
72016f3a
AK
3947 if (r)
3948 gentry = 0;
08e850c6
AK
3949 new = (const u8 *)&gentry;
3950 }
3951
889e5cbc 3952 switch (*bytes) {
08e850c6
AK
3953 case 4:
3954 gentry = *(const u32 *)new;
3955 break;
3956 case 8:
3957 gentry = *(const u64 *)new;
3958 break;
3959 default:
3960 gentry = 0;
3961 break;
72016f3a
AK
3962 }
3963
889e5cbc
XG
3964 return gentry;
3965}
3966
3967/*
3968 * If we're seeing too many writes to a page, it may no longer be a page table,
3969 * or we may be forking, in which case it is better to unmap the page.
3970 */
a138fe75 3971static bool detect_write_flooding(struct kvm_mmu_page *sp)
889e5cbc 3972{
a30f47cb
XG
3973 /*
3974 * Skip write-flooding detected for the sp whose level is 1, because
3975 * it can become unsync, then the guest page is not write-protected.
3976 */
f71fa31f 3977 if (sp->role.level == PT_PAGE_TABLE_LEVEL)
a30f47cb 3978 return false;
3246af0e 3979
a30f47cb 3980 return ++sp->write_flooding_count >= 3;
889e5cbc
XG
3981}
3982
3983/*
3984 * Misaligned accesses are too much trouble to fix up; also, they usually
3985 * indicate a page is not used as a page table.
3986 */
3987static bool detect_write_misaligned(struct kvm_mmu_page *sp, gpa_t gpa,
3988 int bytes)
3989{
3990 unsigned offset, pte_size, misaligned;
3991
3992 pgprintk("misaligned: gpa %llx bytes %d role %x\n",
3993 gpa, bytes, sp->role.word);
3994
3995 offset = offset_in_page(gpa);
3996 pte_size = sp->role.cr4_pae ? 8 : 4;
5d9ca30e
XG
3997
3998 /*
3999 * Sometimes, the OS only writes the last one bytes to update status
4000 * bits, for example, in linux, andb instruction is used in clear_bit().
4001 */
4002 if (!(offset & (pte_size - 1)) && bytes == 1)
4003 return false;
4004
889e5cbc
XG
4005 misaligned = (offset ^ (offset + bytes - 1)) & ~(pte_size - 1);
4006 misaligned |= bytes < 4;
4007
4008 return misaligned;
4009}
4010
4011static u64 *get_written_sptes(struct kvm_mmu_page *sp, gpa_t gpa, int *nspte)
4012{
4013 unsigned page_offset, quadrant;
4014 u64 *spte;
4015 int level;
4016
4017 page_offset = offset_in_page(gpa);
4018 level = sp->role.level;
4019 *nspte = 1;
4020 if (!sp->role.cr4_pae) {
4021 page_offset <<= 1; /* 32->64 */
4022 /*
4023 * A 32-bit pde maps 4MB while the shadow pdes map
4024 * only 2MB. So we need to double the offset again
4025 * and zap two pdes instead of one.
4026 */
4027 if (level == PT32_ROOT_LEVEL) {
4028 page_offset &= ~7; /* kill rounding error */
4029 page_offset <<= 1;
4030 *nspte = 2;
4031 }
4032 quadrant = page_offset >> PAGE_SHIFT;
4033 page_offset &= ~PAGE_MASK;
4034 if (quadrant != sp->role.quadrant)
4035 return NULL;
4036 }
4037
4038 spte = &sp->spt[page_offset / sizeof(*spte)];
4039 return spte;
4040}
4041
4042void kvm_mmu_pte_write(struct kvm_vcpu *vcpu, gpa_t gpa,
4043 const u8 *new, int bytes)
4044{
4045 gfn_t gfn = gpa >> PAGE_SHIFT;
4046 union kvm_mmu_page_role mask = { .word = 0 };
4047 struct kvm_mmu_page *sp;
889e5cbc
XG
4048 LIST_HEAD(invalid_list);
4049 u64 entry, gentry, *spte;
4050 int npte;
a30f47cb 4051 bool remote_flush, local_flush, zap_page;
889e5cbc
XG
4052
4053 /*
4054 * If we don't have indirect shadow pages, it means no page is
4055 * write-protected, so we can exit simply.
4056 */
4057 if (!ACCESS_ONCE(vcpu->kvm->arch.indirect_shadow_pages))
4058 return;
4059
4060 zap_page = remote_flush = local_flush = false;
4061
4062 pgprintk("%s: gpa %llx bytes %d\n", __func__, gpa, bytes);
4063
4064 gentry = mmu_pte_write_fetch_gpte(vcpu, &gpa, new, &bytes);
4065
4066 /*
4067 * No need to care whether allocation memory is successful
4068 * or not since pte prefetch is skiped if it does not have
4069 * enough objects in the cache.
4070 */
4071 mmu_topup_memory_caches(vcpu);
4072
4073 spin_lock(&vcpu->kvm->mmu_lock);
4074 ++vcpu->kvm->stat.mmu_pte_write;
0375f7fa 4075 kvm_mmu_audit(vcpu, AUDIT_PRE_PTE_WRITE);
889e5cbc 4076
fa1de2bf 4077 mask.cr0_wp = mask.cr4_pae = mask.nxe = 1;
b67bfe0d 4078 for_each_gfn_indirect_valid_sp(vcpu->kvm, sp, gfn) {
a30f47cb 4079 if (detect_write_misaligned(sp, gpa, bytes) ||
a138fe75 4080 detect_write_flooding(sp)) {
0671a8e7 4081 zap_page |= !!kvm_mmu_prepare_zap_page(vcpu->kvm, sp,
f41d335a 4082 &invalid_list);
4cee5764 4083 ++vcpu->kvm->stat.mmu_flooded;
0e7bc4b9
AK
4084 continue;
4085 }
889e5cbc
XG
4086
4087 spte = get_written_sptes(sp, gpa, &npte);
4088 if (!spte)
4089 continue;
4090
0671a8e7 4091 local_flush = true;
ac1b714e 4092 while (npte--) {
79539cec 4093 entry = *spte;
38e3b2b2 4094 mmu_page_zap_pte(vcpu->kvm, sp, spte);
fa1de2bf
XG
4095 if (gentry &&
4096 !((sp->role.word ^ vcpu->arch.mmu.base_role.word)
f759e2b4 4097 & mask.word) && rmap_can_add(vcpu))
7c562522 4098 mmu_pte_write_new_pte(vcpu, sp, spte, &gentry);
9bb4f6b1 4099 if (need_remote_flush(entry, *spte))
0671a8e7 4100 remote_flush = true;
ac1b714e 4101 ++spte;
9b7a0325 4102 }
9b7a0325 4103 }
0671a8e7 4104 mmu_pte_write_flush_tlb(vcpu, zap_page, remote_flush, local_flush);
d98ba053 4105 kvm_mmu_commit_zap_page(vcpu->kvm, &invalid_list);
0375f7fa 4106 kvm_mmu_audit(vcpu, AUDIT_POST_PTE_WRITE);
aaee2c94 4107 spin_unlock(&vcpu->kvm->mmu_lock);
da4a00f0
AK
4108}
4109
a436036b
AK
4110int kvm_mmu_unprotect_page_virt(struct kvm_vcpu *vcpu, gva_t gva)
4111{
10589a46
MT
4112 gpa_t gpa;
4113 int r;
a436036b 4114
c5a78f2b 4115 if (vcpu->arch.mmu.direct_map)
60f24784
AK
4116 return 0;
4117
1871c602 4118 gpa = kvm_mmu_gva_to_gpa_read(vcpu, gva, NULL);
10589a46 4119
10589a46 4120 r = kvm_mmu_unprotect_page(vcpu->kvm, gpa >> PAGE_SHIFT);
1cb3f3ae 4121
10589a46 4122 return r;
a436036b 4123}
577bdc49 4124EXPORT_SYMBOL_GPL(kvm_mmu_unprotect_page_virt);
a436036b 4125
81f4f76b 4126static void make_mmu_pages_available(struct kvm_vcpu *vcpu)
ebeace86 4127{
d98ba053 4128 LIST_HEAD(invalid_list);
103ad25a 4129
81f4f76b
TY
4130 if (likely(kvm_mmu_available_pages(vcpu->kvm) >= KVM_MIN_FREE_MMU_PAGES))
4131 return;
4132
5da59607
TY
4133 while (kvm_mmu_available_pages(vcpu->kvm) < KVM_REFILL_PAGES) {
4134 if (!prepare_zap_oldest_mmu_page(vcpu->kvm, &invalid_list))
4135 break;
ebeace86 4136
4cee5764 4137 ++vcpu->kvm->stat.mmu_recycled;
ebeace86 4138 }
aa6bd187 4139 kvm_mmu_commit_zap_page(vcpu->kvm, &invalid_list);
ebeace86 4140}
ebeace86 4141
1cb3f3ae
XG
4142static bool is_mmio_page_fault(struct kvm_vcpu *vcpu, gva_t addr)
4143{
4144 if (vcpu->arch.mmu.direct_map || mmu_is_nested(vcpu))
4145 return vcpu_match_mmio_gpa(vcpu, addr);
4146
4147 return vcpu_match_mmio_gva(vcpu, addr);
4148}
4149
dc25e89e
AP
4150int kvm_mmu_page_fault(struct kvm_vcpu *vcpu, gva_t cr2, u32 error_code,
4151 void *insn, int insn_len)
3067714c 4152{
1cb3f3ae 4153 int r, emulation_type = EMULTYPE_RETRY;
3067714c
AK
4154 enum emulation_result er;
4155
56028d08 4156 r = vcpu->arch.mmu.page_fault(vcpu, cr2, error_code, false);
3067714c
AK
4157 if (r < 0)
4158 goto out;
4159
4160 if (!r) {
4161 r = 1;
4162 goto out;
4163 }
4164
1cb3f3ae
XG
4165 if (is_mmio_page_fault(vcpu, cr2))
4166 emulation_type = 0;
4167
4168 er = x86_emulate_instruction(vcpu, cr2, emulation_type, insn, insn_len);
3067714c
AK
4169
4170 switch (er) {
4171 case EMULATE_DONE:
4172 return 1;
ac0a48c3 4173 case EMULATE_USER_EXIT:
3067714c 4174 ++vcpu->stat.mmio_exits;
6d77dbfc 4175 /* fall through */
3067714c 4176 case EMULATE_FAIL:
3f5d18a9 4177 return 0;
3067714c
AK
4178 default:
4179 BUG();
4180 }
4181out:
3067714c
AK
4182 return r;
4183}
4184EXPORT_SYMBOL_GPL(kvm_mmu_page_fault);
4185
a7052897
MT
4186void kvm_mmu_invlpg(struct kvm_vcpu *vcpu, gva_t gva)
4187{
a7052897 4188 vcpu->arch.mmu.invlpg(vcpu, gva);
a7052897
MT
4189 kvm_mmu_flush_tlb(vcpu);
4190 ++vcpu->stat.invlpg;
4191}
4192EXPORT_SYMBOL_GPL(kvm_mmu_invlpg);
4193
18552672
JR
4194void kvm_enable_tdp(void)
4195{
4196 tdp_enabled = true;
4197}
4198EXPORT_SYMBOL_GPL(kvm_enable_tdp);
4199
5f4cb662
JR
4200void kvm_disable_tdp(void)
4201{
4202 tdp_enabled = false;
4203}
4204EXPORT_SYMBOL_GPL(kvm_disable_tdp);
4205
6aa8b732
AK
4206static void free_mmu_pages(struct kvm_vcpu *vcpu)
4207{
ad312c7c 4208 free_page((unsigned long)vcpu->arch.mmu.pae_root);
81407ca5
JR
4209 if (vcpu->arch.mmu.lm_root != NULL)
4210 free_page((unsigned long)vcpu->arch.mmu.lm_root);
6aa8b732
AK
4211}
4212
4213static int alloc_mmu_pages(struct kvm_vcpu *vcpu)
4214{
17ac10ad 4215 struct page *page;
6aa8b732
AK
4216 int i;
4217
4218 ASSERT(vcpu);
4219
17ac10ad
AK
4220 /*
4221 * When emulating 32-bit mode, cr3 is only 32 bits even on x86_64.
4222 * Therefore we need to allocate shadow page tables in the first
4223 * 4GB of memory, which happens to fit the DMA32 zone.
4224 */
4225 page = alloc_page(GFP_KERNEL | __GFP_DMA32);
4226 if (!page)
d7fa6ab2
WY
4227 return -ENOMEM;
4228
ad312c7c 4229 vcpu->arch.mmu.pae_root = page_address(page);
17ac10ad 4230 for (i = 0; i < 4; ++i)
ad312c7c 4231 vcpu->arch.mmu.pae_root[i] = INVALID_PAGE;
17ac10ad 4232
6aa8b732 4233 return 0;
6aa8b732
AK
4234}
4235
8018c27b 4236int kvm_mmu_create(struct kvm_vcpu *vcpu)
6aa8b732 4237{
6aa8b732 4238 ASSERT(vcpu);
e459e322
XG
4239
4240 vcpu->arch.walk_mmu = &vcpu->arch.mmu;
4241 vcpu->arch.mmu.root_hpa = INVALID_PAGE;
4242 vcpu->arch.mmu.translate_gpa = translate_gpa;
4243 vcpu->arch.nested_mmu.translate_gpa = translate_nested_gpa;
6aa8b732 4244
8018c27b
IM
4245 return alloc_mmu_pages(vcpu);
4246}
6aa8b732 4247
8a3c1a33 4248void kvm_mmu_setup(struct kvm_vcpu *vcpu)
8018c27b
IM
4249{
4250 ASSERT(vcpu);
ad312c7c 4251 ASSERT(!VALID_PAGE(vcpu->arch.mmu.root_hpa));
2c264957 4252
8a3c1a33 4253 init_kvm_mmu(vcpu);
6aa8b732
AK
4254}
4255
90cb0529 4256void kvm_mmu_slot_remove_write_access(struct kvm *kvm, int slot)
6aa8b732 4257{
b99db1d3
TY
4258 struct kvm_memory_slot *memslot;
4259 gfn_t last_gfn;
4260 int i;
6aa8b732 4261
b99db1d3
TY
4262 memslot = id_to_memslot(kvm->memslots, slot);
4263 last_gfn = memslot->base_gfn + memslot->npages - 1;
6aa8b732 4264
9d1beefb
TY
4265 spin_lock(&kvm->mmu_lock);
4266
b99db1d3
TY
4267 for (i = PT_PAGE_TABLE_LEVEL;
4268 i < PT_PAGE_TABLE_LEVEL + KVM_NR_PAGE_SIZES; ++i) {
4269 unsigned long *rmapp;
4270 unsigned long last_index, index;
6aa8b732 4271
b99db1d3
TY
4272 rmapp = memslot->arch.rmap[i - PT_PAGE_TABLE_LEVEL];
4273 last_index = gfn_to_index(last_gfn, memslot->base_gfn, i);
da8dc75f 4274
b99db1d3
TY
4275 for (index = 0; index <= last_index; ++index, ++rmapp) {
4276 if (*rmapp)
4277 __rmap_write_protect(kvm, rmapp, false);
6b81b05e
TY
4278
4279 if (need_resched() || spin_needbreak(&kvm->mmu_lock)) {
4280 kvm_flush_remote_tlbs(kvm);
4281 cond_resched_lock(&kvm->mmu_lock);
4282 }
8234b22e 4283 }
6aa8b732 4284 }
b99db1d3 4285
171d595d 4286 kvm_flush_remote_tlbs(kvm);
9d1beefb 4287 spin_unlock(&kvm->mmu_lock);
6aa8b732 4288}
37a7d8b0 4289
e7d11c7a 4290#define BATCH_ZAP_PAGES 10
5304b8d3
XG
4291static void kvm_zap_obsolete_pages(struct kvm *kvm)
4292{
4293 struct kvm_mmu_page *sp, *node;
e7d11c7a 4294 int batch = 0;
5304b8d3
XG
4295
4296restart:
4297 list_for_each_entry_safe_reverse(sp, node,
4298 &kvm->arch.active_mmu_pages, link) {
e7d11c7a
XG
4299 int ret;
4300
5304b8d3
XG
4301 /*
4302 * No obsolete page exists before new created page since
4303 * active_mmu_pages is the FIFO list.
4304 */
4305 if (!is_obsolete_sp(kvm, sp))
4306 break;
4307
4308 /*
5304b8d3
XG
4309 * Since we are reversely walking the list and the invalid
4310 * list will be moved to the head, skip the invalid page
4311 * can help us to avoid the infinity list walking.
4312 */
4313 if (sp->role.invalid)
4314 continue;
4315
f34d251d
XG
4316 /*
4317 * Need not flush tlb since we only zap the sp with invalid
4318 * generation number.
4319 */
e7d11c7a 4320 if (batch >= BATCH_ZAP_PAGES &&
f34d251d 4321 cond_resched_lock(&kvm->mmu_lock)) {
e7d11c7a 4322 batch = 0;
5304b8d3
XG
4323 goto restart;
4324 }
4325
365c8868
XG
4326 ret = kvm_mmu_prepare_zap_page(kvm, sp,
4327 &kvm->arch.zapped_obsolete_pages);
e7d11c7a
XG
4328 batch += ret;
4329
4330 if (ret)
5304b8d3
XG
4331 goto restart;
4332 }
4333
f34d251d
XG
4334 /*
4335 * Should flush tlb before free page tables since lockless-walking
4336 * may use the pages.
4337 */
365c8868 4338 kvm_mmu_commit_zap_page(kvm, &kvm->arch.zapped_obsolete_pages);
5304b8d3
XG
4339}
4340
4341/*
4342 * Fast invalidate all shadow pages and use lock-break technique
4343 * to zap obsolete pages.
4344 *
4345 * It's required when memslot is being deleted or VM is being
4346 * destroyed, in these cases, we should ensure that KVM MMU does
4347 * not use any resource of the being-deleted slot or all slots
4348 * after calling the function.
4349 */
4350void kvm_mmu_invalidate_zap_all_pages(struct kvm *kvm)
4351{
4352 spin_lock(&kvm->mmu_lock);
35006126 4353 trace_kvm_mmu_invalidate_zap_all_pages(kvm);
5304b8d3
XG
4354 kvm->arch.mmu_valid_gen++;
4355
f34d251d
XG
4356 /*
4357 * Notify all vcpus to reload its shadow page table
4358 * and flush TLB. Then all vcpus will switch to new
4359 * shadow page table with the new mmu_valid_gen.
4360 *
4361 * Note: we should do this under the protection of
4362 * mmu-lock, otherwise, vcpu would purge shadow page
4363 * but miss tlb flush.
4364 */
4365 kvm_reload_remote_mmus(kvm);
4366
5304b8d3
XG
4367 kvm_zap_obsolete_pages(kvm);
4368 spin_unlock(&kvm->mmu_lock);
4369}
4370
365c8868
XG
4371static bool kvm_has_zapped_obsolete_pages(struct kvm *kvm)
4372{
4373 return unlikely(!list_empty_careful(&kvm->arch.zapped_obsolete_pages));
4374}
4375
f8f55942
XG
4376void kvm_mmu_invalidate_mmio_sptes(struct kvm *kvm)
4377{
4378 /*
4379 * The very rare case: if the generation-number is round,
4380 * zap all shadow pages.
f8f55942 4381 */
e6dff7d1 4382 if (unlikely(kvm_current_mmio_generation(kvm) >= MMIO_MAX_GEN)) {
7a2e8aaf 4383 printk_ratelimited(KERN_INFO "kvm: zapping shadow pages for mmio generation wraparound\n");
a8eca9dc 4384 kvm_mmu_invalidate_zap_all_pages(kvm);
7a2e8aaf 4385 }
f8f55942
XG
4386}
4387
70534a73
DC
4388static unsigned long
4389mmu_shrink_scan(struct shrinker *shrink, struct shrink_control *sc)
3ee16c81
IE
4390{
4391 struct kvm *kvm;
1495f230 4392 int nr_to_scan = sc->nr_to_scan;
70534a73 4393 unsigned long freed = 0;
3ee16c81 4394
2f303b74 4395 spin_lock(&kvm_lock);
3ee16c81
IE
4396
4397 list_for_each_entry(kvm, &vm_list, vm_list) {
3d56cbdf 4398 int idx;
d98ba053 4399 LIST_HEAD(invalid_list);
3ee16c81 4400
35f2d16b
TY
4401 /*
4402 * Never scan more than sc->nr_to_scan VM instances.
4403 * Will not hit this condition practically since we do not try
4404 * to shrink more than one VM and it is very unlikely to see
4405 * !n_used_mmu_pages so many times.
4406 */
4407 if (!nr_to_scan--)
4408 break;
19526396
GN
4409 /*
4410 * n_used_mmu_pages is accessed without holding kvm->mmu_lock
4411 * here. We may skip a VM instance errorneosly, but we do not
4412 * want to shrink a VM that only started to populate its MMU
4413 * anyway.
4414 */
365c8868
XG
4415 if (!kvm->arch.n_used_mmu_pages &&
4416 !kvm_has_zapped_obsolete_pages(kvm))
19526396 4417 continue;
19526396 4418
f656ce01 4419 idx = srcu_read_lock(&kvm->srcu);
3ee16c81 4420 spin_lock(&kvm->mmu_lock);
3ee16c81 4421
365c8868
XG
4422 if (kvm_has_zapped_obsolete_pages(kvm)) {
4423 kvm_mmu_commit_zap_page(kvm,
4424 &kvm->arch.zapped_obsolete_pages);
4425 goto unlock;
4426 }
4427
70534a73
DC
4428 if (prepare_zap_oldest_mmu_page(kvm, &invalid_list))
4429 freed++;
d98ba053 4430 kvm_mmu_commit_zap_page(kvm, &invalid_list);
19526396 4431
365c8868 4432unlock:
3ee16c81 4433 spin_unlock(&kvm->mmu_lock);
f656ce01 4434 srcu_read_unlock(&kvm->srcu, idx);
19526396 4435
70534a73
DC
4436 /*
4437 * unfair on small ones
4438 * per-vm shrinkers cry out
4439 * sadness comes quickly
4440 */
19526396
GN
4441 list_move_tail(&kvm->vm_list, &vm_list);
4442 break;
3ee16c81 4443 }
3ee16c81 4444
2f303b74 4445 spin_unlock(&kvm_lock);
70534a73 4446 return freed;
70534a73
DC
4447}
4448
4449static unsigned long
4450mmu_shrink_count(struct shrinker *shrink, struct shrink_control *sc)
4451{
45221ab6 4452 return percpu_counter_read_positive(&kvm_total_used_mmu_pages);
3ee16c81
IE
4453}
4454
4455static struct shrinker mmu_shrinker = {
70534a73
DC
4456 .count_objects = mmu_shrink_count,
4457 .scan_objects = mmu_shrink_scan,
3ee16c81
IE
4458 .seeks = DEFAULT_SEEKS * 10,
4459};
4460
2ddfd20e 4461static void mmu_destroy_caches(void)
b5a33a75 4462{
53c07b18
XG
4463 if (pte_list_desc_cache)
4464 kmem_cache_destroy(pte_list_desc_cache);
d3d25b04
AK
4465 if (mmu_page_header_cache)
4466 kmem_cache_destroy(mmu_page_header_cache);
b5a33a75
AK
4467}
4468
4469int kvm_mmu_module_init(void)
4470{
53c07b18
XG
4471 pte_list_desc_cache = kmem_cache_create("pte_list_desc",
4472 sizeof(struct pte_list_desc),
20c2df83 4473 0, 0, NULL);
53c07b18 4474 if (!pte_list_desc_cache)
b5a33a75
AK
4475 goto nomem;
4476
d3d25b04
AK
4477 mmu_page_header_cache = kmem_cache_create("kvm_mmu_page_header",
4478 sizeof(struct kvm_mmu_page),
20c2df83 4479 0, 0, NULL);
d3d25b04
AK
4480 if (!mmu_page_header_cache)
4481 goto nomem;
4482
45bf21a8
WY
4483 if (percpu_counter_init(&kvm_total_used_mmu_pages, 0))
4484 goto nomem;
4485
3ee16c81
IE
4486 register_shrinker(&mmu_shrinker);
4487
b5a33a75
AK
4488 return 0;
4489
4490nomem:
3ee16c81 4491 mmu_destroy_caches();
b5a33a75
AK
4492 return -ENOMEM;
4493}
4494
3ad82a7e
ZX
4495/*
4496 * Caculate mmu pages needed for kvm.
4497 */
4498unsigned int kvm_mmu_calculate_mmu_pages(struct kvm *kvm)
4499{
3ad82a7e
ZX
4500 unsigned int nr_mmu_pages;
4501 unsigned int nr_pages = 0;
bc6678a3 4502 struct kvm_memslots *slots;
be6ba0f0 4503 struct kvm_memory_slot *memslot;
3ad82a7e 4504
90d83dc3
LJ
4505 slots = kvm_memslots(kvm);
4506
be6ba0f0
XG
4507 kvm_for_each_memslot(memslot, slots)
4508 nr_pages += memslot->npages;
3ad82a7e
ZX
4509
4510 nr_mmu_pages = nr_pages * KVM_PERMILLE_MMU_PAGES / 1000;
4511 nr_mmu_pages = max(nr_mmu_pages,
4512 (unsigned int) KVM_MIN_ALLOC_MMU_PAGES);
4513
4514 return nr_mmu_pages;
4515}
4516
94d8b056
MT
4517int kvm_mmu_get_spte_hierarchy(struct kvm_vcpu *vcpu, u64 addr, u64 sptes[4])
4518{
4519 struct kvm_shadow_walk_iterator iterator;
c2a2ac2b 4520 u64 spte;
94d8b056
MT
4521 int nr_sptes = 0;
4522
37f6a4e2
MT
4523 if (!VALID_PAGE(vcpu->arch.mmu.root_hpa))
4524 return nr_sptes;
4525
c2a2ac2b
XG
4526 walk_shadow_page_lockless_begin(vcpu);
4527 for_each_shadow_entry_lockless(vcpu, addr, iterator, spte) {
4528 sptes[iterator.level-1] = spte;
94d8b056 4529 nr_sptes++;
c2a2ac2b 4530 if (!is_shadow_present_pte(spte))
94d8b056
MT
4531 break;
4532 }
c2a2ac2b 4533 walk_shadow_page_lockless_end(vcpu);
94d8b056
MT
4534
4535 return nr_sptes;
4536}
4537EXPORT_SYMBOL_GPL(kvm_mmu_get_spte_hierarchy);
4538
c42fffe3
XG
4539void kvm_mmu_destroy(struct kvm_vcpu *vcpu)
4540{
4541 ASSERT(vcpu);
4542
95f93af4 4543 kvm_mmu_unload(vcpu);
c42fffe3
XG
4544 free_mmu_pages(vcpu);
4545 mmu_free_memory_caches(vcpu);
b034cf01
XG
4546}
4547
b034cf01
XG
4548void kvm_mmu_module_exit(void)
4549{
4550 mmu_destroy_caches();
4551 percpu_counter_destroy(&kvm_total_used_mmu_pages);
4552 unregister_shrinker(&mmu_shrinker);
c42fffe3
XG
4553 mmu_audit_disable();
4554}