1 files changed, 100 insertions, 0 deletions

diff --git a/userspace/include/unit/unit.h b/userspace/include/unit/unit.h
new file mode 100644
index 00000000..9438d4d9
--- /dev/null
+++ b/userspace/include/unit/unit.h
@@ -0,0 +1,100 @@
+/*
+ * Copyright (c) 2018, NVIDIA CORPORATION.  All rights reserved.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+ * DEALINGS IN THE SOFTWARE.
+ */
+
+#ifndef __UNIT_UNIT_H__
+#define __UNIT_UNIT_H__
+
+struct gk20a;
+
+struct unit_module;
+typedef int (*module_test_fn)(struct unit_module *m,
+                              struct gk20a *g, void *args);
+
+#define UNIT_SUCCESS            0
+#define UNIT_FAIL               -1
+
+struct unit_module_test {
+        /*
+         * Name of the test.
+         */
+        const char *name;
+
+        /*
+         * Function to call to execute the test.
+         */
+        module_test_fn fn;
+
+        /*
+         * A void pointer to arbitrary arguments. Lets the same unit test
+         * function perform multiple tests. This gets passed into the
+         * module_test_fn as @args.
+         */
+        void *args;
+};
+
+/*
+ * Interface to the unit test framework module loader. Each unit test module
+ * will have exactly one of these.
+ */
+struct unit_module {
+        /*
+         * Name of the module.
+         */
+        const char                      *name;
+
+        /*
+         * NULL terminated list of tests within the module.
+         */
+        struct unit_module_test         *tests;
+        unsigned long                    nr_tests;
+
+        /*
+         * For the core FW to use. Not for modules!!!
+         */
+        void                            *lib_handle;
+        struct unit_fw                  *fw;
+};
+
+#define UNIT_MODULE(__name, __tests)                                    \
+        struct unit_module __unit_module__ = {                          \
+                .name = #__name,                                        \
+                .tests = __tests,                                       \
+                .nr_tests = (sizeof(__tests) /                          \
+                             sizeof(struct unit_module_test)),          \
+                .lib_handle = NULL,                                     \
+        }
+
+#define UNIT_TEST(__name, __fn, __args)                                 \
+        {                                                               \
+                .name = #__name,                                        \
+                .fn = __fn,                                             \
+                .args = __args,                                         \
+        }
+
+#define unit_return_fail(m, msg, ...)                                   \
+        do {                                                            \
+                unit_err(m, "%s():%d " msg,                             \
+                         __func__, __LINE__, ##__VA_ARGS__);            \
+                return UNIT_FAIL;                                       \
+        } while (0)
+
+#endif