#pragma once
#include <wtf/Gigacage.h>
#include <wtf/PtrTag.h>
#include <wtf/RawPtrTraits.h>
#include <climits>
namespace WTF {
constexpr bool tagCagedPtr = true;
template<Gigacage::Kind passedKind, typename T, bool shouldTag = false, typename PtrTraits = RawPtrTraits<T>>
class CagedPtr {
public:
static constexpr Gigacage::Kind kind = passedKind;
static constexpr unsigned numberOfPACBits = 25;
static constexpr uintptr_t nonPACBitsMask = (1ull << ((sizeof(T*) * CHAR_BIT) - numberOfPACBits)) - 1;
CagedPtr() : CagedPtr(nullptr) { }
CagedPtr(std::nullptr_t)
: m_ptr(shouldTag ? tagArrayPtr<T>(nullptr, 0) : nullptr)
{ }
CagedPtr(T* ptr, unsigned size)
: m_ptr(shouldTag ? tagArrayPtr(ptr, size) : ptr)
{ }
T* get(unsigned size) const
{
ASSERT(m_ptr);
T* ptr = PtrTraits::unwrap(m_ptr);
T* cagedPtr = Gigacage::caged(kind, ptr);
T* untaggedPtr = shouldTag ? untagArrayPtr(mergePointers(ptr, cagedPtr), size) : cagedPtr;
return untaggedPtr;
}
T* getMayBeNull(unsigned size) const
{
T* ptr = PtrTraits::unwrap(m_ptr);
if (!removeArrayPtrTag(ptr))
return nullptr;
T* cagedPtr = Gigacage::caged(kind, ptr);
T* untaggedPtr = shouldTag ? untagArrayPtr(mergePointers(ptr, cagedPtr), size) : cagedPtr;
return untaggedPtr;
}
T* getUnsafe() const
{
T* ptr = PtrTraits::unwrap(m_ptr);
ptr = shouldTag ? removeArrayPtrTag(ptr) : ptr;
return Gigacage::cagedMayBeNull(kind, ptr);
}
template<typename U = T>
typename std::enable_if<!std::is_same<void, U>::value, T>::type&
at(unsigned index, unsigned size) const { return get(size)[index]; }
void recage(unsigned oldSize, unsigned newSize)
{
auto ptr = get(oldSize);
ASSERT(ptr == getUnsafe());
*this = CagedPtr(ptr, newSize);
}
CagedPtr(CagedPtr& other)
: m_ptr(other.m_ptr)
{
}
CagedPtr& operator=(const CagedPtr& ptr)
{
m_ptr = ptr.m_ptr;
return *this;
}
CagedPtr(CagedPtr&& other)
: m_ptr(PtrTraits::exchange(other.m_ptr, nullptr))
{
}
CagedPtr& operator=(CagedPtr&& ptr)
{
m_ptr = PtrTraits::exchange(ptr.m_ptr, nullptr);
return *this;
}
bool operator==(const CagedPtr& other) const
{
bool result = m_ptr == other.m_ptr;
ASSERT(result == (getUnsafe() == other.getUnsafe()));
return result;
}
bool operator!=(const CagedPtr& other) const
{
return !(*this == other);
}
explicit operator bool() const
{
return getUnsafe() != nullptr;
}
T* rawBits() const
{
return bitwise_cast<T*>(m_ptr);
}
protected:
static inline T* mergePointers(T* sourcePtr, T* cagedPtr)
{
#if CPU(ARM64E)
return reinterpret_cast<T*>((reinterpret_cast<uintptr_t>(sourcePtr) & ~nonPACBitsMask) | (reinterpret_cast<uintptr_t>(cagedPtr) & nonPACBitsMask));
#else
UNUSED_PARAM(sourcePtr);
return cagedPtr;
#endif
}
typename PtrTraits::StorageType m_ptr;
};
}
using WTF::CagedPtr;
using WTF::tagCagedPtr;