git.oblomov.eu Git - wine/blob - dlls/mshtml/tests/script.c

   1 /*
   2  * Copyright 2008-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 #define COBJMACROS
  20 #define CONST_VTABLE
  21
  22 #include <wine/test.h>
  23 #include <stdarg.h>
  24 #include <stdio.h>
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "ole2.h"
  29 #include "wininet.h"
  30 #include "docobj.h"
  31 #include "dispex.h"
  32 #include "hlink.h"
  33 #include "mshtml.h"
  34 #include "mshtmhst.h"
  35 #include "initguid.h"
  36 #include "activscp.h"
  37 #include "activdbg.h"
  38 #include "objsafe.h"
  39 #include "mshtmdid.h"
  40 #include "mshtml_test.h"
  41
  42 DEFINE_GUID(CLSID_IdentityUnmarshal,0x0000001b,0x0000,0x0000,0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
  43
  44 /* Defined as extern in urlmon.idl, but not exported by uuid.lib */
  45 const GUID GUID_CUSTOM_CONFIRMOBJECTSAFETY =
  46     {0x10200490,0xfa38,0x11d0,{0xac,0x0e,0x00,0xa0,0xc9,0xf,0xff,0xc0}};
  47
  48 #ifdef _WIN64
  49
  50 #define CTXARG_T DWORDLONG
  51 #define IActiveScriptParseVtbl IActiveScriptParse64Vtbl
  52 #define IActiveScriptParseProcedure2Vtbl IActiveScriptParseProcedure2_64Vtbl
  53
  54 #else
  55
  56 #define CTXARG_T DWORD
  57 #define IActiveScriptParseVtbl IActiveScriptParse32Vtbl
  58 #define IActiveScriptParseProcedure2Vtbl IActiveScriptParseProcedure2_32Vtbl
  59
  60 #endif
  61
  62 #define DEFINE_EXPECT(func) \
  63     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
  64
  65 #define SET_EXPECT(func) \
  66     do { called_ ## func = FALSE; expect_ ## func = TRUE; } while(0)
  67
  68 #define CHECK_EXPECT2(func) \
  69     do { \
  70         ok(expect_ ##func, "unexpected call " #func "\n"); \
  71         called_ ## func = TRUE; \
  72     }while(0)
  73
  74 #define CHECK_EXPECT(func) \
  75     do { \
  76         CHECK_EXPECT2(func); \
  77         expect_ ## func = FALSE; \
  78     }while(0)
  79
  80 #define CHECK_CALLED(func) \
  81     do { \
  82         ok(called_ ## func, "expected " #func "\n"); \
  83         expect_ ## func = called_ ## func = FALSE; \
  84     }while(0)
  85
  86 #define CHECK_CALLED_BROKEN(func) \
  87     do { \
  88         ok(called_ ## func || broken(!called_ ## func), "expected " #func "\n"); \
  89         expect_ ## func = called_ ## func = FALSE; \
  90     }while(0)
  91
  92 #define CHECK_NOT_CALLED(func) \
  93     do { \
  94         ok(!called_ ## func, "unexpected " #func "\n"); \
  95         expect_ ## func = called_ ## func = FALSE; \
  96     }while(0)
  97
  98 #define CLEAR_CALLED(func) \
  99     expect_ ## func = called_ ## func = FALSE
 100
 101
 102 DEFINE_EXPECT(CreateInstance);
 103 DEFINE_EXPECT(GetInterfaceSafetyOptions);
 104 DEFINE_EXPECT(SetInterfaceSafetyOptions);
 105 DEFINE_EXPECT(InitNew);
 106 DEFINE_EXPECT(Close);
 107 DEFINE_EXPECT(SetProperty_HACK_TRIDENTEVENTSINK);
 108 DEFINE_EXPECT(SetProperty_INVOKEVERSIONING);
 109 DEFINE_EXPECT(SetProperty_ABBREVIATE_GLOBALNAME_RESOLUTION);
 110 DEFINE_EXPECT(SetScriptSite);
 111 DEFINE_EXPECT(GetScriptState);
 112 DEFINE_EXPECT(SetScriptState_STARTED);
 113 DEFINE_EXPECT(SetScriptState_CONNECTED);
 114 DEFINE_EXPECT(SetScriptState_DISCONNECTED);
 115 DEFINE_EXPECT(AddNamedItem);
 116 DEFINE_EXPECT(ParseScriptText);
 117 DEFINE_EXPECT(GetScriptDispatch);
 118 DEFINE_EXPECT(funcDisp);
 119 DEFINE_EXPECT(script_divid_d);
 120 DEFINE_EXPECT(script_testprop_d);
 121 DEFINE_EXPECT(script_testprop_i);
 122 DEFINE_EXPECT(script_testprop2_d);
 123 DEFINE_EXPECT(AXQueryInterface_IActiveScript);
 124 DEFINE_EXPECT(AXQueryInterface_IObjectSafety);
 125 DEFINE_EXPECT(AXGetInterfaceSafetyOptions);
 126 DEFINE_EXPECT(AXSetInterfaceSafetyOptions_IDispatch);
 127 DEFINE_EXPECT(AXSetInterfaceSafetyOptions_IDispatchEx);
 128 DEFINE_EXPECT(external_success);
 129
 130 #define TESTSCRIPT_CLSID "{178fc163-f585-4e24-9c13-4bb7faf80746}"
 131 #define TESTACTIVEX_CLSID "{178fc163-f585-4e24-9c13-4bb7faf80646}"
 132
 133 #define DISPID_SCRIPT_TESTPROP   0x100000
 134 #define DISPID_SCRIPT_TESTPROP2  0x100001
 135
 136 #define DISPID_EXTERNAL_OK             0x300000
 137 #define DISPID_EXTERNAL_TRACE          0x300001
 138 #define DISPID_EXTERNAL_REPORTSUCCESS  0x300002
 139
 140 static const GUID CLSID_TestScript =
 141     {0x178fc163,0xf585,0x4e24,{0x9c,0x13,0x4b,0xb7,0xfa,0xf8,0x07,0x46}};
 142 static const GUID CLSID_TestActiveX =
 143     {0x178fc163,0xf585,0x4e24,{0x9c,0x13,0x4b,0xb7,0xfa,0xf8,0x06,0x46}};
 144
 145 static IHTMLDocument2 *notif_doc;
 146 static IOleDocumentView *view;
 147 static IDispatchEx *window_dispex;
 148 static BOOL doc_complete;
 149 static IDispatch *script_disp;
 150 static BOOL ax_objsafe;
 151 static HWND container_hwnd;
 152 static HRESULT ax_getopt_hres = S_OK, ax_setopt_dispex_hres = S_OK, ax_setopt_disp_hres = S_OK;
 153 static DWORD ax_setopt;
 154
 155 static const char *debugstr_guid(REFIID riid)
 156 {
 157     static char buf[50];
 158
 159     sprintf(buf, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
 160             riid->Data1, riid->Data2, riid->Data3, riid->Data4[0],
 161             riid->Data4[1], riid->Data4[2], riid->Data4[3], riid->Data4[4],
 162             riid->Data4[5], riid->Data4[6], riid->Data4[7]);
 163
 164     return buf;
 165 }
 166
 167 static int strcmp_wa(LPCWSTR strw, const char *stra)
 168 {
 169     CHAR buf[512];
 170     WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), NULL, NULL);
 171     return lstrcmpA(stra, buf);
 172 }
 173
 174 static BSTR a2bstr(const char *str)
 175 {
 176     BSTR ret;
 177     int len;
 178
 179     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
 180     ret = SysAllocStringLen(NULL, len);
 181     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
 182
 183     return ret;
 184 }
 185
 186 static BOOL init_key(const char *key_name, const char *def_value, BOOL init)
 187 {
 188     HKEY hkey;
 189     DWORD res;
 190
 191     if(!init) {
 192         RegDeleteKey(HKEY_CLASSES_ROOT, key_name);
 193         return TRUE;
 194     }
 195
 196     res = RegCreateKeyA(HKEY_CLASSES_ROOT, key_name, &hkey);
 197     if(res != ERROR_SUCCESS)
 198         return FALSE;
 199
 200     if(def_value)
 201         res = RegSetValueA(hkey, NULL, REG_SZ, def_value, strlen(def_value));
 202
 203     RegCloseKey(hkey);
 204
 205     return res == ERROR_SUCCESS;
 206 }
 207
 208 static HRESULT WINAPI PropertyNotifySink_QueryInterface(IPropertyNotifySink *iface,
 209         REFIID riid, void**ppv)
 210 {
 211     if(IsEqualGUID(&IID_IPropertyNotifySink, riid)) {
 212         *ppv = iface;
 213         return S_OK;
 214     }
 215
 216     return E_NOINTERFACE;
 217 }
 218
 219 static ULONG WINAPI PropertyNotifySink_AddRef(IPropertyNotifySink *iface)
 220 {
 221     return 2;
 222 }
 223
 224 static ULONG WINAPI PropertyNotifySink_Release(IPropertyNotifySink *iface)
 225 {
 226     return 1;
 227 }
 228
 229 static HRESULT WINAPI PropertyNotifySink_OnChanged(IPropertyNotifySink *iface, DISPID dispID)
 230 {
 231     if(dispID == DISPID_READYSTATE){
 232         BSTR state;
 233         HRESULT hres;
 234
 235         static const WCHAR completeW[] = {'c','o','m','p','l','e','t','e',0};
 236
 237         hres = IHTMLDocument2_get_readyState(notif_doc, &state);
 238         ok(hres == S_OK, "get_readyState failed: %08x\n", hres);
 239
 240         if(!lstrcmpW(state, completeW))
 241             doc_complete = TRUE;
 242
 243         SysFreeString(state);
 244     }
 245
 246     return S_OK;
 247 }
 248
 249 static HRESULT WINAPI PropertyNotifySink_OnRequestEdit(IPropertyNotifySink *iface, DISPID dispID)
 250 {
 251     ok(0, "unexpected call\n");
 252     return E_NOTIMPL;
 253 }
 254
 255 static IPropertyNotifySinkVtbl PropertyNotifySinkVtbl = {
 256     PropertyNotifySink_QueryInterface,
 257     PropertyNotifySink_AddRef,
 258     PropertyNotifySink_Release,
 259     PropertyNotifySink_OnChanged,
 260     PropertyNotifySink_OnRequestEdit
 261 };
 262
 263 static IPropertyNotifySink PropertyNotifySink = { &PropertyNotifySinkVtbl };
 264
 265 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
 266 {
 267     *ppv = NULL;
 268
 269     if(IsEqualGUID(riid, &IID_IUnknown)
 270        || IsEqualGUID(riid, &IID_IDispatch)
 271        || IsEqualGUID(riid, &IID_IDispatchEx))
 272         *ppv = iface;
 273     else
 274         return E_NOINTERFACE;
 275
 276     return S_OK;
 277 }
 278
 279 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
 280 {
 281     return 2;
 282 }
 283
 284 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
 285 {
 286     return 1;
 287 }
 288
 289 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
 290 {
 291     ok(0, "unexpected call\n");
 292     return E_NOTIMPL;
 293 }
 294
 295 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
 296                                               LCID lcid, ITypeInfo **ppTInfo)
 297 {
 298     ok(0, "unexpected call\n");
 299     return E_NOTIMPL;
 300 }
 301
 302 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
 303                                                 LPOLESTR *rgszNames, UINT cNames,
 304                                                 LCID lcid, DISPID *rgDispId)
 305 {
 306     ok(0, "unexpected call\n");
 307     return E_NOTIMPL;
 308 }
 309
 310 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
 311                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
 312                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
 313 {
 314     ok(0, "unexpected call\n");
 315     return E_NOTIMPL;
 316 }
 317
 318 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
 319 {
 320     ok(0, "unexpected call %s %x\n", wine_dbgstr_w(bstrName), grfdex);
 321     return E_NOTIMPL;
 322 }
 323
 324 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
 325 {
 326     ok(0, "unexpected call\n");
 327     return E_NOTIMPL;
 328 }
 329
 330 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
 331 {
 332     ok(0, "unexpected call\n");
 333     return E_NOTIMPL;
 334 }
 335
 336 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
 337 {
 338     ok(0, "unexpected call\n");
 339     return E_NOTIMPL;
 340 }
 341
 342 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
 343 {
 344     ok(0, "unexpected call\n");
 345     return E_NOTIMPL;
 346 }
 347
 348 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
 349 {
 350     ok(0, "unexpected call\n");
 351     return E_NOTIMPL;
 352 }
 353
 354 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 355 {
 356     ok(0, "unexpected call\n");
 357     return E_NOTIMPL;
 358 }
 359
 360 static HRESULT WINAPI funcDisp_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 361         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 362 {
 363     CHECK_EXPECT(funcDisp);
 364
 365     ok(id == DISPID_VALUE, "id = %d\n", id);
 366     ok(lcid == 0, "lcid = %x\n", lcid);
 367     ok(wFlags == DISPATCH_METHOD, "wFlags = %x\n", wFlags);
 368     ok(pdp != NULL, "pdp == NULL\n");
 369     ok(pdp->cArgs == 2, "pdp->cArgs = %d\n", pdp->cArgs);
 370     ok(pdp->cNamedArgs == 1, "pdp->cNamedArgs = %d\n", pdp->cNamedArgs);
 371     ok(pdp->rgdispidNamedArgs[0] == DISPID_THIS, "pdp->rgdispidNamedArgs[0] = %d\n", pdp->rgdispidNamedArgs[0]);
 372     ok(V_VT(pdp->rgvarg) == VT_DISPATCH, "V_VT(rgvarg) = %d\n", V_VT(pdp->rgvarg));
 373     ok(V_VT(pdp->rgvarg+1) == VT_BOOL, "V_VT(rgvarg[1]) = %d\n", V_VT(pdp->rgvarg));
 374     ok(V_BOOL(pdp->rgvarg+1) == VARIANT_TRUE, "V_BOOL(rgvarg[1]) = %x\n", V_BOOL(pdp->rgvarg));
 375     ok(pvarRes != NULL, "pvarRes == NULL\n");
 376     ok(pei != NULL, "pei == NULL\n");
 377     ok(!pspCaller, "pspCaller != NULL\n");
 378
 379     V_VT(pvarRes) = VT_I4;
 380     V_I4(pvarRes) = 100;
 381     return S_OK;
 382 }
 383
 384 static IDispatchExVtbl testObjVtbl = {
 385     DispatchEx_QueryInterface,
 386     DispatchEx_AddRef,
 387     DispatchEx_Release,
 388     DispatchEx_GetTypeInfoCount,
 389     DispatchEx_GetTypeInfo,
 390     DispatchEx_GetIDsOfNames,
 391     DispatchEx_Invoke,
 392     DispatchEx_GetDispID,
 393     funcDisp_InvokeEx,
 394     DispatchEx_DeleteMemberByName,
 395     DispatchEx_DeleteMemberByDispID,
 396     DispatchEx_GetMemberProperties,
 397     DispatchEx_GetMemberName,
 398     DispatchEx_GetNextDispID,
 399     DispatchEx_GetNameSpaceParent
 400 };
 401
 402 static IDispatchEx funcDisp = { &testObjVtbl };
 403
 404 static HRESULT WINAPI scriptDisp_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 405 {
 406     if(!strcmp_wa(bstrName, "testProp")) {
 407         CHECK_EXPECT(script_testprop_d);
 408         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 409         *pid = DISPID_SCRIPT_TESTPROP;
 410         return S_OK;
 411     }
 412
 413     if(!strcmp_wa(bstrName, "testProp2")) {
 414         CHECK_EXPECT(script_testprop2_d);
 415         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 416         *pid = DISPID_SCRIPT_TESTPROP2;
 417         return S_OK;
 418     }
 419
 420     if(!strcmp_wa(bstrName, "divid")) {
 421         CHECK_EXPECT(script_divid_d);
 422         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 423         return E_FAIL;
 424     }
 425
 426     ok(0, "unexpected call\n");
 427     return E_NOTIMPL;
 428 }
 429
 430 static HRESULT WINAPI scriptDisp_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 431         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 432 {
 433     switch(id) {
 434     case DISPID_SCRIPT_TESTPROP:
 435         CHECK_EXPECT(script_testprop_i);
 436
 437         ok(lcid == 0, "lcid = %x\n", lcid);
 438         ok(wFlags == DISPATCH_PROPERTYGET, "wFlags = %x\n", wFlags);
 439         ok(pdp != NULL, "pdp == NULL\n");
 440         ok(pdp->cArgs == 0, "pdp->cArgs = %d\n", pdp->cArgs);
 441         ok(pdp->cNamedArgs == 0, "pdp->cNamedArgs = %d\n", pdp->cNamedArgs);
 442         ok(!pdp->rgdispidNamedArgs, "pdp->rgdispidNamedArgs != NULL\n");
 443         ok(!pdp->rgvarg, "rgvarg != NULL\n");
 444         ok(pvarRes != NULL, "pvarRes == NULL\n");
 445         ok(pei != NULL, "pei == NULL\n");
 446         ok(!pspCaller, "pspCaller != NULL\n");
 447
 448         V_VT(pvarRes) = VT_NULL;
 449         break;
 450     default:
 451         ok(0, "unexpected call\n");
 452         return E_NOTIMPL;
 453     }
 454
 455     return S_OK;
 456 }
 457
 458 static IDispatchExVtbl scriptDispVtbl = {
 459     DispatchEx_QueryInterface,
 460     DispatchEx_AddRef,
 461     DispatchEx_Release,
 462     DispatchEx_GetTypeInfoCount,
 463     DispatchEx_GetTypeInfo,
 464     DispatchEx_GetIDsOfNames,
 465     DispatchEx_Invoke,
 466     scriptDisp_GetDispID,
 467     scriptDisp_InvokeEx,
 468     DispatchEx_DeleteMemberByName,
 469     DispatchEx_DeleteMemberByDispID,
 470     DispatchEx_GetMemberProperties,
 471     DispatchEx_GetMemberName,
 472     DispatchEx_GetNextDispID,
 473     DispatchEx_GetNameSpaceParent
 474 };
 475
 476 static IDispatchEx scriptDisp = { &scriptDispVtbl };
 477
 478 static HRESULT WINAPI externalDisp_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 479 {
 480     if(!strcmp_wa(bstrName, "ok")) {
 481         *pid = DISPID_EXTERNAL_OK;
 482         return S_OK;
 483     }
 484     if(!strcmp_wa(bstrName, "trace")) {
 485         *pid = DISPID_EXTERNAL_TRACE;
 486         return S_OK;
 487     }
 488     if(!strcmp_wa(bstrName, "reportSuccess")) {
 489         *pid = DISPID_EXTERNAL_REPORTSUCCESS;
 490         return S_OK;
 491     }
 492
 493     ok(0, "unexpected name %s\n", wine_dbgstr_w(bstrName));
 494     return DISP_E_UNKNOWNNAME;
 495 }
 496
 497 static HRESULT WINAPI externalDisp_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 498         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 499 {
 500     switch(id) {
 501     case DISPID_EXTERNAL_OK:
 502         ok(wFlags == INVOKE_FUNC || wFlags == (INVOKE_FUNC|INVOKE_PROPERTYGET), "wFlags = %x\n", wFlags);
 503         ok(pdp != NULL, "pdp == NULL\n");
 504         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
 505         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
 506         ok(pdp->cArgs == 2, "cArgs = %d\n", pdp->cArgs);
 507         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
 508         ok(pei != NULL, "pei == NULL\n");
 509
 510         ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
 511         ok(V_VT(pdp->rgvarg+1) == VT_BOOL, "V_VT(psp->rgvargs+1) = %d\n", V_VT(pdp->rgvarg));
 512         ok(V_BOOL(pdp->rgvarg+1), "%s\n", wine_dbgstr_w(V_BSTR(pdp->rgvarg)));
 513
 514         return S_OK;
 515
 516      case DISPID_EXTERNAL_TRACE:
 517         ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
 518         ok(pdp != NULL, "pdp == NULL\n");
 519         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
 520         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
 521         ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
 522         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
 523         ok(!pvarRes, "pvarRes != NULL\n");
 524         ok(pei != NULL, "pei == NULL\n");
 525
 526         ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
 527         if(V_VT(pdp->rgvarg) == VT_BSTR)
 528             trace("%s\n", wine_dbgstr_w(V_BSTR(pdp->rgvarg)));
 529
 530         return S_OK;
 531
 532     case DISPID_EXTERNAL_REPORTSUCCESS:
 533         CHECK_EXPECT(external_success);
 534
 535         ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
 536         ok(pdp != NULL, "pdp == NULL\n");
 537         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
 538         ok(pdp->cArgs == 0, "cArgs = %d\n", pdp->cArgs);
 539         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
 540         ok(!pvarRes, "pvarRes != NULL\n");
 541         ok(pei != NULL, "pei == NULL\n");
 542
 543         return S_OK;
 544
 545     default:
 546         ok(0, "unexpected call\n");
 547         return E_NOTIMPL;
 548     }
 549
 550     return S_OK;
 551 }
 552
 553 static IDispatchExVtbl externalDispVtbl = {
 554     DispatchEx_QueryInterface,
 555     DispatchEx_AddRef,
 556     DispatchEx_Release,
 557     DispatchEx_GetTypeInfoCount,
 558     DispatchEx_GetTypeInfo,
 559     DispatchEx_GetIDsOfNames,
 560     DispatchEx_Invoke,
 561     externalDisp_GetDispID,
 562     externalDisp_InvokeEx,
 563     DispatchEx_DeleteMemberByName,
 564     DispatchEx_DeleteMemberByDispID,
 565     DispatchEx_GetMemberProperties,
 566     DispatchEx_GetMemberName,
 567     DispatchEx_GetNextDispID,
 568     DispatchEx_GetNameSpaceParent
 569 };
 570
 571 static IDispatchEx externalDisp = { &externalDispVtbl };
 572
 573 static HRESULT QueryInterface(REFIID,void**);
 574
 575 static HRESULT WINAPI DocHostUIHandler_QueryInterface(IDocHostUIHandler2 *iface, REFIID riid, void **ppv)
 576 {
 577     return QueryInterface(riid, ppv);
 578 }
 579
 580 static ULONG WINAPI DocHostUIHandler_AddRef(IDocHostUIHandler2 *iface)
 581 {
 582     return 2;
 583 }
 584
 585 static ULONG WINAPI DocHostUIHandler_Release(IDocHostUIHandler2 *iface)
 586 {
 587     return 1;
 588 }
 589
 590 static HRESULT WINAPI DocHostUIHandler_ShowContextMenu(IDocHostUIHandler2 *iface, DWORD dwID, POINT *ppt,
 591         IUnknown *pcmdtReserved, IDispatch *pdicpReserved)
 592 {
 593     return E_NOTIMPL;
 594 }
 595
 596 static HRESULT WINAPI DocHostUIHandler_GetHostInfo(IDocHostUIHandler2 *iface, DOCHOSTUIINFO *pInfo)
 597 {
 598     return E_NOTIMPL;
 599 }
 600
 601 static HRESULT WINAPI DocHostUIHandler_ShowUI(IDocHostUIHandler2 *iface, DWORD dwID,
 602         IOleInPlaceActiveObject *pActiveObject, IOleCommandTarget *pCommandTarget,
 603         IOleInPlaceFrame *pFrame, IOleInPlaceUIWindow *pDoc)
 604 {
 605     return S_OK;
 606 }
 607
 608 static HRESULT WINAPI DocHostUIHandler_HideUI(IDocHostUIHandler2 *iface)
 609 {
 610     return S_OK;
 611 }
 612
 613 static HRESULT WINAPI DocHostUIHandler_UpdateUI(IDocHostUIHandler2 *iface)
 614 {
 615     return S_OK;
 616 }
 617
 618 static HRESULT WINAPI DocHostUIHandler_EnableModeless(IDocHostUIHandler2 *iface, BOOL fEnable)
 619 {
 620     return E_NOTIMPL;
 621 }
 622
 623 static HRESULT WINAPI DocHostUIHandler_OnDocWindowActivate(IDocHostUIHandler2 *iface, BOOL fActivate)
 624 {
 625     return E_NOTIMPL;
 626 }
 627
 628 static HRESULT WINAPI DocHostUIHandler_OnFrameWindowActivate(IDocHostUIHandler2 *iface, BOOL fActivate)
 629 {
 630     return S_OK;
 631 }
 632
 633 static HRESULT WINAPI DocHostUIHandler_ResizeBorder(IDocHostUIHandler2 *iface, LPCRECT prcBorder,
 634         IOleInPlaceUIWindow *pUIWindow, BOOL fRameWindow)
 635 {
 636     return E_NOTIMPL;
 637 }
 638
 639 static HRESULT WINAPI DocHostUIHandler_TranslateAccelerator(IDocHostUIHandler2 *iface, LPMSG lpMsg,
 640         const GUID *pguidCmdGroup, DWORD nCmdID)
 641 {
 642     return E_NOTIMPL;
 643 }
 644
 645 static HRESULT WINAPI DocHostUIHandler_GetOptionKeyPath(IDocHostUIHandler2 *iface,
 646         LPOLESTR *pchKey, DWORD dw)
 647 {
 648     return S_OK;
 649 }
 650
 651 static HRESULT WINAPI DocHostUIHandler_GetDropTarget(IDocHostUIHandler2 *iface,
 652         IDropTarget *pDropTarget, IDropTarget **ppDropTarget)
 653 {
 654     return E_NOTIMPL;
 655 }
 656
 657 static HRESULT WINAPI DocHostUIHandler_GetExternal(IDocHostUIHandler2 *iface, IDispatch **ppDispatch)
 658 {
 659     *ppDispatch = (IDispatch*)&externalDisp;
 660     return S_OK;
 661 }
 662
 663 static HRESULT WINAPI DocHostUIHandler_TranslateUrl(IDocHostUIHandler2 *iface, DWORD dwTranslate,
 664         OLECHAR *pchURLIn, OLECHAR **ppchURLOut)
 665 {
 666     return S_FALSE;
 667 }
 668
 669 static HRESULT WINAPI DocHostUIHandler_FilterDataObject(IDocHostUIHandler2 *iface, IDataObject *pDO,
 670         IDataObject **ppPORet)
 671 {
 672     return E_NOTIMPL;
 673 }
 674
 675 static HRESULT WINAPI DocHostUIHandler_GetOverrideKeyPath(IDocHostUIHandler2 *iface,
 676         LPOLESTR *pchKey, DWORD dw)
 677 {
 678     return E_NOTIMPL;
 679 }
 680
 681 static const IDocHostUIHandler2Vtbl DocHostUIHandlerVtbl = {
 682     DocHostUIHandler_QueryInterface,
 683     DocHostUIHandler_AddRef,
 684     DocHostUIHandler_Release,
 685     DocHostUIHandler_ShowContextMenu,
 686     DocHostUIHandler_GetHostInfo,
 687     DocHostUIHandler_ShowUI,
 688     DocHostUIHandler_HideUI,
 689     DocHostUIHandler_UpdateUI,
 690     DocHostUIHandler_EnableModeless,
 691     DocHostUIHandler_OnDocWindowActivate,
 692     DocHostUIHandler_OnFrameWindowActivate,
 693     DocHostUIHandler_ResizeBorder,
 694     DocHostUIHandler_TranslateAccelerator,
 695     DocHostUIHandler_GetOptionKeyPath,
 696     DocHostUIHandler_GetDropTarget,
 697     DocHostUIHandler_GetExternal,
 698     DocHostUIHandler_TranslateUrl,
 699     DocHostUIHandler_FilterDataObject,
 700     DocHostUIHandler_GetOverrideKeyPath
 701 };
 702
 703 static IDocHostUIHandler2 DocHostUIHandler = { &DocHostUIHandlerVtbl };
 704
 705 static HRESULT WINAPI InPlaceFrame_QueryInterface(IOleInPlaceFrame *iface, REFIID riid, void **ppv)
 706 {
 707     return E_NOINTERFACE;
 708 }
 709
 710 static ULONG WINAPI InPlaceFrame_AddRef(IOleInPlaceFrame *iface)
 711 {
 712     return 2;
 713 }
 714
 715 static ULONG WINAPI InPlaceFrame_Release(IOleInPlaceFrame *iface)
 716 {
 717     return 1;
 718 }
 719
 720 static HRESULT WINAPI InPlaceFrame_GetWindow(IOleInPlaceFrame *iface, HWND *phwnd)
 721 {
 722     return E_NOTIMPL;
 723 }
 724
 725 static HRESULT WINAPI InPlaceFrame_ContextSensitiveHelp(IOleInPlaceFrame *iface, BOOL fEnterMode)
 726 {
 727     return E_NOTIMPL;
 728 }
 729
 730 static HRESULT WINAPI InPlaceFrame_GetBorder(IOleInPlaceFrame *iface, LPRECT lprectBorder)
 731 {
 732     return E_NOTIMPL;
 733 }
 734
 735 static HRESULT WINAPI InPlaceFrame_RequestBorderSpace(IOleInPlaceFrame *iface,
 736         LPCBORDERWIDTHS pborderwidths)
 737 {
 738     return E_NOTIMPL;
 739 }
 740
 741 static HRESULT WINAPI InPlaceFrame_SetBorderSpace(IOleInPlaceFrame *iface,
 742         LPCBORDERWIDTHS pborderwidths)
 743 {
 744     return S_OK;
 745 }
 746
 747 static HRESULT WINAPI InPlaceFrame_SetActiveObject(IOleInPlaceFrame *iface,
 748         IOleInPlaceActiveObject *pActiveObject, LPCOLESTR pszObjName)
 749 {
 750     return S_OK;
 751 }
 752
 753 static HRESULT WINAPI InPlaceFrame_InsertMenus(IOleInPlaceFrame *iface, HMENU hmenuShared,
 754         LPOLEMENUGROUPWIDTHS lpMenuWidths)
 755 {
 756     return E_NOTIMPL;
 757 }
 758
 759 static HRESULT WINAPI InPlaceFrame_SetMenu(IOleInPlaceFrame *iface, HMENU hmenuShared,
 760         HOLEMENU holemenu, HWND hwndActiveObject)
 761 {
 762     ok(0, "unexpected call\n");
 763     return E_NOTIMPL;
 764 }
 765
 766 static HRESULT WINAPI InPlaceFrame_RemoveMenus(IOleInPlaceFrame *iface, HMENU hmenuShared)
 767 {
 768     ok(0, "unexpected call\n");
 769     return E_NOTIMPL;
 770 }
 771
 772 static HRESULT WINAPI InPlaceFrame_SetStatusText(IOleInPlaceFrame *iface, LPCOLESTR pszStatusText)
 773 {
 774     return S_OK;
 775 }
 776
 777 static HRESULT WINAPI InPlaceFrame_EnableModeless(IOleInPlaceFrame *iface, BOOL fEnable)
 778 {
 779     return E_NOTIMPL;
 780 }
 781
 782 static HRESULT WINAPI InPlaceFrame_TranslateAccelerator(IOleInPlaceFrame *iface, LPMSG lpmsg, WORD wID)
 783 {
 784     ok(0, "unexpected call\n");
 785     return E_NOTIMPL;
 786 }
 787
 788 static const IOleInPlaceFrameVtbl InPlaceFrameVtbl = {
 789     InPlaceFrame_QueryInterface,
 790     InPlaceFrame_AddRef,
 791     InPlaceFrame_Release,
 792     InPlaceFrame_GetWindow,
 793     InPlaceFrame_ContextSensitiveHelp,
 794     InPlaceFrame_GetBorder,
 795     InPlaceFrame_RequestBorderSpace,
 796     InPlaceFrame_SetBorderSpace,
 797     InPlaceFrame_SetActiveObject,
 798     InPlaceFrame_InsertMenus,
 799     InPlaceFrame_SetMenu,
 800     InPlaceFrame_RemoveMenus,
 801     InPlaceFrame_SetStatusText,
 802     InPlaceFrame_EnableModeless,
 803     InPlaceFrame_TranslateAccelerator
 804 };
 805
 806 static IOleInPlaceFrame InPlaceFrame = { &InPlaceFrameVtbl };
 807
 808 static HRESULT WINAPI InPlaceSite_QueryInterface(IOleInPlaceSite *iface, REFIID riid, void **ppv)
 809 {
 810     return QueryInterface(riid, ppv);
 811 }
 812
 813 static ULONG WINAPI InPlaceSite_AddRef(IOleInPlaceSite *iface)
 814 {
 815     return 2;
 816 }
 817
 818 static ULONG WINAPI InPlaceSite_Release(IOleInPlaceSite *iface)
 819 {
 820     return 1;
 821 }
 822
 823 static HRESULT WINAPI InPlaceSite_GetWindow(IOleInPlaceSite *iface, HWND *phwnd)
 824 {
 825     *phwnd = container_hwnd;
 826     return S_OK;
 827 }
 828
 829 static HRESULT WINAPI InPlaceSite_ContextSensitiveHelp(IOleInPlaceSite *iface, BOOL fEnterMode)
 830 {
 831     ok(0, "unexpected call\n");
 832     return E_NOTIMPL;
 833 }
 834
 835 static HRESULT WINAPI InPlaceSite_CanInPlaceActivate(IOleInPlaceSite *iface)
 836 {
 837     return S_OK;
 838 }
 839
 840 static HRESULT WINAPI InPlaceSite_OnInPlaceActivate(IOleInPlaceSite *iface)
 841 {
 842     return S_OK;
 843 }
 844
 845 static HRESULT WINAPI InPlaceSite_OnUIActivate(IOleInPlaceSite *iface)
 846 {
 847     return S_OK;
 848 }
 849
 850 static HRESULT WINAPI InPlaceSite_GetWindowContext(IOleInPlaceSite *iface,
 851         IOleInPlaceFrame **ppFrame, IOleInPlaceUIWindow **ppDoc, LPRECT lprcPosRect,
 852         LPRECT lprcClipRect, LPOLEINPLACEFRAMEINFO lpFrameInfo)
 853 {
 854     static const RECT rect = {0,0,300,300};
 855
 856     *ppFrame = &InPlaceFrame;
 857     *ppDoc = (IOleInPlaceUIWindow*)&InPlaceFrame;
 858     *lprcPosRect = rect;
 859     *lprcClipRect = rect;
 860
 861     lpFrameInfo->cb = sizeof(*lpFrameInfo);
 862     lpFrameInfo->fMDIApp = FALSE;
 863     lpFrameInfo->hwndFrame = container_hwnd;
 864     lpFrameInfo->haccel = NULL;
 865     lpFrameInfo->cAccelEntries = 0;
 866
 867     return S_OK;
 868 }
 869
 870 static HRESULT WINAPI InPlaceSite_Scroll(IOleInPlaceSite *iface, SIZE scrollExtant)
 871 {
 872     return E_NOTIMPL;
 873 }
 874
 875 static HRESULT WINAPI InPlaceSite_OnUIDeactivate(IOleInPlaceSite *iface, BOOL fUndoable)
 876 {
 877     return S_OK;
 878 }
 879
 880 static HRESULT WINAPI InPlaceSite_OnInPlaceDeactivate(IOleInPlaceSite *iface)
 881 {
 882     return S_OK;
 883 }
 884
 885 static HRESULT WINAPI InPlaceSite_DiscardUndoState(IOleInPlaceSite *iface)
 886 {
 887     return E_NOTIMPL;
 888 }
 889
 890 static HRESULT WINAPI InPlaceSite_DeactivateAndUndo(IOleInPlaceSite *iface)
 891 {
 892     return E_NOTIMPL;
 893 }
 894
 895 static HRESULT WINAPI InPlaceSite_OnPosRectChange(IOleInPlaceSite *iface, LPCRECT lprcPosRect)
 896 {
 897     return E_NOTIMPL;
 898 }
 899
 900 static const IOleInPlaceSiteVtbl InPlaceSiteVtbl = {
 901     InPlaceSite_QueryInterface,
 902     InPlaceSite_AddRef,
 903     InPlaceSite_Release,
 904     InPlaceSite_GetWindow,
 905     InPlaceSite_ContextSensitiveHelp,
 906     InPlaceSite_CanInPlaceActivate,
 907     InPlaceSite_OnInPlaceActivate,
 908     InPlaceSite_OnUIActivate,
 909     InPlaceSite_GetWindowContext,
 910     InPlaceSite_Scroll,
 911     InPlaceSite_OnUIDeactivate,
 912     InPlaceSite_OnInPlaceDeactivate,
 913     InPlaceSite_DiscardUndoState,
 914     InPlaceSite_DeactivateAndUndo,
 915     InPlaceSite_OnPosRectChange,
 916 };
 917
 918 static IOleInPlaceSite InPlaceSite = { &InPlaceSiteVtbl };
 919
 920 static HRESULT WINAPI ClientSite_QueryInterface(IOleClientSite *iface, REFIID riid, void **ppv)
 921 {
 922     return QueryInterface(riid, ppv);
 923 }
 924
 925 static ULONG WINAPI ClientSite_AddRef(IOleClientSite *iface)
 926 {
 927     return 2;
 928 }
 929
 930 static ULONG WINAPI ClientSite_Release(IOleClientSite *iface)
 931 {
 932     return 1;
 933 }
 934
 935 static HRESULT WINAPI ClientSite_SaveObject(IOleClientSite *iface)
 936 {
 937     ok(0, "unexpected call\n");
 938     return E_NOTIMPL;
 939 }
 940
 941 static HRESULT WINAPI ClientSite_GetMoniker(IOleClientSite *iface, DWORD dwAssign, DWORD dwWhichMoniker,
 942         IMoniker **ppmon)
 943 {
 944     ok(0, "unexpected call\n");
 945     return E_NOTIMPL;
 946 }
 947
 948 static HRESULT WINAPI ClientSite_GetContainer(IOleClientSite *iface, IOleContainer **ppContainer)
 949 {
 950     return E_NOTIMPL;
 951 }
 952
 953 static HRESULT WINAPI ClientSite_ShowObject(IOleClientSite *iface)
 954 {
 955     ok(0, "unexpected call\n");
 956     return E_NOTIMPL;
 957 }
 958
 959 static HRESULT WINAPI ClientSite_OnShowWindow(IOleClientSite *iface, BOOL fShow)
 960 {
 961     ok(0, "unexpected call\n");
 962     return E_NOTIMPL;
 963 }
 964
 965 static HRESULT WINAPI ClientSite_RequestNewObjectLayout(IOleClientSite *iface)
 966 {
 967     ok(0, "unexpected call\n");
 968     return E_NOTIMPL;
 969 }
 970
 971 static const IOleClientSiteVtbl ClientSiteVtbl = {
 972     ClientSite_QueryInterface,
 973     ClientSite_AddRef,
 974     ClientSite_Release,
 975     ClientSite_SaveObject,
 976     ClientSite_GetMoniker,
 977     ClientSite_GetContainer,
 978     ClientSite_ShowObject,
 979     ClientSite_OnShowWindow,
 980     ClientSite_RequestNewObjectLayout
 981 };
 982
 983 static IOleClientSite ClientSite = { &ClientSiteVtbl };
 984
 985 static HRESULT WINAPI DocumentSite_QueryInterface(IOleDocumentSite *iface, REFIID riid, void **ppv)
 986 {
 987     return QueryInterface(riid, ppv);
 988 }
 989
 990 static ULONG WINAPI DocumentSite_AddRef(IOleDocumentSite *iface)
 991 {
 992     return 2;
 993 }
 994
 995 static ULONG WINAPI DocumentSite_Release(IOleDocumentSite *iface)
 996 {
 997     return 1;
 998 }
 999
1000 static HRESULT WINAPI DocumentSite_ActivateMe(IOleDocumentSite *iface, IOleDocumentView *pViewToActivate)
1001 {
1002     RECT rect = {0,0,300,300};
1003     IOleDocument *document;
1004     HRESULT hres;
1005
1006     hres = IOleDocumentView_QueryInterface(pViewToActivate, &IID_IOleDocument, (void**)&document);
1007     ok(hres == S_OK, "could not get IOleDocument: %08x\n", hres);
1008
1009     hres = IOleDocument_CreateView(document, &InPlaceSite, NULL, 0, &view);
1010     IOleDocument_Release(document);
1011     ok(hres == S_OK, "CreateView failed: %08x\n", hres);
1012
1013     hres = IOleDocumentView_SetInPlaceSite(view, &InPlaceSite);
1014     ok(hres == S_OK, "SetInPlaceSite failed: %08x\n", hres);
1015
1016     hres = IOleDocumentView_UIActivate(view, TRUE);
1017     ok(hres == S_OK, "UIActivate failed: %08x\n", hres);
1018
1019     hres = IOleDocumentView_SetRect(view, &rect);
1020     ok(hres == S_OK, "SetRect failed: %08x\n", hres);
1021
1022     hres = IOleDocumentView_Show(view, TRUE);
1023     ok(hres == S_OK, "Show failed: %08x\n", hres);
1024
1025     return S_OK;
1026 }
1027
1028 static const IOleDocumentSiteVtbl DocumentSiteVtbl = {
1029     DocumentSite_QueryInterface,
1030     DocumentSite_AddRef,
1031     DocumentSite_Release,
1032     DocumentSite_ActivateMe
1033 };
1034
1035 static IOleDocumentSite DocumentSite = { &DocumentSiteVtbl };
1036
1037 static HRESULT QueryInterface(REFIID riid, void **ppv)
1038 {
1039     *ppv = NULL;
1040
1041     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IOleClientSite, riid))
1042         *ppv = &ClientSite;
1043     else if(IsEqualGUID(&IID_IOleDocumentSite, riid))
1044         *ppv = &DocumentSite;
1045     else if(IsEqualGUID(&IID_IOleWindow, riid) || IsEqualGUID(&IID_IOleInPlaceSite, riid))
1046         *ppv = &InPlaceSite;
1047     else if(IsEqualGUID(&IID_IDocHostUIHandler, riid) || IsEqualGUID(&IID_IDocHostUIHandler2, riid))
1048         *ppv = &DocHostUIHandler;
1049
1050     return *ppv ? S_OK : E_NOINTERFACE;
1051 }
1052
1053 static IHTMLDocument2 *create_document(void)
1054 {
1055     IHTMLDocument2 *doc;
1056     IHTMLDocument5 *doc5;
1057     HRESULT hres;
1058
1059     hres = CoCreateInstance(&CLSID_HTMLDocument, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
1060             &IID_IHTMLDocument2, (void**)&doc);
1061     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
1062     if (hres != S_OK) return NULL;
1063
1064     hres = IHTMLDocument2_QueryInterface(doc, &IID_IHTMLDocument5, (void**)&doc5);
1065     if(FAILED(hres)) {
1066         win_skip("Could not get IHTMLDocument5, probably too old IE\n");
1067         IHTMLDocument2_Release(doc);
1068         return NULL;
1069     }
1070
1071     IHTMLDocument5_Release(doc5);
1072     return doc;
1073 }
1074
1075 static void load_string(IHTMLDocument2 *doc, const char *str)
1076 {
1077     IPersistStreamInit *init;
1078     IStream *stream;
1079     HGLOBAL mem;
1080     SIZE_T len;
1081
1082     doc_complete = FALSE;
1083     len = strlen(str);
1084     mem = GlobalAlloc(0, len);
1085     memcpy(mem, str, len);
1086     CreateStreamOnHGlobal(mem, TRUE, &stream);
1087
1088     IHTMLDocument2_QueryInterface(doc, &IID_IPersistStreamInit, (void**)&init);
1089
1090     IPersistStreamInit_Load(init, stream);
1091     IPersistStreamInit_Release(init);
1092     IStream_Release(stream);
1093 }
1094
1095 static void do_advise(IHTMLDocument2 *doc, REFIID riid, IUnknown *unk_advise)
1096 {
1097     IConnectionPointContainer *container;
1098     IConnectionPoint *cp;
1099     DWORD cookie;
1100     HRESULT hres;
1101
1102     hres = IHTMLDocument2_QueryInterface(doc, &IID_IConnectionPointContainer, (void**)&container);
1103     ok(hres == S_OK, "QueryInterface(IID_IConnectionPointContainer) failed: %08x\n", hres);
1104
1105     hres = IConnectionPointContainer_FindConnectionPoint(container, riid, &cp);
1106     IConnectionPointContainer_Release(container);
1107     ok(hres == S_OK, "FindConnectionPoint failed: %08x\n", hres);
1108
1109     notif_doc = doc;
1110
1111     hres = IConnectionPoint_Advise(cp, unk_advise, &cookie);
1112     IConnectionPoint_Release(cp);
1113     ok(hres == S_OK, "Advise failed: %08x\n", hres);
1114 }
1115
1116 static void set_client_site(IHTMLDocument2 *doc, BOOL set)
1117 {
1118     IOleObject *oleobj;
1119     HRESULT hres;
1120
1121     if(!set && view) {
1122         IOleDocumentView_Show(view, FALSE);
1123         IOleDocumentView_CloseView(view, 0);
1124         IOleDocumentView_SetInPlaceSite(view, NULL);
1125         IOleDocumentView_Release(view);
1126         view = NULL;
1127     }
1128
1129     hres = IHTMLDocument2_QueryInterface(doc, &IID_IOleObject, (void**)&oleobj);
1130     ok(hres == S_OK, "Could not et IOleObject: %08x\n", hres);
1131
1132     hres = IOleObject_SetClientSite(oleobj, set ? &ClientSite : NULL);
1133     ok(hres == S_OK, "SetClientSite failed: %08x\n", hres);
1134
1135     if(set) {
1136         IHlinkTarget *hlink;
1137
1138         hres = IOleObject_QueryInterface(oleobj, &IID_IHlinkTarget, (void**)&hlink);
1139         ok(hres == S_OK, "Could not get IHlinkTarget iface: %08x\n", hres);
1140
1141         hres = IHlinkTarget_Navigate(hlink, 0, NULL);
1142         ok(hres == S_OK, "Navgate failed: %08x\n", hres);
1143
1144         IHlinkTarget_Release(hlink);
1145     }
1146
1147     IOleObject_Release(oleobj);
1148 }
1149
1150 typedef void (*domtest_t)(IHTMLDocument2*);
1151
1152 static void load_doc(IHTMLDocument2 *doc, const char *str)
1153 {
1154     IHTMLElement *body = NULL;
1155     MSG msg;
1156     HRESULT hres;
1157     static const WCHAR ucPtr[] = {'b','a','c','k','g','r','o','u','n','d',0};
1158     DISPID dispID = -1;
1159     OLECHAR *name;
1160
1161     load_string(doc, str);
1162     do_advise(doc, &IID_IPropertyNotifySink, (IUnknown*)&PropertyNotifySink);
1163
1164     while(!doc_complete && GetMessage(&msg, NULL, 0, 0)) {
1165         TranslateMessage(&msg);
1166         DispatchMessage(&msg);
1167     }
1168
1169     hres = IHTMLDocument2_get_body(doc, &body);
1170     ok(hres == S_OK, "get_body failed: %08x\n", hres);
1171
1172     /* Check we can query for function on the IHTMLElementBody interface */
1173     name = (WCHAR*)ucPtr;
1174     hres = IHTMLElement_GetIDsOfNames(body, &IID_NULL, &name, 1, LOCALE_USER_DEFAULT, &dispID);
1175     ok(hres == S_OK, "GetIDsOfNames(background) failed %08x\n", hres);
1176     ok(dispID == DISPID_IHTMLBODYELEMENT_BACKGROUND, "Incorrect dispID got (%d)\n", dispID);
1177
1178     IHTMLElement_Release(body);
1179 }
1180
1181 static IActiveScriptSite *site;
1182 static SCRIPTSTATE state;
1183
1184 static HRESULT WINAPI ObjectSafety_QueryInterface(IObjectSafety *iface, REFIID riid, void **ppv)
1185 {
1186     *ppv = NULL;
1187     ok(0, "unexpected call %s\n", debugstr_guid(riid));
1188     return E_NOINTERFACE;
1189 }
1190
1191 static ULONG WINAPI ObjectSafety_AddRef(IObjectSafety *iface)
1192 {
1193     return 2;
1194 }
1195
1196 static ULONG WINAPI ObjectSafety_Release(IObjectSafety *iface)
1197 {
1198     return 1;
1199 }
1200
1201 static HRESULT WINAPI ObjectSafety_GetInterfaceSafetyOptions(IObjectSafety *iface, REFIID riid,
1202         DWORD *pdwSupportedOptions, DWORD *pdwEnabledOptions)
1203 {
1204     CHECK_EXPECT(GetInterfaceSafetyOptions);
1205
1206     ok(IsEqualGUID(&IID_IActiveScriptParse, riid), "unexpected riid %s\n", debugstr_guid(riid));
1207     ok(pdwSupportedOptions != NULL, "pdwSupportedOptions == NULL\n");
1208     ok(pdwEnabledOptions != NULL, "pdwEnabledOptions == NULL\n");
1209
1210     *pdwSupportedOptions = INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER;
1211     *pdwEnabledOptions = INTERFACE_USES_DISPEX;
1212
1213     return S_OK;
1214 }
1215
1216 static HRESULT WINAPI ObjectSafety_SetInterfaceSafetyOptions(IObjectSafety *iface, REFIID riid,
1217         DWORD dwOptionSetMask, DWORD dwEnabledOptions)
1218 {
1219     CHECK_EXPECT(SetInterfaceSafetyOptions);
1220
1221     ok(IsEqualGUID(&IID_IActiveScriptParse, riid), "unexpected riid %s\n", debugstr_guid(riid));
1222
1223     ok(dwOptionSetMask == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
1224        "dwOptionSetMask=%x\n", dwOptionSetMask);
1225     ok(dwEnabledOptions == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
1226        "dwEnabledOptions=%x\n", dwOptionSetMask);
1227
1228     return S_OK;
1229 }
1230
1231 static const IObjectSafetyVtbl ObjectSafetyVtbl = {
1232     ObjectSafety_QueryInterface,
1233     ObjectSafety_AddRef,
1234     ObjectSafety_Release,
1235     ObjectSafety_GetInterfaceSafetyOptions,
1236     ObjectSafety_SetInterfaceSafetyOptions
1237 };
1238
1239 static IObjectSafety ObjectSafety = { &ObjectSafetyVtbl };
1240
1241 static HRESULT WINAPI AXObjectSafety_QueryInterface(IObjectSafety *iface, REFIID riid, void **ppv)
1242 {
1243     *ppv = NULL;
1244
1245     if(IsEqualGUID(&IID_IActiveScript, riid)) {
1246         CHECK_EXPECT(AXQueryInterface_IActiveScript);
1247         return E_NOINTERFACE;
1248     }
1249
1250     if(IsEqualGUID(&IID_IObjectSafety, riid)) {
1251         CHECK_EXPECT(AXQueryInterface_IObjectSafety);
1252         if(!ax_objsafe)
1253             return E_NOINTERFACE;
1254         *ppv = iface;
1255         return S_OK;
1256     }
1257
1258     ok(0, "unexpected call %s\n", debugstr_guid(riid));
1259     return E_NOINTERFACE;
1260 }
1261
1262 static HRESULT WINAPI AXObjectSafety_GetInterfaceSafetyOptions(IObjectSafety *iface, REFIID riid,
1263         DWORD *pdwSupportedOptions, DWORD *pdwEnabledOptions)
1264 {
1265     CHECK_EXPECT(AXGetInterfaceSafetyOptions);
1266
1267     ok(IsEqualGUID(&IID_IDispatchEx, riid), "unexpected riid %s\n", debugstr_guid(riid));
1268     ok(pdwSupportedOptions != NULL, "pdwSupportedOptions == NULL\n");
1269     ok(pdwEnabledOptions != NULL, "pdwEnabledOptions == NULL\n");
1270
1271     if(SUCCEEDED(ax_getopt_hres)) {
1272         *pdwSupportedOptions = INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER;
1273         *pdwEnabledOptions = INTERFACE_USES_DISPEX;
1274     }
1275
1276     return ax_getopt_hres;
1277 }
1278
1279 static HRESULT WINAPI AXObjectSafety_SetInterfaceSafetyOptions(IObjectSafety *iface, REFIID riid,
1280         DWORD dwOptionSetMask, DWORD dwEnabledOptions)
1281 {
1282     if(IsEqualGUID(&IID_IDispatchEx, riid)) {
1283         CHECK_EXPECT(AXSetInterfaceSafetyOptions_IDispatchEx);
1284         ok(dwOptionSetMask == ax_setopt, "dwOptionSetMask=%x, expected %x\n", dwOptionSetMask, ax_setopt);
1285         ok(dwEnabledOptions == ax_setopt, "dwEnabledOptions=%x, expected %x\n", dwOptionSetMask, ax_setopt);
1286         return ax_setopt_dispex_hres;
1287     }
1288
1289     if(IsEqualGUID(&IID_IDispatch, riid)) {
1290         DWORD exopt = ax_setopt & ~INTERFACE_USES_SECURITY_MANAGER;
1291
1292         CHECK_EXPECT(AXSetInterfaceSafetyOptions_IDispatch);
1293         ok(dwOptionSetMask == exopt, "dwOptionSetMask=%x, expected %x\n", dwOptionSetMask, exopt);
1294         ok(dwEnabledOptions == exopt, "dwEnabledOptions=%x, expected %x\n", dwOptionSetMask, exopt);
1295         return ax_setopt_disp_hres;
1296     }
1297
1298     ok(0, "unexpected riid %s\n", debugstr_guid(riid));
1299     return E_NOINTERFACE;
1300 }
1301
1302 static const IObjectSafetyVtbl AXObjectSafetyVtbl = {
1303     AXObjectSafety_QueryInterface,
1304     ObjectSafety_AddRef,
1305     ObjectSafety_Release,
1306     AXObjectSafety_GetInterfaceSafetyOptions,
1307     AXObjectSafety_SetInterfaceSafetyOptions
1308 };
1309
1310 static IObjectSafety AXObjectSafety = { &AXObjectSafetyVtbl };
1311
1312 static BOOL set_safe_reg(BOOL init)
1313 {
1314     return init_key("CLSID\\"TESTACTIVEX_CLSID"\\Implemented Categories\\{7dd95801-9882-11cf-9fa9-00aa006c42c4}",
1315                     NULL, init);
1316 }
1317
1318 static void test_security(void)
1319 {
1320     IInternetHostSecurityManager *sec_mgr;
1321     IServiceProvider *sp;
1322     DWORD policy, policy_size;
1323     struct CONFIRMSAFETY cs;
1324     BYTE *ppolicy;
1325     HRESULT hres;
1326
1327     ax_setopt = INTERFACESAFE_FOR_UNTRUSTED_CALLER|INTERFACE_USES_SECURITY_MANAGER;
1328
1329     hres = IActiveScriptSite_QueryInterface(site, &IID_IServiceProvider, (void**)&sp);
1330     ok(hres == S_OK, "Could not get IServiceProvider iface: %08x\n", hres);
1331
1332     hres = IServiceProvider_QueryService(sp, &SID_SInternetHostSecurityManager,
1333             &IID_IInternetHostSecurityManager, (void**)&sec_mgr);
1334     IServiceProvider_Release(sp);
1335     ok(hres == S_OK, "QueryService failed: %08x\n", hres);
1336
1337     hres = IInternetHostSecurityManager_ProcessUrlAction(sec_mgr, URLACTION_ACTIVEX_RUN, (BYTE*)&policy, sizeof(policy),
1338                                                          (BYTE*)&CLSID_TestActiveX, sizeof(CLSID), 0, 0);
1339     ok(hres == S_OK, "ProcessUrlAction failed: %08x\n", hres);
1340     ok(policy == URLPOLICY_ALLOW, "policy = %x\n", policy);
1341
1342     cs.clsid = CLSID_TestActiveX;
1343     cs.pUnk = (IUnknown*)&AXObjectSafety;
1344     cs.dwFlags = 0;
1345
1346     ax_objsafe = TRUE;
1347     SET_EXPECT(AXQueryInterface_IActiveScript);
1348     SET_EXPECT(AXQueryInterface_IObjectSafety);
1349     SET_EXPECT(AXGetInterfaceSafetyOptions);
1350     SET_EXPECT(AXSetInterfaceSafetyOptions_IDispatchEx);
1351     hres = IInternetHostSecurityManager_QueryCustomPolicy(sec_mgr, &GUID_CUSTOM_CONFIRMOBJECTSAFETY,
1352             &ppolicy, &policy_size, (BYTE*)&cs, sizeof(cs), 0);
1353     CHECK_CALLED(AXQueryInterface_IActiveScript);
1354     CHECK_CALLED(AXQueryInterface_IObjectSafety);
1355     CHECK_CALLED(AXGetInterfaceSafetyOptions);
1356     CHECK_CALLED(AXSetInterfaceSafetyOptions_IDispatchEx);
1357
1358     ok(hres == S_OK, "QueryCusromPolicy failed: %08x\n", hres);
1359     ok(policy_size == sizeof(DWORD), "policy_size = %d\n", policy_size);
1360     ok(*(DWORD*)ppolicy == URLPOLICY_ALLOW, "policy = %x\n", *(DWORD*)ppolicy);
1361     CoTaskMemFree(ppolicy);
1362
1363     ax_objsafe = FALSE;
1364     SET_EXPECT(AXQueryInterface_IActiveScript);
1365     SET_EXPECT(AXQueryInterface_IObjectSafety);
1366     hres = IInternetHostSecurityManager_QueryCustomPolicy(sec_mgr, &GUID_CUSTOM_CONFIRMOBJECTSAFETY,
1367             &ppolicy, &policy_size, (BYTE*)&cs, sizeof(cs), 0);
1368     CHECK_CALLED(AXQueryInterface_IActiveScript);
1369     CHECK_CALLED(AXQueryInterface_IObjectSafety);
1370
1371     ok(hres == S_OK, "QueryCusromPolicy failed: %08x\n", hres);
1372     ok(policy_size == sizeof(DWORD), "policy_size = %d\n", policy_size);
1373     ok(*(DWORD*)ppolicy == URLPOLICY_DISALLOW, "policy = %x\n", *(DWORD*)ppolicy);
1374     CoTaskMemFree(ppolicy);
1375
1376     if(set_safe_reg(TRUE)) {
1377         ax_objsafe = FALSE;
1378         SET_EXPECT(AXQueryInterface_IActiveScript);
1379         SET_EXPECT(AXQueryInterface_IObjectSafety);
1380         hres = IInternetHostSecurityManager_QueryCustomPolicy(sec_mgr, &GUID_CUSTOM_CONFIRMOBJECTSAFETY,
1381                  &ppolicy, &policy_size, (BYTE*)&cs, sizeof(cs), 0);
1382         CHECK_CALLED(AXQueryInterface_IActiveScript);
1383         CHECK_CALLED(AXQueryInterface_IObjectSafety);
1384
1385         ok(hres == S_OK, "QueryCusromPolicy failed: %08x\n", hres);
1386         ok(policy_size == sizeof(DWORD), "policy_size = %d\n", policy_size);
1387         ok(*(DWORD*)ppolicy == URLPOLICY_ALLOW, "policy = %x\n", *(DWORD*)ppolicy);
1388         CoTaskMemFree(ppolicy);
1389
1390         ax_objsafe = TRUE;
1391         SET_EXPECT(AXQueryInterface_IActiveScript);
1392         SET_EXPECT(AXQueryInterface_IObjectSafety);
1393         SET_EXPECT(AXGetInterfaceSafetyOptions);
1394         SET_EXPECT(AXSetInterfaceSafetyOptions_IDispatchEx);
1395         hres = IInternetHostSecurityManager_QueryCustomPolicy(sec_mgr, &GUID_CUSTOM_CONFIRMOBJECTSAFETY,
1396                 &ppolicy, &policy_size, (BYTE*)&cs, sizeof(cs), 0);
1397         CHECK_CALLED(AXQueryInterface_IActiveScript);
1398         CHECK_CALLED(AXQueryInterface_IObjectSafety);
1399         CHECK_CALLED(AXGetInterfaceSafetyOptions);
1400         CHECK_CALLED(AXSetInterfaceSafetyOptions_IDispatchEx);
1401
1402         ok(hres == S_OK, "QueryCusromPolicy failed: %08x\n", hres);
1403         ok(policy_size == sizeof(DWORD), "policy_size = %d\n", policy_size);
1404         ok(*(DWORD*)ppolicy == URLPOLICY_ALLOW, "policy = %x\n", *(DWORD*)ppolicy);
1405         CoTaskMemFree(ppolicy);
1406
1407         set_safe_reg(FALSE);
1408     }else {
1409         skip("Could not set safety registry\n");
1410     }
1411
1412     ax_objsafe = TRUE;
1413
1414     ax_setopt_dispex_hres = E_NOINTERFACE;
1415     SET_EXPECT(AXQueryInterface_IActiveScript);
1416     SET_EXPECT(AXQueryInterface_IObjectSafety);
1417     SET_EXPECT(AXGetInterfaceSafetyOptions);
1418     SET_EXPECT(AXSetInterfaceSafetyOptions_IDispatchEx);
1419     SET_EXPECT(AXSetInterfaceSafetyOptions_IDispatch);
1420     hres = IInternetHostSecurityManager_QueryCustomPolicy(sec_mgr, &GUID_CUSTOM_CONFIRMOBJECTSAFETY,
1421             &ppolicy, &policy_size, (BYTE*)&cs, sizeof(cs), 0);
1422     CHECK_CALLED(AXQueryInterface_IActiveScript);
1423     CHECK_CALLED(AXQueryInterface_IObjectSafety);
1424     CHECK_CALLED(AXGetInterfaceSafetyOptions);
1425     CHECK_CALLED(AXSetInterfaceSafetyOptions_IDispatchEx);
1426     CHECK_CALLED(AXSetInterfaceSafetyOptions_IDispatch);
1427
1428     ok(hres == S_OK, "QueryCusromPolicy failed: %08x\n", hres);
1429     ok(policy_size == sizeof(DWORD), "policy_size = %d\n", policy_size);
1430     ok(*(DWORD*)ppolicy == URLPOLICY_ALLOW, "policy = %x\n", *(DWORD*)ppolicy);
1431     CoTaskMemFree(ppolicy);
1432
1433     ax_setopt_dispex_hres = E_FAIL;
1434     ax_setopt_disp_hres = E_NOINTERFACE;
1435     SET_EXPECT(AXQueryInterface_IActiveScript);
1436     SET_EXPECT(AXQueryInterface_IObjectSafety);
1437     SET_EXPECT(AXGetInterfaceSafetyOptions);
1438     SET_EXPECT(AXSetInterfaceSafetyOptions_IDispatchEx);
1439     SET_EXPECT(AXSetInterfaceSafetyOptions_IDispatch);
1440     hres = IInternetHostSecurityManager_QueryCustomPolicy(sec_mgr, &GUID_CUSTOM_CONFIRMOBJECTSAFETY,
1441             &ppolicy, &policy_size, (BYTE*)&cs, sizeof(cs), 0);
1442     CHECK_CALLED(AXQueryInterface_IActiveScript);
1443     CHECK_CALLED(AXQueryInterface_IObjectSafety);
1444     CHECK_CALLED(AXGetInterfaceSafetyOptions);
1445     CHECK_CALLED(AXSetInterfaceSafetyOptions_IDispatchEx);
1446     CHECK_CALLED(AXSetInterfaceSafetyOptions_IDispatch);
1447
1448     ok(hres == S_OK, "QueryCusromPolicy failed: %08x\n", hres);
1449     ok(policy_size == sizeof(DWORD), "policy_size = %d\n", policy_size);
1450     ok(*(DWORD*)ppolicy == URLPOLICY_DISALLOW, "policy = %x\n", *(DWORD*)ppolicy);
1451     CoTaskMemFree(ppolicy);
1452
1453     ax_setopt_dispex_hres = E_FAIL;
1454     ax_setopt_disp_hres = S_OK;
1455     ax_getopt_hres = E_NOINTERFACE;
1456     ax_setopt = INTERFACESAFE_FOR_UNTRUSTED_CALLER;
1457     SET_EXPECT(AXQueryInterface_IActiveScript);
1458     SET_EXPECT(AXQueryInterface_IObjectSafety);
1459     SET_EXPECT(AXGetInterfaceSafetyOptions);
1460     SET_EXPECT(AXSetInterfaceSafetyOptions_IDispatchEx);
1461     SET_EXPECT(AXSetInterfaceSafetyOptions_IDispatch);
1462     hres = IInternetHostSecurityManager_QueryCustomPolicy(sec_mgr, &GUID_CUSTOM_CONFIRMOBJECTSAFETY,
1463             &ppolicy, &policy_size, (BYTE*)&cs, sizeof(cs), 0);
1464     CHECK_CALLED(AXQueryInterface_IActiveScript);
1465     CHECK_CALLED(AXQueryInterface_IObjectSafety);
1466     CHECK_CALLED(AXGetInterfaceSafetyOptions);
1467     CHECK_CALLED(AXSetInterfaceSafetyOptions_IDispatchEx);
1468     CHECK_CALLED(AXSetInterfaceSafetyOptions_IDispatch);
1469
1470     ok(hres == S_OK, "QueryCusromPolicy failed: %08x\n", hres);
1471     ok(policy_size == sizeof(DWORD), "policy_size = %d\n", policy_size);
1472     ok(*(DWORD*)ppolicy == URLPOLICY_ALLOW, "policy = %x\n", *(DWORD*)ppolicy);
1473     CoTaskMemFree(ppolicy);
1474
1475     IInternetHostSecurityManager_Release(sec_mgr);
1476 }
1477
1478 static HRESULT WINAPI ActiveScriptProperty_QueryInterface(IActiveScriptProperty *iface, REFIID riid, void **ppv)
1479 {
1480     *ppv = NULL;
1481     ok(0, "unexpected call\n");
1482     return E_NOINTERFACE;
1483 }
1484
1485 static ULONG WINAPI ActiveScriptProperty_AddRef(IActiveScriptProperty *iface)
1486 {
1487     return 2;
1488 }
1489
1490 static ULONG WINAPI ActiveScriptProperty_Release(IActiveScriptProperty *iface)
1491 {
1492     return 1;
1493 }
1494
1495 static HRESULT WINAPI ActiveScriptProperty_GetProperty(IActiveScriptProperty *iface, DWORD dwProperty,
1496         VARIANT *pvarIndex, VARIANT *pvarValue)
1497 {
1498     ok(0, "unexpected call\n");
1499     return E_NOTIMPL;
1500 }
1501
1502 static HRESULT WINAPI ActiveScriptProperty_SetProperty(IActiveScriptProperty *iface, DWORD dwProperty,
1503         VARIANT *pvarIndex, VARIANT *pvarValue)
1504 {
1505     switch(dwProperty) {
1506     case SCRIPTPROP_HACK_TRIDENTEVENTSINK:
1507         CHECK_EXPECT(SetProperty_HACK_TRIDENTEVENTSINK);
1508         ok(V_VT(pvarValue) == VT_BOOL, "V_VT(pvarValue)=%d\n", V_VT(pvarValue));
1509         ok(V_BOOL(pvarValue) == VARIANT_TRUE, "V_BOOL(pvarValue)=%x\n", V_BOOL(pvarValue));
1510         break;
1511     case SCRIPTPROP_INVOKEVERSIONING:
1512         CHECK_EXPECT(SetProperty_INVOKEVERSIONING);
1513         ok(V_VT(pvarValue) == VT_I4, "V_VT(pvarValue)=%d\n", V_VT(pvarValue));
1514         ok(V_I4(pvarValue) == 1, "V_I4(pvarValue)=%d\n", V_I4(pvarValue));
1515         break;
1516     case SCRIPTPROP_ABBREVIATE_GLOBALNAME_RESOLUTION:
1517         CHECK_EXPECT(SetProperty_ABBREVIATE_GLOBALNAME_RESOLUTION);
1518         ok(V_VT(pvarValue) == VT_BOOL, "V_VT(pvarValue)=%d\n", V_VT(pvarValue));
1519         ok(V_BOOL(pvarValue) == VARIANT_TRUE, "V_BOOL(pvarValue)=%x\n", V_BOOL(pvarValue));
1520         break;
1521     default:
1522         ok(0, "unexpected property %x\n", dwProperty);
1523         return E_NOTIMPL;
1524     }
1525
1526     ok(!pvarIndex, "pvarIndex != NULL\n");
1527     ok(pvarValue != NULL, "pvarValue == NULL\n");
1528
1529     return S_OK;
1530 }
1531
1532 static const IActiveScriptPropertyVtbl ActiveScriptPropertyVtbl = {
1533     ActiveScriptProperty_QueryInterface,
1534     ActiveScriptProperty_AddRef,
1535     ActiveScriptProperty_Release,
1536     ActiveScriptProperty_GetProperty,
1537     ActiveScriptProperty_SetProperty
1538 };
1539
1540 static IActiveScriptProperty ActiveScriptProperty = { &ActiveScriptPropertyVtbl };
1541
1542 static HRESULT WINAPI ActiveScriptParseProcedure_QueryInterface(IActiveScriptParseProcedure2 *iface, REFIID riid, void **ppv)
1543 {
1544     *ppv = NULL;
1545     ok(0, "unexpected call\n");
1546     return E_NOINTERFACE;
1547 }
1548
1549 static ULONG WINAPI ActiveScriptParseProcedure_AddRef(IActiveScriptParseProcedure2 *iface)
1550 {
1551     return 2;
1552 }
1553
1554 static ULONG WINAPI ActiveScriptParseProcedure_Release(IActiveScriptParseProcedure2 *iface)
1555 {
1556     return 1;
1557 }
1558
1559 static HRESULT WINAPI ActiveScriptParseProcedure_ParseProcedureText(IActiveScriptParseProcedure2 *iface,
1560         LPCOLESTR pstrCode, LPCOLESTR pstrFormalParams, LPCOLESTR pstrProcedureName,
1561         LPCOLESTR pstrItemName, IUnknown *punkContext, LPCOLESTR pstrDelimiter,
1562         CTXARG_T dwSourceContextCookie, ULONG ulStartingLineNumber, DWORD dwFlags, IDispatch **ppdisp)
1563 {
1564     ok(0, "unexpected call\n");
1565     return E_NOTIMPL;
1566 }
1567
1568 static const IActiveScriptParseProcedure2Vtbl ActiveScriptParseProcedureVtbl = {
1569     ActiveScriptParseProcedure_QueryInterface,
1570     ActiveScriptParseProcedure_AddRef,
1571     ActiveScriptParseProcedure_Release,
1572     ActiveScriptParseProcedure_ParseProcedureText
1573 };
1574
1575 static IActiveScriptParseProcedure2 ActiveScriptParseProcedure = { &ActiveScriptParseProcedureVtbl };
1576
1577 static HRESULT WINAPI ActiveScriptParse_QueryInterface(IActiveScriptParse *iface, REFIID riid, void **ppv)
1578 {
1579     *ppv = NULL;
1580     ok(0, "unexpected call\n");
1581     return E_NOINTERFACE;
1582 }
1583
1584 static ULONG WINAPI ActiveScriptParse_AddRef(IActiveScriptParse *iface)
1585 {
1586     return 2;
1587 }
1588
1589 static ULONG WINAPI ActiveScriptParse_Release(IActiveScriptParse *iface)
1590 {
1591     return 1;
1592 }
1593
1594 static HRESULT WINAPI ActiveScriptParse_InitNew(IActiveScriptParse *iface)
1595 {
1596     CHECK_EXPECT(InitNew);
1597     return S_OK;
1598 }
1599
1600 static HRESULT WINAPI ActiveScriptParse_AddScriptlet(IActiveScriptParse *iface,
1601         LPCOLESTR pstrDefaultName, LPCOLESTR pstrCode, LPCOLESTR pstrItemName,
1602         LPCOLESTR pstrSubItemName, LPCOLESTR pstrEventName, LPCOLESTR pstrDelimiter,
1603         CTXARG_T dwSourceContextCookie, ULONG ulStartingLineNumber, DWORD dwFlags,
1604         BSTR *pbstrName, EXCEPINFO *pexcepinfo)
1605 {
1606     ok(0, "unexpected call\n");
1607     return E_NOTIMPL;
1608 }
1609
1610 static HRESULT dispex_propput(IDispatchEx *obj, DISPID id, DWORD flags, VARIANT *var)
1611 {
1612     DISPID propput_arg = DISPID_PROPERTYPUT;
1613     DISPPARAMS dp = {var, &propput_arg, 1, 1};
1614     EXCEPINFO ei = {0};
1615
1616     return IDispatchEx_InvokeEx(obj, id, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT|flags, &dp, NULL, &ei, NULL);
1617 }
1618
1619 static void test_func(IDispatchEx *obj)
1620 {
1621     DISPID id;
1622     IDispatchEx *dispex;
1623     IDispatch *disp;
1624     EXCEPINFO ei;
1625     DISPPARAMS dp;
1626     BSTR str;
1627     VARIANT var;
1628     HRESULT hres;
1629
1630     str = a2bstr("toString");
1631     hres = IDispatchEx_GetDispID(obj, str, fdexNameCaseSensitive, &id);
1632     SysFreeString(str);
1633     ok(hres == S_OK, "GetDispID failed: %08x\n", hres);
1634     ok(id == DISPID_IOMNAVIGATOR_TOSTRING, "id = %x\n", id);
1635
1636     memset(&dp, 0, sizeof(dp));
1637     memset(&ei, 0, sizeof(ei));
1638     VariantInit(&var);
1639     hres = IDispatchEx_InvokeEx(obj, id, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dp, &var, &ei, NULL);
1640     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1641     ok(V_VT(&var) == VT_DISPATCH, "V_VT(var)=%d\n", V_VT(&var));
1642     ok(V_DISPATCH(&var) != NULL, "V_DISPATCH(var) == NULL\n");
1643     disp = V_DISPATCH(&var);
1644
1645     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1646     IDispatch_Release(disp);
1647     ok(hres == S_OK, "Could not get IDispatchEx iface: %08x\n", hres);
1648
1649     /* FIXME: Test InvokeEx(DISPATCH_METHOD) */
1650
1651     memset(&dp, 0, sizeof(dp));
1652     memset(&ei, 0, sizeof(ei));
1653     VariantInit(&var);
1654     hres = IDispatchEx_Invoke(dispex, DISPID_VALUE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dp, &var, &ei, NULL);
1655     ok(hres == S_OK || broken(E_ACCESSDENIED), "InvokeEx failed: %08x\n", hres);
1656     if(SUCCEEDED(hres)) {
1657         ok(V_VT(&var) == VT_BSTR, "V_VT(var)=%d\n", V_VT(&var));
1658         ok(!strcmp_wa(V_BSTR(&var), "[object]"), "V_BSTR(var) = %s\n", wine_dbgstr_w(V_BSTR(&var)));
1659         VariantClear(&var);
1660     }
1661
1662     V_VT(&var) = VT_I4;
1663     V_I4(&var) = 100;
1664     hres = dispex_propput(obj, id, 0, &var);
1665     ok(hres == E_NOTIMPL, "InvokeEx failed: %08x\n", hres);
1666
1667     IDispatchEx_Release(dispex);
1668 }
1669
1670 static void test_nextdispid(IDispatchEx *dispex)
1671 {
1672     DISPID last_id = DISPID_STARTENUM, id, dyn_id;
1673     BSTR name;
1674     VARIANT var;
1675     HRESULT hres;
1676
1677     name = a2bstr("dynVal");
1678     hres = IDispatchEx_GetDispID(dispex, name, fdexNameCaseSensitive|fdexNameEnsure, &dyn_id);
1679     ok(hres == S_OK, "GetDispID failed: %08x\n", hres);
1680     SysFreeString(name);
1681
1682     V_VT(&var) = VT_EMPTY;
1683     hres = dispex_propput(dispex, dyn_id, 0, &var);
1684
1685     while(last_id != dyn_id) {
1686         hres = IDispatchEx_GetNextDispID(dispex, fdexEnumAll, last_id, &id);
1687         ok(hres == S_OK, "GetNextDispID returned: %08x\n", hres);
1688         ok(id != DISPID_STARTENUM, "id == DISPID_STARTENUM\n");
1689         ok(id != DISPID_IOMNAVIGATOR_TOSTRING, "id == DISPID_IOMNAVIGATOR_TOSTRING\n");
1690
1691         hres = IDispatchEx_GetMemberName(dispex, id, &name);
1692         ok(hres == S_OK, "GetMemberName failed: %08x\n", hres);
1693
1694         if(id == dyn_id)
1695             ok(!strcmp_wa(name, "dynVal"), "name = %s\n", wine_dbgstr_w(name));
1696         else if(id == DISPID_IOMNAVIGATOR_PLATFORM)
1697             ok(!strcmp_wa(name, "platform"), "name = %s\n", wine_dbgstr_w(name));
1698
1699         SysFreeString(name);
1700         last_id = id;
1701     }
1702
1703     hres = IDispatchEx_GetNextDispID(dispex, 0, id, &id);
1704     ok(hres == S_FALSE, "GetNextDispID returned: %08x\n", hres);
1705     ok(id == DISPID_STARTENUM, "id != DISPID_STARTENUM\n");
1706 }
1707
1708 static void test_global_id(void)
1709 {
1710     VARIANT var;
1711     DISPPARAMS dp;
1712     EXCEPINFO ei;
1713     BSTR tmp;
1714     DISPID id;
1715     HRESULT hres;
1716
1717     SET_EXPECT(GetScriptDispatch);
1718     SET_EXPECT(script_divid_d);
1719     tmp = a2bstr("divid");
1720     hres = IDispatchEx_GetDispID(window_dispex, tmp, fdexNameCaseSensitive, &id);
1721     ok(hres == S_OK, "GetDispID failed: %08x\n", hres);
1722     SysFreeString(tmp);
1723     CHECK_CALLED(GetScriptDispatch);
1724     CHECK_CALLED(script_divid_d);
1725
1726     VariantInit(&var);
1727     memset(&ei, 0, sizeof(ei));
1728     memset(&dp, 0, sizeof(dp));
1729     hres = IDispatchEx_InvokeEx(window_dispex, id, 0, DISPATCH_PROPERTYGET, &dp, &var, &ei, NULL);
1730     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1731     ok(V_VT(&var) == VT_DISPATCH, "V_VT(var) = %d\n", V_VT(&var));
1732     VariantClear(&var);
1733 }
1734
1735 static HRESULT WINAPI ActiveScriptParse_ParseScriptText(IActiveScriptParse *iface,
1736         LPCOLESTR pstrCode, LPCOLESTR pstrItemName, IUnknown *punkContext,
1737         LPCOLESTR pstrDelimiter, CTXARG_T dwSourceContextCookie, ULONG ulStartingLine,
1738         DWORD dwFlags, VARIANT *pvarResult, EXCEPINFO *pexcepinfo)
1739 {
1740     IDispatchEx *document, *dispex;
1741     IHTMLWindow2 *window;
1742     IOmNavigator *navigator;
1743     IUnknown *unk;
1744     VARIANT var, arg;
1745     DISPPARAMS dp;
1746     EXCEPINFO ei;
1747     DISPID id;
1748     BSTR tmp;
1749     HRESULT hres;
1750
1751     static const WCHAR documentW[] = {'d','o','c','u','m','e','n','t',0};
1752     static const WCHAR testW[] = {'t','e','s','t',0};
1753     static const WCHAR funcW[] = {'f','u','n','c',0};
1754
1755     CHECK_EXPECT(ParseScriptText);
1756
1757     SET_EXPECT(GetScriptDispatch);
1758
1759     tmp = SysAllocString(documentW);
1760     hres = IDispatchEx_GetDispID(window_dispex, tmp, fdexNameCaseSensitive, &id);
1761     SysFreeString(tmp);
1762     ok(hres == S_OK, "GetDispID(document) failed: %08x\n", hres);
1763     ok(id == DISPID_IHTMLWINDOW2_DOCUMENT, "id=%x\n", id);
1764
1765     CHECK_CALLED(GetScriptDispatch);
1766
1767     VariantInit(&var);
1768     memset(&dp, 0, sizeof(dp));
1769     memset(&ei, 0, sizeof(ei));
1770
1771     hres = IDispatchEx_InvokeEx(window_dispex, id, LOCALE_NEUTRAL, INVOKE_PROPERTYGET, &dp, &var, &ei, NULL);
1772     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1773     ok(V_VT(&var) == VT_DISPATCH, "V_VT(var)=%d\n", V_VT(&var));
1774     ok(V_DISPATCH(&var) != NULL, "V_DISPATCH(&var) == NULL\n");
1775
1776     hres = IDispatch_QueryInterface(V_DISPATCH(&var), &IID_IDispatchEx, (void**)&document);
1777     VariantClear(&var);
1778     ok(hres == S_OK, "Could not get DispatchEx: %08x\n", hres);
1779
1780     tmp = SysAllocString(testW);
1781     hres = IDispatchEx_GetDispID(document, tmp, fdexNameCaseSensitive, &id);
1782     ok(hres == DISP_E_UNKNOWNNAME, "GetDispID(document) failed: %08x, expected DISP_E_UNKNOWNNAME\n", hres);
1783     hres = IDispatchEx_GetDispID(document, tmp, fdexNameCaseSensitive | fdexNameImplicit, &id);
1784     ok(hres == DISP_E_UNKNOWNNAME, "GetDispID(document) failed: %08x, expected DISP_E_UNKNOWNNAME\n", hres);
1785     SysFreeString(tmp);
1786
1787     id = 0;
1788     tmp = SysAllocString(testW);
1789     hres = IDispatchEx_GetDispID(document, tmp, fdexNameCaseSensitive|fdexNameEnsure, &id);
1790     SysFreeString(tmp);
1791     ok(hres == S_OK, "GetDispID(document) failed: %08x\n", hres);
1792     ok(id, "id == 0\n");
1793
1794     V_VT(&var) = VT_I4;
1795     V_I4(&var) = 100;
1796     hres = dispex_propput(document, id, 0, &var);
1797     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1798
1799     tmp = SysAllocString(testW);
1800     hres = IDispatchEx_GetDispID(document, tmp, fdexNameCaseSensitive, &id);
1801     SysFreeString(tmp);
1802     ok(hres == S_OK, "GetDispID(document) failed: %08x\n", hres);
1803
1804     VariantInit(&var);
1805     memset(&dp, 0, sizeof(dp));
1806     memset(&ei, 0, sizeof(ei));
1807     hres = IDispatchEx_InvokeEx(document, id, LOCALE_NEUTRAL, INVOKE_PROPERTYGET, &dp, &var, &ei, NULL);
1808     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1809     ok(V_VT(&var) == VT_I4, "V_VT(var)=%d\n", V_VT(&var));
1810     ok(V_I4(&var) == 100, "V_I4(&var) = %d\n", V_I4(&var));
1811
1812     V_VT(&var) = VT_I4;
1813     V_I4(&var) = 200;
1814     hres = dispex_propput(document, id, DISPATCH_PROPERTYPUTREF, &var);
1815     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1816
1817     VariantInit(&var);
1818     memset(&dp, 0, sizeof(dp));
1819     memset(&ei, 0, sizeof(ei));
1820     hres = IDispatchEx_InvokeEx(document, id, LOCALE_NEUTRAL, INVOKE_PROPERTYGET, &dp, &var, &ei, NULL);
1821     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1822     ok(V_VT(&var) == VT_I4, "V_VT(var)=%d\n", V_VT(&var));
1823     ok(V_I4(&var) == 200, "V_I4(&var) = %d\n", V_I4(&var));
1824
1825     memset(&dp, 0, sizeof(dp));
1826     memset(&ei, 0, sizeof(ei));
1827     V_VT(&var) = VT_I4;
1828     V_I4(&var) = 300;
1829     dp.cArgs = 1;
1830     dp.rgvarg = &var;
1831     hres = IDispatchEx_InvokeEx(document, id, LOCALE_NEUTRAL, INVOKE_PROPERTYPUT, &dp, NULL, &ei, NULL);
1832     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1833
1834     VariantInit(&var);
1835     memset(&dp, 0, sizeof(dp));
1836     memset(&ei, 0, sizeof(ei));
1837     hres = IDispatchEx_InvokeEx(document, id, LOCALE_NEUTRAL, INVOKE_PROPERTYGET, &dp, &var, &ei, NULL);
1838     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1839     ok(V_VT(&var) == VT_I4, "V_VT(var)=%d\n", V_VT(&var));
1840     ok(V_I4(&var) == 300, "V_I4(&var) = %d\n", V_I4(&var));
1841
1842     unk = (void*)0xdeadbeef;
1843     hres = IDispatchEx_GetNameSpaceParent(window_dispex, &unk);
1844     ok(hres == S_OK, "GetNameSpaceParent failed: %08x\n", hres);
1845     ok(!unk, "unk=%p, expected NULL\n", unk);
1846
1847     id = 0;
1848     tmp = SysAllocString(funcW);
1849     hres = IDispatchEx_GetDispID(document, tmp, fdexNameCaseSensitive|fdexNameEnsure, &id);
1850     SysFreeString(tmp);
1851     ok(hres == S_OK, "GetDispID(func) failed: %08x\n", hres);
1852     ok(id, "id == 0\n");
1853
1854     dp.cArgs = 1;
1855     dp.rgvarg = &var;
1856     dp.cNamedArgs = 0;
1857     dp.rgdispidNamedArgs = NULL;
1858     V_VT(&var) = VT_DISPATCH;
1859     V_DISPATCH(&var) = (IDispatch*)&funcDisp;
1860     hres = IDispatchEx_InvokeEx(document, id, LOCALE_NEUTRAL, INVOKE_PROPERTYPUT, &dp, NULL, &ei, NULL);
1861     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1862
1863     VariantInit(&var);
1864     memset(&dp, 0, sizeof(dp));
1865     memset(&ei, 0, sizeof(ei));
1866     V_VT(&arg) = VT_BOOL;
1867     V_BOOL(&arg) = VARIANT_TRUE;
1868     dp.cArgs = 1;
1869     dp.rgvarg = &arg;
1870
1871     SET_EXPECT(funcDisp);
1872     hres = IDispatchEx_InvokeEx(document, id, LOCALE_NEUTRAL, INVOKE_FUNC, &dp, &var, &ei, NULL);
1873     CHECK_CALLED(funcDisp);
1874
1875     ok(hres == S_OK, "InvokeEx(INVOKE_FUNC) failed: %08x\n", hres);
1876     ok(V_VT(&var) == VT_I4, "V_VT(var)=%d\n", V_VT(&var));
1877     ok(V_I4(&var) == 100, "V_I4(&var) == NULL\n");
1878
1879     IDispatchEx_Release(document);
1880
1881     hres = IDispatchEx_QueryInterface(window_dispex, &IID_IHTMLWindow2, (void**)&window);
1882     ok(hres == S_OK, "Could not get IHTMLWindow2 iface: %08x\n", hres);
1883
1884     hres = IHTMLWindow2_get_navigator(window, &navigator);
1885     IHTMLWindow2_Release(window);
1886     ok(hres == S_OK, "get_navigator failed: %08x\n", hres);
1887
1888     hres = IOmNavigator_QueryInterface(navigator, &IID_IDispatchEx, (void**)&dispex);
1889     IOmNavigator_Release(navigator);
1890     ok(hres == S_OK, "Could not get IDispatchEx iface: %08x\n", hres);
1891
1892     test_func(dispex);
1893     test_nextdispid(dispex);
1894
1895     tmp = a2bstr("test");
1896     hres = IDispatchEx_DeleteMemberByName(dispex, tmp, fdexNameCaseSensitive);
1897     ok(hres == E_NOTIMPL, "DeleteMemberByName failed: %08x\n", hres);
1898
1899     IDispatchEx_Release(dispex);
1900
1901     script_disp = (IDispatch*)&scriptDisp;
1902
1903     SET_EXPECT(GetScriptDispatch);
1904     SET_EXPECT(script_testprop_d);
1905     tmp = a2bstr("testProp");
1906     hres = IDispatchEx_GetDispID(window_dispex, tmp, fdexNameCaseSensitive, &id);
1907     ok(hres == S_OK, "GetDispID failed: %08x\n", hres);
1908     ok(id != DISPID_SCRIPT_TESTPROP, "id == DISPID_SCRIPT_TESTPROP\n");
1909     CHECK_CALLED(GetScriptDispatch);
1910     CHECK_CALLED(script_testprop_d);
1911     SysFreeString(tmp);
1912
1913     tmp = a2bstr("testProp");
1914     hres = IDispatchEx_GetDispID(window_dispex, tmp, fdexNameCaseSensitive, &id);
1915     ok(hres == S_OK, "GetDispID failed: %08x\n", hres);
1916     ok(id != DISPID_SCRIPT_TESTPROP, "id == DISPID_SCRIPT_TESTPROP\n");
1917     SysFreeString(tmp);
1918
1919     SET_EXPECT(GetScriptDispatch);
1920     SET_EXPECT(script_testprop_i);
1921     memset(&ei, 0, sizeof(ei));
1922     memset(&dp, 0, sizeof(dp));
1923     hres = IDispatchEx_InvokeEx(window_dispex, id, 0, DISPATCH_PROPERTYGET, &dp, &var, &ei, NULL);
1924     ok(hres == S_OK, "InvokeEx failed: %08x\n", hres);
1925     ok(V_VT(&var) == VT_NULL, "V_VT(var) = %d\n", V_VT(&var));
1926     CHECK_CALLED(GetScriptDispatch);
1927     CHECK_CALLED(script_testprop_i);
1928
1929     SET_EXPECT(GetScriptDispatch);
1930     SET_EXPECT(script_testprop2_d);
1931     tmp = a2bstr("testProp2");
1932     hres = IDispatchEx_GetDispID(window_dispex, tmp, fdexNameCaseSensitive|fdexNameEnsure, &id);
1933     ok(hres == S_OK, "GetDispID failed: %08x\n", hres);
1934     ok(id != DISPID_SCRIPT_TESTPROP2, "id == DISPID_SCRIPT_TESTPROP2\n");
1935     CHECK_CALLED(GetScriptDispatch);
1936     CHECK_CALLED(script_testprop2_d);
1937     SysFreeString(tmp);
1938
1939     tmp = a2bstr("test");
1940     hres = IDispatchEx_DeleteMemberByName(window_dispex, tmp, fdexNameCaseSensitive);
1941     ok(hres == E_NOTIMPL, "DeleteMemberByName failed: %08x\n", hres);
1942
1943     test_global_id();
1944
1945     test_security();
1946
1947     return S_OK;
1948 }
1949
1950 static const IActiveScriptParseVtbl ActiveScriptParseVtbl = {
1951     ActiveScriptParse_QueryInterface,
1952     ActiveScriptParse_AddRef,
1953     ActiveScriptParse_Release,
1954     ActiveScriptParse_InitNew,
1955     ActiveScriptParse_AddScriptlet,
1956     ActiveScriptParse_ParseScriptText
1957 };
1958
1959 static IActiveScriptParse ActiveScriptParse = { &ActiveScriptParseVtbl };
1960
1961 static HRESULT WINAPI ActiveScript_QueryInterface(IActiveScript *iface, REFIID riid, void **ppv)
1962 {
1963     *ppv = NULL;
1964
1965     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IActiveScript, riid)) {
1966         *ppv = iface;
1967         return S_OK;
1968     }
1969
1970     if(IsEqualGUID(&IID_IActiveScriptParse, riid)) {
1971         *ppv = &ActiveScriptParse;
1972         return S_OK;
1973     }
1974
1975     if(IsEqualGUID(&IID_IActiveScriptParseProcedure2, riid)) {
1976         *ppv = &ActiveScriptParseProcedure;
1977         return S_OK;
1978     }
1979
1980     if(IsEqualGUID(&IID_IActiveScriptProperty, riid)) {
1981         *ppv = &ActiveScriptProperty;
1982         return S_OK;
1983     }
1984
1985     if(IsEqualGUID(&IID_IObjectSafety, riid)) {
1986         *ppv = &ObjectSafety;
1987         return S_OK;
1988     }
1989
1990     if(IsEqualGUID(&IID_IActiveScriptDebug, riid))
1991         return E_NOINTERFACE;
1992
1993     ok(0, "unexpected riid %s\n", debugstr_guid(riid));
1994     return E_NOINTERFACE;
1995 }
1996
1997 static ULONG WINAPI ActiveScript_AddRef(IActiveScript *iface)
1998 {
1999     return 2;
2000 }
2001
2002 static ULONG WINAPI ActiveScript_Release(IActiveScript *iface)
2003 {
2004     return 1;
2005 }
2006
2007 static HRESULT WINAPI ActiveScript_SetScriptSite(IActiveScript *iface, IActiveScriptSite *pass)
2008 {
2009     IActiveScriptSiteInterruptPoll *poll;
2010     IActiveScriptSiteDebug *debug;
2011     IServiceProvider *service;
2012     ICanHandleException *canexpection;
2013     LCID lcid;
2014     HRESULT hres;
2015
2016     CHECK_EXPECT(SetScriptSite);
2017
2018     ok(pass != NULL, "pass == NULL\n");
2019
2020     hres = IActiveScriptSite_QueryInterface(pass, &IID_IActiveScriptSiteInterruptPoll, (void**)&poll);
2021     ok(hres == S_OK, "Could not get IActiveScriptSiteInterruptPoll interface: %08x\n", hres);
2022     if(FAILED(hres))
2023         IActiveScriptSiteInterruptPoll_Release(poll);
2024
2025     hres = IActiveScriptSite_GetLCID(pass, &lcid);
2026     ok(hres == S_OK, "GetLCID failed: %08x\n", hres);
2027
2028     hres = IActiveScriptSite_OnStateChange(pass, (state = SCRIPTSTATE_INITIALIZED));
2029     ok(hres == S_OK, "OnStateChange failed: %08x\n", hres);
2030
2031     hres = IActiveScriptSite_QueryInterface(pass, &IID_IActiveScriptSiteDebug, (void**)&debug);
2032     ok(hres == S_OK, "Could not get IActiveScriptSiteDebug interface: %08x\n", hres);
2033     if(SUCCEEDED(hres))
2034         IActiveScriptSiteDebug32_Release(debug);
2035
2036     hres = IActiveScriptSite_QueryInterface(pass, &IID_ICanHandleException, (void**)&canexpection);
2037     ok(hres == E_NOINTERFACE, "Could not get IID_ICanHandleException interface: %08x\n", hres);
2038
2039     hres = IActiveScriptSite_QueryInterface(pass, &IID_IServiceProvider, (void**)&service);
2040     ok(hres == S_OK, "Could not get IServiceProvider interface: %08x\n", hres);
2041     if(SUCCEEDED(hres))
2042         IServiceProvider_Release(service);
2043
2044     site = pass;
2045     IActiveScriptSite_AddRef(site);
2046     return S_OK;
2047 }
2048
2049 static HRESULT WINAPI ActiveScript_GetScriptSite(IActiveScript *iface, REFIID riid,
2050                                             void **ppvObject)
2051 {
2052     ok(0, "unexpected call\n");
2053     return E_NOTIMPL;
2054 }
2055
2056 static HRESULT WINAPI ActiveScript_SetScriptState(IActiveScript *iface, SCRIPTSTATE ss)
2057 {
2058     HRESULT hres;
2059
2060     switch(ss) {
2061     case SCRIPTSTATE_STARTED:
2062         CHECK_EXPECT(SetScriptState_STARTED);
2063         break;
2064     case SCRIPTSTATE_CONNECTED:
2065         CHECK_EXPECT(SetScriptState_CONNECTED);
2066         break;
2067     case SCRIPTSTATE_DISCONNECTED:
2068         CHECK_EXPECT(SetScriptState_DISCONNECTED);
2069         break;
2070     default:
2071         ok(0, "unexpected state %d\n", ss);
2072         return E_NOTIMPL;
2073     }
2074
2075     hres = IActiveScriptSite_OnStateChange(site, (state = ss));
2076     ok(hres == S_OK, "OnStateChange failed: %08x\n", hres);
2077
2078     return S_OK;
2079 }
2080
2081 static HRESULT WINAPI ActiveScript_GetScriptState(IActiveScript *iface, SCRIPTSTATE *pssState)
2082 {
2083     CHECK_EXPECT(GetScriptState);
2084
2085     *pssState = state;
2086     return S_OK;
2087 }
2088
2089 static HRESULT WINAPI ActiveScript_Close(IActiveScript *iface)
2090 {
2091     CHECK_EXPECT(Close);
2092     return E_NOTIMPL;
2093 }
2094
2095 static HRESULT WINAPI ActiveScript_AddNamedItem(IActiveScript *iface,
2096         LPCOLESTR pstrName, DWORD dwFlags)
2097 {
2098     IDispatch *disp;
2099     IUnknown *unk = NULL, *unk2;
2100     HRESULT hres;
2101
2102     static const WCHAR windowW[] = {'w','i','n','d','o','w',0};
2103
2104     static const IID unknown_iid = {0x719C3050,0xF9D3,0x11CF,{0xA4,0x93,0x00,0x40,0x05,0x23,0xA8,0xA0}};
2105
2106     CHECK_EXPECT(AddNamedItem);
2107
2108     ok(!lstrcmpW(pstrName, windowW), "pstrName=%s\n", wine_dbgstr_w(pstrName));
2109     ok(dwFlags == (SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS), "dwFlags=%x\n", dwFlags);
2110
2111     hres = IActiveScriptSite_GetItemInfo(site, windowW, SCRIPTINFO_IUNKNOWN, &unk, NULL);
2112     ok(hres == S_OK, "GetItemInfo failed: %08x\n", hres);
2113     ok(unk != NULL, "unk == NULL\n");
2114
2115     hres = IUnknown_QueryInterface(unk, &IID_IDispatch, (void**)&disp);
2116     ok(hres == S_OK, "Could not get IDispatch interface: %08x\n", hres);
2117     if(SUCCEEDED(hres))
2118         IDispatch_Release(disp);
2119
2120     hres = IUnknown_QueryInterface(unk, &unknown_iid, (void**)&unk2);
2121     ok(hres == E_NOINTERFACE, "Got ?? interface: %p\n", unk2);
2122     if(SUCCEEDED(hres))
2123         IUnknown_Release(unk2);
2124
2125     hres = IUnknown_QueryInterface(unk, &IID_IDispatchEx, (void**)&window_dispex);
2126     ok(hres == S_OK, "Could not get IDispatchEx interface: %08x\n", hres);
2127
2128     IUnknown_Release(unk);
2129     return S_OK;
2130 }
2131
2132 static HRESULT WINAPI ActiveScript_AddTypeLib(IActiveScript *iface, REFGUID rguidTypeLib,
2133                                          DWORD dwMajor, DWORD dwMinor, DWORD dwFlags)
2134 {
2135     ok(0, "unexpected call\n");
2136     return E_NOTIMPL;
2137 }
2138
2139 static HRESULT WINAPI ActiveScript_GetScriptDispatch(IActiveScript *iface, LPCOLESTR pstrItemName,
2140                                                 IDispatch **ppdisp)
2141 {
2142     CHECK_EXPECT(GetScriptDispatch);
2143
2144     ok(!strcmp_wa(pstrItemName, "window"), "pstrItemName = %s\n", wine_dbgstr_w(pstrItemName));
2145
2146     if(!script_disp)
2147         return E_NOTIMPL;
2148
2149     *ppdisp = script_disp;
2150     return S_OK;
2151 }
2152
2153 static HRESULT WINAPI ActiveScript_GetCurrentScriptThreadID(IActiveScript *iface,
2154                                                        SCRIPTTHREADID *pstridThread)
2155 {
2156     ok(0, "unexpected call\n");
2157     return E_NOTIMPL;
2158 }
2159
2160 static HRESULT WINAPI ActiveScript_GetScriptThreadID(IActiveScript *iface,
2161                                                 DWORD dwWin32ThreadId, SCRIPTTHREADID *pstidThread)
2162 {
2163     ok(0, "unexpected call\n");
2164     return E_NOTIMPL;
2165 }
2166
2167 static HRESULT WINAPI ActiveScript_GetScriptThreadState(IActiveScript *iface,
2168         SCRIPTTHREADID stidThread, SCRIPTTHREADSTATE *pstsState)
2169 {
2170     ok(0, "unexpected call\n");
2171     return E_NOTIMPL;
2172 }
2173
2174 static HRESULT WINAPI ActiveScript_InterruptScriptThread(IActiveScript *iface,
2175         SCRIPTTHREADID stidThread, const EXCEPINFO *pexcepinfo, DWORD dwFlags)
2176 {
2177     ok(0, "unexpected call\n");
2178     return E_NOTIMPL;
2179 }
2180
2181 static HRESULT WINAPI ActiveScript_Clone(IActiveScript *iface, IActiveScript **ppscript)
2182 {
2183     ok(0, "unexpected call\n");
2184     return E_NOTIMPL;
2185 }
2186
2187 static const IActiveScriptVtbl ActiveScriptVtbl = {
2188     ActiveScript_QueryInterface,
2189     ActiveScript_AddRef,
2190     ActiveScript_Release,
2191     ActiveScript_SetScriptSite,
2192     ActiveScript_GetScriptSite,
2193     ActiveScript_SetScriptState,
2194     ActiveScript_GetScriptState,
2195     ActiveScript_Close,
2196     ActiveScript_AddNamedItem,
2197     ActiveScript_AddTypeLib,
2198     ActiveScript_GetScriptDispatch,
2199     ActiveScript_GetCurrentScriptThreadID,
2200     ActiveScript_GetScriptThreadID,
2201     ActiveScript_GetScriptThreadState,
2202     ActiveScript_InterruptScriptThread,
2203     ActiveScript_Clone
2204 };
2205
2206 static IActiveScript ActiveScript = { &ActiveScriptVtbl };
2207
2208 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
2209 {
2210     *ppv = NULL;
2211
2212     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IClassFactory, riid)) {
2213         *ppv = iface;
2214         return S_OK;
2215     }
2216
2217     if(IsEqualGUID(&IID_IMarshal, riid))
2218         return E_NOINTERFACE;
2219     if(IsEqualGUID(&CLSID_IdentityUnmarshal, riid))
2220         return E_NOINTERFACE;
2221
2222     ok(0, "unexpected riid %s\n", debugstr_guid(riid));
2223     return E_NOTIMPL;
2224 }
2225
2226 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
2227 {
2228     return 2;
2229 }
2230
2231 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
2232 {
2233     return 1;
2234 }
2235
2236 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *outer, REFIID riid, void **ppv)
2237 {
2238     CHECK_EXPECT(CreateInstance);
2239
2240     ok(!outer, "outer = %p\n", outer);
2241     ok(IsEqualGUID(&IID_IActiveScript, riid), "unexpected riid %s\n", debugstr_guid(riid));
2242     *ppv = &ActiveScript;
2243     return S_OK;
2244 }
2245
2246 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
2247 {
2248     ok(0, "unexpected call\n");
2249     return S_OK;
2250 }
2251
2252 static const IClassFactoryVtbl ClassFactoryVtbl = {
2253     ClassFactory_QueryInterface,
2254     ClassFactory_AddRef,
2255     ClassFactory_Release,
2256     ClassFactory_CreateInstance,
2257     ClassFactory_LockServer
2258 };
2259
2260 static IClassFactory script_cf = { &ClassFactoryVtbl };
2261
2262 static const char simple_script_str[] =
2263     "<html><head></head><body>"
2264     "<div id=\"divid\"></div>"
2265     "<script language=\"TestScript\">simple script</script>"
2266     "</body></html>";
2267
2268 static void test_simple_script(void)
2269 {
2270     IHTMLDocument2 *doc;
2271
2272     doc = create_document();
2273     if(!doc)
2274         return;
2275
2276     SET_EXPECT(CreateInstance);
2277     SET_EXPECT(GetInterfaceSafetyOptions);
2278     SET_EXPECT(SetInterfaceSafetyOptions);
2279     SET_EXPECT(SetProperty_INVOKEVERSIONING); /* IE8 */
2280     SET_EXPECT(SetProperty_HACK_TRIDENTEVENTSINK);
2281     SET_EXPECT(InitNew);
2282     SET_EXPECT(SetScriptSite);
2283     SET_EXPECT(GetScriptState);
2284     SET_EXPECT(SetScriptState_STARTED);
2285     SET_EXPECT(AddNamedItem);
2286     SET_EXPECT(SetProperty_ABBREVIATE_GLOBALNAME_RESOLUTION); /* IE8 */
2287     SET_EXPECT(ParseScriptText);
2288     SET_EXPECT(SetScriptState_CONNECTED);
2289
2290     load_doc(doc, simple_script_str);
2291
2292     CHECK_CALLED(CreateInstance);
2293     CHECK_CALLED(GetInterfaceSafetyOptions);
2294     CHECK_CALLED(SetInterfaceSafetyOptions);
2295     CHECK_CALLED_BROKEN(SetProperty_INVOKEVERSIONING); /* IE8 */
2296     CHECK_CALLED(SetProperty_HACK_TRIDENTEVENTSINK);
2297     CHECK_CALLED(InitNew);
2298     CHECK_CALLED(SetScriptSite);
2299     CHECK_CALLED(GetScriptState);
2300     CHECK_CALLED(SetScriptState_STARTED);
2301     CHECK_CALLED(AddNamedItem);
2302     CHECK_CALLED_BROKEN(SetProperty_ABBREVIATE_GLOBALNAME_RESOLUTION); /* IE8 */
2303     CHECK_CALLED(ParseScriptText);
2304     CHECK_CALLED(SetScriptState_CONNECTED);
2305
2306     if(site)
2307         IActiveScriptSite_Release(site);
2308     if(window_dispex)
2309         IDispatchEx_Release(window_dispex);
2310
2311     SET_EXPECT(SetScriptState_DISCONNECTED);
2312     SET_EXPECT(Close);
2313
2314     IHTMLDocument2_Release(doc);
2315
2316     CHECK_CALLED(SetScriptState_DISCONNECTED);
2317     CHECK_CALLED(Close);
2318 }
2319
2320 static void run_js_script(const char *test_name)
2321 {
2322     WCHAR url[INTERNET_MAX_URL_LENGTH];
2323     char urlA[INTERNET_MAX_URL_LENGTH];
2324     IPersistMoniker *persist;
2325     IHTMLDocument2 *doc;
2326     IMoniker *mon;
2327     MSG msg;
2328     HRESULT hres;
2329
2330     static const char res[] = "res://";
2331
2332     trace("running %s...\n", test_name);
2333
2334     doc = create_document();
2335     if(!doc)
2336         return;
2337
2338     set_client_site(doc, TRUE);
2339     do_advise(doc, &IID_IPropertyNotifySink, (IUnknown*)&PropertyNotifySink);
2340
2341     lstrcpyA(urlA, res);
2342     GetModuleFileNameA(NULL, urlA + lstrlenA(res), sizeof(urlA) - lstrlenA(res));
2343     lstrcatA(urlA, "/");
2344     lstrcatA(urlA, test_name);
2345     MultiByteToWideChar(CP_ACP, 0, urlA, -1, url, sizeof(url)/sizeof(WCHAR));
2346
2347     hres = CreateURLMoniker(NULL, url, &mon);
2348     ok(hres == S_OK, "CreateURLMoniker failed: %08x\n", hres);
2349
2350     hres = IHTMLDocument2_QueryInterface(doc, &IID_IPersistMoniker, (void**)&persist);
2351     ok(hres == S_OK, "Could not get IPersistMoniker iface: %08x\n", hres);
2352
2353     hres = IPersistMoniker_Load(persist, FALSE, mon, NULL, 0);
2354     ok(hres == S_OK, "Load failed: %08x\n", hres);
2355
2356     IMoniker_Release(mon);
2357     IPersistMoniker_Release(persist);
2358
2359     SET_EXPECT(external_success);
2360
2361     while(!called_external_success && GetMessage(&msg, NULL, 0, 0)) {
2362         TranslateMessage(&msg);
2363         DispatchMessage(&msg);
2364     }
2365
2366     CHECK_CALLED(external_success);
2367
2368     set_client_site(doc, FALSE);
2369     IHTMLDocument2_Release(doc);
2370 }
2371
2372 static void run_js_tests(void)
2373 {
2374     run_js_script("jstest.html");
2375 }
2376
2377 static BOOL init_registry(BOOL init)
2378 {
2379     return init_key("TestScript\\CLSID", TESTSCRIPT_CLSID, init)
2380         && init_key("CLSID\\"TESTSCRIPT_CLSID"\\Implemented Categories\\{F0B7A1A1-9847-11CF-8F20-00805F2CD064}",
2381                     NULL, init)
2382         && init_key("CLSID\\"TESTSCRIPT_CLSID"\\Implemented Categories\\{F0B7A1A2-9847-11CF-8F20-00805F2CD064}",
2383                     NULL, init);
2384 }
2385
2386 static BOOL register_script_engine(void)
2387 {
2388     DWORD regid;
2389     HRESULT hres;
2390
2391     if(!init_registry(TRUE)) {
2392         init_registry(FALSE);
2393         return FALSE;
2394     }
2395
2396     hres = CoRegisterClassObject(&CLSID_TestScript, (IUnknown *)&script_cf,
2397                                  CLSCTX_INPROC_SERVER, REGCLS_MULTIPLEUSE, &regid);
2398     ok(hres == S_OK, "Could not register screipt engine: %08x\n", hres);
2399
2400     return TRUE;
2401 }
2402
2403 static LRESULT WINAPI wnd_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
2404 {
2405     return DefWindowProc(hwnd, msg, wParam, lParam);
2406 }
2407
2408 static HWND create_container_window(void)
2409 {
2410     static const CHAR szHTMLDocumentTest[] = "HTMLDocumentTest";
2411     static WNDCLASSEXA wndclass = {
2412         sizeof(WNDCLASSEXA),
2413         0,
2414         wnd_proc,
2415         0, 0, NULL, NULL, NULL, NULL, NULL,
2416         szHTMLDocumentTest,
2417         NULL
2418     };
2419
2420     RegisterClassExA(&wndclass);
2421     return CreateWindowA(szHTMLDocumentTest, szHTMLDocumentTest,
2422             WS_OVERLAPPEDWINDOW, CW_USEDEFAULT, CW_USEDEFAULT,
2423             300, 300, NULL, NULL, NULL, NULL);
2424 }
2425
2426 START_TEST(script)
2427 {
2428     CoInitialize(NULL);
2429     container_hwnd = create_container_window();
2430
2431     if(winetest_interactive || ! is_ie_hardened()) {
2432         if(register_script_engine()) {
2433             test_simple_script();
2434             run_js_tests();
2435             init_registry(FALSE);
2436         }else {
2437             skip("Could not register TestScript engine\n");
2438         }
2439     }else {
2440         skip("IE running in Enhanced Security Configuration\n");
2441     }
2442
2443     DestroyWindow(container_hwnd);
2444     CoUninitialize();
2445 }