#pragma once
#include <algorithm>
#include <wtf/StackPointer.h>
#include <wtf/ThreadingPrimitives.h>
namespace WTF {
class StackBounds {
WTF_MAKE_FAST_ALLOCATED;
public:
static constexpr size_t DefaultReservedZone = 64 * 1024;
static constexpr StackBounds emptyBounds() { return StackBounds(); }
#if HAVE(STACK_BOUNDS_FOR_NEW_THREAD)
static StackBounds newThreadStackBounds(PlatformThreadHandle);
#endif
static StackBounds currentThreadStackBounds()
{
auto result = currentThreadStackBoundsInternal();
result.checkConsistency();
return result;
}
void* origin() const
{
ASSERT(m_origin);
return m_origin;
}
void* end() const
{
ASSERT(m_bound);
return m_bound;
}
size_t size() const
{
return static_cast<char*>(m_origin) - static_cast<char*>(m_bound);
}
bool isEmpty() const { return !m_origin; }
bool contains(void* p) const
{
if (isEmpty())
return false;
return (m_origin >= p) && (p > m_bound);
}
void* recursionLimit(size_t minReservedZone = DefaultReservedZone) const
{
checkConsistency();
return static_cast<char*>(m_bound) + minReservedZone;
}
void* recursionLimit(char* startOfUserStack, size_t maxUserStack, size_t reservedZoneSize) const
{
checkConsistency();
if (maxUserStack < reservedZoneSize)
reservedZoneSize = maxUserStack;
size_t maxUserStackWithReservedZone = maxUserStack - reservedZoneSize;
char* endOfStackWithReservedZone = reinterpret_cast<char*>(m_bound) + reservedZoneSize;
if (startOfUserStack < endOfStackWithReservedZone)
return endOfStackWithReservedZone;
size_t availableUserStack = startOfUserStack - endOfStackWithReservedZone;
if (maxUserStackWithReservedZone > availableUserStack)
maxUserStackWithReservedZone = availableUserStack;
return startOfUserStack - maxUserStackWithReservedZone;
}
StackBounds withSoftOrigin(void* origin) const
{
ASSERT(contains(origin));
return StackBounds(origin, m_bound);
}
private:
StackBounds(void* origin, void* end)
: m_origin(origin)
, m_bound(end)
{
ASSERT(isGrowingDownwards());
}
constexpr StackBounds()
: m_origin(nullptr)
, m_bound(nullptr)
{
}
inline bool isGrowingDownwards() const
{
ASSERT(m_origin && m_bound);
return m_bound <= m_origin;
}
WTF_EXPORT_PRIVATE static StackBounds currentThreadStackBoundsInternal();
void checkConsistency() const
{
#if ASSERT_ENABLED
void* currentPosition = currentStackPointer();
ASSERT(m_origin != m_bound);
ASSERT(currentPosition < m_origin && currentPosition > m_bound);
#endif
}
void* m_origin;
void* m_bound;
friend class StackStats;
};
}
using WTF::StackBounds;