kvm_main.c 51.7 KB
Newer Older
A
Avi Kivity 已提交
1 2 3 4 5 6 7
/*
 * 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.
 *
 * Copyright (C) 2006 Qumranet, Inc.
N
Nicolas Kaiser 已提交
8
 * Copyright 2010 Red Hat, Inc. and/or its affiliates.
A
Avi Kivity 已提交
9 10 11 12 13 14 15 16 17 18
 *
 * Authors:
 *   Avi Kivity   <avi@qumranet.com>
 *   Yaniv Kamay  <yaniv@qumranet.com>
 *
 * This work is licensed under the terms of the GNU GPL, version 2.  See
 * the COPYING file in the top-level directory.
 *
 */

19
#include "iodev.h"
A
Avi Kivity 已提交
20

21
#include <linux/kvm_host.h>
A
Avi Kivity 已提交
22 23 24 25 26 27 28 29 30 31 32
#include <linux/kvm.h>
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/percpu.h>
#include <linux/mm.h>
#include <linux/miscdevice.h>
#include <linux/vmalloc.h>
#include <linux/reboot.h>
#include <linux/debugfs.h>
#include <linux/highmem.h>
#include <linux/file.h>
33
#include <linux/sysdev.h>
A
Avi Kivity 已提交
34
#include <linux/cpu.h>
A
Alexey Dobriyan 已提交
35
#include <linux/sched.h>
36 37
#include <linux/cpumask.h>
#include <linux/smp.h>
38
#include <linux/anon_inodes.h>
39
#include <linux/profile.h>
40
#include <linux/kvm_para.h>
41
#include <linux/pagemap.h>
42
#include <linux/mman.h>
43
#include <linux/swap.h>
44
#include <linux/bitops.h>
45
#include <linux/spinlock.h>
46
#include <linux/compat.h>
47
#include <linux/srcu.h>
J
Joerg Roedel 已提交
48
#include <linux/hugetlb.h>
49
#include <linux/slab.h>
A
Avi Kivity 已提交
50

A
Avi Kivity 已提交
51 52 53
#include <asm/processor.h>
#include <asm/io.h>
#include <asm/uaccess.h>
54
#include <asm/pgtable.h>
55
#include <asm-generic/bitops/le.h>
A
Avi Kivity 已提交
56

57 58
#include "coalesced_mmio.h"

59 60 61
#define CREATE_TRACE_POINTS
#include <trace/events/kvm.h>

A
Avi Kivity 已提交
62 63 64
MODULE_AUTHOR("Qumranet");
MODULE_LICENSE("GPL");

65 66 67
/*
 * Ordering of locks:
 *
68
 * 		kvm->lock --> kvm->slots_lock --> kvm->irq_lock
69 70
 */

71 72
DEFINE_SPINLOCK(kvm_lock);
LIST_HEAD(vm_list);
73

74
static cpumask_var_t cpus_hardware_enabled;
75 76
static int kvm_usage_count = 0;
static atomic_t hardware_enable_failed;
77

78 79
struct kmem_cache *kvm_vcpu_cache;
EXPORT_SYMBOL_GPL(kvm_vcpu_cache);
A
Avi Kivity 已提交
80

81 82
static __read_mostly struct preempt_ops kvm_preempt_ops;

83
struct dentry *kvm_debugfs_dir;
A
Avi Kivity 已提交
84

A
Avi Kivity 已提交
85 86
static long kvm_vcpu_ioctl(struct file *file, unsigned int ioctl,
			   unsigned long arg);
87 88
static int hardware_enable_all(void);
static void hardware_disable_all(void);
A
Avi Kivity 已提交
89

M
Marcelo Tosatti 已提交
90 91
static void kvm_io_bus_destroy(struct kvm_io_bus *bus);

H
Hannes Eder 已提交
92
static bool kvm_rebooting;
93

94 95
static bool largepages_enabled = true;

96 97
static struct page *hwpoison_page;
static pfn_t hwpoison_pfn;
98

99 100 101
static struct page *fault_page;
static pfn_t fault_pfn;

102
inline int kvm_is_mmio_pfn(pfn_t pfn)
B
Ben-Ami Yassour 已提交
103
{
104 105 106 107
	if (pfn_valid(pfn)) {
		struct page *page = compound_head(pfn_to_page(pfn));
		return PageReserved(page);
	}
B
Ben-Ami Yassour 已提交
108 109 110 111

	return true;
}

A
Avi Kivity 已提交
112 113 114
/*
 * Switches to specified vcpu, until a matching vcpu_put()
 */
115
void vcpu_load(struct kvm_vcpu *vcpu)
A
Avi Kivity 已提交
116
{
117 118
	int cpu;

A
Avi Kivity 已提交
119
	mutex_lock(&vcpu->mutex);
120 121
	cpu = get_cpu();
	preempt_notifier_register(&vcpu->preempt_notifier);
122
	kvm_arch_vcpu_load(vcpu, cpu);
123
	put_cpu();
A
Avi Kivity 已提交
124 125
}

126
void vcpu_put(struct kvm_vcpu *vcpu)
A
Avi Kivity 已提交
127
{
128
	preempt_disable();
129
	kvm_arch_vcpu_put(vcpu);
130 131
	preempt_notifier_unregister(&vcpu->preempt_notifier);
	preempt_enable();
A
Avi Kivity 已提交
132 133 134
	mutex_unlock(&vcpu->mutex);
}

135 136 137 138
static void ack_flush(void *_completed)
{
}

139
static bool make_all_cpus_request(struct kvm *kvm, unsigned int req)
140
{
141
	int i, cpu, me;
142 143
	cpumask_var_t cpus;
	bool called = true;
144 145
	struct kvm_vcpu *vcpu;

146
	zalloc_cpumask_var(&cpus, GFP_ATOMIC);
147

148
	raw_spin_lock(&kvm->requests_lock);
149
	me = smp_processor_id();
150
	kvm_for_each_vcpu(i, vcpu, kvm) {
151
		if (kvm_make_check_request(req, vcpu))
152 153
			continue;
		cpu = vcpu->cpu;
154 155
		if (cpus != NULL && cpu != -1 && cpu != me)
			cpumask_set_cpu(cpu, cpus);
156
	}
157 158 159 160 161 162
	if (unlikely(cpus == NULL))
		smp_call_function_many(cpu_online_mask, ack_flush, NULL, 1);
	else if (!cpumask_empty(cpus))
		smp_call_function_many(cpus, ack_flush, NULL, 1);
	else
		called = false;
163
	raw_spin_unlock(&kvm->requests_lock);
164
	free_cpumask_var(cpus);
165
	return called;
166 167
}

168
void kvm_flush_remote_tlbs(struct kvm *kvm)
169
{
170 171
	if (make_all_cpus_request(kvm, KVM_REQ_TLB_FLUSH))
		++kvm->stat.remote_tlb_flush;
172 173
}

174 175 176 177
void kvm_reload_remote_mmus(struct kvm *kvm)
{
	make_all_cpus_request(kvm, KVM_REQ_MMU_RELOAD);
}
178

R
Rusty Russell 已提交
179 180 181 182 183 184 185 186 187
int kvm_vcpu_init(struct kvm_vcpu *vcpu, struct kvm *kvm, unsigned id)
{
	struct page *page;
	int r;

	mutex_init(&vcpu->mutex);
	vcpu->cpu = -1;
	vcpu->kvm = kvm;
	vcpu->vcpu_id = id;
E
Eddie Dong 已提交
188
	init_waitqueue_head(&vcpu->wq);
R
Rusty Russell 已提交
189 190 191 192 193 194 195 196

	page = alloc_page(GFP_KERNEL | __GFP_ZERO);
	if (!page) {
		r = -ENOMEM;
		goto fail;
	}
	vcpu->run = page_address(page);

197
	r = kvm_arch_vcpu_init(vcpu);
R
Rusty Russell 已提交
198
	if (r < 0)
199
		goto fail_free_run;
R
Rusty Russell 已提交
200 201 202 203 204
	return 0;

fail_free_run:
	free_page((unsigned long)vcpu->run);
fail:
205
	return r;
R
Rusty Russell 已提交
206 207 208 209 210
}
EXPORT_SYMBOL_GPL(kvm_vcpu_init);

void kvm_vcpu_uninit(struct kvm_vcpu *vcpu)
{
211
	kvm_arch_vcpu_uninit(vcpu);
R
Rusty Russell 已提交
212 213 214 215
	free_page((unsigned long)vcpu->run);
}
EXPORT_SYMBOL_GPL(kvm_vcpu_uninit);

216 217 218 219 220 221 222 223 224 225 226
#if defined(CONFIG_MMU_NOTIFIER) && defined(KVM_ARCH_WANT_MMU_NOTIFIER)
static inline struct kvm *mmu_notifier_to_kvm(struct mmu_notifier *mn)
{
	return container_of(mn, struct kvm, mmu_notifier);
}

static void kvm_mmu_notifier_invalidate_page(struct mmu_notifier *mn,
					     struct mm_struct *mm,
					     unsigned long address)
{
	struct kvm *kvm = mmu_notifier_to_kvm(mn);
227
	int need_tlb_flush, idx;
228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246

	/*
	 * When ->invalidate_page runs, the linux pte has been zapped
	 * already but the page is still allocated until
	 * ->invalidate_page returns. So if we increase the sequence
	 * here the kvm page fault will notice if the spte can't be
	 * established because the page is going to be freed. If
	 * instead the kvm page fault establishes the spte before
	 * ->invalidate_page runs, kvm_unmap_hva will release it
	 * before returning.
	 *
	 * The sequence increase only need to be seen at spin_unlock
	 * time, and not at spin_lock time.
	 *
	 * Increasing the sequence after the spin_unlock would be
	 * unsafe because the kvm page fault could then establish the
	 * pte after kvm_unmap_hva returned, without noticing the page
	 * is going to be freed.
	 */
247
	idx = srcu_read_lock(&kvm->srcu);
248 249 250 251
	spin_lock(&kvm->mmu_lock);
	kvm->mmu_notifier_seq++;
	need_tlb_flush = kvm_unmap_hva(kvm, address);
	spin_unlock(&kvm->mmu_lock);
252
	srcu_read_unlock(&kvm->srcu, idx);
253 254 255 256 257 258 259

	/* we've to flush the tlb before the pages can be freed */
	if (need_tlb_flush)
		kvm_flush_remote_tlbs(kvm);

}

260 261 262 263 264 265
static void kvm_mmu_notifier_change_pte(struct mmu_notifier *mn,
					struct mm_struct *mm,
					unsigned long address,
					pte_t pte)
{
	struct kvm *kvm = mmu_notifier_to_kvm(mn);
266
	int idx;
267

268
	idx = srcu_read_lock(&kvm->srcu);
269 270 271 272
	spin_lock(&kvm->mmu_lock);
	kvm->mmu_notifier_seq++;
	kvm_set_spte_hva(kvm, address, pte);
	spin_unlock(&kvm->mmu_lock);
273
	srcu_read_unlock(&kvm->srcu, idx);
274 275
}

276 277 278 279 280 281
static void kvm_mmu_notifier_invalidate_range_start(struct mmu_notifier *mn,
						    struct mm_struct *mm,
						    unsigned long start,
						    unsigned long end)
{
	struct kvm *kvm = mmu_notifier_to_kvm(mn);
282
	int need_tlb_flush = 0, idx;
283

284
	idx = srcu_read_lock(&kvm->srcu);
285 286 287 288 289 290 291 292 293 294
	spin_lock(&kvm->mmu_lock);
	/*
	 * The count increase must become visible at unlock time as no
	 * spte can be established without taking the mmu_lock and
	 * count is also read inside the mmu_lock critical section.
	 */
	kvm->mmu_notifier_count++;
	for (; start < end; start += PAGE_SIZE)
		need_tlb_flush |= kvm_unmap_hva(kvm, start);
	spin_unlock(&kvm->mmu_lock);
295
	srcu_read_unlock(&kvm->srcu, idx);
296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332

	/* we've to flush the tlb before the pages can be freed */
	if (need_tlb_flush)
		kvm_flush_remote_tlbs(kvm);
}

