git.oblomov.eu Git - wine/blob - dlls/jscript/tests/run.c

   1 /*
   2  * Copyright 2008 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
  28 #include "wine/test.h"
  29
  30 static const CLSID CLSID_JScript =
  31     {0xf414c260,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};
  32
  33 #define DEFINE_EXPECT(func) \
  34     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
  35
  36 #define SET_EXPECT(func) \
  37     expect_ ## func = TRUE
  38
  39 #define SET_CALLED(func) \
  40     called_ ## func = TRUE
  41
  42 #define CHECK_EXPECT2(func) \
  43     do { \
  44         ok(expect_ ##func, "unexpected call " #func "\n"); \
  45         called_ ## func = TRUE; \
  46     }while(0)
  47
  48 #define CHECK_EXPECT(func) \
  49     do { \
  50         CHECK_EXPECT2(func); \
  51         expect_ ## func = FALSE; \
  52     }while(0)
  53
  54 #define CHECK_CALLED(func) \
  55     do { \
  56         ok(called_ ## func, "expected " #func "\n"); \
  57         expect_ ## func = called_ ## func = FALSE; \
  58     }while(0)
  59
  60 DEFINE_EXPECT(global_propget_d);
  61 DEFINE_EXPECT(global_propget_i);
  62 DEFINE_EXPECT(global_propput_d);
  63 DEFINE_EXPECT(global_propput_i);
  64 DEFINE_EXPECT(global_success_d);
  65 DEFINE_EXPECT(global_success_i);
  66 DEFINE_EXPECT(testobj_delete);
  67 DEFINE_EXPECT(GetItemInfo_testVal);
  68
  69 #define DISPID_GLOBAL_TESTPROPGET   0x1000
  70 #define DISPID_GLOBAL_TESTPROPPUT   0x1001
  71 #define DISPID_GLOBAL_REPORTSUCCESS 0x1002
  72 #define DISPID_GLOBAL_TRACE         0x1003
  73 #define DISPID_GLOBAL_OK            0x1004
  74 #define DISPID_GLOBAL_GETVT         0x1005
  75 #define DISPID_GLOBAL_TESTOBJ       0x1006
  76
  77 static const WCHAR testW[] = {'t','e','s','t',0};
  78 static const CHAR testA[] = "test";
  79 static const WCHAR test_valW[] = {'t','e','s','t','V','a','l',0};
  80 static const CHAR test_valA[] = "testVal";
  81
  82 static BOOL strict_dispid_check;
  83 static const char *test_name = "(null)";
  84
  85 static const char *debugstr_w(LPCWSTR str)
  86 {
  87     static char buf[1024];
  88
  89     if(!str)
  90         return "(null)";
  91
  92     WideCharToMultiByte(CP_ACP, 0, str, -1, buf, sizeof(buf), NULL, NULL);
  93     return buf;
  94 }
  95
  96 static BSTR a2bstr(const char *str)
  97 {
  98     BSTR ret;
  99     int len;
 100
 101     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
 102     ret = SysAllocStringLen(NULL, len-1);
 103     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
 104
 105     return ret;
 106 }
 107
 108 static int strcmp_wa(LPCWSTR strw, const char *stra)
 109 {
 110     CHAR buf[512];
 111     WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), 0, 0);
 112     return lstrcmpA(buf, stra);
 113 }
 114
 115 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
 116 {
 117     *ppv = NULL;
 118
 119     if(IsEqualGUID(riid, &IID_IUnknown)
 120        || IsEqualGUID(riid, &IID_IDispatch)
 121        || IsEqualGUID(riid, &IID_IDispatchEx))
 122         *ppv = iface;
 123     else
 124         return E_NOINTERFACE;
 125
 126     return S_OK;
 127 }
 128
 129 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
 130 {
 131     return 2;
 132 }
 133
 134 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
 135 {
 136     return 1;
 137 }
 138
 139 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
 140 {
 141     ok(0, "unexpected call\n");
 142     return E_NOTIMPL;
 143 }
 144
 145 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
 146                                               LCID lcid, ITypeInfo **ppTInfo)
 147 {
 148     ok(0, "unexpected call\n");
 149     return E_NOTIMPL;
 150 }
 151
 152 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
 153                                                 LPOLESTR *rgszNames, UINT cNames,
 154                                                 LCID lcid, DISPID *rgDispId)
 155 {
 156     ok(0, "unexpected call\n");
 157     return E_NOTIMPL;
 158 }
 159
 160 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
 161                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
 162                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
 163 {
 164     ok(0, "unexpected call\n");
 165     return E_NOTIMPL;
 166 }
 167
 168 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
 169 {
 170     ok(0, "unexpected call %s %x\n", debugstr_w(bstrName), grfdex);
 171     return E_NOTIMPL;
 172 }
 173
 174 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
 175 {
 176     ok(0, "unexpected call\n");
 177     return E_NOTIMPL;
 178 }
 179
 180 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
 181 {
 182     ok(0, "unexpected call\n");
 183     return E_NOTIMPL;
 184 }
 185
 186 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
 187 {
 188     ok(0, "unexpected call\n");
 189     return E_NOTIMPL;
 190 }
 191
 192 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
 193 {
 194     ok(0, "unexpected call\n");
 195     return E_NOTIMPL;
 196 }
 197
 198 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
 199 {
 200     ok(0, "unexpected call\n");
 201     return E_NOTIMPL;
 202 }
 203
 204 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 205 {
 206     ok(0, "unexpected call\n");
 207     return E_NOTIMPL;
 208 }
 209
 210 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 211         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 212 {
 213     ok(0, "unexpected call\n");
 214     return E_NOTIMPL;
 215 }
 216
 217 static HRESULT WINAPI testObj_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
 218 {
 219     CHECK_EXPECT(testobj_delete);
 220
 221     ok(!strcmp_wa(bstrName, "deleteTest"), "unexpected name %s\n", debugstr_w(bstrName));
 222     ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 223     return S_OK;
 224 }
 225
 226 static IDispatchExVtbl testObjVtbl = {
 227     DispatchEx_QueryInterface,
 228     DispatchEx_AddRef,
 229     DispatchEx_Release,
 230     DispatchEx_GetTypeInfoCount,
 231     DispatchEx_GetTypeInfo,
 232     DispatchEx_GetIDsOfNames,
 233     DispatchEx_Invoke,
 234     DispatchEx_GetDispID,
 235     DispatchEx_InvokeEx,
 236     testObj_DeleteMemberByName,
 237     DispatchEx_DeleteMemberByDispID,
 238     DispatchEx_GetMemberProperties,
 239     DispatchEx_GetMemberName,
 240     DispatchEx_GetNextDispID,
 241     DispatchEx_GetNameSpaceParent
 242 };
 243
 244 static IDispatchEx testObj = { &testObjVtbl };
 245
 246 static HRESULT WINAPI Global_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 247 {
 248     if(!strcmp_wa(bstrName, "ok")) {
 249         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 250         *pid = DISPID_GLOBAL_OK;
 251         return S_OK;
 252     }
 253     if(!strcmp_wa(bstrName, "trace")) {
 254         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 255         *pid = DISPID_GLOBAL_TRACE;
 256         return S_OK;
 257     }
 258     if(!strcmp_wa(bstrName, "reportSuccess")) {
 259         CHECK_EXPECT(global_success_d);
 260         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 261         *pid = DISPID_GLOBAL_REPORTSUCCESS;
 262         return S_OK;
 263     }
 264     if(!strcmp_wa(bstrName, "testPropGet")) {
 265         CHECK_EXPECT(global_propget_d);
 266         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 267         *pid = DISPID_GLOBAL_TESTPROPGET;
 268         return S_OK;
 269     }
 270     if(!strcmp_wa(bstrName, "testPropPut")) {
 271         CHECK_EXPECT(global_propput_d);
 272         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 273         *pid = DISPID_GLOBAL_TESTPROPPUT;
 274         return S_OK;
 275     }
 276     if(!strcmp_wa(bstrName, "getVT")) {
 277         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 278         *pid = DISPID_GLOBAL_GETVT;
 279         return S_OK;
 280     }
 281     if(!strcmp_wa(bstrName, "testObj")) {
 282         ok(grfdex == fdexNameCaseSensitive, "grfdex = %x\n", grfdex);
 283         *pid = DISPID_GLOBAL_TESTOBJ;
 284         return S_OK;
 285     }
 286
 287     if(strict_dispid_check)
 288         ok(0, "unexpected call %s\n", debugstr_w(bstrName));
 289     return DISP_E_UNKNOWNNAME;
 290 }
 291
 292 static HRESULT WINAPI Global_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 293         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 294 {
 295     switch(id) {
 296     case DISPID_GLOBAL_OK:
 297         ok(wFlags == INVOKE_FUNC || wFlags == (INVOKE_FUNC|INVOKE_PROPERTYGET), "wFlags = %x\n", wFlags);
 298         ok(pdp != NULL, "pdp == NULL\n");
 299         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
 300         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
 301         ok(pdp->cArgs == 2, "cArgs = %d\n", pdp->cArgs);
 302         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
 303         if(wFlags & INVOKE_PROPERTYGET)
 304             ok(pvarRes != NULL, "pvarRes == NULL\n");
 305         else
 306             ok(!pvarRes, "pvarRes != NULL\n");
 307         ok(pei != NULL, "pei == NULL\n");
 308
 309         ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
 310         ok(V_VT(pdp->rgvarg+1) == VT_BOOL, "V_VT(psp->rgvargs+1) = %d\n", V_VT(pdp->rgvarg));
 311         ok(V_BOOL(pdp->rgvarg+1), "%s: %s\n", test_name, debugstr_w(V_BSTR(pdp->rgvarg)));
 312
 313         return S_OK;
 314
 315      case DISPID_GLOBAL_TRACE:
 316         ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
 317         ok(pdp != NULL, "pdp == NULL\n");
 318         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
 319         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
 320         ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
 321         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
 322         ok(!pvarRes, "pvarRes != NULL\n");
 323         ok(pei != NULL, "pei == NULL\n");
 324
 325         ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
 326         if(V_VT(pdp->rgvarg) == VT_BSTR)
 327             trace("%s: %s\n", test_name, debugstr_w(V_BSTR(pdp->rgvarg)));
 328
 329         return S_OK;
 330
 331     case DISPID_GLOBAL_REPORTSUCCESS:
 332         CHECK_EXPECT(global_success_i);
 333
 334         ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
 335         ok(pdp != NULL, "pdp == NULL\n");
 336         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
 337         ok(pdp->cArgs == 0, "cArgs = %d\n", pdp->cArgs);
 338         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
 339         ok(!pvarRes, "pvarRes != NULL\n");
 340         ok(pei != NULL, "pei == NULL\n");
 341
 342         return S_OK;
 343
 344      case DISPID_GLOBAL_TESTPROPGET:
 345         CHECK_EXPECT(global_propget_i);
 346
 347         ok(wFlags == INVOKE_PROPERTYGET, "wFlags = %x\n", wFlags);
 348         ok(pdp != NULL, "pdp == NULL\n");
 349         ok(!pdp->rgvarg, "rgvarg != NULL\n");
 350         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
 351         ok(!pdp->cArgs, "cArgs = %d\n", pdp->cArgs);
 352         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
 353         ok(pvarRes != NULL, "pvarRes == NULL\n");
 354         ok(V_VT(pvarRes) ==  VT_EMPTY, "V_VT(pvarRes) = %d\n", V_VT(pvarRes));
 355         ok(pei != NULL, "pei == NULL\n");
 356
 357         V_VT(pvarRes) = VT_I4;
 358         V_I4(pvarRes) = 1;
 359
 360         return S_OK;
 361
 362     case DISPID_GLOBAL_TESTPROPPUT:
 363         CHECK_EXPECT(global_propput_i);
 364
 365         ok(wFlags == INVOKE_PROPERTYPUT, "wFlags = %x\n", wFlags);
 366         ok(pdp != NULL, "pdp == NULL\n");
 367         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
 368         ok(pdp->rgdispidNamedArgs != NULL, "rgdispidNamedArgs == NULL\n");
 369         ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
 370         ok(pdp->cNamedArgs == 1, "cNamedArgs = %d\n", pdp->cNamedArgs);
 371         ok(pdp->rgdispidNamedArgs[0] == DISPID_PROPERTYPUT, "pdp->rgdispidNamedArgs[0] = %d\n", pdp->rgdispidNamedArgs[0]);
 372         ok(!pvarRes, "pvarRes != NULL\n");
 373
 374         ok(V_VT(pdp->rgvarg) == VT_I4, "V_VT(pdp->rgvarg)=%d\n", V_VT(pdp->rgvarg));
 375         ok(V_I4(pdp->rgvarg) == 1, "V_I4(pdp->rgvarg)=%d\n", V_I4(pdp->rgvarg));
 376         return S_OK;
 377
 378      case DISPID_GLOBAL_GETVT:
 379         ok(pdp != NULL, "pdp == NULL\n");
 380         ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
 381         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
 382         ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
 383         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
 384         ok(pvarRes != NULL, "pvarRes == NULL\n");
 385         ok(V_VT(pvarRes) ==  VT_EMPTY, "V_VT(pvarRes) = %d\n", V_VT(pvarRes));
 386         ok(pei != NULL, "pei == NULL\n");
 387
 388         V_VT(pvarRes) = VT_BSTR;
 389         switch(V_VT(pdp->rgvarg)) {
 390         case VT_EMPTY:
 391             V_BSTR(pvarRes) = a2bstr("VT_EMPTY");
 392             break;
 393         case VT_NULL:
 394             V_BSTR(pvarRes) = a2bstr("VT_NULL");
 395             break;
 396         case VT_I4:
 397             V_BSTR(pvarRes) = a2bstr("VT_I4");
 398             break;
 399         case VT_R8:
 400             V_BSTR(pvarRes) = a2bstr("VT_R8");
 401             break;
 402         case VT_BSTR:
 403             V_BSTR(pvarRes) = a2bstr("VT_BSTR");
 404             break;
 405         case VT_DISPATCH:
 406             V_BSTR(pvarRes) = a2bstr("VT_DISPATCH");
 407             break;
 408         case VT_BOOL:
 409             V_BSTR(pvarRes) = a2bstr("VT_BOOL");
 410             break;
 411         default:
 412             ok(0, "unknown vt %d\n", V_VT(pdp->rgvarg));
 413             return E_FAIL;
 414         }
 415
 416         return S_OK;
 417
 418     case DISPID_GLOBAL_TESTOBJ:
 419         ok(wFlags == INVOKE_PROPERTYGET, "wFlags = %x\n", wFlags);
 420         ok(pdp != NULL, "pdp == NULL\n");
 421         ok(!pdp->rgvarg, "rgvarg != NULL\n");
 422         ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
 423         ok(!pdp->cArgs, "cArgs = %d\n", pdp->cArgs);
 424         ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
 425         ok(pvarRes != NULL, "pvarRes == NULL\n");
 426         ok(V_VT(pvarRes) ==  VT_EMPTY, "V_VT(pvarRes) = %d\n", V_VT(pvarRes));
 427         ok(pei != NULL, "pei == NULL\n");
 428
 429         V_VT(pvarRes) = VT_DISPATCH;
 430         V_DISPATCH(pvarRes) = (IDispatch*)&testObj;
 431         return S_OK;
 432     }
 433
 434     ok(0, "unexpected call %x\n", id);
 435     return DISP_E_MEMBERNOTFOUND;
 436 }
 437
 438 static IDispatchExVtbl GlobalVtbl = {
 439     DispatchEx_QueryInterface,
 440     DispatchEx_AddRef,
 441     DispatchEx_Release,
 442     DispatchEx_GetTypeInfoCount,
 443     DispatchEx_GetTypeInfo,
 444     DispatchEx_GetIDsOfNames,
 445     DispatchEx_Invoke,
 446     Global_GetDispID,
 447     Global_InvokeEx,
 448     DispatchEx_DeleteMemberByName,
 449     DispatchEx_DeleteMemberByDispID,
 450     DispatchEx_GetMemberProperties,
 451     DispatchEx_GetMemberName,
 452     DispatchEx_GetNextDispID,
 453     DispatchEx_GetNameSpaceParent
 454 };
 455
 456 static IDispatchEx Global = { &GlobalVtbl };
 457
 458 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
 459 {
 460     *ppv = NULL;
 461
 462     if(IsEqualGUID(&IID_IUnknown, riid))
 463         *ppv = iface;
 464     else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
 465         *ppv = iface;
 466     else
 467         return E_NOINTERFACE;
 468
 469     IUnknown_AddRef((IUnknown*)*ppv);
 470     return S_OK;
 471 }
 472
 473 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
 474 {
 475     return 2;
 476 }
 477
 478 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
 479 {
 480     return 1;
 481 }
 482
 483 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
 484 {
 485     *plcid = GetUserDefaultLCID();
 486     return S_OK;
 487 }
 488
 489 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
 490         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
 491 {
 492     ok(dwReturnMask == SCRIPTINFO_IUNKNOWN, "unexpected dwReturnMask %x\n", dwReturnMask);
 493     ok(!ppti, "ppti != NULL\n");
 494
 495     if(!strcmp_wa(pstrName, test_valA))
 496         CHECK_EXPECT(GetItemInfo_testVal);
 497     else if(strcmp_wa(pstrName, testA))
 498         ok(0, "unexpected pstrName %s\n", debugstr_w(pstrName));
 499
 500     *ppiunkItem = (IUnknown*)&Global;
 501     return S_OK;
 502 }
 503
 504 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
 505 {
 506     return E_NOTIMPL;
 507 }
 508
 509 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
 510         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
 511 {
 512     return E_NOTIMPL;
 513 }
 514
 515 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
 516 {
 517     return E_NOTIMPL;
 518 }
 519
 520 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
 521 {
 522     return E_NOTIMPL;
 523 }
 524
 525 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 526 {
 527     return E_NOTIMPL;
 528 }
 529
 530 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 531 {
 532     return E_NOTIMPL;
 533 }
 534
 535 #undef ACTSCPSITE_THIS
 536
 537 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
 538     ActiveScriptSite_QueryInterface,
 539     ActiveScriptSite_AddRef,
 540     ActiveScriptSite_Release,
 541     ActiveScriptSite_GetLCID,
 542     ActiveScriptSite_GetItemInfo,
 543     ActiveScriptSite_GetDocVersionString,
 544     ActiveScriptSite_OnScriptTerminate,
 545     ActiveScriptSite_OnStateChange,
 546     ActiveScriptSite_OnScriptError,
 547     ActiveScriptSite_OnEnterScript,
 548     ActiveScriptSite_OnLeaveScript
 549 };
 550
 551 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
 552
 553 static IActiveScript *create_script(void)
 554 {
 555     IActiveScript *script;
 556     HRESULT hres;
 557
 558     hres = CoCreateInstance(&CLSID_JScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 559             &IID_IActiveScript, (void**)&script);
 560     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
 561
 562     return script;
 563 }
 564
 565 static void parse_script(BSTR script_str)
 566 {
 567     IActiveScriptParse *parser;
 568     IActiveScript *engine;
 569     HRESULT hres;
 570
 571     engine = create_script();
 572     if(!engine)
 573         return;
 574
 575     hres = IActiveScript_QueryInterface(engine, &IID_IActiveScriptParse, (void**)&parser);
 576     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
 577     if (FAILED(hres))
 578     {
 579         IActiveScript_Release(engine);
 580         return;
 581     }
 582
 583     hres = IActiveScriptParse64_InitNew(parser);
 584     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 585
 586     hres = IActiveScript_SetScriptSite(engine, &ActiveScriptSite);
 587     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 588
 589     hres = IActiveScript_AddNamedItem(engine, testW,
 590             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
 591     ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
 592
 593     hres = IActiveScript_SetScriptState(engine, SCRIPTSTATE_STARTED);
 594     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
 595
 596     hres = IActiveScriptParse64_ParseScriptText(parser, script_str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
 597     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 598
 599     IActiveScript_Release(engine);
 600     IUnknown_Release(parser);
 601 }
 602
 603 static HRESULT parse_htmlscript(BSTR script_str)
 604 {
 605     IActiveScriptParse *parser;
 606     IActiveScript *engine;
 607     HRESULT hres;
 608     BSTR tmp = a2bstr("</SCRIPT>");
 609
 610     engine = create_script();
 611     if(!engine)
 612         return E_FAIL;
 613
 614     hres = IActiveScript_QueryInterface(engine, &IID_IActiveScriptParse, (void**)&parser);
 615     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
 616     if (FAILED(hres))
 617     {
 618         IActiveScript_Release(engine);
 619         return E_FAIL;
 620     }
 621
 622     hres = IActiveScriptParse64_InitNew(parser);
 623     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 624
 625     hres = IActiveScript_SetScriptSite(engine, &ActiveScriptSite);
 626     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 627
 628     hres = IActiveScript_AddNamedItem(engine, testW,
 629             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
 630     ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
 631
 632     hres = IActiveScript_SetScriptState(engine, SCRIPTSTATE_STARTED);
 633     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
 634
 635     hres = IActiveScriptParse64_ParseScriptText(parser, script_str, NULL, NULL, tmp, 0, 0, 0, NULL, NULL);
 636
 637     IActiveScript_Release(engine);
 638     IUnknown_Release(parser);
 639     SysFreeString(tmp);
 640
 641     return hres;
 642 }
 643
 644 static void parse_script_a(const char *src)
 645 {
 646     BSTR tmp = a2bstr(src);
 647     parse_script(tmp);
 648     SysFreeString(tmp);
 649 }
 650
 651 static HRESULT parse_htmlscript_a(const char *src)
 652 {
 653     HRESULT hres;
 654     BSTR tmp = a2bstr(src);
 655     hres = parse_htmlscript(tmp);
 656     SysFreeString(tmp);
 657
 658     return hres;
 659 }
 660
 661 static BSTR get_script_from_file(const char *filename)
 662 {
 663     DWORD size, len;
 664     HANDLE file, map;
 665     const char *file_map;
 666     BSTR ret;
 667
 668     file = CreateFileA(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
 669     if(file == INVALID_HANDLE_VALUE) {
 670         trace("Could not open file: %u\n", GetLastError());
 671         return NULL;
 672     }
 673
 674     size = GetFileSize(file, NULL);
 675
 676     map = CreateFileMapping(file, NULL, PAGE_READONLY, 0, 0, NULL);
 677     CloseHandle(file);
 678     if(map == INVALID_HANDLE_VALUE) {
 679         trace("Could not create file mapping: %u\n", GetLastError());
 680         return NULL;
 681     }
 682
 683     file_map = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0);
 684     CloseHandle(map);
 685     if(!file_map) {
 686         trace("MapViewOfFile failed: %u\n", GetLastError());
 687         return NULL;
 688     }
 689
 690     len = MultiByteToWideChar(CP_ACP, 0, file_map, size, NULL, 0);
 691     ret = SysAllocStringLen(NULL, len);
 692     MultiByteToWideChar(CP_ACP, 0, file_map, size, ret, len);
 693
 694     UnmapViewOfFile(file_map);
 695
 696     return ret;
 697 }
 698
 699 static void run_from_file(const char *filename)
 700 {
 701     BSTR script_str = get_script_from_file(filename);
 702
 703     strict_dispid_check = FALSE;
 704
 705     if(script_str)
 706         parse_script(script_str);
 707
 708     SysFreeString(script_str);
 709 }
 710
 711 static void run_from_res(const char *name)
 712 {
 713     const char *data;
 714     DWORD size, len;
 715     BSTR str;
 716     HRSRC src;
 717
 718     strict_dispid_check = FALSE;
 719     test_name = name;
 720
 721     src = FindResourceA(NULL, name, (LPCSTR)40);
 722     ok(src != NULL, "Could not find resource %s\n", name);
 723
 724     size = SizeofResource(NULL, src);
 725     data = LoadResource(NULL, src);
 726
 727     len = MultiByteToWideChar(CP_ACP, 0, data, size, NULL, 0);
 728     str = SysAllocStringLen(NULL, len);
 729     len = MultiByteToWideChar(CP_ACP, 0, data, size, str, len);
 730
 731     SET_EXPECT(global_success_d);
 732     SET_EXPECT(global_success_i);
 733     parse_script(str);
 734     CHECK_CALLED(global_success_d);
 735     CHECK_CALLED(global_success_i);
 736
 737     SysFreeString(str);
 738 }
 739
 740 static void test_isvisible(BOOL global_members)
 741 {
 742     IActiveScriptParse *parser;
 743     IActiveScript *engine;
 744     HRESULT hres;
 745
 746     static const WCHAR script_textW[] =
 747         {'v','a','r',' ','v',' ','=',' ','t','e','s','t','V','a','l',';',0};
 748
 749     engine = create_script();
 750     if(!engine)
 751         return;
 752
 753     hres = IActiveScript_QueryInterface(engine, &IID_IActiveScriptParse, (void**)&parser);
 754     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
 755     if (FAILED(hres))
 756     {
 757         IActiveScript_Release(engine);
 758         return;
 759     }
 760
 761     hres = IActiveScriptParse64_InitNew(parser);
 762     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 763
 764     hres = IActiveScript_SetScriptSite(engine, &ActiveScriptSite);
 765     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 766
 767     if(global_members)
 768         SET_EXPECT(GetItemInfo_testVal);
 769     hres = IActiveScript_AddNamedItem(engine, test_valW,
 770             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|
 771             (global_members ? SCRIPTITEM_GLOBALMEMBERS : 0));
 772     ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
 773     if(global_members)
 774         CHECK_CALLED(GetItemInfo_testVal);
 775
 776     hres = IActiveScript_SetScriptState(engine, SCRIPTSTATE_STARTED);
 777     ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
 778
 779     if(!global_members)
 780         SET_EXPECT(GetItemInfo_testVal);
 781     hres = IActiveScriptParse64_ParseScriptText(parser, script_textW, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
 782     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 783     if(!global_members)
 784         CHECK_CALLED(GetItemInfo_testVal);
 785
 786     hres = IActiveScriptParse64_ParseScriptText(parser, script_textW, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
 787     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 788
 789     IActiveScript_Release(engine);
 790     IUnknown_Release(parser);
 791 }
 792
 793 static void run_tests(void)
 794 {
 795     HRESULT hres;
 796
 797     strict_dispid_check = TRUE;
 798
 799     parse_script_a("");
 800     parse_script_a("/* empty */ ;");
 801
 802     SET_EXPECT(global_propget_d);
 803     SET_EXPECT(global_propget_i);
 804     parse_script_a("testPropGet;");
 805     CHECK_CALLED(global_propget_d);
 806     CHECK_CALLED(global_propget_i);
 807
 808     SET_EXPECT(global_propput_d);
 809     SET_EXPECT(global_propput_i);
 810     parse_script_a("testPropPut = 1;");
 811     CHECK_CALLED(global_propput_d);
 812     CHECK_CALLED(global_propput_i);
 813
 814     SET_EXPECT(global_success_d);
 815     SET_EXPECT(global_success_i);
 816     parse_script_a("reportSuccess();");
 817     CHECK_CALLED(global_success_d);
 818     CHECK_CALLED(global_success_i);
 819
 820     SET_EXPECT(testobj_delete);
 821     parse_script_a("delete testObj.deleteTest;");
 822     CHECK_CALLED(testobj_delete);
 823
 824     parse_script_a("ok(typeof(test) === 'object', \"typeof(test) != 'object'\");");
 825
 826     run_from_res("lang.js");
 827     run_from_res("api.js");
 828     run_from_res("regexp.js");
 829
 830     test_isvisible(FALSE);
 831     test_isvisible(TRUE);
 832
 833     hres = parse_htmlscript_a("<!--");
 834     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 835     hres = parse_htmlscript_a("-->");
 836     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 837     hres = parse_htmlscript_a("<!--\nvar a=1;\n-->\n");
 838     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 839     hres = parse_htmlscript_a("<!--\n<!-- ignore this\n-->\n");
 840     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 841     hres = parse_htmlscript_a("var a=1;\nif(a-->0) a=5;\n");
 842     ok(hres == S_OK, "ParseScriptText failed: %08x\n", hres);
 843     hres = parse_htmlscript_a("var a=1;\nif(a\n-->0) a=5;\n");
 844     ok(hres != S_OK, "ParseScriptText have not failed\n");
 845 }
 846
 847 START_TEST(run)
 848 {
 849     int argc;
 850     char **argv;
 851
 852     argc = winetest_get_mainargs(&argv);
 853
 854     CoInitialize(NULL);
 855
 856     if(argc > 2)
 857         run_from_file(argv[2]);
 858     else
 859         run_tests();
 860
 861     CoUninitialize();
 862 }