76 lines
1.8 KiB
C++
76 lines
1.8 KiB
C++
#ifndef SEODISPARATE_COM_ANOTHER_MEMCHECK_H
|
|
#define SEODISPARATE_COM_ANOTHER_MEMCHECK_H
|
|
|
|
extern "C" {
|
|
#include <pthread.h>
|
|
}
|
|
|
|
namespace SC_AM_Internal {
|
|
// Forward declaration.
|
|
struct Stats;
|
|
|
|
extern Stats *stats;
|
|
|
|
constexpr int ANOTHER_MEMCHECK_QUIET_EXISTS = 1;
|
|
constexpr int ANOTHER_MEMCHECK_QUIET_NOT_EXISTS = 2;
|
|
|
|
/// 0 If unknown, 1 if env "ANOTHER_MEMCHECK_QUIET" exists, 2 if not.
|
|
extern int is_env_status;
|
|
|
|
Stats *get_init_stats();
|
|
|
|
extern void *(*real_malloc)(std::size_t size);
|
|
extern void *(*real_calloc)(std::size_t n, std::size_t size);
|
|
extern void *(*real_realloc)(void *ptr, std::size_t size);
|
|
extern void (*real_free)(void *ptr);
|
|
|
|
struct Malloced {
|
|
Malloced();
|
|
Malloced(void *address, std::size_t size);
|
|
|
|
void* address;
|
|
std::size_t size;
|
|
unsigned long long id;
|
|
static unsigned long long count;
|
|
};
|
|
|
|
struct ListNode {
|
|
ListNode() : next(nullptr), prev(nullptr), data(nullptr) {}
|
|
|
|
static void add_to_list(ListNode *tail, Malloced *data);
|
|
static void add_to_list(ListNode *tail, ListNode *node);
|
|
/// Returns true if removed.
|
|
static bool remove_from_list(ListNode *head,
|
|
void *ptr,
|
|
Stats *defer_handler = nullptr);
|
|
|
|
ListNode *next;
|
|
ListNode *prev;
|
|
Malloced *data;
|
|
};
|
|
|
|
|
|
struct Stats {
|
|
Stats();
|
|
|
|
ListNode *malloced_list_head;
|
|
ListNode *malloced_list_tail;
|
|
ListNode *deferred_node;
|
|
pthread_mutex_t pthread_mutex;
|
|
|
|
void initialize();
|
|
void cleanup();
|
|
|
|
void *do_malloc(std::size_t size);
|
|
void *do_calloc(std::size_t n, std::size_t size);
|
|
void *do_realloc(void*, std::size_t size);
|
|
/// true on success.
|
|
bool do_free(void *ptr);
|
|
|
|
void print_status() const;
|
|
};
|
|
}
|
|
|
|
#endif
|
|
|
|
// vim: et sw=2 ts=2 sts=2
|