1 //===- bolt/Core/BinaryData.cpp - Objects in a binary file ----------------===//
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 // This file implements the BinaryData class.
10 //
11 //===----------------------------------------------------------------------===//
12
13 #include "bolt/Core/BinaryData.h"
14 #include "bolt/Core/BinarySection.h"
15 #include "llvm/Support/CommandLine.h"
16 #include "llvm/Support/Regex.h"
17
18 using namespace llvm;
19 using namespace bolt;
20
21 #define DEBUG_TYPE "bolt"
22
23 namespace opts {
24 extern cl::OptionCategory BoltCategory;
25 extern cl::opt<unsigned> Verbosity;
26
27 cl::opt<bool>
28 PrintSymbolAliases("print-aliases",
29 cl::desc("print aliases when printing objects"),
30 cl::Hidden, cl::cat(BoltCategory));
31 }
32
isAbsolute() const33 bool BinaryData::isAbsolute() const { return Flags & SymbolRef::SF_Absolute; }
34
isMoveable() const35 bool BinaryData::isMoveable() const {
36 return (!isAbsolute() && (IsMoveable && (!Parent || isTopLevelJumpTable())));
37 }
38
merge(const BinaryData * Other)39 void BinaryData::merge(const BinaryData *Other) {
40 assert(!Size || !Other->Size || Size == Other->Size);
41 assert(Address == Other->Address);
42 assert(*Section == *Other->Section);
43 assert(OutputOffset == Other->OutputOffset);
44 assert(OutputSection == Other->OutputSection);
45 Symbols.insert(Symbols.end(), Other->Symbols.begin(), Other->Symbols.end());
46 Flags |= Other->Flags;
47 if (!Size)
48 Size = Other->Size;
49 }
50
hasName(StringRef Name) const51 bool BinaryData::hasName(StringRef Name) const {
52 for (const MCSymbol *Symbol : Symbols)
53 if (Name == Symbol->getName())
54 return true;
55 return false;
56 }
57
nameStartsWith(StringRef Prefix) const58 bool BinaryData::nameStartsWith(StringRef Prefix) const {
59 for (const MCSymbol *Symbol : Symbols)
60 if (Symbol->getName().starts_with(Prefix))
61 return true;
62 return false;
63 }
64
getSectionName() const65 StringRef BinaryData::getSectionName() const { return getSection().getName(); }
66
getOutputSectionName() const67 StringRef BinaryData::getOutputSectionName() const {
68 return getOutputSection().getName();
69 }
70
getOutputAddress() const71 uint64_t BinaryData::getOutputAddress() const {
72 assert(OutputSection->getOutputAddress());
73 return OutputSection->getOutputAddress() + OutputOffset;
74 }
75
getOffset() const76 uint64_t BinaryData::getOffset() const {
77 return Address - getSection().getAddress();
78 }
79
setSection(BinarySection & NewSection)80 void BinaryData::setSection(BinarySection &NewSection) {
81 if (OutputSection == Section)
82 OutputSection = &NewSection;
83 Section = &NewSection;
84 }
85
isMoved() const86 bool BinaryData::isMoved() const {
87 return (getOffset() != OutputOffset || OutputSection != Section);
88 }
89
print(raw_ostream & OS) const90 void BinaryData::print(raw_ostream &OS) const { printBrief(OS); }
91
printBrief(raw_ostream & OS) const92 void BinaryData::printBrief(raw_ostream &OS) const {
93 OS << "(";
94
95 if (isJumpTable())
96 OS << "jump-table: ";
97 else
98 OS << "object: ";
99
100 OS << getName();
101
102 if ((opts::PrintSymbolAliases || opts::Verbosity > 1) && Symbols.size() > 1) {
103 OS << ", aliases:";
104 for (unsigned I = 1u; I < Symbols.size(); ++I) {
105 OS << (I == 1 ? " (" : ", ") << Symbols[I]->getName();
106 }
107 OS << ")";
108 }
109
110 if (Parent) {
111 OS << " (parent: ";
112 Parent->printBrief(OS);
113 OS << ")";
114 }
115
116 OS << ", 0x" << Twine::utohexstr(getAddress()) << ":0x"
117 << Twine::utohexstr(getEndAddress()) << "/" << getSize() << "/"
118 << getAlignment() << "/0x" << Twine::utohexstr(Flags);
119
120 OS << ")";
121 }
122
BinaryData(MCSymbol & Symbol,uint64_t Address,uint64_t Size,uint16_t Alignment,BinarySection & Section,unsigned Flags)123 BinaryData::BinaryData(MCSymbol &Symbol, uint64_t Address, uint64_t Size,
124 uint16_t Alignment, BinarySection &Section,
125 unsigned Flags)
126 : Section(&Section), Address(Address), Size(Size), Alignment(Alignment),
127 Flags(Flags), OutputSection(&Section), OutputOffset(getOffset()) {
128 Symbols.push_back(&Symbol);
129 }
130