autarkie 0.8.0

autarkie - instant grammar fuzzing using rust
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use std::{
    collections::{HashMap, VecDeque},
    ops::Range,
    path::PathBuf,
};

use crate::Visitor;

pub fn calculate_subslice_bounds(len: usize, max: usize, visitor: &mut Visitor) -> Range<usize> {
    // minus 1 because we zero index and len is always +1
    let start = visitor.random_range(0, len - 1);
    let mut end = visitor.random_range(start, len);
    if end - start > max {
        end = start + max;
    }
    start..end
}

pub struct FileCache {
    data: HashMap<PathBuf, Vec<u8>>,
    queue: VecDeque<PathBuf>,
    size: usize,
    max_size: usize,
}

impl FileCache {
    pub fn read_cached(&mut self, path: &PathBuf) -> Result<&[u8], std::io::Error> {
        if !self.data.contains_key(path) {
            let data = std::fs::read(path)?;
            self.size += data.len();
            self.data.insert(path.clone(), data);
            self.queue.push_back(path.to_path_buf());
        }
        while self.size > self.max_size {
            let v = self
                .data
                .remove(&self.queue.pop_front().expect("invariant;2"))
                .expect("invariant;");
            self.size -= v.len();
        }
        Ok(self.data.get(path).as_ref().unwrap())
    }

    pub fn new(size: usize) -> Self {
        Self {
            queue: VecDeque::default(),
            max_size: size * 1024 * 1024,
            data: HashMap::new(),
            size: 0,
        }
    }
}