llvm/lib/MC/MCDisassembler/MCDisassembler.cpp

   1 //===- MCDisassembler.cpp - Disassembler interface ------------------------===//
   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 #include "llvm/MC/MCDisassembler/MCDisassembler.h"
  10 #include "llvm/ADT/ArrayRef.h"
  11 #include "llvm/ADT/StringRef.h"
  12 #include "llvm/Support/raw_ostream.h"
  13 #include <algorithm>
  14
  15 using namespace llvm;
  16
  17 MCDisassembler::~MCDisassembler() = default;
  18
  19 Optional<MCDisassembler::DecodeStatus>
  20 MCDisassembler::onSymbolStart(SymbolInfoTy &Symbol, uint64_t &Size,
  21                               ArrayRef<uint8_t> Bytes, uint64_t Address,
  22                               raw_ostream &CStream) const {
  23   return None;
  24 }
  25
  26 bool MCDisassembler::tryAddingSymbolicOperand(MCInst &Inst, int64_t Value,
  27                                               uint64_t Address, bool IsBranch,
  28                                               uint64_t Offset,
  29                                               uint64_t InstSize) const {
  30   if (Symbolizer)
  31     return Symbolizer->tryAddingSymbolicOperand(
  32         Inst, *CommentStream, Value, Address, IsBranch, Offset, InstSize);
  33   return false;
  34 }
  35
  36 void MCDisassembler::tryAddingPcLoadReferenceComment(int64_t Value,
  37                                                      uint64_t Address) const {
  38   if (Symbolizer)
  39     Symbolizer->tryAddingPcLoadReferenceComment(*CommentStream, Value, Address);
  40 }
  41
  42 void MCDisassembler::setSymbolizer(std::unique_ptr<MCSymbolizer> Symzer) {
  43   Symbolizer = std::move(Symzer);
  44 }
  45
  46 #define SMC_PCASE(A, P)                                                         \
  47   case XCOFF::XMC_##A:                                                         \
  48     return P;
  49
  50 static uint8_t getSMCPriority(XCOFF::StorageMappingClass SMC) {
  51   switch (SMC) {
  52     SMC_PCASE(PR, 1)
  53     SMC_PCASE(RO, 1)
  54     SMC_PCASE(DB, 1)
  55     SMC_PCASE(GL, 1)
  56     SMC_PCASE(XO, 1)
  57     SMC_PCASE(SV, 1)
  58     SMC_PCASE(SV64, 1)
  59     SMC_PCASE(SV3264, 1)
  60     SMC_PCASE(TI, 1)
  61     SMC_PCASE(TB, 1)
  62     SMC_PCASE(RW, 1)
  63     SMC_PCASE(TC0, 0)
  64     SMC_PCASE(TC, 1)
  65     SMC_PCASE(TD, 1)
  66     SMC_PCASE(DS, 1)
  67     SMC_PCASE(UA, 1)
  68     SMC_PCASE(BS, 1)
  69     SMC_PCASE(UC, 1)
  70     SMC_PCASE(TL, 1)
  71     SMC_PCASE(UL, 1)
  72     SMC_PCASE(TE, 1)
  73 #undef SMC_PCASE
  74   }
  75   return 0;
  76 }
  77
  78 /// The function is for symbol sorting when symbols have the same address.
  79 /// The symbols in the same section are sorted in ascending order.
  80 /// llvm-objdump -D will choose the highest priority symbol to display when
  81 /// there are symbols with the same address.
  82 bool XCOFFSymbolInfo::operator<(const XCOFFSymbolInfo &SymInfo) const {
  83   // Label symbols have higher priority than non-label symbols.
  84   if (IsLabel != SymInfo.IsLabel)
  85     return SymInfo.IsLabel;
  86
  87   // Symbols with a StorageMappingClass have higher priority than those without.
  88   if (StorageMappingClass.hasValue() != SymInfo.StorageMappingClass.hasValue())
  89     return SymInfo.StorageMappingClass.hasValue();
  90
  91   if (StorageMappingClass.hasValue()) {
  92     return getSMCPriority(StorageMappingClass.getValue()) <
  93            getSMCPriority(SymInfo.StorageMappingClass.getValue());
  94   }
  95
  96   return false;
  97 }