ida: Start new test_ida module

Start transitioning the IDA tests into kernel space.  Framework heavily
cribbed from test_xarray.c.

Signed-off-by: Matthew Wilcox <willy@infradead.org>

3 files changed, 49 insertions, 0 deletions

diff --git a/lib/Kconfig.debug b/lib/Kconfig.debug
index 8838d1158d19..2fff661d9070 100644
--- a/lib/Kconfig.debug
+++ b/lib/Kconfig.debug
@@ -1827,6 +1827,9 @@ config TEST_HASH
           This is intended to help people writing architecture-specific
           optimized versions.  If unsure, say N.
 
+config TEST_IDA
+        tristate "Perform selftest on IDA functions"
+
 config TEST_PARMAN
         tristate "Perform selftest on priority array manager"
         default n
diff --git a/lib/Makefile b/lib/Makefile
index 90dc5520b784..18d87ca4c949 100644
--- a/lib/Makefile
+++ b/lib/Makefile
@@ -50,6 +50,7 @@ obj-$(CONFIG_TEST_BPF) += test_bpf.o
 obj-$(CONFIG_TEST_FIRMWARE) += test_firmware.o
 obj-$(CONFIG_TEST_SYSCTL) += test_sysctl.o
 obj-$(CONFIG_TEST_HASH) += test_hash.o test_siphash.o
+obj-$(CONFIG_TEST_IDA) += test_ida.o
 obj-$(CONFIG_TEST_KASAN) += test_kasan.o
 CFLAGS_test_kasan.o += -fno-builtin
 obj-$(CONFIG_TEST_UBSAN) += test_ubsan.o
diff --git a/lib/test_ida.c b/lib/test_ida.c
new file mode 100644
index 000000000000..8c9a0672696b
--- /dev/null
+++ b/lib/test_ida.c
@@ -0,0 +1,45 @@
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * test_ida.c: Test the IDA API
+ * Copyright (c) 2016-2018 Microsoft Corporation
+ * Copyright (c) 2018 Oracle Corporation
+ * Author: Matthew Wilcox <willy@infradead.org>
+ */
+
+#include <linux/idr.h>
+#include <linux/module.h>
+
+static unsigned int tests_run;
+static unsigned int tests_passed;
+
+#ifdef __KERNEL__
+void ida_dump(struct ida *ida) { }
+#endif
+#define IDA_BUG_ON(ida, x) do {                                         \
+        tests_run++;                                                    \
+        if (x) {                                                        \
+                ida_dump(ida);                                          \
+                dump_stack();                                           \
+        } else {                                                        \
+                tests_passed++;                                         \
+        }                                                               \
+} while (0)
+
+static int ida_checks(void)
+{
+        DEFINE_IDA(ida);
+
+        IDA_BUG_ON(&ida, !ida_is_empty(&ida));
+
+        printk("IDA: %u of %u tests passed\n", tests_passed, tests_run);
+        return (tests_run != tests_passed) ? 0 : -EINVAL;
+}
+
+static void ida_exit(void)
+{
+}
+
+module_init(ida_checks);
+module_exit(ida_exit);
+MODULE_AUTHOR("Matthew Wilcox <willy@infradead.org>");
+MODULE_LICENSE("GPL");