mmu.c 34.6 KB
Newer Older
A
Avi Kivity 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
/*
 * Kernel-based Virtual Machine driver for Linux
 *
 * This module enables machines with Intel VT-x extensions to run virtual
 * machines without emulation or binary translation.
 *
 * MMU support
 *
 * Copyright (C) 2006 Qumranet, Inc.
 *
 * Authors:
 *   Yaniv Kamay  <yaniv@qumranet.com>
 *   Avi Kivity   <avi@qumranet.com>
 *
 * This work is licensed under the terms of the GNU GPL, version 2.  See
 * the COPYING file in the top-level directory.
 *
 */
A
Avi Kivity 已提交
19 20 21 22

#include "vmx.h"
#include "kvm.h"

A
Avi Kivity 已提交
23 24 25 26 27 28
#include <linux/types.h>
#include <linux/string.h>
#include <linux/mm.h>
#include <linux/highmem.h>
#include <linux/module.h>

A
Avi Kivity 已提交
29 30
#include <asm/page.h>
#include <asm/cmpxchg.h>
A
Avi Kivity 已提交
31

32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
#undef MMU_DEBUG

#undef AUDIT

#ifdef AUDIT
static void kvm_mmu_audit(struct kvm_vcpu *vcpu, const char *msg);
#else
static void kvm_mmu_audit(struct kvm_vcpu *vcpu, const char *msg) {}
#endif

#ifdef MMU_DEBUG

#define pgprintk(x...) do { if (dbg) printk(x); } while (0)
#define rmap_printk(x...) do { if (dbg) printk(x); } while (0)

#else

#define pgprintk(x...) do { } while (0)
#define rmap_printk(x...) do { } while (0)

#endif

#if defined(MMU_DEBUG) || defined(AUDIT)
static int dbg = 1;
#endif
A
Avi Kivity 已提交
57

