Fixes the assertion and crash that would happen when inspecting a element from a...
[WebKit-https.git] / WebCore / bindings / js / JSInspectedObjectWrapper.cpp
1 /*
2  * Copyright (C) 2008 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 "JSInspectedObjectWrapper.h"
28
29 #include "JSInspectorCallbackWrapper.h"
30
31 using namespace KJS;
32
33 namespace WebCore {
34
35 typedef HashMap<JSObject*, JSInspectedObjectWrapper*> WrapperMap;
36
37 static WrapperMap& wrappers()
38 {
39     static WrapperMap map;
40     return map;
41 }
42
43 const ClassInfo JSInspectedObjectWrapper::s_info = { "JSInspectedObjectWrapper", &JSQuarantinedObjectWrapper::s_info, 0, 0 };
44
45 JSValue* JSInspectedObjectWrapper::wrap(ExecState* unwrappedExec, JSValue* unwrappedValue)
46 {
47     if (!unwrappedValue->isObject())
48         return unwrappedValue;
49
50     JSObject* unwrappedObject = static_cast<JSObject*>(unwrappedValue);
51
52     if (unwrappedObject->inherits(&JSInspectedObjectWrapper::s_info))
53         return unwrappedObject;
54
55     if (JSInspectedObjectWrapper* wrapper = wrappers().get(unwrappedObject))
56         return wrapper;
57
58     JSValue* prototype = unwrappedObject->prototype();
59     JSValue* wrappedPrototype = prototype ? wrap(unwrappedExec, prototype) : 0;
60
61     return new JSInspectedObjectWrapper(unwrappedExec, unwrappedObject, wrappedPrototype);
62 }
63
64 JSInspectedObjectWrapper::JSInspectedObjectWrapper(ExecState* unwrappedExec, JSObject* unwrappedObject, JSValue* wrappedPrototype)
65     : JSQuarantinedObjectWrapper(unwrappedExec, unwrappedObject, wrappedPrototype)
66 {
67     ASSERT(!wrappers().contains(unwrappedObject));
68     wrappers().set(unwrappedObject, this);
69 }
70
71 JSInspectedObjectWrapper::~JSInspectedObjectWrapper()
72 {
73     wrappers().remove(unwrappedObject());
74 }
75
76 JSValue* JSInspectedObjectWrapper::prepareIncomingValue(ExecState*, JSValue* value) const
77 {
78     // The Inspector is only allowed to pass primitive values and wrapped objects to objects from the inspected page.
79
80     if (!value->isObject())
81         return value;
82
83     JSQuarantinedObjectWrapper* wrapper = asWrapper(value);
84     ASSERT_WITH_MESSAGE(wrapper, "Objects passed to JSInspectedObjectWrapper must be wrapped");
85     if (!wrapper)
86         return jsUndefined();
87
88     if (wrapper->allowsUnwrappedAccessFrom(unwrappedExecState())) {
89         ASSERT_WITH_MESSAGE(wrapper->inherits(&s_info), "A wrapper contains an object from the inspected page but is not a JSInspectedObjectWrapper");
90         if (!wrapper->inherits(&s_info))
91             return jsUndefined();
92
93         // Return the unwrapped object so the inspected page never sees one of its own objects in wrapped form.
94         return wrapper->unwrappedObject();
95     }
96
97     ASSERT_WITH_MESSAGE(wrapper->inherits(&JSInspectorCallbackWrapper::s_info), "A wrapper that was not from the inspected page and is not an Inspector callback was passed to a JSInspectedObjectWrapper");
98     if (!wrapper->inherits(&JSInspectorCallbackWrapper::s_info))
99         return jsUndefined();
100
101     return wrapper;
102 }
103
104 } // namespace WebCore