#pragma once
#if ENABLE(DFG_JIT)
#include "DFGOSRExitBase.h"
#include "DFGVariableEventStream.h"
#include "GPRInfo.h"
#include "MacroAssembler.h"
#include "MethodOfGettingAValueProfile.h"
#include "Operands.h"
#include "ValueRecovery.h"
#include <wtf/RefPtr.h>
namespace JSC {
class ArrayProfile;
class CCallHelpers;
namespace Probe {
class Context;
}
namespace Profiler {
class OSRExit;
}
namespace DFG {
class SpeculativeJIT;
struct BasicBlock;
struct Node;
enum SpeculationRecoveryType : uint8_t {
SpeculativeAdd,
SpeculativeAddImmediate,
BooleanSpeculationCheck
};
class SpeculationRecovery {
public:
SpeculationRecovery(SpeculationRecoveryType type, GPRReg dest, GPRReg src)
: m_src(src)
, m_dest(dest)
, m_type(type)
{
ASSERT(m_type == SpeculativeAdd || m_type == BooleanSpeculationCheck);
}
SpeculationRecovery(SpeculationRecoveryType type, GPRReg dest, int32_t immediate)
: m_immediate(immediate)
, m_dest(dest)
, m_type(type)
{
ASSERT(m_type == SpeculativeAddImmediate);
}
SpeculationRecoveryType type() { return m_type; }
GPRReg dest() { return m_dest; }
GPRReg src() { return m_src; }
int32_t immediate() { return m_immediate; }
private:
union {
GPRReg m_src;
int32_t m_immediate;
};
GPRReg m_dest;
SpeculationRecoveryType m_type;
};
enum class ExtraInitializationLevel;
struct OSRExitState : RefCounted<OSRExitState> {
OSRExitState(OSRExitBase& exit, CodeBlock* codeBlock, CodeBlock* baselineCodeBlock, Operands<ValueRecovery>& operands, Vector<UndefinedOperandSpan>&& undefinedOperandSpans, SpeculationRecovery* recovery, ptrdiff_t stackPointerOffset, int32_t activeThreshold, double memoryUsageAdjustedThreshold, void* jumpTarget, ArrayProfile* arrayProfile)
: exit(exit)
, codeBlock(codeBlock)
, baselineCodeBlock(baselineCodeBlock)
, operands(operands)
, undefinedOperandSpans(undefinedOperandSpans)
, recovery(recovery)
, stackPointerOffset(stackPointerOffset)
, activeThreshold(activeThreshold)
, memoryUsageAdjustedThreshold(memoryUsageAdjustedThreshold)
, jumpTarget(jumpTarget)
, arrayProfile(arrayProfile)
{ }
OSRExitBase& exit;
CodeBlock* codeBlock;
CodeBlock* baselineCodeBlock;
Operands<ValueRecovery> operands;
Vector<UndefinedOperandSpan> undefinedOperandSpans;
SpeculationRecovery* recovery;
ptrdiff_t stackPointerOffset;
uint32_t activeThreshold;
double memoryUsageAdjustedThreshold;
void* jumpTarget;
ArrayProfile* arrayProfile;
ExtraInitializationLevel extraInitializationLevel;
Profiler::OSRExit* profilerExit { nullptr };
};
struct OSRExit : public OSRExitBase {
OSRExit(ExitKind, JSValueSource, MethodOfGettingAValueProfile, SpeculativeJIT*, unsigned streamIndex, unsigned recoveryIndex = UINT_MAX);
static void JIT_OPERATION compileOSRExit(ExecState*) WTF_INTERNAL;
static void executeOSRExit(Probe::Context&);
unsigned m_patchableCodeOffset { 0 };
MacroAssemblerCodeRef m_code;
RefPtr<OSRExitState> exitState;
JSValueSource m_jsValueSource;
MethodOfGettingAValueProfile m_valueProfile;
unsigned m_recoveryIndex;
void setPatchableCodeOffset(MacroAssembler::PatchableJump);
MacroAssembler::Jump getPatchableCodeOffsetAsJump() const;
CodeLocationJump codeLocationForRepatch(CodeBlock*) const;
void correctJump(LinkBuffer&);
unsigned m_streamIndex;
void considerAddingAsFrequentExitSite(CodeBlock* profiledCodeBlock)
{
OSRExitBase::considerAddingAsFrequentExitSite(profiledCodeBlock, ExitFromDFG);
}
private:
static void compileExit(CCallHelpers&, VM&, const OSRExit&, const Operands<ValueRecovery>&, SpeculationRecovery*);
static void emitRestoreArguments(CCallHelpers&, const Operands<ValueRecovery>&);
static void JIT_OPERATION debugOperationPrintSpeculationFailure(ExecState*, void*, void*) WTF_INTERNAL;
};
struct SpeculationFailureDebugInfo {
CodeBlock* codeBlock;
ExitKind kind;
unsigned bytecodeOffset;
};
} }
#endif // ENABLE(DFG_JIT)