#include "config.h"
#include "WasmCodeBlock.h"
#if ENABLE(WEBASSEMBLY)
#include "WasmBBQPlan.h"
#include "WasmCallee.h"
#include "WasmLLIntPlan.h"
#include "WasmWorklist.h"
namespace JSC { namespace Wasm {
Ref<CodeBlock> CodeBlock::create(Context* context, MemoryMode mode, ModuleInformation& moduleInformation, RefPtr<LLIntCallees> llintCallees)
{
auto* result = new (NotNull, fastMalloc(sizeof(CodeBlock))) CodeBlock(context, mode, moduleInformation, llintCallees);
return adoptRef(*result);
}
CodeBlock::CodeBlock(Context* context, MemoryMode mode, ModuleInformation& moduleInformation, RefPtr<LLIntCallees> llintCallees)
: m_calleeCount(moduleInformation.internalFunctionCount())
, m_mode(mode)
, m_llintCallees(llintCallees)
{
RefPtr<CodeBlock> protectedThis = this;
if (Options::useWasmLLInt()) {
m_plan = adoptRef(*new LLIntPlan(context, makeRef(moduleInformation), m_llintCallees->data(), createSharedTask<Plan::CallbackType>([this, protectedThis = WTFMove(protectedThis)] (Plan&) {
auto locker = holdLock(m_lock);
if (m_plan->failed()) {
m_errorMessage = m_plan->errorMessage();
setCompilationFinished();
return;
}
m_bbqCallees.resize(m_calleeCount);
m_omgCallees.resize(m_calleeCount);
m_wasmIndirectCallEntryPoints.resize(m_calleeCount);
for (unsigned i = 0; i < m_calleeCount; ++i)
m_wasmIndirectCallEntryPoints[i] = m_llintCallees->at(i)->entrypoint();
m_wasmToWasmExitStubs = m_plan->takeWasmToWasmExitStubs();
m_wasmToWasmCallsites = m_plan->takeWasmToWasmCallsites();
m_embedderCallees = static_cast<LLIntPlan*>(m_plan.get())->takeEmbedderCallees();
setCompilationFinished();
})));
} else {
m_plan = adoptRef(*new BBQPlan(context, makeRef(moduleInformation), EntryPlan::FullCompile, createSharedTask<Plan::CallbackType>([this, protectedThis = WTFMove(protectedThis)] (Plan&) {
auto locker = holdLock(m_lock);
if (m_plan->failed()) {
m_errorMessage = m_plan->errorMessage();
setCompilationFinished();
return;
}
m_bbqCallees.resize(m_calleeCount);
m_omgCallees.resize(m_calleeCount);
m_wasmIndirectCallEntryPoints.resize(m_calleeCount);
BBQPlan* bbqPlan = static_cast<BBQPlan*>(m_plan.get());
bbqPlan->initializeCallees([&] (unsigned calleeIndex, RefPtr<EmbedderEntrypointCallee>&& embedderEntrypointCallee, RefPtr<BBQCallee>&& wasmEntrypoint) {
if (embedderEntrypointCallee) {
auto result = m_embedderCallees.set(calleeIndex, WTFMove(embedderEntrypointCallee));
ASSERT_UNUSED(result, result.isNewEntry);
}
m_wasmIndirectCallEntryPoints[calleeIndex] = wasmEntrypoint->entrypoint();
m_bbqCallees[calleeIndex] = adoptRef(static_cast<BBQCallee*>(wasmEntrypoint.leakRef()));
});
m_wasmToWasmExitStubs = m_plan->takeWasmToWasmExitStubs();
m_wasmToWasmCallsites = m_plan->takeWasmToWasmCallsites();
setCompilationFinished();
})));
}
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(Context* context, AsyncCompilationCallback&& task)
{
RefPtr<Plan> plan;
{
auto locker = holdLock(m_lock);
plan = m_plan;
}
if (plan) {
RefPtr<CodeBlock> protectedThis = this;
plan->addCompletionTask(context, createSharedTask<Plan::CallbackType>([this, task = WTFMove(task), protectedThis = WTFMove(protectedThis)] (Plan&) {
task->run(makeRef(*this));
}));
} else
task->run(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;
}
RELEASE_ASSERT_NOT_REACHED();
return false;
}
void CodeBlock::setCompilationFinished()
{
m_plan = nullptr;
m_compilationFinished.store(true);
}
} }
#endif // ENABLE(WEBASSEMBLY)