1870 lines
65 KiB
C
1870 lines
65 KiB
C
/*
|
|
* Implementation of OLE Automation for Microsoft Installer (msi.dll)
|
|
*
|
|
* Copyright 2007 Misha Koshelev
|
|
*
|
|
* 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 "winerror.h"
|
|
#include "winuser.h"
|
|
#include "winreg.h"
|
|
#include "msidefs.h"
|
|
#include "msipriv.h"
|
|
#include "activscp.h"
|
|
#include "oleauto.h"
|
|
#include "wine/debug.h"
|
|
#include "wine/unicode.h"
|
|
|
|
#include "msiserver.h"
|
|
#include "msiserver_dispids.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(msi);
|
|
|
|
/*
|
|
* AutomationObject - "base" class for all automation objects. For each interface, we implement Invoke function
|
|
* called from AutomationObject::Invoke, and pass this function to create_automation_object.
|
|
*/
|
|
|
|
typedef interface AutomationObject AutomationObject;
|
|
|
|
interface AutomationObject {
|
|
/*
|
|
* VTables - We provide IDispatch, IProvideClassInfo, IProvideClassInfo2, IProvideMultipleClassInfo
|
|
*/
|
|
const IDispatchVtbl *lpVtbl;
|
|
const IProvideMultipleClassInfoVtbl *lpvtblIProvideMultipleClassInfo;
|
|
|
|
/* Object reference count */
|
|
LONG ref;
|
|
|
|
/* Clsid for this class and it's appropriate ITypeInfo object */
|
|
LPCLSID clsid;
|
|
ITypeInfo *iTypeInfo;
|
|
|
|
/* The MSI handle of the current object */
|
|
MSIHANDLE msiHandle;
|
|
|
|
/* A function that is called from AutomationObject::Invoke, specific to this type of object. */
|
|
HRESULT (STDMETHODCALLTYPE *funcInvoke)(
|
|
AutomationObject* This,
|
|
DISPID dispIdMember,
|
|
REFIID riid,
|
|
LCID lcid,
|
|
WORD wFlags,
|
|
DISPPARAMS* pDispParams,
|
|
VARIANT* pVarResult,
|
|
EXCEPINFO* pExcepInfo,
|
|
UINT* puArgErr);
|
|
|
|
/* A function that is called from AutomationObject::Release when the object is being freed to free any private
|
|
* data structures (or NULL) */
|
|
void (STDMETHODCALLTYPE *funcFree)(AutomationObject* This);
|
|
};
|
|
|
|
/*
|
|
* ListEnumerator - IEnumVARIANT implementation for MSI automation lists.
|
|
*/
|
|
|
|
typedef interface ListEnumerator ListEnumerator;
|
|
|
|
interface ListEnumerator {
|
|
/* VTables */
|
|
const IEnumVARIANTVtbl *lpVtbl;
|
|
|
|
/* Object reference count */
|
|
LONG ref;
|
|
|
|
/* Current position and pointer to AutomationObject that stores actual data */
|
|
ULONG ulPos;
|
|
AutomationObject *pObj;
|
|
};
|
|
|
|
/*
|
|
* Structures for additional data required by specific automation objects
|
|
*/
|
|
|
|
typedef struct {
|
|
ULONG ulCount;
|
|
VARIANT *pVars;
|
|
} ListData;
|
|
|
|
typedef struct {
|
|
/* The parent Installer object */
|
|
IDispatch *pInstaller;
|
|
} SessionData;
|
|
|
|
/* VTables */
|
|
static const struct IDispatchVtbl AutomationObject_Vtbl;
|
|
static const struct IProvideMultipleClassInfoVtbl AutomationObject_IProvideMultipleClassInfo_Vtbl;
|
|
static const struct IEnumVARIANTVtbl ListEnumerator_Vtbl;
|
|
|
|
/* Load type info so we don't have to process GetIDsOfNames */
|
|
HRESULT load_type_info(IDispatch *iface, ITypeInfo **pptinfo, REFIID clsid, LCID lcid)
|
|
{
|
|
HRESULT hr;
|
|
LPTYPELIB pLib = NULL;
|
|
LPTYPEINFO pInfo = NULL;
|
|
static const WCHAR szMsiServer[] = {'m','s','i','s','e','r','v','e','r','.','t','l','b'};
|
|
|
|
TRACE("(%p)->(%s,%d)\n", iface, debugstr_guid(clsid), lcid);
|
|
|
|
/* Load registered type library */
|
|
hr = LoadRegTypeLib(&LIBID_WindowsInstaller, 1, 0, lcid, &pLib);
|
|
if (FAILED(hr)) {
|
|
hr = LoadTypeLib(szMsiServer, &pLib);
|
|
if (FAILED(hr)) {
|
|
ERR("Could not load msiserver.tlb\n");
|
|
return hr;
|
|
}
|
|
}
|
|
|
|
/* Get type information for object */
|
|
hr = ITypeLib_GetTypeInfoOfGuid(pLib, clsid, &pInfo);
|
|
ITypeLib_Release(pLib);
|
|
if (FAILED(hr)) {
|
|
ERR("Could not load ITypeInfo for %s\n", debugstr_guid(clsid));
|
|
return hr;
|
|
}
|
|
*pptinfo = pInfo;
|
|
return S_OK;
|
|
}
|
|
|
|
/* Create the automation object, placing the result in the pointer ppObj. The automation object is created
|
|
* with the appropriate clsid and invocation function. */
|
|
HRESULT create_automation_object(MSIHANDLE msiHandle, IUnknown *pUnkOuter, LPVOID *ppObj, REFIID clsid,
|
|
HRESULT (STDMETHODCALLTYPE *funcInvoke)(AutomationObject*,DISPID,REFIID,LCID,WORD,DISPPARAMS*,
|
|
VARIANT*,EXCEPINFO*,UINT*),
|
|
void (STDMETHODCALLTYPE *funcFree)(AutomationObject*),
|
|
SIZE_T sizetPrivateData)
|
|
{
|
|
AutomationObject *object;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%ld,%p,%p,%s,%p,%p,%ld)\n", (unsigned long)msiHandle, pUnkOuter, ppObj, debugstr_guid(clsid), funcInvoke, funcFree, sizetPrivateData);
|
|
|
|
if( pUnkOuter )
|
|
return CLASS_E_NOAGGREGATION;
|
|
|
|
object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(AutomationObject)+sizetPrivateData);
|
|
|
|
/* Set all the VTable references */
|
|
object->lpVtbl = &AutomationObject_Vtbl;
|
|
object->lpvtblIProvideMultipleClassInfo = &AutomationObject_IProvideMultipleClassInfo_Vtbl;
|
|
object->ref = 1;
|
|
|
|
/* Store data that was passed */
|
|
object->msiHandle = msiHandle;
|
|
object->clsid = (LPCLSID)clsid;
|
|
object->funcInvoke = funcInvoke;
|
|
object->funcFree = funcFree;
|
|
|
|
/* Load our TypeInfo so we don't have to process GetIDsOfNames */
|
|
object->iTypeInfo = NULL;
|
|
hr = load_type_info((IDispatch *)object, &object->iTypeInfo, clsid, 0x0);
|
|
if (FAILED(hr)) {
|
|
HeapFree(GetProcessHeap(), 0, object);
|
|
return hr;
|
|
}
|
|
|
|
*ppObj = object;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
/* Create a list enumerator, placing the result in the pointer ppObj. */
|
|
HRESULT create_list_enumerator(IUnknown *pUnkOuter, LPVOID *ppObj, AutomationObject *pObj, ULONG ulPos)
|
|
{
|
|
ListEnumerator *object;
|
|
|
|
TRACE("(%p,%p,%p,%uld)\n", pUnkOuter, ppObj, pObj, ulPos);
|
|
|
|
if( pUnkOuter )
|
|
return CLASS_E_NOAGGREGATION;
|
|
|
|
object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ListEnumerator));
|
|
|
|
/* Set all the VTable references */
|
|
object->lpVtbl = &ListEnumerator_Vtbl;
|
|
object->ref = 1;
|
|
|
|
/* Store data that was passed */
|
|
object->ulPos = ulPos;
|
|
object->pObj = pObj;
|
|
if (pObj) IDispatch_AddRef((IDispatch *)pObj);
|
|
|
|
*ppObj = object;
|
|
return S_OK;
|
|
}
|
|
|
|
/* Macros to get pointer to AutomationObject from the other VTables. */
|
|
static inline AutomationObject *obj_from_IProvideMultipleClassInfo( IProvideMultipleClassInfo *iface )
|
|
{
|
|
return (AutomationObject *)((char*)iface - FIELD_OFFSET(AutomationObject, lpvtblIProvideMultipleClassInfo));
|
|
}
|
|
|
|
/* Macro to get pointer to private object data */
|
|
static inline void *private_data( AutomationObject *This )
|
|
{
|
|
return This + 1;
|
|
}
|
|
|
|
/*
|
|
* AutomationObject methods
|
|
*/
|
|
|
|
/*** IUnknown methods ***/
|
|
static HRESULT WINAPI AutomationObject_QueryInterface(IDispatch* iface, REFIID riid, void** ppvObject)
|
|
{
|
|
AutomationObject *This = (AutomationObject *)iface;
|
|
|
|
TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
|
|
|
|
if (ppvObject == NULL)
|
|
return E_INVALIDARG;
|
|
|
|
*ppvObject = 0;
|
|
|
|
if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IDispatch) || IsEqualGUID(riid, This->clsid))
|
|
*ppvObject = This;
|
|
else if (IsEqualGUID(riid, &IID_IProvideClassInfo) ||
|
|
IsEqualGUID(riid, &IID_IProvideClassInfo2) ||
|
|
IsEqualGUID(riid, &IID_IProvideMultipleClassInfo))
|
|
*ppvObject = &This->lpvtblIProvideMultipleClassInfo;
|
|
else
|
|
{
|
|
TRACE("() : asking for unsupported interface %s\n",debugstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
/*
|
|
* Query Interface always increases the reference count by one when it is
|
|
* successful
|
|
*/
|
|
IClassFactory_AddRef(iface);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI AutomationObject_AddRef(IDispatch* iface)
|
|
{
|
|
AutomationObject *This = (AutomationObject *)iface;
|
|
|
|
TRACE("(%p/%p)\n", iface, This);
|
|
|
|
return InterlockedIncrement(&This->ref);
|
|
}
|
|
|
|
static ULONG WINAPI AutomationObject_Release(IDispatch* iface)
|
|
{
|
|
AutomationObject *This = (AutomationObject *)iface;
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p/%p)\n", iface, This);
|
|
|
|
if (!ref)
|
|
{
|
|
if (This->funcFree) This->funcFree(This);
|
|
MsiCloseHandle(This->msiHandle);
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
/*** IDispatch methods ***/
|
|
static HRESULT WINAPI AutomationObject_GetTypeInfoCount(
|
|
IDispatch* iface,
|
|
UINT* pctinfo)
|
|
{
|
|
AutomationObject *This = (AutomationObject *)iface;
|
|
|
|
TRACE("(%p/%p)->(%p)\n", iface, This, pctinfo);
|
|
*pctinfo = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI AutomationObject_GetTypeInfo(
|
|
IDispatch* iface,
|
|
UINT iTInfo,
|
|
LCID lcid,
|
|
ITypeInfo** ppTInfo)
|
|
{
|
|
AutomationObject *This = (AutomationObject *)iface;
|
|
TRACE("(%p/%p)->(%d,%d,%p)\n", iface, This, iTInfo, lcid, ppTInfo);
|
|
|
|
ITypeInfo_AddRef(This->iTypeInfo);
|
|
*ppTInfo = This->iTypeInfo;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI AutomationObject_GetIDsOfNames(
|
|
IDispatch* iface,
|
|
REFIID riid,
|
|
LPOLESTR* rgszNames,
|
|
UINT cNames,
|
|
LCID lcid,
|
|
DISPID* rgDispId)
|
|
{
|
|
AutomationObject *This = (AutomationObject *)iface;
|
|
HRESULT hr;
|
|
TRACE("(%p/%p)->(%p,%p,%d,%d,%p)\n", iface, This, riid, rgszNames, cNames, lcid, rgDispId);
|
|
|
|
if (!IsEqualGUID(riid, &IID_NULL)) return E_INVALIDARG;
|
|
hr = ITypeInfo_GetIDsOfNames(This->iTypeInfo, rgszNames, cNames, rgDispId);
|
|
if (hr == DISP_E_UNKNOWNNAME)
|
|
{
|
|
int idx;
|
|
for (idx=0; idx<cNames; idx++)
|
|
{
|
|
if (rgDispId[idx] == DISPID_UNKNOWN)
|
|
FIXME("Unknown member %s, clsid %s\n", debugstr_w(rgszNames[idx]), debugstr_guid(This->clsid));
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
/* Maximum number of allowed function parameters+1 */
|
|
#define MAX_FUNC_PARAMS 20
|
|
|
|
/* Some error checking is done here to simplify individual object function invocation */
|
|
static HRESULT WINAPI AutomationObject_Invoke(
|
|
IDispatch* iface,
|
|
DISPID dispIdMember,
|
|
REFIID riid,
|
|
LCID lcid,
|
|
WORD wFlags,
|
|
DISPPARAMS* pDispParams,
|
|
VARIANT* pVarResult,
|
|
EXCEPINFO* pExcepInfo,
|
|
UINT* puArgErr)
|
|
{
|
|
AutomationObject *This = (AutomationObject *)iface;
|
|
HRESULT hr;
|
|
unsigned int uArgErr;
|
|
VARIANT varResultDummy;
|
|
BSTR bstrName = NULL;
|
|
|
|
TRACE("(%p/%p)->(%d,%p,%d,%d,%p,%p,%p,%p)\n", iface, This, dispIdMember, riid, lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
|
|
if (!IsEqualIID(riid, &IID_NULL))
|
|
{
|
|
ERR("riid was %s instead of IID_NULL\n", debugstr_guid(riid));
|
|
return DISP_E_UNKNOWNNAME;
|
|
}
|
|
|
|
if (wFlags & DISPATCH_PROPERTYGET && !pVarResult)
|
|
{
|
|
ERR("NULL pVarResult not allowed when DISPATCH_PROPERTYGET specified\n");
|
|
return DISP_E_PARAMNOTOPTIONAL;
|
|
}
|
|
|
|
/* This simplifies our individual object invocation functions */
|
|
if (puArgErr == NULL) puArgErr = &uArgErr;
|
|
if (pVarResult == NULL) pVarResult = &varResultDummy;
|
|
|
|
/* Assume return type is void unless determined otherwise */
|
|
VariantInit(pVarResult);
|
|
|
|
/* If we are tracing, we want to see the name of the member we are invoking */
|
|
if (TRACE_ON(msi))
|
|
{
|
|
ITypeInfo_GetDocumentation(This->iTypeInfo, dispIdMember, &bstrName, NULL, NULL, NULL);
|
|
TRACE("Method %d, %s\n", dispIdMember, debugstr_w(bstrName));
|
|
}
|
|
|
|
hr = This->funcInvoke(This,dispIdMember,riid,lcid,wFlags,pDispParams,pVarResult,pExcepInfo,puArgErr);
|
|
|
|
if (hr == DISP_E_MEMBERNOTFOUND) {
|
|
if (bstrName == NULL) ITypeInfo_GetDocumentation(This->iTypeInfo, dispIdMember, &bstrName, NULL, NULL, NULL);
|
|
FIXME("Method %d, %s wflags %d not implemented, clsid %s\n", dispIdMember, debugstr_w(bstrName), wFlags, debugstr_guid(This->clsid));
|
|
}
|
|
else if (pExcepInfo &&
|
|
(hr == DISP_E_PARAMNOTFOUND ||
|
|
hr == DISP_E_EXCEPTION)) {
|
|
static const WCHAR szComma[] = { ',',0 };
|
|
static WCHAR szExceptionSource[] = {'M','s','i',' ','A','P','I',' ','E','r','r','o','r',0};
|
|
WCHAR szExceptionDescription[MAX_PATH];
|
|
BSTR bstrParamNames[MAX_FUNC_PARAMS];
|
|
unsigned namesNo, i;
|
|
BOOL bFirst = TRUE;
|
|
|
|
if (FAILED(ITypeInfo_GetNames(This->iTypeInfo, dispIdMember, bstrParamNames,
|
|
MAX_FUNC_PARAMS, &namesNo)))
|
|
{
|
|
TRACE("Failed to retrieve names for dispIdMember %d\n", dispIdMember);
|
|
}
|
|
else
|
|
{
|
|
memset(szExceptionDescription, 0, sizeof(szExceptionDescription));
|
|
for (i=0; i<namesNo; i++)
|
|
{
|
|
if (bFirst) bFirst = FALSE;
|
|
else {
|
|
lstrcpyW(&szExceptionDescription[lstrlenW(szExceptionDescription)], szComma);
|
|
}
|
|
lstrcpyW(&szExceptionDescription[lstrlenW(szExceptionDescription)], bstrParamNames[i]);
|
|
SysFreeString(bstrParamNames[i]);
|
|
}
|
|
|
|
memset(pExcepInfo, 0, sizeof(EXCEPINFO));
|
|
pExcepInfo->wCode = 1000;
|
|
pExcepInfo->bstrSource = szExceptionSource;
|
|
pExcepInfo->bstrDescription = szExceptionDescription;
|
|
hr = DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
|
|
/* Make sure we free the return variant if it is our dummy variant */
|
|
if (pVarResult == &varResultDummy) VariantClear(pVarResult);
|
|
|
|
/* Free function name if we retrieved it */
|
|
if (bstrName) SysFreeString(bstrName);
|
|
|
|
TRACE("Returning 0x%08x, %s\n", hr, SUCCEEDED(hr) ? "ok" : "not ok");
|
|
|
|
return hr;
|
|
}
|
|
|
|
static const struct IDispatchVtbl AutomationObject_Vtbl =
|
|
{
|
|
AutomationObject_QueryInterface,
|
|
AutomationObject_AddRef,
|
|
AutomationObject_Release,
|
|
AutomationObject_GetTypeInfoCount,
|
|
AutomationObject_GetTypeInfo,
|
|
AutomationObject_GetIDsOfNames,
|
|
AutomationObject_Invoke
|
|
};
|
|
|
|
/*
|
|
* IProvideMultipleClassInfo methods
|
|
*/
|
|
|
|
static HRESULT WINAPI AutomationObject_IProvideMultipleClassInfo_QueryInterface(
|
|
IProvideMultipleClassInfo* iface,
|
|
REFIID riid,
|
|
VOID** ppvoid)
|
|
{
|
|
AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
|
|
return AutomationObject_QueryInterface((IDispatch *)This, riid, ppvoid);
|
|
}
|
|
|
|
static ULONG WINAPI AutomationObject_IProvideMultipleClassInfo_AddRef(IProvideMultipleClassInfo* iface)
|
|
{
|
|
AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
|
|
return AutomationObject_AddRef((IDispatch *)This);
|
|
}
|
|
|
|
static ULONG WINAPI AutomationObject_IProvideMultipleClassInfo_Release(IProvideMultipleClassInfo* iface)
|
|
{
|
|
AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
|
|
return AutomationObject_Release((IDispatch *)This);
|
|
}
|
|
|
|
static HRESULT WINAPI AutomationObject_IProvideMultipleClassInfo_GetClassInfo(IProvideMultipleClassInfo* iface, ITypeInfo** ppTI)
|
|
{
|
|
AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
|
|
TRACE("(%p/%p)->(%p)\n", iface, This, ppTI);
|
|
return load_type_info((IDispatch *)This, ppTI, This->clsid, 0);
|
|
}
|
|
|
|
static HRESULT WINAPI AutomationObject_IProvideMultipleClassInfo_GetGUID(IProvideMultipleClassInfo* iface, DWORD dwGuidKind, GUID* pGUID)
|
|
{
|
|
AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
|
|
TRACE("(%p/%p)->(%d,%s)\n", iface, This, dwGuidKind, debugstr_guid(pGUID));
|
|
|
|
if (dwGuidKind != GUIDKIND_DEFAULT_SOURCE_DISP_IID)
|
|
return E_INVALIDARG;
|
|
else {
|
|
*pGUID = *This->clsid;
|
|
return S_OK;
|
|
}
|
|
}
|
|
|
|
static HRESULT WINAPI AutomationObject_GetMultiTypeInfoCount(IProvideMultipleClassInfo* iface, ULONG* pcti)
|
|
{
|
|
AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
|
|
|
|
TRACE("(%p/%p)->(%p)\n", iface, This, pcti);
|
|
*pcti = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI AutomationObject_GetInfoOfIndex(IProvideMultipleClassInfo* iface,
|
|
ULONG iti,
|
|
DWORD dwFlags,
|
|
ITypeInfo** pptiCoClass,
|
|
DWORD* pdwTIFlags,
|
|
ULONG* pcdispidReserved,
|
|
IID* piidPrimary,
|
|
IID* piidSource)
|
|
{
|
|
AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
|
|
|
|
TRACE("(%p/%p)->(%d,%d,%p,%p,%p,%p,%p)\n", iface, This, iti, dwFlags, pptiCoClass, pdwTIFlags, pcdispidReserved, piidPrimary, piidSource);
|
|
|
|
if (iti != 0)
|
|
return E_INVALIDARG;
|
|
|
|
if (dwFlags & MULTICLASSINFO_GETTYPEINFO)
|
|
load_type_info((IDispatch *)This, pptiCoClass, This->clsid, 0);
|
|
|
|
if (dwFlags & MULTICLASSINFO_GETNUMRESERVEDDISPIDS)
|
|
{
|
|
*pdwTIFlags = 0;
|
|
*pcdispidReserved = 0;
|
|
}
|
|
|
|
if (dwFlags & MULTICLASSINFO_GETIIDPRIMARY){
|
|
*piidPrimary = *This->clsid;
|
|
}
|
|
|
|
if (dwFlags & MULTICLASSINFO_GETIIDSOURCE){
|
|
*piidSource = *This->clsid;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static const IProvideMultipleClassInfoVtbl AutomationObject_IProvideMultipleClassInfo_Vtbl =
|
|
{
|
|
AutomationObject_IProvideMultipleClassInfo_QueryInterface,
|
|
AutomationObject_IProvideMultipleClassInfo_AddRef,
|
|
AutomationObject_IProvideMultipleClassInfo_Release,
|
|
AutomationObject_IProvideMultipleClassInfo_GetClassInfo,
|
|
AutomationObject_IProvideMultipleClassInfo_GetGUID,
|
|
AutomationObject_GetMultiTypeInfoCount,
|
|
AutomationObject_GetInfoOfIndex
|
|
};
|
|
|
|
/*
|
|
* ListEnumerator methods
|
|
*/
|
|
|
|
/*** IUnknown methods ***/
|
|
static HRESULT WINAPI ListEnumerator_QueryInterface(IEnumVARIANT* iface, REFIID riid, void** ppvObject)
|
|
{
|
|
ListEnumerator *This = (ListEnumerator *)iface;
|
|
|
|
TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
|
|
|
|
if (ppvObject == NULL)
|
|
return E_INVALIDARG;
|
|
|
|
*ppvObject = 0;
|
|
|
|
if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IEnumVARIANT))
|
|
*ppvObject = This;
|
|
else
|
|
{
|
|
TRACE("() : asking for unsupported interface %s\n",debugstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
IClassFactory_AddRef(iface);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI ListEnumerator_AddRef(IEnumVARIANT* iface)
|
|
{
|
|
ListEnumerator *This = (ListEnumerator *)iface;
|
|
|
|
TRACE("(%p/%p)\n", iface, This);
|
|
|
|
return InterlockedIncrement(&This->ref);
|
|
}
|
|
|
|
static ULONG WINAPI ListEnumerator_Release(IEnumVARIANT* iface)
|
|
{
|
|
ListEnumerator *This = (ListEnumerator *)iface;
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p/%p)\n", iface, This);
|
|
|
|
if (!ref)
|
|
{
|
|
if (This->pObj) IDispatch_Release((IDispatch *)This->pObj);
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
/* IEnumVARIANT methods */
|
|
|
|
static HRESULT WINAPI ListEnumerator_Next(IEnumVARIANT* iface, ULONG celt, VARIANT *rgVar, ULONG *pCeltFetched)
|
|
{
|
|
ListEnumerator *This = (ListEnumerator *)iface;
|
|
ListData *data = (ListData *)private_data(This->pObj);
|
|
ULONG idx, local;
|
|
|
|
TRACE("(%p,%uld,%p,%p)\n", iface, celt, rgVar, pCeltFetched);
|
|
|
|
if (pCeltFetched != NULL)
|
|
*pCeltFetched = 0;
|
|
|
|
if (rgVar == NULL)
|
|
return S_FALSE;
|
|
|
|
for (local = 0; local < celt; local++)
|
|
VariantInit(&rgVar[local]);
|
|
|
|
for (idx = This->ulPos, local = 0; idx < data->ulCount && local < celt; idx++, local++)
|
|
VariantCopy(&rgVar[local], &data->pVars[idx]);
|
|
|
|
if (pCeltFetched != NULL)
|
|
*pCeltFetched = local;
|
|
This->ulPos = idx;
|
|
|
|
return (local < celt) ? S_FALSE : S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI ListEnumerator_Skip(IEnumVARIANT* iface, ULONG celt)
|
|
{
|
|
ListEnumerator *This = (ListEnumerator *)iface;
|
|
ListData *data = (ListData *)private_data(This->pObj);
|
|
|
|
TRACE("(%p,%uld)\n", iface, celt);
|
|
|
|
This->ulPos += celt;
|
|
if (This->ulPos >= data->ulCount)
|
|
{
|
|
This->ulPos = data->ulCount;
|
|
return S_FALSE;
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI ListEnumerator_Reset(IEnumVARIANT* iface)
|
|
{
|
|
ListEnumerator *This = (ListEnumerator *)iface;
|
|
|
|
TRACE("(%p)\n", iface);
|
|
|
|
This->ulPos = 0;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI ListEnumerator_Clone(IEnumVARIANT* iface, IEnumVARIANT **ppEnum)
|
|
{
|
|
ListEnumerator *This = (ListEnumerator *)iface;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p,%p)\n", iface, ppEnum);
|
|
|
|
if (ppEnum == NULL)
|
|
return S_FALSE;
|
|
|
|
*ppEnum = NULL;
|
|
hr = create_list_enumerator(NULL, (LPVOID *)ppEnum, This->pObj, 0);
|
|
if (FAILED(hr))
|
|
{
|
|
if (*ppEnum)
|
|
IUnknown_Release(*ppEnum);
|
|
return hr;
|
|
}
|
|
|
|
IUnknown_AddRef(*ppEnum);
|
|
return S_OK;
|
|
}
|
|
|
|
static const struct IEnumVARIANTVtbl ListEnumerator_Vtbl =
|
|
{
|
|
ListEnumerator_QueryInterface,
|
|
ListEnumerator_AddRef,
|
|
ListEnumerator_Release,
|
|
ListEnumerator_Next,
|
|
ListEnumerator_Skip,
|
|
ListEnumerator_Reset,
|
|
ListEnumerator_Clone
|
|
};
|
|
|
|
/*
|
|
* Individual Object Invocation Functions
|
|
*/
|
|
|
|
/* Helper function that copies a passed parameter instead of using VariantChangeType like the actual DispGetParam.
|
|
This function is only for VARIANT type parameters that have several types that cannot be properly discriminated
|
|
using DispGetParam/VariantChangeType. */
|
|
HRESULT WINAPI DispGetParam_CopyOnly(
|
|
DISPPARAMS *pdispparams, /* [in] Parameter list */
|
|
UINT *position, /* [in] Position of parameter to copy in pdispparams; on return will contain calculated position */
|
|
VARIANT *pvarResult) /* [out] Destination for resulting variant */
|
|
{
|
|
/* position is counted backwards */
|
|
UINT pos;
|
|
|
|
TRACE("position=%d, cArgs=%d, cNamedArgs=%d\n",
|
|
*position, pdispparams->cArgs, pdispparams->cNamedArgs);
|
|
if (*position < pdispparams->cArgs) {
|
|
/* positional arg? */
|
|
pos = pdispparams->cArgs - *position - 1;
|
|
} else {
|
|
/* FIXME: is this how to handle named args? */
|
|
for (pos=0; pos<pdispparams->cNamedArgs; pos++)
|
|
if (pdispparams->rgdispidNamedArgs[pos] == *position) break;
|
|
|
|
if (pos==pdispparams->cNamedArgs)
|
|
return DISP_E_PARAMNOTFOUND;
|
|
}
|
|
*position = pos;
|
|
return VariantCopyInd(pvarResult,
|
|
&pdispparams->rgvarg[pos]);
|
|
}
|
|
|
|
static HRESULT WINAPI SummaryInfoImpl_Invoke(
|
|
AutomationObject* This,
|
|
DISPID dispIdMember,
|
|
REFIID riid,
|
|
LCID lcid,
|
|
WORD wFlags,
|
|
DISPPARAMS* pDispParams,
|
|
VARIANT* pVarResult,
|
|
EXCEPINFO* pExcepInfo,
|
|
UINT* puArgErr)
|
|
{
|
|
UINT ret;
|
|
VARIANTARG varg0;
|
|
HRESULT hr;
|
|
|
|
VariantInit(&varg0);
|
|
|
|
switch (dispIdMember)
|
|
{
|
|
case DISPID_SUMMARYINFO_PROPERTY:
|
|
if (wFlags & DISPATCH_PROPERTYGET)
|
|
{
|
|
UINT type;
|
|
INT value;
|
|
INT pid;
|
|
|
|
static WCHAR szEmpty[] = {0};
|
|
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
pid = V_I4(&varg0);
|
|
|
|
if (pid == PID_CODEPAGE || (pid >= PID_PAGECOUNT && pid <= PID_CHARCOUNT) || pid == PID_SECURITY)
|
|
{
|
|
if ((ret = MsiSummaryInfoGetPropertyW(This->msiHandle, pid, &type, &value,
|
|
NULL, NULL, NULL)) != ERROR_SUCCESS)
|
|
ERR("MsiSummaryInfoGetProperty returned %d\n", ret);
|
|
else if (type == VT_I2)
|
|
{
|
|
V_VT(pVarResult) = VT_I2;
|
|
V_I2(pVarResult) = value;
|
|
}
|
|
else if (type == VT_I4)
|
|
{
|
|
V_VT(pVarResult) = VT_I4;
|
|
V_I4(pVarResult) = value;
|
|
}
|
|
}
|
|
else if ((pid >= PID_TITLE && pid <= PID_REVNUMBER) || pid == PID_APPNAME)
|
|
{
|
|
LPWSTR str;
|
|
DWORD size = 0;
|
|
|
|
if ((ret = MsiSummaryInfoGetPropertyW(This->msiHandle, pid, &type, NULL,
|
|
NULL, szEmpty, &size)) == ERROR_MORE_DATA)
|
|
{
|
|
if (!(str = msi_alloc(++size * sizeof(WCHAR))))
|
|
ERR("Out of memory\n");
|
|
else if ((ret = MsiSummaryInfoGetPropertyW(This->msiHandle, pid, &type, NULL,
|
|
NULL, str, &size)) == ERROR_SUCCESS)
|
|
{
|
|
V_VT(pVarResult) = VT_BSTR;
|
|
V_BSTR(pVarResult) = SysAllocString(str);
|
|
}
|
|
msi_free(str);
|
|
}
|
|
if (ret != ERROR_SUCCESS && ret != ERROR_MORE_DATA)
|
|
ERR("MsiSummaryInfoGetProperty returned %d\n", ret);
|
|
}
|
|
else if (pid >= PID_EDITTIME && pid <= PID_LASTSAVE_DTM)
|
|
{
|
|
FILETIME ft, ftlocal;
|
|
SYSTEMTIME st;
|
|
DATE date;
|
|
|
|
if ((ret = MsiSummaryInfoGetPropertyW(This->msiHandle, pid, &type, &value,
|
|
&ft, NULL, NULL)) != ERROR_SUCCESS)
|
|
ERR("MsiSummaryInfoGetProperty returned %d\n", ret);
|
|
else if (type == VT_FILETIME)
|
|
{
|
|
FileTimeToLocalFileTime(&ft, &ftlocal);
|
|
FileTimeToSystemTime(&ftlocal, &st);
|
|
SystemTimeToVariantTime(&st, &date);
|
|
|
|
V_VT(pVarResult) = VT_DATE;
|
|
V_DATE(pVarResult) = date;
|
|
}
|
|
}
|
|
else if (pid != PID_DICTIONARY && pid != PID_THUMBNAIL)
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
default:
|
|
return DISP_E_MEMBERNOTFOUND;
|
|
}
|
|
|
|
VariantClear(&varg0);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI RecordImpl_Invoke(
|
|
AutomationObject* This,
|
|
DISPID dispIdMember,
|
|
REFIID riid,
|
|
LCID lcid,
|
|
WORD wFlags,
|
|
DISPPARAMS* pDispParams,
|
|
VARIANT* pVarResult,
|
|
EXCEPINFO* pExcepInfo,
|
|
UINT* puArgErr)
|
|
{
|
|
WCHAR *szString;
|
|
DWORD dwLen;
|
|
UINT ret;
|
|
VARIANTARG varg0, varg1;
|
|
HRESULT hr;
|
|
|
|
VariantInit(&varg0);
|
|
VariantInit(&varg1);
|
|
|
|
switch (dispIdMember)
|
|
{
|
|
case DISPID_RECORD_FIELDCOUNT:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
V_VT(pVarResult) = VT_I4;
|
|
V_I4(pVarResult) = MsiRecordGetFieldCount(This->msiHandle);
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_RECORD_STRINGDATA:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
V_VT(pVarResult) = VT_BSTR;
|
|
V_BSTR(pVarResult) = NULL;
|
|
if ((ret = MsiRecordGetStringW(This->msiHandle, V_I4(&varg0), NULL, &dwLen)) == ERROR_SUCCESS)
|
|
{
|
|
if (!(szString = msi_alloc((++dwLen)*sizeof(WCHAR))))
|
|
ERR("Out of memory\n");
|
|
else if ((ret = MsiRecordGetStringW(This->msiHandle, V_I4(&varg0), szString, &dwLen)) == ERROR_SUCCESS)
|
|
V_BSTR(pVarResult) = SysAllocString(szString);
|
|
msi_free(szString);
|
|
}
|
|
if (ret != ERROR_SUCCESS)
|
|
ERR("MsiRecordGetString returned %d\n", ret);
|
|
} else if (wFlags & DISPATCH_PROPERTYPUT) {
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam(pDispParams, DISPID_PROPERTYPUT, VT_BSTR, &varg1, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
if ((ret = MsiRecordSetStringW(This->msiHandle, V_I4(&varg0), V_BSTR(&varg1))) != ERROR_SUCCESS)
|
|
{
|
|
VariantClear(&varg1);
|
|
ERR("MsiRecordSetString returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_RECORD_INTEGERDATA:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
V_VT(pVarResult) = VT_I4;
|
|
V_I4(pVarResult) = MsiRecordGetInteger(This->msiHandle, V_I4(&varg0));
|
|
} else if (wFlags & DISPATCH_PROPERTYPUT) {
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam(pDispParams, DISPID_PROPERTYPUT, VT_I4, &varg1, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
if ((ret = MsiRecordSetInteger(This->msiHandle, V_I4(&varg0), V_I4(&varg1))) != ERROR_SUCCESS)
|
|
{
|
|
ERR("MsiRecordSetInteger returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
default:
|
|
return DISP_E_MEMBERNOTFOUND;
|
|
}
|
|
|
|
VariantClear(&varg1);
|
|
VariantClear(&varg0);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI ListImpl_Invoke(
|
|
AutomationObject* This,
|
|
DISPID dispIdMember,
|
|
REFIID riid,
|
|
LCID lcid,
|
|
WORD wFlags,
|
|
DISPPARAMS* pDispParams,
|
|
VARIANT* pVarResult,
|
|
EXCEPINFO* pExcepInfo,
|
|
UINT* puArgErr)
|
|
{
|
|
ListData *data = (ListData *)private_data(This);
|
|
HRESULT hr;
|
|
VARIANTARG varg0;
|
|
IUnknown *pUnk = NULL;
|
|
|
|
VariantInit(&varg0);
|
|
|
|
switch (dispIdMember)
|
|
{
|
|
case DISPID_LIST__NEWENUM:
|
|
if (wFlags & DISPATCH_METHOD) {
|
|
V_VT(pVarResult) = VT_UNKNOWN;
|
|
if (SUCCEEDED(hr = create_list_enumerator(NULL, (LPVOID *)&pUnk, This, 0)))
|
|
{
|
|
IUnknown_AddRef(pUnk);
|
|
V_UNKNOWN(pVarResult) = pUnk;
|
|
}
|
|
else
|
|
ERR("Failed to create IEnumVARIANT object, hresult 0x%08x\n", hr);
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_LIST_ITEM:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
if (V_I4(&varg0) < 0 || V_I4(&varg0) >= data->ulCount)
|
|
return DISP_E_BADINDEX;
|
|
VariantCopy(pVarResult, &data->pVars[V_I4(&varg0)]);
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_LIST_COUNT:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
V_VT(pVarResult) = VT_I4;
|
|
V_I4(pVarResult) = data->ulCount;
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
default:
|
|
return DISP_E_MEMBERNOTFOUND;
|
|
}
|
|
|
|
VariantClear(&varg0);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static void WINAPI ListImpl_Free(AutomationObject *This)
|
|
{
|
|
ListData *data = private_data(This);
|
|
ULONG idx;
|
|
|
|
for (idx=0; idx<data->ulCount; idx++)
|
|
VariantClear(&data->pVars[idx]);
|
|
HeapFree(GetProcessHeap(), 0, data->pVars);
|
|
}
|
|
|
|
static HRESULT WINAPI ViewImpl_Invoke(
|
|
AutomationObject* This,
|
|
DISPID dispIdMember,
|
|
REFIID riid,
|
|
LCID lcid,
|
|
WORD wFlags,
|
|
DISPPARAMS* pDispParams,
|
|
VARIANT* pVarResult,
|
|
EXCEPINFO* pExcepInfo,
|
|
UINT* puArgErr)
|
|
{
|
|
MSIHANDLE msiHandle;
|
|
IDispatch *pDispatch = NULL;
|
|
UINT ret;
|
|
VARIANTARG varg0, varg1;
|
|
HRESULT hr;
|
|
|
|
VariantInit(&varg0);
|
|
VariantInit(&varg1);
|
|
|
|
switch (dispIdMember)
|
|
{
|
|
case DISPID_VIEW_EXECUTE:
|
|
if (wFlags & DISPATCH_METHOD)
|
|
{
|
|
hr = DispGetParam(pDispParams, 0, VT_DISPATCH, &varg0, puArgErr);
|
|
if (SUCCEEDED(hr) && V_DISPATCH(&varg0) != NULL)
|
|
MsiViewExecute(This->msiHandle, ((AutomationObject *)V_DISPATCH(&varg0))->msiHandle);
|
|
else
|
|
MsiViewExecute(This->msiHandle, 0);
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_VIEW_FETCH:
|
|
if (wFlags & DISPATCH_METHOD)
|
|
{
|
|
V_VT(pVarResult) = VT_DISPATCH;
|
|
if ((ret = MsiViewFetch(This->msiHandle, &msiHandle)) == ERROR_SUCCESS)
|
|
{
|
|
if (SUCCEEDED(hr = create_automation_object(msiHandle, NULL, (LPVOID*)&pDispatch, &DIID_Record, RecordImpl_Invoke, NULL, 0)))
|
|
{
|
|
IDispatch_AddRef(pDispatch);
|
|
V_DISPATCH(pVarResult) = pDispatch;
|
|
}
|
|
else
|
|
ERR("Failed to create Record object, hresult 0x%08x\n", hr);
|
|
}
|
|
else if (ret == ERROR_NO_MORE_ITEMS)
|
|
V_DISPATCH(pVarResult) = NULL;
|
|
else
|
|
{
|
|
ERR("MsiViewFetch returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_VIEW_MODIFY:
|
|
if (wFlags & DISPATCH_METHOD)
|
|
{
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam(pDispParams, 1, VT_DISPATCH, &varg1, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
if (!V_DISPATCH(&varg1)) return DISP_E_EXCEPTION;
|
|
if ((ret = MsiViewModify(This->msiHandle, V_I4(&varg0), ((AutomationObject *)V_DISPATCH(&varg1))->msiHandle)) != ERROR_SUCCESS)
|
|
{
|
|
VariantClear(&varg1);
|
|
ERR("MsiViewModify returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_VIEW_CLOSE:
|
|
if (wFlags & DISPATCH_METHOD)
|
|
{
|
|
MsiViewClose(This->msiHandle);
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
default:
|
|
return DISP_E_MEMBERNOTFOUND;
|
|
}
|
|
|
|
VariantClear(&varg1);
|
|
VariantClear(&varg0);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI DatabaseImpl_Invoke(
|
|
AutomationObject* This,
|
|
DISPID dispIdMember,
|
|
REFIID riid,
|
|
LCID lcid,
|
|
WORD wFlags,
|
|
DISPPARAMS* pDispParams,
|
|
VARIANT* pVarResult,
|
|
EXCEPINFO* pExcepInfo,
|
|
UINT* puArgErr)
|
|
{
|
|
MSIHANDLE msiHandle;
|
|
IDispatch *pDispatch = NULL;
|
|
UINT ret;
|
|
VARIANTARG varg0, varg1;
|
|
HRESULT hr;
|
|
|
|
VariantInit(&varg0);
|
|
VariantInit(&varg1);
|
|
|
|
switch (dispIdMember)
|
|
{
|
|
case DISPID_DATABASE_SUMMARYINFORMATION:
|
|
if (wFlags & DISPATCH_PROPERTYGET)
|
|
{
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr))
|
|
V_I4(&varg0) = 0;
|
|
|
|
V_VT(pVarResult) = VT_DISPATCH;
|
|
if ((ret = MsiGetSummaryInformationW(This->msiHandle, NULL, V_I4(&varg0), &msiHandle)) == ERROR_SUCCESS)
|
|
{
|
|
hr = create_automation_object(msiHandle, NULL, (LPVOID *)&pDispatch, &DIID_SummaryInfo, SummaryInfoImpl_Invoke, NULL, 0);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
IDispatch_AddRef(pDispatch);
|
|
V_DISPATCH(pVarResult) = pDispatch;
|
|
}
|
|
else
|
|
ERR("Failed to create SummaryInfo object: 0x%08x\n", hr);
|
|
}
|
|
else
|
|
{
|
|
ERR("MsiGetSummaryInformation returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_DATABASE_OPENVIEW:
|
|
if (wFlags & DISPATCH_METHOD)
|
|
{
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
V_VT(pVarResult) = VT_DISPATCH;
|
|
if ((ret = MsiDatabaseOpenViewW(This->msiHandle, V_BSTR(&varg0), &msiHandle)) == ERROR_SUCCESS)
|
|
{
|
|
if (SUCCEEDED(hr = create_automation_object(msiHandle, NULL, (LPVOID*)&pDispatch, &DIID_View, ViewImpl_Invoke, NULL, 0)))
|
|
{
|
|
IDispatch_AddRef(pDispatch);
|
|
V_DISPATCH(pVarResult) = pDispatch;
|
|
}
|
|
else
|
|
ERR("Failed to create View object, hresult 0x%08x\n", hr);
|
|
}
|
|
else
|
|
{
|
|
VariantClear(&varg0);
|
|
ERR("MsiDatabaseOpenView returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
default:
|
|
return DISP_E_MEMBERNOTFOUND;
|
|
}
|
|
|
|
VariantClear(&varg1);
|
|
VariantClear(&varg0);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI SessionImpl_Invoke(
|
|
AutomationObject* This,
|
|
DISPID dispIdMember,
|
|
REFIID riid,
|
|
LCID lcid,
|
|
WORD wFlags,
|
|
DISPPARAMS* pDispParams,
|
|
VARIANT* pVarResult,
|
|
EXCEPINFO* pExcepInfo,
|
|
UINT* puArgErr)
|
|
{
|
|
SessionData *data = private_data(This);
|
|
WCHAR *szString;
|
|
DWORD dwLen;
|
|
IDispatch *pDispatch = NULL;
|
|
MSIHANDLE msiHandle;
|
|
LANGID langId;
|
|
UINT ret;
|
|
INSTALLSTATE iInstalled, iAction;
|
|
VARIANTARG varg0, varg1;
|
|
HRESULT hr;
|
|
|
|
VariantInit(&varg0);
|
|
VariantInit(&varg1);
|
|
|
|
switch (dispIdMember)
|
|
{
|
|
case DISPID_SESSION_INSTALLER:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
V_VT(pVarResult) = VT_DISPATCH;
|
|
IDispatch_AddRef(data->pInstaller);
|
|
V_DISPATCH(pVarResult) = data->pInstaller;
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_SESSION_PROPERTY:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
V_VT(pVarResult) = VT_BSTR;
|
|
V_BSTR(pVarResult) = NULL;
|
|
if ((ret = MsiGetPropertyW(This->msiHandle, V_BSTR(&varg0), NULL, &dwLen)) == ERROR_SUCCESS)
|
|
{
|
|
if (!(szString = msi_alloc((++dwLen)*sizeof(WCHAR))))
|
|
ERR("Out of memory\n");
|
|
else if ((ret = MsiGetPropertyW(This->msiHandle, V_BSTR(&varg0), szString, &dwLen)) == ERROR_SUCCESS)
|
|
V_BSTR(pVarResult) = SysAllocString(szString);
|
|
msi_free(szString);
|
|
}
|
|
if (ret != ERROR_SUCCESS)
|
|
ERR("MsiGetProperty returned %d\n", ret);
|
|
} else if (wFlags & DISPATCH_PROPERTYPUT) {
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam(pDispParams, DISPID_PROPERTYPUT, VT_BSTR, &varg1, puArgErr);
|
|
if (FAILED(hr)) {
|
|
VariantClear(&varg0);
|
|
return hr;
|
|
}
|
|
if ((ret = MsiSetPropertyW(This->msiHandle, V_BSTR(&varg0), V_BSTR(&varg1))) != ERROR_SUCCESS)
|
|
{
|
|
VariantClear(&varg0);
|
|
VariantClear(&varg1);
|
|
ERR("MsiSetProperty returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_SESSION_LANGUAGE:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
langId = MsiGetLanguage(This->msiHandle);
|
|
V_VT(pVarResult) = VT_I4;
|
|
V_I4(pVarResult) = langId;
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_SESSION_MODE:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
V_VT(pVarResult) = VT_BOOL;
|
|
V_BOOL(pVarResult) = MsiGetMode(This->msiHandle, V_I4(&varg0));
|
|
} else if (wFlags & DISPATCH_PROPERTYPUT) {
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam(pDispParams, DISPID_PROPERTYPUT, VT_BOOL, &varg1, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
if ((ret = MsiSetMode(This->msiHandle, V_I4(&varg0), V_BOOL(&varg1))) != ERROR_SUCCESS)
|
|
{
|
|
ERR("MsiSetMode returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_SESSION_DATABASE:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
V_VT(pVarResult) = VT_DISPATCH;
|
|
if ((msiHandle = MsiGetActiveDatabase(This->msiHandle)))
|
|
{
|
|
if (SUCCEEDED(hr = create_automation_object(msiHandle, NULL, (LPVOID*)&pDispatch, &DIID_Database, DatabaseImpl_Invoke, NULL, 0)))
|
|
{
|
|
IDispatch_AddRef(pDispatch);
|
|
V_DISPATCH(pVarResult) = pDispatch;
|
|
}
|
|
else
|
|
ERR("Failed to create Database object, hresult 0x%08x\n", hr);
|
|
}
|
|
else
|
|
{
|
|
ERR("MsiGetActiveDatabase failed\n");
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_SESSION_DOACTION:
|
|
if (wFlags & DISPATCH_METHOD) {
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
ret = MsiDoActionW(This->msiHandle, V_BSTR(&varg0));
|
|
V_VT(pVarResult) = VT_I4;
|
|
switch (ret)
|
|
{
|
|
case ERROR_FUNCTION_NOT_CALLED:
|
|
V_I4(pVarResult) = msiDoActionStatusNoAction;
|
|
break;
|
|
case ERROR_SUCCESS:
|
|
V_I4(pVarResult) = msiDoActionStatusSuccess;
|
|
break;
|
|
case ERROR_INSTALL_USEREXIT:
|
|
V_I4(pVarResult) = msiDoActionStatusUserExit;
|
|
break;
|
|
case ERROR_INSTALL_FAILURE:
|
|
V_I4(pVarResult) = msiDoActionStatusFailure;
|
|
break;
|
|
case ERROR_INSTALL_SUSPEND:
|
|
V_I4(pVarResult) = msiDoActionStatusSuspend;
|
|
break;
|
|
case ERROR_MORE_DATA:
|
|
V_I4(pVarResult) = msiDoActionStatusFinished;
|
|
break;
|
|
case ERROR_INVALID_HANDLE_STATE:
|
|
V_I4(pVarResult) = msiDoActionStatusWrongState;
|
|
break;
|
|
case ERROR_INVALID_DATA:
|
|
V_I4(pVarResult) = msiDoActionStatusBadActionData;
|
|
break;
|
|
default:
|
|
VariantClear(&varg0);
|
|
FIXME("MsiDoAction returned unhandled value %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_SESSION_EVALUATECONDITION:
|
|
if (wFlags & DISPATCH_METHOD) {
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
V_VT(pVarResult) = VT_I4;
|
|
V_I4(pVarResult) = MsiEvaluateConditionW(This->msiHandle, V_BSTR(&varg0));
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_SESSION_SETINSTALLLEVEL:
|
|
if (wFlags & DISPATCH_METHOD) {
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
if ((ret = MsiSetInstallLevel(This->msiHandle, V_I4(&varg0))) != ERROR_SUCCESS)
|
|
{
|
|
ERR("MsiSetInstallLevel returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_SESSION_FEATURECURRENTSTATE:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
V_VT(pVarResult) = VT_I4;
|
|
if ((ret = MsiGetFeatureStateW(This->msiHandle, V_BSTR(&varg0), &iInstalled, &iAction)) == ERROR_SUCCESS)
|
|
V_I4(pVarResult) = iInstalled;
|
|
else
|
|
{
|
|
ERR("MsiGetFeatureState returned %d\n", ret);
|
|
V_I4(pVarResult) = msiInstallStateUnknown;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_SESSION_FEATUREREQUESTSTATE:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
V_VT(pVarResult) = VT_I4;
|
|
if ((ret = MsiGetFeatureStateW(This->msiHandle, V_BSTR(&varg0), &iInstalled, &iAction)) == ERROR_SUCCESS)
|
|
V_I4(pVarResult) = iAction;
|
|
else
|
|
{
|
|
ERR("MsiGetFeatureState returned %d\n", ret);
|
|
V_I4(pVarResult) = msiInstallStateUnknown;
|
|
}
|
|
} else if (wFlags & DISPATCH_PROPERTYPUT) {
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam(pDispParams, DISPID_PROPERTYPUT, VT_I4, &varg1, puArgErr);
|
|
if (FAILED(hr)) {
|
|
VariantClear(&varg0);
|
|
return hr;
|
|
}
|
|
if ((ret = MsiSetFeatureStateW(This->msiHandle, V_BSTR(&varg0), V_I4(&varg1))) != ERROR_SUCCESS)
|
|
{
|
|
VariantClear(&varg0);
|
|
ERR("MsiSetFeatureState returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
default:
|
|
return DISP_E_MEMBERNOTFOUND;
|
|
}
|
|
|
|
VariantClear(&varg1);
|
|
VariantClear(&varg0);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
/* Fill the variant pointed to by pVarResult with the value & size returned by RegQueryValueEx as dictated by the
|
|
* registry value type. Used by Installer::RegistryValue. */
|
|
static void variant_from_registry_value(VARIANT *pVarResult, DWORD dwType, LPBYTE lpData, DWORD dwSize)
|
|
{
|
|
static const WCHAR szREG_BINARY[] = { '(','R','E','G','_','B','I','N','A','R','Y',')',0 };
|
|
static const WCHAR szREG_[] = { '(','R','E','G','_',']',0 };
|
|
WCHAR *szString = (WCHAR *)lpData;
|
|
LPWSTR szNewString = NULL;
|
|
DWORD dwNewSize = 0;
|
|
int idx;
|
|
|
|
switch (dwType)
|
|
{
|
|
/* Registry strings may not be null terminated so we must use SysAllocStringByteLen/Len */
|
|
case REG_MULTI_SZ: /* Multi SZ change internal null characters to newlines */
|
|
idx = (dwSize/sizeof(WCHAR))-1;
|
|
while (idx >= 0 && !szString[idx]) idx--;
|
|
for (; idx >= 0; idx--)
|
|
if (!szString[idx]) szString[idx] = '\n';
|
|
case REG_SZ:
|
|
V_VT(pVarResult) = VT_BSTR;
|
|
V_BSTR(pVarResult) = SysAllocStringByteLen((LPCSTR)szString, dwSize);
|
|
break;
|
|
|
|
case REG_EXPAND_SZ:
|
|
if (!(dwNewSize = ExpandEnvironmentStringsW(szString, szNewString, dwNewSize)))
|
|
ERR("ExpandEnvironmentStrings returned error %d\n", GetLastError());
|
|
else if (!(szNewString = msi_alloc(dwNewSize)))
|
|
ERR("Out of memory\n");
|
|
else if (!(dwNewSize = ExpandEnvironmentStringsW(szString, szNewString, dwNewSize)))
|
|
ERR("ExpandEnvironmentStrings returned error %d\n", GetLastError());
|
|
else
|
|
{
|
|
V_VT(pVarResult) = VT_BSTR;
|
|
V_BSTR(pVarResult) = SysAllocStringLen(szNewString, dwNewSize);
|
|
}
|
|
msi_free(szNewString);
|
|
break;
|
|
|
|
case REG_DWORD:
|
|
V_VT(pVarResult) = VT_I4;
|
|
V_I4(pVarResult) = *((DWORD *)lpData);
|
|
break;
|
|
|
|
case REG_QWORD:
|
|
V_VT(pVarResult) = VT_BSTR;
|
|
V_BSTR(pVarResult) = SysAllocString(szREG_); /* Weird string, don't know why native returns it */
|
|
break;
|
|
|
|
case REG_BINARY:
|
|
V_VT(pVarResult) = VT_BSTR;
|
|
V_BSTR(pVarResult) = SysAllocString(szREG_BINARY);
|
|
break;
|
|
|
|
case REG_NONE:
|
|
V_VT(pVarResult) = VT_EMPTY;
|
|
break;
|
|
|
|
default:
|
|
FIXME("Unhandled registry value type %d\n", dwType);
|
|
}
|
|
}
|
|
|
|
static HRESULT WINAPI InstallerImpl_Invoke(
|
|
AutomationObject* This,
|
|
DISPID dispIdMember,
|
|
REFIID riid,
|
|
LCID lcid,
|
|
WORD wFlags,
|
|
DISPPARAMS* pDispParams,
|
|
VARIANT* pVarResult,
|
|
EXCEPINFO* pExcepInfo,
|
|
UINT* puArgErr)
|
|
{
|
|
MSIHANDLE msiHandle;
|
|
IDispatch *pDispatch = NULL;
|
|
UINT ret;
|
|
VARIANTARG varg0, varg1, varg2;
|
|
HRESULT hr;
|
|
LPWSTR szString = NULL;
|
|
DWORD dwSize = 0;
|
|
|
|
VariantInit(&varg0);
|
|
VariantInit(&varg1);
|
|
VariantInit(&varg2);
|
|
|
|
switch (dispIdMember)
|
|
{
|
|
case DISPID_INSTALLER_CREATERECORD:
|
|
if (wFlags & DISPATCH_METHOD)
|
|
{
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
V_VT(pVarResult) = VT_DISPATCH;
|
|
if ((msiHandle = MsiCreateRecord(V_I4(&varg0))))
|
|
{
|
|
if (SUCCEEDED(hr = create_automation_object(msiHandle, NULL, (LPVOID*)&pDispatch, &DIID_Record, RecordImpl_Invoke, NULL, 0)))
|
|
{
|
|
IDispatch_AddRef(pDispatch);
|
|
V_DISPATCH(pVarResult) = pDispatch;
|
|
}
|
|
else
|
|
ERR("Failed to create Record object, hresult 0x%08x\n", hr);
|
|
}
|
|
else
|
|
{
|
|
ERR("MsiCreateRecord failed\n");
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_INSTALLER_OPENPACKAGE:
|
|
if (wFlags & DISPATCH_METHOD)
|
|
{
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam(pDispParams, 1, VT_I4, &varg1, puArgErr);
|
|
if (FAILED(hr))
|
|
{
|
|
VariantClear(&varg0);
|
|
return hr;
|
|
}
|
|
V_VT(pVarResult) = VT_DISPATCH;
|
|
if ((ret = MsiOpenPackageExW(V_BSTR(&varg0), V_I4(&varg1), &msiHandle)) == ERROR_SUCCESS)
|
|
{
|
|
if (SUCCEEDED(hr = create_session(msiHandle, (IDispatch *)This, &pDispatch)))
|
|
{
|
|
IDispatch_AddRef(pDispatch);
|
|
V_DISPATCH(pVarResult) = pDispatch;
|
|
}
|
|
else
|
|
ERR("Failed to create Session object, hresult 0x%08x\n", hr);
|
|
}
|
|
else
|
|
{
|
|
VariantClear(&varg0);
|
|
ERR("MsiOpenPackageEx returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_INSTALLER_OPENDATABASE:
|
|
if (wFlags & DISPATCH_METHOD)
|
|
{
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
hr = DispGetParam(pDispParams, 1, VT_BSTR, &varg1, puArgErr);
|
|
if (FAILED(hr))
|
|
{
|
|
VariantClear(&varg0);
|
|
return hr;
|
|
}
|
|
|
|
V_VT(pVarResult) = VT_DISPATCH;
|
|
if ((ret = MsiOpenDatabaseW(V_BSTR(&varg0), V_BSTR(&varg1), &msiHandle)) == ERROR_SUCCESS)
|
|
{
|
|
hr = create_automation_object(msiHandle, NULL, (LPVOID *)&pDispatch,
|
|
&DIID_Database, DatabaseImpl_Invoke, NULL, 0);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
IDispatch_AddRef(pDispatch);
|
|
V_DISPATCH(pVarResult) = pDispatch;
|
|
}
|
|
else
|
|
ERR("Failed to create Database object: 0x%08x\n", hr);
|
|
}
|
|
else
|
|
{
|
|
VariantClear(&varg0);
|
|
VariantClear(&varg1);
|
|
ERR("MsiOpenDatabase returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_INSTALLER_INSTALLPRODUCT:
|
|
if (wFlags & DISPATCH_METHOD)
|
|
{
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam(pDispParams, 1, VT_BSTR, &varg1, puArgErr);
|
|
if (FAILED(hr))
|
|
{
|
|
VariantClear(&varg0);
|
|
return hr;
|
|
}
|
|
if ((ret = MsiInstallProductW(V_BSTR(&varg0), V_BSTR(&varg1))) != ERROR_SUCCESS)
|
|
{
|
|
VariantClear(&varg1);
|
|
VariantClear(&varg0);
|
|
ERR("MsiInstallProduct returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_INSTALLER_REGISTRYVALUE:
|
|
if (wFlags & DISPATCH_METHOD) {
|
|
HKEY hkey;
|
|
DWORD dwType;
|
|
UINT posValue = 2; /* Save valuePos so we can save puArgErr if we are unable to do our type conversions */
|
|
|
|
hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam(pDispParams, 1, VT_BSTR, &varg1, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam_CopyOnly(pDispParams, &posValue, &varg2);
|
|
if (FAILED(hr))
|
|
{
|
|
VariantClear(&varg1);
|
|
return hr;
|
|
}
|
|
ret = RegOpenKeyW((HKEY)V_I4(&varg0), V_BSTR(&varg1), &hkey);
|
|
|
|
/* Third parameter can be VT_EMPTY, VT_I4, or VT_BSTR */
|
|
switch (V_VT(&varg2))
|
|
{
|
|
case VT_EMPTY: /* Return VT_BOOL as to whether or not registry key exists */
|
|
V_VT(pVarResult) = VT_BOOL;
|
|
V_BOOL(pVarResult) = (ret == ERROR_SUCCESS);
|
|
break;
|
|
|
|
case VT_BSTR: /* Return value of specified key if it exists */
|
|
if (ret == ERROR_SUCCESS &&
|
|
(ret = RegQueryValueExW(hkey, V_BSTR(&varg2), NULL, NULL, NULL, &dwSize)) == ERROR_SUCCESS)
|
|
{
|
|
if (!(szString = msi_alloc(dwSize)))
|
|
ERR("Out of memory\n");
|
|
else if ((ret = RegQueryValueExW(hkey, V_BSTR(&varg2), NULL, &dwType, (LPBYTE)szString, &dwSize)) == ERROR_SUCCESS)
|
|
variant_from_registry_value(pVarResult, dwType, (LPBYTE)szString, dwSize);
|
|
}
|
|
|
|
if (ret != ERROR_SUCCESS)
|
|
{
|
|
msi_free(szString);
|
|
VariantClear(&varg2);
|
|
VariantClear(&varg1);
|
|
return DISP_E_BADINDEX;
|
|
}
|
|
break;
|
|
|
|
default: /* Try to make it into VT_I4, can use VariantChangeType for this */
|
|
hr = VariantChangeType(&varg2, &varg2, 0, VT_I4);
|
|
if (SUCCEEDED(hr) && ret != ERROR_SUCCESS) hr = DISP_E_BADINDEX; /* Conversion fine, but couldn't find key */
|
|
if (FAILED(hr))
|
|
{
|
|
if (hr == DISP_E_TYPEMISMATCH) *puArgErr = posValue;
|
|
VariantClear(&varg2); /* Unknown type, so let's clear it */
|
|
VariantClear(&varg1);
|
|
return hr;
|
|
}
|
|
|
|
/* Retrieve class name or maximum value name or subkey name size */
|
|
if (!V_I4(&varg2))
|
|
ret = RegQueryInfoKeyW(hkey, NULL, &dwSize, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
|
|
else if (V_I4(&varg2) > 0)
|
|
ret = RegQueryInfoKeyW(hkey, NULL, NULL, NULL, NULL, NULL, NULL, NULL, &dwSize, NULL, NULL, NULL);
|
|
else /* V_I4(&varg2) < 0 */
|
|
ret = RegQueryInfoKeyW(hkey, NULL, NULL, NULL, NULL, &dwSize, NULL, NULL, NULL, NULL, NULL, NULL);
|
|
|
|
if (ret == ERROR_SUCCESS)
|
|
{
|
|
if (!(szString = msi_alloc(++dwSize * sizeof(WCHAR))))
|
|
ERR("Out of memory\n");
|
|
else if (!V_I4(&varg2))
|
|
ret = RegQueryInfoKeyW(hkey, szString, &dwSize, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
|
|
else if (V_I4(&varg2) > 0)
|
|
ret = RegEnumValueW(hkey, V_I4(&varg2)-1, szString, &dwSize, 0, 0, NULL, NULL);
|
|
else /* V_I4(&varg2) < 0 */
|
|
ret = RegEnumKeyW(hkey, -1 - V_I4(&varg2), szString, dwSize);
|
|
|
|
if (szString && ret == ERROR_SUCCESS)
|
|
{
|
|
V_VT(pVarResult) = VT_BSTR;
|
|
V_BSTR(pVarResult) = SysAllocString(szString);
|
|
}
|
|
}
|
|
}
|
|
|
|
msi_free(szString);
|
|
RegCloseKey(hkey);
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_INSTALLER_PRODUCTSTATE:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
V_VT(pVarResult) = VT_I4;
|
|
V_I4(pVarResult) = MsiQueryProductStateW(V_BSTR(&varg0));
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_INSTALLER_PRODUCTINFO:
|
|
if (wFlags & DISPATCH_PROPERTYGET) {
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
hr = DispGetParam(pDispParams, 1, VT_BSTR, &varg1, puArgErr);
|
|
if (FAILED(hr))
|
|
{
|
|
VariantClear(&varg0);
|
|
return hr;
|
|
}
|
|
V_VT(pVarResult) = VT_BSTR;
|
|
V_BSTR(pVarResult) = NULL;
|
|
if ((ret = MsiGetProductInfoW(V_BSTR(&varg0), V_BSTR(&varg1), NULL, &dwSize)) == ERROR_SUCCESS)
|
|
{
|
|
if (!(szString = msi_alloc((++dwSize)*sizeof(WCHAR))))
|
|
ERR("Out of memory\n");
|
|
else if ((ret = MsiGetProductInfoW(V_BSTR(&varg0), V_BSTR(&varg1), szString, &dwSize)) == ERROR_SUCCESS)
|
|
V_BSTR(pVarResult) = SysAllocString(szString);
|
|
msi_free(szString);
|
|
}
|
|
if (ret != ERROR_SUCCESS)
|
|
{
|
|
ERR("MsiGetProductInfo returned %d\n", ret);
|
|
VariantClear(&varg1);
|
|
VariantClear(&varg0);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_INSTALLER_PRODUCTS:
|
|
if (wFlags & DISPATCH_PROPERTYGET)
|
|
{
|
|
ListData *ldata = NULL;
|
|
ULONG idx = 0;
|
|
WCHAR szProductBuf[GUID_SIZE];
|
|
|
|
/* Find number of products */
|
|
while ((ret = MsiEnumProductsW(idx, szProductBuf)) == ERROR_SUCCESS) idx++;
|
|
if (ret != ERROR_NO_MORE_ITEMS)
|
|
{
|
|
ERR("MsiEnumProducts returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
|
|
V_VT(pVarResult) = VT_DISPATCH;
|
|
if (SUCCEEDED(hr = create_automation_object(0, NULL, (LPVOID*)&pDispatch, &DIID_StringList, ListImpl_Invoke, ListImpl_Free, sizeof(ListData))))
|
|
{
|
|
IDispatch_AddRef(pDispatch);
|
|
V_DISPATCH(pVarResult) = pDispatch;
|
|
|
|
/* Save product strings */
|
|
ldata = (ListData *)private_data((AutomationObject *)pDispatch);
|
|
if (!(ldata->pVars = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(VARIANT)*idx)))
|
|
ERR("Out of memory\n");
|
|
else
|
|
{
|
|
ldata->ulCount = idx;
|
|
for (idx = 0; idx < ldata->ulCount; idx++)
|
|
{
|
|
ret = MsiEnumProductsW(idx, szProductBuf);
|
|
VariantInit(&ldata->pVars[idx]);
|
|
V_VT(&ldata->pVars[idx]) = VT_BSTR;
|
|
V_BSTR(&ldata->pVars[idx]) = SysAllocString(szProductBuf);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
ERR("Failed to create StringList object, hresult 0x%08x\n", hr);
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
case DISPID_INSTALLER_RELATEDPRODUCTS:
|
|
if (wFlags & DISPATCH_PROPERTYGET)
|
|
{
|
|
ListData *ldata = NULL;
|
|
ULONG idx = 0;
|
|
WCHAR szProductBuf[GUID_SIZE];
|
|
|
|
hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
/* Find number of related products */
|
|
while ((ret = MsiEnumRelatedProductsW(V_BSTR(&varg0), 0, idx, szProductBuf)) == ERROR_SUCCESS) idx++;
|
|
if (ret != ERROR_NO_MORE_ITEMS)
|
|
{
|
|
VariantClear(&varg0);
|
|
ERR("MsiEnumRelatedProducts returned %d\n", ret);
|
|
return DISP_E_EXCEPTION;
|
|
}
|
|
|
|
V_VT(pVarResult) = VT_DISPATCH;
|
|
if (SUCCEEDED(hr = create_automation_object(0, NULL, (LPVOID*)&pDispatch, &DIID_StringList, ListImpl_Invoke, ListImpl_Free, sizeof(ListData))))
|
|
{
|
|
IDispatch_AddRef(pDispatch);
|
|
V_DISPATCH(pVarResult) = pDispatch;
|
|
|
|
/* Save product strings */
|
|
ldata = (ListData *)private_data((AutomationObject *)pDispatch);
|
|
if (!(ldata->pVars = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(VARIANT)*idx)))
|
|
ERR("Out of memory\n");
|
|
else
|
|
{
|
|
ldata->ulCount = idx;
|
|
for (idx = 0; idx < ldata->ulCount; idx++)
|
|
{
|
|
ret = MsiEnumRelatedProductsW(V_BSTR(&varg0), 0, idx, szProductBuf);
|
|
VariantInit(&ldata->pVars[idx]);
|
|
V_VT(&ldata->pVars[idx]) = VT_BSTR;
|
|
V_BSTR(&ldata->pVars[idx]) = SysAllocString(szProductBuf);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
ERR("Failed to create StringList object, hresult 0x%08x\n", hr);
|
|
}
|
|
else return DISP_E_MEMBERNOTFOUND;
|
|
break;
|
|
|
|
default:
|
|
return DISP_E_MEMBERNOTFOUND;
|
|
}
|
|
|
|
VariantClear(&varg2);
|
|
VariantClear(&varg1);
|
|
VariantClear(&varg0);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
/* Wrapper around create_automation_object to create an installer object. */
|
|
HRESULT create_msiserver(IUnknown *pOuter, LPVOID *ppObj)
|
|
{
|
|
return create_automation_object(0, pOuter, ppObj, &DIID_Installer, InstallerImpl_Invoke, NULL, 0);
|
|
}
|
|
|
|
/* Wrapper around create_automation_object to create a session object. */
|
|
HRESULT create_session(MSIHANDLE msiHandle, IDispatch *pInstaller, IDispatch **pDispatch)
|
|
{
|
|
HRESULT hr = create_automation_object(msiHandle, NULL, (LPVOID)pDispatch, &DIID_Session, SessionImpl_Invoke, NULL, sizeof(SessionData));
|
|
if (SUCCEEDED(hr) && pDispatch && *pDispatch)
|
|
((SessionData *)private_data((AutomationObject *)*pDispatch))->pInstaller = pInstaller;
|
|
return hr;
|
|
}
|