Avoid overflows in testcase malloc_wrappers

devjoa · Feb 1, 2020 · 903ae5b · 903ae5b
1 parent e382ea5
commit 903ae5b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/common/malloc_wrappers.c b/tests/common/malloc_wrappers.c
@@ -31,7 +31,7 @@ void* malloc_with_check(size_t size)
 {
     char *buf = NULL;
 
-    if (g_alloc_bytes + size <= g_max_alloc_bytes)
+    if (size <= g_max_alloc_bytes - g_alloc_bytes)
     {
         buf = malloc(size + GUARD_SIZE);
     }
@@ -89,7 +89,7 @@ void* realloc_with_check(void *ptr, size_t size)
         assert(g_alloc_count > 0);
         assert(g_alloc_bytes >= oldsize);
 
-        if (g_alloc_bytes - oldsize + size <= g_max_alloc_bytes)
+        if (size <= g_max_alloc_bytes - (g_alloc_bytes - oldsize))
         {
             buf = realloc(buf, size + GUARD_SIZE);
         }