Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //===- llvm/MC/MCDisassembler.h - Disassembler interface --------*- 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_MC_MCDISASSEMBLER_MCDISASSEMBLER_H |
||
| 10 | #define LLVM_MC_MCDISASSEMBLER_MCDISASSEMBLER_H |
||
| 11 | |||
| 12 | #include "llvm/ADT/StringRef.h" |
||
| 13 | #include "llvm/BinaryFormat/XCOFF.h" |
||
| 14 | #include "llvm/MC/MCDisassembler/MCSymbolizer.h" |
||
| 15 | #include <cstdint> |
||
| 16 | #include <memory> |
||
| 17 | #include <vector> |
||
| 18 | |||
| 19 | namespace llvm { |
||
| 20 | |||
| 21 | struct XCOFFSymbolInfoTy { |
||
| 22 | std::optional<XCOFF::StorageMappingClass> StorageMappingClass; |
||
| 23 | std::optional<uint32_t> Index; |
||
| 24 | bool IsLabel = false; |
||
| 25 | bool operator<(const XCOFFSymbolInfoTy &SymInfo) const; |
||
| 26 | }; |
||
| 27 | |||
| 28 | struct SymbolInfoTy { |
||
| 29 | uint64_t Addr; |
||
| 30 | StringRef Name; |
||
| 31 | // XCOFF uses XCOFFSymInfo. Other targets use Type. |
||
| 32 | XCOFFSymbolInfoTy XCOFFSymInfo; |
||
| 33 | uint8_t Type; |
||
| 34 | |||
| 35 | private: |
||
| 36 | bool IsXCOFF; |
||
| 37 | bool HasType; |
||
| 38 | |||
| 39 | public: |
||
| 40 | SymbolInfoTy(uint64_t Addr, StringRef Name, |
||
| 41 | std::optional<XCOFF::StorageMappingClass> Smc, |
||
| 42 | std::optional<uint32_t> Idx, bool Label) |
||
| 43 | : Addr(Addr), Name(Name), XCOFFSymInfo{Smc, Idx, Label}, Type(0), |
||
| 44 | IsXCOFF(true), HasType(false) {} |
||
| 45 | SymbolInfoTy(uint64_t Addr, StringRef Name, uint8_t Type, |
||
| 46 | bool IsXCOFF = false) |
||
| 47 | : Addr(Addr), Name(Name), Type(Type), IsXCOFF(IsXCOFF), HasType(true) {} |
||
| 48 | bool isXCOFF() const { return IsXCOFF; } |
||
| 49 | |||
| 50 | private: |
||
| 51 | friend bool operator<(const SymbolInfoTy &P1, const SymbolInfoTy &P2) { |
||
| 52 | assert((P1.IsXCOFF == P2.IsXCOFF && P1.HasType == P2.HasType) && |
||
| 53 | "The value of IsXCOFF and HasType in P1 and P2 should be the same " |
||
| 54 | "respectively."); |
||
| 55 | |||
| 56 | if (P1.IsXCOFF && P1.HasType) |
||
| 57 | return std::tie(P1.Addr, P1.Type, P1.Name) < |
||
| 58 | std::tie(P2.Addr, P2.Type, P2.Name); |
||
| 59 | |||
| 60 | if (P1.IsXCOFF) |
||
| 61 | return std::tie(P1.Addr, P1.XCOFFSymInfo, P1.Name) < |
||
| 62 | std::tie(P2.Addr, P2.XCOFFSymInfo, P2.Name); |
||
| 63 | |||
| 64 | return std::tie(P1.Addr, P1.Name, P1.Type) < |
||
| 65 | std::tie(P2.Addr, P2.Name, P2.Type); |
||
| 66 | } |
||
| 67 | }; |
||
| 68 | |||
| 69 | using SectionSymbolsTy = std::vector<SymbolInfoTy>; |
||
| 70 | |||
| 71 | template <typename T> class ArrayRef; |
||
| 72 | class MCContext; |
||
| 73 | class MCInst; |
||
| 74 | class MCSubtargetInfo; |
||
| 75 | class raw_ostream; |
||
| 76 | |||
| 77 | /// Superclass for all disassemblers. Consumes a memory region and provides an |
||
| 78 | /// array of assembly instructions. |
||
| 79 | class MCDisassembler { |
||
| 80 | public: |
||
| 81 | /// Ternary decode status. Most backends will just use Fail and |
||
| 82 | /// Success, however some have a concept of an instruction with |
||
| 83 | /// understandable semantics but which is architecturally |
||
| 84 | /// incorrect. An example of this is ARM UNPREDICTABLE instructions |
||
| 85 | /// which are disassemblable but cause undefined behaviour. |
||
| 86 | /// |
||
| 87 | /// Because it makes sense to disassemble these instructions, there |
||
| 88 | /// is a "soft fail" failure mode that indicates the MCInst& is |
||
| 89 | /// valid but architecturally incorrect. |
||
| 90 | /// |
||
| 91 | /// The enum numbers are deliberately chosen such that reduction |
||
| 92 | /// from Success->SoftFail ->Fail can be done with a simple |
||
| 93 | /// bitwise-AND: |
||
| 94 | /// |
||
| 95 | /// LEFT & TOP = | Success Unpredictable Fail |
||
| 96 | /// --------------+----------------------------------- |
||
| 97 | /// Success | Success Unpredictable Fail |
||
| 98 | /// Unpredictable | Unpredictable Unpredictable Fail |
||
| 99 | /// Fail | Fail Fail Fail |
||
| 100 | /// |
||
| 101 | /// An easy way of encoding this is as 0b11, 0b01, 0b00 for |
||
| 102 | /// Success, SoftFail, Fail respectively. |
||
| 103 | enum DecodeStatus { |
||
| 104 | Fail = 0, |
||
| 105 | SoftFail = 1, |
||
| 106 | Success = 3 |
||
| 107 | }; |
||
| 108 | |||
| 109 | MCDisassembler(const MCSubtargetInfo &STI, MCContext &Ctx) |
||
| 110 | : Ctx(Ctx), STI(STI) {} |
||
| 111 | |||
| 112 | virtual ~MCDisassembler(); |
||
| 113 | |||
| 114 | /// Returns the disassembly of a single instruction. |
||
| 115 | /// |
||
| 116 | /// \param Instr - An MCInst to populate with the contents of the |
||
| 117 | /// instruction. |
||
| 118 | /// \param Size - A value to populate with the size of the instruction, or |
||
| 119 | /// the number of bytes consumed while attempting to decode |
||
| 120 | /// an invalid instruction. |
||
| 121 | /// \param Address - The address, in the memory space of region, of the first |
||
| 122 | /// byte of the instruction. |
||
| 123 | /// \param Bytes - A reference to the actual bytes of the instruction. |
||
| 124 | /// \param CStream - The stream to print comments and annotations on. |
||
| 125 | /// \return - MCDisassembler::Success if the instruction is valid, |
||
| 126 | /// MCDisassembler::SoftFail if the instruction was |
||
| 127 | /// disassemblable but invalid, |
||
| 128 | /// MCDisassembler::Fail if the instruction was invalid. |
||
| 129 | virtual DecodeStatus getInstruction(MCInst &Instr, uint64_t &Size, |
||
| 130 | ArrayRef<uint8_t> Bytes, uint64_t Address, |
||
| 131 | raw_ostream &CStream) const = 0; |
||
| 132 | |||
| 133 | /// Used to perform separate target specific disassembly for a particular |
||
| 134 | /// symbol. May parse any prelude that precedes instructions after the |
||
| 135 | /// start of a symbol, or the entire symbol. |
||
| 136 | /// This is used for example by WebAssembly to decode preludes. |
||
| 137 | /// |
||
| 138 | /// Base implementation returns std::nullopt. So all targets by default ignore |
||
| 139 | /// to treat symbols separately. |
||
| 140 | /// |
||
| 141 | /// \param Symbol - The symbol. |
||
| 142 | /// \param Size - The number of bytes consumed. |
||
| 143 | /// \param Address - The address, in the memory space of region, of the first |
||
| 144 | /// byte of the symbol. |
||
| 145 | /// \param Bytes - A reference to the actual bytes at the symbol location. |
||
| 146 | /// \param CStream - The stream to print comments and annotations on. |
||
| 147 | /// \return - MCDisassembler::Success if bytes are decoded |
||
| 148 | /// successfully. Size must hold the number of bytes that |
||
| 149 | /// were decoded. |
||
| 150 | /// - MCDisassembler::Fail if the bytes are invalid. Size |
||
| 151 | /// must hold the number of bytes that were decoded before |
||
| 152 | /// failing. The target must print nothing. This can be |
||
| 153 | /// done by buffering the output if needed. |
||
| 154 | /// - std::nullopt if the target doesn't want to handle the |
||
| 155 | /// symbol separately. Value of Size is ignored in this |
||
| 156 | /// case. |
||
| 157 | virtual std::optional<DecodeStatus> |
||
| 158 | onSymbolStart(SymbolInfoTy &Symbol, uint64_t &Size, ArrayRef<uint8_t> Bytes, |
||
| 159 | uint64_t Address, raw_ostream &CStream) const; |
||
| 160 | // TODO: |
||
| 161 | // Implement similar hooks that can be used at other points during |
||
| 162 | // disassembly. Something along the following lines: |
||
| 163 | // - onBeforeInstructionDecode() |
||
| 164 | // - onAfterInstructionDecode() |
||
| 165 | // - onSymbolEnd() |
||
| 166 | // It should help move much of the target specific code from llvm-objdump to |
||
| 167 | // respective target disassemblers. |
||
| 168 | |||
| 169 | /// Suggest a distance to skip in a buffer of data to find the next |
||
| 170 | /// place to look for the start of an instruction. For example, if |
||
| 171 | /// all instructions have a fixed alignment, this might advance to |
||
| 172 | /// the next multiple of that alignment. |
||
| 173 | /// |
||
| 174 | /// If not overridden, the default is 1. |
||
| 175 | /// |
||
| 176 | /// \param Address - The address, in the memory space of region, of the |
||
| 177 | /// starting point (typically the first byte of something |
||
| 178 | /// that did not decode as a valid instruction at all). |
||
| 179 | /// \param Bytes - A reference to the actual bytes at Address. May be |
||
| 180 | /// needed in order to determine the width of an |
||
| 181 | /// unrecognized instruction (e.g. in Thumb this is a simple |
||
| 182 | /// consistent criterion that doesn't require knowing the |
||
| 183 | /// specific instruction). The caller can pass as much data |
||
| 184 | /// as they have available, and the function is required to |
||
| 185 | /// make a reasonable default choice if not enough data is |
||
| 186 | /// available to make a better one. |
||
| 187 | /// \return - A number of bytes to skip. Must always be greater than |
||
| 188 | /// zero. May be greater than the size of Bytes. |
||
| 189 | virtual uint64_t suggestBytesToSkip(ArrayRef<uint8_t> Bytes, |
||
| 190 | uint64_t Address) const; |
||
| 191 | |||
| 192 | private: |
||
| 193 | MCContext &Ctx; |
||
| 194 | |||
| 195 | protected: |
||
| 196 | // Subtarget information, for instruction decoding predicates if required. |
||
| 197 | const MCSubtargetInfo &STI; |
||
| 198 | std::unique_ptr<MCSymbolizer> Symbolizer; |
||
| 199 | |||
| 200 | public: |
||
| 201 | // Helpers around MCSymbolizer |
||
| 202 | bool tryAddingSymbolicOperand(MCInst &Inst, int64_t Value, uint64_t Address, |
||
| 203 | bool IsBranch, uint64_t Offset, uint64_t OpSize, |
||
| 204 | uint64_t InstSize) const; |
||
| 205 | |||
| 206 | void tryAddingPcLoadReferenceComment(int64_t Value, uint64_t Address) const; |
||
| 207 | |||
| 208 | /// Set \p Symzer as the current symbolizer. |
||
| 209 | /// This takes ownership of \p Symzer, and deletes the previously set one. |
||
| 210 | void setSymbolizer(std::unique_ptr<MCSymbolizer> Symzer); |
||
| 211 | |||
| 212 | MCContext& getContext() const { return Ctx; } |
||
| 213 | |||
| 214 | const MCSubtargetInfo& getSubtargetInfo() const { return STI; } |
||
| 215 | |||
| 216 | // Marked mutable because we cache it inside the disassembler, rather than |
||
| 217 | // having to pass it around as an argument through all the autogenerated code. |
||
| 218 | mutable raw_ostream *CommentStream = nullptr; |
||
| 219 | }; |
||
| 220 | |||
| 221 | } // end namespace llvm |
||
| 222 | |||
| 223 | #endif // LLVM_MC_MCDISASSEMBLER_MCDISASSEMBLER_H |