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
|
#include "mpdecimal.h"
#include <stdlib.h>
#include "malloc_fail.h"
/* Test allocation failures */
int alloc_count;
int alloc_fail;
int alloc_idx;
void *
mpd_malloc_count(size_t size)
{
++alloc_count;
return malloc(size);
}
void *
mpd_calloc_count(size_t nmemb, size_t size)
{
++alloc_count;
return calloc(nmemb, size);
}
void *
mpd_realloc_count(void *ptr, size_t size)
{
++alloc_count;
return realloc(ptr, size);
}
void *
mpd_malloc_fail(size_t size)
{
#ifdef TEST_ALLOC
if (++alloc_idx >= alloc_fail) {
return NULL;
}
#endif
return malloc(size);
}
void *
mpd_calloc_fail(size_t nmemb, size_t size)
{
#ifdef TEST_ALLOC
if (++alloc_idx >= alloc_fail) {
return NULL;
}
#endif
if (rand()%2) {
return calloc(nmemb, size);
}
else {
return mpd_callocfunc_em(nmemb, size);
}
}
void *
mpd_realloc_fail(void *ptr, size_t size)
{
#ifdef TEST_ALLOC
if (++alloc_idx >= alloc_fail) {
return NULL;
}
#endif
return realloc(ptr, size);
}
void
mpd_set_alloc_count(mpd_context_t *ctx)
{
mpd_mallocfunc = mpd_malloc_count;
mpd_callocfunc = mpd_calloc_count;
mpd_reallocfunc = mpd_realloc_count;
ctx->traps = MPD_Malloc_error;
alloc_count = 0;
}
void
mpd_set_alloc_fail(mpd_context_t *ctx)
{
mpd_mallocfunc = mpd_malloc_fail;
mpd_callocfunc = mpd_calloc_fail;
mpd_reallocfunc = mpd_realloc_fail;
ctx->traps = 0;
alloc_idx = 0;
}
void
mpd_set_alloc(mpd_context_t *ctx)
{
mpd_mallocfunc = malloc;
mpd_callocfunc = (rand()%2) ? calloc : mpd_callocfunc_em;
mpd_reallocfunc = realloc;
ctx->traps = MPD_Malloc_error;
}
|
