mirror of
https://github.com/smaeul/u-boot.git
synced 2025-11-04 05:50:17 +00:00
467382ca03
After some header file cleanups to add missing include files, remove common.h from all files in the lib directory. This primarily means just dropping the line but in a few cases we need to add in other header files now. Reviewed-by: Simon Glass <sjg@chromium.org> Signed-off-by: Tom Rini <trini@konsulko.com>
62 lines
1.3 KiB
C
62 lines
1.3 KiB
C
// SPDX-License-Identifier: GPL-2.0+
|
|
/*
|
|
* Copyright 2021 Google LLC
|
|
*/
|
|
|
|
#define LOG_CATEGORY LOGC_BOOT
|
|
|
|
#include <abuf.h>
|
|
#include <log.h>
|
|
#include <malloc.h>
|
|
#include <linux/errno.h>
|
|
#include <linux/zstd.h>
|
|
|
|
int zstd_decompress(struct abuf *in, struct abuf *out)
|
|
{
|
|
zstd_dctx *ctx;
|
|
size_t wsize, len;
|
|
void *workspace;
|
|
int ret;
|
|
|
|
wsize = zstd_dctx_workspace_bound();
|
|
workspace = malloc(wsize);
|
|
if (!workspace) {
|
|
debug("%s: cannot allocate workspace of size %zu\n", __func__,
|
|
wsize);
|
|
return -ENOMEM;
|
|
}
|
|
|
|
ctx = zstd_init_dctx(workspace, wsize);
|
|
if (!ctx) {
|
|
log_err("%s: zstd_init_dctx() failed\n", __func__);
|
|
ret = -EPERM;
|
|
goto do_free;
|
|
}
|
|
|
|
/*
|
|
* Find out how large the frame actually is, there may be junk at
|
|
* the end of the frame that zstd_decompress_dctx() can't handle.
|
|
*/
|
|
len = zstd_find_frame_compressed_size(abuf_data(in), abuf_size(in));
|
|
if (zstd_is_error(len)) {
|
|
log_err("%s: failed to detect compressed size: %d\n", __func__,
|
|
zstd_get_error_code(len));
|
|
ret = -EINVAL;
|
|
goto do_free;
|
|
}
|
|
|
|
len = zstd_decompress_dctx(ctx, abuf_data(out), abuf_size(out),
|
|
abuf_data(in), len);
|
|
if (zstd_is_error(len)) {
|
|
log_err("%s: failed to decompress: %d\n", __func__,
|
|
zstd_get_error_code(len));
|
|
ret = -EINVAL;
|
|
goto do_free;
|
|
}
|
|
|
|
ret = len;
|
|
do_free:
|
|
free(workspace);
|
|
return ret;
|
|
}
|