forked from Green-Sky/tomato
40 lines
1.1 KiB
C++
40 lines
1.1 KiB
C++
|
#ifndef C_TOXCORE_TOXCORE_MEM_TEST_UTIL_H
|
||
|
|
#define C_TOXCORE_TOXCORE_MEM_TEST_UTIL_H
|
||
|
|
|
||
|
|
#include "mem.h"
|
||
|
|
#include "test_util.hh"
|
||
|
|
|
||
|
|
struct Memory_Class {
|
||
|
|
static Memory_Funcs const vtable;
|
||
|
|
Memory const self;
|
||
|
|
|
||
|
|
operator Memory const *() const { return &self; }
|
||
|
|
|
||
|
|
Memory_Class(Memory_Class const &) = default;
|
||
|
|
Memory_Class()
|
||
|
|
: self{&vtable, this}
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual ~Memory_Class();
|
||
|
|
virtual mem_malloc_cb malloc = 0;
|
||
|
|
virtual mem_calloc_cb calloc = 0;
|
||
|
|
virtual mem_realloc_cb realloc = 0;
|
||
|
|
virtual mem_free_cb free = 0;
|
||
|
|
};
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Base test Memory class that just forwards to system_memory. Can be
|
||
|
|
* subclassed to override individual (or all) functions.
|
||
|
|
*/
|
||
|
|
class Test_Memory : public Memory_Class {
|
||
|
|
const Memory *mem = REQUIRE_NOT_NULL(system_memory());
|
||
|
|
|
||
|
|
void *malloc(void *obj, uint32_t size) override;
|
||
|
|
void *calloc(void *obj, uint32_t nmemb, uint32_t size) override;
|
||
|
|
void *realloc(void *obj, void *ptr, uint32_t size) override;
|
||
|
|
void free(void *obj, void *ptr) override;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // C_TOXCORE_TOXCORE_MEM_TEST_UTIL_H
|