bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

Register LLVM handlers for bad-alloc / OOM

Browse source 
LLVM's default bad-alloc handler may throw if exceptions are enabled,
and `operator new` isn't hooked at all by default. Now we register our
own handler that prints a message similar to fatal errors, then aborts.
We also call the function that registers the C++ `std::new_handler`.
This commit is contained in:
Josh Stone 2024-03-15 15:49:06 -07:00
parent 1ca424ca43
commit 0ade5a11f5
3 changed files with 26 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

25
compiler/rustc_llvm/llvm-wrapper/RustWrapper.cpp
View file

@ -25,6 +25,13 @@
#include <iostream>
// for raw `write` in the bad-alloc handler
#ifdef _MSC_VER
#include <io.h>
#else
#include <unistd.h>
#endif
//===----------------------------------------------------------------------===
//
// This file defines alternate interfaces to core functions that are more
@ -88,8 +95,24 @@ static void FatalErrorHandler(void *UserData,
exit(101);
}
extern "C" void LLVMRustInstallFatalErrorHandler() {
// Custom error handler for bad-alloc LLVM errors.
//
// It aborts the process without any further allocations, similar to LLVM's
// default except that may be configured to `throw std::bad_alloc()` instead.
static void BadAllocErrorHandler(void *UserData,
const char* Reason,
bool GenCrashDiag) {
const char *OOM = "rustc-LLVM ERROR: out of memory\n";
write(2, OOM, strlen(OOM));
write(2, Reason, strlen(Reason));
write(2, "\n", 1);
abort();
}
extern "C" void LLVMRustInstallErrorHandlers() {
install_fatal_error_handler(FatalErrorHandler);
install_bad_alloc_error_handler(BadAllocErrorHandler);
install_out_of_memory_new_handler();
}
extern "C" void LLVMRustDisableSystemDialogsOnCrash() {