mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
100 lines
2.9 KiB
C++
100 lines
2.9 KiB
C++
/// \file CompileModule.cpp
|
|
/// \brief The isolated kind is used to rappresent isolated root and functions.
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include <optional>
|
|
#include <string>
|
|
|
|
#include "llvm/ADT/STLExtras.h"
|
|
|
|
#include "revng/Model/Binary.h"
|
|
#include "revng/Model/IRHelpers.h"
|
|
#include "revng/Pipeline/AllRegistries.h"
|
|
#include "revng/Pipeline/Context.h"
|
|
#include "revng/Pipeline/PathComponent.h"
|
|
#include "revng/Pipeline/Target.h"
|
|
#include "revng/Pipes/Kinds.h"
|
|
#include "revng/Pipes/ModelGlobal.h"
|
|
#include "revng/Pipes/RootKind.h"
|
|
#include "revng/Pipes/TaggedFunctionKind.h"
|
|
#include "revng/Support/Assert.h"
|
|
#include "revng/Support/FunctionTags.h"
|
|
#include "revng/Support/MetaAddress.h"
|
|
|
|
using namespace pipeline;
|
|
using namespace ::revng::pipes;
|
|
using namespace llvm;
|
|
|
|
pipeline::TargetsList
|
|
TaggedFunctionKind::compactTargets(const pipeline::Context &Ctx,
|
|
pipeline::TargetsList::List &Targets) const {
|
|
|
|
const model::Binary &Model = *getModelFromContext(Ctx);
|
|
|
|
if (Model.Functions.size() == 0) {
|
|
return Targets;
|
|
}
|
|
|
|
std::set<std::string> Set;
|
|
const pipeline::Target AllFunctions({ pipeline::PathComponent("root"),
|
|
pipeline::PathComponent::all() },
|
|
*this);
|
|
|
|
// enumerate the targets
|
|
for (const pipeline::Target &Target : Targets) {
|
|
// if we see a * then no need to check if we have all functions, just
|
|
// return that.
|
|
if (Target.getPathComponents().back().isAll())
|
|
return pipeline::TargetsList({ AllFunctions });
|
|
|
|
Set.insert(Target.getPathComponents().back().getName());
|
|
}
|
|
|
|
// check if all functions in the model are in the targets
|
|
// if they are, return *
|
|
if (llvm::all_of(Model.Functions, [&Set](const model::Function &F) {
|
|
return Set.contains(F.Entry.toString());
|
|
})) {
|
|
|
|
return pipeline::TargetsList({ AllFunctions });
|
|
}
|
|
|
|
return Targets;
|
|
}
|
|
|
|
void TaggedFunctionKind::expandTarget(const Context &Ctx,
|
|
const Target &Input,
|
|
TargetsList &Output) const {
|
|
if (not Input.getPathComponents().back().isAll()) {
|
|
Output.push_back(Input);
|
|
return;
|
|
}
|
|
|
|
const model::Binary &Model = *getModelFromContext(Ctx);
|
|
if (Model.Functions.empty()) {
|
|
Output.push_back(Input);
|
|
return;
|
|
}
|
|
|
|
for (const auto &Function : Model.Functions) {
|
|
|
|
Target ToInsert({ Input.getPathComponents().front(),
|
|
PathComponent(Function.Entry.toString()) },
|
|
*this);
|
|
Output.emplace_back(std::move(ToInsert));
|
|
}
|
|
}
|
|
|
|
std::optional<pipeline::Target>
|
|
TaggedFunctionKind::symbolToTarget(const llvm::Function &Symbol) const {
|
|
if (not Tag->isTagOf(&Symbol) or Symbol.isDeclaration())
|
|
return std::nullopt;
|
|
|
|
auto Address = getMetaAddressOfIsolatedFunction(Symbol);
|
|
revng_assert(Address.isValid());
|
|
return pipeline::Target({ "root", Address.toString() }, *this);
|
|
}
|