515ae5a5e370f4b3af1c09c7e5c95f6a21c78cbc
[fio.git] / t / stest.c
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <assert.h>
4
5 #include "../smalloc.h"
6 #include "../flist.h"
7 #include "../arch/arch.h"
8 #include "debug.h"
9
10 #define MAGIC1  0xa9b1c8d2
11 #define MAGIC2  0xf0a1e9b3
12
13 #define LOOPS   32
14 #define MAXSMALLOC      120*1024*1024UL
15 #define LARGESMALLOC    128*1024U
16
17 struct elem {
18         unsigned int magic1;
19         struct flist_head list;
20         unsigned int magic2;
21         unsigned int size;
22 };
23
24 static FLIST_HEAD(list);
25
26 static int do_rand_allocs(void)
27 {
28         unsigned int size, nr, rounds = 0;
29         unsigned long total;
30         struct elem *e;
31         bool error;
32
33         while (rounds++ < LOOPS) {
34 #ifdef STEST_SEED
35                 srand(MAGIC1);
36 #endif
37                 error = false;
38                 nr = total = 0;
39                 while (total < MAXSMALLOC) {
40                         size = 8 * sizeof(struct elem) + (int) (999.0 * (rand() / (RAND_MAX + 1.0)));
41                         e = smalloc(size);
42                         if (!e) {
43                                 printf("fail at %lu, size %u\n", total, size);
44                                 break;
45                         }
46                         e->magic1 = MAGIC1;
47                         e->magic2 = MAGIC2;
48                         e->size = size;
49                         total += size;
50                         flist_add_tail(&e->list, &list);
51                         nr++;
52                 }
53
54                 printf("Got items: %u\n", nr);
55
56                 while (!flist_empty(&list)) {
57                         e = flist_entry(list.next, struct elem, list);
58                         assert(e->magic1 == MAGIC1);
59                         assert(e->magic2 == MAGIC2);
60                         total -= e->size;
61                         flist_del(&e->list);
62                         sfree(e);
63
64                         if (!error) {
65                                 e = smalloc(LARGESMALLOC);
66                                 if (!e) {
67                                         error = true;
68                                         printf("failure allocating %u bytes at %lu allocated during sfree phase\n",
69                                                 LARGESMALLOC, total);
70                                 }
71                                 else
72                                         sfree(e);
73                         }
74                 }
75         }
76
77         return 0;
78 }
79
80 int main(int argc, char *argv[])
81 {
82         arch_init(argv);
83         sinit();
84         debug_init();
85
86         do_rand_allocs();
87         smalloc_debug(0);       /* free and total blocks should match */
88
89         scleanup();
90         return 0;
91 }