lib: tests: Add a test for sbi_bitmap

Add test suite covering all of the functions from lib/sbi/sbi_bitmap.c:
__bitmap_and, __bitmap_or and __bitmap_xor.

Signed-off-by: Ivan Orlov <ivan.orlov0322@gmail.com>
Reviewed-by: Andrew Jones <ajones@ventanamicro.com>

diff --git a/lib/sbi/objects.mk b/lib/sbi/objects.mk
index 08959f164113b670f4fe55c07a9ec489268e862b..b4c273fb6c8e3493462905043ac43368f0dc930c 100644 (file)
--- a/lib/sbi/objects.mk
+++ b/lib/sbi/objects.mk
@@ -14,6 +14,9 @@ libsbi-objs-y += riscv_locks.o
 libsbi-objs-$(CONFIG_SBIUNIT) += sbi_unit_test.o
 libsbi-objs-$(CONFIG_SBIUNIT) += sbi_unit_tests.o
 
+libsbi-objs-$(CONFIG_SBIUNIT) += sbi_bitmap_test.o
+carray-sbi_unit_tests-$(CONFIG_SBIUNIT) += bitmap_test_suite
+
 libsbi-objs-y += sbi_ecall.o
 libsbi-objs-y += sbi_ecall_exts.o
 

diff --git a/lib/sbi/sbi_bitmap_test.c b/lib/sbi/sbi_bitmap_test.c
new file mode 100644 (file)
index 0000000..d2c3599
--- /dev/null
+++ b/lib/sbi/sbi_bitmap_test.c
@@ -0,0 +1,102 @@
+/*
+ * SPDX-License-Identifier: BSD-2-Clause
+ *
+ * Author: Ivan Orlov <ivan.orlov0322@gmail.com>
+ */
+#include <sbi/sbi_bitmap.h>
+#include <sbi/sbi_unit_test.h>
+
+#define DATA_SIZE sizeof(data_zero)
+#define DATA_BIT_SIZE (DATA_SIZE * 8)
+
+static unsigned long data_a[] = { 0xDEADBEEF, 0x00BAB10C, 0x1BADB002, 0xABADBABE };
+static unsigned long data_b[] = { 0xC00010FF, 0x00BAB10C, 0xBAAAAAAD, 0xBADDCAFE };
+static unsigned long data_zero[] = { 0, 0, 0, 0 };
+
+static void bitmap_and_test(struct sbiunit_test_case *test)
+{
+       unsigned long res[DATA_SIZE];
+       unsigned long a_and_b[] = { data_a[0] & data_b[0], data_a[1] & data_b[1],
+                                   data_a[2] & data_b[2], data_a[3] & data_b[3] };
+
+       __bitmap_and(res, data_a, data_b, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, a_and_b, DATA_SIZE);
+
+       /* a & a = a */
+       __bitmap_and(res, data_a, data_a, DATA_BIT_SIZE);
+       SBIUNIT_ASSERT_MEMEQ(test, res, data_a, DATA_SIZE);
+
+       /* a & 0 = 0 */
+       __bitmap_and(res, data_a, data_zero, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_zero, DATA_SIZE);
+
+       /* 0 & 0 = 0 */
+       __bitmap_and(res, data_zero, data_zero, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_zero, DATA_SIZE);
+
+       sbi_memcpy(res, data_zero, DATA_SIZE);
+       /* Cover zero 'bits' argument */
+       __bitmap_and(res, data_a, data_b, 0);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_zero, DATA_SIZE);
+}
+
+static void bitmap_or_test(struct sbiunit_test_case *test)
+{
+       unsigned long res[DATA_SIZE];
+       unsigned long a_or_b[] = { data_a[0] | data_b[0], data_a[1] | data_b[1],
+                                  data_a[2] | data_b[2], data_a[3] | data_b[3] };
+
+       __bitmap_or(res, data_a, data_b, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, a_or_b, DATA_SIZE);
+
+       /* a | a = a */
+       __bitmap_or(res, data_a, data_a, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_a, DATA_SIZE);
+
+       /* a | 0 = a */
+       __bitmap_or(res, data_a, data_zero, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_a, DATA_SIZE);
+
+       /* 0 | 0 = 0 */
+       __bitmap_or(res, data_zero, data_zero, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_zero, DATA_SIZE);
+
+       sbi_memcpy(res, data_zero, DATA_SIZE);
+       __bitmap_or(res, data_a, data_b, 0);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_zero, DATA_SIZE);
+}
+
+static void bitmap_xor_test(struct sbiunit_test_case *test)
+{
+       unsigned long res[DATA_SIZE];
+       unsigned long a_xor_b[] = { data_a[0] ^ data_b[0], data_a[1] ^ data_b[1],
+                                   data_a[2] ^ data_b[2], data_a[3] ^ data_b[3] };
+
+       __bitmap_xor(res, data_a, data_b, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, a_xor_b, DATA_SIZE);
+
+       /* a ^ 0 = a */
+       __bitmap_xor(res, data_a, data_zero, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_a, DATA_SIZE);
+
+       /* a ^ a = 0 */
+       __bitmap_xor(res, data_a, data_a, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_zero, DATA_SIZE);
+
+       /* 0 ^ 0 = 0 */
+       __bitmap_xor(res, data_zero, data_zero, DATA_BIT_SIZE);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_zero, DATA_SIZE);
+
+       sbi_memcpy(res, data_zero, DATA_SIZE);
+       __bitmap_xor(res, data_a, data_b, 0);
+       SBIUNIT_EXPECT_MEMEQ(test, res, data_zero, DATA_SIZE);
+}
+
+static struct sbiunit_test_case bitmap_test_cases[] = {
+       SBIUNIT_TEST_CASE(bitmap_and_test),
+       SBIUNIT_TEST_CASE(bitmap_or_test),
+       SBIUNIT_TEST_CASE(bitmap_xor_test),
+       SBIUNIT_END_CASE,
+};
+
+SBIUNIT_TEST_SUITE(bitmap_test_suite, bitmap_test_cases);