This is a live mirror of the Perl 5 development currently hosted at https://github.com/perl/perl5
run/locale.t: Add explanation for when tests fail
[perl5.git] / win32 / vmem.h
CommitLineData
7766f137
GS
1/* vmem.h
2 *
3 * (c) 1999 Microsoft Corporation. All rights reserved.
4 * Portions (c) 1999 ActiveState Tool Corp, http://www.ActiveState.com/
5 *
6 * You may distribute under the terms of either the GNU General Public
7 * License or the Artistic License, as specified in the README file.
8 *
f57e8d3b 9 * Options:
7766f137 10 *
f57e8d3b 11 * Defining _USE_MSVCRT_MEM_ALLOC will cause all memory allocations
b75e1177
SH
12 * to be forwarded to the compiler's MSVCR*.DLL. Defining _USE_LINKED_LIST as
13 * well will track all allocations in a doubly linked list, so that the host can
f57e8d3b
GS
14 * free all memory allocated when it goes away.
15 * If _USE_MSVCRT_MEM_ALLOC is not defined then Knuth's boundary tag algorithm
16 * is used; defining _USE_BUDDY_BLOCKS will use Knuth's algorithm R
17 * (Buddy system reservation)
18 *
19 */
20
21#ifndef ___VMEM_H_INC___
22#define ___VMEM_H_INC___
23
45496817 24#ifndef UNDER_CE
222c300a 25#define _USE_MSVCRT_MEM_ALLOC
7bd379e8 26#endif
222c300a 27#define _USE_LINKED_LIST
f57e8d3b
GS
28
29// #define _USE_BUDDY_BLOCKS
30
31// #define _DEBUG_MEM
32#ifdef _DEBUG_MEM
33#define ASSERT(f) if(!(f)) DebugBreak();
34
35inline void MEMODS(char *str)
36{
37 OutputDebugString(str);
38 OutputDebugString("\n");
39}
40
41inline void MEMODSlx(char *str, long x)
42{
43 char szBuffer[512];
44 sprintf(szBuffer, "%s %lx\n", str, x);
45 OutputDebugString(szBuffer);
46}
47
48#define WALKHEAP() WalkHeap(0)
49#define WALKHEAPTRACE() WalkHeap(1)
50
51#else
52
53#define ASSERT(f)
54#define MEMODS(x)
55#define MEMODSlx(x, y)
56#define WALKHEAP()
57#define WALKHEAPTRACE()
58
59#endif
60
61#ifdef _USE_MSVCRT_MEM_ALLOC
62
63#ifndef _USE_LINKED_LIST
64// #define _USE_LINKED_LIST
65#endif
66
67/*
b75e1177
SH
68 * Pass all memory requests through to the compiler's msvcr*.dll.
69 * Optionaly track by using a doubly linked header.
f57e8d3b
GS
70 */
71
f57e8d3b 72#ifdef _USE_LINKED_LIST
222c300a 73class VMem;
f57e8d3b
GS
74typedef struct _MemoryBlockHeader* PMEMORY_BLOCK_HEADER;
75typedef struct _MemoryBlockHeader {
76 PMEMORY_BLOCK_HEADER pNext;
77 PMEMORY_BLOCK_HEADER pPrev;
222c300a 78 VMem *owner;
f57e8d3b
GS
79} MEMORY_BLOCK_HEADER, *PMEMORY_BLOCK_HEADER;
80#endif
81
82class VMem
83{
84public:
85 VMem();
86 ~VMem();
624a1c42 87 void* Malloc(size_t size);
b75e1177
SH
88 void* Realloc(void* pMem, size_t size);
89 void Free(void* pMem);
90 void GetLock(void);
91 void FreeLock(void);
92 int IsLocked(void);
93 long Release(void);
94 long AddRef(void);
f57e8d3b
GS
95
96 inline BOOL CreateOk(void)
97 {
98 return TRUE;
99 };
100
101protected:
102#ifdef _USE_LINKED_LIST
103 void LinkBlock(PMEMORY_BLOCK_HEADER ptr)
104 {
105 PMEMORY_BLOCK_HEADER next = m_Dummy.pNext;
106 m_Dummy.pNext = ptr;
107 ptr->pPrev = &m_Dummy;
108 ptr->pNext = next;
222c300a 109 ptr->owner = this;
f57e8d3b
GS
110 next->pPrev = ptr;
111 }
112 void UnlinkBlock(PMEMORY_BLOCK_HEADER ptr)
113 {
114 PMEMORY_BLOCK_HEADER next = ptr->pNext;
115 PMEMORY_BLOCK_HEADER prev = ptr->pPrev;
116 prev->pNext = next;
117 next->pPrev = prev;
118 }
119
120 MEMORY_BLOCK_HEADER m_Dummy;
624a1c42 121 CRITICAL_SECTION m_cs; // access lock
f57e8d3b
GS
122#endif
123
124 long m_lRefCount; // number of current users
f57e8d3b
GS
125};
126
127VMem::VMem()
128{
129 m_lRefCount = 1;
f57e8d3b 130#ifdef _USE_LINKED_LIST
624a1c42 131 InitializeCriticalSection(&m_cs);
f57e8d3b 132 m_Dummy.pNext = m_Dummy.pPrev = &m_Dummy;
222c300a 133 m_Dummy.owner = this;
f57e8d3b 134#endif
f57e8d3b
GS
135}
136
137VMem::~VMem(void)
138{
139#ifdef _USE_LINKED_LIST
140 while (m_Dummy.pNext != &m_Dummy) {
141 Free(m_Dummy.pNext+1);
142 }
f57e8d3b 143 DeleteCriticalSection(&m_cs);
624a1c42 144#endif
f57e8d3b
GS
145}
146
147void* VMem::Malloc(size_t size)
148{
149#ifdef _USE_LINKED_LIST
222c300a 150 GetLock();
624a1c42 151 PMEMORY_BLOCK_HEADER ptr = (PMEMORY_BLOCK_HEADER)malloc(size+sizeof(MEMORY_BLOCK_HEADER));
9e6e1211
DH
152 if (!ptr) {
153 FreeLock();
154 return NULL;
155 }
f57e8d3b 156 LinkBlock(ptr);
222c300a 157 FreeLock();
f57e8d3b
GS
158 return (ptr+1);
159#else
624a1c42 160 return malloc(size);
f57e8d3b
GS
161#endif
162}
163
164void* VMem::Realloc(void* pMem, size_t size)
165{
166#ifdef _USE_LINKED_LIST
167 if (!pMem)
168 return Malloc(size);
169
170 if (!size) {
171 Free(pMem);
172 return NULL;
173 }
174
222c300a 175 GetLock();
f57e8d3b
GS
176 PMEMORY_BLOCK_HEADER ptr = (PMEMORY_BLOCK_HEADER)(((char*)pMem)-sizeof(MEMORY_BLOCK_HEADER));
177 UnlinkBlock(ptr);
624a1c42 178 ptr = (PMEMORY_BLOCK_HEADER)realloc(ptr, size+sizeof(MEMORY_BLOCK_HEADER));
9e6e1211
DH
179 if (!ptr) {
180 FreeLock();
181 return NULL;
182 }
f57e8d3b 183 LinkBlock(ptr);
222c300a 184 FreeLock();
f57e8d3b
GS
185
186 return (ptr+1);
187#else
624a1c42 188 return realloc(pMem, size);
f57e8d3b
GS
189#endif
190}
191
192void VMem::Free(void* pMem)
193{
194#ifdef _USE_LINKED_LIST
195 if (pMem) {
196 PMEMORY_BLOCK_HEADER ptr = (PMEMORY_BLOCK_HEADER)(((char*)pMem)-sizeof(MEMORY_BLOCK_HEADER));
222c300a 197 if (ptr->owner != this) {
222c300a 198 if (ptr->owner) {
05ec9bb3 199#if 1
05ec9bb3 200 int *nowhere = NULL;
94f17277 201 Perl_warn_nocontext("Free to wrong pool %p not %p",this,ptr->owner);
283d8f99
YO
202 *nowhere = 0; /* this segfault is deliberate,
203 so you can see the stack trace */
05ec9bb3
NIS
204#else
205 ptr->owner->Free(pMem);
206#endif
222c300a
NIS
207 }
208 return;
222c300a
NIS
209 }
210 GetLock();
f57e8d3b 211 UnlinkBlock(ptr);
222c300a 212 ptr->owner = NULL;
624a1c42 213 free(ptr);
222c300a 214 FreeLock();
f57e8d3b 215 }
624a1c42
DD
216#else /*_USE_LINKED_LIST*/
217 free(pMem);
f57e8d3b
GS
218#endif
219}
220
221void VMem::GetLock(void)
222{
624a1c42 223#ifdef _USE_LINKED_LIST
f57e8d3b 224 EnterCriticalSection(&m_cs);
624a1c42 225#endif
f57e8d3b
GS
226}
227
228void VMem::FreeLock(void)
229{
624a1c42 230#ifdef _USE_LINKED_LIST
f57e8d3b 231 LeaveCriticalSection(&m_cs);
624a1c42 232#endif
f57e8d3b
GS
233}
234
235int VMem::IsLocked(void)
236{
237#if 0
238 /* XXX TryEnterCriticalSection() is not available in some versions
239 * of Windows 95. Since this code is not used anywhere yet, we
240 * skirt the issue for now. */
241 BOOL bAccessed = TryEnterCriticalSection(&m_cs);
242 if(bAccessed) {
243 LeaveCriticalSection(&m_cs);
244 }
245 return !bAccessed;
246#else
247 ASSERT(0); /* alarm bells for when somebody calls this */
248 return 0;
249#endif
250}
251
252long VMem::Release(void)
253{
254 long lCount = InterlockedDecrement(&m_lRefCount);
255 if(!lCount)
256 delete this;
257 return lCount;
258}
259
260long VMem::AddRef(void)
261{
262 long lCount = InterlockedIncrement(&m_lRefCount);
263 return lCount;
264}
265
266#else /* _USE_MSVCRT_MEM_ALLOC */
267
268/*
7766f137
GS
269 * Knuth's boundary tag algorithm Vol #1, Page 440.
270 *
271 * Each block in the heap has tag words before and after it,
272 * TAG
273 * block
274 * TAG
275 * The size is stored in these tags as a long word, and includes the 8 bytes
276 * of overhead that the boundary tags consume. Blocks are allocated on long
277 * word boundaries, so the size is always multiples of long words. When the
278 * block is allocated, bit 0, (the tag bit), of the size is set to 1. When
279 * a block is freed, it is merged with adjacent free blocks, and the tag bit
280 * is set to 0.
281 *
282 * A linked list is used to manage the free list. The first two long words of
283 * the block contain double links. These links are only valid when the block
284 * is freed, therefore space needs to be reserved for them. Thus, the minimum
285 * block size (not counting the tags) is 8 bytes.
286 *
f57e8d3b 287 * Since memory allocation may occur on a single threaded, explict locks are not
7766f137
GS
288 * provided.
289 *
290 */
291
f57e8d3b 292const long lAllocStart = 0x00020000; /* start at 128K */
7766f137
GS
293const long minBlockSize = sizeof(void*)*2;
294const long sizeofTag = sizeof(long);
295const long blockOverhead = sizeofTag*2;
296const long minAllocSize = minBlockSize+blockOverhead;
f57e8d3b
GS
297#ifdef _USE_BUDDY_BLOCKS
298const long lSmallBlockSize = 1024;
299const size_t nListEntries = ((lSmallBlockSize-minAllocSize)/sizeof(long));
300
301inline size_t CalcEntry(size_t size)
302{
303 ASSERT((size&(sizeof(long)-1)) == 0);
304 return ((size - minAllocSize) / sizeof(long));
305}
306#endif
7766f137
GS
307
308typedef BYTE* PBLOCK; /* pointer to a memory block */
309
310/*
311 * Macros for accessing hidden fields in a memory block:
312 *
313 * SIZE size of this block (tag bit 0 is 1 if block is allocated)
314 * PSIZE size of previous physical block
315 */
316
317#define SIZE(block) (*(ULONG*)(((PBLOCK)(block))-sizeofTag))
f57e8d3b 318#define PSIZE(block) (*(ULONG*)(((PBLOCK)(block))-(blockOverhead)))
7766f137
GS
319inline void SetTags(PBLOCK block, long size)
320{
321 SIZE(block) = size;
322 PSIZE(block+(size&~1)) = size;
323}
324
325/*
326 * Free list pointers
327 * PREV pointer to previous block
328 * NEXT pointer to next block
329 */
330
331#define PREV(block) (*(PBLOCK*)(block))
332#define NEXT(block) (*(PBLOCK*)((block)+sizeof(PBLOCK)))
333inline void SetLink(PBLOCK block, PBLOCK prev, PBLOCK next)
334{
335 PREV(block) = prev;
336 NEXT(block) = next;
337}
338inline void Unlink(PBLOCK p)
339{
340 PBLOCK next = NEXT(p);
341 PBLOCK prev = PREV(p);
342 NEXT(prev) = next;
343 PREV(next) = prev;
344}
f57e8d3b 345#ifndef _USE_BUDDY_BLOCKS
7766f137
GS
346inline void AddToFreeList(PBLOCK block, PBLOCK pInList)
347{
348 PBLOCK next = NEXT(pInList);
349 NEXT(pInList) = block;
350 SetLink(block, pInList, next);
351 PREV(next) = block;
352}
f57e8d3b 353#endif
7766f137
GS
354
355/* Macro for rounding up to the next sizeof(long) */
356#define ROUND_UP(n) (((ULONG)(n)+sizeof(long)-1)&~(sizeof(long)-1))
357#define ROUND_UP64K(n) (((ULONG)(n)+0x10000-1)&~(0x10000-1))
358#define ROUND_DOWN(n) ((ULONG)(n)&~(sizeof(long)-1))
359
360/*
361 * HeapRec - a list of all non-contiguous heap areas
362 *
363 * Each record in this array contains information about a non-contiguous heap area.
364 */
365
f57e8d3b 366const int maxHeaps = 32; /* 64 was overkill */
7766f137
GS
367const long lAllocMax = 0x80000000; /* max size of allocation */
368
f57e8d3b
GS
369#ifdef _USE_BUDDY_BLOCKS
370typedef struct _FreeListEntry
371{
372 BYTE Dummy[minAllocSize]; // dummy free block
373} FREE_LIST_ENTRY, *PFREE_LIST_ENTRY;
374#endif
375
376#ifndef _USE_BUDDY_BLOCKS
377#define USE_BIGBLOCK_ALLOC
378#endif
379/*
380 * performance tuning
381 * Use VirtualAlloc() for blocks bigger than nMaxHeapAllocSize since
382 * Windows 95/98/Me have heap managers that are designed for memory
383 * blocks smaller than four megabytes.
384 */
385
386#ifdef USE_BIGBLOCK_ALLOC
387const int nMaxHeapAllocSize = (1024*512); /* don't allocate anything larger than this from the heap */
388#endif
389
7766f137
GS
390typedef struct _HeapRec
391{
392 PBLOCK base; /* base of heap area */
393 ULONG len; /* size of heap area */
f57e8d3b
GS
394#ifdef USE_BIGBLOCK_ALLOC
395 BOOL bBigBlock; /* was allocate using VirtualAlloc */
396#endif
7766f137
GS
397} HeapRec;
398
7766f137
GS
399class VMem
400{
401public:
402 VMem();
403 ~VMem();
b75e1177
SH
404 void* Malloc(size_t size);
405 void* Realloc(void* pMem, size_t size);
406 void Free(void* pMem);
407 void GetLock(void);
408 void FreeLock(void);
409 int IsLocked(void);
410 long Release(void);
411 long AddRef(void);
7766f137
GS
412
413 inline BOOL CreateOk(void)
414 {
f57e8d3b
GS
415#ifdef _USE_BUDDY_BLOCKS
416 return TRUE;
417#else
7766f137 418 return m_hHeap != NULL;
f57e8d3b 419#endif
7766f137
GS
420 };
421
422 void ReInit(void);
423
424protected:
425 void Init(void);
426 int Getmem(size_t size);
f57e8d3b
GS
427
428 int HeapAdd(void* ptr, size_t size
429#ifdef USE_BIGBLOCK_ALLOC
430 , BOOL bBigBlock
431#endif
432 );
433
7766f137 434 void* Expand(void* block, size_t size);
7766f137 435
f57e8d3b
GS
436#ifdef _USE_BUDDY_BLOCKS
437 inline PBLOCK GetFreeListLink(int index)
438 {
439 if (index >= nListEntries)
440 index = nListEntries-1;
441 return &m_FreeList[index].Dummy[sizeofTag];
442 }
443 inline PBLOCK GetOverSizeFreeList(void)
444 {
445 return &m_FreeList[nListEntries-1].Dummy[sizeofTag];
446 }
447 inline PBLOCK GetEOLFreeList(void)
448 {
449 return &m_FreeList[nListEntries].Dummy[sizeofTag];
450 }
451
452 void AddToFreeList(PBLOCK block, size_t size)
453 {
454 PBLOCK pFreeList = GetFreeListLink(CalcEntry(size));
455 PBLOCK next = NEXT(pFreeList);
456 NEXT(pFreeList) = block;
457 SetLink(block, pFreeList, next);
458 PREV(next) = block;
459 }
460#endif
461 inline size_t CalcAllocSize(size_t size)
462 {
463 /*
464 * Adjust the real size of the block to be a multiple of sizeof(long), and add
465 * the overhead for the boundary tags. Disallow negative or zero sizes.
466 */
467 return (size < minBlockSize) ? minAllocSize : (size_t)ROUND_UP(size) + blockOverhead;
468 }
469
470#ifdef _USE_BUDDY_BLOCKS
471 FREE_LIST_ENTRY m_FreeList[nListEntries+1]; // free list with dummy end of list entry as well
472#else
52cbf511
JH
473 HANDLE m_hHeap; // memory heap for this script
474 char m_FreeDummy[minAllocSize]; // dummy free block
475 PBLOCK m_pFreeList; // pointer to first block on free list
f57e8d3b 476#endif
52cbf511
JH
477 PBLOCK m_pRover; // roving pointer into the free list
478 HeapRec m_heaps[maxHeaps]; // list of all non-contiguous heap areas
479 int m_nHeaps; // no. of heaps in m_heaps
480 long m_lAllocSize; // current alloc size
481 long m_lRefCount; // number of current users
482 CRITICAL_SECTION m_cs; // access lock
f57e8d3b 483
df3728a2 484#ifdef _DEBUG_MEM
f57e8d3b
GS
485 void WalkHeap(int complete);
486 void MemoryUsageMessage(char *str, long x, long y, int c);
df3728a2
JH
487 FILE* m_pLog;
488#endif
7766f137
GS
489};
490
7766f137
GS
491VMem::VMem()
492{
493 m_lRefCount = 1;
f57e8d3b 494#ifndef _USE_BUDDY_BLOCKS
7766f137
GS
495 BOOL bRet = (NULL != (m_hHeap = HeapCreate(HEAP_NO_SERIALIZE,
496 lAllocStart, /* initial size of heap */
497 0))); /* no upper limit on size of heap */
498 ASSERT(bRet);
f57e8d3b 499#endif
7766f137
GS
500
501 InitializeCriticalSection(&m_cs);
df3728a2
JH
502#ifdef _DEBUG_MEM
503 m_pLog = 0;
504#endif
7766f137
GS
505
506 Init();
507}
508
509VMem::~VMem(void)
510{
f57e8d3b 511#ifndef _USE_BUDDY_BLOCKS
7766f137 512 ASSERT(HeapValidate(m_hHeap, HEAP_NO_SERIALIZE, NULL));
df3728a2 513#endif
f57e8d3b
GS
514 WALKHEAPTRACE();
515
7766f137 516 DeleteCriticalSection(&m_cs);
f57e8d3b
GS
517#ifdef _USE_BUDDY_BLOCKS
518 for(int index = 0; index < m_nHeaps; ++index) {
519 VirtualFree(m_heaps[index].base, 0, MEM_RELEASE);
520 }
521#else /* !_USE_BUDDY_BLOCKS */
522#ifdef USE_BIGBLOCK_ALLOC
523 for(int index = 0; index < m_nHeaps; ++index) {
524 if (m_heaps[index].bBigBlock) {
525 VirtualFree(m_heaps[index].base, 0, MEM_RELEASE);
526 }
527 }
528#endif
7766f137
GS
529 BOOL bRet = HeapDestroy(m_hHeap);
530 ASSERT(bRet);
f57e8d3b 531#endif /* _USE_BUDDY_BLOCKS */
7766f137
GS
532}
533
534void VMem::ReInit(void)
535{
f57e8d3b
GS
536 for(int index = 0; index < m_nHeaps; ++index) {
537#ifdef _USE_BUDDY_BLOCKS
538 VirtualFree(m_heaps[index].base, 0, MEM_RELEASE);
539#else
540#ifdef USE_BIGBLOCK_ALLOC
541 if (m_heaps[index].bBigBlock) {
542 VirtualFree(m_heaps[index].base, 0, MEM_RELEASE);
543 }
544 else
545#endif
546 HeapFree(m_hHeap, HEAP_NO_SERIALIZE, m_heaps[index].base);
547#endif /* _USE_BUDDY_BLOCKS */
548 }
7766f137
GS
549
550 Init();
551}
552
553void VMem::Init(void)
f57e8d3b
GS
554{
555#ifdef _USE_BUDDY_BLOCKS
556 PBLOCK pFreeList;
557 /*
558 * Initialize the free list by placing a dummy zero-length block on it.
559 * Set the end of list marker.
560 * Set the number of non-contiguous heaps to zero.
561 * Set the next allocation size.
562 */
563 for (int index = 0; index < nListEntries; ++index) {
564 pFreeList = GetFreeListLink(index);
565 SIZE(pFreeList) = PSIZE(pFreeList+minAllocSize) = 0;
566 PREV(pFreeList) = NEXT(pFreeList) = pFreeList;
567 }
568 pFreeList = GetEOLFreeList();
569 SIZE(pFreeList) = PSIZE(pFreeList+minAllocSize) = 0;
570 PREV(pFreeList) = NEXT(pFreeList) = NULL;
571 m_pRover = GetOverSizeFreeList();
572#else
573 /*
7766f137
GS
574 * Initialize the free list by placing a dummy zero-length block on it.
575 * Set the number of non-contiguous heaps to zero.
576 */
f57e8d3b
GS
577 m_pFreeList = m_pRover = (PBLOCK)(&m_FreeDummy[sizeofTag]);
578 PSIZE(m_pFreeList+minAllocSize) = SIZE(m_pFreeList) = 0;
7766f137 579 PREV(m_pFreeList) = NEXT(m_pFreeList) = m_pFreeList;
f57e8d3b 580#endif
7766f137
GS
581
582 m_nHeaps = 0;
583 m_lAllocSize = lAllocStart;
584}
585
586void* VMem::Malloc(size_t size)
587{
588 WALKHEAP();
589
f57e8d3b
GS
590 PBLOCK ptr;
591 size_t lsize, rem;
7766f137 592 /*
f57e8d3b 593 * Disallow negative or zero sizes.
7766f137 594 */
f57e8d3b 595 size_t realsize = CalcAllocSize(size);
7766f137
GS
596 if((int)realsize < minAllocSize || size == 0)
597 return NULL;
598
f57e8d3b
GS
599#ifdef _USE_BUDDY_BLOCKS
600 /*
601 * Check the free list of small blocks if this is free use it
602 * Otherwise check the rover if it has no blocks then
603 * Scan the free list entries use the first free block
604 * split the block if needed, stop at end of list marker
605 */
606 {
607 int index = CalcEntry(realsize);
608 if (index < nListEntries-1) {
609 ptr = GetFreeListLink(index);
610 lsize = SIZE(ptr);
611 if (lsize >= realsize) {
612 rem = lsize - realsize;
613 if(rem < minAllocSize) {
614 /* Unlink the block from the free list. */
615 Unlink(ptr);
616 }
617 else {
618 /*
619 * split the block
620 * The remainder is big enough to split off into a new block.
621 * Use the end of the block, resize the beginning of the block
622 * no need to change the free list.
623 */
624 SetTags(ptr, rem);
625 ptr += SIZE(ptr);
626 lsize = realsize;
627 }
628 SetTags(ptr, lsize | 1);
629 return ptr;
630 }
631 ptr = m_pRover;
632 lsize = SIZE(ptr);
633 if (lsize >= realsize) {
634 rem = lsize - realsize;
635 if(rem < minAllocSize) {
636 /* Unlink the block from the free list. */
637 Unlink(ptr);
638 }
639 else {
640 /*
641 * split the block
642 * The remainder is big enough to split off into a new block.
643 * Use the end of the block, resize the beginning of the block
644 * no need to change the free list.
645 */
646 SetTags(ptr, rem);
647 ptr += SIZE(ptr);
648 lsize = realsize;
649 }
650 SetTags(ptr, lsize | 1);
651 return ptr;
652 }
653 ptr = GetFreeListLink(index+1);
654 while (NEXT(ptr)) {
655 lsize = SIZE(ptr);
656 if (lsize >= realsize) {
657 size_t rem = lsize - realsize;
658 if(rem < minAllocSize) {
659 /* Unlink the block from the free list. */
660 Unlink(ptr);
661 }
662 else {
663 /*
664 * split the block
665 * The remainder is big enough to split off into a new block.
666 * Use the end of the block, resize the beginning of the block
667 * no need to change the free list.
668 */
669 SetTags(ptr, rem);
670 ptr += SIZE(ptr);
671 lsize = realsize;
672 }
673 SetTags(ptr, lsize | 1);
674 return ptr;
675 }
676 ptr += sizeof(FREE_LIST_ENTRY);
677 }
678 }
679 }
680#endif
681
7766f137
GS
682 /*
683 * Start searching the free list at the rover. If we arrive back at rover without
684 * finding anything, allocate some memory from the heap and try again.
685 */
f57e8d3b
GS
686 ptr = m_pRover; /* start searching at rover */
687 int loops = 2; /* allow two times through the loop */
7766f137 688 for(;;) {
f57e8d3b 689 lsize = SIZE(ptr);
7766f137
GS
690 ASSERT((lsize&1)==0);
691 /* is block big enough? */
692 if(lsize >= realsize) {
693 /* if the remainder is too small, don't bother splitting the block. */
f57e8d3b 694 rem = lsize - realsize;
7766f137
GS
695 if(rem < minAllocSize) {
696 if(m_pRover == ptr)
697 m_pRover = NEXT(ptr);
698
699 /* Unlink the block from the free list. */
700 Unlink(ptr);
701 }
702 else {
703 /*
704 * split the block
705 * The remainder is big enough to split off into a new block.
706 * Use the end of the block, resize the beginning of the block
707 * no need to change the free list.
708 */
709 SetTags(ptr, rem);
710 ptr += SIZE(ptr);
711 lsize = realsize;
712 }
713 /* Set the boundary tags to mark it as allocated. */
714 SetTags(ptr, lsize | 1);
715 return ((void *)ptr);
716 }
717
718 /*
719 * This block was unsuitable. If we've gone through this list once already without
720 * finding anything, allocate some new memory from the heap and try again.
721 */
722 ptr = NEXT(ptr);
723 if(ptr == m_pRover) {
724 if(!(loops-- && Getmem(realsize))) {
725 return NULL;
726 }
727 ptr = m_pRover;
728 }
729 }
730}
731
732void* VMem::Realloc(void* block, size_t size)
733{
734 WALKHEAP();
735
736 /* if size is zero, free the block. */
737 if(size == 0) {
738 Free(block);
739 return (NULL);
740 }
741
742 /* if block pointer is NULL, do a Malloc(). */
743 if(block == NULL)
744 return Malloc(size);
745
746 /*
747 * Grow or shrink the block in place.
748 * if the block grows then the next block will be used if free
749 */
750 if(Expand(block, size) != NULL)
751 return block;
752
f57e8d3b 753 size_t realsize = CalcAllocSize(size);
7766f137
GS
754 if((int)realsize < minAllocSize)
755 return NULL;
756
757 /*
758 * see if the previous block is free, and is it big enough to cover the new size
759 * if merged with the current block.
760 */
761 PBLOCK ptr = (PBLOCK)block;
762 size_t cursize = SIZE(ptr) & ~1;
763 size_t psize = PSIZE(ptr);
764 if((psize&1) == 0 && (psize + cursize) >= realsize) {
765 PBLOCK prev = ptr - psize;
766 if(m_pRover == prev)
767 m_pRover = NEXT(prev);
768
769 /* Unlink the next block from the free list. */
770 Unlink(prev);
771
772 /* Copy contents of old block to new location, make it the current block. */
773 memmove(prev, ptr, cursize);
774 cursize += psize; /* combine sizes */
775 ptr = prev;
776
777 size_t rem = cursize - realsize;
778 if(rem >= minAllocSize) {
779 /*
780 * The remainder is big enough to be a new block. Set boundary
781 * tags for the resized block and the new block.
782 */
783 prev = ptr + realsize;
784 /*
785 * add the new block to the free list.
786 * next block cannot be free
787 */
788 SetTags(prev, rem);
f57e8d3b
GS
789#ifdef _USE_BUDDY_BLOCKS
790 AddToFreeList(prev, rem);
791#else
7766f137 792 AddToFreeList(prev, m_pFreeList);
f57e8d3b 793#endif
7766f137
GS
794 cursize = realsize;
795 }
796 /* Set the boundary tags to mark it as allocated. */
797 SetTags(ptr, cursize | 1);
798 return ((void *)ptr);
799 }
800
801 /* Allocate a new block, copy the old to the new, and free the old. */
802 if((ptr = (PBLOCK)Malloc(size)) != NULL) {
f57e8d3b 803 memmove(ptr, block, cursize-blockOverhead);
7766f137
GS
804 Free(block);
805 }
806 return ((void *)ptr);
807}
808
809void VMem::Free(void* p)
810{
811 WALKHEAP();
812
813 /* Ignore null pointer. */
814 if(p == NULL)
815 return;
816
817 PBLOCK ptr = (PBLOCK)p;
818
819 /* Check for attempt to free a block that's already free. */
820 size_t size = SIZE(ptr);
821 if((size&1) == 0) {
822 MEMODSlx("Attempt to free previously freed block", (long)p);
823 return;
824 }
825 size &= ~1; /* remove allocated tag */
826
827 /* if previous block is free, add this block to it. */
f57e8d3b 828#ifndef _USE_BUDDY_BLOCKS
7766f137 829 int linked = FALSE;
f57e8d3b 830#endif
7766f137
GS
831 size_t psize = PSIZE(ptr);
832 if((psize&1) == 0) {
833 ptr -= psize; /* point to previous block */
834 size += psize; /* merge the sizes of the two blocks */
f57e8d3b
GS
835#ifdef _USE_BUDDY_BLOCKS
836 Unlink(ptr);
837#else
7766f137 838 linked = TRUE; /* it's already on the free list */
f57e8d3b 839#endif
7766f137
GS
840 }
841
842 /* if the next physical block is free, merge it with this block. */
843 PBLOCK next = ptr + size; /* point to next physical block */
844 size_t nsize = SIZE(next);
845 if((nsize&1) == 0) {
846 /* block is free move rover if needed */
847 if(m_pRover == next)
848 m_pRover = NEXT(next);
849
850 /* unlink the next block from the free list. */
851 Unlink(next);
852
853 /* merge the sizes of this block and the next block. */
854 size += nsize;
855 }
856
857 /* Set the boundary tags for the block; */
858 SetTags(ptr, size);
859
860 /* Link the block to the head of the free list. */
f57e8d3b
GS
861#ifdef _USE_BUDDY_BLOCKS
862 AddToFreeList(ptr, size);
863#else
7766f137
GS
864 if(!linked) {
865 AddToFreeList(ptr, m_pFreeList);
866 }
f57e8d3b 867#endif
7766f137
GS
868}
869
870void VMem::GetLock(void)
871{
872 EnterCriticalSection(&m_cs);
873}
874
875void VMem::FreeLock(void)
876{
877 LeaveCriticalSection(&m_cs);
878}
879
880int VMem::IsLocked(void)
881{
90430aa1
GS
882#if 0
883 /* XXX TryEnterCriticalSection() is not available in some versions
884 * of Windows 95. Since this code is not used anywhere yet, we
885 * skirt the issue for now. */
7766f137
GS
886 BOOL bAccessed = TryEnterCriticalSection(&m_cs);
887 if(bAccessed) {
888 LeaveCriticalSection(&m_cs);
889 }
890 return !bAccessed;
90430aa1
GS
891#else
892 ASSERT(0); /* alarm bells for when somebody calls this */
893 return 0;
894#endif
7766f137
GS
895}
896
897
898long VMem::Release(void)
899{
900 long lCount = InterlockedDecrement(&m_lRefCount);
901 if(!lCount)
902 delete this;
903 return lCount;
904}
905
906long VMem::AddRef(void)
907{
908 long lCount = InterlockedIncrement(&m_lRefCount);
909 return lCount;
910}
911
912
913int VMem::Getmem(size_t requestSize)
914{ /* returns -1 is successful 0 if not */
f57e8d3b
GS
915#ifdef USE_BIGBLOCK_ALLOC
916 BOOL bBigBlock;
917#endif
7766f137
GS
918 void *ptr;
919
920 /* Round up size to next multiple of 64K. */
921 size_t size = (size_t)ROUND_UP64K(requestSize);
f57e8d3b 922
7766f137
GS
923 /*
924 * if the size requested is smaller than our current allocation size
925 * adjust up
926 */
927 if(size < (unsigned long)m_lAllocSize)
928 size = m_lAllocSize;
929
930 /* Update the size to allocate on the next request */
931 if(m_lAllocSize != lAllocMax)
f57e8d3b 932 m_lAllocSize <<= 2;
7766f137 933
f57e8d3b
GS
934#ifndef _USE_BUDDY_BLOCKS
935 if(m_nHeaps != 0
936#ifdef USE_BIGBLOCK_ALLOC
937 && !m_heaps[m_nHeaps-1].bBigBlock
938#endif
939 ) {
7766f137 940 /* Expand the last allocated heap */
f57e8d3b 941 ptr = HeapReAlloc(m_hHeap, HEAP_REALLOC_IN_PLACE_ONLY|HEAP_NO_SERIALIZE,
7766f137
GS
942 m_heaps[m_nHeaps-1].base,
943 m_heaps[m_nHeaps-1].len + size);
944 if(ptr != 0) {
f57e8d3b
GS
945 HeapAdd(((char*)ptr) + m_heaps[m_nHeaps-1].len, size
946#ifdef USE_BIGBLOCK_ALLOC
947 , FALSE
948#endif
949 );
7766f137
GS
950 return -1;
951 }
952 }
f57e8d3b 953#endif /* _USE_BUDDY_BLOCKS */
7766f137
GS
954
955 /*
956 * if we didn't expand a block to cover the requested size
957 * allocate a new Heap
958 * the size of this block must include the additional dummy tags at either end
959 * the above ROUND_UP64K may not have added any memory to include this.
960 */
961 if(size == requestSize)
f57e8d3b
GS
962 size = (size_t)ROUND_UP64K(requestSize+(blockOverhead));
963
964Restart:
965#ifdef _USE_BUDDY_BLOCKS
966 ptr = VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_READWRITE);
967#else
968#ifdef USE_BIGBLOCK_ALLOC
969 bBigBlock = FALSE;
970 if (size >= nMaxHeapAllocSize) {
971 bBigBlock = TRUE;
972 ptr = VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_READWRITE);
973 }
974 else
975#endif
976 ptr = HeapAlloc(m_hHeap, HEAP_NO_SERIALIZE, size);
977#endif /* _USE_BUDDY_BLOCKS */
978
979 if (!ptr) {
980 /* try to allocate a smaller chunk */
981 size >>= 1;
982 if(size > requestSize)
983 goto Restart;
984 }
7766f137 985
7766f137
GS
986 if(ptr == 0) {
987 MEMODSlx("HeapAlloc failed on size!!!", size);
988 return 0;
989 }
990
f57e8d3b
GS
991#ifdef _USE_BUDDY_BLOCKS
992 if (HeapAdd(ptr, size)) {
993 VirtualFree(ptr, 0, MEM_RELEASE);
994 return 0;
995 }
996#else
997#ifdef USE_BIGBLOCK_ALLOC
998 if (HeapAdd(ptr, size, bBigBlock)) {
999 if (bBigBlock) {
1000 VirtualFree(ptr, 0, MEM_RELEASE);
1001 }
1002 }
1003#else
7766f137 1004 HeapAdd(ptr, size);
f57e8d3b
GS
1005#endif
1006#endif /* _USE_BUDDY_BLOCKS */
7766f137
GS
1007 return -1;
1008}
1009
f57e8d3b
GS
1010int VMem::HeapAdd(void* p, size_t size
1011#ifdef USE_BIGBLOCK_ALLOC
1012 , BOOL bBigBlock
1013#endif
1014 )
7766f137
GS
1015{ /* if the block can be succesfully added to the heap, returns 0; otherwise -1. */
1016 int index;
1017
1018 /* Check size, then round size down to next long word boundary. */
1019 if(size < minAllocSize)
1020 return -1;
1021
1022 size = (size_t)ROUND_DOWN(size);
1023 PBLOCK ptr = (PBLOCK)p;
1024
f57e8d3b
GS
1025#ifdef USE_BIGBLOCK_ALLOC
1026 if (!bBigBlock) {
1027#endif
1028 /*
1029 * Search for another heap area that's contiguous with the bottom of this new area.
1030 * (It should be extremely unusual to find one that's contiguous with the top).
1031 */
1032 for(index = 0; index < m_nHeaps; ++index) {
1033 if(ptr == m_heaps[index].base + (int)m_heaps[index].len) {
1034 /*
1035 * The new block is contiguous with a previously allocated heap area. Add its
a6d05634 1036 * length to that of the previous heap. Merge it with the dummy end-of-heap
f57e8d3b
GS
1037 * area marker of the previous heap.
1038 */
1039 m_heaps[index].len += size;
1040 break;
1041 }
7766f137 1042 }
f57e8d3b
GS
1043#ifdef USE_BIGBLOCK_ALLOC
1044 }
1045 else {
1046 index = m_nHeaps;
7766f137 1047 }
f57e8d3b 1048#endif
7766f137
GS
1049
1050 if(index == m_nHeaps) {
f57e8d3b 1051 /* The new block is not contiguous, or is BigBlock. Add it to the heap list. */
7766f137
GS
1052 if(m_nHeaps == maxHeaps) {
1053 return -1; /* too many non-contiguous heaps */
1054 }
1055 m_heaps[m_nHeaps].base = ptr;
1056 m_heaps[m_nHeaps].len = size;
f57e8d3b
GS
1057#ifdef USE_BIGBLOCK_ALLOC
1058 m_heaps[m_nHeaps].bBigBlock = bBigBlock;
1059#endif
7766f137
GS
1060 m_nHeaps++;
1061
1062 /*
1063 * Reserve the first LONG in the block for the ending boundary tag of a dummy
1064 * block at the start of the heap area.
1065 */
f57e8d3b
GS
1066 size -= blockOverhead;
1067 ptr += blockOverhead;
7766f137
GS
1068 PSIZE(ptr) = 1; /* mark the dummy previous block as allocated */
1069 }
1070
1071 /*
1072 * Convert the heap to one large block. Set up its boundary tags, and those of
1073 * marker block after it. The marker block before the heap will already have
1074 * been set up if this heap is not contiguous with the end of another heap.
1075 */
1076 SetTags(ptr, size | 1);
1077 PBLOCK next = ptr + size; /* point to dummy end block */
1078 SIZE(next) = 1; /* mark the dummy end block as allocated */
1079
1080 /*
1081 * Link the block to the start of the free list by calling free().
1082 * This will merge the block with any adjacent free blocks.
1083 */
1084 Free(ptr);
1085 return 0;
1086}
1087
1088
1089void* VMem::Expand(void* block, size_t size)
1090{
1091 /*
f57e8d3b 1092 * Disallow negative or zero sizes.
7766f137 1093 */
f57e8d3b 1094 size_t realsize = CalcAllocSize(size);
7766f137
GS
1095 if((int)realsize < minAllocSize || size == 0)
1096 return NULL;
1097
1098 PBLOCK ptr = (PBLOCK)block;
1099
1100 /* if the current size is the same as requested, do nothing. */
1101 size_t cursize = SIZE(ptr) & ~1;
1102 if(cursize == realsize) {
1103 return block;
1104 }
1105
1106 /* if the block is being shrunk, convert the remainder of the block into a new free block. */
1107 if(realsize <= cursize) {
1108 size_t nextsize = cursize - realsize; /* size of new remainder block */
1109 if(nextsize >= minAllocSize) {
1110 /*
1111 * Split the block
1112 * Set boundary tags for the resized block and the new block.
1113 */
1114 SetTags(ptr, realsize | 1);
1115 ptr += realsize;
1116
1117 /*
1118 * add the new block to the free list.
1119 * call Free to merge this block with next block if free
1120 */
1121 SetTags(ptr, nextsize | 1);
1122 Free(ptr);
1123 }
1124
1125 return block;
1126 }
1127
1128 PBLOCK next = ptr + cursize;
1129 size_t nextsize = SIZE(next);
1130
1131 /* Check the next block for consistency.*/
1132 if((nextsize&1) == 0 && (nextsize + cursize) >= realsize) {
1133 /*
1134 * The next block is free and big enough. Add the part that's needed
1135 * to our block, and split the remainder off into a new block.
1136 */
1137 if(m_pRover == next)
1138 m_pRover = NEXT(next);
1139
1140 /* Unlink the next block from the free list. */
1141 Unlink(next);
1142 cursize += nextsize; /* combine sizes */
1143
1144 size_t rem = cursize - realsize; /* size of remainder */
1145 if(rem >= minAllocSize) {
1146 /*
1147 * The remainder is big enough to be a new block.
1148 * Set boundary tags for the resized block and the new block.
1149 */
1150 next = ptr + realsize;
1151 /*
1152 * add the new block to the free list.
1153 * next block cannot be free
1154 */
1155 SetTags(next, rem);
f57e8d3b
GS
1156#ifdef _USE_BUDDY_BLOCKS
1157 AddToFreeList(next, rem);
1158#else
7766f137 1159 AddToFreeList(next, m_pFreeList);
f57e8d3b 1160#endif
7766f137
GS
1161 cursize = realsize;
1162 }
1163 /* Set the boundary tags to mark it as allocated. */
1164 SetTags(ptr, cursize | 1);
1165 return ((void *)ptr);
1166 }
1167 return NULL;
1168}
1169
1170#ifdef _DEBUG_MEM
df3728a2 1171#define LOG_FILENAME ".\\MemLog.txt"
7766f137 1172
f57e8d3b 1173void VMem::MemoryUsageMessage(char *str, long x, long y, int c)
7766f137 1174{
7766f137
GS
1175 char szBuffer[512];
1176 if(str) {
df3728a2
JH
1177 if(!m_pLog)
1178 m_pLog = fopen(LOG_FILENAME, "w");
7766f137 1179 sprintf(szBuffer, str, x, y, c);
df3728a2 1180 fputs(szBuffer, m_pLog);
7766f137
GS
1181 }
1182 else {
f57e8d3b
GS
1183 if(m_pLog) {
1184 fflush(m_pLog);
1185 fclose(m_pLog);
1186 m_pLog = 0;
1187 }
7766f137
GS
1188 }
1189}
1190
f57e8d3b 1191void VMem::WalkHeap(int complete)
7766f137 1192{
f57e8d3b
GS
1193 if(complete) {
1194 MemoryUsageMessage(NULL, 0, 0, 0);
1195 size_t total = 0;
1196 for(int i = 0; i < m_nHeaps; ++i) {
1197 total += m_heaps[i].len;
1198 }
1199 MemoryUsageMessage("VMem heaps used %d. Total memory %08x\n", m_nHeaps, total, 0);
1200
1201 /* Walk all the heaps - verify structures */
1202 for(int index = 0; index < m_nHeaps; ++index) {
1203 PBLOCK ptr = m_heaps[index].base;
1204 size_t size = m_heaps[index].len;
1205#ifndef _USE_BUDDY_BLOCKS
1206#ifdef USE_BIGBLOCK_ALLOC
1207 if (!m_heaps[m_nHeaps].bBigBlock)
1208#endif
1209 ASSERT(HeapValidate(m_hHeap, HEAP_NO_SERIALIZE, ptr));
1210#endif
7766f137 1211
f57e8d3b
GS
1212 /* set over reserved header block */
1213 size -= blockOverhead;
1214 ptr += blockOverhead;
1215 PBLOCK pLast = ptr + size;
1216 ASSERT(PSIZE(ptr) == 1); /* dummy previous block is allocated */
1217 ASSERT(SIZE(pLast) == 1); /* dummy next block is allocated */
1218 while(ptr < pLast) {
1219 ASSERT(ptr > m_heaps[index].base);
1220 size_t cursize = SIZE(ptr) & ~1;
1221 ASSERT((PSIZE(ptr+cursize) & ~1) == cursize);
1222 MemoryUsageMessage("Memory Block %08x: Size %08x %c\n", (long)ptr, cursize, (SIZE(ptr)&1) ? 'x' : ' ');
1223 if(!(SIZE(ptr)&1)) {
1224 /* this block is on the free list */
1225 PBLOCK tmp = NEXT(ptr);
1226 while(tmp != ptr) {
1227 ASSERT((SIZE(tmp)&1)==0);
1228 if(tmp == m_pFreeList)
1229 break;
1230 ASSERT(NEXT(tmp));
1231 tmp = NEXT(tmp);
1232 }
1233 if(tmp == ptr) {
1234 MemoryUsageMessage("Memory Block %08x: Size %08x free but not in free list\n", (long)ptr, cursize, 0);
1235 }
7766f137 1236 }
f57e8d3b 1237 ptr += cursize;
7766f137 1238 }
7766f137 1239 }
7766f137
GS
1240 MemoryUsageMessage(NULL, 0, 0, 0);
1241 }
1242}
f57e8d3b
GS
1243#endif /* _DEBUG_MEM */
1244
1245#endif /* _USE_MSVCRT_MEM_ALLOC */
7766f137
GS
1246
1247#endif /* ___VMEM_H_INC___ */