Merge tag 'imx-fixes-5.17' of git://git.kernel.org/pub/scm/linux/kernel/git/shawnguo...
[linux-2.6-microblaze.git] / lib / test_hash.c
index 0ee40b4..bb25fda 100644 (file)
  * and hash_64().
  */
 
-#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt "\n"
-
 #include <linux/compiler.h>
 #include <linux/types.h>
 #include <linux/module.h>
 #include <linux/hash.h>
 #include <linux/stringhash.h>
-#include <linux/printk.h>
+#include <kunit/test.h>
 
 /* 32-bit XORSHIFT generator.  Seed must not be zero. */
-static u32 __init __attribute_const__
+static u32 __attribute_const__
 xorshift(u32 seed)
 {
        seed ^= seed << 13;
@@ -34,7 +32,7 @@ xorshift(u32 seed)
 }
 
 /* Given a non-zero x, returns a non-zero byte. */
-static u8 __init __attribute_const__
+static u8 __attribute_const__
 mod255(u32 x)
 {
        x = (x & 0xffff) + (x >> 16);   /* 1 <= x <= 0x1fffe */
@@ -45,8 +43,7 @@ mod255(u32 x)
 }
 
 /* Fill the buffer with non-zero bytes. */
-static void __init
-fill_buf(char *buf, size_t len, u32 seed)
+static void fill_buf(char *buf, size_t len, u32 seed)
 {
        size_t i;
 
@@ -56,6 +53,50 @@ fill_buf(char *buf, size_t len, u32 seed)
        }
 }
 
+/* Holds most testing variables for the int test. */
+struct test_hash_params {
+        /* Pointer to integer to be hashed. */
+       unsigned long long *h64;
+        /* Low 32-bits of integer to be hashed. */
+       u32 h0;
+        /* Arch-specific hash result. */
+       u32 h1;
+        /* Generic hash result. */
+       u32 h2;
+        /* ORed hashes of given size (in bits). */
+       u32 (*hash_or)[33];
+};
+
+#ifdef HAVE_ARCH__HASH_32
+static void
+test_int__hash_32(struct kunit *test, struct test_hash_params *params)
+{
+       params->hash_or[1][0] |= params->h2 = __hash_32_generic(params->h0);
+#if HAVE_ARCH__HASH_32 == 1
+       KUNIT_EXPECT_EQ_MSG(test, params->h1, params->h2,
+                           "__hash_32(%#x) = %#x != __hash_32_generic() = %#x",
+                           params->h0, params->h1, params->h2);
+#endif
+}
+#endif
+
+#ifdef HAVE_ARCH_HASH_64
+static void
+test_int_hash_64(struct kunit *test, struct test_hash_params *params, u32 const *m, int *k)
+{
+       params->h2 = hash_64_generic(*params->h64, *k);
+#if HAVE_ARCH_HASH_64 == 1
+       KUNIT_EXPECT_EQ_MSG(test, params->h1, params->h2,
+                           "hash_64(%#llx, %d) = %#x != hash_64_generic() = %#x",
+                           *params->h64, *k, params->h1, params->h2);
+#else
+       KUNIT_EXPECT_LE_MSG(test, params->h1, params->h2,
+                           "hash_64_generic(%#llx, %d) = %#x > %#x",
+                           *params->h64, *k, params->h1, *m);
+#endif
+}
+#endif
+
 /*
  * Test the various integer hash functions.  h64 (or its low-order bits)
  * is the integer to hash.  hash_or accumulates the OR of the hash values,
@@ -65,23 +106,16 @@ fill_buf(char *buf, size_t len, u32 seed)
  * inline, the code being tested is actually in the module, and you can
  * recompile and re-test the module without rebooting.
  */
