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