466 lines
12 KiB
C
466 lines
12 KiB
C
/*
|
|
* standard IPropertyStore implementation
|
|
*
|
|
* Copyright 2012 Vincent Povirk for CodeWeavers
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#define COBJMACROS
|
|
#include "config.h"
|
|
|
|
#include <stdarg.h>
|
|
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "objbase.h"
|
|
#include "rpcproxy.h"
|
|
#include "propsys.h"
|
|
#include "wine/debug.h"
|
|
#include "wine/unicode.h"
|
|
#include "wine/list.h"
|
|
|
|
#include "initguid.h"
|
|
#include "propsys_private.h"
|
|
|
|
DEFINE_GUID(FMTID_NamedProperties, 0xd5cdd505, 0x2e9c, 0x101b, 0x93, 0x97, 0x08, 0x00, 0x2b, 0x2c, 0xf9, 0xae);
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(propsys);
|
|
|
|
typedef struct {
|
|
struct list entry;
|
|
DWORD pid;
|
|
PROPVARIANT propvar;
|
|
PSC_STATE state;
|
|
} propstore_value;
|
|
|
|
typedef struct {
|
|
struct list entry;
|
|
GUID fmtid;
|
|
struct list values; /* list of struct propstore_value */
|
|
DWORD count;
|
|
} propstore_format;
|
|
|
|
typedef struct {
|
|
IPropertyStoreCache IPropertyStoreCache_iface;
|
|
LONG ref;
|
|
CRITICAL_SECTION lock;
|
|
struct list formats; /* list of struct propstore_format */
|
|
} PropertyStore;
|
|
|
|
static inline PropertyStore *impl_from_IPropertyStoreCache(IPropertyStoreCache *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, PropertyStore, IPropertyStoreCache_iface);
|
|
}
|
|
|
|
static HRESULT WINAPI PropertyStore_QueryInterface(IPropertyStoreCache *iface, REFIID iid,
|
|
void **ppv)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
|
|
|
|
if (!ppv) return E_INVALIDARG;
|
|
|
|
if (IsEqualIID(&IID_IUnknown, iid) || IsEqualIID(&IID_IPropertyStore, iid) ||
|
|
IsEqualIID(&IID_IPropertyStoreCache, iid))
|
|
{
|
|
*ppv = &This->IPropertyStoreCache_iface;
|
|
}
|
|
else
|
|
{
|
|
FIXME("No interface for %s\n", debugstr_guid(iid));
|
|
*ppv = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
IUnknown_AddRef((IUnknown*)*ppv);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI PropertyStore_AddRef(IPropertyStoreCache *iface)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
ULONG ref = InterlockedIncrement(&This->ref);
|
|
|
|
TRACE("(%p) refcount=%u\n", iface, ref);
|
|
|
|
return ref;
|
|
}
|
|
|
|
static void destroy_format(propstore_format *format)
|
|
{
|
|
propstore_value *cursor, *cursor2;
|
|
LIST_FOR_EACH_ENTRY_SAFE(cursor, cursor2, &format->values, propstore_value, entry)
|
|
{
|
|
PropVariantClear(&cursor->propvar);
|
|
HeapFree(GetProcessHeap(), 0, cursor);
|
|
}
|
|
HeapFree(GetProcessHeap(), 0, format);
|
|
}
|
|
|
|
static ULONG WINAPI PropertyStore_Release(IPropertyStoreCache *iface)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p) refcount=%u\n", iface, ref);
|
|
|
|
if (ref == 0)
|
|
{
|
|
propstore_format *cursor, *cursor2;
|
|
This->lock.DebugInfo->Spare[0] = 0;
|
|
DeleteCriticalSection(&This->lock);
|
|
LIST_FOR_EACH_ENTRY_SAFE(cursor, cursor2, &This->formats, propstore_format, entry)
|
|
destroy_format(cursor);
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI PropertyStore_GetCount(IPropertyStoreCache *iface,
|
|
DWORD *cProps)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
propstore_format *format;
|
|
|
|
TRACE("%p,%p\n", iface, cProps);
|
|
|
|
if (!cProps)
|
|
return E_POINTER;
|
|
|
|
*cProps = 0;
|
|
|
|
EnterCriticalSection(&This->lock);
|
|
|
|
LIST_FOR_EACH_ENTRY(format, &This->formats, propstore_format, entry)
|
|
*cProps += format->count;
|
|
|
|
LeaveCriticalSection(&This->lock);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI PropertyStore_GetAt(IPropertyStoreCache *iface,
|
|
DWORD iProp, PROPERTYKEY *pkey)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
propstore_format *format=NULL, *format_candidate;
|
|
propstore_value *value;
|
|
HRESULT hr;
|
|
|
|
TRACE("%p,%d,%p\n", iface, iProp, pkey);
|
|
|
|
if (!pkey)
|
|
return E_POINTER;
|
|
|
|
EnterCriticalSection(&This->lock);
|
|
|
|
LIST_FOR_EACH_ENTRY(format_candidate, &This->formats, propstore_format, entry)
|
|
{
|
|
if (format_candidate->count > iProp)
|
|
{
|
|
format = format_candidate;
|
|
pkey->fmtid = format->fmtid;
|
|
break;
|
|
}
|
|
|
|
iProp -= format_candidate->count;
|
|
}
|
|
|
|
if (format)
|
|
{
|
|
LIST_FOR_EACH_ENTRY(value, &format->values, propstore_value, entry)
|
|
{
|
|
if (iProp == 0)
|
|
{
|
|
pkey->pid = value->pid;
|
|
break;
|
|
}
|
|
|
|
iProp--;
|
|
}
|
|
|
|
hr = S_OK;
|
|
}
|
|
else
|
|
hr = E_INVALIDARG;
|
|
|
|
LeaveCriticalSection(&This->lock);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT PropertyStore_LookupValue(PropertyStore *This, REFPROPERTYKEY key,
|
|
int insert, propstore_value **result)
|
|
{
|
|
propstore_format *format=NULL, *format_candidate;
|
|
propstore_value *value=NULL, *value_candidate;
|
|
|
|
if (IsEqualGUID(&key->fmtid, &FMTID_NamedProperties))
|
|
{
|
|
/* This is used in the property store format [MS-PROPSTORE]
|
|
* for named values and probably gets special treatment. */
|
|
ERR("don't know how to handle FMTID_NamedProperties\n");
|
|
return E_FAIL;
|
|
}
|
|
|
|
LIST_FOR_EACH_ENTRY(format_candidate, &This->formats, propstore_format, entry)
|
|
{
|
|
if (IsEqualGUID(&format_candidate->fmtid, &key->fmtid))
|
|
{
|
|
format = format_candidate;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!format)
|
|
{
|
|
if (!insert)
|
|
return TYPE_E_ELEMENTNOTFOUND;
|
|
|
|
format = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*format));
|
|
if (!format)
|
|
return E_OUTOFMEMORY;
|
|
|
|
format->fmtid = key->fmtid;
|
|
list_init(&format->values);
|
|
list_add_tail(&This->formats, &format->entry);
|
|
}
|
|
|
|
LIST_FOR_EACH_ENTRY(value_candidate, &format->values, propstore_value, entry)
|
|
{
|
|
if (value_candidate->pid == key->pid)
|
|
{
|
|
value = value_candidate;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!value)
|
|
{
|
|
if (!insert)
|
|
return TYPE_E_ELEMENTNOTFOUND;
|
|
|
|
value = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*value));
|
|
if (!value)
|
|
return E_OUTOFMEMORY;
|
|
|
|
value->pid = key->pid;
|
|
list_add_tail(&format->values, &value->entry);
|
|
format->count++;
|
|
}
|
|
|
|
*result = value;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI PropertyStore_GetValue(IPropertyStoreCache *iface,
|
|
REFPROPERTYKEY key, PROPVARIANT *pv)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
propstore_value *value;
|
|
HRESULT hr;
|
|
|
|
TRACE("%p,%p,%p\n", iface, key, pv);
|
|
|
|
if (!pv)
|
|
return E_POINTER;
|
|
|
|
EnterCriticalSection(&This->lock);
|
|
|
|
hr = PropertyStore_LookupValue(This, key, 0, &value);
|
|
|
|
if (SUCCEEDED(hr))
|
|
hr = PropVariantCopy(pv, &value->propvar);
|
|
else if (hr == TYPE_E_ELEMENTNOTFOUND)
|
|
{
|
|
PropVariantInit(pv);
|
|
hr = S_OK;
|
|
}
|
|
|
|
LeaveCriticalSection(&This->lock);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI PropertyStore_SetValue(IPropertyStoreCache *iface,
|
|
REFPROPERTYKEY key, REFPROPVARIANT propvar)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
propstore_value *value;
|
|
HRESULT hr;
|
|
|
|
TRACE("%p,%p,%p\n", iface, key, propvar);
|
|
|
|
EnterCriticalSection(&This->lock);
|
|
|
|
hr = PropertyStore_LookupValue(This, key, 1, &value);
|
|
|
|
if (SUCCEEDED(hr))
|
|
hr = PropVariantCopy(&value->propvar, propvar);
|
|
|
|
LeaveCriticalSection(&This->lock);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI PropertyStore_Commit(IPropertyStoreCache *iface)
|
|
{
|
|
FIXME("%p: stub\n", iface);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI PropertyStore_GetState(IPropertyStoreCache *iface,
|
|
REFPROPERTYKEY key, PSC_STATE *pstate)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
propstore_value *value;
|
|
HRESULT hr;
|
|
|
|
TRACE("%p,%p,%p\n", iface, key, pstate);
|
|
|
|
EnterCriticalSection(&This->lock);
|
|
|
|
hr = PropertyStore_LookupValue(This, key, 0, &value);
|
|
|
|
if (SUCCEEDED(hr))
|
|
*pstate = value->state;
|
|
|
|
LeaveCriticalSection(&This->lock);
|
|
|
|
if (FAILED(hr))
|
|
*pstate = PSC_NORMAL;
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI PropertyStore_GetValueAndState(IPropertyStoreCache *iface,
|
|
REFPROPERTYKEY key, PROPVARIANT *ppropvar, PSC_STATE *pstate)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
propstore_value *value;
|
|
HRESULT hr;
|
|
|
|
TRACE("%p,%p,%p,%p\n", iface, key, ppropvar, pstate);
|
|
|
|
EnterCriticalSection(&This->lock);
|
|
|
|
hr = PropertyStore_LookupValue(This, key, 0, &value);
|
|
|
|
if (SUCCEEDED(hr))
|
|
hr = PropVariantCopy(ppropvar, &value->propvar);
|
|
|
|
if (SUCCEEDED(hr))
|
|
*pstate = value->state;
|
|
|
|
LeaveCriticalSection(&This->lock);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
PropVariantInit(ppropvar);
|
|
*pstate = PSC_NORMAL;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI PropertyStore_SetState(IPropertyStoreCache *iface,
|
|
REFPROPERTYKEY key, PSC_STATE pstate)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
propstore_value *value;
|
|
HRESULT hr;
|
|
|
|
TRACE("%p,%p,%d\n", iface, key, pstate);
|
|
|
|
EnterCriticalSection(&This->lock);
|
|
|
|
hr = PropertyStore_LookupValue(This, key, 0, &value);
|
|
|
|
if (SUCCEEDED(hr))
|
|
value->state = pstate;
|
|
|
|
LeaveCriticalSection(&This->lock);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI PropertyStore_SetValueAndState(IPropertyStoreCache *iface,
|
|
REFPROPERTYKEY key, const PROPVARIANT *ppropvar, PSC_STATE state)
|
|
{
|
|
PropertyStore *This = impl_from_IPropertyStoreCache(iface);
|
|
propstore_value *value;
|
|
HRESULT hr;
|
|
|
|
TRACE("%p,%p,%p,%d\n", iface, key, ppropvar, state);
|
|
|
|
EnterCriticalSection(&This->lock);
|
|
|
|
hr = PropertyStore_LookupValue(This, key, 1, &value);
|
|
|
|
if (SUCCEEDED(hr))
|
|
hr = PropVariantCopy(&value->propvar, ppropvar);
|
|
|
|
if (SUCCEEDED(hr))
|
|
value->state = state;
|
|
|
|
LeaveCriticalSection(&This->lock);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static const IPropertyStoreCacheVtbl PropertyStore_Vtbl = {
|
|
PropertyStore_QueryInterface,
|
|
PropertyStore_AddRef,
|
|
PropertyStore_Release,
|
|
PropertyStore_GetCount,
|
|
PropertyStore_GetAt,
|
|
PropertyStore_GetValue,
|
|
PropertyStore_SetValue,
|
|
PropertyStore_Commit,
|
|
PropertyStore_GetState,
|
|
PropertyStore_GetValueAndState,
|
|
PropertyStore_SetState,
|
|
PropertyStore_SetValueAndState
|
|
};
|
|
|
|
HRESULT PropertyStore_CreateInstance(IUnknown *pUnkOuter, REFIID iid, void** ppv)
|
|
{
|
|
PropertyStore *This;
|
|
HRESULT ret;
|
|
|
|
TRACE("(%p,%s,%p)\n", pUnkOuter, debugstr_guid(iid), ppv);
|
|
|
|
*ppv = NULL;
|
|
|
|
if (pUnkOuter) return CLASS_E_NOAGGREGATION;
|
|
|
|
This = HeapAlloc(GetProcessHeap(), 0, sizeof(PropertyStore));
|
|
if (!This) return E_OUTOFMEMORY;
|
|
|
|
This->IPropertyStoreCache_iface.lpVtbl = &PropertyStore_Vtbl;
|
|
This->ref = 1;
|
|
InitializeCriticalSection(&This->lock);
|
|
This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": PropertyStore.lock");
|
|
list_init(&This->formats);
|
|
|
|
ret = IPropertyStoreCache_QueryInterface(&This->IPropertyStoreCache_iface, iid, ppv);
|
|
IPropertyStoreCache_Release(&This->IPropertyStoreCache_iface);
|
|
|
|
return ret;
|
|
}
|