Add checked casts for ScriptExecutionContext.
[WebKit-https.git] / Source / WebCore / bindings / js / ScheduledAction.cpp
1 /*
2  *  Copyright (C) 2000 Harri Porten (porten@kde.org)
3  *  Copyright (C) 2006 Jon Shier (jshier@iastate.edu)
4  *  Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008, 2009 Apple Inc. All rights reseved.
5  *  Copyright (C) 2006 Alexey Proskuryakov (ap@webkit.org)
6  *  Copyright (C) 2009 Google Inc. All rights reseved.
7  *
8  *  This library is free software; you can redistribute it and/or
9  *  modify it under the terms of the GNU Lesser General Public
10  *  License as published by the Free Software Foundation; either
11  *  version 2 of the License, or (at your option) any later version.
12  *
13  *  This library is distributed in the hope that it will be useful,
14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  *  Lesser General Public License for more details.
17  *
18  *  You should have received a copy of the GNU Lesser General Public
19  *  License along with this library; if not, write to the Free Software
20  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301
21  *  USA
22  */
23
24 #include "config.h"
25 #include "ScheduledAction.h"
26
27 #include "ContentSecurityPolicy.h"
28 #include "DOMWindow.h"
29 #include "Document.h"
30 #include "Frame.h"
31 #include "FrameLoader.h"
32 #include "JSDOMBinding.h"
33 #include "JSDOMWindow.h"
34 #include "JSMainThreadExecState.h"
35 #include "JSMainThreadExecStateInstrumentation.h"
36 #include "JSWorkerGlobalScope.h"
37 #include "ScriptController.h"
38 #include "ScriptExecutionContext.h"
39 #include "ScriptSourceCode.h"
40 #include "WorkerGlobalScope.h"
41 #include "WorkerThread.h"
42 #include <bindings/ScriptValue.h>
43 #include <runtime/JSLock.h>
44
45 using namespace JSC;
46
47 namespace WebCore {
48
49 PassOwnPtr<ScheduledAction> ScheduledAction::create(ExecState* exec, DOMWrapperWorld& isolatedWorld, ContentSecurityPolicy* policy)
50 {
51     JSValue v = exec->argument(0);
52     CallData callData;
53     if (getCallData(v, callData) == CallTypeNone) {
54         if (policy && !policy->allowEval(exec))
55             return nullptr;
56         String string = v.toString(exec)->value(exec);
57         if (exec->hadException())
58             return nullptr;
59         return adoptPtr(new ScheduledAction(string, isolatedWorld));
60     }
61
62     return adoptPtr(new ScheduledAction(exec, v, isolatedWorld));
63 }
64
65 ScheduledAction::ScheduledAction(ExecState* exec, JSValue function, DOMWrapperWorld& isolatedWorld)
66     : m_function(exec->vm(), function)
67     , m_isolatedWorld(&isolatedWorld)
68 {
69     // setTimeout(function, interval, arg0, arg1...).
70     // Start at 2 to skip function and interval.
71     for (size_t i = 2; i < exec->argumentCount(); ++i)
72         m_args.append(Strong<JSC::Unknown>(exec->vm(), exec->uncheckedArgument(i)));
73 }
74
75 void ScheduledAction::execute(ScriptExecutionContext* context)
76 {
77     if (context->isDocument())
78         execute(toDocument(context));
79     else
80         execute(toWorkerGlobalScope(context));
81 }
82
83 void ScheduledAction::executeFunctionInContext(JSGlobalObject* globalObject, JSValue thisValue, ScriptExecutionContext* context)
84 {
85     ASSERT(m_function);
86     JSLockHolder lock(context->vm());
87
88     CallData callData;
89     CallType callType = getCallData(m_function.get(), callData);
90     if (callType == CallTypeNone)
91         return;
92
93     ExecState* exec = globalObject->globalExec();
94
95     MarkedArgumentBuffer args;
96     size_t size = m_args.size();
97     for (size_t i = 0; i < size; ++i)
98         args.append(m_args[i].get());
99
100     InspectorInstrumentationCookie cookie = JSMainThreadExecState::instrumentFunctionCall(context, callType, callData);
101
102     if (context->isDocument())
103         JSMainThreadExecState::call(exec, m_function.get(), callType, callData, thisValue, args);
104     else
105         JSC::call(exec, m_function.get(), callType, callData, thisValue, args);
106
107     InspectorInstrumentation::didCallFunction(cookie, context);
108
109     if (exec->hadException())
110         reportCurrentException(exec);
111 }
112
113 void ScheduledAction::execute(Document* document)
114 {
115     JSDOMWindow* window = toJSDOMWindow(document->frame(), *m_isolatedWorld);
116     if (!window)
117         return;
118
119     RefPtr<Frame> frame = window->impl().frame();
120     if (!frame || !frame->script().canExecuteScripts(AboutToExecuteScript))
121         return;
122
123     if (m_function)
124         executeFunctionInContext(window, window->shell(), document);
125     else
126         frame->script().executeScriptInWorld(*m_isolatedWorld, m_code);
127 }
128
129 void ScheduledAction::execute(WorkerGlobalScope* workerGlobalScope)
130 {
131     // In a Worker, the execution should always happen on a worker thread.
132     ASSERT(workerGlobalScope->thread().threadID() == currentThread());
133
134     WorkerScriptController* scriptController = workerGlobalScope->script();
135
136     if (m_function) {
137         JSWorkerGlobalScope* contextWrapper = scriptController->workerGlobalScopeWrapper();
138         executeFunctionInContext(contextWrapper, contextWrapper, workerGlobalScope);
139     } else {
140         ScriptSourceCode code(m_code, workerGlobalScope->url());
141         scriptController->evaluate(code);
142     }
143 }
144
145 } // namespace WebCore