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"
20 // FIXME: This should be thread local when llvm supports threads.
21 static const PrettyStackTraceEntry *PrettyStackTraceHead = 0;
23 static unsigned PrintStack(const PrettyStackTraceEntry *Entry, raw_ostream &OS){
25 if (Entry->getNextEntry())
26 NextID = PrintStack(Entry->getNextEntry(), OS);
27 OS << NextID << ".\t";
33 /// CrashHandler - This callback is run if a fatal signal is delivered to the
34 /// process, it prints the pretty stack trace.
35 static void CrashHandler(void *Cookie) {
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 raw_ostream &OS = errs();
41 OS << "Stack dump:\n";
43 PrintStack(PrettyStackTraceHead, OS);
47 static bool RegisterCrashPrinter() {
48 sys::AddSignalHandler(CrashHandler, 0);
52 PrettyStackTraceEntry::PrettyStackTraceEntry() {
53 // The first time this is called, we register the crash printer.
54 static bool HandlerRegistered = RegisterCrashPrinter();
55 HandlerRegistered = HandlerRegistered;
58 NextEntry = PrettyStackTraceHead;
59 PrettyStackTraceHead = this;
62 PrettyStackTraceEntry::~PrettyStackTraceEntry() {
63 assert(PrettyStackTraceHead == this &&
64 "Pretty stack trace entry destruction is out of order");
65 PrettyStackTraceHead = getNextEntry();
68 void PrettyStackTraceString::print(raw_ostream &OS) const {
72 void PrettyStackTraceProgram::print(raw_ostream &OS) const {
73 OS << "Program arguments: ";
74 // Print the argument list.
75 for (unsigned i = 0, e = ArgC; i != e; ++i)