void DoCheck (const PolyWord pt)
{
if (pt == PolyWord::FromUnsigned(0)) return;
if (pt.IsTagged()) return;
CheckAddress(pt.AsStackAddr());
}
// This deals with weak references within the run-time system.
void MTGCCheckWeakRef::ScanRuntimeAddress(PolyObject **pt, RtsStrength weak)
{
/* If the object has not been marked and this is only a weak reference */
/* then the pointer is set to zero. This allows streams or windows */
/* to be closed if there is no other reference to them. */
PolyObject *val = *pt;
PolyWord w = val;
if (weak == STRENGTH_STRONG)
return;
LocalMemSpace *space = gMem.LocalSpaceForAddress(w.AsStackAddr());
if (space == 0)
return; // Not in local area
// If it hasn't been marked set it to zero.
if (! space->bitmap.TestBit(space->wordNo(w.AsStackAddr())))
*pt = 0;
}
// Deal with weak objects
void MTGCCheckWeakRef::ScanAddressesInObject(PolyObject *obj, POLYUNSIGNED L)
{
if (! OBJ_IS_WEAKREF_OBJECT(L)) return;
ASSERT(OBJ_IS_MUTABLE_OBJECT(L)); // Should be a mutable.
ASSERT(OBJ_IS_WORD_OBJECT(L)); // Should be a plain object.
// See if any of the SOME objects contain unreferenced refs.
POLYUNSIGNED length = OBJ_OBJECT_LENGTH(L);
PolyWord *baseAddr = (PolyWord*)obj;
for (POLYUNSIGNED i = 0; i < length; i++)
{
PolyWord someAddr = baseAddr[i];
if (someAddr.IsDataPtr())
{
LocalMemSpace *someSpace = gMem.LocalSpaceForAddress(someAddr.AsAddress());
if (someSpace != 0)
{
PolyObject *someObj = someAddr.AsObjPtr();
// If this is a weak object the SOME value may refer to an unreferenced
// ref. If so we have to set this entry to NONE. For safety we also
// set the contents of the SOME to TAGGED(0).
ASSERT(someObj->Length() == 1 && someObj->IsWordObject()); // Should be a SOME node.
PolyWord refAddress = someObj->Get(0);
LocalMemSpace *space = gMem.LocalSpaceForAddress(refAddress.AsAddress());
if (space != 0)
// If the ref is permanent it's always there.
{
POLYUNSIGNED new_bitno = space->wordNo(refAddress.AsStackAddr());
if (! space->bitmap.TestBit(new_bitno))
{
// It wasn't marked so it's otherwise unreferenced.
baseAddr[i] = TAGGED(0); // Set it to NONE.
someObj->Set(0, TAGGED(0)); // For safety.
convertedWeak = true;
}
}
}
}
}
}
// Copy a stack
static void CopyStackFrame(StackObject *old_stack, POLYUNSIGNED old_length, StackObject *new_stack, POLYUNSIGNED new_length)
{
/* Moves a stack, updating all references within the stack */
PolyWord *old_base = (PolyWord *)old_stack;
PolyWord *new_base = (PolyWord*)new_stack;
PolyWord *old_top = old_base + old_length;
/* Calculate the offset of the new stack from the old. If the frame is
being extended objects in the new frame will be further up the stack
than in the old one. */
POLYSIGNED offset = new_base - old_base + new_length - old_length;
/* Copy the registers, changing any that point into the stack. */
new_stack->p_space = old_stack->p_space;
new_stack->p_pc = old_stack->p_pc;
new_stack->p_sp = old_stack->p_sp + offset;
new_stack->p_hr = old_stack->p_hr + offset;
new_stack->p_nreg = old_stack->p_nreg;
/* p_nreg contains contains the number of CHECKED registers */
POLYUNSIGNED i;
for (i = 0; i < new_stack->p_nreg; i++)
{
PolyWord R = old_stack->p_reg[i];
/* if the register points into the old stack, make the new copy
point at the same relative offset within the new stack,
otherwise make the new copy identical to the old version. */
if (R.IsTagged() || R.AsStackAddr() < old_base || R.AsStackAddr() >= old_top)
new_stack->p_reg[i] = R;
else new_stack->p_reg[i] = PolyWord::FromStackAddr(R.AsStackAddr() + offset);
}
/* Copy unchecked registers. - The next "register" is the number of
unchecked registers to copy. Unchecked registers are used for
values that might look like addresses, i.e. don't have tag bits,
but are not. */
POLYUNSIGNED n = old_stack->p_reg[i].AsUnsigned();
new_stack->p_reg[i] = old_stack->p_reg[i];
i++;
ASSERT (n < 100);
while (n--)
{
new_stack->p_reg[i] = old_stack->p_reg[i];
i++;
}
/* Skip the unused part of the stack. */
i = (PolyWord*)old_stack->p_sp - old_base;
ASSERT (i <= old_length);
i = old_length - i;
PolyWord *old = old_stack->p_sp;
PolyWord *newp= new_stack->p_sp;
while (i--)
{
// ASSERT(old >= old_base && old < old_base+old_length);
// ASSERT(newp >= new_base && newp < new_base+new_length);
PolyWord old_word = *old++;
if (old_word.IsTagged() || old_word.AsStackAddr() < old_base || old_word.AsStackAddr() >= old_top)
*newp++ = old_word;
else
*newp++ = PolyWord::FromStackAddr(old_word.AsStackAddr() + offset);
}
ASSERT(old == ((PolyWord*)old_stack)+old_length);
ASSERT(newp == ((PolyWord*)new_stack)+new_length);
}