[PATCH v2 2/2] printf: break kunit into test cases

7 Feb 2025

Use `suite_{init,exit}` and move all tests into `printf_test_cases`.
This gives us nicer output in the event of a failure.
Combine `plain_format` and `plain_hash` into `hash_pointer` since
they're testing the same scenario.
Signed-off-by: Tamir Duberstein tamird@gmail.com
---
 lib/printf_kunit.c | 286 +++++++++++++++++++----------------------------------
 1 file changed, 103 insertions(+), 183 deletions(-)

diff --git a/lib/printf_kunit.c b/lib/printf_kunit.c
index fe6f4df92dd5..b4b6f9943025 100644
--- a/lib/printf_kunit.c
+++ b/lib/printf_kunit.c
@@ -38,13 +38,11 @@
 static char *test_buffer;
 static char *alloced_buffer;
-static struct kunit *kunittest;
-
 #define tc_fail(fmt, ...) \
    KUNIT_FAIL(kunittest, fmt, ##__VA_ARGS__)
-static void __printf(4, 0)
-do_test(int bufsize, const char *expect, int elen,
+static void __printf(5, 0)
+do_test(struct kunit *kunittest, int bufsize, const char *expect, int elen,
    const char *fmt, va_list ap)
 {
    va_list aq;
@@ -99,8 +97,8 @@ do_test(int bufsize, const char *expect, int elen,
    }
 }
-static void __printf(3, 4)
-__test(const char *expect, int elen, const char *fmt, ...)
+static void __printf(4, 5)
+__test(struct kunit *kunittest, const char *expect, int elen, const char *fmt, ...)
 {
    va_list ap;
    int rand;
@@ -120,11 +118,11 @@ __test(const char *expect, int elen, const char *fmt, ...)
     * enough and 0), and then we also test that kvasprintf would
     * be able to print it as expected.
     */
-	do_test(BUF_SIZE, expect, elen, fmt, ap);
+	do_test(kunittest, BUF_SIZE, expect, elen, fmt, ap);
    rand = get_random_u32_inclusive(1, elen + 1);
    /* Since elen < BUF_SIZE, we have 1 <= rand <= BUF_SIZE. */
-	do_test(rand, expect, elen, fmt, ap);
-	do_test(0, expect, elen, fmt, ap);
+	do_test(kunittest, rand, expect, elen, fmt, ap);
+	do_test(kunittest, 0, expect, elen, fmt, ap);
p = kvasprintf(GFP_KERNEL, fmt, ap);
    if (p) {
@@ -138,10 +136,10 @@ __test(const char *expect, int elen, const char *fmt, ...)
 }
#define test(expect, fmt, ...)					\
-	__test(expect, strlen(expect), fmt, ##__VA_ARGS__)
+	__test(kunittest, expect, strlen(expect), fmt, ##__VA_ARGS__)
static void
-test_basic(void)
+test_basic(struct kunit *kunittest)
 {
    /* Work around annoying "warning: zero-length gnu_printf format string". */
    char nul = '\0';
@@ -149,11 +147,11 @@ test_basic(void)
    test("", &nul);
    test("100%", "100%%");
    test("xxx%yyy", "xxx%cyyy", '%');
-	__test("xxx\0yyy", 7, "xxx%cyyy", '\0');
+	__test(kunittest, "xxx\0yyy", 7, "xxx%cyyy", '\0');
 }
static void
-test_number(void)
+test_number(struct kunit *kunittest)
 {
    test("0x1234abcd  ", "%#-12x", 0x1234abcd);
    test("  0x1234abcd", "%#12x", 0x1234abcd);
@@ -175,7 +173,7 @@ test_number(void)
 }
static void
-test_string(void)
+test_string(struct kunit *kunittest)
 {
    test("", "%s%.0s", "", "123");
    test("ABCD|abc|123", "%s|%.3s|%.*s", "ABCD", "abcdef", 3, "123456");
@@ -212,29 +210,6 @@ test_string(void)
 #define ZEROS "00000000"	/* hex 32 zero bits */
 #define ONES "ffffffff"		/* hex 32 one bits */
-static int
-plain_format(void)
-{
-	char buf[PLAIN_BUF_SIZE];
-	int nchars;
-
-	nchars = snprintf(buf, PLAIN_BUF_SIZE, "%p", PTR);
-
-	if (nchars != PTR_WIDTH)
-		return -1;
-
-	if (strncmp(buf, PTR_VAL_NO_CRNG, PTR_WIDTH) == 0) {
-		pr_warn("crng possibly not yet initialized. plain 'p' buffer contains "%s"",
-			PTR_VAL_NO_CRNG);
-		return 0;
-	}
-
-	if (strncmp(buf, ZEROS, strlen(ZEROS)) != 0)
-		return -1;
-
-	return 0;
-}
-
 #else
#define PTR_WIDTH 8
@@ -244,89 +219,44 @@ plain_format(void)
 #define ZEROS ""
 #define ONES ""
-static int
-plain_format(void)
-{
-	/* Format is implicitly tested for 32 bit machines by plain_hash() */
-	return 0;
-}
-
 #endif	/* BITS_PER_LONG == 64 */
-static int
-plain_hash_to_buffer(const void *p, char *buf, size_t len)
+static void
+plain_hash_to_buffer(struct kunit *kunittest, const void *p, char *buf, size_t len)
 {
-	int nchars;
-
-	nchars = snprintf(buf, len, "%p", p);
-
-	if (nchars != PTR_WIDTH)
-		return -1;
+	KUNIT_ASSERT_EQ(kunittest, snprintf(buf, len, "%p", p), PTR_WIDTH);
if (strncmp(buf, PTR_VAL_NO_CRNG, PTR_WIDTH) == 0) {
    	pr_warn("crng possibly not yet initialized. plain 'p' buffer contains "%s"",
    		PTR_VAL_NO_CRNG);
-		return 0;
    }
-
-	return 0;
-}
-
-static int
-plain_hash(void)
-{
-	char buf[PLAIN_BUF_SIZE];
-	int ret;
-
-	ret = plain_hash_to_buffer(PTR, buf, PLAIN_BUF_SIZE);
-	if (ret)
-		return ret;
-
-	if (strncmp(buf, PTR_STR, PTR_WIDTH) == 0)
-		return -1;
-
-	return 0;
 }
-/*
- * We can't use test() to test %p because we don't know what output to expect
- * after an address is hashed.
- */
 static void
-plain(void)
+hash_pointer(struct kunit *kunittest)
 {
-	int err;
+	if (no_hash_pointers)
+		kunit_skip(kunittest, "hash pointers disabled");
-	if (no_hash_pointers) {
-		pr_warn("skipping plain 'p' tests");
-		return;
-	}
+	char buf[PLAIN_BUF_SIZE];
-	err = plain_hash();
-	if (err) {
-		tc_fail("plain 'p' does not appear to be hashed\n");
-		return;
-	}
+	plain_hash_to_buffer(kunittest, PTR, buf, PLAIN_BUF_SIZE);
-	err = plain_format();
-	if (err) {
-		tc_fail("hashing plain 'p' has unexpected format\n");
-	}
+	/*
+	 * We can't use test() to test %p because we don't know what output to expect
+	 * after an address is hashed.
+	 */
+
+	KUNIT_EXPECT_MEMEQ(kunittest, buf, ZEROS, strlen(ZEROS));
+	KUNIT_EXPECT_MEMNEQ(kunittest, buf+strlen(ZEROS), PTR_STR, PTR_WIDTH);
 }
static void
-test_hashed(const char *fmt, const void *p)
+test_hashed(struct kunit *kunittest, const char *fmt, const void *p)
 {
    char buf[PLAIN_BUF_SIZE];
-	int ret;
-	/*
-	 * No need to increase failed test counter since this is assumed
-	 * to be called after plain().
-	 */
-	ret = plain_hash_to_buffer(p, buf, PLAIN_BUF_SIZE);
-	if (ret)
-		return;
+	plain_hash_to_buffer(kunittest, p, buf, PLAIN_BUF_SIZE);
test(buf, fmt, p);
 }
@@ -335,7 +265,7 @@ test_hashed(const char *fmt, const void *p)
  * NULL pointers aren't hashed.
  */
 static void
-null_pointer(void)
+null_pointer(struct kunit *kunittest)
 {
    test(ZEROS "00000000", "%p", NULL);
    test(ZEROS "00000000", "%px", NULL);
@@ -346,7 +276,7 @@ null_pointer(void)
  * Error pointers aren't hashed.
  */
 static void
-error_pointer(void)
+error_pointer(struct kunit *kunittest)
 {
    test(ONES "fffffff5", "%p", ERR_PTR(-11));
    test(ONES "fffffff5", "%px", ERR_PTR(-11));
@@ -356,26 +286,26 @@ error_pointer(void)
 #define PTR_INVALID ((void *)0x000000ab)
static void
-invalid_pointer(void)
+invalid_pointer(struct kunit *kunittest)
 {
-	test_hashed("%p", PTR_INVALID);
+	test_hashed(kunittest, "%p", PTR_INVALID);
    test(ZEROS "000000ab", "%px", PTR_INVALID);
    test("(efault)", "%pE", PTR_INVALID);
 }
static void
-symbol_ptr(void)
+symbol_ptr(struct kunit *kunittest)
 {
 }
static void
-kernel_ptr(void)
+kernel_ptr(struct kunit *kunittest)
 {
    /* We can't test this without access to kptr_restrict. */
 }
static void
-struct_resource(void)
+struct_resource(struct kunit *kunittest)
 {
    struct resource test_resource = {
    	.start = 0xc0ffee00,
@@ -424,7 +354,7 @@ struct_resource(void)
 }
static void
-struct_range(void)
+struct_range(struct kunit *kunittest)
 {
    struct range test_range = DEFINE_RANGE(0xc0ffee00ba5eba11,
    				       0xc0ffee00ba5eba11);
@@ -440,17 +370,17 @@ struct_range(void)
 }
static void
-addr(void)
+addr(struct kunit *kunittest)
 {
 }
static void
-escaped_str(void)
+escaped_str(struct kunit *kunittest)
 {
 }
static void
-hex_string(void)
+hex_string(struct kunit *kunittest)
 {
    const char buf[3] = {0xc0, 0xff, 0xee};
@@ -461,7 +391,7 @@ hex_string(void)
 }
static void
-mac(void)
+mac(struct kunit *kunittest)
 {
    const u8 addr[6] = {0x2d, 0x48, 0xd6, 0xfc, 0x7a, 0x05};
@@ -473,7 +403,7 @@ mac(void)
 }
static void
-ip4(void)
+ip4(struct kunit *kunittest)
 {
    struct sockaddr_in sa;
@@ -488,19 +418,12 @@ ip4(void)
 }
static void
-ip6(void)
+ip6(struct kunit *kunittest)
 {
 }
static void
-ip(void)
-{
-	ip4();
-	ip6();
-}
-
-static void
-uuid(void)
+uuid(struct kunit *kunittest)
 {
    const char uuid[16] = {0x0, 0x1, 0x2, 0x3, 0x4, 0x5, 0x6, 0x7,
    		       0x8, 0x9, 0xa, 0xb, 0xc, 0xd, 0xe, 0xf};
@@ -527,7 +450,7 @@ static struct dentry test_dentry[4] = {
 };
static void
-dentry(void)
+dentry(struct kunit *kunittest)
 {
    test("foo", "%pd", &test_dentry[0]);
    test("foo", "%pd2", &test_dentry[0]);
@@ -548,12 +471,12 @@ dentry(void)
 }
static void
-struct_va_format(void)
+struct_va_format(struct kunit *kunittest)
 {
 }
static void
-time_and_date(void)
+time_and_date(struct kunit *kunittest)
 {
    /* 1543210543 */
    const struct rtc_time tm = {
@@ -587,12 +510,12 @@ time_and_date(void)
 }
static void
-struct_clk(void)
+struct_clk(struct kunit *kunittest)
 {
 }
static void
-large_bitmap(void)
+large_bitmap(struct kunit *kunittest)
 {
    const int nbits = 1 << 16;
    unsigned long *bits = bitmap_zalloc(nbits, GFP_KERNEL);
@@ -606,7 +529,7 @@ large_bitmap(void)
 }
static void
-bitmap(void)
+bitmap(struct kunit *kunittest)
 {
    DECLARE_BITMAP(bits, 20);
    const int primes[] = {2,3,5,7,11,13,17,19};
@@ -625,11 +548,11 @@ bitmap(void)
    test("fffff|fffff", "%20pb|%*pb", bits, 20, bits);
    test("0-19|0-19", "%20pbl|%*pbl", bits, 20, bits);
-	large_bitmap();
+	large_bitmap(kunittest);
 }
static void
-netdev_features(void)
+netdev_features(struct kunit *kunittest)
 {
 }
@@ -655,8 +578,8 @@ static const struct page_flags_test pft[] = {
 };
static void
-page_flags_test(int section, int node, int zone, int last_cpupid,
-		int kasan_tag, unsigned long flags, const char *name,
+page_flags_test(struct kunit *kunittest, int section, int node, int zone,
+		int last_cpupid, int kasan_tag, unsigned long flags, const char *name,
    	char *cmp_buf)
 {
    unsigned long values[] = {section, node, zone, last_cpupid, kasan_tag};
@@ -693,7 +616,7 @@ page_flags_test(int section, int node, int zone, int last_cpupid,
 }
static void
-flags(void)
+flags(struct kunit *kunittest)
 {
    unsigned long flags;
    char *cmp_buffer;
@@ -704,14 +627,14 @@ flags(void)
    	return;
flags = 0;
-	page_flags_test(0, 0, 0, 0, 0, flags, "", cmp_buffer);
+	page_flags_test(kunittest, 0, 0, 0, 0, 0, flags, "", cmp_buffer);
flags = 1UL << NR_PAGEFLAGS;
-	page_flags_test(0, 0, 0, 0, 0, flags, "", cmp_buffer);
+	page_flags_test(kunittest, 0, 0, 0, 0, 0, flags, "", cmp_buffer);
flags |= 1UL << PG_uptodate | 1UL << PG_dirty | 1UL << PG_lru
    	| 1UL << PG_active | 1UL << PG_swapbacked;
-	page_flags_test(1, 1, 1, 0x1fffff, 1, flags,
+	page_flags_test(kunittest, 1, 1, 1, 0x1fffff, 1, flags,
    		"uptodate|dirty|lru|active|swapbacked",
    		cmp_buffer);
@@ -740,7 +663,7 @@ flags(void)
    kfree(cmp_buffer);
 }
-static void fwnode_pointer(void)
+static void fwnode_pointer(struct kunit *kunittest)
 {
    const struct software_node first = { .name = "first" };
    const struct software_node second = { .name = "second", .parent = &first };
@@ -754,8 +677,7 @@ static void fwnode_pointer(void)
rval = software_node_register_node_group(group);
    if (rval) {
-		pr_warn("cannot register softnodes; rval %d\n", rval);
-		return;
+		kunit_skip(kunittest, "cannot register softnodes; rval %d\n", rval);
    }
test(full_name_second, "%pfw", software_node_fwnode(&second));
@@ -767,7 +689,7 @@ static void fwnode_pointer(void)
    software_node_unregister_node_group(group);
 }
-static void fourcc_pointer(void)
+static void fourcc_pointer(struct kunit *kunittest)
 {
    struct {
    	u32 code;
@@ -785,13 +707,13 @@ static void fourcc_pointer(void)
 }
static void
-errptr(void)
+errptr(struct kunit *kunittest)
 {
    test("-1234", "%pe", ERR_PTR(-1234));
/* Check that %pe with a non-ERR_PTR gets treated as ordinary %p. */
    BUILD_BUG_ON(IS_ERR(PTR));
-	test_hashed("%pe", PTR);
+	test_hashed(kunittest, "%pe", PTR);
#ifdef CONFIG_SYMBOLIC_ERRNAME
    test("(-ENOTSOCK)", "(%pe)", ERR_PTR(-ENOTSOCK));
@@ -804,59 +726,57 @@ errptr(void)
 #endif
 }
-static void
-test_pointer(void)
-{
-	plain();
-	null_pointer();
-	error_pointer();
-	invalid_pointer();
-	symbol_ptr();
-	kernel_ptr();
-	struct_resource();
-	struct_range();
-	addr();
-	escaped_str();
-	hex_string();
-	mac();
-	ip();
-	uuid();
-	dentry();
-	struct_va_format();
-	time_and_date();
-	struct_clk();
-	bitmap();
-	netdev_features();
-	flags();
-	errptr();
-	fwnode_pointer();
-	fourcc_pointer();
-}
-
-static void printf_test(struct kunit *test)
+static struct kunit_case printf_test_cases[] = {
+	KUNIT_CASE(test_basic),
+	KUNIT_CASE(test_number),
+	KUNIT_CASE(test_string),
+	KUNIT_CASE(hash_pointer),
+	KUNIT_CASE(null_pointer),
+	KUNIT_CASE(error_pointer),
+	KUNIT_CASE(invalid_pointer),
+	KUNIT_CASE(symbol_ptr),
+	KUNIT_CASE(kernel_ptr),
+	KUNIT_CASE(struct_resource),
+	KUNIT_CASE(struct_range),
+	KUNIT_CASE(addr),
+	KUNIT_CASE(escaped_str),
+	KUNIT_CASE(hex_string),
+	KUNIT_CASE(mac),
+	KUNIT_CASE(ip4),
+	KUNIT_CASE(ip6),
+	KUNIT_CASE(uuid),
+	KUNIT_CASE(dentry),
+	KUNIT_CASE(struct_va_format),
+	KUNIT_CASE(time_and_date),
+	KUNIT_CASE(struct_clk),
+	KUNIT_CASE(bitmap),
+	KUNIT_CASE(netdev_features),
+	KUNIT_CASE(flags),
+	KUNIT_CASE(errptr),
+	KUNIT_CASE(fwnode_pointer),
+	KUNIT_CASE(fourcc_pointer),
+	{}
+};
+
+static int printf_suite_init(struct kunit_suite *suite)
 {
    alloced_buffer = kmalloc(BUF_SIZE + 2*PAD_SIZE, GFP_KERNEL);
    if (!alloced_buffer)
-		return;
+		return -1;
    test_buffer = alloced_buffer + PAD_SIZE;
+	return 0;
+}
-	kunittest = test;
-
-	test_basic();
-	test_number();
-	test_string();
-	test_pointer();
-
+static void printf_suite_exit(struct kunit_suite *suite)
+{
    kfree(alloced_buffer);
 }
-static struct kunit_case printf_test_cases[] = {
-	KUNIT_CASE(printf_test),
-	{}
-};
static struct kunit_suite printf_test_suite = {
    .name = "printf",
+	.suite_init = printf_suite_init,
+	.suite_exit = printf_suite_exit,
    .test_cases = printf_test_cases,
 };
-- 
2.48.1



    

2025

2024

2023

2022

2021

2020

2019

2018

2017

[PATCH v2 2/2] printf: break kunit into test cases