/* * Copyright 2010 Jacek Caban for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include <stdarg.h> #define COBJMACROS #include <windef.h> #include <winbase.h> #include <winreg.h> #include <ole2.h> #include <shellapi.h> #include <activscp.h> #include <initguid.h> #include "wscript.h" #include <wine/debug.h> #ifdef _WIN64 #define IActiveScriptParse_Release IActiveScriptParse64_Release #define IActiveScriptParse_InitNew IActiveScriptParse64_InitNew #define IActiveScriptParse_ParseScriptText IActiveScriptParse64_ParseScriptText #else #define IActiveScriptParse_Release IActiveScriptParse32_Release #define IActiveScriptParse_InitNew IActiveScriptParse32_InitNew #define IActiveScriptParse_ParseScriptText IActiveScriptParse32_ParseScriptText #endif WINE_DEFAULT_DEBUG_CHANNEL(wscript); static const WCHAR wscriptW[] = {'W','S','c','r','i','p','t',0}; static const WCHAR wshW[] = {'W','S','H',0}; WCHAR scriptFullName[MAX_PATH]; ITypeInfo *host_ti; ITypeInfo *arguments_ti; static HRESULT query_interface(REFIID,void**); static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv) { return query_interface(riid, ppv); } static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface) { return 2; } static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface) { return 1; } static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid) { WINE_TRACE("()\n"); *plcid = GetUserDefaultLCID(); return S_OK; } static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName, DWORD dwReturnMask, IUnknown **ppunkItem, ITypeInfo **ppti) { WINE_TRACE("(%s %x %p %p)\n", wine_dbgstr_w(pstrName), dwReturnMask, ppunkItem, ppti); if(lstrcmpW(pstrName, wshW) && lstrcmpW(pstrName, wscriptW)) return E_FAIL; if(dwReturnMask & SCRIPTINFO_ITYPEINFO) { ITypeInfo_AddRef(host_ti); *ppti = host_ti; } if(dwReturnMask & SCRIPTINFO_IUNKNOWN) { IHost_AddRef(&host_obj); *ppunkItem = (IUnknown*)&host_obj; } return S_OK; } static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion) { WINE_FIXME("()\n"); return E_NOTIMPL; } static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface, const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo) { WINE_FIXME("()\n"); return E_NOTIMPL; } static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState) { WINE_TRACE("(%x)\n", ssScriptState); return S_OK; } static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror) { WINE_FIXME("()\n"); return E_NOTIMPL; } static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface) { WINE_TRACE("()\n"); return S_OK; } static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface) { WINE_TRACE("()\n"); return S_OK; } static IActiveScriptSiteVtbl ActiveScriptSiteVtbl = { ActiveScriptSite_QueryInterface, ActiveScriptSite_AddRef, ActiveScriptSite_Release, ActiveScriptSite_GetLCID, ActiveScriptSite_GetItemInfo, ActiveScriptSite_GetDocVersionString, ActiveScriptSite_OnScriptTerminate, ActiveScriptSite_OnStateChange, ActiveScriptSite_OnScriptError, ActiveScriptSite_OnEnterScript, ActiveScriptSite_OnLeaveScript }; static IActiveScriptSite script_site = { &ActiveScriptSiteVtbl }; static HRESULT WINAPI ActiveScriptSiteWindow_QueryInterface(IActiveScriptSiteWindow *iface, REFIID riid, void **ppv) { return query_interface(riid, ppv); } static ULONG WINAPI ActiveScriptSiteWindow_AddRef(IActiveScriptSiteWindow *iface) { return 2; } static ULONG WINAPI ActiveScriptSiteWindow_Release(IActiveScriptSiteWindow *iface) { return 1; } static HRESULT WINAPI ActiveScriptSiteWindow_GetWindow(IActiveScriptSiteWindow *iface, HWND *phwnd) { TRACE("(%p)\n", phwnd); *phwnd = NULL; return S_OK; } static HRESULT WINAPI ActiveScriptSiteWindow_EnableModeless(IActiveScriptSiteWindow *iface, BOOL fEnable) { TRACE("(%x)\n", fEnable); return S_OK; } static const IActiveScriptSiteWindowVtbl ActiveScriptSiteWindowVtbl = { ActiveScriptSiteWindow_QueryInterface, ActiveScriptSiteWindow_AddRef, ActiveScriptSiteWindow_Release, ActiveScriptSiteWindow_GetWindow, ActiveScriptSiteWindow_EnableModeless }; static IActiveScriptSiteWindow script_site_window = { &ActiveScriptSiteWindowVtbl }; static HRESULT query_interface(REFIID riid, void **ppv) { if(IsEqualGUID(riid, &IID_IUnknown)) { TRACE("(IID_IUnknown %p)\n", ppv); *ppv = &script_site; }else if(IsEqualGUID(riid, &IID_IActiveScriptSite)) { TRACE("(IID_IActiveScriptSite %p)\n", ppv); *ppv = &script_site; }else if(IsEqualGUID(riid, &IID_IActiveScriptSiteWindow)) { TRACE("(IID_IActiveScriptSiteWindow %p)\n", ppv); *ppv = &script_site_window; }else { *ppv = NULL; TRACE("(%s %p)\n", wine_dbgstr_guid(riid), ppv); return E_NOINTERFACE; } IUnknown_AddRef((IUnknown*)*ppv); return S_OK; } static BOOL load_typelib(void) { ITypeLib *typelib; HRESULT hres; static const WCHAR wscript_exeW[] = {'w','s','c','r','i','p','t','.','e','x','e',0}; hres = LoadTypeLib(wscript_exeW, &typelib); if(FAILED(hres)) return FALSE; hres = ITypeLib_GetTypeInfoOfGuid(typelib, &IID_IHost, &host_ti); if(SUCCEEDED(hres)) hres = ITypeLib_GetTypeInfoOfGuid(typelib, &IID_IArguments2, &arguments_ti); ITypeLib_Release(typelib); return SUCCEEDED(hres); } static BOOL get_engine_clsid(const WCHAR *ext, CLSID *clsid) { WCHAR fileid[64], progid[64]; DWORD res; LONG size; HKEY hkey; HRESULT hres; static const WCHAR script_engineW[] = {'\\','S','c','r','i','p','t','E','n','g','i','n','e',0}; res = RegOpenKeyW(HKEY_CLASSES_ROOT, ext, &hkey); if(res != ERROR_SUCCESS) return FALSE; size = ARRAY_SIZE(fileid); res = RegQueryValueW(hkey, NULL, fileid, &size); RegCloseKey(hkey); if(res != ERROR_SUCCESS) return FALSE; WINE_TRACE("fileid is %s\n", wine_dbgstr_w(fileid)); lstrcatW(fileid, script_engineW); res = RegOpenKeyW(HKEY_CLASSES_ROOT, fileid, &hkey); if(res != ERROR_SUCCESS) return FALSE; size = ARRAY_SIZE(progid); res = RegQueryValueW(hkey, NULL, progid, &size); RegCloseKey(hkey); if(res != ERROR_SUCCESS) return FALSE; WINE_TRACE("ProgID is %s\n", wine_dbgstr_w(progid)); hres = CLSIDFromProgID(progid, clsid); return SUCCEEDED(hres); } static BOOL create_engine(CLSID *clsid, IActiveScript **script_ret, IActiveScriptParse **parser) { IActiveScript *script; IUnknown *unk; HRESULT hres; hres = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER, &IID_IUnknown, (void**)&unk); if(FAILED(hres)) return FALSE; hres = IUnknown_QueryInterface(unk, &IID_IActiveScript, (void**)&script); IUnknown_Release(unk); if(FAILED(hres)) return FALSE; hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)parser); if(FAILED(hres)) { IActiveScript_Release(script); return FALSE; } *script_ret = script; return TRUE; } static BOOL init_engine(IActiveScript *script, IActiveScriptParse *parser) { HRESULT hres; if(!load_typelib()) return FALSE; hres = IActiveScript_SetScriptSite(script, &script_site); if(FAILED(hres)) return FALSE; hres = IActiveScriptParse_InitNew(parser); if(FAILED(hres)) return FALSE; hres = IActiveScript_AddNamedItem(script, wscriptW, SCRIPTITEM_ISVISIBLE); if(FAILED(hres)) return FALSE; hres = IActiveScript_AddNamedItem(script, wshW, SCRIPTITEM_ISVISIBLE); if(FAILED(hres)) return FALSE; hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_INITIALIZED); return SUCCEEDED(hres); } static BSTR get_script_str(const WCHAR *filename) { const char *file_map; HANDLE file, map; DWORD size, len; BSTR ret; file = CreateFileW(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL); if(file == INVALID_HANDLE_VALUE) return NULL; size = GetFileSize(file, NULL); map = CreateFileMappingW(file, NULL, PAGE_READONLY, 0, 0, NULL); CloseHandle(file); if(map == INVALID_HANDLE_VALUE) return NULL; file_map = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0); CloseHandle(map); if(!file_map) return NULL; len = MultiByteToWideChar(CP_ACP, 0, file_map, size, NULL, 0); ret = SysAllocStringLen(NULL, len); MultiByteToWideChar(CP_ACP, 0, file_map, size, ret, len); UnmapViewOfFile(file_map); return ret; } static void run_script(const WCHAR *filename, IActiveScript *script, IActiveScriptParse *parser) { BSTR text; HRESULT hres; text = get_script_str(filename); if(!text) { WINE_FIXME("Could not get script text\n"); return; } hres = IActiveScriptParse_ParseScriptText(parser, text, NULL, NULL, NULL, 1, 1, SCRIPTTEXT_HOSTMANAGESSOURCE|SCRIPTITEM_ISVISIBLE, NULL, NULL); SysFreeString(text); if(FAILED(hres)) { WINE_FIXME("ParseScriptText failed: %08x\n", hres); return; } hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_STARTED); if(FAILED(hres)) WINE_FIXME("SetScriptState failed: %08x\n", hres); } static BOOL set_host_properties(const WCHAR *prop) { static const WCHAR nologoW[] = {'n','o','l','o','g','o',0}; static const WCHAR iactive[] = {'i',0}; static const WCHAR batch[] = {'b',0}; if(*prop == '/') { ++prop; if(*prop == '/') ++prop; } else ++prop; if(wcsicmp(prop, iactive) == 0) wshInteractive = VARIANT_TRUE; else if(wcsicmp(prop, batch) == 0) wshInteractive = VARIANT_FALSE; else if(wcsicmp(prop, nologoW) == 0) WINE_FIXME("ignored %s switch\n", debugstr_w(nologoW)); else { WINE_FIXME("unsupported switch %s\n", debugstr_w(prop)); return FALSE; } return TRUE; } int WINAPI wWinMain(HINSTANCE hInst, HINSTANCE hPrevInst, LPWSTR cmdline, int cmdshow) { WCHAR *ext, *filepart, *filename = NULL; IActiveScriptParse *parser; IActiveScript *script; WCHAR **argv; CLSID clsid; int argc, i; DWORD res; WINE_TRACE("(%p %p %s %x)\n", hInst, hPrevInst, wine_dbgstr_w(cmdline), cmdshow); argv = CommandLineToArgvW(GetCommandLineW(), &argc); if(!argv) return 1; for(i=1; i<argc; i++) { if(*argv[i] == '/' || *argv[i] == '-') { if(!set_host_properties(argv[i])) return 1; }else { filename = argv[i]; argums = argv+i+1; numOfArgs = argc-i-1; break; } } if(!filename) { WINE_FIXME("No file name specified\n"); return 1; } res = GetFullPathNameW(filename, ARRAY_SIZE(scriptFullName), scriptFullName, &filepart); if(!res || res > ARRAY_SIZE(scriptFullName)) return 1; ext = wcsrchr(filepart, '.'); if(!ext || !get_engine_clsid(ext, &clsid)) { WINE_FIXME("Could not find engine for %s\n", wine_dbgstr_w(ext)); return 1; } CoInitialize(NULL); if(!create_engine(&clsid, &script, &parser)) { WINE_FIXME("Could not create script engine\n"); CoUninitialize(); return 1; } if(init_engine(script, parser)) { run_script(filename, script, parser); IActiveScript_Close(script); ITypeInfo_Release(host_ti); }else { WINE_FIXME("Script initialization failed\n"); } IActiveScript_Release(script); IActiveScriptParse_Release(parser); CoUninitialize(); return 0; }