50 lines
4.2 KiB
C
50 lines
4.2 KiB
C
/**
|
|
* mem — выделение памяти с защитными полями, местом аллокации и учётом живых блоков.
|
|
* Использовать u_malloc/u_calloc/u_realloc/u_strdup и парный u_free; обычный free к этим блокам неприменим.
|
|
* Макросы передают LOCATION автоматически. OOM возвращает NULL; обнаруженное повреждение завершает процесс.
|
|
* realloc живого блока при ошибке сохраняет его, при size=0 освобождает; NULL обрабатывается как malloc.
|
|
* Размеры аргументов _impl — uint32_t; до вызова проверить, что исходные size_t помещаются в этот диапазон.
|
|
* Список аллокаций защищён мьютексом; доступ к одному блоку и чтение счётчика согласует вызывающий.
|
|
* Защитные поля проверяются для живых блоков; обращение к уже освобождённому/чужому указателю недопустимо.
|
|
* Отчёт u_report_unfreed_blocks вызывать после остановки потоков, меняющих список аллокаций.
|
|
*/
|
|
#ifndef MEM_H
|
|
#define MEM_H
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
#define XSTR(s) STR(s)
|
|
#define STR(s) #s
|
|
#define LOCATION __FILE__ ":" XSTR(__LINE__)
|
|
// Проверяет canary/padding живого блока; text/location — строки диагностики. NULL ptr допустим.
|
|
// Обнаруженное повреждение логируется в SYS и завершает процесс через exit(EXIT_FAILURE).
|
|
void u_check(void* ptr, const char* text, const char* location);
|
|
#define u_malloc(size) u_malloc_impl(size, LOCATION)
|
|
#define u_calloc(nmemb, size) u_calloc_impl(nmemb, size, LOCATION)
|
|
#define u_realloc(ptr, size) u_realloc_impl(ptr, size, LOCATION)
|
|
#define u_free(ptr) u_free_impl(ptr, LOCATION)
|
|
#define u_strdup(s) u_strdup_impl(s, LOCATION)
|
|
// Выделяет size байт с метаданными; NULL при OOM/переполнении полного размера. size=0 тоже создаёт учтённый блок.
|
|
// location — обязательная строка места вызова, копируется в метаданные; payload не обнуляется.
|
|
void* u_malloc_impl(uint32_t size, const char* location);
|
|
// Выделяет и обнуляет nmemb*size байт; NULL при произведении > UINT32_MAX или ошибке malloc.
|
|
void* u_calloc_impl(uint32_t nmemb, uint32_t size, const char* location);
|
|
// NULL ptr → malloc(size), включая size=0. Для живого ptr: size=0 → free и NULL;
|
|
// иначе новый блок с min(старый размер, size) байтами данных. При OOM возвращает NULL, старый блок сохранён.
|
|
// Результат сохранять временно: tmp = u_realloc(ptr, size); if (tmp) ptr = tmp; (при size > 0).
|
|
void* u_realloc_impl(void* ptr, uint32_t size, const char* location);
|
|
// Проверяет и освобождает живой блок из этих обёрток; NULL допустим. Обычный free к нему неприменим.
|
|
void u_free_impl(void* ptr, const char* location);
|
|
// Копирует ASCIIZ-строку (strlen(s)+1 должно помещаться в uint32_t); NULL при NULL s/OOM. Освобождать u_free.
|
|
char* u_strdup_impl(const char* s, const char* location);
|
|
// После остановки изменяющих список потоков: статистика и дампы живых блоков; сами блоки не освобождает.
|
|
void u_report_unfreed_blocks(void);
|
|
// Число живых блоков, чтение без блокировки; параллельные аллокации/освобождения нужно исключить извне.
|
|
size_t u_get_allocated_count(void);
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
#endif // MEM_H
|