diff options
Diffstat (limited to 'userspace/src/nvgpu.c')
-rw-r--r-- | userspace/src/nvgpu.c | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/userspace/src/nvgpu.c b/userspace/src/nvgpu.c new file mode 100644 index 00000000..a9b7ced3 --- /dev/null +++ b/userspace/src/nvgpu.c | |||
@@ -0,0 +1,78 @@ | |||
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 | #include <dlfcn.h> | ||
24 | #include <stdlib.h> | ||
25 | |||
26 | #include <unit/io.h> | ||
27 | #include <unit/core.h> | ||
28 | |||
29 | /* | ||
30 | * Load libnvgpu-drv.so. This is done with dlopen() since this will make | ||
31 | * resolving addresses into symbols easier in the future. | ||
32 | * | ||
33 | * Also, this makes people think carefully about what functions to call in | ||
34 | * nvgpu-drv from the unit test FW. The interaction should really be limited | ||
35 | * and doing explicit name lookups is a good way to prevent too much coupling. | ||
36 | */ | ||
37 | int core_load_nvgpu(struct unit_fw *fw) | ||
38 | { | ||
39 | const char *msg; | ||
40 | |||
41 | /* | ||
42 | * Specify a GLOBAL binding so that subsequently loaded unit tests see | ||
43 | * the nvgpu-drv library. They will of course need it (and will access | ||
44 | * it directly). I.e they will link against nvgpu-drv and this should | ||
45 | * satisfy that linkage. | ||
46 | */ | ||
47 | fw->nvgpu_so = dlopen("libnvgpu-drv.so", RTLD_NOW | RTLD_GLOBAL); | ||
48 | |||
49 | if (fw->nvgpu_so == NULL) { | ||
50 | msg = dlerror(); | ||
51 | core_err(fw, "Failed to load nvgpu-drv: %s\n", msg); | ||
52 | return -1; | ||
53 | } | ||
54 | |||
55 | /* | ||
56 | * We directly check the value of the returned symbol for these | ||
57 | * functions against NULL because if it is NULL then something is | ||
58 | * terribly wrong. | ||
59 | */ | ||
60 | |||
61 | fw->nvgpu.nvgpu_posix_probe = dlsym(fw->nvgpu_so, | ||
62 | "nvgpu_posix_probe"); | ||
63 | if (fw->nvgpu.nvgpu_posix_probe == NULL) { | ||
64 | msg = dlerror(); | ||
65 | core_err(fw, "Failed to resolve nvgpu_posix_probe: %s\n", msg); | ||
66 | return -1; | ||
67 | } | ||
68 | |||
69 | fw->nvgpu.nvgpu_posix_cleanup = dlsym(fw->nvgpu_so, | ||
70 | "nvgpu_posix_cleanup"); | ||
71 | if (fw->nvgpu.nvgpu_posix_cleanup == NULL) { | ||
72 | msg = dlerror(); | ||
73 | core_err(fw, "Failed to resolve nvgpu_posix_cleanup: %s\n", msg); | ||
74 | return -1; | ||
75 | } | ||
76 | |||
77 | return 0; | ||
78 | } | ||