mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Alessandro Di Federico
4150afc33f
Split createReturn into a reusable createCall that emits the struct-initializer call without wrapping it in a return. An EmitBody flag lets callers create the declaration but skip the body, useful when the body would later be stripped.
64 lines
2.4 KiB
C++
64 lines
2.4 KiB
C++
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include "revng/FunctionIsolation/StructInitializers.h"
|
|
#include "revng/Model/FunctionTags.h"
|
|
|
|
using namespace llvm;
|
|
|
|
const char *StructInitializerPrefix = "struct_initializer";
|
|
|
|
StructInitializers::StructInitializers(llvm::Module *M, bool EmitBody) :
|
|
Pool(M, false), Context(M->getContext()), EmitBody(EmitBody) {
|
|
Pool.setMemoryEffects(MemoryEffects::none());
|
|
Pool.addFnAttribute(Attribute::NoUnwind);
|
|
Pool.addFnAttribute(Attribute::WillReturn);
|
|
Pool.setTags({ &FunctionTags::StructInitializer,
|
|
&FunctionTags::UniquedByPrototype });
|
|
|
|
// Record existing initializers
|
|
Pool.initializeFromReturnType(FunctionTags::StructInitializer);
|
|
}
|
|
|
|
CallInst *StructInitializers::createCall(revng::IRBuilder &Builder,
|
|
StructType *ReturnType,
|
|
ArrayRef<Value *> Values) {
|
|
SmallVector<Type *, 8> Types;
|
|
llvm::copy(ReturnType->elements(), std::back_inserter(Types));
|
|
|
|
// Create struct_initializer
|
|
Function *Initializer = Pool.get(ReturnType,
|
|
ReturnType,
|
|
Types,
|
|
StructInitializerPrefix);
|
|
|
|
// Lazily populate its body, unless the caller opted out (e.g., the body
|
|
// would otherwise persist past `remove-lifting-artifacts` and confuse later
|
|
// canonicalize passes that expect non-isolated functions to be declarations).
|
|
if (EmitBody and Initializer->isDeclaration()) {
|
|
auto *Entry = BasicBlock::Create(Context, "", Initializer);
|
|
|
|
// TODO: the checks should be enabled conditionally based on the user.
|
|
revng::NonDebugInfoCheckingIRBuilder InitializerBuilder(Entry);
|
|
|
|
SmallVector<Value *, 8> Arguments;
|
|
for (Argument &Arg : Initializer->args())
|
|
Arguments.push_back(&Arg);
|
|
|
|
InitializerBuilder.CreateAggregateRet(Arguments.data(), Arguments.size());
|
|
}
|
|
|
|
// Emit a call in the caller
|
|
return cast<CallInst>(Builder.CreateCall(Initializer, Values));
|
|
}
|
|
|
|
Instruction *StructInitializers::createReturn(revng::IRBuilder &Builder,
|
|
ArrayRef<Value *> Values) {
|
|
// Obtain return StructType
|
|
auto *FT = Builder.GetInsertBlock()->getParent()->getFunctionType();
|
|
auto *ReturnType = cast<StructType>(FT->getReturnType());
|
|
|
|
return Builder.CreateRet(createCall(Builder, ReturnType, Values));
|
|
}
|