+++ /dev/null
-/*
- * 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 "wscript.h"
-
-#include <shellapi.h>
-#include <activscp.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
-
-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(strcmpW(pstrName, wshW) && strcmpW(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 = sizeof(fileid)/sizeof(WCHAR);
- res = RegQueryValueW(hkey, NULL, fileid, &size);
- RegCloseKey(hkey);
- if(res != ERROR_SUCCESS)
- return FALSE;
-
- WINE_TRACE("fileid is %s\n", wine_dbgstr_w(fileid));
-
- strcatW(fileid, script_engineW);
- res = RegOpenKeyW(HKEY_CLASSES_ROOT, fileid, &hkey);
- if(res != ERROR_SUCCESS)
- return FALSE;
-
- size = sizeof(progid)/sizeof(WCHAR);
- 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(strcmpiW(prop, iactive) == 0)
- wshInteractive = VARIANT_TRUE;
- else if(strcmpiW(prop, batch) == 0)
- wshInteractive = VARIANT_FALSE;
- else if(strcmpiW(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, sizeof(scriptFullName)/sizeof(WCHAR), scriptFullName, &filepart);
- if(!res || res > sizeof(scriptFullName)/sizeof(WCHAR))
- return 1;
-
- ext = strrchrW(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;
-}