#include "config.h"
#include "WeakBlock.h"
#include "Heap.h"
#include "HeapRootVisitor.h"
#include "JSObject.h"
#include "ScopeChain.h"
#include "Structure.h"
namespace JSC {
WeakBlock* WeakBlock::create()
{
PageAllocation allocation = PageAllocation::allocate(blockSize, OSAllocator::JSGCHeapPages);
if (!static_cast<bool>(allocation))
CRASH();
return new (NotNull, allocation.base()) WeakBlock(allocation);
}
void WeakBlock::destroy(WeakBlock* block)
{
block->m_allocation.deallocate();
}
WeakBlock::WeakBlock(PageAllocation& allocation)
: m_allocation(allocation)
{
for (size_t i = 0; i < weakImplCount(); ++i) {
WeakImpl* weakImpl = &weakImpls()[i];
new (NotNull, weakImpl) WeakImpl;
addToFreeList(&m_sweepResult.freeList, weakImpl);
}
ASSERT(!m_sweepResult.isNull() && m_sweepResult.blockIsFree);
}
void WeakBlock::finalizeAll()
{
for (size_t i = 0; i < weakImplCount(); ++i) {
WeakImpl* weakImpl = &weakImpls()[i];
if (weakImpl->state() >= WeakImpl::Finalized)
continue;
weakImpl->setState(WeakImpl::Dead);
finalize(weakImpl);
}
}
void WeakBlock::sweep()
{
if (!m_sweepResult.isNull())
return;
SweepResult sweepResult;
for (size_t i = 0; i < weakImplCount(); ++i) {
WeakImpl* weakImpl = &weakImpls()[i];
if (weakImpl->state() == WeakImpl::Dead)
finalize(weakImpl);
if (weakImpl->state() == WeakImpl::Deallocated)
addToFreeList(&sweepResult.freeList, weakImpl);
else
sweepResult.blockIsFree = false;
}
m_sweepResult = sweepResult;
ASSERT(!m_sweepResult.isNull());
}
void WeakBlock::visitLiveWeakImpls(HeapRootVisitor& heapRootVisitor)
{
if (!m_sweepResult.isNull() && m_sweepResult.blockIsFree)
return;
SlotVisitor& visitor = heapRootVisitor.visitor();
for (size_t i = 0; i < weakImplCount(); ++i) {
WeakImpl* weakImpl = &weakImpls()[i];
if (weakImpl->state() != WeakImpl::Live)
continue;
const JSValue& jsValue = weakImpl->jsValue();
if (Heap::isMarked(jsValue.asCell()))
continue;
WeakHandleOwner* weakHandleOwner = weakImpl->weakHandleOwner();
if (!weakHandleOwner)
continue;
if (!weakHandleOwner->isReachableFromOpaqueRoots(Handle<Unknown>::wrapSlot(&const_cast<JSValue&>(jsValue)), weakImpl->context(), visitor))
continue;
heapRootVisitor.visit(&const_cast<JSValue&>(jsValue));
}
}
void WeakBlock::visitDeadWeakImpls(HeapRootVisitor&)
{
if (!m_sweepResult.isNull() && m_sweepResult.blockIsFree)
return;
for (size_t i = 0; i < weakImplCount(); ++i) {
WeakImpl* weakImpl = &weakImpls()[i];
if (weakImpl->state() > WeakImpl::Dead)
continue;
if (Heap::isMarked(weakImpl->jsValue().asCell())) {
ASSERT(weakImpl->state() == WeakImpl::Live);
continue;
}
weakImpl->setState(WeakImpl::Dead);
}
}
}