[llvm-commits] [lld] r149428 - /lld/trunk/tools/lld-core/lld-core.cpp

[Michael J. Spencer]

Author: mspencer
Date: Tue Jan 31 15:46:17 2012
New Revision: 149428

URL: http://llvm.org/viewvc/llvm-project?rev=149428&view=rev
Log:
Add pretty stack tracing and llvm_shutdown.

Modified:
    lld/trunk/tools/lld-core/lld-core.cpp

Modified: lld/trunk/tools/lld-core/lld-core.cpp
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/tools/lld-core/lld-core.cpp?rev=149428&r1=149427&r2=149428&view=diff
==============================================================================
--- lld/trunk/tools/lld-core/lld-core.cpp (original)
+++ lld/trunk/tools/lld-core/lld-core.cpp Tue Jan 31 15:46:17 2012
@@ -23,8 +23,11 @@
 #include "llvm/ADT/SmallString.h"
 #include "llvm/ADT/Twine.h"
 #include "llvm/Support/DataTypes.h"
+#include "llvm/Support/ManagedStatic.h"
 #include "llvm/Support/MemoryBuffer.h"
+#include "llvm/Support/PrettyStackTrace.h"
 #include "llvm/Support/raw_ostream.h"
+#include "llvm/Support/Signals.h"
 #include "llvm/Support/system_error.h"
 #include "llvm/Support/ErrorHandling.h"
 #include "llvm/Support/FileSystem.h"
@@ -198,6 +201,11 @@
 }
 
 int main(int argc, const char *argv[]) {
+  // Print a stack trace if we signal out.
+  llvm::sys::PrintStackTraceOnErrorSignal();
+  llvm::PrettyStackTraceProgram X(argc, argv);
+  llvm::llvm_shutdown_obj Y;  // Call llvm_shutdown() on exit.
+
   // read input YAML doc into object file(s)
   std::vector<File *> files;
   if (error(yaml::parseObjectTextFileOrSTDIN(llvm::StringRef(argv[1]), files)))