Conform the layout, informational and status messages to TAP. No
functional change is intended other than the layout of output messages.

Signed-off-by: Muhammad Usama Anjum <[email protected]>
---
Changes since v3:
- Rebase to next-20240125
- Update newly updated code
---
 tools/testing/selftests/mm/hugepage-vmemmap.c | 39 ++++++++-----------
 1 file changed, 16 insertions(+), 23 deletions(-)

diff --git a/tools/testing/selftests/mm/hugepage-vmemmap.c 
b/tools/testing/selftests/mm/hugepage-vmemmap.c
index 894d28c3dd478..9080e34bb297e 100644
--- a/tools/testing/selftests/mm/hugepage-vmemmap.c
+++ b/tools/testing/selftests/mm/hugepage-vmemmap.c
@@ -11,6 +11,7 @@
 #include <sys/mman.h>
 #include <fcntl.h>
 #include "vm_util.h"
+#include "../kselftest.h"
 
 #define PAGE_COMPOUND_HEAD     (1UL << 15)
 #define PAGE_COMPOUND_TAIL     (1UL << 16)
@@ -77,7 +78,7 @@ static int check_page_flags(unsigned long pfn)
        read(fd, &pageflags, sizeof(pageflags));
        if ((pageflags & HEAD_PAGE_FLAGS) != HEAD_PAGE_FLAGS) {
                close(fd);
-               printf("Head page flags (%lx) is invalid\n", pageflags);
+               ksft_print_msg("Head page flags (%lx) is invalid\n", pageflags);
                return -1;
        }
 
@@ -91,7 +92,7 @@ static int check_page_flags(unsigned long pfn)
                if ((pageflags & TAIL_PAGE_FLAGS) != TAIL_PAGE_FLAGS ||
                    (pageflags & HEAD_PAGE_FLAGS) == HEAD_PAGE_FLAGS) {
                        close(fd);
-                       printf("Tail page flags (%lx) is invalid\n", pageflags);
+                       ksft_print_msg("Tail page flags (%lx) is invalid\n", 
pageflags);
                        return -1;
                }
        }
@@ -106,18 +107,17 @@ int main(int argc, char **argv)
        void *addr;
        unsigned long pfn;
 
+       ksft_print_header();
+       ksft_set_plan(1);
+
        pagesize  = psize();
        maplength = default_huge_page_size();
-       if (!maplength) {
-               printf("Unable to determine huge page size\n");
-               exit(1);
-       }
+       if (!maplength)
+               ksft_exit_fail_msg("Unable to determine huge page size\n");
 
        addr = mmap(MAP_ADDR, maplength, PROT_READ | PROT_WRITE, MAP_FLAGS, -1, 
0);
-       if (addr == MAP_FAILED) {
-               perror("mmap");
-               exit(1);
-       }
+       if (addr == MAP_FAILED)
+               ksft_exit_fail_msg("mmap: %s\n", strerror(errno));
 
        /* Trigger allocation of HugeTLB page. */
        write_bytes(addr, maplength);
@@ -125,23 +125,16 @@ int main(int argc, char **argv)
        pfn = virt_to_pfn(addr);
        if (pfn == -1UL) {
                munmap(addr, maplength);
-               perror("virt_to_pfn");
-               exit(1);
+               ksft_exit_fail_msg("virt_to_pfn: %s\n", strerror(errno));
        }
 
-       printf("Returned address is %p whose pfn is %lx\n", addr, pfn);
+       ksft_print_msg("Returned address is %p whose pfn is %lx\n", addr, pfn);
 
-       if (check_page_flags(pfn) < 0) {
-               munmap(addr, maplength);
-               perror("check_page_flags");
-               exit(1);
-       }
+       ksft_test_result(!check_page_flags(pfn), "check_page_flags\n");
 
        /* munmap() length of MAP_HUGETLB memory must be hugepage aligned */
-       if (munmap(addr, maplength)) {
-               perror("munmap");
-               exit(1);
-       }
+       if (munmap(addr, maplength))
+               ksft_exit_fail_msg("munmap: %s\n", strerror(errno));
 
-       return 0;
+       ksft_finished();
 }
-- 
2.42.0


Reply via email to