ddidderr/zstd-rs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor(cli): move compression status classification to Rust

Browse Source 
Move the aggregate compression-result classification out of the C wrapper and into the Rust fileio module. Keep C responsible for the user-facing EXM_THROW diagnostics and the existing fallback assertion, so optional-format build guards and command-line behavior remain unchanged. The Rust classifier also makes unexpected statuses explicit without widening the format callback boundary.

Test Plan: ulimit -v 41943040; CARGO_BUILD_JOBS=1 cargo clippy --manifest-path rust/Cargo.toml --all-targets -- -D warnings; cargo +nightly fmt --manifest-path rust/Cargo.toml --all -- --check; GCC and Clang syntax-only checks; make -j1 -C tests test (all 41 shell tests, fuzzer, zstd tester, and zstream tester passed).
This commit is contained in:
ddidderr
2026-07-20 17:59:38 +02:00
parent 0bd25053b7
commit bfd61e6ed7
2 changed files with 72 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
programs/fileio.c
+19 -6
View File
@@ -1070,6 +1070,17 @@ enum {
FIO_RUST_COMPRESS_ZSTD_UNSUPPORTED = 4
};
enum {
FIO_RUST_COMPRESS_DIAGNOSTIC_OK = 0,
FIO_RUST_COMPRESS_DIAGNOSTIC_GZIP_UNSUPPORTED = 1,
FIO_RUST_COMPRESS_DIAGNOSTIC_LZMA_UNSUPPORTED = 2,
FIO_RUST_COMPRESS_DIAGNOSTIC_LZ4_UNSUPPORTED = 3,
FIO_RUST_COMPRESS_DIAGNOSTIC_ZSTD_UNSUPPORTED = 4,
FIO_RUST_COMPRESS_DIAGNOSTIC_UNKNOWN = 5
};
int FIO_rust_compressFilenameDiagnostic(int status);
typedef unsigned long long (*FIO_rust_compress_zstd_fn)(
void* fCtx, void* prefs, void* ress,
const char* srcFileName, U64 srcFileSize,
@@ -3005,6 +3016,7 @@ FIO_compressFilename_internal(FIO_ctx_t* const fCtx,
U64 const fileSize = UTIL_getFileSize(srcFileName);
FIO_rust_compress_callbacks_t callbacks;
int status;
int diagnostic;
memset(&callbacks, 0, sizeof(callbacks));
callbacks.opaque = &displayCtx;
@@ -3025,19 +3037,20 @@ FIO_compressFilename_internal(FIO_ctx_t* const fCtx,
fCtx, prefs, &ress, dstFileName, srcFileName, fileSize,
compressionLevel, &callbacks);
switch (status) {
case FIO_RUST_COMPRESS_OK:
diagnostic = FIO_rust_compressFilenameDiagnostic(status);
switch (diagnostic) {
case FIO_RUST_COMPRESS_DIAGNOSTIC_OK:
return 0;
case FIO_RUST_COMPRESS_GZIP_UNSUPPORTED:
case FIO_RUST_COMPRESS_DIAGNOSTIC_GZIP_UNSUPPORTED:
EXM_THROW(20, "zstd: %s: file cannot be compressed as gzip (zstd compiled without ZSTD_GZCOMPRESS) -- ignored \n",
srcFileName);
case FIO_RUST_COMPRESS_LZMA_UNSUPPORTED:
case FIO_RUST_COMPRESS_DIAGNOSTIC_LZMA_UNSUPPORTED:
EXM_THROW(20, "zstd: %s: file cannot be compressed as xz/lzma (zstd compiled without ZSTD_LZMACOMPRESS) -- ignored \n",
srcFileName);
case FIO_RUST_COMPRESS_LZ4_UNSUPPORTED:
case FIO_RUST_COMPRESS_DIAGNOSTIC_LZ4_UNSUPPORTED:
EXM_THROW(20, "zstd: %s: file cannot be compressed as lz4 (zstd compiled without ZSTD_LZ4COMPRESS) -- ignored \n",
srcFileName);
case FIO_RUST_COMPRESS_ZSTD_UNSUPPORTED:
case FIO_RUST_COMPRESS_DIAGNOSTIC_ZSTD_UNSUPPORTED:
default:
assert(status == FIO_RUST_COMPRESS_ZSTD_UNSUPPORTED);
EXM_THROW(20, "zstd: %s: file cannot be compressed as zstd -- ignored \n",