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()
{
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 DefaultPolicyDelegate::QueryInterface(_In_ REFIID riid, _COM_Outptr_ void** ppvObject)
{
if (!ppvObject)
return E_POINTER;
*ppvObject = nullptr;
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 DefaultPolicyDelegate::AddRef()
{
return ++m_refCount;
}
ULONG DefaultPolicyDelegate::Release()
{
ULONG newRef = --m_refCount;
if (!newRef)
delete(this);
return newRef;
}
HRESULT DefaultPolicyDelegate::decidePolicyForNavigationAction(_In_opt_ IWebView* webView, _In_opt_ IPropertyBag* actionInformation,
_In_opt_ IWebURLRequest* request, _In_opt_ IWebFrame* , _In_opt_ 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 DefaultPolicyDelegate::decidePolicyForNewWindowAction(_In_opt_ IWebView* , _In_opt_ IPropertyBag* ,
_In_opt_ IWebURLRequest* , _In_ BSTR , _In_opt_ IWebPolicyDecisionListener* listener)
{
if (!listener)
return E_POINTER;
listener->use();
return S_OK;
}
HRESULT DefaultPolicyDelegate::decidePolicyForMIMEType(_In_opt_ IWebView* webView, _In_ BSTR type, _In_opt_ IWebURLRequest* request,
_In_opt_ IWebFrame* , _In_opt_ 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 DefaultPolicyDelegate::unableToImplementPolicyWithError(_In_opt_ IWebView* , _In_opt_ IWebError* error, _In_opt_ IWebFrame* frame)
{
if (!error || !frame)
return E_POINTER;
BString errorStr;
error->localizedDescription(&errorStr);
BString frameName;
frame->name(&frameName);
LOG_ERROR("called unableToImplementPolicyWithError:%S inFrame:%S", errorStr ? static_cast<BSTR>(errorStr) : TEXT(""), frameName ? static_cast<BSTR>(frameName) : TEXT(""));
return S_OK;
}