blob: 3a5a08298aabd1a06c770c2cf4c2b657fae0af14 [file] [log] [blame]
Avi Kivity6aa8b732006-12-10 02:21:36 -08001/*
2 * Kernel-based Virtual Machine driver for Linux
3 *
4 * This module enables machines with Intel VT-x extensions to run virtual
5 * machines without emulation or binary translation.
6 *
7 * Copyright (C) 2006 Qumranet, Inc.
8 *
9 * Authors:
10 * Avi Kivity <avi@qumranet.com>
11 * Yaniv Kamay <yaniv@qumranet.com>
12 *
13 * This work is licensed under the terms of the GNU GPL, version 2. See
14 * the COPYING file in the top-level directory.
15 *
16 */
17
Hollis Blancharde2174022007-12-03 15:30:24 -060018#include "iodev.h"
Avi Kivity6aa8b732006-12-10 02:21:36 -080019
Avi Kivityedf88412007-12-16 11:02:48 +020020#include <linux/kvm_host.h>
Avi Kivity6aa8b732006-12-10 02:21:36 -080021#include <linux/kvm.h>
22#include <linux/module.h>
23#include <linux/errno.h>
Avi Kivity6aa8b732006-12-10 02:21:36 -080024#include <linux/percpu.h>
25#include <linux/gfp.h>
Avi Kivity6aa8b732006-12-10 02:21:36 -080026#include <linux/mm.h>
27#include <linux/miscdevice.h>
28#include <linux/vmalloc.h>
Avi Kivity6aa8b732006-12-10 02:21:36 -080029#include <linux/reboot.h>
Avi Kivity6aa8b732006-12-10 02:21:36 -080030#include <linux/debugfs.h>
31#include <linux/highmem.h>
32#include <linux/file.h>
Avi Kivity59ae6c62007-02-12 00:54:48 -080033#include <linux/sysdev.h>
Avi Kivity774c47f2007-02-12 00:54:47 -080034#include <linux/cpu.h>
Alexey Dobriyane8edc6e2007-05-21 01:22:52 +040035#include <linux/sched.h>
Avi Kivityd9e368d2007-06-07 19:18:30 +030036#include <linux/cpumask.h>
37#include <linux/smp.h>
Avi Kivityd6d28162007-06-28 08:38:16 -040038#include <linux/anon_inodes.h>
Avi Kivity04d2cc72007-09-10 18:10:54 +030039#include <linux/profile.h>
Anthony Liguori7aa81cc2007-09-17 14:57:50 -050040#include <linux/kvm_para.h>
Izik Eidus6fc138d2007-10-09 19:20:39 +020041#include <linux/pagemap.h>
Anthony Liguori8d4e1282007-10-18 09:59:34 -050042#include <linux/mman.h>
Anthony Liguori35149e22008-04-02 14:46:56 -050043#include <linux/swap.h>
Avi Kivity6aa8b732006-12-10 02:21:36 -080044
Avi Kivitye4956062007-06-28 14:15:57 -040045#include <asm/processor.h>
Avi Kivitye4956062007-06-28 14:15:57 -040046#include <asm/io.h>
47#include <asm/uaccess.h>
Izik Eidus3e021bf2007-11-19 11:16:57 +020048#include <asm/pgtable.h>
Avi Kivity6aa8b732006-12-10 02:21:36 -080049
Sheng Yangf64769e2008-11-24 14:32:55 +080050#ifdef CONFIG_X86
51#include <asm/msidef.h>
52#endif
53
Laurent Vivier5f94c172008-05-30 16:05:54 +020054#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
55#include "coalesced_mmio.h"
56#endif
57
Xiantao Zhang8a98f662008-10-06 13:47:38 +080058#ifdef KVM_CAP_DEVICE_ASSIGNMENT
59#include <linux/pci.h>
60#include <linux/interrupt.h>
61#include "irq.h"
62#endif
63
Avi Kivity6aa8b732006-12-10 02:21:36 -080064MODULE_AUTHOR("Qumranet");
65MODULE_LICENSE("GPL");
66
Sheng Yang5319c662008-11-24 14:32:57 +080067static int msi2intx = 1;
68module_param(msi2intx, bool, 0);
69
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +080070DEFINE_SPINLOCK(kvm_lock);
71LIST_HEAD(vm_list);
Avi Kivity133de902007-02-12 00:54:44 -080072
Rusty Russell7f59f492008-12-07 21:25:45 +103073static cpumask_var_t cpus_hardware_enabled;
Avi Kivity1b6c0162007-05-24 13:03:52 +030074
Rusty Russellc16f8622007-07-30 21:12:19 +100075struct kmem_cache *kvm_vcpu_cache;
76EXPORT_SYMBOL_GPL(kvm_vcpu_cache);
Avi Kivity1165f5f2007-04-19 17:27:43 +030077
Avi Kivity15ad7142007-07-11 18:17:21 +030078static __read_mostly struct preempt_ops kvm_preempt_ops;
79
Hollis Blanchard76f7c872008-04-15 16:05:42 -050080struct dentry *kvm_debugfs_dir;
Avi Kivity6aa8b732006-12-10 02:21:36 -080081
Avi Kivitybccf2152007-02-21 18:04:26 +020082static long kvm_vcpu_ioctl(struct file *file, unsigned int ioctl,
83 unsigned long arg);
84
Hannes Edere8ba5d32008-11-28 17:02:42 +010085static bool kvm_rebooting;
Avi Kivity4ecac3f2008-05-13 13:23:38 +030086
Xiantao Zhang8a98f662008-10-06 13:47:38 +080087#ifdef KVM_CAP_DEVICE_ASSIGNMENT
Sheng Yangf64769e2008-11-24 14:32:55 +080088
89#ifdef CONFIG_X86
90static void assigned_device_msi_dispatch(struct kvm_assigned_dev_kernel *dev)
91{
92 int vcpu_id;
93 struct kvm_vcpu *vcpu;
94 struct kvm_ioapic *ioapic = ioapic_irqchip(dev->kvm);
95 int dest_id = (dev->guest_msi.address_lo & MSI_ADDR_DEST_ID_MASK)
96 >> MSI_ADDR_DEST_ID_SHIFT;
97 int vector = (dev->guest_msi.data & MSI_DATA_VECTOR_MASK)
98 >> MSI_DATA_VECTOR_SHIFT;
99 int dest_mode = test_bit(MSI_ADDR_DEST_MODE_SHIFT,
100 (unsigned long *)&dev->guest_msi.address_lo);
101 int trig_mode = test_bit(MSI_DATA_TRIGGER_SHIFT,
102 (unsigned long *)&dev->guest_msi.data);
103 int delivery_mode = test_bit(MSI_DATA_DELIVERY_MODE_SHIFT,
104 (unsigned long *)&dev->guest_msi.data);
105 u32 deliver_bitmask;
106
107 BUG_ON(!ioapic);
108
109 deliver_bitmask = kvm_ioapic_get_delivery_bitmask(ioapic,
110 dest_id, dest_mode);
111 /* IOAPIC delivery mode value is the same as MSI here */
112 switch (delivery_mode) {
113 case IOAPIC_LOWEST_PRIORITY:
114 vcpu = kvm_get_lowest_prio_vcpu(ioapic->kvm, vector,
115 deliver_bitmask);
116 if (vcpu != NULL)
117 kvm_apic_set_irq(vcpu, vector, trig_mode);
118 else
119 printk(KERN_INFO "kvm: null lowest priority vcpu!\n");
120 break;
121 case IOAPIC_FIXED:
122 for (vcpu_id = 0; deliver_bitmask != 0; vcpu_id++) {
123 if (!(deliver_bitmask & (1 << vcpu_id)))
124 continue;
125 deliver_bitmask &= ~(1 << vcpu_id);
126 vcpu = ioapic->kvm->vcpus[vcpu_id];
127 if (vcpu)
128 kvm_apic_set_irq(vcpu, vector, trig_mode);
129 }
130 break;
131 default:
132 printk(KERN_INFO "kvm: unsupported MSI delivery mode\n");
133 }
134}
135#else
136static void assigned_device_msi_dispatch(struct kvm_assigned_dev_kernel *dev) {}
137#endif
138
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800139static struct kvm_assigned_dev_kernel *kvm_find_assigned_dev(struct list_head *head,
140 int assigned_dev_id)
141{
142 struct list_head *ptr;
143 struct kvm_assigned_dev_kernel *match;
144
145 list_for_each(ptr, head) {
146 match = list_entry(ptr, struct kvm_assigned_dev_kernel, list);
147 if (match->assigned_dev_id == assigned_dev_id)
148 return match;
149 }
150 return NULL;
151}
152
153static void kvm_assigned_dev_interrupt_work_handler(struct work_struct *work)
154{
155 struct kvm_assigned_dev_kernel *assigned_dev;
156
157 assigned_dev = container_of(work, struct kvm_assigned_dev_kernel,
158 interrupt_work);
159
160 /* This is taken to safely inject irq inside the guest. When
161 * the interrupt injection (or the ioapic code) uses a
162 * finer-grained lock, update this
163 */
164 mutex_lock(&assigned_dev->kvm->lock);
Sheng Yang6b9cc7f2008-11-24 14:32:56 +0800165 if (assigned_dev->irq_requested_type & KVM_ASSIGNED_DEV_GUEST_INTX)
166 kvm_set_irq(assigned_dev->kvm,
167 assigned_dev->irq_source_id,
168 assigned_dev->guest_irq, 1);
169 else if (assigned_dev->irq_requested_type &
170 KVM_ASSIGNED_DEV_GUEST_MSI) {
171 assigned_device_msi_dispatch(assigned_dev);
172 enable_irq(assigned_dev->host_irq);
Mark McLoughlindefaf152008-12-02 12:16:33 +0000173 assigned_dev->host_irq_disabled = false;
Sheng Yang6b9cc7f2008-11-24 14:32:56 +0800174 }
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800175 mutex_unlock(&assigned_dev->kvm->lock);
176 kvm_put_kvm(assigned_dev->kvm);
177}
178
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800179static irqreturn_t kvm_assigned_dev_intr(int irq, void *dev_id)
180{
181 struct kvm_assigned_dev_kernel *assigned_dev =
182 (struct kvm_assigned_dev_kernel *) dev_id;
183
184 kvm_get_kvm(assigned_dev->kvm);
Mark McLoughlindefaf152008-12-02 12:16:33 +0000185
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800186 schedule_work(&assigned_dev->interrupt_work);
Mark McLoughlindefaf152008-12-02 12:16:33 +0000187
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800188 disable_irq_nosync(irq);
Mark McLoughlindefaf152008-12-02 12:16:33 +0000189 assigned_dev->host_irq_disabled = true;
190
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800191 return IRQ_HANDLED;
192}
193
194/* Ack the irq line for an assigned device */
195static void kvm_assigned_dev_ack_irq(struct kvm_irq_ack_notifier *kian)
196{
197 struct kvm_assigned_dev_kernel *dev;
198
199 if (kian->gsi == -1)
200 return;
201
202 dev = container_of(kian, struct kvm_assigned_dev_kernel,
203 ack_notifier);
Mark McLoughlindefaf152008-12-02 12:16:33 +0000204
Sheng Yang5550af42008-10-15 20:15:06 +0800205 kvm_set_irq(dev->kvm, dev->irq_source_id, dev->guest_irq, 0);
Mark McLoughlindefaf152008-12-02 12:16:33 +0000206
207 /* The guest irq may be shared so this ack may be
208 * from another device.
209 */
210 if (dev->host_irq_disabled) {
211 enable_irq(dev->host_irq);
212 dev->host_irq_disabled = false;
213 }
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800214}
215
Mark McLoughlin4a643be2008-12-01 13:57:49 +0000216static void kvm_free_assigned_irq(struct kvm *kvm,
217 struct kvm_assigned_dev_kernel *assigned_dev)
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800218{
Mark McLoughlin4a643be2008-12-01 13:57:49 +0000219 if (!irqchip_in_kernel(kvm))
220 return;
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800221
Sheng Yange19e30e2008-10-20 16:07:10 +0800222 kvm_unregister_irq_ack_notifier(&assigned_dev->ack_notifier);
Mark McLoughlinf29b2672008-12-01 13:57:47 +0000223
224 if (assigned_dev->irq_source_id != -1)
225 kvm_free_irq_source_id(kvm, assigned_dev->irq_source_id);
226 assigned_dev->irq_source_id = -1;
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800227
Mark McLoughlin4a643be2008-12-01 13:57:49 +0000228 if (!assigned_dev->irq_requested_type)
229 return;
230
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800231 if (cancel_work_sync(&assigned_dev->interrupt_work))
232 /* We had pending work. That means we will have to take
233 * care of kvm_put_kvm.
234 */
235 kvm_put_kvm(kvm);
236
Mark McLoughlin4a643be2008-12-01 13:57:49 +0000237 free_irq(assigned_dev->host_irq, (void *)assigned_dev);
238
239 if (assigned_dev->irq_requested_type & KVM_ASSIGNED_DEV_HOST_MSI)
240 pci_disable_msi(assigned_dev->dev);
241
242 assigned_dev->irq_requested_type = 0;
243}
244
245
246static void kvm_free_assigned_device(struct kvm *kvm,
247 struct kvm_assigned_dev_kernel
248 *assigned_dev)
249{
250 kvm_free_assigned_irq(kvm, assigned_dev);
251
Sheng Yang6eb55812008-10-31 12:37:41 +0800252 pci_reset_function(assigned_dev->dev);
253
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800254 pci_release_regions(assigned_dev->dev);
255 pci_disable_device(assigned_dev->dev);
256 pci_dev_put(assigned_dev->dev);
257
258 list_del(&assigned_dev->list);
259 kfree(assigned_dev);
260}
261
262void kvm_free_all_assigned_devices(struct kvm *kvm)
263{
264 struct list_head *ptr, *ptr2;
265 struct kvm_assigned_dev_kernel *assigned_dev;
266
267 list_for_each_safe(ptr, ptr2, &kvm->arch.assigned_dev_head) {
268 assigned_dev = list_entry(ptr,
269 struct kvm_assigned_dev_kernel,
270 list);
271
272 kvm_free_assigned_device(kvm, assigned_dev);
273 }
274}
275
Sheng Yang00e3ed32008-11-24 14:32:50 +0800276static int assigned_device_update_intx(struct kvm *kvm,
277 struct kvm_assigned_dev_kernel *adev,
278 struct kvm_assigned_irq *airq)
279{
Sheng Yangfbac7812008-11-24 14:32:52 +0800280 adev->guest_irq = airq->guest_irq;
281 adev->ack_notifier.gsi = airq->guest_irq;
282
283 if (adev->irq_requested_type & KVM_ASSIGNED_DEV_HOST_INTX)
Sheng Yang00e3ed32008-11-24 14:32:50 +0800284 return 0;
Sheng Yang00e3ed32008-11-24 14:32:50 +0800285
286 if (irqchip_in_kernel(kvm)) {
Sheng Yang5319c662008-11-24 14:32:57 +0800287 if (!msi2intx &&
288 adev->irq_requested_type & KVM_ASSIGNED_DEV_HOST_MSI) {
Sheng Yang6b9cc7f2008-11-24 14:32:56 +0800289 free_irq(adev->host_irq, (void *)kvm);
290 pci_disable_msi(adev->dev);
291 }
292
Sheng Yang00e3ed32008-11-24 14:32:50 +0800293 if (!capable(CAP_SYS_RAWIO))
294 return -EPERM;
295
296 if (airq->host_irq)
297 adev->host_irq = airq->host_irq;
298 else
299 adev->host_irq = adev->dev->irq;
Sheng Yang00e3ed32008-11-24 14:32:50 +0800300
301 /* Even though this is PCI, we don't want to use shared
302 * interrupts. Sharing host devices with guest-assigned devices
303 * on the same interrupt line is not a happy situation: there
304 * are going to be long delays in accepting, acking, etc.
305 */
306 if (request_irq(adev->host_irq, kvm_assigned_dev_intr,
307 0, "kvm_assigned_intx_device", (void *)adev))
308 return -EIO;
309 }
310
Sheng Yang4f906c12008-11-24 14:32:51 +0800311 adev->irq_requested_type = KVM_ASSIGNED_DEV_GUEST_INTX |
312 KVM_ASSIGNED_DEV_HOST_INTX;
Sheng Yang00e3ed32008-11-24 14:32:50 +0800313 return 0;
314}
315
Sheng Yang6b9cc7f2008-11-24 14:32:56 +0800316#ifdef CONFIG_X86
317static int assigned_device_update_msi(struct kvm *kvm,
318 struct kvm_assigned_dev_kernel *adev,
319 struct kvm_assigned_irq *airq)
320{
321 int r;
322
Sheng Yang5319c662008-11-24 14:32:57 +0800323 if (airq->flags & KVM_DEV_IRQ_ASSIGN_ENABLE_MSI) {
324 /* x86 don't care upper address of guest msi message addr */
325 adev->irq_requested_type |= KVM_ASSIGNED_DEV_GUEST_MSI;
326 adev->irq_requested_type &= ~KVM_ASSIGNED_DEV_GUEST_INTX;
327 adev->guest_msi.address_lo = airq->guest_msi.addr_lo;
328 adev->guest_msi.data = airq->guest_msi.data;
329 adev->ack_notifier.gsi = -1;
330 } else if (msi2intx) {
331 adev->irq_requested_type |= KVM_ASSIGNED_DEV_GUEST_INTX;
332 adev->irq_requested_type &= ~KVM_ASSIGNED_DEV_GUEST_MSI;
333 adev->guest_irq = airq->guest_irq;
334 adev->ack_notifier.gsi = airq->guest_irq;
335 }
Sheng Yang6b9cc7f2008-11-24 14:32:56 +0800336
337 if (adev->irq_requested_type & KVM_ASSIGNED_DEV_HOST_MSI)
338 return 0;
339
340 if (irqchip_in_kernel(kvm)) {
Sheng Yang5319c662008-11-24 14:32:57 +0800341 if (!msi2intx) {
342 if (adev->irq_requested_type &
343 KVM_ASSIGNED_DEV_HOST_INTX)
344 free_irq(adev->host_irq, (void *)adev);
Sheng Yang6b9cc7f2008-11-24 14:32:56 +0800345
Sheng Yang5319c662008-11-24 14:32:57 +0800346 r = pci_enable_msi(adev->dev);
347 if (r)
348 return r;
349 }
Sheng Yang6b9cc7f2008-11-24 14:32:56 +0800350
351 adev->host_irq = adev->dev->irq;
352 if (request_irq(adev->host_irq, kvm_assigned_dev_intr, 0,
353 "kvm_assigned_msi_device", (void *)adev))
354 return -EIO;
355 }
356
Sheng Yang5319c662008-11-24 14:32:57 +0800357 if (!msi2intx)
358 adev->irq_requested_type = KVM_ASSIGNED_DEV_GUEST_MSI;
359
360 adev->irq_requested_type |= KVM_ASSIGNED_DEV_HOST_MSI;
Sheng Yang6b9cc7f2008-11-24 14:32:56 +0800361 return 0;
362}
363#endif
364
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800365static int kvm_vm_ioctl_assign_irq(struct kvm *kvm,
366 struct kvm_assigned_irq
367 *assigned_irq)
368{
369 int r = 0;
370 struct kvm_assigned_dev_kernel *match;
371
372 mutex_lock(&kvm->lock);
373
374 match = kvm_find_assigned_dev(&kvm->arch.assigned_dev_head,
375 assigned_irq->assigned_dev_id);
376 if (!match) {
377 mutex_unlock(&kvm->lock);
378 return -EINVAL;
379 }
380
Sheng Yang4f906c12008-11-24 14:32:51 +0800381 if (!match->irq_requested_type) {
Sheng Yang342ffb92008-11-24 14:32:49 +0800382 INIT_WORK(&match->interrupt_work,
383 kvm_assigned_dev_interrupt_work_handler);
384 if (irqchip_in_kernel(kvm)) {
385 /* Register ack nofitier */
386 match->ack_notifier.gsi = -1;
387 match->ack_notifier.irq_acked =
388 kvm_assigned_dev_ack_irq;
389 kvm_register_irq_ack_notifier(kvm,
390 &match->ack_notifier);
391
392 /* Request IRQ source ID */
393 r = kvm_request_irq_source_id(kvm);
394 if (r < 0)
395 goto out_release;
396 else
397 match->irq_source_id = r;
Sheng Yang5319c662008-11-24 14:32:57 +0800398
399#ifdef CONFIG_X86
400 /* Determine host device irq type, we can know the
401 * result from dev->msi_enabled */
402 if (msi2intx)
403 pci_enable_msi(match->dev);
404#endif
Sheng Yang342ffb92008-11-24 14:32:49 +0800405 }
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800406 }
407
Sheng Yang5319c662008-11-24 14:32:57 +0800408 if ((!msi2intx &&
409 (assigned_irq->flags & KVM_DEV_IRQ_ASSIGN_ENABLE_MSI)) ||
410 (msi2intx && match->dev->msi_enabled)) {
Sheng Yang6b9cc7f2008-11-24 14:32:56 +0800411#ifdef CONFIG_X86
412 r = assigned_device_update_msi(kvm, match, assigned_irq);
413 if (r) {
414 printk(KERN_WARNING "kvm: failed to enable "
415 "MSI device!\n");
416 goto out_release;
417 }
418#else
419 r = -ENOTTY;
420#endif
421 } else if (assigned_irq->host_irq == 0 && match->dev->irq == 0) {
422 /* Host device IRQ 0 means don't support INTx */
Sheng Yang5319c662008-11-24 14:32:57 +0800423 if (!msi2intx) {
424 printk(KERN_WARNING
425 "kvm: wait device to enable MSI!\n");
426 r = 0;
427 } else {
428 printk(KERN_WARNING
429 "kvm: failed to enable MSI device!\n");
430 r = -ENOTTY;
431 goto out_release;
432 }
Sheng Yang6b9cc7f2008-11-24 14:32:56 +0800433 } else {
434 /* Non-sharing INTx mode */
435 r = assigned_device_update_intx(kvm, match, assigned_irq);
436 if (r) {
437 printk(KERN_WARNING "kvm: failed to enable "
438 "INTx device!\n");
439 goto out_release;
440 }
441 }
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800442
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800443 mutex_unlock(&kvm->lock);
444 return r;
445out_release:
446 mutex_unlock(&kvm->lock);
447 kvm_free_assigned_device(kvm, match);
448 return r;
449}
450
451static int kvm_vm_ioctl_assign_device(struct kvm *kvm,
452 struct kvm_assigned_pci_dev *assigned_dev)
453{
454 int r = 0;
455 struct kvm_assigned_dev_kernel *match;
456 struct pci_dev *dev;
457
458 mutex_lock(&kvm->lock);
459
460 match = kvm_find_assigned_dev(&kvm->arch.assigned_dev_head,
461 assigned_dev->assigned_dev_id);
462 if (match) {
463 /* device already assigned */
464 r = -EINVAL;
465 goto out;
466 }
467
468 match = kzalloc(sizeof(struct kvm_assigned_dev_kernel), GFP_KERNEL);
469 if (match == NULL) {
470 printk(KERN_INFO "%s: Couldn't allocate memory\n",
471 __func__);
472 r = -ENOMEM;
473 goto out;
474 }
475 dev = pci_get_bus_and_slot(assigned_dev->busnr,
476 assigned_dev->devfn);
477 if (!dev) {
478 printk(KERN_INFO "%s: host device not found\n", __func__);
479 r = -EINVAL;
480 goto out_free;
481 }
482 if (pci_enable_device(dev)) {
483 printk(KERN_INFO "%s: Could not enable PCI device\n", __func__);
484 r = -EBUSY;
485 goto out_put;
486 }
487 r = pci_request_regions(dev, "kvm_assigned_device");
488 if (r) {
489 printk(KERN_INFO "%s: Could not get access to device regions\n",
490 __func__);
491 goto out_disable;
492 }
Sheng Yang6eb55812008-10-31 12:37:41 +0800493
494 pci_reset_function(dev);
495
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800496 match->assigned_dev_id = assigned_dev->assigned_dev_id;
497 match->host_busnr = assigned_dev->busnr;
498 match->host_devfn = assigned_dev->devfn;
Weidong Hanb6535742008-12-08 23:29:53 +0800499 match->flags = assigned_dev->flags;
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800500 match->dev = dev;
Mark McLoughlinf29b2672008-12-01 13:57:47 +0000501 match->irq_source_id = -1;
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800502 match->kvm = kvm;
503
504 list_add(&match->list, &kvm->arch.assigned_dev_head);
505
506 if (assigned_dev->flags & KVM_DEV_ASSIGN_ENABLE_IOMMU) {
Joerg Roedel19de40a2008-12-03 14:43:34 +0100507 if (!kvm->arch.iommu_domain) {
Weidong Han260782b2008-12-02 21:03:39 +0800508 r = kvm_iommu_map_guest(kvm);
509 if (r)
510 goto out_list_del;
511 }
512 r = kvm_assign_device(kvm, match);
Xiantao Zhang8a98f662008-10-06 13:47:38 +0800513 if (r)
514 goto out_list_del;
515 }
516
517out:
518 mutex_unlock(&kvm->lock);
519 return r;
520out_list_del:
521 list_del(&match->list);
522 pci_release_regions(dev);
523out_disable:
524 pci_disable_device(dev);
525out_put:
526 pci_dev_put(dev);
527out_free:
528 kfree(match);
529 mutex_unlock(&kvm->lock);
530 return r;
531}
532#endif
533
Weidong Han0a920352008-12-02 21:24:23 +0800534#ifdef KVM_CAP_DEVICE_DEASSIGNMENT
535static int kvm_vm_ioctl_deassign_device(struct kvm *kvm,
536 struct kvm_assigned_pci_dev *assigned_dev)
537{
538 int r = 0;
539 struct kvm_assigned_dev_kernel *match;
540
541 mutex_lock(&kvm->lock);
542
543 match = kvm_find_assigned_dev(&kvm->arch.assigned_dev_head,
544 assigned_dev->assigned_dev_id);
545 if (!match) {
546 printk(KERN_INFO "%s: device hasn't been assigned before, "
547 "so cannot be deassigned\n", __func__);
548 r = -EINVAL;
549 goto out;
550 }
551
552 if (assigned_dev->flags & KVM_DEV_ASSIGN_ENABLE_IOMMU)
553 kvm_deassign_device(kvm, match);
554
555 kvm_free_assigned_device(kvm, match);
556
557out:
558 mutex_unlock(&kvm->lock);
559 return r;
560}
561#endif
562
James Morris5aacf0c2006-12-22 01:04:55 -0800563static inline int valid_vcpu(int n)
564{
565 return likely(n >= 0 && n < KVM_MAX_VCPUS);
566}
567
Xiantao Zhangc77fb9d2008-09-27 10:55:40 +0800568inline int kvm_is_mmio_pfn(pfn_t pfn)
Ben-Ami Yassourcbff90a2008-07-28 19:26:24 +0300569{
570 if (pfn_valid(pfn))
571 return PageReserved(pfn_to_page(pfn));
572
573 return true;
574}
575
Avi Kivity6aa8b732006-12-10 02:21:36 -0800576/*
577 * Switches to specified vcpu, until a matching vcpu_put()
578 */
Carsten Otte313a3dc2007-10-11 19:16:52 +0200579void vcpu_load(struct kvm_vcpu *vcpu)
Avi Kivity6aa8b732006-12-10 02:21:36 -0800580{
Avi Kivity15ad7142007-07-11 18:17:21 +0300581 int cpu;
582
Avi Kivitybccf2152007-02-21 18:04:26 +0200583 mutex_lock(&vcpu->mutex);
Avi Kivity15ad7142007-07-11 18:17:21 +0300584 cpu = get_cpu();
585 preempt_notifier_register(&vcpu->preempt_notifier);
Carsten Otte313a3dc2007-10-11 19:16:52 +0200586 kvm_arch_vcpu_load(vcpu, cpu);
Avi Kivity15ad7142007-07-11 18:17:21 +0300587 put_cpu();
Avi Kivitybccf2152007-02-21 18:04:26 +0200588}
589
Carsten Otte313a3dc2007-10-11 19:16:52 +0200590void vcpu_put(struct kvm_vcpu *vcpu)
Avi Kivity6aa8b732006-12-10 02:21:36 -0800591{
Avi Kivity15ad7142007-07-11 18:17:21 +0300592 preempt_disable();
Carsten Otte313a3dc2007-10-11 19:16:52 +0200593 kvm_arch_vcpu_put(vcpu);
Avi Kivity15ad7142007-07-11 18:17:21 +0300594 preempt_notifier_unregister(&vcpu->preempt_notifier);
595 preempt_enable();
Avi Kivity6aa8b732006-12-10 02:21:36 -0800596 mutex_unlock(&vcpu->mutex);
597}
598
Avi Kivityd9e368d2007-06-07 19:18:30 +0300599static void ack_flush(void *_completed)
600{
Avi Kivityd9e368d2007-06-07 19:18:30 +0300601}
602
Rusty Russell49846892008-12-08 20:26:24 +1030603static bool make_all_cpus_request(struct kvm *kvm, unsigned int req)
Avi Kivityd9e368d2007-06-07 19:18:30 +0300604{
Avi Kivity597a5f52008-07-20 14:24:22 +0300605 int i, cpu, me;
Rusty Russell6ef7a1b2008-12-08 20:28:04 +1030606 cpumask_var_t cpus;
607 bool called = true;
Avi Kivityd9e368d2007-06-07 19:18:30 +0300608 struct kvm_vcpu *vcpu;
Avi Kivityd9e368d2007-06-07 19:18:30 +0300609
Rusty Russell6ef7a1b2008-12-08 20:28:04 +1030610 if (alloc_cpumask_var(&cpus, GFP_ATOMIC))
611 cpumask_clear(cpus);
612
Avi Kivity597a5f52008-07-20 14:24:22 +0300613 me = get_cpu();
Rusty Russellfb3f0f52007-07-27 17:16:56 +1000614 for (i = 0; i < KVM_MAX_VCPUS; ++i) {
615 vcpu = kvm->vcpus[i];
616 if (!vcpu)
617 continue;
Rusty Russell49846892008-12-08 20:26:24 +1030618 if (test_and_set_bit(req, &vcpu->requests))
Avi Kivityd9e368d2007-06-07 19:18:30 +0300619 continue;
620 cpu = vcpu->cpu;
Rusty Russell6ef7a1b2008-12-08 20:28:04 +1030621 if (cpus != NULL && cpu != -1 && cpu != me)
622 cpumask_set_cpu(cpu, cpus);
Avi Kivityd9e368d2007-06-07 19:18:30 +0300623 }
Rusty Russell6ef7a1b2008-12-08 20:28:04 +1030624 if (unlikely(cpus == NULL))
625 smp_call_function_many(cpu_online_mask, ack_flush, NULL, 1);
626 else if (!cpumask_empty(cpus))
627 smp_call_function_many(cpus, ack_flush, NULL, 1);
628 else
629 called = false;
Avi Kivity597a5f52008-07-20 14:24:22 +0300630 put_cpu();
Rusty Russell6ef7a1b2008-12-08 20:28:04 +1030631 free_cpumask_var(cpus);
Rusty Russell49846892008-12-08 20:26:24 +1030632 return called;
633}
634
635void kvm_flush_remote_tlbs(struct kvm *kvm)
636{
637 if (make_all_cpus_request(kvm, KVM_REQ_TLB_FLUSH))
638 ++kvm->stat.remote_tlb_flush;
Avi Kivityd9e368d2007-06-07 19:18:30 +0300639}
640
Marcelo Tosatti2e53d632008-02-20 14:47:24 -0500641void kvm_reload_remote_mmus(struct kvm *kvm)
642{
Rusty Russell49846892008-12-08 20:26:24 +1030643 make_all_cpus_request(kvm, KVM_REQ_MMU_RELOAD);
Marcelo Tosatti2e53d632008-02-20 14:47:24 -0500644}
645
Rusty Russellfb3f0f52007-07-27 17:16:56 +1000646int kvm_vcpu_init(struct kvm_vcpu *vcpu, struct kvm *kvm, unsigned id)
647{
648 struct page *page;
649 int r;
650
651 mutex_init(&vcpu->mutex);
652 vcpu->cpu = -1;
Rusty Russellfb3f0f52007-07-27 17:16:56 +1000653 vcpu->kvm = kvm;
654 vcpu->vcpu_id = id;
Eddie Dongb6958ce2007-07-18 12:15:21 +0300655 init_waitqueue_head(&vcpu->wq);
Rusty Russellfb3f0f52007-07-27 17:16:56 +1000656
657 page = alloc_page(GFP_KERNEL | __GFP_ZERO);
658 if (!page) {
659 r = -ENOMEM;
660 goto fail;
661 }
662 vcpu->run = page_address(page);
663
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +0800664 r = kvm_arch_vcpu_init(vcpu);
Rusty Russellfb3f0f52007-07-27 17:16:56 +1000665 if (r < 0)
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +0800666 goto fail_free_run;
Rusty Russellfb3f0f52007-07-27 17:16:56 +1000667 return 0;
668
Rusty Russellfb3f0f52007-07-27 17:16:56 +1000669fail_free_run:
670 free_page((unsigned long)vcpu->run);
671fail:
Rusty Russell76fafa52007-10-08 10:50:48 +1000672 return r;
Rusty Russellfb3f0f52007-07-27 17:16:56 +1000673}
674EXPORT_SYMBOL_GPL(kvm_vcpu_init);
675
676void kvm_vcpu_uninit(struct kvm_vcpu *vcpu)
677{
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +0800678 kvm_arch_vcpu_uninit(vcpu);
Rusty Russellfb3f0f52007-07-27 17:16:56 +1000679 free_page((unsigned long)vcpu->run);
680}
681EXPORT_SYMBOL_GPL(kvm_vcpu_uninit);
682
Andrea Arcangelie930bff2008-07-25 16:24:52 +0200683#if defined(CONFIG_MMU_NOTIFIER) && defined(KVM_ARCH_WANT_MMU_NOTIFIER)
684static inline struct kvm *mmu_notifier_to_kvm(struct mmu_notifier *mn)
685{
686 return container_of(mn, struct kvm, mmu_notifier);
687}
688
689static void kvm_mmu_notifier_invalidate_page(struct mmu_notifier *mn,
690 struct mm_struct *mm,
691 unsigned long address)
692{
693 struct kvm *kvm = mmu_notifier_to_kvm(mn);
694 int need_tlb_flush;
695
696 /*
697 * When ->invalidate_page runs, the linux pte has been zapped
698 * already but the page is still allocated until
699 * ->invalidate_page returns. So if we increase the sequence
700 * here the kvm page fault will notice if the spte can't be
701 * established because the page is going to be freed. If
702 * instead the kvm page fault establishes the spte before
703 * ->invalidate_page runs, kvm_unmap_hva will release it
704 * before returning.
705 *
706 * The sequence increase only need to be seen at spin_unlock
707 * time, and not at spin_lock time.
708 *
709 * Increasing the sequence after the spin_unlock would be
710 * unsafe because the kvm page fault could then establish the
711 * pte after kvm_unmap_hva returned, without noticing the page
712 * is going to be freed.
713 */
714 spin_lock(&kvm->mmu_lock);
715 kvm->mmu_notifier_seq++;
716 need_tlb_flush = kvm_unmap_hva(kvm, address);
717 spin_unlock(&kvm->mmu_lock);
718
719 /* we've to flush the tlb before the pages can be freed */
720 if (need_tlb_flush)
721 kvm_flush_remote_tlbs(kvm);
722
723}
724
725static void kvm_mmu_notifier_invalidate_range_start(struct mmu_notifier *mn,
726 struct mm_struct *mm,
727 unsigned long start,
728 unsigned long end)
729{
730 struct kvm *kvm = mmu_notifier_to_kvm(mn);
731 int need_tlb_flush = 0;
732
733 spin_lock(&kvm->mmu_lock);
734 /*
735 * The count increase must become visible at unlock time as no
736 * spte can be established without taking the mmu_lock and
737 * count is also read inside the mmu_lock critical section.
738 */
739 kvm->mmu_notifier_count++;
740 for (; start < end; start += PAGE_SIZE)
741 need_tlb_flush |= kvm_unmap_hva(kvm, start);
742 spin_unlock(&kvm->mmu_lock);
743
744 /* we've to flush the tlb before the pages can be freed */
745 if (need_tlb_flush)
746 kvm_flush_remote_tlbs(kvm);
747}
748
749static void kvm_mmu_notifier_invalidate_range_end(struct mmu_notifier *mn,
750 struct mm_struct *mm,
751 unsigned long start,
752 unsigned long end)
753{
754 struct kvm *kvm = mmu_notifier_to_kvm(mn);
755
756 spin_lock(&kvm->mmu_lock);
757 /*
758 * This sequence increase will notify the kvm page fault that
759 * the page that is going to be mapped in the spte could have
760 * been freed.
761 */
762 kvm->mmu_notifier_seq++;
763 /*
764 * The above sequence increase must be visible before the
765 * below count decrease but both values are read by the kvm
766 * page fault under mmu_lock spinlock so we don't need to add
767 * a smb_wmb() here in between the two.
768 */
769 kvm->mmu_notifier_count--;
770 spin_unlock(&kvm->mmu_lock);
771
772 BUG_ON(kvm->mmu_notifier_count < 0);
773}
774
775static int kvm_mmu_notifier_clear_flush_young(struct mmu_notifier *mn,
776 struct mm_struct *mm,
777 unsigned long address)
778{
779 struct kvm *kvm = mmu_notifier_to_kvm(mn);
780 int young;
781
782 spin_lock(&kvm->mmu_lock);
783 young = kvm_age_hva(kvm, address);
784 spin_unlock(&kvm->mmu_lock);
785
786 if (young)
787 kvm_flush_remote_tlbs(kvm);
788
789 return young;
790}
791
792static const struct mmu_notifier_ops kvm_mmu_notifier_ops = {
793 .invalidate_page = kvm_mmu_notifier_invalidate_page,
794 .invalidate_range_start = kvm_mmu_notifier_invalidate_range_start,
795 .invalidate_range_end = kvm_mmu_notifier_invalidate_range_end,
796 .clear_flush_young = kvm_mmu_notifier_clear_flush_young,
797};
798#endif /* CONFIG_MMU_NOTIFIER && KVM_ARCH_WANT_MMU_NOTIFIER */
799
Avi Kivityf17abe92007-02-21 19:28:04 +0200800static struct kvm *kvm_create_vm(void)
Avi Kivity6aa8b732006-12-10 02:21:36 -0800801{
Zhang Xiantaod19a9cd2007-11-18 18:43:45 +0800802 struct kvm *kvm = kvm_arch_create_vm();
Laurent Vivier5f94c172008-05-30 16:05:54 +0200803#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
804 struct page *page;
805#endif
Avi Kivity6aa8b732006-12-10 02:21:36 -0800806
Zhang Xiantaod19a9cd2007-11-18 18:43:45 +0800807 if (IS_ERR(kvm))
808 goto out;
Avi Kivity6aa8b732006-12-10 02:21:36 -0800809
Laurent Vivier5f94c172008-05-30 16:05:54 +0200810#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
811 page = alloc_page(GFP_KERNEL | __GFP_ZERO);
812 if (!page) {
813 kfree(kvm);
814 return ERR_PTR(-ENOMEM);
815 }
816 kvm->coalesced_mmio_ring =
817 (struct kvm_coalesced_mmio_ring *)page_address(page);
818#endif
819
Andrea Arcangelie930bff2008-07-25 16:24:52 +0200820#if defined(CONFIG_MMU_NOTIFIER) && defined(KVM_ARCH_WANT_MMU_NOTIFIER)
821 {
822 int err;
823 kvm->mmu_notifier.ops = &kvm_mmu_notifier_ops;
824 err = mmu_notifier_register(&kvm->mmu_notifier, current->mm);
825 if (err) {
826#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
827 put_page(page);
828#endif
829 kfree(kvm);
830 return ERR_PTR(err);
831 }
832 }
833#endif
834
Avi Kivity6d4e4c42007-11-21 16:41:05 +0200835 kvm->mm = current->mm;
836 atomic_inc(&kvm->mm->mm_count);
Marcelo Tosattiaaee2c92007-12-20 19:18:26 -0500837 spin_lock_init(&kvm->mmu_lock);
Eddie Dong74906342007-06-19 18:05:03 +0300838 kvm_io_bus_init(&kvm->pio_bus);
Shaohua Li11ec2802007-07-23 14:51:37 +0800839 mutex_init(&kvm->lock);
Gregory Haskins2eeb2e92007-05-31 14:08:53 -0400840 kvm_io_bus_init(&kvm->mmio_bus);
Izik Eidus72dc67a2008-02-10 18:04:15 +0200841 init_rwsem(&kvm->slots_lock);
Izik Eidusd39f13b2008-03-30 16:01:25 +0300842 atomic_set(&kvm->users_count, 1);
Rusty Russell5e58cfe2007-07-23 17:08:21 +1000843 spin_lock(&kvm_lock);
844 list_add(&kvm->vm_list, &vm_list);
845 spin_unlock(&kvm_lock);
Laurent Vivier5f94c172008-05-30 16:05:54 +0200846#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
847 kvm_coalesced_mmio_init(kvm);
848#endif
Zhang Xiantaod19a9cd2007-11-18 18:43:45 +0800849out:
Avi Kivityf17abe92007-02-21 19:28:04 +0200850 return kvm;
851}
852
Avi Kivity6aa8b732006-12-10 02:21:36 -0800853/*
854 * Free any memory in @free but not in @dont.
855 */
856static void kvm_free_physmem_slot(struct kvm_memory_slot *free,
857 struct kvm_memory_slot *dont)
858{
Izik Eidus290fc382007-09-27 14:11:22 +0200859 if (!dont || free->rmap != dont->rmap)
860 vfree(free->rmap);
Avi Kivity6aa8b732006-12-10 02:21:36 -0800861
862 if (!dont || free->dirty_bitmap != dont->dirty_bitmap)
863 vfree(free->dirty_bitmap);
864
Marcelo Tosatti05da4552008-02-23 11:44:30 -0300865 if (!dont || free->lpage_info != dont->lpage_info)
866 vfree(free->lpage_info);
867
Avi Kivity6aa8b732006-12-10 02:21:36 -0800868 free->npages = 0;
Al Viro8b6d44c2007-02-09 16:38:40 +0000869 free->dirty_bitmap = NULL;
Anthony Liguori8d4e1282007-10-18 09:59:34 -0500870 free->rmap = NULL;
Marcelo Tosatti05da4552008-02-23 11:44:30 -0300871 free->lpage_info = NULL;
Avi Kivity6aa8b732006-12-10 02:21:36 -0800872}
873
Zhang Xiantaod19a9cd2007-11-18 18:43:45 +0800874void kvm_free_physmem(struct kvm *kvm)
Avi Kivity6aa8b732006-12-10 02:21:36 -0800875{
876 int i;
877
878 for (i = 0; i < kvm->nmemslots; ++i)
Al Viro8b6d44c2007-02-09 16:38:40 +0000879 kvm_free_physmem_slot(&kvm->memslots[i], NULL);
Avi Kivity6aa8b732006-12-10 02:21:36 -0800880}
881
Avi Kivityf17abe92007-02-21 19:28:04 +0200882static void kvm_destroy_vm(struct kvm *kvm)
883{
Avi Kivity6d4e4c42007-11-21 16:41:05 +0200884 struct mm_struct *mm = kvm->mm;
885
Avi Kivity133de902007-02-12 00:54:44 -0800886 spin_lock(&kvm_lock);
887 list_del(&kvm->vm_list);
888 spin_unlock(&kvm_lock);
Eddie Dong74906342007-06-19 18:05:03 +0300889 kvm_io_bus_destroy(&kvm->pio_bus);
Gregory Haskins2eeb2e92007-05-31 14:08:53 -0400890 kvm_io_bus_destroy(&kvm->mmio_bus);
Laurent Vivier5f94c172008-05-30 16:05:54 +0200891#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
892 if (kvm->coalesced_mmio_ring != NULL)
893 free_page((unsigned long)kvm->coalesced_mmio_ring);
894#endif
Andrea Arcangelie930bff2008-07-25 16:24:52 +0200895#if defined(CONFIG_MMU_NOTIFIER) && defined(KVM_ARCH_WANT_MMU_NOTIFIER)
896 mmu_notifier_unregister(&kvm->mmu_notifier, kvm->mm);
897#endif
Zhang Xiantaod19a9cd2007-11-18 18:43:45 +0800898 kvm_arch_destroy_vm(kvm);
Avi Kivity6d4e4c42007-11-21 16:41:05 +0200899 mmdrop(mm);
Avi Kivityf17abe92007-02-21 19:28:04 +0200900}
901
Izik Eidusd39f13b2008-03-30 16:01:25 +0300902void kvm_get_kvm(struct kvm *kvm)
903{
904 atomic_inc(&kvm->users_count);
905}
906EXPORT_SYMBOL_GPL(kvm_get_kvm);
907
908void kvm_put_kvm(struct kvm *kvm)
909{
910 if (atomic_dec_and_test(&kvm->users_count))
911 kvm_destroy_vm(kvm);
912}
913EXPORT_SYMBOL_GPL(kvm_put_kvm);
914
915
Avi Kivityf17abe92007-02-21 19:28:04 +0200916static int kvm_vm_release(struct inode *inode, struct file *filp)
917{
918 struct kvm *kvm = filp->private_data;
919
Izik Eidusd39f13b2008-03-30 16:01:25 +0300920 kvm_put_kvm(kvm);
Avi Kivity6aa8b732006-12-10 02:21:36 -0800921 return 0;
922}
923
Avi Kivity6aa8b732006-12-10 02:21:36 -0800924/*
Avi Kivity6aa8b732006-12-10 02:21:36 -0800925 * Allocate some memory and give it an address in the guest physical address
926 * space.
927 *
928 * Discontiguous memory is allowed, mostly for framebuffers.
Sheng Yangf78e0e22007-10-29 09:40:42 +0800929 *
Marcelo Tosatti10589a42007-12-20 19:18:22 -0500930 * Must be called holding mmap_sem for write.
Avi Kivity6aa8b732006-12-10 02:21:36 -0800931 */
Sheng Yangf78e0e22007-10-29 09:40:42 +0800932int __kvm_set_memory_region(struct kvm *kvm,
933 struct kvm_userspace_memory_region *mem,
934 int user_alloc)
Avi Kivity6aa8b732006-12-10 02:21:36 -0800935{
936 int r;
937 gfn_t base_gfn;
938 unsigned long npages;
939 unsigned long i;
940 struct kvm_memory_slot *memslot;
941 struct kvm_memory_slot old, new;
Avi Kivity6aa8b732006-12-10 02:21:36 -0800942
943 r = -EINVAL;
944 /* General sanity checks */
945 if (mem->memory_size & (PAGE_SIZE - 1))
946 goto out;
947 if (mem->guest_phys_addr & (PAGE_SIZE - 1))
948 goto out;
Sheng Yange7cacd42008-11-11 15:30:40 +0800949 if (user_alloc && (mem->userspace_addr & (PAGE_SIZE - 1)))
Hollis Blanchard78749802008-11-07 13:32:12 -0600950 goto out;
Izik Eiduse0d62c72007-10-24 23:57:46 +0200951 if (mem->slot >= KVM_MEMORY_SLOTS + KVM_PRIVATE_MEM_SLOTS)
Avi Kivity6aa8b732006-12-10 02:21:36 -0800952 goto out;
953 if (mem->guest_phys_addr + mem->memory_size < mem->guest_phys_addr)
954 goto out;
955
956 memslot = &kvm->memslots[mem->slot];
957 base_gfn = mem->guest_phys_addr >> PAGE_SHIFT;
958 npages = mem->memory_size >> PAGE_SHIFT;
959
960 if (!npages)
961 mem->flags &= ~KVM_MEM_LOG_DIRTY_PAGES;
962
Avi Kivity6aa8b732006-12-10 02:21:36 -0800963 new = old = *memslot;
964
965 new.base_gfn = base_gfn;
966 new.npages = npages;
967 new.flags = mem->flags;
968
969 /* Disallow changing a memory slot's size. */
970 r = -EINVAL;
971 if (npages && old.npages && npages != old.npages)
Sheng Yangf78e0e22007-10-29 09:40:42 +0800972 goto out_free;
Avi Kivity6aa8b732006-12-10 02:21:36 -0800973
974 /* Check for overlaps */
975 r = -EEXIST;
976 for (i = 0; i < KVM_MEMORY_SLOTS; ++i) {
977 struct kvm_memory_slot *s = &kvm->memslots[i];
978
979 if (s == memslot)
980 continue;
981 if (!((base_gfn + npages <= s->base_gfn) ||
982 (base_gfn >= s->base_gfn + s->npages)))
Sheng Yangf78e0e22007-10-29 09:40:42 +0800983 goto out_free;
Avi Kivity6aa8b732006-12-10 02:21:36 -0800984 }
Avi Kivity6aa8b732006-12-10 02:21:36 -0800985
Avi Kivity6aa8b732006-12-10 02:21:36 -0800986 /* Free page dirty bitmap if unneeded */
987 if (!(new.flags & KVM_MEM_LOG_DIRTY_PAGES))
Al Viro8b6d44c2007-02-09 16:38:40 +0000988 new.dirty_bitmap = NULL;
Avi Kivity6aa8b732006-12-10 02:21:36 -0800989
990 r = -ENOMEM;
991
992 /* Allocate if a slot is being created */
Carsten Otteeff01142008-06-27 15:05:31 +0200993#ifndef CONFIG_S390
Anthony Liguori8d4e1282007-10-18 09:59:34 -0500994 if (npages && !new.rmap) {
Mike Dayd77c26f2007-10-08 09:02:08 -0400995 new.rmap = vmalloc(npages * sizeof(struct page *));
Izik Eidus290fc382007-09-27 14:11:22 +0200996
997 if (!new.rmap)
Sheng Yangf78e0e22007-10-29 09:40:42 +0800998 goto out_free;
Izik Eidus290fc382007-09-27 14:11:22 +0200999
Izik Eidus290fc382007-09-27 14:11:22 +02001000 memset(new.rmap, 0, npages * sizeof(*new.rmap));
Anthony Liguori8d4e1282007-10-18 09:59:34 -05001001
Izik Eidus80b14b52007-10-25 11:54:04 +02001002 new.user_alloc = user_alloc;
Andrea Arcangeli604b38a2008-07-25 16:32:03 +02001003 /*
1004 * hva_to_rmmap() serialzies with the mmu_lock and to be
1005 * safe it has to ignore memslots with !user_alloc &&
1006 * !userspace_addr.
1007 */
1008 if (user_alloc)
1009 new.userspace_addr = mem->userspace_addr;
1010 else
1011 new.userspace_addr = 0;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001012 }
Marcelo Tosatti05da4552008-02-23 11:44:30 -03001013 if (npages && !new.lpage_info) {
1014 int largepages = npages / KVM_PAGES_PER_HPAGE;
1015 if (npages % KVM_PAGES_PER_HPAGE)
1016 largepages++;
1017 if (base_gfn % KVM_PAGES_PER_HPAGE)
1018 largepages++;
1019
1020 new.lpage_info = vmalloc(largepages * sizeof(*new.lpage_info));
1021
1022 if (!new.lpage_info)
1023 goto out_free;
1024
1025 memset(new.lpage_info, 0, largepages * sizeof(*new.lpage_info));
1026
1027 if (base_gfn % KVM_PAGES_PER_HPAGE)
1028 new.lpage_info[0].write_count = 1;
1029 if ((base_gfn+npages) % KVM_PAGES_PER_HPAGE)
1030 new.lpage_info[largepages-1].write_count = 1;
1031 }
Avi Kivity6aa8b732006-12-10 02:21:36 -08001032
1033 /* Allocate page dirty bitmap if needed */
1034 if ((new.flags & KVM_MEM_LOG_DIRTY_PAGES) && !new.dirty_bitmap) {
1035 unsigned dirty_bytes = ALIGN(npages, BITS_PER_LONG) / 8;
1036
1037 new.dirty_bitmap = vmalloc(dirty_bytes);
1038 if (!new.dirty_bitmap)
Sheng Yangf78e0e22007-10-29 09:40:42 +08001039 goto out_free;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001040 memset(new.dirty_bitmap, 0, dirty_bytes);
1041 }
Carsten Otteeff01142008-06-27 15:05:31 +02001042#endif /* not defined CONFIG_S390 */
Avi Kivity6aa8b732006-12-10 02:21:36 -08001043
Marcelo Tosatti34d4cb82008-07-10 20:49:31 -03001044 if (!npages)
1045 kvm_arch_flush_shadow(kvm);
1046
Andrea Arcangeli604b38a2008-07-25 16:32:03 +02001047 spin_lock(&kvm->mmu_lock);
1048 if (mem->slot >= kvm->nmemslots)
1049 kvm->nmemslots = mem->slot + 1;
1050
Avi Kivity6aa8b732006-12-10 02:21:36 -08001051 *memslot = new;
Andrea Arcangeli604b38a2008-07-25 16:32:03 +02001052 spin_unlock(&kvm->mmu_lock);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001053
Zhang Xiantao0de10342007-11-20 16:25:04 +08001054 r = kvm_arch_set_memory_region(kvm, mem, old, user_alloc);
1055 if (r) {
Andrea Arcangeli604b38a2008-07-25 16:32:03 +02001056 spin_lock(&kvm->mmu_lock);
Zhang Xiantao0de10342007-11-20 16:25:04 +08001057 *memslot = old;
Andrea Arcangeli604b38a2008-07-25 16:32:03 +02001058 spin_unlock(&kvm->mmu_lock);
Zhang Xiantao0de10342007-11-20 16:25:04 +08001059 goto out_free;
Zhang Xiantao3ad82a72007-11-20 13:11:38 +08001060 }
1061
Glauber Costa6f897242008-12-03 13:40:51 -02001062 kvm_free_physmem_slot(&old, npages ? &new : NULL);
1063 /* Slot deletion case: we have to update the current slot */
1064 if (!npages)
1065 *memslot = old;
Xiantao Zhang8a98f662008-10-06 13:47:38 +08001066#ifdef CONFIG_DMAR
Ben-Ami Yassour62c476c2008-09-14 03:48:28 +03001067 /* map the pages in iommu page table */
1068 r = kvm_iommu_map_pages(kvm, base_gfn, npages);
1069 if (r)
1070 goto out;
Xiantao Zhang8a98f662008-10-06 13:47:38 +08001071#endif
Avi Kivity6aa8b732006-12-10 02:21:36 -08001072 return 0;
1073
Sheng Yangf78e0e22007-10-29 09:40:42 +08001074out_free:
Avi Kivity6aa8b732006-12-10 02:21:36 -08001075 kvm_free_physmem_slot(&new, &old);
1076out:
1077 return r;
Izik Eidus210c7c42007-10-24 23:52:57 +02001078
1079}
Sheng Yangf78e0e22007-10-29 09:40:42 +08001080EXPORT_SYMBOL_GPL(__kvm_set_memory_region);
1081
1082int kvm_set_memory_region(struct kvm *kvm,
1083 struct kvm_userspace_memory_region *mem,
1084 int user_alloc)
1085{
1086 int r;
1087
Izik Eidus72dc67a2008-02-10 18:04:15 +02001088 down_write(&kvm->slots_lock);
Sheng Yangf78e0e22007-10-29 09:40:42 +08001089 r = __kvm_set_memory_region(kvm, mem, user_alloc);
Izik Eidus72dc67a2008-02-10 18:04:15 +02001090 up_write(&kvm->slots_lock);
Sheng Yangf78e0e22007-10-29 09:40:42 +08001091 return r;
1092}
Izik Eidus210c7c42007-10-24 23:52:57 +02001093EXPORT_SYMBOL_GPL(kvm_set_memory_region);
1094
Carsten Otte1fe779f2007-10-29 16:08:35 +01001095int kvm_vm_ioctl_set_memory_region(struct kvm *kvm,
1096 struct
1097 kvm_userspace_memory_region *mem,
1098 int user_alloc)
Izik Eidus210c7c42007-10-24 23:52:57 +02001099{
Izik Eiduse0d62c72007-10-24 23:57:46 +02001100 if (mem->slot >= KVM_MEMORY_SLOTS)
1101 return -EINVAL;
Izik Eidus210c7c42007-10-24 23:52:57 +02001102 return kvm_set_memory_region(kvm, mem, user_alloc);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001103}
1104
Zhang Xiantao5bb064d2007-11-18 20:29:43 +08001105int kvm_get_dirty_log(struct kvm *kvm,
1106 struct kvm_dirty_log *log, int *is_dirty)
Avi Kivity6aa8b732006-12-10 02:21:36 -08001107{
1108 struct kvm_memory_slot *memslot;
1109 int r, i;
1110 int n;
1111 unsigned long any = 0;
1112
Avi Kivity6aa8b732006-12-10 02:21:36 -08001113 r = -EINVAL;
1114 if (log->slot >= KVM_MEMORY_SLOTS)
1115 goto out;
1116
1117 memslot = &kvm->memslots[log->slot];
1118 r = -ENOENT;
1119 if (!memslot->dirty_bitmap)
1120 goto out;
1121
Uri Lublincd1a4a92007-02-22 16:43:09 +02001122 n = ALIGN(memslot->npages, BITS_PER_LONG) / 8;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001123
Uri Lublincd1a4a92007-02-22 16:43:09 +02001124 for (i = 0; !any && i < n/sizeof(long); ++i)
Avi Kivity6aa8b732006-12-10 02:21:36 -08001125 any = memslot->dirty_bitmap[i];
1126
1127 r = -EFAULT;
1128 if (copy_to_user(log->dirty_bitmap, memslot->dirty_bitmap, n))
1129 goto out;
1130
Zhang Xiantao5bb064d2007-11-18 20:29:43 +08001131 if (any)
1132 *is_dirty = 1;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001133
1134 r = 0;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001135out:
Avi Kivity6aa8b732006-12-10 02:21:36 -08001136 return r;
1137}
1138
Izik Eiduscea7bb22007-10-17 19:17:48 +02001139int is_error_page(struct page *page)
1140{
1141 return page == bad_page;
1142}
1143EXPORT_SYMBOL_GPL(is_error_page);
1144
Anthony Liguori35149e22008-04-02 14:46:56 -05001145int is_error_pfn(pfn_t pfn)
1146{
1147 return pfn == bad_pfn;
1148}
1149EXPORT_SYMBOL_GPL(is_error_pfn);
1150
Izik Eidusf9d46eb2007-11-11 22:02:22 +02001151static inline unsigned long bad_hva(void)
1152{
1153 return PAGE_OFFSET;
1154}
1155
1156int kvm_is_error_hva(unsigned long addr)
1157{
1158 return addr == bad_hva();
1159}
1160EXPORT_SYMBOL_GPL(kvm_is_error_hva);
1161
Izik Eidus28430992008-10-03 17:40:32 +03001162struct kvm_memory_slot *gfn_to_memslot_unaliased(struct kvm *kvm, gfn_t gfn)
Avi Kivity6aa8b732006-12-10 02:21:36 -08001163{
1164 int i;
1165
1166 for (i = 0; i < kvm->nmemslots; ++i) {
1167 struct kvm_memory_slot *memslot = &kvm->memslots[i];
1168
1169 if (gfn >= memslot->base_gfn
1170 && gfn < memslot->base_gfn + memslot->npages)
1171 return memslot;
1172 }
Al Viro8b6d44c2007-02-09 16:38:40 +00001173 return NULL;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001174}
Izik Eidus28430992008-10-03 17:40:32 +03001175EXPORT_SYMBOL_GPL(gfn_to_memslot_unaliased);
Avi Kivitye8207542007-03-30 16:54:30 +03001176
1177struct kvm_memory_slot *gfn_to_memslot(struct kvm *kvm, gfn_t gfn)
1178{
1179 gfn = unalias_gfn(kvm, gfn);
Izik Eidus28430992008-10-03 17:40:32 +03001180 return gfn_to_memslot_unaliased(kvm, gfn);
Avi Kivitye8207542007-03-30 16:54:30 +03001181}
Avi Kivity6aa8b732006-12-10 02:21:36 -08001182
Izik Eiduse0d62c72007-10-24 23:57:46 +02001183int kvm_is_visible_gfn(struct kvm *kvm, gfn_t gfn)
1184{
1185 int i;
1186
1187 gfn = unalias_gfn(kvm, gfn);
1188 for (i = 0; i < KVM_MEMORY_SLOTS; ++i) {
1189 struct kvm_memory_slot *memslot = &kvm->memslots[i];
1190
1191 if (gfn >= memslot->base_gfn
1192 && gfn < memslot->base_gfn + memslot->npages)
1193 return 1;
1194 }
1195 return 0;
1196}
1197EXPORT_SYMBOL_GPL(kvm_is_visible_gfn);
1198
Marcelo Tosatti05da4552008-02-23 11:44:30 -03001199unsigned long gfn_to_hva(struct kvm *kvm, gfn_t gfn)
Izik Eidus539cb662007-11-11 22:05:04 +02001200{
1201 struct kvm_memory_slot *slot;
1202
1203 gfn = unalias_gfn(kvm, gfn);
Izik Eidus28430992008-10-03 17:40:32 +03001204 slot = gfn_to_memslot_unaliased(kvm, gfn);
Izik Eidus539cb662007-11-11 22:05:04 +02001205 if (!slot)
1206 return bad_hva();
1207 return (slot->userspace_addr + (gfn - slot->base_gfn) * PAGE_SIZE);
1208}
Sheng Yang0d150292008-04-25 21:44:50 +08001209EXPORT_SYMBOL_GPL(gfn_to_hva);
Izik Eidus539cb662007-11-11 22:05:04 +02001210
Anthony Liguori35149e22008-04-02 14:46:56 -05001211pfn_t gfn_to_pfn(struct kvm *kvm, gfn_t gfn)
Avi Kivity954bbbc22007-03-30 14:02:32 +03001212{
Anthony Liguori8d4e1282007-10-18 09:59:34 -05001213 struct page *page[1];
Izik Eidus539cb662007-11-11 22:05:04 +02001214 unsigned long addr;
Anthony Liguori8d4e1282007-10-18 09:59:34 -05001215 int npages;
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001216 pfn_t pfn;
Avi Kivity954bbbc22007-03-30 14:02:32 +03001217
Avi Kivity60395222007-10-21 11:03:36 +02001218 might_sleep();
1219
Izik Eidus539cb662007-11-11 22:05:04 +02001220 addr = gfn_to_hva(kvm, gfn);
1221 if (kvm_is_error_hva(addr)) {
Izik Eidus8a7ae052007-10-18 11:09:33 +02001222 get_page(bad_page);
Anthony Liguori35149e22008-04-02 14:46:56 -05001223 return page_to_pfn(bad_page);
Izik Eidus8a7ae052007-10-18 11:09:33 +02001224 }
Izik Eidus8a7ae052007-10-18 11:09:33 +02001225
Marcelo Tosatti4c2155c2008-09-16 20:54:47 -03001226 npages = get_user_pages_fast(addr, 1, 1, page);
Izik Eidus539cb662007-11-11 22:05:04 +02001227
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001228 if (unlikely(npages != 1)) {
1229 struct vm_area_struct *vma;
Anthony Liguori8d4e1282007-10-18 09:59:34 -05001230
Marcelo Tosatti4c2155c2008-09-16 20:54:47 -03001231 down_read(&current->mm->mmap_sem);
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001232 vma = find_vma(current->mm, addr);
Marcelo Tosatti4c2155c2008-09-16 20:54:47 -03001233
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001234 if (vma == NULL || addr < vma->vm_start ||
1235 !(vma->vm_flags & VM_PFNMAP)) {
Marcelo Tosatti4c2155c2008-09-16 20:54:47 -03001236 up_read(&current->mm->mmap_sem);
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001237 get_page(bad_page);
1238 return page_to_pfn(bad_page);
1239 }
1240
1241 pfn = ((addr - vma->vm_start) >> PAGE_SHIFT) + vma->vm_pgoff;
Marcelo Tosatti4c2155c2008-09-16 20:54:47 -03001242 up_read(&current->mm->mmap_sem);
Xiantao Zhangc77fb9d2008-09-27 10:55:40 +08001243 BUG_ON(!kvm_is_mmio_pfn(pfn));
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001244 } else
1245 pfn = page_to_pfn(page[0]);
1246
1247 return pfn;
Anthony Liguori35149e22008-04-02 14:46:56 -05001248}
1249
1250EXPORT_SYMBOL_GPL(gfn_to_pfn);
1251
1252struct page *gfn_to_page(struct kvm *kvm, gfn_t gfn)
1253{
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001254 pfn_t pfn;
1255
1256 pfn = gfn_to_pfn(kvm, gfn);
Xiantao Zhangc77fb9d2008-09-27 10:55:40 +08001257 if (!kvm_is_mmio_pfn(pfn))
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001258 return pfn_to_page(pfn);
1259
Xiantao Zhangc77fb9d2008-09-27 10:55:40 +08001260 WARN_ON(kvm_is_mmio_pfn(pfn));
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001261
1262 get_page(bad_page);
1263 return bad_page;
Avi Kivity954bbbc22007-03-30 14:02:32 +03001264}
Anthony Liguoriaab61cc2007-10-29 15:15:20 -05001265
Avi Kivity954bbbc22007-03-30 14:02:32 +03001266EXPORT_SYMBOL_GPL(gfn_to_page);
1267
Izik Eidusb4231d62007-11-20 11:49:33 +02001268void kvm_release_page_clean(struct page *page)
1269{
Anthony Liguori35149e22008-04-02 14:46:56 -05001270 kvm_release_pfn_clean(page_to_pfn(page));
Izik Eidusb4231d62007-11-20 11:49:33 +02001271}
1272EXPORT_SYMBOL_GPL(kvm_release_page_clean);
1273
Anthony Liguori35149e22008-04-02 14:46:56 -05001274void kvm_release_pfn_clean(pfn_t pfn)
1275{
Xiantao Zhangc77fb9d2008-09-27 10:55:40 +08001276 if (!kvm_is_mmio_pfn(pfn))
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001277 put_page(pfn_to_page(pfn));
Anthony Liguori35149e22008-04-02 14:46:56 -05001278}
1279EXPORT_SYMBOL_GPL(kvm_release_pfn_clean);
1280
Izik Eidusb4231d62007-11-20 11:49:33 +02001281void kvm_release_page_dirty(struct page *page)
Izik Eidus8a7ae052007-10-18 11:09:33 +02001282{
Anthony Liguori35149e22008-04-02 14:46:56 -05001283 kvm_release_pfn_dirty(page_to_pfn(page));
Izik Eidus8a7ae052007-10-18 11:09:33 +02001284}
Izik Eidusb4231d62007-11-20 11:49:33 +02001285EXPORT_SYMBOL_GPL(kvm_release_page_dirty);
Izik Eidus8a7ae052007-10-18 11:09:33 +02001286
Anthony Liguori35149e22008-04-02 14:46:56 -05001287void kvm_release_pfn_dirty(pfn_t pfn)
1288{
1289 kvm_set_pfn_dirty(pfn);
1290 kvm_release_pfn_clean(pfn);
1291}
1292EXPORT_SYMBOL_GPL(kvm_release_pfn_dirty);
1293
1294void kvm_set_page_dirty(struct page *page)
1295{
1296 kvm_set_pfn_dirty(page_to_pfn(page));
1297}
1298EXPORT_SYMBOL_GPL(kvm_set_page_dirty);
1299
1300void kvm_set_pfn_dirty(pfn_t pfn)
1301{
Xiantao Zhangc77fb9d2008-09-27 10:55:40 +08001302 if (!kvm_is_mmio_pfn(pfn)) {
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001303 struct page *page = pfn_to_page(pfn);
1304 if (!PageReserved(page))
1305 SetPageDirty(page);
1306 }
Anthony Liguori35149e22008-04-02 14:46:56 -05001307}
1308EXPORT_SYMBOL_GPL(kvm_set_pfn_dirty);
1309
1310void kvm_set_pfn_accessed(pfn_t pfn)
1311{
Xiantao Zhangc77fb9d2008-09-27 10:55:40 +08001312 if (!kvm_is_mmio_pfn(pfn))
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001313 mark_page_accessed(pfn_to_page(pfn));
Anthony Liguori35149e22008-04-02 14:46:56 -05001314}
1315EXPORT_SYMBOL_GPL(kvm_set_pfn_accessed);
1316
1317void kvm_get_pfn(pfn_t pfn)
1318{
Xiantao Zhangc77fb9d2008-09-27 10:55:40 +08001319 if (!kvm_is_mmio_pfn(pfn))
Anthony Liguori2e2e3732008-04-30 15:37:07 -05001320 get_page(pfn_to_page(pfn));
Anthony Liguori35149e22008-04-02 14:46:56 -05001321}
1322EXPORT_SYMBOL_GPL(kvm_get_pfn);
1323
Izik Eidus195aefd2007-10-01 22:14:18 +02001324static int next_segment(unsigned long len, int offset)
1325{
1326 if (len > PAGE_SIZE - offset)
1327 return PAGE_SIZE - offset;
1328 else
1329 return len;
1330}
1331
1332int kvm_read_guest_page(struct kvm *kvm, gfn_t gfn, void *data, int offset,
1333 int len)
1334{
Izik Eiduse0506bc2007-11-11 22:10:22 +02001335 int r;
1336 unsigned long addr;
Izik Eidus195aefd2007-10-01 22:14:18 +02001337
Izik Eiduse0506bc2007-11-11 22:10:22 +02001338 addr = gfn_to_hva(kvm, gfn);
1339 if (kvm_is_error_hva(addr))
Izik Eidus195aefd2007-10-01 22:14:18 +02001340 return -EFAULT;
Izik Eiduse0506bc2007-11-11 22:10:22 +02001341 r = copy_from_user(data, (void __user *)addr + offset, len);
1342 if (r)
1343 return -EFAULT;
Izik Eidus195aefd2007-10-01 22:14:18 +02001344 return 0;
1345}
1346EXPORT_SYMBOL_GPL(kvm_read_guest_page);
1347
1348int kvm_read_guest(struct kvm *kvm, gpa_t gpa, void *data, unsigned long len)
1349{
1350 gfn_t gfn = gpa >> PAGE_SHIFT;
1351 int seg;
1352 int offset = offset_in_page(gpa);
1353 int ret;
1354
1355 while ((seg = next_segment(len, offset)) != 0) {
1356 ret = kvm_read_guest_page(kvm, gfn, data, offset, seg);
1357 if (ret < 0)
1358 return ret;
1359 offset = 0;
1360 len -= seg;
1361 data += seg;
1362 ++gfn;
1363 }
1364 return 0;
1365}
1366EXPORT_SYMBOL_GPL(kvm_read_guest);
1367
Marcelo Tosatti7ec54582007-12-20 19:18:23 -05001368int kvm_read_guest_atomic(struct kvm *kvm, gpa_t gpa, void *data,
1369 unsigned long len)
1370{
1371 int r;
1372 unsigned long addr;
1373 gfn_t gfn = gpa >> PAGE_SHIFT;
1374 int offset = offset_in_page(gpa);
1375
1376 addr = gfn_to_hva(kvm, gfn);
1377 if (kvm_is_error_hva(addr))
1378 return -EFAULT;
Andrea Arcangeli0aac03f2008-01-30 19:57:35 +01001379 pagefault_disable();
Marcelo Tosatti7ec54582007-12-20 19:18:23 -05001380 r = __copy_from_user_inatomic(data, (void __user *)addr + offset, len);
Andrea Arcangeli0aac03f2008-01-30 19:57:35 +01001381 pagefault_enable();
Marcelo Tosatti7ec54582007-12-20 19:18:23 -05001382 if (r)
1383 return -EFAULT;
1384 return 0;
1385}
1386EXPORT_SYMBOL(kvm_read_guest_atomic);
1387
Izik Eidus195aefd2007-10-01 22:14:18 +02001388int kvm_write_guest_page(struct kvm *kvm, gfn_t gfn, const void *data,
1389 int offset, int len)
1390{
Izik Eiduse0506bc2007-11-11 22:10:22 +02001391 int r;
1392 unsigned long addr;
Izik Eidus195aefd2007-10-01 22:14:18 +02001393
Izik Eiduse0506bc2007-11-11 22:10:22 +02001394 addr = gfn_to_hva(kvm, gfn);
1395 if (kvm_is_error_hva(addr))
Izik Eidus195aefd2007-10-01 22:14:18 +02001396 return -EFAULT;
Izik Eiduse0506bc2007-11-11 22:10:22 +02001397 r = copy_to_user((void __user *)addr + offset, data, len);
1398 if (r)
1399 return -EFAULT;
Izik Eidus195aefd2007-10-01 22:14:18 +02001400 mark_page_dirty(kvm, gfn);
1401 return 0;
1402}
1403EXPORT_SYMBOL_GPL(kvm_write_guest_page);
1404
1405int kvm_write_guest(struct kvm *kvm, gpa_t gpa, const void *data,
1406 unsigned long len)
1407{
1408 gfn_t gfn = gpa >> PAGE_SHIFT;
1409 int seg;
1410 int offset = offset_in_page(gpa);
1411 int ret;
1412
1413 while ((seg = next_segment(len, offset)) != 0) {
1414 ret = kvm_write_guest_page(kvm, gfn, data, offset, seg);
1415 if (ret < 0)
1416 return ret;
1417 offset = 0;
1418 len -= seg;
1419 data += seg;
1420 ++gfn;
1421 }
1422 return 0;
1423}
1424
1425int kvm_clear_guest_page(struct kvm *kvm, gfn_t gfn, int offset, int len)
1426{
Izik Eidus3e021bf2007-11-19 11:16:57 +02001427 return kvm_write_guest_page(kvm, gfn, empty_zero_page, offset, len);
Izik Eidus195aefd2007-10-01 22:14:18 +02001428}
1429EXPORT_SYMBOL_GPL(kvm_clear_guest_page);
1430
1431int kvm_clear_guest(struct kvm *kvm, gpa_t gpa, unsigned long len)
1432{
1433 gfn_t gfn = gpa >> PAGE_SHIFT;
1434 int seg;
1435 int offset = offset_in_page(gpa);
1436 int ret;
1437
1438 while ((seg = next_segment(len, offset)) != 0) {
1439 ret = kvm_clear_guest_page(kvm, gfn, offset, seg);
1440 if (ret < 0)
1441 return ret;
1442 offset = 0;
1443 len -= seg;
1444 ++gfn;
1445 }
1446 return 0;
1447}
1448EXPORT_SYMBOL_GPL(kvm_clear_guest);
1449
Avi Kivity6aa8b732006-12-10 02:21:36 -08001450void mark_page_dirty(struct kvm *kvm, gfn_t gfn)
1451{
Nguyen Anh Quynh31389942007-06-05 10:35:19 +03001452 struct kvm_memory_slot *memslot;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001453
Uri Lublin3b6fff12007-10-30 10:42:09 +02001454 gfn = unalias_gfn(kvm, gfn);
Izik Eidus28430992008-10-03 17:40:32 +03001455 memslot = gfn_to_memslot_unaliased(kvm, gfn);
Rusty Russell7e9d6192007-07-31 20:41:14 +10001456 if (memslot && memslot->dirty_bitmap) {
1457 unsigned long rel_gfn = gfn - memslot->base_gfn;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001458
Rusty Russell7e9d6192007-07-31 20:41:14 +10001459 /* avoid RMW */
1460 if (!test_bit(rel_gfn, memslot->dirty_bitmap))
1461 set_bit(rel_gfn, memslot->dirty_bitmap);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001462 }
1463}
1464
Eddie Dongb6958ce2007-07-18 12:15:21 +03001465/*
1466 * The vCPU has executed a HLT instruction with in-kernel mode enabled.
1467 */
Hollis Blanchard8776e512007-10-31 17:24:24 -05001468void kvm_vcpu_block(struct kvm_vcpu *vcpu)
Eddie Dongb6958ce2007-07-18 12:15:21 +03001469{
Marcelo Tosattie5c239c2008-05-08 19:47:01 -03001470 DEFINE_WAIT(wait);
Eddie Dongb6958ce2007-07-18 12:15:21 +03001471
Marcelo Tosattie5c239c2008-05-08 19:47:01 -03001472 for (;;) {
1473 prepare_to_wait(&vcpu->wq, &wait, TASK_INTERRUPTIBLE);
Eddie Dongb6958ce2007-07-18 12:15:21 +03001474
Marcelo Tosattid7690172008-09-08 15:23:48 -03001475 if (kvm_cpu_has_interrupt(vcpu) ||
1476 kvm_cpu_has_pending_timer(vcpu) ||
1477 kvm_arch_vcpu_runnable(vcpu)) {
1478 set_bit(KVM_REQ_UNHALT, &vcpu->requests);
Marcelo Tosattie5c239c2008-05-08 19:47:01 -03001479 break;
Marcelo Tosattid7690172008-09-08 15:23:48 -03001480 }
Marcelo Tosattie5c239c2008-05-08 19:47:01 -03001481 if (signal_pending(current))
1482 break;
1483
Eddie Dongb6958ce2007-07-18 12:15:21 +03001484 vcpu_put(vcpu);
1485 schedule();
1486 vcpu_load(vcpu);
1487 }
1488
Marcelo Tosattie5c239c2008-05-08 19:47:01 -03001489 finish_wait(&vcpu->wq, &wait);
Eddie Dongb6958ce2007-07-18 12:15:21 +03001490}
1491
Avi Kivity6aa8b732006-12-10 02:21:36 -08001492void kvm_resched(struct kvm_vcpu *vcpu)
1493{
Yaozu Dong3fca0362007-04-25 16:49:19 +03001494 if (!need_resched())
1495 return;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001496 cond_resched();
Avi Kivity6aa8b732006-12-10 02:21:36 -08001497}
1498EXPORT_SYMBOL_GPL(kvm_resched);
1499
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001500static int kvm_vcpu_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
Avi Kivity9a2bb7f2007-02-22 12:58:31 +02001501{
1502 struct kvm_vcpu *vcpu = vma->vm_file->private_data;
Avi Kivity9a2bb7f2007-02-22 12:58:31 +02001503 struct page *page;
1504
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001505 if (vmf->pgoff == 0)
Avi Kivity039576c2007-03-20 12:46:50 +02001506 page = virt_to_page(vcpu->run);
Avi Kivity09566762008-01-23 18:14:23 +02001507#ifdef CONFIG_X86
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001508 else if (vmf->pgoff == KVM_PIO_PAGE_OFFSET)
Zhang Xiantaoad312c72007-12-13 23:50:52 +08001509 page = virt_to_page(vcpu->arch.pio_data);
Avi Kivity09566762008-01-23 18:14:23 +02001510#endif
Laurent Vivier5f94c172008-05-30 16:05:54 +02001511#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
1512 else if (vmf->pgoff == KVM_COALESCED_MMIO_PAGE_OFFSET)
1513 page = virt_to_page(vcpu->kvm->coalesced_mmio_ring);
1514#endif
Avi Kivity039576c2007-03-20 12:46:50 +02001515 else
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001516 return VM_FAULT_SIGBUS;
Avi Kivity9a2bb7f2007-02-22 12:58:31 +02001517 get_page(page);
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001518 vmf->page = page;
1519 return 0;
Avi Kivity9a2bb7f2007-02-22 12:58:31 +02001520}
1521
1522static struct vm_operations_struct kvm_vcpu_vm_ops = {
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001523 .fault = kvm_vcpu_fault,
Avi Kivity9a2bb7f2007-02-22 12:58:31 +02001524};
1525
1526static int kvm_vcpu_mmap(struct file *file, struct vm_area_struct *vma)
1527{
1528 vma->vm_ops = &kvm_vcpu_vm_ops;
1529 return 0;
1530}
1531
Avi Kivitybccf2152007-02-21 18:04:26 +02001532static int kvm_vcpu_release(struct inode *inode, struct file *filp)
1533{
1534 struct kvm_vcpu *vcpu = filp->private_data;
1535
Al Viro66c0b392008-04-19 20:33:56 +01001536 kvm_put_kvm(vcpu->kvm);
Avi Kivitybccf2152007-02-21 18:04:26 +02001537 return 0;
1538}
1539
Christian Borntraeger3d3aab12008-12-02 11:17:32 +01001540static struct file_operations kvm_vcpu_fops = {
Avi Kivitybccf2152007-02-21 18:04:26 +02001541 .release = kvm_vcpu_release,
1542 .unlocked_ioctl = kvm_vcpu_ioctl,
1543 .compat_ioctl = kvm_vcpu_ioctl,
Avi Kivity9a2bb7f2007-02-22 12:58:31 +02001544 .mmap = kvm_vcpu_mmap,
Avi Kivitybccf2152007-02-21 18:04:26 +02001545};
1546
1547/*
1548 * Allocates an inode for the vcpu.
1549 */
1550static int create_vcpu_fd(struct kvm_vcpu *vcpu)
1551{
Ulrich Drepper7d9dbca2008-07-23 21:29:22 -07001552 int fd = anon_inode_getfd("kvm-vcpu", &kvm_vcpu_fops, vcpu, 0);
Al Viro2030a422008-02-23 06:46:49 -05001553 if (fd < 0)
Al Viro66c0b392008-04-19 20:33:56 +01001554 kvm_put_kvm(vcpu->kvm);
Avi Kivitybccf2152007-02-21 18:04:26 +02001555 return fd;
Avi Kivitybccf2152007-02-21 18:04:26 +02001556}
1557
Avi Kivityc5ea7662007-02-20 18:41:05 +02001558/*
1559 * Creates some virtual cpus. Good luck creating more than one.
1560 */
1561static int kvm_vm_ioctl_create_vcpu(struct kvm *kvm, int n)
1562{
1563 int r;
1564 struct kvm_vcpu *vcpu;
1565
Avi Kivityc5ea7662007-02-20 18:41:05 +02001566 if (!valid_vcpu(n))
Rusty Russellfb3f0f52007-07-27 17:16:56 +10001567 return -EINVAL;
Avi Kivityc5ea7662007-02-20 18:41:05 +02001568
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08001569 vcpu = kvm_arch_vcpu_create(kvm, n);
Rusty Russellfb3f0f52007-07-27 17:16:56 +10001570 if (IS_ERR(vcpu))
1571 return PTR_ERR(vcpu);
Avi Kivityc5ea7662007-02-20 18:41:05 +02001572
Avi Kivity15ad7142007-07-11 18:17:21 +03001573 preempt_notifier_init(&vcpu->preempt_notifier, &kvm_preempt_ops);
1574
Avi Kivity26e52152007-11-20 15:30:24 +02001575 r = kvm_arch_vcpu_setup(vcpu);
1576 if (r)
Glauber Costa7d8fece2008-09-17 23:16:59 -03001577 return r;
Avi Kivity26e52152007-11-20 15:30:24 +02001578
Shaohua Li11ec2802007-07-23 14:51:37 +08001579 mutex_lock(&kvm->lock);
Rusty Russellfb3f0f52007-07-27 17:16:56 +10001580 if (kvm->vcpus[n]) {
1581 r = -EEXIST;
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08001582 goto vcpu_destroy;
Rusty Russellfb3f0f52007-07-27 17:16:56 +10001583 }
1584 kvm->vcpus[n] = vcpu;
Shaohua Li11ec2802007-07-23 14:51:37 +08001585 mutex_unlock(&kvm->lock);
Rusty Russellfb3f0f52007-07-27 17:16:56 +10001586
1587 /* Now it's all set up, let userspace reach it */
Al Viro66c0b392008-04-19 20:33:56 +01001588 kvm_get_kvm(kvm);
Avi Kivitybccf2152007-02-21 18:04:26 +02001589 r = create_vcpu_fd(vcpu);
1590 if (r < 0)
Rusty Russellfb3f0f52007-07-27 17:16:56 +10001591 goto unlink;
Avi Kivitybccf2152007-02-21 18:04:26 +02001592 return r;
Avi Kivityc5ea7662007-02-20 18:41:05 +02001593
Rusty Russellfb3f0f52007-07-27 17:16:56 +10001594unlink:
Shaohua Li11ec2802007-07-23 14:51:37 +08001595 mutex_lock(&kvm->lock);
Rusty Russellfb3f0f52007-07-27 17:16:56 +10001596 kvm->vcpus[n] = NULL;
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08001597vcpu_destroy:
Glauber Costa7d8fece2008-09-17 23:16:59 -03001598 mutex_unlock(&kvm->lock);
Hollis Blanchardd40ccc62007-11-19 14:04:43 -06001599 kvm_arch_vcpu_destroy(vcpu);
Avi Kivityc5ea7662007-02-20 18:41:05 +02001600 return r;
1601}
1602
Avi Kivity1961d272007-03-05 19:46:05 +02001603static int kvm_vcpu_ioctl_set_sigmask(struct kvm_vcpu *vcpu, sigset_t *sigset)
1604{
1605 if (sigset) {
1606 sigdelsetmask(sigset, sigmask(SIGKILL)|sigmask(SIGSTOP));
1607 vcpu->sigset_active = 1;
1608 vcpu->sigset = *sigset;
1609 } else
1610 vcpu->sigset_active = 0;
1611 return 0;
1612}
1613
Avi Kivitybccf2152007-02-21 18:04:26 +02001614static long kvm_vcpu_ioctl(struct file *filp,
1615 unsigned int ioctl, unsigned long arg)
Avi Kivity6aa8b732006-12-10 02:21:36 -08001616{
Avi Kivitybccf2152007-02-21 18:04:26 +02001617 struct kvm_vcpu *vcpu = filp->private_data;
Al Viro2f366982007-02-09 16:38:35 +00001618 void __user *argp = (void __user *)arg;
Carsten Otte313a3dc2007-10-11 19:16:52 +02001619 int r;
Dave Hansenfa3795a2008-08-11 10:01:46 -07001620 struct kvm_fpu *fpu = NULL;
1621 struct kvm_sregs *kvm_sregs = NULL;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001622
Avi Kivity6d4e4c42007-11-21 16:41:05 +02001623 if (vcpu->kvm->mm != current->mm)
1624 return -EIO;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001625 switch (ioctl) {
Avi Kivity9a2bb7f2007-02-22 12:58:31 +02001626 case KVM_RUN:
Avi Kivityf0fe5102007-03-07 13:11:17 +02001627 r = -EINVAL;
1628 if (arg)
1629 goto out;
Hollis Blanchardb6c7a5d2007-11-01 14:16:10 -05001630 r = kvm_arch_vcpu_ioctl_run(vcpu, vcpu->run);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001631 break;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001632 case KVM_GET_REGS: {
Xiantao Zhang3e4bb3a2008-02-25 18:52:20 +08001633 struct kvm_regs *kvm_regs;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001634
Xiantao Zhang3e4bb3a2008-02-25 18:52:20 +08001635 r = -ENOMEM;
1636 kvm_regs = kzalloc(sizeof(struct kvm_regs), GFP_KERNEL);
1637 if (!kvm_regs)
1638 goto out;
1639 r = kvm_arch_vcpu_ioctl_get_regs(vcpu, kvm_regs);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001640 if (r)
Xiantao Zhang3e4bb3a2008-02-25 18:52:20 +08001641 goto out_free1;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001642 r = -EFAULT;
Xiantao Zhang3e4bb3a2008-02-25 18:52:20 +08001643 if (copy_to_user(argp, kvm_regs, sizeof(struct kvm_regs)))
1644 goto out_free1;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001645 r = 0;
Xiantao Zhang3e4bb3a2008-02-25 18:52:20 +08001646out_free1:
1647 kfree(kvm_regs);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001648 break;
1649 }
1650 case KVM_SET_REGS: {
Xiantao Zhang3e4bb3a2008-02-25 18:52:20 +08001651 struct kvm_regs *kvm_regs;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001652
Xiantao Zhang3e4bb3a2008-02-25 18:52:20 +08001653 r = -ENOMEM;
1654 kvm_regs = kzalloc(sizeof(struct kvm_regs), GFP_KERNEL);
1655 if (!kvm_regs)
1656 goto out;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001657 r = -EFAULT;
Xiantao Zhang3e4bb3a2008-02-25 18:52:20 +08001658 if (copy_from_user(kvm_regs, argp, sizeof(struct kvm_regs)))
1659 goto out_free2;
1660 r = kvm_arch_vcpu_ioctl_set_regs(vcpu, kvm_regs);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001661 if (r)
Xiantao Zhang3e4bb3a2008-02-25 18:52:20 +08001662 goto out_free2;
Avi Kivity6aa8b732006-12-10 02:21:36 -08001663 r = 0;
Xiantao Zhang3e4bb3a2008-02-25 18:52:20 +08001664out_free2:
1665 kfree(kvm_regs);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001666 break;
1667 }
1668 case KVM_GET_SREGS: {
Dave Hansenfa3795a2008-08-11 10:01:46 -07001669 kvm_sregs = kzalloc(sizeof(struct kvm_sregs), GFP_KERNEL);
1670 r = -ENOMEM;
1671 if (!kvm_sregs)
1672 goto out;
1673 r = kvm_arch_vcpu_ioctl_get_sregs(vcpu, kvm_sregs);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001674 if (r)
1675 goto out;
1676 r = -EFAULT;
Dave Hansenfa3795a2008-08-11 10:01:46 -07001677 if (copy_to_user(argp, kvm_sregs, sizeof(struct kvm_sregs)))
Avi Kivity6aa8b732006-12-10 02:21:36 -08001678 goto out;
1679 r = 0;
1680 break;
1681 }
1682 case KVM_SET_SREGS: {
Dave Hansenfa3795a2008-08-11 10:01:46 -07001683 kvm_sregs = kmalloc(sizeof(struct kvm_sregs), GFP_KERNEL);
1684 r = -ENOMEM;
1685 if (!kvm_sregs)
Avi Kivity6aa8b732006-12-10 02:21:36 -08001686 goto out;
Dave Hansenfa3795a2008-08-11 10:01:46 -07001687 r = -EFAULT;
1688 if (copy_from_user(kvm_sregs, argp, sizeof(struct kvm_sregs)))
1689 goto out;
1690 r = kvm_arch_vcpu_ioctl_set_sregs(vcpu, kvm_sregs);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001691 if (r)
1692 goto out;
1693 r = 0;
1694 break;
1695 }
Marcelo Tosatti62d9f0d2008-04-11 13:24:45 -03001696 case KVM_GET_MP_STATE: {
1697 struct kvm_mp_state mp_state;
1698
1699 r = kvm_arch_vcpu_ioctl_get_mpstate(vcpu, &mp_state);
1700 if (r)
1701 goto out;
1702 r = -EFAULT;
1703 if (copy_to_user(argp, &mp_state, sizeof mp_state))
1704 goto out;
1705 r = 0;
1706 break;
1707 }
1708 case KVM_SET_MP_STATE: {
1709 struct kvm_mp_state mp_state;
1710
1711 r = -EFAULT;
1712 if (copy_from_user(&mp_state, argp, sizeof mp_state))
1713 goto out;
1714 r = kvm_arch_vcpu_ioctl_set_mpstate(vcpu, &mp_state);
1715 if (r)
1716 goto out;
1717 r = 0;
1718 break;
1719 }
Avi Kivity6aa8b732006-12-10 02:21:36 -08001720 case KVM_TRANSLATE: {
1721 struct kvm_translation tr;
1722
1723 r = -EFAULT;
Al Viro2f366982007-02-09 16:38:35 +00001724 if (copy_from_user(&tr, argp, sizeof tr))
Avi Kivity6aa8b732006-12-10 02:21:36 -08001725 goto out;
Zhang Xiantao8b006792007-11-16 13:05:55 +08001726 r = kvm_arch_vcpu_ioctl_translate(vcpu, &tr);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001727 if (r)
1728 goto out;
1729 r = -EFAULT;
Al Viro2f366982007-02-09 16:38:35 +00001730 if (copy_to_user(argp, &tr, sizeof tr))
Avi Kivity6aa8b732006-12-10 02:21:36 -08001731 goto out;
1732 r = 0;
1733 break;
1734 }
Avi Kivity6aa8b732006-12-10 02:21:36 -08001735 case KVM_DEBUG_GUEST: {
1736 struct kvm_debug_guest dbg;
1737
1738 r = -EFAULT;
Al Viro2f366982007-02-09 16:38:35 +00001739 if (copy_from_user(&dbg, argp, sizeof dbg))
Avi Kivity6aa8b732006-12-10 02:21:36 -08001740 goto out;
Hollis Blanchardb6c7a5d2007-11-01 14:16:10 -05001741 r = kvm_arch_vcpu_ioctl_debug_guest(vcpu, &dbg);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001742 if (r)
1743 goto out;
1744 r = 0;
1745 break;
1746 }
Avi Kivity1961d272007-03-05 19:46:05 +02001747 case KVM_SET_SIGNAL_MASK: {
1748 struct kvm_signal_mask __user *sigmask_arg = argp;
1749 struct kvm_signal_mask kvm_sigmask;
1750 sigset_t sigset, *p;
1751
1752 p = NULL;
1753 if (argp) {
1754 r = -EFAULT;
1755 if (copy_from_user(&kvm_sigmask, argp,
1756 sizeof kvm_sigmask))
1757 goto out;
1758 r = -EINVAL;
1759 if (kvm_sigmask.len != sizeof sigset)
1760 goto out;
1761 r = -EFAULT;
1762 if (copy_from_user(&sigset, sigmask_arg->sigset,
1763 sizeof sigset))
1764 goto out;
1765 p = &sigset;
1766 }
1767 r = kvm_vcpu_ioctl_set_sigmask(vcpu, &sigset);
1768 break;
1769 }
Avi Kivityb8836732007-04-01 16:34:31 +03001770 case KVM_GET_FPU: {
Dave Hansenfa3795a2008-08-11 10:01:46 -07001771 fpu = kzalloc(sizeof(struct kvm_fpu), GFP_KERNEL);
1772 r = -ENOMEM;
1773 if (!fpu)
1774 goto out;
1775 r = kvm_arch_vcpu_ioctl_get_fpu(vcpu, fpu);
Avi Kivityb8836732007-04-01 16:34:31 +03001776 if (r)
1777 goto out;
1778 r = -EFAULT;
Dave Hansenfa3795a2008-08-11 10:01:46 -07001779 if (copy_to_user(argp, fpu, sizeof(struct kvm_fpu)))
Avi Kivityb8836732007-04-01 16:34:31 +03001780 goto out;
1781 r = 0;
1782 break;
1783 }
1784 case KVM_SET_FPU: {
Dave Hansenfa3795a2008-08-11 10:01:46 -07001785 fpu = kmalloc(sizeof(struct kvm_fpu), GFP_KERNEL);
1786 r = -ENOMEM;
1787 if (!fpu)
Avi Kivityb8836732007-04-01 16:34:31 +03001788 goto out;
Dave Hansenfa3795a2008-08-11 10:01:46 -07001789 r = -EFAULT;
1790 if (copy_from_user(fpu, argp, sizeof(struct kvm_fpu)))
1791 goto out;
1792 r = kvm_arch_vcpu_ioctl_set_fpu(vcpu, fpu);
Avi Kivityb8836732007-04-01 16:34:31 +03001793 if (r)
1794 goto out;
1795 r = 0;
1796 break;
1797 }
Avi Kivitybccf2152007-02-21 18:04:26 +02001798 default:
Carsten Otte313a3dc2007-10-11 19:16:52 +02001799 r = kvm_arch_vcpu_ioctl(filp, ioctl, arg);
Avi Kivitybccf2152007-02-21 18:04:26 +02001800 }
1801out:
Dave Hansenfa3795a2008-08-11 10:01:46 -07001802 kfree(fpu);
1803 kfree(kvm_sregs);
Avi Kivitybccf2152007-02-21 18:04:26 +02001804 return r;
1805}
1806
1807static long kvm_vm_ioctl(struct file *filp,
1808 unsigned int ioctl, unsigned long arg)
1809{
1810 struct kvm *kvm = filp->private_data;
1811 void __user *argp = (void __user *)arg;
Carsten Otte1fe779f2007-10-29 16:08:35 +01001812 int r;
Avi Kivitybccf2152007-02-21 18:04:26 +02001813
Avi Kivity6d4e4c42007-11-21 16:41:05 +02001814 if (kvm->mm != current->mm)
1815 return -EIO;
Avi Kivitybccf2152007-02-21 18:04:26 +02001816 switch (ioctl) {
1817 case KVM_CREATE_VCPU:
1818 r = kvm_vm_ioctl_create_vcpu(kvm, arg);
1819 if (r < 0)
1820 goto out;
1821 break;
Izik Eidus6fc138d2007-10-09 19:20:39 +02001822 case KVM_SET_USER_MEMORY_REGION: {
1823 struct kvm_userspace_memory_region kvm_userspace_mem;
1824
1825 r = -EFAULT;
1826 if (copy_from_user(&kvm_userspace_mem, argp,
1827 sizeof kvm_userspace_mem))
1828 goto out;
1829
1830 r = kvm_vm_ioctl_set_memory_region(kvm, &kvm_userspace_mem, 1);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001831 if (r)
1832 goto out;
1833 break;
1834 }
1835 case KVM_GET_DIRTY_LOG: {
1836 struct kvm_dirty_log log;
1837
1838 r = -EFAULT;
Al Viro2f366982007-02-09 16:38:35 +00001839 if (copy_from_user(&log, argp, sizeof log))
Avi Kivity6aa8b732006-12-10 02:21:36 -08001840 goto out;
Avi Kivity2c6f5df2007-02-20 18:27:58 +02001841 r = kvm_vm_ioctl_get_dirty_log(kvm, &log);
Avi Kivity6aa8b732006-12-10 02:21:36 -08001842 if (r)
1843 goto out;
1844 break;
1845 }
Laurent Vivier5f94c172008-05-30 16:05:54 +02001846#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
1847 case KVM_REGISTER_COALESCED_MMIO: {
1848 struct kvm_coalesced_mmio_zone zone;
1849 r = -EFAULT;
1850 if (copy_from_user(&zone, argp, sizeof zone))
1851 goto out;
1852 r = -ENXIO;
1853 r = kvm_vm_ioctl_register_coalesced_mmio(kvm, &zone);
1854 if (r)
1855 goto out;
1856 r = 0;
1857 break;
1858 }
1859 case KVM_UNREGISTER_COALESCED_MMIO: {
1860 struct kvm_coalesced_mmio_zone zone;
1861 r = -EFAULT;
1862 if (copy_from_user(&zone, argp, sizeof zone))
1863 goto out;
1864 r = -ENXIO;
1865 r = kvm_vm_ioctl_unregister_coalesced_mmio(kvm, &zone);
1866 if (r)
1867 goto out;
1868 r = 0;
1869 break;
1870 }
1871#endif
Xiantao Zhang8a98f662008-10-06 13:47:38 +08001872#ifdef KVM_CAP_DEVICE_ASSIGNMENT
1873 case KVM_ASSIGN_PCI_DEVICE: {
1874 struct kvm_assigned_pci_dev assigned_dev;
1875
1876 r = -EFAULT;
1877 if (copy_from_user(&assigned_dev, argp, sizeof assigned_dev))
1878 goto out;
1879 r = kvm_vm_ioctl_assign_device(kvm, &assigned_dev);
1880 if (r)
1881 goto out;
1882 break;
1883 }
1884 case KVM_ASSIGN_IRQ: {
1885 struct kvm_assigned_irq assigned_irq;
1886
1887 r = -EFAULT;
1888 if (copy_from_user(&assigned_irq, argp, sizeof assigned_irq))
1889 goto out;
1890 r = kvm_vm_ioctl_assign_irq(kvm, &assigned_irq);
1891 if (r)
1892 goto out;
1893 break;
1894 }
1895#endif
Weidong Han0a920352008-12-02 21:24:23 +08001896#ifdef KVM_CAP_DEVICE_DEASSIGNMENT
1897 case KVM_DEASSIGN_PCI_DEVICE: {
1898 struct kvm_assigned_pci_dev assigned_dev;
1899
1900 r = -EFAULT;
1901 if (copy_from_user(&assigned_dev, argp, sizeof assigned_dev))
1902 goto out;
1903 r = kvm_vm_ioctl_deassign_device(kvm, &assigned_dev);
1904 if (r)
1905 goto out;
1906 break;
1907 }
1908#endif
Avi Kivityf17abe92007-02-21 19:28:04 +02001909 default:
Carsten Otte1fe779f2007-10-29 16:08:35 +01001910 r = kvm_arch_vm_ioctl(filp, ioctl, arg);
Avi Kivityf17abe92007-02-21 19:28:04 +02001911 }
1912out:
1913 return r;
1914}
1915
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001916static int kvm_vm_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
Avi Kivityf17abe92007-02-21 19:28:04 +02001917{
Marcelo Tosatti777b3f42008-09-16 20:54:46 -03001918 struct page *page[1];
1919 unsigned long addr;
1920 int npages;
1921 gfn_t gfn = vmf->pgoff;
Avi Kivityf17abe92007-02-21 19:28:04 +02001922 struct kvm *kvm = vma->vm_file->private_data;
Avi Kivityf17abe92007-02-21 19:28:04 +02001923
Marcelo Tosatti777b3f42008-09-16 20:54:46 -03001924 addr = gfn_to_hva(kvm, gfn);
1925 if (kvm_is_error_hva(addr))
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001926 return VM_FAULT_SIGBUS;
Marcelo Tosatti777b3f42008-09-16 20:54:46 -03001927
1928 npages = get_user_pages(current, current->mm, addr, 1, 1, 0, page,
1929 NULL);
1930 if (unlikely(npages != 1))
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001931 return VM_FAULT_SIGBUS;
Marcelo Tosatti777b3f42008-09-16 20:54:46 -03001932
1933 vmf->page = page[0];
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001934 return 0;
Avi Kivityf17abe92007-02-21 19:28:04 +02001935}
1936
1937static struct vm_operations_struct kvm_vm_vm_ops = {
npiggin@suse.dee4a533a2007-12-05 18:15:52 +11001938 .fault = kvm_vm_fault,
Avi Kivityf17abe92007-02-21 19:28:04 +02001939};
1940
1941static int kvm_vm_mmap(struct file *file, struct vm_area_struct *vma)
1942{
1943 vma->vm_ops = &kvm_vm_vm_ops;
1944 return 0;
1945}
1946
Christian Borntraeger3d3aab12008-12-02 11:17:32 +01001947static struct file_operations kvm_vm_fops = {
Avi Kivityf17abe92007-02-21 19:28:04 +02001948 .release = kvm_vm_release,
1949 .unlocked_ioctl = kvm_vm_ioctl,
1950 .compat_ioctl = kvm_vm_ioctl,
1951 .mmap = kvm_vm_mmap,
1952};
1953
1954static int kvm_dev_ioctl_create_vm(void)
1955{
Al Viro2030a422008-02-23 06:46:49 -05001956 int fd;
Avi Kivityf17abe92007-02-21 19:28:04 +02001957 struct kvm *kvm;
1958
Avi Kivityf17abe92007-02-21 19:28:04 +02001959 kvm = kvm_create_vm();
Avi Kivityd6d28162007-06-28 08:38:16 -04001960 if (IS_ERR(kvm))
1961 return PTR_ERR(kvm);
Ulrich Drepper7d9dbca2008-07-23 21:29:22 -07001962 fd = anon_inode_getfd("kvm-vm", &kvm_vm_fops, kvm, 0);
Al Viro2030a422008-02-23 06:46:49 -05001963 if (fd < 0)
Al Viro66c0b392008-04-19 20:33:56 +01001964 kvm_put_kvm(kvm);
Avi Kivityf17abe92007-02-21 19:28:04 +02001965
Avi Kivityf17abe92007-02-21 19:28:04 +02001966 return fd;
Avi Kivityf17abe92007-02-21 19:28:04 +02001967}
1968
Avi Kivity1a811b62008-12-08 18:25:27 +02001969static long kvm_dev_ioctl_check_extension_generic(long arg)
1970{
1971 switch (arg) {
Avi Kivityca9edae2008-12-08 18:29:29 +02001972 case KVM_CAP_USER_MEMORY:
Avi Kivity1a811b62008-12-08 18:25:27 +02001973 case KVM_CAP_DESTROY_MEMORY_REGION_WORKS:
1974 return 1;
1975 default:
1976 break;
1977 }
1978 return kvm_dev_ioctl_check_extension(arg);
1979}
1980
Avi Kivityf17abe92007-02-21 19:28:04 +02001981static long kvm_dev_ioctl(struct file *filp,
1982 unsigned int ioctl, unsigned long arg)
1983{
Avi Kivity07c45a32007-03-07 13:05:38 +02001984 long r = -EINVAL;
Avi Kivityf17abe92007-02-21 19:28:04 +02001985
1986 switch (ioctl) {
1987 case KVM_GET_API_VERSION:
Avi Kivityf0fe5102007-03-07 13:11:17 +02001988 r = -EINVAL;
1989 if (arg)
1990 goto out;
Avi Kivityf17abe92007-02-21 19:28:04 +02001991 r = KVM_API_VERSION;
1992 break;
1993 case KVM_CREATE_VM:
Avi Kivityf0fe5102007-03-07 13:11:17 +02001994 r = -EINVAL;
1995 if (arg)
1996 goto out;
Avi Kivityf17abe92007-02-21 19:28:04 +02001997 r = kvm_dev_ioctl_create_vm();
1998 break;
Zhang Xiantao018d00d2007-11-15 23:07:47 +08001999 case KVM_CHECK_EXTENSION:
Avi Kivity1a811b62008-12-08 18:25:27 +02002000 r = kvm_dev_ioctl_check_extension_generic(arg);
Avi Kivity5d308f42007-03-01 17:56:20 +02002001 break;
Avi Kivity07c45a32007-03-07 13:05:38 +02002002 case KVM_GET_VCPU_MMAP_SIZE:
2003 r = -EINVAL;
2004 if (arg)
2005 goto out;
Avi Kivityadb1ff42008-01-24 15:13:08 +02002006 r = PAGE_SIZE; /* struct kvm_run */
2007#ifdef CONFIG_X86
2008 r += PAGE_SIZE; /* pio data page */
2009#endif
Laurent Vivier5f94c172008-05-30 16:05:54 +02002010#ifdef KVM_COALESCED_MMIO_PAGE_OFFSET
2011 r += PAGE_SIZE; /* coalesced mmio ring page */
2012#endif
Avi Kivity07c45a32007-03-07 13:05:38 +02002013 break;
Feng(Eric) Liud4c9ff22008-04-10 08:47:53 -04002014 case KVM_TRACE_ENABLE:
2015 case KVM_TRACE_PAUSE:
2016 case KVM_TRACE_DISABLE:
2017 r = kvm_trace_ioctl(ioctl, arg);
2018 break;
Avi Kivity6aa8b732006-12-10 02:21:36 -08002019 default:
Carsten Otte043405e2007-10-10 17:16:19 +02002020 return kvm_arch_dev_ioctl(filp, ioctl, arg);
Avi Kivity6aa8b732006-12-10 02:21:36 -08002021 }
2022out:
2023 return r;
2024}
2025
Avi Kivity6aa8b732006-12-10 02:21:36 -08002026static struct file_operations kvm_chardev_ops = {
Avi Kivity6aa8b732006-12-10 02:21:36 -08002027 .unlocked_ioctl = kvm_dev_ioctl,
2028 .compat_ioctl = kvm_dev_ioctl,
Avi Kivity6aa8b732006-12-10 02:21:36 -08002029};
2030
2031static struct miscdevice kvm_dev = {
Avi Kivitybbe44322007-03-04 13:27:36 +02002032 KVM_MINOR,
Avi Kivity6aa8b732006-12-10 02:21:36 -08002033 "kvm",
2034 &kvm_chardev_ops,
2035};
2036
Avi Kivity1b6c0162007-05-24 13:03:52 +03002037static void hardware_enable(void *junk)
2038{
2039 int cpu = raw_smp_processor_id();
2040
Rusty Russell7f59f492008-12-07 21:25:45 +10302041 if (cpumask_test_cpu(cpu, cpus_hardware_enabled))
Avi Kivity1b6c0162007-05-24 13:03:52 +03002042 return;
Rusty Russell7f59f492008-12-07 21:25:45 +10302043 cpumask_set_cpu(cpu, cpus_hardware_enabled);
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08002044 kvm_arch_hardware_enable(NULL);
Avi Kivity1b6c0162007-05-24 13:03:52 +03002045}
2046
2047static void hardware_disable(void *junk)
2048{
2049 int cpu = raw_smp_processor_id();
2050
Rusty Russell7f59f492008-12-07 21:25:45 +10302051 if (!cpumask_test_cpu(cpu, cpus_hardware_enabled))
Avi Kivity1b6c0162007-05-24 13:03:52 +03002052 return;
Rusty Russell7f59f492008-12-07 21:25:45 +10302053 cpumask_clear_cpu(cpu, cpus_hardware_enabled);
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08002054 kvm_arch_hardware_disable(NULL);
Avi Kivity1b6c0162007-05-24 13:03:52 +03002055}
2056
Avi Kivity774c47f2007-02-12 00:54:47 -08002057static int kvm_cpu_hotplug(struct notifier_block *notifier, unsigned long val,
2058 void *v)
2059{
2060 int cpu = (long)v;
2061
Avi Kivity1a6f4d72007-11-11 18:37:32 +02002062 val &= ~CPU_TASKS_FROZEN;
Avi Kivity774c47f2007-02-12 00:54:47 -08002063 switch (val) {
Avi Kivitycec9ad22007-05-24 13:11:41 +03002064 case CPU_DYING:
Avi Kivity6ec8a852007-08-19 15:57:26 +03002065 printk(KERN_INFO "kvm: disabling virtualization on CPU%d\n",
2066 cpu);
2067 hardware_disable(NULL);
2068 break;
Avi Kivity774c47f2007-02-12 00:54:47 -08002069 case CPU_UP_CANCELED:
Jeremy Katz43934a32007-02-19 14:37:46 +02002070 printk(KERN_INFO "kvm: disabling virtualization on CPU%d\n",
2071 cpu);
Jens Axboe8691e5a2008-06-06 11:18:06 +02002072 smp_call_function_single(cpu, hardware_disable, NULL, 1);
Avi Kivity774c47f2007-02-12 00:54:47 -08002073 break;
Jeremy Katz43934a32007-02-19 14:37:46 +02002074 case CPU_ONLINE:
2075 printk(KERN_INFO "kvm: enabling virtualization on CPU%d\n",
2076 cpu);
Jens Axboe8691e5a2008-06-06 11:18:06 +02002077 smp_call_function_single(cpu, hardware_enable, NULL, 1);
Avi Kivity774c47f2007-02-12 00:54:47 -08002078 break;
2079 }
2080 return NOTIFY_OK;
2081}
2082
Avi Kivity4ecac3f2008-05-13 13:23:38 +03002083
2084asmlinkage void kvm_handle_fault_on_reboot(void)
2085{
2086 if (kvm_rebooting)
2087 /* spin while reset goes on */
2088 while (true)
2089 ;
2090 /* Fault while not rebooting. We want the trace. */
2091 BUG();
2092}
2093EXPORT_SYMBOL_GPL(kvm_handle_fault_on_reboot);
2094
Rusty Russell9a2b85c2007-07-17 23:17:55 +10002095static int kvm_reboot(struct notifier_block *notifier, unsigned long val,
Mike Dayd77c26f2007-10-08 09:02:08 -04002096 void *v)
Rusty Russell9a2b85c2007-07-17 23:17:55 +10002097{
2098 if (val == SYS_RESTART) {
2099 /*
2100 * Some (well, at least mine) BIOSes hang on reboot if
2101 * in vmx root mode.
2102 */
2103 printk(KERN_INFO "kvm: exiting hardware virtualization\n");
Avi Kivity4ecac3f2008-05-13 13:23:38 +03002104 kvm_rebooting = true;
Jens Axboe15c8b6c2008-05-09 09:39:44 +02002105 on_each_cpu(hardware_disable, NULL, 1);
Rusty Russell9a2b85c2007-07-17 23:17:55 +10002106 }
2107 return NOTIFY_OK;
2108}
2109
2110static struct notifier_block kvm_reboot_notifier = {
2111 .notifier_call = kvm_reboot,
2112 .priority = 0,
2113};
2114
Gregory Haskins2eeb2e92007-05-31 14:08:53 -04002115void kvm_io_bus_init(struct kvm_io_bus *bus)
2116{
2117 memset(bus, 0, sizeof(*bus));
2118}
2119
2120void kvm_io_bus_destroy(struct kvm_io_bus *bus)
2121{
2122 int i;
2123
2124 for (i = 0; i < bus->dev_count; i++) {
2125 struct kvm_io_device *pos = bus->devs[i];
2126
2127 kvm_iodevice_destructor(pos);
2128 }
2129}
2130
Laurent Vivier92760492008-05-30 16:05:53 +02002131struct kvm_io_device *kvm_io_bus_find_dev(struct kvm_io_bus *bus,
2132 gpa_t addr, int len, int is_write)
Gregory Haskins2eeb2e92007-05-31 14:08:53 -04002133{
2134 int i;
2135
2136 for (i = 0; i < bus->dev_count; i++) {
2137 struct kvm_io_device *pos = bus->devs[i];
2138
Laurent Vivier92760492008-05-30 16:05:53 +02002139 if (pos->in_range(pos, addr, len, is_write))
Gregory Haskins2eeb2e92007-05-31 14:08:53 -04002140 return pos;
2141 }
2142
2143 return NULL;
2144}
2145
2146void kvm_io_bus_register_dev(struct kvm_io_bus *bus, struct kvm_io_device *dev)
2147{
2148 BUG_ON(bus->dev_count > (NR_IOBUS_DEVS-1));
2149
2150 bus->devs[bus->dev_count++] = dev;
2151}
2152
Avi Kivity774c47f2007-02-12 00:54:47 -08002153static struct notifier_block kvm_cpu_notifier = {
2154 .notifier_call = kvm_cpu_hotplug,
2155 .priority = 20, /* must be > scheduler priority */
2156};
2157
Christoph Hellwig8b88b092008-02-08 04:20:26 -08002158static int vm_stat_get(void *_offset, u64 *val)
Avi Kivityba1389b2007-11-18 16:24:12 +02002159{
2160 unsigned offset = (long)_offset;
Avi Kivityba1389b2007-11-18 16:24:12 +02002161 struct kvm *kvm;
2162
Christoph Hellwig8b88b092008-02-08 04:20:26 -08002163 *val = 0;
Avi Kivityba1389b2007-11-18 16:24:12 +02002164 spin_lock(&kvm_lock);
2165 list_for_each_entry(kvm, &vm_list, vm_list)
Christoph Hellwig8b88b092008-02-08 04:20:26 -08002166 *val += *(u32 *)((void *)kvm + offset);
Avi Kivityba1389b2007-11-18 16:24:12 +02002167 spin_unlock(&kvm_lock);
Christoph Hellwig8b88b092008-02-08 04:20:26 -08002168 return 0;
Avi Kivityba1389b2007-11-18 16:24:12 +02002169}
2170
2171DEFINE_SIMPLE_ATTRIBUTE(vm_stat_fops, vm_stat_get, NULL, "%llu\n");
2172
Christoph Hellwig8b88b092008-02-08 04:20:26 -08002173static int vcpu_stat_get(void *_offset, u64 *val)
Avi Kivity1165f5f2007-04-19 17:27:43 +03002174{
2175 unsigned offset = (long)_offset;
Avi Kivity1165f5f2007-04-19 17:27:43 +03002176 struct kvm *kvm;
2177 struct kvm_vcpu *vcpu;
2178 int i;
2179
Christoph Hellwig8b88b092008-02-08 04:20:26 -08002180 *val = 0;
Avi Kivity1165f5f2007-04-19 17:27:43 +03002181 spin_lock(&kvm_lock);
2182 list_for_each_entry(kvm, &vm_list, vm_list)
2183 for (i = 0; i < KVM_MAX_VCPUS; ++i) {
Rusty Russellfb3f0f52007-07-27 17:16:56 +10002184 vcpu = kvm->vcpus[i];
2185 if (vcpu)
Christoph Hellwig8b88b092008-02-08 04:20:26 -08002186 *val += *(u32 *)((void *)vcpu + offset);
Avi Kivity1165f5f2007-04-19 17:27:43 +03002187 }
2188 spin_unlock(&kvm_lock);
Christoph Hellwig8b88b092008-02-08 04:20:26 -08002189 return 0;
Avi Kivity1165f5f2007-04-19 17:27:43 +03002190}
2191
Avi Kivityba1389b2007-11-18 16:24:12 +02002192DEFINE_SIMPLE_ATTRIBUTE(vcpu_stat_fops, vcpu_stat_get, NULL, "%llu\n");
2193
2194static struct file_operations *stat_fops[] = {
2195 [KVM_STAT_VCPU] = &vcpu_stat_fops,
2196 [KVM_STAT_VM] = &vm_stat_fops,
2197};
Avi Kivity1165f5f2007-04-19 17:27:43 +03002198
Zhang Xiantaoa16b0432007-11-16 14:38:21 +08002199static void kvm_init_debug(void)
Avi Kivity6aa8b732006-12-10 02:21:36 -08002200{
2201 struct kvm_stats_debugfs_item *p;
2202
Hollis Blanchard76f7c872008-04-15 16:05:42 -05002203 kvm_debugfs_dir = debugfs_create_dir("kvm", NULL);
Avi Kivity6aa8b732006-12-10 02:21:36 -08002204 for (p = debugfs_entries; p->name; ++p)
Hollis Blanchard76f7c872008-04-15 16:05:42 -05002205 p->dentry = debugfs_create_file(p->name, 0444, kvm_debugfs_dir,
Avi Kivity1165f5f2007-04-19 17:27:43 +03002206 (void *)(long)p->offset,
Avi Kivityba1389b2007-11-18 16:24:12 +02002207 stat_fops[p->kind]);
Avi Kivity6aa8b732006-12-10 02:21:36 -08002208}
2209
2210static void kvm_exit_debug(void)
2211{
2212 struct kvm_stats_debugfs_item *p;
2213
2214 for (p = debugfs_entries; p->name; ++p)
2215 debugfs_remove(p->dentry);
Hollis Blanchard76f7c872008-04-15 16:05:42 -05002216 debugfs_remove(kvm_debugfs_dir);
Avi Kivity6aa8b732006-12-10 02:21:36 -08002217}
2218
Avi Kivity59ae6c62007-02-12 00:54:48 -08002219static int kvm_suspend(struct sys_device *dev, pm_message_t state)
2220{
Avi Kivity4267c412007-05-24 13:09:41 +03002221 hardware_disable(NULL);
Avi Kivity59ae6c62007-02-12 00:54:48 -08002222 return 0;
2223}
2224
2225static int kvm_resume(struct sys_device *dev)
2226{
Avi Kivity4267c412007-05-24 13:09:41 +03002227 hardware_enable(NULL);
Avi Kivity59ae6c62007-02-12 00:54:48 -08002228 return 0;
2229}
2230
2231static struct sysdev_class kvm_sysdev_class = {
Kay Sieversaf5ca3f2007-12-20 02:09:39 +01002232 .name = "kvm",
Avi Kivity59ae6c62007-02-12 00:54:48 -08002233 .suspend = kvm_suspend,
2234 .resume = kvm_resume,
2235};
2236
2237static struct sys_device kvm_sysdev = {
2238 .id = 0,
2239 .cls = &kvm_sysdev_class,
2240};
2241
Izik Eiduscea7bb22007-10-17 19:17:48 +02002242struct page *bad_page;
Anthony Liguori35149e22008-04-02 14:46:56 -05002243pfn_t bad_pfn;
Avi Kivity6aa8b732006-12-10 02:21:36 -08002244
Avi Kivity15ad7142007-07-11 18:17:21 +03002245static inline
2246struct kvm_vcpu *preempt_notifier_to_vcpu(struct preempt_notifier *pn)
2247{
2248 return container_of(pn, struct kvm_vcpu, preempt_notifier);
2249}
2250
2251static void kvm_sched_in(struct preempt_notifier *pn, int cpu)
2252{
2253 struct kvm_vcpu *vcpu = preempt_notifier_to_vcpu(pn);
2254
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08002255 kvm_arch_vcpu_load(vcpu, cpu);
Avi Kivity15ad7142007-07-11 18:17:21 +03002256}
2257
2258static void kvm_sched_out(struct preempt_notifier *pn,
2259 struct task_struct *next)
2260{
2261 struct kvm_vcpu *vcpu = preempt_notifier_to_vcpu(pn);
2262
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08002263 kvm_arch_vcpu_put(vcpu);
Avi Kivity15ad7142007-07-11 18:17:21 +03002264}
2265
Zhang Xiantaof8c16bb2007-11-14 20:40:21 +08002266int kvm_init(void *opaque, unsigned int vcpu_size,
Rusty Russellc16f8622007-07-30 21:12:19 +10002267 struct module *module)
Avi Kivity6aa8b732006-12-10 02:21:36 -08002268{
2269 int r;
Yang, Sheng002c7f72007-07-31 14:23:01 +03002270 int cpu;
Avi Kivity6aa8b732006-12-10 02:21:36 -08002271
Zhang Xiantaocb498ea2007-11-14 20:39:31 +08002272 kvm_init_debug();
2273
Zhang Xiantaof8c16bb2007-11-14 20:40:21 +08002274 r = kvm_arch_init(opaque);
2275 if (r)
Zhang Xiantaod23087842007-11-29 15:35:39 +08002276 goto out_fail;
Zhang Xiantaocb498ea2007-11-14 20:39:31 +08002277
2278 bad_page = alloc_page(GFP_KERNEL | __GFP_ZERO);
2279
2280 if (bad_page == NULL) {
2281 r = -ENOMEM;
2282 goto out;
2283 }
2284
Anthony Liguori35149e22008-04-02 14:46:56 -05002285 bad_pfn = page_to_pfn(bad_page);
2286
Rusty Russell7f59f492008-12-07 21:25:45 +10302287 if (!alloc_cpumask_var(&cpus_hardware_enabled, GFP_KERNEL)) {
2288 r = -ENOMEM;
2289 goto out_free_0;
2290 }
2291
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08002292 r = kvm_arch_hardware_setup();
Avi Kivity6aa8b732006-12-10 02:21:36 -08002293 if (r < 0)
Rusty Russell7f59f492008-12-07 21:25:45 +10302294 goto out_free_0a;
Avi Kivity6aa8b732006-12-10 02:21:36 -08002295
Yang, Sheng002c7f72007-07-31 14:23:01 +03002296 for_each_online_cpu(cpu) {
2297 smp_call_function_single(cpu,
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08002298 kvm_arch_check_processor_compat,
Jens Axboe8691e5a2008-06-06 11:18:06 +02002299 &r, 1);
Yang, Sheng002c7f72007-07-31 14:23:01 +03002300 if (r < 0)
Zhang Xiantaod23087842007-11-29 15:35:39 +08002301 goto out_free_1;
Yang, Sheng002c7f72007-07-31 14:23:01 +03002302 }
2303
Jens Axboe15c8b6c2008-05-09 09:39:44 +02002304 on_each_cpu(hardware_enable, NULL, 1);
Avi Kivity774c47f2007-02-12 00:54:47 -08002305 r = register_cpu_notifier(&kvm_cpu_notifier);
2306 if (r)
Zhang Xiantaod23087842007-11-29 15:35:39 +08002307 goto out_free_2;
Avi Kivity6aa8b732006-12-10 02:21:36 -08002308 register_reboot_notifier(&kvm_reboot_notifier);
2309
Avi Kivity59ae6c62007-02-12 00:54:48 -08002310 r = sysdev_class_register(&kvm_sysdev_class);
2311 if (r)
Zhang Xiantaod23087842007-11-29 15:35:39 +08002312 goto out_free_3;
Avi Kivity59ae6c62007-02-12 00:54:48 -08002313
2314 r = sysdev_register(&kvm_sysdev);
2315 if (r)
Zhang Xiantaod23087842007-11-29 15:35:39 +08002316 goto out_free_4;
Avi Kivity59ae6c62007-02-12 00:54:48 -08002317
Rusty Russellc16f8622007-07-30 21:12:19 +10002318 /* A kmem cache lets us meet the alignment requirements of fx_save. */
2319 kvm_vcpu_cache = kmem_cache_create("kvm_vcpu", vcpu_size,
Joe Perches56919c52007-11-12 20:06:51 -08002320 __alignof__(struct kvm_vcpu),
2321 0, NULL);
Rusty Russellc16f8622007-07-30 21:12:19 +10002322 if (!kvm_vcpu_cache) {
2323 r = -ENOMEM;
Zhang Xiantaod23087842007-11-29 15:35:39 +08002324 goto out_free_5;
Rusty Russellc16f8622007-07-30 21:12:19 +10002325 }
2326
Avi Kivity6aa8b732006-12-10 02:21:36 -08002327 kvm_chardev_ops.owner = module;
Christian Borntraeger3d3aab12008-12-02 11:17:32 +01002328 kvm_vm_fops.owner = module;
2329 kvm_vcpu_fops.owner = module;
Avi Kivity6aa8b732006-12-10 02:21:36 -08002330
2331 r = misc_register(&kvm_dev);
2332 if (r) {
Mike Dayd77c26f2007-10-08 09:02:08 -04002333 printk(KERN_ERR "kvm: misc device register failed\n");
Avi Kivity6aa8b732006-12-10 02:21:36 -08002334 goto out_free;
2335 }
2336
Avi Kivity15ad7142007-07-11 18:17:21 +03002337 kvm_preempt_ops.sched_in = kvm_sched_in;
2338 kvm_preempt_ops.sched_out = kvm_sched_out;
Sheng Yang5319c662008-11-24 14:32:57 +08002339#ifndef CONFIG_X86
2340 msi2intx = 0;
2341#endif
Avi Kivity15ad7142007-07-11 18:17:21 +03002342
Avi Kivityc7addb92007-09-16 18:58:32 +02002343 return 0;
Avi Kivity6aa8b732006-12-10 02:21:36 -08002344
2345out_free:
Rusty Russellc16f8622007-07-30 21:12:19 +10002346 kmem_cache_destroy(kvm_vcpu_cache);
Zhang Xiantaod23087842007-11-29 15:35:39 +08002347out_free_5:
Avi Kivity59ae6c62007-02-12 00:54:48 -08002348 sysdev_unregister(&kvm_sysdev);
Zhang Xiantaod23087842007-11-29 15:35:39 +08002349out_free_4:
Avi Kivity59ae6c62007-02-12 00:54:48 -08002350 sysdev_class_unregister(&kvm_sysdev_class);
Zhang Xiantaod23087842007-11-29 15:35:39 +08002351out_free_3:
Avi Kivity6aa8b732006-12-10 02:21:36 -08002352 unregister_reboot_notifier(&kvm_reboot_notifier);
Avi Kivity774c47f2007-02-12 00:54:47 -08002353 unregister_cpu_notifier(&kvm_cpu_notifier);
Zhang Xiantaod23087842007-11-29 15:35:39 +08002354out_free_2:
Jens Axboe15c8b6c2008-05-09 09:39:44 +02002355 on_each_cpu(hardware_disable, NULL, 1);
Zhang Xiantaod23087842007-11-29 15:35:39 +08002356out_free_1:
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08002357 kvm_arch_hardware_unsetup();
Rusty Russell7f59f492008-12-07 21:25:45 +10302358out_free_0a:
2359 free_cpumask_var(cpus_hardware_enabled);
Zhang Xiantaod23087842007-11-29 15:35:39 +08002360out_free_0:
2361 __free_page(bad_page);
Avi Kivityca45aaa2007-03-01 19:21:03 +02002362out:
Zhang Xiantaof8c16bb2007-11-14 20:40:21 +08002363 kvm_arch_exit();
Zhang Xiantaocb498ea2007-11-14 20:39:31 +08002364 kvm_exit_debug();
Zhang Xiantaod23087842007-11-29 15:35:39 +08002365out_fail:
Avi Kivity6aa8b732006-12-10 02:21:36 -08002366 return r;
2367}
Zhang Xiantaocb498ea2007-11-14 20:39:31 +08002368EXPORT_SYMBOL_GPL(kvm_init);
Avi Kivity6aa8b732006-12-10 02:21:36 -08002369
Zhang Xiantaocb498ea2007-11-14 20:39:31 +08002370void kvm_exit(void)
Avi Kivity6aa8b732006-12-10 02:21:36 -08002371{
Feng(Eric) Liud4c9ff22008-04-10 08:47:53 -04002372 kvm_trace_cleanup();
Avi Kivity6aa8b732006-12-10 02:21:36 -08002373 misc_deregister(&kvm_dev);
Rusty Russellc16f8622007-07-30 21:12:19 +10002374 kmem_cache_destroy(kvm_vcpu_cache);
Avi Kivity59ae6c62007-02-12 00:54:48 -08002375 sysdev_unregister(&kvm_sysdev);
2376 sysdev_class_unregister(&kvm_sysdev_class);
Avi Kivity6aa8b732006-12-10 02:21:36 -08002377 unregister_reboot_notifier(&kvm_reboot_notifier);
Avi Kivity59ae6c62007-02-12 00:54:48 -08002378 unregister_cpu_notifier(&kvm_cpu_notifier);
Jens Axboe15c8b6c2008-05-09 09:39:44 +02002379 on_each_cpu(hardware_disable, NULL, 1);
Zhang Xiantaoe9b11c12007-11-14 20:38:21 +08002380 kvm_arch_hardware_unsetup();
Zhang Xiantaof8c16bb2007-11-14 20:40:21 +08002381 kvm_arch_exit();
Avi Kivity6aa8b732006-12-10 02:21:36 -08002382 kvm_exit_debug();
Rusty Russell7f59f492008-12-07 21:25:45 +10302383 free_cpumask_var(cpus_hardware_enabled);
Izik Eiduscea7bb22007-10-17 19:17:48 +02002384 __free_page(bad_page);
Avi Kivity6aa8b732006-12-10 02:21:36 -08002385}
Zhang Xiantaocb498ea2007-11-14 20:39:31 +08002386EXPORT_SYMBOL_GPL(kvm_exit);