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