OLD | NEW |
1 // Copyright 2014 PDFium Authors. All rights reserved. | 1 // Copyright 2014 PDFium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
6 | 6 |
7 #include <stdlib.h> // For abort(). | 7 #include <stdlib.h> // For abort(). |
8 | 8 |
9 #include "core/include/fxcrt/fx_memory.h" | 9 #include "core/include/fxcrt/fx_memory.h" |
10 | 10 |
11 void* FXMEM_DefaultAlloc(size_t byte_size, int flags) { | 11 void* FXMEM_DefaultAlloc(size_t byte_size, int flags) { |
12 return (void*)malloc(byte_size); | 12 return (void*)malloc(byte_size); |
13 } | 13 } |
14 void* FXMEM_DefaultRealloc(void* pointer, size_t new_size, int flags) { | 14 void* FXMEM_DefaultRealloc(void* pointer, size_t new_size, int flags) { |
15 return realloc(pointer, new_size); | 15 return realloc(pointer, new_size); |
16 } | 16 } |
17 void FXMEM_DefaultFree(void* pointer, int flags) { | 17 void FXMEM_DefaultFree(void* pointer, int flags) { |
18 free(pointer); | 18 free(pointer); |
19 } | 19 } |
20 | 20 |
21 NEVER_INLINE void FX_OutOfMemoryTerminate() { | 21 NEVER_INLINE void FX_OutOfMemoryTerminate() { |
22 // Termimate cleanly if we can, else crash at a specific address (0xbd). | 22 // Termimate cleanly if we can, else crash at a specific address (0xbd). |
23 abort(); | 23 abort(); |
24 reinterpret_cast<void (*)()>(0xbd)(); | 24 reinterpret_cast<void (*)()>(0xbd)(); |
25 } | 25 } |
26 | |
27 CFX_GrowOnlyPool::CFX_GrowOnlyPool(size_t trunk_size) { | |
28 m_TrunkSize = trunk_size; | |
29 m_pFirstTrunk = NULL; | |
30 } | |
31 CFX_GrowOnlyPool::~CFX_GrowOnlyPool() { | |
32 FreeAll(); | |
33 } | |
34 struct _FX_GrowOnlyTrunk { | |
35 size_t m_Size; | |
36 size_t m_Allocated; | |
37 _FX_GrowOnlyTrunk* m_pNext; | |
38 }; | |
39 void CFX_GrowOnlyPool::FreeAll() { | |
40 _FX_GrowOnlyTrunk* pTrunk = (_FX_GrowOnlyTrunk*)m_pFirstTrunk; | |
41 while (pTrunk) { | |
42 _FX_GrowOnlyTrunk* pNext = pTrunk->m_pNext; | |
43 FX_Free(pTrunk); | |
44 pTrunk = pNext; | |
45 } | |
46 m_pFirstTrunk = NULL; | |
47 } | |
48 void* CFX_GrowOnlyPool::Alloc(size_t size) { | |
49 size = (size + 3) / 4 * 4; | |
50 _FX_GrowOnlyTrunk* pTrunk = (_FX_GrowOnlyTrunk*)m_pFirstTrunk; | |
51 while (pTrunk) { | |
52 if (pTrunk->m_Size - pTrunk->m_Allocated >= size) { | |
53 void* p = (uint8_t*)(pTrunk + 1) + pTrunk->m_Allocated; | |
54 pTrunk->m_Allocated += size; | |
55 return p; | |
56 } | |
57 pTrunk = pTrunk->m_pNext; | |
58 } | |
59 size_t alloc_size = size > m_TrunkSize ? size : m_TrunkSize; | |
60 pTrunk = (_FX_GrowOnlyTrunk*)FX_Alloc(uint8_t, | |
61 sizeof(_FX_GrowOnlyTrunk) + alloc_size); | |
62 pTrunk->m_Size = alloc_size; | |
63 pTrunk->m_Allocated = size; | |
64 pTrunk->m_pNext = (_FX_GrowOnlyTrunk*)m_pFirstTrunk; | |
65 m_pFirstTrunk = pTrunk; | |
66 return pTrunk + 1; | |
67 } | |
OLD | NEW |