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 "InjectedBundleDOMWindowExtension.h"
28
29#include "InjectedBundleScriptWorld.h"
30#include "WebFrame.h"
31#include "WebFrameLoaderClient.h"
32#include <WebCore/DOMWindowExtension.h>
33#include <WebCore/DOMWrapperWorld.h>
34#include <WebCore/Frame.h>
35#include <WebCore/FrameLoader.h>
36#include <wtf/HashMap.h>
37#include <wtf/NeverDestroyed.h>
38
39namespace WebKit {
40using namespace WebCore;
41
42typedef HashMap<WebCore::DOMWindowExtension*, InjectedBundleDOMWindowExtension*> ExtensionMap;
43static ExtensionMap& allExtensions()
44{
45 static NeverDestroyed<ExtensionMap> map;
46 return map;
47}
48
49Ref<InjectedBundleDOMWindowExtension> InjectedBundleDOMWindowExtension::create(WebFrame* frame, InjectedBundleScriptWorld* world)
50{
51 return adoptRef(*new InjectedBundleDOMWindowExtension(frame, world));
52}
53
54InjectedBundleDOMWindowExtension* InjectedBundleDOMWindowExtension::get(DOMWindowExtension* extension)
55{
56 ASSERT(allExtensions().contains(extension));
57 return allExtensions().get(extension);
58}
59
60InjectedBundleDOMWindowExtension::InjectedBundleDOMWindowExtension(WebFrame* frame, InjectedBundleScriptWorld* world)
61 : m_coreExtension(DOMWindowExtension::create(frame->coreFrame() ? frame->coreFrame()->window() : nullptr, world->coreWorld()))
62{
63 allExtensions().add(m_coreExtension.get(), this);
64}
65
66InjectedBundleDOMWindowExtension::~InjectedBundleDOMWindowExtension()
67{
68 ASSERT(allExtensions().contains(m_coreExtension.get()));
69 allExtensions().remove(m_coreExtension.get());
70}
71
72WebFrame* InjectedBundleDOMWindowExtension::frame() const
73{
74 Frame* frame = m_coreExtension->frame();
75 if (!frame)
76 return nullptr;
77
78 return WebFrame::fromCoreFrame(*frame);
79}
80
81InjectedBundleScriptWorld* InjectedBundleDOMWindowExtension::world() const
82{
83 if (!m_world)
84 m_world = InjectedBundleScriptWorld::getOrCreate(m_coreExtension->world());
85
86 return m_world.get();
87}
88
89} // namespace WebKit
90