Avoid overflows in testcase malloc_wrappers
diff --git a/tests/common/malloc_wrappers.c b/tests/common/malloc_wrappers.c
index 2cfde9d..3f2cc38 100644
--- a/tests/common/malloc_wrappers.c
+++ b/tests/common/malloc_wrappers.c
@@ -31,7 +31,7 @@
{
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 @@
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);
}