mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Alessandro Di Federico
9fa311bd1b
Attach a `func.entry` metadata node to the entry basic block of a function and rename `func` to `func.member.of`.
47 lines
1.0 KiB
C++
47 lines
1.0 KiB
C++
#ifndef _FUNCTIONBOUNDARIESDETECTION_H
|
|
#define _FUNCTIONBOUNDARIESDETECTION_H
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
// Standard includes
|
|
#include <map>
|
|
#include <string>
|
|
|
|
// LLVM includes
|
|
#include "llvm/Pass.h"
|
|
|
|
namespace llvm {
|
|
class BasicBlock;
|
|
}
|
|
|
|
class JumpTargetManager;
|
|
|
|
class FunctionBoundariesDetectionPass : public llvm::FunctionPass {
|
|
public:
|
|
static char ID;
|
|
|
|
public:
|
|
FunctionBoundariesDetectionPass() : llvm::FunctionPass(ID), JTM(nullptr) { }
|
|
FunctionBoundariesDetectionPass(JumpTargetManager *JTM,
|
|
std::string SerializePath) :
|
|
llvm::FunctionPass(ID), JTM(JTM), SerializePath(SerializePath) { }
|
|
|
|
void getAnalysisUsage(llvm::AnalysisUsage &AU) const override {
|
|
AU.setPreservesAll();
|
|
}
|
|
|
|
bool runOnFunction(llvm::Function &F) override;
|
|
|
|
private:
|
|
void serialize() const;
|
|
|
|
private:
|
|
JumpTargetManager *JTM;
|
|
std::string SerializePath;
|
|
std::map<llvm::BasicBlock *, std::vector<llvm::BasicBlock *>> Functions;
|
|
};
|
|
|
|
#endif // _FUNCTIONBOUNDARIESDETECTION_H
|