lib/Target/X86/AsmParser/X86AsmInstrumentation.h

   1 //===- X86AsmInstrumentation.h - Instrument X86 inline assembly *- 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 #ifndef LLVM_LIB_TARGET_X86_ASMPARSER_X86ASMINSTRUMENTATION_H
  11 #define LLVM_LIB_TARGET_X86_ASMPARSER_X86ASMINSTRUMENTATION_H
  12
  13 #include "llvm/ADT/SmallVector.h"
  14
  15 #include <memory>
  16
  17 namespace llvm {
  18
  19 class MCContext;
  20 class MCInst;
  21 class MCInstrInfo;
  22 class MCParsedAsmOperand;
  23 class MCStreamer;
  24 class MCSubtargetInfo;
  25 class MCTargetOptions;
  26
  27 class X86AsmInstrumentation;
  28
  29 X86AsmInstrumentation *
  30 CreateX86AsmInstrumentation(const MCTargetOptions &MCOptions,
  31                             const MCContext &Ctx, const MCSubtargetInfo &STI);
  32
  33 class X86AsmInstrumentation {
  34 public:
  35   virtual ~X86AsmInstrumentation();
  36
  37   void SetFrameRegister(unsigned RegNo) {
  38     FrameReg = RegNo;
  39   }
  40
  41   // Tries to instrument and emit instruction.
  42   virtual void InstrumentAndEmitInstruction(
  43       const MCInst &Inst,
  44       SmallVectorImpl<std::unique_ptr<MCParsedAsmOperand> > &Operands,
  45       MCContext &Ctx, const MCInstrInfo &MII, MCStreamer &Out);
  46
  47 protected:
  48   friend X86AsmInstrumentation *
  49   CreateX86AsmInstrumentation(const MCTargetOptions &MCOptions,
  50                               const MCContext &Ctx, const MCSubtargetInfo &STI);
  51
  52   X86AsmInstrumentation(const MCSubtargetInfo &STI);
  53
  54   void EmitInstruction(MCStreamer &Out, const MCInst &Inst);
  55
  56   const MCSubtargetInfo &STI;
  57
  58   unsigned FrameReg;
  59 };
  60
  61 } // End llvm namespace
  62
  63 #endif