blob: a321c52f15eca52213ecd6ee92172719811b325a [file] [log] [blame]
/*
* Copyright (C) 2016-2017 Apple Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "config.h"
#include "WasmCallee.h"
#include "runtime/VM.h"
#if ENABLE(WEBASSEMBLY)
#include "LLIntExceptions.h"
#include "WasmCalleeRegistry.h"
#include "WasmCallingConvention.h"
namespace JSC { namespace Wasm {
Callee::Callee(Wasm::CompilationMode compilationMode)
: m_compilationMode(compilationMode)
{
CalleeRegistry::singleton().registerCallee(this);
}
Callee::Callee(Wasm::CompilationMode compilationMode, size_t index, std::pair<const Name*, RefPtr<NameSection>>&& name)
: m_compilationMode(compilationMode)
, m_indexOrName(index, WTFMove(name))
{
CalleeRegistry::singleton().registerCallee(this);
}
Callee::~Callee()
{
CalleeRegistry::singleton().unregisterCallee(this);
}
void Callee::dump(PrintStream& out) const
{
out.print(makeString(m_indexOrName));
}
const HandlerInfo* Callee::handlerForIndex(Instance& instance, unsigned index, const Tag* tag)
{
ASSERT(hasExceptionHandlers());
return HandlerInfo::handlerForIndex(instance, m_exceptionHandlers, index, tag);
}
JITCallee::JITCallee(Wasm::CompilationMode compilationMode, Entrypoint&& entrypoint)
: Callee(compilationMode)
, m_entrypoint(WTFMove(entrypoint))
{
}
JITCallee::JITCallee(Wasm::CompilationMode compilationMode, Entrypoint&& entrypoint, size_t index, std::pair<const Name*, RefPtr<NameSection>>&& name, Vector<UnlinkedWasmToWasmCall>&& unlinkedCalls)
: Callee(compilationMode, index, WTFMove(name))
, m_wasmToWasmCallsites(WTFMove(unlinkedCalls))
, m_entrypoint(WTFMove(entrypoint))
{
}
LLIntCallee::LLIntCallee(FunctionCodeBlockGenerator& generator, size_t index, std::pair<const Name*, RefPtr<NameSection>>&& name)
: Callee(Wasm::CompilationMode::LLIntMode, index, WTFMove(name))
, m_functionIndex(generator.m_functionIndex)
, m_numVars(generator.m_numVars)
, m_numCalleeLocals(generator.m_numCalleeLocals)
, m_numArguments(generator.m_numArguments)
, m_constantTypes(WTFMove(generator.m_constantTypes))
, m_constants(WTFMove(generator.m_constants))
, m_instructions(WTFMove(generator.m_instructions))
, m_instructionsRawPointer(generator.m_instructionsRawPointer)
, m_jumpTargets(WTFMove(generator.m_jumpTargets))
, m_signatures(WTFMove(generator.m_signatures))
, m_outOfLineJumpTargets(WTFMove(generator.m_outOfLineJumpTargets))
, m_tierUpCounter(WTFMove(generator.m_tierUpCounter))
, m_jumpTables(WTFMove(generator.m_jumpTables))
{
if (size_t count = generator.numberOfExceptionHandlers()) {
m_exceptionHandlers = FixedVector<HandlerInfo>(count);
for (size_t i = 0; i < count; i++) {
const UnlinkedHandlerInfo& unlinkedHandler = generator.exceptionHandler(i);
HandlerInfo& handler = m_exceptionHandlers[i];
auto& instruction = *m_instructions->at(unlinkedHandler.m_target).ptr();
CodeLocationLabel<ExceptionHandlerPtrTag> target;
if (unlinkedHandler.m_type == HandlerType::Catch)
target = CodeLocationLabel<ExceptionHandlerPtrTag>(LLInt::handleWasmCatch(instruction.width()).code());
else
target = CodeLocationLabel<ExceptionHandlerPtrTag>(LLInt::handleWasmCatchAll(instruction.width()).code());
handler.initialize(unlinkedHandler, target);
}
}
}
void LLIntCallee::setEntrypoint(MacroAssemblerCodePtr<WasmEntryPtrTag> entrypoint)
{
m_entrypoint = entrypoint;
}
MacroAssemblerCodePtr<WasmEntryPtrTag> LLIntCallee::entrypoint() const
{
return m_entrypoint;
}
RegisterAtOffsetList* LLIntCallee::calleeSaveRegisters()
{
static LazyNeverDestroyed<RegisterAtOffsetList> calleeSaveRegisters;
static std::once_flag initializeFlag;
std::call_once(initializeFlag, [] {
RegisterSet registers;
registers.set(GPRInfo::regCS0); // Wasm::Instance
#if CPU(X86_64)
registers.set(GPRInfo::regCS2); // PB
#elif CPU(ARM64) || CPU(RISCV64)
registers.set(GPRInfo::regCS7); // PB
#else
#error Unsupported architecture.
#endif
ASSERT(registers.numberOfSetRegisters() == numberOfLLIntCalleeSaveRegisters);
calleeSaveRegisters.construct(WTFMove(registers));
});
return &calleeSaveRegisters.get();
}
std::tuple<void*, void*> LLIntCallee::range() const
{
return { nullptr, nullptr };
}
WasmInstructionStream::Offset LLIntCallee::outOfLineJumpOffset(WasmInstructionStream::Offset bytecodeOffset)
{
ASSERT(m_outOfLineJumpTargets.contains(bytecodeOffset));
return m_outOfLineJumpTargets.get(bytecodeOffset);
}
const WasmInstruction* LLIntCallee::outOfLineJumpTarget(const WasmInstruction* pc)
{
int offset = bytecodeOffset(pc);
int target = outOfLineJumpOffset(offset);
return m_instructions->at(offset + target).ptr();
}
void OptimizingJITCallee::linkExceptionHandlers(Vector<UnlinkedHandlerInfo> unlinkedExceptionHandlers, Vector<CodeLocationLabel<ExceptionHandlerPtrTag>> exceptionHandlerLocations)
{
size_t count = unlinkedExceptionHandlers.size();
m_exceptionHandlers = FixedVector<HandlerInfo>(count);
for (size_t i = 0; i < count; i++) {
HandlerInfo& handler = m_exceptionHandlers[i];
const UnlinkedHandlerInfo& unlinkedHandler = unlinkedExceptionHandlers[i];
CodeLocationLabel<ExceptionHandlerPtrTag> location = exceptionHandlerLocations[i];
handler.initialize(unlinkedHandler, location);
}
}
const StackMap& OptimizingJITCallee::stackmap(CallSiteIndex callSiteIndex) const
{
auto iter = m_stackmaps.find(callSiteIndex);
if (iter == m_stackmaps.end()) {
for (auto pair : m_stackmaps) {
dataLog(pair.key.bits(), ": ");
for (auto value : pair.value)
dataLog(value, ", ");
dataLogLn("");
}
}
RELEASE_ASSERT(iter != m_stackmaps.end());
return iter->value;
}
} } // namespace JSC::Wasm
#endif // ENABLE(WEBASSEMBLY)