static void kvm_mmu_notifier_invalidate_range_end(struct mmu_notifier *mn,
						  struct mm_struct *mm,
						  unsigned long start,
						  unsigned long end)
{
	struct kvm *kvm = mmu_notifier_to_kvm(mn);

	spin_lock(&kvm->mmu_lock);
	/*
	 * This sequence increase will notify the kvm page fault that
	 * the page that is going to be mapped in the spte could have
	 * been freed.
	 */
	kvm->mmu_notifier_seq++;
	/*
	 * The above sequence increase must be visible before the
	 * below count decrease but both values are read by the kvm
	 * page fault under mmu_lock spinlock so we don't need to add
	 * a smb_wmb() here in between the two.
	 */
	kvm->mmu_notifier_count--;
	spin_unlock(&kvm->mmu_lock);

	BUG_ON(kvm->mmu_notifier_count < 0);
}

static int kvm_mmu_notifier_clear_flush_young(struct mmu_notifier *mn,
					      struct mm_struct *mm,
					      unsigned long address)
{
	struct kvm *kvm = mmu_notifier_to_kvm(mn);
333
	int young, idx;
334

335
	idx = srcu_read_lock(&kvm->srcu);
336 337 338
	spin_lock(&kvm->mmu_lock);
	young = kvm_age_hva(kvm, address);
	spin_unlock(&kvm->mmu_lock);
339
	srcu_read_unlock(&kvm->srcu, idx);
340 341 342 343 344 345 346

	if (young)
		kvm_flush_remote_tlbs(kvm);

	return young;
}

347 348 349 350
static void kvm_mmu_notifier_release(struct mmu_notifier *mn,
				     struct mm_struct *mm)
{
	struct kvm *kvm = mmu_notifier_to_kvm(mn);
351 352 353
	int idx;

	idx = srcu_read_lock(&kvm->srcu);
354
	kvm_arch_flush_shadow(kvm);
355
	srcu_read_unlock(&kvm->srcu, idx);
356 357
}

358 359 360 361 362
static const struct mmu_notifier_ops kvm_mmu_notifier_ops = {
	.invalidate_page	= kvm_mmu_notifier_invalidate_page,
	.invalidate_range_start	= kvm_mmu_notifier_invalidate_range_start,
	.invalidate_range_end	= kvm_mmu_notifier_invalidate_range_end,
	.clear_flush_young	= kvm_mmu_notifier_clear_flush_young,
363
	.change_pte		= kvm_mmu_notifier_change_pte,
364
	.release		= kvm_mmu_notifier_release,
365
};
366 367 368 369 370 371 372 373 374 375 376 377 378 379

static int kvm_init_mmu_notifier(struct kvm *kvm)
{
	kvm->mmu_notifier.ops = &kvm_mmu_notifier_ops;
	return mmu_notifier_register(&kvm->mmu_notifier, current->mm);
}

#else  /* !(CONFIG_MMU_NOTIFIER && KVM_ARCH_WANT_MMU_NOTIFIER) */

static int kvm_init_mmu_notifier(struct kvm *kvm)
{
	return 0;
}

380 381
#endif /* CONFIG_MMU_NOTIFIER && KVM_ARCH_WANT_MMU_NOTIFIER */

382
static struct kvm *kvm_create_vm(void)
A
Avi Kivity 已提交
383
{
M
Marcelo Tosatti 已提交
384
	int r = 0, i;
385
	struct kvm *kvm = kvm_arch_create_vm();
A
Avi Kivity 已提交
386

387 388
	if (IS_ERR(kvm))
		goto out;
389 390 391 392 393

	r = hardware_enable_all();
	if (r)
		goto out_err_nodisable;

394 395
#ifdef CONFIG_HAVE_KVM_IRQCHIP
	INIT_HLIST_HEAD(&kvm->mask_notifier_list);
396
	INIT_HLIST_HEAD(&kvm->irq_ack_notifier_list);
397
#endif
A
Avi Kivity 已提交
398

399 400 401 402
	r = -ENOMEM;
	kvm->memslots = kzalloc(sizeof(struct kvm_memslots), GFP_KERNEL);
	if (!kvm->memslots)
		goto out_err;
403 404
	if (init_srcu_struct(&kvm->srcu))
		goto out_err;
M
Marcelo Tosatti 已提交
405 406 407 408 409 410 411 412
	for (i = 0; i < KVM_NR_BUSES; i++) {
		kvm->buses[i] = kzalloc(sizeof(struct kvm_io_bus),
					GFP_KERNEL);
		if (!kvm->buses[i]) {
			cleanup_srcu_struct(&kvm->srcu);
			goto out_err;
		}
	}
413

414
	r = kvm_init_mmu_notifier(kvm);
415
	if (r) {
416
		cleanup_srcu_struct(&kvm->srcu);
417
		goto out_err;
418 419
	}

420 421
	kvm->mm = current->mm;
	atomic_inc(&kvm->mm->mm_count);
422
	spin_lock_init(&kvm->mmu_lock);
423
	raw_spin_lock_init(&kvm->requests_lock);
G
Gregory Haskins 已提交
424
	kvm_eventfd_init(kvm);
S
Shaohua Li 已提交
425
	mutex_init(&kvm->lock);
426
	mutex_init(&kvm->irq_lock);
427
	mutex_init(&kvm->slots_lock);
I
Izik Eidus 已提交
428
	atomic_set(&kvm->users_count, 1);
429 430 431
	spin_lock(&kvm_lock);
	list_add(&kvm->vm_list, &vm_list);
	spin_unlock(&kvm_lock);
432
out:
433
	return kvm;
434 435 436 437

out_err:
	hardware_disable_all();
out_err_nodisable:
M
Marcelo Tosatti 已提交
438 439
	for (i = 0; i < KVM_NR_BUSES; i++)
		kfree(kvm->buses[i]);
440
	kfree(kvm->memslots);
441 442
	kfree(kvm);
	return ERR_PTR(r);
443 444
}

A
Avi Kivity 已提交
445 446 447 448 449 450
/*
 * Free any memory in @free but not in @dont.
 */
static void kvm_free_physmem_slot(struct kvm_memory_slot *free,
				  struct kvm_memory_slot *dont)
{
451 452
	int i;

453 454
	if (!dont || free->rmap != dont->rmap)
		vfree(free->rmap);
A
Avi Kivity 已提交
455 456 457 458

	if (!dont || free->dirty_bitmap != dont->dirty_bitmap)
		vfree(free->dirty_bitmap);

459 460 461 462 463 464 465

	for (i = 0; i < KVM_NR_PAGE_SIZES - 1; ++i) {
		if (!dont || free->lpage_info[i] != dont->lpage_info[i]) {
			vfree(free->lpage_info[i]);
			free->lpage_info[i] = NULL;
		}
	}
M
Marcelo Tosatti 已提交
466

A
Avi Kivity 已提交
467
	free->npages = 0;
A
Al Viro 已提交
468
	free->dirty_bitmap = NULL;
469
	free->rmap = NULL;
A
Avi Kivity 已提交
470 471
}

472
void kvm_free_physmem(struct kvm *kvm)
A
Avi Kivity 已提交
473 474
{
	int i;
475 476 477 478
	struct kvm_memslots *slots = kvm->memslots;

	for (i = 0; i < slots->nmemslots; ++i)
		kvm_free_physmem_slot(&slots->memslots[i], NULL);
A
Avi Kivity 已提交
479

480
	kfree(kvm->memslots);
A
Avi Kivity 已提交
481 482
}

483 484
static void kvm_destroy_vm(struct kvm *kvm)
{
M
Marcelo Tosatti 已提交
485
	int i;
486 487
	struct mm_struct *mm = kvm->mm;

488
	kvm_arch_sync_events(kvm);
489 490 491
	spin_lock(&kvm_lock);
	list_del(&kvm->vm_list);
	spin_unlock(&kvm_lock);
492
	kvm_free_irq_routing(kvm);
M
Marcelo Tosatti 已提交
493 494
	for (i = 0; i < KVM_NR_BUSES; i++)
		kvm_io_bus_destroy(kvm->buses[i]);
495
	kvm_coalesced_mmio_free(kvm);
496 497
#if defined(CONFIG_MMU_NOTIFIER) && defined(KVM_ARCH_WANT_MMU_NOTIFIER)
	mmu_notifier_unregister(&kvm->mmu_notifier, kvm->mm);
498 499
#else
	kvm_arch_flush_shadow(kvm);
500
#endif
501
	kvm_arch_destroy_vm(kvm);
502
	hardware_disable_all();
503
	mmdrop(mm);
504 505
}

I
Izik Eidus 已提交
506 507 508 509 510 511 512 513 514 515 516 517 518 519
void kvm_get_kvm(struct kvm *kvm)
{
	atomic_inc(&kvm->users_count);
}
EXPORT_SYMBOL_GPL(kvm_get_kvm);

void kvm_put_kvm(struct kvm *kvm)
{
	if (atomic_dec_and_test(&kvm->users_count))
		kvm_destroy_vm(kvm);
}
EXPORT_SYMBOL_GPL(kvm_put_kvm);


520 521 522 523
static int kvm_vm_release(struct inode *inode, struct file *filp)
{
	struct kvm *kvm = filp->private_data;

G
Gregory Haskins 已提交
524 525
	kvm_irqfd_release(kvm);

I
Izik Eidus 已提交
526
	kvm_put_kvm(kvm);
A
Avi Kivity 已提交
527 528 529 530 531 532 533 534
	return 0;
}

/*
 * Allocate some memory and give it an address in the guest physical address
 * space.
 *
 * Discontiguous memory is allowed, mostly for framebuffers.
535
 *
536
 * Must be called holding mmap_sem for write.
A
Avi Kivity 已提交
537
 */
538 539 540
int __kvm_set_memory_region(struct kvm *kvm,
			    struct kvm_userspace_memory_region *mem,
			    int user_alloc)
