mfmediaengine: Remove unnecessary import library.
[wine.git] / dlls / vbscript / tests / createobj.c
blobc0a653fb1eae4bbe6f701feeff5834eb498e6a96
1 /*
2 * Copyright 2009,2011 Jacek Caban for CodeWeavers
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 #include <stdio.h>
21 #define COBJMACROS
22 #define CONST_VTABLE
24 #include <ole2.h>
25 #include <dispex.h>
26 #include <activscp.h>
27 #include <objsafe.h>
28 #include <urlmon.h>
29 #include <mshtmhst.h>
31 #include "wine/test.h"
33 #ifdef _WIN64
35 #define IActiveScriptParse_QueryInterface IActiveScriptParse64_QueryInterface
36 #define IActiveScriptParse_Release IActiveScriptParse64_Release
37 #define IActiveScriptParse_InitNew IActiveScriptParse64_InitNew
38 #define IActiveScriptParse_ParseScriptText IActiveScriptParse64_ParseScriptText
39 #define IActiveScriptParseProcedure2_Release IActiveScriptParseProcedure2_64_Release
41 #else
43 #define IActiveScriptParse_QueryInterface IActiveScriptParse32_QueryInterface
44 #define IActiveScriptParse_Release IActiveScriptParse32_Release
45 #define IActiveScriptParse_InitNew IActiveScriptParse32_InitNew
46 #define IActiveScriptParse_ParseScriptText IActiveScriptParse32_ParseScriptText
47 #define IActiveScriptParseProcedure2_Release IActiveScriptParseProcedure2_32_Release
49 #endif
51 extern const CLSID CLSID_VBScript;
53 #define VB_E_ACCESS_DENIED 0x800a0046
55 #define DEFINE_EXPECT(func) \
56 static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
58 #define SET_EXPECT(func) \
59 expect_ ## func = TRUE
61 #define CHECK_EXPECT2(func) \
62 do { \
63 ok(expect_ ##func, "unexpected call " #func "\n"); \
64 called_ ## func = TRUE; \
65 }while(0)
67 #define CHECK_EXPECT(func) \
68 do { \
69 CHECK_EXPECT2(func); \
70 expect_ ## func = FALSE; \
71 }while(0)
73 #define CHECK_CALLED(func) \
74 do { \
75 ok(called_ ## func, "expected " #func "\n"); \
76 expect_ ## func = called_ ## func = FALSE; \
77 }while(0)
79 #define CLEAR_CALLED(func) \
80 expect_ ## func = called_ ## func = FALSE
82 DEFINE_EXPECT(CreateInstance);
83 DEFINE_EXPECT(ProcessUrlAction);
84 DEFINE_EXPECT(QueryCustomPolicy);
85 DEFINE_EXPECT(reportSuccess);
86 DEFINE_EXPECT(Host_QS_SecMgr);
87 DEFINE_EXPECT(Caller_QS_SecMgr);
88 DEFINE_EXPECT(QI_IObjectWithSite);
89 DEFINE_EXPECT(SetSite);
91 static HRESULT QS_SecMgr_hres;
92 static HRESULT ProcessUrlAction_hres;
93 static DWORD ProcessUrlAction_policy;
94 static HRESULT CreateInstance_hres;
95 static HRESULT QueryCustomPolicy_hres;
96 static DWORD QueryCustomPolicy_psize;
97 static DWORD QueryCustomPolicy_policy;
98 static HRESULT QI_IDispatch_hres;
99 static HRESULT QI_IObjectWithSite_hres;
100 static HRESULT SetSite_hres;
102 #define TESTOBJ_CLSID "{178fc163-f585-4e24-9c13-4bb7faf80646}"
103 #define TESTOBJINST_CLSID "{178fc163-f585-4e24-9c13-4bb7faf80647}"
105 static const GUID CLSID_TestObj =
106 {0x178fc163,0xf585,0x4e24,{0x9c,0x13,0x4b,0xb7,0xfa,0xf8,0x06,0x46}};
107 static const GUID CLSID_TestObjInst =
108 {0x178fc163,0xf585,0x4e24,{0x9c,0x13,0x4b,0xb7,0xfa,0xf8,0x06,0x47}};
110 /* Defined as extern in urlmon.idl, but not exported by uuid.lib */
111 const GUID GUID_CUSTOM_CONFIRMOBJECTSAFETY =
112 {0x10200490,0xfa38,0x11d0,{0xac,0x0e,0x00,0xa0,0xc9,0xf,0xff,0xc0}};
114 #define DISPID_TEST_REPORTSUCCESS 0x1000
116 #define DISPID_GLOBAL_OK 0x2000
118 #define VB_E_CANNOT_CREATE_OBJ 0x800a01ad
120 static BSTR a2bstr(const char *str)
122 BSTR ret;
123 int len;
125 len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
126 ret = SysAllocStringLen(NULL, len-1);
127 MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
129 return ret;
132 static HRESULT WINAPI ObjectWithSite_QueryInterface(IObjectWithSite *iface, REFIID riid, void **ppv)
134 ok(0, "unexpected call\n");
135 return E_NOTIMPL;
138 static ULONG WINAPI ObjectWithSite_AddRef(IObjectWithSite *iface)
140 return 2;
143 static ULONG WINAPI ObjectWithSite_Release(IObjectWithSite *iface)
145 return 1;
148 static HRESULT WINAPI ObjectWithSite_SetSite(IObjectWithSite *iface, IUnknown *pUnkSite)
150 IServiceProvider *sp;
151 HRESULT hres;
154 CHECK_EXPECT(SetSite);
155 ok(pUnkSite != NULL, "pUnkSite == NULL\n");
157 hres = IUnknown_QueryInterface(pUnkSite, &IID_IServiceProvider, (void**)&sp);
158 ok(hres == S_OK, "Could not get IServiceProvider iface: %08x\n", hres);
159 IServiceProvider_Release(sp);
161 return SetSite_hres;
164 static HRESULT WINAPI ObjectWithSite_GetSite(IObjectWithSite *iface, REFIID riid, void **ppvSite)
166 ok(0, "unexpected call\n");
167 return E_NOTIMPL;
170 static const IObjectWithSiteVtbl ObjectWithSiteVtbl = {
171 ObjectWithSite_QueryInterface,
172 ObjectWithSite_AddRef,
173 ObjectWithSite_Release,
174 ObjectWithSite_SetSite,
175 ObjectWithSite_GetSite
178 static IObjectWithSite ObjectWithSite = { &ObjectWithSiteVtbl };
180 static IObjectWithSite *object_with_site;
182 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
184 *ppv = NULL;
186 if(IsEqualGUID(riid, &IID_IUnknown)) {
187 *ppv = iface;
188 }else if(IsEqualGUID(riid, &IID_IDispatch) || IsEqualGUID(riid, &IID_IDispatchEx)) {
189 if(FAILED(QI_IDispatch_hres))
190 return QI_IDispatch_hres;
191 *ppv = iface;
192 }else if(IsEqualGUID(&IID_IObjectWithSite, riid)) {
193 CHECK_EXPECT(QI_IObjectWithSite);
194 if(FAILED(QI_IObjectWithSite_hres))
195 return QI_IObjectWithSite_hres;
196 *ppv = object_with_site;
197 }else if(IsEqualGUID(&IID_IObjectSafety, riid)) {
198 ok(0, "Unexpected IID_IObjectSafety query\n");
201 return *ppv ? S_OK : E_NOINTERFACE;
204 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
206 return 2;
209 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
211 return 1;
214 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
216 ok(0, "unexpected call\n");
217 return E_NOTIMPL;
220 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
221 LCID lcid, ITypeInfo **ppTInfo)
223 return E_NOTIMPL;
226 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
227 LPOLESTR *rgszNames, UINT cNames,
228 LCID lcid, DISPID *rgDispId)
230 ok(0, "unexpected call\n");
231 return E_NOTIMPL;
234 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
235 REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
236 VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
238 ok(0, "unexpected call\n");
239 return E_NOTIMPL;
242 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
244 ok(0, "unexpected call %s %x\n", wine_dbgstr_w(bstrName), grfdex);
245 return E_NOTIMPL;
248 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
250 ok(0, "unexpected call\n");
251 return E_NOTIMPL;
254 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
256 ok(0, "unexpected call\n");
257 return E_NOTIMPL;
260 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
262 ok(0, "unexpected call\n");
263 return E_NOTIMPL;
266 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
268 ok(0, "unexpected call\n");
269 return E_NOTIMPL;
272 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
274 ok(0, "unexpected call\n");
275 return E_NOTIMPL;
278 static HRESULT WINAPI Test_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
280 if(!lstrcmpW(bstrName, L"reportSuccess")) {
281 ok(grfdex == fdexNameCaseInsensitive, "grfdex = %x\n", grfdex);
282 *pid = DISPID_TEST_REPORTSUCCESS;
283 return S_OK;
286 ok(0, "unexpected name %s\n", wine_dbgstr_w(bstrName));
287 return E_NOTIMPL;
290 static HRESULT WINAPI Test_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
291 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
293 switch(id) {
294 case DISPID_TEST_REPORTSUCCESS:
295 CHECK_EXPECT(reportSuccess);
297 ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
298 ok(pdp != NULL, "pdp == NULL\n");
299 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
300 ok(pdp->cArgs == 0, "cArgs = %d\n", pdp->cArgs);
301 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
302 ok(!pvarRes, "pvarRes != NULL\n");
303 ok(pei != NULL, "pei == NULL\n");
304 break;
306 default:
307 ok(0, "unexpected call\n");
308 return E_NOTIMPL;
311 return S_OK;
314 static IDispatchExVtbl testObjVtbl = {
315 DispatchEx_QueryInterface,
316 DispatchEx_AddRef,
317 DispatchEx_Release,
318 DispatchEx_GetTypeInfoCount,
319 DispatchEx_GetTypeInfo,
320 DispatchEx_GetIDsOfNames,
321 DispatchEx_Invoke,
322 Test_GetDispID,
323 Test_InvokeEx,
324 DispatchEx_DeleteMemberByName,
325 DispatchEx_DeleteMemberByDispID,
326 DispatchEx_GetMemberProperties,
327 DispatchEx_GetMemberName,
328 DispatchEx_GetNextDispID,
329 DispatchEx_GetNameSpaceParent
332 static IDispatchEx testObj = { &testObjVtbl };
334 static HRESULT WINAPI Global_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
336 if(!lstrcmpW(bstrName, L"ok")) {
337 ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
338 *pid = DISPID_GLOBAL_OK;
339 return S_OK;
342 ok(0, "unexpected name %s\n", wine_dbgstr_w(bstrName));
343 return E_NOTIMPL;
346 static HRESULT WINAPI Global_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
347 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
349 switch(id) {
350 case DISPID_GLOBAL_OK:
351 ok(wFlags == INVOKE_FUNC || wFlags == (INVOKE_FUNC|INVOKE_PROPERTYGET), "wFlags = %x\n", wFlags);
352 ok(pdp != NULL, "pdp == NULL\n");
353 ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
354 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
355 ok(pdp->cArgs == 2, "cArgs = %d\n", pdp->cArgs);
356 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
357 ok(pei != NULL, "pei == NULL\n");
359 ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
360 ok(V_VT(pdp->rgvarg+1) == VT_BOOL, "V_VT(psp->rgvargs+1) = %d\n", V_VT(pdp->rgvarg));
361 ok(V_BOOL(pdp->rgvarg+1), "%s\n", wine_dbgstr_w(V_BSTR(pdp->rgvarg)));
362 break;
364 default:
365 ok(0, "unexpected call\n");
366 return E_NOTIMPL;
369 return S_OK;
372 static IDispatchExVtbl globalObjVtbl = {
373 DispatchEx_QueryInterface,
374 DispatchEx_AddRef,
375 DispatchEx_Release,
376 DispatchEx_GetTypeInfoCount,
377 DispatchEx_GetTypeInfo,
378 DispatchEx_GetIDsOfNames,
379 DispatchEx_Invoke,
380 Global_GetDispID,
381 Global_InvokeEx,
382 DispatchEx_DeleteMemberByName,
383 DispatchEx_DeleteMemberByDispID,
384 DispatchEx_GetMemberProperties,
385 DispatchEx_GetMemberName,
386 DispatchEx_GetNextDispID,
387 DispatchEx_GetNameSpaceParent
390 static IDispatchEx globalObj = { &globalObjVtbl };
392 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
394 if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IClassFactory, riid)) {
395 *ppv = iface;
396 return S_OK;
399 /* TODO: IClassFactoryEx */
400 *ppv = NULL;
401 return E_NOINTERFACE;
404 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
406 return 2;
409 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
411 return 1;
414 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *outer, REFIID riid, void **ppv)
416 CHECK_EXPECT(CreateInstance);
418 ok(!outer, "outer = %p\n", outer);
419 ok(IsEqualGUID(&IID_IUnknown, riid), "unexpected riid %s\n", wine_dbgstr_guid(riid));
421 if(SUCCEEDED(CreateInstance_hres))
422 *ppv = &testObj;
423 return CreateInstance_hres;
426 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
428 ok(0, "unexpected call\n");
429 return S_OK;
432 static const IClassFactoryVtbl ClassFactoryVtbl = {
433 ClassFactory_QueryInterface,
434 ClassFactory_AddRef,
435 ClassFactory_Release,
436 ClassFactory_CreateInstance,
437 ClassFactory_LockServer
440 static IClassFactory activex_cf = { &ClassFactoryVtbl };
442 static HRESULT WINAPI InternetHostSecurityManager_QueryInterface(IInternetHostSecurityManager *iface, REFIID riid, void **ppv)
444 ok(0, "unexpected call\n");
445 return E_NOINTERFACE;
448 static ULONG WINAPI InternetHostSecurityManager_AddRef(IInternetHostSecurityManager *iface)
450 return 2;
453 static ULONG WINAPI InternetHostSecurityManager_Release(IInternetHostSecurityManager *iface)
455 return 1;
458 static HRESULT WINAPI InternetHostSecurityManager_GetSecurityId(IInternetHostSecurityManager *iface, BYTE *pbSecurityId,
459 DWORD *pcbSecurityId, DWORD_PTR dwReserved)
461 ok(0, "unexpected call\n");
462 return E_NOTIMPL;
465 static HRESULT WINAPI InternetHostSecurityManager_ProcessUrlAction(IInternetHostSecurityManager *iface, DWORD dwAction,
466 BYTE *pPolicy, DWORD cbPolicy, BYTE *pContext, DWORD cbContext, DWORD dwFlags, DWORD dwReserved)
468 CHECK_EXPECT(ProcessUrlAction);
470 ok(dwAction == URLACTION_ACTIVEX_RUN, "dwAction = %x\n", dwAction);
471 ok(pPolicy != NULL, "pPolicy == NULL\n");
472 ok(cbPolicy == sizeof(DWORD), "cbPolicy = %d\n", cbPolicy);
473 ok(pContext != NULL, "pContext == NULL\n");
474 ok(cbContext == sizeof(GUID), "cbContext = %d\n", cbContext);
475 ok(IsEqualGUID(pContext, &CLSID_TestObj), "pContext = %s\n", wine_dbgstr_guid((const IID*)pContext));
476 ok(!dwFlags, "dwFlags = %x\n", dwFlags);
477 ok(!dwReserved, "dwReserved = %x\n", dwReserved);
479 if(SUCCEEDED(ProcessUrlAction_hres))
480 *(DWORD*)pPolicy = ProcessUrlAction_policy;
481 return ProcessUrlAction_hres;
484 static HRESULT WINAPI InternetHostSecurityManager_QueryCustomPolicy(IInternetHostSecurityManager *iface, REFGUID guidKey,
485 BYTE **ppPolicy, DWORD *pcbPolicy, BYTE *pContext, DWORD cbContext, DWORD dwReserved)
487 const struct CONFIRMSAFETY *cs = (const struct CONFIRMSAFETY*)pContext;
488 DWORD *ret;
490 CHECK_EXPECT(QueryCustomPolicy);
492 ok(IsEqualGUID(&GUID_CUSTOM_CONFIRMOBJECTSAFETY, guidKey), "guidKey = %s\n", wine_dbgstr_guid(guidKey));
494 ok(ppPolicy != NULL, "ppPolicy == NULL\n");
495 ok(pcbPolicy != NULL, "pcbPolicy == NULL\n");
496 ok(pContext != NULL, "pContext == NULL\n");
497 ok(cbContext == sizeof(struct CONFIRMSAFETY), "cbContext = %d\n", cbContext);
498 ok(!dwReserved, "dwReserved = %x\n", dwReserved);
500 /* TODO: CLSID */
501 ok(cs->pUnk != NULL, "cs->pUnk == NULL\n");
502 ok(!cs->dwFlags, "dwFlags = %x\n", cs->dwFlags);
504 if(FAILED(QueryCustomPolicy_hres))
505 return QueryCustomPolicy_hres;
507 ret = CoTaskMemAlloc(QueryCustomPolicy_psize);
508 *ppPolicy = (BYTE*)ret;
509 *pcbPolicy = QueryCustomPolicy_psize;
510 memset(ret, 0, QueryCustomPolicy_psize);
511 if(QueryCustomPolicy_psize >= sizeof(DWORD))
512 *ret = QueryCustomPolicy_policy;
514 return QueryCustomPolicy_hres;
517 static const IInternetHostSecurityManagerVtbl InternetHostSecurityManagerVtbl = {
518 InternetHostSecurityManager_QueryInterface,
519 InternetHostSecurityManager_AddRef,
520 InternetHostSecurityManager_Release,
521 InternetHostSecurityManager_GetSecurityId,
522 InternetHostSecurityManager_ProcessUrlAction,
523 InternetHostSecurityManager_QueryCustomPolicy
526 static IInternetHostSecurityManager InternetHostSecurityManager = { &InternetHostSecurityManagerVtbl };
528 static IServiceProvider ServiceProvider;
530 static HRESULT WINAPI ServiceProvider_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv)
532 ok(0, "unexpected call\n");
533 return E_NOINTERFACE;
536 static ULONG WINAPI ServiceProvider_AddRef(IServiceProvider *iface)
538 return 2;
541 static ULONG WINAPI ServiceProvider_Release(IServiceProvider *iface)
543 return 1;
546 static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface,
547 REFGUID guidService, REFIID riid, void **ppv)
549 if(IsEqualGUID(&SID_GetCaller, guidService))
550 return E_NOINTERFACE;
552 if(IsEqualGUID(&SID_SInternetHostSecurityManager, guidService)) {
553 if(iface == &ServiceProvider)
554 CHECK_EXPECT(Host_QS_SecMgr);
555 else
556 CHECK_EXPECT(Caller_QS_SecMgr);
557 ok(IsEqualGUID(&IID_IInternetHostSecurityManager, riid), "unexpected riid %s\n", wine_dbgstr_guid(riid));
558 if(SUCCEEDED(QS_SecMgr_hres))
559 *ppv = &InternetHostSecurityManager;
560 return QS_SecMgr_hres;
563 ok(0, "unexpected service %s\n", wine_dbgstr_guid(guidService));
564 return E_NOINTERFACE;
567 static IServiceProviderVtbl ServiceProviderVtbl = {
568 ServiceProvider_QueryInterface,
569 ServiceProvider_AddRef,
570 ServiceProvider_Release,
571 ServiceProvider_QueryService
574 static IServiceProvider ServiceProvider = { &ServiceProviderVtbl };
576 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
578 if(IsEqualGUID(&IID_IUnknown, riid)) {
579 *ppv = iface;
580 }else if(IsEqualGUID(&IID_IActiveScriptSite, riid)) {
581 *ppv = iface;
582 }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
583 *ppv = &ServiceProvider;
584 }else {
585 *ppv = NULL;
586 return E_NOINTERFACE;
589 IUnknown_AddRef((IUnknown*)*ppv);
590 return S_OK;
593 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
595 return 2;
598 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
600 return 1;
603 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
605 *plcid = GetUserDefaultLCID();
606 return S_OK;
609 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
610 DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
612 ok(dwReturnMask == SCRIPTINFO_IUNKNOWN, "unexpected dwReturnMask %x\n", dwReturnMask);
613 ok(!ppti, "ppti != NULL\n");
614 ok(!lstrcmpW(pstrName, L"test"), "pstrName = %s\n", wine_dbgstr_w(pstrName));
616 *ppiunkItem = (IUnknown*)&globalObj;
617 return S_OK;
620 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
622 return E_NOTIMPL;
625 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
626 const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
628 return E_NOTIMPL;
631 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
633 return E_NOTIMPL;
636 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
638 return E_NOTIMPL;
641 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
643 return E_NOTIMPL;
646 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
648 return E_NOTIMPL;
651 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
652 ActiveScriptSite_QueryInterface,
653 ActiveScriptSite_AddRef,
654 ActiveScriptSite_Release,
655 ActiveScriptSite_GetLCID,
656 ActiveScriptSite_GetItemInfo,
657 ActiveScriptSite_GetDocVersionString,
658 ActiveScriptSite_OnScriptTerminate,
659 ActiveScriptSite_OnStateChange,
660 ActiveScriptSite_OnScriptError,
661 ActiveScriptSite_OnEnterScript,
662 ActiveScriptSite_OnLeaveScript
665 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
667 static void set_safety_options(IUnknown *unk, BOOL use_sec_mgr)
669 IObjectSafety *safety;
670 DWORD supported, enabled, options_all, options_set;
671 HRESULT hres;
673 hres = IUnknown_QueryInterface(unk, &IID_IObjectSafety, (void**)&safety);
674 ok(hres == S_OK, "Could not get IObjectSafety: %08x\n", hres);
675 if(FAILED(hres))
676 return;
678 options_all = INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER;
679 if(use_sec_mgr)
680 options_set = options_all;
681 else
682 options_set = INTERFACE_USES_DISPEX;
684 hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, options_all, options_set);
685 ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
687 supported = enabled = 0xdeadbeef;
688 hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
689 ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
690 ok(supported == options_all, "supported=%x, expected %x\n", supported, options_all);
691 ok(enabled == options_set, "enabled=%x, expected %x\n", enabled, options_set);
693 IObjectSafety_Release(safety);
696 #define parse_script_a(p,s) _parse_script_a(__LINE__,p,s)
697 static void _parse_script_a(unsigned line, IActiveScriptParse *parser, const char *script)
699 BSTR str;
700 HRESULT hres;
702 str = a2bstr(script);
703 hres = IActiveScriptParse_ParseScriptText(parser, str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
704 SysFreeString(str);
705 ok_(__FILE__,line)(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
708 static HRESULT parse_script_ae(IActiveScriptParse *parser, const char *script)
710 BSTR str;
711 HRESULT hres;
713 str = a2bstr(script);
714 hres = IActiveScriptParse_ParseScriptText(parser, str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
715 SysFreeString(str);
717 return hres;
720 static IActiveScriptParse *create_script(BOOL use_sec_mgr)
722 IActiveScriptParse *parser;
723 IActiveScript *script;
724 HRESULT hres;
726 QS_SecMgr_hres = S_OK;
727 ProcessUrlAction_hres = S_OK;
728 ProcessUrlAction_policy = URLPOLICY_ALLOW;
729 CreateInstance_hres = S_OK;
730 QueryCustomPolicy_hres = S_OK;
731 QueryCustomPolicy_psize = sizeof(DWORD);
732 QueryCustomPolicy_policy = URLPOLICY_ALLOW;
733 QI_IDispatch_hres = S_OK;
734 QI_IObjectWithSite_hres = S_OK;
735 SetSite_hres = S_OK;
737 hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
738 &IID_IActiveScript, (void**)&script);
739 ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
740 if(FAILED(hres))
741 return NULL;
743 set_safety_options((IUnknown*)script, use_sec_mgr);
745 hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parser);
746 ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
748 hres = IActiveScriptParse_InitNew(parser);
749 ok(hres == S_OK, "InitNew failed: %08x\n", hres);
751 hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
752 ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
754 hres = IActiveScript_AddNamedItem(script, L"test",
755 SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
756 ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
758 hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_STARTED);
759 ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
761 IActiveScript_Release(script);
763 return parser;
766 static void test_CreateObject(void)
768 IActiveScriptParse *parser;
769 HRESULT hres;
771 parser = create_script(TRUE);
773 SET_EXPECT(Host_QS_SecMgr);
774 SET_EXPECT(ProcessUrlAction);
775 SET_EXPECT(CreateInstance);
776 SET_EXPECT(QueryCustomPolicy);
777 SET_EXPECT(QI_IObjectWithSite);
778 SET_EXPECT(reportSuccess);
779 parse_script_a(parser, "Call CreateObject(\"Wine.Test\").reportSuccess()");
780 CHECK_CALLED(Host_QS_SecMgr);
781 CHECK_CALLED(ProcessUrlAction);
782 CHECK_CALLED(CreateInstance);
783 CHECK_CALLED(QueryCustomPolicy);
784 CHECK_CALLED(QI_IObjectWithSite);
785 CHECK_CALLED(reportSuccess);
787 IActiveScriptParse_Release(parser);
789 parser = create_script(TRUE);
791 hres = parse_script_ae(parser, "Call CreateObject(\"Wine.TestABC\")");
792 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
794 IActiveScriptParse_Release(parser);
796 parser = create_script(TRUE);
797 QS_SecMgr_hres = E_NOINTERFACE;
799 SET_EXPECT(Host_QS_SecMgr);
800 hres = parse_script_ae(parser, "Call CreateObject(\"Wine.Test\")");
801 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
802 CHECK_CALLED(Host_QS_SecMgr);
804 IActiveScriptParse_Release(parser);
806 parser = create_script(TRUE);
807 ProcessUrlAction_hres = E_FAIL;
809 SET_EXPECT(Host_QS_SecMgr);
810 SET_EXPECT(ProcessUrlAction);
811 hres = parse_script_ae(parser, "Call CreateObject(\"Wine.Test\")");
812 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
813 CHECK_CALLED(Host_QS_SecMgr);
814 CHECK_CALLED(ProcessUrlAction);
816 IActiveScriptParse_Release(parser);
818 parser = create_script(TRUE);
819 ProcessUrlAction_policy = URLPOLICY_DISALLOW;
821 SET_EXPECT(Host_QS_SecMgr);
822 SET_EXPECT(ProcessUrlAction);
823 hres = parse_script_ae(parser, "Call CreateObject(\"Wine.Test\")");
824 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
825 CHECK_CALLED(Host_QS_SecMgr);
826 CHECK_CALLED(ProcessUrlAction);
828 IActiveScriptParse_Release(parser);
830 parser = create_script(TRUE);
831 CreateInstance_hres = E_FAIL;
833 SET_EXPECT(Host_QS_SecMgr);
834 SET_EXPECT(ProcessUrlAction);
835 SET_EXPECT(CreateInstance);
836 hres = parse_script_ae(parser, "Call CreateObject(\"Wine.Test\")");
837 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
838 CHECK_CALLED(Host_QS_SecMgr);
839 CHECK_CALLED(ProcessUrlAction);
840 CHECK_CALLED(CreateInstance);
842 IActiveScriptParse_Release(parser);
844 parser = create_script(TRUE);
845 QueryCustomPolicy_hres = E_FAIL;
847 SET_EXPECT(Host_QS_SecMgr);
848 SET_EXPECT(ProcessUrlAction);
849 SET_EXPECT(CreateInstance);
850 SET_EXPECT(QueryCustomPolicy);
851 hres = parse_script_ae(parser, "Call CreateObject(\"Wine.Test\")");
852 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
853 CHECK_CALLED(Host_QS_SecMgr);
854 CHECK_CALLED(ProcessUrlAction);
855 CHECK_CALLED(CreateInstance);
856 CHECK_CALLED(QueryCustomPolicy);
858 IActiveScriptParse_Release(parser);
860 parser = create_script(TRUE);
861 QueryCustomPolicy_psize = 6;
863 SET_EXPECT(Host_QS_SecMgr);
864 SET_EXPECT(ProcessUrlAction);
865 SET_EXPECT(CreateInstance);
866 SET_EXPECT(QueryCustomPolicy);
867 SET_EXPECT(QI_IObjectWithSite);
868 SET_EXPECT(reportSuccess);
869 parse_script_a(parser, "Call CreateObject(\"Wine.Test\").reportSuccess()");
870 CHECK_CALLED(Host_QS_SecMgr);
871 CHECK_CALLED(ProcessUrlAction);
872 CHECK_CALLED(CreateInstance);
873 CHECK_CALLED(QueryCustomPolicy);
874 CHECK_CALLED(QI_IObjectWithSite);
875 CHECK_CALLED(reportSuccess);
877 IActiveScriptParse_Release(parser);
879 parser = create_script(TRUE);
880 QueryCustomPolicy_policy = URLPOLICY_DISALLOW;
882 SET_EXPECT(Host_QS_SecMgr);
883 SET_EXPECT(ProcessUrlAction);
884 SET_EXPECT(CreateInstance);
885 SET_EXPECT(QueryCustomPolicy);
886 hres = parse_script_ae(parser, "Call CreateObject(\"Wine.Test\")");
887 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
888 CHECK_CALLED(Host_QS_SecMgr);
889 CHECK_CALLED(ProcessUrlAction);
890 CHECK_CALLED(CreateInstance);
891 CHECK_CALLED(QueryCustomPolicy);
893 QueryCustomPolicy_psize = 6;
895 SET_EXPECT(ProcessUrlAction);
896 SET_EXPECT(CreateInstance);
897 SET_EXPECT(QueryCustomPolicy);
898 hres = parse_script_ae(parser, "Call CreateObject(\"Wine.Test\")");
899 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
900 CHECK_CALLED(ProcessUrlAction);
901 CHECK_CALLED(CreateInstance);
902 CHECK_CALLED(QueryCustomPolicy);
904 QueryCustomPolicy_policy = URLPOLICY_ALLOW;
905 QueryCustomPolicy_psize = 3;
907 SET_EXPECT(ProcessUrlAction);
908 SET_EXPECT(CreateInstance);
909 SET_EXPECT(QueryCustomPolicy);
910 hres = parse_script_ae(parser, "Call CreateObject(\"Wine.Test\")");
911 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
912 CHECK_CALLED(ProcessUrlAction);
913 CHECK_CALLED(CreateInstance);
914 CHECK_CALLED(QueryCustomPolicy);
916 IActiveScriptParse_Release(parser);
918 parser = create_script(FALSE);
920 SET_EXPECT(CreateInstance);
921 SET_EXPECT(QI_IObjectWithSite);
922 SET_EXPECT(reportSuccess);
923 parse_script_a(parser, "Call CreateObject(\"Wine.Test\").reportSuccess()");
924 CHECK_CALLED(CreateInstance);
925 CHECK_CALLED(QI_IObjectWithSite);
926 CHECK_CALLED(reportSuccess);
928 IActiveScriptParse_Release(parser);
930 parser = create_script(TRUE);
931 object_with_site = &ObjectWithSite;
933 SET_EXPECT(Host_QS_SecMgr);
934 SET_EXPECT(ProcessUrlAction);
935 SET_EXPECT(CreateInstance);
936 SET_EXPECT(QueryCustomPolicy);
937 SET_EXPECT(QI_IObjectWithSite);
938 SET_EXPECT(SetSite);
939 SET_EXPECT(reportSuccess);
940 parse_script_a(parser, "Call CreateObject(\"Wine.Test\").reportSuccess()");
941 CHECK_CALLED(Host_QS_SecMgr);
942 CHECK_CALLED(ProcessUrlAction);
943 CHECK_CALLED(CreateInstance);
944 CHECK_CALLED(QueryCustomPolicy);
945 CHECK_CALLED(QI_IObjectWithSite);
946 CHECK_CALLED(SetSite);
947 CHECK_CALLED(reportSuccess);
949 SetSite_hres = E_FAIL;
950 SET_EXPECT(ProcessUrlAction);
951 SET_EXPECT(CreateInstance);
952 SET_EXPECT(QueryCustomPolicy);
953 SET_EXPECT(QI_IObjectWithSite);
954 SET_EXPECT(SetSite);
955 hres = parse_script_ae(parser, "Call CreateObject(\"Wine.Test\")");
956 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
957 CHECK_CALLED(ProcessUrlAction);
958 CHECK_CALLED(CreateInstance);
959 CHECK_CALLED(QueryCustomPolicy);
960 CHECK_CALLED(QI_IObjectWithSite);
961 CHECK_CALLED(SetSite);
963 IActiveScriptParse_Release(parser);
966 static void test_GetObject(void)
968 IActiveScriptParse *parser;
969 HRESULT hres;
971 /* Never allowed with security manager */
972 parser = create_script(TRUE);
973 hres = parse_script_ae(parser, "Call GetObject(\"clsid:" TESTOBJINST_CLSID "\").reportSuccess()");
974 ok(hres == VB_E_CANNOT_CREATE_OBJ, "hres = %08x\n", hres);
975 IActiveScriptParse_Release(parser);
977 parser = create_script(FALSE);
979 SET_EXPECT(QI_IObjectWithSite);
980 SET_EXPECT(SetSite);
981 SET_EXPECT(reportSuccess);
982 hres = parse_script_ae(parser, "Call GetObject(\"clsid:" TESTOBJINST_CLSID "\").reportSuccess()");
983 if(broken(hres == VB_E_ACCESS_DENIED)) /* 64-bit win8 fails on the first try */
984 hres = parse_script_ae(parser, "Call GetObject(\"clsid:" TESTOBJINST_CLSID "\").reportSuccess()");
985 if(hres == HRESULT_FROM_WIN32(ERROR_MOD_NOT_FOUND)) { /* Workaround for broken win2k */
986 win_skip("got unexpected error %08x\n", hres);
987 CLEAR_CALLED(QI_IObjectWithSite);
988 CLEAR_CALLED(SetSite);
989 CLEAR_CALLED(reportSuccess);
990 IActiveScriptParse_Release(parser);
991 return;
993 ok(hres == S_OK, "hres = %08x\n", hres);
994 CHECK_CALLED(QI_IObjectWithSite);
995 CHECK_CALLED(SetSite);
996 CHECK_CALLED(reportSuccess);
998 SetSite_hres = E_FAIL;
999 SET_EXPECT(QI_IObjectWithSite);
1000 SET_EXPECT(SetSite);
1001 hres = parse_script_ae(parser, "Call GetObject(\"clsid:" TESTOBJINST_CLSID "\").reportSuccess()");
1002 ok(hres == E_FAIL, "hres = %08x\n", hres);
1003 CHECK_CALLED(QI_IObjectWithSite);
1004 CHECK_CALLED(SetSite);
1006 QI_IObjectWithSite_hres = E_NOINTERFACE;
1007 SET_EXPECT(QI_IObjectWithSite);
1008 SET_EXPECT(reportSuccess);
1009 parse_script_a(parser, "Call GetObject(\"clsid:" TESTOBJINST_CLSID "\").reportSuccess()");
1010 CHECK_CALLED(QI_IObjectWithSite);
1011 CHECK_CALLED(reportSuccess);
1013 IActiveScriptParse_Release(parser);
1015 /* Invalid moniker */
1016 parser = create_script(FALSE);
1017 hres = parse_script_ae(parser, "Call GetObject(\"nonexistent:test\").reportSuccess()");
1018 ok(hres == MK_E_SYNTAX, "hres = %08x\n", hres);
1019 IActiveScriptParse_Release(parser);
1022 static BOOL init_key(const char *key_name, const char *def_value, BOOL init)
1024 HKEY hkey;
1025 DWORD res;
1027 if(!init) {
1028 RegDeleteKeyA(HKEY_CLASSES_ROOT, key_name);
1029 return TRUE;
1032 res = RegCreateKeyA(HKEY_CLASSES_ROOT, key_name, &hkey);
1033 if(res != ERROR_SUCCESS)
1034 return FALSE;
1036 if(def_value)
1037 res = RegSetValueA(hkey, NULL, REG_SZ, def_value, strlen(def_value));
1039 RegCloseKey(hkey);
1041 return res == ERROR_SUCCESS;
1044 static BOOL init_registry(BOOL init)
1046 return init_key("Wine.Test\\CLSID", TESTOBJ_CLSID, init);
1049 static BOOL register_activex(void)
1051 DWORD regid;
1052 HRESULT hres;
1054 if(!init_registry(TRUE)) {
1055 init_registry(FALSE);
1056 return FALSE;
1059 hres = CoRegisterClassObject(&CLSID_TestObj, (IUnknown *)&activex_cf,
1060 CLSCTX_INPROC_SERVER, REGCLS_MULTIPLEUSE, &regid);
1061 ok(hres == S_OK, "Could not register script engine: %08x\n", hres);
1063 hres = CoRegisterClassObject(&CLSID_TestObjInst, (IUnknown *)&testObj,
1064 CLSCTX_INPROC_SERVER, REGCLS_MULTIPLEUSE, &regid);
1065 ok(hres == S_OK, "Could not register script engine: %08x\n", hres);
1067 return TRUE;
1070 static BOOL check_vbscript(void)
1072 IActiveScriptParseProcedure2 *vbscript;
1073 HRESULT hres;
1075 hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
1076 &IID_IActiveScriptParseProcedure2, (void**)&vbscript);
1077 if(SUCCEEDED(hres))
1078 IActiveScriptParseProcedure2_Release(vbscript);
1080 return hres == S_OK;
1084 START_TEST(createobj)
1086 CoInitialize(NULL);
1088 if(check_vbscript()) {
1089 if(register_activex()) {
1090 test_CreateObject();
1091 test_GetObject();
1092 init_registry(FALSE);
1093 }else {
1094 skip("Could not register ActiveX object.\n");
1096 }else {
1097 win_skip("Broken engine, probably too old\n");
1100 CoUninitialize();