include/llvm/Transforms/Instrumentation/TraceValues.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

//===- llvm/Transforms/Instrumentation/TraceValues.h - Tracing ---*- C++ -*--=//
//
// Support for inserting LLVM code to print values at basic block and method
// exits.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_TRANSFORMS_INSTRUMENTATION_TRACEVALUES_H
#define LLVM_TRANSFORMS_INSTRUMENTATION_TRACEVALUES_H

#include "llvm/Pass.h"
class Method;

class InsertTraceCode : public MethodPass {
  bool TraceBasicBlockExits, TraceMethodExits;
  Method *PrintfMeth;
public:
  InsertTraceCode(bool traceBasicBlockExits, bool traceMethodExits)
    : TraceBasicBlockExits(traceBasicBlockExits), 
      TraceMethodExits(traceMethodExits) {}

  // Add a prototype for printf if it is not already in the program.
  //
  bool doInitialization(Module *M);

  //--------------------------------------------------------------------------
  // Function InsertCodeToTraceValues
  // 
  // Inserts tracing code for all live values at basic block and/or method exits
  // as specified by `traceBasicBlockExits' and `traceMethodExits'.
  //
  static bool doit(Method *M, bool traceBasicBlockExits,
                   bool traceMethodExits, Method *Printf);

  // runOnMethod - This method does the work.  Always successful.
  //
  bool runOnMethod(Method *M) {
    return doit(M, TraceBasicBlockExits, TraceMethodExits, PrintfMeth);
  }
};

#endif