llvm-project/clang/lib/AST/ByteCode/FunctionPointer.h
Timm Baeder db94852b9b
[clang][bytecode] Allow adding offsets to function pointers (#105641)
Convert them to Pointers, do the offset calculation and then convert
them back to function pointers.
2024-08-22 15:23:50 +02:00

77 lines
2.1 KiB
C++

//===--- FunctionPointer.h - Types for the constexpr VM ---------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_CLANG_AST_INTERP_FUNCTION_POINTER_H
#define LLVM_CLANG_AST_INTERP_FUNCTION_POINTER_H
#include "Function.h"
#include "Primitives.h"
namespace clang {
class ASTContext;
class APValue;
namespace interp {
class FunctionPointer final {
private:
const Function *Func;
uint64_t Offset;
bool Valid;
public:
FunctionPointer() = default;
FunctionPointer(const Function *Func, uint64_t Offset = 0)
: Func(Func), Offset(Offset), Valid(true) {}
FunctionPointer(uintptr_t IntVal, const Descriptor *Desc = nullptr)
: Func(reinterpret_cast<const Function *>(IntVal)), Offset(0),
Valid(false) {}
const Function *getFunction() const { return Func; }
uint64_t getOffset() const { return Offset; }
bool isZero() const { return !Func; }
bool isValid() const { return Valid; }
bool isWeak() const {
if (!Func || !Valid || !Func->getDecl())
return false;
return Func->getDecl()->isWeak();
}
APValue toAPValue(const ASTContext &) const;
void print(llvm::raw_ostream &OS) const;
std::string toDiagnosticString(const ASTContext &Ctx) const {
if (!Func)
return "nullptr";
return toAPValue(Ctx).getAsString(Ctx, Func->getDecl()->getType());
}
uint64_t getIntegerRepresentation() const {
return static_cast<uint64_t>(reinterpret_cast<uintptr_t>(Func));
}
ComparisonCategoryResult compare(const FunctionPointer &RHS) const {
if (Func == RHS.Func && Offset == RHS.Offset)
return ComparisonCategoryResult::Equal;
return ComparisonCategoryResult::Unordered;
}
};
inline llvm::raw_ostream &operator<<(llvm::raw_ostream &OS,
FunctionPointer FP) {
FP.print(OS);
return OS;
}
} // namespace interp
} // namespace clang
#endif