1e0c1b49fSNick Terrell // SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause
2e0c1b49fSNick Terrell /*
3*65d1f550SNick Terrell * Copyright (c) Meta Platforms, Inc. and affiliates.
4e0c1b49fSNick Terrell * All rights reserved.
5e0c1b49fSNick Terrell *
6e0c1b49fSNick Terrell * This source code is licensed under both the BSD-style license (found in the
7e0c1b49fSNick Terrell * LICENSE file in the root directory of this source tree) and the GPLv2 (found
8e0c1b49fSNick Terrell * in the COPYING file in the root directory of this source tree).
9e0c1b49fSNick Terrell * You may select, at your option, one of the above-listed licenses.
10e0c1b49fSNick Terrell */
11e0c1b49fSNick Terrell
12e0c1b49fSNick Terrell #include <linux/kernel.h>
13e0c1b49fSNick Terrell #include <linux/module.h>
14e0c1b49fSNick Terrell #include <linux/string.h>
15e0c1b49fSNick Terrell #include <linux/zstd.h>
16e0c1b49fSNick Terrell
17e0c1b49fSNick Terrell #include "common/zstd_deps.h"
18e0c1b49fSNick Terrell
19e0c1b49fSNick Terrell /* Common symbols. zstd_compress must depend on zstd_decompress. */
20e0c1b49fSNick Terrell
zstd_is_error(size_t code)21e0c1b49fSNick Terrell unsigned int zstd_is_error(size_t code)
22e0c1b49fSNick Terrell {
23e0c1b49fSNick Terrell return ZSTD_isError(code);
24e0c1b49fSNick Terrell }
25e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_is_error);
26e0c1b49fSNick Terrell
zstd_get_error_code(size_t code)27e0c1b49fSNick Terrell zstd_error_code zstd_get_error_code(size_t code)
28e0c1b49fSNick Terrell {
29e0c1b49fSNick Terrell return ZSTD_getErrorCode(code);
30e0c1b49fSNick Terrell }
31e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_get_error_code);
32e0c1b49fSNick Terrell
zstd_get_error_name(size_t code)33e0c1b49fSNick Terrell const char *zstd_get_error_name(size_t code)
34e0c1b49fSNick Terrell {
35e0c1b49fSNick Terrell return ZSTD_getErrorName(code);
36e0c1b49fSNick Terrell }
37e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_get_error_name);
38e0c1b49fSNick Terrell
39e0c1b49fSNick Terrell /* Decompression symbols. */
40e0c1b49fSNick Terrell
zstd_dctx_workspace_bound(void)41e0c1b49fSNick Terrell size_t zstd_dctx_workspace_bound(void)
42e0c1b49fSNick Terrell {
43e0c1b49fSNick Terrell return ZSTD_estimateDCtxSize();
44e0c1b49fSNick Terrell }
45e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_dctx_workspace_bound);
46e0c1b49fSNick Terrell
zstd_create_dctx_advanced(zstd_custom_mem custom_mem)474fc41879SSergey Senozhatsky zstd_dctx *zstd_create_dctx_advanced(zstd_custom_mem custom_mem)
484fc41879SSergey Senozhatsky {
494fc41879SSergey Senozhatsky return ZSTD_createDCtx_advanced(custom_mem);
504fc41879SSergey Senozhatsky }
514fc41879SSergey Senozhatsky EXPORT_SYMBOL(zstd_create_dctx_advanced);
524fc41879SSergey Senozhatsky
zstd_free_dctx(zstd_dctx * dctx)534fc41879SSergey Senozhatsky size_t zstd_free_dctx(zstd_dctx *dctx)
544fc41879SSergey Senozhatsky {
554fc41879SSergey Senozhatsky return ZSTD_freeDCtx(dctx);
564fc41879SSergey Senozhatsky }
574fc41879SSergey Senozhatsky EXPORT_SYMBOL(zstd_free_dctx);
584fc41879SSergey Senozhatsky
zstd_create_ddict_byreference(const void * dict,size_t dict_size,zstd_custom_mem custom_mem)594fc41879SSergey Senozhatsky zstd_ddict *zstd_create_ddict_byreference(const void *dict, size_t dict_size,
604fc41879SSergey Senozhatsky zstd_custom_mem custom_mem)
614fc41879SSergey Senozhatsky {
624fc41879SSergey Senozhatsky return ZSTD_createDDict_advanced(dict, dict_size, ZSTD_dlm_byRef,
634fc41879SSergey Senozhatsky ZSTD_dct_auto, custom_mem);
644fc41879SSergey Senozhatsky
654fc41879SSergey Senozhatsky }
664fc41879SSergey Senozhatsky EXPORT_SYMBOL(zstd_create_ddict_byreference);
674fc41879SSergey Senozhatsky
zstd_free_ddict(zstd_ddict * ddict)684fc41879SSergey Senozhatsky size_t zstd_free_ddict(zstd_ddict *ddict)
694fc41879SSergey Senozhatsky {
704fc41879SSergey Senozhatsky return ZSTD_freeDDict(ddict);
714fc41879SSergey Senozhatsky }
724fc41879SSergey Senozhatsky EXPORT_SYMBOL(zstd_free_ddict);
734fc41879SSergey Senozhatsky
zstd_init_dctx(void * workspace,size_t workspace_size)74e0c1b49fSNick Terrell zstd_dctx *zstd_init_dctx(void *workspace, size_t workspace_size)
75e0c1b49fSNick Terrell {
76e0c1b49fSNick Terrell if (workspace == NULL)
77e0c1b49fSNick Terrell return NULL;
78e0c1b49fSNick Terrell return ZSTD_initStaticDCtx(workspace, workspace_size);
79e0c1b49fSNick Terrell }
80e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_init_dctx);
81e0c1b49fSNick Terrell
zstd_decompress_dctx(zstd_dctx * dctx,void * dst,size_t dst_capacity,const void * src,size_t src_size)82e0c1b49fSNick Terrell size_t zstd_decompress_dctx(zstd_dctx *dctx, void *dst, size_t dst_capacity,
83e0c1b49fSNick Terrell const void *src, size_t src_size)
84e0c1b49fSNick Terrell {
85e0c1b49fSNick Terrell return ZSTD_decompressDCtx(dctx, dst, dst_capacity, src, src_size);
86e0c1b49fSNick Terrell }
87e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_decompress_dctx);
88e0c1b49fSNick Terrell
zstd_decompress_using_ddict(zstd_dctx * dctx,void * dst,size_t dst_capacity,const void * src,size_t src_size,const zstd_ddict * ddict)894fc41879SSergey Senozhatsky size_t zstd_decompress_using_ddict(zstd_dctx *dctx,
904fc41879SSergey Senozhatsky void *dst, size_t dst_capacity, const void* src, size_t src_size,
914fc41879SSergey Senozhatsky const zstd_ddict* ddict)
924fc41879SSergey Senozhatsky {
934fc41879SSergey Senozhatsky return ZSTD_decompress_usingDDict(dctx, dst, dst_capacity, src,
944fc41879SSergey Senozhatsky src_size, ddict);
954fc41879SSergey Senozhatsky }
964fc41879SSergey Senozhatsky EXPORT_SYMBOL(zstd_decompress_using_ddict);
974fc41879SSergey Senozhatsky
zstd_dstream_workspace_bound(size_t max_window_size)98e0c1b49fSNick Terrell size_t zstd_dstream_workspace_bound(size_t max_window_size)
99e0c1b49fSNick Terrell {
100e0c1b49fSNick Terrell return ZSTD_estimateDStreamSize(max_window_size);
101e0c1b49fSNick Terrell }
102e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_dstream_workspace_bound);
103e0c1b49fSNick Terrell
zstd_init_dstream(size_t max_window_size,void * workspace,size_t workspace_size)104e0c1b49fSNick Terrell zstd_dstream *zstd_init_dstream(size_t max_window_size, void *workspace,
105e0c1b49fSNick Terrell size_t workspace_size)
106e0c1b49fSNick Terrell {
107e0c1b49fSNick Terrell if (workspace == NULL)
108e0c1b49fSNick Terrell return NULL;
109e0c1b49fSNick Terrell (void)max_window_size;
110e0c1b49fSNick Terrell return ZSTD_initStaticDStream(workspace, workspace_size);
111e0c1b49fSNick Terrell }
112e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_init_dstream);
113e0c1b49fSNick Terrell
zstd_reset_dstream(zstd_dstream * dstream)114e0c1b49fSNick Terrell size_t zstd_reset_dstream(zstd_dstream *dstream)
115e0c1b49fSNick Terrell {
116*65d1f550SNick Terrell return ZSTD_DCtx_reset(dstream, ZSTD_reset_session_only);
117e0c1b49fSNick Terrell }
118e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_reset_dstream);
119e0c1b49fSNick Terrell
zstd_decompress_stream(zstd_dstream * dstream,zstd_out_buffer * output,zstd_in_buffer * input)120e0c1b49fSNick Terrell size_t zstd_decompress_stream(zstd_dstream *dstream, zstd_out_buffer *output,
121e0c1b49fSNick Terrell zstd_in_buffer *input)
122e0c1b49fSNick Terrell {
123e0c1b49fSNick Terrell return ZSTD_decompressStream(dstream, output, input);
124e0c1b49fSNick Terrell }
125e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_decompress_stream);
126e0c1b49fSNick Terrell
zstd_find_frame_compressed_size(const void * src,size_t src_size)127e0c1b49fSNick Terrell size_t zstd_find_frame_compressed_size(const void *src, size_t src_size)
128e0c1b49fSNick Terrell {
129e0c1b49fSNick Terrell return ZSTD_findFrameCompressedSize(src, src_size);
130e0c1b49fSNick Terrell }
131e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_find_frame_compressed_size);
132e0c1b49fSNick Terrell
zstd_get_frame_header(zstd_frame_header * header,const void * src,size_t src_size)133e0c1b49fSNick Terrell size_t zstd_get_frame_header(zstd_frame_header *header, const void *src,
134e0c1b49fSNick Terrell size_t src_size)
135e0c1b49fSNick Terrell {
136e0c1b49fSNick Terrell return ZSTD_getFrameHeader(header, src, src_size);
137e0c1b49fSNick Terrell }
138e0c1b49fSNick Terrell EXPORT_SYMBOL(zstd_get_frame_header);
139e0c1b49fSNick Terrell
140e0c1b49fSNick Terrell MODULE_LICENSE("Dual BSD/GPL");
141e0c1b49fSNick Terrell MODULE_DESCRIPTION("Zstd Decompressor");
142