]> git.proxmox.com Git - mirror_ubuntu-artful-kernel.git/blame - virt/kvm/kvm_main.c
KVM: IA64: Carry non-ia64 changes into ia64
[mirror_ubuntu-artful-kernel.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.
9611c187 8 * Copyright 2010 Red Hat, Inc. and/or its affiliates.
6aa8b732
AK
9 *
10 * Authors:
11 * Avi Kivity <avi@qumranet.com>
12 * Yaniv Kamay <yaniv@qumranet.com>
13 *
14 * This work is licensed under the terms of the GNU GPL, version 2. See
15 * the COPYING file in the top-level directory.
16 *
17 */
18
e2174021 19#include "iodev.h"
6aa8b732 20
edf88417 21#include <linux/kvm_host.h>
6aa8b732
AK
22#include <linux/kvm.h>
23#include <linux/module.h>
24#include <linux/errno.h>
6aa8b732 25#include <linux/percpu.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>
fb3600cc 33#include <linux/syscore_ops.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>
e56d532f 44#include <linux/bitops.h>
547de29e 45#include <linux/spinlock.h>
6ff5894c 46#include <linux/compat.h>
bc6678a3 47#include <linux/srcu.h>
8f0b1ab6 48#include <linux/hugetlb.h>
5a0e3ad6 49#include <linux/slab.h>
743eeb0b
SL
50#include <linux/sort.h>
51#include <linux/bsearch.h>
6aa8b732 52
e495606d 53#include <asm/processor.h>
e495606d
AK
54#include <asm/io.h>
55#include <asm/uaccess.h>
3e021bf5 56#include <asm/pgtable.h>
6aa8b732 57
5f94c174 58#include "coalesced_mmio.h"
af585b92 59#include "async_pf.h"
5f94c174 60
229456fc
MT
61#define CREATE_TRACE_POINTS
62#include <trace/events/kvm.h>
63
6aa8b732
AK
64MODULE_AUTHOR("Qumranet");
65MODULE_LICENSE("GPL");
66
fa40a821
MT
67/*
68 * Ordering of locks:
69 *
fae3a353 70 * kvm->lock --> kvm->slots_lock --> kvm->irq_lock
fa40a821
MT
71 */
72
e935b837 73DEFINE_RAW_SPINLOCK(kvm_lock);
e9b11c17 74LIST_HEAD(vm_list);
133de902 75
7f59f492 76static cpumask_var_t cpus_hardware_enabled;
10474ae8
AG
77static int kvm_usage_count = 0;
78static atomic_t hardware_enable_failed;
1b6c0168 79
c16f862d
RR
80struct kmem_cache *kvm_vcpu_cache;
81EXPORT_SYMBOL_GPL(kvm_vcpu_cache);
1165f5fe 82
15ad7146
AK
83static __read_mostly struct preempt_ops kvm_preempt_ops;
84
76f7c879 85struct dentry *kvm_debugfs_dir;
6aa8b732 86
bccf2150
AK
87static long kvm_vcpu_ioctl(struct file *file, unsigned int ioctl,
88 unsigned long arg);
1dda606c
AG
89#ifdef CONFIG_COMPAT
90static long kvm_vcpu_compat_ioctl(struct file *file, unsigned int ioctl,
91 unsigned long arg);
92#endif
10474ae8
AG
93static int hardware_enable_all(void);
94static void hardware_disable_all(void);
bccf2150 95
e93f8a0f
MT
96static void kvm_io_bus_destroy(struct kvm_io_bus *bus);
97
b7c4145b
AK
98bool kvm_rebooting;
99EXPORT_SYMBOL_GPL(kvm_rebooting);
4ecac3fd 100
54dee993
MT
101static bool largepages_enabled = true;
102
a2766325 103bool kvm_is_mmio_pfn(pfn_t pfn)
cbff90a7 104{
fc5659c8 105 if (pfn_valid(pfn)) {
22e5c47e 106 int reserved;
936a5fe6 107 struct page *tail = pfn_to_page(pfn);
22e5c47e
AA
108 struct page *head = compound_trans_head(tail);
109 reserved = PageReserved(head);
936a5fe6 110 if (head != tail) {
936a5fe6 111 /*
22e5c47e
AA
112 * "head" is not a dangling pointer
113 * (compound_trans_head takes care of that)
114 * but the hugepage may have been splitted
115 * from under us (and we may not hold a
116 * reference count on the head page so it can
117 * be reused before we run PageReferenced), so
118 * we've to check PageTail before returning
119 * what we just read.
936a5fe6 120 */
22e5c47e
AA
121 smp_rmb();
122 if (PageTail(tail))
123 return reserved;
936a5fe6
AA
124 }
125 return PageReserved(tail);
fc5659c8 126 }
cbff90a7
BAY
127
128 return true;
129}
130
bccf2150
AK
131/*
132 * Switches to specified vcpu, until a matching vcpu_put()
133 */
9fc77441 134int vcpu_load(struct kvm_vcpu *vcpu)
6aa8b732 135{
15ad7146
AK
136 int cpu;
137
9fc77441
MT
138 if (mutex_lock_killable(&vcpu->mutex))
139 return -EINTR;
34bb10b7
RR
140 if (unlikely(vcpu->pid != current->pids[PIDTYPE_PID].pid)) {
141 /* The thread running this VCPU changed. */
142 struct pid *oldpid = vcpu->pid;
143 struct pid *newpid = get_task_pid(current, PIDTYPE_PID);
144 rcu_assign_pointer(vcpu->pid, newpid);
145 synchronize_rcu();
146 put_pid(oldpid);
147 }
15ad7146
AK
148 cpu = get_cpu();
149 preempt_notifier_register(&vcpu->preempt_notifier);
313a3dc7 150 kvm_arch_vcpu_load(vcpu, cpu);
15ad7146 151 put_cpu();
9fc77441 152 return 0;
6aa8b732
AK
153}
154
313a3dc7 155void vcpu_put(struct kvm_vcpu *vcpu)
6aa8b732 156{
15ad7146 157 preempt_disable();
313a3dc7 158 kvm_arch_vcpu_put(vcpu);
15ad7146
AK
159 preempt_notifier_unregister(&vcpu->preempt_notifier);
160 preempt_enable();
6aa8b732
AK
161 mutex_unlock(&vcpu->mutex);
162}
163
d9e368d6
AK
164static void ack_flush(void *_completed)
165{
d9e368d6
AK
166}
167
49846896 168static bool make_all_cpus_request(struct kvm *kvm, unsigned int req)
d9e368d6 169{
597a5f55 170 int i, cpu, me;
6ef7a1bc
RR
171 cpumask_var_t cpus;
172 bool called = true;
d9e368d6 173 struct kvm_vcpu *vcpu;
d9e368d6 174
79f55997 175 zalloc_cpumask_var(&cpus, GFP_ATOMIC);
6ef7a1bc 176
3cba4130 177 me = get_cpu();
988a2cae 178 kvm_for_each_vcpu(i, vcpu, kvm) {
3cba4130 179 kvm_make_request(req, vcpu);
d9e368d6 180 cpu = vcpu->cpu;
6b7e2d09
XG
181
182 /* Set ->requests bit before we read ->mode */
183 smp_mb();
184
185 if (cpus != NULL && cpu != -1 && cpu != me &&
186 kvm_vcpu_exiting_guest_mode(vcpu) != OUTSIDE_GUEST_MODE)
6ef7a1bc 187 cpumask_set_cpu(cpu, cpus);
49846896 188 }
6ef7a1bc
RR
189 if (unlikely(cpus == NULL))
190 smp_call_function_many(cpu_online_mask, ack_flush, NULL, 1);
191 else if (!cpumask_empty(cpus))
192 smp_call_function_many(cpus, ack_flush, NULL, 1);
193 else
194 called = false;
3cba4130 195 put_cpu();
6ef7a1bc 196 free_cpumask_var(cpus);
49846896 197 return called;
d9e368d6
AK
198}
199
49846896 200void kvm_flush_remote_tlbs(struct kvm *kvm)
2e53d63a 201{
bec87d6e 202 long dirty_count = kvm->tlbs_dirty;
a4ee1ca4
XG
203
204 smp_mb();
49846896
RR
205 if (make_all_cpus_request(kvm, KVM_REQ_TLB_FLUSH))
206 ++kvm->stat.remote_tlb_flush;
a4ee1ca4 207 cmpxchg(&kvm->tlbs_dirty, dirty_count, 0);
2e53d63a
MT
208}
209
49846896
RR
210void kvm_reload_remote_mmus(struct kvm *kvm)
211{
212 make_all_cpus_request(kvm, KVM_REQ_MMU_RELOAD);
213}
2e53d63a 214
d828199e
MT
215void kvm_make_mclock_inprogress_request(struct kvm *kvm)
216{
217 make_all_cpus_request(kvm, KVM_REQ_MCLOCK_INPROGRESS);
218}
219
3d81bc7e 220void kvm_make_scan_ioapic_request(struct kvm *kvm)
c7c9c56c 221{
3d81bc7e 222 make_all_cpus_request(kvm, KVM_REQ_SCAN_IOAPIC);
c7c9c56c
YZ
223}
224
fb3f0f51
RR
225int kvm_vcpu_init(struct kvm_vcpu *vcpu, struct kvm *kvm, unsigned id)
226{
227 struct page *page;
228 int r;
229
230 mutex_init(&vcpu->mutex);
231 vcpu->cpu = -1;
fb3f0f51
RR
232 vcpu->kvm = kvm;
233 vcpu->vcpu_id = id;
34bb10b7 234 vcpu->pid = NULL;
b6958ce4 235 init_waitqueue_head(&vcpu->wq);
af585b92 236 kvm_async_pf_vcpu_init(vcpu);
fb3f0f51
RR
237
238 page = alloc_page(GFP_KERNEL | __GFP_ZERO);
239 if (!page) {
240 r = -ENOMEM;
241 goto fail;
242 }
243 vcpu->run = page_address(page);
244
4c088493
R
245 kvm_vcpu_set_in_spin_loop(vcpu, false);
246 kvm_vcpu_set_dy_eligible(vcpu, false);
3a08a8f9 247 vcpu->preempted = false;
4c088493 248
e9b11c17 249 r = kvm_arch_vcpu_init(vcpu);
fb3f0f51 250 if (r < 0)
e9b11c17 251 goto fail_free_run;
fb3f0f51
RR
252 return 0;
253
fb3f0f51
RR
254fail_free_run:
255 free_page((unsigned long)vcpu->run);
256fail:
76fafa5e 257 return r;
fb3f0f51
RR
258}
259EXPORT_SYMBOL_GPL(kvm_vcpu_init);
260
261void kvm_vcpu_uninit(struct kvm_vcpu *vcpu)
262{
34bb10b7 263 put_pid(vcpu->pid);
e9b11c17 264 kvm_arch_vcpu_uninit(vcpu);
fb3f0f51
RR
265 free_page((unsigned long)vcpu->run);
266}
267EXPORT_SYMBOL_GPL(kvm_vcpu_uninit);
268
e930bffe
AA
269#if defined(CONFIG_MMU_NOTIFIER) && defined(KVM_ARCH_WANT_MMU_NOTIFIER)
270static inline struct kvm *mmu_notifier_to_kvm(struct mmu_notifier *mn)
271{
272 return container_of(mn, struct kvm, mmu_notifier);
273}
274
275static void kvm_mmu_notifier_invalidate_page(struct mmu_notifier *mn,
276 struct mm_struct *mm,
277 unsigned long address)
278{
279 struct kvm *kvm = mmu_notifier_to_kvm(mn);
bc6678a3 280 int need_tlb_flush, idx;
e930bffe
AA
281
282 /*
283 * When ->invalidate_page runs, the linux pte has been zapped
284 * already but the page is still allocated until
285 * ->invalidate_page returns. So if we increase the sequence
286 * here the kvm page fault will notice if the spte can't be
287 * established because the page is going to be freed. If
288 * instead the kvm page fault establishes the spte before
289 * ->invalidate_page runs, kvm_unmap_hva will release it
290 * before returning.
291 *
292 * The sequence increase only need to be seen at spin_unlock
293 * time, and not at spin_lock time.
294 *
295 * Increasing the sequence after the spin_unlock would be
296 * unsafe because the kvm page fault could then establish the
297 * pte after kvm_unmap_hva returned, without noticing the page
298 * is going to be freed.
299 */
bc6678a3 300 idx = srcu_read_lock(&kvm->srcu);
e930bffe 301 spin_lock(&kvm->mmu_lock);
565f3be2 302
e930bffe 303 kvm->mmu_notifier_seq++;
a4ee1ca4 304 need_tlb_flush = kvm_unmap_hva(kvm, address) | kvm->tlbs_dirty;
e930bffe
AA
305 /* we've to flush the tlb before the pages can be freed */
306 if (need_tlb_flush)
307 kvm_flush_remote_tlbs(kvm);
308
565f3be2
TY
309 spin_unlock(&kvm->mmu_lock);
310 srcu_read_unlock(&kvm->srcu, idx);
e930bffe
AA
311}
312
3da0dd43
IE
313static void kvm_mmu_notifier_change_pte(struct mmu_notifier *mn,
314 struct mm_struct *mm,
315 unsigned long address,
316 pte_t pte)
317{
318 struct kvm *kvm = mmu_notifier_to_kvm(mn);
bc6678a3 319 int idx;
3da0dd43 320
bc6678a3 321 idx = srcu_read_lock(&kvm->srcu);
3da0dd43
IE
322 spin_lock(&kvm->mmu_lock);
323 kvm->mmu_notifier_seq++;
324 kvm_set_spte_hva(kvm, address, pte);
325 spin_unlock(&kvm->mmu_lock);
bc6678a3 326 srcu_read_unlock(&kvm->srcu, idx);
3da0dd43
IE
327}
328
e930bffe
AA
329static void kvm_mmu_notifier_invalidate_range_start(struct mmu_notifier *mn,
330 struct mm_struct *mm,
331 unsigned long start,
332 unsigned long end)
333{
334 struct kvm *kvm = mmu_notifier_to_kvm(mn);
bc6678a3 335 int need_tlb_flush = 0, idx;
e930bffe 336
bc6678a3 337 idx = srcu_read_lock(&kvm->srcu);
e930bffe
AA
338 spin_lock(&kvm->mmu_lock);
339 /*
340 * The count increase must become visible at unlock time as no
341 * spte can be established without taking the mmu_lock and
342 * count is also read inside the mmu_lock critical section.
343 */
344 kvm->mmu_notifier_count++;
b3ae2096 345 need_tlb_flush = kvm_unmap_hva_range(kvm, start, end);
a4ee1ca4 346 need_tlb_flush |= kvm->tlbs_dirty;
e930bffe
AA
347 /* we've to flush the tlb before the pages can be freed */
348 if (need_tlb_flush)
349 kvm_flush_remote_tlbs(kvm);
565f3be2
TY
350
351 spin_unlock(&kvm->mmu_lock);
352 srcu_read_unlock(&kvm->srcu, idx);
e930bffe
AA
353}
354
355static void kvm_mmu_notifier_invalidate_range_end(struct mmu_notifier *mn,
356 struct mm_struct *mm,
357 unsigned long start,
358 unsigned long end)
359{
360 struct kvm *kvm = mmu_notifier_to_kvm(mn);
361
362 spin_lock(&kvm->mmu_lock);
363 /*
364 * This sequence increase will notify the kvm page fault that
365 * the page that is going to be mapped in the spte could have
366 * been freed.
367 */
368 kvm->mmu_notifier_seq++;
a355aa54 369 smp_wmb();
e930bffe
AA
370 /*
371 * The above sequence increase must be visible before the
a355aa54
PM
372 * below count decrease, which is ensured by the smp_wmb above
373 * in conjunction with the smp_rmb in mmu_notifier_retry().
e930bffe
AA
374 */
375 kvm->mmu_notifier_count--;
376 spin_unlock(&kvm->mmu_lock);
377
378 BUG_ON(kvm->mmu_notifier_count < 0);
379}
380
381static int kvm_mmu_notifier_clear_flush_young(struct mmu_notifier *mn,
382 struct mm_struct *mm,
383 unsigned long address)
384{
385 struct kvm *kvm = mmu_notifier_to_kvm(mn);
bc6678a3 386 int young, idx;
e930bffe 387
bc6678a3 388 idx = srcu_read_lock(&kvm->srcu);
e930bffe 389 spin_lock(&kvm->mmu_lock);
e930bffe 390
565f3be2 391 young = kvm_age_hva(kvm, address);
e930bffe
AA
392 if (young)
393 kvm_flush_remote_tlbs(kvm);
394
565f3be2
TY
395 spin_unlock(&kvm->mmu_lock);
396 srcu_read_unlock(&kvm->srcu, idx);
397
e930bffe
AA
398 return young;
399}
400
8ee53820
AA
401static int kvm_mmu_notifier_test_young(struct mmu_notifier *mn,
402 struct mm_struct *mm,
403 unsigned long address)
404{
405 struct kvm *kvm = mmu_notifier_to_kvm(mn);
406 int young, idx;
407
408 idx = srcu_read_lock(&kvm->srcu);
409 spin_lock(&kvm->mmu_lock);
410 young = kvm_test_age_hva(kvm, address);
411 spin_unlock(&kvm->mmu_lock);
412 srcu_read_unlock(&kvm->srcu, idx);
413
414 return young;
415}
416
85db06e5
MT
417static void kvm_mmu_notifier_release(struct mmu_notifier *mn,
418 struct mm_struct *mm)
419{
420 struct kvm *kvm = mmu_notifier_to_kvm(mn);
eda2beda
LJ
421 int idx;
422
423 idx = srcu_read_lock(&kvm->srcu);
2df72e9b 424 kvm_arch_flush_shadow_all(kvm);
eda2beda 425 srcu_read_unlock(&kvm->srcu, idx);
85db06e5
MT
426}
427
e930bffe
AA
428static const struct mmu_notifier_ops kvm_mmu_notifier_ops = {
429 .invalidate_page = kvm_mmu_notifier_invalidate_page,
430 .invalidate_range_start = kvm_mmu_notifier_invalidate_range_start,
431 .invalidate_range_end = kvm_mmu_notifier_invalidate_range_end,
432 .clear_flush_young = kvm_mmu_notifier_clear_flush_young,
8ee53820 433 .test_young = kvm_mmu_notifier_test_young,
3da0dd43 434 .change_pte = kvm_mmu_notifier_change_pte,
85db06e5 435 .release = kvm_mmu_notifier_release,
e930bffe 436};
4c07b0a4
AK
437
438static int kvm_init_mmu_notifier(struct kvm *kvm)
439{
440 kvm->mmu_notifier.ops = &kvm_mmu_notifier_ops;
441 return mmu_notifier_register(&kvm->mmu_notifier, current->mm);
442}
443
444#else /* !(CONFIG_MMU_NOTIFIER && KVM_ARCH_WANT_MMU_NOTIFIER) */
445
446static int kvm_init_mmu_notifier(struct kvm *kvm)
447{
448 return 0;
449}
450
e930bffe
AA
451#endif /* CONFIG_MMU_NOTIFIER && KVM_ARCH_WANT_MMU_NOTIFIER */
452
bf3e05bc
XG
453static void kvm_init_memslots_id(struct kvm *kvm)
454{
455 int i;
456 struct kvm_memslots *slots = kvm->memslots;
457
458 for (i = 0; i < KVM_MEM_SLOTS_NUM; i++)
f85e2cb5 459 slots->id_to_index[i] = slots->memslots[i].id = i;
bf3e05bc
XG
460}
461
e08b9637 462static struct kvm *kvm_create_vm(unsigned long type)
6aa8b732 463{
d89f5eff
JK
464 int r, i;
465 struct kvm *kvm = kvm_arch_alloc_vm();
6aa8b732 466
d89f5eff
JK
467 if (!kvm)
468 return ERR_PTR(-ENOMEM);
469
e08b9637 470 r = kvm_arch_init_vm(kvm, type);
d89f5eff
JK
471 if (r)
472 goto out_err_nodisable;
10474ae8
AG
473
474 r = hardware_enable_all();
475 if (r)
476 goto out_err_nodisable;
477
75858a84
AK
478#ifdef CONFIG_HAVE_KVM_IRQCHIP
479 INIT_HLIST_HEAD(&kvm->mask_notifier_list);
136bdfee 480 INIT_HLIST_HEAD(&kvm->irq_ack_notifier_list);
75858a84 481#endif
6aa8b732 482
1e702d9a
AW
483 BUILD_BUG_ON(KVM_MEM_SLOTS_NUM > SHRT_MAX);
484
46a26bf5
MT
485 r = -ENOMEM;
486 kvm->memslots = kzalloc(sizeof(struct kvm_memslots), GFP_KERNEL);
487 if (!kvm->memslots)
57e7fbee 488 goto out_err_nosrcu;
bf3e05bc 489 kvm_init_memslots_id(kvm);
bc6678a3 490 if (init_srcu_struct(&kvm->srcu))
57e7fbee 491 goto out_err_nosrcu;
e93f8a0f
MT
492 for (i = 0; i < KVM_NR_BUSES; i++) {
493 kvm->buses[i] = kzalloc(sizeof(struct kvm_io_bus),
494 GFP_KERNEL);
57e7fbee 495 if (!kvm->buses[i])
e93f8a0f 496 goto out_err;
e93f8a0f 497 }
e930bffe 498
74b5c5bf 499 spin_lock_init(&kvm->mmu_lock);
6d4e4c4f
AK
500 kvm->mm = current->mm;
501 atomic_inc(&kvm->mm->mm_count);
d34e6b17 502 kvm_eventfd_init(kvm);
11ec2804 503 mutex_init(&kvm->lock);
60eead79 504 mutex_init(&kvm->irq_lock);
79fac95e 505 mutex_init(&kvm->slots_lock);
d39f13b0 506 atomic_set(&kvm->users_count, 1);
74b5c5bf
MW
507
508 r = kvm_init_mmu_notifier(kvm);
509 if (r)
510 goto out_err;
511
e935b837 512 raw_spin_lock(&kvm_lock);
5e58cfe4 513 list_add(&kvm->vm_list, &vm_list);
e935b837 514 raw_spin_unlock(&kvm_lock);
d89f5eff 515
f17abe9a 516 return kvm;
10474ae8
AG
517
518out_err:
57e7fbee
JK
519 cleanup_srcu_struct(&kvm->srcu);
520out_err_nosrcu:
10474ae8
AG
521 hardware_disable_all();
522out_err_nodisable:
e93f8a0f
MT
523 for (i = 0; i < KVM_NR_BUSES; i++)
524 kfree(kvm->buses[i]);
46a26bf5 525 kfree(kvm->memslots);
d89f5eff 526 kvm_arch_free_vm(kvm);
10474ae8 527 return ERR_PTR(r);
f17abe9a
AK
528}
529
92eca8fa
TY
530/*
531 * Avoid using vmalloc for a small buffer.
532 * Should not be used when the size is statically known.
533 */
c1a7b32a 534void *kvm_kvzalloc(unsigned long size)
92eca8fa
TY
535{
536 if (size > PAGE_SIZE)
537 return vzalloc(size);
538 else
539 return kzalloc(size, GFP_KERNEL);
540}
541
c1a7b32a 542void kvm_kvfree(const void *addr)
92eca8fa
TY
543{
544 if (is_vmalloc_addr(addr))
545 vfree(addr);
546 else
547 kfree(addr);
548}
549
a36a57b1
TY
550static void kvm_destroy_dirty_bitmap(struct kvm_memory_slot *memslot)
551{
552 if (!memslot->dirty_bitmap)
553 return;
554
92eca8fa 555 kvm_kvfree(memslot->dirty_bitmap);
a36a57b1
TY
556 memslot->dirty_bitmap = NULL;
557}
558
6aa8b732
AK
559/*
560 * Free any memory in @free but not in @dont.
561 */
562static void kvm_free_physmem_slot(struct kvm_memory_slot *free,
563 struct kvm_memory_slot *dont)
564{
6aa8b732 565 if (!dont || free->dirty_bitmap != dont->dirty_bitmap)
a36a57b1 566 kvm_destroy_dirty_bitmap(free);
6aa8b732 567
db3fe4eb 568 kvm_arch_free_memslot(free, dont);
05da4558 569
6aa8b732 570 free->npages = 0;
6aa8b732
AK
571}
572
d19a9cd2 573void kvm_free_physmem(struct kvm *kvm)
6aa8b732 574{
46a26bf5 575 struct kvm_memslots *slots = kvm->memslots;
be6ba0f0 576 struct kvm_memory_slot *memslot;
46a26bf5 577
be6ba0f0
XG
578 kvm_for_each_memslot(memslot, slots)
579 kvm_free_physmem_slot(memslot, NULL);
6aa8b732 580
46a26bf5 581 kfree(kvm->memslots);
6aa8b732
AK
582}
583
f17abe9a
AK
584static void kvm_destroy_vm(struct kvm *kvm)
585{
e93f8a0f 586 int i;
6d4e4c4f
AK
587 struct mm_struct *mm = kvm->mm;
588
ad8ba2cd 589 kvm_arch_sync_events(kvm);
e935b837 590 raw_spin_lock(&kvm_lock);
133de902 591 list_del(&kvm->vm_list);
e935b837 592 raw_spin_unlock(&kvm_lock);
399ec807 593 kvm_free_irq_routing(kvm);
e93f8a0f
MT
594 for (i = 0; i < KVM_NR_BUSES; i++)
595 kvm_io_bus_destroy(kvm->buses[i]);
980da6ce 596 kvm_coalesced_mmio_free(kvm);
e930bffe
AA
597#if defined(CONFIG_MMU_NOTIFIER) && defined(KVM_ARCH_WANT_MMU_NOTIFIER)
598 mmu_notifier_unregister(&kvm->mmu_notifier, kvm->mm);
f00be0ca 599#else
2df72e9b 600 kvm_arch_flush_shadow_all(kvm);
5f94c174 601#endif
d19a9cd2 602 kvm_arch_destroy_vm(kvm);
d89f5eff
JK
603 kvm_free_physmem(kvm);
604 cleanup_srcu_struct(&kvm->srcu);
605 kvm_arch_free_vm(kvm);
10474ae8 606 hardware_disable_all();
6d4e4c4f 607 mmdrop(mm);
f17abe9a
AK
608}
609
d39f13b0
IE
610void kvm_get_kvm(struct kvm *kvm)
611{
612 atomic_inc(&kvm->users_count);
613}
614EXPORT_SYMBOL_GPL(kvm_get_kvm);
615
616void kvm_put_kvm(struct kvm *kvm)
617{
618 if (atomic_dec_and_test(&kvm->users_count))
619 kvm_destroy_vm(kvm);
620}
621EXPORT_SYMBOL_GPL(kvm_put_kvm);
622
623
f17abe9a
AK
624static int kvm_vm_release(struct inode *inode, struct file *filp)
625{
626 struct kvm *kvm = filp->private_data;
627
721eecbf
GH
628 kvm_irqfd_release(kvm);
629
d39f13b0 630 kvm_put_kvm(kvm);
6aa8b732
AK
631 return 0;
632}
633
515a0127
TY
634/*
635 * Allocation size is twice as large as the actual dirty bitmap size.
93474b25 636 * See x86's kvm_vm_ioctl_get_dirty_log() why this is needed.
515a0127 637 */
a36a57b1
TY
638static int kvm_create_dirty_bitmap(struct kvm_memory_slot *memslot)
639{
189a2f7b 640#ifndef CONFIG_S390
515a0127 641 unsigned long dirty_bytes = 2 * kvm_dirty_bitmap_bytes(memslot);
a36a57b1 642
92eca8fa 643 memslot->dirty_bitmap = kvm_kvzalloc(dirty_bytes);
a36a57b1
TY
644 if (!memslot->dirty_bitmap)
645 return -ENOMEM;
646
189a2f7b 647#endif /* !CONFIG_S390 */
a36a57b1
TY
648 return 0;
649}
650
bf3e05bc
XG
651static int cmp_memslot(const void *slot1, const void *slot2)
652{
653 struct kvm_memory_slot *s1, *s2;
654
655 s1 = (struct kvm_memory_slot *)slot1;
656 s2 = (struct kvm_memory_slot *)slot2;
657
658 if (s1->npages < s2->npages)
659 return 1;
660 if (s1->npages > s2->npages)
661 return -1;
662
663 return 0;
664}
665
666/*
667 * Sort the memslots base on its size, so the larger slots
668 * will get better fit.
669 */
670static void sort_memslots(struct kvm_memslots *slots)
671{
f85e2cb5
XG
672 int i;
673
bf3e05bc
XG
674 sort(slots->memslots, KVM_MEM_SLOTS_NUM,
675 sizeof(struct kvm_memory_slot), cmp_memslot, NULL);
f85e2cb5
XG
676
677 for (i = 0; i < KVM_MEM_SLOTS_NUM; i++)
678 slots->id_to_index[slots->memslots[i].id] = i;
bf3e05bc
XG
679}
680
116c14c0
AW
681void update_memslots(struct kvm_memslots *slots, struct kvm_memory_slot *new,
682 u64 last_generation)
be593d62
XG
683{
684 if (new) {
685 int id = new->id;
28a37544 686 struct kvm_memory_slot *old = id_to_memslot(slots, id);
bf3e05bc 687 unsigned long npages = old->npages;
be593d62 688
28a37544 689 *old = *new;
bf3e05bc
XG
690 if (new->npages != npages)
691 sort_memslots(slots);
be593d62
XG
692 }
693
116c14c0 694 slots->generation = last_generation + 1;
be593d62
XG
695}
696
a50d64d6
XG
697static int check_memory_region_flags(struct kvm_userspace_memory_region *mem)
698{
4d8b81ab
XG
699 u32 valid_flags = KVM_MEM_LOG_DIRTY_PAGES;
700
701#ifdef KVM_CAP_READONLY_MEM
702 valid_flags |= KVM_MEM_READONLY;
703#endif
704
705 if (mem->flags & ~valid_flags)
a50d64d6
XG
706 return -EINVAL;
707
708 return 0;
709}
710
7ec4fb44
GN
711static struct kvm_memslots *install_new_memslots(struct kvm *kvm,
712 struct kvm_memslots *slots, struct kvm_memory_slot *new)
713{
714 struct kvm_memslots *old_memslots = kvm->memslots;
715
716 update_memslots(slots, new, kvm->memslots->generation);
717 rcu_assign_pointer(kvm->memslots, slots);
718 synchronize_srcu_expedited(&kvm->srcu);
719 return old_memslots;
720}
721
6aa8b732
AK
722/*
723 * Allocate some memory and give it an address in the guest physical address
724 * space.
725 *
726 * Discontiguous memory is allowed, mostly for framebuffers.
f78e0e2e 727 *
10589a46 728 * Must be called holding mmap_sem for write.
6aa8b732 729 */
f78e0e2e 730int __kvm_set_memory_region(struct kvm *kvm,
47ae31e2 731 struct kvm_userspace_memory_region *mem)
6aa8b732 732{
8234b22e 733 int r;
6aa8b732 734 gfn_t base_gfn;
28bcb112 735 unsigned long npages;
a843fac2 736 struct kvm_memory_slot *slot;
6aa8b732 737 struct kvm_memory_slot old, new;
b7f69c55 738 struct kvm_memslots *slots = NULL, *old_memslots;
f64c0398 739 enum kvm_mr_change change;
6aa8b732 740
a50d64d6
XG
741 r = check_memory_region_flags(mem);
742 if (r)
743 goto out;
744
6aa8b732
AK
745 r = -EINVAL;
746 /* General sanity checks */
747 if (mem->memory_size & (PAGE_SIZE - 1))
748 goto out;
749 if (mem->guest_phys_addr & (PAGE_SIZE - 1))
750 goto out;
fa3d315a 751 /* We can read the guest memory with __xxx_user() later on. */
47ae31e2 752 if ((mem->slot < KVM_USER_MEM_SLOTS) &&
fa3d315a 753 ((mem->userspace_addr & (PAGE_SIZE - 1)) ||
9e3bb6b6
HC
754 !access_ok(VERIFY_WRITE,
755 (void __user *)(unsigned long)mem->userspace_addr,
756 mem->memory_size)))
78749809 757 goto out;
93a5cef0 758 if (mem->slot >= KVM_MEM_SLOTS_NUM)
6aa8b732
AK
759 goto out;
760 if (mem->guest_phys_addr + mem->memory_size < mem->guest_phys_addr)
761 goto out;
762
a843fac2 763 slot = id_to_memslot(kvm->memslots, mem->slot);
6aa8b732
AK
764 base_gfn = mem->guest_phys_addr >> PAGE_SHIFT;
765 npages = mem->memory_size >> PAGE_SHIFT;
766
660c22c4
TY
767 r = -EINVAL;
768 if (npages > KVM_MEM_MAX_NR_PAGES)
769 goto out;
770
6aa8b732
AK
771 if (!npages)
772 mem->flags &= ~KVM_MEM_LOG_DIRTY_PAGES;
773
a843fac2 774 new = old = *slot;
6aa8b732 775
e36d96f7 776 new.id = mem->slot;
6aa8b732
AK
777 new.base_gfn = base_gfn;
778 new.npages = npages;
779 new.flags = mem->flags;
780
6aa8b732 781 r = -EINVAL;
f64c0398
TY
782 if (npages) {
783 if (!old.npages)
784 change = KVM_MR_CREATE;
785 else { /* Modify an existing slot. */
786 if ((mem->userspace_addr != old.userspace_addr) ||
75d61fbc
TY
787 (npages != old.npages) ||
788 ((new.flags ^ old.flags) & KVM_MEM_READONLY))
f64c0398
TY
789 goto out;
790
791 if (base_gfn != old.base_gfn)
792 change = KVM_MR_MOVE;
793 else if (new.flags != old.flags)
794 change = KVM_MR_FLAGS_ONLY;
795 else { /* Nothing to change. */
796 r = 0;
797 goto out;
798 }
799 }
800 } else if (old.npages) {
801 change = KVM_MR_DELETE;
802 } else /* Modify a non-existent slot: disallowed. */
0ea75e1d 803 goto out;
6aa8b732 804
f64c0398 805 if ((change == KVM_MR_CREATE) || (change == KVM_MR_MOVE)) {
0a706bee
TY
806 /* Check for overlaps */
807 r = -EEXIST;
808 kvm_for_each_memslot(slot, kvm->memslots) {
a843fac2
TY
809 if ((slot->id >= KVM_USER_MEM_SLOTS) ||
810 (slot->id == mem->slot))
0a706bee
TY
811 continue;
812 if (!((base_gfn + npages <= slot->base_gfn) ||
813 (base_gfn >= slot->base_gfn + slot->npages)))
814 goto out;
815 }
6aa8b732 816 }
6aa8b732 817
6aa8b732
AK
818 /* Free page dirty bitmap if unneeded */
819 if (!(new.flags & KVM_MEM_LOG_DIRTY_PAGES))
8b6d44c7 820 new.dirty_bitmap = NULL;
6aa8b732
AK
821
822 r = -ENOMEM;
f64c0398 823 if (change == KVM_MR_CREATE) {
189a2f7b 824 new.userspace_addr = mem->userspace_addr;
d89cc617 825
db3fe4eb
TY
826 if (kvm_arch_create_memslot(&new, npages))
827 goto out_free;
6aa8b732 828 }
ec04b260 829
6aa8b732
AK
830 /* Allocate page dirty bitmap if needed */
831 if ((new.flags & KVM_MEM_LOG_DIRTY_PAGES) && !new.dirty_bitmap) {
a36a57b1 832 if (kvm_create_dirty_bitmap(&new) < 0)
f78e0e2e 833 goto out_free;
6aa8b732
AK
834 }
835
f64c0398 836 if ((change == KVM_MR_DELETE) || (change == KVM_MR_MOVE)) {
bc6678a3 837 r = -ENOMEM;
6da64fdb
TM
838 slots = kmemdup(kvm->memslots, sizeof(struct kvm_memslots),
839 GFP_KERNEL);
bc6678a3
MT
840 if (!slots)
841 goto out_free;
28a37544
XG
842 slot = id_to_memslot(slots, mem->slot);
843 slot->flags |= KVM_MEMSLOT_INVALID;
844
7ec4fb44 845 old_memslots = install_new_memslots(kvm, slots, NULL);
bc6678a3 846
e40f193f
AW
847 /* slot was deleted or moved, clear iommu mapping */
848 kvm_iommu_unmap_pages(kvm, &old);
12d6e753
MT
849 /* From this point no new shadow pages pointing to a deleted,
850 * or moved, memslot will be created.
bc6678a3
MT
851 *
852 * validation of sp->gfn happens in:
853 * - gfn_to_hva (kvm_read_guest, gfn_to_pfn)
854 * - kvm_is_visible_gfn (mmu_check_roots)
855 */
2df72e9b 856 kvm_arch_flush_shadow_memslot(kvm, slot);
b7f69c55 857 slots = old_memslots;
bc6678a3 858 }
34d4cb8f 859
7b6195a9 860 r = kvm_arch_prepare_memory_region(kvm, &new, mem, change);
f7784b8e 861 if (r)
b7f69c55 862 goto out_slots;
f7784b8e 863
bc6678a3 864 r = -ENOMEM;
b7f69c55
AW
865 /*
866 * We can re-use the old_memslots from above, the only difference
867 * from the currently installed memslots is the invalid flag. This
868 * will get overwritten by update_memslots anyway.
869 */
870 if (!slots) {
871 slots = kmemdup(kvm->memslots, sizeof(struct kvm_memslots),
872 GFP_KERNEL);
873 if (!slots)
874 goto out_free;
875 }
bc6678a3 876
261874b0
AW
877 /*
878 * IOMMU mapping: New slots need to be mapped. Old slots need to be
75d61fbc
TY
879 * un-mapped and re-mapped if their base changes. Since base change
880 * unmapping is handled above with slot deletion, mapping alone is
881 * needed here. Anything else the iommu might care about for existing
882 * slots (size changes, userspace addr changes and read-only flag
883 * changes) is disallowed above, so any other attribute changes getting
884 * here can be skipped.
261874b0 885 */
75d61fbc
TY
886 if ((change == KVM_MR_CREATE) || (change == KVM_MR_MOVE)) {
887 r = kvm_iommu_map_pages(kvm, &new);
888 if (r)
889 goto out_slots;
e40f193f
AW
890 }
891
bc6678a3 892 /* actual memory is freed via old in kvm_free_physmem_slot below */
f64c0398 893 if (change == KVM_MR_DELETE) {
bc6678a3 894 new.dirty_bitmap = NULL;
db3fe4eb 895 memset(&new.arch, 0, sizeof(new.arch));
bc6678a3
MT
896 }
897
7ec4fb44 898 old_memslots = install_new_memslots(kvm, slots, &new);
3ad82a7e 899
8482644a 900 kvm_arch_commit_memory_region(kvm, mem, &old, change);
82ce2c96 901
bc6678a3
MT
902 kvm_free_physmem_slot(&old, &new);
903 kfree(old_memslots);
904
6aa8b732
AK
905 return 0;
906
e40f193f
AW
907out_slots:
908 kfree(slots);
f78e0e2e 909out_free:
6aa8b732
AK
910 kvm_free_physmem_slot(&new, &old);
911out:
912 return r;
210c7c4d 913}
f78e0e2e
SY
914EXPORT_SYMBOL_GPL(__kvm_set_memory_region);
915
916int kvm_set_memory_region(struct kvm *kvm,
47ae31e2 917 struct kvm_userspace_memory_region *mem)
f78e0e2e
SY
918{
919 int r;
920
79fac95e 921 mutex_lock(&kvm->slots_lock);
47ae31e2 922 r = __kvm_set_memory_region(kvm, mem);
79fac95e 923 mutex_unlock(&kvm->slots_lock);
f78e0e2e
SY
924 return r;
925}
210c7c4d
IE
926EXPORT_SYMBOL_GPL(kvm_set_memory_region);
927
1fe779f8 928int kvm_vm_ioctl_set_memory_region(struct kvm *kvm,
47ae31e2 929 struct kvm_userspace_memory_region *mem)
210c7c4d 930{
bbacc0c1 931 if (mem->slot >= KVM_USER_MEM_SLOTS)
e0d62c7f 932 return -EINVAL;
47ae31e2 933 return kvm_set_memory_region(kvm, mem);
6aa8b732
AK
934}
935
5bb064dc
ZX
936int kvm_get_dirty_log(struct kvm *kvm,
937 struct kvm_dirty_log *log, int *is_dirty)
6aa8b732
AK
938{
939 struct kvm_memory_slot *memslot;
940 int r, i;
87bf6e7d 941 unsigned long n;
6aa8b732
AK
942 unsigned long any = 0;
943
6aa8b732 944 r = -EINVAL;
bbacc0c1 945 if (log->slot >= KVM_USER_MEM_SLOTS)
6aa8b732
AK
946 goto out;
947
28a37544 948 memslot = id_to_memslot(kvm->memslots, log->slot);
6aa8b732
AK
949 r = -ENOENT;
950 if (!memslot->dirty_bitmap)
951 goto out;
952
87bf6e7d 953 n = kvm_dirty_bitmap_bytes(memslot);
6aa8b732 954
cd1a4a98 955 for (i = 0; !any && i < n/sizeof(long); ++i)
6aa8b732
AK
956 any = memslot->dirty_bitmap[i];
957
958 r = -EFAULT;
959 if (copy_to_user(log->dirty_bitmap, memslot->dirty_bitmap, n))
960 goto out;
961
5bb064dc
ZX
962 if (any)
963 *is_dirty = 1;
6aa8b732
AK
964
965 r = 0;
6aa8b732 966out:
6aa8b732
AK
967 return r;
968}
969
db3fe4eb
TY
970bool kvm_largepages_enabled(void)
971{
972 return largepages_enabled;
973}
974
54dee993
MT
975void kvm_disable_largepages(void)
976{
977 largepages_enabled = false;
978}
979EXPORT_SYMBOL_GPL(kvm_disable_largepages);
980
49c7754c
GN
981struct kvm_memory_slot *gfn_to_memslot(struct kvm *kvm, gfn_t gfn)
982{
983 return __gfn_to_memslot(kvm_memslots(kvm), gfn);
984}
a1f4d395 985EXPORT_SYMBOL_GPL(gfn_to_memslot);
6aa8b732 986
e0d62c7f
IE
987int kvm_is_visible_gfn(struct kvm *kvm, gfn_t gfn)
988{
bf3e05bc 989 struct kvm_memory_slot *memslot = gfn_to_memslot(kvm, gfn);
e0d62c7f 990
bbacc0c1 991 if (!memslot || memslot->id >= KVM_USER_MEM_SLOTS ||
bf3e05bc
XG
992 memslot->flags & KVM_MEMSLOT_INVALID)
993 return 0;
e0d62c7f 994
bf3e05bc 995 return 1;
e0d62c7f
IE
996}
997EXPORT_SYMBOL_GPL(kvm_is_visible_gfn);
998
8f0b1ab6
JR
999unsigned long kvm_host_page_size(struct kvm *kvm, gfn_t gfn)
1000{
1001 struct vm_area_struct *vma;
1002 unsigned long addr, size;
1003
1004 size = PAGE_SIZE;
1005
1006 addr = gfn_to_hva(kvm, gfn);
1007 if (kvm_is_error_hva(addr))
1008 return PAGE_SIZE;
1009
1010 down_read(&current->mm->mmap_sem);
1011 vma = find_vma(current->mm, addr);
1012 if (!vma)
1013 goto out;
1014
1015 size = vma_kernel_pagesize(vma);
1016
1017out:
1018 up_read(&current->mm->mmap_sem);
1019
1020 return size;
1021}
1022
4d8b81ab
XG
1023static bool memslot_is_readonly(struct kvm_memory_slot *slot)
1024{
1025 return slot->flags & KVM_MEM_READONLY;
1026}
1027
4d8b81ab
XG
1028static unsigned long __gfn_to_hva_many(struct kvm_memory_slot *slot, gfn_t gfn,
1029 gfn_t *nr_pages, bool write)
539cb660 1030{
bc6678a3 1031 if (!slot || slot->flags & KVM_MEMSLOT_INVALID)
ca3a490c 1032 return KVM_HVA_ERR_BAD;
48987781 1033
4d8b81ab
XG
1034 if (memslot_is_readonly(slot) && write)
1035 return KVM_HVA_ERR_RO_BAD;
48987781
XG
1036
1037 if (nr_pages)
1038 *nr_pages = slot->npages - (gfn - slot->base_gfn);
1039
4d8b81ab 1040 return __gfn_to_hva_memslot(slot, gfn);
539cb660 1041}
48987781 1042
4d8b81ab
XG
1043static unsigned long gfn_to_hva_many(struct kvm_memory_slot *slot, gfn_t gfn,
1044 gfn_t *nr_pages)
1045{
1046 return __gfn_to_hva_many(slot, gfn, nr_pages, true);
539cb660 1047}
48987781 1048
4d8b81ab
XG
1049unsigned long gfn_to_hva_memslot(struct kvm_memory_slot *slot,
1050 gfn_t gfn)
1051{
1052 return gfn_to_hva_many(slot, gfn, NULL);
1053}
1054EXPORT_SYMBOL_GPL(gfn_to_hva_memslot);
1055
48987781
XG
1056unsigned long gfn_to_hva(struct kvm *kvm, gfn_t gfn)
1057{
49c7754c 1058 return gfn_to_hva_many(gfn_to_memslot(kvm, gfn), gfn, NULL);
48987781 1059}
0d150298 1060EXPORT_SYMBOL_GPL(gfn_to_hva);
539cb660 1061
86ab8cff
XG
1062/*
1063 * The hva returned by this function is only allowed to be read.
1064 * It should pair with kvm_read_hva() or kvm_read_hva_atomic().
1065 */
1066static unsigned long gfn_to_hva_read(struct kvm *kvm, gfn_t gfn)
1067{
4d8b81ab 1068 return __gfn_to_hva_many(gfn_to_memslot(kvm, gfn), gfn, NULL, false);
86ab8cff
XG
1069}
1070
1071static int kvm_read_hva(void *data, void __user *hva, int len)
8030089f 1072{
86ab8cff
XG
1073 return __copy_from_user(data, hva, len);
1074}
1075
1076static int kvm_read_hva_atomic(void *data, void __user *hva, int len)
1077{
1078 return __copy_from_user_inatomic(data, hva, len);
8030089f
GN
1079}
1080
39369f7a 1081static int get_user_page_nowait(struct task_struct *tsk, struct mm_struct *mm,
0857b9e9
GN
1082 unsigned long start, int write, struct page **page)
1083{
1084 int flags = FOLL_TOUCH | FOLL_NOWAIT | FOLL_HWPOISON | FOLL_GET;
1085
1086 if (write)
1087 flags |= FOLL_WRITE;
1088
1089 return __get_user_pages(tsk, mm, start, 1, flags, page, NULL, NULL);
1090}
1091
fafc3dba
HY
1092static inline int check_user_page_hwpoison(unsigned long addr)
1093{
1094 int rc, flags = FOLL_TOUCH | FOLL_HWPOISON | FOLL_WRITE;
1095
1096 rc = __get_user_pages(current, current->mm, addr, 1,
1097 flags, NULL, NULL, NULL);
1098 return rc == -EHWPOISON;
1099}
1100
2fc84311
XG
1101/*
1102 * The atomic path to get the writable pfn which will be stored in @pfn,
1103 * true indicates success, otherwise false is returned.
1104 */
1105static bool hva_to_pfn_fast(unsigned long addr, bool atomic, bool *async,
1106 bool write_fault, bool *writable, pfn_t *pfn)
954bbbc2 1107{
8d4e1288 1108 struct page *page[1];
2fc84311 1109 int npages;
954bbbc2 1110
2fc84311
XG
1111 if (!(async || atomic))
1112 return false;
af585b92 1113
12ce13fe
XG
1114 /*
1115 * Fast pin a writable pfn only if it is a write fault request
1116 * or the caller allows to map a writable pfn for a read fault
1117 * request.
1118 */
1119 if (!(write_fault || writable))
1120 return false;
612819c3 1121
2fc84311
XG
1122 npages = __get_user_pages_fast(addr, 1, 1, page);
1123 if (npages == 1) {
1124 *pfn = page_to_pfn(page[0]);
612819c3 1125
2fc84311
XG
1126 if (writable)
1127 *writable = true;
1128 return true;
1129 }
af585b92 1130
2fc84311
XG
1131 return false;
1132}
612819c3 1133
2fc84311
XG
1134/*
1135 * The slow path to get the pfn of the specified host virtual address,
1136 * 1 indicates success, -errno is returned if error is detected.
1137 */
1138static int hva_to_pfn_slow(unsigned long addr, bool *async, bool write_fault,
1139 bool *writable, pfn_t *pfn)
1140{
1141 struct page *page[1];
1142 int npages = 0;
612819c3 1143
2fc84311
XG
1144 might_sleep();
1145
1146 if (writable)
1147 *writable = write_fault;
1148
1149 if (async) {
1150 down_read(&current->mm->mmap_sem);
1151 npages = get_user_page_nowait(current, current->mm,
1152 addr, write_fault, page);
1153 up_read(&current->mm->mmap_sem);
1154 } else
1155 npages = get_user_pages_fast(addr, 1, write_fault,
1156 page);
1157 if (npages != 1)
1158 return npages;
1159
1160 /* map read fault as writable if possible */
12ce13fe 1161 if (unlikely(!write_fault) && writable) {
2fc84311
XG
1162 struct page *wpage[1];
1163
1164 npages = __get_user_pages_fast(addr, 1, 1, wpage);
1165 if (npages == 1) {
1166 *writable = true;
1167 put_page(page[0]);
1168 page[0] = wpage[0];
612819c3 1169 }
2fc84311
XG
1170
1171 npages = 1;
887c08ac 1172 }
2fc84311
XG
1173 *pfn = page_to_pfn(page[0]);
1174 return npages;
1175}
539cb660 1176
4d8b81ab
XG
1177static bool vma_is_valid(struct vm_area_struct *vma, bool write_fault)
1178{
1179 if (unlikely(!(vma->vm_flags & VM_READ)))
1180 return false;
2e2e3738 1181
4d8b81ab
XG
1182 if (write_fault && (unlikely(!(vma->vm_flags & VM_WRITE))))
1183 return false;
887c08ac 1184
4d8b81ab
XG
1185 return true;
1186}
bf998156 1187
12ce13fe
XG
1188/*
1189 * Pin guest page in memory and return its pfn.
1190 * @addr: host virtual address which maps memory to the guest
1191 * @atomic: whether this function can sleep
1192 * @async: whether this function need to wait IO complete if the
1193 * host page is not in the memory
1194 * @write_fault: whether we should get a writable host page
1195 * @writable: whether it allows to map a writable host page for !@write_fault
1196 *
1197 * The function will map a writable host page for these two cases:
1198 * 1): @write_fault = true
1199 * 2): @write_fault = false && @writable, @writable will tell the caller
1200 * whether the mapping is writable.
1201 */
2fc84311
XG
1202static pfn_t hva_to_pfn(unsigned long addr, bool atomic, bool *async,
1203 bool write_fault, bool *writable)
1204{
1205 struct vm_area_struct *vma;
1206 pfn_t pfn = 0;
1207 int npages;
2e2e3738 1208
2fc84311
XG
1209 /* we can do it either atomically or asynchronously, not both */
1210 BUG_ON(atomic && async);
8d4e1288 1211
2fc84311
XG
1212 if (hva_to_pfn_fast(addr, atomic, async, write_fault, writable, &pfn))
1213 return pfn;
1214
1215 if (atomic)
1216 return KVM_PFN_ERR_FAULT;
1217
1218 npages = hva_to_pfn_slow(addr, async, write_fault, writable, &pfn);
1219 if (npages == 1)
1220 return pfn;
8d4e1288 1221
2fc84311
XG
1222 down_read(&current->mm->mmap_sem);
1223 if (npages == -EHWPOISON ||
1224 (!async && check_user_page_hwpoison(addr))) {
1225 pfn = KVM_PFN_ERR_HWPOISON;
1226 goto exit;
1227 }
1228
1229 vma = find_vma_intersection(current->mm, addr, addr + 1);
1230
1231 if (vma == NULL)
1232 pfn = KVM_PFN_ERR_FAULT;
1233 else if ((vma->vm_flags & VM_PFNMAP)) {
1234 pfn = ((addr - vma->vm_start) >> PAGE_SHIFT) +
1235 vma->vm_pgoff;
1236 BUG_ON(!kvm_is_mmio_pfn(pfn));
1237 } else {
4d8b81ab 1238 if (async && vma_is_valid(vma, write_fault))
2fc84311
XG
1239 *async = true;
1240 pfn = KVM_PFN_ERR_FAULT;
1241 }
1242exit:
1243 up_read(&current->mm->mmap_sem);
2e2e3738 1244 return pfn;
35149e21
AL
1245}
1246
4d8b81ab
XG
1247static pfn_t
1248__gfn_to_pfn_memslot(struct kvm_memory_slot *slot, gfn_t gfn, bool atomic,
1249 bool *async, bool write_fault, bool *writable)
887c08ac 1250{
4d8b81ab
XG
1251 unsigned long addr = __gfn_to_hva_many(slot, gfn, NULL, write_fault);
1252
1253 if (addr == KVM_HVA_ERR_RO_BAD)
1254 return KVM_PFN_ERR_RO_FAULT;
1255
1256 if (kvm_is_error_hva(addr))
81c52c56 1257 return KVM_PFN_NOSLOT;
4d8b81ab
XG
1258
1259 /* Do not map writable pfn in the readonly memslot. */
1260 if (writable && memslot_is_readonly(slot)) {
1261 *writable = false;
1262 writable = NULL;
1263 }
1264
1265 return hva_to_pfn(addr, atomic, async, write_fault,
1266 writable);
887c08ac 1267}
887c08ac 1268
612819c3
MT
1269static pfn_t __gfn_to_pfn(struct kvm *kvm, gfn_t gfn, bool atomic, bool *async,
1270 bool write_fault, bool *writable)
506f0d6f 1271{
4d8b81ab 1272 struct kvm_memory_slot *slot;
506f0d6f 1273
af585b92
GN
1274 if (async)
1275 *async = false;
1276
4d8b81ab 1277 slot = gfn_to_memslot(kvm, gfn);
506f0d6f 1278
4d8b81ab
XG
1279 return __gfn_to_pfn_memslot(slot, gfn, atomic, async, write_fault,
1280 writable);
365fb3fd
XG
1281}
1282
1283pfn_t gfn_to_pfn_atomic(struct kvm *kvm, gfn_t gfn)
1284{
612819c3 1285 return __gfn_to_pfn(kvm, gfn, true, NULL, true, NULL);
365fb3fd
XG
1286}
1287EXPORT_SYMBOL_GPL(gfn_to_pfn_atomic);
1288
612819c3
MT
1289pfn_t gfn_to_pfn_async(struct kvm *kvm, gfn_t gfn, bool *async,
1290 bool write_fault, bool *writable)
af585b92 1291{
612819c3 1292 return __gfn_to_pfn(kvm, gfn, false, async, write_fault, writable);
af585b92
GN
1293}
1294EXPORT_SYMBOL_GPL(gfn_to_pfn_async);
1295
365fb3fd
XG
1296pfn_t gfn_to_pfn(struct kvm *kvm, gfn_t gfn)
1297{
612819c3 1298 return __gfn_to_pfn(kvm, gfn, false, NULL, true, NULL);
506f0d6f 1299}
35149e21
AL
1300EXPORT_SYMBOL_GPL(gfn_to_pfn);
1301
612819c3
MT
1302pfn_t gfn_to_pfn_prot(struct kvm *kvm, gfn_t gfn, bool write_fault,
1303 bool *writable)
1304{
1305 return __gfn_to_pfn(kvm, gfn, false, NULL, write_fault, writable);
1306}
1307EXPORT_SYMBOL_GPL(gfn_to_pfn_prot);
1308
d5661048 1309pfn_t gfn_to_pfn_memslot(struct kvm_memory_slot *slot, gfn_t gfn)
506f0d6f 1310{
4d8b81ab 1311 return __gfn_to_pfn_memslot(slot, gfn, false, NULL, true, NULL);
506f0d6f
MT
1312}
1313
037d92dc 1314pfn_t gfn_to_pfn_memslot_atomic(struct kvm_memory_slot *slot, gfn_t gfn)
506f0d6f 1315{
4d8b81ab 1316 return __gfn_to_pfn_memslot(slot, gfn, true, NULL, true, NULL);
506f0d6f 1317}
037d92dc 1318EXPORT_SYMBOL_GPL(gfn_to_pfn_memslot_atomic);
506f0d6f 1319
48987781
XG
1320int gfn_to_page_many_atomic(struct kvm *kvm, gfn_t gfn, struct page **pages,
1321 int nr_pages)
1322{
1323 unsigned long addr;
1324 gfn_t entry;
1325
49c7754c 1326 addr = gfn_to_hva_many(gfn_to_memslot(kvm, gfn), gfn, &entry);
48987781
XG
1327 if (kvm_is_error_hva(addr))
1328 return -1;
1329
1330 if (entry < nr_pages)
1331 return 0;
1332
1333 return __get_user_pages_fast(addr, nr_pages, 1, pages);
1334}
1335EXPORT_SYMBOL_GPL(gfn_to_page_many_atomic);
1336
a2766325
XG
1337static struct page *kvm_pfn_to_page(pfn_t pfn)
1338{
81c52c56 1339 if (is_error_noslot_pfn(pfn))
cb9aaa30 1340 return KVM_ERR_PTR_BAD_PAGE;
a2766325 1341
cb9aaa30
XG
1342 if (kvm_is_mmio_pfn(pfn)) {
1343 WARN_ON(1);
6cede2e6 1344 return KVM_ERR_PTR_BAD_PAGE;
cb9aaa30 1345 }
a2766325
XG
1346
1347 return pfn_to_page(pfn);
1348}
1349
35149e21
AL
1350struct page *gfn_to_page(struct kvm *kvm, gfn_t gfn)
1351{
2e2e3738
AL
1352 pfn_t pfn;
1353
1354 pfn = gfn_to_pfn(kvm, gfn);
2e2e3738 1355
a2766325 1356 return kvm_pfn_to_page(pfn);
954bbbc2 1357}
aab61cc0 1358
954bbbc2
AK
1359EXPORT_SYMBOL_GPL(gfn_to_page);
1360
b4231d61
IE
1361void kvm_release_page_clean(struct page *page)
1362{
32cad84f
XG
1363 WARN_ON(is_error_page(page));
1364
35149e21 1365 kvm_release_pfn_clean(page_to_pfn(page));
b4231d61
IE
1366}
1367EXPORT_SYMBOL_GPL(kvm_release_page_clean);
1368
35149e21
AL
1369void kvm_release_pfn_clean(pfn_t pfn)
1370{
81c52c56 1371 if (!is_error_noslot_pfn(pfn) && !kvm_is_mmio_pfn(pfn))
2e2e3738 1372 put_page(pfn_to_page(pfn));
35149e21
AL
1373}
1374EXPORT_SYMBOL_GPL(kvm_release_pfn_clean);
1375
b4231d61 1376void kvm_release_page_dirty(struct page *page)
8a7ae055 1377{
a2766325
XG
1378 WARN_ON(is_error_page(page));
1379
35149e21
AL
1380 kvm_release_pfn_dirty(page_to_pfn(page));
1381}
1382EXPORT_SYMBOL_GPL(kvm_release_page_dirty);
1383
1384void kvm_release_pfn_dirty(pfn_t pfn)
1385{
1386 kvm_set_pfn_dirty(pfn);
1387 kvm_release_pfn_clean(pfn);
1388}
1389EXPORT_SYMBOL_GPL(kvm_release_pfn_dirty);
1390
1391void kvm_set_page_dirty(struct page *page)
1392{
1393 kvm_set_pfn_dirty(page_to_pfn(page));
1394}
1395EXPORT_SYMBOL_GPL(kvm_set_page_dirty);
1396
1397void kvm_set_pfn_dirty(pfn_t pfn)
1398{
c77fb9dc 1399 if (!kvm_is_mmio_pfn(pfn)) {
2e2e3738
AL
1400 struct page *page = pfn_to_page(pfn);
1401 if (!PageReserved(page))
1402 SetPageDirty(page);
1403 }
8a7ae055 1404}
35149e21
AL
1405EXPORT_SYMBOL_GPL(kvm_set_pfn_dirty);
1406
1407void kvm_set_pfn_accessed(pfn_t pfn)
1408{
c77fb9dc 1409 if (!kvm_is_mmio_pfn(pfn))
2e2e3738 1410 mark_page_accessed(pfn_to_page(pfn));
35149e21
AL
1411}
1412EXPORT_SYMBOL_GPL(kvm_set_pfn_accessed);
1413
1414void kvm_get_pfn(pfn_t pfn)
1415{
c77fb9dc 1416 if (!kvm_is_mmio_pfn(pfn))
2e2e3738 1417 get_page(pfn_to_page(pfn));
35149e21
AL
1418}
1419EXPORT_SYMBOL_GPL(kvm_get_pfn);
8a7ae055 1420
195aefde
IE
1421static int next_segment(unsigned long len, int offset)
1422{
1423 if (len > PAGE_SIZE - offset)
1424 return PAGE_SIZE - offset;
1425 else
1426 return len;
1427}
1428
1429int kvm_read_guest_page(struct kvm *kvm, gfn_t gfn, void *data, int offset,
1430 int len)
1431{
e0506bcb
IE
1432 int r;
1433 unsigned long addr;
195aefde 1434
86ab8cff 1435 addr = gfn_to_hva_read(kvm, gfn);
e0506bcb
IE
1436 if (kvm_is_error_hva(addr))
1437 return -EFAULT;
86ab8cff 1438 r = kvm_read_hva(data, (void __user *)addr + offset, len);
e0506bcb 1439 if (r)
195aefde 1440 return -EFAULT;
195aefde
IE
1441 return 0;
1442}
1443EXPORT_SYMBOL_GPL(kvm_read_guest_page);
1444
1445int kvm_read_guest(struct kvm *kvm, gpa_t gpa, void *data, unsigned long len)
1446{
1447 gfn_t gfn = gpa >> PAGE_SHIFT;
1448 int seg;
1449 int offset = offset_in_page(gpa);
1450 int ret;
1451
1452 while ((seg = next_segment(len, offset)) != 0) {
1453 ret = kvm_read_guest_page(kvm, gfn, data, offset, seg);
1454 if (ret < 0)
1455 return ret;
1456 offset = 0;
1457 len -= seg;
1458 data += seg;
1459 ++gfn;
1460 }
1461 return 0;
1462}
1463EXPORT_SYMBOL_GPL(kvm_read_guest);
1464
7ec54588
MT
1465int kvm_read_guest_atomic(struct kvm *kvm, gpa_t gpa, void *data,
1466 unsigned long len)
1467{
1468 int r;
1469 unsigned long addr;
1470 gfn_t gfn = gpa >> PAGE_SHIFT;
1471 int offset = offset_in_page(gpa);
1472
86ab8cff 1473 addr = gfn_to_hva_read(kvm, gfn);
7ec54588
MT
1474 if (kvm_is_error_hva(addr))
1475 return -EFAULT;
0aac03f0 1476 pagefault_disable();
86ab8cff 1477 r = kvm_read_hva_atomic(data, (void __user *)addr + offset, len);
0aac03f0 1478 pagefault_enable();
7ec54588
MT
1479 if (r)
1480 return -EFAULT;
1481 return 0;
1482}
1483EXPORT_SYMBOL(kvm_read_guest_atomic);
1484
195aefde
IE
1485int kvm_write_guest_page(struct kvm *kvm, gfn_t gfn, const void *data,
1486 int offset, int len)
1487{
e0506bcb
IE
1488 int r;
1489 unsigned long addr;
195aefde 1490
e0506bcb
IE
1491 addr = gfn_to_hva(kvm, gfn);
1492 if (kvm_is_error_hva(addr))
1493 return -EFAULT;
8b0cedff 1494 r = __copy_to_user((void __user *)addr + offset, data, len);
e0506bcb 1495 if (r)
195aefde 1496 return -EFAULT;
195aefde
IE
1497 mark_page_dirty(kvm, gfn);
1498 return 0;
1499}
1500EXPORT_SYMBOL_GPL(kvm_write_guest_page);
1501
1502int kvm_write_guest(struct kvm *kvm, gpa_t gpa, const void *data,
1503 unsigned long len)
1504{
1505 gfn_t gfn = gpa >> PAGE_SHIFT;
1506 int seg;
1507 int offset = offset_in_page(gpa);
1508 int ret;
1509
1510 while ((seg = next_segment(len, offset)) != 0) {
1511 ret = kvm_write_guest_page(kvm, gfn, data, offset, seg);
1512 if (ret < 0)
1513 return ret;
1514 offset = 0;
1515 len -= seg;
1516 data += seg;
1517 ++gfn;
1518 }
1519 return 0;
1520}
1521
49c7754c
GN
1522int kvm_gfn_to_hva_cache_init(struct kvm *kvm, struct gfn_to_hva_cache *ghc,
1523 gpa_t gpa)
1524{
1525 struct kvm_memslots *slots = kvm_memslots(kvm);
1526 int offset = offset_in_page(gpa);
1527 gfn_t gfn = gpa >> PAGE_SHIFT;
1528
1529 ghc->gpa = gpa;
1530 ghc->generation = slots->generation;
9d4cba7f 1531 ghc->memslot = gfn_to_memslot(kvm, gfn);
49c7754c
GN
1532 ghc->hva = gfn_to_hva_many(ghc->memslot, gfn, NULL);
1533 if (!kvm_is_error_hva(ghc->hva))
1534 ghc->hva += offset;
1535 else
1536 return -EFAULT;
1537
1538 return 0;
1539}
1540EXPORT_SYMBOL_GPL(kvm_gfn_to_hva_cache_init);
1541
1542int kvm_write_guest_cached(struct kvm *kvm, struct gfn_to_hva_cache *ghc,
1543 void *data, unsigned long len)
1544{
1545 struct kvm_memslots *slots = kvm_memslots(kvm);
1546 int r;
1547
1548 if (slots->generation != ghc->generation)
1549 kvm_gfn_to_hva_cache_init(kvm, ghc, ghc->gpa);
1550
1551 if (kvm_is_error_hva(ghc->hva))
1552 return -EFAULT;
1553
8b0cedff 1554 r = __copy_to_user((void __user *)ghc->hva, data, len);
49c7754c
GN
1555 if (r)
1556 return -EFAULT;
1557 mark_page_dirty_in_slot(kvm, ghc->memslot, ghc->gpa >> PAGE_SHIFT);
1558
1559 return 0;
1560}
1561EXPORT_SYMBOL_GPL(kvm_write_guest_cached);
1562
e03b644f
GN
1563int kvm_read_guest_cached(struct kvm *kvm, struct gfn_to_hva_cache *ghc,
1564 void *data, unsigned long len)
1565{
1566 struct kvm_memslots *slots = kvm_memslots(kvm);
1567 int r;
1568
1569 if (slots->generation != ghc->generation)
1570 kvm_gfn_to_hva_cache_init(kvm, ghc, ghc->gpa);
1571
1572 if (kvm_is_error_hva(ghc->hva))
1573 return -EFAULT;
1574
1575 r = __copy_from_user(data, (void __user *)ghc->hva, len);
1576 if (r)
1577 return -EFAULT;
1578
1579 return 0;
1580}
1581EXPORT_SYMBOL_GPL(kvm_read_guest_cached);
1582
195aefde
IE
1583int kvm_clear_guest_page(struct kvm *kvm, gfn_t gfn, int offset, int len)
1584{
3bcc8a8c
HC
1585 return kvm_write_guest_page(kvm, gfn, (const void *) empty_zero_page,
1586 offset, len);
195aefde
IE
1587}
1588EXPORT_SYMBOL_GPL(kvm_clear_guest_page);
1589
1590int kvm_clear_guest(struct kvm *kvm, gpa_t gpa, unsigned long len)
1591{
1592 gfn_t gfn = gpa >> PAGE_SHIFT;
1593 int seg;
1594 int offset = offset_in_page(gpa);
1595 int ret;
1596
1597 while ((seg = next_segment(len, offset)) != 0) {
1598 ret = kvm_clear_guest_page(kvm, gfn, offset, seg);
1599 if (ret < 0)
1600 return ret;
1601 offset = 0;
1602 len -= seg;
1603 ++gfn;
1604 }
1605 return 0;
1606}
1607EXPORT_SYMBOL_GPL(kvm_clear_guest);
1608
49c7754c
GN
1609void mark_page_dirty_in_slot(struct kvm *kvm, struct kvm_memory_slot *memslot,
1610 gfn_t gfn)
6aa8b732 1611{
7e9d619d
RR
1612 if (memslot && memslot->dirty_bitmap) {
1613 unsigned long rel_gfn = gfn - memslot->base_gfn;
6aa8b732 1614
b74ca3b3 1615 set_bit_le(rel_gfn, memslot->dirty_bitmap);
6aa8b732
AK
1616 }
1617}
1618
49c7754c
GN
1619void mark_page_dirty(struct kvm *kvm, gfn_t gfn)
1620{
1621 struct kvm_memory_slot *memslot;
1622
1623 memslot = gfn_to_memslot(kvm, gfn);
1624 mark_page_dirty_in_slot(kvm, memslot, gfn);
1625}
1626
b6958ce4
ED
1627/*
1628 * The vCPU has executed a HLT instruction with in-kernel mode enabled.
1629 */
8776e519 1630void kvm_vcpu_block(struct kvm_vcpu *vcpu)
d3bef15f 1631{
e5c239cf
MT
1632 DEFINE_WAIT(wait);
1633
1634 for (;;) {
1635 prepare_to_wait(&vcpu->wq, &wait, TASK_INTERRUPTIBLE);
1636
a1b37100 1637 if (kvm_arch_vcpu_runnable(vcpu)) {
a8eeb04a 1638 kvm_make_request(KVM_REQ_UNHALT, vcpu);
e5c239cf 1639 break;
d7690175 1640 }
09cec754
GN
1641 if (kvm_cpu_has_pending_timer(vcpu))
1642 break;
e5c239cf
MT
1643 if (signal_pending(current))
1644 break;
1645
b6958ce4 1646 schedule();
b6958ce4 1647 }
d3bef15f 1648
e5c239cf 1649 finish_wait(&vcpu->wq, &wait);
b6958ce4
ED
1650}
1651
8c84780d 1652#ifndef CONFIG_S390
b6d33834
CD
1653/*
1654 * Kick a sleeping VCPU, or a guest VCPU in guest mode, into host kernel mode.
1655 */
1656void kvm_vcpu_kick(struct kvm_vcpu *vcpu)
1657{
1658 int me;
1659 int cpu = vcpu->cpu;
1660 wait_queue_head_t *wqp;
1661
1662 wqp = kvm_arch_vcpu_wq(vcpu);
1663 if (waitqueue_active(wqp)) {
1664 wake_up_interruptible(wqp);
1665 ++vcpu->stat.halt_wakeup;
1666 }
1667
1668 me = get_cpu();
1669 if (cpu != me && (unsigned)cpu < nr_cpu_ids && cpu_online(cpu))
1670 if (kvm_arch_vcpu_should_kick(vcpu))
1671 smp_send_reschedule(cpu);
1672 put_cpu();
1673}
a20ed54d 1674EXPORT_SYMBOL_GPL(kvm_vcpu_kick);
8c84780d 1675#endif /* !CONFIG_S390 */
b6d33834 1676
6aa8b732
AK
1677void kvm_resched(struct kvm_vcpu *vcpu)
1678{
3fca0365
YD
1679 if (!need_resched())
1680 return;
6aa8b732 1681 cond_resched();
6aa8b732
AK
1682}
1683EXPORT_SYMBOL_GPL(kvm_resched);
1684
41628d33
KW
1685bool kvm_vcpu_yield_to(struct kvm_vcpu *target)
1686{
1687 struct pid *pid;
1688 struct task_struct *task = NULL;
c45c528e 1689 bool ret = false;
41628d33
KW
1690
1691 rcu_read_lock();
1692 pid = rcu_dereference(target->pid);
1693 if (pid)
1694 task = get_pid_task(target->pid, PIDTYPE_PID);
1695 rcu_read_unlock();
1696 if (!task)
c45c528e 1697 return ret;
41628d33
KW
1698 if (task->flags & PF_VCPU) {
1699 put_task_struct(task);
c45c528e 1700 return ret;
41628d33 1701 }
c45c528e 1702 ret = yield_to(task, 1);
41628d33 1703 put_task_struct(task);
c45c528e
R
1704
1705 return ret;
41628d33
KW
1706}
1707EXPORT_SYMBOL_GPL(kvm_vcpu_yield_to);
1708
06e48c51
R
1709#ifdef CONFIG_HAVE_KVM_CPU_RELAX_INTERCEPT
1710/*
1711 * Helper that checks whether a VCPU is eligible for directed yield.
1712 * Most eligible candidate to yield is decided by following heuristics:
1713 *
1714 * (a) VCPU which has not done pl-exit or cpu relax intercepted recently
1715 * (preempted lock holder), indicated by @in_spin_loop.
1716 * Set at the beiginning and cleared at the end of interception/PLE handler.
1717 *
1718 * (b) VCPU which has done pl-exit/ cpu relax intercepted but did not get
1719 * chance last time (mostly it has become eligible now since we have probably
1720 * yielded to lockholder in last iteration. This is done by toggling
1721 * @dy_eligible each time a VCPU checked for eligibility.)
1722 *
1723 * Yielding to a recently pl-exited/cpu relax intercepted VCPU before yielding
1724 * to preempted lock-holder could result in wrong VCPU selection and CPU
1725 * burning. Giving priority for a potential lock-holder increases lock
1726 * progress.
1727 *
1728 * Since algorithm is based on heuristics, accessing another VCPU data without
1729 * locking does not harm. It may result in trying to yield to same VCPU, fail
1730 * and continue with next VCPU and so on.
1731 */
1732bool kvm_vcpu_eligible_for_directed_yield(struct kvm_vcpu *vcpu)
1733{
1734 bool eligible;
1735
1736 eligible = !vcpu->spin_loop.in_spin_loop ||
1737 (vcpu->spin_loop.in_spin_loop &&
1738 vcpu->spin_loop.dy_eligible);
1739
1740 if (vcpu->spin_loop.in_spin_loop)
1741 kvm_vcpu_set_dy_eligible(vcpu, !vcpu->spin_loop.dy_eligible);
1742
1743 return eligible;
1744}
1745#endif
c45c528e 1746
217ece61 1747void kvm_vcpu_on_spin(struct kvm_vcpu *me)
d255f4f2 1748{
217ece61
RR
1749 struct kvm *kvm = me->kvm;
1750 struct kvm_vcpu *vcpu;
1751 int last_boosted_vcpu = me->kvm->last_boosted_vcpu;
1752 int yielded = 0;
c45c528e 1753 int try = 3;
217ece61
RR
1754 int pass;
1755 int i;
d255f4f2 1756
4c088493 1757 kvm_vcpu_set_in_spin_loop(me, true);
217ece61
RR
1758 /*
1759 * We boost the priority of a VCPU that is runnable but not
1760 * currently running, because it got preempted by something
1761 * else and called schedule in __vcpu_run. Hopefully that
1762 * VCPU is holding the lock that we need and will release it.
1763 * We approximate round-robin by starting at the last boosted VCPU.
1764 */
c45c528e 1765 for (pass = 0; pass < 2 && !yielded && try; pass++) {
217ece61 1766 kvm_for_each_vcpu(i, vcpu, kvm) {
5cfc2aab 1767 if (!pass && i <= last_boosted_vcpu) {
217ece61
RR
1768 i = last_boosted_vcpu;
1769 continue;
1770 } else if (pass && i > last_boosted_vcpu)
1771 break;
7bc7ae25
R
1772 if (!ACCESS_ONCE(vcpu->preempted))
1773 continue;
217ece61
RR
1774 if (vcpu == me)
1775 continue;
1776 if (waitqueue_active(&vcpu->wq))
1777 continue;
06e48c51
R
1778 if (!kvm_vcpu_eligible_for_directed_yield(vcpu))
1779 continue;
c45c528e
R
1780
1781 yielded = kvm_vcpu_yield_to(vcpu);
1782 if (yielded > 0) {
217ece61 1783 kvm->last_boosted_vcpu = i;
217ece61 1784 break;
c45c528e
R
1785 } else if (yielded < 0) {
1786 try--;
1787 if (!try)
1788 break;
217ece61 1789 }
217ece61
RR
1790 }
1791 }
4c088493 1792 kvm_vcpu_set_in_spin_loop(me, false);
06e48c51
R
1793
1794 /* Ensure vcpu is not eligible during next spinloop */
1795 kvm_vcpu_set_dy_eligible(me, false);
d255f4f2
ZE
1796}
1797EXPORT_SYMBOL_GPL(kvm_vcpu_on_spin);
1798
e4a533a4 1799static int kvm_vcpu_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
9a2bb7f4
AK
1800{
1801 struct kvm_vcpu *vcpu = vma->vm_file->private_data;
9a2bb7f4
AK
1802 struct page *page;
1803
e4a533a4 1804 if (vmf->pgoff == 0)
039576c0 1805 page = virt_to_page(vcpu->run);
09566765 1806#ifdef CONFIG_X86
e4a533a4 1807 else if (vmf->pgoff == KVM_PIO_PAGE_OFFSET)
ad312c7c 1808 page = virt_to_page(vcpu->arch.pio_data);
5f94c174
LV
1809#endif
1810#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
1811 else if (vmf->pgoff == KVM_COALESCED_MMIO_PAGE_OFFSET)
1812 page = virt_to_page(vcpu->kvm->coalesced_mmio_ring);
09566765 1813#endif
039576c0 1814 else
5b1c1493 1815 return kvm_arch_vcpu_fault(vcpu, vmf);
9a2bb7f4 1816 get_page(page);
e4a533a4 1817 vmf->page = page;
1818 return 0;
9a2bb7f4
AK
1819}
1820
f0f37e2f 1821static const struct vm_operations_struct kvm_vcpu_vm_ops = {
e4a533a4 1822 .fault = kvm_vcpu_fault,
9a2bb7f4
AK
1823};
1824
1825static int kvm_vcpu_mmap(struct file *file, struct vm_area_struct *vma)
1826{
1827 vma->vm_ops = &kvm_vcpu_vm_ops;
1828 return 0;
1829}
1830
bccf2150
AK
1831static int kvm_vcpu_release(struct inode *inode, struct file *filp)
1832{
1833 struct kvm_vcpu *vcpu = filp->private_data;
1834
66c0b394 1835 kvm_put_kvm(vcpu->kvm);
bccf2150
AK
1836 return 0;
1837}
1838
3d3aab1b 1839static struct file_operations kvm_vcpu_fops = {
bccf2150
AK
1840 .release = kvm_vcpu_release,
1841 .unlocked_ioctl = kvm_vcpu_ioctl,
1dda606c
AG
1842#ifdef CONFIG_COMPAT
1843 .compat_ioctl = kvm_vcpu_compat_ioctl,
1844#endif
9a2bb7f4 1845 .mmap = kvm_vcpu_mmap,
6038f373 1846 .llseek = noop_llseek,
bccf2150
AK
1847};
1848
1849/*
1850 * Allocates an inode for the vcpu.
1851 */
1852static int create_vcpu_fd(struct kvm_vcpu *vcpu)
1853{
628ff7c1 1854 return anon_inode_getfd("kvm-vcpu", &kvm_vcpu_fops, vcpu, O_RDWR);
bccf2150
AK
1855}
1856
c5ea7660
AK
1857/*
1858 * Creates some virtual cpus. Good luck creating more than one.
1859 */
73880c80 1860static int kvm_vm_ioctl_create_vcpu(struct kvm *kvm, u32 id)
c5ea7660
AK
1861{
1862 int r;
988a2cae 1863 struct kvm_vcpu *vcpu, *v;
c5ea7660 1864
73880c80 1865 vcpu = kvm_arch_vcpu_create(kvm, id);
fb3f0f51
RR
1866 if (IS_ERR(vcpu))
1867 return PTR_ERR(vcpu);
c5ea7660 1868
15ad7146
AK
1869 preempt_notifier_init(&vcpu->preempt_notifier, &kvm_preempt_ops);
1870
26e5215f
AK
1871 r = kvm_arch_vcpu_setup(vcpu);
1872 if (r)
d780592b 1873 goto vcpu_destroy;
26e5215f 1874
11ec2804 1875 mutex_lock(&kvm->lock);
3e515705
AK
1876 if (!kvm_vcpu_compatible(vcpu)) {
1877 r = -EINVAL;
1878 goto unlock_vcpu_destroy;
1879 }
73880c80
GN
1880 if (atomic_read(&kvm->online_vcpus) == KVM_MAX_VCPUS) {
1881 r = -EINVAL;
d780592b 1882 goto unlock_vcpu_destroy;
fb3f0f51 1883 }
73880c80 1884
988a2cae
GN
1885 kvm_for_each_vcpu(r, v, kvm)
1886 if (v->vcpu_id == id) {
73880c80 1887 r = -EEXIST;
d780592b 1888 goto unlock_vcpu_destroy;
73880c80
GN
1889 }
1890
1891 BUG_ON(kvm->vcpus[atomic_read(&kvm->online_vcpus)]);
c5ea7660 1892
fb3f0f51 1893 /* Now it's all set up, let userspace reach it */
66c0b394 1894 kvm_get_kvm(kvm);
bccf2150 1895 r = create_vcpu_fd(vcpu);
73880c80
GN
1896 if (r < 0) {
1897 kvm_put_kvm(kvm);
d780592b 1898 goto unlock_vcpu_destroy;
73880c80
GN
1899 }
1900
1901 kvm->vcpus[atomic_read(&kvm->online_vcpus)] = vcpu;
1902 smp_wmb();
1903 atomic_inc(&kvm->online_vcpus);
1904
73880c80 1905 mutex_unlock(&kvm->lock);
42897d86 1906 kvm_arch_vcpu_postcreate(vcpu);
fb3f0f51 1907 return r;
39c3b86e 1908
d780592b 1909unlock_vcpu_destroy:
7d8fece6 1910 mutex_unlock(&kvm->lock);
d780592b 1911vcpu_destroy:
d40ccc62 1912 kvm_arch_vcpu_destroy(vcpu);
c5ea7660
AK
1913 return r;
1914}
1915
1961d276
AK
1916static int kvm_vcpu_ioctl_set_sigmask(struct kvm_vcpu *vcpu, sigset_t *sigset)
1917{
1918 if (sigset) {
1919 sigdelsetmask(sigset, sigmask(SIGKILL)|sigmask(SIGSTOP));
1920 vcpu->sigset_active = 1;
1921 vcpu->sigset = *sigset;
1922 } else
1923 vcpu->sigset_active = 0;
1924 return 0;
1925}
1926
bccf2150
AK
1927static long kvm_vcpu_ioctl(struct file *filp,
1928 unsigned int ioctl, unsigned long arg)
6aa8b732 1929{
bccf2150 1930 struct kvm_vcpu *vcpu = filp->private_data;
2f366987 1931 void __user *argp = (void __user *)arg;
313a3dc7 1932 int r;
fa3795a7
DH
1933 struct kvm_fpu *fpu = NULL;
1934 struct kvm_sregs *kvm_sregs = NULL;
6aa8b732 1935
6d4e4c4f
AK
1936 if (vcpu->kvm->mm != current->mm)
1937 return -EIO;
2122ff5e
AK
1938
1939#if defined(CONFIG_S390) || defined(CONFIG_PPC)
1940 /*
1941 * Special cases: vcpu ioctls that are asynchronous to vcpu execution,
1942 * so vcpu_load() would break it.
1943 */
1944 if (ioctl == KVM_S390_INTERRUPT || ioctl == KVM_INTERRUPT)
1945 return kvm_arch_vcpu_ioctl(filp, ioctl, arg);
1946#endif
1947
1948
9fc77441
MT
1949 r = vcpu_load(vcpu);
1950 if (r)
1951 return r;
6aa8b732 1952 switch (ioctl) {
9a2bb7f4 1953 case KVM_RUN:
f0fe5108
AK
1954 r = -EINVAL;
1955 if (arg)
1956 goto out;
b6c7a5dc 1957 r = kvm_arch_vcpu_ioctl_run(vcpu, vcpu->run);
64be5007 1958 trace_kvm_userspace_exit(vcpu->run->exit_reason, r);
6aa8b732 1959 break;
6aa8b732 1960 case KVM_GET_REGS: {
3e4bb3ac 1961 struct kvm_regs *kvm_regs;
6aa8b732 1962
3e4bb3ac
XZ
1963 r = -ENOMEM;
1964 kvm_regs = kzalloc(sizeof(struct kvm_regs), GFP_KERNEL);
1965 if (!kvm_regs)
6aa8b732 1966 goto out;
3e4bb3ac
XZ
1967 r = kvm_arch_vcpu_ioctl_get_regs(vcpu, kvm_regs);
1968 if (r)
1969 goto out_free1;
6aa8b732 1970 r = -EFAULT;
3e4bb3ac
XZ
1971 if (copy_to_user(argp, kvm_regs, sizeof(struct kvm_regs)))
1972 goto out_free1;
6aa8b732 1973 r = 0;
3e4bb3ac
XZ
1974out_free1:
1975 kfree(kvm_regs);
6aa8b732
AK
1976 break;
1977 }
1978 case KVM_SET_REGS: {
3e4bb3ac 1979 struct kvm_regs *kvm_regs;
6aa8b732 1980
3e4bb3ac 1981 r = -ENOMEM;
ff5c2c03
SL
1982 kvm_regs = memdup_user(argp, sizeof(*kvm_regs));
1983 if (IS_ERR(kvm_regs)) {
1984 r = PTR_ERR(kvm_regs);
6aa8b732 1985 goto out;
ff5c2c03 1986 }
3e4bb3ac 1987 r = kvm_arch_vcpu_ioctl_set_regs(vcpu, kvm_regs);
3e4bb3ac 1988 kfree(kvm_regs);
6aa8b732
AK
1989 break;
1990 }
1991 case KVM_GET_SREGS: {
fa3795a7
DH
1992 kvm_sregs = kzalloc(sizeof(struct kvm_sregs), GFP_KERNEL);
1993 r = -ENOMEM;
1994 if (!kvm_sregs)
1995 goto out;
1996 r = kvm_arch_vcpu_ioctl_get_sregs(vcpu, kvm_sregs);
6aa8b732
AK
1997 if (r)
1998 goto out;
1999 r = -EFAULT;
fa3795a7 2000 if (copy_to_user(argp, kvm_sregs, sizeof(struct kvm_sregs)))
6aa8b732
AK
2001 goto out;
2002 r = 0;
2003 break;
2004 }
2005 case KVM_SET_SREGS: {
ff5c2c03
SL
2006 kvm_sregs = memdup_user(argp, sizeof(*kvm_sregs));
2007 if (IS_ERR(kvm_sregs)) {
2008 r = PTR_ERR(kvm_sregs);
18595411 2009 kvm_sregs = NULL;
6aa8b732 2010 goto out;
ff5c2c03 2011 }
fa3795a7 2012 r = kvm_arch_vcpu_ioctl_set_sregs(vcpu, kvm_sregs);
6aa8b732
AK
2013 break;
2014 }
62d9f0db
MT
2015 case KVM_GET_MP_STATE: {
2016 struct kvm_mp_state mp_state;
2017
2018 r = kvm_arch_vcpu_ioctl_get_mpstate(vcpu, &mp_state);
2019 if (r)
2020 goto out;
2021 r = -EFAULT;
2022 if (copy_to_user(argp, &mp_state, sizeof mp_state))
2023 goto out;
2024 r = 0;
2025 break;
2026 }
2027 case KVM_SET_MP_STATE: {
2028 struct kvm_mp_state mp_state;
2029
2030 r = -EFAULT;
2031 if (copy_from_user(&mp_state, argp, sizeof mp_state))
2032 goto out;
2033 r = kvm_arch_vcpu_ioctl_set_mpstate(vcpu, &mp_state);
62d9f0db
MT
2034 break;
2035 }
6aa8b732
AK
2036 case KVM_TRANSLATE: {
2037 struct kvm_translation tr;
2038
2039 r = -EFAULT;
2f366987 2040 if (copy_from_user(&tr, argp, sizeof tr))
6aa8b732 2041 goto out;
8b006791 2042 r = kvm_arch_vcpu_ioctl_translate(vcpu, &tr);
6aa8b732
AK
2043 if (r)
2044 goto out;
2045 r = -EFAULT;
2f366987 2046 if (copy_to_user(argp, &tr, sizeof tr))
6aa8b732
AK
2047 goto out;
2048 r = 0;
2049 break;
2050 }
d0bfb940
JK
2051 case KVM_SET_GUEST_DEBUG: {
2052 struct kvm_guest_debug dbg;
6aa8b732
AK
2053
2054 r = -EFAULT;
2f366987 2055 if (copy_from_user(&dbg, argp, sizeof dbg))
6aa8b732 2056 goto out;
d0bfb940 2057 r = kvm_arch_vcpu_ioctl_set_guest_debug(vcpu, &dbg);
6aa8b732
AK
2058 break;
2059 }
1961d276
AK
2060 case KVM_SET_SIGNAL_MASK: {
2061 struct kvm_signal_mask __user *sigmask_arg = argp;
2062 struct kvm_signal_mask kvm_sigmask;
2063 sigset_t sigset, *p;
2064
2065 p = NULL;
2066 if (argp) {
2067 r = -EFAULT;
2068 if (copy_from_user(&kvm_sigmask, argp,
2069 sizeof kvm_sigmask))
2070 goto out;
2071 r = -EINVAL;
2072 if (kvm_sigmask.len != sizeof sigset)
2073 goto out;
2074 r = -EFAULT;
2075 if (copy_from_user(&sigset, sigmask_arg->sigset,
2076 sizeof sigset))
2077 goto out;
2078 p = &sigset;
2079 }
376d41ff 2080 r = kvm_vcpu_ioctl_set_sigmask(vcpu, p);
1961d276
AK
2081 break;
2082 }
b8836737 2083 case KVM_GET_FPU: {
fa3795a7
DH
2084 fpu = kzalloc(sizeof(struct kvm_fpu), GFP_KERNEL);
2085 r = -ENOMEM;
2086 if (!fpu)
2087 goto out;
2088 r = kvm_arch_vcpu_ioctl_get_fpu(vcpu, fpu);
b8836737
AK
2089 if (r)
2090 goto out;
2091 r = -EFAULT;
fa3795a7 2092 if (copy_to_user(argp, fpu, sizeof(struct kvm_fpu)))
b8836737
AK
2093 goto out;
2094 r = 0;
2095 break;
2096 }
2097 case KVM_SET_FPU: {
ff5c2c03
SL
2098 fpu = memdup_user(argp, sizeof(*fpu));
2099 if (IS_ERR(fpu)) {
2100 r = PTR_ERR(fpu);
18595411 2101 fpu = NULL;
b8836737 2102 goto out;
ff5c2c03 2103 }
fa3795a7 2104 r = kvm_arch_vcpu_ioctl_set_fpu(vcpu, fpu);
b8836737
AK
2105 break;
2106 }
bccf2150 2107 default:
313a3dc7 2108 r = kvm_arch_vcpu_ioctl(filp, ioctl, arg);
bccf2150
AK
2109 }
2110out:
2122ff5e 2111 vcpu_put(vcpu);
fa3795a7
DH
2112 kfree(fpu);
2113 kfree(kvm_sregs);
bccf2150
AK
2114 return r;
2115}
2116
1dda606c
AG
2117#ifdef CONFIG_COMPAT
2118static long kvm_vcpu_compat_ioctl(struct file *filp,
2119 unsigned int ioctl, unsigned long arg)
2120{
2121 struct kvm_vcpu *vcpu = filp->private_data;
2122 void __user *argp = compat_ptr(arg);
2123 int r;
2124
2125 if (vcpu->kvm->mm != current->mm)
2126 return -EIO;
2127
2128 switch (ioctl) {
2129 case KVM_SET_SIGNAL_MASK: {
2130 struct kvm_signal_mask __user *sigmask_arg = argp;
2131 struct kvm_signal_mask kvm_sigmask;
2132 compat_sigset_t csigset;
2133 sigset_t sigset;
2134
2135 if (argp) {
2136 r = -EFAULT;
2137 if (copy_from_user(&kvm_sigmask, argp,
2138 sizeof kvm_sigmask))
2139 goto out;
2140 r = -EINVAL;
2141 if (kvm_sigmask.len != sizeof csigset)
2142 goto out;
2143 r = -EFAULT;
2144 if (copy_from_user(&csigset, sigmask_arg->sigset,
2145 sizeof csigset))
2146 goto out;
760a9a30
AC
2147 sigset_from_compat(&sigset, &csigset);
2148 r = kvm_vcpu_ioctl_set_sigmask(vcpu, &sigset);
2149 } else
2150 r = kvm_vcpu_ioctl_set_sigmask(vcpu, NULL);
1dda606c
AG
2151 break;
2152 }
2153 default:
2154 r = kvm_vcpu_ioctl(filp, ioctl, arg);
2155 }
2156
2157out:
2158 return r;
2159}
2160#endif
2161
852b6d57
SW
2162static int kvm_device_ioctl_attr(struct kvm_device *dev,
2163 int (*accessor)(struct kvm_device *dev,
2164 struct kvm_device_attr *attr),
2165 unsigned long arg)
2166{
2167 struct kvm_device_attr attr;
2168
2169 if (!accessor)
2170 return -EPERM;
2171
2172 if (copy_from_user(&attr, (void __user *)arg, sizeof(attr)))
2173 return -EFAULT;
2174
2175 return accessor(dev, &attr);
2176}
2177
2178static long kvm_device_ioctl(struct file *filp, unsigned int ioctl,
2179 unsigned long arg)
2180{
2181 struct kvm_device *dev = filp->private_data;
2182
2183 switch (ioctl) {
2184 case KVM_SET_DEVICE_ATTR:
2185 return kvm_device_ioctl_attr(dev, dev->ops->set_attr, arg);
2186 case KVM_GET_DEVICE_ATTR:
2187 return kvm_device_ioctl_attr(dev, dev->ops->get_attr, arg);
2188 case KVM_HAS_DEVICE_ATTR:
2189 return kvm_device_ioctl_attr(dev, dev->ops->has_attr, arg);
2190 default:
2191 if (dev->ops->ioctl)
2192 return dev->ops->ioctl(dev, ioctl, arg);
2193
2194 return -ENOTTY;
2195 }
2196}
2197
2198void kvm_device_get(struct kvm_device *dev)
2199{
2200 atomic_inc(&dev->users);
2201}
2202
2203void kvm_device_put(struct kvm_device *dev)
2204{
2205 if (atomic_dec_and_test(&dev->users))
2206 dev->ops->destroy(dev);
2207}
2208
2209static int kvm_device_release(struct inode *inode, struct file *filp)
2210{
2211 struct kvm_device *dev = filp->private_data;
2212 struct kvm *kvm = dev->kvm;
2213
2214 kvm_device_put(dev);
2215 kvm_put_kvm(kvm);
2216 return 0;
2217}
2218
2219static const struct file_operations kvm_device_fops = {
2220 .unlocked_ioctl = kvm_device_ioctl,
2221 .release = kvm_device_release,
2222};
2223
2224struct kvm_device *kvm_device_from_filp(struct file *filp)
2225{
2226 if (filp->f_op != &kvm_device_fops)
2227 return NULL;
2228
2229 return filp->private_data;
2230}
2231
2232static int kvm_ioctl_create_device(struct kvm *kvm,
2233 struct kvm_create_device *cd)
2234{
2235 struct kvm_device_ops *ops = NULL;
2236 struct kvm_device *dev;
2237 bool test = cd->flags & KVM_CREATE_DEVICE_TEST;
2238 int ret;
2239
2240 switch (cd->type) {
5df554ad
SW
2241#ifdef CONFIG_KVM_MPIC
2242 case KVM_DEV_TYPE_FSL_MPIC_20:
2243 case KVM_DEV_TYPE_FSL_MPIC_42:
2244 ops = &kvm_mpic_ops;
2245 break;
2246#endif
852b6d57
SW
2247 default:
2248 return -ENODEV;
2249 }
2250
2251 if (test)
2252 return 0;
2253
2254 dev = kzalloc(sizeof(*dev), GFP_KERNEL);
2255 if (!dev)
2256 return -ENOMEM;
2257
2258 dev->ops = ops;
2259 dev->kvm = kvm;
2260 atomic_set(&dev->users, 1);
2261
2262 ret = ops->create(dev, cd->type);
2263 if (ret < 0) {
2264 kfree(dev);
2265 return ret;
2266 }
2267
2268 ret = anon_inode_getfd(ops->name, &kvm_device_fops, dev, O_RDWR);
2269 if (ret < 0) {
2270 ops->destroy(dev);
2271 return ret;
2272 }
2273
2274 kvm_get_kvm(kvm);
2275 cd->fd = ret;
2276 return 0;
2277}
2278
bccf2150
AK
2279static long kvm_vm_ioctl(struct file *filp,
2280 unsigned int ioctl, unsigned long arg)
2281{
2282 struct kvm *kvm = filp->private_data;
2283 void __user *argp = (void __user *)arg;
1fe779f8 2284 int r;
bccf2150 2285
6d4e4c4f
AK
2286 if (kvm->mm != current->mm)
2287 return -EIO;
bccf2150
AK
2288 switch (ioctl) {
2289 case KVM_CREATE_VCPU:
2290 r = kvm_vm_ioctl_create_vcpu(kvm, arg);
bccf2150 2291 break;
6fc138d2
IE
2292 case KVM_SET_USER_MEMORY_REGION: {
2293 struct kvm_userspace_memory_region kvm_userspace_mem;
2294
2295 r = -EFAULT;
2296 if (copy_from_user(&kvm_userspace_mem, argp,
2297 sizeof kvm_userspace_mem))
2298 goto out;
2299
47ae31e2 2300 r = kvm_vm_ioctl_set_memory_region(kvm, &kvm_userspace_mem);
6aa8b732
AK
2301 break;
2302 }
2303 case KVM_GET_DIRTY_LOG: {
2304 struct kvm_dirty_log log;
2305
2306 r = -EFAULT;
2f366987 2307 if (copy_from_user(&log, argp, sizeof log))
6aa8b732 2308 goto out;
2c6f5df9 2309 r = kvm_vm_ioctl_get_dirty_log(kvm, &log);
6aa8b732
AK
2310 break;
2311 }
5f94c174
LV
2312#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
2313 case KVM_REGISTER_COALESCED_MMIO: {
2314 struct kvm_coalesced_mmio_zone zone;
2315 r = -EFAULT;
2316 if (copy_from_user(&zone, argp, sizeof zone))
2317 goto out;
5f94c174 2318 r = kvm_vm_ioctl_register_coalesced_mmio(kvm, &zone);
5f94c174
LV
2319 break;
2320 }
2321 case KVM_UNREGISTER_COALESCED_MMIO: {
2322 struct kvm_coalesced_mmio_zone zone;
2323 r = -EFAULT;
2324 if (copy_from_user(&zone, argp, sizeof zone))
2325 goto out;
5f94c174 2326 r = kvm_vm_ioctl_unregister_coalesced_mmio(kvm, &zone);
5f94c174
LV
2327 break;
2328 }
2329#endif
721eecbf
GH
2330 case KVM_IRQFD: {
2331 struct kvm_irqfd data;
2332
2333 r = -EFAULT;
2334 if (copy_from_user(&data, argp, sizeof data))
2335 goto out;
d4db2935 2336 r = kvm_irqfd(kvm, &data);
721eecbf
GH
2337 break;
2338 }
d34e6b17
GH
2339 case KVM_IOEVENTFD: {
2340 struct kvm_ioeventfd data;
2341
2342 r = -EFAULT;
2343 if (copy_from_user(&data, argp, sizeof data))
2344 goto out;
2345 r = kvm_ioeventfd(kvm, &data);
2346 break;
2347 }
73880c80
GN
2348#ifdef CONFIG_KVM_APIC_ARCHITECTURE
2349 case KVM_SET_BOOT_CPU_ID:
2350 r = 0;
894a9c55 2351 mutex_lock(&kvm->lock);
73880c80
GN
2352 if (atomic_read(&kvm->online_vcpus) != 0)
2353 r = -EBUSY;
2354 else
2355 kvm->bsp_vcpu_id = arg;
894a9c55 2356 mutex_unlock(&kvm->lock);
73880c80 2357 break;
07975ad3
JK
2358#endif
2359#ifdef CONFIG_HAVE_KVM_MSI
2360 case KVM_SIGNAL_MSI: {
2361 struct kvm_msi msi;
2362
2363 r = -EFAULT;
2364 if (copy_from_user(&msi, argp, sizeof msi))
2365 goto out;
2366 r = kvm_send_userspace_msi(kvm, &msi);
2367 break;
2368 }
23d43cf9
CD
2369#endif
2370#ifdef __KVM_HAVE_IRQ_LINE
2371 case KVM_IRQ_LINE_STATUS:
2372 case KVM_IRQ_LINE: {
2373 struct kvm_irq_level irq_event;
2374
2375 r = -EFAULT;
2376 if (copy_from_user(&irq_event, argp, sizeof irq_event))
2377 goto out;
2378
aa2fbe6d
YZ
2379 r = kvm_vm_ioctl_irq_line(kvm, &irq_event,
2380 ioctl == KVM_IRQ_LINE_STATUS);
23d43cf9
CD
2381 if (r)
2382 goto out;
2383
2384 r = -EFAULT;
2385 if (ioctl == KVM_IRQ_LINE_STATUS) {
2386 if (copy_to_user(argp, &irq_event, sizeof irq_event))
2387 goto out;
2388 }
2389
2390 r = 0;
2391 break;
2392 }
73880c80 2393#endif
aa8d5944
AG
2394#ifdef CONFIG_HAVE_KVM_IRQ_ROUTING
2395 case KVM_SET_GSI_ROUTING: {
2396 struct kvm_irq_routing routing;
2397 struct kvm_irq_routing __user *urouting;
2398 struct kvm_irq_routing_entry *entries;
2399
2400 r = -EFAULT;
2401 if (copy_from_user(&routing, argp, sizeof(routing)))
2402 goto out;
2403 r = -EINVAL;
2404 if (routing.nr >= KVM_MAX_IRQ_ROUTES)
2405 goto out;
2406 if (routing.flags)
2407 goto out;
2408 r = -ENOMEM;
2409 entries = vmalloc(routing.nr * sizeof(*entries));
2410 if (!entries)
2411 goto out;
2412 r = -EFAULT;
2413 urouting = argp;
2414 if (copy_from_user(entries, urouting->entries,
2415 routing.nr * sizeof(*entries)))
2416 goto out_free_irq_routing;
2417 r = kvm_set_irq_routing(kvm, entries, routing.nr,
2418 routing.flags);
2419 out_free_irq_routing:
2420 vfree(entries);
2421 break;
2422 }
2423#endif /* CONFIG_HAVE_KVM_IRQ_ROUTING */
852b6d57
SW
2424 case KVM_CREATE_DEVICE: {
2425 struct kvm_create_device cd;
2426
2427 r = -EFAULT;
2428 if (copy_from_user(&cd, argp, sizeof(cd)))
2429 goto out;
2430
2431 r = kvm_ioctl_create_device(kvm, &cd);
2432 if (r)
2433 goto out;
2434
2435 r = -EFAULT;
2436 if (copy_to_user(argp, &cd, sizeof(cd)))
2437 goto out;
2438
2439 r = 0;
2440 break;
2441 }
f17abe9a 2442 default:
1fe779f8 2443 r = kvm_arch_vm_ioctl(filp, ioctl, arg);
bfd99ff5
AK
2444 if (r == -ENOTTY)
2445 r = kvm_vm_ioctl_assigned_device(kvm, ioctl, arg);
f17abe9a
AK
2446 }
2447out:
2448 return r;
2449}
2450
6ff5894c
AB
2451#ifdef CONFIG_COMPAT
2452struct compat_kvm_dirty_log {
2453 __u32 slot;
2454 __u32 padding1;
2455 union {
2456 compat_uptr_t dirty_bitmap; /* one bit per page */
2457 __u64 padding2;
2458 };
2459};
2460
2461static long kvm_vm_compat_ioctl(struct file *filp,
2462 unsigned int ioctl, unsigned long arg)
2463{
2464 struct kvm *kvm = filp->private_data;
2465 int r;
2466
2467 if (kvm->mm != current->mm)
2468 return -EIO;
2469 switch (ioctl) {
2470 case KVM_GET_DIRTY_LOG: {
2471 struct compat_kvm_dirty_log compat_log;
2472 struct kvm_dirty_log log;
2473
2474 r = -EFAULT;
2475 if (copy_from_user(&compat_log, (void __user *)arg,
2476 sizeof(compat_log)))
2477 goto out;
2478 log.slot = compat_log.slot;
2479 log.padding1 = compat_log.padding1;
2480 log.padding2 = compat_log.padding2;
2481 log.dirty_bitmap = compat_ptr(compat_log.dirty_bitmap);
2482
2483 r = kvm_vm_ioctl_get_dirty_log(kvm, &log);
6ff5894c
AB
2484 break;
2485 }
2486 default:
2487 r = kvm_vm_ioctl(filp, ioctl, arg);
2488 }
2489
2490out:
2491 return r;
2492}
2493#endif
2494
e4a533a4 2495static int kvm_vm_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
f17abe9a 2496{
777b3f49
MT
2497 struct page *page[1];
2498 unsigned long addr;
2499 int npages;
2500 gfn_t gfn = vmf->pgoff;
f17abe9a 2501 struct kvm *kvm = vma->vm_file->private_data;
f17abe9a 2502
777b3f49
MT
2503 addr = gfn_to_hva(kvm, gfn);
2504 if (kvm_is_error_hva(addr))
e4a533a4 2505 return VM_FAULT_SIGBUS;
777b3f49
MT
2506
2507 npages = get_user_pages(current, current->mm, addr, 1, 1, 0, page,
2508 NULL);
2509 if (unlikely(npages != 1))
e4a533a4 2510 return VM_FAULT_SIGBUS;
777b3f49
MT
2511
2512 vmf->page = page[0];
e4a533a4 2513 return 0;
f17abe9a
AK
2514}
2515
f0f37e2f 2516static const struct vm_operations_struct kvm_vm_vm_ops = {
e4a533a4 2517 .fault = kvm_vm_fault,
f17abe9a
AK
2518};
2519
2520static int kvm_vm_mmap(struct file *file, struct vm_area_struct *vma)
2521{
2522 vma->vm_ops = &kvm_vm_vm_ops;
2523 return 0;
2524}
2525
3d3aab1b 2526static struct file_operations kvm_vm_fops = {
f17abe9a
AK
2527 .release = kvm_vm_release,
2528 .unlocked_ioctl = kvm_vm_ioctl,
6ff5894c
AB
2529#ifdef CONFIG_COMPAT
2530 .compat_ioctl = kvm_vm_compat_ioctl,
2531#endif
f17abe9a 2532 .mmap = kvm_vm_mmap,
6038f373 2533 .llseek = noop_llseek,
f17abe9a
AK
2534};
2535
e08b9637 2536static int kvm_dev_ioctl_create_vm(unsigned long type)
f17abe9a 2537{
aac87636 2538 int r;
f17abe9a
AK
2539 struct kvm *kvm;
2540
e08b9637 2541 kvm = kvm_create_vm(type);
d6d28168
AK
2542 if (IS_ERR(kvm))
2543 return PTR_ERR(kvm);
6ce5a090
TY
2544#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
2545 r = kvm_coalesced_mmio_init(kvm);
2546 if (r < 0) {
2547 kvm_put_kvm(kvm);
2548 return r;
2549 }
2550#endif
aac87636
HC
2551 r = anon_inode_getfd("kvm-vm", &kvm_vm_fops, kvm, O_RDWR);
2552 if (r < 0)
66c0b394 2553 kvm_put_kvm(kvm);
f17abe9a 2554
aac87636 2555 return r;
f17abe9a
AK
2556}
2557
1a811b61
AK
2558static long kvm_dev_ioctl_check_extension_generic(long arg)
2559{
2560 switch (arg) {
ca9edaee 2561 case KVM_CAP_USER_MEMORY:
1a811b61 2562 case KVM_CAP_DESTROY_MEMORY_REGION_WORKS:
4cd481f6 2563 case KVM_CAP_JOIN_MEMORY_REGIONS_WORKS:
73880c80
GN
2564#ifdef CONFIG_KVM_APIC_ARCHITECTURE
2565 case KVM_CAP_SET_BOOT_CPU_ID:
2566#endif
a9c7399d 2567 case KVM_CAP_INTERNAL_ERROR_DATA:
07975ad3
JK
2568#ifdef CONFIG_HAVE_KVM_MSI
2569 case KVM_CAP_SIGNAL_MSI:
7df35f54
AG
2570#endif
2571#ifdef CONFIG_HAVE_KVM_IRQ_ROUTING
2572 case KVM_CAP_IRQFD_RESAMPLE:
07975ad3 2573#endif
1a811b61 2574 return 1;
a725d56a 2575#ifdef CONFIG_HAVE_KVM_IRQ_ROUTING
399ec807 2576 case KVM_CAP_IRQ_ROUTING:
36463146 2577 return KVM_MAX_IRQ_ROUTES;
399ec807 2578#endif
1a811b61
AK
2579 default:
2580 break;
2581 }
2582 return kvm_dev_ioctl_check_extension(arg);
2583}
2584
f17abe9a
AK
2585static long kvm_dev_ioctl(struct file *filp,
2586 unsigned int ioctl, unsigned long arg)
2587{
07c45a36 2588 long r = -EINVAL;
f17abe9a
AK
2589
2590 switch (ioctl) {
2591 case KVM_GET_API_VERSION:
f0fe5108
AK
2592 r = -EINVAL;
2593 if (arg)
2594 goto out;
f17abe9a
AK
2595 r = KVM_API_VERSION;
2596 break;
2597 case KVM_CREATE_VM:
e08b9637 2598 r = kvm_dev_ioctl_create_vm(arg);
f17abe9a 2599 break;
018d00d2 2600 case KVM_CHECK_EXTENSION:
1a811b61 2601 r = kvm_dev_ioctl_check_extension_generic(arg);
5d308f45 2602 break;
07c45a36
AK
2603 case KVM_GET_VCPU_MMAP_SIZE:
2604 r = -EINVAL;
2605 if (arg)
2606 goto out;
adb1ff46
AK
2607 r = PAGE_SIZE; /* struct kvm_run */
2608#ifdef CONFIG_X86
2609 r += PAGE_SIZE; /* pio data page */
5f94c174
LV
2610#endif
2611#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
2612 r += PAGE_SIZE; /* coalesced mmio ring page */
adb1ff46 2613#endif
07c45a36 2614 break;
d4c9ff2d
FEL
2615 case KVM_TRACE_ENABLE:
2616 case KVM_TRACE_PAUSE:
2617 case KVM_TRACE_DISABLE:
2023a29c 2618 r = -EOPNOTSUPP;
d4c9ff2d 2619 break;
6aa8b732 2620 default:
043405e1 2621 return kvm_arch_dev_ioctl(filp, ioctl, arg);
6aa8b732
AK
2622 }
2623out:
2624 return r;
2625}
2626
6aa8b732 2627static struct file_operations kvm_chardev_ops = {
6aa8b732
AK
2628 .unlocked_ioctl = kvm_dev_ioctl,
2629 .compat_ioctl = kvm_dev_ioctl,
6038f373 2630 .llseek = noop_llseek,
6aa8b732
AK
2631};
2632
2633static struct miscdevice kvm_dev = {
bbe4432e 2634 KVM_MINOR,
6aa8b732
AK
2635 "kvm",
2636 &kvm_chardev_ops,
2637};
2638
75b7127c 2639static void hardware_enable_nolock(void *junk)
1b6c0168
AK
2640{
2641 int cpu = raw_smp_processor_id();
10474ae8 2642 int r;
1b6c0168 2643
7f59f492 2644 if (cpumask_test_cpu(cpu, cpus_hardware_enabled))
1b6c0168 2645 return;
10474ae8 2646
7f59f492 2647 cpumask_set_cpu(cpu, cpus_hardware_enabled);
10474ae8
AG
2648
2649 r = kvm_arch_hardware_enable(NULL);
2650
2651 if (r) {
2652 cpumask_clear_cpu(cpu, cpus_hardware_enabled);
2653 atomic_inc(&hardware_enable_failed);
2654 printk(KERN_INFO "kvm: enabling virtualization on "
2655 "CPU%d failed\n", cpu);
2656 }
1b6c0168
AK
2657}
2658
75b7127c
TY
2659static void hardware_enable(void *junk)
2660{
e935b837 2661 raw_spin_lock(&kvm_lock);
75b7127c 2662 hardware_enable_nolock(junk);
e935b837 2663 raw_spin_unlock(&kvm_lock);
75b7127c
TY
2664}
2665
2666static void hardware_disable_nolock(void *junk)
1b6c0168
AK
2667{
2668 int cpu = raw_smp_processor_id();
2669
7f59f492 2670 if (!cpumask_test_cpu(cpu, cpus_hardware_enabled))
1b6c0168 2671 return;
7f59f492 2672 cpumask_clear_cpu(cpu, cpus_hardware_enabled);
e9b11c17 2673 kvm_arch_hardware_disable(NULL);
1b6c0168
AK
2674}
2675
75b7127c
TY
2676static void hardware_disable(void *junk)
2677{
e935b837 2678 raw_spin_lock(&kvm_lock);
75b7127c 2679 hardware_disable_nolock(junk);
e935b837 2680 raw_spin_unlock(&kvm_lock);
75b7127c
TY
2681}
2682
10474ae8
AG
2683static void hardware_disable_all_nolock(void)
2684{
2685 BUG_ON(!kvm_usage_count);
2686
2687 kvm_usage_count--;
2688 if (!kvm_usage_count)
75b7127c 2689 on_each_cpu(hardware_disable_nolock, NULL, 1);
10474ae8
AG
2690}
2691
2692static void hardware_disable_all(void)
2693{
e935b837 2694 raw_spin_lock(&kvm_lock);
10474ae8 2695 hardware_disable_all_nolock();
e935b837 2696 raw_spin_unlock(&kvm_lock);
10474ae8
AG
2697}
2698
2699static int hardware_enable_all(void)
2700{
2701 int r = 0;
2702
e935b837 2703 raw_spin_lock(&kvm_lock);
10474ae8
AG
2704
2705 kvm_usage_count++;
2706 if (kvm_usage_count == 1) {
2707 atomic_set(&hardware_enable_failed, 0);
75b7127c 2708 on_each_cpu(hardware_enable_nolock, NULL, 1);
10474ae8
AG
2709
2710 if (atomic_read(&hardware_enable_failed)) {
2711 hardware_disable_all_nolock();
2712 r = -EBUSY;
2713 }
2714 }
2715
e935b837 2716 raw_spin_unlock(&kvm_lock);
10474ae8
AG
2717
2718 return r;
2719}
2720
774c47f1
AK
2721static int kvm_cpu_hotplug(struct notifier_block *notifier, unsigned long val,
2722 void *v)
2723{
2724 int cpu = (long)v;
2725
10474ae8
AG
2726 if (!kvm_usage_count)
2727 return NOTIFY_OK;
2728
1a6f4d7f 2729 val &= ~CPU_TASKS_FROZEN;
774c47f1 2730 switch (val) {
cec9ad27 2731 case CPU_DYING:
6ec8a856
AK
2732 printk(KERN_INFO "kvm: disabling virtualization on CPU%d\n",
2733 cpu);
2734 hardware_disable(NULL);
2735 break;
da908f2f 2736 case CPU_STARTING:
43934a38
JK
2737 printk(KERN_INFO "kvm: enabling virtualization on CPU%d\n",
2738 cpu);
da908f2f 2739 hardware_enable(NULL);
774c47f1
AK
2740 break;
2741 }
2742 return NOTIFY_OK;
2743}
2744
9a2b85c6 2745static int kvm_reboot(struct notifier_block *notifier, unsigned long val,
d77c26fc 2746 void *v)
9a2b85c6 2747{
8e1c1815
SY
2748 /*
2749 * Some (well, at least mine) BIOSes hang on reboot if
2750 * in vmx root mode.
2751 *
2752 * And Intel TXT required VMX off for all cpu when system shutdown.
2753 */
2754 printk(KERN_INFO "kvm: exiting hardware virtualization\n");
2755 kvm_rebooting = true;
75b7127c 2756 on_each_cpu(hardware_disable_nolock, NULL, 1);
9a2b85c6
RR
2757 return NOTIFY_OK;
2758}
2759
2760static struct notifier_block kvm_reboot_notifier = {
2761 .notifier_call = kvm_reboot,
2762 .priority = 0,
2763};
2764
e93f8a0f 2765static void kvm_io_bus_destroy(struct kvm_io_bus *bus)
2eeb2e94
GH
2766{
2767 int i;
2768
2769 for (i = 0; i < bus->dev_count; i++) {
743eeb0b 2770 struct kvm_io_device *pos = bus->range[i].dev;
2eeb2e94
GH
2771
2772 kvm_iodevice_destructor(pos);
2773 }
e93f8a0f 2774 kfree(bus);
2eeb2e94
GH
2775}
2776
39369f7a 2777static int kvm_io_bus_sort_cmp(const void *p1, const void *p2)
743eeb0b
SL
2778{
2779 const struct kvm_io_range *r1 = p1;
2780 const struct kvm_io_range *r2 = p2;
2781
2782 if (r1->addr < r2->addr)
2783 return -1;
2784 if (r1->addr + r1->len > r2->addr + r2->len)
2785 return 1;
2786 return 0;
2787}
2788
39369f7a 2789static int kvm_io_bus_insert_dev(struct kvm_io_bus *bus, struct kvm_io_device *dev,
743eeb0b
SL
2790 gpa_t addr, int len)
2791{
743eeb0b
SL
2792 bus->range[bus->dev_count++] = (struct kvm_io_range) {
2793 .addr = addr,
2794 .len = len,
2795 .dev = dev,
2796 };
2797
2798 sort(bus->range, bus->dev_count, sizeof(struct kvm_io_range),
2799 kvm_io_bus_sort_cmp, NULL);
2800
2801 return 0;
2802}
2803
39369f7a 2804static int kvm_io_bus_get_first_dev(struct kvm_io_bus *bus,
743eeb0b
SL
2805 gpa_t addr, int len)
2806{
2807 struct kvm_io_range *range, key;
2808 int off;
2809
2810 key = (struct kvm_io_range) {
2811 .addr = addr,
2812 .len = len,
2813 };
2814
2815 range = bsearch(&key, bus->range, bus->dev_count,
2816 sizeof(struct kvm_io_range), kvm_io_bus_sort_cmp);
2817 if (range == NULL)
2818 return -ENOENT;
2819
2820 off = range - bus->range;
2821
2822 while (off > 0 && kvm_io_bus_sort_cmp(&key, &bus->range[off-1]) == 0)
2823 off--;
2824
2825 return off;
2826}
2827
bda9020e 2828/* kvm_io_bus_write - called under kvm->slots_lock */
e93f8a0f 2829int kvm_io_bus_write(struct kvm *kvm, enum kvm_bus bus_idx, gpa_t addr,
bda9020e 2830 int len, const void *val)
2eeb2e94 2831{
743eeb0b 2832 int idx;
90d83dc3 2833 struct kvm_io_bus *bus;
743eeb0b
SL
2834 struct kvm_io_range range;
2835
2836 range = (struct kvm_io_range) {
2837 .addr = addr,
2838 .len = len,
2839 };
90d83dc3
LJ
2840
2841 bus = srcu_dereference(kvm->buses[bus_idx], &kvm->srcu);
743eeb0b
SL
2842 idx = kvm_io_bus_get_first_dev(bus, addr, len);
2843 if (idx < 0)
2844 return -EOPNOTSUPP;
2845
2846 while (idx < bus->dev_count &&
2847 kvm_io_bus_sort_cmp(&range, &bus->range[idx]) == 0) {
2848 if (!kvm_iodevice_write(bus->range[idx].dev, addr, len, val))
bda9020e 2849 return 0;
743eeb0b
SL
2850 idx++;
2851 }
2852
bda9020e
MT
2853 return -EOPNOTSUPP;
2854}
2eeb2e94 2855
bda9020e 2856/* kvm_io_bus_read - called under kvm->slots_lock */
e93f8a0f
MT
2857int kvm_io_bus_read(struct kvm *kvm, enum kvm_bus bus_idx, gpa_t addr,
2858 int len, void *val)
bda9020e 2859{
743eeb0b 2860 int idx;
90d83dc3 2861 struct kvm_io_bus *bus;
743eeb0b
SL
2862 struct kvm_io_range range;
2863
2864 range = (struct kvm_io_range) {
2865 .addr = addr,
2866 .len = len,
2867 };
e93f8a0f 2868
90d83dc3 2869 bus = srcu_dereference(kvm->buses[bus_idx], &kvm->srcu);
743eeb0b
SL
2870 idx = kvm_io_bus_get_first_dev(bus, addr, len);
2871 if (idx < 0)
2872 return -EOPNOTSUPP;
2873
2874 while (idx < bus->dev_count &&
2875 kvm_io_bus_sort_cmp(&range, &bus->range[idx]) == 0) {
2876 if (!kvm_iodevice_read(bus->range[idx].dev, addr, len, val))
bda9020e 2877 return 0;
743eeb0b
SL
2878 idx++;
2879 }
2880
bda9020e 2881 return -EOPNOTSUPP;
2eeb2e94
GH
2882}
2883
79fac95e 2884/* Caller must hold slots_lock. */
743eeb0b
SL
2885int kvm_io_bus_register_dev(struct kvm *kvm, enum kvm_bus bus_idx, gpa_t addr,
2886 int len, struct kvm_io_device *dev)
6c474694 2887{
e93f8a0f 2888 struct kvm_io_bus *new_bus, *bus;
090b7aff 2889
e93f8a0f 2890 bus = kvm->buses[bus_idx];
a1300716 2891 if (bus->dev_count > NR_IOBUS_DEVS - 1)
090b7aff 2892 return -ENOSPC;
2eeb2e94 2893
a1300716
AK
2894 new_bus = kzalloc(sizeof(*bus) + ((bus->dev_count + 1) *
2895 sizeof(struct kvm_io_range)), GFP_KERNEL);
e93f8a0f
MT
2896 if (!new_bus)
2897 return -ENOMEM;
a1300716
AK
2898 memcpy(new_bus, bus, sizeof(*bus) + (bus->dev_count *
2899 sizeof(struct kvm_io_range)));
743eeb0b 2900 kvm_io_bus_insert_dev(new_bus, dev, addr, len);
e93f8a0f
MT
2901 rcu_assign_pointer(kvm->buses[bus_idx], new_bus);
2902 synchronize_srcu_expedited(&kvm->srcu);
2903 kfree(bus);
090b7aff
GH
2904
2905 return 0;
2906}
2907
79fac95e 2908/* Caller must hold slots_lock. */
e93f8a0f
MT
2909int kvm_io_bus_unregister_dev(struct kvm *kvm, enum kvm_bus bus_idx,
2910 struct kvm_io_device *dev)
090b7aff 2911{
e93f8a0f
MT
2912 int i, r;
2913 struct kvm_io_bus *new_bus, *bus;
090b7aff 2914
cdfca7b3 2915 bus = kvm->buses[bus_idx];
e93f8a0f 2916 r = -ENOENT;
a1300716
AK
2917 for (i = 0; i < bus->dev_count; i++)
2918 if (bus->range[i].dev == dev) {
e93f8a0f 2919 r = 0;
090b7aff
GH
2920 break;
2921 }
e93f8a0f 2922
a1300716 2923 if (r)
e93f8a0f 2924 return r;
a1300716
AK
2925
2926 new_bus = kzalloc(sizeof(*bus) + ((bus->dev_count - 1) *
2927 sizeof(struct kvm_io_range)), GFP_KERNEL);
2928 if (!new_bus)
2929 return -ENOMEM;
2930
2931 memcpy(new_bus, bus, sizeof(*bus) + i * sizeof(struct kvm_io_range));
2932 new_bus->dev_count--;
2933 memcpy(new_bus->range + i, bus->range + i + 1,
2934 (new_bus->dev_count - i) * sizeof(struct kvm_io_range));
e93f8a0f
MT
2935
2936 rcu_assign_pointer(kvm->buses[bus_idx], new_bus);
2937 synchronize_srcu_expedited(&kvm->srcu);
2938 kfree(bus);
2939 return r;
2eeb2e94
GH
2940}
2941
774c47f1
AK
2942static struct notifier_block kvm_cpu_notifier = {
2943 .notifier_call = kvm_cpu_hotplug,
774c47f1
AK
2944};
2945
8b88b099 2946static int vm_stat_get(void *_offset, u64 *val)
ba1389b7
AK
2947{
2948 unsigned offset = (long)_offset;
ba1389b7
AK
2949 struct kvm *kvm;
2950
8b88b099 2951 *val = 0;
e935b837 2952 raw_spin_lock(&kvm_lock);
ba1389b7 2953 list_for_each_entry(kvm, &vm_list, vm_list)
8b88b099 2954 *val += *(u32 *)((void *)kvm + offset);
e935b837 2955 raw_spin_unlock(&kvm_lock);
8b88b099 2956 return 0;
ba1389b7
AK
2957}
2958
2959DEFINE_SIMPLE_ATTRIBUTE(vm_stat_fops, vm_stat_get, NULL, "%llu\n");
2960
8b88b099 2961static int vcpu_stat_get(void *_offset, u64 *val)
1165f5fe
AK
2962{
2963 unsigned offset = (long)_offset;
1165f5fe
AK
2964 struct kvm *kvm;
2965 struct kvm_vcpu *vcpu;
2966 int i;
2967
8b88b099 2968 *val = 0;
e935b837 2969 raw_spin_lock(&kvm_lock);
1165f5fe 2970 list_for_each_entry(kvm, &vm_list, vm_list)
988a2cae
GN
2971 kvm_for_each_vcpu(i, vcpu, kvm)
2972 *val += *(u32 *)((void *)vcpu + offset);
2973
e935b837 2974 raw_spin_unlock(&kvm_lock);
8b88b099 2975 return 0;
1165f5fe
AK
2976}
2977
ba1389b7
AK
2978DEFINE_SIMPLE_ATTRIBUTE(vcpu_stat_fops, vcpu_stat_get, NULL, "%llu\n");
2979
828c0950 2980static const struct file_operations *stat_fops[] = {
ba1389b7
AK
2981 [KVM_STAT_VCPU] = &vcpu_stat_fops,
2982 [KVM_STAT_VM] = &vm_stat_fops,
2983};
1165f5fe 2984
4f69b680 2985static int kvm_init_debug(void)
6aa8b732 2986{
4f69b680 2987 int r = -EFAULT;
6aa8b732
AK
2988 struct kvm_stats_debugfs_item *p;
2989
76f7c879 2990 kvm_debugfs_dir = debugfs_create_dir("kvm", NULL);
4f69b680
H
2991 if (kvm_debugfs_dir == NULL)
2992 goto out;
2993
2994 for (p = debugfs_entries; p->name; ++p) {
76f7c879 2995 p->dentry = debugfs_create_file(p->name, 0444, kvm_debugfs_dir,
1165f5fe 2996 (void *)(long)p->offset,
ba1389b7 2997 stat_fops[p->kind]);
4f69b680
H
2998 if (p->dentry == NULL)
2999 goto out_dir;
3000 }
3001
3002 return 0;
3003
3004out_dir:
3005 debugfs_remove_recursive(kvm_debugfs_dir);
3006out:
3007 return r;
6aa8b732
AK
3008}
3009
3010static void kvm_exit_debug(void)
3011{
3012 struct kvm_stats_debugfs_item *p;
3013
3014 for (p = debugfs_entries; p->name; ++p)
3015 debugfs_remove(p->dentry);
76f7c879 3016 debugfs_remove(kvm_debugfs_dir);
6aa8b732
AK
3017}
3018
fb3600cc 3019static int kvm_suspend(void)
59ae6c6b 3020{
10474ae8 3021 if (kvm_usage_count)
75b7127c 3022 hardware_disable_nolock(NULL);
59ae6c6b
AK
3023 return 0;
3024}
3025
fb3600cc 3026static void kvm_resume(void)
59ae6c6b 3027{
ca84d1a2 3028 if (kvm_usage_count) {
e935b837 3029 WARN_ON(raw_spin_is_locked(&kvm_lock));
75b7127c 3030 hardware_enable_nolock(NULL);
ca84d1a2 3031 }
59ae6c6b
AK
3032}
3033
fb3600cc 3034static struct syscore_ops kvm_syscore_ops = {
59ae6c6b
AK
3035 .suspend = kvm_suspend,
3036 .resume = kvm_resume,
3037};
3038
15ad7146
AK
3039static inline
3040struct kvm_vcpu *preempt_notifier_to_vcpu(struct preempt_notifier *pn)
3041{
3042 return container_of(pn, struct kvm_vcpu, preempt_notifier);
3043}
3044
3045static void kvm_sched_in(struct preempt_notifier *pn, int cpu)
3046{
3047 struct kvm_vcpu *vcpu = preempt_notifier_to_vcpu(pn);
3a08a8f9
R
3048 if (vcpu->preempted)
3049 vcpu->preempted = false;
15ad7146 3050
e9b11c17 3051 kvm_arch_vcpu_load(vcpu, cpu);
15ad7146
AK
3052}
3053
3054static void kvm_sched_out(struct preempt_notifier *pn,
3055 struct task_struct *next)
3056{
3057 struct kvm_vcpu *vcpu = preempt_notifier_to_vcpu(pn);
3058
3a08a8f9
R
3059 if (current->state == TASK_RUNNING)
3060 vcpu->preempted = true;
e9b11c17 3061 kvm_arch_vcpu_put(vcpu);
15ad7146
AK
3062}
3063
0ee75bea 3064int kvm_init(void *opaque, unsigned vcpu_size, unsigned vcpu_align,
c16f862d 3065 struct module *module)
6aa8b732
AK
3066{
3067 int r;
002c7f7c 3068 int cpu;
6aa8b732 3069
a0f155e9
CH
3070 r = kvm_irqfd_init();
3071 if (r)
3072 goto out_irqfd;
f8c16bba
ZX
3073 r = kvm_arch_init(opaque);
3074 if (r)
d2308784 3075 goto out_fail;
cb498ea2 3076
8437a617 3077 if (!zalloc_cpumask_var(&cpus_hardware_enabled, GFP_KERNEL)) {
7f59f492
RR
3078 r = -ENOMEM;
3079 goto out_free_0;
3080 }
3081
e9b11c17 3082 r = kvm_arch_hardware_setup();
6aa8b732 3083 if (r < 0)
7f59f492 3084 goto out_free_0a;
6aa8b732 3085
002c7f7c
YS
3086 for_each_online_cpu(cpu) {
3087 smp_call_function_single(cpu,
e9b11c17 3088 kvm_arch_check_processor_compat,
8691e5a8 3089 &r, 1);
002c7f7c 3090 if (r < 0)
d2308784 3091 goto out_free_1;
002c7f7c
YS
3092 }
3093
774c47f1
AK
3094 r = register_cpu_notifier(&kvm_cpu_notifier);
3095 if (r)
d2308784 3096 goto out_free_2;
6aa8b732
AK
3097 register_reboot_notifier(&kvm_reboot_notifier);
3098
c16f862d 3099 /* A kmem cache lets us meet the alignment requirements of fx_save. */
0ee75bea
AK
3100 if (!vcpu_align)
3101 vcpu_align = __alignof__(struct kvm_vcpu);
3102 kvm_vcpu_cache = kmem_cache_create("kvm_vcpu", vcpu_size, vcpu_align,
56919c5c 3103 0, NULL);
c16f862d
RR
3104 if (!kvm_vcpu_cache) {
3105 r = -ENOMEM;
fb3600cc 3106 goto out_free_3;
c16f862d
RR
3107 }
3108
af585b92
GN
3109 r = kvm_async_pf_init();
3110 if (r)
3111 goto out_free;
3112
6aa8b732 3113 kvm_chardev_ops.owner = module;
3d3aab1b
CB
3114 kvm_vm_fops.owner = module;
3115 kvm_vcpu_fops.owner = module;
6aa8b732
AK
3116
3117 r = misc_register(&kvm_dev);
3118 if (r) {
d77c26fc 3119 printk(KERN_ERR "kvm: misc device register failed\n");
af585b92 3120 goto out_unreg;
6aa8b732
AK
3121 }
3122
fb3600cc
RW
3123 register_syscore_ops(&kvm_syscore_ops);
3124
15ad7146
AK
3125 kvm_preempt_ops.sched_in = kvm_sched_in;
3126 kvm_preempt_ops.sched_out = kvm_sched_out;
3127
4f69b680
H
3128 r = kvm_init_debug();
3129 if (r) {
3130 printk(KERN_ERR "kvm: create debugfs files failed\n");
3131 goto out_undebugfs;
3132 }
0ea4ed8e 3133
c7addb90 3134 return 0;
6aa8b732 3135
4f69b680
H
3136out_undebugfs:
3137 unregister_syscore_ops(&kvm_syscore_ops);
af585b92
GN
3138out_unreg:
3139 kvm_async_pf_deinit();
6aa8b732 3140out_free:
c16f862d 3141 kmem_cache_destroy(kvm_vcpu_cache);
d2308784 3142out_free_3:
6aa8b732 3143 unregister_reboot_notifier(&kvm_reboot_notifier);
774c47f1 3144 unregister_cpu_notifier(&kvm_cpu_notifier);
d2308784 3145out_free_2:
d2308784 3146out_free_1:
e9b11c17 3147 kvm_arch_hardware_unsetup();
7f59f492
RR
3148out_free_0a:
3149 free_cpumask_var(cpus_hardware_enabled);
d2308784 3150out_free_0:
f8c16bba 3151 kvm_arch_exit();
d2308784 3152out_fail:
a0f155e9
CH
3153 kvm_irqfd_exit();
3154out_irqfd:
6aa8b732
AK
3155 return r;
3156}
cb498ea2 3157EXPORT_SYMBOL_GPL(kvm_init);
6aa8b732 3158
cb498ea2 3159void kvm_exit(void)
6aa8b732 3160{
0ea4ed8e 3161 kvm_exit_debug();
6aa8b732 3162 misc_deregister(&kvm_dev);
c16f862d 3163 kmem_cache_destroy(kvm_vcpu_cache);
af585b92 3164 kvm_async_pf_deinit();
fb3600cc 3165 unregister_syscore_ops(&kvm_syscore_ops);
6aa8b732 3166 unregister_reboot_notifier(&kvm_reboot_notifier);
59ae6c6b 3167 unregister_cpu_notifier(&kvm_cpu_notifier);
75b7127c 3168 on_each_cpu(hardware_disable_nolock, NULL, 1);
e9b11c17 3169 kvm_arch_hardware_unsetup();
f8c16bba 3170 kvm_arch_exit();
a0f155e9 3171 kvm_irqfd_exit();
7f59f492 3172 free_cpumask_var(cpus_hardware_enabled);
6aa8b732 3173}
cb498ea2 3174EXPORT_SYMBOL_GPL(kvm_exit);