mirror of
https://github.com/morgan9e/systemd
synced 2026-04-15 00:47:10 +09:00
5c672e90a6
When looking at how dlopen for various libs is implemented, I found that the macros hide too much. I find it much easier to see what is going on if 'extern' and '= NULL' are written explicitly. After all, we don't hide those for other definitions, e.g. our style guide says that static variables should be initialized with '= NULL'. With that change, it's much more obvious what is a variable declaration and what is a variable initialization.
134 lines
4.8 KiB
C
134 lines
4.8 KiB
C
/* SPDX-License-Identifier: LGPL-2.1-or-later */
|
|
#pragma once
|
|
|
|
#include <errno.h>
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <unistd.h>
|
|
|
|
#if HAVE_LZ4
|
|
#include <lz4.h>
|
|
#include <lz4frame.h>
|
|
#endif
|
|
|
|
#include "dlfcn-util.h"
|
|
|
|
typedef enum Compression {
|
|
COMPRESSION_NONE,
|
|
COMPRESSION_XZ,
|
|
COMPRESSION_LZ4,
|
|
COMPRESSION_ZSTD,
|
|
_COMPRESSION_MAX,
|
|
_COMPRESSION_INVALID = -EINVAL,
|
|
} Compression;
|
|
|
|
const char* compression_to_string(Compression compression);
|
|
Compression compression_from_string(const char *compression);
|
|
|
|
bool compression_supported(Compression c);
|
|
|
|
int compress_blob_xz(const void *src, uint64_t src_size,
|
|
void *dst, size_t dst_alloc_size, size_t *dst_size);
|
|
int compress_blob_lz4(const void *src, uint64_t src_size,
|
|
void *dst, size_t dst_alloc_size, size_t *dst_size);
|
|
int compress_blob_zstd(const void *src, uint64_t src_size,
|
|
void *dst, size_t dst_alloc_size, size_t *dst_size);
|
|
|
|
int decompress_blob_xz(const void *src, uint64_t src_size,
|
|
void **dst, size_t* dst_size, size_t dst_max);
|
|
int decompress_blob_lz4(const void *src, uint64_t src_size,
|
|
void **dst, size_t* dst_size, size_t dst_max);
|
|
int decompress_blob_zstd(const void *src, uint64_t src_size,
|
|
void **dst, size_t* dst_size, size_t dst_max);
|
|
int decompress_blob(Compression compression,
|
|
const void *src, uint64_t src_size,
|
|
void **dst, size_t* dst_size, size_t dst_max);
|
|
|
|
int decompress_startswith_xz(const void *src, uint64_t src_size,
|
|
void **buffer,
|
|
const void *prefix, size_t prefix_len,
|
|
uint8_t extra);
|
|
int decompress_startswith_lz4(const void *src, uint64_t src_size,
|
|
void **buffer,
|
|
const void *prefix, size_t prefix_len,
|
|
uint8_t extra);
|
|
int decompress_startswith_zstd(const void *src, uint64_t src_size,
|
|
void **buffer,
|
|
const void *prefix, size_t prefix_len,
|
|
uint8_t extra);
|
|
int decompress_startswith(Compression compression,
|
|
const void *src, uint64_t src_size,
|
|
void **buffer,
|
|
const void *prefix, size_t prefix_len,
|
|
uint8_t extra);
|
|
|
|
int compress_stream_xz(int fdf, int fdt, uint64_t max_bytes, uint64_t *ret_uncompressed_size);
|
|
int compress_stream_lz4(int fdf, int fdt, uint64_t max_bytes, uint64_t *ret_uncompressed_size);
|
|
int compress_stream_zstd(int fdf, int fdt, uint64_t max_bytes, uint64_t *ret_uncompressed_size);
|
|
|
|
int decompress_stream_xz(int fdf, int fdt, uint64_t max_size);
|
|
int decompress_stream_lz4(int fdf, int fdt, uint64_t max_size);
|
|
int decompress_stream_zstd(int fdf, int fdt, uint64_t max_size);
|
|
|
|
#if HAVE_LZ4
|
|
extern DLSYM_PROTOTYPE(LZ4_compress_default);
|
|
extern DLSYM_PROTOTYPE(LZ4_decompress_safe);
|
|
extern DLSYM_PROTOTYPE(LZ4_decompress_safe_partial);
|
|
extern DLSYM_PROTOTYPE(LZ4_versionNumber);
|
|
|
|
int dlopen_lz4(void);
|
|
#endif
|
|
|
|
#if HAVE_ZSTD
|
|
int dlopen_zstd(void);
|
|
#endif
|
|
|
|
#if HAVE_XZ
|
|
int dlopen_lzma(void);
|
|
#endif
|
|
|
|
static inline int compress_blob(
|
|
Compression compression,
|
|
const void *src, uint64_t src_size,
|
|
void *dst, size_t dst_alloc_size, size_t *dst_size) {
|
|
|
|
switch (compression) {
|
|
case COMPRESSION_ZSTD:
|
|
return compress_blob_zstd(src, src_size, dst, dst_alloc_size, dst_size);
|
|
case COMPRESSION_LZ4:
|
|
return compress_blob_lz4(src, src_size, dst, dst_alloc_size, dst_size);
|
|
case COMPRESSION_XZ:
|
|
return compress_blob_xz(src, src_size, dst, dst_alloc_size, dst_size);
|
|
default:
|
|
return -EOPNOTSUPP;
|
|
}
|
|
}
|
|
|
|
static inline int compress_stream(int fdf, int fdt, uint64_t max_bytes, uint64_t *ret_uncompressed_size) {
|
|
switch (DEFAULT_COMPRESSION) {
|
|
case COMPRESSION_ZSTD:
|
|
return compress_stream_zstd(fdf, fdt, max_bytes, ret_uncompressed_size);
|
|
case COMPRESSION_LZ4:
|
|
return compress_stream_lz4(fdf, fdt, max_bytes, ret_uncompressed_size);
|
|
case COMPRESSION_XZ:
|
|
return compress_stream_xz(fdf, fdt, max_bytes, ret_uncompressed_size);
|
|
default:
|
|
return -EOPNOTSUPP;
|
|
}
|
|
}
|
|
|
|
static inline const char* default_compression_extension(void) {
|
|
switch (DEFAULT_COMPRESSION) {
|
|
case COMPRESSION_ZSTD:
|
|
return ".zst";
|
|
case COMPRESSION_LZ4:
|
|
return ".lz4";
|
|
case COMPRESSION_XZ:
|
|
return ".xz";
|
|
default:
|
|
return "";
|
|
}
|
|
}
|
|
|
|
int decompress_stream(const char *filename, int fdf, int fdt, uint64_t max_bytes);
|