diff -r 000000000000 -r dd21522fd290 webengine/osswebengine/WebKit/win/HTTPHeaderPropertyBag.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/webengine/osswebengine/WebKit/win/HTTPHeaderPropertyBag.cpp Mon Mar 30 12:54:55 2009 +0300 @@ -0,0 +1,120 @@ +/* + * Copyright (C) 2007 Apple Inc. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY + * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR + * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR + * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, + * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, + * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR + * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY + * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include "config.h" +#include "HTTPHeaderPropertyBag.h" + +#include "HTTPHeaderMap.h" + +#include + +// HTTPHeaderPropertyBag ----------------------------------------------- + +HTTPHeaderPropertyBag::HTTPHeaderPropertyBag(WebURLResponse* response) + : m_refCount(1) + , m_response(response) +{ +} + +HTTPHeaderPropertyBag* HTTPHeaderPropertyBag::createInstance(WebURLResponse* response) +{ + return new HTTPHeaderPropertyBag(response); +} + +// IUnknown ------------------------------------------------------------------- + +HRESULT STDMETHODCALLTYPE HTTPHeaderPropertyBag::QueryInterface(REFIID riid, void** ppvObject) +{ + *ppvObject = 0; + if (IsEqualGUID(riid, IID_IUnknown)) + *ppvObject = static_cast(this); + else if (IsEqualGUID(riid, IID_IPropertyBag)) + *ppvObject = static_cast(this); + else if (IsEqualGUID(riid, __uuidof(HTTPHeaderPropertyBag))) + *ppvObject = this; + else + return E_NOINTERFACE; + + AddRef(); + return S_OK; +} + +ULONG STDMETHODCALLTYPE HTTPHeaderPropertyBag::AddRef(void) +{ + return ++m_refCount; +} + +ULONG STDMETHODCALLTYPE HTTPHeaderPropertyBag::Release(void) +{ + ULONG newRef = --m_refCount; + if (!newRef) + delete this; + + return newRef; +} + +static inline WebCore::String ConvertFromLPCOLESTRToString(LPCOLESTR pszPropName) +{ + return WebCore::String(pszPropName); +} + +static bool ConvertFromStringToVariant(VARIANT* pVar, const WebCore::String& stringVal) +{ + if (!pVar) + return false; + + if (V_VT(pVar) == VT_BSTR) { + V_BSTR(pVar) = WebCore::BString(stringVal); + return true; + } + + return false; +} + +// IPropertyBag ------------------------------------------------------------ + +HRESULT STDMETHODCALLTYPE HTTPHeaderPropertyBag::Read(LPCOLESTR pszPropName, VARIANT* pVar, IErrorLog* /*pErrorLog*/) +{ + if (!pszPropName) + return E_POINTER; + + if (!m_response) + return E_FAIL; + + const WebCore::HTTPHeaderMap& headerMap = m_response->resourceResponse().httpHeaderFields(); + WebCore::String key = ConvertFromLPCOLESTRToString(pszPropName); + WebCore::String value = headerMap.get(key); + + if (!ConvertFromStringToVariant(pVar, value)) + return E_INVALIDARG; + + return S_OK; +} + +HRESULT STDMETHODCALLTYPE HTTPHeaderPropertyBag::Write(LPCOLESTR /*pszPropName*/, VARIANT* /*pVar*/) +{ + // We cannot add to the Resource Response's header hash map + return E_FAIL; +}