A
Avi Kivity 已提交
541
{
542
	int r, flush_shadow = 0;
A
Avi Kivity 已提交
543
	gfn_t base_gfn;
544 545
	unsigned long npages;
	unsigned long i;
A
Avi Kivity 已提交
546 547
	struct kvm_memory_slot *memslot;
	struct kvm_memory_slot old, new;
548
	struct kvm_memslots *slots, *old_memslots;
A
Avi Kivity 已提交
549 550 551 552 553 554 555

	r = -EINVAL;
	/* General sanity checks */
	if (mem->memory_size & (PAGE_SIZE - 1))
		goto out;
	if (mem->guest_phys_addr & (PAGE_SIZE - 1))
		goto out;
S
Sheng Yang 已提交
556
	if (user_alloc && (mem->userspace_addr & (PAGE_SIZE - 1)))
557
		goto out;
558
	if (mem->slot >= KVM_MEMORY_SLOTS + KVM_PRIVATE_MEM_SLOTS)
A
Avi Kivity 已提交
559 560 561 562
		goto out;
	if (mem->guest_phys_addr + mem->memory_size < mem->guest_phys_addr)
		goto out;

563
	memslot = &kvm->memslots->memslots[mem->slot];
A
Avi Kivity 已提交
564 565 566
	base_gfn = mem->guest_phys_addr >> PAGE_SHIFT;
	npages = mem->memory_size >> PAGE_SHIFT;

567 568 569 570
	r = -EINVAL;
	if (npages > KVM_MEM_MAX_NR_PAGES)
		goto out;

A
Avi Kivity 已提交
571 572 573 574 575
	if (!npages)
		mem->flags &= ~KVM_MEM_LOG_DIRTY_PAGES;

	new = old = *memslot;

576
	new.id = mem->slot;
A
Avi Kivity 已提交
577 578 579 580 581 582 583
	new.base_gfn = base_gfn;
	new.npages = npages;
	new.flags = mem->flags;

	/* Disallow changing a memory slot's size. */
	r = -EINVAL;
	if (npages && old.npages && npages != old.npages)
584
		goto out_free;
A
Avi Kivity 已提交
585 586 587 588

	/* Check for overlaps */
	r = -EEXIST;
	for (i = 0; i < KVM_MEMORY_SLOTS; ++i) {
589
		struct kvm_memory_slot *s = &kvm->memslots->memslots[i];
A
Avi Kivity 已提交
590

591
		if (s == memslot || !s->npages)
A
Avi Kivity 已提交
592 593 594
			continue;
		if (!((base_gfn + npages <= s->base_gfn) ||
		      (base_gfn >= s->base_gfn + s->npages)))
595
			goto out_free;
A
Avi Kivity 已提交
596 597 598 599
	}

	/* Free page dirty bitmap if unneeded */
	if (!(new.flags & KVM_MEM_LOG_DIRTY_PAGES))
A
Al Viro 已提交
600
		new.dirty_bitmap = NULL;
A
Avi Kivity 已提交
601 602 603 604

	r = -ENOMEM;

	/* Allocate if a slot is being created */
605
#ifndef CONFIG_S390
606
	if (npages && !new.rmap) {
L
Lai Jiangshan 已提交
607
		new.rmap = vmalloc(npages * sizeof(*new.rmap));
608 609

		if (!new.rmap)
610
			goto out_free;
611 612

		memset(new.rmap, 0, npages * sizeof(*new.rmap));
613

614
		new.user_alloc = user_alloc;
615
		new.userspace_addr = mem->userspace_addr;
A
Avi Kivity 已提交
616
	}
617 618
	if (!npages)
		goto skip_lpage;
M
Marcelo Tosatti 已提交
619

620
	for (i = 0; i < KVM_NR_PAGE_SIZES - 1; ++i) {
621 622 623
		unsigned long ugfn;
		unsigned long j;
		int lpages;
624
		int level = i + 2;
M
Marcelo Tosatti 已提交
625

626 627 628 629 630 631
		/* Avoid unused variable warning if no large pages */
		(void)level;

		if (new.lpage_info[i])
			continue;

632 633 634
		lpages = 1 + ((base_gfn + npages - 1)
			     >> KVM_HPAGE_GFN_SHIFT(level));
		lpages -= base_gfn >> KVM_HPAGE_GFN_SHIFT(level);
635 636 637 638

		new.lpage_info[i] = vmalloc(lpages * sizeof(*new.lpage_info[i]));

		if (!new.lpage_info[i])
M
Marcelo Tosatti 已提交
639 640
			goto out_free;

641 642
		memset(new.lpage_info[i], 0,
		       lpages * sizeof(*new.lpage_info[i]));
M
Marcelo Tosatti 已提交
643

644
		if (base_gfn & (KVM_PAGES_PER_HPAGE(level) - 1))
645
			new.lpage_info[i][0].write_count = 1;
646
		if ((base_gfn+npages) & (KVM_PAGES_PER_HPAGE(level) - 1))
647
			new.lpage_info[i][lpages - 1].write_count = 1;
648 649 650
		ugfn = new.userspace_addr >> PAGE_SHIFT;
		/*
		 * If the gfn and userspace address are not aligned wrt each
651 652
		 * other, or if explicitly asked to, disable large page
		 * support for this slot
653
		 */
654
		if ((base_gfn ^ ugfn) & (KVM_PAGES_PER_HPAGE(level) - 1) ||
655
		    !largepages_enabled)
656 657
			for (j = 0; j < lpages; ++j)
				new.lpage_info[i][j].write_count = 1;
M
Marcelo Tosatti 已提交
658
	}
A
Avi Kivity 已提交
659

660 661
skip_lpage:

A
Avi Kivity 已提交
662 663
	/* Allocate page dirty bitmap if needed */
	if ((new.flags & KVM_MEM_LOG_DIRTY_PAGES) && !new.dirty_bitmap) {
664
		unsigned long dirty_bytes = kvm_dirty_bitmap_bytes(&new);
A
Avi Kivity 已提交
665 666 667

		new.dirty_bitmap = vmalloc(dirty_bytes);
		if (!new.dirty_bitmap)
668
			goto out_free;
A
Avi Kivity 已提交
669
		memset(new.dirty_bitmap, 0, dirty_bytes);
670
		/* destroy any largepage mappings for dirty tracking */
671
		if (old.npages)
672
			flush_shadow = 1;
A
Avi Kivity 已提交
673
	}
674 675 676 677
#else  /* not defined CONFIG_S390 */
	new.user_alloc = user_alloc;
	if (user_alloc)
		new.userspace_addr = mem->userspace_addr;
678
#endif /* not defined CONFIG_S390 */
A
Avi Kivity 已提交
679

680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699
	if (!npages) {
		r = -ENOMEM;
		slots = kzalloc(sizeof(struct kvm_memslots), GFP_KERNEL);
		if (!slots)
			goto out_free;
		memcpy(slots, kvm->memslots, sizeof(struct kvm_memslots));
		if (mem->slot >= slots->nmemslots)
			slots->nmemslots = mem->slot + 1;
		slots->memslots[mem->slot].flags |= KVM_MEMSLOT_INVALID;

		old_memslots = kvm->memslots;
		rcu_assign_pointer(kvm->memslots, slots);
		synchronize_srcu_expedited(&kvm->srcu);
		/* From this point no new shadow pages pointing to a deleted
		 * memslot will be created.
		 *
		 * validation of sp->gfn happens in:
		 * 	- gfn_to_hva (kvm_read_guest, gfn_to_pfn)
		 * 	- kvm_is_visible_gfn (mmu_check_roots)
		 */
700
		kvm_arch_flush_shadow(kvm);
701 702
		kfree(old_memslots);
	}
703

704 705 706 707
	r = kvm_arch_prepare_memory_region(kvm, &new, old, mem, user_alloc);
	if (r)
		goto out_free;

708 709 710 711 712 713
	/* map the pages in iommu page table */
	if (npages) {
		r = kvm_iommu_map_pages(kvm, &new);
		if (r)
			goto out_free;
	}
714

715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734
	r = -ENOMEM;
	slots = kzalloc(sizeof(struct kvm_memslots), GFP_KERNEL);
	if (!slots)
		goto out_free;
	memcpy(slots, kvm->memslots, sizeof(struct kvm_memslots));
	if (mem->slot >= slots->nmemslots)
		slots->nmemslots = mem->slot + 1;

	/* actual memory is freed via old in kvm_free_physmem_slot below */
	if (!npages) {
		new.rmap = NULL;
		new.dirty_bitmap = NULL;
		for (i = 0; i < KVM_NR_PAGE_SIZES - 1; ++i)
			new.lpage_info[i] = NULL;
	}

	slots->memslots[mem->slot] = new;
	old_memslots = kvm->memslots;
	rcu_assign_pointer(kvm->memslots, slots);
	synchronize_srcu_expedited(&kvm->srcu);
735

736
	kvm_arch_commit_memory_region(kvm, mem, old, user_alloc);
737

738 739 740 741 742 743
	kvm_free_physmem_slot(&old, &new);
	kfree(old_memslots);

	if (flush_shadow)
		kvm_arch_flush_shadow(kvm);

A
Avi Kivity 已提交
744 745
	return 0;

746
out_free:
A
Avi Kivity 已提交
747 748 749
	kvm_free_physmem_slot(&new, &old);
out:
	return r;
750 751

}
752 753 754 755 756 757 758 759
EXPORT_SYMBOL_GPL(__kvm_set_memory_region);

int kvm_set_memory_region(struct kvm *kvm,
			  struct kvm_userspace_memory_region *mem,
			  int user_alloc)
{
	int r;

760
	mutex_lock(&kvm->slots_lock);
761
	r = __kvm_set_memory_region(kvm, mem, user_alloc);
762
	mutex_unlock(&kvm->slots_lock);
763 764
	return r;
}
765 766
EXPORT_SYMBOL_GPL(kvm_set_memory_region);

767 768 769 770
int kvm_vm_ioctl_set_memory_region(struct kvm *kvm,
				   struct
				   kvm_userspace_memory_region *mem,
				   int user_alloc)
771
{
772 773
	if (mem->slot >= KVM_MEMORY_SLOTS)
		return -EINVAL;
774
	return kvm_set_memory_region(kvm, mem, user_alloc);
A
Avi Kivity 已提交
775 776
}

777 778
int kvm_get_dirty_log(struct kvm *kvm,
			struct kvm_dirty_log *log, int *is_dirty)
A
Avi Kivity 已提交
779 780 781
{
	struct kvm_memory_slot *memslot;
	int r, i;
782
	unsigned long n;
A
Avi Kivity 已提交
783 784 785 786 787 788
	unsigned long any = 0;

	r = -EINVAL;
	if (log->slot >= KVM_MEMORY_SLOTS)
		goto out;

789
	memslot = &kvm->memslots->memslots[log->slot];
A
Avi Kivity 已提交
790 791 792 793
	r = -ENOENT;
	if (!memslot->dirty_bitmap)
		goto out;

794
	n = kvm_dirty_bitmap_bytes(memslot);
A
Avi Kivity 已提交
795

796
	for (i = 0; !any && i < n/sizeof(long); ++i)
A
Avi Kivity 已提交
797 798 799 800 801 802
		any = memslot->dirty_bitmap[i];

	r = -EFAULT;
	if (copy_to_user(log->dirty_bitmap, memslot->dirty_bitmap, n))
		goto out;

803 804
	if (any)
		*is_dirty = 1;
A
Avi Kivity 已提交
805 806 807 808 809 810

	r = 0;
out:
	return r;
}

811 812 813 814 815 816
void kvm_disable_largepages(void)
{
	largepages_enabled = false;
}
EXPORT_SYMBOL_GPL(kvm_disable_largepages);

817 818
int is_error_page(struct page *page)
{
819
	return page == bad_page || page == hwpoison_page || page == fault_page;
820 821 822
}
EXPORT_SYMBOL_GPL(is_error_page);

823 824
int is_error_pfn(pfn_t pfn)
{
825
	return pfn == bad_pfn || pfn == hwpoison_pfn || pfn == fault_pfn;
826 827 828
}
EXPORT_SYMBOL_GPL(is_error_pfn);

829 830 831 832 833 834
int is_hwpoison_pfn(pfn_t pfn)
{
	return pfn == hwpoison_pfn;
}
EXPORT_SYMBOL_GPL(is_hwpoison_pfn);

835 836 837 838 839 840
int is_fault_pfn(pfn_t pfn)
{
	return pfn == fault_pfn;
}
EXPORT_SYMBOL_GPL(is_fault_pfn);

I
Izik Eidus 已提交
841 842 843 844 845 846 847 848 849 850 851
static inline unsigned long bad_hva(void)
{
	return PAGE_OFFSET;
}

int kvm_is_error_hva(unsigned long addr)
{
	return addr == bad_hva();
}
EXPORT_SYMBOL_GPL(kvm_is_error_hva);

A
Avi Kivity 已提交
852
struct kvm_memory_slot *gfn_to_memslot(struct kvm *kvm, gfn_t gfn)
A
Avi Kivity 已提交
853 854
{
	int i;
855
	struct kvm_memslots *slots = kvm_memslots(kvm);
A
Avi Kivity 已提交
856

857 858
	for (i = 0; i < slots->nmemslots; ++i) {
		struct kvm_memory_slot *memslot = &slots->memslots[i];
A
Avi Kivity 已提交
859 860 861 862 863

		if (gfn >= memslot->base_gfn
		    && gfn < memslot->base_gfn + memslot->npages)
			return memslot;
	}
A
Al Viro 已提交
864
	return NULL;
A
Avi Kivity 已提交
865
}
A
Avi Kivity 已提交
866
EXPORT_SYMBOL_GPL(gfn_to_memslot);
A
Avi Kivity 已提交
867

868 869 870
int kvm_is_visible_gfn(struct kvm *kvm, gfn_t gfn)
{
	int i;
871
	struct kvm_memslots *slots = kvm_memslots(kvm);
872 873

	for (i = 0; i < KVM_MEMORY_SLOTS; ++i) {
874
		struct kvm_memory_slot *memslot = &slots->memslots[i];
875

876 877 878
		if (memslot->flags & KVM_MEMSLOT_INVALID)
			continue;

879 880 881 882 883 884 885 886
		if (gfn >= memslot->base_gfn
		    && gfn < memslot->base_gfn + memslot->npages)
			return 1;
	}
	return 0;
}
EXPORT_SYMBOL_GPL(kvm_is_visible_gfn);

