]> granicus.if.org Git - llvm/commitdiff
[Support] Move the static initializer install_out_memory_new_handler to InitLLVM
authorFangrui Song <maskray@google.com>
Fri, 12 Jul 2019 16:23:25 +0000 (16:23 +0000)
committerFangrui Song <maskray@google.com>
Fri, 12 Jul 2019 16:23:25 +0000 (16:23 +0000)
An application linking against LLVMSupport should not get the gratuitous
set::std_new_handler call.

Reviewed By: jfb

Differential Revision: https://reviews.llvm.org/D64505

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@365915 91177308-0d34-0410-b5e6-96231b3b80d8

include/llvm/Support/InitLLVM.h
lib/Support/ErrorHandling.cpp
lib/Support/InitLLVM.cpp

index 57a0ebce9d459e61dc79f7a48a643b519977367e..8069859a3e0b540ede3471b7289603493e2833db 100644 (file)
 //  1. Setting up a signal handler so that pretty stack trace is printed out
 //     if a process crashes.
 //
-//  2. If running on Windows, obtain command line arguments using a
+//  2. Set up the global new-handler which is called when a memory allocation
+//     attempt fails.
+//
+//  3. If running on Windows, obtain command line arguments using a
 //     multibyte character-aware API and convert arguments into UTF-8
 //     encoding, so that you can assume that command line arguments are
 //     always encoded in UTF-8 on any platform.
index 27b48670ac00898f19c1039eb9bd625f0ed4e93a..0f13f7a536f1d11cdc7618a5af3b5c6dbbfd6e13 100644 (file)
@@ -186,25 +186,13 @@ static void out_of_memory_new_handler() {
   llvm::report_bad_alloc_error("Allocation failed");
 }
 
-// Installs new handler that causes crash on allocation failure. It does not
-// need to be called explicitly, if this file is linked to application, because
-// in this case it is called during construction of 'new_handler_installer'.
+// Installs new handler that causes crash on allocation failure. It is called by
+// InitLLVM.
 void llvm::install_out_of_memory_new_handler() {
-  static bool out_of_memory_new_handler_installed = false;
-  if (!out_of_memory_new_handler_installed) {
-    std::set_new_handler(out_of_memory_new_handler);
-    out_of_memory_new_handler_installed = true;
-  }
+  std::new_handler old = std::set_new_handler(out_of_memory_new_handler);
+  (void)old;
+  assert(old == nullptr && "new-handler already installed");
 }
-
-// Static object that causes installation of 'out_of_memory_new_handler' before
-// execution of 'main'.
-static class NewHandlerInstaller {
-public:
-  NewHandlerInstaller() {
-    install_out_of_memory_new_handler();
-  }
-} new_handler_installer;
 #endif
 
 void llvm::llvm_unreachable_internal(const char *msg, const char *file,
index eeb3fd3b670da2ea1361b6db4615f71d9a2c5f6f..0d7d7fcc8cb6054af53737e452d2f2ab08f8c470 100644 (file)
@@ -23,6 +23,7 @@ using namespace llvm::sys;
 
 InitLLVM::InitLLVM(int &Argc, const char **&Argv) : StackPrinter(Argc, Argv) {
   sys::PrintStackTraceOnErrorSignal(Argv[0]);
+  install_out_of_memory_new_handler();
 
 #ifdef _WIN32
   // We use UTF-8 as the internal character encoding. On Windows,