Browse Source

WARNING: TEST: Security check to address potential overflow cocerns

Ray 1 năm trước cách đây
mục cha
commit
2bc75b877a
1 tập tin đã thay đổi với 3 bổ sung2 xóa
  1. 3 2
      src/raylib.h

+ 3 - 2
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)