DefaultPolicyDelegate.cpp [plain text]
#include "WebKitDLL.h"
#include "DefaultPolicyDelegate.h"
#include <comutil.h>
#include <WebCore/BString.h>
#include <WebCore/COMPtr.h>
#include <wtf/text/WTFString.h>
using namespace WebCore;
typedef enum WebExtraNavigationType {
WebNavigationTypePlugInRequest = WebNavigationTypeOther + 1
} WebExtraNavigationType;
DefaultPolicyDelegate::DefaultPolicyDelegate()
: m_refCount(0)
{
gClassCount++;
gClassNameCount().add("DefaultPolicyDelegate");
}
DefaultPolicyDelegate::~DefaultPolicyDelegate()
{
gClassCount--;
gClassNameCount().remove("DefaultPolicyDelegate");
}
DefaultPolicyDelegate* DefaultPolicyDelegate::sharedInstance()
{
static COMPtr<DefaultPolicyDelegate> shared;
if (!shared)
shared.adoptRef(DefaultPolicyDelegate::createInstance());
return shared.get();
}
DefaultPolicyDelegate* DefaultPolicyDelegate::createInstance()
{
DefaultPolicyDelegate* instance = new DefaultPolicyDelegate();
instance->AddRef();
return instance;
}
HRESULT STDMETHODCALLTYPE DefaultPolicyDelegate::QueryInterface(REFIID riid, void** ppvObject)
{
*ppvObject = 0;
if (IsEqualGUID(riid, IID_IUnknown))
*ppvObject = static_cast<IUnknown*>(this);
else if (IsEqualGUID(riid, IID_IWebPolicyDelegate))
*ppvObject = static_cast<IWebPolicyDelegate*>(this);
else
return E_NOINTERFACE;
AddRef();
return S_OK;
}
ULONG STDMETHODCALLTYPE DefaultPolicyDelegate::AddRef()
{
return ++m_refCount;
}
ULONG STDMETHODCALLTYPE DefaultPolicyDelegate::Release()
{
ULONG newRef = --m_refCount;
if (!newRef)
delete(this);
return newRef;
}
HRESULT DefaultPolicyDelegate::decidePolicyForNavigationAction(IWebView* webView, IPropertyBag* actionInformation,
IWebURLRequest* request, IWebFrame* , IWebPolicyDecisionListener* listener)
{
int navType = 0;
_variant_t var;
if (SUCCEEDED(actionInformation->Read(WebActionNavigationTypeKey, &var.GetVARIANT(), nullptr))) {
var.ChangeType(VT_I4, nullptr);
navType = V_I4(&var);
}
COMPtr<IWebViewPrivate2> wvPrivate(Query, webView);
if (wvPrivate) {
BOOL canHandleRequest = FALSE;
if (SUCCEEDED(wvPrivate->canHandleRequest(request, &canHandleRequest)) && canHandleRequest)
listener->use();
else if (navType == WebNavigationTypePlugInRequest)
listener->use();
else {
BString url;
if (SUCCEEDED(request->URL(&url)) && !String(url, SysStringLen(url)).startsWith("file:")) {
;
}
listener->ignore();
}
}
return S_OK;
}
HRESULT STDMETHODCALLTYPE DefaultPolicyDelegate::decidePolicyForNewWindowAction(
IWebView* ,
IPropertyBag* ,
IWebURLRequest* ,
BSTR ,
IWebPolicyDecisionListener* listener)
{
listener->use();
return S_OK;
}
HRESULT STDMETHODCALLTYPE DefaultPolicyDelegate::decidePolicyForMIMEType(
IWebView* webView,
BSTR type,
IWebURLRequest* request,
IWebFrame* ,
IWebPolicyDecisionListener* listener)
{
BOOL canShowMIMEType;
if (FAILED(webView->canShowMIMEType(type, &canShowMIMEType)))
canShowMIMEType = FALSE;
BString url;
request->URL(&url);
if (String(url, SysStringLen(url)).startsWith("file:")) {
BOOL isDirectory = FALSE;
WIN32_FILE_ATTRIBUTE_DATA attrs;
if (GetFileAttributesEx(url, GetFileExInfoStandard, &attrs))
isDirectory = !!(attrs.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY);
if (isDirectory)
listener->ignore();
else if(canShowMIMEType)
listener->use();
else
listener->ignore();
} else if (canShowMIMEType)
listener->use();
else
listener->ignore();
return S_OK;
}
HRESULT STDMETHODCALLTYPE DefaultPolicyDelegate::unableToImplementPolicyWithError(
IWebView* ,
IWebError* error,
IWebFrame* frame)
{
BString errorStr;
error->localizedDescription(&errorStr);
BString frameName;
frame->name(&frameName);
LOG_ERROR("called unableToImplementPolicyWithError:%S inFrame:%S", errorStr ? errorStr : TEXT(""), frameName ? frameName : TEXT(""));
return S_OK;
}