git.oblomov.eu Git - wine/blob - 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
 724     set_safety_options((IUnknown*)script);
 725
 726     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)&parser);
 727     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
 728
 729     hres = IActiveScriptParse64_InitNew(parser);
 730     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 731
 732     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 733     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 734
 735     hres = IActiveScript_AddNamedItem(script, testW,
 736             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
 737     ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
 738
 739     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_STARTED);
 740     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
 741
 742     IActiveScript_Release(script);
 743
 744     parse_script_a(parser,
 745             "function testException(func, type, number) {\n"
 746             "try {\n"
 747             "    func();\n"
 748             "}catch(e) {\n"
 749             "    ok(e.name === type, 'e.name = ' + e.name + ', expected ' + type)\n"
 750             "    ok(e.number === number, 'e.number = ' + e.number + ', expected ' + number);\n"
 751             "    return;\n"
 752             "}"
 753             "ok(false, 'exception expected');\n"
 754             "}");
 755
 756     return parser;
 757 }
 758
 759 static IDispatchEx *parse_procedure_a(IActiveScriptParse *parser, const char *src)
 760 {
 761     IActiveScriptParseProcedure2 *parse_proc;
 762     IDispatchEx *dispex;
 763     IDispatch *disp;
 764     BSTR str;
 765     HRESULT hres;
 766
 767     hres = IUnknown_QueryInterface(parser, &IID_IActiveScriptParseProcedure2, (void**)&parse_proc);
 768     ok(hres == S_OK, "Coult not get IActiveScriptParseProcedure2: %08x\n", hres);
 769
 770     str = a2bstr(src);
 771     hres = IActiveScriptParseProcedure2_64_ParseProcedureText(parse_proc, str, NULL, NULL, NULL, NULL, NULL, 0, 0, 0, &disp);
 772     SysFreeString(str);
 773     IUnknown_Release(parse_proc);
 774     ok(hres == S_OK, "ParseProcedureText failed: %08x\n", hres);
 775     ok(disp != NULL, "disp == NULL\n");
 776
 777     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
 778     IDispatch_Release(dispex);
 779     ok(hres == S_OK, "Could not get IDispatchEx iface: %08x\n", hres);
 780
 781     return dispex;
 782 }
 783
 784 #define call_procedure(p,c) _call_procedure(__LINE__,p,c)
 785 static void _call_procedure(unsigned line, IDispatchEx *proc, IServiceProvider *caller)
 786 {
 787     DISPPARAMS dp = {NULL,NULL,0,0};
 788     EXCEPINFO ei = {0};
 789     HRESULT hres;
 790
 791     hres = IDispatchEx_InvokeEx(proc, DISPID_VALUE, 0, DISPATCH_METHOD, &dp, NULL, &ei, caller);
 792     ok_(__FILE__,line)(hres == S_OK, "InvokeEx failed: %08x\n", hres);
 793
 794 }
 795
 796 static void test_ActiveXObject(void)
 797 {
 798     IActiveScriptParse *parser;
 799     IDispatchEx *proc;
 800
 801     parser = create_script();
 802
 803     SET_EXPECT(Host_QS_SecMgr);
 804     SET_EXPECT(ProcessUrlAction);
 805     SET_EXPECT(CreateInstance);
 806     SET_EXPECT(QueryCustomPolicy);
 807     SET_EXPECT(QI_IObjectWithSite);
 808     SET_EXPECT(reportSuccess);
 809     parse_script_a(parser, "(new ActiveXObject('Wine.Test')).reportSuccess();");
 810     CHECK_CALLED(Host_QS_SecMgr);
 811     CHECK_CALLED(ProcessUrlAction);
 812     CHECK_CALLED(CreateInstance);
 813     CHECK_CALLED(QueryCustomPolicy);
 814     CHECK_CALLED(QI_IObjectWithSite);
 815     CHECK_CALLED(reportSuccess);
 816
 817     proc = parse_procedure_a(parser, "(new ActiveXObject('Wine.Test')).reportSuccess();");
 818
 819     SET_EXPECT(ProcessUrlAction);
 820     SET_EXPECT(CreateInstance);
 821     SET_EXPECT(QueryCustomPolicy);
 822     SET_EXPECT(QI_IObjectWithSite);
 823     SET_EXPECT(reportSuccess);
 824     call_procedure(proc, NULL);
 825     CHECK_CALLED(ProcessUrlAction);
 826     CHECK_CALLED(CreateInstance);
 827     CHECK_CALLED(QueryCustomPolicy);
 828     CHECK_CALLED(QI_IObjectWithSite);
 829     CHECK_CALLED(reportSuccess);
 830
 831     SET_EXPECT(ProcessUrlAction);
 832     SET_EXPECT(CreateInstance);
 833     SET_EXPECT(QueryCustomPolicy);
 834     SET_EXPECT(QI_IObjectWithSite);
 835     SET_EXPECT(reportSuccess);
 836     call_procedure(proc, &caller_sp);
 837     CHECK_CALLED(ProcessUrlAction);
 838     CHECK_CALLED(CreateInstance);
 839     CHECK_CALLED(QueryCustomPolicy);
 840     CHECK_CALLED(QI_IObjectWithSite);
 841     CHECK_CALLED(reportSuccess);
 842
 843     IDispatchEx_Release(proc);
 844     IUnknown_Release(parser);
 845
 846     parser = create_script();
 847     proc = parse_procedure_a(parser, "(new ActiveXObject('Wine.Test')).reportSuccess();");
 848
 849     SET_EXPECT(Host_QS_SecMgr);
 850     SET_EXPECT(ProcessUrlAction);
 851     SET_EXPECT(CreateInstance);
 852     SET_EXPECT(QueryCustomPolicy);
 853     SET_EXPECT(QI_IObjectWithSite);
 854     SET_EXPECT(reportSuccess);
 855     call_procedure(proc, &caller_sp);
 856     CHECK_CALLED(Host_QS_SecMgr);
 857     CHECK_CALLED(ProcessUrlAction);
 858     CHECK_CALLED(CreateInstance);
 859     CHECK_CALLED(QueryCustomPolicy);
 860     CHECK_CALLED(QI_IObjectWithSite);
 861     CHECK_CALLED(reportSuccess);
 862
 863     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.TestABC'); }, 'Error', -2146827859);");
 864
 865     IDispatchEx_Release(proc);
 866     IUnknown_Release(parser);
 867
 868     parser = create_script();
 869     QS_SecMgr_hres = E_NOINTERFACE;
 870
 871     SET_EXPECT(Host_QS_SecMgr);
 872     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
 873     CHECK_CALLED(Host_QS_SecMgr);
 874
 875     IUnknown_Release(parser);
 876
 877     parser = create_script();
 878     ProcessUrlAction_hres = E_FAIL;
 879
 880     SET_EXPECT(Host_QS_SecMgr);
 881     SET_EXPECT(ProcessUrlAction);
 882     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
 883     CHECK_CALLED(Host_QS_SecMgr);
 884     CHECK_CALLED(ProcessUrlAction);
 885
 886     IUnknown_Release(parser);
 887
 888     parser = create_script();
 889     ProcessUrlAction_policy = URLPOLICY_DISALLOW;
 890
 891     SET_EXPECT(Host_QS_SecMgr);
 892     SET_EXPECT(ProcessUrlAction);
 893     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
 894     CHECK_CALLED(Host_QS_SecMgr);
 895     CHECK_CALLED(ProcessUrlAction);
 896
 897     IUnknown_Release(parser);
 898
 899     parser = create_script();
 900     CreateInstance_hres = E_FAIL;
 901
 902     SET_EXPECT(Host_QS_SecMgr);
 903     SET_EXPECT(ProcessUrlAction);
 904     SET_EXPECT(CreateInstance);
 905     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
 906     CHECK_CALLED(Host_QS_SecMgr);
 907     CHECK_CALLED(ProcessUrlAction);
 908     CHECK_CALLED(CreateInstance);
 909
 910     IUnknown_Release(parser);
 911
 912     parser = create_script();
 913     QueryCustomPolicy_hres = E_FAIL;
 914
 915     SET_EXPECT(Host_QS_SecMgr);
 916     SET_EXPECT(ProcessUrlAction);
 917     SET_EXPECT(CreateInstance);
 918     SET_EXPECT(QueryCustomPolicy);
 919     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
 920     CHECK_CALLED(Host_QS_SecMgr);
 921     CHECK_CALLED(ProcessUrlAction);
 922     CHECK_CALLED(CreateInstance);
 923     CHECK_CALLED(QueryCustomPolicy);
 924
 925     IUnknown_Release(parser);
 926
 927     parser = create_script();
 928     QueryCustomPolicy_psize = 6;
 929
 930     SET_EXPECT(Host_QS_SecMgr);
 931     SET_EXPECT(ProcessUrlAction);
 932     SET_EXPECT(CreateInstance);
 933     SET_EXPECT(QueryCustomPolicy);
 934     SET_EXPECT(QI_IObjectWithSite);
 935     SET_EXPECT(reportSuccess);
 936     parse_script_a(parser, "(new ActiveXObject('Wine.Test')).reportSuccess();");
 937     CHECK_CALLED(Host_QS_SecMgr);
 938     CHECK_CALLED(ProcessUrlAction);
 939     CHECK_CALLED(CreateInstance);
 940     CHECK_CALLED(QueryCustomPolicy);
 941     CHECK_CALLED(QI_IObjectWithSite);
 942     CHECK_CALLED(reportSuccess);
 943
 944     IUnknown_Release(parser);
 945
 946     parser = create_script();
 947     QueryCustomPolicy_policy = URLPOLICY_DISALLOW;
 948
 949     SET_EXPECT(Host_QS_SecMgr);
 950     SET_EXPECT(ProcessUrlAction);
 951     SET_EXPECT(CreateInstance);
 952     SET_EXPECT(QueryCustomPolicy);
 953     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
 954     CHECK_CALLED(Host_QS_SecMgr);
 955     CHECK_CALLED(ProcessUrlAction);
 956     CHECK_CALLED(CreateInstance);
 957     CHECK_CALLED(QueryCustomPolicy);
 958
 959     QueryCustomPolicy_psize = 6;
 960
 961     SET_EXPECT(ProcessUrlAction);
 962     SET_EXPECT(CreateInstance);
 963     SET_EXPECT(QueryCustomPolicy);
 964     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
 965     CHECK_CALLED(ProcessUrlAction);
 966     CHECK_CALLED(CreateInstance);
 967     CHECK_CALLED(QueryCustomPolicy);
 968
 969     QueryCustomPolicy_policy = URLPOLICY_ALLOW;
 970     QueryCustomPolicy_psize = 3;
 971
 972     SET_EXPECT(ProcessUrlAction);
 973     SET_EXPECT(CreateInstance);
 974     SET_EXPECT(QueryCustomPolicy);
 975     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
 976     CHECK_CALLED(ProcessUrlAction);
 977     CHECK_CALLED(CreateInstance);
 978     CHECK_CALLED(QueryCustomPolicy);
 979
 980     IUnknown_Release(parser);
 981
 982     parser = create_script();
 983     object_with_site = &ObjectWithSite;
 984
 985     SET_EXPECT(Host_QS_SecMgr);
 986     SET_EXPECT(ProcessUrlAction);
 987     SET_EXPECT(CreateInstance);
 988     SET_EXPECT(QueryCustomPolicy);
 989     SET_EXPECT(QI_IObjectWithSite);
 990     SET_EXPECT(SetSite);
 991     SET_EXPECT(reportSuccess);
 992     parse_script_a(parser, "(new ActiveXObject('Wine.Test')).reportSuccess();");
 993     CHECK_CALLED(Host_QS_SecMgr);
 994     CHECK_CALLED(ProcessUrlAction);
 995     CHECK_CALLED(CreateInstance);
 996     CHECK_CALLED(QueryCustomPolicy);
 997     CHECK_CALLED(QI_IObjectWithSite);
 998     CHECK_CALLED(SetSite);
 999     CHECK_CALLED(reportSuccess);
1000
1001     SetSite_hres = E_FAIL;
1002     SET_EXPECT(ProcessUrlAction);
1003     SET_EXPECT(CreateInstance);
1004     SET_EXPECT(QueryCustomPolicy);
1005     SET_EXPECT(QI_IObjectWithSite);
1006     SET_EXPECT(SetSite);
1007     parse_script_a(parser, "testException(function() { new ActiveXObject('Wine.Test'); }, 'Error', -2146827859);");
1008     CHECK_CALLED(ProcessUrlAction);
1009     CHECK_CALLED(CreateInstance);
1010     CHECK_CALLED(QueryCustomPolicy);
1011     CHECK_CALLED(QI_IObjectWithSite);
1012     CHECK_CALLED(SetSite);
1013
1014     IUnknown_Release(parser);
1015 }
1016
1017 static BOOL init_key(const char *key_name, const char *def_value, BOOL init)
1018 {
1019     HKEY hkey;
1020     DWORD res;
1021
1022     if(!init) {
1023         RegDeleteKey(HKEY_CLASSES_ROOT, key_name);
1024         return TRUE;
1025     }
1026
1027     res = RegCreateKeyA(HKEY_CLASSES_ROOT, key_name, &hkey);
1028     if(res != ERROR_SUCCESS)
1029         return FALSE;
1030
1031     if(def_value)
1032         res = RegSetValueA(hkey, NULL, REG_SZ, def_value, strlen(def_value));
1033
1034     RegCloseKey(hkey);
1035
1036     return res == ERROR_SUCCESS;
1037 }
1038
1039 static BOOL init_registry(BOOL init)
1040 {
1041     return init_key("Wine.Test\\CLSID", TESTOBJ_CLSID, init);
1042 }
1043
1044 static BOOL register_activex(void)
1045 {
1046     DWORD regid;
1047     HRESULT hres;
1048
1049     if(!init_registry(TRUE)) {
1050         init_registry(FALSE);
1051         return FALSE;
1052     }
1053
1054     hres = CoRegisterClassObject(&CLSID_TestObj, (IUnknown *)&activex_cf,
1055                                  CLSCTX_INPROC_SERVER, REGCLS_MULTIPLEUSE, &regid);
1056     ok(hres == S_OK, "Could not register screipt engine: %08x\n", hres);
1057
1058     return TRUE;
1059 }
1060
1061 START_TEST(activex)
1062 {
1063     CoInitialize(NULL);
1064     register_activex();
1065
1066     test_ActiveXObject();
1067
1068     init_registry(FALSE);
1069     CoUninitialize();
1070 }