J
Joerg Roedel 已提交
887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910
unsigned long kvm_host_page_size(struct kvm *kvm, gfn_t gfn)
{
	struct vm_area_struct *vma;
	unsigned long addr, size;

	size = PAGE_SIZE;

	addr = gfn_to_hva(kvm, gfn);
	if (kvm_is_error_hva(addr))
		return PAGE_SIZE;

	down_read(&current->mm->mmap_sem);
	vma = find_vma(current->mm, addr);
	if (!vma)
		goto out;

	size = vma_kernel_pagesize(vma);

out:
	up_read(&current->mm->mmap_sem);

	return size;
}

911 912 913
int memslot_id(struct kvm *kvm, gfn_t gfn)
{
	int i;
914
	struct kvm_memslots *slots = kvm_memslots(kvm);
915 916 917 918 919 920 921 922 923 924 925 926 927
	struct kvm_memory_slot *memslot = NULL;

	for (i = 0; i < slots->nmemslots; ++i) {
		memslot = &slots->memslots[i];

		if (gfn >= memslot->base_gfn
		    && gfn < memslot->base_gfn + memslot->npages)
			break;
	}

	return memslot - slots->memslots;
}

928 929
static unsigned long gfn_to_hva_many(struct kvm *kvm, gfn_t gfn,
				     gfn_t *nr_pages)
I
Izik Eidus 已提交
930 931 932
{
	struct kvm_memory_slot *slot;

A
Avi Kivity 已提交
933
	slot = gfn_to_memslot(kvm, gfn);
934
	if (!slot || slot->flags & KVM_MEMSLOT_INVALID)
I
Izik Eidus 已提交
935
		return bad_hva();
936 937 938 939

	if (nr_pages)
		*nr_pages = slot->npages - (gfn - slot->base_gfn);

940
	return gfn_to_hva_memslot(slot, gfn);
I
Izik Eidus 已提交
941
}
942 943 944 945 946

unsigned long gfn_to_hva(struct kvm *kvm, gfn_t gfn)
{
	return gfn_to_hva_many(kvm, gfn, NULL);
}
947
EXPORT_SYMBOL_GPL(gfn_to_hva);
I
Izik Eidus 已提交
948

949
static pfn_t hva_to_pfn(struct kvm *kvm, unsigned long addr, bool atomic)
A
Avi Kivity 已提交
950
{
951 952
	struct page *page[1];
	int npages;
953
	pfn_t pfn;
A
Avi Kivity 已提交
954

955 956 957 958 959 960
	if (atomic)
		npages = __get_user_pages_fast(addr, 1, 1, page);
	else {
		might_sleep();
		npages = get_user_pages_fast(addr, 1, 1, page);
	}
I
Izik Eidus 已提交
961

962 963 964
	if (unlikely(npages != 1)) {
		struct vm_area_struct *vma;

965 966 967
		if (atomic)
			goto return_fault_page;

968
		down_read(&current->mm->mmap_sem);
969
		if (is_hwpoison_address(addr)) {
970
			up_read(&current->mm->mmap_sem);
971 972 973 974
			get_page(hwpoison_page);
			return page_to_pfn(hwpoison_page);
		}

975
		vma = find_vma(current->mm, addr);
976

977 978
		if (vma == NULL || addr < vma->vm_start ||
		    !(vma->vm_flags & VM_PFNMAP)) {
979
			up_read(&current->mm->mmap_sem);
980
return_fault_page:
981 982
			get_page(fault_page);
			return page_to_pfn(fault_page);
983 984 985
		}

		pfn = ((addr - vma->vm_start) >> PAGE_SHIFT) + vma->vm_pgoff;
986
		up_read(&current->mm->mmap_sem);
987
		BUG_ON(!kvm_is_mmio_pfn(pfn));
988 989
	} else
		pfn = page_to_pfn(page[0]);
990

991
	return pfn;
992 993
}

994 995 996 997 998 999
pfn_t hva_to_pfn_atomic(struct kvm *kvm, unsigned long addr)
{
	return hva_to_pfn(kvm, addr, true);
}
EXPORT_SYMBOL_GPL(hva_to_pfn_atomic);

1000
static pfn_t __gfn_to_pfn(struct kvm *kvm, gfn_t gfn, bool atomic)
1001 1002 1003 1004 1005 1006 1007 1008 1009
{
	unsigned long addr;

	addr = gfn_to_hva(kvm, gfn);
	if (kvm_is_error_hva(addr)) {
		get_page(bad_page);
		return page_to_pfn(bad_page);
	}

1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021
	return hva_to_pfn(kvm, addr, atomic);
}

pfn_t gfn_to_pfn_atomic(struct kvm *kvm, gfn_t gfn)
{
	return __gfn_to_pfn(kvm, gfn, true);
}
EXPORT_SYMBOL_GPL(gfn_to_pfn_atomic);

pfn_t gfn_to_pfn(struct kvm *kvm, gfn_t gfn)
{
	return __gfn_to_pfn(kvm, gfn, false);
1022
}
1023 1024
EXPORT_SYMBOL_GPL(gfn_to_pfn);

1025 1026 1027 1028
pfn_t gfn_to_pfn_memslot(struct kvm *kvm,
			 struct kvm_memory_slot *slot, gfn_t gfn)
{
	unsigned long addr = gfn_to_hva_memslot(slot, gfn);
1029
	return hva_to_pfn(kvm, addr, false);
1030 1031
}

1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048
int gfn_to_page_many_atomic(struct kvm *kvm, gfn_t gfn, struct page **pages,
								  int nr_pages)
{
	unsigned long addr;
	gfn_t entry;

	addr = gfn_to_hva_many(kvm, gfn, &entry);
	if (kvm_is_error_hva(addr))
		return -1;

	if (entry < nr_pages)
		return 0;

	return __get_user_pages_fast(addr, nr_pages, 1, pages);
}
EXPORT_SYMBOL_GPL(gfn_to_page_many_atomic);

1049 1050
struct page *gfn_to_page(struct kvm *kvm, gfn_t gfn)
{
1051 1052 1053
	pfn_t pfn;

	pfn = gfn_to_pfn(kvm, gfn);
1054
	if (!kvm_is_mmio_pfn(pfn))
1055 1056
		return pfn_to_page(pfn);

1057
	WARN_ON(kvm_is_mmio_pfn(pfn));
1058 1059 1060

	get_page(bad_page);
	return bad_page;
A
Avi Kivity 已提交
1061
}
1062

A
Avi Kivity 已提交
1063 1064
EXPORT_SYMBOL_GPL(gfn_to_page);

1065 1066
void kvm_release_page_clean(struct page *page)
{
1067
	kvm_release_pfn_clean(page_to_pfn(page));
1068 1069 1070
}
EXPORT_SYMBOL_GPL(kvm_release_page_clean);

1071 1072
void kvm_release_pfn_clean(pfn_t pfn)
{
1073
	if (!kvm_is_mmio_pfn(pfn))
1074
		put_page(pfn_to_page(pfn));
1075 1076 1077
}
EXPORT_SYMBOL_GPL(kvm_release_pfn_clean);

1078
void kvm_release_page_dirty(struct page *page)
1079
{
1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
	kvm_release_pfn_dirty(page_to_pfn(page));
}
EXPORT_SYMBOL_GPL(kvm_release_page_dirty);

void kvm_release_pfn_dirty(pfn_t pfn)
{
	kvm_set_pfn_dirty(pfn);
	kvm_release_pfn_clean(pfn);
}
EXPORT_SYMBOL_GPL(kvm_release_pfn_dirty);

void kvm_set_page_dirty(struct page *page)
{
	kvm_set_pfn_dirty(page_to_pfn(page));
}
EXPORT_SYMBOL_GPL(kvm_set_page_dirty);

void kvm_set_pfn_dirty(pfn_t pfn)
{
1099
	if (!kvm_is_mmio_pfn(pfn)) {
1100 1101 1102 1103
		struct page *page = pfn_to_page(pfn);
		if (!PageReserved(page))
			SetPageDirty(page);
	}
1104
}
1105 1106 1107 1108
EXPORT_SYMBOL_GPL(kvm_set_pfn_dirty);

void kvm_set_pfn_accessed(pfn_t pfn)
{
1109
	if (!kvm_is_mmio_pfn(pfn))
1110
		mark_page_accessed(pfn_to_page(pfn));
1111 1112 1113 1114 1115
}
EXPORT_SYMBOL_GPL(kvm_set_pfn_accessed);

void kvm_get_pfn(pfn_t pfn)
{
1116
	if (!kvm_is_mmio_pfn(pfn))
1117
		get_page(pfn_to_page(pfn));
1118 1119
}
EXPORT_SYMBOL_GPL(kvm_get_pfn);
1120

1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131
static int next_segment(unsigned long len, int offset)
{
	if (len > PAGE_SIZE - offset)
		return PAGE_SIZE - offset;
	else
		return len;
}

int kvm_read_guest_page(struct kvm *kvm, gfn_t gfn, void *data, int offset,
			int len)
{
1132 1133
	int r;
	unsigned long addr;
1134

1135 1136 1137 1138 1139
	addr = gfn_to_hva(kvm, gfn);
	if (kvm_is_error_hva(addr))
		return -EFAULT;
	r = copy_from_user(data, (void __user *)addr + offset, len);
	if (r)
1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164
		return -EFAULT;
	return 0;
}
EXPORT_SYMBOL_GPL(kvm_read_guest_page);

int kvm_read_guest(struct kvm *kvm, gpa_t gpa, void *data, unsigned long len)
{
	gfn_t gfn = gpa >> PAGE_SHIFT;
	int seg;
	int offset = offset_in_page(gpa);
	int ret;

	while ((seg = next_segment(len, offset)) != 0) {
		ret = kvm_read_guest_page(kvm, gfn, data, offset, seg);
		if (ret < 0)
			return ret;
		offset = 0;
		len -= seg;
		data += seg;
		++gfn;
	}
	return 0;
}
EXPORT_SYMBOL_GPL(kvm_read_guest);

1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175
int kvm_read_guest_atomic(struct kvm *kvm, gpa_t gpa, void *data,
			  unsigned long len)
{
	int r;
	unsigned long addr;
	gfn_t gfn = gpa >> PAGE_SHIFT;
	int offset = offset_in_page(gpa);

	addr = gfn_to_hva(kvm, gfn);
	if (kvm_is_error_hva(addr))
		return -EFAULT;
1176
	pagefault_disable();
1177
	r = __copy_from_user_inatomic(data, (void __user *)addr + offset, len);
1178
	pagefault_enable();
1179 1180 1181 1182 1183 1184
	if (r)
		return -EFAULT;
	return 0;
}
EXPORT_SYMBOL(kvm_read_guest_atomic);

1185 1186 1187
int kvm_write_guest_page(struct kvm *kvm, gfn_t gfn, const void *data,
			 int offset, int len)
{
1188 1189
	int r;
	unsigned long addr;
1190

1191 1192 1193 1194 1195
	addr = gfn_to_hva(kvm, gfn);
	if (kvm_is_error_hva(addr))
		return -EFAULT;
	r = copy_to_user((void __user *)addr + offset, data, len);
	if (r)
1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223
		return -EFAULT;
	mark_page_dirty(kvm, gfn);
	return 0;
}
EXPORT_SYMBOL_GPL(kvm_write_guest_page);

int kvm_write_guest(struct kvm *kvm, gpa_t gpa, const void *data,
		    unsigned long len)
{
	gfn_t gfn = gpa >> PAGE_SHIFT;
	int seg;
	int offset = offset_in_page(gpa);
	int ret;

	while ((seg = next_segment(len, offset)) != 0) {
		ret = kvm_write_guest_page(kvm, gfn, data, offset, seg);
		if (ret < 0)
			return ret;
		offset = 0;
		len -= seg;
		data += seg;
		++gfn;
	}
	return 0;
}

