add some helper classes for building light-weight symbolic stack traces
[oota-llvm.git] / include / llvm / Support / PrettyStackTrace.h
1 //===- llvm/Support/PrettyStackTrace.h - Pretty Crash Handling --*- C++ -*-===//
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 the PrettyStackTraceEntry class, which is used to make
11 // crashes give more contextual information about what the program was doing
12 // when it crashed.
13 //
14 //===----------------------------------------------------------------------===//
15
16 #ifndef LLVM_SUPPORT_PRETTYSTACKTRACE_H
17 #define LLVM_SUPPORT_PRETTYSTACKTRACE_H
18
19 namespace llvm {
20   class raw_ostream;
21   
22   /// PrettyStackTraceEntry - This class is used to represent a frame of the
23   /// "pretty" stack trace that is dumped when a program crashes. You can define
24   /// subclasses of this and declare them on the program stack: when they are 
25   /// constructed and destructed, they will add their symbolic frames to a
26   /// virtual stack trace.  This gets dumped out if the program crashes.
27   class PrettyStackTraceEntry {
28     const PrettyStackTraceEntry *NextEntry;
29     PrettyStackTraceEntry(const PrettyStackTraceEntry &); // DO NOT IMPLEMENT
30     void operator=(const PrettyStackTraceEntry&);         // DO NOT IMPLEMENT
31   public:
32     PrettyStackTraceEntry();
33     virtual ~PrettyStackTraceEntry();
34     
35     /// print - Emit information about this stack frame to OS.
36     virtual void print(raw_ostream &OS) const = 0;
37     
38     /// getNextEntry - Return the next entry in the list of frames.
39     const PrettyStackTraceEntry *getNextEntry() const { return NextEntry; }
40   };
41   
42   /// PrettyStackTraceString - This object prints a specified string (which
43   /// should not contain newlines) to the stream as the stack trace when a crash
44   /// occurs.
45   class PrettyStackTraceString : public PrettyStackTraceEntry {
46     const char *Str;
47   public:
48     PrettyStackTraceString(const char *str) : Str(str) {}
49     virtual void print(raw_ostream &OS) const;
50   };
51   
52   /// PrettyStackTraceProgram - This object prints a specified program arguments
53   /// to the stream as the stack trace when a crash occurs.
54   class PrettyStackTraceProgram : public PrettyStackTraceEntry {
55     int ArgC;
56     const char *const *ArgV;
57   public:
58     PrettyStackTraceProgram(int argc, const char * const*argv)
59       : ArgC(argc), ArgV(argv) {}
60     virtual void print(raw_ostream &OS) const;
61   };
62   
63 } // end namespace llvm
64
65 #endif