lib/Target/X86/AsmParser/X86AsmInstrumentation.h

   1 //===- X86AsmInstrumentation.h - Instrument X86 inline assembly -*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #ifndef LLVM_LIB_TARGET_X86_ASMPARSER_X86ASMINSTRUMENTATION_H
  10 #define LLVM_LIB_TARGET_X86_ASMPARSER_X86ASMINSTRUMENTATION_H
  11
  12 #include "llvm/ADT/SmallVector.h"
  13 #include <memory>
  14
  15 namespace llvm {
  16
  17 class MCContext;
  18 class MCInst;
  19 class MCInstrInfo;
  20 class MCParsedAsmOperand;
  21 class MCStreamer;
  22 class MCSubtargetInfo;
  23 class MCTargetOptions;
  24 class X86AsmInstrumentation;
  25
  26 X86AsmInstrumentation *
  27 CreateX86AsmInstrumentation(const MCTargetOptions &MCOptions,
  28                             const MCContext &Ctx,
  29                             const MCSubtargetInfo *&STI);
  30
  31 class X86AsmInstrumentation {
  32 public:
  33   virtual ~X86AsmInstrumentation();
  34
  35   // Sets frame register corresponding to a current frame.
  36   void SetInitialFrameRegister(unsigned RegNo) {
  37     InitialFrameReg = RegNo;
  38   }
  39
  40   // Tries to instrument and emit instruction.
  41   virtual void InstrumentAndEmitInstruction(
  42       const MCInst &Inst,
  43       SmallVectorImpl<std::unique_ptr<MCParsedAsmOperand>> &Operands,
  44       MCContext &Ctx, const MCInstrInfo &MII, MCStreamer &Out);
  45
  46 protected:
  47   friend X86AsmInstrumentation *
  48   CreateX86AsmInstrumentation(const MCTargetOptions &MCOptions,
  49                               const MCContext &Ctx,
  50                               const MCSubtargetInfo *&STI);
  51
  52   X86AsmInstrumentation(const MCSubtargetInfo *&STI);
  53
  54   unsigned GetFrameRegGeneric(const MCContext &Ctx, MCStreamer &Out);
  55
  56   void EmitInstruction(MCStreamer &Out, const MCInst &Inst);
  57
  58   const MCSubtargetInfo *&STI;
  59
  60   unsigned InitialFrameReg = 0;
  61 };
  62
  63 } // end namespace llvm
  64
  65 #endif // LLVM_LIB_TARGET_X86_ASMPARSER_X86ASMINSTRUMENTATION_H