-
Notifications
You must be signed in to change notification settings - Fork 93
/
alloc.h
39 lines (32 loc) · 896 Bytes
/
alloc.h
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
/*
* by Scarlett. public domain.
* replacements for w3m's allocation macros which add overflow
* detection and concentrate the macros in one file
*/
#ifndef W3_ALLOC_H
#define W3_ALLOC_H
#include <gc.h>
#include <stdlib.h>
#include <stdio.h>
#include <limits.h>
static inline size_t
z_mult_no_oflow_(size_t n, size_t size)
{
if (size != 0 && n > ULONG_MAX / size) {
fprintf(stderr,
"w3m: overflow in malloc, %lu*%lu\n", (unsigned long)n, (unsigned long)size);
exit(1);
}
return n * size;
}
#define New(type) \
(GC_MALLOC(sizeof(type)))
#define NewAtom(type) \
(GC_MALLOC_ATOMIC(sizeof(type)))
#define New_N(type, n) \
(GC_MALLOC(z_mult_no_oflow_((n), sizeof(type))))
#define NewAtom_N(type, n) \
(GC_MALLOC_ATOMIC(z_mult_no_oflow_((n), sizeof(type))))
#define New_Reuse(type, ptr, n) \
(GC_REALLOC((ptr), z_mult_no_oflow_((n), sizeof(type))))
#endif /* W3_ALLOC_H */