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