PrettyStackTrace.cpp [plain text]
#include "llvm/Support/PrettyStackTrace.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/System/Signals.h"
#include "llvm/ADT/SmallString.h"
using namespace llvm;
static const PrettyStackTraceEntry *PrettyStackTraceHead = 0;
static unsigned PrintStack(const PrettyStackTraceEntry *Entry, raw_ostream &OS){
unsigned NextID = 0;
if (Entry->getNextEntry())
NextID = PrintStack(Entry->getNextEntry(), OS);
OS << NextID << ".\t";
Entry->print(OS);
return NextID+1;
}
static void PrintCurStackTrace(raw_ostream &OS) {
if (PrettyStackTraceHead == 0) return;
OS << "Stack dump:\n";
PrintStack(PrettyStackTraceHead, OS);
OS.flush();
}
#ifdef __APPLE__
extern "C" const char *__crashreporter_info__;
const char *__crashreporter_info__ = 0;
#endif
static void CrashHandler(void *Cookie) {
#ifndef __APPLE__
PrintCurStackTrace(errs());
#else
SmallString<2048> TmpStr;
{
raw_svector_ostream Stream(TmpStr);
PrintCurStackTrace(Stream);
}
if (!TmpStr.empty()) {
__crashreporter_info__ = strdup(TmpStr.c_str());
errs() << __crashreporter_info__;
}
#endif
}
static bool RegisterCrashPrinter() {
sys::AddSignalHandler(CrashHandler, 0);
return false;
}
PrettyStackTraceEntry::PrettyStackTraceEntry() {
static bool HandlerRegistered = RegisterCrashPrinter();
HandlerRegistered = HandlerRegistered;
NextEntry = PrettyStackTraceHead;
PrettyStackTraceHead = this;
}
PrettyStackTraceEntry::~PrettyStackTraceEntry() {
assert(PrettyStackTraceHead == this &&
"Pretty stack trace entry destruction is out of order");
PrettyStackTraceHead = getNextEntry();
}
void PrettyStackTraceString::print(raw_ostream &OS) const {
OS << Str << "\n";
}
void PrettyStackTraceProgram::print(raw_ostream &OS) const {
OS << "Program arguments: ";
for (unsigned i = 0, e = ArgC; i != e; ++i)
OS << ArgV[i] << ' ';
OS << '\n';
}