#include "config.h"
#include "WasmCodeBlock.h"
#if ENABLE(WEBASSEMBLY)
#include "WasmBBQPlanInlines.h"
#include "WasmCallee.h"
#include "WasmWorklist.h"
namespace JSC { namespace Wasm {
CodeBlock::CodeBlock(MemoryMode mode, ModuleInformation& moduleInformation)
: m_calleeCount(moduleInformation.internalFunctionCount())
, m_mode(mode)
{
RefPtr<CodeBlock> protectedThis = this;
m_plan = adoptRef(*new BBQPlan(nullptr, makeRef(moduleInformation), BBQPlan::FullCompile, createSharedTask<Plan::CallbackType>([this, protectedThis = WTFMove(protectedThis)] (VM*, Plan&) {
auto locker = holdLock(m_lock);
if (m_plan->failed()) {
m_errorMessage = m_plan->errorMessage();
m_plan = nullptr;
return;
}
m_callees.resize(m_calleeCount);
m_optimizedCallees.resize(m_calleeCount);
m_wasmIndirectCallEntryPoints.resize(m_calleeCount);
m_plan->initializeCallees([&] (unsigned calleeIndex, RefPtr<Wasm::Callee>&& jsEntrypointCallee, Ref<Wasm::Callee>&& wasmEntrypointCallee) {
if (jsEntrypointCallee) {
auto result = m_jsCallees.set(calleeIndex, WTFMove(jsEntrypointCallee));
ASSERT_UNUSED(result, result.isNewEntry);
}
m_callees[calleeIndex] = WTFMove(wasmEntrypointCallee);
m_wasmIndirectCallEntryPoints[calleeIndex] = m_callees[calleeIndex]->entrypoint();
});
m_wasmToWasmExitStubs = m_plan->takeWasmToWasmExitStubs();
m_wasmToWasmCallsites = m_plan->takeWasmToWasmCallsites();
m_tierUpCounts = m_plan->takeTierUpCounts();
m_plan = nullptr;
})));
m_plan->setMode(mode);
auto& worklist = Wasm::ensureWorklist();
worklist.enqueue(makeRef(*m_plan.get()));
}
CodeBlock::~CodeBlock() { }
void CodeBlock::waitUntilFinished()
{
RefPtr<Plan> plan;
{
auto locker = holdLock(m_lock);
plan = m_plan;
}
if (plan) {
auto& worklist = Wasm::ensureWorklist();
worklist.completePlanSynchronously(*plan.get());
}
}
void CodeBlock::compileAsync(VM& vm, AsyncCompilationCallback&& task)
{
RefPtr<Plan> plan;
{
auto locker = holdLock(m_lock);
plan = m_plan;
}
if (plan) {
RefPtr<CodeBlock> protectedThis = this;
plan->addCompletionTask(vm, createSharedTask<Plan::CallbackType>([this, task = WTFMove(task), protectedThis = WTFMove(protectedThis)] (VM* vm, Plan&) {
ASSERT(vm);
task->run(*vm, makeRef(*this));
}));
} else
task->run(vm, makeRef(*this));
}
bool CodeBlock::isSafeToRun(MemoryMode memoryMode)
{
if (!runnable())
return false;
switch (m_mode) {
case Wasm::MemoryMode::BoundsChecking:
return true;
case Wasm::MemoryMode::Signaling:
return memoryMode == Wasm::MemoryMode::Signaling;
case Wasm::MemoryMode::NumberOfMemoryModes:
break;
}
RELEASE_ASSERT_NOT_REACHED();
return false;
}
} }
#endif // ENABLE(WEBASSEMBLY)