selftests/mm: pagemap_ioctl: add setup of HugeTLB pages

pagemap-ioctl skips HugeTLB tests if there are no free huge pages prepared
by a wrapper script.

Add setup of HugeTLB pages to the test and make sure that the original
settings are restored on the test exit.

Link: https://lore.kernel.org/20260511162840.375890-47-rppt@kernel.org
Signed-off-by: Mike Rapoport (Microsoft) <rppt@kernel.org>
Tested-by: Luiz Capitulino <luizcap@redhat.com>
Tested-by: Sarthak Sharma <sarthak.sharma@arm.com>
Cc: Baolin Wang <baolin.wang@linux.alibaba.com>
Cc: Barry Song <baohua@kernel.org>
Cc: David Hildenbrand <david@kernel.org>
Cc: Dev Jain <dev.jain@arm.com>
Cc: Donet Tom <donettom@linux.ibm.com>
Cc: Jason Gunthorpe <jgg@ziepe.ca>
Cc: John Hubbard <jhubbard@nvidia.com>
Cc: Lance Yang <lance.yang@linux.dev>
Cc: Leon Romanovsky <leon@kernel.org>
Cc: Liam Howlett <liam@infradead.org>
Cc: Li Wang <li.wang@linux.dev>
Cc: Lorenzo Stoakes <ljs@kernel.org>
Cc: Mark Brown <broonie@kernel.org>
Cc: Michal Hocko <mhocko@suse.com>
Cc: Nico Pache <npache@redhat.com>
Cc: Peter Xu <peterx@redhat.com>
Cc: Ryan Roberts <ryan.roberts@arm.com>
Cc: Shuah Khan <shuah@kernel.org>
Cc: Suren Baghdasaryan <surenb@google.com>
Cc: Vlastimil Babka <vbabka@kernel.org>
Cc: Zi Yan <ziy@nvidia.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>

diff --git a/tools/testing/selftests/mm/pagemap_ioctl.c b/tools/testing/selftests/mm/pagemap_ioctl.c
index 7f9428d6062cab21194bcae114bb04fc8a857c31..762306177ad813eeabf0bc6d71faf76d34cf464d 100644 (file)
--- a/tools/testing/selftests/mm/pagemap_ioctl.c
+++ b/tools/testing/selftests/mm/pagemap_ioctl.c
@@ -7,8 +7,6 @@
 #include <sys/mman.h>
 #include <errno.h>
 #include <malloc.h>
-#include "vm_util.h"
-#include "kselftest.h"
 #include <linux/types.h>
 #include <linux/memfd.h>
 #include <linux/userfaultfd.h>
@@ -23,6 +21,10 @@
 #include <sys/ipc.h>
 #include <sys/shm.h>
 
+#include "vm_util.h"
+#include "kselftest.h"
+#include "hugepage_settings.h"
+
 #define PAGEMAP_BITS_ALL               (PAGE_IS_WPALLOWED | PAGE_IS_WRITTEN |  \
                                         PAGE_IS_FILE | PAGE_IS_PRESENT |       \
                                         PAGE_IS_SWAPPED | PAGE_IS_PFNZERO |    \
@@ -1554,6 +1556,9 @@ int main(int __attribute__((unused)) argc, char *argv[])
        if (init_uffd())
                ksft_exit_skip("Failed to initialize userfaultfd\n");
 
+       if (!hugetlb_setup_default(4))
+               ksft_print_msg("HugeTLB test will be skipped\n");
+
        ksft_set_plan(117);
 
        page_size = getpagesize();
@@ -1605,7 +1610,7 @@ int main(int __attribute__((unused)) argc, char *argv[])
        }
 
        /* 5. SHM Hugetlb page testing */
-       mem_size = 2*1024*1024;
+       mem_size = default_huge_page_size();
        mem = gethugetlb_mem(mem_size, &shmid);
        if (mem) {
                wp_init(mem, mem_size);
@@ -1633,7 +1638,7 @@ int main(int __attribute__((unused)) argc, char *argv[])
        }
 
        /* 7. File Hugetlb testing */
-       mem_size = 2*1024*1024;
+       mem_size = default_huge_page_size();
        fd = memfd_create("uffd-test", MFD_HUGETLB | MFD_NOEXEC_SEAL);
        if (fd < 0)
                ksft_exit_fail_msg("uffd-test creation failed %d %s\n", errno, strerror(errno));