static void vmx_sync_pir_to_irr_dummy(struct kvm_vcpu *vcpu);
 static void copy_vmcs12_to_shadow(struct vcpu_vmx *vmx);
 static void copy_shadow_to_vmcs12(struct vcpu_vmx *vmx);
+static int alloc_identity_pagetable(struct kvm *kvm);
 
 static DEFINE_PER_CPU(struct vmcs *, vmxarea);
 static DEFINE_PER_CPU(struct vmcs *, current_vmcs);
 
 static int init_rmode_identity_map(struct kvm *kvm)
 {
-       int i, idx, r, ret;
+       int i, idx, r, ret = 0;
        pfn_t identity_map_pfn;
        u32 tmp;
 
        if (!enable_ept)
                return 1;
-       if (unlikely(!kvm->arch.ept_identity_pagetable)) {
-               printk(KERN_ERR "EPT: identity-mapping pagetable "
-                       "haven't been allocated!\n");
-               return 0;
+
+       /* Protect kvm->arch.ept_identity_pagetable_done. */
+       mutex_lock(&kvm->slots_lock);
+
+       if (likely(kvm->arch.ept_identity_pagetable_done)) {
+               ret = 1;
+               goto out2;
        }
-       if (likely(kvm->arch.ept_identity_pagetable_done))
-               return 1;
-       ret = 0;
+
        identity_map_pfn = kvm->arch.ept_identity_map_addr >> PAGE_SHIFT;
+
+       r = alloc_identity_pagetable(kvm);
+       if (r)
+               goto out2;
+
        idx = srcu_read_lock(&kvm->srcu);
        r = kvm_clear_guest_page(kvm, identity_map_pfn, 0, PAGE_SIZE);
        if (r < 0)
        ret = 1;
 out:
        srcu_read_unlock(&kvm->srcu, idx);
+
+out2:
+       mutex_unlock(&kvm->slots_lock);
        return ret;
 }
 
 
 static int alloc_identity_pagetable(struct kvm *kvm)
 {
-       struct page *page;
+       /* Called with kvm->slots_lock held. */
+
        struct kvm_userspace_memory_region kvm_userspace_mem;
        int r = 0;
 
-       mutex_lock(&kvm->slots_lock);
-       if (kvm->arch.ept_identity_pagetable)
-               goto out;
+       BUG_ON(kvm->arch.ept_identity_pagetable_done);
+
        kvm_userspace_mem.slot = IDENTITY_PAGETABLE_PRIVATE_MEMSLOT;
        kvm_userspace_mem.flags = 0;
        kvm_userspace_mem.guest_phys_addr =
                kvm->arch.ept_identity_map_addr;
        kvm_userspace_mem.memory_size = PAGE_SIZE;
        r = __kvm_set_memory_region(kvm, &kvm_userspace_mem);
-       if (r)
-               goto out;
-
-       page = gfn_to_page(kvm, kvm->arch.ept_identity_map_addr >> PAGE_SHIFT);
-       if (is_error_page(page)) {
-               r = -EFAULT;
-               goto out;
-       }
 
-       kvm->arch.ept_identity_pagetable = page;
-out:
-       mutex_unlock(&kvm->slots_lock);
        return r;
 }
 
                        kvm->arch.ept_identity_map_addr =
                                VMX_EPT_IDENTITY_PAGETABLE_ADDR;
                err = -ENOMEM;
-               if (alloc_identity_pagetable(kvm) != 0)
-                       goto free_vmcs;
                if (!init_rmode_identity_map(kvm))
                        goto free_vmcs;
        }