2 * Copyright 2010 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
26 #define IActiveScriptParse_Release IActiveScriptParse64_Release
27 #define IActiveScriptParse_InitNew IActiveScriptParse64_InitNew
28 #define IActiveScriptParse_ParseScriptText IActiveScriptParse64_ParseScriptText
32 #define IActiveScriptParse_Release IActiveScriptParse32_Release
33 #define IActiveScriptParse_InitNew IActiveScriptParse32_InitNew
34 #define IActiveScriptParse_ParseScriptText IActiveScriptParse32_ParseScriptText
38 static const WCHAR wscriptW
[] = {'W','S','c','r','i','p','t',0};
39 static const WCHAR wshW
[] = {'W','S','H',0};
40 WCHAR scriptFullName
[MAX_PATH
];
43 ITypeInfo
*arguments_ti
;
45 static HRESULT
query_interface(REFIID
,void**);
47 static HRESULT WINAPI
ActiveScriptSite_QueryInterface(IActiveScriptSite
*iface
,
48 REFIID riid
, void **ppv
)
50 return query_interface(riid
, ppv
);
53 static ULONG WINAPI
ActiveScriptSite_AddRef(IActiveScriptSite
*iface
)
58 static ULONG WINAPI
ActiveScriptSite_Release(IActiveScriptSite
*iface
)
63 static HRESULT WINAPI
ActiveScriptSite_GetLCID(IActiveScriptSite
*iface
, LCID
*plcid
)
67 *plcid
= GetUserDefaultLCID();
71 static HRESULT WINAPI
ActiveScriptSite_GetItemInfo(IActiveScriptSite
*iface
,
72 LPCOLESTR pstrName
, DWORD dwReturnMask
, IUnknown
**ppunkItem
, ITypeInfo
**ppti
)
74 WINE_TRACE("(%s %x %p %p)\n", wine_dbgstr_w(pstrName
), dwReturnMask
, ppunkItem
, ppti
);
76 if(strcmpW(pstrName
, wshW
) && strcmpW(pstrName
, wscriptW
))
79 if(dwReturnMask
& SCRIPTINFO_ITYPEINFO
) {
80 ITypeInfo_AddRef(host_ti
);
84 if(dwReturnMask
& SCRIPTINFO_IUNKNOWN
) {
85 IHost_AddRef(&host_obj
);
86 *ppunkItem
= (IUnknown
*)&host_obj
;
92 static HRESULT WINAPI
ActiveScriptSite_GetDocVersionString(IActiveScriptSite
*iface
,
99 static HRESULT WINAPI
ActiveScriptSite_OnScriptTerminate(IActiveScriptSite
*iface
,
100 const VARIANT
*pvarResult
, const EXCEPINFO
*pexcepinfo
)
106 static HRESULT WINAPI
ActiveScriptSite_OnStateChange(IActiveScriptSite
*iface
,
107 SCRIPTSTATE ssScriptState
)
109 WINE_TRACE("(%x)\n", ssScriptState
);
113 static HRESULT WINAPI
ActiveScriptSite_OnScriptError(IActiveScriptSite
*iface
,
114 IActiveScriptError
*pscripterror
)
120 static HRESULT WINAPI
ActiveScriptSite_OnEnterScript(IActiveScriptSite
*iface
)
126 static HRESULT WINAPI
ActiveScriptSite_OnLeaveScript(IActiveScriptSite
*iface
)
132 static IActiveScriptSiteVtbl ActiveScriptSiteVtbl
= {
133 ActiveScriptSite_QueryInterface
,
134 ActiveScriptSite_AddRef
,
135 ActiveScriptSite_Release
,
136 ActiveScriptSite_GetLCID
,
137 ActiveScriptSite_GetItemInfo
,
138 ActiveScriptSite_GetDocVersionString
,
139 ActiveScriptSite_OnScriptTerminate
,
140 ActiveScriptSite_OnStateChange
,
141 ActiveScriptSite_OnScriptError
,
142 ActiveScriptSite_OnEnterScript
,
143 ActiveScriptSite_OnLeaveScript
146 static IActiveScriptSite script_site
= { &ActiveScriptSiteVtbl
};
148 static HRESULT WINAPI
ActiveScriptSiteWindow_QueryInterface(IActiveScriptSiteWindow
*iface
, REFIID riid
, void **ppv
)
150 return query_interface(riid
, ppv
);
153 static ULONG WINAPI
ActiveScriptSiteWindow_AddRef(IActiveScriptSiteWindow
*iface
)
158 static ULONG WINAPI
ActiveScriptSiteWindow_Release(IActiveScriptSiteWindow
*iface
)
163 static HRESULT WINAPI
ActiveScriptSiteWindow_GetWindow(IActiveScriptSiteWindow
*iface
, HWND
*phwnd
)
165 TRACE("(%p)\n", phwnd
);
171 static HRESULT WINAPI
ActiveScriptSiteWindow_EnableModeless(IActiveScriptSiteWindow
*iface
, BOOL fEnable
)
173 TRACE("(%x)\n", fEnable
);
177 static IActiveScriptSiteWindowVtbl ActiveScriptSiteWindowVtbl
= {
178 ActiveScriptSiteWindow_QueryInterface
,
179 ActiveScriptSiteWindow_AddRef
,
180 ActiveScriptSiteWindow_Release
,
181 ActiveScriptSiteWindow_GetWindow
,
182 ActiveScriptSiteWindow_EnableModeless
185 static IActiveScriptSiteWindow script_site_window
= { &ActiveScriptSiteWindowVtbl
};
187 static HRESULT
query_interface(REFIID riid
, void **ppv
)
189 if(IsEqualGUID(riid
, &IID_IUnknown
)) {
190 TRACE("(IID_IUnknown %p)\n", ppv
);
192 }else if(IsEqualGUID(riid
, &IID_IActiveScriptSite
)) {
193 TRACE("(IID_IActiveScriptSite %p)\n", ppv
);
195 }else if(IsEqualGUID(riid
, &IID_IActiveScriptSiteWindow
)) {
196 TRACE("(IID_IActiveScriptSiteWindow %p)\n", ppv
);
197 *ppv
= &script_site_window
;
200 TRACE("(%s %p)\n", wine_dbgstr_guid(riid
), ppv
);
201 return E_NOINTERFACE
;
204 IUnknown_AddRef((IUnknown
*)*ppv
);
208 static BOOL
load_typelib(void)
213 static const WCHAR wscript_exeW
[] = {'w','s','c','r','i','p','t','.','e','x','e',0};
215 hres
= LoadTypeLib(wscript_exeW
, &typelib
);
219 hres
= ITypeLib_GetTypeInfoOfGuid(typelib
, &IID_IHost
, &host_ti
);
221 hres
= ITypeLib_GetTypeInfoOfGuid(typelib
, &IID_IArguments2
, &arguments_ti
);
223 ITypeLib_Release(typelib
);
224 return SUCCEEDED(hres
);
227 static BOOL
get_engine_clsid(const WCHAR
*ext
, CLSID
*clsid
)
229 WCHAR fileid
[64], progid
[64];
235 static const WCHAR script_engineW
[] =
236 {'\\','S','c','r','i','p','t','E','n','g','i','n','e',0};
238 res
= RegOpenKeyW(HKEY_CLASSES_ROOT
, ext
, &hkey
);
239 if(res
!= ERROR_SUCCESS
)
242 size
= sizeof(fileid
)/sizeof(WCHAR
);
243 res
= RegQueryValueW(hkey
, NULL
, fileid
, &size
);
245 if(res
!= ERROR_SUCCESS
)
248 WINE_TRACE("fileid is %s\n", wine_dbgstr_w(fileid
));
250 strcatW(fileid
, script_engineW
);
251 res
= RegOpenKeyW(HKEY_CLASSES_ROOT
, fileid
, &hkey
);
252 if(res
!= ERROR_SUCCESS
)
255 size
= sizeof(progid
)/sizeof(WCHAR
);
256 res
= RegQueryValueW(hkey
, NULL
, progid
, &size
);
258 if(res
!= ERROR_SUCCESS
)
261 WINE_TRACE("ProgID is %s\n", wine_dbgstr_w(progid
));
263 hres
= CLSIDFromProgID(progid
, clsid
);
264 return SUCCEEDED(hres
);
267 static BOOL
create_engine(CLSID
*clsid
, IActiveScript
**script_ret
,
268 IActiveScriptParse
**parser
)
270 IActiveScript
*script
;
274 hres
= CoCreateInstance(clsid
, NULL
, CLSCTX_INPROC_SERVER
|CLSCTX_INPROC_HANDLER
,
275 &IID_IUnknown
, (void**)&unk
);
279 hres
= IUnknown_QueryInterface(unk
, &IID_IActiveScript
, (void**)&script
);
280 IUnknown_Release(unk
);
284 hres
= IActiveScript_QueryInterface(script
, &IID_IActiveScriptParse
, (void**)parser
);
286 IActiveScript_Release(script
);
290 *script_ret
= script
;
294 static BOOL
init_engine(IActiveScript
*script
, IActiveScriptParse
*parser
)
301 hres
= IActiveScript_SetScriptSite(script
, &script_site
);
305 hres
= IActiveScriptParse_InitNew(parser
);
309 hres
= IActiveScript_AddNamedItem(script
, wscriptW
, SCRIPTITEM_ISVISIBLE
);
313 hres
= IActiveScript_AddNamedItem(script
, wshW
, SCRIPTITEM_ISVISIBLE
);
317 hres
= IActiveScript_SetScriptState(script
, SCRIPTSTATE_INITIALIZED
);
318 return SUCCEEDED(hres
);
321 static BSTR
get_script_str(const WCHAR
*filename
)
323 const char *file_map
;
328 file
= CreateFileW(filename
, GENERIC_READ
, 0, NULL
, OPEN_EXISTING
, FILE_ATTRIBUTE_READONLY
, NULL
);
329 if(file
== INVALID_HANDLE_VALUE
)
332 size
= GetFileSize(file
, NULL
);
333 map
= CreateFileMappingW(file
, NULL
, PAGE_READONLY
, 0, 0, NULL
);
335 if(map
== INVALID_HANDLE_VALUE
)
338 file_map
= MapViewOfFile(map
, FILE_MAP_READ
, 0, 0, 0);
343 len
= MultiByteToWideChar(CP_ACP
, 0, file_map
, size
, NULL
, 0);
344 ret
= SysAllocStringLen(NULL
, len
);
345 MultiByteToWideChar(CP_ACP
, 0, file_map
, size
, ret
, len
);
347 UnmapViewOfFile(file_map
);
351 static void run_script(const WCHAR
*filename
, IActiveScript
*script
, IActiveScriptParse
*parser
)
356 text
= get_script_str(filename
);
358 WINE_FIXME("Could not get script text\n");
362 hres
= IActiveScriptParse_ParseScriptText(parser
, text
, NULL
, NULL
, NULL
, 1, 1,
363 SCRIPTTEXT_HOSTMANAGESSOURCE
|SCRIPTITEM_ISVISIBLE
, NULL
, NULL
);
366 WINE_FIXME("ParseScriptText failed: %08x\n", hres
);
370 hres
= IActiveScript_SetScriptState(script
, SCRIPTSTATE_STARTED
);
372 WINE_FIXME("SetScriptState failed: %08x\n", hres
);
375 static BOOL
set_host_properties(const WCHAR
*prop
)
377 static const WCHAR nologoW
[] = {'n','o','l','o','g','o',0};
378 static const WCHAR iactive
[] = {'i',0};
379 static const WCHAR batch
[] = {'b',0};
389 if(strcmpiW(prop
, iactive
) == 0)
390 wshInteractive
= VARIANT_TRUE
;
391 else if(strcmpiW(prop
, batch
) == 0)
392 wshInteractive
= VARIANT_FALSE
;
393 else if(strcmpiW(prop
, nologoW
) == 0)
394 WINE_FIXME("ignored %s switch\n", debugstr_w(nologoW
));
397 WINE_FIXME("unsupported switch %s\n", debugstr_w(prop
));
403 int WINAPI
wWinMain(HINSTANCE hInst
, HINSTANCE hPrevInst
, LPWSTR cmdline
, int cmdshow
)
405 WCHAR
*ext
, *filepart
, *filename
= NULL
;
406 IActiveScriptParse
*parser
;
407 IActiveScript
*script
;
413 WINE_TRACE("(%p %p %s %x)\n", hInst
, hPrevInst
, wine_dbgstr_w(cmdline
), cmdshow
);
415 argv
= CommandLineToArgvW(cmdline
, &argc
);
419 for(i
=0; i
<argc
; i
++) {
420 if(*argv
[i
] == '/' || *argv
[i
] == '-') {
421 if(!set_host_properties(argv
[i
]))
426 numOfArgs
= argc
-i
-1;
432 WINE_FIXME("No file name specified\n");
435 res
= GetFullPathNameW(filename
, sizeof(scriptFullName
)/sizeof(WCHAR
), scriptFullName
, &filepart
);
436 if(!res
|| res
> sizeof(scriptFullName
)/sizeof(WCHAR
))
439 ext
= strrchrW(filepart
, '.');
440 if(!ext
|| !get_engine_clsid(ext
, &clsid
)) {
441 WINE_FIXME("Could not find engine for %s\n", wine_dbgstr_w(ext
));
447 if(!create_engine(&clsid
, &script
, &parser
)) {
448 WINE_FIXME("Could not create script engine\n");
453 if(init_engine(script
, parser
)) {
454 run_script(filename
, script
, parser
);
455 IActiveScript_Close(script
);
456 ITypeInfo_Release(host_ti
);
458 WINE_FIXME("Script initialization failed\n");
461 IActiveScript_Release(script
);
462 IActiveScriptParse_Release(parser
);