KVM: use get_task_mm() instead of atomic_inc the mm_count
Signed-off-by: Izik Eidus <izike@qumranet.com>
Signed-off-by: Avi Kivity <avi@qumranet.com>
diff --git a/virt/kvm/kvm_main.c b/virt/kvm/kvm_main.c
index 30bf832..a96a44f 100644
--- a/virt/kvm/kvm_main.c
+++ b/virt/kvm/kvm_main.c
@@ -186,8 +186,7 @@
if (IS_ERR(kvm))
goto out;
- kvm->mm = current->mm;
- atomic_inc(&kvm->mm->mm_count);
+ kvm->mm = get_task_mm(current);
spin_lock_init(&kvm->mmu_lock);
kvm_io_bus_init(&kvm->pio_bus);
mutex_init(&kvm->lock);
@@ -239,7 +238,7 @@
kvm_io_bus_destroy(&kvm->pio_bus);
kvm_io_bus_destroy(&kvm->mmio_bus);
kvm_arch_destroy_vm(kvm);
- mmdrop(mm);
+ mmput(mm);
}
static int kvm_vm_release(struct inode *inode, struct file *filp)
