This repository has been archived by the owner on Jan 21, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 9
/
xalloc.c
119 lines (104 loc) · 1.7 KB
/
xalloc.c
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
/*
* Copyright (c) 2004-2009 Smithsonian Astrophysical Observatory
*/
/*
*
* xalloc -- safe memory allocation with error checking
*
*/
/* this module is compiled within a funtools filter and must not require
the header files */
#ifdef FILTER_PTYPE
#define ANSI_FUNC 1
#else
#include <xalloc.h>
#endif
#define XALLOC_ERROR "ERROR: can't allocate memory (xalloc)\n"
#if XALLOC_SETJMP
static jmp_buf *xalloc_envptr=NULL;
#ifdef ANSI_FUNC
void xalloc_savejmp(jmp_buf *env)
#else
void xalloc_savejmp(env)
jmp_buf *env;
#endif
{
xalloc_envptr = env;
}
#endif
#ifdef ANSI_FUNC
static void _xalloc_error(void)
#else
static void _xalloc_error()
#endif
{
if( write(1, XALLOC_ERROR, strlen(XALLOC_ERROR)) < 0 ){
exit(1);
}
#if XALLOC_SETJMP
if( xalloc_envptr )
longjmp(*xalloc_envptr, XALLOC_SETJMP);
else
#endif
exit(1);
}
#ifdef ANSI_FUNC
void *xmalloc(size_t n)
#else
void *xmalloc(n)
size_t n;
#endif
{
void *p;
if( !(p = (void *)malloc(n)) )
_xalloc_error();
return p;
}
#ifdef ANSI_FUNC
void *xcalloc (size_t n, size_t s)
#else
void *xcalloc (n, s)
size_t n, s;
#endif
{
void *p;
if( !(p = (void *)calloc(n, s)) )
_xalloc_error();
return p;
}
#ifdef ANSI_FUNC
void *xrealloc (void *p, size_t n)
#else
void *xrealloc (p, n)
void *p;
size_t n;
#endif
{
if( !p )
return xmalloc(n);
if( !(p = (void *)realloc(p, n)) )
_xalloc_error();
return p;
}
#ifdef ANSI_FUNC
void xfree (void *p)
#else
void xfree (p)
void *p;
#endif
{
if( p )
free(p);
}
#ifdef ANSI_FUNC
char *xstrdup (char *s)
#else
char *xstrdup (s)
char *s;
#endif
{
if( s )
return((char *)strcpy((char *)xmalloc((size_t)strlen(s)+1), s));
else
return NULL;
}