mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
ab125b35b0
Change company name to "rev.ng Labs Srl" in all license headers to reflect changed company name and legal status Add missing license headers to files that didn't have one
69 lines
1.7 KiB
C++
69 lines
1.7 KiB
C++
#pragma once
|
|
|
|
//
|
|
// Copyright (c) rev.ng Labs Srl. See LICENSE.md for details.
|
|
//
|
|
|
|
#include <variant>
|
|
|
|
#include "llvm/IR/Argument.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/Instructions.h"
|
|
|
|
#include "revng/Model/Binary.h"
|
|
#include "revng/Model/Type.h"
|
|
|
|
#include "revng-c/DataLayoutAnalysis/DLATypeSystem.h"
|
|
|
|
namespace dla {
|
|
|
|
// Some APIs to make Function and CallInst homogeneous
|
|
const inline llvm::Type *getRetType(const llvm::Function *F) {
|
|
return F->getReturnType();
|
|
}
|
|
const inline llvm::Type *getRetType(const llvm::CallInst *C) {
|
|
return C->getType();
|
|
}
|
|
|
|
inline auto getArgs(const llvm::Function *F) {
|
|
return F->args();
|
|
}
|
|
inline auto getArgs(const llvm::CallInst *C) {
|
|
return C->arg_operands();
|
|
}
|
|
|
|
inline unsigned long arg_size(const llvm::Function *F) {
|
|
return F->arg_size();
|
|
}
|
|
inline unsigned long arg_size(const llvm::CallInst *C) {
|
|
return llvm::size(C->arg_operands());
|
|
}
|
|
|
|
/// Wrapper around a variant between Function and CallInst
|
|
class FuncOrCallInst {
|
|
private:
|
|
std::variant<const llvm::Function *, const llvm::CallInst *> Val;
|
|
|
|
public:
|
|
FuncOrCallInst() : Val((llvm::Function *) nullptr){};
|
|
FuncOrCallInst(const llvm::Function *F) : Val(F){};
|
|
FuncOrCallInst(const llvm::CallInst *F) : Val(F){};
|
|
FuncOrCallInst(const FuncOrCallInst &F) : Val(F.Val){};
|
|
|
|
~FuncOrCallInst() = default;
|
|
FuncOrCallInst(FuncOrCallInst &&) = default;
|
|
FuncOrCallInst &operator=(const FuncOrCallInst &) = default;
|
|
FuncOrCallInst &operator=(FuncOrCallInst &&) = default;
|
|
|
|
bool isNull() const;
|
|
|
|
const llvm::Value *getVal() const;
|
|
const llvm::Type *getRetType() const;
|
|
|
|
/// Get the \a Idx-th argument
|
|
const llvm::Value *getArg(unsigned Idx) const;
|
|
unsigned long arg_size() const;
|
|
};
|
|
|
|
} // namespace dla
|