blob: 9a1f011221bc21400bee887ed1119b369ffe6003 [file] [log] [blame]
/*
* Copyright (C) 2006-2019 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 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 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 "WebKitDLL.h"
#include "CFDictionaryPropertyBag.h"
#include "MarshallingHelpers.h"
#include "WebKit.h"
// CFDictionaryPropertyBag -----------------------------------------------
CFDictionaryPropertyBag::CFDictionaryPropertyBag()
{
gClassCount++;
gClassNameCount().add("CFDictionaryPropertyBag");
}
CFDictionaryPropertyBag::~CFDictionaryPropertyBag()
{
gClassCount--;
gClassNameCount().remove("CFDictionaryPropertyBag");
}
COMPtr<CFDictionaryPropertyBag> CFDictionaryPropertyBag::createInstance()
{
return new CFDictionaryPropertyBag;
}
#if USE(CF)
void CFDictionaryPropertyBag::setDictionary(CFMutableDictionaryRef dictionary)
{
m_dictionary = dictionary;
}
CFMutableDictionaryRef CFDictionaryPropertyBag::dictionary() const
{
return m_dictionary.get();
}
#endif
// IUnknown -------------------------------------------------------------------
HRESULT CFDictionaryPropertyBag::QueryInterface(_In_ REFIID riid, _COM_Outptr_ void** ppvObject)
{
if (!ppvObject)
return E_POINTER;
*ppvObject = nullptr;
if (IsEqualGUID(riid, IID_IUnknown))
*ppvObject = static_cast<IPropertyBag*>(this);
else if (IsEqualGUID(riid, IID_IPropertyBag))
*ppvObject = static_cast<IPropertyBag*>(this);
else if (IsEqualGUID(riid, __uuidof(this)))
*ppvObject = this;
else
return E_NOINTERFACE;
AddRef();
return S_OK;
}
ULONG CFDictionaryPropertyBag::AddRef()
{
return ++m_refCount;
}
ULONG CFDictionaryPropertyBag::Release()
{
ULONG newRef = --m_refCount;
if (!newRef)
delete(this);
return newRef;
}
// IPropertyBag ------------------------------------------------------------
static bool ConvertCFTypeToVariant(VARIANT* pVar, void* cfObj)
{
#if USE(CF)
if (!cfObj) {
V_VT(pVar) = VT_NULL;
return true;
}
else {
// if caller expects a string, retrieve BSTR from CFStringRef
if (V_VT(pVar) == VT_BSTR) {
V_BSTR(pVar) = MarshallingHelpers::CFStringRefToBSTR((CFStringRef) cfObj);
return true;
} else if (V_VT(pVar) == VT_I4) {
V_I4(pVar) = MarshallingHelpers::CFNumberRefToInt((CFNumberRef) cfObj);
return true;
} else if (!!(V_VT(pVar)&VT_ARRAY)) {
if ((V_VT(pVar)&~VT_ARRAY) == VT_BSTR) {
V_ARRAY(pVar) = MarshallingHelpers::stringArrayToSafeArray((CFArrayRef) cfObj);
return true;
} else if ((V_VT(pVar)&~VT_ARRAY) == VT_I4) {
V_ARRAY(pVar) = MarshallingHelpers::intArrayToSafeArray((CFArrayRef) cfObj);
return true;
} else if ((V_VT(pVar)&~VT_ARRAY) == VT_UNKNOWN) {
V_ARRAY(pVar) = MarshallingHelpers::iunknownArrayToSafeArray((CFArrayRef) cfObj);
return true;
}
}
}
#endif
return false;
}
#if USE(CF)
static bool ConvertVariantToCFType(VARIANT* pVar, void** cfObj)
{
if (V_VT(pVar) == VT_NULL) {
*cfObj = 0;
return true;
}
else {
// if caller expects a string, retrieve BSTR from CFStringRef
if (V_VT(pVar) == VT_BSTR) {
*cfObj = (void*) MarshallingHelpers::BSTRToCFStringRef(V_BSTR(pVar)).leakRef();
return true;
} else if (V_VT(pVar) == VT_I4) {
*cfObj = (void*) MarshallingHelpers::intToCFNumberRef(V_I4(pVar)).leakRef();
return true;
} else if (!!(V_VT(pVar)&VT_ARRAY)) {
if ((V_VT(pVar)&~VT_ARRAY) == VT_BSTR) {
*cfObj = (void*) MarshallingHelpers::safeArrayToStringArray(V_ARRAY(pVar)).leakRef();
return true;
} else if ((V_VT(pVar)&~VT_ARRAY) == VT_I4) {
*cfObj = (void*) MarshallingHelpers::safeArrayToIntArray(V_ARRAY(pVar)).leakRef();
return true;
} else if ((V_VT(pVar)&~VT_ARRAY) == VT_UNKNOWN) {
*cfObj = (void*) MarshallingHelpers::safeArrayToIUnknownArray(V_ARRAY(pVar)).leakRef();
return true;
}
}
}
return false;
}
#endif
HRESULT CFDictionaryPropertyBag::Read(LPCOLESTR pszPropName, VARIANT *pVar, IErrorLog * /*pErrorLog*/)
{
if (!pszPropName)
return E_POINTER;
#if USE(CF)
if (m_dictionary) {
void* value;
auto key = MarshallingHelpers::LPCOLESTRToCFStringRef(pszPropName);
HRESULT hr = E_FAIL;
if (CFDictionaryGetValueIfPresent(m_dictionary.get(), key.get(), (const void**) &value)) {
if (ConvertCFTypeToVariant(pVar, value))
hr = S_OK;
} else
hr = E_INVALIDARG;
return hr;
}
#endif
return E_FAIL;
}
HRESULT CFDictionaryPropertyBag::Write(_In_ LPCOLESTR pszPropName, _In_ VARIANT* pVar)
{
if (!pszPropName || !pVar)
return E_POINTER;
#if USE(CF)
if (!m_dictionary) {
m_dictionary = adoptCF(CFDictionaryCreateMutable(0, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks));
}
void* cfObj;
if (ConvertVariantToCFType(pVar, &cfObj)) {
auto key = MarshallingHelpers::LPCOLESTRToCFStringRef(pszPropName);
CFDictionaryAddValue(m_dictionary.get(), key.get(), cfObj);
// CFDictionaryAddValue should automatically retain the CF objects passed in, so release them here
CFRelease(cfObj);
return S_OK;
}
#endif
return E_FAIL;
}