1
2#include <stdlib.h>
3#include <stdint.h>
4#include <assert.h>
5#include <string.h>
6
7#include "../memcheck.h"
8
9struct cell
10{
11  struct cell *next;
12  int x;
13};
14
15struct pool
16{
17  size_t allocated;
18  size_t used;
19  uint8_t *buf;
20};
21
22void *
23allocate_from_pool(struct pool *p, size_t n)
24{
25  void *a = p->buf + p->used;
26  assert(p->used + n < p->allocated);
27  VALGRIND_MEMPOOL_ALLOC(p, a, n);
28  p->used += n;
29  return a;
30}
31
32struct pool *
33allocate_pool()
34{
35  struct pool *p = malloc(sizeof(struct pool));
36  assert(p);
37  p->allocated = 4096;
38  p->used = 0;
39  p->buf = malloc(p->allocated);
40  assert(p->buf);
41  memset(p->buf, 0, p->allocated);
42  VALGRIND_CREATE_MEMPOOL(p, 0, 0);
43  (void) VALGRIND_MAKE_MEM_NOACCESS(p->buf, p->allocated);
44  return p;
45}
46
47#define N 100
48
49/* flags */
50int static_roots = 0;
51int trim_pool    = 0;
52int destroy_pool = 0;
53void set_flags ( int n )
54{
55  switch (n) {
56     case 0:
57        static_roots = 0;
58        trim_pool    = 0;
59        destroy_pool = 0;
60        break;
61     case 1:
62        static_roots = 0;
63        trim_pool    = 1;
64        destroy_pool = 0;
65        break;
66     case 2:
67        static_roots = 0;
68        trim_pool    = 0;
69        destroy_pool = 1;
70        break;
71     case 3:
72        static_roots = 1;
73        trim_pool    = 0;
74        destroy_pool = 0;
75        break;
76     case 4:
77        static_roots = 1;
78        trim_pool    = 1;
79        destroy_pool = 0;
80        break;
81     case 5:
82        static_roots = 1;
83        trim_pool    = 0;
84        destroy_pool = 1;
85        break;
86     default:
87        assert(0);
88  }
89}
90
91
92struct cell *cells_static[N];
93
94
95int main( int argc, char** argv )
96{
97  struct cell *cells_local[N];
98  int arg;
99  size_t i;
100  struct pool *p      = allocate_pool();
101  struct cell **cells = static_roots ? cells_static : cells_local;
102
103  assert(argc == 2);
104  assert(argv[1]);
105  assert(strlen(argv[1]) == 1);
106  assert(argv[1][0] >= '0' && argv[1][0] <= '5');
107  arg = atoi( argv[1] );
108  set_flags( arg );
109
110  memset(cells_static, 0, sizeof(cells_static));
111  memset(cells_local,  0, sizeof(cells_local));
112
113  for (i = 0; i < N; ++i) {
114    cells[i] = allocate_from_pool(p, sizeof(struct cell));
115  }
116
117  if (trim_pool)
118  VALGRIND_MEMPOOL_TRIM(p,
119			p->buf+(10 * sizeof(struct cell)),
120			20 * sizeof(struct cell) + 2);
121
122  if (destroy_pool)
123  VALGRIND_DESTROY_MEMPOOL(p);
124
125  return 0;
126}
127