contrib/llvm/include/llvm/Analysis/IndirectCallSiteVisitor.h

   1 //===-- IndirectCallSiteVisitor.h - indirect call-sites visitor -----------===//
   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 implements defines a visitor class and a helper function that find
  11 // all indirect call-sites in a function.
  12
  13 #include "llvm/IR/InstVisitor.h"
  14 #include <vector>
  15
  16 namespace llvm {
  17 // Visitor class that finds all indirect call sites.
  18 struct PGOIndirectCallSiteVisitor
  19     : public InstVisitor<PGOIndirectCallSiteVisitor> {
  20   std::vector<Instruction *> IndirectCallInsts;
  21   PGOIndirectCallSiteVisitor() {}
  22
  23   void visitCallSite(CallSite CS) {
  24     if (CS.isIndirectCall())
  25       IndirectCallInsts.push_back(CS.getInstruction());
  26   }
  27 };
  28
  29 // Helper function that finds all indirect call sites.
  30 inline std::vector<Instruction *> findIndirectCallSites(Function &F) {
  31   PGOIndirectCallSiteVisitor ICV;
  32   ICV.visit(F);
  33   return ICV.IndirectCallInsts;
  34 }
  35 }