precomp2 0.2.0

Reversible preprocessing for compressed and container data.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

use crate::error::{Error, Result};

#[derive(Default)]
pub(super) struct EncodeBudget {
  pub(super) total_output: u64,
}

impl EncodeBudget {
  pub(super) fn consume(&mut self, len: usize, config: &super::EncodeConfig) -> Result<()> {
    self.total_output = self.total_output.saturating_add(len as u64);
    if self.total_output > config.max_total_output {
      return Err(Error::Other("encode output exceeds budget".to_string()));
    }
    Ok(())
  }
}

#[derive(Default)]
pub(super) struct DecodeBudget {
  pub(super) total_output: u64,
}

impl DecodeBudget {
  pub(super) fn consume(&mut self, len: usize, config: &super::DecodeConfig) -> Result<()> {
    self.total_output = self.total_output.saturating_add(len as u64);
    if self.total_output > config.max_total_output {
      return Err(Error::Other("decode output exceeds budget".to_string()));
    }
    Ok(())
  }
}