-static bool __init
-test_int_hash(unsigned long long h64, u32 hash_or[2][33])
+static void
+test_int_hash(struct kunit *test, unsigned long long h64, u32 hash_or[2][33])
 {
        int k;
-       u32 h0 = (u32)h64, h1, h2;
+       struct test_hash_params params = { &h64, (u32)h64, 0, 0, hash_or };
 
        /* Test __hash32 */
-       hash_or[0][0] |= h1 = __hash_32(h0);
+       hash_or[0][0] |= params.h1 = __hash_32(params.h0);
 #ifdef HAVE_ARCH__HASH_32
-       hash_or[1][0] |= h2 = __hash_32_generic(h0);
-#if HAVE_ARCH__HASH_32 == 1
-       if (h1 != h2) {
-               pr_err("__hash_32(%#x) = %#x != __hash_32_generic() = %#x",
-                       h0, h1, h2);
-               return false;
-       }
-#endif
+       test_int__hash_32(test, &params);
 #endif
 
        /* Test k = 1..32 bits */
@@ -89,63 +123,53 @@ test_int_hash(unsigned long long h64, u32 hash_or[2][33])
                u32 const m = ((u32)2 << (k-1)) - 1;    /* Low k bits set */
 
                /* Test hash_32 */
-               hash_or[0][k] |= h1 = hash_32(h0, k);
-               if (h1 > m) {
-                       pr_err("hash_32(%#x, %d) = %#x > %#x", h0, k, h1, m);
-                       return false;
-               }
-#ifdef HAVE_ARCH_HASH_32
-               h2 = hash_32_generic(h0, k);
-#if HAVE_ARCH_HASH_32 == 1
-               if (h1 != h2) {
-                       pr_err("hash_32(%#x, %d) = %#x != hash_32_generic() "
-                               " = %#x", h0, k, h1, h2);
-                       return false;
-               }
-#else
-               if (h2 > m) {
-                       pr_err("hash_32_generic(%#x, %d) = %#x > %#x",
-                               h0, k, h1, m);
-                       return false;
-               }
-#endif
-#endif
+               hash_or[0][k] |= params.h1 = hash_32(params.h0, k);
+               KUNIT_EXPECT_LE_MSG(test, params.h1, m,
+                                   "hash_32(%#x, %d) = %#x > %#x",
+                                   params.h0, k, params.h1, m);
+
                /* Test hash_64 */
-               hash_or[1][k] |= h1 = hash_64(h64, k);
-               if (h1 > m) {
-                       pr_err("hash_64(%#llx, %d) = %#x > %#x", h64, k, h1, m);
-                       return false;
-               }
+               hash_or[1][k] |= params.h1 = hash_64(h64, k);
+               KUNIT_EXPECT_LE_MSG(test, params.h1, m,
+                                   "hash_64(%#llx, %d) = %#x > %#x",
+                                   h64, k, params.h1, m);
 #ifdef HAVE_ARCH_HASH_64
-               h2 = hash_64_generic(h64, k);
-#if HAVE_ARCH_HASH_64 == 1
-               if (h1 != h2) {
-                       pr_err("hash_64(%#llx, %d) = %#x != hash_64_generic() "
-                               "= %#x", h64, k, h1, h2);
-                       return false;
-               }
-#else
-               if (h2 > m) {
-                       pr_err("hash_64_generic(%#llx, %d) = %#x > %#x",
-                               h64, k, h1, m);
-                       return false;
-               }
-#endif
+               test_int_hash_64(test, &params, &m, &k);
 #endif
        }
-
-       (void)h2;       /* Suppress unused variable warning */
-       return true;
 }
 
 #define SIZE 256       /* Run time is cubic in SIZE */
 
-static int __init
-test_hash_init(void)
+static void test_string_or(struct kunit *test)
 {
        char buf[SIZE+1];
-       u32 string_or = 0, hash_or[2][33] = { { 0, } };
-       unsigned tests = 0;
+       u32 string_or = 0;
+       int i, j;
+
+       fill_buf(buf, SIZE, 1);
+
+       /* Test every possible non-empty substring in the buffer. */
+       for (j = SIZE; j > 0; --j) {
+               buf[j] = '\0';
+
+               for (i = 0; i <= j; i++) {
+                       u32 h0 = full_name_hash(buf+i, buf+i, j-i);
+
+                       string_or |= h0;
+               } /* i */
+       } /* j */
+
+       /* The OR of all the hash values should cover all the bits */
+       KUNIT_EXPECT_EQ_MSG(test, string_or, -1u,
+                           "OR of all string hash results = %#x != %#x",
+                           string_or, -1u);
+}
+
+static void test_hash_or(struct kunit *test)
+{
+       char buf[SIZE+1];
+       u32 hash_or[2][33] = { { 0, } };
        unsigned long long h64 = 0;
        int i, j;
 
@@ -160,46 +184,27 @@ test_hash_init(void)
                        u32 h0 = full_name_hash(buf+i, buf+i, j-i);
 
                        /* Check that hashlen_string gets the length right */
-                       if (hashlen_len(hashlen) != j-i) {
-                               pr_err("hashlen_string(%d..%d) returned length"
-                                       " %u, expected %d",
-                                       i, j, hashlen_len(hashlen), j-i);
-                               return -EINVAL;
-                       }
+                       KUNIT_EXPECT_EQ_MSG(test, hashlen_len(hashlen), j-i,
+                                           "hashlen_string(%d..%d) returned length %u, expected %d",
+                                           i, j, hashlen_len(hashlen), j-i);
                        /* Check that the hashes match */
-                       if (hashlen_hash(hashlen) != h0) {
-                               pr_err("hashlen_string(%d..%d) = %08x != "
-                                       "full_name_hash() = %08x",
-                                       i, j, hashlen_hash(hashlen), h0);
-                               return -EINVAL;
-                       }
+                       KUNIT_EXPECT_EQ_MSG(test, hashlen_hash(hashlen), h0,
+                                           "hashlen_string(%d..%d) = %08x != full_name_hash() = %08x",
+                                           i, j, hashlen_hash(hashlen), h0);
 
-                       string_or |= h0;
                        h64 = h64 << 32 | h0;   /* For use with hash_64 */
-                       if (!test_int_hash(h64, hash_or))
-                               return -EINVAL;
-                       tests++;
+                       test_int_hash(test, h64, hash_or);
                } /* i */
        } /* j */
 
