Set errno to ENOMEM on reallocarray overflow.

2 files changed, 3 insertions, 0 deletions

diff --git a/vespamalloc/src/tests/test1/new_test.cpp b/vespamalloc/src/tests/test1/new_test.cpp
index 131f78cbe05..ac053323255 100644
--- a/vespamalloc/src/tests/test1/new_test.cpp
+++ b/vespamalloc/src/tests/test1/new_test.cpp
@@ -101,9 +101,11 @@ TEST("verify realloarray") {
     void *arr2 = reallocarray(arr, 800, 5);
     EXPECT_NOT_EQUAL(arr, arr2);
     EXPECT_NOT_EQUAL(nullptr, arr2);
+    EXPECT_NOT_EQUAL(ENOMEM, errno);
 
     void *arr3 = reallocarray(arr2, 1u << 33, 1u << 33);
     EXPECT_EQUAL(nullptr, arr3);
+    EXPECT_EQUAL(ENOMEM, errno);
 }
 #endif
 
diff --git a/vespamalloc/src/vespamalloc/malloc/overload.h b/vespamalloc/src/vespamalloc/malloc/overload.h
index b6a0d0316ae..8815a11a84f 100644
--- a/vespamalloc/src/vespamalloc/malloc/overload.h
+++ b/vespamalloc/src/vespamalloc/malloc/overload.h
@@ -159,6 +159,7 @@ void * reallocarray(void * ptr, size_t nemb, size_t elemSize) __THROW
 {
     size_t sz = nemb * elemSize;
     if (nemb != 0 && (sz/nemb != elemSize)) {
+        errno = ENOMEM;
         return nullptr;
     }
     return vespamalloc::createAllocator()->realloc(ptr, sz);