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