alloc.h 896 Bytes
Newer Older
Scarlett's avatar
Scarlett committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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)
{
Tatsuya Kinoshita's avatar
Tatsuya Kinoshita committed
16
	if (size != 0 && n > ULONG_MAX / size) {
Scarlett's avatar
Scarlett committed
17
		fprintf(stderr,
Tatsuya Kinoshita's avatar
Tatsuya Kinoshita committed
18
		    "w3m: overflow in malloc, %lu*%lu\n", (unsigned long)n, (unsigned long)size);
Scarlett's avatar
Scarlett committed
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39
		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 */