1// SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause
2/*
3 * Copyright (c) Meta Platforms, Inc. and affiliates.
4 * All rights reserved.
5 *
6 * This source code is licensed under both the BSD-style license (found in the
7 * LICENSE file in the root directory of this source tree) and the GPLv2 (found
8 * in the COPYING file in the root directory of this source tree).
9 * You may select, at your option, one of the above-listed licenses.
10 */
11
12#include <linux/kernel.h>
13#include <linux/module.h>
14#include <linux/string.h>
15#include <linux/zstd.h>
16
17#include "common/zstd_deps.h"
18
19/* Common symbols. zstd_compress must depend on zstd_decompress. */
20
21unsigned int zstd_is_error(size_t code)
22{
23 return ZSTD_isError(result: code);
24}
25EXPORT_SYMBOL(zstd_is_error);
26
27zstd_error_code zstd_get_error_code(size_t code)
28{
29 return ZSTD_getErrorCode(functionResult: code);
30}
31EXPORT_SYMBOL(zstd_get_error_code);
32
33const char *zstd_get_error_name(size_t code)
34{
35 return ZSTD_getErrorName(result: code);
36}
37EXPORT_SYMBOL(zstd_get_error_name);
38
39/* Decompression symbols. */
40
41size_t zstd_dctx_workspace_bound(void)
42{
43 return ZSTD_estimateDCtxSize();
44}
45EXPORT_SYMBOL(zstd_dctx_workspace_bound);
46
47zstd_dctx *zstd_create_dctx_advanced(zstd_custom_mem custom_mem)
48{
49 return ZSTD_createDCtx_advanced(customMem: custom_mem);
50}
51EXPORT_SYMBOL(zstd_create_dctx_advanced);
52
53size_t zstd_free_dctx(zstd_dctx *dctx)
54{
55 return ZSTD_freeDCtx(dctx);
56}
57EXPORT_SYMBOL(zstd_free_dctx);
58
59zstd_ddict *zstd_create_ddict_byreference(const void *dict, size_t dict_size,
60 zstd_custom_mem custom_mem)
61{
62 return ZSTD_createDDict_advanced(dict, dictSize: dict_size, dictLoadMethod: ZSTD_dlm_byRef,
63 dictContentType: ZSTD_dct_auto, customMem: custom_mem);
64
65}
66EXPORT_SYMBOL(zstd_create_ddict_byreference);
67
68size_t zstd_free_ddict(zstd_ddict *ddict)
69{
70 return ZSTD_freeDDict(ddict);
71}
72EXPORT_SYMBOL(zstd_free_ddict);
73
74zstd_dctx *zstd_init_dctx(void *workspace, size_t workspace_size)
75{
76 if (workspace == NULL)
77 return NULL;
78 return ZSTD_initStaticDCtx(workspace, workspaceSize: workspace_size);
79}
80EXPORT_SYMBOL(zstd_init_dctx);
81
82size_t zstd_decompress_dctx(zstd_dctx *dctx, void *dst, size_t dst_capacity,
83 const void *src, size_t src_size)
84{
85 return ZSTD_decompressDCtx(dctx, dst, dstCapacity: dst_capacity, src, srcSize: src_size);
86}
87EXPORT_SYMBOL(zstd_decompress_dctx);
88
89size_t zstd_decompress_using_ddict(zstd_dctx *dctx,
90 void *dst, size_t dst_capacity, const void* src, size_t src_size,
91 const zstd_ddict* ddict)
92{
93 return ZSTD_decompress_usingDDict(dctx, dst, dstCapacity: dst_capacity, src,
94 srcSize: src_size, ddict);
95}
96EXPORT_SYMBOL(zstd_decompress_using_ddict);
97
98size_t zstd_dstream_workspace_bound(size_t max_window_size)
99{
100 return ZSTD_estimateDStreamSize(maxWindowSize: max_window_size);
101}
102EXPORT_SYMBOL(zstd_dstream_workspace_bound);
103
104zstd_dstream *zstd_init_dstream(size_t max_window_size, void *workspace,
105 size_t workspace_size)
106{
107 if (workspace == NULL)
108 return NULL;
109 (void)max_window_size;
110 return ZSTD_initStaticDStream(workspace, workspaceSize: workspace_size);
111}
112EXPORT_SYMBOL(zstd_init_dstream);
113
114size_t zstd_reset_dstream(zstd_dstream *dstream)
115{
116 return ZSTD_DCtx_reset(dctx: dstream, reset: ZSTD_reset_session_only);
117}
118EXPORT_SYMBOL(zstd_reset_dstream);
119
120size_t zstd_decompress_stream(zstd_dstream *dstream, zstd_out_buffer *output,
121 zstd_in_buffer *input)
122{
123 return ZSTD_decompressStream(zds: dstream, output, input);
124}
125EXPORT_SYMBOL(zstd_decompress_stream);
126
127size_t zstd_find_frame_compressed_size(const void *src, size_t src_size)
128{
129 return ZSTD_findFrameCompressedSize(src, srcSize: src_size);
130}
131EXPORT_SYMBOL(zstd_find_frame_compressed_size);
132
133size_t zstd_get_frame_header(zstd_frame_header *header, const void *src,
134 size_t src_size)
135{
136 return ZSTD_getFrameHeader(zfhPtr: header, src, srcSize: src_size);
137}
138EXPORT_SYMBOL(zstd_get_frame_header);
139
140MODULE_LICENSE("Dual BSD/GPL");
141MODULE_DESCRIPTION("Zstd Decompressor");
142