Memory allocator: Clarify consistency check function
[blender-staging.git] / intern / guardedalloc / test / simpletest / memtest.c
1 /**
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): none yet.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /**
29  * Copyright (C) 2001 NaN Technologies B.V.
30  * Simple test of memory.
31  */
32
33 /* To compile run:
34  * gcc -DWITH_GUARDEDALLOC -I../../ -I../../../atomic/ memtest.c  ../../intern/mallocn.c -o simpletest
35  */
36
37 /* Number of chunks to test with */
38 #define NUM_BLOCKS 10
39
40 #include <stdio.h>
41 #include <string.h>
42 #include <stdlib.h>
43 #include "MEM_guardedalloc.h"
44
45 static void mem_error_cb(const char *errorStr)
46 {
47         fprintf(stderr, "%s", errorStr);
48         fflush(stderr);
49 }
50
51 int main(int argc, char *argv[])
52 {
53         int verbose       = 0;
54         int error_status  = 0;
55         int retval        = 0;
56         int *ip;
57
58         void *p[NUM_BLOCKS];
59         int i = 0;
60
61         /* ----------------------------------------------------------------- */
62         switch (argc) {
63         case 2:         
64                 verbose = atoi(argv[1]);
65                 if (verbose < 0) verbose = 0;
66                 break;          
67         case 1:
68         default:
69                 verbose = 0;
70         }
71         if (verbose) {
72                 fprintf(stderr,"\n*** Simple memory test\n|\n");
73         }
74
75         /* ----------------------------------------------------------------- */
76         /* Round one, do a normal allocation, and free the blocks again.     */
77         /* ----------------------------------------------------------------- */
78         /* flush mem lib output to stderr */
79         MEM_set_error_callback(mem_error_cb);
80         
81         for (i = 0; i < NUM_BLOCKS; i++) {
82                 int blocksize = 10000;
83                 char tagstring[1000];
84                 if (verbose > 1) printf("|--* Allocating block %d\n", i);
85                 sprintf(tagstring,"Memblock no. %d : ", i);
86                 p[i]= MEM_callocN(blocksize, strdup(tagstring));
87         }
88
89         /* report on that */
90         if (verbose > 1) MEM_printmemlist();
91
92         /* memory is there: test it */
93         error_status = MEM_consistency_check();
94
95         if (verbose) {
96                 if (error_status) {
97                         fprintf(stderr, "|--* Memory test FAILED\n|\n");
98                 }
99                 else {
100                         fprintf(stderr, "|--* Memory tested as good (as it should be)\n|\n");
101                 }
102         } 
103
104         for (i = 0; i < NUM_BLOCKS; i++) {
105                 MEM_freeN(p[i]);
106         }
107
108         /* ----------------------------------------------------------------- */
109         /* Round two, do a normal allocation, and corrupt some blocks.       */
110         /* ----------------------------------------------------------------- */
111         /* switch off, because it will complain about some things.           */
112         MEM_set_error_callback(NULL);
113
114         for (i = 0; i < NUM_BLOCKS; i++) {
115                 int blocksize = 10000;
116                 char tagstring[1000];
117                 if (verbose > 1) printf("|--* Allocating block %d\n", i);
118                 sprintf(tagstring,"Memblock no. %d : ", i);
119                 p[i]= MEM_callocN(blocksize, strdup(tagstring));
120         }
121
122         /* now corrupt a few blocks...*/
123         ip = (int*) p[5] - 50;
124         for (i = 0; i< 1000; i++,ip++) *ip = i+1;
125         ip = (int*) p[6];
126         *(ip+10005) = 0;
127         
128         retval = MEM_consistency_check();
129
130         /* the test should have failed */
131         error_status |= !retval;                
132         if (verbose) {
133                 if (retval) {
134                         fprintf(stderr, "|--* Memory test failed (as it should be)\n");
135                 }
136                 else {
137                         fprintf(stderr, "|--* Memory test FAILED to find corrupted blocks \n");
138                 }
139         } 
140         
141         for (i = 0; i < NUM_BLOCKS; i++) {
142                 MEM_freeN(p[i]);
143         }
144
145
146         if (verbose && error_status) {
147                 fprintf(stderr,"|--* Memory was corrupted\n");
148         }
149         /* ----------------------------------------------------------------- */ 
150         if (verbose) {
151                 if (error_status) {
152                         fprintf(stderr,"|\n|--* Errors were detected\n");
153                 }
154                 else {
155                         fprintf(stderr,"|\n|--* Test exited succesfully\n");
156                 }
157                 
158                 fprintf(stderr,"|\n*** Finished test\n\n");
159         }
160         return error_status;
161 }
162
163