void ptdump_walk(struct seq_file *s, struct ptdump_info *info);
 void note_page(struct ptdump_state *pt_st, unsigned long addr, int level,
-              u64 val);
+              pteval_t val);
+void note_page_pte(struct ptdump_state *st, unsigned long addr, pte_t pte);
+void note_page_pmd(struct ptdump_state *st, unsigned long addr, pmd_t pmd);
+void note_page_pud(struct ptdump_state *st, unsigned long addr, pud_t pud);
+void note_page_p4d(struct ptdump_state *st, unsigned long addr, p4d_t p4d);
+void note_page_pgd(struct ptdump_state *st, unsigned long addr, pgd_t pgd);
+void note_page_flush(struct ptdump_state *st);
 #ifdef CONFIG_PTDUMP_DEBUGFS
 #define EFI_RUNTIME_MAP_END    DEFAULT_MAP_WINDOW_64
 void __init ptdump_debugfs_register(struct ptdump_info *info, const char *name);
 #endif /* CONFIG_PTDUMP_DEBUGFS */
 #else
 static inline void note_page(struct ptdump_state *pt_st, unsigned long addr,
-                            int level, u64 val) { }
+                            int level, pteval_t val) { }
+static inline void note_page_pte(struct ptdump_state *st, unsigned long addr, pte_t pte) { }
+static inline void note_page_pmd(struct ptdump_state *st, unsigned long addr, pmd_t pmd) { }
+static inline void note_page_pud(struct ptdump_state *st, unsigned long addr, pud_t pud) { }
+static inline void note_page_p4d(struct ptdump_state *st, unsigned long addr, p4d_t p4d) { }
+static inline void note_page_pgd(struct ptdump_state *st, unsigned long addr, pgd_t pgd) { }
+static inline void note_page_flush(struct ptdump_state *st) { }
 #endif /* CONFIG_PTDUMP */
 
 #endif /* __ASM_PTDUMP_H */
 
 }
 
 void note_page(struct ptdump_state *pt_st, unsigned long addr, int level,
-              u64 val)
+              pteval_t val)
 {
        struct ptdump_pg_state *st = container_of(pt_st, struct ptdump_pg_state, ptdump);
        struct ptdump_pg_level *pg_level = st->pg_level;
 
 }
 
