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 }
485