lib/Transforms/Instrumentation/Instrumentation.cpp

   1 //===-- Instrumentation.cpp - TransformUtils Infrastructure ---------------===//
   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 common initialization infrastructure for the
  11 // Instrumentation library.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/Transforms/Instrumentation.h"
  16 #include "llvm-c/Initialization.h"
  17 #include "llvm/IR/IntrinsicInst.h"
  18 #include "llvm/InitializePasses.h"
  19 #include "llvm/PassRegistry.h"
  20
  21 using namespace llvm;
  22
  23 /// Moves I before IP. Returns new insert point.
  24 static BasicBlock::iterator moveBeforeInsertPoint(BasicBlock::iterator I, BasicBlock::iterator IP) {
  25   // If I is IP, move the insert point down.
  26   if (I == IP)
  27     return ++IP;
  28   // Otherwise, move I before IP and return IP.
  29   I->moveBefore(&*IP);
  30   return IP;
  31 }
  32
  33 /// Instrumentation passes often insert conditional checks into entry blocks.
  34 /// Call this function before splitting the entry block to move instructions
  35 /// that must remain in the entry block up before the split point. Static
  36 /// allocas and llvm.localescape calls, for example, must remain in the entry
  37 /// block.
  38 BasicBlock::iterator llvm::PrepareToSplitEntryBlock(BasicBlock &BB,
  39                                                     BasicBlock::iterator IP) {
  40   assert(&BB.getParent()->getEntryBlock() == &BB);
  41   for (auto I = IP, E = BB.end(); I != E; ++I) {
  42     bool KeepInEntry = false;
  43     if (auto *AI = dyn_cast<AllocaInst>(I)) {
  44       if (AI->isStaticAlloca())
  45         KeepInEntry = true;
  46     } else if (auto *II = dyn_cast<IntrinsicInst>(I)) {
  47       if (II->getIntrinsicID() == llvm::Intrinsic::localescape)
  48         KeepInEntry = true;
  49     }
  50     if (KeepInEntry)
  51       IP = moveBeforeInsertPoint(I, IP);
  52   }
  53   return IP;
  54 }
  55
  56 /// initializeInstrumentation - Initialize all passes in the TransformUtils
  57 /// library.
  58 void llvm::initializeInstrumentation(PassRegistry &Registry) {
  59   initializeAddressSanitizerPass(Registry);
  60   initializeAddressSanitizerModulePass(Registry);
  61   initializeBoundsCheckingPass(Registry);
  62   initializeGCOVProfilerPass(Registry);
  63   initializePGOInstrumentationGenPass(Registry);
  64   initializePGOInstrumentationUsePass(Registry);
  65   initializeInstrProfilingPass(Registry);
  66   initializeMemorySanitizerPass(Registry);
  67   initializeThreadSanitizerPass(Registry);
  68   initializeSanitizerCoverageModulePass(Registry);
  69   initializeDataFlowSanitizerPass(Registry);
  70   initializeSafeStackPass(Registry);
  71 }
  72
  73 /// LLVMInitializeInstrumentation - C binding for
  74 /// initializeInstrumentation.
  75 void LLVMInitializeInstrumentation(LLVMPassRegistryRef R) {
  76   initializeInstrumentation(*unwrap(R));
  77 }