blob: 68aa246c5c9a5b7ad3b407dda2a66bc7403dd8e9 (
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
|
/** \ingroup rpmio
* \file rpmio/rpmmalloc.c
*/
#include "system.h"
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include "debug.h"
static rpmMemFailFunc failfunc = NULL;
static void *failfunc_data = NULL;
/*
* Give memfail callback a chance to try to give us memory or perform
* it's own cleanup. If we dont get memory we die anyway as rpm doesn't
* check for NULL returns from allocations.
*/
static void *vmefail(size_t size)
{
void *val = failfunc ? (*failfunc)(size, failfunc_data) : NULL;
if (val == NULL) {
fprintf(stderr, _("memory alloc (%u bytes) returned NULL.\n"),
(unsigned)size);
exit(EXIT_FAILURE);
}
return val;
}
void * rpmSetMemFail(rpmMemFailFunc func, void *data)
{
void *ofunc = failfunc;
failfunc = func;
failfunc_data = data;
return ofunc;
}
void * rmalloc (size_t size)
{
register void *value;
if (size == 0) size++;
value = malloc (size);
if (value == NULL)
value = vmefail(size);
return value;
}
void * rcalloc (size_t nmemb, size_t size)
{
register void *value;
if (size == 0) size++;
if (nmemb == 0) nmemb++;
value = calloc (nmemb, size);
if (value == NULL)
value = vmefail(size);
return value;
}
void * rrealloc (void *ptr, size_t size)
{
register void *value;
if (size == 0) size++;
value = realloc (ptr, size);
if (value == NULL)
value = vmefail(size);
return value;
}
char * rstrdup (const char *str)
{
size_t size = strlen(str) + 1;
char *newstr = (char *) malloc (size);
if (newstr == NULL)
newstr = (char *) vmefail(size);
strcpy (newstr, str);
return newstr;
}
void * rfree (void *ptr)
{
free(ptr);
return NULL;
}
|
