Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Fix HAVE_FOOBAR flags handling
[simgrid.git] / teshsuite / xbt / mmalloc / mmalloc_test.c
1 /* Copyright (c) 2012-2014. The SimGrid Team.
2  * All rights reserved.                                                     */
3
4 /* This program is free software; you can redistribute it and/or modify it
5  * under the terms of the license (GNU LGPL) which comes with this package. */
6
7 #include "xbt/mmalloc.h"
8 #include "xbt.h"
9 #include <stdio.h>
10 #include <assert.h>
11 #include <fcntl.h>
12 #include <sys/stat.h>
13 #include <stdlib.h>
14 #include <string.h>
15 #include <unistd.h>
16
17 XBT_LOG_NEW_DEFAULT_CATEGORY(test,"this test");
18
19 #define BUFFSIZE 204800
20 #define TESTSIZE 100
21 #define size_of_block(i) (((i % 50)+1)* 100)
22
23 int main(int argc, char**argv)
24 {
25   void *heapA;
26   void *pointers[TESTSIZE];
27   xbt_init(&argc,argv);
28
29   XBT_INFO("Allocating a new heap");
30   unsigned long mask = ~((unsigned long)xbt_pagesize - 1);
31   void *addr = (void*)(((unsigned long)sbrk(0) + BUFFSIZE) & mask);
32   heapA = xbt_mheap_new(-1, addr);
33   if (heapA == NULL) {
34     perror("attach 1 failed");
35     fprintf(stderr, "bye\n");
36     exit(1);
37   }
38
39   XBT_INFO("HeapA allocated");
40
41   int i, size;
42   for (i = 0; i < TESTSIZE; i++) {
43     size = size_of_block(i);
44     pointers[i] = mmalloc(heapA, size);
45     XBT_INFO("%d bytes allocated with offset %tx", size, ((char*)pointers[i])-((char*)heapA));
46   }
47   XBT_INFO("All blocks were correctly allocated. Free every second block");
48   for (i = 0; i < TESTSIZE; i+=2) {
49     mfree(heapA,pointers[i]);
50   }
51   XBT_INFO("Memset every second block to zero (yeah, they are not currently allocated :)");
52   for (i = 0; i < TESTSIZE; i+=2) {
53     size = size_of_block(i);
54     memset(pointers[i],0, size);
55   }
56   XBT_INFO("Re-allocate every second block");
57   for (i = 0; i < TESTSIZE; i+=2) {
58     size = size_of_block(i);
59     pointers[i] = mmalloc(heapA, size);
60   }
61
62   XBT_INFO("free all blocks (each one twice, to check that double free are correctly catched)");
63   for (i = 0; i < TESTSIZE; i++) {
64     xbt_ex_t e;
65     int gotit = 1;
66
67     mfree(heapA, pointers[i]);
68     TRY {
69       mfree(heapA, pointers[i]);
70       gotit = 0;
71     } CATCH(e) {
72       xbt_ex_free(e);
73     }
74     if (!gotit)
75       xbt_die("FAIL: A double-free went undetected (for size:%d)",size_of_block(i));
76   }
77
78   XBT_INFO("free again all blocks (to really check that double free are correctly catched)");
79   for (i = 0; i < TESTSIZE; i++) {
80     xbt_ex_t e;
81     int gotit = 1;
82
83     TRY {
84       mfree(heapA, pointers[i]);
85       gotit = 0;
86     } CATCH(e) {
87       xbt_ex_free(e);
88     }
89     if (!gotit)
90       xbt_die("FAIL: A double-free went undetected (for size:%d)",size_of_block(i));
91   }
92
93   XBT_INFO("Damnit, I cannot break mmalloc this time. That's SO disappointing.");
94   return 0;
95 }