e919b0d240c7790aadb6f575829bc5a69a8564f0
[reactos.git] / reactos / dll / win32 / msi / automation.c
1 /*
2 * Implementation of OLE Automation for Microsoft Installer (msi.dll)
3 *
4 * Copyright 2007 Misha Koshelev
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 */
20
21 #define COBJMACROS
22
23 #include <stdarg.h>
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winerror.h"
27 #include "winuser.h"
28 #include "winreg.h"
29 #include "msidefs.h"
30 #include "msipriv.h"
31 #include "activscp.h"
32 #include "oleauto.h"
33 #include "shlwapi.h"
34 #include "wine/debug.h"
35 #include "wine/unicode.h"
36
37 #include "msiserver.h"
38 #include "msiserver_dispids.h"
39
40 WINE_DEFAULT_DEBUG_CHANNEL(msi);
41
42 #define REG_INDEX_CLASSES_ROOT 0
43 #define REG_INDEX_DYN_DATA 6
44
45 /*
46 * AutomationObject - "base" class for all automation objects. For each interface, we implement Invoke function
47 * called from AutomationObject::Invoke, and pass this function to create_automation_object.
48 */
49
50 typedef interface AutomationObject AutomationObject;
51
52 interface AutomationObject {
53 /*
54 * VTables - We provide IDispatch, IProvideClassInfo, IProvideClassInfo2, IProvideMultipleClassInfo
55 */
56 const IDispatchVtbl *lpVtbl;
57 const IProvideMultipleClassInfoVtbl *lpvtblIProvideMultipleClassInfo;
58
59 /* Object reference count */
60 LONG ref;
61
62 /* Clsid for this class and it's appropriate ITypeInfo object */
63 LPCLSID clsid;
64 ITypeInfo *iTypeInfo;
65
66 /* The MSI handle of the current object */
67 MSIHANDLE msiHandle;
68
69 /* A function that is called from AutomationObject::Invoke, specific to this type of object. */
70 HRESULT (STDMETHODCALLTYPE *funcInvoke)(
71 AutomationObject* This,
72 DISPID dispIdMember,
73 REFIID riid,
74 LCID lcid,
75 WORD wFlags,
76 DISPPARAMS* pDispParams,
77 VARIANT* pVarResult,
78 EXCEPINFO* pExcepInfo,
79 UINT* puArgErr);
80
81 /* A function that is called from AutomationObject::Release when the object is being freed to free any private
82 * data structures (or NULL) */
83 void (STDMETHODCALLTYPE *funcFree)(AutomationObject* This);
84 };
85
86 /*
87 * ListEnumerator - IEnumVARIANT implementation for MSI automation lists.
88 */
89
90 typedef interface ListEnumerator ListEnumerator;
91
92 interface ListEnumerator {
93 /* VTables */
94 const IEnumVARIANTVtbl *lpVtbl;
95
96 /* Object reference count */
97 LONG ref;
98
99 /* Current position and pointer to AutomationObject that stores actual data */
100 ULONG ulPos;
101 AutomationObject *pObj;
102 };
103
104 /*
105 * Structures for additional data required by specific automation objects
106 */
107
108 typedef struct {
109 ULONG ulCount;
110 VARIANT *pVars;
111 } ListData;
112
113 typedef struct {
114 /* The parent Installer object */
115 IDispatch *pInstaller;
116 } SessionData;
117
118 /* VTables */
119 static const struct IDispatchVtbl AutomationObject_Vtbl;
120 static const struct IProvideMultipleClassInfoVtbl AutomationObject_IProvideMultipleClassInfo_Vtbl;
121 static const struct IEnumVARIANTVtbl ListEnumerator_Vtbl;
122
123 /* Load type info so we don't have to process GetIDsOfNames */
124 HRESULT load_type_info(IDispatch *iface, ITypeInfo **pptinfo, REFIID clsid, LCID lcid)
125 {
126 HRESULT hr;
127 LPTYPELIB pLib = NULL;
128 LPTYPEINFO pInfo = NULL;
129 static const WCHAR szMsiServer[] = {'m','s','i','s','e','r','v','e','r','.','t','l','b'};
130
131 TRACE("(%p)->(%s,%d)\n", iface, debugstr_guid(clsid), lcid);
132
133 /* Load registered type library */
134 hr = LoadRegTypeLib(&LIBID_WindowsInstaller, 1, 0, lcid, &pLib);
135 if (FAILED(hr)) {
136 hr = LoadTypeLib(szMsiServer, &pLib);
137 if (FAILED(hr)) {
138 ERR("Could not load msiserver.tlb\n");
139 return hr;
140 }
141 }
142
143 /* Get type information for object */
144 hr = ITypeLib_GetTypeInfoOfGuid(pLib, clsid, &pInfo);
145 ITypeLib_Release(pLib);
146 if (FAILED(hr)) {
147 ERR("Could not load ITypeInfo for %s\n", debugstr_guid(clsid));
148 return hr;
149 }
150 *pptinfo = pInfo;
151 return S_OK;
152 }
153
154 /* Create the automation object, placing the result in the pointer ppObj. The automation object is created
155 * with the appropriate clsid and invocation function. */
156 static HRESULT create_automation_object(MSIHANDLE msiHandle, IUnknown *pUnkOuter, LPVOID *ppObj, REFIID clsid,
157 HRESULT (STDMETHODCALLTYPE *funcInvoke)(AutomationObject*,DISPID,REFIID,LCID,WORD,DISPPARAMS*,
158 VARIANT*,EXCEPINFO*,UINT*),
159 void (STDMETHODCALLTYPE *funcFree)(AutomationObject*),
160 SIZE_T sizetPrivateData)
161 {
162 AutomationObject *object;
163 HRESULT hr;
164
165 TRACE("(%ld,%p,%p,%s,%p,%p,%ld)\n", (unsigned long)msiHandle, pUnkOuter, ppObj, debugstr_guid(clsid), funcInvoke, funcFree, sizetPrivateData);
166
167 if( pUnkOuter )
168 return CLASS_E_NOAGGREGATION;
169
170 object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(AutomationObject)+sizetPrivateData);
171
172 /* Set all the VTable references */
173 object->lpVtbl = &AutomationObject_Vtbl;
174 object->lpvtblIProvideMultipleClassInfo = &AutomationObject_IProvideMultipleClassInfo_Vtbl;
175 object->ref = 1;
176
177 /* Store data that was passed */
178 object->msiHandle = msiHandle;
179 object->clsid = (LPCLSID)clsid;
180 object->funcInvoke = funcInvoke;
181 object->funcFree = funcFree;
182
183 /* Load our TypeInfo so we don't have to process GetIDsOfNames */
184 object->iTypeInfo = NULL;
185 hr = load_type_info((IDispatch *)object, &object->iTypeInfo, clsid, 0x0);
186 if (FAILED(hr)) {
187 HeapFree(GetProcessHeap(), 0, object);
188 return hr;
189 }
190
191 *ppObj = object;
192
193 return S_OK;
194 }
195
196 /* Create a list enumerator, placing the result in the pointer ppObj. */
197 static HRESULT create_list_enumerator(IUnknown *pUnkOuter, LPVOID *ppObj, AutomationObject *pObj, ULONG ulPos)
198 {
199 ListEnumerator *object;
200
201 TRACE("(%p,%p,%p,%uld)\n", pUnkOuter, ppObj, pObj, ulPos);
202
203 if( pUnkOuter )
204 return CLASS_E_NOAGGREGATION;
205
206 object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ListEnumerator));
207
208 /* Set all the VTable references */
209 object->lpVtbl = &ListEnumerator_Vtbl;
210 object->ref = 1;
211
212 /* Store data that was passed */
213 object->ulPos = ulPos;
214 object->pObj = pObj;
215 if (pObj) IDispatch_AddRef((IDispatch *)pObj);
216
217 *ppObj = object;
218 return S_OK;
219 }
220
221 /* Macros to get pointer to AutomationObject from the other VTables. */
222 static inline AutomationObject *obj_from_IProvideMultipleClassInfo( IProvideMultipleClassInfo *iface )
223 {
224 return (AutomationObject *)((char*)iface - FIELD_OFFSET(AutomationObject, lpvtblIProvideMultipleClassInfo));
225 }
226
227 /* Macro to get pointer to private object data */
228 static inline void *private_data( AutomationObject *This )
229 {
230 return This + 1;
231 }
232
233 /*
234 * AutomationObject methods
235 */
236
237 /*** IUnknown methods ***/
238 static HRESULT WINAPI AutomationObject_QueryInterface(IDispatch* iface, REFIID riid, void** ppvObject)
239 {
240 AutomationObject *This = (AutomationObject *)iface;
241
242 TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
243
244 if (ppvObject == NULL)
245 return E_INVALIDARG;
246
247 *ppvObject = 0;
248
249 if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IDispatch) || IsEqualGUID(riid, This->clsid))
250 *ppvObject = This;
251 else if (IsEqualGUID(riid, &IID_IProvideClassInfo) ||
252 IsEqualGUID(riid, &IID_IProvideClassInfo2) ||
253 IsEqualGUID(riid, &IID_IProvideMultipleClassInfo))
254 *ppvObject = &This->lpvtblIProvideMultipleClassInfo;
255 else
256 {
257 TRACE("() : asking for unsupported interface %s\n",debugstr_guid(riid));
258 return E_NOINTERFACE;
259 }
260
261 /*
262 * Query Interface always increases the reference count by one when it is
263 * successful
264 */
265 IClassFactory_AddRef(iface);
266
267 return S_OK;
268 }
269
270 static ULONG WINAPI AutomationObject_AddRef(IDispatch* iface)
271 {
272 AutomationObject *This = (AutomationObject *)iface;
273
274 TRACE("(%p/%p)\n", iface, This);
275
276 return InterlockedIncrement(&This->ref);
277 }
278
279 static ULONG WINAPI AutomationObject_Release(IDispatch* iface)
280 {
281 AutomationObject *This = (AutomationObject *)iface;
282 ULONG ref = InterlockedDecrement(&This->ref);
283
284 TRACE("(%p/%p)\n", iface, This);
285
286 if (!ref)
287 {
288 if (This->funcFree) This->funcFree(This);
289 ITypeInfo_Release(This->iTypeInfo);
290 MsiCloseHandle(This->msiHandle);
291 HeapFree(GetProcessHeap(), 0, This);
292 }
293
294 return ref;
295 }
296
297 /*** IDispatch methods ***/
298 static HRESULT WINAPI AutomationObject_GetTypeInfoCount(
299 IDispatch* iface,
300 UINT* pctinfo)
301 {
302 AutomationObject *This = (AutomationObject *)iface;
303
304 TRACE("(%p/%p)->(%p)\n", iface, This, pctinfo);
305 *pctinfo = 1;
306 return S_OK;
307 }
308
309 static HRESULT WINAPI AutomationObject_GetTypeInfo(
310 IDispatch* iface,
311 UINT iTInfo,
312 LCID lcid,
313 ITypeInfo** ppTInfo)
314 {
315 AutomationObject *This = (AutomationObject *)iface;
316 TRACE("(%p/%p)->(%d,%d,%p)\n", iface, This, iTInfo, lcid, ppTInfo);
317
318 ITypeInfo_AddRef(This->iTypeInfo);
319 *ppTInfo = This->iTypeInfo;
320 return S_OK;
321 }
322
323 static HRESULT WINAPI AutomationObject_GetIDsOfNames(
324 IDispatch* iface,
325 REFIID riid,
326 LPOLESTR* rgszNames,
327 UINT cNames,
328 LCID lcid,
329 DISPID* rgDispId)
330 {
331 AutomationObject *This = (AutomationObject *)iface;
332 HRESULT hr;
333 TRACE("(%p/%p)->(%p,%p,%d,%d,%p)\n", iface, This, riid, rgszNames, cNames, lcid, rgDispId);
334
335 if (!IsEqualGUID(riid, &IID_NULL)) return E_INVALIDARG;
336 hr = ITypeInfo_GetIDsOfNames(This->iTypeInfo, rgszNames, cNames, rgDispId);
337 if (hr == DISP_E_UNKNOWNNAME)
338 {
339 int idx;
340 for (idx=0; idx<cNames; idx++)
341 {
342 if (rgDispId[idx] == DISPID_UNKNOWN)
343 FIXME("Unknown member %s, clsid %s\n", debugstr_w(rgszNames[idx]), debugstr_guid(This->clsid));
344 }
345 }
346 return hr;
347 }
348
349 /* Maximum number of allowed function parameters+1 */
350 #define MAX_FUNC_PARAMS 20
351
352 /* Some error checking is done here to simplify individual object function invocation */
353 static HRESULT WINAPI AutomationObject_Invoke(
354 IDispatch* iface,
355 DISPID dispIdMember,
356 REFIID riid,
357 LCID lcid,
358 WORD wFlags,
359 DISPPARAMS* pDispParams,
360 VARIANT* pVarResult,
361 EXCEPINFO* pExcepInfo,
362 UINT* puArgErr)
363 {
364 AutomationObject *This = (AutomationObject *)iface;
365 HRESULT hr;
366 unsigned int uArgErr;
367 VARIANT varResultDummy;
368 BSTR bstrName = NULL;
369
370 TRACE("(%p/%p)->(%d,%p,%d,%d,%p,%p,%p,%p)\n", iface, This, dispIdMember, riid, lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
371
372 if (!IsEqualIID(riid, &IID_NULL))
373 {
374 ERR("riid was %s instead of IID_NULL\n", debugstr_guid(riid));
375 return DISP_E_UNKNOWNNAME;
376 }
377
378 if (wFlags & DISPATCH_PROPERTYGET && !pVarResult)
379 {
380 ERR("NULL pVarResult not allowed when DISPATCH_PROPERTYGET specified\n");
381 return DISP_E_PARAMNOTOPTIONAL;
382 }
383
384 /* This simplifies our individual object invocation functions */
385 if (puArgErr == NULL) puArgErr = &uArgErr;
386 if (pVarResult == NULL) pVarResult = &varResultDummy;
387
388 /* Assume return type is void unless determined otherwise */
389 VariantInit(pVarResult);
390
391 /* If we are tracing, we want to see the name of the member we are invoking */
392 if (TRACE_ON(msi))
393 {
394 ITypeInfo_GetDocumentation(This->iTypeInfo, dispIdMember, &bstrName, NULL, NULL, NULL);
395 TRACE("Method %d, %s\n", dispIdMember, debugstr_w(bstrName));
396 }
397
398 hr = This->funcInvoke(This,dispIdMember,riid,lcid,wFlags,pDispParams,pVarResult,pExcepInfo,puArgErr);
399
400 if (hr == DISP_E_MEMBERNOTFOUND) {
401 if (bstrName == NULL) ITypeInfo_GetDocumentation(This->iTypeInfo, dispIdMember, &bstrName, NULL, NULL, NULL);
402 FIXME("Method %d, %s wflags %d not implemented, clsid %s\n", dispIdMember, debugstr_w(bstrName), wFlags, debugstr_guid(This->clsid));
403 }
404 else if (pExcepInfo &&
405 (hr == DISP_E_PARAMNOTFOUND ||
406 hr == DISP_E_EXCEPTION)) {
407 static const WCHAR szComma[] = { ',',0 };
408 static const WCHAR szExceptionSource[] = {'M','s','i',' ','A','P','I',' ','E','r','r','o','r',0};
409 WCHAR szExceptionDescription[MAX_PATH];
410 BSTR bstrParamNames[MAX_FUNC_PARAMS];
411 unsigned namesNo, i;
412 BOOL bFirst = TRUE;
413
414 if (FAILED(ITypeInfo_GetNames(This->iTypeInfo, dispIdMember, bstrParamNames,
415 MAX_FUNC_PARAMS, &namesNo)))
416 {
417 TRACE("Failed to retrieve names for dispIdMember %d\n", dispIdMember);
418 }
419 else
420 {
421 memset(szExceptionDescription, 0, sizeof(szExceptionDescription));
422 for (i=0; i<namesNo; i++)
423 {
424 if (bFirst) bFirst = FALSE;
425 else {
426 lstrcpyW(&szExceptionDescription[lstrlenW(szExceptionDescription)], szComma);
427 }
428 lstrcpyW(&szExceptionDescription[lstrlenW(szExceptionDescription)], bstrParamNames[i]);
429 SysFreeString(bstrParamNames[i]);
430 }
431
432 memset(pExcepInfo, 0, sizeof(EXCEPINFO));
433 pExcepInfo->wCode = 1000;
434 pExcepInfo->bstrSource = SysAllocString(szExceptionSource);
435 pExcepInfo->bstrDescription = SysAllocString(szExceptionDescription);
436 hr = DISP_E_EXCEPTION;
437 }
438 }
439
440 /* Make sure we free the return variant if it is our dummy variant */
441 if (pVarResult == &varResultDummy) VariantClear(pVarResult);
442
443 /* Free function name if we retrieved it */
444 if (bstrName) SysFreeString(bstrName);
445
446 TRACE("Returning 0x%08x, %s\n", hr, SUCCEEDED(hr) ? "ok" : "not ok");
447
448 return hr;
449 }
450
451 static const struct IDispatchVtbl AutomationObject_Vtbl =
452 {
453 AutomationObject_QueryInterface,
454 AutomationObject_AddRef,
455 AutomationObject_Release,
456 AutomationObject_GetTypeInfoCount,
457 AutomationObject_GetTypeInfo,
458 AutomationObject_GetIDsOfNames,
459 AutomationObject_Invoke
460 };
461
462 /*
463 * IProvideMultipleClassInfo methods
464 */
465
466 static HRESULT WINAPI AutomationObject_IProvideMultipleClassInfo_QueryInterface(
467 IProvideMultipleClassInfo* iface,
468 REFIID riid,
469 VOID** ppvoid)
470 {
471 AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
472 return AutomationObject_QueryInterface((IDispatch *)This, riid, ppvoid);
473 }
474
475 static ULONG WINAPI AutomationObject_IProvideMultipleClassInfo_AddRef(IProvideMultipleClassInfo* iface)
476 {
477 AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
478 return AutomationObject_AddRef((IDispatch *)This);
479 }
480
481 static ULONG WINAPI AutomationObject_IProvideMultipleClassInfo_Release(IProvideMultipleClassInfo* iface)
482 {
483 AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
484 return AutomationObject_Release((IDispatch *)This);
485 }
486
487 static HRESULT WINAPI AutomationObject_IProvideMultipleClassInfo_GetClassInfo(IProvideMultipleClassInfo* iface, ITypeInfo** ppTI)
488 {
489 AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
490 TRACE("(%p/%p)->(%p)\n", iface, This, ppTI);
491 return load_type_info((IDispatch *)This, ppTI, This->clsid, 0);
492 }
493
494 static HRESULT WINAPI AutomationObject_IProvideMultipleClassInfo_GetGUID(IProvideMultipleClassInfo* iface, DWORD dwGuidKind, GUID* pGUID)
495 {
496 AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
497 TRACE("(%p/%p)->(%d,%s)\n", iface, This, dwGuidKind, debugstr_guid(pGUID));
498
499 if (dwGuidKind != GUIDKIND_DEFAULT_SOURCE_DISP_IID)
500 return E_INVALIDARG;
501 else {
502 *pGUID = *This->clsid;
503 return S_OK;
504 }
505 }
506
507 static HRESULT WINAPI AutomationObject_GetMultiTypeInfoCount(IProvideMultipleClassInfo* iface, ULONG* pcti)
508 {
509 AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
510
511 TRACE("(%p/%p)->(%p)\n", iface, This, pcti);
512 *pcti = 1;
513 return S_OK;
514 }
515
516 static HRESULT WINAPI AutomationObject_GetInfoOfIndex(IProvideMultipleClassInfo* iface,
517 ULONG iti,
518 DWORD dwFlags,
519 ITypeInfo** pptiCoClass,
520 DWORD* pdwTIFlags,
521 ULONG* pcdispidReserved,
522 IID* piidPrimary,
523 IID* piidSource)
524 {
525 AutomationObject *This = obj_from_IProvideMultipleClassInfo(iface);
526
527 TRACE("(%p/%p)->(%d,%d,%p,%p,%p,%p,%p)\n", iface, This, iti, dwFlags, pptiCoClass, pdwTIFlags, pcdispidReserved, piidPrimary, piidSource);
528
529 if (iti != 0)
530 return E_INVALIDARG;
531
532 if (dwFlags & MULTICLASSINFO_GETTYPEINFO)
533 load_type_info((IDispatch *)This, pptiCoClass, This->clsid, 0);
534
535 if (dwFlags & MULTICLASSINFO_GETNUMRESERVEDDISPIDS)
536 {
537 *pdwTIFlags = 0;
538 *pcdispidReserved = 0;
539 }
540
541 if (dwFlags & MULTICLASSINFO_GETIIDPRIMARY){
542 *piidPrimary = *This->clsid;
543 }
544
545 if (dwFlags & MULTICLASSINFO_GETIIDSOURCE){
546 *piidSource = *This->clsid;
547 }
548
549 return S_OK;
550 }
551
552 static const IProvideMultipleClassInfoVtbl AutomationObject_IProvideMultipleClassInfo_Vtbl =
553 {
554 AutomationObject_IProvideMultipleClassInfo_QueryInterface,
555 AutomationObject_IProvideMultipleClassInfo_AddRef,
556 AutomationObject_IProvideMultipleClassInfo_Release,
557 AutomationObject_IProvideMultipleClassInfo_GetClassInfo,
558 AutomationObject_IProvideMultipleClassInfo_GetGUID,
559 AutomationObject_GetMultiTypeInfoCount,
560 AutomationObject_GetInfoOfIndex
561 };
562
563 /*
564 * ListEnumerator methods
565 */
566
567 /*** IUnknown methods ***/
568 static HRESULT WINAPI ListEnumerator_QueryInterface(IEnumVARIANT* iface, REFIID riid, void** ppvObject)
569 {
570 ListEnumerator *This = (ListEnumerator *)iface;
571
572 TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
573
574 if (ppvObject == NULL)
575 return E_INVALIDARG;
576
577 *ppvObject = 0;
578
579 if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IEnumVARIANT))
580 *ppvObject = This;
581 else
582 {
583 TRACE("() : asking for unsupported interface %s\n",debugstr_guid(riid));
584 return E_NOINTERFACE;
585 }
586
587 IClassFactory_AddRef(iface);
588 return S_OK;
589 }
590
591 static ULONG WINAPI ListEnumerator_AddRef(IEnumVARIANT* iface)
592 {
593 ListEnumerator *This = (ListEnumerator *)iface;
594
595 TRACE("(%p/%p)\n", iface, This);
596
597 return InterlockedIncrement(&This->ref);
598 }
599
600 static ULONG WINAPI ListEnumerator_Release(IEnumVARIANT* iface)
601 {
602 ListEnumerator *This = (ListEnumerator *)iface;
603 ULONG ref = InterlockedDecrement(&This->ref);
604
605 TRACE("(%p/%p)\n", iface, This);
606
607 if (!ref)
608 {
609 if (This->pObj) IDispatch_Release((IDispatch *)This->pObj);
610 HeapFree(GetProcessHeap(), 0, This);
611 }
612
613 return ref;
614 }
615
616 /* IEnumVARIANT methods */
617
618 static HRESULT WINAPI ListEnumerator_Next(IEnumVARIANT* iface, ULONG celt, VARIANT *rgVar, ULONG *pCeltFetched)
619 {
620 ListEnumerator *This = (ListEnumerator *)iface;
621 ListData *data = (ListData *)private_data(This->pObj);
622 ULONG idx, local;
623
624 TRACE("(%p,%uld,%p,%p)\n", iface, celt, rgVar, pCeltFetched);
625
626 if (pCeltFetched != NULL)
627 *pCeltFetched = 0;
628
629 if (rgVar == NULL)
630 return S_FALSE;
631
632 for (local = 0; local < celt; local++)
633 VariantInit(&rgVar[local]);
634
635 for (idx = This->ulPos, local = 0; idx < data->ulCount && local < celt; idx++, local++)
636 VariantCopy(&rgVar[local], &data->pVars[idx]);
637
638 if (pCeltFetched != NULL)
639 *pCeltFetched = local;
640 This->ulPos = idx;
641
642 return (local < celt) ? S_FALSE : S_OK;
643 }
644
645 static HRESULT WINAPI ListEnumerator_Skip(IEnumVARIANT* iface, ULONG celt)
646 {
647 ListEnumerator *This = (ListEnumerator *)iface;
648 ListData *data = (ListData *)private_data(This->pObj);
649
650 TRACE("(%p,%uld)\n", iface, celt);
651
652 This->ulPos += celt;
653 if (This->ulPos >= data->ulCount)
654 {
655 This->ulPos = data->ulCount;
656 return S_FALSE;
657 }
658 return S_OK;
659 }
660
661 static HRESULT WINAPI ListEnumerator_Reset(IEnumVARIANT* iface)
662 {
663 ListEnumerator *This = (ListEnumerator *)iface;
664
665 TRACE("(%p)\n", iface);
666
667 This->ulPos = 0;
668 return S_OK;
669 }
670
671 static HRESULT WINAPI ListEnumerator_Clone(IEnumVARIANT* iface, IEnumVARIANT **ppEnum)
672 {
673 ListEnumerator *This = (ListEnumerator *)iface;
674 HRESULT hr;
675
676 TRACE("(%p,%p)\n", iface, ppEnum);
677
678 if (ppEnum == NULL)
679 return S_FALSE;
680
681 *ppEnum = NULL;
682 hr = create_list_enumerator(NULL, (LPVOID *)ppEnum, This->pObj, 0);
683 if (FAILED(hr))
684 {
685 if (*ppEnum)
686 IUnknown_Release(*ppEnum);
687 return hr;
688 }
689
690 return S_OK;
691 }
692
693 static const struct IEnumVARIANTVtbl ListEnumerator_Vtbl =
694 {
695 ListEnumerator_QueryInterface,
696 ListEnumerator_AddRef,
697 ListEnumerator_Release,
698 ListEnumerator_Next,
699 ListEnumerator_Skip,
700 ListEnumerator_Reset,
701 ListEnumerator_Clone
702 };
703
704 /*
705 * Individual Object Invocation Functions
706 */
707
708 /* Helper function that copies a passed parameter instead of using VariantChangeType like the actual DispGetParam.
709 This function is only for VARIANT type parameters that have several types that cannot be properly discriminated
710 using DispGetParam/VariantChangeType. */
711 static HRESULT WINAPI DispGetParam_CopyOnly(
712 DISPPARAMS *pdispparams, /* [in] Parameter list */
713 UINT *position, /* [in] Position of parameter to copy in pdispparams; on return will contain calculated position */
714 VARIANT *pvarResult) /* [out] Destination for resulting variant */
715 {
716 /* position is counted backwards */
717 UINT pos;
718
719 TRACE("position=%d, cArgs=%d, cNamedArgs=%d\n",
720 *position, pdispparams->cArgs, pdispparams->cNamedArgs);
721 if (*position < pdispparams->cArgs) {
722 /* positional arg? */
723 pos = pdispparams->cArgs - *position - 1;
724 } else {
725 /* FIXME: is this how to handle named args? */
726 for (pos=0; pos<pdispparams->cNamedArgs; pos++)
727 if (pdispparams->rgdispidNamedArgs[pos] == *position) break;
728
729 if (pos==pdispparams->cNamedArgs)
730 return DISP_E_PARAMNOTFOUND;
731 }
732 *position = pos;
733 return VariantCopyInd(pvarResult,
734 &pdispparams->rgvarg[pos]);
735 }
736
737 static HRESULT WINAPI SummaryInfoImpl_Invoke(
738 AutomationObject* This,
739 DISPID dispIdMember,
740 REFIID riid,
741 LCID lcid,
742 WORD wFlags,
743 DISPPARAMS* pDispParams,
744 VARIANT* pVarResult,
745 EXCEPINFO* pExcepInfo,
746 UINT* puArgErr)
747 {
748 UINT ret;
749 VARIANTARG varg0, varg1;
750 FILETIME ft, ftlocal;
751 SYSTEMTIME st;
752 HRESULT hr;
753
754 VariantInit(&varg0);
755 VariantInit(&varg1);
756
757 switch (dispIdMember)
758 {
759 case DISPID_SUMMARYINFO_PROPERTY:
760 if (wFlags & DISPATCH_PROPERTYGET)
761 {
762 UINT type;
763 INT value;
764 DWORD size = 0;
765 DATE date;
766 LPWSTR str;
767
768 static WCHAR szEmpty[] = {0};
769
770 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
771 if (FAILED(hr)) return hr;
772 ret = MsiSummaryInfoGetPropertyW(This->msiHandle, V_I4(&varg0), &type, &value,
773 &ft, szEmpty, &size);
774 if (ret != ERROR_SUCCESS &&
775 ret != ERROR_MORE_DATA)
776 {
777 ERR("MsiSummaryInfoGetProperty returned %d\n", ret);
778 return DISP_E_EXCEPTION;
779 }
780
781 switch (type)
782 {
783 case VT_EMPTY:
784 break;
785
786 case VT_I2:
787 case VT_I4:
788 V_VT(pVarResult) = VT_I4;
789 V_I4(pVarResult) = value;
790 break;
791
792 case VT_LPSTR:
793 if (!(str = msi_alloc(++size * sizeof(WCHAR))))
794 ERR("Out of memory\n");
795 else if ((ret = MsiSummaryInfoGetPropertyW(This->msiHandle, V_I4(&varg0), &type, NULL,
796 NULL, str, &size)) != ERROR_SUCCESS)
797 ERR("MsiSummaryInfoGetProperty returned %d\n", ret);
798 else
799 {
800 V_VT(pVarResult) = VT_BSTR;
801 V_BSTR(pVarResult) = SysAllocString(str);
802 }
803 msi_free(str);
804 break;
805
806 case VT_FILETIME:
807 FileTimeToLocalFileTime(&ft, &ftlocal);
808 FileTimeToSystemTime(&ftlocal, &st);
809 SystemTimeToVariantTime(&st, &date);
810
811 V_VT(pVarResult) = VT_DATE;
812 V_DATE(pVarResult) = date;
813 break;
814
815 default:
816 ERR("Unhandled variant type %d\n", type);
817 }
818 }
819 else if (wFlags & DISPATCH_PROPERTYPUT)
820 {
821 UINT posValue = DISPID_PROPERTYPUT;
822
823 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
824 if (FAILED(hr)) return hr;
825 hr = DispGetParam_CopyOnly(pDispParams, &posValue, &varg1);
826 if (FAILED(hr))
827 {
828 *puArgErr = posValue;
829 return hr;
830 }
831
832 switch (V_VT(&varg1))
833 {
834 case VT_I2:
835 case VT_I4:
836 ret = MsiSummaryInfoSetPropertyW(This->msiHandle, V_I4(&varg0), V_VT(&varg1), V_I4(&varg1), NULL, NULL);
837 break;
838
839 case VT_DATE:
840 VariantTimeToSystemTime(V_DATE(&varg1), &st);
841 SystemTimeToFileTime(&st, &ftlocal);
842 LocalFileTimeToFileTime(&ftlocal, &ft);
843 ret = MsiSummaryInfoSetPropertyW(This->msiHandle, V_I4(&varg0), VT_FILETIME, 0, &ft, NULL);
844 break;
845
846 case VT_BSTR:
847 ret = MsiSummaryInfoSetPropertyW(This->msiHandle, V_I4(&varg0), VT_LPSTR, 0, NULL, V_BSTR(&varg1));
848 break;
849
850 default:
851 FIXME("Unhandled variant type %d\n", V_VT(&varg1));
852 VariantClear(&varg1);
853 return DISP_E_EXCEPTION;
854 }
855
856 if (ret != ERROR_SUCCESS)
857 {
858 ERR("MsiSummaryInfoSetPropertyW returned %d\n", ret);
859 return DISP_E_EXCEPTION;
860 }
861 }
862 else return DISP_E_MEMBERNOTFOUND;
863 break;
864
865 case DISPID_SUMMARYINFO_PROPERTYCOUNT:
866 if (wFlags & DISPATCH_PROPERTYGET) {
867 UINT count;
868 if ((ret = MsiSummaryInfoGetPropertyCount(This->msiHandle, &count)) != ERROR_SUCCESS)
869 ERR("MsiSummaryInfoGetPropertyCount returned %d\n", ret);
870 else
871 {
872 V_VT(pVarResult) = VT_I4;
873 V_I4(pVarResult) = count;
874 }
875 }
876 else return DISP_E_MEMBERNOTFOUND;
877 break;
878
879 default:
880 return DISP_E_MEMBERNOTFOUND;
881 }
882
883 VariantClear(&varg1);
884 VariantClear(&varg0);
885
886 return S_OK;
887 }
888
889 static HRESULT WINAPI RecordImpl_Invoke(
890 AutomationObject* This,
891 DISPID dispIdMember,
892 REFIID riid,
893 LCID lcid,
894 WORD wFlags,
895 DISPPARAMS* pDispParams,
896 VARIANT* pVarResult,
897 EXCEPINFO* pExcepInfo,
898 UINT* puArgErr)
899 {
900 WCHAR *szString;
901 DWORD dwLen;
902 UINT ret;
903 VARIANTARG varg0, varg1;
904 HRESULT hr;
905
906 VariantInit(&varg0);
907 VariantInit(&varg1);
908
909 switch (dispIdMember)
910 {
911 case DISPID_RECORD_FIELDCOUNT:
912 if (wFlags & DISPATCH_PROPERTYGET) {
913 V_VT(pVarResult) = VT_I4;
914 V_I4(pVarResult) = MsiRecordGetFieldCount(This->msiHandle);
915 }
916 else return DISP_E_MEMBERNOTFOUND;
917 break;
918
919 case DISPID_RECORD_STRINGDATA:
920 if (wFlags & DISPATCH_PROPERTYGET) {
921 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
922 if (FAILED(hr)) return hr;
923 V_VT(pVarResult) = VT_BSTR;
924 V_BSTR(pVarResult) = NULL;
925 if ((ret = MsiRecordGetStringW(This->msiHandle, V_I4(&varg0), NULL, &dwLen)) == ERROR_SUCCESS)
926 {
927 if (!(szString = msi_alloc((++dwLen)*sizeof(WCHAR))))
928 ERR("Out of memory\n");
929 else if ((ret = MsiRecordGetStringW(This->msiHandle, V_I4(&varg0), szString, &dwLen)) == ERROR_SUCCESS)
930 V_BSTR(pVarResult) = SysAllocString(szString);
931 msi_free(szString);
932 }
933 if (ret != ERROR_SUCCESS)
934 ERR("MsiRecordGetString returned %d\n", ret);
935 } else if (wFlags & DISPATCH_PROPERTYPUT) {
936 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
937 if (FAILED(hr)) return hr;
938 hr = DispGetParam(pDispParams, DISPID_PROPERTYPUT, VT_BSTR, &varg1, puArgErr);
939 if (FAILED(hr)) return hr;
940 if ((ret = MsiRecordSetStringW(This->msiHandle, V_I4(&varg0), V_BSTR(&varg1))) != ERROR_SUCCESS)
941 {
942 VariantClear(&varg1);
943 ERR("MsiRecordSetString returned %d\n", ret);
944 return DISP_E_EXCEPTION;
945 }
946 }
947 else return DISP_E_MEMBERNOTFOUND;
948 break;
949
950 case DISPID_RECORD_INTEGERDATA:
951 if (wFlags & DISPATCH_PROPERTYGET) {
952 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
953 if (FAILED(hr)) return hr;
954 V_VT(pVarResult) = VT_I4;
955 V_I4(pVarResult) = MsiRecordGetInteger(This->msiHandle, V_I4(&varg0));
956 } else if (wFlags & DISPATCH_PROPERTYPUT) {
957 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
958 if (FAILED(hr)) return hr;
959 hr = DispGetParam(pDispParams, DISPID_PROPERTYPUT, VT_I4, &varg1, puArgErr);
960 if (FAILED(hr)) return hr;
961 if ((ret = MsiRecordSetInteger(This->msiHandle, V_I4(&varg0), V_I4(&varg1))) != ERROR_SUCCESS)
962 {
963 ERR("MsiRecordSetInteger returned %d\n", ret);
964 return DISP_E_EXCEPTION;
965 }
966 }
967 else return DISP_E_MEMBERNOTFOUND;
968 break;
969
970 default:
971 return DISP_E_MEMBERNOTFOUND;
972 }
973
974 VariantClear(&varg1);
975 VariantClear(&varg0);
976
977 return S_OK;
978 }
979
980 static HRESULT WINAPI ListImpl_Invoke(
981 AutomationObject* This,
982 DISPID dispIdMember,
983 REFIID riid,
984 LCID lcid,
985 WORD wFlags,
986 DISPPARAMS* pDispParams,
987 VARIANT* pVarResult,
988 EXCEPINFO* pExcepInfo,
989 UINT* puArgErr)
990 {
991 ListData *data = (ListData *)private_data(This);
992 HRESULT hr;
993 VARIANTARG varg0;
994 IUnknown *pUnk = NULL;
995
996 VariantInit(&varg0);
997
998 switch (dispIdMember)
999 {
1000 case DISPID_LIST__NEWENUM:
1001 if (wFlags & DISPATCH_METHOD) {
1002 V_VT(pVarResult) = VT_UNKNOWN;
1003 if (SUCCEEDED(hr = create_list_enumerator(NULL, (LPVOID *)&pUnk, This, 0)))
1004 V_UNKNOWN(pVarResult) = pUnk;
1005 else
1006 ERR("Failed to create IEnumVARIANT object, hresult 0x%08x\n", hr);
1007 }
1008 else return DISP_E_MEMBERNOTFOUND;
1009 break;
1010
1011 case DISPID_LIST_ITEM:
1012 if (wFlags & DISPATCH_PROPERTYGET) {
1013 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
1014 if (FAILED(hr)) return hr;
1015 if (V_I4(&varg0) < 0 || V_I4(&varg0) >= data->ulCount)
1016 return DISP_E_BADINDEX;
1017 VariantCopy(pVarResult, &data->pVars[V_I4(&varg0)]);
1018 }
1019 else return DISP_E_MEMBERNOTFOUND;
1020 break;
1021
1022 case DISPID_LIST_COUNT:
1023 if (wFlags & DISPATCH_PROPERTYGET) {
1024 V_VT(pVarResult) = VT_I4;
1025 V_I4(pVarResult) = data->ulCount;
1026 }
1027 else return DISP_E_MEMBERNOTFOUND;
1028 break;
1029
1030 default:
1031 return DISP_E_MEMBERNOTFOUND;
1032 }
1033
1034 VariantClear(&varg0);
1035
1036 return S_OK;
1037 }
1038
1039 static void WINAPI ListImpl_Free(AutomationObject *This)
1040 {
1041 ListData *data = private_data(This);
1042 ULONG idx;
1043
1044 for (idx=0; idx<data->ulCount; idx++)
1045 VariantClear(&data->pVars[idx]);
1046 HeapFree(GetProcessHeap(), 0, data->pVars);
1047 }
1048
1049 static HRESULT WINAPI ViewImpl_Invoke(
1050 AutomationObject* This,
1051 DISPID dispIdMember,
1052 REFIID riid,
1053 LCID lcid,
1054 WORD wFlags,
1055 DISPPARAMS* pDispParams,
1056 VARIANT* pVarResult,
1057 EXCEPINFO* pExcepInfo,
1058 UINT* puArgErr)
1059 {
1060 MSIHANDLE msiHandle;
1061 IDispatch *pDispatch = NULL;
1062 UINT ret;
1063 VARIANTARG varg0, varg1;
1064 HRESULT hr;
1065
1066 VariantInit(&varg0);
1067 VariantInit(&varg1);
1068
1069 switch (dispIdMember)
1070 {
1071 case DISPID_VIEW_EXECUTE:
1072 if (wFlags & DISPATCH_METHOD)
1073 {
1074 hr = DispGetParam(pDispParams, 0, VT_DISPATCH, &varg0, puArgErr);
1075 if (SUCCEEDED(hr) && V_DISPATCH(&varg0) != NULL)
1076 MsiViewExecute(This->msiHandle, ((AutomationObject *)V_DISPATCH(&varg0))->msiHandle);
1077 else
1078 MsiViewExecute(This->msiHandle, 0);
1079 }
1080 else return DISP_E_MEMBERNOTFOUND;
1081 break;
1082
1083 case DISPID_VIEW_FETCH:
1084 if (wFlags & DISPATCH_METHOD)
1085 {
1086 V_VT(pVarResult) = VT_DISPATCH;
1087 if ((ret = MsiViewFetch(This->msiHandle, &msiHandle)) == ERROR_SUCCESS)
1088 {
1089 if (SUCCEEDED(hr = create_automation_object(msiHandle, NULL, (LPVOID*)&pDispatch, &DIID_Record, RecordImpl_Invoke, NULL, 0)))
1090 V_DISPATCH(pVarResult) = pDispatch;
1091 else
1092 ERR("Failed to create Record object, hresult 0x%08x\n", hr);
1093 }
1094 else if (ret == ERROR_NO_MORE_ITEMS)
1095 V_DISPATCH(pVarResult) = NULL;
1096 else
1097 {
1098 ERR("MsiViewFetch returned %d\n", ret);
1099 return DISP_E_EXCEPTION;
1100 }
1101 }
1102 else return DISP_E_MEMBERNOTFOUND;
1103 break;
1104
1105 case DISPID_VIEW_MODIFY:
1106 if (wFlags & DISPATCH_METHOD)
1107 {
1108 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
1109 if (FAILED(hr)) return hr;
1110 hr = DispGetParam(pDispParams, 1, VT_DISPATCH, &varg1, puArgErr);
1111 if (FAILED(hr)) return hr;
1112 if (!V_DISPATCH(&varg1)) return DISP_E_EXCEPTION;
1113 if ((ret = MsiViewModify(This->msiHandle, V_I4(&varg0), ((AutomationObject *)V_DISPATCH(&varg1))->msiHandle)) != ERROR_SUCCESS)
1114 {
1115 VariantClear(&varg1);
1116 ERR("MsiViewModify returned %d\n", ret);
1117 return DISP_E_EXCEPTION;
1118 }
1119 }
1120 else return DISP_E_MEMBERNOTFOUND;
1121 break;
1122
1123 case DISPID_VIEW_CLOSE:
1124 if (wFlags & DISPATCH_METHOD)
1125 {
1126 MsiViewClose(This->msiHandle);
1127 }
1128 else return DISP_E_MEMBERNOTFOUND;
1129 break;
1130
1131 default:
1132 return DISP_E_MEMBERNOTFOUND;
1133 }
1134
1135 VariantClear(&varg1);
1136 VariantClear(&varg0);
1137
1138 return S_OK;
1139 }
1140
1141 static HRESULT WINAPI DatabaseImpl_Invoke(
1142 AutomationObject* This,
1143 DISPID dispIdMember,
1144 REFIID riid,
1145 LCID lcid,
1146 WORD wFlags,
1147 DISPPARAMS* pDispParams,
1148 VARIANT* pVarResult,
1149 EXCEPINFO* pExcepInfo,
1150 UINT* puArgErr)
1151 {
1152 MSIHANDLE msiHandle;
1153 IDispatch *pDispatch = NULL;
1154 UINT ret;
1155 VARIANTARG varg0, varg1;
1156 HRESULT hr;
1157
1158 VariantInit(&varg0);
1159 VariantInit(&varg1);
1160
1161 switch (dispIdMember)
1162 {
1163 case DISPID_DATABASE_SUMMARYINFORMATION:
1164 if (wFlags & DISPATCH_PROPERTYGET)
1165 {
1166 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
1167 if (FAILED(hr))
1168 V_I4(&varg0) = 0;
1169
1170 V_VT(pVarResult) = VT_DISPATCH;
1171 if ((ret = MsiGetSummaryInformationW(This->msiHandle, NULL, V_I4(&varg0), &msiHandle)) == ERROR_SUCCESS)
1172 {
1173 hr = create_automation_object(msiHandle, NULL, (LPVOID *)&pDispatch, &DIID_SummaryInfo, SummaryInfoImpl_Invoke, NULL, 0);
1174 if (SUCCEEDED(hr))
1175 V_DISPATCH(pVarResult) = pDispatch;
1176 else
1177 ERR("Failed to create SummaryInfo object: 0x%08x\n", hr);
1178 }
1179 else
1180 {
1181 ERR("MsiGetSummaryInformation returned %d\n", ret);
1182 return DISP_E_EXCEPTION;
1183 }
1184 }
1185 else return DISP_E_MEMBERNOTFOUND;
1186 break;
1187
1188 case DISPID_DATABASE_OPENVIEW:
1189 if (wFlags & DISPATCH_METHOD)
1190 {
1191 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1192 if (FAILED(hr)) return hr;
1193 V_VT(pVarResult) = VT_DISPATCH;
1194 if ((ret = MsiDatabaseOpenViewW(This->msiHandle, V_BSTR(&varg0), &msiHandle)) == ERROR_SUCCESS)
1195 {
1196 if (SUCCEEDED(hr = create_automation_object(msiHandle, NULL, (LPVOID*)&pDispatch, &DIID_View, ViewImpl_Invoke, NULL, 0)))
1197 V_DISPATCH(pVarResult) = pDispatch;
1198 else
1199 ERR("Failed to create View object, hresult 0x%08x\n", hr);
1200 }
1201 else
1202 {
1203 VariantClear(&varg0);
1204 ERR("MsiDatabaseOpenView returned %d\n", ret);
1205 return DISP_E_EXCEPTION;
1206 }
1207 }
1208 else return DISP_E_MEMBERNOTFOUND;
1209 break;
1210
1211 default:
1212 return DISP_E_MEMBERNOTFOUND;
1213 }
1214
1215 VariantClear(&varg1);
1216 VariantClear(&varg0);
1217
1218 return S_OK;
1219 }
1220
1221 static HRESULT WINAPI SessionImpl_Invoke(
1222 AutomationObject* This,
1223 DISPID dispIdMember,
1224 REFIID riid,
1225 LCID lcid,
1226 WORD wFlags,
1227 DISPPARAMS* pDispParams,
1228 VARIANT* pVarResult,
1229 EXCEPINFO* pExcepInfo,
1230 UINT* puArgErr)
1231 {
1232 SessionData *data = private_data(This);
1233 WCHAR *szString;
1234 DWORD dwLen;
1235 IDispatch *pDispatch = NULL;
1236 MSIHANDLE msiHandle;
1237 LANGID langId;
1238 UINT ret;
1239 INSTALLSTATE iInstalled, iAction;
1240 VARIANTARG varg0, varg1;
1241 HRESULT hr;
1242
1243 VariantInit(&varg0);
1244 VariantInit(&varg1);
1245
1246 switch (dispIdMember)
1247 {
1248 case DISPID_SESSION_INSTALLER:
1249 if (wFlags & DISPATCH_PROPERTYGET) {
1250 V_VT(pVarResult) = VT_DISPATCH;
1251 IDispatch_AddRef(data->pInstaller);
1252 V_DISPATCH(pVarResult) = data->pInstaller;
1253 }
1254 else return DISP_E_MEMBERNOTFOUND;
1255 break;
1256
1257 case DISPID_SESSION_PROPERTY:
1258 if (wFlags & DISPATCH_PROPERTYGET) {
1259 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1260 if (FAILED(hr)) return hr;
1261 V_VT(pVarResult) = VT_BSTR;
1262 V_BSTR(pVarResult) = NULL;
1263 if ((ret = MsiGetPropertyW(This->msiHandle, V_BSTR(&varg0), NULL, &dwLen)) == ERROR_SUCCESS)
1264 {
1265 if (!(szString = msi_alloc((++dwLen)*sizeof(WCHAR))))
1266 ERR("Out of memory\n");
1267 else if ((ret = MsiGetPropertyW(This->msiHandle, V_BSTR(&varg0), szString, &dwLen)) == ERROR_SUCCESS)
1268 V_BSTR(pVarResult) = SysAllocString(szString);
1269 msi_free(szString);
1270 }
1271 if (ret != ERROR_SUCCESS)
1272 ERR("MsiGetProperty returned %d\n", ret);
1273 } else if (wFlags & DISPATCH_PROPERTYPUT) {
1274 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1275 if (FAILED(hr)) return hr;
1276 hr = DispGetParam(pDispParams, DISPID_PROPERTYPUT, VT_BSTR, &varg1, puArgErr);
1277 if (FAILED(hr)) {
1278 VariantClear(&varg0);
1279 return hr;
1280 }
1281 if ((ret = MsiSetPropertyW(This->msiHandle, V_BSTR(&varg0), V_BSTR(&varg1))) != ERROR_SUCCESS)
1282 {
1283 VariantClear(&varg0);
1284 VariantClear(&varg1);
1285 ERR("MsiSetProperty returned %d\n", ret);
1286 return DISP_E_EXCEPTION;
1287 }
1288 }
1289 else return DISP_E_MEMBERNOTFOUND;
1290 break;
1291
1292 case DISPID_SESSION_LANGUAGE:
1293 if (wFlags & DISPATCH_PROPERTYGET) {
1294 langId = MsiGetLanguage(This->msiHandle);
1295 V_VT(pVarResult) = VT_I4;
1296 V_I4(pVarResult) = langId;
1297 }
1298 else return DISP_E_MEMBERNOTFOUND;
1299 break;
1300
1301 case DISPID_SESSION_MODE:
1302 if (wFlags & DISPATCH_PROPERTYGET) {
1303 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
1304 if (FAILED(hr)) return hr;
1305 V_VT(pVarResult) = VT_BOOL;
1306 V_BOOL(pVarResult) = MsiGetMode(This->msiHandle, V_I4(&varg0));
1307 } else if (wFlags & DISPATCH_PROPERTYPUT) {
1308 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
1309 if (FAILED(hr)) return hr;
1310 hr = DispGetParam(pDispParams, DISPID_PROPERTYPUT, VT_BOOL, &varg1, puArgErr);
1311 if (FAILED(hr)) return hr;
1312 if ((ret = MsiSetMode(This->msiHandle, V_I4(&varg0), V_BOOL(&varg1))) != ERROR_SUCCESS)
1313 {
1314 ERR("MsiSetMode returned %d\n", ret);
1315 return DISP_E_EXCEPTION;
1316 }
1317 }
1318 else return DISP_E_MEMBERNOTFOUND;
1319 break;
1320
1321 case DISPID_SESSION_DATABASE:
1322 if (wFlags & DISPATCH_PROPERTYGET) {
1323 V_VT(pVarResult) = VT_DISPATCH;
1324 if ((msiHandle = MsiGetActiveDatabase(This->msiHandle)))
1325 {
1326 if (SUCCEEDED(hr = create_automation_object(msiHandle, NULL, (LPVOID*)&pDispatch, &DIID_Database, DatabaseImpl_Invoke, NULL, 0)))
1327 V_DISPATCH(pVarResult) = pDispatch;
1328 else
1329 ERR("Failed to create Database object, hresult 0x%08x\n", hr);
1330 }
1331 else
1332 {
1333 ERR("MsiGetActiveDatabase failed\n");
1334 return DISP_E_EXCEPTION;
1335 }
1336 }
1337 else return DISP_E_MEMBERNOTFOUND;
1338 break;
1339
1340 case DISPID_SESSION_DOACTION:
1341 if (wFlags & DISPATCH_METHOD) {
1342 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1343 if (FAILED(hr)) return hr;
1344 ret = MsiDoActionW(This->msiHandle, V_BSTR(&varg0));
1345 V_VT(pVarResult) = VT_I4;
1346 switch (ret)
1347 {
1348 case ERROR_FUNCTION_NOT_CALLED:
1349 V_I4(pVarResult) = msiDoActionStatusNoAction;
1350 break;
1351 case ERROR_SUCCESS:
1352 V_I4(pVarResult) = msiDoActionStatusSuccess;
1353 break;
1354 case ERROR_INSTALL_USEREXIT:
1355 V_I4(pVarResult) = msiDoActionStatusUserExit;
1356 break;
1357 case ERROR_INSTALL_FAILURE:
1358 V_I4(pVarResult) = msiDoActionStatusFailure;
1359 break;
1360 case ERROR_INSTALL_SUSPEND:
1361 V_I4(pVarResult) = msiDoActionStatusSuspend;
1362 break;
1363 case ERROR_MORE_DATA:
1364 V_I4(pVarResult) = msiDoActionStatusFinished;
1365 break;
1366 case ERROR_INVALID_HANDLE_STATE:
1367 V_I4(pVarResult) = msiDoActionStatusWrongState;
1368 break;
1369 case ERROR_INVALID_DATA:
1370 V_I4(pVarResult) = msiDoActionStatusBadActionData;
1371 break;
1372 default:
1373 VariantClear(&varg0);
1374 FIXME("MsiDoAction returned unhandled value %d\n", ret);
1375 return DISP_E_EXCEPTION;
1376 }
1377 }
1378 else return DISP_E_MEMBERNOTFOUND;
1379 break;
1380
1381 case DISPID_SESSION_EVALUATECONDITION:
1382 if (wFlags & DISPATCH_METHOD) {
1383 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1384 if (FAILED(hr)) return hr;
1385 V_VT(pVarResult) = VT_I4;
1386 V_I4(pVarResult) = MsiEvaluateConditionW(This->msiHandle, V_BSTR(&varg0));
1387 }
1388 else return DISP_E_MEMBERNOTFOUND;
1389 break;
1390
1391 case DISPID_SESSION_SETINSTALLLEVEL:
1392 if (wFlags & DISPATCH_METHOD) {
1393 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
1394 if (FAILED(hr)) return hr;
1395 if ((ret = MsiSetInstallLevel(This->msiHandle, V_I4(&varg0))) != ERROR_SUCCESS)
1396 {
1397 ERR("MsiSetInstallLevel returned %d\n", ret);
1398 return DISP_E_EXCEPTION;
1399 }
1400 }
1401 else return DISP_E_MEMBERNOTFOUND;
1402 break;
1403
1404 case DISPID_SESSION_FEATURECURRENTSTATE:
1405 if (wFlags & DISPATCH_PROPERTYGET) {
1406 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1407 if (FAILED(hr)) return hr;
1408 V_VT(pVarResult) = VT_I4;
1409 if ((ret = MsiGetFeatureStateW(This->msiHandle, V_BSTR(&varg0), &iInstalled, &iAction)) == ERROR_SUCCESS)
1410 V_I4(pVarResult) = iInstalled;
1411 else
1412 {
1413 ERR("MsiGetFeatureState returned %d\n", ret);
1414 V_I4(pVarResult) = msiInstallStateUnknown;
1415 }
1416 }
1417 else return DISP_E_MEMBERNOTFOUND;
1418 break;
1419
1420 case DISPID_SESSION_FEATUREREQUESTSTATE:
1421 if (wFlags & DISPATCH_PROPERTYGET) {
1422 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1423 if (FAILED(hr)) return hr;
1424 V_VT(pVarResult) = VT_I4;
1425 if ((ret = MsiGetFeatureStateW(This->msiHandle, V_BSTR(&varg0), &iInstalled, &iAction)) == ERROR_SUCCESS)
1426 V_I4(pVarResult) = iAction;
1427 else
1428 {
1429 ERR("MsiGetFeatureState returned %d\n", ret);
1430 V_I4(pVarResult) = msiInstallStateUnknown;
1431 }
1432 } else if (wFlags & DISPATCH_PROPERTYPUT) {
1433 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1434 if (FAILED(hr)) return hr;
1435 hr = DispGetParam(pDispParams, DISPID_PROPERTYPUT, VT_I4, &varg1, puArgErr);
1436 if (FAILED(hr)) {
1437 VariantClear(&varg0);
1438 return hr;
1439 }
1440 if ((ret = MsiSetFeatureStateW(This->msiHandle, V_BSTR(&varg0), V_I4(&varg1))) != ERROR_SUCCESS)
1441 {
1442 VariantClear(&varg0);
1443 ERR("MsiSetFeatureState returned %d\n", ret);
1444 return DISP_E_EXCEPTION;
1445 }
1446 }
1447 else return DISP_E_MEMBERNOTFOUND;
1448 break;
1449
1450 default:
1451 return DISP_E_MEMBERNOTFOUND;
1452 }
1453
1454 VariantClear(&varg1);
1455 VariantClear(&varg0);
1456
1457 return S_OK;
1458 }
1459
1460 /* Fill the variant pointed to by pVarResult with the value & size returned by RegQueryValueEx as dictated by the
1461 * registry value type. Used by Installer::RegistryValue. */
1462 static void variant_from_registry_value(VARIANT *pVarResult, DWORD dwType, LPBYTE lpData, DWORD dwSize)
1463 {
1464 static const WCHAR szREG_BINARY[] = { '(','R','E','G','_','B','I','N','A','R','Y',')',0 };
1465 static const WCHAR szREG_[] = { '(','R','E','G','_',']',0 };
1466 WCHAR *szString = (WCHAR *)lpData;
1467 LPWSTR szNewString = NULL;
1468 DWORD dwNewSize = 0;
1469 int idx;
1470
1471 switch (dwType)
1472 {
1473 /* Registry strings may not be null terminated so we must use SysAllocStringByteLen/Len */
1474 case REG_MULTI_SZ: /* Multi SZ change internal null characters to newlines */
1475 idx = (dwSize/sizeof(WCHAR))-1;
1476 while (idx >= 0 && !szString[idx]) idx--;
1477 for (; idx >= 0; idx--)
1478 if (!szString[idx]) szString[idx] = '\n';
1479 case REG_SZ:
1480 V_VT(pVarResult) = VT_BSTR;
1481 V_BSTR(pVarResult) = SysAllocStringByteLen((LPCSTR)szString, dwSize);
1482 break;
1483
1484 case REG_EXPAND_SZ:
1485 if (!(dwNewSize = ExpandEnvironmentStringsW(szString, szNewString, dwNewSize)))
1486 ERR("ExpandEnvironmentStrings returned error %d\n", GetLastError());
1487 else if (!(szNewString = msi_alloc(dwNewSize)))
1488 ERR("Out of memory\n");
1489 else if (!(dwNewSize = ExpandEnvironmentStringsW(szString, szNewString, dwNewSize)))
1490 ERR("ExpandEnvironmentStrings returned error %d\n", GetLastError());
1491 else
1492 {
1493 V_VT(pVarResult) = VT_BSTR;
1494 V_BSTR(pVarResult) = SysAllocStringLen(szNewString, dwNewSize);
1495 }
1496 msi_free(szNewString);
1497 break;
1498
1499 case REG_DWORD:
1500 V_VT(pVarResult) = VT_I4;
1501 V_I4(pVarResult) = *((DWORD *)lpData);
1502 break;
1503
1504 case REG_QWORD:
1505 V_VT(pVarResult) = VT_BSTR;
1506 V_BSTR(pVarResult) = SysAllocString(szREG_); /* Weird string, don't know why native returns it */
1507 break;
1508
1509 case REG_BINARY:
1510 V_VT(pVarResult) = VT_BSTR;
1511 V_BSTR(pVarResult) = SysAllocString(szREG_BINARY);
1512 break;
1513
1514 case REG_NONE:
1515 V_VT(pVarResult) = VT_EMPTY;
1516 break;
1517
1518 default:
1519 FIXME("Unhandled registry value type %d\n", dwType);
1520 }
1521 }
1522
1523 static HRESULT WINAPI InstallerImpl_Invoke(
1524 AutomationObject* This,
1525 DISPID dispIdMember,
1526 REFIID riid,
1527 LCID lcid,
1528 WORD wFlags,
1529 DISPPARAMS* pDispParams,
1530 VARIANT* pVarResult,
1531 EXCEPINFO* pExcepInfo,
1532 UINT* puArgErr)
1533 {
1534 MSIHANDLE msiHandle;
1535 IDispatch *pDispatch = NULL;
1536 UINT ret;
1537 VARIANTARG varg0, varg1, varg2;
1538 HRESULT hr;
1539 LPWSTR szString = NULL;
1540 DWORD dwSize = 0;
1541 INSTALLUILEVEL ui;
1542
1543 VariantInit(&varg0);
1544 VariantInit(&varg1);
1545 VariantInit(&varg2);
1546
1547 switch (dispIdMember)
1548 {
1549 case DISPID_INSTALLER_CREATERECORD:
1550 if (wFlags & DISPATCH_METHOD)
1551 {
1552 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
1553 if (FAILED(hr)) return hr;
1554 V_VT(pVarResult) = VT_DISPATCH;
1555 if ((msiHandle = MsiCreateRecord(V_I4(&varg0))))
1556 {
1557 if (SUCCEEDED(hr = create_automation_object(msiHandle, NULL, (LPVOID*)&pDispatch, &DIID_Record, RecordImpl_Invoke, NULL, 0)))
1558 V_DISPATCH(pVarResult) = pDispatch;
1559 else
1560 ERR("Failed to create Record object, hresult 0x%08x\n", hr);
1561 }
1562 else
1563 {
1564 ERR("MsiCreateRecord failed\n");
1565 return DISP_E_EXCEPTION;
1566 }
1567 }
1568 else return DISP_E_MEMBERNOTFOUND;
1569 break;
1570
1571 case DISPID_INSTALLER_OPENPACKAGE:
1572 if (wFlags & DISPATCH_METHOD)
1573 {
1574 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1575 if (FAILED(hr)) return hr;
1576 hr = DispGetParam(pDispParams, 1, VT_I4, &varg1, puArgErr);
1577 if (FAILED(hr))
1578 {
1579 VariantClear(&varg0);
1580 return hr;
1581 }
1582 V_VT(pVarResult) = VT_DISPATCH;
1583 if ((ret = MsiOpenPackageExW(V_BSTR(&varg0), V_I4(&varg1), &msiHandle)) == ERROR_SUCCESS)
1584 {
1585 if (SUCCEEDED(hr = create_session(msiHandle, (IDispatch *)This, &pDispatch)))
1586 V_DISPATCH(pVarResult) = pDispatch;
1587 else
1588 ERR("Failed to create Session object, hresult 0x%08x\n", hr);
1589 }
1590 else
1591 {
1592 VariantClear(&varg0);
1593 ERR("MsiOpenPackageEx returned %d\n", ret);
1594 return DISP_E_EXCEPTION;
1595 }
1596 }
1597 else return DISP_E_MEMBERNOTFOUND;
1598 break;
1599
1600 case DISPID_INSTALLER_OPENDATABASE:
1601 if (wFlags & DISPATCH_METHOD)
1602 {
1603 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1604 if (FAILED(hr)) return hr;
1605
1606 hr = DispGetParam(pDispParams, 1, VT_BSTR, &varg1, puArgErr);
1607 if (FAILED(hr))
1608 {
1609 VariantClear(&varg0);
1610 return hr;
1611 }
1612
1613 V_VT(pVarResult) = VT_DISPATCH;
1614 if ((ret = MsiOpenDatabaseW(V_BSTR(&varg0), V_BSTR(&varg1), &msiHandle)) == ERROR_SUCCESS)
1615 {
1616 hr = create_automation_object(msiHandle, NULL, (LPVOID *)&pDispatch,
1617 &DIID_Database, DatabaseImpl_Invoke, NULL, 0);
1618 if (SUCCEEDED(hr))
1619 V_DISPATCH(pVarResult) = pDispatch;
1620 else
1621 ERR("Failed to create Database object: 0x%08x\n", hr);
1622 }
1623 else
1624 {
1625 VariantClear(&varg0);
1626 VariantClear(&varg1);
1627 ERR("MsiOpenDatabase returned %d\n", ret);
1628 return DISP_E_EXCEPTION;
1629 }
1630 }
1631 else return DISP_E_MEMBERNOTFOUND;
1632 break;
1633
1634 case DISPID_INSTALLER_UILEVEL:
1635 if (wFlags & DISPATCH_PROPERTYPUT)
1636 {
1637 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
1638 if (FAILED(hr)) return hr;
1639 if ((ui = MsiSetInternalUI(V_I4(&varg0), NULL) == INSTALLUILEVEL_NOCHANGE))
1640 {
1641 ERR("MsiSetInternalUI failed\n");
1642 return DISP_E_EXCEPTION;
1643 }
1644 }
1645 else if (wFlags & DISPATCH_PROPERTYGET)
1646 {
1647 if ((ui = MsiSetInternalUI(INSTALLUILEVEL_NOCHANGE, NULL) == INSTALLUILEVEL_NOCHANGE))
1648 {
1649 ERR("MsiSetInternalUI failed\n");
1650 return DISP_E_EXCEPTION;
1651 }
1652
1653 V_VT(pVarResult) = VT_I4;
1654 V_I4(pVarResult) = ui;
1655 }
1656 else return DISP_E_MEMBERNOTFOUND;
1657 break;
1658
1659 case DISPID_INSTALLER_INSTALLPRODUCT:
1660 if (wFlags & DISPATCH_METHOD)
1661 {
1662 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1663 if (FAILED(hr)) return hr;
1664 hr = DispGetParam(pDispParams, 1, VT_BSTR, &varg1, puArgErr);
1665 if (FAILED(hr))
1666 {
1667 VariantClear(&varg0);
1668 return hr;
1669 }
1670 if ((ret = MsiInstallProductW(V_BSTR(&varg0), V_BSTR(&varg1))) != ERROR_SUCCESS)
1671 {
1672 VariantClear(&varg1);
1673 VariantClear(&varg0);
1674 ERR("MsiInstallProduct returned %d\n", ret);
1675 return DISP_E_EXCEPTION;
1676 }
1677 }
1678 else return DISP_E_MEMBERNOTFOUND;
1679 break;
1680
1681 case DISPID_INSTALLER_VERSION:
1682 if (wFlags & DISPATCH_PROPERTYGET) {
1683 DLLVERSIONINFO verinfo;
1684 WCHAR version[MAX_PATH];
1685
1686 static const WCHAR format[] = {'%','d','.','%','d','.','%','d','.','%','d',0};
1687
1688 verinfo.cbSize = sizeof(DLLVERSIONINFO);
1689 hr = DllGetVersion(&verinfo);
1690 if (FAILED(hr)) return hr;
1691
1692 sprintfW(version, format, verinfo.dwMajorVersion, verinfo.dwMinorVersion,
1693 verinfo.dwBuildNumber, verinfo.dwPlatformID);
1694
1695 V_VT(pVarResult) = VT_BSTR;
1696 V_BSTR(pVarResult) = SysAllocString(version);
1697 }
1698 else return DISP_E_MEMBERNOTFOUND;
1699 break;
1700
1701 case DISPID_INSTALLER_REGISTRYVALUE:
1702 if (wFlags & DISPATCH_METHOD) {
1703 HKEY hkey;
1704 DWORD dwType;
1705 UINT posValue = 2; /* Save valuePos so we can save puArgErr if we are unable to do our type conversions */
1706
1707 hr = DispGetParam(pDispParams, 0, VT_I4, &varg0, puArgErr);
1708 if (FAILED(hr)) return hr;
1709 hr = DispGetParam(pDispParams, 1, VT_BSTR, &varg1, puArgErr);
1710 if (FAILED(hr)) return hr;
1711 hr = DispGetParam_CopyOnly(pDispParams, &posValue, &varg2);
1712 if (FAILED(hr))
1713 {
1714 VariantClear(&varg1);
1715 return hr;
1716 }
1717
1718 if (V_I4(&varg0) >= REG_INDEX_CLASSES_ROOT &&
1719 V_I4(&varg0) <= REG_INDEX_DYN_DATA)
1720 V_I4(&varg0) |= (UINT)HKEY_CLASSES_ROOT;
1721
1722 ret = RegOpenKeyW((HKEY)V_I4(&varg0), V_BSTR(&varg1), &hkey);
1723
1724 /* Third parameter can be VT_EMPTY, VT_I4, or VT_BSTR */
1725 switch (V_VT(&varg2))
1726 {
1727 case VT_EMPTY: /* Return VT_BOOL as to whether or not registry key exists */
1728 V_VT(pVarResult) = VT_BOOL;
1729 V_BOOL(pVarResult) = (ret == ERROR_SUCCESS);
1730 break;
1731
1732 case VT_BSTR: /* Return value of specified key if it exists */
1733 if (ret == ERROR_SUCCESS &&
1734 (ret = RegQueryValueExW(hkey, V_BSTR(&varg2), NULL, NULL, NULL, &dwSize)) == ERROR_SUCCESS)
1735 {
1736 if (!(szString = msi_alloc(dwSize)))
1737 ERR("Out of memory\n");
1738 else if ((ret = RegQueryValueExW(hkey, V_BSTR(&varg2), NULL, &dwType, (LPBYTE)szString, &dwSize)) == ERROR_SUCCESS)
1739 variant_from_registry_value(pVarResult, dwType, (LPBYTE)szString, dwSize);
1740 }
1741
1742 if (ret != ERROR_SUCCESS)
1743 {
1744 msi_free(szString);
1745 VariantClear(&varg2);
1746 VariantClear(&varg1);
1747 return DISP_E_BADINDEX;
1748 }
1749 break;
1750
1751 default: /* Try to make it into VT_I4, can use VariantChangeType for this */
1752 hr = VariantChangeType(&varg2, &varg2, 0, VT_I4);
1753 if (SUCCEEDED(hr) && ret != ERROR_SUCCESS) hr = DISP_E_BADINDEX; /* Conversion fine, but couldn't find key */
1754 if (FAILED(hr))
1755 {
1756 if (hr == DISP_E_TYPEMISMATCH) *puArgErr = posValue;
1757 VariantClear(&varg2); /* Unknown type, so let's clear it */
1758 VariantClear(&varg1);
1759 return hr;
1760 }
1761
1762 /* Retrieve class name or maximum value name or subkey name size */
1763 if (!V_I4(&varg2))
1764 ret = RegQueryInfoKeyW(hkey, NULL, &dwSize, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
1765 else if (V_I4(&varg2) > 0)
1766 ret = RegQueryInfoKeyW(hkey, NULL, NULL, NULL, NULL, NULL, NULL, NULL, &dwSize, NULL, NULL, NULL);
1767 else /* V_I4(&varg2) < 0 */
1768 ret = RegQueryInfoKeyW(hkey, NULL, NULL, NULL, NULL, &dwSize, NULL, NULL, NULL, NULL, NULL, NULL);
1769
1770 if (ret == ERROR_SUCCESS)
1771 {
1772 if (!(szString = msi_alloc(++dwSize * sizeof(WCHAR))))
1773 ERR("Out of memory\n");
1774 else if (!V_I4(&varg2))
1775 ret = RegQueryInfoKeyW(hkey, szString, &dwSize, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
1776 else if (V_I4(&varg2) > 0)
1777 ret = RegEnumValueW(hkey, V_I4(&varg2)-1, szString, &dwSize, 0, 0, NULL, NULL);
1778 else /* V_I4(&varg2) < 0 */
1779 ret = RegEnumKeyW(hkey, -1 - V_I4(&varg2), szString, dwSize);
1780
1781 if (szString && ret == ERROR_SUCCESS)
1782 {
1783 V_VT(pVarResult) = VT_BSTR;
1784 V_BSTR(pVarResult) = SysAllocString(szString);
1785 }
1786 }
1787 }
1788
1789 msi_free(szString);
1790 RegCloseKey(hkey);
1791 }
1792 else return DISP_E_MEMBERNOTFOUND;
1793 break;
1794
1795 case DISPID_INSTALLER_PRODUCTSTATE:
1796 if (wFlags & DISPATCH_PROPERTYGET) {
1797 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1798 if (FAILED(hr)) return hr;
1799 V_VT(pVarResult) = VT_I4;
1800 V_I4(pVarResult) = MsiQueryProductStateW(V_BSTR(&varg0));
1801 }
1802 else return DISP_E_MEMBERNOTFOUND;
1803 break;
1804
1805 case DISPID_INSTALLER_PRODUCTINFO:
1806 if (wFlags & DISPATCH_PROPERTYGET) {
1807 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1808 if (FAILED(hr)) return hr;
1809 hr = DispGetParam(pDispParams, 1, VT_BSTR, &varg1, puArgErr);
1810 if (FAILED(hr))
1811 {
1812 VariantClear(&varg0);
1813 return hr;
1814 }
1815 V_VT(pVarResult) = VT_BSTR;
1816 V_BSTR(pVarResult) = NULL;
1817 if ((ret = MsiGetProductInfoW(V_BSTR(&varg0), V_BSTR(&varg1), NULL, &dwSize)) == ERROR_SUCCESS)
1818 {
1819 if (!(szString = msi_alloc((++dwSize)*sizeof(WCHAR))))
1820 ERR("Out of memory\n");
1821 else if ((ret = MsiGetProductInfoW(V_BSTR(&varg0), V_BSTR(&varg1), szString, &dwSize)) == ERROR_SUCCESS)
1822 V_BSTR(pVarResult) = SysAllocString(szString);
1823 msi_free(szString);
1824 }
1825 if (ret != ERROR_SUCCESS)
1826 {
1827 ERR("MsiGetProductInfo returned %d\n", ret);
1828 VariantClear(&varg1);
1829 VariantClear(&varg0);
1830 return DISP_E_EXCEPTION;
1831 }
1832 }
1833 else return DISP_E_MEMBERNOTFOUND;
1834 break;
1835
1836 case DISPID_INSTALLER_PRODUCTS:
1837 if (wFlags & DISPATCH_PROPERTYGET)
1838 {
1839 ListData *ldata = NULL;
1840 ULONG idx = 0;
1841 WCHAR szProductBuf[GUID_SIZE];
1842
1843 /* Find number of products */
1844 while ((ret = MsiEnumProductsW(idx, szProductBuf)) == ERROR_SUCCESS) idx++;
1845 if (ret != ERROR_NO_MORE_ITEMS)
1846 {
1847 ERR("MsiEnumProducts returned %d\n", ret);
1848 return DISP_E_EXCEPTION;
1849 }
1850
1851 V_VT(pVarResult) = VT_DISPATCH;
1852 if (SUCCEEDED(hr = create_automation_object(0, NULL, (LPVOID*)&pDispatch, &DIID_StringList, ListImpl_Invoke, ListImpl_Free, sizeof(ListData))))
1853 {
1854 V_DISPATCH(pVarResult) = pDispatch;
1855
1856 /* Save product strings */
1857 ldata = (ListData *)private_data((AutomationObject *)pDispatch);
1858 if (!(ldata->pVars = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(VARIANT)*idx)))
1859 ERR("Out of memory\n");
1860 else
1861 {
1862 ldata->ulCount = idx;
1863 for (idx = 0; idx < ldata->ulCount; idx++)
1864 {
1865 ret = MsiEnumProductsW(idx, szProductBuf);
1866 VariantInit(&ldata->pVars[idx]);
1867 V_VT(&ldata->pVars[idx]) = VT_BSTR;
1868 V_BSTR(&ldata->pVars[idx]) = SysAllocString(szProductBuf);
1869 }
1870 }
1871 }
1872 else
1873 ERR("Failed to create StringList object, hresult 0x%08x\n", hr);
1874 }
1875 else return DISP_E_MEMBERNOTFOUND;
1876 break;
1877
1878 case DISPID_INSTALLER_RELATEDPRODUCTS:
1879 if (wFlags & DISPATCH_PROPERTYGET)
1880 {
1881 ListData *ldata = NULL;
1882 ULONG idx = 0;
1883 WCHAR szProductBuf[GUID_SIZE];
1884
1885 hr = DispGetParam(pDispParams, 0, VT_BSTR, &varg0, puArgErr);
1886 if (FAILED(hr)) return hr;
1887
1888 /* Find number of related products */
1889 while ((ret = MsiEnumRelatedProductsW(V_BSTR(&varg0), 0, idx, szProductBuf)) == ERROR_SUCCESS) idx++;
1890 if (ret != ERROR_NO_MORE_ITEMS)
1891 {
1892 VariantClear(&varg0);
1893 ERR("MsiEnumRelatedProducts returned %d\n", ret);
1894 return DISP_E_EXCEPTION;
1895 }
1896
1897 V_VT(pVarResult) = VT_DISPATCH;
1898 if (SUCCEEDED(hr = create_automation_object(0, NULL, (LPVOID*)&pDispatch, &DIID_StringList, ListImpl_Invoke, ListImpl_Free, sizeof(ListData))))
1899 {
1900 V_DISPATCH(pVarResult) = pDispatch;
1901
1902 /* Save product strings */
1903 ldata = (ListData *)private_data((AutomationObject *)pDispatch);
1904 if (!(ldata->pVars = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(VARIANT)*idx)))
1905 ERR("Out of memory\n");
1906 else
1907 {
1908 ldata->ulCount = idx;
1909 for (idx = 0; idx < ldata->ulCount; idx++)
1910 {
1911 ret = MsiEnumRelatedProductsW(V_BSTR(&varg0), 0, idx, szProductBuf);
1912 VariantInit(&ldata->pVars[idx]);
1913 V_VT(&ldata->pVars[idx]) = VT_BSTR;
1914 V_BSTR(&ldata->pVars[idx]) = SysAllocString(szProductBuf);
1915 }
1916 }
1917 }
1918 else
1919 ERR("Failed to create StringList object, hresult 0x%08x\n", hr);
1920 }
1921 else return DISP_E_MEMBERNOTFOUND;
1922 break;
1923
1924 default:
1925 return DISP_E_MEMBERNOTFOUND;
1926 }
1927
1928 VariantClear(&varg2);
1929 VariantClear(&varg1);
1930 VariantClear(&varg0);
1931
1932 return S_OK;
1933 }
1934
1935 /* Wrapper around create_automation_object to create an installer object. */
1936 HRESULT create_msiserver(IUnknown *pOuter, LPVOID *ppObj)
1937 {
1938 return create_automation_object(0, pOuter, ppObj, &DIID_Installer, InstallerImpl_Invoke, NULL, 0);
1939 }
1940
1941 /* Wrapper around create_automation_object to create a session object. */
1942 HRESULT create_session(MSIHANDLE msiHandle, IDispatch *pInstaller, IDispatch **pDispatch)
1943 {
1944 HRESULT hr = create_automation_object(msiHandle, NULL, (LPVOID)pDispatch, &DIID_Session, SessionImpl_Invoke, NULL, sizeof(SessionData));
1945 if (SUCCEEDED(hr) && pDispatch && *pDispatch)
1946 ((SessionData *)private_data((AutomationObject *)*pDispatch))->pInstaller = pInstaller;
1947 return hr;
1948 }