71 lines
1.6 KiB
Rust
71 lines
1.6 KiB
Rust
use libc::{c_int, size_t};
|
|
|
|
extern "C" {
|
|
pub fn bshuf_compress_lz4(
|
|
inp: *const u8,
|
|
out: *const u8,
|
|
size: size_t,
|
|
elem_size: size_t,
|
|
block_size: size_t,
|
|
) -> i64;
|
|
|
|
pub fn bshuf_decompress_lz4(
|
|
inp: *const u8,
|
|
out: *const u8,
|
|
size: size_t,
|
|
elem_size: size_t,
|
|
block_size: size_t,
|
|
) -> i64;
|
|
|
|
pub fn LZ4_decompress_safe(
|
|
source: *const u8,
|
|
dest: *mut u8,
|
|
compressedSize: c_int,
|
|
maxDecompressedSize: c_int,
|
|
) -> c_int;
|
|
}
|
|
|
|
pub fn bitshuffle_compress(
|
|
inp: &[u8],
|
|
out: &mut [u8],
|
|
size: usize,
|
|
elem_size: usize,
|
|
block_size: usize,
|
|
) -> Result<usize, isize> {
|
|
unsafe {
|
|
let n = bshuf_compress_lz4(inp.as_ptr(), out.as_mut_ptr(), size, elem_size, block_size);
|
|
if n >= 0 {
|
|
Ok(n as usize)
|
|
} else {
|
|
Err(n as isize)
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn bitshuffle_decompress(
|
|
inp: &[u8],
|
|
out: &mut [u8],
|
|
size: usize,
|
|
elem_size: usize,
|
|
block_size: usize,
|
|
) -> Result<usize, isize> {
|
|
unsafe {
|
|
let n = bshuf_decompress_lz4(inp.as_ptr(), out.as_mut_ptr(), size, elem_size, block_size);
|
|
if n >= 0 {
|
|
Ok(n as usize)
|
|
} else {
|
|
Err(n as isize)
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn lz4_decompress(inp: &[u8], out: &mut [u8]) -> Result<usize, isize> {
|
|
let max_out = out.len() as _;
|
|
let ec = unsafe { LZ4_decompress_safe(inp.as_ptr(), out.as_mut_ptr(), inp.len() as _, max_out) };
|
|
if ec < 0 {
|
|
Err(ec as _)
|
|
} else {
|
|
Ok(ec as _)
|
|
}
|
|
}
|