38 lines
740 B
C
38 lines
740 B
C
|
/* $Revision: 1.1 $
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "unroff.h"
|
||
|
|
|
||
|
|
static void (*err_handler)(char *, size_t);
|
||
|
|
static void default_err_handler(char *, size_t);
|
||
|
|
|
||
|
|
void *safe_malloc(size_t n) {
|
||
|
|
void *p;
|
||
|
|
|
||
|
|
if ((p = malloc(n)) == 0) {
|
||
|
|
(err_handler ? err_handler : default_err_handler)("malloc", n);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
return p;
|
||
|
|
}
|
||
|
|
|
||
|
|
void *safe_realloc(void *old, size_t n) {
|
||
|
|
void *p;
|
||
|
|
|
||
|
|
if ((p = realloc(old, n)) == 0) {
|
||
|
|
(err_handler ? err_handler : default_err_handler)("realloc", n);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
return p;
|
||
|
|
}
|
||
|
|
|
||
|
|
void set_alloc_failure(void (*func)(char *, size_t)) {
|
||
|
|
err_handler = func;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void default_err_handler(char *what, size_t n) {
|
||
|
|
fatal_error("cannot %s %lu bytes--virtual memory exhausted",
|
||
|
|
what, (unsigned long)n);
|
||
|
|
}
|
||
|
|
|