jemalloc/test/integration/cpp/infallible_new.cpp
Slobodan Predolac 160ab9d7f6 Replace experimental_infallible_new with compile-time flag
The runtime option aborted on every OOM, breaking new(std::nothrow)
semantics. Replace with configure-time --enable-cxx-infallible-new
(default off): when on, throwing new aborts (size logged) and
nothrow returns null; when off, standard new_handler + bad_alloc /
null behavior is preserved. Under LTO the on-path lets the compiler
prove operator new is no-throw.
2026-06-06 09:50:37 -04:00

64 lines
1.7 KiB
C++

#include <stdio.h>
#include "test/jemalloc_test.h"
/*
* Verifies that, when jemalloc is built with --enable-cxx-infallible-new,
* throwing operator new on OOM aborts via safety_check_fail. The test hook
* intercepts the abort and asserts the size-bearing message prefix.
*/
bool fake_abort_called;
void
fake_abort(const char *message) {
const char *expected_start = "<jemalloc>: Allocation of size";
if (strncmp(message, expected_start, strlen(expected_start)) != 0) {
abort();
}
fake_abort_called = true;
}
static bool
own_operator_new(void) {
uint64_t before, after;
size_t sz = sizeof(before);
/* thread.allocated is always available, even w/o config_stats. */
expect_d_eq(mallctl("thread.allocated", (void *)&before, &sz, NULL, 0),
0, "Unexpected mallctl failure reading stats");
void *volatile ptr = ::operator new((size_t)8);
expect_ptr_not_null(ptr, "Unexpected allocation failure");
expect_d_eq(mallctl("thread.allocated", (void *)&after, &sz, NULL, 0),
0, "Unexpected mallctl failure reading stats");
return (after != before);
}
TEST_BEGIN(test_failing_alloc) {
test_hooks_safety_check_abort = &fake_abort;
/*
* Not owning operator new is only expected to happen on MinGW which
* does not support operator new / delete replacement.
*/
#ifdef _WIN32
test_skip_if(!own_operator_new());
#else
expect_true(own_operator_new(), "No operator new overload");
#endif
void *volatile ptr = (void *)1;
try {
/* Too big of an allocation to succeed. */
ptr = ::operator new((size_t)-1);
} catch (...) {
abort();
}
expect_ptr_null(ptr, "Allocation should have failed");
expect_b_eq(fake_abort_called, true, "Abort hook not invoked");
test_hooks_safety_check_abort = NULL;
}
TEST_END
int
main(void) {
return test(test_failing_alloc);
}