code.vuplus.com Git - vuplus_webkit/blob - Source/WebKit/win/WebFramePolicyListener.cpp

   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