int kvm_clear_guest_page(struct kvm *kvm, gfn_t gfn, int offset, int len)
{
1224
	return kvm_write_guest_page(kvm, gfn, empty_zero_page, offset, len);
1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246
}
EXPORT_SYMBOL_GPL(kvm_clear_guest_page);

int kvm_clear_guest(struct kvm *kvm, gpa_t gpa, unsigned long len)
{
	gfn_t gfn = gpa >> PAGE_SHIFT;
	int seg;
	int offset = offset_in_page(gpa);
	int ret;

        while ((seg = next_segment(len, offset)) != 0) {
		ret = kvm_clear_guest_page(kvm, gfn, offset, seg);
		if (ret < 0)
			return ret;
		offset = 0;
		len -= seg;
		++gfn;
	}
	return 0;
}
EXPORT_SYMBOL_GPL(kvm_clear_guest);

A
Avi Kivity 已提交
1247 1248
void mark_page_dirty(struct kvm *kvm, gfn_t gfn)
{
1249
	struct kvm_memory_slot *memslot;
A
Avi Kivity 已提交
1250

A
Avi Kivity 已提交
1251
	memslot = gfn_to_memslot(kvm, gfn);
R
Rusty Russell 已提交
1252 1253
	if (memslot && memslot->dirty_bitmap) {
		unsigned long rel_gfn = gfn - memslot->base_gfn;
A
Avi Kivity 已提交
1254

1255
		generic___set_le_bit(rel_gfn, memslot->dirty_bitmap);
A
Avi Kivity 已提交
1256 1257 1258
	}
}

E
Eddie Dong 已提交
1259 1260 1261
/*
 * The vCPU has executed a HLT instruction with in-kernel mode enabled.
 */
1262
void kvm_vcpu_block(struct kvm_vcpu *vcpu)
1263
{
1264 1265 1266 1267 1268
	DEFINE_WAIT(wait);

	for (;;) {
		prepare_to_wait(&vcpu->wq, &wait, TASK_INTERRUPTIBLE);

1269
		if (kvm_arch_vcpu_runnable(vcpu)) {
1270
			kvm_make_request(KVM_REQ_UNHALT, vcpu);
1271
			break;
1272
		}
1273 1274
		if (kvm_cpu_has_pending_timer(vcpu))
			break;
1275 1276 1277
		if (signal_pending(current))
			break;

E
Eddie Dong 已提交
1278 1279
		schedule();
	}
1280

1281
	finish_wait(&vcpu->wq, &wait);
E
Eddie Dong 已提交
1282 1283
}

A
Avi Kivity 已提交
1284 1285
void kvm_resched(struct kvm_vcpu *vcpu)
{
1286 1287
	if (!need_resched())
		return;
A
Avi Kivity 已提交
1288 1289 1290 1291
	cond_resched();
}
EXPORT_SYMBOL_GPL(kvm_resched);

Z
Zhai, Edwin 已提交
1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306
void kvm_vcpu_on_spin(struct kvm_vcpu *vcpu)
{
	ktime_t expires;
	DEFINE_WAIT(wait);

	prepare_to_wait(&vcpu->wq, &wait, TASK_INTERRUPTIBLE);

	/* Sleep for 100 us, and hope lock-holder got scheduled */
	expires = ktime_add_ns(ktime_get(), 100000UL);
	schedule_hrtimeout(&expires, HRTIMER_MODE_ABS);

	finish_wait(&vcpu->wq, &wait);
}
EXPORT_SYMBOL_GPL(kvm_vcpu_on_spin);

1307
static int kvm_vcpu_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
1308 1309 1310 1311
{
	struct kvm_vcpu *vcpu = vma->vm_file->private_data;
	struct page *page;

1312
	if (vmf->pgoff == 0)
1313
		page = virt_to_page(vcpu->run);
A
Avi Kivity 已提交
1314
#ifdef CONFIG_X86
1315
	else if (vmf->pgoff == KVM_PIO_PAGE_OFFSET)
1316
		page = virt_to_page(vcpu->arch.pio_data);
1317 1318 1319 1320
#endif
#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
	else if (vmf->pgoff == KVM_COALESCED_MMIO_PAGE_OFFSET)
		page = virt_to_page(vcpu->kvm->coalesced_mmio_ring);
A
Avi Kivity 已提交
1321
#endif
1322
	else
1323
		return VM_FAULT_SIGBUS;
1324
	get_page(page);
1325 1326
	vmf->page = page;
	return 0;
1327 1328
}

1329
static const struct vm_operations_struct kvm_vcpu_vm_ops = {
1330
	.fault = kvm_vcpu_fault,
1331 1332 1333 1334 1335 1336 1337 1338
};

static int kvm_vcpu_mmap(struct file *file, struct vm_area_struct *vma)
{
	vma->vm_ops = &kvm_vcpu_vm_ops;
	return 0;
}

A
Avi Kivity 已提交
1339 1340 1341 1342
static int kvm_vcpu_release(struct inode *inode, struct file *filp)
{
	struct kvm_vcpu *vcpu = filp->private_data;

A
Al Viro 已提交
1343
	kvm_put_kvm(vcpu->kvm);
A
Avi Kivity 已提交
1344 1345 1346
	return 0;
}

1347
static struct file_operations kvm_vcpu_fops = {
A
Avi Kivity 已提交
1348 1349 1350
	.release        = kvm_vcpu_release,
	.unlocked_ioctl = kvm_vcpu_ioctl,
	.compat_ioctl   = kvm_vcpu_ioctl,
1351
	.mmap           = kvm_vcpu_mmap,
1352
	.llseek		= noop_llseek,
A
Avi Kivity 已提交
1353 1354 1355 1356 1357 1358 1359
};

/*
 * Allocates an inode for the vcpu.
 */
static int create_vcpu_fd(struct kvm_vcpu *vcpu)
{
1360
	return anon_inode_getfd("kvm-vcpu", &kvm_vcpu_fops, vcpu, O_RDWR);
A
Avi Kivity 已提交
1361 1362
}

1363 1364 1365
/*
 * Creates some virtual cpus.  Good luck creating more than one.
 */
1366
static int kvm_vm_ioctl_create_vcpu(struct kvm *kvm, u32 id)
1367 1368
{
	int r;
1369
	struct kvm_vcpu *vcpu, *v;
1370

1371
	vcpu = kvm_arch_vcpu_create(kvm, id);
R
Rusty Russell 已提交
1372 1373
	if (IS_ERR(vcpu))
		return PTR_ERR(vcpu);
1374

1375 1376
	preempt_notifier_init(&vcpu->preempt_notifier, &kvm_preempt_ops);

1377 1378
	r = kvm_arch_vcpu_setup(vcpu);
	if (r)
1379
		return r;
1380

S
Shaohua Li 已提交
1381
	mutex_lock(&kvm->lock);
1382 1383
	if (atomic_read(&kvm->online_vcpus) == KVM_MAX_VCPUS) {
		r = -EINVAL;
1384
		goto vcpu_destroy;
R
Rusty Russell 已提交
1385
	}
1386

1387 1388
	kvm_for_each_vcpu(r, v, kvm)
		if (v->vcpu_id == id) {
1389 1390 1391 1392 1393
			r = -EEXIST;
			goto vcpu_destroy;
		}

	BUG_ON(kvm->vcpus[atomic_read(&kvm->online_vcpus)]);
1394

R
Rusty Russell 已提交
1395
	/* Now it's all set up, let userspace reach it */
A
Al Viro 已提交
1396
	kvm_get_kvm(kvm);
A
Avi Kivity 已提交
1397
	r = create_vcpu_fd(vcpu);
1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411
	if (r < 0) {
		kvm_put_kvm(kvm);
		goto vcpu_destroy;
	}

	kvm->vcpus[atomic_read(&kvm->online_vcpus)] = vcpu;
	smp_wmb();
	atomic_inc(&kvm->online_vcpus);

#ifdef CONFIG_KVM_APIC_ARCHITECTURE
	if (kvm->bsp_vcpu_id == id)
		kvm->bsp_vcpu = vcpu;
#endif
	mutex_unlock(&kvm->lock);
R
Rusty Russell 已提交
1412
	return r;
1413

1414
vcpu_destroy:
1415
	mutex_unlock(&kvm->lock);
1416
	kvm_arch_vcpu_destroy(vcpu);
1417 1418 1419
	return r;
}

A
Avi Kivity 已提交
1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430
static int kvm_vcpu_ioctl_set_sigmask(struct kvm_vcpu *vcpu, sigset_t *sigset)
{
	if (sigset) {
		sigdelsetmask(sigset, sigmask(SIGKILL)|sigmask(SIGSTOP));
		vcpu->sigset_active = 1;
		vcpu->sigset = *sigset;
	} else
		vcpu->sigset_active = 0;
	return 0;
}

A
Avi Kivity 已提交
1431 1432
static long kvm_vcpu_ioctl(struct file *filp,
			   unsigned int ioctl, unsigned long arg)
