rust/src/rt/rust_exchange_alloc.cpp
2013-05-15 12:19:14 -07:00

52 lines
1.3 KiB
C++

// Copyright 2013 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
#include "rust_exchange_alloc.h"
#include "sync/sync.h"
#include <stdlib.h>
#include <assert.h>
#include <string.h>
#include <stdio.h>
extern uintptr_t rust_exchange_count;
uintptr_t rust_exchange_count = 0;
void *
rust_exchange_alloc::malloc(size_t size) {
void *value = ::malloc(size);
assert(value);
sync::increment(rust_exchange_count);
return value;
}
void *
rust_exchange_alloc::realloc(void *ptr, size_t size) {
void *new_ptr = ::realloc(ptr, size);
assert(new_ptr);
return new_ptr;
}
void
rust_exchange_alloc::free(void *ptr) {
sync::decrement(rust_exchange_count);
::free(ptr);
}
void
rust_check_exchange_count_on_exit() {
if (rust_exchange_count != 0) {
printf("exchange heap not empty on exit\n");
printf("%d dangling allocations\n", (int)rust_exchange_count);
abort();
}
}