Cleaned up alot more memory leaks. (still get 720 leaks just running demo1.dem)
[xonotic/darkplaces.git] / zone.c
1 /*
2 Copyright (C) 1996-1997 Id Software, Inc.
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.
12
13 See the GNU General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the Free Software
17 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
18
19 */
20 // Z_zone.c
21
22 #include "quakedef.h"
23
24 cvar_t developer_memory = {0, "developer_memory", "0"};
25 cvar_t developer_memorydebug = {0, "developer_memorydebug", "0"};
26
27 mempool_t *poolchain = NULL;
28
29 void *_Mem_Alloc(mempool_t *pool, int size, const char *filename, int fileline)
30 {
31 #if MEMCLUMPING
32         int i, j, k, needed, endbit, largest;
33         memclump_t *clump, **clumpchainpointer;
34 #endif
35         memheader_t *mem;
36         if (size <= 0)
37                 return NULL;
38         if (pool == NULL)
39                 Sys_Error("Mem_Alloc: pool == NULL (alloc at %s:%i)", filename, fileline);
40         if (developer.integer && developer_memory.integer)
41                 Con_Printf("Mem_Alloc: pool %s, file %s:%i, size %i bytes\n", pool->name, filename, fileline, size);
42         if (developer.integer && developer_memorydebug.integer)
43                 _Mem_CheckSentinelsGlobal(filename, fileline);
44         pool->totalsize += size;
45 #if MEMCLUMPING
46         if (size < 4096)
47         {
48                 // clumping
49                 needed = (sizeof(memheader_t) + size + sizeof(int) + (MEMUNIT - 1)) / MEMUNIT;
50                 endbit = MEMBITS - needed;
51                 for (clumpchainpointer = &pool->clumpchain;*clumpchainpointer;clumpchainpointer = &(*clumpchainpointer)->chain)
52                 {
53                         clump = *clumpchainpointer;
54                         if (clump->sentinel1 != MEMCLUMP_SENTINEL)
55                                 Sys_Error("Mem_Alloc: trashed clump sentinel 1 (alloc at %s:%d)", filename, fileline);
56                         if (clump->sentinel2 != MEMCLUMP_SENTINEL)
57                                 Sys_Error("Mem_Alloc: trashed clump sentinel 2 (alloc at %s:%d)", filename, fileline);
58                         if (clump->largestavailable >= needed)
59                         {
60                                 largest = 0;
61                                 for (i = 0;i < endbit;i++)
62                                 {
63                                         if (clump->bits[i >> 5] & (1 << (i & 31)))
64                                                 continue;
65                                         k = i + needed;
66                                         for (j = i;i < k;i++)
67                                                 if (clump->bits[i >> 5] & (1 << (i & 31)))
68                                                         goto loopcontinue;
69                                         goto choseclump;
70         loopcontinue:;
71                                         if (largest < j - i)
72                                                 largest = j - i;
73                                 }
74                                 // since clump falsely advertised enough space (nothing wrong
75                                 // with that), update largest count to avoid wasting time in
76                                 // later allocations
77                                 clump->largestavailable = largest;
78                         }
79                 }
80                 pool->realsize += sizeof(memclump_t);
81                 clump = malloc(sizeof(memclump_t));
82                 if (clump == NULL)
83                         Sys_Error("Mem_Alloc: out of memory (alloc at %s:%i)", filename, fileline);
84                 memset(clump, 0, sizeof(memclump_t));
85                 *clumpchainpointer = clump;
86                 clump->sentinel1 = MEMCLUMP_SENTINEL;
87                 clump->sentinel2 = MEMCLUMP_SENTINEL;
88                 clump->chain = NULL;
89                 clump->blocksinuse = 0;
90                 clump->largestavailable = MEMBITS - needed;
91                 j = 0;
92 choseclump:
93                 mem = (memheader_t *)((qbyte *) clump->block + j * MEMUNIT);
94                 mem->clump = clump;
95                 clump->blocksinuse += needed;
96                 for (i = j + needed;j < i;j++)
97                         clump->bits[j >> 5] |= (1 << (j & 31));
98         }
99         else
100         {
101                 // big allocations are not clumped
102 #endif
103                 pool->realsize += sizeof(memheader_t) + size + sizeof(int);
104                 mem = malloc(sizeof(memheader_t) + size + sizeof(int));
105                 if (mem == NULL)
106                         Sys_Error("Mem_Alloc: out of memory (alloc at %s:%i)", filename, fileline);
107 #if MEMCLUMPING
108                 mem->clump = NULL;
109         }
110 #endif
111         mem->filename = filename;
112         mem->fileline = fileline;
113         mem->size = size;
114         mem->pool = pool;
115         mem->sentinel1 = MEMHEADER_SENTINEL1;
116         // we have to use only a single byte for this sentinel, because it may not be aligned, and some platforms can't use unaligned accesses
117         *((qbyte *) mem + sizeof(memheader_t) + mem->size) = MEMHEADER_SENTINEL2;
118         // append to head of list
119         mem->next = pool->chain;
120         mem->prev = NULL;
121         pool->chain = mem;
122         if (mem->next)
123                 mem->next->prev = mem;
124         memset((void *)((qbyte *) mem + sizeof(memheader_t)), 0, mem->size);
125         return (void *)((qbyte *) mem + sizeof(memheader_t));
126 }
127
128 void _Mem_Free(void *data, const char *filename, int fileline)
129 {
130 #if MEMCLUMPING
131         int i, firstblock, endblock;
132         memclump_t *clump, **clumpchainpointer;
133 #endif
134         memheader_t *mem;
135         mempool_t *pool;
136         if (data == NULL)
137                 Sys_Error("Mem_Free: data == NULL (called at %s:%i)", filename, fileline);
138
139         mem = (memheader_t *)((qbyte *) data - sizeof(memheader_t));
140         if (mem->sentinel1 != MEMHEADER_SENTINEL1)
141                 Sys_Error("Mem_Free: trashed header sentinel 1 (alloc at %s:%i, free at %s:%i)", mem->filename, mem->fileline, filename, fileline);
142         if (*((qbyte *) mem + sizeof(memheader_t) + mem->size) != MEMHEADER_SENTINEL2)
143                 Sys_Error("Mem_Free: trashed header sentinel 2 (alloc at %s:%i, free at %s:%i)", mem->filename, mem->fileline, filename, fileline);
144         pool = mem->pool;
145         if (developer.integer && developer_memory.integer)
146                 Con_Printf("Mem_Free: pool %s, alloc %s:%i, free %s:%i, size %i bytes\n", pool->name, mem->filename, mem->fileline, filename, fileline, mem->size);
147         // unlink memheader from doubly linked list
148         if ((mem->prev ? mem->prev->next != mem : pool->chain != mem) || (mem->next && mem->next->prev != mem))
149                 Sys_Error("Mem_Free: not allocated or double freed (free at %s:%i)", filename, fileline);
150         if (mem->prev)
151                 mem->prev->next = mem->next;
152         else
153                 pool->chain = mem->next;
154         if (mem->next)
155                 mem->next->prev = mem->prev;
156         // memheader has been unlinked, do the actual free now
157         pool->totalsize -= mem->size;
158 #if MEMCLUMPING
159         if ((clump = mem->clump))
160         {
161                 if (clump->sentinel1 != MEMCLUMP_SENTINEL)
162                         Sys_Error("Mem_Free: trashed clump sentinel 1 (free at %s:%i)", filename, fileline);
163                 if (clump->sentinel2 != MEMCLUMP_SENTINEL)
164                         Sys_Error("Mem_Free: trashed clump sentinel 2 (free at %s:%i)", filename, fileline);
165                 firstblock = ((qbyte *) mem - (qbyte *) clump->block);
166                 if (firstblock & (MEMUNIT - 1))
167                         Sys_Error("Mem_Free: address not valid in clump (free at %s:%i)", filename, fileline);
168                 firstblock /= MEMUNIT;
169                 endblock = firstblock + ((sizeof(memheader_t) + mem->size + sizeof(int) + (MEMUNIT - 1)) / MEMUNIT);
170                 clump->blocksinuse -= endblock - firstblock;
171                 // could use &, but we know the bit is set
172                 for (i = firstblock;i < endblock;i++)
173                         clump->bits[i >> 5] -= (1 << (i & 31));
174                 if (clump->blocksinuse <= 0)
175                 {
176                         // unlink from chain
177                         for (clumpchainpointer = &pool->clumpchain;*clumpchainpointer;clumpchainpointer = &(*clumpchainpointer)->chain)
178                         {
179                                 if (*clumpchainpointer == clump)
180                                 {
181                                         *clumpchainpointer = clump->chain;
182                                         break;
183                                 }
184                         }
185                         pool->realsize -= sizeof(memclump_t);
186                         memset(clump, 0xBF, sizeof(memclump_t));
187                         free(clump);
188                 }
189                 else
190                 {
191                         // clump still has some allocations
192                         // force re-check of largest available space on next alloc
193                         clump->largestavailable = MEMBITS - clump->blocksinuse;
194                 }
195         }
196         else
197         {
198 #endif
199                 pool->realsize -= sizeof(memheader_t) + mem->size + sizeof(int);
200                 if (developer.integer)
201                         memset(mem, 0xBF, sizeof(memheader_t) + mem->size + sizeof(int));
202                 free(mem);
203 #if MEMCLUMPING
204         }
205 #endif
206 }
207
208 mempool_t *_Mem_AllocPool(const char *name, int flags, mempool_t *parent, const char *filename, int fileline)
209 {
210         mempool_t *pool;
211         pool = malloc(sizeof(mempool_t));
212         if (pool == NULL)
213                 Sys_Error("Mem_AllocPool: out of memory (allocpool at %s:%i)", filename, fileline);
214         memset(pool, 0, sizeof(mempool_t));
215         pool->sentinel1 = MEMHEADER_SENTINEL1;
216         pool->sentinel2 = MEMHEADER_SENTINEL1;
217         pool->filename = filename;
218         pool->fileline = fileline;
219         pool->flags = flags;
220         pool->chain = NULL;
221         pool->totalsize = 0;
222         pool->realsize = sizeof(mempool_t);
223         strlcpy (pool->name, name, sizeof (pool->name));
224         pool->parent = parent;
225         pool->next = poolchain;
226         poolchain = pool;
227         return pool;
228 }
229
230 void _Mem_FreePool(mempool_t **pool, const char *filename, int fileline)
231 {
232         mempool_t **chainaddress, *iter, *temp;
233         
234         if (*pool)
235         {
236                 if ((*pool)->sentinel1 != MEMHEADER_SENTINEL1)
237                         Sys_Error("Mem_FreePool: trashed pool sentinel 1 (allocpool at %s:%i, freepool at %s:%i)", (*pool)->filename, (*pool)->fileline, filename, fileline);
238                 if ((*pool)->sentinel2 != MEMHEADER_SENTINEL1)
239                         Sys_Error("Mem_FreePool: trashed pool sentinel 2 (allocpool at %s:%i, freepool at %s:%i)", (*pool)->filename, (*pool)->fileline, filename, fileline);
240                 // unlink pool from chain
241                 for (chainaddress = &poolchain;*chainaddress && *chainaddress != *pool;chainaddress = &((*chainaddress)->next));
242                 if (*chainaddress != *pool)
243                         Sys_Error("Mem_FreePool: pool already free (freepool at %s:%i)", filename, fileline);
244                 *chainaddress = (*pool)->next;
245
246                 // free memory owned by the pool
247                 while ((*pool)->chain)
248                         Mem_Free((void *)((qbyte *) (*pool)->chain + sizeof(memheader_t)));
249
250                 // free child pools, too
251                 for(iter = poolchain; iter; temp = iter = iter->next)
252                         if(iter->parent == *pool)
253                                 _Mem_FreePool(&temp, filename, fileline);
254                 
255                 // free the pool itself
256                 memset(*pool, 0xBF, sizeof(mempool_t));
257                 free(*pool);
258                 *pool = NULL;
259         }
260 }
261
262 void _Mem_EmptyPool(mempool_t *pool, const char *filename, int fileline)
263 {
264         mempool_t *chainaddress;
265         
266         if (pool == NULL)
267                 Sys_Error("Mem_EmptyPool: pool == NULL (emptypool at %s:%i)", filename, fileline);
268         if (pool->sentinel1 != MEMHEADER_SENTINEL1)
269                 Sys_Error("Mem_EmptyPool: trashed pool sentinel 1 (allocpool at %s:%i, emptypool at %s:%i)", pool->filename, pool->fileline, filename, fileline);
270         if (pool->sentinel2 != MEMHEADER_SENTINEL1)
271                 Sys_Error("Mem_EmptyPool: trashed pool sentinel 2 (allocpool at %s:%i, emptypool at %s:%i)", pool->filename, pool->fileline, filename, fileline);
272
273         // free memory owned by the pool
274         while (pool->chain)
275                 Mem_Free((void *)((qbyte *) pool->chain + sizeof(memheader_t)));
276
277         // empty child pools, too
278         for(chainaddress = poolchain; chainaddress; chainaddress = chainaddress->next)
279                 if(chainaddress->parent == pool)
280                         _Mem_EmptyPool(chainaddress, filename, fileline);
281
282 }
283
284 void _Mem_CheckSentinels(void *data, const char *filename, int fileline)
285 {
286         memheader_t *mem;
287
288         if (data == NULL)
289                 Sys_Error("Mem_CheckSentinels: data == NULL (sentinel check at %s:%i)", filename, fileline);
290
291         mem = (memheader_t *)((qbyte *) data - sizeof(memheader_t));
292         if (mem->sentinel1 != MEMHEADER_SENTINEL1)
293                 Sys_Error("Mem_CheckSentinels: trashed header sentinel 1 (block allocated at %s:%i, sentinel check at %s:%i)", mem->filename, mem->fileline, filename, fileline);
294         if (*((qbyte *) mem + sizeof(memheader_t) + mem->size) != MEMHEADER_SENTINEL2)
295                 Sys_Error("Mem_CheckSentinels: trashed header sentinel 2 (block allocated at %s:%i, sentinel check at %s:%i)", mem->filename, mem->fileline, filename, fileline);
296 }
297
298 #if MEMCLUMPING
299 static void _Mem_CheckClumpSentinels(memclump_t *clump, const char *filename, int fileline)
300 {
301         // this isn't really very useful
302         if (clump->sentinel1 != MEMCLUMP_SENTINEL)
303                 Sys_Error("Mem_CheckClumpSentinels: trashed sentinel 1 (sentinel check at %s:%i)", filename, fileline);
304         if (clump->sentinel2 != MEMCLUMP_SENTINEL)
305                 Sys_Error("Mem_CheckClumpSentinels: trashed sentinel 2 (sentinel check at %s:%i)", filename, fileline);
306 }
307 #endif
308
309 void _Mem_CheckSentinelsGlobal(const char *filename, int fileline)
310 {
311         memheader_t *mem;
312 #if MEMCLUMPING
313         memclump_t *clump;
314 #endif
315         mempool_t *pool;
316         for (pool = poolchain;pool;pool = pool->next)
317         {
318                 if (pool->sentinel1 != MEMHEADER_SENTINEL1)
319                         Sys_Error("Mem_CheckSentinelsGlobal: trashed pool sentinel 1 (allocpool at %s:%i, sentinel check at %s:%i)", pool->filename, pool->fileline, filename, fileline);
320                 if (pool->sentinel2 != MEMHEADER_SENTINEL1)
321                         Sys_Error("Mem_CheckSentinelsGlobal: trashed pool sentinel 2 (allocpool at %s:%i, sentinel check at %s:%i)", pool->filename, pool->fileline, filename, fileline);
322         }
323         for (pool = poolchain;pool;pool = pool->next)
324                 for (mem = pool->chain;mem;mem = mem->next)
325                         _Mem_CheckSentinels((void *)((qbyte *) mem + sizeof(memheader_t)), filename, fileline);
326 #if MEMCLUMPING
327         for (pool = poolchain;pool;pool = pool->next)
328                 for (clump = pool->clumpchain;clump;clump = clump->chain)
329                         _Mem_CheckClumpSentinels(clump, filename, fileline);
330 #endif
331 }
332
333 qboolean Mem_IsAllocated(mempool_t *pool, void *data)
334 {
335         memheader_t *header;
336         memheader_t *target;
337
338     target = (memheader_t *)((qbyte *) data - sizeof(memheader_t));
339         for( header = pool->chain ; header ; header = header->next )
340                 if( header == target )
341                         return true;
342         return false;
343 }
344
345
346 // used for temporary memory allocations around the engine, not for longterm
347 // storage, if anything in this pool stays allocated during gameplay, it is
348 // considered a leak
349 mempool_t *tempmempool;
350 // only for zone
351 mempool_t *zonemempool;
352
353 void Mem_PrintStats(void)
354 {
355         int count = 0, size = 0;
356         mempool_t *pool;
357         memheader_t *mem;
358         Mem_CheckSentinelsGlobal();
359         for (pool = poolchain;pool;pool = pool->next)
360         {
361                 count++;
362                 size += pool->totalsize;
363         }
364         Con_Printf("%i memory pools, totalling %i bytes (%.3fMB)\n", count, size, size / 1048576.0);
365         for (pool = poolchain;pool;pool = pool->next)
366         {
367                 if ((pool->flags & POOLFLAG_TEMP) && pool->chain)
368                 {
369                         Con_Printf("Memory pool %p has sprung a leak totalling %i bytes (%.3fMB)!  Listing contents...\n", pool, pool->totalsize, pool->totalsize / 1048576.0);
370                         for (mem = pool->chain;mem;mem = mem->next)
371                                 Con_Printf("%10i bytes allocated at %s:%i\n", mem->size, mem->filename, mem->fileline);
372                 }
373         }
374 }
375
376 void Mem_PrintList(int listallocations)
377 {
378         mempool_t *pool;
379         memheader_t *mem;
380         Mem_CheckSentinelsGlobal();
381         Con_Print("memory pool list:\n"
382                    "size    name\n");
383         for (pool = poolchain;pool;pool = pool->next)
384         {
385                 Con_Printf("%10ik (%10ik actual) %s (%+i byte change) %s\n", (pool->totalsize + 1023) / 1024, (pool->realsize + 1023) / 1024, pool->name, pool->totalsize - pool->lastchecksize, (pool->flags & POOLFLAG_TEMP) ? "TEMP" : "");
386                 pool->lastchecksize = pool->totalsize;
387                 if (listallocations)
388                         for (mem = pool->chain;mem;mem = mem->next)
389                                 Con_Printf("%10i bytes allocated at %s:%i\n", mem->size, mem->filename, mem->fileline);
390         }
391 }
392
393 void MemList_f(void)
394 {
395         switch(Cmd_Argc())
396         {
397         case 1:
398                 Mem_PrintList(false);
399                 Mem_PrintStats();
400                 break;
401         case 2:
402                 if (!strcmp(Cmd_Argv(1), "all"))
403                 {
404                         Mem_PrintList(true);
405                         Mem_PrintStats();
406                         break;
407                 }
408                 // drop through
409         default:
410                 Con_Print("MemList_f: unrecognized options\nusage: memlist [all]\n");
411                 break;
412         }
413 }
414
415 extern void R_TextureStats_PrintTotal(void);
416 void MemStats_f(void)
417 {
418         Mem_CheckSentinelsGlobal();
419         R_TextureStats_PrintTotal();
420         Mem_PrintStats();
421 }
422
423
424 /*
425 ========================
426 Memory_Init
427 ========================
428 */
429 void Memory_Init (void)
430 {
431         tempmempool = Mem_AllocPool("Temporary Memory", POOLFLAG_TEMP, NULL);
432         zonemempool = Mem_AllocPool("Zone", 0, NULL);
433         poolchain = NULL;
434 }
435
436 void Memory_Shutdown (void)
437 {
438 //      Mem_FreePool (&zonemempool);
439 //      Mem_FreePool (&tempmempool);
440 }
441
442 void Memory_Init_Commands (void)
443 {
444         Cmd_AddCommand ("memstats", MemStats_f);
445         Cmd_AddCommand ("memlist", MemList_f);
446         Cvar_RegisterVariable (&developer_memory);
447         Cvar_RegisterVariable (&developer_memorydebug);
448 }
449