git.oblomov.eu Git - wine/blob - dlls/vbscript/tests/createobj.c

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