1 /* 2 * Copyright (C) 2007 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 COMPUTER, 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 COMPUTER, 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 "WebKitDLL.h" 28 #include "WebFramePolicyListener.h" 29 30 #include "WebFrame.h" 31 #include <WebCore/Frame.h> 32 #include <WebCore/FrameLoader.h> 33 #include <WebCore/FrameLoaderClient.h> 34 35 using namespace WebCore; 36 37 // WebFramePolicyListener ---------------------------------------------------------------- 38 39 WebFramePolicyListener::WebFramePolicyListener(PassRefPtr<Frame> frame) 40 : m_refCount(0) 41 , m_frame(frame) 42 { 43 gClassCount++; 44 gClassNameCount.add("WebFramePolicyListener"); 45 } 46 47 WebFramePolicyListener::~WebFramePolicyListener() 48 { 49 gClassCount--; 50 gClassNameCount.remove("WebFramePolicyListener"); 51 } 52 53 WebFramePolicyListener* WebFramePolicyListener::createInstance(PassRefPtr<Frame> frame) 54 { 55 WebFramePolicyListener* instance = new WebFramePolicyListener(frame); 56 instance->AddRef(); 57 return instance; 58 } 59 60 // IUnknown ------------------------------------------------------------------- 61 62 HRESULT STDMETHODCALLTYPE WebFramePolicyListener::QueryInterface(REFIID riid, void** ppvObject) 63 { 64 *ppvObject = 0; 65 if (IsEqualGUID(riid, IID_IUnknown)) 66 *ppvObject = static_cast<IWebPolicyDecisionListener*>(this); 67 else if (IsEqualGUID(riid, IID_IWebPolicyDecisionListener)) 68 *ppvObject = static_cast<IWebPolicyDecisionListener*>(this); 69 else if (IsEqualGUID(riid, IID_IWebFormSubmissionListener)) 70 *ppvObject = static_cast<IWebFormSubmissionListener*>(this); 71 else 72 return E_NOINTERFACE; 73 74 AddRef(); 75 return S_OK; 76 } 77 78 ULONG STDMETHODCALLTYPE WebFramePolicyListener::AddRef(void) 79 { 80 return ++m_refCount; 81 } 82 83 ULONG STDMETHODCALLTYPE WebFramePolicyListener::Release(void) 84 { 85 ULONG newRef = --m_refCount; 86 if (!newRef) 87 delete(this); 88 89 return newRef; 90 } 91 92 // IWebPolicyDecisionListener ------------------------------------------------------------ 93 94 HRESULT STDMETHODCALLTYPE WebFramePolicyListener::use(void) 95 { 96 receivedPolicyDecision(PolicyUse); 97 return S_OK; 98 } 99 100 HRESULT STDMETHODCALLTYPE WebFramePolicyListener::download(void) 101 { 102 receivedPolicyDecision(PolicyDownload); 103 return S_OK; 104 } 105 106 HRESULT STDMETHODCALLTYPE WebFramePolicyListener::ignore(void) 107 { 108 receivedPolicyDecision(PolicyIgnore); 109 return S_OK; 110 } 111 112 // IWebFormSubmissionListener ------------------------------------------------------------ 113 114 HRESULT STDMETHODCALLTYPE WebFramePolicyListener::continueSubmit(void) 115 { 116 receivedPolicyDecision(PolicyUse); 117 return S_OK; 118 } 119 120 // WebFramePolicyListener ---------------------------------------------------------------- 121 void WebFramePolicyListener::receivedPolicyDecision(PolicyAction action) 122 { 123 RefPtr<Frame> frame = m_frame.release(); 124 if (frame) 125 static_cast<WebFrame*>(frame->loader()->client())->receivedPolicyDecision(action); 126 } 127 128 void WebFramePolicyListener::invalidate() 129 { 130 m_frame = 0; 131 } 132 133