mirror of
https://github.com/AsahiLinux/u-boot
synced 2024-11-28 15:41:40 +00:00
btrfs: Use U-Boot API for decompression
Use the common function to avoid code duplication. Acked-by: Qu Wenruo <wqu@suse.com> Signed-off-by: Simon Glass <sjg@chromium.org>
This commit is contained in:
parent
94d0a2efc0
commit
918adf8e07
1 changed files with 5 additions and 46 deletions
|
@ -6,6 +6,7 @@
|
|||
*/
|
||||
|
||||
#include "btrfs.h"
|
||||
#include <abuf.h>
|
||||
#include <log.h>
|
||||
#include <malloc.h>
|
||||
#include <linux/lzo.h>
|
||||
|
@ -136,54 +137,12 @@ static u32 decompress_zlib(const u8 *_cbuf, u32 clen, u8 *dbuf, u32 dlen)
|
|||
|
||||
static u32 decompress_zstd(const u8 *cbuf, u32 clen, u8 *dbuf, u32 dlen)
|
||||
{
|
||||
ZSTD_DStream *dstream;
|
||||
ZSTD_inBuffer in_buf;
|
||||
ZSTD_outBuffer out_buf;
|
||||
void *workspace;
|
||||
size_t wsize;
|
||||
u32 res = -1;
|
||||
struct abuf in, out;
|
||||
|
||||
wsize = ZSTD_DStreamWorkspaceBound(ZSTD_BTRFS_MAX_INPUT);
|
||||
workspace = malloc(wsize);
|
||||
if (!workspace) {
|
||||
debug("%s: cannot allocate workspace of size %zu\n", __func__,
|
||||
wsize);
|
||||
return -1;
|
||||
}
|
||||
abuf_init_set(&in, (u8 *)cbuf, clen);
|
||||
abuf_init_set(&out, dbuf, dlen);
|
||||
|
||||
dstream = ZSTD_initDStream(ZSTD_BTRFS_MAX_INPUT, workspace, wsize);
|
||||
if (!dstream) {
|
||||
printf("%s: ZSTD_initDStream failed\n", __func__);
|
||||
goto err_free;
|
||||
}
|
||||
|
||||
in_buf.src = cbuf;
|
||||
in_buf.pos = 0;
|
||||
in_buf.size = clen;
|
||||
|
||||
out_buf.dst = dbuf;
|
||||
out_buf.pos = 0;
|
||||
out_buf.size = dlen;
|
||||
|
||||
while (1) {
|
||||
size_t ret;
|
||||
|
||||
ret = ZSTD_decompressStream(dstream, &out_buf, &in_buf);
|
||||
if (ZSTD_isError(ret)) {
|
||||
printf("%s: ZSTD_decompressStream error %d\n", __func__,
|
||||
ZSTD_getErrorCode(ret));
|
||||
goto err_free;
|
||||
}
|
||||
|
||||
if (in_buf.pos >= clen || !ret)
|
||||
break;
|
||||
}
|
||||
|
||||
res = out_buf.pos;
|
||||
|
||||
err_free:
|
||||
free(workspace);
|
||||
return res;
|
||||
return zstd_decompress(&in, &out);
|
||||
}
|
||||
|
||||
u32 btrfs_decompress(u8 type, const char *c, u32 clen, char *d, u32 dlen)
|
||||
|
|
Loading…
Reference in a new issue