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