comctl32: Only support an alpha channel for 32-bpp DIB-based imagelists.
[wine] / dlls / jscript / tests / activex.c
1 /*
2  * Copyright 2009 Jacek Caban for CodeWeavers
3  *
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.
8  *
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.
13  *
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
17  */
18
19 #include <stdio.h>
20
21 #define COBJMACROS
22 #define CONST_VTABLE
23
24 #include <ole2.h>
25 #include <dispex.h>
26 #include <activscp.h>
27 #include <objsafe.h>
28 #include <urlmon.h>
29 #include <mshtmhst.h>
30
31 #include "wine/test.h"
32
33 static const CLSID CLSID_JScript =
34     {0xf414c260,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};
35
36 #define DEFINE_EXPECT(func) \
37     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
38
39 #define SET_EXPECT(func) \
40     expect_ ## func = TRUE
41
42 #define SET_CALLED(func) \
43     called_ ## func = TRUE
44
45 #define CHECK_EXPECT2(func) \
46     do { \
47         ok(expect_ ##func, "unexpected call " #func "\n"); \
48         called_ ## func = TRUE; \
49     }while(0)
50
51 #define CHECK_EXPECT(func) \
52     do { \
53         CHECK_EXPECT2(func); \
54         expect_ ## func = FALSE; \
55     }while(0)
56
57 #define CHECK_CALLED(func) \
58     do { \
59         ok(called_ ## func, "expected " #func "\n"); \
60         expect_ ## func = called_ ## func = FALSE; \
61     }while(0)
62
63 DEFINE_EXPECT(CreateInstance);
64 DEFINE_EXPECT(ProcessUrlAction);
65 DEFINE_EXPECT(QueryCustomPolicy);
66 DEFINE_EXPECT(reportSuccess);
67 DEFINE_EXPECT(Host_QS_SecMgr);
68 DEFINE_EXPECT(Caller_QS_SecMgr);
69 DEFINE_EXPECT(QI_IObjectWithSite);
70 DEFINE_EXPECT(SetSite);
71
72 static const WCHAR testW[] = {'t','e','s','t',0};
73
74 static HRESULT QS_SecMgr_hres;
75 static HRESULT ProcessUrlAction_hres;
76 static DWORD ProcessUrlAction_policy;
77 static HRESULT CreateInstance_hres;
78 static HRESULT QueryCustomPolicy_hres;
79 static DWORD QueryCustomPolicy_psize;
80 static DWORD QueryCustomPolicy_policy;
81 static HRESULT QI_IDispatch_hres;
82 static HRESULT SetSite_hres;
83
84 #define TESTOBJ_CLSID "{178fc163-f585-4e24-9c13-4bb7faf80646}"
85
86 static const GUID CLSID_TestObj =
87     {0x178fc163,0xf585,0x4e24,{0x9c,0x13,0x4b,0xb7,0xfa,0xf8,0x06,0x46}};
88
89 /* Defined as extern in urlmon.idl, but not exported by uuid.lib */
90 const GUID GUID_CUSTOM_CONFIRMOBJECTSAFETY =
91     {0x10200490,0xfa38,0x11d0,{0xac,0x0e,0x00,0xa0,0xc9,0xf,0xff,0xc0}};
92
93 #define DISPID_TEST_REPORTSUCCESS    0x1000
94
95 #define DISPID_GLOBAL_OK             0x2000
96
97 static const char *debugstr_guid(REFIID riid)
98 {
99     static char buf[50];
100
101     sprintf(buf, "{%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",
102             riid->Data1, riid->Data2, riid->Data3, riid->Data4[0],
103             riid->Data4[1], riid->Data4[2], riid->Data4[3], riid->Data4[4],
104             riid->Data4[5], riid->Data4[6], riid->Data4[7]);
105
106     return buf;
107 }
108
109 static BSTR a2bstr(const char *str)
110 {
111     BSTR ret;
112     int len;
113
114     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
115     ret = SysAllocStringLen(NULL, len-1);
116     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
117
118     return ret;
119 }
120
121 static int strcmp_wa(LPCWSTR strw, const char *stra)
122 {
123     CHAR buf[512];
124     WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), 0, 0);
125     return lstrcmpA(buf, stra);
126 }
127
128 static HRESULT WINAPI ObjectWithSite_QueryInterface(IObjectWithSite *iface, REFIID riid, void **ppv)
129 {
130     ok(0, "unexpected call\n");
131     return E_NOTIMPL;
132 }
133
134 static ULONG WINAPI ObjectWithSite_AddRef(IObjectWithSite *iface)
135 {
136     return 2;
137 }
138
139 static ULONG WINAPI ObjectWithSite_Release(IObjectWithSite *iface)
140 {
141     return 1;
142 }
143
144 static HRESULT WINAPI ObjectWithSite_SetSite(IObjectWithSite *iface, IUnknown *pUnkSite)
145 {
146     IServiceProvider *sp;
147     HRESULT hres;
148
149
150     CHECK_EXPECT(SetSite);
151     ok(pUnkSite != NULL, "pUnkSite == NULL\n");
152
153     hres = IUnknown_QueryInterface(pUnkSite, &IID_IServiceProvider, (void**)&sp);
154     ok(hres == S_OK, "Could not get IServiceProvider iface: %08x\n", hres);
155     IServiceProvider_Release(sp);
156
157     return SetSite_hres;
158 }
159
160 static HRESULT WINAPI ObjectWithSite_GetSite(IObjectWithSite *iface, REFIID riid, void **ppvSite)
161 {
162     ok(0, "unexpected call\n");
163     return E_NOTIMPL;
164 }
165
166 static const IObjectWithSiteVtbl ObjectWithSiteVtbl = {
167     ObjectWithSite_QueryInterface,
168     ObjectWithSite_AddRef,
169     ObjectWithSite_Release,
170     ObjectWithSite_SetSite,
171     ObjectWithSite_GetSite
172 };
173
174 static IObjectWithSite ObjectWithSite = { &ObjectWithSiteVtbl };
175
176 static IObjectWithSite *object_with_site;
177
178 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
179 {
180     *ppv = NULL;
181
182     if(IsEqualGUID(riid, &IID_IUnknown)) {
183        *ppv = iface;
184     }else if(IsEqualGUID(riid, &IID_IDispatch) || IsEqualGUID(riid, &IID_IDispatchEx)) {
185         if(FAILED(QI_IDispatch_hres))
186             return QI_IDispatch_hres;
187         *ppv = iface;
188     }else if(IsEqualGUID(&IID_IObjectWithSite, riid)) {
189         CHECK_EXPECT(QI_IObjectWithSite);
190         *ppv = object_with_site;
191     }else {
192         return E_NOINTERFACE;
193     }
194
195     return *ppv ? S_OK : E_NOINTERFACE;
196 }
197
198 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
199 {
200     return 2;
201 }
202
203 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
204 {
205     return 1;
206 }
207
208 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
209 {
210     ok(0, "unexpected call\n");
211     return E_NOTIMPL;
212 }
213
214 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
215                                               LCID lcid, ITypeInfo **ppTInfo)
216 {
217     ok(0, "unexpected call\n");
218     return E_NOTIMPL;
219 }
220
221 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
222                                                 LPOLESTR *rgszNames, UINT cNames,
223                                                 LCID lcid, DISPID *rgDispId)
224 {
225     ok(0, "unexpected call\n");
226     return E_NOTIMPL;
227 }
228
229 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
230                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
231                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
232 {
233     ok(0, "unexpected call\n");
234     return E_NOTIMPL;
235 }
236
237 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
238 {
239     ok(0, "unexpected call %s %x\n", wine_dbgstr_w(bstrName), grfdex);
240     return E_NOTIMPL;
241 }
242
243 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
244 {
245     ok(0, "unexpected call\n");
246     return E_NOTIMPL;
247 }
248
249 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
250 {
251     ok(0, "unexpected call\n");
252     return E_NOTIMPL;
253 }
254
255 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
256 {
257     ok(0, "unexpected call\n");
258     return E_NOTIMPL;
259 }
260
261 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
262 {
263     ok(0, "unexpected call\n");
264     return E_NOTIMPL;
265 }
266
267 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
268 {
269     ok(0, "unexpected call\n");
270     return E_NOTIMPL;
271 }
272
273 static HRESULT WINAPI Test_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
274 {
275     if(!strcmp_wa(bstrName, "reportSuccess")) {
276         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
277         *pid = DISPID_TEST_REPORTSUCCESS;
278         return S_OK;
279     }
280
281     ok(0, "unexpected name %s\n", wine_dbgstr_w(bstrName));
282     return E_NOTIMPL;
283 }
284
285 static HRESULT WINAPI Test_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
286         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
287 {
288     switch(id) {
289     case DISPID_TEST_REPORTSUCCESS:
290         CHECK_EXPECT(reportSuccess);
291
292         ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
293         ok(pdp != NULL, "pdp == NULL\n");
294         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
295         ok(pdp->cArgs == 0, "cArgs = %d\n", pdp->cArgs);
296         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
297         ok(!pvarRes, "pvarRes != NULL\n");
298         ok(pei != NULL, "pei == NULL\n");
299         break;
300
301     default:
302         ok(0, "unexpected call\n");
303         return E_NOTIMPL;
304     }
305
306     return S_OK;
307 }
308
309 static IDispatchExVtbl testObjVtbl = {
310     DispatchEx_QueryInterface,
311     DispatchEx_AddRef,
312     DispatchEx_Release,
313     DispatchEx_GetTypeInfoCount,
314     DispatchEx_GetTypeInfo,
315     DispatchEx_GetIDsOfNames,
316     DispatchEx_Invoke,
317     Test_GetDispID,
318     Test_InvokeEx,
319     DispatchEx_DeleteMemberByName,
320     DispatchEx_DeleteMemberByDispID,
321     DispatchEx_GetMemberProperties,
322     DispatchEx_GetMemberName,
323     DispatchEx_GetNextDispID,
324     DispatchEx_GetNameSpaceParent
325 };
326
327 static IDispatchEx testObj = { &testObjVtbl };
328
329 static HRESULT WINAPI Global_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
330 {
331     if(!strcmp_wa(bstrName, "ok")) {
332         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
333         *pid = DISPID_GLOBAL_OK;
334         return S_OK;
335     }
336
337     ok(0, "unexpected name %s\n", wine_dbgstr_w(bstrName));
338     return E_NOTIMPL;
339 }
340
341 static HRESULT WINAPI Global_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
342         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
343 {
344     switch(id) {
345     case DISPID_GLOBAL_OK:
346         ok(wFlags == INVOKE_FUNC || wFlags == (INVOKE_FUNC|INVOKE_PROPERTYGET), "wFlags = %x\n", wFlags);
347         ok(pdp != NULL, "pdp == NULL\n");
348         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
349         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
350         ok(pdp->cArgs == 2, "cArgs = %d\n", pdp->cArgs);
351         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
352         ok(pei != NULL, "pei == NULL\n");
353
354         ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
355         ok(V_VT(pdp->rgvarg+1) == VT_BOOL, "V_VT(psp->rgvargs+1) = %d\n", V_VT(pdp->rgvarg));
356         ok(V_BOOL(pdp->rgvarg+1), "%s\n", wine_dbgstr_w(V_BSTR(pdp->rgvarg)));
357         break;
358
359     default:
360         ok(0, "unexpected call\n");
361         return E_NOTIMPL;
362     }
363
364     return S_OK;
365 }
366
367 static IDispatchExVtbl globalObjVtbl = {
368     DispatchEx_QueryInterface,
369     DispatchEx_AddRef,
370     DispatchEx_Release,
371     DispatchEx_GetTypeInfoCount,
372     DispatchEx_GetTypeInfo,
373     DispatchEx_GetIDsOfNames,
374     DispatchEx_Invoke,
375     Global_GetDispID,
376     Global_InvokeEx,
377     DispatchEx_DeleteMemberByName,
378     DispatchEx_DeleteMemberByDispID,
379     DispatchEx_GetMemberProperties,
380     DispatchEx_GetMemberName,
381     DispatchEx_GetNextDispID,
382     DispatchEx_GetNameSpaceParent
383 };
384
385 static IDispatchEx globalObj = { &globalObjVtbl };
386
387 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
388 {
389     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IClassFactory, riid)) {
390         *ppv = iface;
391         return S_OK;
392     }
393
394     /* TODO: IClassFactoryEx */
395     *ppv = NULL;
396     return E_NOINTERFACE;
397 }
398
399 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
400 {
401     return 2;
402 }
403
404 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
405 {
406     return 1;
407 }
408
409 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *outer, REFIID riid, void **ppv)
410 {
411     CHECK_EXPECT(CreateInstance);
412
413     ok(!outer, "outer = %p\n", outer);
414     ok(IsEqualGUID(&IID_IUnknown, riid), "unexpected riid %s\n", debugstr_guid(riid));
415
416     if(SUCCEEDED(CreateInstance_hres))
417         *ppv = &testObj;
418     return CreateInstance_hres;
419 }
420
421 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
422 {
423     ok(0, "unexpected call\n");
424     return S_OK;
425 }
426
427 static const IClassFactoryVtbl ClassFactoryVtbl = {
428     ClassFactory_QueryInterface,
429     ClassFactory_AddRef,
430     ClassFactory_Release,
431     ClassFactory_CreateInstance,
432     ClassFactory_LockServer
433 };
434
435 static IClassFactory activex_cf = { &ClassFactoryVtbl };
436
437 static HRESULT WINAPI InternetHostSecurityManager_QueryInterface(IInternetHostSecurityManager *iface, REFIID riid, void **ppv)
438 {
439     ok(0, "unexpected call\n");
440     return E_NOINTERFACE;
441 }
442
443 static ULONG WINAPI InternetHostSecurityManager_AddRef(IInternetHostSecurityManager *iface)
444 {
445     return 2;
446 }
447
448 static ULONG WINAPI InternetHostSecurityManager_Release(IInternetHostSecurityManager *iface)
449 {
450     return 1;
451 }
452
453 static HRESULT WINAPI InternetHostSecurityManager_GetSecurityId(IInternetHostSecurityManager *iface,  BYTE *pbSecurityId,
454         DWORD *pcbSecurityId, DWORD_PTR dwReserved)
455 {
456     ok(0, "unexpected call\n");
457     return E_NOTIMPL;
458 }
459
460 static HRESULT WINAPI InternetHostSecurityManager_ProcessUrlAction(IInternetHostSecurityManager *iface, DWORD dwAction,
461         BYTE *pPolicy, DWORD cbPolicy, BYTE *pContext, DWORD cbContext, DWORD dwFlags, DWORD dwReserved)
462 {
463     CHECK_EXPECT(ProcessUrlAction);
464
465     ok(dwAction == URLACTION_ACTIVEX_RUN, "dwAction = %x\n", dwAction);
466     ok(pPolicy != NULL, "pPolicy == NULL\n");
467     ok(cbPolicy == sizeof(DWORD), "cbPolicy = %d\n", cbPolicy);
468     ok(pContext != NULL, "pContext == NULL\n");
469     ok(cbContext == sizeof(GUID), "cbContext = %d\n", cbContext);
470     ok(IsEqualGUID(pContext, &CLSID_TestObj), "pContext = %s\n", debugstr_guid((const IID*)pContext));
471     ok(!dwFlags, "dwFlags = %x\n", dwFlags);
472     ok(!dwReserved, "dwReserved = %x\n", dwReserved);
473
474     if(SUCCEEDED(ProcessUrlAction_hres))
475         *(DWORD*)pPolicy = ProcessUrlAction_policy;
476     return ProcessUrlAction_hres;
477 }
478
479 static HRESULT WINAPI InternetHostSecurityManager_QueryCustomPolicy(IInternetHostSecurityManager *iface, REFGUID guidKey,
480         BYTE **ppPolicy, DWORD *pcbPolicy, BYTE *pContext, DWORD cbContext, DWORD dwReserved)
481 {
482     const struct CONFIRMSAFETY *cs = (const struct CONFIRMSAFETY*)pContext;
483     DWORD *ret;
484
485     CHECK_EXPECT(QueryCustomPolicy);
486
487     ok(IsEqualGUID(&GUID_CUSTOM_CONFIRMOBJECTSAFETY, guidKey), "guidKey = %s\n", debugstr_guid(guidKey));
488
489     ok(ppPolicy != NULL, "ppPolicy == NULL\n");
490     ok(pcbPolicy != NULL, "pcbPolicy == NULL\n");
491     ok(pContext != NULL, "pContext == NULL\n");
492     ok(cbContext == sizeof(struct CONFIRMSAFETY), "cbContext = %d\n", cbContext);
493     ok(!dwReserved, "dwReserved = %x\n", dwReserved);
494
495     /* TODO: CLSID */
496     ok(cs->pUnk != NULL, "cs->pUnk == NULL\n");
497     ok(!cs->dwFlags, "dwFlags = %x\n", cs->dwFlags);
498
499     if(FAILED(QueryCustomPolicy_hres))
500         return QueryCustomPolicy_hres;
501
502     ret = CoTaskMemAlloc(QueryCustomPolicy_psize);
503     *ppPolicy = (BYTE*)ret;
504     *pcbPolicy = QueryCustomPolicy_psize;
505     memset(ret, 0, QueryCustomPolicy_psize);
506     if(QueryCustomPolicy_psize >= sizeof(DWORD))
507         *ret = QueryCustomPolicy_policy;
508
509     return QueryCustomPolicy_hres;
510 }
511
512 static const IInternetHostSecurityManagerVtbl InternetHostSecurityManagerVtbl = {
513     InternetHostSecurityManager_QueryInterface,
514     InternetHostSecurityManager_AddRef,
515     InternetHostSecurityManager_Release,
516     InternetHostSecurityManager_GetSecurityId,
517     InternetHostSecurityManager_ProcessUrlAction,
518     InternetHostSecurityManager_QueryCustomPolicy
519 };
520
521 static IInternetHostSecurityManager InternetHostSecurityManager = { &InternetHostSecurityManagerVtbl };
522
523 static IServiceProvider ServiceProvider;
524
525 static HRESULT WINAPI ServiceProvider_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv)
526 {
527     ok(0, "unexpected call\n");
528     return E_NOINTERFACE;
529 }
530
531 static ULONG WINAPI ServiceProvider_AddRef(IServiceProvider *iface)
532 {
533     return 2;
534 }
535
536 static ULONG WINAPI ServiceProvider_Release(IServiceProvider *iface)
537 {
538     return 1;
539 }
540
541 static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface,
542         REFGUID guidService, REFIID riid, void **ppv)
543 {
544     if(IsEqualGUID(&SID_GetCaller, guidService))
545         return E_NOINTERFACE;
546
547     if(IsEqualGUID(&SID_SInternetHostSecurityManager, guidService)) {
548         if(iface == &ServiceProvider)
549             CHECK_EXPECT(Host_QS_SecMgr);
550         else
551             CHECK_EXPECT(Caller_QS_SecMgr);
552         ok(IsEqualGUID(&IID_IInternetHostSecurityManager, riid), "unexpected riid %s\n", debugstr_guid(riid));
553         if(SUCCEEDED(QS_SecMgr_hres))
554             *ppv = &InternetHostSecurityManager;
555         return QS_SecMgr_hres;
556     }
557
558     ok(0, "unexpected service %s\n", debugstr_guid(guidService));
559     return E_NOINTERFACE;
560 }
561
562 static IServiceProviderVtbl ServiceProviderVtbl = {
563     ServiceProvider_QueryInterface,
564     ServiceProvider_AddRef,
565     ServiceProvider_Release,
566     ServiceProvider_QueryService
567 };
568
569 static IServiceProvider ServiceProvider = { &ServiceProviderVtbl };
570 static IServiceProvider caller_sp = { &ServiceProviderVtbl };
571
572 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
573 {
574     if(IsEqualGUID(&IID_IUnknown, riid)) {
575         *ppv = iface;
576     }else if(IsEqualGUID(&IID_IActiveScriptSite, riid)) {
577         *ppv = iface;
578     }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
579         *ppv = &ServiceProvider;
580     }else {
581         *ppv = NULL;
582         return E_NOINTERFACE;
583     }
584
585     IUnknown_AddRef((IUnknown*)*ppv);
586     return S_OK;
587 }
588
589 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
590 {
591     return 2;
592 }
593
594 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
595 {
596     return 1;
597 }
598
599 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
600 {
601     *plcid = GetUserDefaultLCID();
602     return S_OK;
603 }
604
605 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
606         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
607 {
608     ok(dwReturnMask == SCRIPTINFO_IUNKNOWN, "unexpected dwReturnMask %x\n", dwReturnMask);
609     ok(!ppti, "ppti != NULL\n");
610     ok(!strcmp_wa(pstrName, "test"), "pstrName = %s\n", wine_dbgstr_w(pstrName));
611
612     *ppiunkItem = (IUnknown*)&globalObj;
613     return S_OK;
614 }
615
616 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
617 {
618     return E_NOTIMPL;
619 }
620
621 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
622         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
623 {
624     return E_NOTIMPL;
625 }
626
627 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
628 {
629     return E_NOTIMPL;
630 }
631
632 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
633 {
634     return E_NOTIMPL;
635 }
636
637 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
638 {
639     return E_NOTIMPL;
640 }
641
642 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
643 {
644     return E_NOTIMPL;
645 }
646
647 #undef ACTSCPSITE_THIS
648
649 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
650     ActiveScriptSite_QueryInterface,
651     ActiveScriptSite_AddRef,
652     ActiveScriptSite_Release,
653     ActiveScriptSite_GetLCID,
654     ActiveScriptSite_GetItemInfo,
655     ActiveScriptSite_GetDocVersionString,
656     ActiveScriptSite_OnScriptTerminate,
657     ActiveScriptSite_OnStateChange,
658     ActiveScriptSite_OnScriptError,
659     ActiveScriptSite_OnEnterScript,
660     ActiveScriptSite_OnLeaveScript
661 };
662
663 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
664
665 static void set_safety_options(IUnknown *unk)
666 {
667     IObjectSafety *safety;
668     DWORD supported, enabled;
669     HRESULT hres;
670
671     hres = IUnknown_QueryInterface(unk, &IID_IObjectSafety, (void**)&safety);
672     ok(hres == S_OK, "Could not get IObjectSafety: %08x\n", hres);
673     if(FAILED(hres))
674         return;
675
676     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
677             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER,
678             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
679     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
680
681     supported = enabled = 0xdeadbeef;
682     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
683     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
684     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
685        "supported=%x\n", supported);
686     ok(enabled == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
687        "enabled=%x\n", enabled);
688
689     IObjectSafety_Release(safety);
690 }
691
692 #define parse_script_a(p,s) _parse_script_a(__LINE__,p,s)
693 static void _parse_script_a(unsigned line, IActiveScriptParse *parser, const char *script)
694 {
695     BSTR str;
696     HRESULT hres;
697
698     str = a2bstr(script);
699     hres = IActiveScriptParse64_ParseScriptText(parser, str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
700     SysFreeString(str);
701     ok_(__FILE__,line)(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
702 }
703
704 static IActiveScriptParse *create_script(void)
705 {
706     IActiveScriptParse *parser;
707     IActiveScript *script;
708     HRESULT hres;
709
710     QS_SecMgr_hres = S_OK;
711     ProcessUrlAction_hres = S_OK;
712     ProcessUrlAction_policy = URLPOLICY_ALLOW;
713     CreateInstance_hres = S_OK;
714     QueryCustomPolicy_hres = S_OK;
715     QueryCustomPolicy_psize = sizeof(DWORD);
716     QueryCustomPolicy_policy = URLPOLICY_ALLOW;
717     QI_IDispatch_hres = S_OK;
718     SetSite_hres = S_OK;
719
720     hres = CoCreateInstance(&CLSID_JScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
721             &IID_IActiveScript, (void**)&script);
722     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
723     if(FAILED(hres))
724         return NULL;
725
726     set_safety_options((IUnknown*)script);
727
728     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parser);
729     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
730
731     hres = IActiveScriptParse64_InitNew(parser);
732     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
733
734     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
735     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
736
737     hres = IActiveScript_AddNamedItem(script, testW,
738             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
739     ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
740
741     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_STARTED);
742     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
743
744     IActiveScript_Release(script);
745
746     parse_script_a(parser,
747             "function testException(func, type, number) {\n"
748             "try {\n"
749             "    func();\n"
750             "}catch(e) {\n"
751             "    ok(e.name === type, 'e.name = ' + e.name + ', expected ' + type)\n"
752             "    ok(e.number === number, 'e.number = ' + e.number + ', expected ' + number);\n"
753             "    return;\n"
754             "}"
755             "ok(false, 'exception expected');\n"
756             "}");
757
758     return parser;
759 }
760
761 static IDispatchEx *parse_procedure_a(IActiveScriptParse *parser, const char *src)
762 {
763     IActiveScriptParseProcedure2 *parse_proc;
764     IDispatchEx *dispex;
765     IDispatch *disp;
766     BSTR str;
767     HRESULT hres;
768
769     hres = IUnknown_QueryInterface(parser, &IID_IActiveScriptParseProcedure2, (void**)&parse_proc);
770     ok(hres == S_OK, "Coult not get IActiveScriptParseProcedure2: %08x\n", hres);
771
772     str = a2bstr(src);
773     hres = IActiveScriptParseProcedure2_64_ParseProcedureText(parse_proc, str, NULL, NULL, NULL, NULL, NULL, 0, 0, 0, &disp);
774     SysFreeString(str);
775     IUnknown_Release(parse_proc);
776     ok(hres == S_OK, "ParseProcedureText failed: %08x\n", hres);
777     ok(disp != NULL, "disp == NULL\n");
778
779     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
780     IDispatch_Release(dispex);
781     ok(hres == S_OK, "Could not get IDispatchEx iface: %08x\n", hres);
782
783     return dispex;
784 }
785
786 #define call_procedure(p,c) _call_procedure(__LINE__,p,c)
787 static void _call_procedure(unsigned line, IDispatchEx *proc, IServiceProvider *caller)
788 {
789     DISPPARAMS dp = {NULL,NULL,0,0};
790     EXCEPINFO ei = {0};
791     HRESULT hres;
792
793     hres = IDispatchEx_InvokeEx(proc, DISPID_VALUE, 0, DISPATCH_METHOD, &dp, NULL, &ei, caller);
794     ok_(__FILE__,line)(hres == S_OK, "InvokeEx failed: %08x\n", hres);
795
796 }
797
798 static void test_ActiveXObject(void)
799 {
800     IActiveScriptParse *parser;
801     IDispatchEx *proc;
802
803     parser = create_script();
804
805     SET_EXPECT(Host_QS_SecMgr);
806     SET_EXPECT(ProcessUrlAction);
807     SET_EXPECT(CreateInstance);
808     SET_EXPECT(QueryCustomPolicy);
809     SET_EXPECT(QI_IObjectWithSite);
810     SET_EXPECT(reportSuccess);
811     parse_script_a(parser, "(new ActiveXObject('Wine.Test')).reportSuccess();");
812     CHECK_CALLED(Host_QS_SecMgr);
813     CHECK_CALLED(ProcessUrlAction);
814     CHECK_CALLED(CreateInstance);
815     CHECK_CALLED(QueryCustomPolicy);
816     CHECK_CALLED(QI_IObjectWithSite);
817     CHECK_CALLED(reportSuccess);
818
819     proc = parse_procedure_a(parser, "(new ActiveXObject('Wine.Test')).reportSuccess();");
820
821     SET_EXPECT(ProcessUrlAction);
822     SET_EXPECT(CreateInstance);
823     SET_EXPECT(QueryCustomPolicy);
824     SET_EXPECT(QI_IObjectWithSite);
825     SET_EXPECT(reportSuccess);
826     call_procedure(proc, NULL);
827     CHECK_CALLED(ProcessUrlAction);
828     CHECK_CALLED(CreateInstance);
829     CHECK_CALLED(QueryCustomPolicy);
830     CHECK_CALLED(QI_IObjectWithSite);
831     CHECK_CALLED(reportSuccess);
832
833     SET_EXPECT(ProcessUrlAction);
834     SET_EXPECT(CreateInstance);
835     SET_EXPECT(QueryCustomPolicy);
836     SET_EXPECT(QI_IObjectWithSite);
837     SET_EXPECT(reportSuccess);
838     call_procedure(proc, &caller_sp);
839     CHECK_CALLED(ProcessUrlAction);
840     CHECK_CALLED(CreateInstance);
841     CHECK_CALLED(QueryCustomPolicy);
842     CHECK_CALLED(QI_IObjectWithSite);
843     CHECK_CALLED(reportSuccess);
844
845     IDispatchEx_Release(proc);
846     IUnknown_Release(parser);
847
848     parser = create_script();
849     proc = parse_procedure_a(parser, "(new ActiveXObject('Wine.Test')).reportSuccess();");
850
851     SET_EXPECT(Host_QS_SecMgr);
852     SET_EXPECT(ProcessUrlAction);
853     SET_EXPECT(CreateInstance);
854     SET_EXPECT(QueryCustomPolicy);
855     SET_EXPECT(QI_IObjectWithSite);
856     SET_EXPECT(reportSuccess);
857     call_procedure(proc, &caller_sp);
858     CHECK_CALLED(Host_QS_SecMgr);
859     CHECK_CALLED(ProcessUrlAction);
860     CHECK_CALLED(CreateInstance);
861     CHECK_CALLED(QueryCustomPolicy);
862     CHECK_CALLED(QI_IObjectWithSite);
863     CHECK_CALLED(reportSuccess);
864
865     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.TestABC'); }, 'Error', -2146827859);");
866
867     IDispatchEx_Release(proc);
868     IUnknown_Release(parser);
869
870     parser = create_script();
871     QS_SecMgr_hres = E_NOINTERFACE;
872
873     SET_EXPECT(Host_QS_SecMgr);
874     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
875     CHECK_CALLED(Host_QS_SecMgr);
876
877     IUnknown_Release(parser);
878
879     parser = create_script();
880     ProcessUrlAction_hres = E_FAIL;
881
882     SET_EXPECT(Host_QS_SecMgr);
883     SET_EXPECT(ProcessUrlAction);
884     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
885     CHECK_CALLED(Host_QS_SecMgr);
886     CHECK_CALLED(ProcessUrlAction);
887
888     IUnknown_Release(parser);
889
890     parser = create_script();
891     ProcessUrlAction_policy = URLPOLICY_DISALLOW;
892
893     SET_EXPECT(Host_QS_SecMgr);
894     SET_EXPECT(ProcessUrlAction);
895     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
896     CHECK_CALLED(Host_QS_SecMgr);
897     CHECK_CALLED(ProcessUrlAction);
898
899     IUnknown_Release(parser);
900
901     parser = create_script();
902     CreateInstance_hres = E_FAIL;
903
904     SET_EXPECT(Host_QS_SecMgr);
905     SET_EXPECT(ProcessUrlAction);
906     SET_EXPECT(CreateInstance);
907     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
908     CHECK_CALLED(Host_QS_SecMgr);
909     CHECK_CALLED(ProcessUrlAction);
910     CHECK_CALLED(CreateInstance);
911
912     IUnknown_Release(parser);
913
914     parser = create_script();
915     QueryCustomPolicy_hres = E_FAIL;
916
917     SET_EXPECT(Host_QS_SecMgr);
918     SET_EXPECT(ProcessUrlAction);
919     SET_EXPECT(CreateInstance);
920     SET_EXPECT(QueryCustomPolicy);
921     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
922     CHECK_CALLED(Host_QS_SecMgr);
923     CHECK_CALLED(ProcessUrlAction);
924     CHECK_CALLED(CreateInstance);
925     CHECK_CALLED(QueryCustomPolicy);
926
927     IUnknown_Release(parser);
928
929     parser = create_script();
930     QueryCustomPolicy_psize = 6;
931
932     SET_EXPECT(Host_QS_SecMgr);
933     SET_EXPECT(ProcessUrlAction);
934     SET_EXPECT(CreateInstance);
935     SET_EXPECT(QueryCustomPolicy);
936     SET_EXPECT(QI_IObjectWithSite);
937     SET_EXPECT(reportSuccess);
938     parse_script_a(parser, "(new ActiveXObject('Wine.Test')).reportSuccess();");
939     CHECK_CALLED(Host_QS_SecMgr);
940     CHECK_CALLED(ProcessUrlAction);
941     CHECK_CALLED(CreateInstance);
942     CHECK_CALLED(QueryCustomPolicy);
943     CHECK_CALLED(QI_IObjectWithSite);
944     CHECK_CALLED(reportSuccess);
945
946     IUnknown_Release(parser);
947
948     parser = create_script();
949     QueryCustomPolicy_policy = URLPOLICY_DISALLOW;
950
951     SET_EXPECT(Host_QS_SecMgr);
952     SET_EXPECT(ProcessUrlAction);
953     SET_EXPECT(CreateInstance);
954     SET_EXPECT(QueryCustomPolicy);
955     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
956     CHECK_CALLED(Host_QS_SecMgr);
957     CHECK_CALLED(ProcessUrlAction);
958     CHECK_CALLED(CreateInstance);
959     CHECK_CALLED(QueryCustomPolicy);
960
961     QueryCustomPolicy_psize = 6;
962
963     SET_EXPECT(ProcessUrlAction);
964     SET_EXPECT(CreateInstance);
965     SET_EXPECT(QueryCustomPolicy);
966     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
967     CHECK_CALLED(ProcessUrlAction);
968     CHECK_CALLED(CreateInstance);
969     CHECK_CALLED(QueryCustomPolicy);
970
971     QueryCustomPolicy_policy = URLPOLICY_ALLOW;
972     QueryCustomPolicy_psize = 3;
973
974     SET_EXPECT(ProcessUrlAction);
975     SET_EXPECT(CreateInstance);
976     SET_EXPECT(QueryCustomPolicy);
977     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
978     CHECK_CALLED(ProcessUrlAction);
979     CHECK_CALLED(CreateInstance);
980     CHECK_CALLED(QueryCustomPolicy);
981
982     IUnknown_Release(parser);
983
984     parser = create_script();
985     object_with_site = &ObjectWithSite;
986
987     SET_EXPECT(Host_QS_SecMgr);
988     SET_EXPECT(ProcessUrlAction);
989     SET_EXPECT(CreateInstance);
990     SET_EXPECT(QueryCustomPolicy);
991     SET_EXPECT(QI_IObjectWithSite);
992     SET_EXPECT(SetSite);
993     SET_EXPECT(reportSuccess);
994     parse_script_a(parser, "(new ActiveXObject('Wine.Test')).reportSuccess();");
995     CHECK_CALLED(Host_QS_SecMgr);
996     CHECK_CALLED(ProcessUrlAction);
997     CHECK_CALLED(CreateInstance);
998     CHECK_CALLED(QueryCustomPolicy);
999     CHECK_CALLED(QI_IObjectWithSite);
1000     CHECK_CALLED(SetSite);
1001     CHECK_CALLED(reportSuccess);
1002
1003     SetSite_hres = E_FAIL;
1004     SET_EXPECT(ProcessUrlAction);
1005     SET_EXPECT(CreateInstance);
1006     SET_EXPECT(QueryCustomPolicy);
1007     SET_EXPECT(QI_IObjectWithSite);
1008     SET_EXPECT(SetSite);
1009     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
1010     CHECK_CALLED(ProcessUrlAction);
1011     CHECK_CALLED(CreateInstance);
1012     CHECK_CALLED(QueryCustomPolicy);
1013     CHECK_CALLED(QI_IObjectWithSite);
1014     CHECK_CALLED(SetSite);
1015
1016     IUnknown_Release(parser);
1017 }
1018
1019 static BOOL init_key(const char *key_name, const char *def_value, BOOL init)
1020 {
1021     HKEY hkey;
1022     DWORD res;
1023
1024     if(!init) {
1025         RegDeleteKey(HKEY_CLASSES_ROOT, key_name);
1026         return TRUE;
1027     }
1028
1029     res = RegCreateKeyA(HKEY_CLASSES_ROOT, key_name, &hkey);
1030     if(res != ERROR_SUCCESS)
1031         return FALSE;
1032
1033     if(def_value)
1034         res = RegSetValueA(hkey, NULL, REG_SZ, def_value, strlen(def_value));
1035
1036     RegCloseKey(hkey);
1037
1038     return res == ERROR_SUCCESS;
1039 }
1040
1041 static BOOL init_registry(BOOL init)
1042 {
1043     return init_key("Wine.Test\\CLSID", TESTOBJ_CLSID, init);
1044 }
1045
1046 static BOOL register_activex(void)
1047 {
1048     DWORD regid;
1049     HRESULT hres;
1050
1051     if(!init_registry(TRUE)) {
1052         init_registry(FALSE);
1053         return FALSE;
1054     }
1055
1056     hres = CoRegisterClassObject(&CLSID_TestObj, (IUnknown *)&activex_cf,
1057                                  CLSCTX_INPROC_SERVER, REGCLS_MULTIPLEUSE, &regid);
1058     ok(hres == S_OK, "Could not register screipt engine: %08x\n", hres);
1059
1060     return TRUE;
1061 }
1062
1063 static BOOL check_jscript(void)
1064 {
1065     IActiveScriptParse *parser;
1066     BSTR str;
1067     HRESULT hres;
1068
1069     parser = create_script();
1070     if(!parser)
1071         return FALSE;
1072
1073     str = a2bstr("if(!('localeCompare' in String.prototype)) throw 1;");
1074     hres = IActiveScriptParse64_ParseScriptText(parser, str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
1075     SysFreeString(str);
1076     IUnknown_Release(parser);
1077
1078     return hres == S_OK;
1079 }
1080
1081 START_TEST(activex)
1082 {
1083     CoInitialize(NULL);
1084
1085     if(check_jscript()) {
1086         register_activex();
1087
1088         test_ActiveXObject();
1089
1090         init_registry(FALSE);
1091     }else {
1092         win_skip("Broken engine, probably too old\n");
1093     }
1094
1095     CoUninitialize();
1096 }