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

Subsequent to commit 15f1b5e {"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]> Signed-off-by: Tashfin Shakeer Rhythm <[email protected]>
Atom-X-Devs · Mar 23, 2022 · 0ee0333 · 0ee0333
1 parent 67fff45
commit 0ee0333
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);