summaryrefslogtreecommitdiff
path: root/prism/internal/allocator_debug.h
blob: 846e96ba2da5ee9cc4388dc43f10c196b6d3d573 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#ifndef PRISM_INTERNAL_ALLOCATOR_DEBUG_H
#define PRISM_INTERNAL_ALLOCATOR_DEBUG_H

#include <string.h>
#include <stdio.h>
#include <stdlib.h>

static inline void *
pm_allocator_debug_malloc(size_t size) {
    size_t *memory = xmalloc(size + sizeof(size_t));
    memory[0] = size;
    return memory + 1;
}

static inline void *
pm_allocator_debug_calloc(size_t nmemb, size_t size) {
    size_t total_size = nmemb * size;
    void *ptr = pm_allocator_debug_malloc(total_size);
    memset(ptr, 0, total_size);
    return ptr;
}

static inline void *
pm_allocator_debug_realloc(void *ptr, size_t size) {
    if (ptr == NULL) {
        return pm_allocator_debug_malloc(size);
    }

    size_t *memory = (size_t *)ptr;
    void *raw_memory = memory - 1;
    memory = (size_t *)xrealloc(raw_memory, size + sizeof(size_t));
    memory[0] = size;
    return memory + 1;
}

static inline void
pm_allocator_debug_free(void *ptr) {
    if (ptr != NULL) {
        size_t *memory = (size_t *)ptr;
        xfree(memory - 1);
    }
}

static inline void
pm_allocator_debug_free_sized(void *ptr, size_t old_size) {
    if (ptr != NULL) {
        size_t *memory = (size_t *)ptr;
        if (old_size != memory[-1]) {
            fprintf(stderr, "[BUG] buffer %p was allocated with size %lu but freed with size %lu\n", ptr, memory[-1], old_size);
            abort();
        }
        xfree_sized(memory - 1, old_size + sizeof(size_t));
    }
}

static inline void *
pm_allocator_debug_realloc_sized(void *ptr, size_t size, size_t old_size) {
    if (ptr == NULL) {
        if (old_size != 0) {
            fprintf(stderr, "[BUG] realloc_sized called with NULL pointer and old size %lu\n", old_size);
            abort();
        }
        return pm_allocator_debug_malloc(size);
    }

    size_t *memory = (size_t *)ptr;
    if (old_size != memory[-1]) {
        fprintf(stderr, "[BUG] buffer %p was allocated with size %lu but realloced with size %lu\n", ptr, memory[-1], old_size);
        abort();
    }
    return pm_allocator_debug_realloc(ptr, size);
}

#undef xmalloc
#undef xrealloc
#undef xcalloc
#undef xfree
#undef xrealloc_sized
#undef xfree_sized

#define xmalloc          pm_allocator_debug_malloc
#define xrealloc         pm_allocator_debug_realloc
#define xcalloc          pm_allocator_debug_calloc
#define xfree            pm_allocator_debug_free
#define xrealloc_sized   pm_allocator_debug_realloc_sized
#define xfree_sized      pm_allocator_debug_free_sized

#endif