A
Avi Kivity 已提交
1433
{
A
Avi Kivity 已提交
1434
	struct kvm_vcpu *vcpu = filp->private_data;
A
Al Viro 已提交
1435
	void __user *argp = (void __user *)arg;
1436
	int r;
1437 1438
	struct kvm_fpu *fpu = NULL;
	struct kvm_sregs *kvm_sregs = NULL;
A
Avi Kivity 已提交
1439

1440 1441
	if (vcpu->kvm->mm != current->mm)
		return -EIO;
1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453

#if defined(CONFIG_S390) || defined(CONFIG_PPC)
	/*
	 * Special cases: vcpu ioctls that are asynchronous to vcpu execution,
	 * so vcpu_load() would break it.
	 */
	if (ioctl == KVM_S390_INTERRUPT || ioctl == KVM_INTERRUPT)
		return kvm_arch_vcpu_ioctl(filp, ioctl, arg);
#endif


	vcpu_load(vcpu);
A
Avi Kivity 已提交
1454
	switch (ioctl) {
1455
	case KVM_RUN:
1456 1457 1458
		r = -EINVAL;
		if (arg)
			goto out;
1459
		r = kvm_arch_vcpu_ioctl_run(vcpu, vcpu->run);
A
Avi Kivity 已提交
1460 1461
		break;
	case KVM_GET_REGS: {
1462
		struct kvm_regs *kvm_regs;
A
Avi Kivity 已提交
1463

1464 1465 1466
		r = -ENOMEM;
		kvm_regs = kzalloc(sizeof(struct kvm_regs), GFP_KERNEL);
		if (!kvm_regs)
A
Avi Kivity 已提交
1467
			goto out;
1468 1469 1470
		r = kvm_arch_vcpu_ioctl_get_regs(vcpu, kvm_regs);
		if (r)
			goto out_free1;
A
Avi Kivity 已提交
1471
		r = -EFAULT;
1472 1473
		if (copy_to_user(argp, kvm_regs, sizeof(struct kvm_regs)))
			goto out_free1;
A
Avi Kivity 已提交
1474
		r = 0;
1475 1476
out_free1:
		kfree(kvm_regs);
A
Avi Kivity 已提交
1477 1478 1479
		break;
	}
	case KVM_SET_REGS: {
1480
		struct kvm_regs *kvm_regs;
A
Avi Kivity 已提交
1481

1482 1483 1484
		r = -ENOMEM;
		kvm_regs = kzalloc(sizeof(struct kvm_regs), GFP_KERNEL);
		if (!kvm_regs)
A
Avi Kivity 已提交
1485
			goto out;
1486 1487 1488 1489
		r = -EFAULT;
		if (copy_from_user(kvm_regs, argp, sizeof(struct kvm_regs)))
			goto out_free2;
		r = kvm_arch_vcpu_ioctl_set_regs(vcpu, kvm_regs);
A
Avi Kivity 已提交
1490
		if (r)
1491
			goto out_free2;
A
Avi Kivity 已提交
1492
		r = 0;
1493 1494
out_free2:
		kfree(kvm_regs);
A
Avi Kivity 已提交
1495 1496 1497
		break;
	}
	case KVM_GET_SREGS: {
1498 1499 1500 1501 1502
		kvm_sregs = kzalloc(sizeof(struct kvm_sregs), GFP_KERNEL);
		r = -ENOMEM;
		if (!kvm_sregs)
			goto out;
		r = kvm_arch_vcpu_ioctl_get_sregs(vcpu, kvm_sregs);
A
Avi Kivity 已提交
1503 1504 1505
		if (r)
			goto out;
		r = -EFAULT;
1506
		if (copy_to_user(argp, kvm_sregs, sizeof(struct kvm_sregs)))
A
Avi Kivity 已提交
1507 1508 1509 1510 1511
			goto out;
		r = 0;
		break;
	}
	case KVM_SET_SREGS: {
1512 1513 1514 1515
		kvm_sregs = kmalloc(sizeof(struct kvm_sregs), GFP_KERNEL);
		r = -ENOMEM;
		if (!kvm_sregs)
			goto out;
A
Avi Kivity 已提交
1516
		r = -EFAULT;
1517
		if (copy_from_user(kvm_sregs, argp, sizeof(struct kvm_sregs)))
A
Avi Kivity 已提交
1518
			goto out;
1519
		r = kvm_arch_vcpu_ioctl_set_sregs(vcpu, kvm_sregs);
A
Avi Kivity 已提交
1520 1521 1522 1523 1524
		if (r)
			goto out;
		r = 0;
		break;
	}
1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548
	case KVM_GET_MP_STATE: {
		struct kvm_mp_state mp_state;

		r = kvm_arch_vcpu_ioctl_get_mpstate(vcpu, &mp_state);
		if (r)
			goto out;
		r = -EFAULT;
		if (copy_to_user(argp, &mp_state, sizeof mp_state))
			goto out;
		r = 0;
		break;
	}
	case KVM_SET_MP_STATE: {
		struct kvm_mp_state mp_state;

		r = -EFAULT;
		if (copy_from_user(&mp_state, argp, sizeof mp_state))
			goto out;
		r = kvm_arch_vcpu_ioctl_set_mpstate(vcpu, &mp_state);
		if (r)
			goto out;
		r = 0;
		break;
	}
A
Avi Kivity 已提交
1549 1550 1551 1552
	case KVM_TRANSLATE: {
		struct kvm_translation tr;

		r = -EFAULT;
A
Al Viro 已提交
1553
		if (copy_from_user(&tr, argp, sizeof tr))
A
Avi Kivity 已提交
1554
			goto out;
1555
		r = kvm_arch_vcpu_ioctl_translate(vcpu, &tr);
A
Avi Kivity 已提交
1556 1557 1558
		if (r)
			goto out;
		r = -EFAULT;
A
Al Viro 已提交
1559
		if (copy_to_user(argp, &tr, sizeof tr))
A
Avi Kivity 已提交
1560 1561 1562 1563
			goto out;
		r = 0;
		break;
	}
J
Jan Kiszka 已提交
1564 1565
	case KVM_SET_GUEST_DEBUG: {
		struct kvm_guest_debug dbg;
A
Avi Kivity 已提交
1566 1567

		r = -EFAULT;
A
Al Viro 已提交
1568
		if (copy_from_user(&dbg, argp, sizeof dbg))
A
Avi Kivity 已提交
1569
			goto out;
J
Jan Kiszka 已提交
1570
		r = kvm_arch_vcpu_ioctl_set_guest_debug(vcpu, &dbg);
A
Avi Kivity 已提交
1571 1572 1573 1574 1575
		if (r)
			goto out;
		r = 0;
		break;
	}
A
Avi Kivity 已提交
1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595
	case KVM_SET_SIGNAL_MASK: {
		struct kvm_signal_mask __user *sigmask_arg = argp;
		struct kvm_signal_mask kvm_sigmask;
		sigset_t sigset, *p;

		p = NULL;
		if (argp) {
			r = -EFAULT;
			if (copy_from_user(&kvm_sigmask, argp,
					   sizeof kvm_sigmask))
				goto out;
			r = -EINVAL;
			if (kvm_sigmask.len != sizeof sigset)
				goto out;
			r = -EFAULT;
			if (copy_from_user(&sigset, sigmask_arg->sigset,
					   sizeof sigset))
				goto out;
			p = &sigset;
		}
1596
		r = kvm_vcpu_ioctl_set_sigmask(vcpu, p);
A
Avi Kivity 已提交
1597 1598
		break;
	}
A
Avi Kivity 已提交
1599
	case KVM_GET_FPU: {
1600 1601 1602 1603 1604
		fpu = kzalloc(sizeof(struct kvm_fpu), GFP_KERNEL);
		r = -ENOMEM;
		if (!fpu)
			goto out;
		r = kvm_arch_vcpu_ioctl_get_fpu(vcpu, fpu);
A
Avi Kivity 已提交
1605 1606 1607
		if (r)
			goto out;
		r = -EFAULT;
1608
		if (copy_to_user(argp, fpu, sizeof(struct kvm_fpu)))
A
Avi Kivity 已提交
1609 1610 1611 1612 1613
			goto out;
		r = 0;
		break;
	}
	case KVM_SET_FPU: {
1614 1615 1616 1617
		fpu = kmalloc(sizeof(struct kvm_fpu), GFP_KERNEL);
		r = -ENOMEM;
		if (!fpu)
			goto out;
A
Avi Kivity 已提交
1618
		r = -EFAULT;
1619
		if (copy_from_user(fpu, argp, sizeof(struct kvm_fpu)))
A
Avi Kivity 已提交
1620
			goto out;
1621
		r = kvm_arch_vcpu_ioctl_set_fpu(vcpu, fpu);
A
Avi Kivity 已提交
1622 1623 1624 1625 1626
		if (r)
			goto out;
		r = 0;
		break;
	}
A
Avi Kivity 已提交
1627
	default:
1628
		r = kvm_arch_vcpu_ioctl(filp, ioctl, arg);
A
Avi Kivity 已提交
1629 1630
	}
out:
1631
	vcpu_put(vcpu);
1632 1633
	kfree(fpu);
	kfree(kvm_sregs);
A
Avi Kivity 已提交
1634 1635 1636 1637 1638 1639 1640 1641
	return r;
}

static long kvm_vm_ioctl(struct file *filp,
			   unsigned int ioctl, unsigned long arg)
{
	struct kvm *kvm = filp->private_data;
	void __user *argp = (void __user *)arg;
1642
	int r;
A
Avi Kivity 已提交
1643

1644 1645
	if (kvm->mm != current->mm)
		return -EIO;
A
Avi Kivity 已提交
1646 1647 1648 1649 1650 1651
	switch (ioctl) {
	case KVM_CREATE_VCPU:
		r = kvm_vm_ioctl_create_vcpu(kvm, arg);
		if (r < 0)
			goto out;
		break;
1652 1653 1654 1655 1656 1657 1658 1659 1660
	case KVM_SET_USER_MEMORY_REGION: {
		struct kvm_userspace_memory_region kvm_userspace_mem;

		r = -EFAULT;
		if (copy_from_user(&kvm_userspace_mem, argp,
						sizeof kvm_userspace_mem))
			goto out;

		r = kvm_vm_ioctl_set_memory_region(kvm, &kvm_userspace_mem, 1);
A
Avi Kivity 已提交
1661 1662 1663 1664 1665 1666 1667 1668
		if (r)
			goto out;
		break;
	}
	case KVM_GET_DIRTY_LOG: {
		struct kvm_dirty_log log;

		r = -EFAULT;
A
Al Viro 已提交
1669
		if (copy_from_user(&log, argp, sizeof log))
A
Avi Kivity 已提交
1670
			goto out;
1671
		r = kvm_vm_ioctl_get_dirty_log(kvm, &log);
A
Avi Kivity 已提交
1672 1673 1674 1675
		if (r)
			goto out;
		break;
	}
1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699
#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
	case KVM_REGISTER_COALESCED_MMIO: {
		struct kvm_coalesced_mmio_zone zone;
		r = -EFAULT;
		if (copy_from_user(&zone, argp, sizeof zone))
			goto out;
		r = kvm_vm_ioctl_register_coalesced_mmio(kvm, &zone);
		if (r)
			goto out;
		r = 0;
		break;
	}
	case KVM_UNREGISTER_COALESCED_MMIO: {
		struct kvm_coalesced_mmio_zone zone;
		r = -EFAULT;
		if (copy_from_user(&zone, argp, sizeof zone))
			goto out;
		r = kvm_vm_ioctl_unregister_coalesced_mmio(kvm, &zone);
		if (r)
			goto out;
		r = 0;
		break;
	}
#endif
G
Gregory Haskins 已提交
1700 1701 1702 1703 1704 1705 1706 1707 1708
	case KVM_IRQFD: {
		struct kvm_irqfd data;

		r = -EFAULT;
		if (copy_from_user(&data, argp, sizeof data))
			goto out;
		r = kvm_irqfd(kvm, data.fd, data.gsi, data.flags);
		break;
	}
G
Gregory Haskins 已提交
1709 1710 1711 1712 1713 1714 1715 1716 1717
	case KVM_IOEVENTFD: {
		struct kvm_ioeventfd data;

		r = -EFAULT;
		if (copy_from_user(&data, argp, sizeof data))
			goto out;
		r = kvm_ioeventfd(kvm, &data);
		break;
	}
1718 1719 1720
#ifdef CONFIG_KVM_APIC_ARCHITECTURE
	case KVM_SET_BOOT_CPU_ID:
		r = 0;
1721
		mutex_lock(&kvm->lock);
1722 1723 1724 1725
		if (atomic_read(&kvm->online_vcpus) != 0)
			r = -EBUSY;
		else
			kvm->bsp_vcpu_id = arg;
1726
		mutex_unlock(&kvm->lock);
1727 1728
		break;
#endif
1729
	default:
1730
		r = kvm_arch_vm_ioctl(filp, ioctl, arg);
1731 1732
		if (r == -ENOTTY)
			r = kvm_vm_ioctl_assigned_device(kvm, ioctl, arg);
1733 1734 1735 1736 1737
	}
out:
	return r;
}

1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783
#ifdef CONFIG_COMPAT
struct compat_kvm_dirty_log {
	__u32 slot;
	__u32 padding1;
	union {
		compat_uptr_t dirty_bitmap; /* one bit per page */
		__u64 padding2;
	};
};

static long kvm_vm_compat_ioctl(struct file *filp,
			   unsigned int ioctl, unsigned long arg)
{
	struct kvm *kvm = filp->private_data;
	int r;

	if (kvm->mm != current->mm)
		return -EIO;
	switch (ioctl) {
	case KVM_GET_DIRTY_LOG: {
		struct compat_kvm_dirty_log compat_log;
		struct kvm_dirty_log log;

		r = -EFAULT;
		if (copy_from_user(&compat_log, (void __user *)arg,
				   sizeof(compat_log)))
			goto out;
		log.slot	 = compat_log.slot;
		log.padding1	 = compat_log.padding1;
		log.padding2	 = compat_log.padding2;
		log.dirty_bitmap = compat_ptr(compat_log.dirty_bitmap);

		r = kvm_vm_ioctl_get_dirty_log(kvm, &log);
		if (r)
			goto out;
		break;
	}
	default:
		r = kvm_vm_ioctl(filp, ioctl, arg);
	}

out:
	return r;
}
#endif

1784
static int kvm_vm_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
1785
{
1786 1787 1788 1789
	struct page *page[1];
	unsigned long addr;
	int npages;
	gfn_t gfn = vmf->pgoff;
1790 1791
	struct kvm *kvm = vma->vm_file->private_data;

1792 1793
	addr = gfn_to_hva(kvm, gfn);
	if (kvm_is_error_hva(addr))
1794
		return VM_FAULT_SIGBUS;
1795 1796 1797 1798

	npages = get_user_pages(current, current->mm, addr, 1, 1, 0, page,
				NULL);
	if (unlikely(npages != 1))
1799
		return VM_FAULT_SIGBUS;
1800 1801

	vmf->page = page[0];
1802
	return 0;
1803 1804
}

1805
static const struct vm_operations_struct kvm_vm_vm_ops = {
1806
	.fault = kvm_vm_fault,
1807 1808 1809 1810 1811 1812 1813 1814
};

