#include "config.h"
#include "WeakBlock.h"
#include "CellContainerInlines.h"
#include "Heap.h"
#include "HeapAnalyzer.h"
#include "JSCInlines.h"
#include "JSObject.h"
#include "WeakHandleOwner.h"
namespace JSC {
DEFINE_ALLOCATOR_WITH_HEAP_IDENTIFIER(WeakBlock);
WeakBlock* WeakBlock::create(Heap& heap, CellContainer container)
{
heap.didAllocateBlock(WeakBlock::blockSize);
return new (NotNull, WeakBlockMalloc::malloc(blockSize)) WeakBlock(container);
}
void WeakBlock::destroy(Heap& heap, WeakBlock* block)
{
block->~WeakBlock();
WeakBlockMalloc::free(block);
heap.didFreeBlock(WeakBlock::blockSize);
}
WeakBlock::WeakBlock(CellContainer container)
: DoublyLinkedListNode<WeakBlock>()
, m_container(container)
{
for (size_t i = 0; i < weakImplCount(); ++i) {
WeakImpl* weakImpl = &weakImpls()[i];
new (NotNull, weakImpl) WeakImpl;
addToFreeList(&m_sweepResult.freeList, weakImpl);
}
ASSERT(isEmpty());
}
void WeakBlock::lastChanceToFinalize()
{
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 (isEmpty())
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;
if (weakImpl->state() == WeakImpl::Live)
sweepResult.blockIsLogicallyEmpty = false;
}
}
m_sweepResult = sweepResult;
ASSERT(!m_sweepResult.isNull());
}
template<typename ContainerType>
void WeakBlock::specializedVisit(ContainerType& container, SlotVisitor& visitor)
{
HeapVersion markingVersion = visitor.markingVersion();
size_t count = weakImplCount();
for (size_t i = 0; i < count; ++i) {
WeakImpl* weakImpl = &weakImpls()[i];
if (weakImpl->state() != WeakImpl::Live)
continue;
WeakHandleOwner* weakHandleOwner = weakImpl->weakHandleOwner();
if (!weakHandleOwner)
continue;
JSValue jsValue = weakImpl->jsValue();
if (container.isMarked(markingVersion, jsValue.asCell()))
continue;
const char* reason = "";
const char** reasonPtr = nullptr;
if (UNLIKELY(visitor.isAnalyzingHeap()))
reasonPtr = &reason;
if (!weakHandleOwner->isReachableFromOpaqueRoots(Handle<Unknown>::wrapSlot(&const_cast<JSValue&>(jsValue)), weakImpl->context(), visitor, reasonPtr))
continue;
visitor.appendUnbarriered(jsValue);
if (UNLIKELY(visitor.isAnalyzingHeap())) {
if (jsValue.isCell())
visitor.heapAnalyzer()->setOpaqueRootReachabilityReasonForCell(jsValue.asCell(), *reasonPtr);
}
}
}
void WeakBlock::visit(SlotVisitor& visitor)
{
if (isEmpty())
return;
ASSERT(m_container);
if (m_container.isPreciseAllocation())
specializedVisit(m_container.preciseAllocation(), visitor);
else
specializedVisit(m_container.markedBlock(), visitor);
}
void WeakBlock::reap()
{
if (isEmpty())
return;
ASSERT(m_container);
HeapVersion markingVersion = m_container.heap()->objectSpace().markingVersion();
for (size_t i = 0; i < weakImplCount(); ++i) {
WeakImpl* weakImpl = &weakImpls()[i];
if (weakImpl->state() > WeakImpl::Dead)
continue;
if (m_container.isMarked(markingVersion, weakImpl->jsValue().asCell())) {
ASSERT(weakImpl->state() == WeakImpl::Live);
continue;
}
weakImpl->setState(WeakImpl::Dead);
}
}
}