haz-alloc 0.2.1

A general-purpose allocator written in 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

use haz_alloc_core::backend::TlsCallback;
use std::cell::Cell;
use std::ptr;

thread_local! {
    static ATTACHED: Attached = Attached(Cell::new(ptr::null()));
}

struct Attached(Cell<*const TlsCallback>);

impl Drop for Attached {
    fn drop(&mut self) {
        let mut callback = self.0.get();
        while !callback.is_null() {
            unsafe {
                ((*callback).func)();
                callback = (*callback).next.get();
            }
        }
    }
}

pub unsafe fn tls_attach(callback: *const TlsCallback) {
    ATTACHED.with(|attached| {
        attached.0.set(callback);
    });
}