blob: d3bc173fd4539f13a0035263fde7397aa66efaf7 (
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
86
87
88
89
90
91
92
|
/* SPDX-License-Identifier: LGPL-2.1-or-later */
#include "mempool.h"
#include "random-util.h"
#include "tests.h"
struct element {
uint64_t value;
};
DEFINE_MEMPOOL(test_mempool, struct element, 8);
TEST(mempool_trim) {
#define NN 4000
struct element *a[NN];
size_t n_freed = 0;
assert_se(!test_mempool.first_pool);
assert_se(!test_mempool.freelist);
mempool_trim(&test_mempool);
for (size_t i = 0; i < NN; i++) {
assert_se(a[i] = mempool_alloc_tile(&test_mempool));
a[i]->value = i;
}
mempool_trim(&test_mempool);
/* free up to one third randomly */
size_t x = 0;
for (size_t i = 0; i < NN/3; i++) {
x = (x + random_u64()) % ELEMENTSOF(a);
assert_se(!a[x] || a[x]->value == x);
if (a[x])
n_freed ++;
a[x] = mempool_free_tile(&test_mempool, a[x]);
}
mempool_trim(&test_mempool);
/* free definitely at least one third */
for (size_t i = 2; i < NN; i += 3) {
assert_se(!a[i] || a[i]->value == i);
if (a[i])
n_freed ++;
a[i] = mempool_free_tile(&test_mempool, a[i]);
}
mempool_trim(&test_mempool);
/* Allocate another set of tiles, which will fill up the free list and allocate some new tiles */
struct element *b[NN];
for (size_t i = 0; i < NN; i++) {
assert_se(b[i] = mempool_alloc_tile(&test_mempool));
b[i]->value = ~(uint64_t) i;
}
mempool_trim(&test_mempool);
/* free everything from the original set*/
for (size_t i = 0; i < NN; i += 1) {
assert_se(!a[i] || a[i]->value == i);
if (a[i])
n_freed ++;
a[i] = mempool_free_tile(&test_mempool, a[i]);
}
mempool_trim(&test_mempool);
/* and now everything from the second set too */
for (size_t i = 0; i < NN; i += 1) {
assert_se(!b[i] || b[i]->value == ~(uint64_t) i);
if (b[i])
n_freed ++;
b[i] = mempool_free_tile(&test_mempool, b[i]);
}
assert_se(n_freed == NN * 2);
mempool_trim(&test_mempool);
assert_se(!test_mempool.first_pool);
assert_se(!test_mempool.freelist);
}
DEFINE_TEST_MAIN(LOG_DEBUG);
|
