void ScriptContextTTD::RegisterEvalScript(Js::FunctionBody* body, uint32 bodyCtrId)
{
TTD::TopLevelFunctionInContextRelation relation;
relation.TopLevelBodyCtr = bodyCtrId;
relation.ContextSpecificBodyPtrId = TTD_CONVERT_FUNCTIONBODY_TO_PTR_ID(body);
this->m_ttdTopLevelEval.Add(relation.ContextSpecificBodyPtrId, relation);
}
void SnapshotExtractor::ExtractSlotArrayIfNeeded(Js::ScriptContext* ctx, Js::Var* scope)
{
if(this->m_marks.IsMarked(scope))
{
NSSnapValues::SlotArrayInfo* slotInfo = this->m_pendingSnap->GetNextAvailableSlotArrayEntry();
Js::ScopeSlots slots(scope);
slotInfo->SlotId = TTD_CONVERT_VAR_TO_PTR_ID(scope);
slotInfo->ScriptContextLogId = ctx->ScriptContextLogTag;
slotInfo->SlotCount = slots.GetCount();
slotInfo->Slots = this->m_pendingSnap->GetSnapshotSlabAllocator().SlabAllocateArray<TTDVar>(slotInfo->SlotCount);
for(uint32 j = 0; j < slotInfo->SlotCount; ++j)
{
slotInfo->Slots[j] = slots.Get(j);
}
if(slots.IsFunctionScopeSlotArray())
{
Js::FunctionBody* fb = slots.GetFunctionBody();
slotInfo->isFunctionBodyMetaData = true;
slotInfo->OptFunctionBodyId = TTD_CONVERT_FUNCTIONBODY_TO_PTR_ID(fb);
#if ENABLE_TTD_INTERNAL_DIAGNOSTICS
Js::PropertyId* propertyIds = fb->GetPropertyIdsForScopeSlotArray();
slotInfo->DebugPIDArray = this->m_pendingSnap->GetSnapshotSlabAllocator().SlabAllocateArray<Js::PropertyId>(slotInfo->SlotCount);
for(uint32 j = 0; j < slotInfo->SlotCount; ++j)
{
slotInfo->DebugPIDArray[j] = propertyIds[j];
}
#endif
}
else
{
slotInfo->isFunctionBodyMetaData = false;
slotInfo->OptFunctionBodyId = TTD_INVALID_PTR_ID;
#if ENABLE_TTD_INTERNAL_DIAGNOSTICS
slotInfo->DebugPIDArray = this->m_pendingSnap->GetSnapshotSlabAllocator().SlabAllocateArray<Js::PropertyId>(slotInfo->SlotCount);
for(uint32 j = 0; j < slotInfo->SlotCount; ++j)
{
slotInfo->DebugPIDArray[j] = (Js::PropertyId)0;
}
#endif
}
this->m_marks.ClearMark(scope);
}
}
uint32 ScriptContextTTD::FindTopLevelCtrForBody(Js::FunctionBody* body) const
{
Js::FunctionBody* rootBody = body;
while(this->ResolveParentBody(rootBody) != nullptr)
{
rootBody = this->ResolveParentBody(rootBody);
}
TTD_PTR_ID trgtid = TTD_CONVERT_FUNCTIONBODY_TO_PTR_ID(rootBody);
for(auto iter = this->m_ttdTopLevelScriptLoad.GetIterator(); iter.IsValid(); iter.MoveNext())
{
if(iter.CurrentValue().ContextSpecificBodyPtrId == trgtid)
{
return iter.CurrentValue().TopLevelBodyCtr;
}
}
for(auto iter = this->m_ttdTopLevelNewFunction.GetIterator(); iter.IsValid(); iter.MoveNext())
{
if(iter.CurrentValue().ContextSpecificBodyPtrId == trgtid)
{
return iter.CurrentValue().TopLevelBodyCtr;
}
}
for(auto iter = this->m_ttdTopLevelEval.GetIterator(); iter.IsValid(); iter.MoveNext())
{
if(iter.CurrentValue().ContextSpecificBodyPtrId == trgtid)
{
return iter.CurrentValue().TopLevelBodyCtr;
}
}
TTDAssert(false, "We are missing a top-level function reference.");
return 0;
}
