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
31 #include "wine/test.h"
33 extern const CLSID CLSID_VBScript
;
35 #define DEFINE_EXPECT(func) \
36 static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
38 #define SET_EXPECT(func) \
39 expect_ ## func = TRUE
41 #define CHECK_EXPECT2(func) \
43 ok(expect_ ##func, "unexpected call " #func "\n"); \
44 called_ ## func = TRUE; \
47 #define CHECK_EXPECT(func) \
49 CHECK_EXPECT2(func); \
50 expect_ ## func = FALSE; \
53 #define CHECK_CALLED(func) \
55 ok(called_ ## func, "expected " #func "\n"); \
56 expect_ ## func = called_ ## func = FALSE; \
59 DEFINE_EXPECT(CreateInstance
);
60 DEFINE_EXPECT(ProcessUrlAction
);
61 DEFINE_EXPECT(QueryCustomPolicy
);
62 DEFINE_EXPECT(reportSuccess
);
63 DEFINE_EXPECT(Host_QS_SecMgr
);
64 DEFINE_EXPECT(Caller_QS_SecMgr
);
65 DEFINE_EXPECT(QI_IObjectWithSite
);
66 DEFINE_EXPECT(SetSite
);
68 static const WCHAR testW
[] = {'t','e','s','t',0};
70 static HRESULT QS_SecMgr_hres
;
71 static HRESULT ProcessUrlAction_hres
;
72 static DWORD ProcessUrlAction_policy
;
73 static HRESULT CreateInstance_hres
;
74 static HRESULT QueryCustomPolicy_hres
;
75 static DWORD QueryCustomPolicy_psize
;
76 static DWORD QueryCustomPolicy_policy
;
77 static HRESULT QI_IDispatch_hres
;
78 static HRESULT SetSite_hres
;
80 #define TESTOBJ_CLSID "{178fc163-f585-4e24-9c13-4bb7faf80646}"
82 static const GUID CLSID_TestObj
=
83 {0x178fc163,0xf585,0x4e24,{0x9c,0x13,0x4b,0xb7,0xfa,0xf8,0x06,0x46}};
85 /* Defined as extern in urlmon.idl, but not exported by uuid.lib */
86 const GUID GUID_CUSTOM_CONFIRMOBJECTSAFETY
=
87 {0x10200490,0xfa38,0x11d0,{0xac,0x0e,0x00,0xa0,0xc9,0xf,0xff,0xc0}};
89 #define DISPID_TEST_REPORTSUCCESS 0x1000
91 #define DISPID_GLOBAL_OK 0x2000
93 #define VB_E_CANNOT_CREATE_OBJ 0x800a01ad
95 static const char *debugstr_guid(REFIID riid
)
99 sprintf(buf
, "{%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",
100 riid
->Data1
, riid
->Data2
, riid
->Data3
, riid
->Data4
[0],
101 riid
->Data4
[1], riid
->Data4
[2], riid
->Data4
[3], riid
->Data4
[4],
102 riid
->Data4
[5], riid
->Data4
[6], riid
->Data4
[7]);
107 static BSTR
a2bstr(const char *str
)
112 len
= MultiByteToWideChar(CP_ACP
, 0, str
, -1, NULL
, 0);
113 ret
= SysAllocStringLen(NULL
, len
-1);
114 MultiByteToWideChar(CP_ACP
, 0, str
, -1, ret
, len
);
119 static int strcmp_wa(LPCWSTR strw
, const char *stra
)
122 WideCharToMultiByte(CP_ACP
, 0, strw
, -1, buf
, sizeof(buf
), 0, 0);
123 return lstrcmpA(buf
, stra
);
126 static HRESULT WINAPI
ObjectWithSite_QueryInterface(IObjectWithSite
*iface
, REFIID riid
, void **ppv
)
128 ok(0, "unexpected call\n");
132 static ULONG WINAPI
ObjectWithSite_AddRef(IObjectWithSite
*iface
)
137 static ULONG WINAPI
ObjectWithSite_Release(IObjectWithSite
*iface
)
142 static HRESULT WINAPI
ObjectWithSite_SetSite(IObjectWithSite
*iface
, IUnknown
*pUnkSite
)
144 IServiceProvider
*sp
;
148 CHECK_EXPECT(SetSite
);
149 ok(pUnkSite
!= NULL
, "pUnkSite == NULL\n");
151 hres
= IUnknown_QueryInterface(pUnkSite
, &IID_IServiceProvider
, (void**)&sp
);
152 ok(hres
== S_OK
, "Could not get IServiceProvider iface: %08x\n", hres
);
153 IServiceProvider_Release(sp
);
158 static HRESULT WINAPI
ObjectWithSite_GetSite(IObjectWithSite
*iface
, REFIID riid
, void **ppvSite
)
160 ok(0, "unexpected call\n");
164 static const IObjectWithSiteVtbl ObjectWithSiteVtbl
= {
165 ObjectWithSite_QueryInterface
,
166 ObjectWithSite_AddRef
,
167 ObjectWithSite_Release
,
168 ObjectWithSite_SetSite
,
169 ObjectWithSite_GetSite
172 static IObjectWithSite ObjectWithSite
= { &ObjectWithSiteVtbl
};
174 static IObjectWithSite
*object_with_site
;
176 static HRESULT WINAPI
DispatchEx_QueryInterface(IDispatchEx
*iface
, REFIID riid
, void **ppv
)
180 if(IsEqualGUID(riid
, &IID_IUnknown
)) {
182 }else if(IsEqualGUID(riid
, &IID_IDispatch
) || IsEqualGUID(riid
, &IID_IDispatchEx
)) {
183 if(FAILED(QI_IDispatch_hres
))
184 return QI_IDispatch_hres
;
186 }else if(IsEqualGUID(&IID_IObjectWithSite
, riid
)) {
187 CHECK_EXPECT(QI_IObjectWithSite
);
188 *ppv
= object_with_site
;
189 }else if(IsEqualGUID(&IID_IObjectSafety
, riid
)) {
190 ok(0, "Unexpected IID_IObjectSafety query\n");
193 return *ppv
? S_OK
: E_NOINTERFACE
;
196 static ULONG WINAPI
DispatchEx_AddRef(IDispatchEx
*iface
)
201 static ULONG WINAPI
DispatchEx_Release(IDispatchEx
*iface
)
206 static HRESULT WINAPI
DispatchEx_GetTypeInfoCount(IDispatchEx
*iface
, UINT
*pctinfo
)
208 ok(0, "unexpected call\n");
212 static HRESULT WINAPI
DispatchEx_GetTypeInfo(IDispatchEx
*iface
, UINT iTInfo
,
213 LCID lcid
, ITypeInfo
**ppTInfo
)
215 ok(0, "unexpected call\n");
219 static HRESULT WINAPI
DispatchEx_GetIDsOfNames(IDispatchEx
*iface
, REFIID riid
,
220 LPOLESTR
*rgszNames
, UINT cNames
,
221 LCID lcid
, DISPID
*rgDispId
)
223 ok(0, "unexpected call\n");
227 static HRESULT WINAPI
DispatchEx_Invoke(IDispatchEx
*iface
, DISPID dispIdMember
,
228 REFIID riid
, LCID lcid
, WORD wFlags
, DISPPARAMS
*pDispParams
,
229 VARIANT
*pVarResult
, EXCEPINFO
*pExcepInfo
, UINT
*puArgErr
)
231 ok(0, "unexpected call\n");
235 static HRESULT WINAPI
DispatchEx_DeleteMemberByName(IDispatchEx
*iface
, BSTR bstrName
, DWORD grfdex
)
237 ok(0, "unexpected call %s %x\n", wine_dbgstr_w(bstrName
), grfdex
);
241 static HRESULT WINAPI
DispatchEx_DeleteMemberByDispID(IDispatchEx
*iface
, DISPID id
)
243 ok(0, "unexpected call\n");
247 static HRESULT WINAPI
DispatchEx_GetMemberProperties(IDispatchEx
*iface
, DISPID id
, DWORD grfdexFetch
, DWORD
*pgrfdex
)
249 ok(0, "unexpected call\n");
253 static HRESULT WINAPI
DispatchEx_GetMemberName(IDispatchEx
*iface
, DISPID id
, BSTR
*pbstrName
)
255 ok(0, "unexpected call\n");
259 static HRESULT WINAPI
DispatchEx_GetNextDispID(IDispatchEx
*iface
, DWORD grfdex
, DISPID id
, DISPID
*pid
)
261 ok(0, "unexpected call\n");
265 static HRESULT WINAPI
DispatchEx_GetNameSpaceParent(IDispatchEx
*iface
, IUnknown
**ppunk
)
267 ok(0, "unexpected call\n");
271 static HRESULT WINAPI
Test_GetDispID(IDispatchEx
*iface
, BSTR bstrName
, DWORD grfdex
, DISPID
*pid
)
273 if(!strcmp_wa(bstrName
, "reportSuccess")) {
274 ok(grfdex
== fdexNameCaseInsensitive
, "grfdex = %x\n", grfdex
);
275 *pid
= DISPID_TEST_REPORTSUCCESS
;
279 ok(0, "unexpected name %s\n", wine_dbgstr_w(bstrName
));
283 static HRESULT WINAPI
Test_InvokeEx(IDispatchEx
*iface
, DISPID id
, LCID lcid
, WORD wFlags
, DISPPARAMS
*pdp
,
284 VARIANT
*pvarRes
, EXCEPINFO
*pei
, IServiceProvider
*pspCaller
)
287 case DISPID_TEST_REPORTSUCCESS
:
288 CHECK_EXPECT(reportSuccess
);
290 ok(wFlags
== INVOKE_FUNC
, "wFlags = %x\n", wFlags
);
291 ok(pdp
!= NULL
, "pdp == NULL\n");
292 ok(!pdp
->rgdispidNamedArgs
, "rgdispidNamedArgs != NULL\n");
293 ok(pdp
->cArgs
== 0, "cArgs = %d\n", pdp
->cArgs
);
294 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
295 ok(!pvarRes
, "pvarRes != NULL\n");
296 ok(pei
!= NULL
, "pei == NULL\n");
300 ok(0, "unexpected call\n");
307 static IDispatchExVtbl testObjVtbl
= {
308 DispatchEx_QueryInterface
,
311 DispatchEx_GetTypeInfoCount
,
312 DispatchEx_GetTypeInfo
,
313 DispatchEx_GetIDsOfNames
,
317 DispatchEx_DeleteMemberByName
,
318 DispatchEx_DeleteMemberByDispID
,
319 DispatchEx_GetMemberProperties
,
320 DispatchEx_GetMemberName
,
321 DispatchEx_GetNextDispID
,
322 DispatchEx_GetNameSpaceParent
325 static IDispatchEx testObj
= { &testObjVtbl
};
327 static HRESULT WINAPI
Global_GetDispID(IDispatchEx
*iface
, BSTR bstrName
, DWORD grfdex
, DISPID
*pid
)
329 if(!strcmp_wa(bstrName
, "ok")) {
330 ok(grfdex
== fdexNameCaseSensitive
, "grfdex = %x\n", grfdex
);
331 *pid
= DISPID_GLOBAL_OK
;
335 ok(0, "unexpected name %s\n", wine_dbgstr_w(bstrName
));
339 static HRESULT WINAPI
Global_InvokeEx(IDispatchEx
*iface
, DISPID id
, LCID lcid
, WORD wFlags
, DISPPARAMS
*pdp
,
340 VARIANT
*pvarRes
, EXCEPINFO
*pei
, IServiceProvider
*pspCaller
)
343 case DISPID_GLOBAL_OK
:
344 ok(wFlags
== INVOKE_FUNC
|| wFlags
== (INVOKE_FUNC
|INVOKE_PROPERTYGET
), "wFlags = %x\n", wFlags
);
345 ok(pdp
!= NULL
, "pdp == NULL\n");
346 ok(pdp
->rgvarg
!= NULL
, "rgvarg == NULL\n");
347 ok(!pdp
->rgdispidNamedArgs
, "rgdispidNamedArgs != NULL\n");
348 ok(pdp
->cArgs
== 2, "cArgs = %d\n", pdp
->cArgs
);
349 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
350 ok(pei
!= NULL
, "pei == NULL\n");
352 ok(V_VT(pdp
->rgvarg
) == VT_BSTR
, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp
->rgvarg
));
353 ok(V_VT(pdp
->rgvarg
+1) == VT_BOOL
, "V_VT(psp->rgvargs+1) = %d\n", V_VT(pdp
->rgvarg
));
354 ok(V_BOOL(pdp
->rgvarg
+1), "%s\n", wine_dbgstr_w(V_BSTR(pdp
->rgvarg
)));
358 ok(0, "unexpected call\n");
365 static IDispatchExVtbl globalObjVtbl
= {
366 DispatchEx_QueryInterface
,
369 DispatchEx_GetTypeInfoCount
,
370 DispatchEx_GetTypeInfo
,
371 DispatchEx_GetIDsOfNames
,
375 DispatchEx_DeleteMemberByName
,
376 DispatchEx_DeleteMemberByDispID
,
377 DispatchEx_GetMemberProperties
,
378 DispatchEx_GetMemberName
,
379 DispatchEx_GetNextDispID
,
380 DispatchEx_GetNameSpaceParent
383 static IDispatchEx globalObj
= { &globalObjVtbl
};
385 static HRESULT WINAPI
ClassFactory_QueryInterface(IClassFactory
*iface
, REFIID riid
, void **ppv
)
387 if(IsEqualGUID(&IID_IUnknown
, riid
) || IsEqualGUID(&IID_IClassFactory
, riid
)) {
392 /* TODO: IClassFactoryEx */
394 return E_NOINTERFACE
;
397 static ULONG WINAPI
ClassFactory_AddRef(IClassFactory
*iface
)
402 static ULONG WINAPI
ClassFactory_Release(IClassFactory
*iface
)
407 static HRESULT WINAPI
ClassFactory_CreateInstance(IClassFactory
*iface
, IUnknown
*outer
, REFIID riid
, void **ppv
)
409 CHECK_EXPECT(CreateInstance
);
411 ok(!outer
, "outer = %p\n", outer
);
412 ok(IsEqualGUID(&IID_IUnknown
, riid
), "unexpected riid %s\n", debugstr_guid(riid
));
414 if(SUCCEEDED(CreateInstance_hres
))
416 return CreateInstance_hres
;
419 static HRESULT WINAPI
ClassFactory_LockServer(IClassFactory
*iface
, BOOL dolock
)
421 ok(0, "unexpected call\n");
425 static const IClassFactoryVtbl ClassFactoryVtbl
= {
426 ClassFactory_QueryInterface
,
428 ClassFactory_Release
,
429 ClassFactory_CreateInstance
,
430 ClassFactory_LockServer
433 static IClassFactory activex_cf
= { &ClassFactoryVtbl
};
435 static HRESULT WINAPI
InternetHostSecurityManager_QueryInterface(IInternetHostSecurityManager
*iface
, REFIID riid
, void **ppv
)
437 ok(0, "unexpected call\n");
438 return E_NOINTERFACE
;
441 static ULONG WINAPI
InternetHostSecurityManager_AddRef(IInternetHostSecurityManager
*iface
)
446 static ULONG WINAPI
InternetHostSecurityManager_Release(IInternetHostSecurityManager
*iface
)
451 static HRESULT WINAPI
InternetHostSecurityManager_GetSecurityId(IInternetHostSecurityManager
*iface
, BYTE
*pbSecurityId
,
452 DWORD
*pcbSecurityId
, DWORD_PTR dwReserved
)
454 ok(0, "unexpected call\n");
458 static HRESULT WINAPI
InternetHostSecurityManager_ProcessUrlAction(IInternetHostSecurityManager
*iface
, DWORD dwAction
,
459 BYTE
*pPolicy
, DWORD cbPolicy
, BYTE
*pContext
, DWORD cbContext
, DWORD dwFlags
, DWORD dwReserved
)
461 CHECK_EXPECT(ProcessUrlAction
);
463 ok(dwAction
== URLACTION_ACTIVEX_RUN
, "dwAction = %x\n", dwAction
);
464 ok(pPolicy
!= NULL
, "pPolicy == NULL\n");
465 ok(cbPolicy
== sizeof(DWORD
), "cbPolicy = %d\n", cbPolicy
);
466 ok(pContext
!= NULL
, "pContext == NULL\n");
467 ok(cbContext
== sizeof(GUID
), "cbContext = %d\n", cbContext
);
468 ok(IsEqualGUID(pContext
, &CLSID_TestObj
), "pContext = %s\n", debugstr_guid((const IID
*)pContext
));
469 ok(!dwFlags
, "dwFlags = %x\n", dwFlags
);
470 ok(!dwReserved
, "dwReserved = %x\n", dwReserved
);
472 if(SUCCEEDED(ProcessUrlAction_hres
))
473 *(DWORD
*)pPolicy
= ProcessUrlAction_policy
;
474 return ProcessUrlAction_hres
;
477 static HRESULT WINAPI
InternetHostSecurityManager_QueryCustomPolicy(IInternetHostSecurityManager
*iface
, REFGUID guidKey
,
478 BYTE
**ppPolicy
, DWORD
*pcbPolicy
, BYTE
*pContext
, DWORD cbContext
, DWORD dwReserved
)
480 const struct CONFIRMSAFETY
*cs
= (const struct CONFIRMSAFETY
*)pContext
;
483 CHECK_EXPECT(QueryCustomPolicy
);
485 ok(IsEqualGUID(&GUID_CUSTOM_CONFIRMOBJECTSAFETY
, guidKey
), "guidKey = %s\n", debugstr_guid(guidKey
));
487 ok(ppPolicy
!= NULL
, "ppPolicy == NULL\n");
488 ok(pcbPolicy
!= NULL
, "pcbPolicy == NULL\n");
489 ok(pContext
!= NULL
, "pContext == NULL\n");
490 ok(cbContext
== sizeof(struct CONFIRMSAFETY
), "cbContext = %d\n", cbContext
);
491 ok(!dwReserved
, "dwReserved = %x\n", dwReserved
);
494 ok(cs
->pUnk
!= NULL
, "cs->pUnk == NULL\n");
495 ok(!cs
->dwFlags
, "dwFlags = %x\n", cs
->dwFlags
);
497 if(FAILED(QueryCustomPolicy_hres
))
498 return QueryCustomPolicy_hres
;
500 ret
= CoTaskMemAlloc(QueryCustomPolicy_psize
);
501 *ppPolicy
= (BYTE
*)ret
;
502 *pcbPolicy
= QueryCustomPolicy_psize
;
503 memset(ret
, 0, QueryCustomPolicy_psize
);
504 if(QueryCustomPolicy_psize
>= sizeof(DWORD
))
505 *ret
= QueryCustomPolicy_policy
;
507 return QueryCustomPolicy_hres
;
510 static const IInternetHostSecurityManagerVtbl InternetHostSecurityManagerVtbl
= {
511 InternetHostSecurityManager_QueryInterface
,
512 InternetHostSecurityManager_AddRef
,
513 InternetHostSecurityManager_Release
,
514 InternetHostSecurityManager_GetSecurityId
,
515 InternetHostSecurityManager_ProcessUrlAction
,
516 InternetHostSecurityManager_QueryCustomPolicy
519 static IInternetHostSecurityManager InternetHostSecurityManager
= { &InternetHostSecurityManagerVtbl
};
521 static IServiceProvider ServiceProvider
;
523 static HRESULT WINAPI
ServiceProvider_QueryInterface(IServiceProvider
*iface
, REFIID riid
, void **ppv
)
525 ok(0, "unexpected call\n");
526 return E_NOINTERFACE
;
529 static ULONG WINAPI
ServiceProvider_AddRef(IServiceProvider
*iface
)
534 static ULONG WINAPI
ServiceProvider_Release(IServiceProvider
*iface
)
539 static HRESULT WINAPI
ServiceProvider_QueryService(IServiceProvider
*iface
,
540 REFGUID guidService
, REFIID riid
, void **ppv
)
542 if(IsEqualGUID(&SID_GetCaller
, guidService
))
543 return E_NOINTERFACE
;
545 if(IsEqualGUID(&SID_SInternetHostSecurityManager
, guidService
)) {
546 if(iface
== &ServiceProvider
)
547 CHECK_EXPECT(Host_QS_SecMgr
);
549 CHECK_EXPECT(Caller_QS_SecMgr
);
550 ok(IsEqualGUID(&IID_IInternetHostSecurityManager
, riid
), "unexpected riid %s\n", debugstr_guid(riid
));
551 if(SUCCEEDED(QS_SecMgr_hres
))
552 *ppv
= &InternetHostSecurityManager
;
553 return QS_SecMgr_hres
;
556 ok(0, "unexpected service %s\n", debugstr_guid(guidService
));
557 return E_NOINTERFACE
;
560 static IServiceProviderVtbl ServiceProviderVtbl
= {
561 ServiceProvider_QueryInterface
,
562 ServiceProvider_AddRef
,
563 ServiceProvider_Release
,
564 ServiceProvider_QueryService
567 static IServiceProvider ServiceProvider
= { &ServiceProviderVtbl
};
569 static HRESULT WINAPI
ActiveScriptSite_QueryInterface(IActiveScriptSite
*iface
, REFIID riid
, void **ppv
)
571 if(IsEqualGUID(&IID_IUnknown
, riid
)) {
573 }else if(IsEqualGUID(&IID_IActiveScriptSite
, riid
)) {
575 }else if(IsEqualGUID(&IID_IServiceProvider
, riid
)) {
576 *ppv
= &ServiceProvider
;
579 return E_NOINTERFACE
;
582 IUnknown_AddRef((IUnknown
*)*ppv
);
586 static ULONG WINAPI
ActiveScriptSite_AddRef(IActiveScriptSite
*iface
)
591 static ULONG WINAPI
ActiveScriptSite_Release(IActiveScriptSite
*iface
)
596 static HRESULT WINAPI
ActiveScriptSite_GetLCID(IActiveScriptSite
*iface
, LCID
*plcid
)
598 *plcid
= GetUserDefaultLCID();
602 static HRESULT WINAPI
ActiveScriptSite_GetItemInfo(IActiveScriptSite
*iface
, LPCOLESTR pstrName
,
603 DWORD dwReturnMask
, IUnknown
**ppiunkItem
, ITypeInfo
**ppti
)
605 ok(dwReturnMask
== SCRIPTINFO_IUNKNOWN
, "unexpected dwReturnMask %x\n", dwReturnMask
);
606 ok(!ppti
, "ppti != NULL\n");
607 ok(!strcmp_wa(pstrName
, "test"), "pstrName = %s\n", wine_dbgstr_w(pstrName
));
609 *ppiunkItem
= (IUnknown
*)&globalObj
;
613 static HRESULT WINAPI
ActiveScriptSite_GetDocVersionString(IActiveScriptSite
*iface
, BSTR
*pbstrVersion
)
618 static HRESULT WINAPI
ActiveScriptSite_OnScriptTerminate(IActiveScriptSite
*iface
,
619 const VARIANT
*pvarResult
, const EXCEPINFO
*pexcepinfo
)
624 static HRESULT WINAPI
ActiveScriptSite_OnStateChange(IActiveScriptSite
*iface
, SCRIPTSTATE ssScriptState
)
629 static HRESULT WINAPI
ActiveScriptSite_OnScriptError(IActiveScriptSite
*iface
, IActiveScriptError
*pscripterror
)
634 static HRESULT WINAPI
ActiveScriptSite_OnEnterScript(IActiveScriptSite
*iface
)
639 static HRESULT WINAPI
ActiveScriptSite_OnLeaveScript(IActiveScriptSite
*iface
)
644 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl
= {
645 ActiveScriptSite_QueryInterface
,
646 ActiveScriptSite_AddRef
,
647 ActiveScriptSite_Release
,
648 ActiveScriptSite_GetLCID
,
649 ActiveScriptSite_GetItemInfo
,
650 ActiveScriptSite_GetDocVersionString
,
651 ActiveScriptSite_OnScriptTerminate
,
652 ActiveScriptSite_OnStateChange
,
653 ActiveScriptSite_OnScriptError
,
654 ActiveScriptSite_OnEnterScript
,
655 ActiveScriptSite_OnLeaveScript
658 static IActiveScriptSite ActiveScriptSite
= { &ActiveScriptSiteVtbl
};
660 static void set_safety_options(IUnknown
*unk
, BOOL use_sec_mgr
)
662 IObjectSafety
*safety
;
663 DWORD supported
, enabled
, options_all
, options_set
;
666 hres
= IUnknown_QueryInterface(unk
, &IID_IObjectSafety
, (void**)&safety
);
667 ok(hres
== S_OK
, "Could not get IObjectSafety: %08x\n", hres
);
671 options_all
= INTERFACESAFE_FOR_UNTRUSTED_DATA
|INTERFACE_USES_DISPEX
|INTERFACE_USES_SECURITY_MANAGER
;
673 options_set
= options_all
;
675 options_set
= INTERFACE_USES_DISPEX
;
677 hres
= IObjectSafety_SetInterfaceSafetyOptions(safety
, &IID_IActiveScriptParse
, options_all
, options_set
);
678 ok(hres
== S_OK
, "SetInterfaceSafetyOptions failed: %08x\n", hres
);
680 supported
= enabled
= 0xdeadbeef;
681 hres
= IObjectSafety_GetInterfaceSafetyOptions(safety
, &IID_IActiveScriptParse
, &supported
, &enabled
);
682 ok(hres
== S_OK
, "GetInterfaceSafetyOptions failed: %08x\n", hres
);
683 ok(supported
== options_all
, "supported=%x, expected %x\n", supported
, options_all
);
684 ok(enabled
== options_set
, "enabled=%x, expected %x\n", enabled
, options_set
);
686 IObjectSafety_Release(safety
);
689 #define parse_script_a(p,s) _parse_script_a(__LINE__,p,s)
690 static void _parse_script_a(unsigned line
, IActiveScriptParse
*parser
, const char *script
)
695 str
= a2bstr(script
);
696 hres
= IActiveScriptParse64_ParseScriptText(parser
, str
, NULL
, NULL
, NULL
, 0, 0, 0, NULL
, NULL
);
698 ok_(__FILE__
,line
)(hres
== S_OK
, "ParseScriptText failed: %08x\n", hres
);
701 static HRESULT
parse_script_ae(IActiveScriptParse
*parser
, const char *script
)
706 str
= a2bstr(script
);
707 hres
= IActiveScriptParse64_ParseScriptText(parser
, str
, NULL
, NULL
, NULL
, 0, 0, 0, NULL
, NULL
);
713 static IActiveScriptParse
*create_script(BOOL use_sec_mgr
)
715 IActiveScriptParse
*parser
;
716 IActiveScript
*script
;
719 QS_SecMgr_hres
= S_OK
;
720 ProcessUrlAction_hres
= S_OK
;
721 ProcessUrlAction_policy
= URLPOLICY_ALLOW
;
722 CreateInstance_hres
= S_OK
;
723 QueryCustomPolicy_hres
= S_OK
;
724 QueryCustomPolicy_psize
= sizeof(DWORD
);
725 QueryCustomPolicy_policy
= URLPOLICY_ALLOW
;
726 QI_IDispatch_hres
= S_OK
;
729 hres
= CoCreateInstance(&CLSID_VBScript
, NULL
, CLSCTX_INPROC_SERVER
|CLSCTX_INPROC_HANDLER
,
730 &IID_IActiveScript
, (void**)&script
);
731 ok(hres
== S_OK
, "CoCreateInstance failed: %08x\n", hres
);
735 set_safety_options((IUnknown
*)script
, use_sec_mgr
);
737 hres
= IActiveScript_QueryInterface(script
, &IID_IActiveScriptParse
, (void**)&parser
);
738 ok(hres
== S_OK
, "Could not get IActiveScriptParse: %08x\n", hres
);
740 hres
= IActiveScriptParse64_InitNew(parser
);
741 ok(hres
== S_OK
, "InitNew failed: %08x\n", hres
);
743 hres
= IActiveScript_SetScriptSite(script
, &ActiveScriptSite
);
744 ok(hres
== S_OK
, "SetScriptSite failed: %08x\n", hres
);
746 hres
= IActiveScript_AddNamedItem(script
, testW
,
747 SCRIPTITEM_ISVISIBLE
|SCRIPTITEM_ISSOURCE
|SCRIPTITEM_GLOBALMEMBERS
);
748 ok(hres
== S_OK
, "AddNamedItem failed: %08x\n", hres
);
750 hres
= IActiveScript_SetScriptState(script
, SCRIPTSTATE_STARTED
);
751 ok(hres
== S_OK
, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres
);
753 IActiveScript_Release(script
);
758 static void test_CreateObject(void)
760 IActiveScriptParse
*parser
;
763 parser
= create_script(TRUE
);
765 SET_EXPECT(Host_QS_SecMgr
);
766 SET_EXPECT(ProcessUrlAction
);
767 SET_EXPECT(CreateInstance
);
768 SET_EXPECT(QueryCustomPolicy
);
769 SET_EXPECT(QI_IObjectWithSite
);
770 SET_EXPECT(reportSuccess
);
771 parse_script_a(parser
, "Call CreateObject(\"Wine.Test\").reportSuccess()");
772 CHECK_CALLED(Host_QS_SecMgr
);
773 CHECK_CALLED(ProcessUrlAction
);
774 CHECK_CALLED(CreateInstance
);
775 CHECK_CALLED(QueryCustomPolicy
);
776 CHECK_CALLED(QI_IObjectWithSite
);
777 CHECK_CALLED(reportSuccess
);
779 IUnknown_Release(parser
);
781 parser
= create_script(TRUE
);
783 hres
= parse_script_ae(parser
, "Call CreateObject(\"Wine.TestABC\")");
784 ok(hres
== VB_E_CANNOT_CREATE_OBJ
, "hres = %08x\n", hres
);
786 IUnknown_Release(parser
);
788 parser
= create_script(TRUE
);
789 QS_SecMgr_hres
= E_NOINTERFACE
;
791 SET_EXPECT(Host_QS_SecMgr
);
792 hres
= parse_script_ae(parser
, "Call CreateObject(\"Wine.Test\")");
793 ok(hres
== VB_E_CANNOT_CREATE_OBJ
, "hres = %08x\n", hres
);
794 CHECK_CALLED(Host_QS_SecMgr
);
796 IUnknown_Release(parser
);
798 parser
= create_script(TRUE
);
799 ProcessUrlAction_hres
= E_FAIL
;
801 SET_EXPECT(Host_QS_SecMgr
);
802 SET_EXPECT(ProcessUrlAction
);
803 hres
= parse_script_ae(parser
, "Call CreateObject(\"Wine.Test\")");
804 ok(hres
== VB_E_CANNOT_CREATE_OBJ
, "hres = %08x\n", hres
);
805 CHECK_CALLED(Host_QS_SecMgr
);
806 CHECK_CALLED(ProcessUrlAction
);
808 IUnknown_Release(parser
);
810 parser
= create_script(TRUE
);
811 ProcessUrlAction_policy
= URLPOLICY_DISALLOW
;
813 SET_EXPECT(Host_QS_SecMgr
);
814 SET_EXPECT(ProcessUrlAction
);
815 hres
= parse_script_ae(parser
, "Call CreateObject(\"Wine.Test\")");
816 ok(hres
== VB_E_CANNOT_CREATE_OBJ
, "hres = %08x\n", hres
);
817 CHECK_CALLED(Host_QS_SecMgr
);
818 CHECK_CALLED(ProcessUrlAction
);
820 IUnknown_Release(parser
);
822 parser
= create_script(TRUE
);
823 CreateInstance_hres
= E_FAIL
;
825 SET_EXPECT(Host_QS_SecMgr
);
826 SET_EXPECT(ProcessUrlAction
);
827 SET_EXPECT(CreateInstance
);
828 hres
= parse_script_ae(parser
, "Call CreateObject(\"Wine.Test\")");
829 ok(hres
== VB_E_CANNOT_CREATE_OBJ
, "hres = %08x\n", hres
);
830 CHECK_CALLED(Host_QS_SecMgr
);
831 CHECK_CALLED(ProcessUrlAction
);
832 CHECK_CALLED(CreateInstance
);
834 IUnknown_Release(parser
);
836 parser
= create_script(TRUE
);
837 QueryCustomPolicy_hres
= E_FAIL
;
839 SET_EXPECT(Host_QS_SecMgr
);
840 SET_EXPECT(ProcessUrlAction
);
841 SET_EXPECT(CreateInstance
);
842 SET_EXPECT(QueryCustomPolicy
);
843 hres
= parse_script_ae(parser
, "Call CreateObject(\"Wine.Test\")");
844 ok(hres
== VB_E_CANNOT_CREATE_OBJ
, "hres = %08x\n", hres
);
845 CHECK_CALLED(Host_QS_SecMgr
);
846 CHECK_CALLED(ProcessUrlAction
);
847 CHECK_CALLED(CreateInstance
);
848 CHECK_CALLED(QueryCustomPolicy
);
850 IUnknown_Release(parser
);
852 parser
= create_script(TRUE
);
853 QueryCustomPolicy_psize
= 6;
855 SET_EXPECT(Host_QS_SecMgr
);
856 SET_EXPECT(ProcessUrlAction
);
857 SET_EXPECT(CreateInstance
);
858 SET_EXPECT(QueryCustomPolicy
);
859 SET_EXPECT(QI_IObjectWithSite
);
860 SET_EXPECT(reportSuccess
);
861 parse_script_a(parser
, "Call CreateObject(\"Wine.Test\").reportSuccess()");
862 CHECK_CALLED(Host_QS_SecMgr
);
863 CHECK_CALLED(ProcessUrlAction
);
864 CHECK_CALLED(CreateInstance
);
865 CHECK_CALLED(QueryCustomPolicy
);
866 CHECK_CALLED(QI_IObjectWithSite
);
867 CHECK_CALLED(reportSuccess
);
869 IUnknown_Release(parser
);
871 parser
= create_script(TRUE
);
872 QueryCustomPolicy_policy
= URLPOLICY_DISALLOW
;
874 SET_EXPECT(Host_QS_SecMgr
);
875 SET_EXPECT(ProcessUrlAction
);
876 SET_EXPECT(CreateInstance
);
877 SET_EXPECT(QueryCustomPolicy
);
878 hres
= parse_script_ae(parser
, "Call CreateObject(\"Wine.Test\")");
879 ok(hres
== VB_E_CANNOT_CREATE_OBJ
, "hres = %08x\n", hres
);
880 CHECK_CALLED(Host_QS_SecMgr
);
881 CHECK_CALLED(ProcessUrlAction
);
882 CHECK_CALLED(CreateInstance
);
883 CHECK_CALLED(QueryCustomPolicy
);
885 QueryCustomPolicy_psize
= 6;
887 SET_EXPECT(ProcessUrlAction
);
888 SET_EXPECT(CreateInstance
);
889 SET_EXPECT(QueryCustomPolicy
);
890 hres
= parse_script_ae(parser
, "Call CreateObject(\"Wine.Test\")");
891 ok(hres
== VB_E_CANNOT_CREATE_OBJ
, "hres = %08x\n", hres
);
892 CHECK_CALLED(ProcessUrlAction
);
893 CHECK_CALLED(CreateInstance
);
894 CHECK_CALLED(QueryCustomPolicy
);
896 QueryCustomPolicy_policy
= URLPOLICY_ALLOW
;
897 QueryCustomPolicy_psize
= 3;
899 SET_EXPECT(ProcessUrlAction
);
900 SET_EXPECT(CreateInstance
);
901 SET_EXPECT(QueryCustomPolicy
);
902 hres
= parse_script_ae(parser
, "Call CreateObject(\"Wine.Test\")");
903 ok(hres
== VB_E_CANNOT_CREATE_OBJ
, "hres = %08x\n", hres
);
904 CHECK_CALLED(ProcessUrlAction
);
905 CHECK_CALLED(CreateInstance
);
906 CHECK_CALLED(QueryCustomPolicy
);
908 IUnknown_Release(parser
);
910 parser
= create_script(FALSE
);
912 SET_EXPECT(CreateInstance
);
913 SET_EXPECT(QI_IObjectWithSite
);
914 SET_EXPECT(reportSuccess
);
915 parse_script_a(parser
, "Call CreateObject(\"Wine.Test\").reportSuccess()");
916 CHECK_CALLED(CreateInstance
);
917 CHECK_CALLED(QI_IObjectWithSite
);
918 CHECK_CALLED(reportSuccess
);
920 IUnknown_Release(parser
);
922 parser
= create_script(TRUE
);
923 object_with_site
= &ObjectWithSite
;
925 SET_EXPECT(Host_QS_SecMgr
);
926 SET_EXPECT(ProcessUrlAction
);
927 SET_EXPECT(CreateInstance
);
928 SET_EXPECT(QueryCustomPolicy
);
929 SET_EXPECT(QI_IObjectWithSite
);
931 SET_EXPECT(reportSuccess
);
932 parse_script_a(parser
, "Call CreateObject(\"Wine.Test\").reportSuccess()");
933 CHECK_CALLED(Host_QS_SecMgr
);
934 CHECK_CALLED(ProcessUrlAction
);
935 CHECK_CALLED(CreateInstance
);
936 CHECK_CALLED(QueryCustomPolicy
);
937 CHECK_CALLED(QI_IObjectWithSite
);
938 CHECK_CALLED(SetSite
);
939 CHECK_CALLED(reportSuccess
);
941 SetSite_hres
= E_FAIL
;
942 SET_EXPECT(ProcessUrlAction
);
943 SET_EXPECT(CreateInstance
);
944 SET_EXPECT(QueryCustomPolicy
);
945 SET_EXPECT(QI_IObjectWithSite
);
947 hres
= parse_script_ae(parser
, "Call CreateObject(\"Wine.Test\")");
948 ok(hres
== VB_E_CANNOT_CREATE_OBJ
, "hres = %08x\n", hres
);
949 CHECK_CALLED(ProcessUrlAction
);
950 CHECK_CALLED(CreateInstance
);
951 CHECK_CALLED(QueryCustomPolicy
);
952 CHECK_CALLED(QI_IObjectWithSite
);
953 CHECK_CALLED(SetSite
);
955 IUnknown_Release(parser
);
958 static BOOL
init_key(const char *key_name
, const char *def_value
, BOOL init
)
964 RegDeleteKey(HKEY_CLASSES_ROOT
, key_name
);
968 res
= RegCreateKeyA(HKEY_CLASSES_ROOT
, key_name
, &hkey
);
969 if(res
!= ERROR_SUCCESS
)
973 res
= RegSetValueA(hkey
, NULL
, REG_SZ
, def_value
, strlen(def_value
));
977 return res
== ERROR_SUCCESS
;
980 static BOOL
init_registry(BOOL init
)
982 return init_key("Wine.Test\\CLSID", TESTOBJ_CLSID
, init
);
985 static BOOL
register_activex(void)
990 if(!init_registry(TRUE
)) {
991 init_registry(FALSE
);
995 hres
= CoRegisterClassObject(&CLSID_TestObj
, (IUnknown
*)&activex_cf
,
996 CLSCTX_INPROC_SERVER
, REGCLS_MULTIPLEUSE
, ®id
);
997 ok(hres
== S_OK
, "Could not register script engine: %08x\n", hres
);
1002 START_TEST(createobj
)
1008 test_CreateObject();
1010 init_registry(FALSE
);