llvm-project/bolt/lib/Core/BinaryData.cpp
Rafael Auler a34c753fe7 Rebase: [NFC] Refactor sources to be buildable in shared mode
Summary:
Moves source files into separate components, and make explicit
component dependency on each other, so LLVM build system knows how to
build BOLT in BUILD_SHARED_LIBS=ON.

Please use the -c merge.renamelimit=230 git option when rebasing your
work on top of this change.

To achieve this, we create a new library to hold core IR files (most
classes beginning with Binary in their names), a new library to hold
Utils, some command line options shared across both RewriteInstance
and core IR files, a new library called Rewrite to hold most classes
concerned with running top-level functions coordinating the binary
rewriting process, and a new library called Profile to hold classes
dealing with profile reading and writing.

To remove the dependency from BinaryContext into X86-specific classes,
we do some refactoring on the BinaryContext constructor to receive a
reference to the specific backend directly from RewriteInstance. Then,
the dependency on X86 or AArch64-specific classes is transfered to the
Rewrite library. We can't have the Core library depend on targets
because targets depend on Core (which would create a cycle).

Files implementing the entry point of a tool are transferred to the
tools/ folder. All header files are transferred to the include/
folder. The src/ folder was renamed to lib/.

(cherry picked from FBD32746834)
2021-10-08 11:47:10 -07:00

160 lines
4.0 KiB
C++

//===--- BinaryData.cpp - Representation of section data objects ----------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
//===----------------------------------------------------------------------===//
#include "bolt/Core/BinaryData.h"
#include "bolt/Core/BinarySection.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/Regex.h"
using namespace llvm;
using namespace bolt;
#undef DEBUG_TYPE
#define DEBUG_TYPE "bolt"
namespace opts {
extern cl::OptionCategory BoltCategory;
extern cl::opt<unsigned> Verbosity;
cl::opt<bool>
PrintSymbolAliases("print-aliases",
cl::desc("print aliases when printing objects"),
cl::Hidden,
cl::ZeroOrMore,
cl::cat(BoltCategory));
}
bool BinaryData::isAbsolute() const {
return Flags & SymbolRef::SF_Absolute;
}
bool BinaryData::isMoveable() const {
return (!isAbsolute() &&
(IsMoveable &&
(!Parent || isTopLevelJumpTable())));
}
void BinaryData::merge(const BinaryData *Other) {
assert(!Size || !Other->Size || Size == Other->Size);
assert(Address == Other->Address);
assert(*Section == *Other->Section);
assert(OutputOffset == Other->OutputOffset);
assert(OutputSection == Other->OutputSection);
Symbols.insert(Symbols.end(), Other->Symbols.begin(), Other->Symbols.end());
Flags |= Other->Flags;
if (!Size)
Size = Other->Size;
}
bool BinaryData::hasName(StringRef Name) const {
for (const MCSymbol *Symbol : Symbols) {
if (Name == Symbol->getName())
return true;
}
return false;
}
bool BinaryData::hasNameRegex(StringRef NameRegex) const {
Regex MatchName(NameRegex);
for (const MCSymbol *Symbol : Symbols) {
if (MatchName.match(Symbol->getName()))
return true;
}
return false;
}
bool BinaryData::nameStartsWith(StringRef Prefix) const {
for (const MCSymbol *Symbol : Symbols) {
if (Symbol->getName().startswith(Prefix))
return true;
}
return false;
}
StringRef BinaryData::getSectionName() const {
return getSection().getName();
}
StringRef BinaryData::getOutputSectionName() const {
return getOutputSection().getName();
}
uint64_t BinaryData::getOutputAddress() const {
assert(OutputSection->getOutputAddress());
return OutputSection->getOutputAddress() + OutputOffset;
}
uint64_t BinaryData::getOffset() const {
return Address - getSection().getAddress();
}
void BinaryData::setSection(BinarySection &NewSection) {
if (OutputSection == Section)
OutputSection = &NewSection;
Section = &NewSection;
}
bool BinaryData::isMoved() const {
return (getOffset() != OutputOffset || OutputSection != Section);
}
void BinaryData::print(raw_ostream &OS) const {
printBrief(OS);
}
void BinaryData::printBrief(raw_ostream &OS) const {
OS << "(";
if (isJumpTable())
OS << "jump-table: ";
else
OS << "object: ";
OS << getName();
if ((opts::PrintSymbolAliases || opts::Verbosity > 1) && Symbols.size() > 1) {
OS << ", aliases:";
for (unsigned I = 1u; I < Symbols.size(); ++I) {
OS << (I == 1 ? " (" : ", ") << Symbols[I]->getName();
}
OS << ")";
}
if (Parent) {
OS << " (parent: ";
Parent->printBrief(OS);
OS << ")";
}
OS << ", 0x" << Twine::utohexstr(getAddress())
<< ":0x" << Twine::utohexstr(getEndAddress())
<< "/" << getSize() << "/" << getAlignment()
<< "/0x" << Twine::utohexstr(Flags);
OS << ")";
}
BinaryData::BinaryData(MCSymbol &Symbol,
uint64_t Address,
uint64_t Size,
uint16_t Alignment,
BinarySection &Section,
unsigned Flags)
: Section(&Section),
Address(Address),
Size(Size),
Alignment(Alignment),
Flags(Flags),
OutputSection(&Section),
OutputOffset(getOffset())
{
Symbols.push_back(&Symbol);
}