mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Alessandro Di Federico
73 lines
2.6 KiB
C++
73 lines
2.6 KiB
C++
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include "revng/Model/Binary.h"
|
|
#include "revng/Model/VerifyHelper.h"
|
|
|
|
// NOTE: there's a really similar function for computing alignment in
|
|
// `lib/ABI/Definition.cpp`. It's better if two are kept in sync, so
|
|
// when modifying this function, please apply corresponding modifications
|
|
// to its little brother as well.
|
|
RecursiveCoroutine<std::optional<uint64_t>>
|
|
model::TypeDefinition::trySize(VerifyHelper &VH) const {
|
|
if (auto MaybeSize = VH.size(*this))
|
|
rc_return MaybeSize;
|
|
|
|
std::optional<uint64_t> Result = 0;
|
|
if (llvm::isa<model::CABIFunctionDefinition>(this)
|
|
|| llvm::isa<model::RawFunctionDefinition>(this)) {
|
|
rc_return 0;
|
|
|
|
} else if (auto *E = llvm::dyn_cast<model::EnumDefinition>(this)) {
|
|
Result = rc_recur E->UnderlyingType()->trySize(VH);
|
|
|
|
} else if (auto *T = llvm::dyn_cast<model::TypedefDefinition>(this)) {
|
|
// TODO: in case of a recursive type (not via a pointer), we'll get an
|
|
// assertion, instead of a clean failure
|
|
Result = rc_recur T->UnderlyingType()->trySize(VH);
|
|
|
|
} else if (auto *S = llvm::dyn_cast<model::StructDefinition>(this)) {
|
|
Result = S->Size();
|
|
|
|
} else if (auto *U = llvm::dyn_cast<model::UnionDefinition>(this)) {
|
|
for (const auto &Field : U->Fields()) {
|
|
if (auto Size = rc_recur Field.Type()->trySize(VH))
|
|
Result = std::max(Result.value(), Size.value());
|
|
else
|
|
rc_return std::nullopt;
|
|
}
|
|
|
|
} else {
|
|
revng_abort("Unsupported type definition kind.");
|
|
}
|
|
|
|
if (!Result.has_value())
|
|
rc_return std::nullopt;
|
|
|
|
rc_return VH.setSize(*this, Result.value());
|
|
}
|
|
|
|
std::optional<uint64_t> model::TypeDefinition::trySize() const {
|
|
model::VerifyHelper VH;
|
|
return trySize(VH);
|
|
}
|
|
|
|
using DefR = model::TypeDefinitionReference;
|
|
template DefR DefR::fromString<model::Binary>(model::Binary *Root,
|
|
llvm::StringRef Path);
|
|
template DefR DefR::fromString<const model::Binary>(const model::Binary *Root,
|
|
llvm::StringRef Path);
|
|
|
|
llvm::SmallVector<const model::Type *, 4> model::TypeDefinition::edges() const {
|
|
const auto *PtrCopy = this;
|
|
auto GetEdges = [](const auto &Upcasted) { return Upcasted.edges(); };
|
|
return upcast(PtrCopy, GetEdges, llvm::SmallVector<const model::Type *, 4>());
|
|
}
|
|
|
|
llvm::SmallVector<model::Type *, 4> model::TypeDefinition::edges() {
|
|
auto *PtrCopy = this;
|
|
auto GetEdges = [](auto &Upcasted) { return Upcasted.edges(); };
|
|
return upcast(PtrCopy, GetEdges, llvm::SmallVector<model::Type *, 4>());
|
|
}
|