#ifndef StackBounds_h
#define StackBounds_h
#include <algorithm>
#include <wtf/ThreadingPrimitives.h>
namespace WTF {
class StackBounds {
const static size_t s_defaultAvailabilityDelta = 64 * 1024;
public:
enum class StackDirection { Upward, Downward };
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
{
if (isGrowingDownward())
return static_cast<char*>(m_origin) - static_cast<char*>(m_bound);
return static_cast<char*>(m_bound) - static_cast<char*>(m_origin);
}
bool isEmpty() const { return !m_origin; }
bool contains(void* p) const
{
if (isEmpty())
return false;
if (isGrowingDownward())
return (m_origin >= p) && (p > m_bound);
return (m_bound > p) && (p >= m_origin);
}
void* recursionLimit(size_t minAvailableDelta = s_defaultAvailabilityDelta) const
{
checkConsistency();
if (isGrowingDownward())
return static_cast<char*>(m_bound) + minAvailableDelta;
return static_cast<char*>(m_bound) - minAvailableDelta;
}
void* recursionLimit(char* startOfUserStack, size_t maxUserStack, size_t reservedZoneSize) const
{
checkConsistency();
if (maxUserStack < reservedZoneSize)
reservedZoneSize = maxUserStack;
size_t maxUserStackWithReservedZone = maxUserStack - reservedZoneSize;
if (isGrowingDownward()) {
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;
}
char* endOfStackWithReservedZone = reinterpret_cast<char*>(m_bound) - reservedZoneSize;
if (startOfUserStack > endOfStackWithReservedZone)
return endOfStackWithReservedZone;
size_t availableUserStack = endOfStackWithReservedZone - startOfUserStack;
if (maxUserStackWithReservedZone > availableUserStack)
maxUserStackWithReservedZone = availableUserStack;
return startOfUserStack + maxUserStackWithReservedZone;
}
bool isGrowingDownward() const
{
ASSERT(m_origin && m_bound);
return m_bound <= m_origin;
}
private:
StackBounds(void* origin, void* end)
: m_origin(origin)
, m_bound(end)
{
}
constexpr StackBounds()
: m_origin(nullptr)
, m_bound(nullptr)
{
}
static StackDirection stackDirection();
WTF_EXPORT_PRIVATE static StackBounds currentThreadStackBoundsInternal();
void checkConsistency() const
{
#if !ASSERT_DISABLED
void* currentPosition = ¤tPosition;
ASSERT(m_origin != m_bound);
ASSERT(isGrowingDownward()
? (currentPosition < m_origin && currentPosition > m_bound)
: (currentPosition > m_origin && currentPosition < m_bound));
#endif
}
void* m_origin;
void* m_bound;
friend class StackStats;
};
}
using WTF::StackBounds;
#endif