tesla::FnCallerInstrumenter Class Reference

Detailed Description

Instruments function calls in the caller context.

Definition at line 56 of file Caller.h.

#include "Caller.h"

Inheritance diagram for tesla::FnCallerInstrumenter:

Collaboration diagram for tesla::FnCallerInstrumenter:

Public Member Functions
	FnCallerInstrumenter (const Manifest &M, bool SuppressDI)
	~FnCallerInstrumenter ()
const char *	getPassName () const
virtual bool	doInitialization (llvm::Module &M)
virtual bool	runOnFunction (llvm::Function &Fn)
virtual bool	runOnBasicBlock (llvm::BasicBlock &Block)
Public Member Functions inherited from tesla::Instrumenter
	Instrumenter (const Manifest &M, bool SuppressDI)
virtual	~Instrumenter ()

Static Public Attributes
static char	ID = 0

Additional Inherited Members
Protected Attributes inherited from tesla::Instrumenter
const Manifest &	M
	TESLA manifest that describes automata. More...
const bool	SuppressDebugInstr
	Don't produce debug instrumentation (e.g., printf() statements). More...

Constructor & Destructor Documentation

tesla::FnCallerInstrumenter::FnCallerInstrumenter ( const Manifest &  M, bool  SuppressDI  )

inline

Definition at line 59 of file Caller.h.

    : Instrumenter(M, SuppressDI), FunctionPass(ID) {}

tesla::FnCallerInstrumenter::~FnCallerInstrumenter

(

)

Definition at line 53 of file Caller.cpp.

                                            {
  ::google::protobuf::ShutdownProtobufLibrary();
}

Member Function Documentation

bool tesla::FnCallerInstrumenter::doInitialization ( llvm::Module &  M )

virtual

Definition at line 57 of file Caller.cpp.

References tesla::FnTransition::FnEvent().

                                                       {
  bool ModifiedIR = true;

  for (auto i : M.RootAutomata()) {
    auto& A = *M.FindAutomaton(i->identifier());
    for (auto EquivClass : A) {
      assert(!EquivClass.empty());

      auto *Head = dyn_cast<FnTransition>(*EquivClass.begin());
      if (!Head)
        continue;

      auto& FnEvent = Head->FnEvent();
      if (FnEvent.context() != FunctionEvent::Caller)
        continue;

      Function *Target = Mod.getFunction(FnEvent.function().name());
      if (!Target)
        continue;

      GetOrCreateInstr(Mod, Target, FnEvent.direction())
        ->AppendInstrumentation(A, FnEvent, EquivClass);

      ModifiedIR = true;
    }
  }

  return ModifiedIR;
}

Here is the call graph for this function:

const char* tesla::FnCallerInstrumenter::getPassName ( ) const

inline

Definition at line 63 of file Caller.h.

                                  {
    return "TESLA function instrumenter (caller-side)";
  }

bool tesla::FnCallerInstrumenter::runOnBasicBlock ( llvm::BasicBlock &  Block )

virtual

Definition at line 114 of file Caller.cpp.

References tesla::CallerInstrumentation::Instrument().

                                                            {
  bool ModifiedIR = false;

  for (auto &Inst : Block) {
    if (!isa<CallInst>(Inst)) continue;
    CallInst &Call = cast<CallInst>(Inst);
    Function *Callee = Call.getCalledFunction();

    // TODO: handle indirection (e.g. function pointers)?
    if (!Callee)
      continue;

    StringRef Name = Callee->getName();
    if (auto Instr = Calls.lookup(Name))
      ModifiedIR |= Instr->Instrument(Call);

    if (auto Instr = Returns.lookup(Name))
      ModifiedIR |= Instr->Instrument(Call);
  }

  return ModifiedIR;
}

Here is the call graph for this function:

bool tesla::FnCallerInstrumenter::runOnFunction ( llvm::Function &  Fn )

virtual

Definition at line 104 of file Caller.cpp.

                                                     {
  bool modifiedIR = false;

  for (auto &Block : Fn) {
    modifiedIR |= runOnBasicBlock(Block);
  }

  return modifiedIR;
}

Member Data Documentation

char tesla::FnCallerInstrumenter::ID = 0

static

Definition at line 58 of file Caller.h.

---

The documentation for this class was generated from the following files:

• tesla/instrumenter/Caller.h
• tesla/instrumenter/Caller.cpp