[CSCRIPT][WSCRIPT] Sync with Wine Staging 3.17. CORE-15127
[reactos.git] / base / applications / cmdutils / wscript / main.c
1 /*
2 * Copyright 2010 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 <stdarg.h>
20
21 #define COBJMACROS
22 #ifdef __REACTOS__
23 #define CONST_VTABLE
24 #endif
25
26 #include <windef.h>
27 #include <winbase.h>
28 #include <winreg.h>
29 #include <ole2.h>
30 #include <shellapi.h>
31 #include <activscp.h>
32 #include <initguid.h>
33
34 #include "wscript.h"
35
36 #include <wine/debug.h>
37 #include <wine/unicode.h>
38
39 #ifdef _WIN64
40
41 #define IActiveScriptParse_Release IActiveScriptParse64_Release
42 #define IActiveScriptParse_InitNew IActiveScriptParse64_InitNew
43 #define IActiveScriptParse_ParseScriptText IActiveScriptParse64_ParseScriptText
44
45 #else
46
47 #define IActiveScriptParse_Release IActiveScriptParse32_Release
48 #define IActiveScriptParse_InitNew IActiveScriptParse32_InitNew
49 #define IActiveScriptParse_ParseScriptText IActiveScriptParse32_ParseScriptText
50
51 #endif
52
53 WINE_DEFAULT_DEBUG_CHANNEL(wscript);
54
55 static const WCHAR wscriptW[] = {'W','S','c','r','i','p','t',0};
56 static const WCHAR wshW[] = {'W','S','H',0};
57 WCHAR scriptFullName[MAX_PATH];
58
59 ITypeInfo *host_ti;
60 ITypeInfo *arguments_ti;
61
62 static HRESULT query_interface(REFIID,void**);
63
64 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface,
65 REFIID riid, void **ppv)
66 {
67 return query_interface(riid, ppv);
68 }
69
70 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
71 {
72 return 2;
73 }
74
75 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
76 {
77 return 1;
78 }
79
80 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
81 {
82 WINE_TRACE("()\n");
83
84 *plcid = GetUserDefaultLCID();
85 return S_OK;
86 }
87
88 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface,
89 LPCOLESTR pstrName, DWORD dwReturnMask, IUnknown **ppunkItem, ITypeInfo **ppti)
90 {
91 WINE_TRACE("(%s %x %p %p)\n", wine_dbgstr_w(pstrName), dwReturnMask, ppunkItem, ppti);
92
93 if(strcmpW(pstrName, wshW) && strcmpW(pstrName, wscriptW))
94 return E_FAIL;
95
96 if(dwReturnMask & SCRIPTINFO_ITYPEINFO) {
97 ITypeInfo_AddRef(host_ti);
98 *ppti = host_ti;
99 }
100
101 if(dwReturnMask & SCRIPTINFO_IUNKNOWN) {
102 IHost_AddRef(&host_obj);
103 *ppunkItem = (IUnknown*)&host_obj;
104 }
105
106 return S_OK;
107 }
108
109 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface,
110 BSTR *pbstrVersion)
111 {
112 WINE_FIXME("()\n");
113 return E_NOTIMPL;
114 }
115
116 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
117 const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
118 {
119 WINE_FIXME("()\n");
120 return E_NOTIMPL;
121 }
122
123 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface,
124 SCRIPTSTATE ssScriptState)
125 {
126 WINE_TRACE("(%x)\n", ssScriptState);
127 return S_OK;
128 }
129
130 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface,
131 IActiveScriptError *pscripterror)
132 {
133 WINE_FIXME("()\n");
134 return E_NOTIMPL;
135 }
136
137 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
138 {
139 WINE_TRACE("()\n");
140 return S_OK;
141 }
142
143 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
144 {
145 WINE_TRACE("()\n");
146 return S_OK;
147 }
148
149 static IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
150 ActiveScriptSite_QueryInterface,
151 ActiveScriptSite_AddRef,
152 ActiveScriptSite_Release,
153 ActiveScriptSite_GetLCID,
154 ActiveScriptSite_GetItemInfo,
155 ActiveScriptSite_GetDocVersionString,
156 ActiveScriptSite_OnScriptTerminate,
157 ActiveScriptSite_OnStateChange,
158 ActiveScriptSite_OnScriptError,
159 ActiveScriptSite_OnEnterScript,
160 ActiveScriptSite_OnLeaveScript
161 };
162
163 static IActiveScriptSite script_site = { &ActiveScriptSiteVtbl };
164
165 static HRESULT WINAPI ActiveScriptSiteWindow_QueryInterface(IActiveScriptSiteWindow *iface, REFIID riid, void **ppv)
166 {
167 return query_interface(riid, ppv);
168 }
169
170 static ULONG WINAPI ActiveScriptSiteWindow_AddRef(IActiveScriptSiteWindow *iface)
171 {
172 return 2;
173 }
174
175 static ULONG WINAPI ActiveScriptSiteWindow_Release(IActiveScriptSiteWindow *iface)
176 {
177 return 1;
178 }
179
180 static HRESULT WINAPI ActiveScriptSiteWindow_GetWindow(IActiveScriptSiteWindow *iface, HWND *phwnd)
181 {
182 TRACE("(%p)\n", phwnd);
183
184 *phwnd = NULL;
185 return S_OK;
186 }
187
188 static HRESULT WINAPI ActiveScriptSiteWindow_EnableModeless(IActiveScriptSiteWindow *iface, BOOL fEnable)
189 {
190 TRACE("(%x)\n", fEnable);
191 return S_OK;
192 }
193
194 static const IActiveScriptSiteWindowVtbl ActiveScriptSiteWindowVtbl = {
195 ActiveScriptSiteWindow_QueryInterface,
196 ActiveScriptSiteWindow_AddRef,
197 ActiveScriptSiteWindow_Release,
198 ActiveScriptSiteWindow_GetWindow,
199 ActiveScriptSiteWindow_EnableModeless
200 };
201
202 static IActiveScriptSiteWindow script_site_window = { &ActiveScriptSiteWindowVtbl };
203
204 static HRESULT query_interface(REFIID riid, void **ppv)
205 {
206 if(IsEqualGUID(riid, &IID_IUnknown)) {
207 TRACE("(IID_IUnknown %p)\n", ppv);
208 *ppv = &script_site;
209 }else if(IsEqualGUID(riid, &IID_IActiveScriptSite)) {
210 TRACE("(IID_IActiveScriptSite %p)\n", ppv);
211 *ppv = &script_site;
212 }else if(IsEqualGUID(riid, &IID_IActiveScriptSiteWindow)) {
213 TRACE("(IID_IActiveScriptSiteWindow %p)\n", ppv);
214 *ppv = &script_site_window;
215 }else {
216 *ppv = NULL;
217 TRACE("(%s %p)\n", wine_dbgstr_guid(riid), ppv);
218 return E_NOINTERFACE;
219 }
220
221 IUnknown_AddRef((IUnknown*)*ppv);
222 return S_OK;
223 }
224
225 static BOOL load_typelib(void)
226 {
227 ITypeLib *typelib;
228 HRESULT hres;
229
230 static const WCHAR wscript_exeW[] = {'w','s','c','r','i','p','t','.','e','x','e',0};
231
232 hres = LoadTypeLib(wscript_exeW, &typelib);
233 if(FAILED(hres))
234 return FALSE;
235
236 hres = ITypeLib_GetTypeInfoOfGuid(typelib, &IID_IHost, &host_ti);
237 if(SUCCEEDED(hres))
238 hres = ITypeLib_GetTypeInfoOfGuid(typelib, &IID_IArguments2, &arguments_ti);
239
240 ITypeLib_Release(typelib);
241 return SUCCEEDED(hres);
242 }
243
244 static BOOL get_engine_clsid(const WCHAR *ext, CLSID *clsid)
245 {
246 WCHAR fileid[64], progid[64];
247 DWORD res;
248 LONG size;
249 HKEY hkey;
250 HRESULT hres;
251
252 static const WCHAR script_engineW[] =
253 {'\\','S','c','r','i','p','t','E','n','g','i','n','e',0};
254
255 res = RegOpenKeyW(HKEY_CLASSES_ROOT, ext, &hkey);
256 if(res != ERROR_SUCCESS)
257 return FALSE;
258
259 size = ARRAY_SIZE(fileid);
260 res = RegQueryValueW(hkey, NULL, fileid, &size);
261 RegCloseKey(hkey);
262 if(res != ERROR_SUCCESS)
263 return FALSE;
264
265 WINE_TRACE("fileid is %s\n", wine_dbgstr_w(fileid));
266
267 strcatW(fileid, script_engineW);
268 res = RegOpenKeyW(HKEY_CLASSES_ROOT, fileid, &hkey);
269 if(res != ERROR_SUCCESS)
270 return FALSE;
271
272 size = ARRAY_SIZE(progid);
273 res = RegQueryValueW(hkey, NULL, progid, &size);
274 RegCloseKey(hkey);
275 if(res != ERROR_SUCCESS)
276 return FALSE;
277
278 WINE_TRACE("ProgID is %s\n", wine_dbgstr_w(progid));
279
280 hres = CLSIDFromProgID(progid, clsid);
281 return SUCCEEDED(hres);
282 }
283
284 static BOOL create_engine(CLSID *clsid, IActiveScript **script_ret,
285 IActiveScriptParse **parser)
286 {
287 IActiveScript *script;
288 IUnknown *unk;
289 HRESULT hres;
290
291 hres = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
292 &IID_IUnknown, (void**)&unk);
293 if(FAILED(hres))
294 return FALSE;
295
296 hres = IUnknown_QueryInterface(unk, &IID_IActiveScript, (void**)&script);
297 IUnknown_Release(unk);
298 if(FAILED(hres))
299 return FALSE;
300
301 hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)parser);
302 if(FAILED(hres)) {
303 IActiveScript_Release(script);
304 return FALSE;
305 }
306
307 *script_ret = script;
308 return TRUE;
309 }
310
311 static BOOL init_engine(IActiveScript *script, IActiveScriptParse *parser)
312 {
313 HRESULT hres;
314
315 if(!load_typelib())
316 return FALSE;
317
318 hres = IActiveScript_SetScriptSite(script, &script_site);
319 if(FAILED(hres))
320 return FALSE;
321
322 hres = IActiveScriptParse_InitNew(parser);
323 if(FAILED(hres))
324 return FALSE;
325
326 hres = IActiveScript_AddNamedItem(script, wscriptW, SCRIPTITEM_ISVISIBLE);
327 if(FAILED(hres))
328 return FALSE;
329
330 hres = IActiveScript_AddNamedItem(script, wshW, SCRIPTITEM_ISVISIBLE);
331 if(FAILED(hres))
332 return FALSE;
333
334 hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_INITIALIZED);
335 return SUCCEEDED(hres);
336 }
337
338 static BSTR get_script_str(const WCHAR *filename)
339 {
340 const char *file_map;
341 HANDLE file, map;
342 DWORD size, len;
343 BSTR ret;
344
345 file = CreateFileW(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
346 if(file == INVALID_HANDLE_VALUE)
347 return NULL;
348
349 size = GetFileSize(file, NULL);
350 map = CreateFileMappingW(file, NULL, PAGE_READONLY, 0, 0, NULL);
351 CloseHandle(file);
352 if(map == INVALID_HANDLE_VALUE)
353 return NULL;
354
355 file_map = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0);
356 CloseHandle(map);
357 if(!file_map)
358 return NULL;
359
360 len = MultiByteToWideChar(CP_ACP, 0, file_map, size, NULL, 0);
361 ret = SysAllocStringLen(NULL, len);
362 MultiByteToWideChar(CP_ACP, 0, file_map, size, ret, len);
363
364 UnmapViewOfFile(file_map);
365 return ret;
366 }
367
368 static void run_script(const WCHAR *filename, IActiveScript *script, IActiveScriptParse *parser)
369 {
370 BSTR text;
371 HRESULT hres;
372
373 text = get_script_str(filename);
374 if(!text) {
375 WINE_FIXME("Could not get script text\n");
376 return;
377 }
378
379 hres = IActiveScriptParse_ParseScriptText(parser, text, NULL, NULL, NULL, 1, 1,
380 SCRIPTTEXT_HOSTMANAGESSOURCE|SCRIPTITEM_ISVISIBLE, NULL, NULL);
381 SysFreeString(text);
382 if(FAILED(hres)) {
383 WINE_FIXME("ParseScriptText failed: %08x\n", hres);
384 return;
385 }
386
387 hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_STARTED);
388 if(FAILED(hres))
389 WINE_FIXME("SetScriptState failed: %08x\n", hres);
390 }
391
392 static BOOL set_host_properties(const WCHAR *prop)
393 {
394 static const WCHAR nologoW[] = {'n','o','l','o','g','o',0};
395 static const WCHAR iactive[] = {'i',0};
396 static const WCHAR batch[] = {'b',0};
397
398 if(*prop == '/') {
399 ++prop;
400 if(*prop == '/')
401 ++prop;
402 }
403 else
404 ++prop;
405
406 if(strcmpiW(prop, iactive) == 0)
407 wshInteractive = VARIANT_TRUE;
408 else if(strcmpiW(prop, batch) == 0)
409 wshInteractive = VARIANT_FALSE;
410 else if(strcmpiW(prop, nologoW) == 0)
411 WINE_FIXME("ignored %s switch\n", debugstr_w(nologoW));
412 else
413 {
414 WINE_FIXME("unsupported switch %s\n", debugstr_w(prop));
415 return FALSE;
416 }
417 return TRUE;
418 }
419
420 int WINAPI wWinMain(HINSTANCE hInst, HINSTANCE hPrevInst, LPWSTR cmdline, int cmdshow)
421 {
422 WCHAR *ext, *filepart, *filename = NULL;
423 IActiveScriptParse *parser;
424 IActiveScript *script;
425 WCHAR **argv;
426 CLSID clsid;
427 int argc, i;
428 DWORD res;
429
430 WINE_TRACE("(%p %p %s %x)\n", hInst, hPrevInst, wine_dbgstr_w(cmdline), cmdshow);
431
432 argv = CommandLineToArgvW(GetCommandLineW(), &argc);
433 if(!argv)
434 return 1;
435
436 for(i=1; i<argc; i++) {
437 if(*argv[i] == '/' || *argv[i] == '-') {
438 if(!set_host_properties(argv[i]))
439 return 1;
440 }else {
441 filename = argv[i];
442 argums = argv+i+1;
443 numOfArgs = argc-i-1;
444 break;
445 }
446 }
447
448 if(!filename) {
449 WINE_FIXME("No file name specified\n");
450 return 1;
451 }
452 res = GetFullPathNameW(filename, ARRAY_SIZE(scriptFullName), scriptFullName, &filepart);
453 if(!res || res > ARRAY_SIZE(scriptFullName))
454 return 1;
455
456 ext = strrchrW(filepart, '.');
457 if(!ext || !get_engine_clsid(ext, &clsid)) {
458 WINE_FIXME("Could not find engine for %s\n", wine_dbgstr_w(ext));
459 return 1;
460 }
461
462 CoInitialize(NULL);
463
464 if(!create_engine(&clsid, &script, &parser)) {
465 WINE_FIXME("Could not create script engine\n");
466 CoUninitialize();
467 return 1;
468 }
469
470 if(init_engine(script, parser)) {
471 run_script(filename, script, parser);
472 IActiveScript_Close(script);
473 ITypeInfo_Release(host_ti);
474 }else {
475 WINE_FIXME("Script initialization failed\n");
476 }
477
478 IActiveScript_Release(script);
479 IActiveScriptParse_Release(parser);
480
481 CoUninitialize();
482
483 return 0;
484 }