busywork 0.1.0

Sleep replacement that executes real, varied work to break behavioral pattern matching
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

#[cfg(feature = "cat-compute")]
pub mod compute;
#[cfg(feature = "cat-crypto")]
pub mod crypto;
#[cfg(feature = "cat-filesystem")]
pub mod filesystem;
#[cfg(feature = "cat-memory")]
pub mod memory;
#[cfg(feature = "cat-network")]
pub mod network;
#[cfg(feature = "cat-registry")]
pub mod registry;
#[cfg(feature = "cat-winapi")]
pub mod winapi_tasks;

use crate::categories::Categories;
use rand::rngs::ThreadRng;

pub struct TaskParams {
    pub iterations: usize,
    pub buffer_size: usize,
    pub call_depth: usize,
}

pub type TaskFn = fn(&TaskParams, &mut ThreadRng);

#[allow(dead_code)]
pub struct TaskDescriptor {
    pub name: &'static str,
    pub category: Categories,
    pub func: TaskFn,
}

pub fn all_tasks() -> Vec<TaskDescriptor> {
    let mut tasks = Vec::new();
    #[cfg(feature = "cat-compute")]
    tasks.extend(compute::register());
    #[cfg(feature = "cat-memory")]
    tasks.extend(memory::register());
    #[cfg(feature = "cat-filesystem")]
    tasks.extend(filesystem::register());
    #[cfg(feature = "cat-registry")]
    tasks.extend(registry::register());
    #[cfg(feature = "cat-winapi")]
    tasks.extend(winapi_tasks::register());
    #[cfg(feature = "cat-network")]
    tasks.extend(network::register());
    #[cfg(feature = "cat-crypto")]
    tasks.extend(crypto::register());
    tasks
}