WebURLProtectionSpace.cpp [plain text]
#include "WebKit.h"
#include "WebKitDLL.h"
#include "WebURLProtectionSpace.h"
#include <WebCore/BString.h>
using namespace WebCore;
WebURLProtectionSpace::WebURLProtectionSpace(const ProtectionSpace& protectionSpace)
: m_refCount(0)
, m_protectionSpace(protectionSpace)
{
gClassCount++;
gClassNameCount().add("WebURLProtectionSpace");
}
WebURLProtectionSpace::~WebURLProtectionSpace()
{
gClassCount--;
gClassNameCount().remove("WebURLProtectionSpace");
}
WebURLProtectionSpace* WebURLProtectionSpace::createInstance()
{
WebURLProtectionSpace* instance = new WebURLProtectionSpace(ProtectionSpace());
instance->AddRef();
return instance;
}
WebURLProtectionSpace* WebURLProtectionSpace::createInstance(const ProtectionSpace& protectionSpace)
{
WebURLProtectionSpace* instance = new WebURLProtectionSpace(protectionSpace);
instance->AddRef();
return instance;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::QueryInterface(REFIID riid, void** ppvObject)
{
*ppvObject = 0;
if (IsEqualGUID(riid, IID_IUnknown))
*ppvObject = static_cast<IUnknown*>(this);
else if (IsEqualGUID(riid, CLSID_WebURLProtectionSpace))
*ppvObject = static_cast<WebURLProtectionSpace*>(this);
else if (IsEqualGUID(riid, IID_IWebURLProtectionSpace))
*ppvObject = static_cast<IWebURLProtectionSpace*>(this);
else
return E_NOINTERFACE;
AddRef();
return S_OK;
}
ULONG STDMETHODCALLTYPE WebURLProtectionSpace::AddRef(void)
{
return ++m_refCount;
}
ULONG STDMETHODCALLTYPE WebURLProtectionSpace::Release(void)
{
ULONG newRef = --m_refCount;
if (!newRef)
delete(this);
return newRef;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::authenticationMethod(
BSTR* result)
{
switch (m_protectionSpace.authenticationScheme()) {
case ProtectionSpaceAuthenticationSchemeDefault:
*result = SysAllocString(WebURLAuthenticationMethodDefault);
break;
case ProtectionSpaceAuthenticationSchemeHTTPBasic:
*result = SysAllocString(WebURLAuthenticationMethodHTTPBasic);
break;
case ProtectionSpaceAuthenticationSchemeHTTPDigest:
*result = SysAllocString(WebURLAuthenticationMethodHTTPDigest);
break;
case ProtectionSpaceAuthenticationSchemeHTMLForm:
*result = SysAllocString(WebURLAuthenticationMethodHTMLForm);
break;
default:
ASSERT_NOT_REACHED();
return E_FAIL;
}
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::host(
BSTR* result)
{
BString str = m_protectionSpace.host();
*result = str.release();
return S_OK;
}
static ProtectionSpaceAuthenticationScheme coreScheme(BSTR authenticationMethod)
{
ProtectionSpaceAuthenticationScheme scheme = ProtectionSpaceAuthenticationSchemeDefault;
if (BString(authenticationMethod) == BString(WebURLAuthenticationMethodDefault))
scheme = ProtectionSpaceAuthenticationSchemeDefault;
else if (BString(authenticationMethod) == BString(WebURLAuthenticationMethodHTTPBasic))
scheme = ProtectionSpaceAuthenticationSchemeHTTPBasic;
else if (BString(authenticationMethod) == BString(WebURLAuthenticationMethodHTTPDigest))
scheme = ProtectionSpaceAuthenticationSchemeHTTPDigest;
else if (BString(authenticationMethod) == BString(WebURLAuthenticationMethodHTMLForm))
scheme = ProtectionSpaceAuthenticationSchemeHTMLForm;
else
ASSERT_NOT_REACHED();
return scheme;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::initWithHost(
BSTR host,
int port,
BSTR protocol,
BSTR realm,
BSTR authenticationMethod)
{
static BString& webURLProtectionSpaceHTTPBString = *new BString(WebURLProtectionSpaceHTTP);
static BString& webURLProtectionSpaceHTTPSBString = *new BString(WebURLProtectionSpaceHTTPS);
static BString& webURLProtectionSpaceFTPBString = *new BString(WebURLProtectionSpaceFTP);
static BString& webURLProtectionSpaceFTPSBString = *new BString(WebURLProtectionSpaceFTPS);
ProtectionSpaceServerType serverType = ProtectionSpaceServerHTTP;
if (BString(protocol) == webURLProtectionSpaceHTTPBString)
serverType = ProtectionSpaceServerHTTP;
else if (BString(protocol) == webURLProtectionSpaceHTTPSBString)
serverType = ProtectionSpaceServerHTTPS;
else if (BString(protocol) == webURLProtectionSpaceFTPBString)
serverType = ProtectionSpaceServerFTP;
else if (BString(protocol) == webURLProtectionSpaceFTPSBString)
serverType = ProtectionSpaceServerFTPS;
m_protectionSpace = ProtectionSpace(String(host, SysStringLen(host)), port, serverType,
String(realm, SysStringLen(realm)), coreScheme(authenticationMethod));
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::initWithProxyHost(
BSTR host,
int port,
BSTR proxyType,
BSTR realm,
BSTR authenticationMethod)
{
static BString& webURLProtectionSpaceHTTPProxyBString = *new BString(WebURLProtectionSpaceHTTPProxy);
static BString& webURLProtectionSpaceHTTPSProxyBString = *new BString(WebURLProtectionSpaceHTTPSProxy);
static BString& webURLProtectionSpaceFTPProxyBString = *new BString(WebURLProtectionSpaceFTPProxy);
static BString& webURLProtectionSpaceSOCKSProxyBString = *new BString(WebURLProtectionSpaceSOCKSProxy);
ProtectionSpaceServerType serverType = ProtectionSpaceProxyHTTP;
if (BString(proxyType) == webURLProtectionSpaceHTTPProxyBString)
serverType = ProtectionSpaceProxyHTTP;
else if (BString(proxyType) == webURLProtectionSpaceHTTPSProxyBString)
serverType = ProtectionSpaceProxyHTTPS;
else if (BString(proxyType) == webURLProtectionSpaceFTPProxyBString)
serverType = ProtectionSpaceProxyFTP;
else if (BString(proxyType) == webURLProtectionSpaceSOCKSProxyBString)
serverType = ProtectionSpaceProxySOCKS;
else
ASSERT_NOT_REACHED();
m_protectionSpace = ProtectionSpace(String(host, SysStringLen(host)), port, serverType,
String(realm, SysStringLen(realm)), coreScheme(authenticationMethod));
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::isProxy(
BOOL* result)
{
*result = m_protectionSpace.isProxy();
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::port(
int* result)
{
*result = m_protectionSpace.port();
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::protocol(
BSTR* result)
{
switch (m_protectionSpace.serverType()) {
case ProtectionSpaceServerHTTP:
*result = SysAllocString(WebURLProtectionSpaceHTTP);
break;
case ProtectionSpaceServerHTTPS:
*result = SysAllocString(WebURLProtectionSpaceHTTPS);
break;
case ProtectionSpaceServerFTP:
*result = SysAllocString(WebURLProtectionSpaceFTP);
break;
case ProtectionSpaceServerFTPS:
*result = SysAllocString(WebURLProtectionSpaceFTPS);
break;
default:
ASSERT_NOT_REACHED();
return E_FAIL;
}
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::proxyType(
BSTR* result)
{
switch (m_protectionSpace.serverType()) {
case ProtectionSpaceProxyHTTP:
*result = SysAllocString(WebURLProtectionSpaceHTTPProxy);
break;
case ProtectionSpaceProxyHTTPS:
*result = SysAllocString(WebURLProtectionSpaceHTTPSProxy);
break;
case ProtectionSpaceProxyFTP:
*result = SysAllocString(WebURLProtectionSpaceFTPProxy);
break;
case ProtectionSpaceProxySOCKS:
*result = SysAllocString(WebURLProtectionSpaceSOCKSProxy);
break;
default:
ASSERT_NOT_REACHED();
return E_FAIL;
}
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::realm(
BSTR* result)
{
BString bstring = m_protectionSpace.realm();
*result = bstring.release();
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebURLProtectionSpace::receivesCredentialSecurely(
BOOL* result)
{
*result = m_protectionSpace.receivesCredentialSecurely();
return S_OK;
}
const ProtectionSpace& WebURLProtectionSpace::protectionSpace() const
{
return m_protectionSpace;
}