58 59 60
#ifndef MMU_DEBUG
#define ASSERT(x) do { } while (0)
#else
A
Avi Kivity 已提交
61 62 63 64 65
#define ASSERT(x)							\
	if (!(x)) {							\
		printk(KERN_WARNING "assertion failed %s:%d: %s\n",	\
		       __FILE__, __LINE__, #x);				\
	}
66
#endif
A
Avi Kivity 已提交
67

68 69 70 71
#define PT64_PT_BITS 9
#define PT64_ENT_PER_PAGE (1 << PT64_PT_BITS)
#define PT32_PT_BITS 10
#define PT32_ENT_PER_PAGE (1 << PT32_PT_BITS)
A
Avi Kivity 已提交
72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126

#define PT_WRITABLE_SHIFT 1

#define PT_PRESENT_MASK (1ULL << 0)
#define PT_WRITABLE_MASK (1ULL << PT_WRITABLE_SHIFT)
#define PT_USER_MASK (1ULL << 2)
#define PT_PWT_MASK (1ULL << 3)
#define PT_PCD_MASK (1ULL << 4)
#define PT_ACCESSED_MASK (1ULL << 5)
#define PT_DIRTY_MASK (1ULL << 6)
#define PT_PAGE_SIZE_MASK (1ULL << 7)
#define PT_PAT_MASK (1ULL << 7)
#define PT_GLOBAL_MASK (1ULL << 8)
#define PT64_NX_MASK (1ULL << 63)

#define PT_PAT_SHIFT 7
#define PT_DIR_PAT_SHIFT 12
#define PT_DIR_PAT_MASK (1ULL << PT_DIR_PAT_SHIFT)

#define PT32_DIR_PSE36_SIZE 4
#define PT32_DIR_PSE36_SHIFT 13
#define PT32_DIR_PSE36_MASK (((1ULL << PT32_DIR_PSE36_SIZE) - 1) << PT32_DIR_PSE36_SHIFT)


#define PT_FIRST_AVAIL_BITS_SHIFT 9
#define PT64_SECOND_AVAIL_BITS_SHIFT 52

#define PT_SHADOW_IO_MARK (1ULL << PT_FIRST_AVAIL_BITS_SHIFT)

#define VALID_PAGE(x) ((x) != INVALID_PAGE)

#define PT64_LEVEL_BITS 9

#define PT64_LEVEL_SHIFT(level) \
		( PAGE_SHIFT + (level - 1) * PT64_LEVEL_BITS )

#define PT64_LEVEL_MASK(level) \
		(((1ULL << PT64_LEVEL_BITS) - 1) << PT64_LEVEL_SHIFT(level))

#define PT64_INDEX(address, level)\
	(((address) >> PT64_LEVEL_SHIFT(level)) & ((1 << PT64_LEVEL_BITS) - 1))


#define PT32_LEVEL_BITS 10

#define PT32_LEVEL_SHIFT(level) \
		( PAGE_SHIFT + (level - 1) * PT32_LEVEL_BITS )

#define PT32_LEVEL_MASK(level) \
		(((1ULL << PT32_LEVEL_BITS) - 1) << PT32_LEVEL_SHIFT(level))

#define PT32_INDEX(address, level)\
	(((address) >> PT32_LEVEL_SHIFT(level)) & ((1 << PT32_LEVEL_BITS) - 1))


127
#define PT64_BASE_ADDR_MASK (((1ULL << 52) - 1) & ~(u64)(PAGE_SIZE-1))
A
Avi Kivity 已提交
128 129 130 131 132 133 134 135 136 137 138
#define PT64_DIR_BASE_ADDR_MASK \
	(PT64_BASE_ADDR_MASK & ~((1ULL << (PAGE_SHIFT + PT64_LEVEL_BITS)) - 1))

#define PT32_BASE_ADDR_MASK PAGE_MASK
#define PT32_DIR_BASE_ADDR_MASK \
	(PAGE_MASK & ~((1ULL << (PAGE_SHIFT + PT32_LEVEL_BITS)) - 1))


#define PFERR_PRESENT_MASK (1U << 0)
#define PFERR_WRITE_MASK (1U << 1)
#define PFERR_USER_MASK (1U << 2)
139
#define PFERR_FETCH_MASK (1U << 4)
A
Avi Kivity 已提交
140 141 142 143 144 145 146 147

#define PT64_ROOT_LEVEL 4
#define PT32_ROOT_LEVEL 2
#define PT32E_ROOT_LEVEL 3

#define PT_DIRECTORY_LEVEL 2
#define PT_PAGE_TABLE_LEVEL 1

148 149 150 151 152 153 154
#define RMAP_EXT 4

struct kvm_rmap_desc {
	u64 *shadow_ptes[RMAP_EXT];
	struct kvm_rmap_desc *more;
};

155 156
static struct kmem_cache *pte_chain_cache;
static struct kmem_cache *rmap_desc_cache;
157
static struct kmem_cache *mmu_page_header_cache;
158

A
Avi Kivity 已提交
159 160
static int is_write_protection(struct kvm_vcpu *vcpu)
{
161
	return vcpu->cr0 & X86_CR0_WP;
A
Avi Kivity 已提交
162 163 164 165 166 167 168
}

static int is_cpuid_PSE36(void)
{
	return 1;
}

169 170 171 172 173
static int is_nx(struct kvm_vcpu *vcpu)
{
	return vcpu->shadow_efer & EFER_NX;
}

A
Avi Kivity 已提交
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188
static int is_present_pte(unsigned long pte)
{
	return pte & PT_PRESENT_MASK;
}

static int is_writeble_pte(unsigned long pte)
{
	return pte & PT_WRITABLE_MASK;
}

static int is_io_pte(unsigned long pte)
{
	return pte & PT_SHADOW_IO_MARK;
}

189 190 191 192 193 194
static int is_rmap_pte(u64 pte)
{
	return (pte & (PT_WRITABLE_MASK | PT_PRESENT_MASK))
		== (PT_WRITABLE_MASK | PT_PRESENT_MASK);
}

195 196 197 198 199 200 201 202 203
static void set_shadow_pte(u64 *sptep, u64 spte)
{
#ifdef CONFIG_X86_64
	set_64bit((unsigned long *)sptep, spte);
#else
	set_64bit((unsigned long long *)sptep, spte);
#endif
}

204
static int mmu_topup_memory_cache(struct kvm_mmu_memory_cache *cache,
205
				  struct kmem_cache *base_cache, int min)
206 207 208 209
{
	void *obj;

	if (cache->nobjs >= min)
210
		return 0;
211
	while (cache->nobjs < ARRAY_SIZE(cache->objects)) {
212
		obj = kmem_cache_zalloc(base_cache, GFP_KERNEL);
213
		if (!obj)
214
			return -ENOMEM;
215 216
		cache->objects[cache->nobjs++] = obj;
	}
217
	return 0;
218 219 220 221 222 223 224 225
}

static void mmu_free_memory_cache(struct kvm_mmu_memory_cache *mc)
{
	while (mc->nobjs)
		kfree(mc->objects[--mc->nobjs]);
}

A
Avi Kivity 已提交
226
static int mmu_topup_memory_cache_page(struct kvm_mmu_memory_cache *cache,
227
				       int min)
A
Avi Kivity 已提交
228 229 230 231 232 233
{
	struct page *page;

	if (cache->nobjs >= min)
		return 0;
	while (cache->nobjs < ARRAY_SIZE(cache->objects)) {
234
		page = alloc_page(GFP_KERNEL);
A
Avi Kivity 已提交
235 236 237 238 239 240 241 242 243 244 245
		if (!page)
			return -ENOMEM;
		set_page_private(page, 0);
		cache->objects[cache->nobjs++] = page_address(page);
	}
	return 0;
}

static void mmu_free_memory_cache_page(struct kvm_mmu_memory_cache *mc)
{
	while (mc->nobjs)
246
		free_page((unsigned long)mc->objects[--mc->nobjs]);
A
Avi Kivity 已提交
247 248
}

249
static int mmu_topup_memory_caches(struct kvm_vcpu *vcpu)
250
{
251 252
	int r;

253
	kvm_mmu_free_some_pages(vcpu);
254
	r = mmu_topup_memory_cache(&vcpu->mmu_pte_chain_cache,
255
				   pte_chain_cache, 4);
256 257 258
	if (r)
		goto out;
	r = mmu_topup_memory_cache(&vcpu->mmu_rmap_desc_cache,
259
				   rmap_desc_cache, 1);
260 261
	if (r)
		goto out;
262
	r = mmu_topup_memory_cache_page(&vcpu->mmu_page_cache, 4);
263 264 265
	if (r)
		goto out;
	r = mmu_topup_memory_cache(&vcpu->mmu_page_header_cache,
266
				   mmu_page_header_cache, 4);
267 268
out:
	return r;
269 270 271 272 273 274
}

static void mmu_free_memory_caches(struct kvm_vcpu *vcpu)
{
	mmu_free_memory_cache(&vcpu->mmu_pte_chain_cache);
	mmu_free_memory_cache(&vcpu->mmu_rmap_desc_cache);
A
Avi Kivity 已提交
275
	mmu_free_memory_cache_page(&vcpu->mmu_page_cache);
276
	mmu_free_memory_cache(&vcpu->mmu_page_header_cache);
277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295
}

static void *mmu_memory_cache_alloc(struct kvm_mmu_memory_cache *mc,
				    size_t size)
{
	void *p;

	BUG_ON(!mc->nobjs);
	p = mc->objects[--mc->nobjs];
	memset(p, 0, size);
	return p;
}

static struct kvm_pte_chain *mmu_alloc_pte_chain(struct kvm_vcpu *vcpu)
{
	return mmu_memory_cache_alloc(&vcpu->mmu_pte_chain_cache,
				      sizeof(struct kvm_pte_chain));
}

296
static void mmu_free_pte_chain(struct kvm_pte_chain *pc)
297
{
298
	kfree(pc);
299 300 301 302 303 304 305 306
}

static struct kvm_rmap_desc *mmu_alloc_rmap_desc(struct kvm_vcpu *vcpu)
{
	return mmu_memory_cache_alloc(&vcpu->mmu_rmap_desc_cache,
				      sizeof(struct kvm_rmap_desc));
}

307
static void mmu_free_rmap_desc(struct kvm_rmap_desc *rd)
308
{
309
	kfree(rd);
310 311
}

312 313 314 315 316 317 318 319 320
/*
 * Reverse mapping data structures:
 *
 * If page->private bit zero is zero, then page->private points to the
 * shadow page table entry that points to page_address(page).
 *
 * If page->private bit zero is one, (then page->private & ~1) points
 * to a struct kvm_rmap_desc containing more mappings.
 */
321
static void rmap_add(struct kvm_vcpu *vcpu, u64 *spte)
322 323 324 325 326 327 328 329
{
	struct page *page;
	struct kvm_rmap_desc *desc;
	int i;

	if (!is_rmap_pte(*spte))
		return;
	page = pfn_to_page((*spte & PT64_BASE_ADDR_MASK) >> PAGE_SHIFT);
330
	if (!page_private(page)) {
331
		rmap_printk("rmap_add: %p %llx 0->1\n", spte, *spte);
332 333
		set_page_private(page,(unsigned long)spte);
	} else if (!(page_private(page) & 1)) {
334
		rmap_printk("rmap_add: %p %llx 1->many\n", spte, *spte);
335
		desc = mmu_alloc_rmap_desc(vcpu);
336
		desc->shadow_ptes[0] = (u64 *)page_private(page);
337
		desc->shadow_ptes[1] = spte;
338
		set_page_private(page,(unsigned long)desc | 1);
339 340
	} else {
		rmap_printk("rmap_add: %p %llx many->many\n", spte, *spte);
341
		desc = (struct kvm_rmap_desc *)(page_private(page) & ~1ul);
342 343 344
		while (desc->shadow_ptes[RMAP_EXT-1] && desc->more)
			desc = desc->more;
		if (desc->shadow_ptes[RMAP_EXT-1]) {
345
			desc->more = mmu_alloc_rmap_desc(vcpu);
346 347 348 349 350 351 352 353
			desc = desc->more;
		}
		for (i = 0; desc->shadow_ptes[i]; ++i)
			;
		desc->shadow_ptes[i] = spte;
	}
}

354
static void rmap_desc_remove_entry(struct page *page,
355 356 357 358 359 360 361 362 363
				   struct kvm_rmap_desc *desc,
				   int i,
				   struct kvm_rmap_desc *prev_desc)
{
	int j;

	for (j = RMAP_EXT - 1; !desc->shadow_ptes[j] && j > i; --j)
		;
	desc->shadow_ptes[i] = desc->shadow_ptes[j];
A
Al Viro 已提交
364
	desc->shadow_ptes[j] = NULL;
365 366 367
	if (j != 0)
		return;
	if (!prev_desc && !desc->more)
368
		set_page_private(page,(unsigned long)desc->shadow_ptes[0]);
369 370 371 372
	else
		if (prev_desc)
			prev_desc->more = desc->more;
		else
373
			set_page_private(page,(unsigned long)desc->more | 1);
374
	mmu_free_rmap_desc(desc);
375 376
}

377
static void rmap_remove(u64 *spte)
378 379 380 381 382 383 384 385 386
{
	struct page *page;
	struct kvm_rmap_desc *desc;
	struct kvm_rmap_desc *prev_desc;
	int i;

	if (!is_rmap_pte(*spte))
		return;
	page = pfn_to_page((*spte & PT64_BASE_ADDR_MASK) >> PAGE_SHIFT);
387
	if (!page_private(page)) {
388 389
		printk(KERN_ERR "rmap_remove: %p %llx 0->BUG\n", spte, *spte);
		BUG();
390
	} else if (!(page_private(page) & 1)) {
391
		rmap_printk("rmap_remove:  %p %llx 1->0\n", spte, *spte);
392
		if ((u64 *)page_private(page) != spte) {
393 394 395 396
			printk(KERN_ERR "rmap_remove:  %p %llx 1->BUG\n",
			       spte, *spte);
			BUG();
		}
397
		set_page_private(page,0);
398 399
	} else {
		rmap_printk("rmap_remove:  %p %llx many->many\n", spte, *spte);
400
		desc = (struct kvm_rmap_desc *)(page_private(page) & ~1ul);
401 402 403 404
		prev_desc = NULL;
		while (desc) {
			for (i = 0; i < RMAP_EXT && desc->shadow_ptes[i]; ++i)
				if (desc->shadow_ptes[i] == spte) {
405
					rmap_desc_remove_entry(page,
406
							       desc, i,
407 408 409 410 411 412 413 414 415 416
							       prev_desc);
					return;
				}
			prev_desc = desc;
			desc = desc->more;
		}
		BUG();
	}
}

417
static void rmap_write_protect(struct kvm_vcpu *vcpu, u64 gfn)
418
{
419
	struct kvm *kvm = vcpu->kvm;
420 421 422 423
	struct page *page;
	struct kvm_rmap_desc *desc;
	u64 *spte;

A
Avi Kivity 已提交
424 425
	page = gfn_to_page(kvm, gfn);
	BUG_ON(!page);
426

427 428 429
	while (page_private(page)) {
		if (!(page_private(page) & 1))
			spte = (u64 *)page_private(page);
430
		else {
431
			desc = (struct kvm_rmap_desc *)(page_private(page) & ~1ul);
432 433 434
			spte = desc->shadow_ptes[0];
		}
		BUG_ON(!spte);
435 436
		BUG_ON((*spte & PT64_BASE_ADDR_MASK) >> PAGE_SHIFT
		       != page_to_pfn(page));
437 438 439
		BUG_ON(!(*spte & PT_PRESENT_MASK));
		BUG_ON(!(*spte & PT_WRITABLE_MASK));
		rmap_printk("rmap_write_protect: spte %p %llx\n", spte, *spte);
440
		rmap_remove(spte);
441
		set_shadow_pte(spte, *spte & ~PT_WRITABLE_MASK);
S
Shaohua Li 已提交
442
		kvm_flush_remote_tlbs(vcpu->kvm);
443 444 445
	}
}

446
#ifdef MMU_DEBUG
447
static int is_empty_shadow_page(u64 *spt)
A
Avi Kivity 已提交
448
{
449 450 451
	u64 *pos;
	u64 *end;

452
	for (pos = spt, end = pos + PAGE_SIZE / sizeof(u64); pos != end; pos++)
453 454 455
		if (*pos != 0) {
			printk(KERN_ERR "%s: %p %llx\n", __FUNCTION__,
			       pos, *pos);
A
Avi Kivity 已提交
456
			return 0;
457
		}
A
Avi Kivity 已提交
458 459
	return 1;
}
460
#endif
A
Avi Kivity 已提交
461

462
static void kvm_mmu_free_page(struct kvm *kvm,
463
			      struct kvm_mmu_page *page_head)
464
{
465
	ASSERT(is_empty_shadow_page(page_head->spt));
466
	list_del(&page_head->link);
A
Avi Kivity 已提交
467
	__free_page(virt_to_page(page_head->spt));
468 469
	kfree(page_head);
	++kvm->n_free_mmu_pages;
470 471
}

472 473 474 475 476
static unsigned kvm_page_table_hashfn(gfn_t gfn)
{
	return gfn;
}

477 478
static struct kvm_mmu_page *kvm_mmu_alloc_page(struct kvm_vcpu *vcpu,
					       u64 *parent_pte)
A
Avi Kivity 已提交
479 480 481
{
	struct kvm_mmu_page *page;

482
	if (!vcpu->kvm->n_free_mmu_pages)
483
		return NULL;
A
Avi Kivity 已提交
484

485 486 487 488 489
	page = mmu_memory_cache_alloc(&vcpu->mmu_page_header_cache,
				      sizeof *page);
	page->spt = mmu_memory_cache_alloc(&vcpu->mmu_page_cache, PAGE_SIZE);
	set_page_private(virt_to_page(page->spt), (unsigned long)page);
	list_add(&page->link, &vcpu->kvm->active_mmu_pages);
490
	ASSERT(is_empty_shadow_page(page->spt));
A
Avi Kivity 已提交
491
	page->slot_bitmap = 0;
492
	page->multimapped = 0;
A
Avi Kivity 已提交
493
	page->parent_pte = parent_pte;
A
Avi Kivity 已提交
494
	--vcpu->kvm->n_free_mmu_pages;
495
	return page;
A
Avi Kivity 已提交
496 497
}

498 499
static void mmu_page_add_parent_pte(struct kvm_vcpu *vcpu,
				    struct kvm_mmu_page *page, u64 *parent_pte)
500 501 502 503 504 505 506 507 508 509 510 511 512 513 514
{
	struct kvm_pte_chain *pte_chain;
	struct hlist_node *node;
	int i;

	if (!parent_pte)
		return;
	if (!page->multimapped) {
		u64 *old = page->parent_pte;

		if (!old) {
			page->parent_pte = parent_pte;
			return;
		}
		page->multimapped = 1;
515
		pte_chain = mmu_alloc_pte_chain(vcpu);
516 517 518 519 520 521 522 523 524 525 526 527 528
		INIT_HLIST_HEAD(&page->parent_ptes);
		hlist_add_head(&pte_chain->link, &page->parent_ptes);
		pte_chain->parent_ptes[0] = old;
	}
	hlist_for_each_entry(pte_chain, node, &page->parent_ptes, link) {
		if (pte_chain->parent_ptes[NR_PTE_CHAIN_ENTRIES-1])
			continue;
		for (i = 0; i < NR_PTE_CHAIN_ENTRIES; ++i)
			if (!pte_chain->parent_ptes[i]) {
				pte_chain->parent_ptes[i] = parent_pte;
				return;
			}
	}
529
	pte_chain = mmu_alloc_pte_chain(vcpu);
530 531 532 533 534
	BUG_ON(!pte_chain);
	hlist_add_head(&pte_chain->link, &page->parent_ptes);
	pte_chain->parent_ptes[0] = parent_pte;
}

535
static void mmu_page_remove_parent_pte(struct kvm_mmu_page *page,
536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552
				       u64 *parent_pte)
{
	struct kvm_pte_chain *pte_chain;
	struct hlist_node *node;
	int i;

	if (!page->multimapped) {
		BUG_ON(page->parent_pte != parent_pte);
		page->parent_pte = NULL;
		return;
	}
	hlist_for_each_entry(pte_chain, node, &page->parent_ptes, link)
		for (i = 0; i < NR_PTE_CHAIN_ENTRIES; ++i) {
			if (!pte_chain->parent_ptes[i])
				break;
			if (pte_chain->parent_ptes[i] != parent_pte)
				continue;
553 554
			while (i + 1 < NR_PTE_CHAIN_ENTRIES
				&& pte_chain->parent_ptes[i + 1]) {
555 556 557 558 559
				pte_chain->parent_ptes[i]
					= pte_chain->parent_ptes[i + 1];
				++i;
			}
			pte_chain->parent_ptes[i] = NULL;
560 561
			if (i == 0) {
				hlist_del(&pte_chain->link);
562
				mmu_free_pte_chain(pte_chain);
563 564 565 566 567
				if (hlist_empty(&page->parent_ptes)) {
					page->multimapped = 0;
					page->parent_pte = NULL;
				}
			}
568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597
			return;
		}
	BUG();
}

static struct kvm_mmu_page *kvm_mmu_lookup_page(struct kvm_vcpu *vcpu,
						gfn_t gfn)
{
	unsigned index;
	struct hlist_head *bucket;
	struct kvm_mmu_page *page;
	struct hlist_node *node;

	pgprintk("%s: looking for gfn %lx\n", __FUNCTION__, gfn);
	index = kvm_page_table_hashfn(gfn) % KVM_NUM_MMU_PAGES;
	bucket = &vcpu->kvm->mmu_page_hash[index];
	hlist_for_each_entry(page, node, bucket, hash_link)
		if (page->gfn == gfn && !page->role.metaphysical) {
			pgprintk("%s: found role %x\n",
				 __FUNCTION__, page->role.word);
			return page;
		}
	return NULL;
}

static struct kvm_mmu_page *kvm_mmu_get_page(struct kvm_vcpu *vcpu,
					     gfn_t gfn,
					     gva_t gaddr,
					     unsigned level,
					     int metaphysical,
598
					     unsigned hugepage_access,
599 600 601 602 603 604 605 606 607 608 609 610 611
					     u64 *parent_pte)
{
	union kvm_mmu_page_role role;
	unsigned index;
	unsigned quadrant;
	struct hlist_head *bucket;
	struct kvm_mmu_page *page;
	struct hlist_node *node;

	role.word = 0;
	role.glevels = vcpu->mmu.root_level;
	role.level = level;
	role.metaphysical = metaphysical;
612
	role.hugepage_access = hugepage_access;
613 614 615 616 617 618 619 620 621 622 623
	if (vcpu->mmu.root_level <= PT32_ROOT_LEVEL) {
		quadrant = gaddr >> (PAGE_SHIFT + (PT64_PT_BITS * level));
		quadrant &= (1 << ((PT32_PT_BITS - PT64_PT_BITS) * level)) - 1;
		role.quadrant = quadrant;
	}
	pgprintk("%s: looking gfn %lx role %x\n", __FUNCTION__,
		 gfn, role.word);
	index = kvm_page_table_hashfn(gfn) % KVM_NUM_MMU_PAGES;
	bucket = &vcpu->kvm->mmu_page_hash[index];
	hlist_for_each_entry(page, node, bucket, hash_link)
		if (page->gfn == gfn && page->role.word == role.word) {
624
			mmu_page_add_parent_pte(vcpu, page, parent_pte);
625 626 627 628 629 630 631 632 633 634
			pgprintk("%s: found\n", __FUNCTION__);
			return page;
		}
	page = kvm_mmu_alloc_page(vcpu, parent_pte);
	if (!page)
		return page;
	pgprintk("%s: adding gfn %lx role %x\n", __FUNCTION__, gfn, role.word);
	page->gfn = gfn;
	page->role = role;
	hlist_add_head(&page->hash_link, bucket);
635
	if (!metaphysical)
636
		rmap_write_protect(vcpu, gfn);
637 638 639
	return page;
}

640
static void kvm_mmu_page_unlink_children(struct kvm *kvm,
641 642
					 struct kvm_mmu_page *page)
{
643 644 645 646
	unsigned i;
	u64 *pt;
	u64 ent;

647
	pt = page->spt;
648 649 650 651

	if (page->role.level == PT_PAGE_TABLE_LEVEL) {
		for (i = 0; i < PT64_ENT_PER_PAGE; ++i) {
			if (pt[i] & PT_PRESENT_MASK)
652
				rmap_remove(&pt[i]);
653 654
			pt[i] = 0;
		}
655
		kvm_flush_remote_tlbs(kvm);
656 657 658 659 660 661 662 663 664 665
		return;
	}

	for (i = 0; i < PT64_ENT_PER_PAGE; ++i) {
		ent = pt[i];

		pt[i] = 0;
		if (!(ent & PT_PRESENT_MASK))
			continue;
		ent &= PT64_BASE_ADDR_MASK;
666
		mmu_page_remove_parent_pte(page_header(ent), &pt[i]);
667
	}
668
	kvm_flush_remote_tlbs(kvm);
669 670
}

671
static void kvm_mmu_put_page(struct kvm_mmu_page *page,
672 673
			     u64 *parent_pte)
{
674
	mmu_page_remove_parent_pte(page, parent_pte);
675 676
}

677
static void kvm_mmu_zap_page(struct kvm *kvm,
678 679 680 681 682 683 684 685 686 687 688 689 690 691
			     struct kvm_mmu_page *page)
{
	u64 *parent_pte;

	while (page->multimapped || page->parent_pte) {
		if (!page->multimapped)
			parent_pte = page->parent_pte;
		else {
			struct kvm_pte_chain *chain;

			chain = container_of(page->parent_ptes.first,
					     struct kvm_pte_chain, link);
			parent_pte = chain->parent_ptes[0];
		}
692
		BUG_ON(!parent_pte);
693
		kvm_mmu_put_page(page, parent_pte);
694
		set_shadow_pte(parent_pte, 0);
695
	}
696
	kvm_mmu_page_unlink_children(kvm, page);
697 698
	if (!page->root_count) {
		hlist_del(&page->hash_link);
699
		kvm_mmu_free_page(kvm, page);
A
Avi Kivity 已提交
700
	} else
701
		list_move(&page->link, &kvm->active_mmu_pages);
702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717
}

static int kvm_mmu_unprotect_page(struct kvm_vcpu *vcpu, gfn_t gfn)
{
	unsigned index;
	struct hlist_head *bucket;
	struct kvm_mmu_page *page;
	struct hlist_node *node, *n;
	int r;

	pgprintk("%s: looking for gfn %lx\n", __FUNCTION__, gfn);
	r = 0;
	index = kvm_page_table_hashfn(gfn) % KVM_NUM_MMU_PAGES;
	bucket = &vcpu->kvm->mmu_page_hash[index];
	hlist_for_each_entry_safe(page, node, n, bucket, hash_link)
		if (page->gfn == gfn && !page->role.metaphysical) {
718 719
			pgprintk("%s: gfn %lx role %x\n", __FUNCTION__, gfn,
				 page->role.word);
720
			kvm_mmu_zap_page(vcpu->kvm, page);
721 722 723
			r = 1;
		}
	return r;
724 725
}

726 727 728 729 730 731 732
static void mmu_unshadow(struct kvm_vcpu *vcpu, gfn_t gfn)
{
	struct kvm_mmu_page *page;

	while ((page = kvm_mmu_lookup_page(vcpu, gfn)) != NULL) {
		pgprintk("%s: zap %lx %x\n",
			 __FUNCTION__, gfn, page->role.word);
733
		kvm_mmu_zap_page(vcpu->kvm, page);
734 735 736
	}
}

A
Avi Kivity 已提交
737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756
static void page_header_update_slot(struct kvm *kvm, void *pte, gpa_t gpa)
{
	int slot = memslot_id(kvm, gfn_to_memslot(kvm, gpa >> PAGE_SHIFT));
	struct kvm_mmu_page *page_head = page_header(__pa(pte));

	__set_bit(slot, &page_head->slot_bitmap);
}

hpa_t safe_gpa_to_hpa(struct kvm_vcpu *vcpu, gpa_t gpa)
{
	hpa_t hpa = gpa_to_hpa(vcpu, gpa);

	return is_error_hpa(hpa) ? bad_page_address | (gpa & ~PAGE_MASK): hpa;
}

hpa_t gpa_to_hpa(struct kvm_vcpu *vcpu, gpa_t gpa)
{
	struct page *page;

	ASSERT((gpa & HPA_ERR_MASK) == 0);
A
Avi Kivity 已提交
757 758
	page = gfn_to_page(vcpu->kvm, gpa >> PAGE_SHIFT);
	if (!page)
A
Avi Kivity 已提交
759 760 761 762 763 764 765 766 767 768 769 770 771 772
		return gpa | HPA_ERR_MASK;
	return ((hpa_t)page_to_pfn(page) << PAGE_SHIFT)
		| (gpa & (PAGE_SIZE-1));
}

hpa_t gva_to_hpa(struct kvm_vcpu *vcpu, gva_t gva)
{
	gpa_t gpa = vcpu->mmu.gva_to_gpa(vcpu, gva);

	if (gpa == UNMAPPED_GVA)
		return UNMAPPED_GVA;
	return gpa_to_hpa(vcpu, gpa);
}

773 774 775 776 777 778 779 780 781
struct page *gva_to_page(struct kvm_vcpu *vcpu, gva_t gva)
{
	gpa_t gpa = vcpu->mmu.gva_to_gpa(vcpu, gva);

	if (gpa == UNMAPPED_GVA)
		return NULL;
	return pfn_to_page(gpa_to_hpa(vcpu, gpa) >> PAGE_SHIFT);
}

A
Avi Kivity 已提交
782 783 784 785 786 787 788 789 790 791 792 793
static void nonpaging_new_cr3(struct kvm_vcpu *vcpu)
{
}

static int nonpaging_map(struct kvm_vcpu *vcpu, gva_t v, hpa_t p)
{
	int level = PT32E_ROOT_LEVEL;
	hpa_t table_addr = vcpu->mmu.root_hpa;

	for (; ; level--) {
		u32 index = PT64_INDEX(v, level);
		u64 *table;
794
		u64 pte;
A
Avi Kivity 已提交
795 796 797 798 799

		ASSERT(VALID_PAGE(table_addr));
		table = __va(table_addr);

		if (level == 1) {
800 801 802
			pte = table[index];
			if (is_present_pte(pte) && is_writeble_pte(pte))
				return 0;
A
Avi Kivity 已提交
803 804 805 806
			mark_page_dirty(vcpu->kvm, v >> PAGE_SHIFT);
			page_header_update_slot(vcpu->kvm, table, v);
			table[index] = p | PT_PRESENT_MASK | PT_WRITABLE_MASK |
								PT_USER_MASK;
807
			rmap_add(vcpu, &table[index]);
A
Avi Kivity 已提交
808 809 810 811
			return 0;
		}

		if (table[index] == 0) {
812
			struct kvm_mmu_page *new_table;
813
			gfn_t pseudo_gfn;
A
Avi Kivity 已提交
814

815 816 817 818
			pseudo_gfn = (v & PT64_DIR_BASE_ADDR_MASK)
				>> PAGE_SHIFT;
			new_table = kvm_mmu_get_page(vcpu, pseudo_gfn,
						     v, level - 1,
819
						     1, 0, &table[index]);
820
			if (!new_table) {
A
Avi Kivity 已提交
821 822 823 824
				pgprintk("nonpaging_map: ENOMEM\n");
				return -ENOMEM;
			}

825
			table[index] = __pa(new_table->spt) | PT_PRESENT_MASK
826
				| PT_WRITABLE_MASK | PT_USER_MASK;
A
Avi Kivity 已提交
827 828 829 830 831
		}
		table_addr = table[index] & PT64_BASE_ADDR_MASK;
	}
}

832 833 834
static void mmu_free_roots(struct kvm_vcpu *vcpu)
{
	int i;
835
	struct kvm_mmu_page *page;
836

A
Avi Kivity 已提交
837 838
	if (!VALID_PAGE(vcpu->mmu.root_hpa))
		return;
839 840 841 842
#ifdef CONFIG_X86_64
	if (vcpu->mmu.shadow_root_level == PT64_ROOT_LEVEL) {
		hpa_t root = vcpu->mmu.root_hpa;

843 844
		page = page_header(root);
		--page->root_count;
845 846 847 848 849 850 851
		vcpu->mmu.root_hpa = INVALID_PAGE;
		return;
	}
#endif
	for (i = 0; i < 4; ++i) {
		hpa_t root = vcpu->mmu.pae_root[i];

A
Avi Kivity 已提交
852 853 854 855 856
		if (root) {
			root &= PT64_BASE_ADDR_MASK;
			page = page_header(root);
			--page->root_count;
		}
857 858 859 860 861 862 863 864
		vcpu->mmu.pae_root[i] = INVALID_PAGE;
	}
	vcpu->mmu.root_hpa = INVALID_PAGE;
}

static void mmu_alloc_roots(struct kvm_vcpu *vcpu)
{
	int i;
865
	gfn_t root_gfn;
866 867
	struct kvm_mmu_page *page;

868
	root_gfn = vcpu->cr3 >> PAGE_SHIFT;
869 870 871 872 873 874

#ifdef CONFIG_X86_64
	if (vcpu->mmu.shadow_root_level == PT64_ROOT_LEVEL) {
		hpa_t root = vcpu->mmu.root_hpa;

		ASSERT(!VALID_PAGE(root));
875
		page = kvm_mmu_get_page(vcpu, root_gfn, 0,
876
					PT64_ROOT_LEVEL, 0, 0, NULL);
877
		root = __pa(page->spt);
878
		++page->root_count;
879 880 881 882 883 884 885 886
		vcpu->mmu.root_hpa = root;
		return;
	}
#endif
	for (i = 0; i < 4; ++i) {
		hpa_t root = vcpu->mmu.pae_root[i];

		ASSERT(!VALID_PAGE(root));
A
Avi Kivity 已提交
887 888 889 890 891
		if (vcpu->mmu.root_level == PT32E_ROOT_LEVEL) {
			if (!is_present_pte(vcpu->pdptrs[i])) {
				vcpu->mmu.pae_root[i] = 0;
				continue;
			}
892
			root_gfn = vcpu->pdptrs[i] >> PAGE_SHIFT;
A
Avi Kivity 已提交
893
		} else if (vcpu->mmu.root_level == 0)
894
			root_gfn = 0;
895
		page = kvm_mmu_get_page(vcpu, root_gfn, i << 30,
896
					PT32_ROOT_LEVEL, !is_paging(vcpu),
897
					0, NULL);
898
		root = __pa(page->spt);
899
		++page->root_count;
900 901 902 903 904
		vcpu->mmu.pae_root[i] = root | PT_PRESENT_MASK;
	}
	vcpu->mmu.root_hpa = __pa(vcpu->mmu.pae_root);
}

A
Avi Kivity 已提交
905 906 907 908 909 910 911 912 913
static gpa_t nonpaging_gva_to_gpa(struct kvm_vcpu *vcpu, gva_t vaddr)
{
	return vaddr;
}

static int nonpaging_page_fault(struct kvm_vcpu *vcpu, gva_t gva,
			       u32 error_code)
{
	gpa_t addr = gva;
A
Avi Kivity 已提交
914
	hpa_t paddr;
915
	int r;
A
Avi Kivity 已提交
916

917 918 919
	r = mmu_topup_memory_caches(vcpu);
	if (r)
		return r;
920

A
Avi Kivity 已提交
921 922 923 924
	ASSERT(vcpu);
	ASSERT(VALID_PAGE(vcpu->mmu.root_hpa));


A
Avi Kivity 已提交
925
	paddr = gpa_to_hpa(vcpu , addr & PT64_BASE_ADDR_MASK);
A
Avi Kivity 已提交
926

A
Avi Kivity 已提交
927 928
	if (is_error_hpa(paddr))
		return 1;
A
Avi Kivity 已提交
929

A
Avi Kivity 已提交
930
	return nonpaging_map(vcpu, addr & PAGE_MASK, paddr);
A
Avi Kivity 已提交
931 932 933 934
}

static void nonpaging_free(struct kvm_vcpu *vcpu)
{
935
	mmu_free_roots(vcpu);
A
Avi Kivity 已提交
936 937 938 939 940 941 942 943 944 945
}

static int nonpaging_init_context(struct kvm_vcpu *vcpu)
{
	struct kvm_mmu *context = &vcpu->mmu;

	context->new_cr3 = nonpaging_new_cr3;
	context->page_fault = nonpaging_page_fault;
	context->gva_to_gpa = nonpaging_gva_to_gpa;
	context->free = nonpaging_free;
946
	context->root_level = 0;
A
Avi Kivity 已提交
947
	context->shadow_root_level = PT32E_ROOT_LEVEL;
A
Avi Kivity 已提交
948
	context->root_hpa = INVALID_PAGE;
A
Avi Kivity 已提交
949 950 951 952 953
	return 0;
}

static void kvm_mmu_flush_tlb(struct kvm_vcpu *vcpu)
{
A
Avi Kivity 已提交
954
	++vcpu->stat.tlb_flush;
955
	kvm_x86_ops->tlb_flush(vcpu);
A
Avi Kivity 已提交
956 957 958 959
}

static void paging_new_cr3(struct kvm_vcpu *vcpu)
{
960
	pgprintk("%s: cr3 %lx\n", __FUNCTION__, vcpu->cr3);
961
	mmu_free_roots(vcpu);
A
Avi Kivity 已提交
962 963 964 965 966 967
}

static void inject_page_fault(struct kvm_vcpu *vcpu,
			      u64 addr,
			      u32 err_code)
{
968
	kvm_x86_ops->inject_page_fault(vcpu, addr, err_code);
A
Avi Kivity 已提交
969 970 971 972 973 974 975 976 977 978 979 980 981 982 983
}

static void paging_free(struct kvm_vcpu *vcpu)
{
	nonpaging_free(vcpu);
}

#define PTTYPE 64
#include "paging_tmpl.h"
#undef PTTYPE

#define PTTYPE 32
#include "paging_tmpl.h"
#undef PTTYPE

984
static int paging64_init_context_common(struct kvm_vcpu *vcpu, int level)
A
Avi Kivity 已提交
985 986 987 988 989 990 991 992
{
	struct kvm_mmu *context = &vcpu->mmu;

	ASSERT(is_pae(vcpu));
	context->new_cr3 = paging_new_cr3;
	context->page_fault = paging64_page_fault;
	context->gva_to_gpa = paging64_gva_to_gpa;
	context->free = paging_free;
993 994
	context->root_level = level;
	context->shadow_root_level = level;
A
Avi Kivity 已提交
995
	context->root_hpa = INVALID_PAGE;
A
Avi Kivity 已提交
996 997 998
	return 0;
}

999 1000 1001 1002 1003
static int paging64_init_context(struct kvm_vcpu *vcpu)
{
	return paging64_init_context_common(vcpu, PT64_ROOT_LEVEL);
}

A
Avi Kivity 已提交
1004 1005 1006 1007 1008 1009 1010 1011 1012 1013
static int paging32_init_context(struct kvm_vcpu *vcpu)
{
	struct kvm_mmu *context = &vcpu->mmu;

	context->new_cr3 = paging_new_cr3;
	context->page_fault = paging32_page_fault;
	context->gva_to_gpa = paging32_gva_to_gpa;
	context->free = paging_free;
	context->root_level = PT32_ROOT_LEVEL;
	context->shadow_root_level = PT32E_ROOT_LEVEL;
A
Avi Kivity 已提交
1014
	context->root_hpa = INVALID_PAGE;
A
Avi Kivity 已提交
1015 1016 1017 1018 1019
	return 0;
}

static int paging32E_init_context(struct kvm_vcpu *vcpu)
{
1020
	return paging64_init_context_common(vcpu, PT32E_ROOT_LEVEL);
A
Avi Kivity 已提交
1021 1022 1023 1024 1025 1026 1027 1028 1029
}

static int init_kvm_mmu(struct kvm_vcpu *vcpu)
{
	ASSERT(vcpu);
	ASSERT(!VALID_PAGE(vcpu->mmu.root_hpa));

	if (!is_paging(vcpu))
		return nonpaging_init_context(vcpu);
A
Avi Kivity 已提交
1030
	else if (is_long_mode(vcpu))
A
Avi Kivity 已提交
1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047
		return paging64_init_context(vcpu);
	else if (is_pae(vcpu))
		return paging32E_init_context(vcpu);
	else
		return paging32_init_context(vcpu);
}

static void destroy_kvm_mmu(struct kvm_vcpu *vcpu)
{
	ASSERT(vcpu);
	if (VALID_PAGE(vcpu->mmu.root_hpa)) {
		vcpu->mmu.free(vcpu);
		vcpu->mmu.root_hpa = INVALID_PAGE;
	}
}

int kvm_mmu_reset_context(struct kvm_vcpu *vcpu)
A
Avi Kivity 已提交
1048 1049 1050 1051 1052 1053
{
	destroy_kvm_mmu(vcpu);
	return init_kvm_mmu(vcpu);
}

int kvm_mmu_load(struct kvm_vcpu *vcpu)
A
Avi Kivity 已提交
1054
{
1055 1056
	int r;

S
Shaohua Li 已提交
1057
	mutex_lock(&vcpu->kvm->lock);
1058
	r = mmu_topup_memory_caches(vcpu);
A
Avi Kivity 已提交
1059 1060 1061
	if (r)
		goto out;
	mmu_alloc_roots(vcpu);
1062
	kvm_x86_ops->set_cr3(vcpu, vcpu->mmu.root_hpa);
A
Avi Kivity 已提交
1063
	kvm_mmu_flush_tlb(vcpu);
1064
out:
S
Shaohua Li 已提交
1065
	mutex_unlock(&vcpu->kvm->lock);
1066
	return r;
A
Avi Kivity 已提交
1067
}
A
Avi Kivity 已提交
1068 1069 1070 1071 1072 1073
EXPORT_SYMBOL_GPL(kvm_mmu_load);

void kvm_mmu_unload(struct kvm_vcpu *vcpu)
{
	mmu_free_roots(vcpu);
}
A
Avi Kivity 已提交
1074

1075
static void mmu_pte_write_zap_pte(struct kvm_vcpu *vcpu,
1076 1077 1078 1079 1080 1081 1082 1083 1084
				  struct kvm_mmu_page *page,
				  u64 *spte)
{
	u64 pte;
	struct kvm_mmu_page *child;

	pte = *spte;
	if (is_present_pte(pte)) {
		if (page->role.level == PT_PAGE_TABLE_LEVEL)
1085
			rmap_remove(spte);
1086 1087
		else {
			child = page_header(pte & PT64_BASE_ADDR_MASK);
1088
			mmu_page_remove_parent_pte(child, spte);
1089 1090
		}
	}
1091
	set_shadow_pte(spte, 0);
1092
	kvm_flush_remote_tlbs(vcpu->kvm);
1093 1094
}

1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108
static void mmu_pte_write_new_pte(struct kvm_vcpu *vcpu,
				  struct kvm_mmu_page *page,
				  u64 *spte,
				  const void *new, int bytes)
{
	if (page->role.level != PT_PAGE_TABLE_LEVEL)
		return;

	if (page->role.glevels == PT32_ROOT_LEVEL)
		paging32_update_pte(vcpu, page, spte, new, bytes);
	else
		paging64_update_pte(vcpu, page, spte, new, bytes);
}

1109
void kvm_mmu_pte_write(struct kvm_vcpu *vcpu, gpa_t gpa,
1110
		       const u8 *new, int bytes)
1111
{
1112 1113
	gfn_t gfn = gpa >> PAGE_SHIFT;
	struct kvm_mmu_page *page;
1114
	struct hlist_node *node, *n;
1115 1116 1117 1118
	struct hlist_head *bucket;
	unsigned index;
	u64 *spte;
	unsigned offset = offset_in_page(gpa);
1119
	unsigned pte_size;
1120
	unsigned page_offset;
1121
	unsigned misaligned;
1122
	unsigned quadrant;
1123
	int level;
1124
	int flooded = 0;
1125
	int npte;
1126

1127
	pgprintk("%s: gpa %llx bytes %d\n", __FUNCTION__, gpa, bytes);
1128 1129 1130 1131 1132 1133 1134 1135
	if (gfn == vcpu->last_pt_write_gfn) {
		++vcpu->last_pt_write_count;
		if (vcpu->last_pt_write_count >= 3)
			flooded = 1;
	} else {
		vcpu->last_pt_write_gfn = gfn;
		vcpu->last_pt_write_count = 1;
	}
1136 1137
	index = kvm_page_table_hashfn(gfn) % KVM_NUM_MMU_PAGES;
	bucket = &vcpu->kvm->mmu_page_hash[index];
1138
	hlist_for_each_entry_safe(page, node, n, bucket, hash_link) {
1139 1140
		if (page->gfn != gfn || page->role.metaphysical)
			continue;
1141 1142
		pte_size = page->role.glevels == PT32_ROOT_LEVEL ? 4 : 8;
		misaligned = (offset ^ (offset + bytes - 1)) & ~(pte_size - 1);
1143
		misaligned |= bytes < 4;
1144
		if (misaligned || flooded) {
1145 1146 1147 1148
			/*
			 * Misaligned accesses are too much trouble to fix
			 * up; also, they usually indicate a page is not used
			 * as a page table.
1149 1150 1151 1152 1153
			 *
			 * If we're seeing too many writes to a page,
			 * it may no longer be a page table, or we may be
			 * forking, in which case it is better to unmap the
			 * page.
1154 1155 1156
			 */
			pgprintk("misaligned: gpa %llx bytes %d role %x\n",
				 gpa, bytes, page->role.word);
1157
			kvm_mmu_zap_page(vcpu->kvm, page);
1158 1159
			continue;
		}
1160 1161
		page_offset = offset;
		level = page->role.level;
1162
		npte = 1;
1163
		if (page->role.glevels == PT32_ROOT_LEVEL) {
1164 1165 1166 1167 1168 1169 1170
			page_offset <<= 1;	/* 32->64 */
			/*
			 * A 32-bit pde maps 4MB while the shadow pdes map
			 * only 2MB.  So we need to double the offset again
			 * and zap two pdes instead of one.
			 */
			if (level == PT32_ROOT_LEVEL) {
1171
				page_offset &= ~7; /* kill rounding error */
1172 1173 1174
				page_offset <<= 1;
				npte = 2;
			}
1175
			quadrant = page_offset >> PAGE_SHIFT;
1176
			page_offset &= ~PAGE_MASK;
1177 1178
			if (quadrant != page->role.quadrant)
				continue;
1179
		}
1180
		spte = &page->spt[page_offset / sizeof(*spte)];
1181
		while (npte--) {
1182
			mmu_pte_write_zap_pte(vcpu, page, spte);
1183
			mmu_pte_write_new_pte(vcpu, page, spte, new, bytes);
1184
			++spte;
1185 1186
		}
	}
1187 1188
}

1189 1190 1191 1192 1193 1194 1195
int kvm_mmu_unprotect_page_virt(struct kvm_vcpu *vcpu, gva_t gva)
{
	gpa_t gpa = vcpu->mmu.gva_to_gpa(vcpu, gva);

	return kvm_mmu_unprotect_page(vcpu, gpa >> PAGE_SHIFT);
}

1196
void __kvm_mmu_free_some_pages(struct kvm_vcpu *vcpu)
A
Avi Kivity 已提交
1197 1198 1199 1200 1201 1202
{
	while (vcpu->kvm->n_free_mmu_pages < KVM_REFILL_PAGES) {
		struct kvm_mmu_page *page;

		page = container_of(vcpu->kvm->active_mmu_pages.prev,
				    struct kvm_mmu_page, link);
1203
		kvm_mmu_zap_page(vcpu->kvm, page);
A
Avi Kivity 已提交
1204 1205 1206
	}
}

A
Avi Kivity 已提交
1207 1208
static void free_mmu_pages(struct kvm_vcpu *vcpu)
{
1209
	struct kvm_mmu_page *page;
A
Avi Kivity 已提交
1210

1211 1212 1213
	while (!list_empty(&vcpu->kvm->active_mmu_pages)) {
		page = container_of(vcpu->kvm->active_mmu_pages.next,
				    struct kvm_mmu_page, link);
1214
		kvm_mmu_zap_page(vcpu->kvm, page);
1215
	}
1216
	free_page((unsigned long)vcpu->mmu.pae_root);
A
Avi Kivity 已提交
1217 1218 1219 1220
}

static int alloc_mmu_pages(struct kvm_vcpu *vcpu)
{
1221
	struct page *page;
A
Avi Kivity 已提交
1222 1223 1224 1225
	int i;

	ASSERT(vcpu);

1226
	vcpu->kvm->n_free_mmu_pages = KVM_NUM_MMU_PAGES;
1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239

	/*
	 * When emulating 32-bit mode, cr3 is only 32 bits even on x86_64.
	 * Therefore we need to allocate shadow page tables in the first
	 * 4GB of memory, which happens to fit the DMA32 zone.
	 */
	page = alloc_page(GFP_KERNEL | __GFP_DMA32);
	if (!page)
		goto error_1;
	vcpu->mmu.pae_root = page_address(page);
	for (i = 0; i < 4; ++i)
		vcpu->mmu.pae_root[i] = INVALID_PAGE;

A
Avi Kivity 已提交
1240 1241 1242 1243 1244 1245 1246
	return 0;

error_1:
	free_mmu_pages(vcpu);
	return -ENOMEM;
}

1247
int kvm_mmu_create(struct kvm_vcpu *vcpu)
A
Avi Kivity 已提交
1248 1249 1250 1251
{
	ASSERT(vcpu);
	ASSERT(!VALID_PAGE(vcpu->mmu.root_hpa));

1252 1253
	return alloc_mmu_pages(vcpu);
}
A
Avi Kivity 已提交
1254

1255 1256 1257 1258
int kvm_mmu_setup(struct kvm_vcpu *vcpu)
{
	ASSERT(vcpu);
	ASSERT(!VALID_PAGE(vcpu->mmu.root_hpa));
1259

1260
	return init_kvm_mmu(vcpu);
A
Avi Kivity 已提交
1261 1262 1263 1264 1265 1266 1267 1268
}

void kvm_mmu_destroy(struct kvm_vcpu *vcpu)
{
	ASSERT(vcpu);

	destroy_kvm_mmu(vcpu);
	free_mmu_pages(vcpu);
1269
	mmu_free_memory_caches(vcpu);
A
Avi Kivity 已提交
1270 1271
}

1272
void kvm_mmu_slot_remove_write_access(struct kvm *kvm, int slot)
A
Avi Kivity 已提交
1273 1274 1275 1276 1277 1278 1279 1280 1281 1282
{
	struct kvm_mmu_page *page;

	list_for_each_entry(page, &kvm->active_mmu_pages, link) {
		int i;
		u64 *pt;

		if (!test_bit(slot, &page->slot_bitmap))
			continue;

1283
		pt = page->spt;
A
Avi Kivity 已提交
1284 1285
		for (i = 0; i < PT64_ENT_PER_PAGE; ++i)
			/* avoid RMW */
1286
			if (pt[i] & PT_WRITABLE_MASK) {
1287
				rmap_remove(&pt[i]);
A
Avi Kivity 已提交
1288
				pt[i] &= ~PT_WRITABLE_MASK;
1289
			}
A
Avi Kivity 已提交
1290 1291
	}
}
1292

1293
void kvm_mmu_zap_all(struct kvm *kvm)
D
Dor Laor 已提交
1294
{
1295
	struct kvm_mmu_page *page, *node;
D
Dor Laor 已提交
1296

1297 1298
	list_for_each_entry_safe(page, node, &kvm->active_mmu_pages, link)
		kvm_mmu_zap_page(kvm, page);
D
Dor Laor 已提交
1299

1300
	kvm_flush_remote_tlbs(kvm);
D
Dor Laor 已提交
1301 1302
}

1303 1304 1305 1306 1307 1308
void kvm_mmu_module_exit(void)
{
	if (pte_chain_cache)
		kmem_cache_destroy(pte_chain_cache);
	if (rmap_desc_cache)
		kmem_cache_destroy(rmap_desc_cache);
1309 1310
	if (mmu_page_header_cache)
		kmem_cache_destroy(mmu_page_header_cache);
1311 1312 1313 1314 1315 1316
}

int kvm_mmu_module_init(void)
{
	pte_chain_cache = kmem_cache_create("kvm_pte_chain",
					    sizeof(struct kvm_pte_chain),
1317
					    0, 0, NULL);
1318 1319 1320 1321
	if (!pte_chain_cache)
		goto nomem;
	rmap_desc_cache = kmem_cache_create("kvm_rmap_desc",
					    sizeof(struct kvm_rmap_desc),
1322
					    0, 0, NULL);
1323 1324 1325
	if (!rmap_desc_cache)
		goto nomem;

1326 1327
	mmu_page_header_cache = kmem_cache_create("kvm_mmu_page_header",
						  sizeof(struct kvm_mmu_page),
1328
						  0, 0, NULL);
1329 1330 1331
	if (!mmu_page_header_cache)
		goto nomem;

1332 1333 1334 1335 1336 1337 1338
	return 0;

nomem:
	kvm_mmu_module_exit();
	return -ENOMEM;
}

1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360
#ifdef AUDIT

static const char *audit_msg;

static gva_t canonicalize(gva_t gva)
{
#ifdef CONFIG_X86_64
	gva = (long long)(gva << 16) >> 16;
#endif
	return gva;
}

static void audit_mappings_page(struct kvm_vcpu *vcpu, u64 page_pte,
				gva_t va, int level)
{
	u64 *pt = __va(page_pte & PT64_BASE_ADDR_MASK);
	int i;
	gva_t va_delta = 1ul << (PAGE_SHIFT + 9 * (level - 1));

	for (i = 0; i < PT64_ENT_PER_PAGE; ++i, va += va_delta) {
		u64 ent = pt[i];

A
Adrian Bunk 已提交
1361
		if (!(ent & PT_PRESENT_MASK))
1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382
			continue;

		va = canonicalize(va);
		if (level > 1)
			audit_mappings_page(vcpu, ent, va, level - 1);
		else {
			gpa_t gpa = vcpu->mmu.gva_to_gpa(vcpu, va);
			hpa_t hpa = gpa_to_hpa(vcpu, gpa);

			if ((ent & PT_PRESENT_MASK)
			    && (ent & PT64_BASE_ADDR_MASK) != hpa)
				printk(KERN_ERR "audit error: (%s) levels %d"
				       " gva %lx gpa %llx hpa %llx ent %llx\n",
				       audit_msg, vcpu->mmu.root_level,
				       va, gpa, hpa, ent);
		}
	}
}

static void audit_mappings(struct kvm_vcpu *vcpu)
{
1383
	unsigned i;
1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434

	if (vcpu->mmu.root_level == 4)
		audit_mappings_page(vcpu, vcpu->mmu.root_hpa, 0, 4);
	else
		for (i = 0; i < 4; ++i)
			if (vcpu->mmu.pae_root[i] & PT_PRESENT_MASK)
				audit_mappings_page(vcpu,
						    vcpu->mmu.pae_root[i],
						    i << 30,
						    2);
}

static int count_rmaps(struct kvm_vcpu *vcpu)
{
	int nmaps = 0;
	int i, j, k;

	for (i = 0; i < KVM_MEMORY_SLOTS; ++i) {
		struct kvm_memory_slot *m = &vcpu->kvm->memslots[i];
		struct kvm_rmap_desc *d;

		for (j = 0; j < m->npages; ++j) {
			struct page *page = m->phys_mem[j];

			if (!page->private)
				continue;
			if (!(page->private & 1)) {
				++nmaps;
				continue;
			}
			d = (struct kvm_rmap_desc *)(page->private & ~1ul);
			while (d) {
				for (k = 0; k < RMAP_EXT; ++k)
					if (d->shadow_ptes[k])
						++nmaps;
					else
						break;
				d = d->more;
			}
		}
	}
	return nmaps;
}

static int count_writable_mappings(struct kvm_vcpu *vcpu)
{
	int nmaps = 0;
	struct kvm_mmu_page *page;
	int i;

	list_for_each_entry(page, &vcpu->kvm->active_mmu_pages, link) {
1435
		u64 *pt = page->spt;
1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497

		if (page->role.level != PT_PAGE_TABLE_LEVEL)
			continue;

		for (i = 0; i < PT64_ENT_PER_PAGE; ++i) {
			u64 ent = pt[i];

			if (!(ent & PT_PRESENT_MASK))
				continue;
			if (!(ent & PT_WRITABLE_MASK))
				continue;
			++nmaps;
		}
	}
	return nmaps;
}

static void audit_rmap(struct kvm_vcpu *vcpu)
{
	int n_rmap = count_rmaps(vcpu);
	int n_actual = count_writable_mappings(vcpu);

	if (n_rmap != n_actual)
		printk(KERN_ERR "%s: (%s) rmap %d actual %d\n",
		       __FUNCTION__, audit_msg, n_rmap, n_actual);
}

static void audit_write_protection(struct kvm_vcpu *vcpu)
{
	struct kvm_mmu_page *page;

	list_for_each_entry(page, &vcpu->kvm->active_mmu_pages, link) {
		hfn_t hfn;
		struct page *pg;

		if (page->role.metaphysical)
			continue;

		hfn = gpa_to_hpa(vcpu, (gpa_t)page->gfn << PAGE_SHIFT)
			>> PAGE_SHIFT;
		pg = pfn_to_page(hfn);
		if (pg->private)
			printk(KERN_ERR "%s: (%s) shadow page has writable"
			       " mappings: gfn %lx role %x\n",
			       __FUNCTION__, audit_msg, page->gfn,
			       page->role.word);
	}
}

static void kvm_mmu_audit(struct kvm_vcpu *vcpu, const char *msg)
{
	int olddbg = dbg;

	dbg = 0;
	audit_msg = msg;
	audit_rmap(vcpu);
	audit_write_protection(vcpu);
	audit_mappings(vcpu);
	dbg = olddbg;
}

#endif