2969ca571a
GitOrigin-RevId: 257cbbcd3ab7bd96f5d24d50adc807de7c82e06d
130 lines
5.4 KiB
Diff
130 lines
5.4 KiB
Diff
diff --git a/src/ldcache.c b/src/ldcache.c
|
|
index 38bab05..e1abc89 100644
|
|
--- a/src/ldcache.c
|
|
+++ b/src/ldcache.c
|
|
@@ -108,40 +108,27 @@ ldcache_close(struct ldcache *ctx)
|
|
|
|
int
|
|
ldcache_resolve(struct ldcache *ctx, uint32_t arch, const char *root, const char * const libs[],
|
|
- char *paths[], size_t size, ldcache_select_fn select, void *select_ctx)
|
|
+ char *paths[], size_t size, const char* version)
|
|
{
|
|
char path[PATH_MAX];
|
|
- struct header_libc6 *h;
|
|
- int override;
|
|
+ char dir[PATH_MAX];
|
|
+ char lib[PATH_MAX];
|
|
|
|
- h = (struct header_libc6 *)ctx->ptr;
|
|
memset(paths, 0, size * sizeof(*paths));
|
|
|
|
- for (uint32_t i = 0; i < h->nlibs; ++i) {
|
|
- int32_t flags = h->libs[i].flags;
|
|
- char *key = (char *)ctx->ptr + h->libs[i].key;
|
|
- char *value = (char *)ctx->ptr + h->libs[i].value;
|
|
-
|
|
- if (!(flags & LD_ELF) || (flags & LD_ARCH_MASK) != arch)
|
|
+ for (size_t j = 0; j < size; ++j) {
|
|
+ snprintf(dir, 100, "/run/opengl-driver%s/lib",
|
|
+ arch == LD_I386_LIB32 ? "-32" : "");
|
|
+ if (!strncmp(libs[j], "libvdpau_nvidia.so", 100))
|
|
+ strcat(dir, "/vdpau");
|
|
+ snprintf(lib, 100, "%s/%s.%s", dir, libs[j], version);
|
|
+ if (path_resolve_full(ctx->err, path, "/", lib) < 0)
|
|
+ return (-1);
|
|
+ if (!file_exists(ctx->err, path))
|
|
continue;
|
|
-
|
|
- for (size_t j = 0; j < size; ++j) {
|
|
- if (!str_has_prefix(key, libs[j]))
|
|
- continue;
|
|
- if (path_resolve(ctx->err, path, root, value) < 0)
|
|
- return (-1);
|
|
- if (paths[j] != NULL && str_equal(paths[j], path))
|
|
- continue;
|
|
- if ((override = select(ctx->err, select_ctx, root, paths[j], path)) < 0)
|
|
- return (-1);
|
|
- if (override) {
|
|
- free(paths[j]);
|
|
- paths[j] = xstrdup(ctx->err, path);
|
|
- if (paths[j] == NULL)
|
|
- return (-1);
|
|
- }
|
|
- break;
|
|
- }
|
|
+ paths[j] = xstrdup(ctx->err, path);
|
|
+ if (paths[j] == NULL)
|
|
+ return (-1);
|
|
}
|
|
return (0);
|
|
}
|
|
diff --git a/src/ldcache.h b/src/ldcache.h
|
|
index 33d78dd..2b087db 100644
|
|
--- a/src/ldcache.h
|
|
+++ b/src/ldcache.h
|
|
@@ -50,6 +50,6 @@ void ldcache_init(struct ldcache *, struct error *, const char *);
|
|
int ldcache_open(struct ldcache *);
|
|
int ldcache_close(struct ldcache *);
|
|
int ldcache_resolve(struct ldcache *, uint32_t, const char *, const char * const [],
|
|
- char *[], size_t, ldcache_select_fn, void *);
|
|
+ char *[], size_t, const char*);
|
|
|
|
#endif /* HEADER_LDCACHE_H */
|
|
diff --git a/src/nvc_info.c b/src/nvc_info.c
|
|
index 30e3cfd..6d12a50 100644
|
|
--- a/src/nvc_info.c
|
|
+++ b/src/nvc_info.c
|
|
@@ -167,15 +167,13 @@ find_library_paths(struct error *err, struct nvc_driver_info *info, const char *
|
|
if (path_resolve_full(err, path, root, ldcache) < 0)
|
|
return (-1);
|
|
ldcache_init(&ld, err, path);
|
|
- if (ldcache_open(&ld) < 0)
|
|
- return (-1);
|
|
|
|
info->nlibs = size;
|
|
info->libs = array_new(err, size);
|
|
if (info->libs == NULL)
|
|
goto fail;
|
|
if (ldcache_resolve(&ld, LIB_ARCH, root, libs,
|
|
- info->libs, info->nlibs, select_libraries_fn, info) < 0)
|
|
+ info->libs, info->nlibs, info->nvrm_version) < 0)
|
|
goto fail;
|
|
|
|
info->nlibs32 = size;
|
|
@@ -183,13 +181,11 @@ find_library_paths(struct error *err, struct nvc_driver_info *info, const char *
|
|
if (info->libs32 == NULL)
|
|
goto fail;
|
|
if (ldcache_resolve(&ld, LIB32_ARCH, root, libs,
|
|
- info->libs32, info->nlibs32, select_libraries_fn, info) < 0)
|
|
+ info->libs32, info->nlibs32, info->nvrm_version) < 0)
|
|
goto fail;
|
|
rv = 0;
|
|
|
|
fail:
|
|
- if (ldcache_close(&ld) < 0)
|
|
- return (-1);
|
|
return (rv);
|
|
}
|
|
|
|
@@ -203,7 +199,7 @@ find_binary_paths(struct error *err, struct nvc_driver_info *info, const char *r
|
|
char path[PATH_MAX];
|
|
int rv = -1;
|
|
|
|
- if ((env = secure_getenv("PATH")) == NULL) {
|
|
+ if ((env = "/run/nvidia-docker/bin:/run/nvidia-docker/extras/bin") == NULL) {
|
|
error_setx(err, "environment variable PATH not found");
|
|
return (-1);
|
|
}
|
|
diff --git a/src/nvc_ldcache.c b/src/nvc_ldcache.c
|
|
index 6ff380f..cbe6a69 100644
|
|
--- a/src/nvc_ldcache.c
|
|
+++ b/src/nvc_ldcache.c
|
|
@@ -340,7 +340,7 @@ nvc_ldcache_update(struct nvc_context *ctx, const struct nvc_container *cnt)
|
|
if (validate_args(ctx, cnt != NULL) < 0)
|
|
return (-1);
|
|
|
|
- argv = (char * []){cnt->cfg.ldconfig, cnt->cfg.libs_dir, cnt->cfg.libs32_dir, NULL};
|
|
+ argv = (char * []){cnt->cfg.ldconfig, "-f", "/tmp/ld.so.conf.nvidia-host", "-C", "/tmp/ld.so.cache.nvidia-host", cnt->cfg.libs_dir, cnt->cfg.libs32_dir, NULL};
|
|
if (*argv[0] == '@') {
|
|
/*
|
|
* We treat this path specially to be relative to the host filesystem.
|