#pragma once
#include "AuxiliaryBarrier.h"
#include <type_traits>
#include <wtf/CagedPtr.h>
namespace WTF {
template<typename Poison, typename T> struct PoisonedPtrTraits;
}
namespace JSC {
class JSCell;
class VM;
template<Gigacage::Kind passedKind, typename T, typename PtrTraits = WTF::DumbPtrTraits<T>>
class CagedBarrierPtr {
public:
static constexpr Gigacage::Kind kind = passedKind;
typedef T Type;
CagedBarrierPtr() { }
template<typename U>
CagedBarrierPtr(VM& vm, JSCell* cell, U&& value)
{
m_barrier.set(vm, cell, std::forward<U>(value));
}
void clear() { m_barrier.clear(); }
template<typename U>
void set(VM& vm, JSCell* cell, U&& value)
{
m_barrier.set(vm, cell, std::forward<U>(value));
}
T* get() const { return m_barrier.get().get(); }
T* getMayBeNull() const { return m_barrier.get().getMayBeNull(); }
bool operator==(const CagedBarrierPtr& other) const
{
return getMayBeNull() == other.getMayBeNull();
}
bool operator!=(const CagedBarrierPtr& other) const
{
return !(*this == other);
}
explicit operator bool() const
{
return *this != CagedBarrierPtr();
}
template<typename U>
void setWithoutBarrier(U&& value) { m_barrier.setWithoutBarrier(std::forward<U>(value)); }
T& operator*() const { return *get(); }
T* operator->() const { return get(); }
template<typename IndexType>
T& operator[](IndexType index) const { return get()[index]; }
private:
AuxiliaryBarrier<CagedPtr<kind, T, PtrTraits>> m_barrier;
};
template<Gigacage::Kind passedKind, typename PtrTraits>
class CagedBarrierPtr<passedKind, void, PtrTraits> {
public:
static constexpr Gigacage::Kind kind = passedKind;
typedef void Type;
CagedBarrierPtr() { }
template<typename U>
CagedBarrierPtr(VM& vm, JSCell* cell, U&& value)
{
m_barrier.set(vm, cell, std::forward<U>(value));
}
void clear() { m_barrier.clear(); }
template<typename U>
void set(VM& vm, JSCell* cell, U&& value)
{
m_barrier.set(vm, cell, std::forward<U>(value));
}
void* get() const { return m_barrier.get().get(); }
void* getMayBeNull() const { return m_barrier.get().getMayBeNull(); }
bool operator==(const CagedBarrierPtr& other) const
{
return getMayBeNull() == other.getMayBeNull();
}
bool operator!=(const CagedBarrierPtr& other) const
{
return !(*this == other);
}
explicit operator bool() const
{
return *this != CagedBarrierPtr();
}
template<typename U>
void setWithoutBarrier(U&& value) { m_barrier.setWithoutBarrier(std::forward<U>(value)); }
private:
AuxiliaryBarrier<CagedPtr<kind, void, PtrTraits>> m_barrier;
};
template<typename Poison, Gigacage::Kind passedKind, typename T>
using PoisonedCagedBarrierPtr = CagedBarrierPtr<passedKind, T, WTF::PoisonedPtrTraits<Poison, T>>;
}