From: "Mike Rapoport (Microsoft)" <[email protected]>

hugetlb-mmap test fails 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.

Reviewed-by: Luiz Capitulino <[email protected]>
Tested-by: Luiz Capitulino <[email protected]>
Tested-by: Sarthak Sharma <[email protected]>
Signed-off-by: Mike Rapoport (Microsoft) <[email protected]>
---
 tools/testing/selftests/mm/hugetlb-mmap.c | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/tools/testing/selftests/mm/hugetlb-mmap.c 
b/tools/testing/selftests/mm/hugetlb-mmap.c
index 395be5d0dc1a..0f2aad1b7dbd 100644
--- a/tools/testing/selftests/mm/hugetlb-mmap.c
+++ b/tools/testing/selftests/mm/hugetlb-mmap.c
@@ -105,28 +105,35 @@ int main(int argc, char **argv)
 {
        size_t hugepage_size;
        size_t length = LENGTH;
-       int shift = 0;
+       int shift = 0, nr;
 
        ksft_print_header();
-       ksft_set_plan(2);
 
        if (argc > 1)
                length = atol(argv[1]) << 20;
        if (argc > 2)
                shift = atoi(argv[2]);
 
+       hugetlb_save_settings();
        if (shift) {
                hugepage_size = (1UL << shift);
                ksft_print_msg("%lu kB hugepages\n", 1UL << (shift - 10));
        } else {
                hugepage_size = default_huge_page_size();
+               if (!hugepage_size)
+                       ksft_exit_skip("Could not detect default hugetlb page 
size.");
                ksft_print_msg("Default size hugepages (%lu kB)\n", 
hugepage_size >> 10);
        }
 
        /* munmap will fail if the length is not page aligned */
-       if (hugepage_size > length)
-               length = hugepage_size;
+       length = (length + hugepage_size - 1) & ~(hugepage_size - 1);
+       nr = length / hugepage_size;
+
+       hugetlb_set_nr_pages(hugepage_size, nr);
+       if (hugetlb_free_pages(hugepage_size) < nr)
+               ksft_exit_skip("Not enough %lu Kb pages\n", hugepage_size >> 
10);
 
+       ksft_set_plan(2);
        ksft_print_msg("Mapping %lu Mbytes\n", (unsigned long)length >> 20);
 
        test_anon_mmap(length, shift);
-- 
2.53.0


Reply via email to