StructureStubInfo should be able to reset itself without going through CodeBlock
[WebKit-https.git] / Source / JavaScriptCore / bytecode / CodeBlock.h
1 /*
2  * Copyright (C) 2008-2015 Apple Inc. All rights reserved.
3  * Copyright (C) 2008 Cameron Zwarich <cwzwarich@uwaterloo.ca>
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1.  Redistributions of source code must retain the above copyright
10  *     notice, this list of conditions and the following disclaimer.
11  * 2.  Redistributions in binary form must reproduce the above copyright
12  *     notice, this list of conditions and the following disclaimer in the
13  *     documentation and/or other materials provided with the distribution.
14  * 3.  Neither the name of Apple Inc. ("Apple") nor the names of
15  *     its contributors may be used to endorse or promote products derived
16  *     from this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
19  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
21  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
22  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
23  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
24  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
25  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29
30 #ifndef CodeBlock_h
31 #define CodeBlock_h
32
33 #include "ArrayProfile.h"
34 #include "ByValInfo.h"
35 #include "BytecodeConventions.h"
36 #include "BytecodeLivenessAnalysis.h"
37 #include "CallLinkInfo.h"
38 #include "CallReturnOffsetToBytecodeOffset.h"
39 #include "CodeBlockHash.h"
40 #include "CodeBlockSet.h"
41 #include "ConcurrentJITLock.h"
42 #include "CodeOrigin.h"
43 #include "CodeType.h"
44 #include "CompactJITCodeMap.h"
45 #include "DFGCommon.h"
46 #include "DFGCommonData.h"
47 #include "DFGExitProfile.h"
48 #include "DeferredCompilationCallback.h"
49 #include "EvalCodeCache.h"
50 #include "ExecutionCounter.h"
51 #include "ExpressionRangeInfo.h"
52 #include "HandlerInfo.h"
53 #include "ObjectAllocationProfile.h"
54 #include "Options.h"
55 #include "PutPropertySlot.h"
56 #include "Instruction.h"
57 #include "JITCode.h"
58 #include "JITWriteBarrier.h"
59 #include "JSGlobalObject.h"
60 #include "JumpTable.h"
61 #include "LLIntCallLinkInfo.h"
62 #include "LazyOperandValueProfile.h"
63 #include "ProfilerCompilation.h"
64 #include "ProfilerJettisonReason.h"
65 #include "RegExpObject.h"
66 #include "StructureStubInfo.h"
67 #include "UnconditionalFinalizer.h"
68 #include "ValueProfile.h"
69 #include "VirtualRegister.h"
70 #include "Watchpoint.h"
71 #include <wtf/Bag.h>
72 #include <wtf/FastMalloc.h>
73 #include <wtf/RefCountedArray.h>
74 #include <wtf/RefPtr.h>
75 #include <wtf/SegmentedVector.h>
76 #include <wtf/Vector.h>
77 #include <wtf/text/WTFString.h>
78
79 namespace JSC {
80
81 class ExecState;
82 class LLIntOffsetsExtractor;
83 class TypeLocation;
84
85 enum ReoptimizationMode { DontCountReoptimization, CountReoptimization };
86
87 class CodeBlock : public ThreadSafeRefCounted<CodeBlock>, public UnconditionalFinalizer, public WeakReferenceHarvester {
88     WTF_MAKE_FAST_ALLOCATED;
89     friend class BytecodeLivenessAnalysis;
90     friend class JIT;
91     friend class LLIntOffsetsExtractor;
92 public:
93     enum CopyParsedBlockTag { CopyParsedBlock };
94 protected:
95     CodeBlock(CopyParsedBlockTag, CodeBlock& other);
96         
97     CodeBlock(ScriptExecutable* ownerExecutable, UnlinkedCodeBlock*, JSScope*, PassRefPtr<SourceProvider>, unsigned sourceOffset, unsigned firstLineColumnOffset);
98 #if ENABLE(WEBASSEMBLY)
99     CodeBlock(WebAssemblyExecutable* ownerExecutable, VM&, JSGlobalObject*);
100 #endif
101
102     WriteBarrier<JSGlobalObject> m_globalObject;
103     Heap* m_heap;
104
105 public:
106     JS_EXPORT_PRIVATE virtual ~CodeBlock();
107
108     UnlinkedCodeBlock* unlinkedCodeBlock() const { return m_unlinkedCode.get(); }
109
110     CString inferredName() const;
111     CodeBlockHash hash() const;
112     bool hasHash() const;
113     bool isSafeToComputeHash() const;
114     CString hashAsStringIfPossible() const;
115     CString sourceCodeForTools() const; // Not quite the actual source we parsed; this will do things like prefix the source for a function with a reified signature.
116     CString sourceCodeOnOneLine() const; // As sourceCodeForTools(), but replaces all whitespace runs with a single space.
117     void dumpAssumingJITType(PrintStream&, JITCode::JITType) const;
118     void dump(PrintStream&) const;
119
120     int numParameters() const { return m_numParameters; }
121     void setNumParameters(int newValue);
122
123     int* addressOfNumParameters() { return &m_numParameters; }
124     static ptrdiff_t offsetOfNumParameters() { return OBJECT_OFFSETOF(CodeBlock, m_numParameters); }
125
126     CodeBlock* alternative() { return m_alternative.get(); }
127     PassRefPtr<CodeBlock> releaseAlternative() { return m_alternative.release(); }
128     void setAlternative(PassRefPtr<CodeBlock> alternative) { m_alternative = alternative; }
129
130     template <typename Functor> void forEachRelatedCodeBlock(Functor&& functor)
131     {
132         Functor f(std::forward<Functor>(functor));
133         Vector<CodeBlock*, 4> codeBlocks;
134         codeBlocks.append(this);
135
136         while (!codeBlocks.isEmpty()) {
137             CodeBlock* currentCodeBlock = codeBlocks.takeLast();
138             f(currentCodeBlock);
139
140             if (CodeBlock* alternative = currentCodeBlock->alternative())
141                 codeBlocks.append(alternative);
142             if (CodeBlock* osrEntryBlock = currentCodeBlock->specialOSREntryBlockOrNull())
143                 codeBlocks.append(osrEntryBlock);
144         }
145     }
146     
147     CodeSpecializationKind specializationKind() const
148     {
149         return specializationFromIsConstruct(m_isConstructor);
150     }
151     
152     CodeBlock* baselineAlternative();
153     
154     // FIXME: Get rid of this.
155     // https://bugs.webkit.org/show_bug.cgi?id=123677
156     CodeBlock* baselineVersion();
157
158     void visitAggregate(SlotVisitor&);
159
160     void dumpSource();
161     void dumpSource(PrintStream&);
162
163     void dumpBytecode();
164     void dumpBytecode(PrintStream&);
165     void dumpBytecode(
166         PrintStream&, unsigned bytecodeOffset,
167         const StubInfoMap& = StubInfoMap(), const CallLinkInfoMap& = CallLinkInfoMap());
168     void printStructures(PrintStream&, const Instruction*);
169     void printStructure(PrintStream&, const char* name, const Instruction*, int operand);
170
171     bool isStrictMode() const { return m_isStrictMode; }
172     ECMAMode ecmaMode() const { return isStrictMode() ? StrictMode : NotStrictMode; }
173
174     inline bool isKnownNotImmediate(int index)
175     {
176         if (index == m_thisRegister.offset() && !m_isStrictMode)
177             return true;
178
179         if (isConstantRegisterIndex(index))
180             return getConstant(index).isCell();
181
182         return false;
183     }
184
185     ALWAYS_INLINE bool isTemporaryRegisterIndex(int index)
186     {
187         return index >= m_numVars;
188     }
189
190     enum class RequiredHandler {
191         CatchHandler,
192         AnyHandler
193     };
194     HandlerInfo* handlerForBytecodeOffset(unsigned bytecodeOffset, RequiredHandler = RequiredHandler::AnyHandler);
195     unsigned lineNumberForBytecodeOffset(unsigned bytecodeOffset);
196     unsigned columnNumberForBytecodeOffset(unsigned bytecodeOffset);
197     void expressionRangeForBytecodeOffset(unsigned bytecodeOffset, int& divot,
198                                           int& startOffset, int& endOffset, unsigned& line, unsigned& column);
199
200     void getStubInfoMap(const ConcurrentJITLocker&, StubInfoMap& result);
201     void getStubInfoMap(StubInfoMap& result);
202     
203     void getCallLinkInfoMap(const ConcurrentJITLocker&, CallLinkInfoMap& result);
204     void getCallLinkInfoMap(CallLinkInfoMap& result);
205
206     void getByValInfoMap(const ConcurrentJITLocker&, ByValInfoMap& result);
207     void getByValInfoMap(ByValInfoMap& result);
208     
209 #if ENABLE(JIT)
210     StructureStubInfo* addStubInfo();
211     Bag<StructureStubInfo>::iterator stubInfoBegin() { return m_stubInfos.begin(); }
212     Bag<StructureStubInfo>::iterator stubInfoEnd() { return m_stubInfos.end(); }
213     
214     // O(n) operation. Use getStubInfoMap() unless you really only intend to get one
215     // stub info.
216     StructureStubInfo* findStubInfo(CodeOrigin);
217
218     ByValInfo* addByValInfo();
219
220     CallLinkInfo* addCallLinkInfo();
221     Bag<CallLinkInfo>::iterator callLinkInfosBegin() { return m_callLinkInfos.begin(); }
222     Bag<CallLinkInfo>::iterator callLinkInfosEnd() { return m_callLinkInfos.end(); }
223
224     // This is a slow function call used primarily for compiling OSR exits in the case
225     // that there had been inlining. Chances are if you want to use this, you're really
226     // looking for a CallLinkInfoMap to amortize the cost of calling this.
227     CallLinkInfo* getCallLinkInfoForBytecodeIndex(unsigned bytecodeIndex);
228 #endif // ENABLE(JIT)
229
230     void unlinkIncomingCalls();
231
232 #if ENABLE(JIT)
233     void linkIncomingCall(ExecState* callerFrame, CallLinkInfo*);
234     void linkIncomingPolymorphicCall(ExecState* callerFrame, PolymorphicCallNode*);
235 #endif // ENABLE(JIT)
236
237     void linkIncomingCall(ExecState* callerFrame, LLIntCallLinkInfo*);
238
239     void setJITCodeMap(std::unique_ptr<CompactJITCodeMap> jitCodeMap)
240     {
241         m_jitCodeMap = WTF::move(jitCodeMap);
242     }
243     CompactJITCodeMap* jitCodeMap()
244     {
245         return m_jitCodeMap.get();
246     }
247     
248     unsigned bytecodeOffset(Instruction* returnAddress)
249     {
250         RELEASE_ASSERT(returnAddress >= instructions().begin() && returnAddress < instructions().end());
251         return static_cast<Instruction*>(returnAddress) - instructions().begin();
252     }
253
254     unsigned numberOfInstructions() const { return m_instructions.size(); }
255     RefCountedArray<Instruction>& instructions() { return m_instructions; }
256     const RefCountedArray<Instruction>& instructions() const { return m_instructions; }
257
258     size_t predictedMachineCodeSize();
259
260     bool usesOpcode(OpcodeID);
261
262     unsigned instructionCount() const { return m_instructions.size(); }
263
264     // Exactly equivalent to codeBlock->ownerExecutable()->installCode(codeBlock);
265     void install();
266     
267     // Exactly equivalent to codeBlock->ownerExecutable()->newReplacementCodeBlockFor(codeBlock->specializationKind())
268     PassRefPtr<CodeBlock> newReplacement();
269     
270     void setJITCode(PassRefPtr<JITCode> code)
271     {
272         ASSERT(m_heap->isDeferred());
273         m_heap->reportExtraMemoryAllocated(code->size());
274         ConcurrentJITLocker locker(m_lock);
275         WTF::storeStoreFence(); // This is probably not needed because the lock will also do something similar, but it's good to be paranoid.
276         m_jitCode = code;
277     }
278     PassRefPtr<JITCode> jitCode() { return m_jitCode; }
279     JITCode::JITType jitType() const
280     {
281         JITCode* jitCode = m_jitCode.get();
282         WTF::loadLoadFence();
283         JITCode::JITType result = JITCode::jitTypeFor(jitCode);
284         WTF::loadLoadFence(); // This probably isn't needed. Oh well, paranoia is good.
285         return result;
286     }
287
288     bool hasBaselineJITProfiling() const
289     {
290         return jitType() == JITCode::BaselineJIT;
291     }
292     
293 #if ENABLE(JIT)
294     virtual CodeBlock* replacement() = 0;
295
296     virtual DFG::CapabilityLevel capabilityLevelInternal() = 0;
297     DFG::CapabilityLevel capabilityLevel();
298     DFG::CapabilityLevel capabilityLevelState() { return m_capabilityLevelState; }
299
300     bool hasOptimizedReplacement(JITCode::JITType typeToReplace);
301     bool hasOptimizedReplacement(); // the typeToReplace is my JITType
302 #endif
303
304     void jettison(Profiler::JettisonReason, ReoptimizationMode = DontCountReoptimization, const FireDetail* = nullptr);
305     
306     ExecutableBase* ownerExecutable() const { return m_ownerExecutable.get(); }
307     ScriptExecutable* ownerScriptExecutable() const { return jsCast<ScriptExecutable*>(m_ownerExecutable.get()); }
308
309     void setVM(VM* vm) { m_vm = vm; }
310     VM* vm() { return m_vm; }
311
312     void setThisRegister(VirtualRegister thisRegister) { m_thisRegister = thisRegister; }
313     VirtualRegister thisRegister() const { return m_thisRegister; }
314
315     bool usesEval() const { return m_unlinkedCode->usesEval(); }
316
317     void setScopeRegister(VirtualRegister scopeRegister)
318     {
319         ASSERT(scopeRegister.isLocal() || !scopeRegister.isValid());
320         m_scopeRegister = scopeRegister;
321     }
322
323     VirtualRegister scopeRegister() const
324     {
325         return m_scopeRegister;
326     }
327
328     void setActivationRegister(VirtualRegister activationRegister)
329     {
330         m_lexicalEnvironmentRegister = activationRegister;
331     }
332
333     VirtualRegister activationRegister() const
334     {
335         ASSERT(m_lexicalEnvironmentRegister.isValid());
336         return m_lexicalEnvironmentRegister;
337     }
338
339     VirtualRegister uncheckedActivationRegister()
340     {
341         return m_lexicalEnvironmentRegister;
342     }
343
344     bool needsActivation() const
345     {
346         ASSERT(m_lexicalEnvironmentRegister.isValid() == m_needsActivation);
347         return m_needsActivation;
348     }
349     
350     CodeType codeType() const
351     {
352 #if ENABLE(WEBASSEMBLY)
353         if (m_ownerExecutable->isWebAssemblyExecutable())
354             return FunctionCode;
355 #endif
356         return m_unlinkedCode->codeType();
357     }
358
359     PutPropertySlot::Context putByIdContext() const
360     {
361         if (codeType() == EvalCode)
362             return PutPropertySlot::PutByIdEval;
363         return PutPropertySlot::PutById;
364     }
365
366     SourceProvider* source() const { return m_source.get(); }
367     unsigned sourceOffset() const { return m_sourceOffset; }
368     unsigned firstLineColumnOffset() const { return m_firstLineColumnOffset; }
369
370     size_t numberOfJumpTargets() const { return m_unlinkedCode->numberOfJumpTargets(); }
371     unsigned jumpTarget(int index) const { return m_unlinkedCode->jumpTarget(index); }
372
373     String nameForRegister(VirtualRegister);
374
375     unsigned numberOfArgumentValueProfiles()
376     {
377         ASSERT(m_numParameters >= 0);
378         ASSERT(m_argumentValueProfiles.size() == static_cast<unsigned>(m_numParameters));
379         return m_argumentValueProfiles.size();
380     }
381     ValueProfile* valueProfileForArgument(unsigned argumentIndex)
382     {
383         ValueProfile* result = &m_argumentValueProfiles[argumentIndex];
384         ASSERT(result->m_bytecodeOffset == -1);
385         return result;
386     }
387
388     unsigned numberOfValueProfiles() { return m_valueProfiles.size(); }
389     ValueProfile* valueProfile(int index) { return &m_valueProfiles[index]; }
390     ValueProfile* valueProfileForBytecodeOffset(int bytecodeOffset);
391     SpeculatedType valueProfilePredictionForBytecodeOffset(const ConcurrentJITLocker& locker, int bytecodeOffset)
392     {
393         return valueProfileForBytecodeOffset(bytecodeOffset)->computeUpdatedPrediction(locker);
394     }
395
396     unsigned totalNumberOfValueProfiles()
397     {
398         return numberOfArgumentValueProfiles() + numberOfValueProfiles();
399     }
400     ValueProfile* getFromAllValueProfiles(unsigned index)
401     {
402         if (index < numberOfArgumentValueProfiles())
403             return valueProfileForArgument(index);
404         return valueProfile(index - numberOfArgumentValueProfiles());
405     }
406
407     RareCaseProfile* addRareCaseProfile(int bytecodeOffset)
408     {
409         m_rareCaseProfiles.append(RareCaseProfile(bytecodeOffset));
410         return &m_rareCaseProfiles.last();
411     }
412     unsigned numberOfRareCaseProfiles() { return m_rareCaseProfiles.size(); }
413     RareCaseProfile* rareCaseProfile(int index) { return &m_rareCaseProfiles[index]; }
414     RareCaseProfile* rareCaseProfileForBytecodeOffset(int bytecodeOffset);
415
416     bool likelyToTakeSlowCase(int bytecodeOffset)
417     {
418         if (!hasBaselineJITProfiling())
419             return false;
420         unsigned value = rareCaseProfileForBytecodeOffset(bytecodeOffset)->m_counter;
421         return value >= Options::likelyToTakeSlowCaseMinimumCount();
422     }
423
424     bool couldTakeSlowCase(int bytecodeOffset)
425     {
426         if (!hasBaselineJITProfiling())
427             return false;
428         unsigned value = rareCaseProfileForBytecodeOffset(bytecodeOffset)->m_counter;
429         return value >= Options::couldTakeSlowCaseMinimumCount();
430     }
431
432     RareCaseProfile* addSpecialFastCaseProfile(int bytecodeOffset)
433     {
434         m_specialFastCaseProfiles.append(RareCaseProfile(bytecodeOffset));
435         return &m_specialFastCaseProfiles.last();
436     }
437     unsigned numberOfSpecialFastCaseProfiles() { return m_specialFastCaseProfiles.size(); }
438     RareCaseProfile* specialFastCaseProfile(int index) { return &m_specialFastCaseProfiles[index]; }
439     RareCaseProfile* specialFastCaseProfileForBytecodeOffset(int bytecodeOffset)
440     {
441         return tryBinarySearch<RareCaseProfile, int>(
442             m_specialFastCaseProfiles, m_specialFastCaseProfiles.size(), bytecodeOffset,
443             getRareCaseProfileBytecodeOffset);
444     }
445
446     bool likelyToTakeSpecialFastCase(int bytecodeOffset)
447     {
448         if (!hasBaselineJITProfiling())
449             return false;
450         unsigned specialFastCaseCount = specialFastCaseProfileForBytecodeOffset(bytecodeOffset)->m_counter;
451         return specialFastCaseCount >= Options::likelyToTakeSlowCaseMinimumCount();
452     }
453
454     bool couldTakeSpecialFastCase(int bytecodeOffset)
455     {
456         if (!hasBaselineJITProfiling())
457             return false;
458         unsigned specialFastCaseCount = specialFastCaseProfileForBytecodeOffset(bytecodeOffset)->m_counter;
459         return specialFastCaseCount >= Options::couldTakeSlowCaseMinimumCount();
460     }
461
462     bool likelyToTakeDeepestSlowCase(int bytecodeOffset)
463     {
464         if (!hasBaselineJITProfiling())
465             return false;
466         unsigned slowCaseCount = rareCaseProfileForBytecodeOffset(bytecodeOffset)->m_counter;
467         unsigned specialFastCaseCount = specialFastCaseProfileForBytecodeOffset(bytecodeOffset)->m_counter;
468         unsigned value = slowCaseCount - specialFastCaseCount;
469         return value >= Options::likelyToTakeSlowCaseMinimumCount();
470     }
471
472     bool likelyToTakeAnySlowCase(int bytecodeOffset)
473     {
474         if (!hasBaselineJITProfiling())
475             return false;
476         unsigned slowCaseCount = rareCaseProfileForBytecodeOffset(bytecodeOffset)->m_counter;
477         unsigned specialFastCaseCount = specialFastCaseProfileForBytecodeOffset(bytecodeOffset)->m_counter;
478         unsigned value = slowCaseCount + specialFastCaseCount;
479         return value >= Options::likelyToTakeSlowCaseMinimumCount();
480     }
481
482     unsigned numberOfArrayProfiles() const { return m_arrayProfiles.size(); }
483     const ArrayProfileVector& arrayProfiles() { return m_arrayProfiles; }
484     ArrayProfile* addArrayProfile(unsigned bytecodeOffset)
485     {
486         m_arrayProfiles.append(ArrayProfile(bytecodeOffset));
487         return &m_arrayProfiles.last();
488     }
489     ArrayProfile* getArrayProfile(unsigned bytecodeOffset);
490     ArrayProfile* getOrAddArrayProfile(unsigned bytecodeOffset);
491
492     // Exception handling support
493
494     size_t numberOfExceptionHandlers() const { return m_rareData ? m_rareData->m_exceptionHandlers.size() : 0; }
495     HandlerInfo& exceptionHandler(int index) { RELEASE_ASSERT(m_rareData); return m_rareData->m_exceptionHandlers[index]; }
496
497     bool hasExpressionInfo() { return m_unlinkedCode->hasExpressionInfo(); }
498
499 #if ENABLE(DFG_JIT)
500     Vector<CodeOrigin, 0, UnsafeVectorOverflow>& codeOrigins()
501     {
502         return m_jitCode->dfgCommon()->codeOrigins;
503     }
504     
505     // Having code origins implies that there has been some inlining.
506     bool hasCodeOrigins()
507     {
508         return JITCode::isOptimizingJIT(jitType());
509     }
510         
511     bool canGetCodeOrigin(CallSiteIndex index)
512     {
513         if (!hasCodeOrigins())
514             return false;
515         return index.bits() < codeOrigins().size();
516     }
517
518     CodeOrigin codeOrigin(CallSiteIndex index)
519     {
520         return codeOrigins()[index.bits()];
521     }
522
523     bool addFrequentExitSite(const DFG::FrequentExitSite& site)
524     {
525         ASSERT(JITCode::isBaselineCode(jitType()));
526         ConcurrentJITLocker locker(m_lock);
527         return m_exitProfile.add(locker, site);
528     }
529
530     bool hasExitSite(const ConcurrentJITLocker& locker, const DFG::FrequentExitSite& site) const
531     {
532         return m_exitProfile.hasExitSite(locker, site);
533     }
534     bool hasExitSite(const DFG::FrequentExitSite& site) const
535     {
536         ConcurrentJITLocker locker(m_lock);
537         return hasExitSite(locker, site);
538     }
539
540     DFG::ExitProfile& exitProfile() { return m_exitProfile; }
541
542     CompressedLazyOperandValueProfileHolder& lazyOperandValueProfiles()
543     {
544         return m_lazyOperandValueProfiles;
545     }
546 #endif // ENABLE(DFG_JIT)
547
548     // Constant Pool
549 #if ENABLE(DFG_JIT)
550     size_t numberOfIdentifiers() const { return m_unlinkedCode->numberOfIdentifiers() + numberOfDFGIdentifiers(); }
551     size_t numberOfDFGIdentifiers() const
552     {
553         if (!JITCode::isOptimizingJIT(jitType()))
554             return 0;
555
556         return m_jitCode->dfgCommon()->dfgIdentifiers.size();
557     }
558
559     const Identifier& identifier(int index) const
560     {
561         size_t unlinkedIdentifiers = m_unlinkedCode->numberOfIdentifiers();
562         if (static_cast<unsigned>(index) < unlinkedIdentifiers)
563             return m_unlinkedCode->identifier(index);
564         ASSERT(JITCode::isOptimizingJIT(jitType()));
565         return m_jitCode->dfgCommon()->dfgIdentifiers[index - unlinkedIdentifiers];
566     }
567 #else
568     size_t numberOfIdentifiers() const { return m_unlinkedCode->numberOfIdentifiers(); }
569     const Identifier& identifier(int index) const { return m_unlinkedCode->identifier(index); }
570 #endif
571
572     Vector<WriteBarrier<Unknown>>& constants() { return m_constantRegisters; }
573     Vector<SourceCodeRepresentation>& constantsSourceCodeRepresentation() { return m_constantsSourceCodeRepresentation; }
574     unsigned addConstant(JSValue v)
575     {
576         unsigned result = m_constantRegisters.size();
577         m_constantRegisters.append(WriteBarrier<Unknown>());
578         m_constantRegisters.last().set(m_globalObject->vm(), m_ownerExecutable.get(), v);
579         m_constantsSourceCodeRepresentation.append(SourceCodeRepresentation::Other);
580         return result;
581     }
582
583     unsigned addConstantLazily()
584     {
585         unsigned result = m_constantRegisters.size();
586         m_constantRegisters.append(WriteBarrier<Unknown>());
587         m_constantsSourceCodeRepresentation.append(SourceCodeRepresentation::Other);
588         return result;
589     }
590
591     WriteBarrier<Unknown>& constantRegister(int index) { return m_constantRegisters[index - FirstConstantRegisterIndex]; }
592     ALWAYS_INLINE bool isConstantRegisterIndex(int index) const { return index >= FirstConstantRegisterIndex; }
593     ALWAYS_INLINE JSValue getConstant(int index) const { return m_constantRegisters[index - FirstConstantRegisterIndex].get(); }
594     ALWAYS_INLINE SourceCodeRepresentation constantSourceCodeRepresentation(int index) const { return m_constantsSourceCodeRepresentation[index - FirstConstantRegisterIndex]; }
595
596     FunctionExecutable* functionDecl(int index) { return m_functionDecls[index].get(); }
597     int numberOfFunctionDecls() { return m_functionDecls.size(); }
598     FunctionExecutable* functionExpr(int index) { return m_functionExprs[index].get(); }
599     
600     RegExp* regexp(int index) const { return m_unlinkedCode->regexp(index); }
601
602     unsigned numberOfConstantBuffers() const
603     {
604         if (!m_rareData)
605             return 0;
606         return m_rareData->m_constantBuffers.size();
607     }
608     unsigned addConstantBuffer(const Vector<JSValue>& buffer)
609     {
610         createRareDataIfNecessary();
611         unsigned size = m_rareData->m_constantBuffers.size();
612         m_rareData->m_constantBuffers.append(buffer);
613         return size;
614     }
615
616     Vector<JSValue>& constantBufferAsVector(unsigned index)
617     {
618         ASSERT(m_rareData);
619         return m_rareData->m_constantBuffers[index];
620     }
621     JSValue* constantBuffer(unsigned index)
622     {
623         return constantBufferAsVector(index).data();
624     }
625
626     Heap* heap() const { return m_heap; }
627     JSGlobalObject* globalObject() { return m_globalObject.get(); }
628
629     JSGlobalObject* globalObjectFor(CodeOrigin);
630
631     BytecodeLivenessAnalysis& livenessAnalysis()
632     {
633         {
634             ConcurrentJITLocker locker(m_lock);
635             if (!!m_livenessAnalysis)
636                 return *m_livenessAnalysis;
637         }
638         std::unique_ptr<BytecodeLivenessAnalysis> analysis =
639             std::make_unique<BytecodeLivenessAnalysis>(this);
640         {
641             ConcurrentJITLocker locker(m_lock);
642             if (!m_livenessAnalysis)
643                 m_livenessAnalysis = WTF::move(analysis);
644             return *m_livenessAnalysis;
645         }
646     }
647     
648     void validate();
649
650     // Jump Tables
651
652     size_t numberOfSwitchJumpTables() const { return m_rareData ? m_rareData->m_switchJumpTables.size() : 0; }
653     SimpleJumpTable& addSwitchJumpTable() { createRareDataIfNecessary(); m_rareData->m_switchJumpTables.append(SimpleJumpTable()); return m_rareData->m_switchJumpTables.last(); }
654     SimpleJumpTable& switchJumpTable(int tableIndex) { RELEASE_ASSERT(m_rareData); return m_rareData->m_switchJumpTables[tableIndex]; }
655     void clearSwitchJumpTables()
656     {
657         if (!m_rareData)
658             return;
659         m_rareData->m_switchJumpTables.clear();
660     }
661
662     size_t numberOfStringSwitchJumpTables() const { return m_rareData ? m_rareData->m_stringSwitchJumpTables.size() : 0; }
663     StringJumpTable& addStringSwitchJumpTable() { createRareDataIfNecessary(); m_rareData->m_stringSwitchJumpTables.append(StringJumpTable()); return m_rareData->m_stringSwitchJumpTables.last(); }
664     StringJumpTable& stringSwitchJumpTable(int tableIndex) { RELEASE_ASSERT(m_rareData); return m_rareData->m_stringSwitchJumpTables[tableIndex]; }
665
666     EvalCodeCache& evalCodeCache() { createRareDataIfNecessary(); return m_rareData->m_evalCodeCache; }
667
668     enum ShrinkMode {
669         // Shrink prior to generating machine code that may point directly into vectors.
670         EarlyShrink,
671
672         // Shrink after generating machine code, and after possibly creating new vectors
673         // and appending to others. At this time it is not safe to shrink certain vectors
674         // because we would have generated machine code that references them directly.
675         LateShrink
676     };
677     void shrinkToFit(ShrinkMode);
678
679     // Functions for controlling when JITting kicks in, in a mixed mode
680     // execution world.
681
682     bool checkIfJITThresholdReached()
683     {
684         return m_llintExecuteCounter.checkIfThresholdCrossedAndSet(this);
685     }
686
687     void dontJITAnytimeSoon()
688     {
689         m_llintExecuteCounter.deferIndefinitely();
690     }
691
692     void jitAfterWarmUp()
693     {
694         m_llintExecuteCounter.setNewThreshold(Options::thresholdForJITAfterWarmUp(), this);
695     }
696
697     void jitSoon()
698     {
699         m_llintExecuteCounter.setNewThreshold(Options::thresholdForJITSoon(), this);
700     }
701
702     const BaselineExecutionCounter& llintExecuteCounter() const
703     {
704         return m_llintExecuteCounter;
705     }
706
707     // Functions for controlling when tiered compilation kicks in. This
708     // controls both when the optimizing compiler is invoked and when OSR
709     // entry happens. Two triggers exist: the loop trigger and the return
710     // trigger. In either case, when an addition to m_jitExecuteCounter
711     // causes it to become non-negative, the optimizing compiler is
712     // invoked. This includes a fast check to see if this CodeBlock has
713     // already been optimized (i.e. replacement() returns a CodeBlock
714     // that was optimized with a higher tier JIT than this one). In the
715     // case of the loop trigger, if the optimized compilation succeeds
716     // (or has already succeeded in the past) then OSR is attempted to
717     // redirect program flow into the optimized code.
718
719     // These functions are called from within the optimization triggers,
720     // and are used as a single point at which we define the heuristics
721     // for how much warm-up is mandated before the next optimization
722     // trigger files. All CodeBlocks start out with optimizeAfterWarmUp(),
723     // as this is called from the CodeBlock constructor.
724
725     // When we observe a lot of speculation failures, we trigger a
726     // reoptimization. But each time, we increase the optimization trigger
727     // to avoid thrashing.
728     JS_EXPORT_PRIVATE unsigned reoptimizationRetryCounter() const;
729     void countReoptimization();
730 #if ENABLE(JIT)
731     unsigned numberOfDFGCompiles();
732
733     int32_t codeTypeThresholdMultiplier() const;
734
735     int32_t adjustedCounterValue(int32_t desiredThreshold);
736
737     int32_t* addressOfJITExecuteCounter()
738     {
739         return &m_jitExecuteCounter.m_counter;
740     }
741
742     static ptrdiff_t offsetOfJITExecuteCounter() { return OBJECT_OFFSETOF(CodeBlock, m_jitExecuteCounter) + OBJECT_OFFSETOF(BaselineExecutionCounter, m_counter); }
743     static ptrdiff_t offsetOfJITExecutionActiveThreshold() { return OBJECT_OFFSETOF(CodeBlock, m_jitExecuteCounter) + OBJECT_OFFSETOF(BaselineExecutionCounter, m_activeThreshold); }
744     static ptrdiff_t offsetOfJITExecutionTotalCount() { return OBJECT_OFFSETOF(CodeBlock, m_jitExecuteCounter) + OBJECT_OFFSETOF(BaselineExecutionCounter, m_totalCount); }
745
746     const BaselineExecutionCounter& jitExecuteCounter() const { return m_jitExecuteCounter; }
747
748     unsigned optimizationDelayCounter() const { return m_optimizationDelayCounter; }
749
750     // Check if the optimization threshold has been reached, and if not,
751     // adjust the heuristics accordingly. Returns true if the threshold has
752     // been reached.
753     bool checkIfOptimizationThresholdReached();
754
755     // Call this to force the next optimization trigger to fire. This is
756     // rarely wise, since optimization triggers are typically more
757     // expensive than executing baseline code.
758     void optimizeNextInvocation();
759
760     // Call this to prevent optimization from happening again. Note that
761     // optimization will still happen after roughly 2^29 invocations,
762     // so this is really meant to delay that as much as possible. This
763     // is called if optimization failed, and we expect it to fail in
764     // the future as well.
765     void dontOptimizeAnytimeSoon();
766
767     // Call this to reinitialize the counter to its starting state,
768     // forcing a warm-up to happen before the next optimization trigger
769     // fires. This is called in the CodeBlock constructor. It also
770     // makes sense to call this if an OSR exit occurred. Note that
771     // OSR exit code is code generated, so the value of the execute
772     // counter that this corresponds to is also available directly.
773     void optimizeAfterWarmUp();
774
775     // Call this to force an optimization trigger to fire only after
776     // a lot of warm-up.
777     void optimizeAfterLongWarmUp();
778
779     // Call this to cause an optimization trigger to fire soon, but
780     // not necessarily the next one. This makes sense if optimization
781     // succeeds. Successfuly optimization means that all calls are
782     // relinked to the optimized code, so this only affects call
783     // frames that are still executing this CodeBlock. The value here
784     // is tuned to strike a balance between the cost of OSR entry
785     // (which is too high to warrant making every loop back edge to
786     // trigger OSR immediately) and the cost of executing baseline
787     // code (which is high enough that we don't necessarily want to
788     // have a full warm-up). The intuition for calling this instead of
789     // optimizeNextInvocation() is for the case of recursive functions
790     // with loops. Consider that there may be N call frames of some
791     // recursive function, for a reasonably large value of N. The top
792     // one triggers optimization, and then returns, and then all of
793     // the others return. We don't want optimization to be triggered on
794     // each return, as that would be superfluous. It only makes sense
795     // to trigger optimization if one of those functions becomes hot
796     // in the baseline code.
797     void optimizeSoon();
798
799     void forceOptimizationSlowPathConcurrently();
800
801     void setOptimizationThresholdBasedOnCompilationResult(CompilationResult);
802     
803     uint32_t osrExitCounter() const { return m_osrExitCounter; }
804
805     void countOSRExit() { m_osrExitCounter++; }
806
807     uint32_t* addressOfOSRExitCounter() { return &m_osrExitCounter; }
808
809     static ptrdiff_t offsetOfOSRExitCounter() { return OBJECT_OFFSETOF(CodeBlock, m_osrExitCounter); }
810
811     uint32_t adjustedExitCountThreshold(uint32_t desiredThreshold);
812     uint32_t exitCountThresholdForReoptimization();
813     uint32_t exitCountThresholdForReoptimizationFromLoop();
814     bool shouldReoptimizeNow();
815     bool shouldReoptimizeFromLoopNow();
816 #else // No JIT
817     void optimizeAfterWarmUp() { }
818     unsigned numberOfDFGCompiles() { return 0; }
819 #endif
820
821     bool shouldOptimizeNow();
822     void updateAllValueProfilePredictions();
823     void updateAllArrayPredictions();
824     void updateAllPredictions();
825
826     unsigned frameRegisterCount();
827     int stackPointerOffset();
828
829     bool hasOpDebugForLineAndColumn(unsigned line, unsigned column);
830
831     bool hasDebuggerRequests() const { return m_debuggerRequests; }
832     void* debuggerRequestsAddress() { return &m_debuggerRequests; }
833
834     void addBreakpoint(unsigned numBreakpoints);
835     void removeBreakpoint(unsigned numBreakpoints)
836     {
837         ASSERT(m_numBreakpoints >= numBreakpoints);
838         m_numBreakpoints -= numBreakpoints;
839     }
840
841     enum SteppingMode {
842         SteppingModeDisabled,
843         SteppingModeEnabled
844     };
845     void setSteppingMode(SteppingMode);
846
847     void clearDebuggerRequests()
848     {
849         m_steppingMode = SteppingModeDisabled;
850         m_numBreakpoints = 0;
851     }
852     
853     // FIXME: Make these remaining members private.
854
855     int m_numCalleeRegisters;
856     int m_numVars;
857     bool m_isConstructor : 1;
858     
859     // This is intentionally public; it's the responsibility of anyone doing any
860     // of the following to hold the lock:
861     //
862     // - Modifying any inline cache in this code block.
863     //
864     // - Quering any inline cache in this code block, from a thread other than
865     //   the main thread.
866     //
867     // Additionally, it's only legal to modify the inline cache on the main
868     // thread. This means that the main thread can query the inline cache without
869     // locking. This is crucial since executing the inline cache is effectively
870     // "querying" it.
871     //
872     // Another exception to the rules is that the GC can do whatever it wants
873     // without holding any locks, because the GC is guaranteed to wait until any
874     // concurrent compilation threads finish what they're doing.
875     mutable ConcurrentJITLock m_lock;
876     
877     bool m_shouldAlwaysBeInlined; // Not a bitfield because the JIT wants to store to it.
878     bool m_allTransitionsHaveBeenMarked : 1; // Initialized and used on every GC.
879     
880     bool m_didFailFTLCompilation : 1;
881     bool m_hasBeenCompiledWithFTL : 1;
882
883     // Internal methods for use by validation code. It would be private if it wasn't
884     // for the fact that we use it from anonymous namespaces.
885     void beginValidationDidFail();
886     NO_RETURN_DUE_TO_CRASH void endValidationDidFail();
887
888     bool isKnownToBeLiveDuringGC(); // Will only return valid results when called during GC. Assumes that you've already established that the owner executable is live.
889
890     struct RareData {
891         WTF_MAKE_FAST_ALLOCATED;
892     public:
893         Vector<HandlerInfo> m_exceptionHandlers;
894
895         // Buffers used for large array literals
896         Vector<Vector<JSValue>> m_constantBuffers;
897
898         // Jump Tables
899         Vector<SimpleJumpTable> m_switchJumpTables;
900         Vector<StringJumpTable> m_stringSwitchJumpTables;
901
902         EvalCodeCache m_evalCodeCache;
903     };
904
905 protected:
906     virtual void visitWeakReferences(SlotVisitor&) override;
907     virtual void finalizeUnconditionally() override;
908
909 #if ENABLE(DFG_JIT)
910     void tallyFrequentExitSites();
911 #else
912     void tallyFrequentExitSites() { }
913 #endif
914
915 private:
916     friend class CodeBlockSet;
917     
918     CodeBlock* specialOSREntryBlockOrNull();
919     
920     void noticeIncomingCall(ExecState* callerFrame);
921     
922     double optimizationThresholdScalingFactor();
923
924     void updateAllPredictionsAndCountLiveness(unsigned& numberOfLiveNonArgumentValueProfiles, unsigned& numberOfSamplesInProfiles);
925
926     void setConstantRegisters(const Vector<WriteBarrier<Unknown>>& constants, const Vector<SourceCodeRepresentation>& constantsSourceCodeRepresentation)
927     {
928         ASSERT(constants.size() == constantsSourceCodeRepresentation.size());
929         size_t count = constants.size();
930         m_constantRegisters.resizeToFit(count);
931         for (size_t i = 0; i < count; i++)
932             m_constantRegisters[i].set(*m_vm, ownerExecutable(), constants[i].get());
933         m_constantsSourceCodeRepresentation = constantsSourceCodeRepresentation;
934     }
935
936     void replaceConstant(int index, JSValue value)
937     {
938         ASSERT(isConstantRegisterIndex(index) && static_cast<size_t>(index - FirstConstantRegisterIndex) < m_constantRegisters.size());
939         m_constantRegisters[index - FirstConstantRegisterIndex].set(m_globalObject->vm(), m_ownerExecutable.get(), value);
940     }
941
942     void dumpBytecode(
943         PrintStream&, ExecState*, const Instruction* begin, const Instruction*&,
944         const StubInfoMap& = StubInfoMap(), const CallLinkInfoMap& = CallLinkInfoMap());
945
946     CString registerName(int r) const;
947     CString constantName(int index) const;
948     void printUnaryOp(PrintStream&, ExecState*, int location, const Instruction*&, const char* op);
949     void printBinaryOp(PrintStream&, ExecState*, int location, const Instruction*&, const char* op);
950     void printConditionalJump(PrintStream&, ExecState*, const Instruction*, const Instruction*&, int location, const char* op);
951     void printGetByIdOp(PrintStream&, ExecState*, int location, const Instruction*&);
952     void printGetByIdCacheStatus(PrintStream&, ExecState*, int location, const StubInfoMap&);
953     enum CacheDumpMode { DumpCaches, DontDumpCaches };
954     void printCallOp(PrintStream&, ExecState*, int location, const Instruction*&, const char* op, CacheDumpMode, bool& hasPrintedProfiling, const CallLinkInfoMap&);
955     void printPutByIdOp(PrintStream&, ExecState*, int location, const Instruction*&, const char* op);
956     void printPutByIdCacheStatus(PrintStream&, ExecState*, int location, const StubInfoMap&);
957     void printLocationAndOp(PrintStream&, ExecState*, int location, const Instruction*&, const char* op);
958     void printLocationOpAndRegisterOperand(PrintStream&, ExecState*, int location, const Instruction*& it, const char* op, int operand);
959
960     void beginDumpProfiling(PrintStream&, bool& hasPrintedProfiling);
961     void dumpValueProfiling(PrintStream&, const Instruction*&, bool& hasPrintedProfiling);
962     void dumpArrayProfiling(PrintStream&, const Instruction*&, bool& hasPrintedProfiling);
963     void dumpRareCaseProfile(PrintStream&, const char* name, RareCaseProfile*, bool& hasPrintedProfiling);
964         
965     bool shouldImmediatelyAssumeLivenessDuringScan();
966     
967     void propagateTransitions(SlotVisitor&);
968     void determineLiveness(SlotVisitor&);
969         
970     void stronglyVisitStrongReferences(SlotVisitor&);
971     void stronglyVisitWeakReferences(SlotVisitor&);
972
973     void createRareDataIfNecessary()
974     {
975         if (!m_rareData)
976             m_rareData = std::make_unique<RareData>();
977     }
978
979     void insertBasicBlockBoundariesForControlFlowProfiler(Vector<Instruction, 0, UnsafeVectorOverflow>&);
980
981     WriteBarrier<UnlinkedCodeBlock> m_unlinkedCode;
982     int m_numParameters;
983     union {
984         unsigned m_debuggerRequests;
985         struct {
986             unsigned m_hasDebuggerStatement : 1;
987             unsigned m_steppingMode : 1;
988             unsigned m_numBreakpoints : 30;
989         };
990     };
991     WriteBarrier<ExecutableBase> m_ownerExecutable;
992     VM* m_vm;
993
994     RefCountedArray<Instruction> m_instructions;
995     VirtualRegister m_thisRegister;
996     VirtualRegister m_scopeRegister;
997     VirtualRegister m_lexicalEnvironmentRegister;
998
999     bool m_isStrictMode;
1000     bool m_needsActivation;
1001
1002     bool m_mayBeExecuting;
1003     bool m_isStronglyReferenced;
1004     Atomic<bool> m_visitAggregateHasBeenCalled;
1005
1006     RefPtr<SourceProvider> m_source;
1007     unsigned m_sourceOffset;
1008     unsigned m_firstLineColumnOffset;
1009     unsigned m_codeType;
1010
1011     Vector<LLIntCallLinkInfo> m_llintCallLinkInfos;
1012     SentinelLinkedList<LLIntCallLinkInfo, BasicRawSentinelNode<LLIntCallLinkInfo>> m_incomingLLIntCalls;
1013     RefPtr<JITCode> m_jitCode;
1014 #if ENABLE(JIT)
1015     Bag<StructureStubInfo> m_stubInfos;
1016     Bag<ByValInfo> m_byValInfos;
1017     Bag<CallLinkInfo> m_callLinkInfos;
1018     SentinelLinkedList<CallLinkInfo, BasicRawSentinelNode<CallLinkInfo>> m_incomingCalls;
1019     SentinelLinkedList<PolymorphicCallNode, BasicRawSentinelNode<PolymorphicCallNode>> m_incomingPolymorphicCalls;
1020 #endif
1021     std::unique_ptr<CompactJITCodeMap> m_jitCodeMap;
1022 #if ENABLE(DFG_JIT)
1023     // This is relevant to non-DFG code blocks that serve as the profiled code block
1024     // for DFG code blocks.
1025     DFG::ExitProfile m_exitProfile;
1026     CompressedLazyOperandValueProfileHolder m_lazyOperandValueProfiles;
1027 #endif
1028     Vector<ValueProfile> m_argumentValueProfiles;
1029     Vector<ValueProfile> m_valueProfiles;
1030     SegmentedVector<RareCaseProfile, 8> m_rareCaseProfiles;
1031     SegmentedVector<RareCaseProfile, 8> m_specialFastCaseProfiles;
1032     Vector<ArrayAllocationProfile> m_arrayAllocationProfiles;
1033     ArrayProfileVector m_arrayProfiles;
1034     Vector<ObjectAllocationProfile> m_objectAllocationProfiles;
1035
1036     // Constant Pool
1037     COMPILE_ASSERT(sizeof(Register) == sizeof(WriteBarrier<Unknown>), Register_must_be_same_size_as_WriteBarrier_Unknown);
1038     // TODO: This could just be a pointer to m_unlinkedCodeBlock's data, but the DFG mutates
1039     // it, so we're stuck with it for now.
1040     Vector<WriteBarrier<Unknown>> m_constantRegisters;
1041     Vector<SourceCodeRepresentation> m_constantsSourceCodeRepresentation;
1042     Vector<WriteBarrier<FunctionExecutable>> m_functionDecls;
1043     Vector<WriteBarrier<FunctionExecutable>> m_functionExprs;
1044
1045     RefPtr<CodeBlock> m_alternative;
1046     
1047     BaselineExecutionCounter m_llintExecuteCounter;
1048
1049     BaselineExecutionCounter m_jitExecuteCounter;
1050     int32_t m_totalJITExecutions;
1051     uint32_t m_osrExitCounter;
1052     uint16_t m_optimizationDelayCounter;
1053     uint16_t m_reoptimizationRetryCounter;
1054     
1055     mutable CodeBlockHash m_hash;
1056
1057     std::unique_ptr<BytecodeLivenessAnalysis> m_livenessAnalysis;
1058
1059     std::unique_ptr<RareData> m_rareData;
1060 #if ENABLE(JIT)
1061     DFG::CapabilityLevel m_capabilityLevelState;
1062 #endif
1063 };
1064
1065 // Program code is not marked by any function, so we make the global object
1066 // responsible for marking it.
1067
1068 class GlobalCodeBlock : public CodeBlock {
1069 protected:
1070     GlobalCodeBlock(CopyParsedBlockTag, GlobalCodeBlock& other)
1071     : CodeBlock(CopyParsedBlock, other)
1072     {
1073     }
1074         
1075     GlobalCodeBlock(ScriptExecutable* ownerExecutable, UnlinkedCodeBlock* unlinkedCodeBlock, JSScope* scope, PassRefPtr<SourceProvider> sourceProvider, unsigned sourceOffset, unsigned firstLineColumnOffset)
1076         : CodeBlock(ownerExecutable, unlinkedCodeBlock, scope, sourceProvider, sourceOffset, firstLineColumnOffset)
1077     {
1078     }
1079 };
1080
1081 class ProgramCodeBlock : public GlobalCodeBlock {
1082 public:
1083     ProgramCodeBlock(CopyParsedBlockTag, ProgramCodeBlock& other)
1084     : GlobalCodeBlock(CopyParsedBlock, other)
1085     {
1086     }
1087
1088     ProgramCodeBlock(ProgramExecutable* ownerExecutable, UnlinkedProgramCodeBlock* unlinkedCodeBlock, JSScope* scope, PassRefPtr<SourceProvider> sourceProvider, unsigned firstLineColumnOffset)
1089         : GlobalCodeBlock(ownerExecutable, unlinkedCodeBlock, scope, sourceProvider, 0, firstLineColumnOffset)
1090     {
1091     }
1092
1093 #if ENABLE(JIT)
1094 protected:
1095     virtual CodeBlock* replacement() override;
1096     virtual DFG::CapabilityLevel capabilityLevelInternal() override;
1097 #endif
1098 };
1099
1100 class ModuleProgramCodeBlock : public GlobalCodeBlock {
1101 public:
1102     ModuleProgramCodeBlock(CopyParsedBlockTag, ModuleProgramCodeBlock& other)
1103         : GlobalCodeBlock(CopyParsedBlock, other)
1104     {
1105     }
1106
1107     ModuleProgramCodeBlock(ModuleProgramExecutable* ownerExecutable, UnlinkedModuleProgramCodeBlock* unlinkedCodeBlock, JSScope* scope, PassRefPtr<SourceProvider> sourceProvider, unsigned firstLineColumnOffset)
1108         : GlobalCodeBlock(ownerExecutable, unlinkedCodeBlock, scope, sourceProvider, 0, firstLineColumnOffset)
1109     {
1110     }
1111
1112 #if ENABLE(JIT)
1113 protected:
1114     virtual CodeBlock* replacement() override;
1115     virtual DFG::CapabilityLevel capabilityLevelInternal() override;
1116 #endif
1117 };
1118
1119 class EvalCodeBlock : public GlobalCodeBlock {
1120 public:
1121     EvalCodeBlock(CopyParsedBlockTag, EvalCodeBlock& other)
1122         : GlobalCodeBlock(CopyParsedBlock, other)
1123     {
1124     }
1125         
1126     EvalCodeBlock(EvalExecutable* ownerExecutable, UnlinkedEvalCodeBlock* unlinkedCodeBlock, JSScope* scope, PassRefPtr<SourceProvider> sourceProvider)
1127         : GlobalCodeBlock(ownerExecutable, unlinkedCodeBlock, scope, sourceProvider, 0, 1)
1128     {
1129     }
1130     
1131     const Identifier& variable(unsigned index) { return unlinkedEvalCodeBlock()->variable(index); }
1132     unsigned numVariables() { return unlinkedEvalCodeBlock()->numVariables(); }
1133     
1134 #if ENABLE(JIT)
1135 protected:
1136     virtual CodeBlock* replacement() override;
1137     virtual DFG::CapabilityLevel capabilityLevelInternal() override;
1138 #endif
1139     
1140 private:
1141     UnlinkedEvalCodeBlock* unlinkedEvalCodeBlock() const { return jsCast<UnlinkedEvalCodeBlock*>(unlinkedCodeBlock()); }
1142 };
1143
1144 class FunctionCodeBlock : public CodeBlock {
1145 public:
1146     FunctionCodeBlock(CopyParsedBlockTag, FunctionCodeBlock& other)
1147         : CodeBlock(CopyParsedBlock, other)
1148     {
1149     }
1150
1151     FunctionCodeBlock(FunctionExecutable* ownerExecutable, UnlinkedFunctionCodeBlock* unlinkedCodeBlock, JSScope* scope, PassRefPtr<SourceProvider> sourceProvider, unsigned sourceOffset, unsigned firstLineColumnOffset)
1152         : CodeBlock(ownerExecutable, unlinkedCodeBlock, scope, sourceProvider, sourceOffset, firstLineColumnOffset)
1153     {
1154     }
1155     
1156 #if ENABLE(JIT)
1157 protected:
1158     virtual CodeBlock* replacement() override;
1159     virtual DFG::CapabilityLevel capabilityLevelInternal() override;
1160 #endif
1161 };
1162
1163 #if ENABLE(WEBASSEMBLY)
1164 class WebAssemblyCodeBlock : public CodeBlock {
1165 public:
1166     WebAssemblyCodeBlock(CopyParsedBlockTag, WebAssemblyCodeBlock& other)
1167         : CodeBlock(CopyParsedBlock, other)
1168     {
1169     }
1170
1171     WebAssemblyCodeBlock(WebAssemblyExecutable* ownerExecutable, VM& vm, JSGlobalObject* globalObject)
1172         : CodeBlock(ownerExecutable, vm, globalObject)
1173     {
1174     }
1175
1176 #if ENABLE(JIT)
1177 protected:
1178     virtual CodeBlock* replacement() override;
1179     virtual DFG::CapabilityLevel capabilityLevelInternal() override;
1180 #endif
1181 };
1182 #endif
1183
1184 inline Register& ExecState::r(int index)
1185 {
1186     CodeBlock* codeBlock = this->codeBlock();
1187     if (codeBlock->isConstantRegisterIndex(index))
1188         return *reinterpret_cast<Register*>(&codeBlock->constantRegister(index));
1189     return this[index];
1190 }
1191
1192 inline Register& ExecState::r(VirtualRegister reg)
1193 {
1194     return r(reg.offset());
1195 }
1196
1197 inline Register& ExecState::uncheckedR(int index)
1198 {
1199     RELEASE_ASSERT(index < FirstConstantRegisterIndex);
1200     return this[index];
1201 }
1202
1203 inline Register& ExecState::uncheckedR(VirtualRegister reg)
1204 {
1205     return uncheckedR(reg.offset());
1206 }
1207
1208 inline void CodeBlockSet::clearMarks(CodeBlock* codeBlock)
1209 {
1210     if (!codeBlock)
1211         return;
1212
1213     codeBlock->m_mayBeExecuting = false;
1214     codeBlock->m_isStronglyReferenced = false;
1215     codeBlock->m_visitAggregateHasBeenCalled.store(false, std::memory_order_relaxed);
1216 }
1217
1218 inline void CodeBlockSet::mark(void* candidateCodeBlock)
1219 {
1220     // We have to check for 0 and -1 because those are used by the HashMap as markers.
1221     uintptr_t value = reinterpret_cast<uintptr_t>(candidateCodeBlock);
1222     
1223     // This checks for both of those nasty cases in one go.
1224     // 0 + 1 = 1
1225     // -1 + 1 = 0
1226     if (value + 1 <= 1)
1227         return;
1228
1229     CodeBlock* codeBlock = static_cast<CodeBlock*>(candidateCodeBlock); 
1230     if (!m_oldCodeBlocks.contains(codeBlock) && !m_newCodeBlocks.contains(codeBlock))
1231         return;
1232
1233     mark(codeBlock);
1234 }
1235
1236 inline void CodeBlockSet::mark(CodeBlock* codeBlock)
1237 {
1238     if (!codeBlock)
1239         return;
1240     
1241     if (codeBlock->m_mayBeExecuting)
1242         return;
1243     
1244     codeBlock->m_mayBeExecuting = true;
1245
1246     // We might not have cleared the marks for this CodeBlock, but we need to visit it.
1247     codeBlock->m_visitAggregateHasBeenCalled.store(false, std::memory_order_relaxed);
1248
1249     // For simplicity, we don't attempt to jettison code blocks during GC if
1250     // they are executing. Instead we strongly mark their weak references to
1251     // allow them to continue to execute soundly.
1252     codeBlock->m_isStronglyReferenced = true;
1253
1254 #if ENABLE(GGC)
1255     m_currentlyExecuting.append(codeBlock);
1256 #endif
1257 }
1258
1259 template <typename Functor> inline void ScriptExecutable::forEachCodeBlock(Functor&& functor)
1260 {
1261     switch (type()) {
1262     case ProgramExecutableType: {
1263         if (CodeBlock* codeBlock = jsCast<ProgramExecutable*>(this)->m_programCodeBlock.get())
1264             codeBlock->forEachRelatedCodeBlock(std::forward<Functor>(functor));
1265         break;
1266     }
1267         
1268     case EvalExecutableType: {
1269         if (CodeBlock* codeBlock = jsCast<EvalExecutable*>(this)->m_evalCodeBlock.get())
1270             codeBlock->forEachRelatedCodeBlock(std::forward<Functor>(functor));
1271         break;
1272     }
1273         
1274     case FunctionExecutableType: {
1275         Functor f(std::forward<Functor>(functor));
1276         FunctionExecutable* executable = jsCast<FunctionExecutable*>(this);
1277         if (CodeBlock* codeBlock = executable->m_codeBlockForCall.get())
1278             codeBlock->forEachRelatedCodeBlock(f);
1279         if (CodeBlock* codeBlock = executable->m_codeBlockForConstruct.get())
1280             codeBlock->forEachRelatedCodeBlock(f);
1281         break;
1282     }
1283     default:
1284         RELEASE_ASSERT_NOT_REACHED();
1285     }
1286 }
1287
1288 } // namespace JSC
1289
1290 #endif // CodeBlock_h