huangcm
2024-12-18 9d29be7f7249789d6ffd0440067187a9f040c2cd
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
#include "test/jemalloc_test.h"
 
#ifdef JEMALLOC_FILL
const char *malloc_conf =
    "abort:false,junk:false,zero:true,redzone:false,quarantine:0";
#endif
 
static void
test_zero(size_t sz_min, size_t sz_max)
{
   uint8_t *s;
   size_t sz_prev, sz, i;
#define    MAGIC    ((uint8_t)0x61)
 
   sz_prev = 0;
   s = (uint8_t *)mallocx(sz_min, 0);
   assert_ptr_not_null((void *)s, "Unexpected mallocx() failure");
 
   for (sz = sallocx(s, 0); sz <= sz_max;
       sz_prev = sz, sz = sallocx(s, 0)) {
       if (sz_prev > 0) {
           assert_u_eq(s[0], MAGIC,
               "Previously allocated byte %zu/%zu is corrupted",
               ZU(0), sz_prev);
           assert_u_eq(s[sz_prev-1], MAGIC,
               "Previously allocated byte %zu/%zu is corrupted",
               sz_prev-1, sz_prev);
       }
 
       for (i = sz_prev; i < sz; i++) {
           assert_u_eq(s[i], 0x0,
               "Newly allocated byte %zu/%zu isn't zero-filled",
               i, sz);
           s[i] = MAGIC;
       }
 
       if (xallocx(s, sz+1, 0, 0) == sz) {
           s = (uint8_t *)rallocx(s, sz+1, 0);
           assert_ptr_not_null((void *)s,
               "Unexpected rallocx() failure");
       }
   }
 
   dallocx(s, 0);
#undef MAGIC
}
 
TEST_BEGIN(test_zero_small)
{
 
   test_skip_if(!config_fill);
   test_zero(1, SMALL_MAXCLASS-1);
}
TEST_END
 
TEST_BEGIN(test_zero_large)
{
 
   test_skip_if(!config_fill);
   test_zero(SMALL_MAXCLASS+1, large_maxclass);
}
TEST_END
 
TEST_BEGIN(test_zero_huge)
{
 
   test_skip_if(!config_fill);
   test_zero(large_maxclass+1, chunksize*2);
}
TEST_END
 
int
main(void)
{
 
   return (test(
       test_zero_small,
       test_zero_large,
       test_zero_huge));
}