of TEXT_OFFSET and platforms must not require a specific
          value.
 
+config DEBUG_WX
+       bool "Warn on W+X mappings at boot"
+       select ARM64_PTDUMP_CORE
+       ---help---
+         Generate a warning if any W+X mappings are found at boot.
+
+         This is useful for discovering cases where the kernel is leaving
+         W+X mappings after applying NX, as such mappings are a security risk.
+         This check also includes UXN, which should be set on all kernel
+         mappings.
+
+         Look for a message in dmesg output like this:
+
+           arm64/mm: Checked W+X mappings: passed, no W+X pages found.
+
+         or like this, if the check failed:
+
+           arm64/mm: Checked W+X mappings: FAILED, <N> W+X pages found.
+
+         Note that even if the check fails, your kernel is possibly
+         still fine, as W+X mappings are not a security hole in
+         themselves, what they do is that they make the exploitation
+         of other unfixed kernel bugs easier.
+
+         There is no runtime or memory usage effect of this option
+         once the kernel has booted up - it's a one time check.
+
+         If in doubt, say "Y".
+
 config DEBUG_SET_MODULE_RONX
        bool "Set loadable kernel module data as NX and text as RO"
        depends on MODULES
 
        unsigned long start_address;
        unsigned level;
        u64 current_prot;
+       bool check_wx;
+       unsigned long wx_pages;
+       unsigned long uxn_pages;
 };
 
 struct prot_bits {
        }
 }
 
+static void note_prot_uxn(struct pg_state *st, unsigned long addr)
+{
+       if (!st->check_wx)
+               return;
+
+       if ((st->current_prot & PTE_UXN) == PTE_UXN)
+               return;
+
+       WARN_ONCE(1, "arm64/mm: Found non-UXN mapping at address %p/%pS\n",
+                 (void *)st->start_address, (void *)st->start_address);
+
+       st->uxn_pages += (addr - st->start_address) / PAGE_SIZE;
+}
+
+static void note_prot_wx(struct pg_state *st, unsigned long addr)
+{
+       if (!st->check_wx)
+               return;
+       if ((st->current_prot & PTE_RDONLY) == PTE_RDONLY)
+               return;
+       if ((st->current_prot & PTE_PXN) == PTE_PXN)
+               return;
+
+       WARN_ONCE(1, "arm64/mm: Found insecure W+X mapping at address %p/%pS\n",
+                 (void *)st->start_address, (void *)st->start_address);
+
+       st->wx_pages += (addr - st->start_address) / PAGE_SIZE;
+}
+
 static void note_page(struct pg_state *st, unsigned long addr, unsigned level,
                                u64 val)
 {
                unsigned long delta;
 
                if (st->current_prot) {
+                       note_prot_uxn(st, addr);
+                       note_prot_wx(st, addr);
                        pt_dump_seq_printf(st->seq, "0x%016lx-0x%016lx   ",
                                   st->start_address, addr);
 
        .base_addr      = VA_START,
 };
 
+void ptdump_check_wx(void)
+{
+       struct pg_state st = {
+               .seq = NULL,
+               .marker = (struct addr_marker[]) {
+                       { 0, NULL},
+                       { -1, NULL},
+               },
+               .check_wx = true,
+       };
+
+       walk_pgd(&st, &init_mm, 0);
+       note_page(&st, 0, 0, 0);
+       if (st.wx_pages || st.uxn_pages)
+               pr_warn("Checked W+X mappings: FAILED, %lu W+X pages found, %lu non-UXN pages found\n",
+                       st.wx_pages, st.uxn_pages);
+       else
+               pr_info("Checked W+X mappings: passed, no W+X pages found\n");
+}
+
 static int ptdump_init(void)
 {
        ptdump_initialize();