Add checked casts for ScriptExecutionContext.
[WebKit-https.git] / Source / WebCore / bindings / js / JSEventListener.cpp
1 /*
2  *  Copyright (C) 2001 Peter Kelly (pmk@post.com)
3  *  Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2013 Apple Inc. All Rights Reserved.
4  *
5  *  This library is free software; you can redistribute it and/or
6  *  modify it under the terms of the GNU Lesser General Public
7  *  License as published by the Free Software Foundation; either
8  *  version 2 of the License, or (at your option) any later version.
9  *
10  *  This library is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  *  Lesser General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Lesser General Public
16  *  License along with this library; if not, write to the Free Software
17  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
18  */
19
20 #include "config.h"
21 #include "JSEventListener.h"
22
23 #include "BeforeUnloadEvent.h"
24 #include "Event.h"
25 #include "Frame.h"
26 #include "InspectorCounters.h"
27 #include "JSEvent.h"
28 #include "JSEventTarget.h"
29 #include "JSMainThreadExecState.h"
30 #include "JSMainThreadExecStateInstrumentation.h"
31 #include "ScriptController.h"
32 #include "WorkerGlobalScope.h"
33 #include <runtime/ExceptionHelpers.h>
34 #include <runtime/JSLock.h>
35 #include <runtime/VMEntryScope.h>
36 #include <wtf/Ref.h>
37 #include <wtf/RefCountedLeakCounter.h>
38
39 using namespace JSC;
40
41 namespace WebCore {
42
43 JSEventListener::JSEventListener(JSObject* function, JSObject* wrapper, bool isAttribute, DOMWrapperWorld& isolatedWorld)
44     : EventListener(JSEventListenerType)
45     , m_wrapper(wrapper)
46     , m_isAttribute(isAttribute)
47     , m_isolatedWorld(&isolatedWorld)
48 {
49     if (wrapper) {
50         JSC::Heap::heap(wrapper)->writeBarrier(wrapper, function);
51         m_jsFunction = JSC::Weak<JSC::JSObject>(function);
52     } else
53         ASSERT(!function);
54 #if ENABLE(INSPECTOR)
55     ThreadLocalInspectorCounters::current().incrementCounter(ThreadLocalInspectorCounters::JSEventListenerCounter);
56 #endif
57 }
58
59 JSEventListener::~JSEventListener()
60 {
61 #if ENABLE(INSPECTOR)
62     ThreadLocalInspectorCounters::current().decrementCounter(ThreadLocalInspectorCounters::JSEventListenerCounter);
63 #endif
64 }
65
66 JSObject* JSEventListener::initializeJSFunction(ScriptExecutionContext*) const
67 {
68     return 0;
69 }
70
71 void JSEventListener::visitJSFunction(SlotVisitor& visitor)
72 {
73     // If m_wrapper is 0, then m_jsFunction is zombied, and should never be accessed.
74     if (!m_wrapper)
75         return;
76
77     visitor.appendUnbarrieredWeak(&m_jsFunction);
78 }
79
80 void JSEventListener::handleEvent(ScriptExecutionContext* scriptExecutionContext, Event* event)
81 {
82     ASSERT(scriptExecutionContext);
83     if (!scriptExecutionContext || scriptExecutionContext->isJSExecutionForbidden())
84         return;
85
86     JSLockHolder lock(scriptExecutionContext->vm());
87
88     JSObject* jsFunction = this->jsFunction(scriptExecutionContext);
89     if (!jsFunction)
90         return;
91
92     JSDOMGlobalObject* globalObject = toJSDOMGlobalObject(scriptExecutionContext, *m_isolatedWorld);
93     if (!globalObject)
94         return;
95
96     if (scriptExecutionContext->isDocument()) {
97         JSDOMWindow* window = jsCast<JSDOMWindow*>(globalObject);
98         if (!window->impl().isCurrentlyDisplayedInFrame())
99             return;
100         // FIXME: Is this check needed for other contexts?
101         ScriptController& script = window->impl().frame()->script();
102         if (!script.canExecuteScripts(AboutToExecuteScript) || script.isPaused())
103             return;
104     }
105
106     ExecState* exec = globalObject->globalExec();
107     JSValue handleEventFunction = jsFunction;
108
109     CallData callData;
110     CallType callType = getCallData(handleEventFunction, callData);
111     // If jsFunction is not actually a function, see if it implements the EventListener interface and use that
112     if (callType == CallTypeNone) {
113         handleEventFunction = jsFunction->get(exec, Identifier(exec, "handleEvent"));
114         callType = getCallData(handleEventFunction, callData);
115     }
116
117     if (callType != CallTypeNone) {
118         Ref<JSEventListener> protect(*this);
119
120         MarkedArgumentBuffer args;
121         args.append(toJS(exec, globalObject, event));
122
123         Event* savedEvent = globalObject->currentEvent();
124         globalObject->setCurrentEvent(event);
125
126         VM& vm = globalObject->vm();
127         VMEntryScope entryScope(vm, vm.entryScope ? vm.entryScope->globalObject() : globalObject);
128
129         InspectorInstrumentationCookie cookie = JSMainThreadExecState::instrumentFunctionCall(scriptExecutionContext, callType, callData);
130
131         JSValue thisValue = handleEventFunction == jsFunction ? toJS(exec, globalObject, event->currentTarget()) : jsFunction;
132         JSValue retval = scriptExecutionContext->isDocument()
133             ? JSMainThreadExecState::call(exec, handleEventFunction, callType, callData, thisValue, args)
134             : JSC::call(exec, handleEventFunction, callType, callData, thisValue, args);
135
136         InspectorInstrumentation::didCallFunction(cookie, scriptExecutionContext);
137
138         globalObject->setCurrentEvent(savedEvent);
139
140         if (scriptExecutionContext->isWorkerGlobalScope()) {
141             bool terminatorCausedException = (exec->hadException() && isTerminatedExecutionException(exec->exception()));
142             if (terminatorCausedException || vm.watchdog.didFire())
143                 toWorkerGlobalScope(scriptExecutionContext)->script()->forbidExecution();
144         }
145
146         if (exec->hadException()) {
147             event->target()->uncaughtExceptionInEventHandler();
148             reportCurrentException(exec);
149         } else {
150             if (!retval.isUndefinedOrNull() && event->isBeforeUnloadEvent())
151                 toBeforeUnloadEvent(event)->setReturnValue(retval.toString(exec)->value(exec));
152             if (m_isAttribute) {
153                 if (retval.isFalse())
154                     event->preventDefault();
155             }
156         }
157     }
158 }
159
160 bool JSEventListener::virtualisAttribute() const
161 {
162     return m_isAttribute;
163 }
164
165 bool JSEventListener::operator==(const EventListener& listener)
166 {
167     if (const JSEventListener* jsEventListener = JSEventListener::cast(&listener))
168         return m_jsFunction == jsEventListener->m_jsFunction && m_isAttribute == jsEventListener->m_isAttribute;
169     return false;
170 }
171
172 } // namespace WebCore