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
use std::alloc::{GlobalAlloc, Layout};
use std::os::raw::c_void;
#[link(name = "tcmalloc")]
extern "C" {
pub fn tc_memalign(alignment: usize, size: usize) -> *mut c_void;
pub fn tc_free(ptr: *mut c_void);
}
pub struct TCMalloc;
unsafe impl GlobalAlloc for TCMalloc {
unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
tc_memalign(layout.align(), layout.size()) as *mut u8
}
unsafe fn dealloc(&self, ptr: *mut u8, _layout: Layout) {
tc_free(ptr as *mut c_void);
}
}