include/llvm/MC/MCDisassembler.h

   1 //===-- llvm/MC/MCDisassembler.h - Disassembler interface -------*- 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 #ifndef LLVM_MC_MCDISASSEMBLER_H
  10 #define LLVM_MC_MCDISASSEMBLER_H
  11
  12 #include "llvm-c/Disassembler.h"
  13 #include "llvm/ADT/OwningPtr.h"
  14 #include "llvm/MC/MCRelocationInfo.h"
  15 #include "llvm/MC/MCSymbolizer.h"
  16 #include "llvm/Support/DataTypes.h"
  17
  18 namespace llvm {
  19
  20 class MCInst;
  21 class MCSubtargetInfo;
  22 class MemoryObject;
  23 class raw_ostream;
  24 class MCContext;
  25
  26 /// MCDisassembler - Superclass for all disassemblers.  Consumes a memory region
  27 ///   and provides an array of assembly instructions.
  28 class MCDisassembler {
  29 public:
  30   /// Ternary decode status. Most backends will just use Fail and
  31   /// Success, however some have a concept of an instruction with
  32   /// understandable semantics but which is architecturally
  33   /// incorrect. An example of this is ARM UNPREDICTABLE instructions
  34   /// which are disassemblable but cause undefined behaviour.
  35   ///
  36   /// Because it makes sense to disassemble these instructions, there
  37   /// is a "soft fail" failure mode that indicates the MCInst& is
  38   /// valid but architecturally incorrect.
  39   ///
  40   /// The enum numbers are deliberately chosen such that reduction
  41   /// from Success->SoftFail ->Fail can be done with a simple
  42   /// bitwise-AND:
  43   ///
  44   ///   LEFT & TOP =  | Success       Unpredictable   Fail
  45   ///   --------------+-----------------------------------
  46   ///   Success       | Success       Unpredictable   Fail
  47   ///   Unpredictable | Unpredictable Unpredictable   Fail
  48   ///   Fail          | Fail          Fail            Fail
  49   ///
  50   /// An easy way of encoding this is as 0b11, 0b01, 0b00 for
  51   /// Success, SoftFail, Fail respectively.
  52   enum DecodeStatus {
  53     Fail = 0,
  54     SoftFail = 1,
  55     Success = 3
  56   };
  57
  58   /// Constructor     - Performs initial setup for the disassembler.
  59   MCDisassembler(const MCSubtargetInfo &STI) : GetOpInfo(0), SymbolLookUp(0),
  60                                                DisInfo(0), Ctx(0),
  61                                                STI(STI), Symbolizer(0),
  62                                                CommentStream(0) {}
  63
  64   virtual ~MCDisassembler();
  65
  66   /// getInstruction  - Returns the disassembly of a single instruction.
  67   ///
  68   /// @param instr    - An MCInst to populate with the contents of the
  69   ///                   instruction.
  70   /// @param size     - A value to populate with the size of the instruction, or
  71   ///                   the number of bytes consumed while attempting to decode
  72   ///                   an invalid instruction.
  73   /// @param region   - The memory object to use as a source for machine code.
  74   /// @param address  - The address, in the memory space of region, of the first
  75   ///                   byte of the instruction.
  76   /// @param vStream  - The stream to print warnings and diagnostic messages on.
  77   /// @param cStream  - The stream to print comments and annotations on.
  78   /// @return         - MCDisassembler::Success if the instruction is valid,
  79   ///                   MCDisassembler::SoftFail if the instruction was
  80   ///                                            disassemblable but invalid,
  81   ///                   MCDisassembler::Fail if the instruction was invalid.
  82   virtual DecodeStatus  getInstruction(MCInst& instr,
  83                                        uint64_t& size,
  84                                        const MemoryObject &region,
  85                                        uint64_t address,
  86                                        raw_ostream &vStream,
  87                                        raw_ostream &cStream) const = 0;
  88 private:
  89   //
  90   // Hooks for symbolic disassembly via the public 'C' interface.
  91   //
  92   // The function to get the symbolic information for operands.
  93   LLVMOpInfoCallback GetOpInfo;
  94   // The function to lookup a symbol name.
  95   LLVMSymbolLookupCallback SymbolLookUp;
  96   // The pointer to the block of symbolic information for above call back.
  97   void *DisInfo;
  98   // The assembly context for creating symbols and MCExprs in place of
  99   // immediate operands when there is symbolic information.
 100   MCContext *Ctx;
 101
 102 protected:
 103   // Subtarget information, for instruction decoding predicates if required.
 104   const MCSubtargetInfo &STI;
 105   OwningPtr<MCSymbolizer> Symbolizer;
 106
 107 public:
 108   // Helpers around MCSymbolizer
 109   bool tryAddingSymbolicOperand(MCInst &Inst,
 110                                 int64_t Value,
 111                                 uint64_t Address, bool IsBranch,
 112                                 uint64_t Offset, uint64_t InstSize) const;
 113
 114   void tryAddingPcLoadReferenceComment(int64_t Value, uint64_t Address) const;
 115
 116   /// Set \p Symzer as the current symbolizer.
 117   /// This takes ownership of \p Symzer, and deletes the previously set one.
 118   void setSymbolizer(OwningPtr<MCSymbolizer> &Symzer);
 119
 120   /// Sets up an external symbolizer that uses the C API callbacks.
 121   void setupForSymbolicDisassembly(LLVMOpInfoCallback GetOpInfo,
 122                                    LLVMSymbolLookupCallback SymbolLookUp,
 123                                    void *DisInfo,
 124                                    MCContext *Ctx,
 125                                    OwningPtr<MCRelocationInfo> &RelInfo);
 126   void setupForSymbolicDisassembly(LLVMOpInfoCallback GetOpInfo,
 127                                    LLVMSymbolLookupCallback SymbolLookUp,
 128                                    void *DisInfo,
 129                                    MCContext *Ctx,
 130                                    std::unique_ptr<MCRelocationInfo> &RelInfo);
 131
 132   LLVMOpInfoCallback getLLVMOpInfoCallback() const { return GetOpInfo; }
 133   LLVMSymbolLookupCallback getLLVMSymbolLookupCallback() const {
 134     return SymbolLookUp;
 135   }
 136   void *getDisInfoBlock() const { return DisInfo; }
 137   MCContext *getMCContext() const { return Ctx; }
 138
 139   const MCSubtargetInfo& getSubtargetInfo() const { return STI; }
 140
 141   // Marked mutable because we cache it inside the disassembler, rather than
 142   // having to pass it around as an argument through all the autogenerated code.
 143   mutable raw_ostream *CommentStream;
 144 };
 145
 146 } // namespace llvm
 147
 148 #endif