1 // SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause 2 /* 3 * Copyright (c) Facebook, Inc. 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 #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 27 static 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 57 int zstd_min_clevel(void) 58 { 59 return ZSTD_minCLevel(); 60 } 61 EXPORT_SYMBOL(zstd_min_clevel); 62 63 int zstd_max_clevel(void) 64 { 65 return ZSTD_maxCLevel(); 66 } 67 EXPORT_SYMBOL(zstd_max_clevel); 68 69 size_t zstd_compress_bound(size_t src_size) 70 { 71 return ZSTD_compressBound(src_size); 72 } 73 EXPORT_SYMBOL(zstd_compress_bound); 74 75 zstd_parameters zstd_get_params(int level, 76 unsigned long long estimated_src_size) 77 { 78 return ZSTD_getParams(level, estimated_src_size, 0); 79 } 80 EXPORT_SYMBOL(zstd_get_params); 81 82 size_t zstd_cctx_workspace_bound(const zstd_compression_parameters *cparams) 83 { 84 return ZSTD_estimateCCtxSize_usingCParams(*cparams); 85 } 86 EXPORT_SYMBOL(zstd_cctx_workspace_bound); 87 88 zstd_cctx *zstd_init_cctx(void *workspace, size_t workspace_size) 89 { 90 if (workspace == NULL) 91 return NULL; 92 return ZSTD_initStaticCCtx(workspace, workspace_size); 93 } 94 EXPORT_SYMBOL(zstd_init_cctx); 95 96 size_t zstd_compress_cctx(zstd_cctx *cctx, void *dst, size_t dst_capacity, 97 const void *src, size_t src_size, const zstd_parameters *parameters) 98 { 99 ZSTD_FORWARD_IF_ERR(zstd_cctx_init(cctx, parameters, src_size)); 100 return ZSTD_compress2(cctx, dst, dst_capacity, src, src_size); 101 } 102 EXPORT_SYMBOL(zstd_compress_cctx); 103 104 size_t zstd_cstream_workspace_bound(const zstd_compression_parameters *cparams) 105 { 106 return ZSTD_estimateCStreamSize_usingCParams(*cparams); 107 } 108 EXPORT_SYMBOL(zstd_cstream_workspace_bound); 109 110 zstd_cstream *zstd_init_cstream(const zstd_parameters *parameters, 111 unsigned long long pledged_src_size, void *workspace, size_t workspace_size) 112 { 113 zstd_cstream *cstream; 114 115 if (workspace == NULL) 116 return NULL; 117 118 cstream = ZSTD_initStaticCStream(workspace, workspace_size); 119 if (cstream == NULL) 120 return NULL; 121 122 /* 0 means unknown in linux zstd API but means 0 in new zstd API */ 123 if (pledged_src_size == 0) 124 pledged_src_size = ZSTD_CONTENTSIZE_UNKNOWN; 125 126 if (ZSTD_isError(zstd_cctx_init(cstream, parameters, pledged_src_size))) 127 return NULL; 128 129 return cstream; 130 } 131 EXPORT_SYMBOL(zstd_init_cstream); 132 133 size_t zstd_reset_cstream(zstd_cstream *cstream, 134 unsigned long long pledged_src_size) 135 { 136 if (pledged_src_size == 0) 137 pledged_src_size = ZSTD_CONTENTSIZE_UNKNOWN; 138 ZSTD_FORWARD_IF_ERR( ZSTD_CCtx_reset(cstream, ZSTD_reset_session_only) ); 139 ZSTD_FORWARD_IF_ERR( ZSTD_CCtx_setPledgedSrcSize(cstream, pledged_src_size) ); 140 return 0; 141 } 142 EXPORT_SYMBOL(zstd_reset_cstream); 143 144 size_t zstd_compress_stream(zstd_cstream *cstream, zstd_out_buffer *output, 145 zstd_in_buffer *input) 146 { 147 return ZSTD_compressStream(cstream, output, input); 148 } 149 EXPORT_SYMBOL(zstd_compress_stream); 150 151 size_t zstd_flush_stream(zstd_cstream *cstream, zstd_out_buffer *output) 152 { 153 return ZSTD_flushStream(cstream, output); 154 } 155 EXPORT_SYMBOL(zstd_flush_stream); 156 157 size_t zstd_end_stream(zstd_cstream *cstream, zstd_out_buffer *output) 158 { 159 return ZSTD_endStream(cstream, output); 160 } 161 EXPORT_SYMBOL(zstd_end_stream); 162 163 MODULE_LICENSE("Dual BSD/GPL"); 164 MODULE_DESCRIPTION("Zstd Compressor"); 165