jemallocator/tests/malloctl.rs

extern crate jemallocator;
extern crate libc;

use jemallocator::Jemalloc;
use std::alloc::{GlobalAlloc, Layout};

#[global_allocator]
static A: Jemalloc = Jemalloc;

#[test]
fn smoke() {
    let layout = Layout::from_size_align(100, 8).unwrap();
    unsafe {
        let ptr = Jemalloc.alloc(layout.clone());
        assert!(!ptr.is_null());
        Jemalloc.dealloc(ptr, layout);
    }
}

#[test]
fn test_mallctl() {
    let mut epoch: u64 = 0;
    unsafe {
        assert_eq!(
            jemallocator::mallctl_fetch(b"", &mut epoch),
            Err(libc::EINVAL)
        );
        assert_eq!(
            jemallocator::mallctl_fetch(b"epoch", &mut epoch),
            Err(libc::EINVAL)
        );
        jemallocator::mallctl_fetch(b"epoch\0", &mut epoch).unwrap();
        assert!(epoch > 0);
        assert_eq!(jemallocator::mallctl_set(b"", epoch), Err(libc::EINVAL));
        assert_eq!(
            jemallocator::mallctl_set(b"epoch", epoch),
            Err(libc::EINVAL)
        );
        jemallocator::mallctl_set(b"epoch\0", epoch).unwrap();
    }
}