KVM: ia64: Stub out kvmtrace
[GitHub/exynos8895/android_kernel_samsung_universal8895.git] / virt / kvm / kvm_main.c
CommitLineData
6aa8b732
AK
1/*
2 * Kernel-based Virtual Machine driver for Linux
3 *
4 * This module enables machines with Intel VT-x extensions to run virtual
5 * machines without emulation or binary translation.
6 *
7 * Copyright (C) 2006 Qumranet, Inc.
8 *
9 * Authors:
10 * Avi Kivity <avi@qumranet.com>
11 * Yaniv Kamay <yaniv@qumranet.com>
12 *
13 * This work is licensed under the terms of the GNU GPL, version 2. See
14 * the COPYING file in the top-level directory.
15 *
16 */
17
e2174021 18#include "iodev.h"
6aa8b732 19
edf88417 20#include <linux/kvm_host.h>
6aa8b732
AK
21#include <linux/kvm.h>
22#include <linux/module.h>
23#include <linux/errno.h>
6aa8b732
AK
24#include <linux/percpu.h>
25#include <linux/gfp.h>
6aa8b732
AK
26#include <linux/mm.h>
27#include <linux/miscdevice.h>
28#include <linux/vmalloc.h>
6aa8b732 29#include <linux/reboot.h>
6aa8b732
AK
30#include <linux/debugfs.h>
31#include <linux/highmem.h>
32#include <linux/file.h>
59ae6c6b 33#include <linux/sysdev.h>
774c47f1 34#include <linux/cpu.h>
e8edc6e0 35#include <linux/sched.h>
d9e368d6
AK
36#include <linux/cpumask.h>
37#include <linux/smp.h>
d6d28168 38#include <linux/anon_inodes.h>
04d2cc77 39#include <linux/profile.h>
7aa81cc0 40#include <linux/kvm_para.h>
6fc138d2 41#include <linux/pagemap.h>
8d4e1288 42#include <linux/mman.h>
35149e21 43#include <linux/swap.h>
6aa8b732 44
e495606d 45#include <asm/processor.h>
e495606d
AK
46#include <asm/io.h>
47#include <asm/uaccess.h>
3e021bf5 48#include <asm/pgtable.h>
6aa8b732
AK
49
50MODULE_AUTHOR("Qumranet");
51MODULE_LICENSE("GPL");
52
e9b11c17
ZX
53DEFINE_SPINLOCK(kvm_lock);
54LIST_HEAD(vm_list);
133de902 55
1b6c0168
AK
56static cpumask_t cpus_hardware_enabled;
57
c16f862d
RR
58struct kmem_cache *kvm_vcpu_cache;
59EXPORT_SYMBOL_GPL(kvm_vcpu_cache);
1165f5fe 60
15ad7146
AK
61static __read_mostly struct preempt_ops kvm_preempt_ops;
62
6aa8b732
AK
63static struct dentry *debugfs_dir;
64
bccf2150
AK
65static long kvm_vcpu_ioctl(struct file *file, unsigned int ioctl,
66 unsigned long arg);
67
5aacf0ca
JM
68static inline int valid_vcpu(int n)
69{
70 return likely(n >= 0 && n < KVM_MAX_VCPUS);
71}
72
bccf2150
AK
73/*
74 * Switches to specified vcpu, until a matching vcpu_put()
75 */
313a3dc7 76void vcpu_load(struct kvm_vcpu *vcpu)
6aa8b732 77{
15ad7146
AK
78 int cpu;
79
bccf2150 80 mutex_lock(&vcpu->mutex);
15ad7146
AK
81 cpu = get_cpu();
82 preempt_notifier_register(&vcpu->preempt_notifier);
313a3dc7 83 kvm_arch_vcpu_load(vcpu, cpu);
15ad7146 84 put_cpu();
6aa8b732
AK
85}
86
313a3dc7 87void vcpu_put(struct kvm_vcpu *vcpu)
6aa8b732 88{
15ad7146 89 preempt_disable();
313a3dc7 90 kvm_arch_vcpu_put(vcpu);
15ad7146
AK
91 preempt_notifier_unregister(&vcpu->preempt_notifier);
92 preempt_enable();
6aa8b732
AK
93 mutex_unlock(&vcpu->mutex);
94}
95
d9e368d6
AK
96static void ack_flush(void *_completed)
97{
d9e368d6
AK
98}
99
100void kvm_flush_remote_tlbs(struct kvm *kvm)
101{
49d3bd7e 102 int i, cpu;
d9e368d6
AK
103 cpumask_t cpus;
104 struct kvm_vcpu *vcpu;
d9e368d6 105
d9e368d6 106 cpus_clear(cpus);
fb3f0f51
RR
107 for (i = 0; i < KVM_MAX_VCPUS; ++i) {
108 vcpu = kvm->vcpus[i];
109 if (!vcpu)
110 continue;
3176bc3e 111 if (test_and_set_bit(KVM_REQ_TLB_FLUSH, &vcpu->requests))
d9e368d6
AK
112 continue;
113 cpu = vcpu->cpu;
114 if (cpu != -1 && cpu != raw_smp_processor_id())
49d3bd7e 115 cpu_set(cpu, cpus);
d9e368d6 116 }
0f74a24c
AK
117 if (cpus_empty(cpus))
118 return;
119 ++kvm->stat.remote_tlb_flush;
49d3bd7e 120 smp_call_function_mask(cpus, ack_flush, NULL, 1);
d9e368d6
AK
121}
122
2e53d63a
MT
123void kvm_reload_remote_mmus(struct kvm *kvm)
124{
125 int i, cpu;
126 cpumask_t cpus;
127 struct kvm_vcpu *vcpu;
128
129 cpus_clear(cpus);
130 for (i = 0; i < KVM_MAX_VCPUS; ++i) {
131 vcpu = kvm->vcpus[i];
132 if (!vcpu)
133 continue;
134 if (test_and_set_bit(KVM_REQ_MMU_RELOAD, &vcpu->requests))
135 continue;
136 cpu = vcpu->cpu;
137 if (cpu != -1 && cpu != raw_smp_processor_id())
138 cpu_set(cpu, cpus);
139 }
140 if (cpus_empty(cpus))
141 return;
142 smp_call_function_mask(cpus, ack_flush, NULL, 1);
143}
144
145
fb3f0f51
RR
146int kvm_vcpu_init(struct kvm_vcpu *vcpu, struct kvm *kvm, unsigned id)
147{
148 struct page *page;
149 int r;
150
151 mutex_init(&vcpu->mutex);
152 vcpu->cpu = -1;
fb3f0f51
RR
153 vcpu->kvm = kvm;
154 vcpu->vcpu_id = id;
b6958ce4 155 init_waitqueue_head(&vcpu->wq);
fb3f0f51
RR
156
157 page = alloc_page(GFP_KERNEL | __GFP_ZERO);
158 if (!page) {
159 r = -ENOMEM;
160 goto fail;
161 }
162 vcpu->run = page_address(page);
163
e9b11c17 164 r = kvm_arch_vcpu_init(vcpu);
fb3f0f51 165 if (r < 0)
e9b11c17 166 goto fail_free_run;
fb3f0f51
RR
167 return 0;
168
fb3f0f51
RR
169fail_free_run:
170 free_page((unsigned long)vcpu->run);
171fail:
76fafa5e 172 return r;
fb3f0f51
RR
173}
174EXPORT_SYMBOL_GPL(kvm_vcpu_init);
175
176void kvm_vcpu_uninit(struct kvm_vcpu *vcpu)
177{
e9b11c17 178 kvm_arch_vcpu_uninit(vcpu);
fb3f0f51
RR
179 free_page((unsigned long)vcpu->run);
180}
181EXPORT_SYMBOL_GPL(kvm_vcpu_uninit);
182
f17abe9a 183static struct kvm *kvm_create_vm(void)
6aa8b732 184{
d19a9cd2 185 struct kvm *kvm = kvm_arch_create_vm();
6aa8b732 186
d19a9cd2
ZX
187 if (IS_ERR(kvm))
188 goto out;
6aa8b732 189
6d4e4c4f
AK
190 kvm->mm = current->mm;
191 atomic_inc(&kvm->mm->mm_count);
aaee2c94 192 spin_lock_init(&kvm->mmu_lock);
74906345 193 kvm_io_bus_init(&kvm->pio_bus);
11ec2804 194 mutex_init(&kvm->lock);
2eeb2e94 195 kvm_io_bus_init(&kvm->mmio_bus);
72dc67a6 196 init_rwsem(&kvm->slots_lock);
d39f13b0 197 atomic_set(&kvm->users_count, 1);
5e58cfe4
RR
198 spin_lock(&kvm_lock);
199 list_add(&kvm->vm_list, &vm_list);
200 spin_unlock(&kvm_lock);
d19a9cd2 201out:
f17abe9a
AK
202 return kvm;
203}
204
6aa8b732
AK
205/*
206 * Free any memory in @free but not in @dont.
207 */
208static void kvm_free_physmem_slot(struct kvm_memory_slot *free,
209 struct kvm_memory_slot *dont)
210{
290fc38d
IE
211 if (!dont || free->rmap != dont->rmap)
212 vfree(free->rmap);
6aa8b732
AK
213
214 if (!dont || free->dirty_bitmap != dont->dirty_bitmap)
215 vfree(free->dirty_bitmap);
216
05da4558
MT
217 if (!dont || free->lpage_info != dont->lpage_info)
218 vfree(free->lpage_info);
219
6aa8b732 220 free->npages = 0;
8b6d44c7 221 free->dirty_bitmap = NULL;
8d4e1288 222 free->rmap = NULL;
05da4558 223 free->lpage_info = NULL;
6aa8b732
AK
224}
225
d19a9cd2 226void kvm_free_physmem(struct kvm *kvm)
6aa8b732
AK
227{
228 int i;
229
230 for (i = 0; i < kvm->nmemslots; ++i)
8b6d44c7 231 kvm_free_physmem_slot(&kvm->memslots[i], NULL);
6aa8b732
AK
232}
233
f17abe9a
AK
234static void kvm_destroy_vm(struct kvm *kvm)
235{
6d4e4c4f
AK
236 struct mm_struct *mm = kvm->mm;
237
133de902
AK
238 spin_lock(&kvm_lock);
239 list_del(&kvm->vm_list);
240 spin_unlock(&kvm_lock);
74906345 241 kvm_io_bus_destroy(&kvm->pio_bus);
2eeb2e94 242 kvm_io_bus_destroy(&kvm->mmio_bus);
d19a9cd2 243 kvm_arch_destroy_vm(kvm);
6d4e4c4f 244 mmdrop(mm);
f17abe9a
AK
245}
246
d39f13b0
IE
247void kvm_get_kvm(struct kvm *kvm)
248{
249 atomic_inc(&kvm->users_count);
250}
251EXPORT_SYMBOL_GPL(kvm_get_kvm);
252
253void kvm_put_kvm(struct kvm *kvm)
254{
255 if (atomic_dec_and_test(&kvm->users_count))
256 kvm_destroy_vm(kvm);
257}
258EXPORT_SYMBOL_GPL(kvm_put_kvm);
259
260
f17abe9a
AK
261static int kvm_vm_release(struct inode *inode, struct file *filp)
262{
263 struct kvm *kvm = filp->private_data;
264
d39f13b0 265 kvm_put_kvm(kvm);
6aa8b732
AK
266 return 0;
267}
268
6aa8b732
AK
269/*
270 * Allocate some memory and give it an address in the guest physical address
271 * space.
272 *
273 * Discontiguous memory is allowed, mostly for framebuffers.
f78e0e2e 274 *
10589a46 275 * Must be called holding mmap_sem for write.
6aa8b732 276 */
f78e0e2e
SY
277int __kvm_set_memory_region(struct kvm *kvm,
278 struct kvm_userspace_memory_region *mem,
279 int user_alloc)
6aa8b732
AK
280{
281 int r;
282 gfn_t base_gfn;
283 unsigned long npages;
284 unsigned long i;
285 struct kvm_memory_slot *memslot;
286 struct kvm_memory_slot old, new;
6aa8b732
AK
287
288 r = -EINVAL;
289 /* General sanity checks */
290 if (mem->memory_size & (PAGE_SIZE - 1))
291 goto out;
292 if (mem->guest_phys_addr & (PAGE_SIZE - 1))
293 goto out;
e0d62c7f 294 if (mem->slot >= KVM_MEMORY_SLOTS + KVM_PRIVATE_MEM_SLOTS)
6aa8b732
AK
295 goto out;
296 if (mem->guest_phys_addr + mem->memory_size < mem->guest_phys_addr)
297 goto out;
298
299 memslot = &kvm->memslots[mem->slot];
300 base_gfn = mem->guest_phys_addr >> PAGE_SHIFT;
301 npages = mem->memory_size >> PAGE_SHIFT;
302
303 if (!npages)
304 mem->flags &= ~KVM_MEM_LOG_DIRTY_PAGES;
305
6aa8b732
AK
306 new = old = *memslot;
307
308 new.base_gfn = base_gfn;
309 new.npages = npages;
310 new.flags = mem->flags;
311
312 /* Disallow changing a memory slot's size. */
313 r = -EINVAL;
314 if (npages && old.npages && npages != old.npages)
f78e0e2e 315 goto out_free;
6aa8b732
AK
316
317 /* Check for overlaps */
318 r = -EEXIST;
319 for (i = 0; i < KVM_MEMORY_SLOTS; ++i) {
320 struct kvm_memory_slot *s = &kvm->memslots[i];
321
322 if (s == memslot)
323 continue;
324 if (!((base_gfn + npages <= s->base_gfn) ||
325 (base_gfn >= s->base_gfn + s->npages)))
f78e0e2e 326 goto out_free;
6aa8b732 327 }
6aa8b732 328
6aa8b732
AK
329 /* Free page dirty bitmap if unneeded */
330 if (!(new.flags & KVM_MEM_LOG_DIRTY_PAGES))
8b6d44c7 331 new.dirty_bitmap = NULL;
6aa8b732
AK
332
333 r = -ENOMEM;
334
335 /* Allocate if a slot is being created */
8d4e1288 336 if (npages && !new.rmap) {
d77c26fc 337 new.rmap = vmalloc(npages * sizeof(struct page *));
290fc38d
IE
338
339 if (!new.rmap)
f78e0e2e 340 goto out_free;
290fc38d 341
290fc38d 342 memset(new.rmap, 0, npages * sizeof(*new.rmap));
8d4e1288 343
80b14b5b 344 new.user_alloc = user_alloc;
0de10343 345 new.userspace_addr = mem->userspace_addr;
6aa8b732 346 }
05da4558
MT
347 if (npages && !new.lpage_info) {
348 int largepages = npages / KVM_PAGES_PER_HPAGE;
349 if (npages % KVM_PAGES_PER_HPAGE)
350 largepages++;
351 if (base_gfn % KVM_PAGES_PER_HPAGE)
352 largepages++;
353
354 new.lpage_info = vmalloc(largepages * sizeof(*new.lpage_info));
355
356 if (!new.lpage_info)
357 goto out_free;
358
359 memset(new.lpage_info, 0, largepages * sizeof(*new.lpage_info));
360
361 if (base_gfn % KVM_PAGES_PER_HPAGE)
362 new.lpage_info[0].write_count = 1;
363 if ((base_gfn+npages) % KVM_PAGES_PER_HPAGE)
364 new.lpage_info[largepages-1].write_count = 1;
365 }
6aa8b732
AK
366
367 /* Allocate page dirty bitmap if needed */
368 if ((new.flags & KVM_MEM_LOG_DIRTY_PAGES) && !new.dirty_bitmap) {
369 unsigned dirty_bytes = ALIGN(npages, BITS_PER_LONG) / 8;
370
371 new.dirty_bitmap = vmalloc(dirty_bytes);
372 if (!new.dirty_bitmap)
f78e0e2e 373 goto out_free;
6aa8b732
AK
374 memset(new.dirty_bitmap, 0, dirty_bytes);
375 }
376
6aa8b732
AK
377 if (mem->slot >= kvm->nmemslots)
378 kvm->nmemslots = mem->slot + 1;
379
3ad82a7e
ZX
380 *memslot = new;
381
0de10343
ZX
382 r = kvm_arch_set_memory_region(kvm, mem, old, user_alloc);
383 if (r) {
384 *memslot = old;
385 goto out_free;
82ce2c96
IE
386 }
387
6aa8b732
AK
388 kvm_free_physmem_slot(&old, &new);
389 return 0;
390
f78e0e2e 391out_free:
6aa8b732
AK
392 kvm_free_physmem_slot(&new, &old);
393out:
394 return r;
210c7c4d
IE
395
396}
f78e0e2e
SY
397EXPORT_SYMBOL_GPL(__kvm_set_memory_region);
398
399int kvm_set_memory_region(struct kvm *kvm,
400 struct kvm_userspace_memory_region *mem,
401 int user_alloc)
402{
403 int r;
404
72dc67a6 405 down_write(&kvm->slots_lock);
f78e0e2e 406 r = __kvm_set_memory_region(kvm, mem, user_alloc);
72dc67a6 407 up_write(&kvm->slots_lock);
f78e0e2e
SY
408 return r;
409}
210c7c4d
IE
410EXPORT_SYMBOL_GPL(kvm_set_memory_region);
411
1fe779f8
CO
412int kvm_vm_ioctl_set_memory_region(struct kvm *kvm,
413 struct
414 kvm_userspace_memory_region *mem,
415 int user_alloc)
210c7c4d 416{
e0d62c7f
IE
417 if (mem->slot >= KVM_MEMORY_SLOTS)
418 return -EINVAL;
210c7c4d 419 return kvm_set_memory_region(kvm, mem, user_alloc);
6aa8b732
AK
420}
421
5bb064dc
ZX
422int kvm_get_dirty_log(struct kvm *kvm,
423 struct kvm_dirty_log *log, int *is_dirty)
6aa8b732
AK
424{
425 struct kvm_memory_slot *memslot;
426 int r, i;
427 int n;
428 unsigned long any = 0;
429
6aa8b732
AK
430 r = -EINVAL;
431 if (log->slot >= KVM_MEMORY_SLOTS)
432 goto out;
433
434 memslot = &kvm->memslots[log->slot];
435 r = -ENOENT;
436 if (!memslot->dirty_bitmap)
437 goto out;
438
cd1a4a98 439 n = ALIGN(memslot->npages, BITS_PER_LONG) / 8;
6aa8b732 440
cd1a4a98 441 for (i = 0; !any && i < n/sizeof(long); ++i)
6aa8b732
AK
442 any = memslot->dirty_bitmap[i];
443
444 r = -EFAULT;
445 if (copy_to_user(log->dirty_bitmap, memslot->dirty_bitmap, n))
446 goto out;
447
5bb064dc
ZX
448 if (any)
449 *is_dirty = 1;
6aa8b732
AK
450
451 r = 0;
6aa8b732 452out:
6aa8b732
AK
453 return r;
454}
455
cea7bb21
IE
456int is_error_page(struct page *page)
457{
458 return page == bad_page;
459}
460EXPORT_SYMBOL_GPL(is_error_page);
461
35149e21
AL
462int is_error_pfn(pfn_t pfn)
463{
464 return pfn == bad_pfn;
465}
466EXPORT_SYMBOL_GPL(is_error_pfn);
467
f9d46eb0
IE
468static inline unsigned long bad_hva(void)
469{
470 return PAGE_OFFSET;
471}
472
473int kvm_is_error_hva(unsigned long addr)
474{
475 return addr == bad_hva();
476}
477EXPORT_SYMBOL_GPL(kvm_is_error_hva);
478
e8207547 479static struct kvm_memory_slot *__gfn_to_memslot(struct kvm *kvm, gfn_t gfn)
6aa8b732
AK
480{
481 int i;
482
483 for (i = 0; i < kvm->nmemslots; ++i) {
484 struct kvm_memory_slot *memslot = &kvm->memslots[i];
485
486 if (gfn >= memslot->base_gfn
487 && gfn < memslot->base_gfn + memslot->npages)
488 return memslot;
489 }
8b6d44c7 490 return NULL;
6aa8b732 491}
e8207547
AK
492
493struct kvm_memory_slot *gfn_to_memslot(struct kvm *kvm, gfn_t gfn)
494{
495 gfn = unalias_gfn(kvm, gfn);
496 return __gfn_to_memslot(kvm, gfn);
497}
6aa8b732 498
e0d62c7f
IE
499int kvm_is_visible_gfn(struct kvm *kvm, gfn_t gfn)
500{
501 int i;
502
503 gfn = unalias_gfn(kvm, gfn);
504 for (i = 0; i < KVM_MEMORY_SLOTS; ++i) {
505 struct kvm_memory_slot *memslot = &kvm->memslots[i];
506
507 if (gfn >= memslot->base_gfn
508 && gfn < memslot->base_gfn + memslot->npages)
509 return 1;
510 }
511 return 0;
512}
513EXPORT_SYMBOL_GPL(kvm_is_visible_gfn);
514
05da4558 515unsigned long gfn_to_hva(struct kvm *kvm, gfn_t gfn)
539cb660
IE
516{
517 struct kvm_memory_slot *slot;
518
519 gfn = unalias_gfn(kvm, gfn);
520 slot = __gfn_to_memslot(kvm, gfn);
521 if (!slot)
522 return bad_hva();
523 return (slot->userspace_addr + (gfn - slot->base_gfn) * PAGE_SIZE);
524}
525
aab61cc0
AL
526/*
527 * Requires current->mm->mmap_sem to be held
528 */
35149e21 529pfn_t gfn_to_pfn(struct kvm *kvm, gfn_t gfn)
954bbbc2 530{
8d4e1288 531 struct page *page[1];
539cb660 532 unsigned long addr;
8d4e1288 533 int npages;
954bbbc2 534
60395224
AK
535 might_sleep();
536
539cb660
IE
537 addr = gfn_to_hva(kvm, gfn);
538 if (kvm_is_error_hva(addr)) {
8a7ae055 539 get_page(bad_page);
35149e21 540 return page_to_pfn(bad_page);
8a7ae055 541 }
8d4e1288 542
539cb660
IE
543 npages = get_user_pages(current, current->mm, addr, 1, 1, 1, page,
544 NULL);
545
8d4e1288
AL
546 if (npages != 1) {
547 get_page(bad_page);
35149e21 548 return page_to_pfn(bad_page);
8a7ae055 549 }
8d4e1288 550
35149e21
AL
551 return page_to_pfn(page[0]);
552}
553
554EXPORT_SYMBOL_GPL(gfn_to_pfn);
555
556struct page *gfn_to_page(struct kvm *kvm, gfn_t gfn)
557{
558 return pfn_to_page(gfn_to_pfn(kvm, gfn));
954bbbc2 559}
aab61cc0 560
954bbbc2
AK
561EXPORT_SYMBOL_GPL(gfn_to_page);
562
b4231d61
IE
563void kvm_release_page_clean(struct page *page)
564{
35149e21 565 kvm_release_pfn_clean(page_to_pfn(page));
b4231d61
IE
566}
567EXPORT_SYMBOL_GPL(kvm_release_page_clean);
568
35149e21
AL
569void kvm_release_pfn_clean(pfn_t pfn)
570{
571 put_page(pfn_to_page(pfn));
572}
573EXPORT_SYMBOL_GPL(kvm_release_pfn_clean);
574
b4231d61 575void kvm_release_page_dirty(struct page *page)
8a7ae055 576{
35149e21
AL
577 kvm_release_pfn_dirty(page_to_pfn(page));
578}
579EXPORT_SYMBOL_GPL(kvm_release_page_dirty);
580
581void kvm_release_pfn_dirty(pfn_t pfn)
582{
583 kvm_set_pfn_dirty(pfn);
584 kvm_release_pfn_clean(pfn);
585}
586EXPORT_SYMBOL_GPL(kvm_release_pfn_dirty);
587
588void kvm_set_page_dirty(struct page *page)
589{
590 kvm_set_pfn_dirty(page_to_pfn(page));
591}
592EXPORT_SYMBOL_GPL(kvm_set_page_dirty);
593
594void kvm_set_pfn_dirty(pfn_t pfn)
595{
596 struct page *page = pfn_to_page(pfn);
8a7ae055
IE
597 if (!PageReserved(page))
598 SetPageDirty(page);
8a7ae055 599}
35149e21
AL
600EXPORT_SYMBOL_GPL(kvm_set_pfn_dirty);
601
602void kvm_set_pfn_accessed(pfn_t pfn)
603{
604 mark_page_accessed(pfn_to_page(pfn));
605}
606EXPORT_SYMBOL_GPL(kvm_set_pfn_accessed);
607
608void kvm_get_pfn(pfn_t pfn)
609{
610 get_page(pfn_to_page(pfn));
611}
612EXPORT_SYMBOL_GPL(kvm_get_pfn);
8a7ae055 613
195aefde
IE
614static int next_segment(unsigned long len, int offset)
615{
616 if (len > PAGE_SIZE - offset)
617 return PAGE_SIZE - offset;
618 else
619 return len;
620}
621
622int kvm_read_guest_page(struct kvm *kvm, gfn_t gfn, void *data, int offset,
623 int len)
624{
e0506bcb
IE
625 int r;
626 unsigned long addr;
195aefde 627
e0506bcb
IE
628 addr = gfn_to_hva(kvm, gfn);
629 if (kvm_is_error_hva(addr))
630 return -EFAULT;
631 r = copy_from_user(data, (void __user *)addr + offset, len);
632 if (r)
195aefde 633 return -EFAULT;
195aefde
IE
634 return 0;
635}
636EXPORT_SYMBOL_GPL(kvm_read_guest_page);
637
638int kvm_read_guest(struct kvm *kvm, gpa_t gpa, void *data, unsigned long len)
639{
640 gfn_t gfn = gpa >> PAGE_SHIFT;
641 int seg;
642 int offset = offset_in_page(gpa);
643 int ret;
644
645 while ((seg = next_segment(len, offset)) != 0) {
646 ret = kvm_read_guest_page(kvm, gfn, data, offset, seg);
647 if (ret < 0)
648 return ret;
649 offset = 0;
650 len -= seg;
651 data += seg;
652 ++gfn;
653 }
654 return 0;
655}
656EXPORT_SYMBOL_GPL(kvm_read_guest);
657
7ec54588
MT
658int kvm_read_guest_atomic(struct kvm *kvm, gpa_t gpa, void *data,
659 unsigned long len)
660{
661 int r;
662 unsigned long addr;
663 gfn_t gfn = gpa >> PAGE_SHIFT;
664 int offset = offset_in_page(gpa);
665
666 addr = gfn_to_hva(kvm, gfn);
667 if (kvm_is_error_hva(addr))
668 return -EFAULT;
0aac03f0 669 pagefault_disable();
7ec54588 670 r = __copy_from_user_inatomic(data, (void __user *)addr + offset, len);
0aac03f0 671 pagefault_enable();
7ec54588
MT
672 if (r)
673 return -EFAULT;
674 return 0;
675}
676EXPORT_SYMBOL(kvm_read_guest_atomic);
677
195aefde
IE
678int kvm_write_guest_page(struct kvm *kvm, gfn_t gfn, const void *data,
679 int offset, int len)
680{
e0506bcb
IE
681 int r;
682 unsigned long addr;
195aefde 683
e0506bcb
IE
684 addr = gfn_to_hva(kvm, gfn);
685 if (kvm_is_error_hva(addr))
686 return -EFAULT;
687 r = copy_to_user((void __user *)addr + offset, data, len);
688 if (r)
195aefde 689 return -EFAULT;
195aefde
IE
690 mark_page_dirty(kvm, gfn);
691 return 0;
692}
693EXPORT_SYMBOL_GPL(kvm_write_guest_page);
694
695int kvm_write_guest(struct kvm *kvm, gpa_t gpa, const void *data,
696 unsigned long len)
697{
698 gfn_t gfn = gpa >> PAGE_SHIFT;
699 int seg;
700 int offset = offset_in_page(gpa);
701 int ret;
702
703 while ((seg = next_segment(len, offset)) != 0) {
704 ret = kvm_write_guest_page(kvm, gfn, data, offset, seg);
705 if (ret < 0)
706 return ret;
707 offset = 0;
708 len -= seg;
709 data += seg;
710 ++gfn;
711 }
712 return 0;
713}
714
715int kvm_clear_guest_page(struct kvm *kvm, gfn_t gfn, int offset, int len)
716{
3e021bf5 717 return kvm_write_guest_page(kvm, gfn, empty_zero_page, offset, len);
195aefde
IE
718}
719EXPORT_SYMBOL_GPL(kvm_clear_guest_page);
720
721int kvm_clear_guest(struct kvm *kvm, gpa_t gpa, unsigned long len)
722{
723 gfn_t gfn = gpa >> PAGE_SHIFT;
724 int seg;
725 int offset = offset_in_page(gpa);
726 int ret;
727
728 while ((seg = next_segment(len, offset)) != 0) {
729 ret = kvm_clear_guest_page(kvm, gfn, offset, seg);
730 if (ret < 0)
731 return ret;
732 offset = 0;
733 len -= seg;
734 ++gfn;
735 }
736 return 0;
737}
738EXPORT_SYMBOL_GPL(kvm_clear_guest);
739
6aa8b732
AK
740void mark_page_dirty(struct kvm *kvm, gfn_t gfn)
741{
31389947 742 struct kvm_memory_slot *memslot;
6aa8b732 743
3b6fff19 744 gfn = unalias_gfn(kvm, gfn);
7e9d619d
RR
745 memslot = __gfn_to_memslot(kvm, gfn);
746 if (memslot && memslot->dirty_bitmap) {
747 unsigned long rel_gfn = gfn - memslot->base_gfn;
6aa8b732 748
7e9d619d
RR
749 /* avoid RMW */
750 if (!test_bit(rel_gfn, memslot->dirty_bitmap))
751 set_bit(rel_gfn, memslot->dirty_bitmap);
6aa8b732
AK
752 }
753}
754
b6958ce4
ED
755/*
756 * The vCPU has executed a HLT instruction with in-kernel mode enabled.
757 */
8776e519 758void kvm_vcpu_block(struct kvm_vcpu *vcpu)
d3bef15f 759{
b6958ce4
ED
760 DECLARE_WAITQUEUE(wait, current);
761
762 add_wait_queue(&vcpu->wq, &wait);
763
764 /*
765 * We will block until either an interrupt or a signal wakes us up
766 */
c5ec1534
HQ
767 while (!kvm_cpu_has_interrupt(vcpu)
768 && !signal_pending(current)
53e0aa7b 769 && !kvm_arch_vcpu_runnable(vcpu)) {
b6958ce4
ED
770 set_current_state(TASK_INTERRUPTIBLE);
771 vcpu_put(vcpu);
772 schedule();
773 vcpu_load(vcpu);
774 }
d3bef15f 775
c5ec1534 776 __set_current_state(TASK_RUNNING);
b6958ce4 777 remove_wait_queue(&vcpu->wq, &wait);
b6958ce4
ED
778}
779
6aa8b732
AK
780void kvm_resched(struct kvm_vcpu *vcpu)
781{
3fca0365
YD
782 if (!need_resched())
783 return;
6aa8b732 784 cond_resched();
6aa8b732
AK
785}
786EXPORT_SYMBOL_GPL(kvm_resched);
787
e4a533a4 788static int kvm_vcpu_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
9a2bb7f4
AK
789{
790 struct kvm_vcpu *vcpu = vma->vm_file->private_data;
9a2bb7f4
AK
791 struct page *page;
792
e4a533a4 793 if (vmf->pgoff == 0)
039576c0 794 page = virt_to_page(vcpu->run);
09566765 795#ifdef CONFIG_X86
e4a533a4 796 else if (vmf->pgoff == KVM_PIO_PAGE_OFFSET)
ad312c7c 797 page = virt_to_page(vcpu->arch.pio_data);
09566765 798#endif
039576c0 799 else
e4a533a4 800 return VM_FAULT_SIGBUS;
9a2bb7f4 801 get_page(page);
e4a533a4 802 vmf->page = page;
803 return 0;
9a2bb7f4
AK
804}
805
806static struct vm_operations_struct kvm_vcpu_vm_ops = {
e4a533a4 807 .fault = kvm_vcpu_fault,
9a2bb7f4
AK
808};
809
810static int kvm_vcpu_mmap(struct file *file, struct vm_area_struct *vma)
811{
812 vma->vm_ops = &kvm_vcpu_vm_ops;
813 return 0;
814}
815
bccf2150
AK
816static int kvm_vcpu_release(struct inode *inode, struct file *filp)
817{
818 struct kvm_vcpu *vcpu = filp->private_data;
819
820 fput(vcpu->kvm->filp);
821 return 0;
822}
823
5c502742 824static const struct file_operations kvm_vcpu_fops = {
bccf2150
AK
825 .release = kvm_vcpu_release,
826 .unlocked_ioctl = kvm_vcpu_ioctl,
827 .compat_ioctl = kvm_vcpu_ioctl,
9a2bb7f4 828 .mmap = kvm_vcpu_mmap,
bccf2150
AK
829};
830
831/*
832 * Allocates an inode for the vcpu.
833 */
834static int create_vcpu_fd(struct kvm_vcpu *vcpu)
835{
836 int fd, r;
837 struct inode *inode;
838 struct file *file;
839
d6d28168
AK
840 r = anon_inode_getfd(&fd, &inode, &file,
841 "kvm-vcpu", &kvm_vcpu_fops, vcpu);
842 if (r)
843 return r;
bccf2150 844 atomic_inc(&vcpu->kvm->filp->f_count);
bccf2150 845 return fd;
bccf2150
AK
846}
847
c5ea7660
AK
848/*
849 * Creates some virtual cpus. Good luck creating more than one.
850 */
851static int kvm_vm_ioctl_create_vcpu(struct kvm *kvm, int n)
852{
853 int r;
854 struct kvm_vcpu *vcpu;
855
c5ea7660 856 if (!valid_vcpu(n))
fb3f0f51 857 return -EINVAL;
c5ea7660 858
e9b11c17 859 vcpu = kvm_arch_vcpu_create(kvm, n);
fb3f0f51
RR
860 if (IS_ERR(vcpu))
861 return PTR_ERR(vcpu);
c5ea7660 862
15ad7146
AK
863 preempt_notifier_init(&vcpu->preempt_notifier, &kvm_preempt_ops);
864
26e5215f
AK
865 r = kvm_arch_vcpu_setup(vcpu);
866 if (r)
867 goto vcpu_destroy;
868
11ec2804 869 mutex_lock(&kvm->lock);
fb3f0f51
RR
870 if (kvm->vcpus[n]) {
871 r = -EEXIST;
11ec2804 872 mutex_unlock(&kvm->lock);
e9b11c17 873 goto vcpu_destroy;
fb3f0f51
RR
874 }
875 kvm->vcpus[n] = vcpu;
11ec2804 876 mutex_unlock(&kvm->lock);
c5ea7660 877
fb3f0f51 878 /* Now it's all set up, let userspace reach it */
bccf2150
AK
879 r = create_vcpu_fd(vcpu);
880 if (r < 0)
fb3f0f51
RR
881 goto unlink;
882 return r;
39c3b86e 883
fb3f0f51 884unlink:
11ec2804 885 mutex_lock(&kvm->lock);
fb3f0f51 886 kvm->vcpus[n] = NULL;
11ec2804 887 mutex_unlock(&kvm->lock);
e9b11c17 888vcpu_destroy:
d40ccc62 889 kvm_arch_vcpu_destroy(vcpu);
c5ea7660
AK
890 return r;
891}
892
1961d276
AK
893static int kvm_vcpu_ioctl_set_sigmask(struct kvm_vcpu *vcpu, sigset_t *sigset)
894{
895 if (sigset) {
896 sigdelsetmask(sigset, sigmask(SIGKILL)|sigmask(SIGSTOP));
897 vcpu->sigset_active = 1;
898 vcpu->sigset = *sigset;
899 } else
900 vcpu->sigset_active = 0;
901 return 0;
902}
903
bccf2150
AK
904static long kvm_vcpu_ioctl(struct file *filp,
905 unsigned int ioctl, unsigned long arg)
6aa8b732 906{
bccf2150 907 struct kvm_vcpu *vcpu = filp->private_data;
2f366987 908 void __user *argp = (void __user *)arg;
313a3dc7 909 int r;
6aa8b732 910
6d4e4c4f
AK
911 if (vcpu->kvm->mm != current->mm)
912 return -EIO;
6aa8b732 913 switch (ioctl) {
9a2bb7f4 914 case KVM_RUN:
f0fe5108
AK
915 r = -EINVAL;
916 if (arg)
917 goto out;
b6c7a5dc 918 r = kvm_arch_vcpu_ioctl_run(vcpu, vcpu->run);
6aa8b732 919 break;
6aa8b732 920 case KVM_GET_REGS: {
3e4bb3ac 921 struct kvm_regs *kvm_regs;
6aa8b732 922
3e4bb3ac
XZ
923 r = -ENOMEM;
924 kvm_regs = kzalloc(sizeof(struct kvm_regs), GFP_KERNEL);
925 if (!kvm_regs)
6aa8b732 926 goto out;
3e4bb3ac
XZ
927 r = kvm_arch_vcpu_ioctl_get_regs(vcpu, kvm_regs);
928 if (r)
929 goto out_free1;
6aa8b732 930 r = -EFAULT;
3e4bb3ac
XZ
931 if (copy_to_user(argp, kvm_regs, sizeof(struct kvm_regs)))
932 goto out_free1;
6aa8b732 933 r = 0;
3e4bb3ac
XZ
934out_free1:
935 kfree(kvm_regs);
6aa8b732
AK
936 break;
937 }
938 case KVM_SET_REGS: {
3e4bb3ac 939 struct kvm_regs *kvm_regs;
6aa8b732 940
3e4bb3ac
XZ
941 r = -ENOMEM;
942 kvm_regs = kzalloc(sizeof(struct kvm_regs), GFP_KERNEL);
943 if (!kvm_regs)
6aa8b732 944 goto out;
3e4bb3ac
XZ
945 r = -EFAULT;
946 if (copy_from_user(kvm_regs, argp, sizeof(struct kvm_regs)))
947 goto out_free2;
948 r = kvm_arch_vcpu_ioctl_set_regs(vcpu, kvm_regs);
6aa8b732 949 if (r)
3e4bb3ac 950 goto out_free2;
6aa8b732 951 r = 0;
3e4bb3ac
XZ
952out_free2:
953 kfree(kvm_regs);
6aa8b732
AK
954 break;
955 }
956 case KVM_GET_SREGS: {
957 struct kvm_sregs kvm_sregs;
958
bccf2150 959 memset(&kvm_sregs, 0, sizeof kvm_sregs);
b6c7a5dc 960 r = kvm_arch_vcpu_ioctl_get_sregs(vcpu, &kvm_sregs);
6aa8b732
AK
961 if (r)
962 goto out;
963 r = -EFAULT;
2f366987 964 if (copy_to_user(argp, &kvm_sregs, sizeof kvm_sregs))
6aa8b732
AK
965 goto out;
966 r = 0;
967 break;
968 }
969 case KVM_SET_SREGS: {
970 struct kvm_sregs kvm_sregs;
971
972 r = -EFAULT;
2f366987 973 if (copy_from_user(&kvm_sregs, argp, sizeof kvm_sregs))
6aa8b732 974 goto out;
b6c7a5dc 975 r = kvm_arch_vcpu_ioctl_set_sregs(vcpu, &kvm_sregs);
6aa8b732
AK
976 if (r)
977 goto out;
978 r = 0;
979 break;
980 }
981 case KVM_TRANSLATE: {
982 struct kvm_translation tr;
983
984 r = -EFAULT;
2f366987 985 if (copy_from_user(&tr, argp, sizeof tr))
6aa8b732 986 goto out;
8b006791 987 r = kvm_arch_vcpu_ioctl_translate(vcpu, &tr);
6aa8b732
AK
988 if (r)
989 goto out;
990 r = -EFAULT;
2f366987 991 if (copy_to_user(argp, &tr, sizeof tr))
6aa8b732
AK
992 goto out;
993 r = 0;
994 break;
995 }
6aa8b732
AK
996 case KVM_DEBUG_GUEST: {
997 struct kvm_debug_guest dbg;
998
999 r = -EFAULT;
2f366987 1000 if (copy_from_user(&dbg, argp, sizeof dbg))
6aa8b732 1001 goto out;
b6c7a5dc 1002 r = kvm_arch_vcpu_ioctl_debug_guest(vcpu, &dbg);
6aa8b732
AK
1003 if (r)
1004 goto out;
1005 r = 0;
1006 break;
1007 }
1961d276
AK
1008 case KVM_SET_SIGNAL_MASK: {
1009 struct kvm_signal_mask __user *sigmask_arg = argp;
1010 struct kvm_signal_mask kvm_sigmask;
1011 sigset_t sigset, *p;
1012
1013 p = NULL;
1014 if (argp) {
1015 r = -EFAULT;
1016 if (copy_from_user(&kvm_sigmask, argp,
1017 sizeof kvm_sigmask))
1018 goto out;
1019 r = -EINVAL;
1020 if (kvm_sigmask.len != sizeof sigset)
1021 goto out;
1022 r = -EFAULT;
1023 if (copy_from_user(&sigset, sigmask_arg->sigset,
1024 sizeof sigset))
1025 goto out;
1026 p = &sigset;
1027 }
1028 r = kvm_vcpu_ioctl_set_sigmask(vcpu, &sigset);
1029 break;
1030 }
b8836737
AK
1031 case KVM_GET_FPU: {
1032 struct kvm_fpu fpu;
1033
1034 memset(&fpu, 0, sizeof fpu);
d0752060 1035 r = kvm_arch_vcpu_ioctl_get_fpu(vcpu, &fpu);
b8836737
AK
1036 if (r)
1037 goto out;
1038 r = -EFAULT;
1039 if (copy_to_user(argp, &fpu, sizeof fpu))
1040 goto out;
1041 r = 0;
1042 break;
1043 }
1044 case KVM_SET_FPU: {
1045 struct kvm_fpu fpu;
1046
1047 r = -EFAULT;
1048 if (copy_from_user(&fpu, argp, sizeof fpu))
1049 goto out;
d0752060 1050 r = kvm_arch_vcpu_ioctl_set_fpu(vcpu, &fpu);
b8836737
AK
1051 if (r)
1052 goto out;
1053 r = 0;
1054 break;
1055 }
bccf2150 1056 default:
313a3dc7 1057 r = kvm_arch_vcpu_ioctl(filp, ioctl, arg);
bccf2150
AK
1058 }
1059out:
1060 return r;
1061}
1062
1063static long kvm_vm_ioctl(struct file *filp,
1064 unsigned int ioctl, unsigned long arg)
1065{
1066 struct kvm *kvm = filp->private_data;
1067 void __user *argp = (void __user *)arg;
1fe779f8 1068 int r;
bccf2150 1069
6d4e4c4f
AK
1070 if (kvm->mm != current->mm)
1071 return -EIO;
bccf2150
AK
1072 switch (ioctl) {
1073 case KVM_CREATE_VCPU:
1074 r = kvm_vm_ioctl_create_vcpu(kvm, arg);
1075 if (r < 0)
1076 goto out;
1077 break;
6fc138d2
IE
1078 case KVM_SET_USER_MEMORY_REGION: {
1079 struct kvm_userspace_memory_region kvm_userspace_mem;
1080
1081 r = -EFAULT;
1082 if (copy_from_user(&kvm_userspace_mem, argp,
1083 sizeof kvm_userspace_mem))
1084 goto out;
1085
1086 r = kvm_vm_ioctl_set_memory_region(kvm, &kvm_userspace_mem, 1);
6aa8b732
AK
1087 if (r)
1088 goto out;
1089 break;
1090 }
1091 case KVM_GET_DIRTY_LOG: {
1092 struct kvm_dirty_log log;
1093
1094 r = -EFAULT;
2f366987 1095 if (copy_from_user(&log, argp, sizeof log))
6aa8b732 1096 goto out;
2c6f5df9 1097 r = kvm_vm_ioctl_get_dirty_log(kvm, &log);
6aa8b732
AK
1098 if (r)
1099 goto out;
1100 break;
1101 }
f17abe9a 1102 default:
1fe779f8 1103 r = kvm_arch_vm_ioctl(filp, ioctl, arg);
f17abe9a
AK
1104 }
1105out:
1106 return r;
1107}
1108
e4a533a4 1109static int kvm_vm_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
f17abe9a
AK
1110{
1111 struct kvm *kvm = vma->vm_file->private_data;
f17abe9a
AK
1112 struct page *page;
1113
e4a533a4 1114 if (!kvm_is_visible_gfn(kvm, vmf->pgoff))
1115 return VM_FAULT_SIGBUS;
10589a46 1116 page = gfn_to_page(kvm, vmf->pgoff);
8a7ae055 1117 if (is_error_page(page)) {
b4231d61 1118 kvm_release_page_clean(page);
e4a533a4 1119 return VM_FAULT_SIGBUS;
8a7ae055 1120 }
e4a533a4 1121 vmf->page = page;
1122 return 0;
f17abe9a
AK
1123}
1124
1125static struct vm_operations_struct kvm_vm_vm_ops = {
e4a533a4 1126 .fault = kvm_vm_fault,
f17abe9a
AK
1127};
1128
1129static int kvm_vm_mmap(struct file *file, struct vm_area_struct *vma)
1130{
1131 vma->vm_ops = &kvm_vm_vm_ops;
1132 return 0;
1133}
1134
5c502742 1135static const struct file_operations kvm_vm_fops = {
f17abe9a
AK
1136 .release = kvm_vm_release,
1137 .unlocked_ioctl = kvm_vm_ioctl,
1138 .compat_ioctl = kvm_vm_ioctl,
1139 .mmap = kvm_vm_mmap,
1140};
1141
1142static int kvm_dev_ioctl_create_vm(void)
1143{
1144 int fd, r;
1145 struct inode *inode;
1146 struct file *file;
1147 struct kvm *kvm;
1148
f17abe9a 1149 kvm = kvm_create_vm();
d6d28168
AK
1150 if (IS_ERR(kvm))
1151 return PTR_ERR(kvm);
1152 r = anon_inode_getfd(&fd, &inode, &file, "kvm-vm", &kvm_vm_fops, kvm);
1153 if (r) {
1154 kvm_destroy_vm(kvm);
1155 return r;
f17abe9a
AK
1156 }
1157
bccf2150 1158 kvm->filp = file;
f17abe9a 1159
f17abe9a 1160 return fd;
f17abe9a
AK
1161}
1162
1163static long kvm_dev_ioctl(struct file *filp,
1164 unsigned int ioctl, unsigned long arg)
1165{
1166 void __user *argp = (void __user *)arg;
07c45a36 1167 long r = -EINVAL;
f17abe9a
AK
1168
1169 switch (ioctl) {
1170 case KVM_GET_API_VERSION:
f0fe5108
AK
1171 r = -EINVAL;
1172 if (arg)
1173 goto out;
f17abe9a
AK
1174 r = KVM_API_VERSION;
1175 break;
1176 case KVM_CREATE_VM:
f0fe5108
AK
1177 r = -EINVAL;
1178 if (arg)
1179 goto out;
f17abe9a
AK
1180 r = kvm_dev_ioctl_create_vm();
1181 break;
018d00d2
ZX
1182 case KVM_CHECK_EXTENSION:
1183 r = kvm_dev_ioctl_check_extension((long)argp);
5d308f45 1184 break;
07c45a36
AK
1185 case KVM_GET_VCPU_MMAP_SIZE:
1186 r = -EINVAL;
1187 if (arg)
1188 goto out;
adb1ff46
AK
1189 r = PAGE_SIZE; /* struct kvm_run */
1190#ifdef CONFIG_X86
1191 r += PAGE_SIZE; /* pio data page */
1192#endif
07c45a36 1193 break;
6aa8b732 1194 default:
043405e1 1195 return kvm_arch_dev_ioctl(filp, ioctl, arg);
6aa8b732
AK
1196 }
1197out:
1198 return r;
1199}
1200
6aa8b732 1201static struct file_operations kvm_chardev_ops = {
6aa8b732
AK
1202 .unlocked_ioctl = kvm_dev_ioctl,
1203 .compat_ioctl = kvm_dev_ioctl,
6aa8b732
AK
1204};
1205
1206static struct miscdevice kvm_dev = {
bbe4432e 1207 KVM_MINOR,
6aa8b732
AK
1208 "kvm",
1209 &kvm_chardev_ops,
1210};
1211
1b6c0168
AK
1212static void hardware_enable(void *junk)
1213{
1214 int cpu = raw_smp_processor_id();
1215
1216 if (cpu_isset(cpu, cpus_hardware_enabled))
1217 return;
1218 cpu_set(cpu, cpus_hardware_enabled);
e9b11c17 1219 kvm_arch_hardware_enable(NULL);
1b6c0168
AK
1220}
1221
1222static void hardware_disable(void *junk)
1223{
1224 int cpu = raw_smp_processor_id();
1225
1226 if (!cpu_isset(cpu, cpus_hardware_enabled))
1227 return;
1228 cpu_clear(cpu, cpus_hardware_enabled);
1229 decache_vcpus_on_cpu(cpu);
e9b11c17 1230 kvm_arch_hardware_disable(NULL);
1b6c0168
AK
1231}
1232
774c47f1
AK
1233static int kvm_cpu_hotplug(struct notifier_block *notifier, unsigned long val,
1234 void *v)
1235{
1236 int cpu = (long)v;
1237
1a6f4d7f 1238 val &= ~CPU_TASKS_FROZEN;
774c47f1 1239 switch (val) {
cec9ad27 1240 case CPU_DYING:
6ec8a856
AK
1241 printk(KERN_INFO "kvm: disabling virtualization on CPU%d\n",
1242 cpu);
1243 hardware_disable(NULL);
1244 break;
774c47f1 1245 case CPU_UP_CANCELED:
43934a38
JK
1246 printk(KERN_INFO "kvm: disabling virtualization on CPU%d\n",
1247 cpu);
1b6c0168 1248 smp_call_function_single(cpu, hardware_disable, NULL, 0, 1);
774c47f1 1249 break;
43934a38
JK
1250 case CPU_ONLINE:
1251 printk(KERN_INFO "kvm: enabling virtualization on CPU%d\n",
1252 cpu);
1b6c0168 1253 smp_call_function_single(cpu, hardware_enable, NULL, 0, 1);
774c47f1
AK
1254 break;
1255 }
1256 return NOTIFY_OK;
1257}
1258
9a2b85c6 1259static int kvm_reboot(struct notifier_block *notifier, unsigned long val,
d77c26fc 1260 void *v)
9a2b85c6
RR
1261{
1262 if (val == SYS_RESTART) {
1263 /*
1264 * Some (well, at least mine) BIOSes hang on reboot if
1265 * in vmx root mode.
1266 */
1267 printk(KERN_INFO "kvm: exiting hardware virtualization\n");
1268 on_each_cpu(hardware_disable, NULL, 0, 1);
1269 }
1270 return NOTIFY_OK;
1271}
1272
1273static struct notifier_block kvm_reboot_notifier = {
1274 .notifier_call = kvm_reboot,
1275 .priority = 0,
1276};
1277
2eeb2e94
GH
1278void kvm_io_bus_init(struct kvm_io_bus *bus)
1279{
1280 memset(bus, 0, sizeof(*bus));
1281}
1282
1283void kvm_io_bus_destroy(struct kvm_io_bus *bus)
1284{
1285 int i;
1286
1287 for (i = 0; i < bus->dev_count; i++) {
1288 struct kvm_io_device *pos = bus->devs[i];
1289
1290 kvm_iodevice_destructor(pos);
1291 }
1292}
1293
1294struct kvm_io_device *kvm_io_bus_find_dev(struct kvm_io_bus *bus, gpa_t addr)
1295{
1296 int i;
1297
1298 for (i = 0; i < bus->dev_count; i++) {
1299 struct kvm_io_device *pos = bus->devs[i];
1300
1301 if (pos->in_range(pos, addr))
1302 return pos;
1303 }
1304
1305 return NULL;
1306}
1307
1308void kvm_io_bus_register_dev(struct kvm_io_bus *bus, struct kvm_io_device *dev)
1309{
1310 BUG_ON(bus->dev_count > (NR_IOBUS_DEVS-1));
1311
1312 bus->devs[bus->dev_count++] = dev;
1313}
1314
774c47f1
AK
1315static struct notifier_block kvm_cpu_notifier = {
1316 .notifier_call = kvm_cpu_hotplug,
1317 .priority = 20, /* must be > scheduler priority */
1318};
1319
8b88b099 1320static int vm_stat_get(void *_offset, u64 *val)
ba1389b7
AK
1321{
1322 unsigned offset = (long)_offset;
ba1389b7
AK
1323 struct kvm *kvm;
1324
8b88b099 1325 *val = 0;
ba1389b7
AK
1326 spin_lock(&kvm_lock);
1327 list_for_each_entry(kvm, &vm_list, vm_list)
8b88b099 1328 *val += *(u32 *)((void *)kvm + offset);
ba1389b7 1329 spin_unlock(&kvm_lock);
8b88b099 1330 return 0;
ba1389b7
AK
1331}
1332
1333DEFINE_SIMPLE_ATTRIBUTE(vm_stat_fops, vm_stat_get, NULL, "%llu\n");
1334
8b88b099 1335static int vcpu_stat_get(void *_offset, u64 *val)
1165f5fe
AK
1336{
1337 unsigned offset = (long)_offset;
1165f5fe
AK
1338 struct kvm *kvm;
1339 struct kvm_vcpu *vcpu;
1340 int i;
1341
8b88b099 1342 *val = 0;
1165f5fe
AK
1343 spin_lock(&kvm_lock);
1344 list_for_each_entry(kvm, &vm_list, vm_list)
1345 for (i = 0; i < KVM_MAX_VCPUS; ++i) {
fb3f0f51
RR
1346 vcpu = kvm->vcpus[i];
1347 if (vcpu)
8b88b099 1348 *val += *(u32 *)((void *)vcpu + offset);
1165f5fe
AK
1349 }
1350 spin_unlock(&kvm_lock);
8b88b099 1351 return 0;
1165f5fe
AK
1352}
1353
ba1389b7
AK
1354DEFINE_SIMPLE_ATTRIBUTE(vcpu_stat_fops, vcpu_stat_get, NULL, "%llu\n");
1355
1356static struct file_operations *stat_fops[] = {
1357 [KVM_STAT_VCPU] = &vcpu_stat_fops,
1358 [KVM_STAT_VM] = &vm_stat_fops,
1359};
1165f5fe 1360
a16b043c 1361static void kvm_init_debug(void)
6aa8b732
AK
1362{
1363 struct kvm_stats_debugfs_item *p;
1364
8b6d44c7 1365 debugfs_dir = debugfs_create_dir("kvm", NULL);
6aa8b732 1366 for (p = debugfs_entries; p->name; ++p)
1165f5fe
AK
1367 p->dentry = debugfs_create_file(p->name, 0444, debugfs_dir,
1368 (void *)(long)p->offset,
ba1389b7 1369 stat_fops[p->kind]);
6aa8b732
AK
1370}
1371
1372static void kvm_exit_debug(void)
1373{
1374 struct kvm_stats_debugfs_item *p;
1375
1376 for (p = debugfs_entries; p->name; ++p)
1377 debugfs_remove(p->dentry);
1378 debugfs_remove(debugfs_dir);
1379}
1380
59ae6c6b
AK
1381static int kvm_suspend(struct sys_device *dev, pm_message_t state)
1382{
4267c41a 1383 hardware_disable(NULL);
59ae6c6b
AK
1384 return 0;
1385}
1386
1387static int kvm_resume(struct sys_device *dev)
1388{
4267c41a 1389 hardware_enable(NULL);
59ae6c6b
AK
1390 return 0;
1391}
1392
1393static struct sysdev_class kvm_sysdev_class = {
af5ca3f4 1394 .name = "kvm",
59ae6c6b
AK
1395 .suspend = kvm_suspend,
1396 .resume = kvm_resume,
1397};
1398
1399static struct sys_device kvm_sysdev = {
1400 .id = 0,
1401 .cls = &kvm_sysdev_class,
1402};
1403
cea7bb21 1404struct page *bad_page;
35149e21 1405pfn_t bad_pfn;
6aa8b732 1406
15ad7146
AK
1407static inline
1408struct kvm_vcpu *preempt_notifier_to_vcpu(struct preempt_notifier *pn)
1409{
1410 return container_of(pn, struct kvm_vcpu, preempt_notifier);
1411}
1412
1413static void kvm_sched_in(struct preempt_notifier *pn, int cpu)
1414{
1415 struct kvm_vcpu *vcpu = preempt_notifier_to_vcpu(pn);
1416
e9b11c17 1417 kvm_arch_vcpu_load(vcpu, cpu);
15ad7146
AK
1418}
1419
1420static void kvm_sched_out(struct preempt_notifier *pn,
1421 struct task_struct *next)
1422{
1423 struct kvm_vcpu *vcpu = preempt_notifier_to_vcpu(pn);
1424
e9b11c17 1425 kvm_arch_vcpu_put(vcpu);
15ad7146
AK
1426}
1427
f8c16bba 1428int kvm_init(void *opaque, unsigned int vcpu_size,
c16f862d 1429 struct module *module)
6aa8b732
AK
1430{
1431 int r;
002c7f7c 1432 int cpu;
6aa8b732 1433
cb498ea2
ZX
1434 kvm_init_debug();
1435
f8c16bba
ZX
1436 r = kvm_arch_init(opaque);
1437 if (r)
d2308784 1438 goto out_fail;
cb498ea2
ZX
1439
1440 bad_page = alloc_page(GFP_KERNEL | __GFP_ZERO);
1441
1442 if (bad_page == NULL) {
1443 r = -ENOMEM;
1444 goto out;
1445 }
1446
35149e21
AL
1447 bad_pfn = page_to_pfn(bad_page);
1448
e9b11c17 1449 r = kvm_arch_hardware_setup();
6aa8b732 1450 if (r < 0)
d2308784 1451 goto out_free_0;
6aa8b732 1452
002c7f7c
YS
1453 for_each_online_cpu(cpu) {
1454 smp_call_function_single(cpu,
e9b11c17 1455 kvm_arch_check_processor_compat,
002c7f7c
YS
1456 &r, 0, 1);
1457 if (r < 0)
d2308784 1458 goto out_free_1;
002c7f7c
YS
1459 }
1460
1b6c0168 1461 on_each_cpu(hardware_enable, NULL, 0, 1);
774c47f1
AK
1462 r = register_cpu_notifier(&kvm_cpu_notifier);
1463 if (r)
d2308784 1464 goto out_free_2;
6aa8b732
AK
1465 register_reboot_notifier(&kvm_reboot_notifier);
1466
59ae6c6b
AK
1467 r = sysdev_class_register(&kvm_sysdev_class);
1468 if (r)
d2308784 1469 goto out_free_3;
59ae6c6b
AK
1470
1471 r = sysdev_register(&kvm_sysdev);
1472 if (r)
d2308784 1473 goto out_free_4;
59ae6c6b 1474
c16f862d
RR
1475 /* A kmem cache lets us meet the alignment requirements of fx_save. */
1476 kvm_vcpu_cache = kmem_cache_create("kvm_vcpu", vcpu_size,
56919c5c
JP
1477 __alignof__(struct kvm_vcpu),
1478 0, NULL);
c16f862d
RR
1479 if (!kvm_vcpu_cache) {
1480 r = -ENOMEM;
d2308784 1481 goto out_free_5;
c16f862d
RR
1482 }
1483
6aa8b732
AK
1484 kvm_chardev_ops.owner = module;
1485
1486 r = misc_register(&kvm_dev);
1487 if (r) {
d77c26fc 1488 printk(KERN_ERR "kvm: misc device register failed\n");
6aa8b732
AK
1489 goto out_free;
1490 }
1491
15ad7146
AK
1492 kvm_preempt_ops.sched_in = kvm_sched_in;
1493 kvm_preempt_ops.sched_out = kvm_sched_out;
1494
c7addb90 1495 return 0;
6aa8b732
AK
1496
1497out_free:
c16f862d 1498 kmem_cache_destroy(kvm_vcpu_cache);
d2308784 1499out_free_5:
59ae6c6b 1500 sysdev_unregister(&kvm_sysdev);
d2308784 1501out_free_4:
59ae6c6b 1502 sysdev_class_unregister(&kvm_sysdev_class);
d2308784 1503out_free_3:
6aa8b732 1504 unregister_reboot_notifier(&kvm_reboot_notifier);
774c47f1 1505 unregister_cpu_notifier(&kvm_cpu_notifier);
d2308784 1506out_free_2:
1b6c0168 1507 on_each_cpu(hardware_disable, NULL, 0, 1);
d2308784 1508out_free_1:
e9b11c17 1509 kvm_arch_hardware_unsetup();
d2308784
ZX
1510out_free_0:
1511 __free_page(bad_page);
ca45aaae 1512out:
f8c16bba 1513 kvm_arch_exit();
cb498ea2 1514 kvm_exit_debug();
d2308784 1515out_fail:
6aa8b732
AK
1516 return r;
1517}
cb498ea2 1518EXPORT_SYMBOL_GPL(kvm_init);
6aa8b732 1519
cb498ea2 1520void kvm_exit(void)
6aa8b732
AK
1521{
1522 misc_deregister(&kvm_dev);
c16f862d 1523 kmem_cache_destroy(kvm_vcpu_cache);
59ae6c6b
AK
1524 sysdev_unregister(&kvm_sysdev);
1525 sysdev_class_unregister(&kvm_sysdev_class);
6aa8b732 1526 unregister_reboot_notifier(&kvm_reboot_notifier);
59ae6c6b 1527 unregister_cpu_notifier(&kvm_cpu_notifier);
1b6c0168 1528 on_each_cpu(hardware_disable, NULL, 0, 1);
e9b11c17 1529 kvm_arch_hardware_unsetup();
f8c16bba 1530 kvm_arch_exit();
6aa8b732 1531 kvm_exit_debug();
cea7bb21 1532 __free_page(bad_page);
6aa8b732 1533}
cb498ea2 1534EXPORT_SYMBOL_GPL(kvm_exit);