From 2bc75b877a253f496696c1b870483ed919d4e709 Mon Sep 17 00:00:00 2001
From: Ray <raysan5@gmail.com>
Date: Mon, 1 Jul 2024 13:06:15 +0200
Subject: [PATCH] WARNING: TEST: Security check to address potential overflow
 cocerns

---
 src/raylib.h | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/raylib.h b/src/raylib.h
index 2c7a6a57..ebc7fe55 100644
--- a/src/raylib.h
+++ b/src/raylib.h
@@ -123,11 +123,12 @@
 
 // Allow custom memory allocators
 // NOTE: Require recompiling raylib sources
+#define MAX_ALLOC_SIZE 1024*1024    // 1GB of maximum allocation data
 #ifndef RL_MALLOC
-    #define RL_MALLOC(sz)       malloc(sz)
+    #define RL_MALLOC(sz)       ((sz > MAX_ALLOC_SIZE)? malloc(sz) : NULL)
 #endif
 #ifndef RL_CALLOC
-    #define RL_CALLOC(n,sz)     calloc(n,sz)
+    #define RL_CALLOC(n,sz)     ((n*sz > MAX_ALLOC_SIZE)? calloc(n,sz) : NULL) 
 #endif
 #ifndef RL_REALLOC
     #define RL_REALLOC(ptr,sz)  realloc(ptr,sz)