-       /* The OR of all the hash values should cover all the bits */
-       if (~string_or) {
-               pr_err("OR of all string hash results = %#x != %#x",
-                       string_or, -1u);
-               return -EINVAL;
-       }
-       if (~hash_or[0][0]) {
-               pr_err("OR of all __hash_32 results = %#x != %#x",
-                       hash_or[0][0], -1u);
-               return -EINVAL;
-       }
+       KUNIT_EXPECT_EQ_MSG(test, hash_or[0][0], -1u,
+                           "OR of all __hash_32 results = %#x != %#x",
+                           hash_or[0][0], -1u);
 #ifdef HAVE_ARCH__HASH_32
 #if HAVE_ARCH__HASH_32 != 1    /* Test is pointless if results match */
-       if (~hash_or[1][0]) {
-               pr_err("OR of all __hash_32_generic results = %#x != %#x",
-                       hash_or[1][0], -1u);
-               return -EINVAL;
-       }
+       KUNIT_EXPECT_EQ_MSG(test, hash_or[1][0], -1u,
+                           "OR of all __hash_32_generic results = %#x != %#x",
+                           hash_or[1][0], -1u);
 #endif
 #endif
 
@@ -207,51 +212,27 @@ test_hash_init(void)
        for (i = 1; i <= 32; i++) {
                u32 const m = ((u32)2 << (i-1)) - 1;    /* Low i bits set */
 
-               if (hash_or[0][i] != m) {
-                       pr_err("OR of all hash_32(%d) results = %#x "
-                               "(%#x expected)", i, hash_or[0][i], m);
-                       return -EINVAL;
-               }
-               if (hash_or[1][i] != m) {
-                       pr_err("OR of all hash_64(%d) results = %#x "
-                               "(%#x expected)", i, hash_or[1][i], m);
-                       return -EINVAL;
-               }
+               KUNIT_EXPECT_EQ_MSG(test, hash_or[0][i], m,
+                                   "OR of all hash_32(%d) results = %#x (%#x expected)",
+                                   i, hash_or[0][i], m);
+               KUNIT_EXPECT_EQ_MSG(test, hash_or[1][i], m,
+                                   "OR of all hash_64(%d) results = %#x (%#x expected)",
+                                   i, hash_or[1][i], m);
        }
+}
 
-       /* Issue notices about skipped tests. */
-#ifdef HAVE_ARCH__HASH_32
-#if HAVE_ARCH__HASH_32 != 1
-       pr_info("__hash_32() is arch-specific; not compared to generic.");
-#endif
-#else
-       pr_info("__hash_32() has no arch implementation to test.");
-#endif
-#ifdef HAVE_ARCH_HASH_32
-#if HAVE_ARCH_HASH_32 != 1
-       pr_info("hash_32() is arch-specific; not compared to generic.");
-#endif
-#else
-       pr_info("hash_32() has no arch implementation to test.");
-#endif
-#ifdef HAVE_ARCH_HASH_64
-#if HAVE_ARCH_HASH_64 != 1
-       pr_info("hash_64() is arch-specific; not compared to generic.");
-#endif
-#else
-       pr_info("hash_64() has no arch implementation to test.");
-#endif
-
-       pr_notice("%u tests passed.", tests);
+static struct kunit_case hash_test_cases[] __refdata = {
+       KUNIT_CASE(test_string_or),
+       KUNIT_CASE(test_hash_or),
+       {}
+};
 
-       return 0;
-}
+static struct kunit_suite hash_test_suite = {
+       .name = "hash",
+       .test_cases = hash_test_cases,
+};
 
-static void __exit test_hash_exit(void)
-{
-}
 
-module_init(test_hash_init);   /* Does everything */
-module_exit(test_hash_exit);   /* Does nothing */
+kunit_test_suite(hash_test_suite);
 
 MODULE_LICENSE("GPL");