WebFramePolicyListener.cpp [plain text]
#include "WebKitDLL.h"
#include "WebFramePolicyListener.h"
#include "WebFrameLoaderClient.h"
#include <WebCore/Frame.h>
#include <WebCore/FrameLoader.h>
#include <WebCore/FrameLoaderClient.h>
using namespace WebCore;
WebFramePolicyListener::WebFramePolicyListener(PassRefPtr<Frame> frame)
: m_frame(frame)
{
gClassCount++;
gClassNameCount().add("WebFramePolicyListener");
}
WebFramePolicyListener::~WebFramePolicyListener()
{
gClassCount--;
gClassNameCount().remove("WebFramePolicyListener");
}
WebFramePolicyListener* WebFramePolicyListener::createInstance(PassRefPtr<Frame> frame)
{
WebFramePolicyListener* instance = new WebFramePolicyListener(frame);
instance->AddRef();
return instance;
}
HRESULT WebFramePolicyListener::QueryInterface(_In_ REFIID riid, _COM_Outptr_ void** ppvObject)
{
if (!ppvObject)
return E_POINTER;
*ppvObject = nullptr;
if (IsEqualGUID(riid, IID_IUnknown))
*ppvObject = static_cast<IWebPolicyDecisionListener*>(this);
else if (IsEqualGUID(riid, IID_IWebPolicyDecisionListener))
*ppvObject = static_cast<IWebPolicyDecisionListener*>(this);
else if (IsEqualGUID(riid, IID_IWebFormSubmissionListener))
*ppvObject = static_cast<IWebFormSubmissionListener*>(this);
else
return E_NOINTERFACE;
AddRef();
return S_OK;
}
ULONG WebFramePolicyListener::AddRef()
{
return ++m_refCount;
}
ULONG WebFramePolicyListener::Release()
{
ULONG newRef = --m_refCount;
if (!newRef)
delete(this);
return newRef;
}
HRESULT WebFramePolicyListener::use()
{
receivedPolicyDecision(PolicyUse);
return S_OK;
}
HRESULT WebFramePolicyListener::download()
{
receivedPolicyDecision(PolicyDownload);
return S_OK;
}
HRESULT WebFramePolicyListener::ignore()
{
receivedPolicyDecision(PolicyIgnore);
return S_OK;
}
HRESULT WebFramePolicyListener::continueSubmit()
{
receivedPolicyDecision(PolicyUse);
return S_OK;
}
void WebFramePolicyListener::receivedPolicyDecision(PolicyAction action)
{
RefPtr<Frame> frame = m_frame.release();
if (frame)
static_cast<WebFrameLoaderClient&>(frame->loader().client()).receivedPolicyDecision(action);
}
void WebFramePolicyListener::invalidate()
{
m_frame = nullptr;
}