1*01826a49SYabin Cui // SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause
2*01826a49SYabin Cui /*
3*01826a49SYabin Cui * Copyright (c) Meta Platforms, Inc. and affiliates.
4*01826a49SYabin Cui * All rights reserved.
5*01826a49SYabin Cui *
6*01826a49SYabin Cui * This source code is licensed under both the BSD-style license (found in the
7*01826a49SYabin Cui * LICENSE file in the root directory of this source tree) and the GPLv2 (found
8*01826a49SYabin Cui * in the COPYING file in the root directory of this source tree).
9*01826a49SYabin Cui * You may select, at your option, one of the above-listed licenses.
10*01826a49SYabin Cui */
11*01826a49SYabin Cui
12*01826a49SYabin Cui #include <linux/kernel.h>
13*01826a49SYabin Cui #include <linux/module.h>
14*01826a49SYabin Cui #include <linux/string.h>
15*01826a49SYabin Cui #include <linux/zstd.h>
16*01826a49SYabin Cui
17*01826a49SYabin Cui #include "common/zstd_deps.h"
18*01826a49SYabin Cui
19*01826a49SYabin Cui /* Common symbols. zstd_compress must depend on zstd_decompress. */
20*01826a49SYabin Cui
zstd_is_error(size_t code)21*01826a49SYabin Cui unsigned int zstd_is_error(size_t code)
22*01826a49SYabin Cui {
23*01826a49SYabin Cui return ZSTD_isError(code);
24*01826a49SYabin Cui }
25*01826a49SYabin Cui EXPORT_SYMBOL(zstd_is_error);
26*01826a49SYabin Cui
zstd_get_error_code(size_t code)27*01826a49SYabin Cui zstd_error_code zstd_get_error_code(size_t code)
28*01826a49SYabin Cui {
29*01826a49SYabin Cui return ZSTD_getErrorCode(code);
30*01826a49SYabin Cui }
31*01826a49SYabin Cui EXPORT_SYMBOL(zstd_get_error_code);
32*01826a49SYabin Cui
zstd_get_error_name(size_t code)33*01826a49SYabin Cui const char *zstd_get_error_name(size_t code)
34*01826a49SYabin Cui {
35*01826a49SYabin Cui return ZSTD_getErrorName(code);
36*01826a49SYabin Cui }
37*01826a49SYabin Cui EXPORT_SYMBOL(zstd_get_error_name);
38*01826a49SYabin Cui
39*01826a49SYabin Cui /* Decompression symbols. */
40*01826a49SYabin Cui
zstd_dctx_workspace_bound(void)41*01826a49SYabin Cui size_t zstd_dctx_workspace_bound(void)
42*01826a49SYabin Cui {
43*01826a49SYabin Cui return ZSTD_estimateDCtxSize();
44*01826a49SYabin Cui }
45*01826a49SYabin Cui EXPORT_SYMBOL(zstd_dctx_workspace_bound);
46*01826a49SYabin Cui
zstd_init_dctx(void * workspace,size_t workspace_size)47*01826a49SYabin Cui zstd_dctx *zstd_init_dctx(void *workspace, size_t workspace_size)
48*01826a49SYabin Cui {
49*01826a49SYabin Cui if (workspace == NULL)
50*01826a49SYabin Cui return NULL;
51*01826a49SYabin Cui return ZSTD_initStaticDCtx(workspace, workspace_size);
52*01826a49SYabin Cui }
53*01826a49SYabin Cui EXPORT_SYMBOL(zstd_init_dctx);
54*01826a49SYabin Cui
zstd_decompress_dctx(zstd_dctx * dctx,void * dst,size_t dst_capacity,const void * src,size_t src_size)55*01826a49SYabin Cui size_t zstd_decompress_dctx(zstd_dctx *dctx, void *dst, size_t dst_capacity,
56*01826a49SYabin Cui const void *src, size_t src_size)
57*01826a49SYabin Cui {
58*01826a49SYabin Cui return ZSTD_decompressDCtx(dctx, dst, dst_capacity, src, src_size);
59*01826a49SYabin Cui }
60*01826a49SYabin Cui EXPORT_SYMBOL(zstd_decompress_dctx);
61*01826a49SYabin Cui
zstd_dstream_workspace_bound(size_t max_window_size)62*01826a49SYabin Cui size_t zstd_dstream_workspace_bound(size_t max_window_size)
63*01826a49SYabin Cui {
64*01826a49SYabin Cui return ZSTD_estimateDStreamSize(max_window_size);
65*01826a49SYabin Cui }
66*01826a49SYabin Cui EXPORT_SYMBOL(zstd_dstream_workspace_bound);
67*01826a49SYabin Cui
zstd_init_dstream(size_t max_window_size,void * workspace,size_t workspace_size)68*01826a49SYabin Cui zstd_dstream *zstd_init_dstream(size_t max_window_size, void *workspace,
69*01826a49SYabin Cui size_t workspace_size)
70*01826a49SYabin Cui {
71*01826a49SYabin Cui if (workspace == NULL)
72*01826a49SYabin Cui return NULL;
73*01826a49SYabin Cui (void)max_window_size;
74*01826a49SYabin Cui return ZSTD_initStaticDStream(workspace, workspace_size);
75*01826a49SYabin Cui }
76*01826a49SYabin Cui EXPORT_SYMBOL(zstd_init_dstream);
77*01826a49SYabin Cui
zstd_reset_dstream(zstd_dstream * dstream)78*01826a49SYabin Cui size_t zstd_reset_dstream(zstd_dstream *dstream)
79*01826a49SYabin Cui {
80*01826a49SYabin Cui return ZSTD_DCtx_reset(dstream, ZSTD_reset_session_only);
81*01826a49SYabin Cui }
82*01826a49SYabin Cui EXPORT_SYMBOL(zstd_reset_dstream);
83*01826a49SYabin Cui
zstd_decompress_stream(zstd_dstream * dstream,zstd_out_buffer * output,zstd_in_buffer * input)84*01826a49SYabin Cui size_t zstd_decompress_stream(zstd_dstream *dstream, zstd_out_buffer *output,
85*01826a49SYabin Cui zstd_in_buffer *input)
86*01826a49SYabin Cui {
87*01826a49SYabin Cui return ZSTD_decompressStream(dstream, output, input);
88*01826a49SYabin Cui }
89*01826a49SYabin Cui EXPORT_SYMBOL(zstd_decompress_stream);
90*01826a49SYabin Cui
zstd_find_frame_compressed_size(const void * src,size_t src_size)91*01826a49SYabin Cui size_t zstd_find_frame_compressed_size(const void *src, size_t src_size)
92*01826a49SYabin Cui {
93*01826a49SYabin Cui return ZSTD_findFrameCompressedSize(src, src_size);
94*01826a49SYabin Cui }
95*01826a49SYabin Cui EXPORT_SYMBOL(zstd_find_frame_compressed_size);
96*01826a49SYabin Cui
zstd_get_frame_header(zstd_frame_header * header,const void * src,size_t src_size)97*01826a49SYabin Cui size_t zstd_get_frame_header(zstd_frame_header *header, const void *src,
98*01826a49SYabin Cui size_t src_size)
99*01826a49SYabin Cui {
100*01826a49SYabin Cui return ZSTD_getFrameHeader(header, src, src_size);
101*01826a49SYabin Cui }
102*01826a49SYabin Cui EXPORT_SYMBOL(zstd_get_frame_header);
103*01826a49SYabin Cui
104*01826a49SYabin Cui MODULE_LICENSE("Dual BSD/GPL");
105*01826a49SYabin Cui MODULE_DESCRIPTION("Zstd Decompressor");
106