diff options
Diffstat (limited to 'userspace/src/unit_main.c')
-rw-r--r-- | userspace/src/unit_main.c | 76 |
1 files changed, 76 insertions, 0 deletions
diff --git a/userspace/src/unit_main.c b/userspace/src/unit_main.c new file mode 100644 index 00000000..31c31d50 --- /dev/null +++ b/userspace/src/unit_main.c | |||
@@ -0,0 +1,76 @@ | |||
1 | /* | ||
2 | * Copyright (c) 2018, NVIDIA CORPORATION. All rights reserved. | ||
3 | * | ||
4 | * Permission is hereby granted, free of charge, to any person obtaining a | ||
5 | * copy of this software and associated documentation files (the "Software"), | ||
6 | * to deal in the Software without restriction, including without limitation | ||
7 | * the rights to use, copy, modify, merge, publish, distribute, sublicense, | ||
8 | * and/or sell copies of the Software, and to permit persons to whom the | ||
9 | * Software is furnished to do so, subject to the following conditions: | ||
10 | * | ||
11 | * The above copyright notice and this permission notice shall be included in | ||
12 | * all copies or substantial portions of the Software. | ||
13 | * | ||
14 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | ||
15 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | ||
16 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL | ||
17 | * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | ||
18 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING | ||
19 | * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER | ||
20 | * DEALINGS IN THE SOFTWARE. | ||
21 | */ | ||
22 | |||
23 | /** | ||
24 | * NvGpu unit testing framework! | ||
25 | */ | ||
26 | |||
27 | #include <stdlib.h> | ||
28 | #include <string.h> | ||
29 | |||
30 | #include <unit/io.h> | ||
31 | #include <unit/core.h> | ||
32 | #include <unit/args.h> | ||
33 | #include <unit/module.h> | ||
34 | #include <unit/results.h> | ||
35 | |||
36 | int main(int argc, char **argv) | ||
37 | { | ||
38 | struct unit_fw *fw; | ||
39 | int ret; | ||
40 | |||
41 | fw = malloc(sizeof(*fw)); | ||
42 | if (!fw) | ||
43 | return 1; | ||
44 | |||
45 | memset(fw, 0, sizeof(*fw)); | ||
46 | |||
47 | ret = core_parse_args(fw, argc, argv); | ||
48 | if (ret) { | ||
49 | core_err(fw, "Enable to parse args.\n"); | ||
50 | core_err(fw, "Exiting!\n"); | ||
51 | return 1; | ||
52 | } | ||
53 | |||
54 | core_vbs(fw, 1, "Welcome to the nvgpu unit testing framework!\n"); | ||
55 | |||
56 | if (args(fw)->help) { | ||
57 | core_print_help(fw); | ||
58 | return 1; | ||
59 | } | ||
60 | |||
61 | ret = core_load_nvgpu(fw); | ||
62 | if (ret != 0) | ||
63 | return ret; | ||
64 | |||
65 | fw->modules = core_load_modules(fw); | ||
66 | if (fw->modules == NULL) | ||
67 | return -1; | ||
68 | |||
69 | ret = core_exec(fw); | ||
70 | if (ret != 0) | ||
71 | return ret; | ||
72 | |||
73 | core_print_test_status(fw); | ||
74 | |||
75 | return 0; | ||
76 | } | ||