admin/revng-revng
1
0
Fork 0
mirror of https://github.com/revng/revng synced 2026-06-21 14:07:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

GeneratedCodeBasicInfo: parse root lazily

Browse Source 
This commit makes sure that GeneratedCodeBasicInfo can be used even in
absence of the `root`.

This also ensure that no time is wasted on brief/focused pipelines that
do not care about analyzing each basic block in the `root` function.
This commit is contained in:
Alessandro Di Federico
2021-11-29 14:55:23 +01:00
parent e113cb579e
commit a2364bc6a0
2 changed files with 58 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/BasicAnalyses/GeneratedCodeBasicInfo.cpp
+28 -21
View File
@@ -26,16 +26,13 @@ using RegisterGCBI = RegisterPass<GeneratedCodeBasicInfoWrapperPass>;
static RegisterGCBI X("gcbi", "Generated Code Basic Info", true, true);
void GeneratedCodeBasicInfo::run(Module &M) {
Function &F = *M.getFunction("root");
RootFunction = M.getFunction("root");
NewPC = M.getFunction("newpc");
if (NewPC != nullptr) {
if (NewPC != nullptr)
MetaAddressStruct = cast<StructType>(NewPC->arg_begin()->getType());
}
revng_log(PassesLog, "Starting GeneratedCodeBasicInfo");
RootFunction = &F;
const char *MDName = "revng.input.architecture";
NamedMDNode *InputArchMD = M.getOrInsertNamedMetadata(MDName);
auto *Tuple = dyn_cast<MDTuple>(InputArchMD->getOperand(0));
@@ -65,7 +62,26 @@ void GeneratedCodeBasicInfo::run(Module &M) {
Type *PCType = PC->getType()->getPointerElementType();
PCRegSize = M.getDataLayout().getTypeAllocSize(PCType);
for (BasicBlock &BB : F) {
if (auto *NamedMD = M.getNamedMetadata("revng.csv")) {
auto *Tuple = cast<MDTuple>(NamedMD->getOperand(0));
for (const MDOperand &Operand : Tuple->operands()) {
if (Operand.get() == nullptr)
continue;
auto *CSV = cast<GlobalVariable>(QMD.extract<Constant *>(Operand.get()));
CSVs.push_back(CSV);
}
}
revng_log(PassesLog, "Ending GeneratedCodeBasicInfo");
}
void GeneratedCodeBasicInfo::parseRoot() {
if (RootParsed)
return;
RootParsed = true;
for (BasicBlock &BB : *RootFunction) {
if (!BB.empty()) {
switch (getType(&BB)) {
case BlockType::RootDispatcherBlock:
@@ -90,7 +106,7 @@ void GeneratedCodeBasicInfo::run(Module &M) {
case BlockType::JumpTargetBlock: {
auto *Call = cast<CallInst>(&*BB.begin());
revng_assert(Call->getCalledFunction()->getName() == "newpc");
revng_assert(Call->getCalledFunction() == NewPC);
JumpTargets[MetaAddress::fromConstant(Call->getArgOperand(0))] = &BB;
break;
}
@@ -104,23 +120,14 @@ void GeneratedCodeBasicInfo::run(Module &M) {
}
}
}
if (auto *NamedMD = M.getNamedMetadata("revng.csv")) {
auto *Tuple = cast<MDTuple>(NamedMD->getOperand(0));
for (const MDOperand &Operand : Tuple->operands()) {
if (Operand.get() == nullptr)
continue;
auto *CSV = cast<GlobalVariable>(QMD.extract<Constant *>(Operand.get()));
CSVs.push_back(CSV);
}
}
revng_log(PassesLog, "Ending GeneratedCodeBasicInfo");
}
GeneratedCodeBasicInfo::SuccessorsList
GeneratedCodeBasicInfo::getSuccessors(BasicBlock *BB) const {
GeneratedCodeBasicInfo::getSuccessors(BasicBlock *BB) {
parseRoot();
revng_assert(BB->getParent() == RootFunction);
SuccessorsList Result;
df_iterator_default_set<BasicBlock *> Visited;