style cleanup
[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
30  * Copyright (C) 2001 NaN Technologies B.V.
31  * Simple test of memory.
32  */
33
34
35
36 /* Number of chunks to test with */
37 #define NUM_BLOCKS 10
38
39 #include <stdio.h>
40 #include <string.h>
41 #include <stdlib.h>
42 #include "MEM_guardedalloc.h"
43
44 static void mem_error_cb(const char *errorStr)
45 {
46         fprintf(stderr, "%s", errorStr);
47         fflush(stderr);
48 }
49
50 int main(int argc, char *argv[])
51 {
52         int verbose       = 0;
53         int error_status  = 0;
54         int retval        = 0;
55         int *ip;
56
57         void *p[NUM_BLOCKS];
58         int i = 0;
59
60         /* ----------------------------------------------------------------- */
61         switch (argc) {
62         case 2:         
63                 verbose = atoi(argv[1]);
64                 if (verbose < 0) verbose = 0;
65                 break;          
66         case 1:
67         default:
68                 verbose = 0;
69         }
70         if (verbose) {
71                 fprintf(stderr,"\n*** Simple memory test\n|\n");
72         }
73
74         /* ----------------------------------------------------------------- */
75         /* Round one, do a normal allocation, and free the blocks again.     */
76         /* ----------------------------------------------------------------- */
77         /* flush mem lib output to stderr */
78         MEM_set_error_callback(mem_error_cb);
79         
80         for (i = 0; i < NUM_BLOCKS; i++) {
81                 int blocksize = 10000;
82                 char tagstring[1000];
83                 if (verbose > 1) printf("|--* Allocating block %d\n", i);
84                 sprintf(tagstring,"Memblock no. %d : ", i);
85                 p[i]= MEM_callocN(blocksize, strdup(tagstring));
86         }
87
88         /* report on that */
89         if (verbose > 1) MEM_printmemlist();
90
91         /* memory is there: test it */
92         error_status = MEM_check_memory_integrity();
93
94         if (verbose) {
95                 if (error_status) {
96                         fprintf(stderr, "|--* Memory test FAILED\n|\n");
97                 }
98                 else {
99                         fprintf(stderr, "|--* Memory tested as good (as it should be)\n|\n");
100                 }
101         } 
102
103         for (i = 0; i < NUM_BLOCKS; i++) {
104                 MEM_freeN(p[i]);
105         }
106
107         /* ----------------------------------------------------------------- */
108         /* Round two, do a normal allocation, and corrupt some blocks.       */
109         /* ----------------------------------------------------------------- */
110         /* switch off, because it will complain about some things.           */
111         MEM_set_error_callback(NULL);
112
113         for (i = 0; i < NUM_BLOCKS; i++) {
114                 int blocksize = 10000;
115                 char tagstring[1000];
116                 if (verbose > 1) printf("|--* Allocating block %d\n", i);
117                 sprintf(tagstring,"Memblock no. %d : ", i);
118                 p[i]= MEM_callocN(blocksize, strdup(tagstring));
119         }
120
121         /* now corrupt a few blocks...*/
122         ip = (int*) p[5] - 50;
123         for (i = 0; i< 1000; i++,ip++) *ip = i+1;
124         ip = (int*) p[6];
125         *(ip+10005) = 0;
126         
127         retval = MEM_check_memory_integrity();
128
129         /* the test should have failed */
130         error_status |= !retval;                
131         if (verbose) {
132                 if (retval) {
133                         fprintf(stderr, "|--* Memory test failed (as it should be)\n");
134                 }
135                 else {
136                         fprintf(stderr, "|--* Memory test FAILED to find corrupted blocks \n");
137                 }
138         } 
139         
140         for (i = 0; i < NUM_BLOCKS; i++) {
141                 MEM_freeN(p[i]);
142         }
143
144
145         if (verbose && error_status) {
146                 fprintf(stderr,"|--* Memory was corrupted\n");
147         }
148         /* ----------------------------------------------------------------- */ 
149         if (verbose) {
150                 if (error_status) {
151                         fprintf(stderr,"|\n|--* Errors were detected\n");
152                 }
153                 else {
154                         fprintf(stderr,"|\n|--* Test exited succesfully\n");
155                 }
156                 
157                 fprintf(stderr,"|\n*** Finished test\n\n");
158         }
159         return error_status;
160 }
161
162