1 files changed, 10 insertions, 5 deletions
diff --git a/lib/alloc/alloc_heap.c b/lib/alloc/alloc_heap.c
index 46fbdc9..fccfcfc 100644
--- a/lib/alloc/alloc_heap.c
+++ b/lib/alloc/alloc_heap.c
@@ -1,21 +1,26 @@
-#include <stddef.h>
+#include <errno.h>
 #include <setjmp.h>
+#include <stdckdint.h>
+#include <stddef.h>
 #include <stdlib.h>
 
 #include "alloc.h"
 #include "errors.h"
 
 void *
-alloc_heap(void *raw_ctx, void *ptr, size_t, size_t new, size_t)
+alloc_heap(void *raw_ctx, void *ptr, size_t, size_t new, size_t elemsz, size_t)
 {
 	if (new == 0) {
 		free(ptr);
 		return nullptr;
 	}
 
-	void *p = realloc(ptr, new);
-	if (p != nullptr)
-		return p;
+	if (!ckd_mul(&new, new, elemsz)) {
+		void *p = realloc(ptr, new);
+		if (p != nullptr)
+			return p;
+	} else
+		errno = EOVERFLOW;
 
 	struct heap_ctx *ctx = raw_ctx;
 	if (ctx == nullptr || ctx->jmp == nullptr)