Subtests 6 and 7 of the udmabuf test require free huge pages. If there
are not enough huge pages available, the entire test fails on an mmap:
        ok 4 drivers/dma-buf/udmabuf: [PASS,test-4]
        ok 5 drivers/dma-buf/udmabuf: [PASS,test-5]
        drivers/dma-buf/udmabuf: ubuf_fd mmap fail

Instead of failing, check if there are enough huge pages available
beforehand and potentially skip the subtests 6 and 7.

Signed-off-by: Chris Gellermann <[email protected]>
---
 .../selftests/drivers/dma-buf/udmabuf.c       | 36 +++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/tools/testing/selftests/drivers/dma-buf/udmabuf.c 
b/tools/testing/selftests/drivers/dma-buf/udmabuf.c
index 2def04ed3451..daa2fc6c2c27 100644
--- a/tools/testing/selftests/drivers/dma-buf/udmabuf.c
+++ b/tools/testing/selftests/drivers/dma-buf/udmabuf.c
@@ -25,6 +25,34 @@
 
 static unsigned int page_size;
 
+static bool enough_free_huge_pages(off_t size)
+{
+       unsigned long free_pages = 0, hpage_size_kb = 0;
+       bool found_free = false, found_size = false;
+       char line[256];
+       FILE *f;
+
+       f = fopen("/proc/meminfo", "r");
+       if (!f)
+               return false;
+
+       while (fgets(line, sizeof(line), f)) {
+               if (sscanf(line, "HugePages_Free: %lu", &free_pages) == 1)
+                       found_free = true;
+               if (sscanf(line, "Hugepagesize: %lu kB", &hpage_size_kb) == 1)
+                       found_size = true;
+               if (found_free && found_size)
+                       break;
+       }
+       fclose(f);
+
+       if (!found_free || !found_size)
+               return false;
+
+       return ((unsigned long long)free_pages * hpage_size_kb * 1024ULL) >=
+              (unsigned long long)size;
+}
+
 static int create_memfd_with_seals(off_t size, bool hpage)
 {
        int memfd, ret;
@@ -236,6 +264,13 @@ int main(int argc, char *argv[])
        /* should work (migration of 2MB size huge pages)*/
        page_size = getpagesize() * 512; /* 2 MB */
        size = MEMFD_SIZE * page_size;
+       if (!enough_free_huge_pages(size)) {
+               ksft_test_result_skip("%s: [SKIP,test-6] not enough huge 
pages\n",
+                                     TEST_PREFIX);
+               ksft_test_result_skip("%s: [SKIP,test-7] not enough huge 
pages\n",
+                                     TEST_PREFIX);
+               goto out;
+       }
        memfd = create_memfd_with_seals(size, true);
        addr1 = mmap_fd(memfd, size);
        write_to_memfd(addr1, size, 'a');
@@ -268,6 +303,7 @@ int main(int argc, char *argv[])
 
        close(buf);
        close(memfd);
+out:
        close(devfd);
 
        ksft_print_msg("%s: ok\n", TEST_PREFIX);
-- 
2.47.3


Reply via email to