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 #define DEFINE_EXPECT(func) \
31 static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
33 #define SET_EXPECT(func) \
34 expect_ ## func = TRUE
36 #define SET_CALLED(func) \
37 called_ ## func = TRUE
39 #define CHECK_EXPECT2(func) \
41 ok(expect_ ##func, "unexpected call " #func "\n"); \
42 called_ ## func = TRUE; \
45 #define CHECK_EXPECT(func) \
47 CHECK_EXPECT2(func); \
48 expect_ ## func = FALSE; \
51 #define CHECK_CALLED(func) \
53 ok(called_ ## func, "expected " #func "\n"); \
54 expect_ ## func = called_ ## func = FALSE; \
57 DEFINE_EXPECT(reportSuccess
);
59 #define DISPID_TESTOBJ_OK 10000
60 #define DISPID_TESTOBJ_TRACE 10001
61 #define DISPID_TESTOBJ_REPORTSUCCESS 10002
62 #define DISPID_TESTOBJ_WSCRIPTFULLNAME 10003
63 #define DISPID_TESTOBJ_WSCRIPTPATH 10004
64 #define DISPID_TESTOBJ_WSCRIPTSCRIPTNAME 10005
66 #define TESTOBJ_CLSID "{178fc166-f585-4e24-9c13-4bb7faf80646}"
68 static const GUID CLSID_TestObj
=
69 {0x178fc166,0xf585,0x4e24,{0x9c,0x13,0x4b,0xb7,0xfa,0xf8,0x06,0x46}};
71 static const char *script_name
;
73 static int strcmp_wa(LPCWSTR strw
, const char *stra
)
76 MultiByteToWideChar(CP_ACP
, 0, stra
, -1, buf
, sizeof(buf
)/sizeof(WCHAR
));
77 return lstrcmpW(strw
, buf
);
80 static const WCHAR
* mystrrchr(const WCHAR
*str
, WCHAR ch
)
82 const WCHAR
*pos
= NULL
, *current
= str
;
83 while(*current
!= 0) {
91 static BSTR
a2bstr(const char *str
)
96 len
= MultiByteToWideChar(CP_ACP
, 0, str
, -1, NULL
, 0);
97 ret
= SysAllocStringLen(NULL
, len
-1);
98 MultiByteToWideChar(CP_ACP
, 0, str
, -1, ret
, len
);
103 static HRESULT WINAPI
Dispatch_QueryInterface(IDispatch
*iface
, REFIID riid
, void **ppv
)
105 if(IsEqualGUID(riid
, &IID_IUnknown
) || IsEqualGUID(riid
, &IID_IDispatch
)) {
111 return E_NOINTERFACE
;
114 static ULONG WINAPI
Dispatch_AddRef(IDispatch
*iface
)
119 static ULONG WINAPI
Dispatch_Release(IDispatch
*iface
)
124 static HRESULT WINAPI
Dispatch_GetTypeInfoCount(IDispatch
*iface
, UINT
*pctinfo
)
126 ok(0, "unexpected call\n");
130 static HRESULT WINAPI
Dispatch_GetTypeInfo(IDispatch
*iface
, UINT iTInfo
,
131 LCID lcid
, ITypeInfo
**ppTInfo
)
133 ok(0, "unexpected call\n");
137 static HRESULT WINAPI
Dispatch_GetIDsOfNames(IDispatch
*iface
, REFIID riid
,
138 LPOLESTR
*rgszNames
, UINT cNames
, LCID lcid
, DISPID
*rgDispId
)
142 for(i
=0; i
<cNames
; i
++) {
143 if(!strcmp_wa(rgszNames
[i
], "ok")) {
144 rgDispId
[i
] = DISPID_TESTOBJ_OK
;
145 }else if(!strcmp_wa(rgszNames
[i
], "trace")) {
146 rgDispId
[i
] = DISPID_TESTOBJ_TRACE
;
147 }else if(!strcmp_wa(rgszNames
[i
], "reportSuccess")) {
148 rgDispId
[i
] = DISPID_TESTOBJ_REPORTSUCCESS
;
149 }else if(!strcmp_wa(rgszNames
[i
], "wscriptFullName")) {
150 rgDispId
[i
] = DISPID_TESTOBJ_WSCRIPTFULLNAME
;
151 }else if(!strcmp_wa(rgszNames
[i
], "wscriptPath")) {
152 rgDispId
[i
] = DISPID_TESTOBJ_WSCRIPTPATH
;
153 }else if(!strcmp_wa(rgszNames
[i
], "wscriptScriptName")) {
154 rgDispId
[i
] = DISPID_TESTOBJ_WSCRIPTSCRIPTNAME
;
156 ok(0, "unexpected name %s\n", wine_dbgstr_w(rgszNames
[i
]));
157 return DISP_E_UNKNOWNNAME
;
164 static HRESULT WINAPI
Dispatch_Invoke(IDispatch
*iface
, DISPID dispIdMember
, REFIID riid
, LCID lcid
,
165 WORD wFlags
, DISPPARAMS
*pdp
, VARIANT
*pVarResult
, EXCEPINFO
*pExcepInfo
, UINT
*puArgErr
)
167 switch(dispIdMember
) {
168 case DISPID_TESTOBJ_OK
:
169 ok(wFlags
== INVOKE_FUNC
, "wFlags = %x\n", wFlags
);
170 ok(pdp
->cArgs
== 2, "cArgs = %d\n", pdp
->cArgs
);
171 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
172 ok(V_VT(pdp
->rgvarg
) == VT_BSTR
, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp
->rgvarg
));
173 ok(V_VT(pdp
->rgvarg
+1) == VT_BOOL
, "V_VT(psp->rgvargs+1) = %d\n", V_VT(pdp
->rgvarg
));
174 ok(V_BOOL(pdp
->rgvarg
+1), "%s: %s\n", script_name
, wine_dbgstr_w(V_BSTR(pdp
->rgvarg
)));
176 V_VT(pVarResult
) = VT_EMPTY
;
178 case DISPID_TESTOBJ_TRACE
:
179 ok(wFlags
== INVOKE_FUNC
, "wFlags = %x\n", wFlags
);
180 ok(pdp
->cArgs
== 1, "cArgs = %d\n", pdp
->cArgs
);
181 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
182 ok(V_VT(pdp
->rgvarg
) == VT_BSTR
, "V_VT(psp->rgvargs) = %d\n", V_VT(pdp
->rgvarg
));
183 trace("%s: %s\n", script_name
, wine_dbgstr_w(V_BSTR(pdp
->rgvarg
)));
185 V_VT(pVarResult
) = VT_EMPTY
;
187 case DISPID_TESTOBJ_REPORTSUCCESS
:
188 CHECK_EXPECT(reportSuccess
);
190 ok(wFlags
== INVOKE_FUNC
, "wFlags = %x\n", wFlags
);
191 ok(pdp
->cArgs
== 0, "cArgs = %d\n", pdp
->cArgs
);
192 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
194 V_VT(pVarResult
) = VT_EMPTY
;
196 case DISPID_TESTOBJ_WSCRIPTFULLNAME
:
198 WCHAR fullName
[MAX_PATH
];
199 const WCHAR wscriptexe
[] = {'w','s','c','r','i','p','t','.','e','x','e',0};
202 ok(wFlags
== INVOKE_PROPERTYGET
, "wFlags = %x\n", wFlags
);
203 ok(pdp
->cArgs
== 0, "cArgs = %d\n", pdp
->cArgs
);
204 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
205 V_VT(pVarResult
) = VT_BSTR
;
206 res
= SearchPathW(NULL
, wscriptexe
, NULL
, sizeof(fullName
)/sizeof(WCHAR
), fullName
, NULL
);
209 if(!(V_BSTR(pVarResult
) = SysAllocString(fullName
)))
210 return E_OUTOFMEMORY
;
213 case DISPID_TESTOBJ_WSCRIPTPATH
:
215 WCHAR fullPath
[MAX_PATH
];
216 const WCHAR wscriptexe
[] = {'w','s','c','r','i','p','t','.','e','x','e',0};
220 ok(wFlags
== INVOKE_PROPERTYGET
, "wFlags = %x\n", wFlags
);
221 ok(pdp
->cArgs
== 0, "cArgs = %d\n", pdp
->cArgs
);
222 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
223 V_VT(pVarResult
) = VT_BSTR
;
224 res
= SearchPathW(NULL
, wscriptexe
, NULL
, sizeof(fullPath
)/sizeof(WCHAR
), fullPath
, NULL
);
227 pos
= mystrrchr(fullPath
, '\\');
228 if(!(V_BSTR(pVarResult
) = SysAllocStringLen(fullPath
, pos
-fullPath
)))
229 return E_OUTOFMEMORY
;
232 case DISPID_TESTOBJ_WSCRIPTSCRIPTNAME
:
234 char fullPath
[MAX_PATH
];
238 ok(wFlags
== INVOKE_PROPERTYGET
, "wFlags = %x\n", wFlags
);
239 ok(pdp
->cArgs
== 0, "cArgs = %d\n", pdp
->cArgs
);
240 ok(!pdp
->cNamedArgs
, "cNamedArgs = %d\n", pdp
->cNamedArgs
);
241 V_VT(pVarResult
) = VT_BSTR
;
242 res
= GetFullPathNameA(script_name
, sizeof(fullPath
)/sizeof(WCHAR
), fullPath
, &pos
);
243 if(!res
|| res
> sizeof(fullPath
)/sizeof(WCHAR
))
245 if(!(V_BSTR(pVarResult
) = SysAllocString(a2bstr(pos
))))
246 return E_OUTOFMEMORY
;
250 ok(0, "unexpected dispIdMember %d\n", dispIdMember
);
257 static IDispatchVtbl testobj_vtbl
= {
258 Dispatch_QueryInterface
,
261 Dispatch_GetTypeInfoCount
,
262 Dispatch_GetTypeInfo
,
263 Dispatch_GetIDsOfNames
,
267 static IDispatch testobj
= { &testobj_vtbl
};
269 static HRESULT WINAPI
ClassFactory_QueryInterface(IClassFactory
*iface
, REFIID riid
, void **ppv
)
271 if(IsEqualGUID(&IID_IUnknown
, riid
) || IsEqualGUID(&IID_IClassFactory
, riid
)) {
277 return E_NOINTERFACE
;
280 static ULONG WINAPI
ClassFactory_AddRef(IClassFactory
*iface
)
285 static ULONG WINAPI
ClassFactory_Release(IClassFactory
*iface
)
290 static HRESULT WINAPI
ClassFactory_CreateInstance(IClassFactory
*iface
, IUnknown
*outer
, REFIID riid
, void **ppv
)
292 ok(!outer
, "outer = %p\n", outer
);
293 return IDispatch_QueryInterface(&testobj
, riid
, ppv
);
296 static HRESULT WINAPI
ClassFactory_LockServer(IClassFactory
*iface
, BOOL dolock
)
301 static const IClassFactoryVtbl ClassFactoryVtbl
= {
302 ClassFactory_QueryInterface
,
304 ClassFactory_Release
,
305 ClassFactory_CreateInstance
,
306 ClassFactory_LockServer
309 static IClassFactory testobj_cf
= { &ClassFactoryVtbl
};
311 static void run_test(const char *file_name
)
313 SECURITY_ATTRIBUTES sa
= {sizeof(sa
), 0, TRUE
};
314 char command
[MAX_PATH
];
315 STARTUPINFOA si
= {sizeof(si
)};
316 PROCESS_INFORMATION pi
;
319 script_name
= file_name
;
320 sprintf(command
, "wscript.exe %s", file_name
);
322 SET_EXPECT(reportSuccess
);
324 bres
= CreateProcessA(NULL
, command
, NULL
, NULL
, TRUE
, 0, NULL
, NULL
, &si
, &pi
);
326 win_skip("script.exe is not available\n");
327 SET_CALLED(reportSuccess
);
331 WaitForSingleObject(pi
.hProcess
, INFINITE
);
332 CloseHandle(pi
.hThread
);
333 CloseHandle(pi
.hProcess
);
335 CHECK_CALLED(reportSuccess
);
338 static BOOL WINAPI
test_enum_proc(HMODULE module
, LPCTSTR type
, LPSTR name
, LONG_PTR param
)
340 const char *script_data
, *ext
;
341 DWORD script_size
, size
;
342 char file_name
[MAX_PATH
];
347 trace("running %s test...\n", name
);
349 src
= FindResourceA(NULL
, name
, type
);
350 ok(src
!= NULL
, "Could not find resource %s: %u\n", name
, GetLastError());
354 script_data
= LoadResource(NULL
, src
);
355 script_size
= SizeofResource(NULL
, src
);
356 while(script_size
&& !script_data
[script_size
-1])
359 ext
= strrchr(name
, '.');
360 ok(ext
!= NULL
, "no script extension\n");
364 sprintf(file_name
, "test%s", ext
);
366 file
= CreateFileA(file_name
, GENERIC_WRITE
, 0, NULL
, CREATE_ALWAYS
,
367 FILE_ATTRIBUTE_NORMAL
, NULL
);
368 ok(file
!= INVALID_HANDLE_VALUE
, "CreateFile failed: %u\n", GetLastError());
369 if(file
== INVALID_HANDLE_VALUE
)
372 res
= WriteFile(file
, script_data
, script_size
, &size
, NULL
);
374 ok(res
, "Could not write to file: %u\n", GetLastError());
380 DeleteFileA(file_name
);
384 static BOOL
init_key(const char *key_name
, const char *def_value
, BOOL init
)
390 RegDeleteKey(HKEY_CLASSES_ROOT
, key_name
);
394 res
= RegCreateKeyA(HKEY_CLASSES_ROOT
, key_name
, &hkey
);
395 if(res
!= ERROR_SUCCESS
)
399 res
= RegSetValueA(hkey
, NULL
, REG_SZ
, def_value
, strlen(def_value
));
402 return res
== ERROR_SUCCESS
;
405 static BOOL
init_registry(BOOL init
)
407 return init_key("Wine.Test\\CLSID", TESTOBJ_CLSID
, init
);
410 static BOOL
register_activex(void)
415 if(!init_registry(TRUE
)) {
416 init_registry(FALSE
);
420 hres
= CoRegisterClassObject(&CLSID_TestObj
, (IUnknown
*)&testobj_cf
,
421 CLSCTX_SERVER
, REGCLS_MULTIPLEUSE
, ®id
);
422 ok(hres
== S_OK
, "Could not register script engine: %08x\n", hres
);
431 CoInitializeEx(NULL
, COINIT_MULTITHREADED
);
434 argc
= winetest_get_mainargs(&argv
);
438 EnumResourceNamesA(NULL
, "TESTSCRIPT", test_enum_proc
, 0);
440 init_registry(FALSE
);