lib: zstd: Revise the warning fix at zstd_reset_cstream()

Subsequent to 15f1b5e137 {"lib: zstd: Fix a warning in zstd_reset_cstream()"} - As referenced by Nick Terrelln ~ the ZSTD maintainer in the linux kernel [1], making zstd_reset_cstream() functionally identical to ZSTD_resetCStream() would be the perfect way to fix the warning without touching any core functions or breaking other parts of the code. [1]: facebook/zstd#3088 (comment) Suggested-by: Nick Terrell <[email protected]> Signed-off-by: Cyber Knight <[email protected]>
Stratosphere-Kernel · Mar 25, 2022 · add7b85 · add7b85
1 parent 34f7dc5
commit add7b85
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/zstd/zstd_compress_module.c b/lib/zstd/zstd_compress_module.c
@@ -133,7 +133,11 @@ EXPORT_SYMBOL(zstd_init_cstream);
 size_t zstd_reset_cstream(zstd_cstream *cstream,
 	unsigned long long pledged_src_size)
 {
-        return ZSTD_CCtx_reset(cstream, pledged_src_size);
+	if (pledged_src_size == 0)
+		pledged_src_size = ZSTD_CONTENTSIZE_UNKNOWN;
+	ZSTD_FORWARD_IF_ERR( ZSTD_CCtx_reset(cstream, ZSTD_reset_session_only) );
+	ZSTD_FORWARD_IF_ERR( ZSTD_CCtx_setPledgedSrcSize(cstream, pledged_src_size) );
+	return 0;
 }
 EXPORT_SYMBOL(zstd_reset_cstream);