96 lines
1.9 KiB
C++
96 lines
1.9 KiB
C++
|
#include <stdlib.h>
|
||
|
|
#include <map>
|
||
|
|
#include <set>
|
||
|
|
#include "bandit/bandit.h"
|
||
|
|
|
||
|
|
using std::map;
|
||
|
|
using std::set;
|
||
|
|
|
||
|
|
bool _enabled = false;
|
||
|
|
static size_t _allocation_count = 0;
|
||
|
|
static map<void *, size_t> _outstanding_allocations;
|
||
|
|
static size_t _allocation_failure_index = -1;
|
||
|
|
|
||
|
|
namespace record_alloc {
|
||
|
|
|
||
|
|
void start() {
|
||
|
|
_enabled = true;
|
||
|
|
_allocation_count = 0;
|
||
|
|
_outstanding_allocations.clear();
|
||
|
|
_allocation_failure_index = -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
void stop() {
|
||
|
|
_enabled = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
void fail_at_allocation_index(size_t failure_index) {
|
||
|
|
_allocation_failure_index = failure_index;
|
||
|
|
}
|
||
|
|
|
||
|
|
set<size_t> outstanding_allocation_indices() {
|
||
|
|
set<size_t> result;
|
||
|
|
for (const auto &entry : _outstanding_allocations) {
|
||
|
|
result.insert(entry.second);
|
||
|
|
}
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t allocation_count() {
|
||
|
|
return _allocation_count;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace record_alloc
|
||
|
|
|
||
|
|
static void *record_allocation(void *result) {
|
||
|
|
if (!_enabled)
|
||
|
|
return result;
|
||
|
|
|
||
|
|
if (_allocation_count > _allocation_failure_index) {
|
||
|
|
free(result);
|
||
|
|
Assert::Failure("Allocated after a previous allocation failed!");
|
||
|
|
}
|
||
|
|
|
||
|
|
if (_allocation_count == _allocation_failure_index) {
|
||
|
|
_allocation_count++;
|
||
|
|
free(result);
|
||
|
|
return nullptr;
|
||
|
|
}
|
||
|
|
|
||
|
|
_outstanding_allocations[result] = _allocation_count;
|
||
|
|
_allocation_count++;
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void record_deallocation(void *pointer) {
|
||
|
|
if (!_enabled)
|
||
|
|
return;
|
||
|
|
|
||
|
|
auto entry = _outstanding_allocations.find(pointer);
|
||
|
|
if (entry != _outstanding_allocations.end()) {
|
||
|
|
_outstanding_allocations.erase(entry);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
extern "C" {
|
||
|
|
|
||
|
|
void *ts_record_malloc(size_t size) {
|
||
|
|
return record_allocation(malloc(size));
|
||
|
|
}
|
||
|
|
|
||
|
|
void *ts_record_realloc(void *pointer, size_t size) {
|
||
|
|
record_deallocation(pointer);
|
||
|
|
return record_allocation(realloc(pointer, size));
|
||
|
|
}
|
||
|
|
|
||
|
|
void *ts_record_calloc(size_t count, size_t size) {
|
||
|
|
return record_allocation(calloc(count, size));
|
||
|
|
}
|
||
|
|
|
||
|
|
void ts_record_free(void *pointer) {
|
||
|
|
free(pointer);
|
||
|
|
record_deallocation(pointer);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|