From d99e0eb6789b1497cbd4597212436faaf0b175c5 Mon Sep 17 00:00:00 2001
From: Lai Jiangshan <laijs@cn.fujitsu.com>
Date: Wed, 26 May 2010 16:48:25 +0800
Subject: [PATCH] --- yaml --- r: 202301 b: refs/heads/master c:
 c9fa0b3bef9a0b117b3c3f958ec553c21f609a9f h: refs/heads/master i:   202299:
 bb1e399c0f8a4bfee6153c03214590a14db1c14d v: v3

---
 [refs]                   | 2 +-
 trunk/arch/x86/kvm/mmu.c | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/[refs] b/[refs]
index ff9fa5bcfe17..b3ca40702474 100644
--- a/[refs]
+++ b/[refs]
@@ -1,2 +1,2 @@
 ---
-refs/heads/master: 2032a93d66fa282ba0f2ea9152eeff9511fa9a96
+refs/heads/master: c9fa0b3bef9a0b117b3c3f958ec553c21f609a9f
diff --git a/trunk/arch/x86/kvm/mmu.c b/trunk/arch/x86/kvm/mmu.c
index f46b6c9aff27..c0350be52c91 100644
--- a/trunk/arch/x86/kvm/mmu.c
+++ b/trunk/arch/x86/kvm/mmu.c
@@ -2020,7 +2020,10 @@ static int __direct_map(struct kvm_vcpu *vcpu, gpa_t v, int write,
 		}
 
 		if (*iterator.sptep == shadow_trap_nonpresent_pte) {
-			pseudo_gfn = (iterator.addr & PT64_DIR_BASE_ADDR_MASK) >> PAGE_SHIFT;
+			u64 base_addr = iterator.addr;
+
+			base_addr &= PT64_LVL_ADDR_MASK(iterator.level);
+			pseudo_gfn = base_addr >> PAGE_SHIFT;
 			sp = kvm_mmu_get_page(vcpu, pseudo_gfn, iterator.addr,
 					      iterator.level - 1,
 					      1, ACC_ALL, iterator.sptep);