214 lines · c
1// SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause2/*3 * Copyright (c) Facebook, Inc.4 * All rights reserved.5 *6 * This source code is licensed under both the BSD-style license (found in the7 * LICENSE file in the root directory of this source tree) and the GPLv2 (found8 * 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#include "common/zstd_internal.h"19 20#define ZSTD_FORWARD_IF_ERR(ret) \21 do { \22 size_t const __ret = (ret); \23 if (ZSTD_isError(__ret)) \24 return __ret; \25 } while (0)26 27static size_t zstd_cctx_init(zstd_cctx *cctx, const zstd_parameters *parameters,28 unsigned long long pledged_src_size)29{30 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_reset(31 cctx, ZSTD_reset_session_and_parameters));32 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setPledgedSrcSize(33 cctx, pledged_src_size));34 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setParameter(35 cctx, ZSTD_c_windowLog, parameters->cParams.windowLog));36 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setParameter(37 cctx, ZSTD_c_hashLog, parameters->cParams.hashLog));38 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setParameter(39 cctx, ZSTD_c_chainLog, parameters->cParams.chainLog));40 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setParameter(41 cctx, ZSTD_c_searchLog, parameters->cParams.searchLog));42 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setParameter(43 cctx, ZSTD_c_minMatch, parameters->cParams.minMatch));44 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setParameter(45 cctx, ZSTD_c_targetLength, parameters->cParams.targetLength));46 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setParameter(47 cctx, ZSTD_c_strategy, parameters->cParams.strategy));48 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setParameter(49 cctx, ZSTD_c_contentSizeFlag, parameters->fParams.contentSizeFlag));50 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setParameter(51 cctx, ZSTD_c_checksumFlag, parameters->fParams.checksumFlag));52 ZSTD_FORWARD_IF_ERR(ZSTD_CCtx_setParameter(53 cctx, ZSTD_c_dictIDFlag, !parameters->fParams.noDictIDFlag));54 return 0;55}56 57int zstd_min_clevel(void)58{59 return ZSTD_minCLevel();60}61EXPORT_SYMBOL(zstd_min_clevel);62 63int zstd_max_clevel(void)64{65 return ZSTD_maxCLevel();66}67EXPORT_SYMBOL(zstd_max_clevel);68 69int zstd_default_clevel(void)70{71 return ZSTD_defaultCLevel();72}73EXPORT_SYMBOL(zstd_default_clevel);74 75size_t zstd_compress_bound(size_t src_size)76{77 return ZSTD_compressBound(src_size);78}79EXPORT_SYMBOL(zstd_compress_bound);80 81zstd_parameters zstd_get_params(int level,82 unsigned long long estimated_src_size)83{84 return ZSTD_getParams(level, estimated_src_size, 0);85}86EXPORT_SYMBOL(zstd_get_params);87 88zstd_compression_parameters zstd_get_cparams(int level,89 unsigned long long estimated_src_size, size_t dict_size)90{91 return ZSTD_getCParams(level, estimated_src_size, dict_size);92}93EXPORT_SYMBOL(zstd_get_cparams);94 95size_t zstd_cctx_workspace_bound(const zstd_compression_parameters *cparams)96{97 return ZSTD_estimateCCtxSize_usingCParams(*cparams);98}99EXPORT_SYMBOL(zstd_cctx_workspace_bound);100 101zstd_cctx *zstd_init_cctx(void *workspace, size_t workspace_size)102{103 if (workspace == NULL)104 return NULL;105 return ZSTD_initStaticCCtx(workspace, workspace_size);106}107EXPORT_SYMBOL(zstd_init_cctx);108 109zstd_cctx *zstd_create_cctx_advanced(zstd_custom_mem custom_mem)110{111 return ZSTD_createCCtx_advanced(custom_mem);112}113EXPORT_SYMBOL(zstd_create_cctx_advanced);114 115size_t zstd_free_cctx(zstd_cctx *cctx)116{117 return ZSTD_freeCCtx(cctx);118}119EXPORT_SYMBOL(zstd_free_cctx);120 121zstd_cdict *zstd_create_cdict_byreference(const void *dict, size_t dict_size,122 zstd_compression_parameters cparams,123 zstd_custom_mem custom_mem)124{125 return ZSTD_createCDict_advanced(dict, dict_size, ZSTD_dlm_byRef,126 ZSTD_dct_auto, cparams, custom_mem);127}128EXPORT_SYMBOL(zstd_create_cdict_byreference);129 130size_t zstd_free_cdict(zstd_cdict *cdict)131{132 return ZSTD_freeCDict(cdict);133}134EXPORT_SYMBOL(zstd_free_cdict);135 136size_t zstd_compress_cctx(zstd_cctx *cctx, void *dst, size_t dst_capacity,137 const void *src, size_t src_size, const zstd_parameters *parameters)138{139 ZSTD_FORWARD_IF_ERR(zstd_cctx_init(cctx, parameters, src_size));140 return ZSTD_compress2(cctx, dst, dst_capacity, src, src_size);141}142EXPORT_SYMBOL(zstd_compress_cctx);143 144size_t zstd_compress_using_cdict(zstd_cctx *cctx, void *dst,145 size_t dst_capacity, const void *src, size_t src_size,146 const ZSTD_CDict *cdict)147{148 return ZSTD_compress_usingCDict(cctx, dst, dst_capacity,149 src, src_size, cdict);150}151EXPORT_SYMBOL(zstd_compress_using_cdict);152 153size_t zstd_cstream_workspace_bound(const zstd_compression_parameters *cparams)154{155 return ZSTD_estimateCStreamSize_usingCParams(*cparams);156}157EXPORT_SYMBOL(zstd_cstream_workspace_bound);158 159zstd_cstream *zstd_init_cstream(const zstd_parameters *parameters,160 unsigned long long pledged_src_size, void *workspace, size_t workspace_size)161{162 zstd_cstream *cstream;163 164 if (workspace == NULL)165 return NULL;166 167 cstream = ZSTD_initStaticCStream(workspace, workspace_size);168 if (cstream == NULL)169 return NULL;170 171 /* 0 means unknown in linux zstd API but means 0 in new zstd API */172 if (pledged_src_size == 0)173 pledged_src_size = ZSTD_CONTENTSIZE_UNKNOWN;174 175 if (ZSTD_isError(zstd_cctx_init(cstream, parameters, pledged_src_size)))176 return NULL;177 178 return cstream;179}180EXPORT_SYMBOL(zstd_init_cstream);181 182size_t zstd_reset_cstream(zstd_cstream *cstream,183 unsigned long long pledged_src_size)184{185 if (pledged_src_size == 0)186 pledged_src_size = ZSTD_CONTENTSIZE_UNKNOWN;187 ZSTD_FORWARD_IF_ERR( ZSTD_CCtx_reset(cstream, ZSTD_reset_session_only) );188 ZSTD_FORWARD_IF_ERR( ZSTD_CCtx_setPledgedSrcSize(cstream, pledged_src_size) );189 return 0;190}191EXPORT_SYMBOL(zstd_reset_cstream);192 193size_t zstd_compress_stream(zstd_cstream *cstream, zstd_out_buffer *output,194 zstd_in_buffer *input)195{196 return ZSTD_compressStream(cstream, output, input);197}198EXPORT_SYMBOL(zstd_compress_stream);199 200size_t zstd_flush_stream(zstd_cstream *cstream, zstd_out_buffer *output)201{202 return ZSTD_flushStream(cstream, output);203}204EXPORT_SYMBOL(zstd_flush_stream);205 206size_t zstd_end_stream(zstd_cstream *cstream, zstd_out_buffer *output)207{208 return ZSTD_endStream(cstream, output);209}210EXPORT_SYMBOL(zstd_end_stream);211 212MODULE_LICENSE("Dual BSD/GPL");213MODULE_DESCRIPTION("Zstd Compressor");214