KVM: MMU: Make use of is_large_pte() in walker

Make use of is_large_pte() instead of checking PT_PAGE_SIZE_MASK
bit directly.

Signed-off-by: Gui Jianfeng <guijianfeng@cn.fujitsu.com>
Signed-off-by: Avi Kivity <avi@redhat.com>

1 files changed, 2 insertions, 2 deletions

diff --git a/arch/x86/kvm/paging_tmpl.h b/arch/x86/kvm/paging_tmpl.h
index 5910557b3f3..d0cc07eb6ed 100644
--- a/arch/x86/kvm/paging_tmpl.h
+++ b/arch/x86/kvm/paging_tmpl.h
@@ -190,10 +190,10 @@ walk:
 
                 if ((walker->level == PT_PAGE_TABLE_LEVEL) ||
                     ((walker->level == PT_DIRECTORY_LEVEL) &&
-                                (pte & PT_PAGE_SIZE_MASK)  &&
+                                is_large_pte(pte) &&
                                 (PTTYPE == 64 || is_pse(vcpu))) ||
                     ((walker->level == PT_PDPE_LEVEL) &&
-                                (pte & PT_PAGE_SIZE_MASK)  &&
+                                is_large_pte(pte) &&
                                 is_long_mode(vcpu))) {
                         int lvl = walker->level;