1 //===- PrettyStackTrace.cpp - Pretty Crash Handling -----------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file defines some helpful functions for dealing with the possibility of
11 // Unix signals occuring while your program is running.
13 //===----------------------------------------------------------------------===//
15 #include "llvm/Support/PrettyStackTrace.h"
16 #include "llvm/Support/raw_ostream.h"
17 #include "llvm/System/Signals.h"
18 #include "llvm/ADT/SmallString.h"
21 // FIXME: This should be thread local when llvm supports threads.
22 static const PrettyStackTraceEntry *PrettyStackTraceHead = 0;
24 static unsigned PrintStack(const PrettyStackTraceEntry *Entry, raw_ostream &OS){
26 if (Entry->getNextEntry())
27 NextID = PrintStack(Entry->getNextEntry(), OS);
28 OS << NextID << ".\t";
34 /// PrintCurStackTrace - Print the current stack trace to the specified stream.
35 static void PrintCurStackTrace(raw_ostream &OS) {
36 // Don't print an empty trace.
37 if (PrettyStackTraceHead == 0) return;
39 // If there are pretty stack frames registered, walk and emit them.
40 OS << "Stack dump:\n";
42 PrintStack(PrettyStackTraceHead, OS);
46 // Integrate with crash reporter.
48 extern "C" const char *__crashreporter_info__;
49 const char *__crashreporter_info__ = 0;
53 /// CrashHandler - This callback is run if a fatal signal is delivered to the
54 /// process, it prints the pretty stack trace.
55 static void CrashHandler(void *Cookie) {
57 // On non-apple systems, just emit the crash stack trace to stderr.
58 PrintCurStackTrace(errs());
60 // Otherwise, emit to a smallvector of chars, send *that* to stderr, but also
61 // put it into __crashreporter_info__.
62 SmallString<2048> TmpStr;
64 raw_svector_ostream Stream(TmpStr);
65 PrintCurStackTrace(Stream);
68 if (!TmpStr.empty()) {
69 __crashreporter_info__ = strdup(TmpStr.c_str());
70 errs() << __crashreporter_info__;
77 static bool RegisterCrashPrinter() {
78 sys::AddSignalHandler(CrashHandler, 0);
82 PrettyStackTraceEntry::PrettyStackTraceEntry() {
83 // The first time this is called, we register the crash printer.
84 static bool HandlerRegistered = RegisterCrashPrinter();
85 HandlerRegistered = HandlerRegistered;
88 NextEntry = PrettyStackTraceHead;
89 PrettyStackTraceHead = this;
92 PrettyStackTraceEntry::~PrettyStackTraceEntry() {
93 assert(PrettyStackTraceHead == this &&
94 "Pretty stack trace entry destruction is out of order");
95 PrettyStackTraceHead = getNextEntry();
98 void PrettyStackTraceString::print(raw_ostream &OS) const {
102 void PrettyStackTraceProgram::print(raw_ostream &OS) const {
103 OS << "Program arguments: ";
104 // Print the argument list.
105 for (unsigned i = 0, e = ArgC; i != e; ++i)
106 OS << ArgV[i] << ' ';