add an example for compression (for benchmarking)

Author: folkertdev

fuzz/Cargo.lock

@@ -12,6 +12,7 @@ fuzz_target!(|data: String| {
             &mut length,
             data.as_ptr().cast(),
             data.len() as _,
+            9,
         )
     };
 

fuzz/fuzz_targets/compress.rs

@@ -32,6 +32,7 @@ fuzz_target!(|data: String| {
             &mut length,
             data.as_ptr().cast(),
             data.len() as _,
+            9,
         )
     };
 

fuzz/fuzz_targets/decompress.rs

@@ -0,0 +1,61 @@
+use core::ffi::c_uint;
+
+use test_libbz2_rs_sys::{compress_c, compress_rs};
+
+fn main() {
+    let mut it = std::env::args();
+
+    let _ = it.next().unwrap();
+
+    match it.next().unwrap().as_str() {
+        "c" => {
+            let level: i32 = it.next().unwrap().parse().unwrap();
+
+            let path = it.next().unwrap();
+            let input = std::fs::read(&path).unwrap();
+
+            let mut dest_vec = vec![0u8; 1 << 28];
+
+            let mut dest_len = dest_vec.len() as c_uint;
+            let dest = dest_vec.as_mut_ptr();
+
+            let source = input.as_ptr();
+            let source_len = input.len() as _;
+
+            let err = unsafe { compress_c(dest, &mut dest_len, source, source_len, level) };
+
+            if err != 0 {
+                panic!("error {err}");
+            }
+
+            dest_vec.truncate(dest_len as usize);
+
+            drop(dest_vec)
+        }
+        "rs" => {
+            let level: i32 = it.next().unwrap().parse().unwrap();
+
+            let path = it.next().unwrap();
+            let input = std::fs::read(&path).unwrap();
+
+            let mut dest_vec = vec![0u8; 1 << 28];
+
+            let mut dest_len = dest_vec.len() as std::ffi::c_uint;
+            let dest = dest_vec.as_mut_ptr();
+
+            let source = input.as_ptr();
+            let source_len = input.len() as _;
+
+            let err = unsafe { compress_rs(dest, &mut dest_len, source, source_len, level) };
+
+            if err != 0 {
+                panic!("error {err}");
+            }
+
+            dest_vec.truncate(dest_len as usize);
+
+            drop(dest_vec)
+        }
+        other => panic!("invalid option '{other}', expected one of 'c' or 'rs'"),
+    }
+}

test-libbz2-rs-sys/examples/compress.rs

@@ -141,6 +141,7 @@ fn compress_chunked_input() {
                 &mut dest_len,
                 SAMPLE1_REF.as_ptr(),
                 SAMPLE1_REF.len() as _,
+                9,
             )
         };
         assert_eq!(err, 0);
@@ -293,6 +294,7 @@ fn compress_chunked_output() {
                 &mut dest_len,
                 SAMPLE1_REF.as_ptr(),
                 SAMPLE1_REF.len() as _,
+                9,
             )
         };
         assert_eq!(err, 0);

test-libbz2-rs-sys/src/chunked.rs

@@ -83,6 +83,7 @@ macro_rules! assert_eq_compress {
                 &mut dest_len,
                 input.as_ptr(),
                 input.len() as core::ffi::c_uint,
+                9,
             );
 
             dest.truncate(dest_len as usize);
@@ -375,6 +376,7 @@ pub unsafe fn compress_c(
     dest_len: *mut libc::c_uint,
     source: *const u8,
     source_len: libc::c_uint,
+    blockSize100k: i32,
 ) -> i32 {
     use bzip2_sys::*;
 
@@ -400,7 +402,7 @@ pub unsafe fn compress_c(
     strm.bzfree = None;
     strm.opaque = std::ptr::null_mut::<libc::c_void>();
     unsafe {
-        ret = BZ2_bzCompressInit(&mut strm, 9, 0, 30);
+        ret = BZ2_bzCompressInit(&mut strm, blockSize100k, 0, 30);
         if ret != 0 as libc::c_int {
             return ret;
         }
@@ -428,6 +430,7 @@ pub unsafe fn compress_rs(
     dest_len: *mut libc::c_uint,
     source: *const u8,
     source_len: libc::c_uint,
+    blockSize100k: i32,
 ) -> i32 {
     use libbz2_rs_sys::*;
 
@@ -453,7 +456,7 @@ pub unsafe fn compress_rs(
     strm.bzfree = None;
     strm.opaque = std::ptr::null_mut::<libc::c_void>();
     unsafe {
-        ret = BZ2_bzCompressInit(&mut strm, 9, 0, 30);
+        ret = BZ2_bzCompressInit(&mut strm, blockSize100k, 0, 30);
         if ret != 0 as libc::c_int {
             return ret;
         }
@@ -1307,6 +1310,7 @@ mod high_level_interface {
                 &mut expected_len,
                 SAMPLE1_BZ2.as_ptr(),
                 SAMPLE1_BZ2.len() as _,
+                9,
             )
         };
         assert_eq!(err, 0);