60c0c5514e037b18f653563711ebe58f051a1231
[WebKit-https.git] / Source / JavaScriptCore / jit / GCAwareJITStubRoutine.cpp
1 /*
2  * Copyright (C) 2012 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #include "config.h"
27 #include "GCAwareJITStubRoutine.h"
28
29 #if ENABLE(JIT)
30
31 #include "CodeBlock.h"
32 #include "DFGCommonData.h"
33 #include "Heap.h"
34 #include "VM.h"
35 #include "JSCInlines.h"
36 #include "SlotVisitor.h"
37 #include "Structure.h"
38
39 namespace JSC {
40
41 GCAwareJITStubRoutine::GCAwareJITStubRoutine(
42     const MacroAssemblerCodeRef& code, VM& vm)
43     : JITStubRoutine(code)
44     , m_mayBeExecuting(false)
45     , m_isJettisoned(false)
46 {
47     vm.heap.m_jitStubRoutines.add(this);
48 }
49
50 GCAwareJITStubRoutine::~GCAwareJITStubRoutine() { }
51
52 void GCAwareJITStubRoutine::observeZeroRefCount()
53 {
54     if (m_isJettisoned) {
55         // This case is needed for when the system shuts down. It may be that
56         // the JIT stub routine set gets deleted before we get around to deleting
57         // this guy. In that case the GC informs us that we're jettisoned already
58         // and that we should delete ourselves as soon as the ref count reaches
59         // zero.
60         delete this;
61         return;
62     }
63     
64     RELEASE_ASSERT(!m_refCount);
65
66     m_isJettisoned = true;
67 }
68
69 void GCAwareJITStubRoutine::deleteFromGC()
70 {
71     ASSERT(m_isJettisoned);
72     ASSERT(!m_refCount);
73     ASSERT(!m_mayBeExecuting);
74     
75     delete this;
76 }
77
78 void GCAwareJITStubRoutine::markRequiredObjectsInternal(SlotVisitor&)
79 {
80 }
81
82 MarkingGCAwareJITStubRoutineWithOneObject::MarkingGCAwareJITStubRoutineWithOneObject(
83     const MacroAssemblerCodeRef& code, VM& vm, const JSCell* owner,
84     JSCell* object)
85     : GCAwareJITStubRoutine(code, vm)
86     , m_object(vm, owner, object)
87 {
88 }
89
90 MarkingGCAwareJITStubRoutineWithOneObject::~MarkingGCAwareJITStubRoutineWithOneObject()
91 {
92 }
93
94 void MarkingGCAwareJITStubRoutineWithOneObject::markRequiredObjectsInternal(SlotVisitor& visitor)
95 {
96     visitor.append(&m_object);
97 }
98
99
100 GCAwareJITStubRoutineWithExceptionHandler::GCAwareJITStubRoutineWithExceptionHandler(
101     const MacroAssemblerCodeRef& code, VM& vm, 
102     CodeBlock* codeBlockForExceptionHandlers, CallSiteIndex exceptionHandlerCallSiteIndex)
103     : GCAwareJITStubRoutine(code, vm)
104     , m_codeBlockWithExceptionHandler(codeBlockForExceptionHandlers)
105     , m_exceptionHandlerCallSiteIndex(exceptionHandlerCallSiteIndex)
106 {
107     RELEASE_ASSERT(m_codeBlockWithExceptionHandler);
108     ASSERT(!!m_codeBlockWithExceptionHandler->handlerForIndex(exceptionHandlerCallSiteIndex.bits()));
109 }
110
111 void GCAwareJITStubRoutineWithExceptionHandler::aboutToDie()
112 {
113     m_codeBlockWithExceptionHandler = nullptr;
114 }
115
116 void GCAwareJITStubRoutineWithExceptionHandler::observeZeroRefCount()
117 {
118 #if ENABLE(DFG_JIT)
119     if (m_codeBlockWithExceptionHandler) {
120         m_codeBlockWithExceptionHandler->jitCode()->dfgCommon()->removeCallSiteIndex(m_exceptionHandlerCallSiteIndex);
121         m_codeBlockWithExceptionHandler->removeExceptionHandlerForCallSite(m_exceptionHandlerCallSiteIndex);
122         m_codeBlockWithExceptionHandler = nullptr;
123     }
124 #endif
125
126     Base::observeZeroRefCount();
127 }
128
129
130 PassRefPtr<JITStubRoutine> createJITStubRoutine(
131     const MacroAssemblerCodeRef& code,
132     VM& vm,
133     const JSCell* owner,
134     bool makesCalls,
135     JSCell* object,
136     CodeBlock* codeBlockForExceptionHandlers,
137     CallSiteIndex exceptionHandlerCallSiteIndex)
138 {
139     if (!makesCalls)
140         return adoptRef(new JITStubRoutine(code));
141     
142     if (codeBlockForExceptionHandlers) {
143         RELEASE_ASSERT(!object); // We're not a marking stub routine.
144         RELEASE_ASSERT(JITCode::isOptimizingJIT(codeBlockForExceptionHandlers->jitType()));
145         return static_pointer_cast<JITStubRoutine>(
146             adoptRef(new GCAwareJITStubRoutineWithExceptionHandler(code, vm, codeBlockForExceptionHandlers, exceptionHandlerCallSiteIndex)));
147     }
148
149     if (!object) {
150         return static_pointer_cast<JITStubRoutine>(
151             adoptRef(new GCAwareJITStubRoutine(code, vm)));
152     }
153     
154     return static_pointer_cast<JITStubRoutine>(
155         adoptRef(new MarkingGCAwareJITStubRoutineWithOneObject(code, vm, owner, object)));
156 }
157
158 } // namespace JSC
159
160 #endif // ENABLE(JIT)
161