Sweden-Number/dlls/wbemdisp/locator.c

2463 lines
64 KiB
C

/*
* Copyright 2013 Hans Leidekker 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 <stdarg.h>
#include "windef.h"
#include "winbase.h"
#include "initguid.h"
#include "objbase.h"
#include "wmiutils.h"
#include "wbemcli.h"
#include "wbemdisp.h"
#include "wine/debug.h"
#include "wine/heap.h"
#include "wbemdisp_private.h"
#include "wbemdisp_classes.h"
WINE_DEFAULT_DEBUG_CHANNEL(wbemdisp);
static HRESULT EnumVARIANT_create( IEnumWbemClassObject *, IEnumVARIANT ** );
static HRESULT ISWbemSecurity_create( ISWbemSecurity ** );
enum type_id
{
ISWbemLocator_tid,
ISWbemObject_tid,
ISWbemObjectSet_tid,
ISWbemProperty_tid,
ISWbemPropertySet_tid,
ISWbemServices_tid,
ISWbemSecurity_tid,
last_tid
};
static ITypeLib *wbemdisp_typelib;
static ITypeInfo *wbemdisp_typeinfo[last_tid];
static REFIID wbemdisp_tid_id[] =
{
&IID_ISWbemLocator,
&IID_ISWbemObject,
&IID_ISWbemObjectSet,
&IID_ISWbemProperty,
&IID_ISWbemPropertySet,
&IID_ISWbemServices,
&IID_ISWbemSecurity
};
static HRESULT get_typeinfo( enum type_id tid, ITypeInfo **ret )
{
HRESULT hr;
if (!wbemdisp_typelib)
{
ITypeLib *typelib;
hr = LoadRegTypeLib( &LIBID_WbemScripting, 1, 2, LOCALE_SYSTEM_DEFAULT, &typelib );
if (FAILED( hr ))
{
ERR( "LoadRegTypeLib failed: %08x\n", hr );
return hr;
}
if (InterlockedCompareExchangePointer( (void **)&wbemdisp_typelib, typelib, NULL ))
ITypeLib_Release( typelib );
}
if (!wbemdisp_typeinfo[tid])
{
ITypeInfo *typeinfo;
hr = ITypeLib_GetTypeInfoOfGuid( wbemdisp_typelib, wbemdisp_tid_id[tid], &typeinfo );
if (FAILED( hr ))
{
ERR( "GetTypeInfoOfGuid(%s) failed: %08x\n", debugstr_guid(wbemdisp_tid_id[tid]), hr );
return hr;
}
if (InterlockedCompareExchangePointer( (void **)(wbemdisp_typeinfo + tid), typeinfo, NULL ))
ITypeInfo_Release( typeinfo );
}
*ret = wbemdisp_typeinfo[tid];
ITypeInfo_AddRef( *ret );
return S_OK;
}
struct property
{
ISWbemProperty ISWbemProperty_iface;
LONG refs;
IWbemClassObject *object;
BSTR name;
};
static inline struct property *impl_from_ISWbemProperty( ISWbemProperty *iface )
{
return CONTAINING_RECORD( iface, struct property, ISWbemProperty_iface );
}
static ULONG WINAPI property_AddRef( ISWbemProperty *iface )
{
struct property *property = impl_from_ISWbemProperty( iface );
return InterlockedIncrement( &property->refs );
}
static ULONG WINAPI property_Release( ISWbemProperty *iface )
{
struct property *property = impl_from_ISWbemProperty( iface );
LONG refs = InterlockedDecrement( &property->refs );
if (!refs)
{
TRACE( "destroying %p\n", property );
IWbemClassObject_Release( property->object );
SysFreeString( property->name );
heap_free( property );
}
return refs;
}
static HRESULT WINAPI property_QueryInterface( ISWbemProperty *iface, REFIID riid, void **obj )
{
struct property *property = impl_from_ISWbemProperty( iface );
TRACE( "%p %s %p\n", property, debugstr_guid(riid), obj );
if (IsEqualGUID( riid, &IID_ISWbemProperty ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ))
{
*obj = iface;
}
else
{
FIXME( "interface %s not implemented\n", debugstr_guid(riid) );
return E_NOINTERFACE;
}
ISWbemProperty_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI property_GetTypeInfoCount( ISWbemProperty *iface, UINT *count )
{
struct property *property = impl_from_ISWbemProperty( iface );
TRACE( "%p, %p\n", property, count );
*count = 1;
return S_OK;
}
static HRESULT WINAPI property_GetTypeInfo( ISWbemProperty *iface, UINT index,
LCID lcid, ITypeInfo **info )
{
struct property *property = impl_from_ISWbemProperty( iface );
TRACE( "%p, %u, %u, %p\n", property, index, lcid, info );
return get_typeinfo( ISWbemProperty_tid, info );
}
static HRESULT WINAPI property_GetIDsOfNames( ISWbemProperty *iface, REFIID riid, LPOLESTR *names,
UINT count, LCID lcid, DISPID *dispid )
{
struct property *property = impl_from_ISWbemProperty( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %s, %p, %u, %u, %p\n", property, debugstr_guid(riid), names, count, lcid, dispid );
if (!names || !count || !dispid) return E_INVALIDARG;
hr = get_typeinfo( ISWbemProperty_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, names, count, dispid );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI property_Invoke( ISWbemProperty *iface, DISPID member, REFIID riid,
LCID lcid, WORD flags, DISPPARAMS *params,
VARIANT *result, EXCEPINFO *excep_info, UINT *arg_err )
{
struct property *property = impl_from_ISWbemProperty( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %d, %s, %d, %d, %p, %p, %p, %p\n", property, member, debugstr_guid(riid),
lcid, flags, params, result, excep_info, arg_err );
hr = get_typeinfo( ISWbemProperty_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &property->ISWbemProperty_iface, member, flags,
params, result, excep_info, arg_err );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI property_get_Value( ISWbemProperty *iface, VARIANT *value )
{
struct property *property = impl_from_ISWbemProperty( iface );
TRACE( "%p %p\n", property, value );
return IWbemClassObject_Get( property->object, property->name, 0, value, NULL, NULL );
}
static HRESULT WINAPI property_put_Value( ISWbemProperty *iface, VARIANT *varValue )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI property_get_Name( ISWbemProperty *iface, BSTR *strName )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI property_get_IsLocal( ISWbemProperty *iface, VARIANT_BOOL *bIsLocal )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI property_get_Origin( ISWbemProperty *iface, BSTR *strOrigin )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI property_get_CIMType( ISWbemProperty *iface, WbemCimtypeEnum *iCimType )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI property_get_Qualifiers_( ISWbemProperty *iface, ISWbemQualifierSet **objWbemQualifierSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI property_get_IsArray( ISWbemProperty *iface, VARIANT_BOOL *bIsArray )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static const ISWbemPropertyVtbl property_vtbl =
{
property_QueryInterface,
property_AddRef,
property_Release,
property_GetTypeInfoCount,
property_GetTypeInfo,
property_GetIDsOfNames,
property_Invoke,
property_get_Value,
property_put_Value,
property_get_Name,
property_get_IsLocal,
property_get_Origin,
property_get_CIMType,
property_get_Qualifiers_,
property_get_IsArray
};
static HRESULT SWbemProperty_create( IWbemClassObject *wbem_object, BSTR name, ISWbemProperty **obj )
{
struct property *property;
TRACE( "%p, %p\n", obj, wbem_object );
if (!(property = heap_alloc( sizeof(*property) ))) return E_OUTOFMEMORY;
property->ISWbemProperty_iface.lpVtbl = &property_vtbl;
property->refs = 1;
property->object = wbem_object;
IWbemClassObject_AddRef( property->object );
property->name = SysAllocStringLen( name, SysStringLen( name ) );
*obj = &property->ISWbemProperty_iface;
TRACE( "returning iface %p\n", *obj );
return S_OK;
}
struct propertyset
{
ISWbemPropertySet ISWbemPropertySet_iface;
LONG refs;
IWbemClassObject *object;
};
static inline struct propertyset *impl_from_ISWbemPropertySet(
ISWbemPropertySet *iface )
{
return CONTAINING_RECORD( iface, struct propertyset, ISWbemPropertySet_iface );
}
static ULONG WINAPI propertyset_AddRef( ISWbemPropertySet *iface )
{
struct propertyset *propertyset = impl_from_ISWbemPropertySet( iface );
return InterlockedIncrement( &propertyset->refs );
}
static ULONG WINAPI propertyset_Release( ISWbemPropertySet *iface )
{
struct propertyset *propertyset = impl_from_ISWbemPropertySet( iface );
LONG refs = InterlockedDecrement( &propertyset->refs );
if (!refs)
{
TRACE( "destroying %p\n", propertyset );
IWbemClassObject_Release( propertyset->object );
heap_free( propertyset );
}
return refs;
}
static HRESULT WINAPI propertyset_QueryInterface( ISWbemPropertySet *iface,
REFIID riid, void **obj )
{
struct propertyset *propertyset = impl_from_ISWbemPropertySet( iface );
TRACE( "%p %s %p\n", propertyset, debugstr_guid(riid), obj );
if (IsEqualGUID( riid, &IID_ISWbemPropertySet ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ))
{
*obj = iface;
}
else
{
FIXME( "interface %s not implemented\n", debugstr_guid(riid) );
return E_NOINTERFACE;
}
ISWbemPropertySet_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI propertyset_GetTypeInfoCount( ISWbemPropertySet *iface, UINT *count )
{
struct propertyset *propertyset = impl_from_ISWbemPropertySet( iface );
TRACE( "%p, %p\n", propertyset, count );
*count = 1;
return S_OK;
}
static HRESULT WINAPI propertyset_GetTypeInfo( ISWbemPropertySet *iface,
UINT index, LCID lcid, ITypeInfo **info )
{
struct propertyset *propertyset = impl_from_ISWbemPropertySet( iface );
TRACE( "%p, %u, %u, %p\n", propertyset, index, lcid, info );
return get_typeinfo( ISWbemPropertySet_tid, info );
}
static HRESULT WINAPI propertyset_GetIDsOfNames( ISWbemPropertySet *iface, REFIID riid, LPOLESTR *names,
UINT count, LCID lcid, DISPID *dispid )
{
struct propertyset *propertyset = impl_from_ISWbemPropertySet( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %s, %p, %u, %u, %p\n", propertyset, debugstr_guid(riid), names, count, lcid, dispid );
if (!names || !count || !dispid) return E_INVALIDARG;
hr = get_typeinfo( ISWbemPropertySet_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, names, count, dispid );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI propertyset_Invoke( ISWbemPropertySet *iface, DISPID member, REFIID riid,
LCID lcid, WORD flags, DISPPARAMS *params,
VARIANT *result, EXCEPINFO *excep_info, UINT *arg_err )
{
struct propertyset *propertyset = impl_from_ISWbemPropertySet( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %d, %s, %d, %d, %p, %p, %p, %p\n", propertyset, member, debugstr_guid(riid),
lcid, flags, params, result, excep_info, arg_err );
hr = get_typeinfo( ISWbemPropertySet_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &propertyset->ISWbemPropertySet_iface, member, flags,
params, result, excep_info, arg_err );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI propertyset_get__NewEnum( ISWbemPropertySet *iface, IUnknown **unk )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI propertyset_Item( ISWbemPropertySet *iface, BSTR name,
LONG flags, ISWbemProperty **prop )
{
struct propertyset *propertyset = impl_from_ISWbemPropertySet( iface );
HRESULT hr;
VARIANT var;
TRACE( "%p, %s, %08x, %p\n", propertyset, debugstr_w(name), flags, prop );
hr = IWbemClassObject_Get( propertyset->object, name, 0, &var, NULL, NULL );
if (SUCCEEDED(hr))
{
hr = SWbemProperty_create( propertyset->object, name, prop );
VariantClear( &var );
}
return hr;
}
static HRESULT WINAPI propertyset_get_Count( ISWbemPropertySet *iface, LONG *count )
{
static const WCHAR propcountW[] = {'_','_','P','R','O','P','E','R','T','Y','_','C','O','U','N','T',0};
struct propertyset *propertyset = impl_from_ISWbemPropertySet( iface );
HRESULT hr;
VARIANT val;
TRACE( "%p, %p\n", propertyset, count );
hr = IWbemClassObject_Get( propertyset->object, propcountW, 0, &val, NULL, NULL );
if (SUCCEEDED(hr))
{
*count = V_I4( &val );
}
return hr;
}
static HRESULT WINAPI propertyset_Add( ISWbemPropertySet *iface, BSTR name, WbemCimtypeEnum type,
VARIANT_BOOL is_array, LONG flags, ISWbemProperty **prop )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI propertyset_Remove( ISWbemPropertySet *iface, BSTR name, LONG flags )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static const ISWbemPropertySetVtbl propertyset_vtbl =
{
propertyset_QueryInterface,
propertyset_AddRef,
propertyset_Release,
propertyset_GetTypeInfoCount,
propertyset_GetTypeInfo,
propertyset_GetIDsOfNames,
propertyset_Invoke,
propertyset_get__NewEnum,
propertyset_Item,
propertyset_get_Count,
propertyset_Add,
propertyset_Remove
};
static HRESULT SWbemPropertySet_create( IWbemClassObject *wbem_object, ISWbemPropertySet **obj )
{
struct propertyset *propertyset;
TRACE( "%p, %p\n", obj, wbem_object );
if (!(propertyset = heap_alloc( sizeof(*propertyset) ))) return E_OUTOFMEMORY;
propertyset->ISWbemPropertySet_iface.lpVtbl = &propertyset_vtbl;
propertyset->refs = 1;
propertyset->object = wbem_object;
IWbemClassObject_AddRef( propertyset->object );
*obj = &propertyset->ISWbemPropertySet_iface;
TRACE( "returning iface %p\n", *obj );
return S_OK;
}
#define DISPID_BASE 0x1800000
#define DISPID_BASE_METHOD 0x1000000
struct member
{
BSTR name;
BOOL is_method;
DISPID dispid;
};
struct object
{
ISWbemObject ISWbemObject_iface;
LONG refs;
IWbemClassObject *object;
struct member *members;
UINT nb_members;
DISPID last_dispid;
DISPID last_dispid_method;
};
static inline struct object *impl_from_ISWbemObject(
ISWbemObject *iface )
{
return CONTAINING_RECORD( iface, struct object, ISWbemObject_iface );
}
static ULONG WINAPI object_AddRef(
ISWbemObject *iface )
{
struct object *object = impl_from_ISWbemObject( iface );
return InterlockedIncrement( &object->refs );
}
static ULONG WINAPI object_Release(
ISWbemObject *iface )
{
struct object *object = impl_from_ISWbemObject( iface );
LONG refs = InterlockedDecrement( &object->refs );
if (!refs)
{
UINT i;
TRACE( "destroying %p\n", object );
IWbemClassObject_Release( object->object );
for (i = 0; i < object->nb_members; i++) SysFreeString( object->members[i].name );
heap_free( object->members );
heap_free( object );
}
return refs;
}
static HRESULT WINAPI object_QueryInterface(
ISWbemObject *iface,
REFIID riid,
void **ppvObject )
{
struct object *object = impl_from_ISWbemObject( iface );
TRACE( "%p %s %p\n", object, debugstr_guid(riid), ppvObject );
if (IsEqualGUID( riid, &IID_ISWbemObject ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ))
{
*ppvObject = iface;
}
else
{
FIXME( "interface %s not implemented\n", debugstr_guid(riid) );
return E_NOINTERFACE;
}
ISWbemObject_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI object_GetTypeInfoCount(
ISWbemObject *iface,
UINT *count )
{
struct object *object = impl_from_ISWbemObject( iface );
TRACE( "%p, %p\n", object, count );
*count = 1;
return S_OK;
}
static HRESULT WINAPI object_GetTypeInfo(
ISWbemObject *iface,
UINT index,
LCID lcid,
ITypeInfo **info )
{
struct object *object = impl_from_ISWbemObject( iface );
FIXME( "%p, %u, %u, %p\n", object, index, lcid, info );
return E_NOTIMPL;
}
static HRESULT init_members( struct object *object )
{
IWbemClassObject *sig_in, *sig_out;
LONG i = 0, count = 0;
BSTR name;
HRESULT hr;
if (object->members) return S_OK;
hr = IWbemClassObject_BeginEnumeration( object->object, 0 );
if (SUCCEEDED( hr ))
{
while (IWbemClassObject_Next( object->object, 0, NULL, NULL, NULL, NULL ) == S_OK) count++;
IWbemClassObject_EndEnumeration( object->object );
}
hr = IWbemClassObject_BeginMethodEnumeration( object->object, 0 );
if (SUCCEEDED( hr ))
{
while (IWbemClassObject_NextMethod( object->object, 0, &name, &sig_in, &sig_out ) == S_OK)
{
count++;
SysFreeString( name );
if (sig_in) IWbemClassObject_Release( sig_in );
if (sig_out) IWbemClassObject_Release( sig_out );
}
IWbemClassObject_EndMethodEnumeration( object->object );
}
if (!(object->members = heap_alloc( sizeof(struct member) * count ))) return E_OUTOFMEMORY;
hr = IWbemClassObject_BeginEnumeration( object->object, 0 );
if (SUCCEEDED( hr ))
{
while (IWbemClassObject_Next( object->object, 0, &name, NULL, NULL, NULL ) == S_OK)
{
object->members[i].name = name;
object->members[i].is_method = FALSE;
object->members[i].dispid = 0;
if (++i > count)
{
IWbemClassObject_EndEnumeration( object->object );
goto error;
}
TRACE( "added property %s\n", debugstr_w(name) );
}
IWbemClassObject_EndEnumeration( object->object );
}
hr = IWbemClassObject_BeginMethodEnumeration( object->object, 0 );
if (SUCCEEDED( hr ))
{
while (IWbemClassObject_NextMethod( object->object, 0, &name, &sig_in, &sig_out ) == S_OK)
{
object->members[i].name = name;
object->members[i].is_method = TRUE;
object->members[i].dispid = 0;
if (++i > count)
{
IWbemClassObject_EndMethodEnumeration( object->object );
goto error;
}
if (sig_in) IWbemClassObject_Release( sig_in );
if (sig_out) IWbemClassObject_Release( sig_out );
TRACE( "added method %s\n", debugstr_w(name) );
}
IWbemClassObject_EndMethodEnumeration( object->object );
}
object->nb_members = count;
TRACE( "added %u members\n", object->nb_members );
return S_OK;
error:
for (--i; i >= 0; i--) SysFreeString( object->members[i].name );
heap_free( object->members );
object->members = NULL;
object->nb_members = 0;
return E_FAIL;
}
static DISPID get_member_dispid( struct object *object, const WCHAR *name )
{
UINT i;
for (i = 0; i < object->nb_members; i++)
{
if (!wcsicmp( object->members[i].name, name ))
{
if (!object->members[i].dispid)
{
if (object->members[i].is_method)
object->members[i].dispid = ++object->last_dispid_method;
else
object->members[i].dispid = ++object->last_dispid;
}
return object->members[i].dispid;
}
}
return DISPID_UNKNOWN;
}
static HRESULT WINAPI object_GetIDsOfNames(
ISWbemObject *iface,
REFIID riid,
LPOLESTR *names,
UINT count,
LCID lcid,
DISPID *dispid )
{
struct object *object = impl_from_ISWbemObject( iface );
HRESULT hr;
UINT i;
ITypeInfo *typeinfo;
TRACE( "%p, %s, %p, %u, %u, %p\n", object, debugstr_guid(riid), names, count, lcid, dispid );
if (!names || !count || !dispid) return E_INVALIDARG;
hr = init_members( object );
if (FAILED( hr )) return hr;
hr = get_typeinfo( ISWbemObject_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, names, count, dispid );
ITypeInfo_Release( typeinfo );
}
if (SUCCEEDED(hr)) return hr;
for (i = 0; i < count; i++)
{
if ((dispid[i] = get_member_dispid( object, names[i] )) == DISPID_UNKNOWN) break;
}
if (i != count) return DISP_E_UNKNOWNNAME;
return S_OK;
}
static BSTR get_member_name( struct object *object, DISPID dispid )
{
UINT i;
for (i = 0; i < object->nb_members; i++)
{
if (object->members[i].dispid == dispid) return object->members[i].name;
}
return NULL;
}
static HRESULT WINAPI object_Invoke(
ISWbemObject *iface,
DISPID member,
REFIID riid,
LCID lcid,
WORD flags,
DISPPARAMS *params,
VARIANT *result,
EXCEPINFO *excep_info,
UINT *arg_err )
{
struct object *object = impl_from_ISWbemObject( iface );
BSTR name;
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %x, %s, %u, %x, %p, %p, %p, %p\n", object, member, debugstr_guid(riid),
lcid, flags, params, result, excep_info, arg_err );
if (member <= DISPID_BASE_METHOD)
{
hr = get_typeinfo( ISWbemObject_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &object->ISWbemObject_iface, member, flags,
params, result, excep_info, arg_err );
ITypeInfo_Release( typeinfo );
}
return hr;
}
if (flags != (DISPATCH_METHOD|DISPATCH_PROPERTYGET))
{
FIXME( "flags %x not supported\n", flags );
return E_NOTIMPL;
}
if (!(name = get_member_name( object, member )))
return DISP_E_MEMBERNOTFOUND;
memset( params, 0, sizeof(*params) );
return IWbemClassObject_Get( object->object, name, 0, result, NULL, NULL );
}
static HRESULT WINAPI object_Put_(
ISWbemObject *iface,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObjectPath **objWbemObjectPath )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_PutAsync_(
ISWbemObject *iface,
IDispatch *objWbemSink,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_Delete_(
ISWbemObject *iface,
LONG iFlags,
IDispatch *objWbemNamedValueSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_DeleteAsync_(
ISWbemObject *iface,
IDispatch *objWbemSink,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_Instances_(
ISWbemObject *iface,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObjectSet **objWbemObjectSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_InstancesAsync_(
ISWbemObject *iface,
IDispatch *objWbemSink,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_Subclasses_(
ISWbemObject *iface,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObjectSet **objWbemObjectSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_SubclassesAsync_(
ISWbemObject *iface,
IDispatch *objWbemSink,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_Associators_(
ISWbemObject *iface,
BSTR strAssocClass,
BSTR strResultClass,
BSTR strResultRole,
BSTR strRole,
VARIANT_BOOL bClassesOnly,
VARIANT_BOOL bSchemaOnly,
BSTR strRequiredAssocQualifier,
BSTR strRequiredQualifier,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObjectSet **objWbemObjectSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_AssociatorsAsync_(
ISWbemObject *iface,
IDispatch *objWbemSink,
BSTR strAssocClass,
BSTR strResultClass,
BSTR strResultRole,
BSTR strRole,
VARIANT_BOOL bClassesOnly,
VARIANT_BOOL bSchemaOnly,
BSTR strRequiredAssocQualifier,
BSTR strRequiredQualifier,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_References_(
ISWbemObject *iface,
BSTR strResultClass,
BSTR strRole,
VARIANT_BOOL bClassesOnly,
VARIANT_BOOL bSchemaOnly,
BSTR strRequiredQualifier,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObjectSet **objWbemObjectSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_ReferencesAsync_(
ISWbemObject *iface,
IDispatch *objWbemSink,
BSTR strResultClass,
BSTR strRole,
VARIANT_BOOL bClassesOnly,
VARIANT_BOOL bSchemaOnly,
BSTR strRequiredQualifier,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_ExecMethod_(
ISWbemObject *iface,
BSTR strMethodName,
IDispatch *objWbemInParameters,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObject **objWbemOutParameters )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_ExecMethodAsync_(
ISWbemObject *iface,
IDispatch *objWbemSink,
BSTR strMethodName,
IDispatch *objWbemInParameters,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_Clone_(
ISWbemObject *iface,
ISWbemObject **objWbemObject )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_GetObjectText_(
ISWbemObject *iface,
LONG iFlags,
BSTR *strObjectText )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_SpawnDerivedClass_(
ISWbemObject *iface,
LONG iFlags,
ISWbemObject **objWbemObject )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_SpawnInstance_(
ISWbemObject *iface,
LONG iFlags,
ISWbemObject **objWbemObject )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_CompareTo_(
ISWbemObject *iface,
IDispatch *objWbemObject,
LONG iFlags,
VARIANT_BOOL *bResult )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_get_Qualifiers_(
ISWbemObject *iface,
ISWbemQualifierSet **objWbemQualifierSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_get_Properties_( ISWbemObject *iface, ISWbemPropertySet **prop_set )
{
struct object *object = impl_from_ISWbemObject( iface );
TRACE( "%p, %p\n", object, prop_set );
return SWbemPropertySet_create( object->object, prop_set );
}
static HRESULT WINAPI object_get_Methods_(
ISWbemObject *iface,
ISWbemMethodSet **objWbemMethodSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_get_Derivation_(
ISWbemObject *iface,
VARIANT *strClassNameArray )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_get_Path_(
ISWbemObject *iface,
ISWbemObjectPath **objWbemObjectPath )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI object_get_Security_(
ISWbemObject *iface,
ISWbemSecurity **objWbemSecurity )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static const ISWbemObjectVtbl object_vtbl =
{
object_QueryInterface,
object_AddRef,
object_Release,
object_GetTypeInfoCount,
object_GetTypeInfo,
object_GetIDsOfNames,
object_Invoke,
object_Put_,
object_PutAsync_,
object_Delete_,
object_DeleteAsync_,
object_Instances_,
object_InstancesAsync_,
object_Subclasses_,
object_SubclassesAsync_,
object_Associators_,
object_AssociatorsAsync_,
object_References_,
object_ReferencesAsync_,
object_ExecMethod_,
object_ExecMethodAsync_,
object_Clone_,
object_GetObjectText_,
object_SpawnDerivedClass_,
object_SpawnInstance_,
object_CompareTo_,
object_get_Qualifiers_,
object_get_Properties_,
object_get_Methods_,
object_get_Derivation_,
object_get_Path_,
object_get_Security_
};
static HRESULT SWbemObject_create( IWbemClassObject *wbem_object, ISWbemObject **obj )
{
struct object *object;
TRACE( "%p, %p\n", obj, wbem_object );
if (!(object = heap_alloc( sizeof(*object) ))) return E_OUTOFMEMORY;
object->ISWbemObject_iface.lpVtbl = &object_vtbl;
object->refs = 1;
object->object = wbem_object;
IWbemClassObject_AddRef( object->object );
object->members = NULL;
object->nb_members = 0;
object->last_dispid = DISPID_BASE;
object->last_dispid_method = DISPID_BASE_METHOD;
*obj = &object->ISWbemObject_iface;
TRACE( "returning iface %p\n", *obj );
return S_OK;
}
struct objectset
{
ISWbemObjectSet ISWbemObjectSet_iface;
LONG refs;
IEnumWbemClassObject *objectenum;
LONG count;
};
static inline struct objectset *impl_from_ISWbemObjectSet(
ISWbemObjectSet *iface )
{
return CONTAINING_RECORD( iface, struct objectset, ISWbemObjectSet_iface );
}
static ULONG WINAPI objectset_AddRef(
ISWbemObjectSet *iface )
{
struct objectset *objectset = impl_from_ISWbemObjectSet( iface );
return InterlockedIncrement( &objectset->refs );
}
static ULONG WINAPI objectset_Release(
ISWbemObjectSet *iface )
{
struct objectset *objectset = impl_from_ISWbemObjectSet( iface );
LONG refs = InterlockedDecrement( &objectset->refs );
if (!refs)
{
TRACE( "destroying %p\n", objectset );
IEnumWbemClassObject_Release( objectset->objectenum );
heap_free( objectset );
}
return refs;
}
static HRESULT WINAPI objectset_QueryInterface(
ISWbemObjectSet *iface,
REFIID riid,
void **ppvObject )
{
struct objectset *objectset = impl_from_ISWbemObjectSet( iface );
TRACE( "%p %s %p\n", objectset, debugstr_guid(riid), ppvObject );
if (IsEqualGUID( riid, &IID_ISWbemObjectSet ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ))
{
*ppvObject = iface;
}
else
{
FIXME( "interface %s not implemented\n", debugstr_guid(riid) );
return E_NOINTERFACE;
}
ISWbemObjectSet_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI objectset_GetTypeInfoCount(
ISWbemObjectSet *iface,
UINT *count )
{
struct objectset *objectset = impl_from_ISWbemObjectSet( iface );
TRACE( "%p, %p\n", objectset, count );
*count = 1;
return S_OK;
}
static HRESULT WINAPI objectset_GetTypeInfo(
ISWbemObjectSet *iface,
UINT index,
LCID lcid,
ITypeInfo **info )
{
struct objectset *objectset = impl_from_ISWbemObjectSet( iface );
TRACE( "%p, %u, %u, %p\n", objectset, index, lcid, info );
return get_typeinfo( ISWbemObjectSet_tid, info );
}
static HRESULT WINAPI objectset_GetIDsOfNames(
ISWbemObjectSet *iface,
REFIID riid,
LPOLESTR *names,
UINT count,
LCID lcid,
DISPID *dispid )
{
struct objectset *objectset = impl_from_ISWbemObjectSet( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %s, %p, %u, %u, %p\n", objectset, debugstr_guid(riid), names, count, lcid, dispid );
if (!names || !count || !dispid) return E_INVALIDARG;
hr = get_typeinfo( ISWbemObjectSet_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, names, count, dispid );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI objectset_Invoke(
ISWbemObjectSet *iface,
DISPID member,
REFIID riid,
LCID lcid,
WORD flags,
DISPPARAMS *params,
VARIANT *result,
EXCEPINFO *excep_info,
UINT *arg_err )
{
struct objectset *objectset = impl_from_ISWbemObjectSet( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %d, %s, %d, %d, %p, %p, %p, %p\n", objectset, member, debugstr_guid(riid),
lcid, flags, params, result, excep_info, arg_err );
hr = get_typeinfo( ISWbemObjectSet_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &objectset->ISWbemObjectSet_iface, member, flags,
params, result, excep_info, arg_err );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI objectset_get__NewEnum(
ISWbemObjectSet *iface,
IUnknown **pUnk )
{
struct objectset *objectset = impl_from_ISWbemObjectSet( iface );
IEnumWbemClassObject *objectenum;
HRESULT hr;
TRACE( "%p, %p\n", objectset, pUnk );
hr = IEnumWbemClassObject_Clone( objectset->objectenum, &objectenum );
if (FAILED( hr )) return hr;
hr = EnumVARIANT_create( objectenum, (IEnumVARIANT **)pUnk );
IEnumWbemClassObject_Release( objectenum );
return hr;
}
static HRESULT WINAPI objectset_Item(
ISWbemObjectSet *iface,
BSTR strObjectPath,
LONG iFlags,
ISWbemObject **objWbemObject )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI objectset_get_Count(
ISWbemObjectSet *iface,
LONG *iCount )
{
struct objectset *objectset = impl_from_ISWbemObjectSet( iface );
TRACE( "%p, %p\n", objectset, iCount );
*iCount = objectset->count;
return S_OK;
}
static HRESULT WINAPI objectset_get_Security_(
ISWbemObjectSet *iface,
ISWbemSecurity **objWbemSecurity )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI objectset_ItemIndex(
ISWbemObjectSet *iface,
LONG lIndex,
ISWbemObject **objWbemObject )
{
struct objectset *objectset = impl_from_ISWbemObjectSet( iface );
LONG count;
HRESULT hr;
IEnumVARIANT *enum_var;
VARIANT var;
TRACE( "%p, %d, %p\n", objectset, lIndex, objWbemObject );
*objWbemObject = NULL;
hr = ISWbemObjectSet_get_Count( iface, &count );
if (FAILED(hr)) return hr;
if (lIndex >= count) return WBEM_E_NOT_FOUND;
hr = ISWbemObjectSet_get__NewEnum( iface, (IUnknown **)&enum_var );
if (FAILED(hr)) return hr;
IEnumVARIANT_Reset( enum_var );
hr = IEnumVARIANT_Skip( enum_var, lIndex );
if (SUCCEEDED(hr))
hr = IEnumVARIANT_Next( enum_var, 1, &var, NULL );
IEnumVARIANT_Release( enum_var );
if (SUCCEEDED(hr))
{
if (V_VT( &var ) == VT_DISPATCH)
hr = IDispatch_QueryInterface( V_DISPATCH( &var ), &IID_ISWbemObject, (void **)objWbemObject );
else
hr = WBEM_E_NOT_FOUND;
VariantClear( &var );
}
return hr;
}
static const ISWbemObjectSetVtbl objectset_vtbl =
{
objectset_QueryInterface,
objectset_AddRef,
objectset_Release,
objectset_GetTypeInfoCount,
objectset_GetTypeInfo,
objectset_GetIDsOfNames,
objectset_Invoke,
objectset_get__NewEnum,
objectset_Item,
objectset_get_Count,
objectset_get_Security_,
objectset_ItemIndex
};
static LONG get_object_count( IEnumWbemClassObject *iter )
{
LONG count = 0;
while (IEnumWbemClassObject_Skip( iter, WBEM_INFINITE, 1 ) == S_OK) count++;
IEnumWbemClassObject_Reset( iter );
return count;
}
static HRESULT SWbemObjectSet_create( IEnumWbemClassObject *wbem_objectenum, ISWbemObjectSet **obj )
{
struct objectset *objectset;
TRACE( "%p, %p\n", obj, wbem_objectenum );
if (!(objectset = heap_alloc( sizeof(*objectset) ))) return E_OUTOFMEMORY;
objectset->ISWbemObjectSet_iface.lpVtbl = &objectset_vtbl;
objectset->refs = 1;
objectset->objectenum = wbem_objectenum;
IEnumWbemClassObject_AddRef( objectset->objectenum );
objectset->count = get_object_count( objectset->objectenum );
*obj = &objectset->ISWbemObjectSet_iface;
TRACE( "returning iface %p\n", *obj );
return S_OK;
}
struct enumvar
{
IEnumVARIANT IEnumVARIANT_iface;
LONG refs;
IEnumWbemClassObject *objectenum;
};
static inline struct enumvar *impl_from_IEnumVARIANT(
IEnumVARIANT *iface )
{
return CONTAINING_RECORD( iface, struct enumvar, IEnumVARIANT_iface );
}
static ULONG WINAPI enumvar_AddRef(
IEnumVARIANT *iface )
{
struct enumvar *enumvar = impl_from_IEnumVARIANT( iface );
return InterlockedIncrement( &enumvar->refs );
}
static ULONG WINAPI enumvar_Release(
IEnumVARIANT *iface )
{
struct enumvar *enumvar = impl_from_IEnumVARIANT( iface );
LONG refs = InterlockedDecrement( &enumvar->refs );
if (!refs)
{
TRACE( "destroying %p\n", enumvar );
IEnumWbemClassObject_Release( enumvar->objectenum );
heap_free( enumvar );
}
return refs;
}
static HRESULT WINAPI enumvar_QueryInterface(
IEnumVARIANT *iface,
REFIID riid,
void **ppvObject )
{
struct enumvar *enumvar = impl_from_IEnumVARIANT( iface );
TRACE( "%p %s %p\n", enumvar, debugstr_guid(riid), ppvObject );
if (IsEqualGUID( riid, &IID_IEnumVARIANT ) ||
IsEqualGUID( riid, &IID_IUnknown ))
{
*ppvObject = iface;
}
else
{
FIXME( "interface %s not implemented\n", debugstr_guid(riid) );
return E_NOINTERFACE;
}
IEnumVARIANT_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI enumvar_Next( IEnumVARIANT *iface, ULONG celt, VARIANT *var, ULONG *fetched )
{
struct enumvar *enumvar = impl_from_IEnumVARIANT( iface );
IWbemClassObject *obj;
ULONG count = 0;
TRACE( "%p, %u, %p, %p\n", iface, celt, var, fetched );
if (celt) IEnumWbemClassObject_Next( enumvar->objectenum, WBEM_INFINITE, 1, &obj, &count );
if (count)
{
ISWbemObject *sobj;
HRESULT hr;
hr = SWbemObject_create( obj, &sobj );
IWbemClassObject_Release( obj );
if (FAILED( hr )) return hr;
V_VT( var ) = VT_DISPATCH;
V_DISPATCH( var ) = (IDispatch *)sobj;
}
if (fetched) *fetched = count;
return (count < celt) ? S_FALSE : S_OK;
}
static HRESULT WINAPI enumvar_Skip( IEnumVARIANT *iface, ULONG celt )
{
struct enumvar *enumvar = impl_from_IEnumVARIANT( iface );
TRACE( "%p, %u\n", iface, celt );
return IEnumWbemClassObject_Skip( enumvar->objectenum, WBEM_INFINITE, celt );
}
static HRESULT WINAPI enumvar_Reset( IEnumVARIANT *iface )
{
struct enumvar *enumvar = impl_from_IEnumVARIANT( iface );
TRACE( "%p\n", iface );
return IEnumWbemClassObject_Reset( enumvar->objectenum );
}
static HRESULT WINAPI enumvar_Clone( IEnumVARIANT *iface, IEnumVARIANT **penum )
{
FIXME( "%p, %p\n", iface, penum );
return E_NOTIMPL;
}
static const struct IEnumVARIANTVtbl enumvar_vtbl =
{
enumvar_QueryInterface,
enumvar_AddRef,
enumvar_Release,
enumvar_Next,
enumvar_Skip,
enumvar_Reset,
enumvar_Clone
};
static HRESULT EnumVARIANT_create( IEnumWbemClassObject *objectenum, IEnumVARIANT **obj )
{
struct enumvar *enumvar;
if (!(enumvar = heap_alloc( sizeof(*enumvar) ))) return E_OUTOFMEMORY;
enumvar->IEnumVARIANT_iface.lpVtbl = &enumvar_vtbl;
enumvar->refs = 1;
enumvar->objectenum = objectenum;
IEnumWbemClassObject_AddRef( enumvar->objectenum );
*obj = &enumvar->IEnumVARIANT_iface;
TRACE( "returning iface %p\n", *obj );
return S_OK;
}
struct services
{
ISWbemServices ISWbemServices_iface;
LONG refs;
IWbemServices *services;
};
static inline struct services *impl_from_ISWbemServices(
ISWbemServices *iface )
{
return CONTAINING_RECORD( iface, struct services, ISWbemServices_iface );
}
static ULONG WINAPI services_AddRef(
ISWbemServices *iface )
{
struct services *services = impl_from_ISWbemServices( iface );
return InterlockedIncrement( &services->refs );
}
static ULONG WINAPI services_Release(
ISWbemServices *iface )
{
struct services *services = impl_from_ISWbemServices( iface );
LONG refs = InterlockedDecrement( &services->refs );
if (!refs)
{
TRACE( "destroying %p\n", services );
IWbemServices_Release( services->services );
heap_free( services );
}
return refs;
}
static HRESULT WINAPI services_QueryInterface(
ISWbemServices *iface,
REFIID riid,
void **ppvObject )
{
struct services *services = impl_from_ISWbemServices( iface );
TRACE( "%p %s %p\n", services, debugstr_guid(riid), ppvObject );
if (IsEqualGUID( riid, &IID_ISWbemServices ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ))
{
*ppvObject = iface;
}
else
{
FIXME( "interface %s not implemented\n", debugstr_guid(riid) );
return E_NOINTERFACE;
}
ISWbemServices_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI services_GetTypeInfoCount(
ISWbemServices *iface,
UINT *count )
{
struct services *services = impl_from_ISWbemServices( iface );
TRACE( "%p, %p\n", services, count );
*count = 1;
return S_OK;
}
static HRESULT WINAPI services_GetTypeInfo(
ISWbemServices *iface,
UINT index,
LCID lcid,
ITypeInfo **info )
{
struct services *services = impl_from_ISWbemServices( iface );
TRACE( "%p, %u, %u, %p\n", services, index, lcid, info );
return get_typeinfo( ISWbemServices_tid, info );
}
static HRESULT WINAPI services_GetIDsOfNames(
ISWbemServices *iface,
REFIID riid,
LPOLESTR *names,
UINT count,
LCID lcid,
DISPID *dispid )
{
struct services *services = impl_from_ISWbemServices( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %s, %p, %u, %u, %p\n", services, debugstr_guid(riid), names, count, lcid, dispid );
if (!names || !count || !dispid) return E_INVALIDARG;
hr = get_typeinfo( ISWbemServices_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, names, count, dispid );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI services_Invoke(
ISWbemServices *iface,
DISPID member,
REFIID riid,
LCID lcid,
WORD flags,
DISPPARAMS *params,
VARIANT *result,
EXCEPINFO *excep_info,
UINT *arg_err )
{
struct services *services = impl_from_ISWbemServices( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %d, %s, %d, %d, %p, %p, %p, %p\n", services, member, debugstr_guid(riid),
lcid, flags, params, result, excep_info, arg_err );
hr = get_typeinfo( ISWbemServices_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &services->ISWbemServices_iface, member, flags,
params, result, excep_info, arg_err );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI services_Get(
ISWbemServices *iface,
BSTR strObjectPath,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObject **objWbemObject )
{
struct services *services = impl_from_ISWbemServices( iface );
IWbemClassObject *obj;
HRESULT hr;
TRACE( "%p, %s, %d, %p, %p\n", iface, debugstr_w(strObjectPath), iFlags, objWbemNamedValueSet,
objWbemObject );
if (objWbemNamedValueSet) FIXME( "ignoring context\n" );
hr = IWbemServices_GetObject( services->services, strObjectPath, iFlags, NULL, &obj, NULL );
if (hr != S_OK) return hr;
hr = SWbemObject_create( obj, objWbemObject );
IWbemClassObject_Release( obj );
return hr;
}
static HRESULT WINAPI services_GetAsync(
ISWbemServices *iface,
IDispatch *objWbemSink,
BSTR strObjectPath,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_Delete(
ISWbemServices *iface,
BSTR strObjectPath,
LONG iFlags,
IDispatch *objWbemNamedValueSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_DeleteAsync(
ISWbemServices* This,
IDispatch *objWbemSink,
BSTR strObjectPath,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static BSTR build_query_string( const WCHAR *class )
{
static const WCHAR selectW[] = {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',0};
UINT len = lstrlenW(class) + ARRAY_SIZE(selectW);
BSTR ret;
if (!(ret = SysAllocStringLen( NULL, len ))) return NULL;
lstrcpyW( ret, selectW );
lstrcatW( ret, class );
return ret;
}
static HRESULT WINAPI services_InstancesOf(
ISWbemServices *iface,
BSTR strClass,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObjectSet **objWbemObjectSet )
{
static const WCHAR wqlW[] = {'W','Q','L',0};
BSTR query, wql = SysAllocString( wqlW );
HRESULT hr;
TRACE( "%p, %s, %x, %p, %p\n", iface, debugstr_w(strClass), iFlags, objWbemNamedValueSet,
objWbemObjectSet );
if (!(query = build_query_string( strClass )))
{
SysFreeString( wql );
return E_OUTOFMEMORY;
}
hr = ISWbemServices_ExecQuery( iface, query, wql, iFlags, objWbemNamedValueSet, objWbemObjectSet );
SysFreeString( wql );
SysFreeString( query );
return hr;
}
static HRESULT WINAPI services_InstancesOfAsync(
ISWbemServices *iface,
IDispatch *objWbemSink,
BSTR strClass,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_SubclassesOf(
ISWbemServices *iface,
BSTR strSuperclass,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObjectSet **objWbemObjectSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_SubclassesOfAsync(
ISWbemServices *iface,
IDispatch *objWbemSink,
BSTR strSuperclass,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_ExecQuery(
ISWbemServices *iface,
BSTR strQuery,
BSTR strQueryLanguage,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObjectSet **objWbemObjectSet )
{
struct services *services = impl_from_ISWbemServices( iface );
IEnumWbemClassObject *iter;
HRESULT hr;
TRACE( "%p, %s, %s, %x, %p, %p\n", iface, debugstr_w(strQuery), debugstr_w(strQueryLanguage),
iFlags, objWbemNamedValueSet, objWbemObjectSet );
if (objWbemNamedValueSet) FIXME( "ignoring context\n" );
hr = IWbemServices_ExecQuery( services->services, strQueryLanguage, strQuery, iFlags, NULL, &iter );
if (hr != S_OK) return hr;
hr = SWbemObjectSet_create( iter, objWbemObjectSet );
IEnumWbemClassObject_Release( iter );
return hr;
}
static HRESULT WINAPI services_ExecQueryAsync(
ISWbemServices *iface,
IDispatch *objWbemSink,
BSTR strQuery,
BSTR strQueryLanguage,
LONG lFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_AssociatorsOf(
ISWbemServices *iface,
BSTR strObjectPath,
BSTR strAssocClass,
BSTR strResultClass,
BSTR strResultRole,
BSTR strRole,
VARIANT_BOOL bClassesOnly,
VARIANT_BOOL bSchemaOnly,
BSTR strRequiredAssocQualifier,
BSTR strRequiredQualifier,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObjectSet **objWbemObjectSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_AssociatorsOfAsync(
ISWbemServices *iface,
IDispatch *objWbemSink,
BSTR strObjectPath,
BSTR strAssocClass,
BSTR strResultClass,
BSTR strResultRole,
BSTR strRole,
VARIANT_BOOL bClassesOnly,
VARIANT_BOOL bSchemaOnly,
BSTR strRequiredAssocQualifier,
BSTR strRequiredQualifier,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_ReferencesTo(
ISWbemServices *iface,
BSTR strObjectPath,
BSTR strResultClass,
BSTR strRole,
VARIANT_BOOL bClassesOnly,
VARIANT_BOOL bSchemaOnly,
BSTR strRequiredQualifier,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObjectSet **objWbemObjectSet )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_ReferencesToAsync(
ISWbemServices *iface,
IDispatch *objWbemSink,
BSTR strObjectPath,
BSTR strResultClass,
BSTR strRole,
VARIANT_BOOL bClassesOnly,
VARIANT_BOOL bSchemaOnly,
BSTR strRequiredQualifier,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_ExecNotificationQuery(
ISWbemServices *iface,
BSTR strQuery,
BSTR strQueryLanguage,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemEventSource **objWbemEventSource )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_ExecNotificationQueryAsync(
ISWbemServices *iface,
IDispatch *objWbemSink,
BSTR strQuery,
BSTR strQueryLanguage,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_ExecMethod(
ISWbemServices *iface,
BSTR strObjectPath,
BSTR strMethodName,
IDispatch *objWbemInParameters,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
ISWbemObject **objWbemOutParameters )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_ExecMethodAsync(
ISWbemServices *iface,
IDispatch *objWbemSink,
BSTR strObjectPath,
BSTR strMethodName,
IDispatch *objWbemInParameters,
LONG iFlags,
IDispatch *objWbemNamedValueSet,
IDispatch *objWbemAsyncContext )
{
FIXME( "\n" );
return E_NOTIMPL;
}
static HRESULT WINAPI services_get_Security_(
ISWbemServices *iface,
ISWbemSecurity **objWbemSecurity )
{
TRACE( "%p, %p\n", iface, objWbemSecurity );
if (!objWbemSecurity)
return E_INVALIDARG;
return ISWbemSecurity_create( objWbemSecurity );
}
static const ISWbemServicesVtbl services_vtbl =
{
services_QueryInterface,
services_AddRef,
services_Release,
services_GetTypeInfoCount,
services_GetTypeInfo,
services_GetIDsOfNames,
services_Invoke,
services_Get,
services_GetAsync,
services_Delete,
services_DeleteAsync,
services_InstancesOf,
services_InstancesOfAsync,
services_SubclassesOf,
services_SubclassesOfAsync,
services_ExecQuery,
services_ExecQueryAsync,
services_AssociatorsOf,
services_AssociatorsOfAsync,
services_ReferencesTo,
services_ReferencesToAsync,
services_ExecNotificationQuery,
services_ExecNotificationQueryAsync,
services_ExecMethod,
services_ExecMethodAsync,
services_get_Security_
};
static HRESULT SWbemServices_create( IWbemServices *wbem_services, ISWbemServices **obj )
{
struct services *services;
TRACE( "%p, %p\n", obj, wbem_services );
if (!(services = heap_alloc( sizeof(*services) ))) return E_OUTOFMEMORY;
services->ISWbemServices_iface.lpVtbl = &services_vtbl;
services->refs = 1;
services->services = wbem_services;
IWbemServices_AddRef( services->services );
*obj = &services->ISWbemServices_iface;
TRACE( "returning iface %p\n", *obj );
return S_OK;
}
struct locator
{
ISWbemLocator ISWbemLocator_iface;
LONG refs;
IWbemLocator *locator;
};
static inline struct locator *impl_from_ISWbemLocator( ISWbemLocator *iface )
{
return CONTAINING_RECORD( iface, struct locator, ISWbemLocator_iface );
}
static ULONG WINAPI locator_AddRef(
ISWbemLocator *iface )
{
struct locator *locator = impl_from_ISWbemLocator( iface );
return InterlockedIncrement( &locator->refs );
}
static ULONG WINAPI locator_Release(
ISWbemLocator *iface )
{
struct locator *locator = impl_from_ISWbemLocator( iface );
LONG refs = InterlockedDecrement( &locator->refs );
if (!refs)
{
TRACE( "destroying %p\n", locator );
if (locator->locator)
IWbemLocator_Release( locator->locator );
heap_free( locator );
}
return refs;
}
static HRESULT WINAPI locator_QueryInterface(
ISWbemLocator *iface,
REFIID riid,
void **ppvObject )
{
struct locator *locator = impl_from_ISWbemLocator( iface );
TRACE( "%p, %s, %p\n", locator, debugstr_guid( riid ), ppvObject );
if (IsEqualGUID( riid, &IID_ISWbemLocator ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ))
{
*ppvObject = iface;
}
else
{
FIXME( "interface %s not implemented\n", debugstr_guid(riid) );
return E_NOINTERFACE;
}
ISWbemLocator_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI locator_GetTypeInfoCount(
ISWbemLocator *iface,
UINT *count )
{
struct locator *locator = impl_from_ISWbemLocator( iface );
TRACE( "%p, %p\n", locator, count );
*count = 1;
return S_OK;
}
static HRESULT WINAPI locator_GetTypeInfo(
ISWbemLocator *iface,
UINT index,
LCID lcid,
ITypeInfo **info )
{
struct locator *locator = impl_from_ISWbemLocator( iface );
TRACE( "%p, %u, %u, %p\n", locator, index, lcid, info );
return get_typeinfo( ISWbemLocator_tid, info );
}
static HRESULT WINAPI locator_GetIDsOfNames(
ISWbemLocator *iface,
REFIID riid,
LPOLESTR *names,
UINT count,
LCID lcid,
DISPID *dispid )
{
struct locator *locator = impl_from_ISWbemLocator( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %s, %p, %u, %u, %p\n", locator, debugstr_guid(riid), names, count, lcid, dispid );
if (!names || !count || !dispid) return E_INVALIDARG;
hr = get_typeinfo( ISWbemLocator_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, names, count, dispid );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI locator_Invoke(
ISWbemLocator *iface,
DISPID member,
REFIID riid,
LCID lcid,
WORD flags,
DISPPARAMS *params,
VARIANT *result,
EXCEPINFO *excep_info,
UINT *arg_err )
{
struct locator *locator = impl_from_ISWbemLocator( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %d, %s, %d, %d, %p, %p, %p, %p\n", locator, member, debugstr_guid(riid),
lcid, flags, params, result, excep_info, arg_err );
hr = get_typeinfo( ISWbemLocator_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &locator->ISWbemLocator_iface, member, flags,
params, result, excep_info, arg_err );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static BSTR build_resource_string( BSTR server, BSTR namespace )
{
static const WCHAR defaultW[] = {'r','o','o','t','\\','d','e','f','a','u','l','t',0};
ULONG len, len_server = 0, len_namespace = 0;
BSTR ret;
if (server && *server) len_server = lstrlenW( server );
else len_server = 1;
if (namespace && *namespace) len_namespace = lstrlenW( namespace );
else len_namespace = ARRAY_SIZE(defaultW) - 1;
if (!(ret = SysAllocStringLen( NULL, 2 + len_server + 1 + len_namespace ))) return NULL;
ret[0] = ret[1] = '\\';
if (server && *server) lstrcpyW( ret + 2, server );
else ret[2] = '.';
len = len_server + 2;
ret[len++] = '\\';
if (namespace && *namespace) lstrcpyW( ret + len, namespace );
else lstrcpyW( ret + len, defaultW );
return ret;
}
static HRESULT WINAPI locator_ConnectServer(
ISWbemLocator *iface,
BSTR strServer,
BSTR strNamespace,
BSTR strUser,
BSTR strPassword,
BSTR strLocale,
BSTR strAuthority,
LONG iSecurityFlags,
IDispatch *objWbemNamedValueSet,
ISWbemServices **objWbemServices )
{
struct locator *locator = impl_from_ISWbemLocator( iface );
IWbemServices *services;
BSTR resource;
HRESULT hr;
TRACE( "%p, %s, %s, %s, %p, %s, %s, 0x%08x, %p, %p\n", iface, debugstr_w(strServer),
debugstr_w(strNamespace), debugstr_w(strUser), strPassword, debugstr_w(strLocale),
debugstr_w(strAuthority), iSecurityFlags, objWbemNamedValueSet, objWbemServices );
if (objWbemNamedValueSet) FIXME( "context not supported\n" );
if (!locator->locator)
{
hr = CoCreateInstance( &CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER, &IID_IWbemLocator,
(void **)&locator->locator );
if (hr != S_OK) return hr;
}
if (!(resource = build_resource_string( strServer, strNamespace ))) return E_OUTOFMEMORY;
hr = IWbemLocator_ConnectServer( locator->locator, resource, strUser, strPassword, strLocale,
iSecurityFlags, strAuthority, NULL, &services );
SysFreeString( resource );
if (hr != S_OK) return hr;
hr = SWbemServices_create( services, objWbemServices );
IWbemServices_Release( services );
return hr;
}
static HRESULT WINAPI locator_get_Security_(
ISWbemLocator *iface,
ISWbemSecurity **objWbemSecurity )
{
TRACE( "%p, %p\n", iface, objWbemSecurity );
if (!objWbemSecurity)
return E_INVALIDARG;
return ISWbemSecurity_create( objWbemSecurity );
}
static const ISWbemLocatorVtbl locator_vtbl =
{
locator_QueryInterface,
locator_AddRef,
locator_Release,
locator_GetTypeInfoCount,
locator_GetTypeInfo,
locator_GetIDsOfNames,
locator_Invoke,
locator_ConnectServer,
locator_get_Security_
};
HRESULT SWbemLocator_create( void **obj )
{
struct locator *locator;
TRACE( "%p\n", obj );
if (!(locator = heap_alloc( sizeof(*locator) ))) return E_OUTOFMEMORY;
locator->ISWbemLocator_iface.lpVtbl = &locator_vtbl;
locator->refs = 1;
locator->locator = NULL;
*obj = &locator->ISWbemLocator_iface;
TRACE( "returning iface %p\n", *obj );
return S_OK;
}
struct security
{
ISWbemSecurity ISWbemSecurity_iface;
LONG refs;
WbemImpersonationLevelEnum implevel;
WbemAuthenticationLevelEnum authlevel;
};
static inline struct security *impl_from_ISWbemSecurity( ISWbemSecurity *iface )
{
return CONTAINING_RECORD( iface, struct security, ISWbemSecurity_iface );
}
static ULONG WINAPI security_AddRef(
ISWbemSecurity *iface )
{
struct security *security = impl_from_ISWbemSecurity( iface );
return InterlockedIncrement( &security->refs );
}
static ULONG WINAPI security_Release(
ISWbemSecurity *iface )
{
struct security *security = impl_from_ISWbemSecurity( iface );
LONG refs = InterlockedDecrement( &security->refs );
if (!refs)
{
TRACE( "destroying %p\n", security );
heap_free( security );
}
return refs;
}
static HRESULT WINAPI security_QueryInterface(
ISWbemSecurity *iface,
REFIID riid,
void **ppvObject )
{
struct security *security = impl_from_ISWbemSecurity( iface );
TRACE( "%p, %s, %p\n", security, debugstr_guid( riid ), ppvObject );
if (IsEqualGUID( riid, &IID_ISWbemSecurity ) ||
IsEqualGUID( riid, &IID_IDispatch ) ||
IsEqualGUID( riid, &IID_IUnknown ))
{
*ppvObject = iface;
}
else
{
FIXME( "interface %s not implemented\n", debugstr_guid(riid) );
return E_NOINTERFACE;
}
ISWbemSecurity_AddRef( iface );
return S_OK;
}
static HRESULT WINAPI security_GetTypeInfoCount(
ISWbemSecurity *iface,
UINT *count )
{
struct security *security = impl_from_ISWbemSecurity( iface );
TRACE( "%p, %p\n", security, count );
*count = 1;
return S_OK;
}
static HRESULT WINAPI security_GetTypeInfo(
ISWbemSecurity *iface,
UINT index,
LCID lcid,
ITypeInfo **info )
{
struct security *security = impl_from_ISWbemSecurity( iface );
TRACE( "%p, %u, %u, %p\n", security, index, lcid, info );
return get_typeinfo( ISWbemSecurity_tid, info );
}
static HRESULT WINAPI security_GetIDsOfNames(
ISWbemSecurity *iface,
REFIID riid,
LPOLESTR *names,
UINT count,
LCID lcid,
DISPID *dispid )
{
struct security *security = impl_from_ISWbemSecurity( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %s, %p, %u, %u, %p\n", security, debugstr_guid(riid), names, count, lcid, dispid );
if (!names || !count || !dispid) return E_INVALIDARG;
hr = get_typeinfo( ISWbemSecurity_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_GetIDsOfNames( typeinfo, names, count, dispid );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI security_Invoke(
ISWbemSecurity *iface,
DISPID member,
REFIID riid,
LCID lcid,
WORD flags,
DISPPARAMS *params,
VARIANT *result,
EXCEPINFO *excep_info,
UINT *arg_err )
{
struct security *security = impl_from_ISWbemSecurity( iface );
ITypeInfo *typeinfo;
HRESULT hr;
TRACE( "%p, %d, %s, %d, %d, %p, %p, %p, %p\n", security, member, debugstr_guid(riid),
lcid, flags, params, result, excep_info, arg_err );
hr = get_typeinfo( ISWbemSecurity_tid, &typeinfo );
if (SUCCEEDED(hr))
{
hr = ITypeInfo_Invoke( typeinfo, &security->ISWbemSecurity_iface, member, flags,
params, result, excep_info, arg_err );
ITypeInfo_Release( typeinfo );
}
return hr;
}
static HRESULT WINAPI security_get_ImpersonationLevel(
ISWbemSecurity *iface,
WbemImpersonationLevelEnum *impersonation_level )
{
struct security *security = impl_from_ISWbemSecurity( iface );
FIXME( "%p, %p: stub\n", security, impersonation_level );
if (!impersonation_level)
return E_INVALIDARG;
*impersonation_level = security->implevel;
return S_OK;
}
static HRESULT WINAPI security_put_ImpersonationLevel(
ISWbemSecurity *iface,
WbemImpersonationLevelEnum impersonation_level )
{
struct security *security = impl_from_ISWbemSecurity( iface );
FIXME( "%p, %d: stub\n", security, impersonation_level );
security->implevel = impersonation_level;
return S_OK;
}
static HRESULT WINAPI security_get_AuthenticationLevel(
ISWbemSecurity *iface,
WbemAuthenticationLevelEnum *authentication_level )
{
struct security *security = impl_from_ISWbemSecurity( iface );
FIXME( "%p, %p: stub\n", security, authentication_level );
if (!authentication_level)
return E_INVALIDARG;
*authentication_level = security->authlevel;
return S_OK;
}
static HRESULT WINAPI security_put_AuthenticationLevel(
ISWbemSecurity *iface,
WbemAuthenticationLevelEnum authentication_level )
{
struct security *security = impl_from_ISWbemSecurity( iface );
FIXME( "%p, %d: stub\n", security, authentication_level );
security->authlevel = authentication_level;
return S_OK;
}
static HRESULT WINAPI security_get_Privileges(
ISWbemSecurity *iface,
ISWbemPrivilegeSet **privilege_set )
{
struct security *security = impl_from_ISWbemSecurity( iface );
FIXME( "%p, %p: stub\n", security, privilege_set );
if (!privilege_set)
return E_INVALIDARG;
return E_NOTIMPL;
}
static const ISWbemSecurityVtbl security_vtbl =
{
security_QueryInterface,
security_AddRef,
security_Release,
security_GetTypeInfoCount,
security_GetTypeInfo,
security_GetIDsOfNames,
security_Invoke,
security_get_ImpersonationLevel,
security_put_ImpersonationLevel,
security_get_AuthenticationLevel,
security_put_AuthenticationLevel,
security_get_Privileges
};
static HRESULT ISWbemSecurity_create( ISWbemSecurity **obj )
{
struct security *security;
TRACE( "%p\n", obj );
if (!(security = heap_alloc( sizeof(*security) ))) return E_OUTOFMEMORY;
security->ISWbemSecurity_iface.lpVtbl = &security_vtbl;
security->refs = 1;
security->implevel = wbemImpersonationLevelImpersonate;
security->authlevel = wbemAuthenticationLevelPktPrivacy;
*obj = &security->ISWbemSecurity_iface;
TRACE( "returning iface %p\n", *obj );
return S_OK;
}