mirror of
https://github.com/kmod-project/kmod.git
synced 2026-01-26 07:37:54 +00:00
bbab061e51
Add a dlopen option that allows toggling what libraries libkmod should attempt to dlopen. If -Ddlopen=foo is passed, it means that library is required to build, regardless of -Dfoo=*. However that library will only be linked in if it's not set as dlopen. Signed-off-by: Lucas De Marchi <lucas.de.marchi@gmail.com> [ with disagreement on the need to toggle each one individually, it'd be better to be all-or-nothing dlopen'ed ] Reviewed-by: Emil Velikov <emil.l.velikov@gmail.com> Link: https://github.com/kmod-project/kmod/pull/262
108 lines
1.9 KiB
C
108 lines
1.9 KiB
C
// SPDX-License-Identifier: LGPL-2.1-or-later
|
|
/*
|
|
* Copyright © 2024 Intel Corporation
|
|
*/
|
|
|
|
#define DLSYM_LOCALLY_ENABLED ENABLE_ZLIB_DLOPEN
|
|
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#include <zlib.h>
|
|
|
|
#include <shared/util.h>
|
|
|
|
#include "libkmod.h"
|
|
#include "libkmod-internal.h"
|
|
#include "libkmod-internal-file.h"
|
|
|
|
#define READ_STEP (4 * 1024 * 1024)
|
|
|
|
#define DL_SYMBOL_TABLE(M) \
|
|
M(gzclose) \
|
|
M(gzdopen) \
|
|
M(gzerror) \
|
|
M(gzread)
|
|
|
|
DL_SYMBOL_TABLE(DECLARE_SYM)
|
|
|
|
static int dlopen_zlib(void)
|
|
{
|
|
static void *dl = NULL;
|
|
|
|
if (!DLSYM_LOCALLY_ENABLED)
|
|
return 0;
|
|
|
|
return dlsym_many(&dl, "libz.so.1", DL_SYMBOL_TABLE(DLSYM_ARG) NULL);
|
|
}
|
|
|
|
int kmod_file_load_zlib(struct kmod_file *file)
|
|
{
|
|
_cleanup_free_ unsigned char *p = NULL;
|
|
int ret = 0;
|
|
off_t did = 0, total = 0;
|
|
gzFile gzf;
|
|
int gzfd;
|
|
|
|
ret = dlopen_zlib();
|
|
if (ret < 0) {
|
|
ERR(file->ctx, "zlib: can't load and resolve symbols (%s)",
|
|
strerror(-ret));
|
|
return -EINVAL;
|
|
}
|
|
|
|
errno = 0;
|
|
gzfd = fcntl(file->fd, F_DUPFD_CLOEXEC, 3);
|
|
if (gzfd < 0)
|
|
return -errno;
|
|
|
|
gzf = sym_gzdopen(gzfd, "rb"); /* takes ownership of the fd */
|
|
if (gzf == NULL) {
|
|
close(gzfd);
|
|
return -errno;
|
|
}
|
|
|
|
for (;;) {
|
|
int r;
|
|
|
|
if (did == total) {
|
|
void *tmp = realloc(p, total + READ_STEP);
|
|
if (tmp == NULL) {
|
|
ret = -errno;
|
|
goto error;
|
|
}
|
|
total += READ_STEP;
|
|
p = tmp;
|
|
}
|
|
|
|
r = sym_gzread(gzf, p + did, total - did);
|
|
if (r == 0)
|
|
break;
|
|
else if (r < 0) {
|
|
int gzerr;
|
|
const char *gz_errmsg = sym_gzerror(gzf, &gzerr);
|
|
|
|
ERR(file->ctx, "gzip: %s\n", gz_errmsg);
|
|
|
|
/* gzip might not set errno here */
|
|
ret = gzerr == Z_ERRNO ? -errno : -EINVAL;
|
|
goto error;
|
|
}
|
|
did += r;
|
|
}
|
|
|
|
file->memory = TAKE_PTR(p);
|
|
file->size = did;
|
|
sym_gzclose(gzf);
|
|
|
|
return 0;
|
|
|
|
error:
|
|
sym_gzclose(gzf);
|
|
return ret;
|
|
}
|