[libclang] Install a fatal error handler that will call abort() instead of exit()
when there is a report_fatal_error() call.
rdar://10507984
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@146641 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/tools/libclang/CIndex.cpp b/tools/libclang/CIndex.cpp
index 9c057e5..bccd927 100644
--- a/tools/libclang/CIndex.cpp
+++ b/tools/libclang/CIndex.cpp
@@ -2336,6 +2336,17 @@
static llvm::sys::Mutex EnableMultithreadingMutex;
static bool EnabledMultithreading;
+static void fatal_error_handler(void *user_data, const std::string& reason) {
+ llvm::SmallString<64> Buffer;
+ llvm::raw_svector_ostream OS(Buffer);
+ OS << "LIBCLANG FATAL ERROR: " << reason << "\n";
+ StringRef MessageStr = OS.str();
+ // Write the result out to stderr avoiding errs() because raw_ostreams can
+ // call report_fatal_error.
+ ::write(2, MessageStr.data(), MessageStr.size());
+ ::abort();
+}
+
extern "C" {
CXIndex clang_createIndex(int excludeDeclarationsFromPCH,
int displayDiagnostics) {
@@ -2351,6 +2362,7 @@
{
llvm::sys::ScopedLock L(EnableMultithreadingMutex);
if (!EnabledMultithreading) {
+ llvm::install_fatal_error_handler(fatal_error_handler, 0);
llvm::llvm_start_multithreaded();
EnabledMultithreading = true;
}