report_fatal_error can't use errs(), because errs() can call
[oota-llvm.git] / lib / Support / ErrorHandling.cpp
1 //===- lib/Support/ErrorHandling.cpp - Callbacks for errors ---------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file defines an API used to indicate fatal error conditions.  Non-fatal
11 // errors (most of them) should be handled through LLVMContext.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #include "llvm/ADT/Twine.h"
16 #include "llvm/Support/Debug.h"
17 #include "llvm/Support/ErrorHandling.h"
18 #include "llvm/Support/raw_ostream.h"
19 #include "llvm/System/Signals.h"
20 #include "llvm/System/Threading.h"
21 #include "llvm/ADT/SmallVector.h"
22 #include <cassert>
23 #include <cstdlib>
24
25 #if defined(HAVE_UNISTD_H)
26 # include <unistd.h>
27 #endif
28 #if defined(_MSC_VER)
29 # include <io.h>
30 # include <fcntl.h>
31 #endif
32
33 using namespace llvm;
34 using namespace std;
35
36 static fatal_error_handler_t ErrorHandler = 0;
37 static void *ErrorHandlerUserData = 0;
38
39 void llvm::install_fatal_error_handler(fatal_error_handler_t handler,
40                                        void *user_data) {
41   assert(!llvm_is_multithreaded() &&
42          "Cannot register error handlers after starting multithreaded mode!\n");
43   assert(!ErrorHandler && "Error handler already registered!\n");
44   ErrorHandler = handler;
45   ErrorHandlerUserData = user_data;
46 }
47
48 void llvm::remove_fatal_error_handler() {
49   ErrorHandler = 0;
50 }
51
52 void llvm::report_fatal_error(const char *Reason) {
53   report_fatal_error(Twine(Reason));
54 }
55
56 void llvm::report_fatal_error(const std::string &Reason) {
57   report_fatal_error(Twine(Reason));
58 }
59
60 void llvm::report_fatal_error(const Twine &Reason) {
61   if (ErrorHandler) {
62     ErrorHandler(ErrorHandlerUserData, Reason.str());
63   } else {
64     // Blast the result out to stderr.  We don't try hard to make sure this
65     // succeeds (e.g. handling EINTR) and we can't use errs() here because
66     // raw ostreams can call report_fatal_error.
67     SmallVector<char, 64> Buffer;
68     StringRef ReasonStr = Reason.toStringRef(Buffer);
69     
70     ::write(2, "LLVM ERROR: ", 12);
71     ::write(2, ReasonStr.data(), ReasonStr.size());
72     ::write(2, "\n", 1);
73   }
74
75   // If we reached here, we are failing ungracefully. Run the interrupt handlers
76   // to make sure any special cleanups get done, in particular that we remove
77   // files registered with RemoveFileOnSignal.
78   sys::RunInterruptHandlers();
79
80   exit(1);
81 }
82
83 void llvm::llvm_unreachable_internal(const char *msg, const char *file,
84                                      unsigned line) {
85   // This code intentionally doesn't call the ErrorHandler callback, because
86   // llvm_unreachable is intended to be used to indicate "impossible"
87   // situations, and not legitimate runtime errors.
88   if (msg)
89     dbgs() << msg << "\n";
90   dbgs() << "UNREACHABLE executed";
91   if (file)
92     dbgs() << " at " << file << ":" << line;
93   dbgs() << "!\n";
94   abort();
95 }