+void note_page_pte(struct ptdump_state *pt_st, unsigned long addr, pte_t pte)
+{
+       note_page(pt_st, addr, 4, pte_val(pte));
+}
+
+void note_page_pmd(struct ptdump_state *pt_st, unsigned long addr, pmd_t pmd)
+{
+       note_page(pt_st, addr, 3, pmd_val(pmd));
+}
+
+void note_page_pud(struct ptdump_state *pt_st, unsigned long addr, pud_t pud)
+{
+       note_page(pt_st, addr, 2, pud_val(pud));
+}
+
+void note_page_p4d(struct ptdump_state *pt_st, unsigned long addr, p4d_t p4d)
+{
+       note_page(pt_st, addr, 1, p4d_val(p4d));
+}
+
+void note_page_pgd(struct ptdump_state *pt_st, unsigned long addr, pgd_t pgd)
+{
+       note_page(pt_st, addr, 0, pgd_val(pgd));
+}
+
+void note_page_flush(struct ptdump_state *pt_st)
+{
+       pte_t pte_zero = {0};
+
+       note_page(pt_st, 0, -1, pte_val(pte_zero));
+}
+
 void ptdump_walk(struct seq_file *s, struct ptdump_info *info)
 {
        unsigned long end = ~0UL;
                .pg_level = &kernel_pg_levels[0],
                .level = -1,
                .ptdump = {
-                       .note_page = note_page,
+                       .note_page_pte = note_page_pte,
+                       .note_page_pmd = note_page_pmd,
+                       .note_page_pud = note_page_pud,
+                       .note_page_p4d = note_page_p4d,
+                       .note_page_pgd = note_page_pgd,
+                       .note_page_flush = note_page_flush,
                        .range = (struct ptdump_range[]){
                                {info->base_addr, end},
                                {0, 0}
                .level = -1,
                .check_wx = true,
                .ptdump = {
-                       .note_page = note_page,
+                       .note_page_pte = note_page_pte,
+                       .note_page_pmd = note_page_pmd,
+                       .note_page_pud = note_page_pud,
+                       .note_page_p4d = note_page_p4d,
+                       .note_page_pgd = note_page_pgd,
+                       .note_page_flush = note_page_flush,
                        .range = (struct ptdump_range[]) {
                                {_PAGE_OFFSET(vabits_actual), ~0UL},
                                {0, 0}
 
 #endif
 }
 
+static void note_page_pte(struct ptdump_state *pt_st, unsigned long addr, pte_t pte)
+{
+       note_page(pt_st, addr, 4, pte_val(pte));
+}
+
+static void note_page_pmd(struct ptdump_state *pt_st, unsigned long addr, pmd_t pmd)
+{
+       note_page(pt_st, addr, 3, pmd_val(pmd));
+}
+
+static void note_page_pud(struct ptdump_state *pt_st, unsigned long addr, pud_t pud)
+{
+       note_page(pt_st, addr, 2, pud_val(pud));
+}
+
+static void note_page_p4d(struct ptdump_state *pt_st, unsigned long addr, p4d_t p4d)
+{
+       note_page(pt_st, addr, 1, p4d_val(p4d));
+}
+
+static void note_page_pgd(struct ptdump_state *pt_st, unsigned long addr, pgd_t pgd)
+{
+       note_page(pt_st, addr, 0, pgd_val(pgd));
+}
+
+static void note_page_flush(struct ptdump_state *pt_st)
+{
+       pte_t pte_zero = {0};
+
+       note_page(pt_st, 0, -1, pte_val(pte_zero));
+}
+
 static int ptdump_show(struct seq_file *m, void *v)
 {
        struct pg_state st = {
                .marker = address_markers,
                .level = -1,
                .ptdump = {
-                       .note_page = note_page,
+                       .note_page_pte = note_page_pte,
+                       .note_page_pmd = note_page_pmd,
+                       .note_page_pud = note_page_pud,
+                       .note_page_p4d = note_page_p4d,
+                       .note_page_pgd = note_page_pgd,
+                       .note_page_flush = note_page_flush,
                        .range = ptdump_range,
                }
        };
                .level = -1,
                .check_wx = true,
                .ptdump = {
-                       .note_page = note_page,
+                       .note_page_pte = note_page_pte,
+                       .note_page_pmd = note_page_pmd,
+                       .note_page_pud = note_page_pud,
+                       .note_page_p4d = note_page_p4d,
+                       .note_page_pgd = note_page_pgd,
+                       .note_page_flush = note_page_flush,
                        .range = ptdump_range,
                }
        };
 
        }
 }
 
+static void note_page_pte(struct ptdump_state *pt_st, unsigned long addr, pte_t pte)
+{
+       note_page(pt_st, addr, 4, pte_val(pte));
+}
+
+static void note_page_pmd(struct ptdump_state *pt_st, unsigned long addr, pmd_t pmd)
+{
+       note_page(pt_st, addr, 3, pmd_val(pmd));
+}
+
+static void note_page_pud(struct ptdump_state *pt_st, unsigned long addr, pud_t pud)
+{
+       note_page(pt_st, addr, 2, pud_val(pud));
+}
+
+static void note_page_p4d(struct ptdump_state *pt_st, unsigned long addr, p4d_t p4d)
+{
+       note_page(pt_st, addr, 1, p4d_val(p4d));
+}
+
+static void note_page_pgd(struct ptdump_state *pt_st, unsigned long addr, pgd_t pgd)
+{
+       note_page(pt_st, addr, 0, pgd_val(pgd));
+}
+
+static void note_page_flush(struct ptdump_state *pt_st)
+{
+       pte_t pte_zero = {0};
+
+       note_page(pt_st, 0, -1, pte_val(pte_zero));
+}
+
 static void ptdump_walk(struct seq_file *s, struct ptd_mm_info *pinfo)
 {
        struct pg_state st = {
                .marker = pinfo->markers,
                .level = -1,
                .ptdump = {
-                       .note_page = note_page,
+                       .note_page_pte = note_page_pte,
+                       .note_page_pmd = note_page_pmd,
+                       .note_page_pud = note_page_pud,
+                       .note_page_p4d = note_page_p4d,
+                       .note_page_pgd = note_page_pgd,
+                       .note_page_flush = note_page_flush,
                        .range = (struct ptdump_range[]) {
                                {pinfo->base_addr, pinfo->end},
                                {0, 0}
                .level = -1,
                .check_wx = true,
                .ptdump = {
-                       .note_page = note_page,
+                       .note_page_pte = note_page_pte,
+                       .note_page_pmd = note_page_pmd,
+                       .note_page_pud = note_page_pud,
+                       .note_page_p4d = note_page_p4d,
+                       .note_page_pgd = note_page_pgd,
+                       .note_page_flush = note_page_flush,
                        .range = (struct ptdump_range[]) {
                                {KERN_VIRT_START, ULONG_MAX},
                                {0, 0}
 
        }
 }
 
+static void note_page_pte(struct ptdump_state *pt_st, unsigned long addr, pte_t pte)
+{
+       note_page(pt_st, addr, 4, pte_val(pte));
+}
+
+static void note_page_pmd(struct ptdump_state *pt_st, unsigned long addr, pmd_t pmd)
+{
+       note_page(pt_st, addr, 3, pmd_val(pmd));
+}
+
+static void note_page_pud(struct ptdump_state *pt_st, unsigned long addr, pud_t pud)
+{
+       note_page(pt_st, addr, 2, pud_val(pud));
+}
+
+static void note_page_p4d(struct ptdump_state *pt_st, unsigned long addr, p4d_t p4d)
+{
+       note_page(pt_st, addr, 1, p4d_val(p4d));
+}
+
+static void note_page_pgd(struct ptdump_state *pt_st, unsigned long addr, pgd_t pgd)
+{
+       note_page(pt_st, addr, 0, pgd_val(pgd));
+}
+
+static void note_page_flush(struct ptdump_state *pt_st)
+{
+       pte_t pte_zero = {0};
+
+       note_page(pt_st, 0, -1, pte_val(pte_zero));
+}
+
 bool ptdump_check_wx(void)
 {
        struct pg_state st = {
                .ptdump = {
-                       .note_page = note_page,
+                       .note_page_pte = note_page_pte,
+                       .note_page_pmd = note_page_pmd,
+                       .note_page_pud = note_page_pud,
+                       .note_page_p4d = note_page_p4d,
+                       .note_page_pgd = note_page_pgd,
+                       .note_page_flush = note_page_flush,
                        .range = (struct ptdump_range[]) {
                                {.start = 0, .end = max_addr},
                                {.start = 0, .end = 0},
 {
        struct pg_state st = {
                .ptdump = {
-                       .note_page = note_page,
+                       .note_page_pte = note_page_pte,
+                       .note_page_pmd = note_page_pmd,
+                       .note_page_pud = note_page_pud,
+                       .note_page_p4d = note_page_p4d,
+                       .note_page_pgd = note_page_pgd,
+                       .note_page_flush = note_page_flush,
                        .range = (struct ptdump_range[]) {
                                {.start = 0, .end = max_addr},
                                {.start = 0, .end = 0},
 
        }
 }
 
+static void note_page_pte(struct ptdump_state *pt_st, unsigned long addr, pte_t pte)
+{
+       note_page(pt_st, addr, 4, pte_val(pte));
+}
+
+static void note_page_pmd(struct ptdump_state *pt_st, unsigned long addr, pmd_t pmd)
+{
+       note_page(pt_st, addr, 3, pmd_val(pmd));
+}
+
+static void note_page_pud(struct ptdump_state *pt_st, unsigned long addr, pud_t pud)
+{
+       note_page(pt_st, addr, 2, pud_val(pud));
+}
+
+static void note_page_p4d(struct ptdump_state *pt_st, unsigned long addr, p4d_t p4d)
+{
+       note_page(pt_st, addr, 1, p4d_val(p4d));
+}
+
+static void note_page_pgd(struct ptdump_state *pt_st, unsigned long addr, pgd_t pgd)
+{
+       note_page(pt_st, addr, 0, pgd_val(pgd));
+}
+
+static void note_page_flush(struct ptdump_state *pt_st)
+{
+       pte_t pte_zero = {0};
+
+       note_page(pt_st, 0, -1, pte_val(pte_zero));
+}
+
 bool ptdump_walk_pgd_level_core(struct seq_file *m,
                                struct mm_struct *mm, pgd_t *pgd,
                                bool checkwx, bool dmesg)
 
        struct pg_state st = {
                .ptdump = {
-                       .note_page      = note_page,
+                       .note_page_pte = note_page_pte,
+                       .note_page_pmd = note_page_pmd,
+                       .note_page_pud = note_page_pud,
+                       .note_page_p4d = note_page_p4d,
+                       .note_page_pgd = note_page_pgd,
+                       .note_page_flush = note_page_flush,
                        .effective_prot = effective_prot,
                        .range          = ptdump_ranges
                },
 
 };
 
 struct ptdump_state {
-       /* level is 0:PGD to 4:PTE, or -1 if unknown */
-       void (*note_page)(struct ptdump_state *st, unsigned long addr,
-                         int level, u64 val);
+       void (*note_page_pte)(struct ptdump_state *st, unsigned long addr, pte_t pte);
+       void (*note_page_pmd)(struct ptdump_state *st, unsigned long addr, pmd_t pmd);
+       void (*note_page_pud)(struct ptdump_state *st, unsigned long addr, pud_t pud);
+       void (*note_page_p4d)(struct ptdump_state *st, unsigned long addr, p4d_t p4d);
+       void (*note_page_pgd)(struct ptdump_state *st, unsigned long addr, pgd_t pgd);
+       void (*note_page_flush)(struct ptdump_state *st);
        void (*effective_prot)(struct ptdump_state *st, int level, u64 val);
        const struct ptdump_range *range;
 };
 
 {
        struct ptdump_state *st = walk->private;
 
-       st->note_page(st, addr, 4, pte_val(kasan_early_shadow_pte[0]));
+       st->note_page_pte(st, addr, kasan_early_shadow_pte[0]);
 
        walk->action = ACTION_CONTINUE;
 
                st->effective_prot(st, 0, pgd_val(val));
 
        if (pgd_leaf(val)) {
-               st->note_page(st, addr, 0, pgd_val(val));
+               st->note_page_pgd(st, addr, val);
                walk->action = ACTION_CONTINUE;
        }
 
                st->effective_prot(st, 1, p4d_val(val));
 
        if (p4d_leaf(val)) {
-               st->note_page(st, addr, 1, p4d_val(val));
+               st->note_page_p4d(st, addr, val);
                walk->action = ACTION_CONTINUE;
        }
 
                st->effective_prot(st, 2, pud_val(val));
 
        if (pud_leaf(val)) {
-               st->note_page(st, addr, 2, pud_val(val));
+               st->note_page_pud(st, addr, val);
                walk->action = ACTION_CONTINUE;
        }
 
        if (st->effective_prot)
                st->effective_prot(st, 3, pmd_val(val));
        if (pmd_leaf(val)) {
-               st->note_page(st, addr, 3, pmd_val(val));
+               st->note_page_pmd(st, addr, val);
                walk->action = ACTION_CONTINUE;
        }
 
        if (st->effective_prot)
                st->effective_prot(st, 4, pte_val(val));
 
-       st->note_page(st, addr, 4, pte_val(val));
+       st->note_page_pte(st, addr, val);
 
        return 0;
 }
                       int depth, struct mm_walk *walk)
 {
        struct ptdump_state *st = walk->private;
-
-       st->note_page(st, addr, depth, 0);
-
+       pte_t pte_zero = {0};
+       pmd_t pmd_zero = {0};
+       pud_t pud_zero = {0};
+       p4d_t p4d_zero = {0};
+       pgd_t pgd_zero = {0};
+
+       switch (depth) {
+       case 4:
+               st->note_page_pte(st, addr, pte_zero);
+               break;
+       case 3:
+               st->note_page_pmd(st, addr, pmd_zero);
+               break;
+       case 2:
+               st->note_page_pud(st, addr, pud_zero);
+               break;
+       case 1:
+               st->note_page_p4d(st, addr, p4d_zero);
+               break;
+       case 0:
+               st->note_page_pgd(st, addr, pgd_zero);
+               break;
+       default:
+               break;
+       }
        return 0;
 }
 
        mmap_write_unlock(mm);
 
        /* Flush out the last page */
-       st->note_page(st, 0, -1, 0);
+       st->note_page_flush(st);
 }
 
 static int check_wx_show(struct seq_file *m, void *v)