static int kvm_vm_mmap(struct file *file, struct vm_area_struct *vma)
{
	vma->vm_ops = &kvm_vm_vm_ops;
	return 0;
}

1815
static struct file_operations kvm_vm_fops = {
1816 1817
	.release        = kvm_vm_release,
	.unlocked_ioctl = kvm_vm_ioctl,
1818 1819 1820
#ifdef CONFIG_COMPAT
	.compat_ioctl   = kvm_vm_compat_ioctl,
#endif
1821
	.mmap           = kvm_vm_mmap,
1822
	.llseek		= noop_llseek,
1823 1824 1825 1826
};

static int kvm_dev_ioctl_create_vm(void)
{
1827
	int fd, r;
1828 1829 1830
	struct kvm *kvm;

	kvm = kvm_create_vm();
1831 1832
	if (IS_ERR(kvm))
		return PTR_ERR(kvm);
1833 1834 1835 1836 1837 1838 1839
#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
	r = kvm_coalesced_mmio_init(kvm);
	if (r < 0) {
		kvm_put_kvm(kvm);
		return r;
	}
#endif
1840
	fd = anon_inode_getfd("kvm-vm", &kvm_vm_fops, kvm, O_RDWR);
A
Al Viro 已提交
1841
	if (fd < 0)
A
Al Viro 已提交
1842
		kvm_put_kvm(kvm);
1843 1844 1845 1846

	return fd;
}

1847 1848 1849
static long kvm_dev_ioctl_check_extension_generic(long arg)
{
	switch (arg) {
1850
	case KVM_CAP_USER_MEMORY:
1851
	case KVM_CAP_DESTROY_MEMORY_REGION_WORKS:
1852
	case KVM_CAP_JOIN_MEMORY_REGIONS_WORKS:
1853 1854 1855
#ifdef CONFIG_KVM_APIC_ARCHITECTURE
	case KVM_CAP_SET_BOOT_CPU_ID:
#endif
1856
	case KVM_CAP_INTERNAL_ERROR_DATA:
1857
		return 1;
1858 1859
#ifdef CONFIG_HAVE_KVM_IRQCHIP
	case KVM_CAP_IRQ_ROUTING:
1860
		return KVM_MAX_IRQ_ROUTES;
1861
#endif
1862 1863 1864 1865 1866 1867
	default:
		break;
	}
	return kvm_dev_ioctl_check_extension(arg);
}

1868 1869 1870
static long kvm_dev_ioctl(struct file *filp,
			  unsigned int ioctl, unsigned long arg)
{
1871
	long r = -EINVAL;
1872 1873 1874

	switch (ioctl) {
	case KVM_GET_API_VERSION:
1875 1876 1877
		r = -EINVAL;
		if (arg)
			goto out;
1878 1879 1880
		r = KVM_API_VERSION;
		break;
	case KVM_CREATE_VM:
1881 1882 1883
		r = -EINVAL;
		if (arg)
			goto out;
1884 1885
		r = kvm_dev_ioctl_create_vm();
		break;
1886
	case KVM_CHECK_EXTENSION:
1887
		r = kvm_dev_ioctl_check_extension_generic(arg);
1888
		break;
1889 1890 1891 1892
	case KVM_GET_VCPU_MMAP_SIZE:
		r = -EINVAL;
		if (arg)
			goto out;
1893 1894 1895
		r = PAGE_SIZE;     /* struct kvm_run */
#ifdef CONFIG_X86
		r += PAGE_SIZE;    /* pio data page */
1896 1897 1898
#endif
#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
		r += PAGE_SIZE;    /* coalesced mmio ring page */
1899
#endif
1900
		break;
1901 1902 1903
	case KVM_TRACE_ENABLE:
	case KVM_TRACE_PAUSE:
	case KVM_TRACE_DISABLE:
1904
		r = -EOPNOTSUPP;
1905
		break;
A
Avi Kivity 已提交
1906
	default:
1907
		return kvm_arch_dev_ioctl(filp, ioctl, arg);
A
Avi Kivity 已提交
1908 1909 1910 1911 1912 1913 1914 1915
	}
out:
	return r;
}

static struct file_operations kvm_chardev_ops = {
	.unlocked_ioctl = kvm_dev_ioctl,
	.compat_ioctl   = kvm_dev_ioctl,
1916
	.llseek		= noop_llseek,
A
Avi Kivity 已提交
1917 1918 1919
};

static struct miscdevice kvm_dev = {
A
Avi Kivity 已提交
1920
	KVM_MINOR,
A
Avi Kivity 已提交
1921 1922 1923 1924
	"kvm",
	&kvm_chardev_ops,
};

1925 1926 1927
static void hardware_enable(void *junk)
{
	int cpu = raw_smp_processor_id();
1928
	int r;
1929

1930
	if (cpumask_test_cpu(cpu, cpus_hardware_enabled))
1931
		return;
1932

1933
	cpumask_set_cpu(cpu, cpus_hardware_enabled);
1934 1935 1936 1937 1938 1939 1940 1941 1942

	r = kvm_arch_hardware_enable(NULL);

	if (r) {
		cpumask_clear_cpu(cpu, cpus_hardware_enabled);
		atomic_inc(&hardware_enable_failed);
		printk(KERN_INFO "kvm: enabling virtualization on "
				 "CPU%d failed\n", cpu);
	}
1943 1944 1945 1946 1947 1948
}

static void hardware_disable(void *junk)
{
	int cpu = raw_smp_processor_id();

1949
	if (!cpumask_test_cpu(cpu, cpus_hardware_enabled))
1950
		return;
1951
	cpumask_clear_cpu(cpu, cpus_hardware_enabled);
1952
	kvm_arch_hardware_disable(NULL);
1953 1954
}

1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992
static void hardware_disable_all_nolock(void)
{
	BUG_ON(!kvm_usage_count);

	kvm_usage_count--;
	if (!kvm_usage_count)
		on_each_cpu(hardware_disable, NULL, 1);
}

static void hardware_disable_all(void)
{
	spin_lock(&kvm_lock);
	hardware_disable_all_nolock();
	spin_unlock(&kvm_lock);
}

static int hardware_enable_all(void)
{
	int r = 0;

	spin_lock(&kvm_lock);

	kvm_usage_count++;
	if (kvm_usage_count == 1) {
		atomic_set(&hardware_enable_failed, 0);
		on_each_cpu(hardware_enable, NULL, 1);

		if (atomic_read(&hardware_enable_failed)) {
			hardware_disable_all_nolock();
			r = -EBUSY;
		}
	}

	spin_unlock(&kvm_lock);

	return r;
}

A
Avi Kivity 已提交
1993 1994 1995 1996 1997
static int kvm_cpu_hotplug(struct notifier_block *notifier, unsigned long val,
			   void *v)
{
	int cpu = (long)v;

1998 1999 2000
	if (!kvm_usage_count)
		return NOTIFY_OK;

2001
	val &= ~CPU_TASKS_FROZEN;
A
Avi Kivity 已提交
2002
	switch (val) {
2003
	case CPU_DYING:
2004 2005 2006 2007
		printk(KERN_INFO "kvm: disabling virtualization on CPU%d\n",
		       cpu);
		hardware_disable(NULL);
		break;
2008
	case CPU_STARTING:
2009 2010
		printk(KERN_INFO "kvm: enabling virtualization on CPU%d\n",
		       cpu);
2011
		spin_lock(&kvm_lock);
2012
		hardware_enable(NULL);
2013
		spin_unlock(&kvm_lock);
A
Avi Kivity 已提交
2014 2015 2016 2017 2018
		break;
	}
	return NOTIFY_OK;
}

2019 2020 2021

asmlinkage void kvm_handle_fault_on_reboot(void)
{
A
Avi Kivity 已提交
2022
	if (kvm_rebooting) {
2023
		/* spin while reset goes on */
A
Avi Kivity 已提交
2024
		local_irq_enable();
2025
		while (true)
2026
			cpu_relax();
A
Avi Kivity 已提交
2027
	}
2028 2029 2030 2031 2032
	/* Fault while not rebooting.  We want the trace. */
	BUG();
}
EXPORT_SYMBOL_GPL(kvm_handle_fault_on_reboot);

2033
static int kvm_reboot(struct notifier_block *notifier, unsigned long val,
M
Mike Day 已提交
2034
		      void *v)
2035
{
2036 2037 2038 2039 2040 2041 2042 2043 2044
	/*
	 * Some (well, at least mine) BIOSes hang on reboot if
	 * in vmx root mode.
	 *
	 * And Intel TXT required VMX off for all cpu when system shutdown.
	 */
	printk(KERN_INFO "kvm: exiting hardware virtualization\n");
	kvm_rebooting = true;
	on_each_cpu(hardware_disable, NULL, 1);
2045 2046 2047 2048 2049 2050 2051 2052
	return NOTIFY_OK;
}

static struct notifier_block kvm_reboot_notifier = {
	.notifier_call = kvm_reboot,
	.priority = 0,
};

M
Marcelo Tosatti 已提交
2053
static void kvm_io_bus_destroy(struct kvm_io_bus *bus)
2054 2055 2056 2057 2058 2059 2060 2061
{
	int i;

	for (i = 0; i < bus->dev_count; i++) {
		struct kvm_io_device *pos = bus->devs[i];

		kvm_iodevice_destructor(pos);
	}
M
Marcelo Tosatti 已提交
2062
	kfree(bus);
2063 2064
}

2065
/* kvm_io_bus_write - called under kvm->slots_lock */
M
Marcelo Tosatti 已提交
2066
int kvm_io_bus_write(struct kvm *kvm, enum kvm_bus bus_idx, gpa_t addr,
2067
		     int len, const void *val)
2068 2069
{
	int i;
2070 2071 2072
	struct kvm_io_bus *bus;

	bus = srcu_dereference(kvm->buses[bus_idx], &kvm->srcu);
2073 2074 2075 2076 2077
	for (i = 0; i < bus->dev_count; i++)
		if (!kvm_iodevice_write(bus->devs[i], addr, len, val))
			return 0;
	return -EOPNOTSUPP;
}
2078

2079
/* kvm_io_bus_read - called under kvm->slots_lock */
M
Marcelo Tosatti 已提交
2080 2081
int kvm_io_bus_read(struct kvm *kvm, enum kvm_bus bus_idx, gpa_t addr,
		    int len, void *val)
2082 2083
{
	int i;
2084
	struct kvm_io_bus *bus;
M
Marcelo Tosatti 已提交
2085

2086
	bus = srcu_dereference(kvm->buses[bus_idx], &kvm->srcu);
2087 2088 2089 2090
	for (i = 0; i < bus->dev_count; i++)
		if (!kvm_iodevice_read(bus->devs[i], addr, len, val))
			return 0;
	return -EOPNOTSUPP;
2091 2092
}

2093
/* Caller must hold slots_lock. */
M
Marcelo Tosatti 已提交
2094 2095
int kvm_io_bus_register_dev(struct kvm *kvm, enum kvm_bus bus_idx,
			    struct kvm_io_device *dev)
2096
{
M
Marcelo Tosatti 已提交
2097
	struct kvm_io_bus *new_bus, *bus;
2098

M
Marcelo Tosatti 已提交
2099
	bus = kvm->buses[bus_idx];
2100 2101
	if (bus->dev_count > NR_IOBUS_DEVS-1)
		return -ENOSPC;
2102

M
Marcelo Tosatti 已提交
2103 2104 2105 2106 2107 2108 2109 2110
	new_bus = kzalloc(sizeof(struct kvm_io_bus), GFP_KERNEL);
	if (!new_bus)
		return -ENOMEM;
	memcpy(new_bus, bus, sizeof(struct kvm_io_bus));
	new_bus->devs[new_bus->dev_count++] = dev;
	rcu_assign_pointer(kvm->buses[bus_idx], new_bus);
	synchronize_srcu_expedited(&kvm->srcu);
	kfree(bus);
2111 2112 2113 2114

	return 0;
}

