Index: core/include/fxcrt/fx_memory.h |
diff --git a/core/include/fxcrt/fx_memory.h b/core/include/fxcrt/fx_memory.h |
index 4116e17450da0571f8b1598577f70952c8923cec..f03e0ac87e9a9b2cff2b00970983dc07a673f7b3 100644 |
--- a/core/include/fxcrt/fx_memory.h |
+++ b/core/include/fxcrt/fx_memory.h |
@@ -10,23 +10,60 @@ |
#include "fx_system.h" |
#ifdef __cplusplus |
-#include <new> |
extern "C" { |
#endif |
-#define FX_Alloc(type, size) (type*)calloc(size, sizeof(type)) |
-#define FX_Realloc(type, ptr, size) (type*)realloc(ptr, sizeof(type) * (size)) |
-#define FX_AllocNL(type, size) FX_Alloc(type, size) |
-#define FX_ReallocNL(type, ptr, size) FX_Realloc(type, ptr, size) |
-#define FX_Free(ptr) free(ptr) |
+// For external C libraries to malloc through PDFium. These may return NULL. |
void* FXMEM_DefaultAlloc(size_t byte_size, int flags); |
void* FXMEM_DefaultRealloc(void* pointer, size_t new_size, int flags); |
void FXMEM_DefaultFree(void* pointer, int flags); |
#ifdef __cplusplus |
+} // extern "C" |
+ |
+#include <stdlib.h> |
+#include <limits> |
+#include <new> |
+ |
+NEVER_INLINE void FX_OutOfMemoryTerminate(); |
+ |
+inline void* FX_SafeRealloc(void* ptr, size_t num_members, size_t member_size) { |
+ if (num_members < std::numeric_limits<size_t>::max() / member_size) { |
+ return realloc(ptr, num_members * member_size); |
+ } |
+ return nullptr; |
+} |
+ |
+inline void* FX_AllocOrDie(size_t num_members, size_t member_size) { |
+ // TODO(tsepez): See if we can avoid the implicit memset(0). |
+ if (void* result = calloc(num_members, member_size)) { |
+ return result; |
+ } |
+ FX_OutOfMemoryTerminate(); // Never returns. |
+ return nullptr; // Suppress compiler warning. |
} |
#define FX_NEW_VECTOR(Pointer, Class, Count) (Pointer = new Class[Count]) |
#define FX_DELETE_VECTOR(Pointer, Class, Count) delete[] Pointer |
+inline void* FX_ReallocOrDie(void* ptr, size_t num_members, size_t member_size) { |
+ if (void* result = FX_SafeRealloc(ptr, num_members, member_size)) { |
+ return result; |
+ } |
+ FX_OutOfMemoryTerminate(); // Never returns. |
+ return nullptr; // Suppress compiler warning. |
+} |
+ |
+// Never returns NULL. |
+#define FX_Alloc(type, size) (type*)FX_AllocOrDie(size, sizeof(type)) |
+#define FX_Realloc(type, ptr, size) \ |
+ (type*)FX_ReallocOrDie(ptr, size, sizeof(type)) |
+ |
+// May return NULL. |
+#define FX_TryAlloc(type, size) (type*)calloc(size, sizeof(type)) |
+#define FX_TryRealloc(type, ptr, size) \ |
+ (type*)FX_SafeRealloc(ptr, size, sizeof(type)) |
+ |
+#define FX_Free(ptr) free(ptr) |
+ |
class CFX_DestructObject |
{ |
public: |
@@ -72,5 +109,5 @@ private: |
void* m_pFirstTrunk; |
}; |
-#endif |
-#endif |
+#endif // __cplusplust |
+#endif // _FX_MEMORY_H_ |