1/*
2 * Copyright (C) 2010 Google 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. AND ITS CONTRIBUTORS ``AS IS''
14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23 * THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#ifndef JSMainThreadExecState_h
27#define JSMainThreadExecState_h
28
29#include "InspectorInstrumentation.h"
30#include "JSDOMBinding.h"
31#include <runtime/Completion.h>
32#include <runtime/Executable.h>
33#ifndef NDEBUG
34#include <wtf/MainThread.h>
35#endif
36
37namespace WebCore {
38
39class ScriptExecutionContext;
40
41class JSMainThreadExecState {
42    WTF_MAKE_NONCOPYABLE(JSMainThreadExecState);
43public:
44    static JSC::ExecState* currentState()
45    {
46        ASSERT(isMainThread());
47        return s_mainThreadState;
48    };
49
50    static JSC::JSValue call(JSC::ExecState* exec, JSC::JSValue functionObject, JSC::CallType callType, const JSC::CallData& callData, JSC::JSValue thisValue, const JSC::ArgList& args)
51    {
52        JSMainThreadExecState currentState(exec);
53        // Ensure DOM global object is unwrapped to the shell.
54        if (thisValue.isObject())
55            thisValue = thisValue.toThisObject(exec);
56        return JSC::call(exec, functionObject, callType, callData, thisValue, args);
57    };
58
59    static inline InspectorInstrumentationCookie instrumentFunctionCall(ScriptExecutionContext* context, JSC::CallType callType, const JSC::CallData& callData)
60    {
61        if (!InspectorInstrumentation::timelineAgentEnabled(context))
62            return InspectorInstrumentationCookie();
63        String resourceName;
64        int lineNumber = 1;
65        if (callType == JSC::CallTypeJS) {
66            resourceName = callData.js.functionExecutable->sourceURL();
67            lineNumber = callData.js.functionExecutable->lineNo();
68        } else
69            resourceName = "undefined";
70        return InspectorInstrumentation::willCallFunction(context, resourceName, lineNumber);
71    }
72
73    static JSC::JSValue evaluate(JSC::ExecState* exec, const JSC::SourceCode& source, JSC::JSValue thisValue, JSC::JSValue* exception)
74    {
75        JSMainThreadExecState currentState(exec);
76        JSC::JSLockHolder lock(exec);
77        return JSC::evaluate(exec, source, thisValue, exception);
78    };
79
80protected:
81    explicit JSMainThreadExecState(JSC::ExecState* exec)
82        : m_previousState(s_mainThreadState)
83    {
84        ASSERT(isMainThread());
85        s_mainThreadState = exec;
86    };
87
88    ~JSMainThreadExecState()
89    {
90        ASSERT(isMainThread());
91
92        bool didExitJavaScript = s_mainThreadState && !m_previousState;
93
94        s_mainThreadState = m_previousState;
95
96        if (didExitJavaScript)
97            didLeaveScriptContext();
98    }
99
100private:
101    static JSC::ExecState* s_mainThreadState;
102    JSC::ExecState* m_previousState;
103
104    static void didLeaveScriptContext();
105};
106
107// Null state prevents origin security checks.
108// Used by non-JavaScript bindings (ObjC, GObject).
109class JSMainThreadNullState : private JSMainThreadExecState {
110public:
111    explicit JSMainThreadNullState() : JSMainThreadExecState(0) {};
112};
113
114} // namespace WebCore
115
116#endif // JSMainThreadExecState_h
117