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
120
121
122
123
|
/* $OpenBSD: malloc_general.c,v 1.7 2022/01/09 07:18:50 otto Exp $ */
/*
* Copyright (c) 2017 Otto Moerbeek <otto@drijf.net>
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* #define VERBOSE */
#define N 1000
size_t
size(void)
{
int p = arc4random_uniform(17) + 3;
return arc4random_uniform(1 << p);
}
struct { void *p; size_t sz; } a[N];
void
fill(u_char *p, size_t sz)
{
size_t i;
for (i = 0; i < sz; i++)
p[i] = i % 256;
}
void
check(u_char *p, size_t sz)
{
size_t i;
for (i = 0; i < sz; i++)
if (p[i] != i % 256)
errx(1, "check");
}
int
main(int argc, char *argv[])
{
int count, p, r, i;
void * q;
size_t sz;
for (count = 0; count < 800000; count++) {
if (count % 10000 == 0) {
printf(".");
fflush(stdout);
}
p = arc4random_uniform(2);
i = arc4random_uniform(N);
switch (p) {
case 0:
if (a[i].p) {
#ifdef VERBOSE
printf("F %p\n", a[i].p);
#endif
if (a[i].p)
check(a[i].p, a[i].sz);
free(a[i].p);
a[i].p = NULL;
}
sz = size();
#ifdef VERBOSE
printf("M %zu=", sz);
#endif
r = arc4random_uniform(2);
a[i].p = r == 0 ? malloc_conceal(sz) : malloc(sz);
a[i].sz = sz;
#ifdef VERBOSE
printf("%p\n", a[i].p);
#endif
if (a[i].p)
fill(a[i].p, sz);
break;
case 1:
sz = size();
#ifdef VERBOSE
printf("R %p %zu=", a[i].p, sz);
#endif
q = realloc(a[i].p, sz);
#ifdef VERBOSE
printf("%p\n", q);
#endif
if (a[i].p && q)
check(q, a[i].sz < sz ? a[i].sz : sz);
if (q) {
a[i].p = q;
a[i].sz = sz;
fill(a[i].p, sz);
}
break;
}
}
for (i = 0; i < N; i++) {
if (a[i].p)
check(a[i].p, a[i].sz);
r = arc4random_uniform(2);
if (r)
free(a[i].p);
else
freezero(a[i].p, a[i].sz);
}
printf("\n");
return 0;
}
|
