#include "config.h"
#include "WasmCodeBlock.h"
#if ENABLE(WEBASSEMBLY)
#include "WasmBBQPlanInlines.h"
#include "WasmCallee.h"
#include "WasmFormat.h"
#include "WasmWorklist.h"
namespace JSC { namespace Wasm {
Ref<CodeBlock> CodeBlock::create(Context* context, MemoryMode mode, ModuleInformation& moduleInformation, CreateEmbedderWrapper&& createEmbedderWrapper, ThrowWasmException throwWasmException)
{
auto* result = new (NotNull, fastMalloc(sizeof(CodeBlock))) CodeBlock(context, mode, moduleInformation, WTFMove(createEmbedderWrapper), throwWasmException);
return adoptRef(*result);
}
CodeBlock::CodeBlock(Context* context, MemoryMode mode, ModuleInformation& moduleInformation, CreateEmbedderWrapper&& createEmbedderWrapper, ThrowWasmException throwWasmException)
: m_calleeCount(moduleInformation.internalFunctionCount())
, m_mode(mode)
{
RefPtr<CodeBlock> protectedThis = this;
m_plan = adoptRef(*new BBQPlan(context, makeRef(moduleInformation), BBQPlan::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_callees.resize(m_calleeCount);
m_optimizedCallees.resize(m_calleeCount);
m_wasmIndirectCallEntryPoints.resize(m_calleeCount);
m_plan->initializeCallees([&] (unsigned calleeIndex, RefPtr<Wasm::Callee>&& embedderEntrypointCallee, Ref<Wasm::Callee>&& wasmEntrypointCallee) {
if (embedderEntrypointCallee) {
auto result = m_embedderCallees.set(calleeIndex, WTFMove(embedderEntrypointCallee));
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();
setCompilationFinished();
}), WTFMove(createEmbedderWrapper), throwWasmException));
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)