mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-12-15 07:39:31 +00:00
eedbfb8aab
This patch builds on some existing code to do CFG reconstruction from a disassembled binary: - MCModule represents the binary, and has a list of MCAtoms. - MCAtom represents either disassembled instructions (MCTextAtom), or contiguous data (MCDataAtom), and covers a specific range of addresses. - MCBasicBlock and MCFunction form the reconstructed CFG. An MCBB is backed by an MCTextAtom, and has the usual successors/predecessors. - MCObjectDisassembler creates a module from an ObjectFile using a disassembler. It first builds an atom for each section. It can also construct the CFG, and this splits the text atoms into basic blocks. MCModule and MCAtom were only sketched out; MCFunction and MCBB were implemented under the experimental "-cfg" llvm-objdump -macho option. This cleans them up for further use; llvm-objdump -d -cfg now generates graphviz files for each function found in the binary. In the future, MCObjectDisassembler may be the right place to do "intelligent" disassembly: for example, handling constant islands is just a matter of splitting the atom, using information that may be available in the ObjectFile. Also, better initial atom formation than just using sections is possible using symbols (and things like Mach-O's function_starts load command). This brings two minor regressions in llvm-objdump -macho -cfg: - The printing of a relocation's referenced symbol. - An annotation on loop BBs, i.e., which are their own successor. Relocation printing is replaced by the MCSymbolizer; the basic CFG annotation will be superseded by more related functionality. llvm-svn: 182628
111 lines
3.2 KiB
C++
111 lines
3.2 KiB
C++
//===- lib/MC/MCAtom.cpp - MCAtom implementation --------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/MC/MCAtom.h"
|
|
#include "llvm/MC/MCModule.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include <iterator>
|
|
|
|
using namespace llvm;
|
|
|
|
void MCAtom::remap(uint64_t NewBegin, uint64_t NewEnd) {
|
|
Parent->remap(this, NewBegin, NewEnd);
|
|
}
|
|
|
|
void MCAtom::remapForTruncate(uint64_t TruncPt) {
|
|
assert((TruncPt >= Begin && TruncPt < End) &&
|
|
"Truncation point not contained in atom!");
|
|
remap(Begin, TruncPt);
|
|
}
|
|
|
|
void MCAtom::remapForSplit(uint64_t SplitPt,
|
|
uint64_t &LBegin, uint64_t &LEnd,
|
|
uint64_t &RBegin, uint64_t &REnd) {
|
|
assert((SplitPt > Begin && SplitPt <= End) &&
|
|
"Splitting at point not contained in atom!");
|
|
|
|
// Compute the new begin/end points.
|
|
LBegin = Begin;
|
|
LEnd = SplitPt - 1;
|
|
RBegin = SplitPt;
|
|
REnd = End;
|
|
|
|
// Remap this atom to become the lower of the two new ones.
|
|
remap(LBegin, LEnd);
|
|
}
|
|
|
|
// MCDataAtom
|
|
|
|
void MCDataAtom::addData(const MCData &D) {
|
|
Data.push_back(D);
|
|
if (Data.size() > Begin - End)
|
|
remap(Begin, End + 1);
|
|
}
|
|
|
|
void MCDataAtom::truncate(uint64_t TruncPt) {
|
|
remapForTruncate(TruncPt);
|
|
|
|
Data.resize(TruncPt - Begin + 1);
|
|
}
|
|
|
|
MCDataAtom *MCDataAtom::split(uint64_t SplitPt) {
|
|
uint64_t LBegin, LEnd, RBegin, REnd;
|
|
remapForSplit(SplitPt, LBegin, LEnd, RBegin, REnd);
|
|
|
|
MCDataAtom *RightAtom = Parent->createDataAtom(RBegin, REnd);
|
|
RightAtom->setName(getName());
|
|
|
|
std::vector<MCData>::iterator I = Data.begin() + (RBegin - LBegin);
|
|
assert(I != Data.end() && "Split point not found in range!");
|
|
|
|
std::copy(I, Data.end(), std::back_inserter(RightAtom->Data));
|
|
Data.erase(I, Data.end());
|
|
return RightAtom;
|
|
}
|
|
|
|
// MCTextAtom
|
|
|
|
void MCTextAtom::addInst(const MCInst &I, uint64_t Size) {
|
|
if (NextInstAddress > End)
|
|
remap(Begin, NextInstAddress);
|
|
Insts.push_back(MCDecodedInst(I, NextInstAddress, Size));
|
|
NextInstAddress += Size;
|
|
}
|
|
|
|
void MCTextAtom::truncate(uint64_t TruncPt) {
|
|
remapForTruncate(TruncPt);
|
|
|
|
InstListTy::iterator I = Insts.begin();
|
|
while (I != Insts.end() && I->Address <= TruncPt) ++I;
|
|
|
|
assert(I != Insts.end() && "Truncation point not found in disassembly!");
|
|
assert(I->Address == TruncPt + 1 &&
|
|
"Truncation point does not fall on instruction boundary");
|
|
|
|
Insts.erase(I, Insts.end());
|
|
}
|
|
|
|
MCTextAtom *MCTextAtom::split(uint64_t SplitPt) {
|
|
uint64_t LBegin, LEnd, RBegin, REnd;
|
|
remapForSplit(SplitPt, LBegin, LEnd, RBegin, REnd);
|
|
|
|
MCTextAtom *RightAtom = Parent->createTextAtom(RBegin, REnd);
|
|
RightAtom->setName(getName());
|
|
|
|
InstListTy::iterator I = Insts.begin();
|
|
while (I != Insts.end() && I->Address < SplitPt) ++I;
|
|
assert(I != Insts.end() && "Split point not found in disassembly!");
|
|
assert(I->Address == SplitPt &&
|
|
"Split point does not fall on instruction boundary!");
|
|
|
|
std::copy(I, Insts.end(), std::back_inserter(RightAtom->Insts));
|
|
Insts.erase(I, Insts.end());
|
|
return RightAtom;
|
|
}
|