2 * Copyright 2011 Jacek Caban for CodeWeavers
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.
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.
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
28 #include "wine/test.h"
30 extern const CLSID CLSID_VBScript;
32 #define DEFINE_EXPECT(func) \
33 static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
35 #define SET_EXPECT(func) \
36 expect_ ## func = TRUE
38 #define SET_CALLED(func) \
39 called_ ## func = TRUE
41 #define CHECK_EXPECT2(func) \
43 ok(expect_ ##func, "unexpected call " #func "\n"); \
44 called_ ## func = TRUE; \
47 #define CHECK_EXPECT(func) \
49 CHECK_EXPECT2(func); \
50 expect_ ## func = FALSE; \
53 #define CHECK_CALLED(func) \
55 ok(called_ ## func, "expected " #func "\n"); \
56 expect_ ## func = called_ ## func = FALSE; \
59 DEFINE_EXPECT(global_success_d);
60 DEFINE_EXPECT(global_success_i);
62 #define DISPID_GLOBAL_REPORTSUCCESS 1000
63 #define DISPID_GLOBAL_TRACE 1001
64 #define DISPID_GLOBAL_OK 1002
65 #define DISPID_GLOBAL_GETVT 1003
67 static const WCHAR testW[] = {'t','e','s','t',0};
69 static BOOL strict_dispid_check;
70 static const char *test_name = "(null)";
72 static BSTR a2bstr(const char *str)
77 len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
78 ret = SysAllocStringLen(NULL, len-1);
79 MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
84 static int strcmp_wa(LPCWSTR strw, const char *stra)
87 WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), 0, 0);
88 return lstrcmpA(buf, stra);
91 static const char *vt2a(VARIANT *v)
93 if(V_VT(v) == (VT_BYREF|VT_VARIANT)) {
95 sprintf(buf, "%s*", vt2a(V_BYREF(v)));
113 return "VT_DISPATCH";
116 case VT_ARRAY|VT_VARIANT:
117 return "VT_ARRAY|VT_VARIANT";
119 ok(0, "unknown vt %d\n", V_VT(v));
124 #define test_grfdex(a,b) _test_grfdex(__LINE__,a,b)
125 static void _test_grfdex(unsigned line, DWORD grfdex, DWORD expect)
127 ok_(__FILE__,line)(grfdex == expect, "grfdex = %x, expected %x\n", grfdex, expect);
130 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
134 if(IsEqualGUID(riid, &IID_IUnknown)
135 || IsEqualGUID(riid, &IID_IDispatch)
136 || IsEqualGUID(riid, &IID_IDispatchEx))
139 return E_NOINTERFACE;
144 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
149 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
154 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
156 ok(0, "unexpected call\n");
160 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
161 LCID lcid, ITypeInfo **ppTInfo)
163 ok(0, "unexpected call\n");
167 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
168 LPOLESTR *rgszNames, UINT cNames,
169 LCID lcid, DISPID *rgDispId)
171 ok(0, "unexpected call\n");
175 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
176 REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
177 VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
179 ok(0, "unexpected call\n");
183 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
185 ok(0, "unexpected call %s %x\n", wine_dbgstr_w(bstrName), grfdex);
189 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
191 ok(0, "unexpected call\n");
195 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
197 ok(0, "unexpected call\n");
201 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
203 ok(0, "unexpected call\n");
207 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
209 ok(0, "unexpected call\n");
213 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
215 ok(0, "unexpected call\n");
220 static HRESULT WINAPI Global_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
222 if(!strcmp_wa(bstrName, "ok")) {
223 test_grfdex(grfdex, fdexNameCaseInsensitive);
224 *pid = DISPID_GLOBAL_OK;
227 if(!strcmp_wa(bstrName, "trace")) {
228 test_grfdex(grfdex, fdexNameCaseInsensitive);
229 *pid = DISPID_GLOBAL_TRACE;
232 if(!strcmp_wa(bstrName, "reportSuccess")) {
233 CHECK_EXPECT(global_success_d);
234 test_grfdex(grfdex, fdexNameCaseInsensitive);
235 *pid = DISPID_GLOBAL_REPORTSUCCESS;
238 if(!strcmp_wa(bstrName, "getVT")) {
239 test_grfdex(grfdex, fdexNameCaseInsensitive);
240 *pid = DISPID_GLOBAL_GETVT;
244 if(strict_dispid_check && strcmp_wa(bstrName, "x"))
245 ok(0, "unexpected call %s %x\n", wine_dbgstr_w(bstrName), grfdex);
246 return DISP_E_UNKNOWNNAME;
249 static HRESULT WINAPI Global_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
250 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
253 case DISPID_GLOBAL_OK: {
256 ok(wFlags == INVOKE_FUNC || wFlags == (INVOKE_FUNC|INVOKE_PROPERTYGET), "wFlags = %x\n", wFlags);
257 ok(pdp != NULL, "pdp == NULL\n");
258 ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
259 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
260 ok(pdp->cArgs == 2, "cArgs = %d\n", pdp->cArgs);
261 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
262 if(wFlags & INVOKE_PROPERTYGET)
263 ok(pvarRes != NULL, "pvarRes == NULL\n");
265 ok(!pvarRes, "pvarRes != NULL\n");
266 ok(pei != NULL, "pei == NULL\n");
268 ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
271 if(V_VT(b) == (VT_BYREF|VT_VARIANT))
274 ok(V_VT(b) == VT_BOOL, "V_VT(b) = %d\n", V_VT(b));
276 ok(V_BOOL(b), "%s: %s\n", test_name, wine_dbgstr_w(V_BSTR(pdp->rgvarg)));
280 case DISPID_GLOBAL_TRACE:
281 ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
282 ok(pdp != NULL, "pdp == NULL\n");
283 ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
284 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
285 ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
286 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
287 ok(!pvarRes, "pvarRes != NULL\n");
288 ok(pei != NULL, "pei == NULL\n");
290 ok(V_VT(pdp->rgvarg) == VT_BSTR, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp->rgvarg));
291 if(V_VT(pdp->rgvarg) == VT_BSTR)
292 trace("%s: %s\n", test_name, wine_dbgstr_w(V_BSTR(pdp->rgvarg)));
296 case DISPID_GLOBAL_REPORTSUCCESS:
297 CHECK_EXPECT(global_success_i);
299 ok(wFlags == INVOKE_FUNC, "wFlags = %x\n", wFlags);
300 ok(pdp != NULL, "pdp == NULL\n");
301 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
302 ok(pdp->cArgs == 0, "cArgs = %d\n", pdp->cArgs);
303 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
304 ok(!pvarRes, "pvarRes != NULL\n");
305 ok(pei != NULL, "pei == NULL\n");
309 case DISPID_GLOBAL_GETVT:
310 ok(pdp != NULL, "pdp == NULL\n");
311 ok(pdp->rgvarg != NULL, "rgvarg == NULL\n");
312 ok(!pdp->rgdispidNamedArgs, "rgdispidNamedArgs != NULL\n");
313 ok(pdp->cArgs == 1, "cArgs = %d\n", pdp->cArgs);
314 ok(!pdp->cNamedArgs, "cNamedArgs = %d\n", pdp->cNamedArgs);
315 ok(pvarRes != NULL, "pvarRes == NULL\n");
316 ok(V_VT(pvarRes) == VT_EMPTY, "V_VT(pvarRes) = %d\n", V_VT(pvarRes));
317 ok(pei != NULL, "pei == NULL\n");
319 V_VT(pvarRes) = VT_BSTR;
320 V_BSTR(pvarRes) = a2bstr(vt2a(pdp->rgvarg));
324 ok(0, "unexpected call %d\n", id);
325 return DISP_E_MEMBERNOTFOUND;
328 static IDispatchExVtbl GlobalVtbl = {
329 DispatchEx_QueryInterface,
332 DispatchEx_GetTypeInfoCount,
333 DispatchEx_GetTypeInfo,
334 DispatchEx_GetIDsOfNames,
338 DispatchEx_DeleteMemberByName,
339 DispatchEx_DeleteMemberByDispID,
340 DispatchEx_GetMemberProperties,
341 DispatchEx_GetMemberName,
342 DispatchEx_GetNextDispID,
343 DispatchEx_GetNameSpaceParent
346 static IDispatchEx Global = { &GlobalVtbl };
348 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
352 if(IsEqualGUID(&IID_IUnknown, riid))
354 else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
357 return E_NOINTERFACE;
359 IUnknown_AddRef((IUnknown*)*ppv);
363 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
368 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
373 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
375 *plcid = GetUserDefaultLCID();
379 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
380 DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
382 ok(dwReturnMask == SCRIPTINFO_IUNKNOWN, "unexpected dwReturnMask %x\n", dwReturnMask);
383 ok(!ppti, "ppti != NULL\n");
385 if(strcmp_wa(pstrName, "test"))
386 ok(0, "unexpected pstrName %s\n", wine_dbgstr_w(pstrName));
388 *ppiunkItem = (IUnknown*)&Global;
392 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
397 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
398 const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
403 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
408 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
413 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
418 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
423 #undef ACTSCPSITE_THIS
425 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
426 ActiveScriptSite_QueryInterface,
427 ActiveScriptSite_AddRef,
428 ActiveScriptSite_Release,
429 ActiveScriptSite_GetLCID,
430 ActiveScriptSite_GetItemInfo,
431 ActiveScriptSite_GetDocVersionString,
432 ActiveScriptSite_OnScriptTerminate,
433 ActiveScriptSite_OnStateChange,
434 ActiveScriptSite_OnScriptError,
435 ActiveScriptSite_OnEnterScript,
436 ActiveScriptSite_OnLeaveScript
439 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
441 static IActiveScript *create_script(void)
443 IActiveScript *script;
446 hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
447 &IID_IActiveScript, (void**)&script);
448 ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
453 static HRESULT parse_script(DWORD flags, BSTR script_str)
455 IActiveScriptParse *parser;
456 IActiveScript *engine;
457 IDispatch *script_disp;
460 engine = create_script();
464 hres = IActiveScript_QueryInterface(engine, &IID_IActiveScriptParse, (void**)&parser);
465 ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
468 IActiveScript_Release(engine);
472 hres = IActiveScriptParse64_InitNew(parser);
473 ok(hres == S_OK, "InitNew failed: %08x\n", hres);
475 hres = IActiveScript_SetScriptSite(engine, &ActiveScriptSite);
476 ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
478 hres = IActiveScript_AddNamedItem(engine, testW,
479 SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|flags);
480 ok(hres == S_OK, "AddNamedItem failed: %08x\n", hres);
482 hres = IActiveScript_SetScriptState(engine, SCRIPTSTATE_STARTED);
483 ok(hres == S_OK, "SetScriptState(SCRIPTSTATE_STARTED) failed: %08x\n", hres);
485 hres = IActiveScript_GetScriptDispatch(engine, NULL, &script_disp);
486 ok(hres == S_OK, "GetScriptDispatch failed: %08x\n", hres);
487 ok(script_disp != NULL, "script_disp == NULL\n");
488 ok(script_disp != (IDispatch*)&Global, "script_disp == Global\n");
490 hres = IActiveScriptParse64_ParseScriptText(parser, script_str, NULL, NULL, NULL, 0, 0, 0, NULL, NULL);
492 IDispatch_Release(script_disp);
493 IActiveScript_Release(engine);
494 IUnknown_Release(parser);
499 static void parse_script_af(DWORD flags, const char *src)
505 hres = parse_script(flags, tmp);
507 ok(hres == S_OK, "parse_script failed: %08x\n", hres);
510 static void parse_script_a(const char *src)
512 parse_script_af(SCRIPTITEM_GLOBALMEMBERS, src);
515 static BSTR get_script_from_file(const char *filename)
519 const char *file_map;
522 file = CreateFileA(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
523 if(file == INVALID_HANDLE_VALUE) {
524 trace("Could not open file: %u\n", GetLastError());
528 size = GetFileSize(file, NULL);
530 map = CreateFileMapping(file, NULL, PAGE_READONLY, 0, 0, NULL);
532 if(map == INVALID_HANDLE_VALUE) {
533 trace("Could not create file mapping: %u\n", GetLastError());
537 file_map = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0);
540 trace("MapViewOfFile failed: %u\n", GetLastError());
544 len = MultiByteToWideChar(CP_ACP, 0, file_map, size, NULL, 0);
545 ret = SysAllocStringLen(NULL, len);
546 MultiByteToWideChar(CP_ACP, 0, file_map, size, ret, len);
548 UnmapViewOfFile(file_map);
553 static void run_from_file(const char *filename)
558 script_str = get_script_from_file(filename);
562 strict_dispid_check = FALSE;
563 hres = parse_script(SCRIPTITEM_GLOBALMEMBERS, script_str);
564 SysFreeString(script_str);
565 ok(hres == S_OK, "parse_script failed: %08x\n", hres);
568 static void run_from_res(const char *name)
576 strict_dispid_check = FALSE;
579 src = FindResourceA(NULL, name, (LPCSTR)40);
580 ok(src != NULL, "Could not find resource %s\n", name);
582 size = SizeofResource(NULL, src);
583 data = LoadResource(NULL, src);
585 len = MultiByteToWideChar(CP_ACP, 0, data, size, NULL, 0);
586 str = SysAllocStringLen(NULL, len);
587 MultiByteToWideChar(CP_ACP, 0, data, size, str, len);
589 SET_EXPECT(global_success_d);
590 SET_EXPECT(global_success_i);
591 hres = parse_script(SCRIPTITEM_GLOBALMEMBERS, str);
592 CHECK_CALLED(global_success_d);
593 CHECK_CALLED(global_success_i);
595 ok(hres == S_OK, "parse_script failed: %08x\n", hres);
599 static void run_tests(void)
601 strict_dispid_check = TRUE;
604 parse_script_a("' empty ;");
606 SET_EXPECT(global_success_d);
607 SET_EXPECT(global_success_i);
608 parse_script_a("reportSuccess");
609 CHECK_CALLED(global_success_d);
610 CHECK_CALLED(global_success_i);
612 SET_EXPECT(global_success_d);
613 SET_EXPECT(global_success_i);
614 parse_script_a("reportSuccess()");
615 CHECK_CALLED(global_success_d);
616 CHECK_CALLED(global_success_i);
618 SET_EXPECT(global_success_d);
619 SET_EXPECT(global_success_i);
620 parse_script_a("Call reportSuccess");
621 CHECK_CALLED(global_success_d);
622 CHECK_CALLED(global_success_i);
624 run_from_res("lang.vbs");
627 static BOOL check_vbscript(void)
629 IActiveScript *vbscript;
632 hres = CoCreateInstance(&CLSID_VBScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
633 &IID_IActiveScript, (void**)&vbscript);
635 IActiveScript_Release(vbscript);
645 argc = winetest_get_mainargs(&argv);
649 if(!check_vbscript()) {
650 win_skip("Broken engine, probably too old\n");
652 run_from_file(argv[2]);