nalloc 0.1.2

An allocator wrapper that can be turned on and off
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

use std::process::Command;
use std::{env, os::unix::process::ExitStatusExt};

use nalloc::NAlloc;

#[global_allocator]
static ALLOCATOR: NAlloc<std::alloc::System> =
    NAlloc::<std::alloc::System>::new(std::alloc::System);

fn test_abort(test_name: &str, f: impl FnOnce()) {
    if env::var_os("NALLOC_TEST").is_none() {
        let status = Command::new(env::current_exe().expect("Could not get current executable"))
            .arg("--exact")
            .arg(test_name)
            .env("NALLOC_TEST", "true")
            .output()
            .expect("Recursive call failed");

        assert_eq!(status.status.signal(), Some(6));

        return;
    }

    f();
}

#[test]
fn off() {
    let _ = Vec::<u128>::with_capacity(10);
}

#[test]
fn on() {
    test_abort("on", || {
        let _lock = ALLOCATOR.deny();
        let _ = Vec::<u128>::with_capacity(10);
    });
}

#[test]
fn on_and_off() {
    {
        let _lock = ALLOCATOR.deny();
    }

    let _ = Vec::<u128>::with_capacity(10);
}

#[test]
fn multiple_and_off() {
    {
        let _lock = ALLOCATOR.deny();
        let _lock = ALLOCATOR.deny();
        let _lock = ALLOCATOR.deny();
        let _lock = ALLOCATOR.deny();
        let _lock = ALLOCATOR.deny();
    }

    let _ = Vec::<u128>::with_capacity(10);
}

#[test]
fn multiple_and_partial_off_before() {
    test_abort("multiple_and_partial_off_before", || {
        let _lock = ALLOCATOR.deny();

        {
            let _lock = ALLOCATOR.deny();
            let _lock = ALLOCATOR.deny();
            let _lock = ALLOCATOR.deny();
            let _lock = ALLOCATOR.deny();
        }

        let _ = Vec::<u128>::with_capacity(10);
    });
}

#[test]
fn multiple_and_partial_off_after() {
    test_abort("multiple_and_partial_off_after", || {
        {
            let _lock = ALLOCATOR.deny();
            let _lock = ALLOCATOR.deny();
            let _lock = ALLOCATOR.deny();
            let _lock = ALLOCATOR.deny();
        }

        let _lock = ALLOCATOR.deny();
        let _ = Vec::<u128>::with_capacity(10);
    });
}