Index: core/include/fxcrt/fx_memory.h |
diff --git a/core/include/fxcrt/fx_memory.h b/core/include/fxcrt/fx_memory.h |
index c1c0740ae24973c843a543533739575d54020a7a..e4b6e98039308aa26fc3aa7e7c15b59f6b6d1100 100644 |
--- a/core/include/fxcrt/fx_memory.h |
+++ b/core/include/fxcrt/fx_memory.h |
@@ -43,6 +43,14 @@ inline void* FX_AllocOrDie(size_t num_members, size_t member_size) { |
return nullptr; // Suppress compiler warning. |
} |
+inline void* FX_AllocOrDie2D(size_t w, size_t h, size_t member_size) { |
+ if (w < std::numeric_limits<size_t>::max() / h) { |
+ return FX_AllocOrDie(w * h, member_size); |
+ } |
+ FX_OutOfMemoryTerminate(); // Never returns. |
+ return nullptr; // Suppress compiler warning. |
+} |
+ |
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; |
@@ -53,6 +61,7 @@ inline void* FX_ReallocOrDie(void* ptr, size_t num_members, size_t member_size) |
// Never returns NULL. |
#define FX_Alloc(type, size) (type*)FX_AllocOrDie(size, sizeof(type)) |
+#define FX_Alloc2D(type, w, h) (type*)FX_AllocOrDie2D(w, h, sizeof(type)) |
#define FX_Realloc(type, ptr, size) \ |
(type*)FX_ReallocOrDie(ptr, size, sizeof(type)) |