mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
63 lines
2.0 KiB
C++
63 lines
2.0 KiB
C++
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include "revng/Model/Binary.h"
|
|
#include "revng/Pipeline/Pipe.h"
|
|
#include "revng/Pipeline/RegisterContainerFactory.h"
|
|
#include "revng/Pipeline/RegisterPipe.h"
|
|
#include "revng/Pipes/Kinds.h"
|
|
#include "revng/Pipes/ModelGlobal.h"
|
|
#include "revng/Yield/Function.h"
|
|
#include "revng/Yield/Pipes/YieldControlFlow.h"
|
|
#include "revng/Yield/SVG.h"
|
|
|
|
namespace revng::pipes {
|
|
|
|
void YieldControlFlow::run(pipeline::ExecutionContext &Context,
|
|
const FunctionAssemblyStringMap &Input,
|
|
FunctionControlFlowStringMap &Output) {
|
|
// Access the model
|
|
const auto &Model = revng::getModelFromContext(Context);
|
|
ptml::MarkupBuilder B;
|
|
|
|
for (const model::Function &Function :
|
|
getFunctionsAndCommit(Context, Output.name())) {
|
|
MetaAddress Address = Function.Entry();
|
|
llvm::StringRef YamlText = Input.at(Address);
|
|
auto MaybeFunction = TupleTree<yield::Function>::fromString(YamlText);
|
|
|
|
revng_assert(MaybeFunction && MaybeFunction->verify());
|
|
revng_assert((*MaybeFunction)->Entry() == Address);
|
|
|
|
Output.insert_or_assign((*MaybeFunction)->Entry(),
|
|
yield::svg::controlFlowGraph(B,
|
|
**MaybeFunction,
|
|
*Model));
|
|
}
|
|
}
|
|
|
|
} // end namespace revng::pipes
|
|
|
|
using namespace revng::pipes;
|
|
using namespace pipeline;
|
|
static RegisterDefaultConstructibleContainer<FunctionControlFlowStringMap>
|
|
GraphContainer;
|
|
|
|
static pipeline::RegisterPipe<revng::pipes::YieldControlFlow> CFGPipe;
|
|
|
|
namespace revng::pypeline::piperuns {
|
|
|
|
void YieldCFG::runOnFunction(const model::Function &Function) {
|
|
ObjectID Object(Function.Entry());
|
|
auto YieldFunction = Input.getElement(Object);
|
|
|
|
revng_assert(YieldFunction->verify());
|
|
revng_assert(YieldFunction->Entry() == Function.Entry());
|
|
|
|
auto OS = Output.getOStream(Object);
|
|
*OS << yield::svg::controlFlowGraph(B, *YieldFunction, Binary);
|
|
}
|
|
|
|
} // namespace revng::pypeline::piperuns
|