2115
/* Caller must hold slots_lock. */
M
Marcelo Tosatti 已提交
2116 2117
int kvm_io_bus_unregister_dev(struct kvm *kvm, enum kvm_bus bus_idx,
			      struct kvm_io_device *dev)
2118
{
M
Marcelo Tosatti 已提交
2119 2120
	int i, r;
	struct kvm_io_bus *new_bus, *bus;
2121

M
Marcelo Tosatti 已提交
2122 2123 2124
	new_bus = kzalloc(sizeof(struct kvm_io_bus), GFP_KERNEL);
	if (!new_bus)
		return -ENOMEM;
2125

M
Marcelo Tosatti 已提交
2126 2127 2128 2129 2130 2131 2132 2133
	bus = kvm->buses[bus_idx];
	memcpy(new_bus, bus, sizeof(struct kvm_io_bus));

	r = -ENOENT;
	for (i = 0; i < new_bus->dev_count; i++)
		if (new_bus->devs[i] == dev) {
			r = 0;
			new_bus->devs[i] = new_bus->devs[--new_bus->dev_count];
2134 2135
			break;
		}
M
Marcelo Tosatti 已提交
2136 2137 2138 2139 2140 2141 2142 2143 2144 2145

	if (r) {
		kfree(new_bus);
		return r;
	}

	rcu_assign_pointer(kvm->buses[bus_idx], new_bus);
	synchronize_srcu_expedited(&kvm->srcu);
	kfree(bus);
	return r;
2146 2147
}

A
Avi Kivity 已提交
2148 2149 2150 2151
static struct notifier_block kvm_cpu_notifier = {
	.notifier_call = kvm_cpu_hotplug,
};

2152
static int vm_stat_get(void *_offset, u64 *val)
2153 2154 2155 2156
{
	unsigned offset = (long)_offset;
	struct kvm *kvm;

2157
	*val = 0;
2158 2159
	spin_lock(&kvm_lock);
	list_for_each_entry(kvm, &vm_list, vm_list)
2160
		*val += *(u32 *)((void *)kvm + offset);
2161
	spin_unlock(&kvm_lock);
2162
	return 0;
2163 2164 2165 2166
}

DEFINE_SIMPLE_ATTRIBUTE(vm_stat_fops, vm_stat_get, NULL, "%llu\n");

2167
static int vcpu_stat_get(void *_offset, u64 *val)
A
Avi Kivity 已提交
2168 2169 2170 2171 2172 2173
{
	unsigned offset = (long)_offset;
	struct kvm *kvm;
	struct kvm_vcpu *vcpu;
	int i;

2174
	*val = 0;
A
Avi Kivity 已提交
2175 2176
	spin_lock(&kvm_lock);
	list_for_each_entry(kvm, &vm_list, vm_list)
2177 2178 2179
		kvm_for_each_vcpu(i, vcpu, kvm)
			*val += *(u32 *)((void *)vcpu + offset);

A
Avi Kivity 已提交
2180
	spin_unlock(&kvm_lock);
2181
	return 0;
A
Avi Kivity 已提交
2182 2183
}

2184 2185
DEFINE_SIMPLE_ATTRIBUTE(vcpu_stat_fops, vcpu_stat_get, NULL, "%llu\n");

2186
static const struct file_operations *stat_fops[] = {
2187 2188 2189
	[KVM_STAT_VCPU] = &vcpu_stat_fops,
	[KVM_STAT_VM]   = &vm_stat_fops,
};
A
Avi Kivity 已提交
2190

2191
static void kvm_init_debug(void)
A
Avi Kivity 已提交
2192 2193 2194
{
	struct kvm_stats_debugfs_item *p;

2195
	kvm_debugfs_dir = debugfs_create_dir("kvm", NULL);
A
Avi Kivity 已提交
2196
	for (p = debugfs_entries; p->name; ++p)
2197
		p->dentry = debugfs_create_file(p->name, 0444, kvm_debugfs_dir,
A
Avi Kivity 已提交
2198
						(void *)(long)p->offset,
2199
						stat_fops[p->kind]);
A
Avi Kivity 已提交
2200 2201 2202 2203 2204 2205 2206 2207
}

static void kvm_exit_debug(void)
{
	struct kvm_stats_debugfs_item *p;

	for (p = debugfs_entries; p->name; ++p)
		debugfs_remove(p->dentry);
2208
	debugfs_remove(kvm_debugfs_dir);
A
Avi Kivity 已提交
2209 2210
}

2211 2212
static int kvm_suspend(struct sys_device *dev, pm_message_t state)
{
2213 2214
	if (kvm_usage_count)
		hardware_disable(NULL);
2215 2216 2217 2218 2219
	return 0;
}

static int kvm_resume(struct sys_device *dev)
{
2220 2221
	if (kvm_usage_count) {
		WARN_ON(spin_is_locked(&kvm_lock));
2222
		hardware_enable(NULL);
2223
	}
2224 2225 2226 2227
	return 0;
}

static struct sysdev_class kvm_sysdev_class = {
2228
	.name = "kvm",
2229 2230 2231 2232 2233 2234 2235 2236 2237
	.suspend = kvm_suspend,
	.resume = kvm_resume,
};

static struct sys_device kvm_sysdev = {
	.id = 0,
	.cls = &kvm_sysdev_class,
};

2238
struct page *bad_page;
2239
pfn_t bad_pfn;
A
Avi Kivity 已提交
2240

2241 2242 2243 2244 2245 2246 2247 2248 2249 2250
static inline
struct kvm_vcpu *preempt_notifier_to_vcpu(struct preempt_notifier *pn)
{
	return container_of(pn, struct kvm_vcpu, preempt_notifier);
}

static void kvm_sched_in(struct preempt_notifier *pn, int cpu)
{
	struct kvm_vcpu *vcpu = preempt_notifier_to_vcpu(pn);

2251
	kvm_arch_vcpu_load(vcpu, cpu);
2252 2253 2254 2255 2256 2257 2258
}

static void kvm_sched_out(struct preempt_notifier *pn,
			  struct task_struct *next)
{
	struct kvm_vcpu *vcpu = preempt_notifier_to_vcpu(pn);

2259
	kvm_arch_vcpu_put(vcpu);
2260 2261
}

2262
int kvm_init(void *opaque, unsigned vcpu_size, unsigned vcpu_align,
2263
		  struct module *module)
A
Avi Kivity 已提交
2264 2265
{
	int r;
Y
Yang, Sheng 已提交
2266
	int cpu;
A
Avi Kivity 已提交
2267

2268 2269
	r = kvm_arch_init(opaque);
	if (r)
2270
		goto out_fail;
2271 2272 2273 2274 2275 2276 2277 2278

	bad_page = alloc_page(GFP_KERNEL | __GFP_ZERO);

	if (bad_page == NULL) {
		r = -ENOMEM;
		goto out;
	}

2279 2280
	bad_pfn = page_to_pfn(bad_page);

2281 2282 2283 2284 2285 2286 2287 2288 2289
	hwpoison_page = alloc_page(GFP_KERNEL | __GFP_ZERO);

	if (hwpoison_page == NULL) {
		r = -ENOMEM;
		goto out_free_0;
	}

	hwpoison_pfn = page_to_pfn(hwpoison_page);

2290 2291 2292 2293 2294 2295 2296 2297 2298
	fault_page = alloc_page(GFP_KERNEL | __GFP_ZERO);

	if (fault_page == NULL) {
		r = -ENOMEM;
		goto out_free_0;
	}

	fault_pfn = page_to_pfn(fault_page);

2299
	if (!zalloc_cpumask_var(&cpus_hardware_enabled, GFP_KERNEL)) {
2300 2301 2302 2303
		r = -ENOMEM;
		goto out_free_0;
	}

2304
	r = kvm_arch_hardware_setup();
A
Avi Kivity 已提交
2305
	if (r < 0)
2306
		goto out_free_0a;
A
Avi Kivity 已提交
2307

Y
Yang, Sheng 已提交
2308 2309
	for_each_online_cpu(cpu) {
		smp_call_function_single(cpu,
2310
				kvm_arch_check_processor_compat,
2311
				&r, 1);
Y
Yang, Sheng 已提交
2312
		if (r < 0)
2313
			goto out_free_1;
Y
Yang, Sheng 已提交
2314 2315
	}

A
Avi Kivity 已提交
2316 2317
	r = register_cpu_notifier(&kvm_cpu_notifier);
	if (r)
2318
		goto out_free_2;
A
Avi Kivity 已提交
2319 2320
	register_reboot_notifier(&kvm_reboot_notifier);

2321 2322
	r = sysdev_class_register(&kvm_sysdev_class);
	if (r)
2323
		goto out_free_3;
2324 2325 2326

	r = sysdev_register(&kvm_sysdev);
	if (r)
2327
		goto out_free_4;
2328

2329
	/* A kmem cache lets us meet the alignment requirements of fx_save. */
2330 2331 2332
	if (!vcpu_align)
		vcpu_align = __alignof__(struct kvm_vcpu);
	kvm_vcpu_cache = kmem_cache_create("kvm_vcpu", vcpu_size, vcpu_align,
J
Joe Perches 已提交
2333
					   0, NULL);
2334 2335
	if (!kvm_vcpu_cache) {
		r = -ENOMEM;
2336
		goto out_free_5;
2337 2338
	}

A
Avi Kivity 已提交
2339
	kvm_chardev_ops.owner = module;
2340 2341
	kvm_vm_fops.owner = module;
	kvm_vcpu_fops.owner = module;
A
Avi Kivity 已提交
2342 2343 2344

	r = misc_register(&kvm_dev);
	if (r) {
M
Mike Day 已提交
2345
		printk(KERN_ERR "kvm: misc device register failed\n");
A
Avi Kivity 已提交
2346 2347 2348
		goto out_free;
	}

2349 2350 2351
	kvm_preempt_ops.sched_in = kvm_sched_in;
	kvm_preempt_ops.sched_out = kvm_sched_out;

2352 2353
	kvm_init_debug();

2354
	return 0;
A
Avi Kivity 已提交
2355 2356

out_free:
2357
	kmem_cache_destroy(kvm_vcpu_cache);
2358
out_free_5:
2359
	sysdev_unregister(&kvm_sysdev);
2360
out_free_4:
2361
	sysdev_class_unregister(&kvm_sysdev_class);
2362
out_free_3:
A
Avi Kivity 已提交
2363
	unregister_reboot_notifier(&kvm_reboot_notifier);
A
Avi Kivity 已提交
2364
	unregister_cpu_notifier(&kvm_cpu_notifier);
2365 2366
out_free_2:
out_free_1:
2367
	kvm_arch_hardware_unsetup();
2368 2369
out_free_0a:
	free_cpumask_var(cpus_hardware_enabled);
2370
out_free_0:
2371 2372
	if (fault_page)
		__free_page(fault_page);
2373 2374
	if (hwpoison_page)
		__free_page(hwpoison_page);
2375
	__free_page(bad_page);
2376
out:
2377
	kvm_arch_exit();
2378
out_fail:
A
Avi Kivity 已提交
2379 2380
	return r;
}
2381
EXPORT_SYMBOL_GPL(kvm_init);
A
Avi Kivity 已提交
2382

2383
void kvm_exit(void)
A
Avi Kivity 已提交
2384
{
2385
	kvm_exit_debug();
A
Avi Kivity 已提交
2386
	misc_deregister(&kvm_dev);
2387
	kmem_cache_destroy(kvm_vcpu_cache);
2388 2389
	sysdev_unregister(&kvm_sysdev);
	sysdev_class_unregister(&kvm_sysdev_class);
A
Avi Kivity 已提交
2390
	unregister_reboot_notifier(&kvm_reboot_notifier);
2391
	unregister_cpu_notifier(&kvm_cpu_notifier);
2392
	on_each_cpu(hardware_disable, NULL, 1);
2393
	kvm_arch_hardware_unsetup();
2394
	kvm_arch_exit();
2395
	free_cpumask_var(cpus_hardware_enabled);
2396
	__free_page(hwpoison_page);
2397
	__free_page(bad_page);
A
Avi Kivity 已提交
2398
}
2399
EXPORT_SYMBOL_GPL(kvm_exit);