flate2/ffi/
mod.rs

1//! This module contains backend-specific code.
2
3use crate::mem::{CompressError, DecompressError, FlushCompress, FlushDecompress, Status};
4use crate::Compression;
5use std::mem::MaybeUninit;
6
7fn initialize_buffer(output: &mut [MaybeUninit<u8>]) -> &mut [u8] {
8    // SAFETY: Here we zero-initialize the output and cast it to [u8]
9    unsafe {
10        output.as_mut_ptr().write_bytes(0, output.len());
11        &mut *(output as *mut [MaybeUninit<u8>] as *mut [u8])
12    }
13}
14
15/// Traits specifying the interface of the backends.
16///
17/// Sync + Send are added as a condition to ensure they are available
18/// for the frontend.
19pub trait Backend: Sync + Send {
20    fn total_in(&self) -> u64;
21    fn total_out(&self) -> u64;
22}
23
24pub trait InflateBackend: Backend {
25    fn make(zlib_header: bool, window_bits: u8) -> Self;
26    fn decompress(
27        &mut self,
28        input: &[u8],
29        output: &mut [u8],
30        flush: FlushDecompress,
31    ) -> Result<Status, DecompressError>;
32    fn decompress_uninit(
33        &mut self,
34        input: &[u8],
35        output: &mut [MaybeUninit<u8>],
36        flush: FlushDecompress,
37    ) -> Result<Status, DecompressError> {
38        self.decompress(input, initialize_buffer(output), flush)
39    }
40    fn reset(&mut self, zlib_header: bool);
41}
42
43pub trait DeflateBackend: Backend {
44    fn make(level: Compression, zlib_header: bool, window_bits: u8) -> Self;
45    fn compress(
46        &mut self,
47        input: &[u8],
48        output: &mut [u8],
49        flush: FlushCompress,
50    ) -> Result<Status, CompressError>;
51    fn compress_uninit(
52        &mut self,
53        input: &[u8],
54        output: &mut [MaybeUninit<u8>],
55        flush: FlushCompress,
56    ) -> Result<Status, CompressError> {
57        self.compress(input, initialize_buffer(output), flush)
58    }
59    fn reset(&mut self);
60}
61
62// Default to Rust implementation unless explicitly opted in to a different backend.
63#[cfg(feature = "any_zlib")]
64mod c;
65#[cfg(feature = "any_zlib")]
66pub use self::c::*;
67
68#[cfg(all(not(feature = "any_zlib"), feature = "miniz_oxide"))]
69mod rust;
70#[cfg(all(not(feature = "any_zlib"), feature = "miniz_oxide"))]
71pub use self::rust::*;
72
73impl std::fmt::Debug for ErrorMessage {
74    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
75        self.get().fmt(f)
76    }
77}
flate2/ffi/mod.rs

flate2/ffi/
mod.rs
