[PATCH] D64505: [Support] Move the static initializer install_out_memory_new_handler to InitLLVM

[Fangrui Song via Phabricator via llvm-commits]

This revision was automatically updated to reflect the committed changes.
Closed by commit rL365915: [Support] Move the static initializer install_out_memory_new_handler to InitLLVM (authored by MaskRay, committed by ).

Repository:
  rL LLVM

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D64505/new/

https://reviews.llvm.org/D64505

Files:
  llvm/trunk/include/llvm/Support/InitLLVM.h
  llvm/trunk/lib/Support/ErrorHandling.cpp
  llvm/trunk/lib/Support/InitLLVM.cpp


Index: llvm/trunk/include/llvm/Support/InitLLVM.h
===================================================================
--- llvm/trunk/include/llvm/Support/InitLLVM.h
+++ llvm/trunk/include/llvm/Support/InitLLVM.h
@@ -19,7 +19,10 @@
 //  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: llvm/trunk/lib/Support/ErrorHandling.cpp
===================================================================
--- llvm/trunk/lib/Support/ErrorHandling.cpp
+++ llvm/trunk/lib/Support/ErrorHandling.cpp
@@ -186,25 +186,13 @@
   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: llvm/trunk/lib/Support/InitLLVM.cpp
===================================================================
--- llvm/trunk/lib/Support/InitLLVM.cpp
+++ llvm/trunk/lib/Support/InitLLVM.cpp
@@ -23,6 +23,7 @@
 
 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,


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D64505.209511.patch
Type: text/x-patch
Size: 2617 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20190712/74bfc9f1/attachment.bin>