autarkie 0.3.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
53
54
55
56
57
58
59
60
61
62
63
64
65

//! Stage that wraps mutating stages for stats and cleanup
use crate::fuzzer::context::Context;
use crate::Visitor;
use core::{marker::PhantomData, time::Duration};
use libafl_bolts::current_time;
use std::cell::RefCell;
use std::rc::Rc;

use libafl::{
    stages::{Restartable, Stage},
    Error, HasMetadata,
};
/// Track an inner Stage's execution time
#[derive(Debug)]
pub struct MutatingStageWrapper<S, ST> {
    inner: ST,
    visitor: Rc<RefCell<Visitor>>,
    phantom: PhantomData<S>,
}

impl<S, ST> MutatingStageWrapper<S, ST> {
    /// Create a `MutatingStageWrapper`
    pub fn new(inner: ST, visitor: Rc<RefCell<Visitor>>) -> Self {
        Self {
            inner,
            visitor,
            phantom: PhantomData,
        }
    }
}

impl<E, M, Z, S, ST> Stage<E, M, S, Z> for MutatingStageWrapper<S, ST>
where
    S: HasMetadata,
    ST: Stage<E, M, S, Z>,
{
    fn perform(
        &mut self,
        fuzzer: &mut Z,
        executor: &mut E,
        state: &mut S,
        manager: &mut M,
    ) -> Result<(), Error> {
        self.inner.perform(fuzzer, executor, state, manager)?;
        let _ = state
            .metadata_mut::<Context>()
            .expect("773RwiGF____")
            .clear_mutations();
        let _ = self.visitor.borrow_mut().serialized();
        Ok(())
    }
}

impl<S, ST> Restartable<S> for MutatingStageWrapper<S, ST>
where
    ST: Restartable<S>,
{
    fn should_restart(&mut self, state: &mut S) -> Result<bool, Error> {
        self.inner.should_restart(state)
    }

    fn clear_progress(&mut self, state: &mut S) -> Result<(), Error> {
        self.inner.clear_progress(state)
    }
}