programs/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
  23 #include <windef.h>
  24 #include <winbase.h>
  25 #include <winreg.h>
  26 #include <ole2.h>
  27 #include <activscp.h>
  28
  29 #include <wine/debug.h>
  30 #include <wine/unicode.h>
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(wscript);
  33
  34 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface,
  35                                                       REFIID riid, void **ppv)
  36 {
  37     if(IsEqualGUID(riid, &IID_IUnknown)) {
  38         WINE_TRACE("(IID_IUnknown %p)\n", ppv);
  39         *ppv = iface;
  40     }else if(IsEqualGUID(riid, &IID_IActiveScriptSite)) {
  41         WINE_TRACE("(IID_IActiveScriptSite %p)\n", ppv);
  42         *ppv = iface;
  43     }else {
  44         *ppv = NULL;
  45         WINE_TRACE("(%s %p)\n", wine_dbgstr_guid(riid), ppv);
  46         return E_NOINTERFACE;
  47     }
  48
  49     IUnknown_AddRef((IUnknown*)*ppv);
  50     return S_OK;
  51 }
  52
  53 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
  54 {
  55     return 2;
  56 }
  57
  58 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
  59 {
  60     return 1;
  61 }
  62
  63 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
  64 {
  65     WINE_TRACE("()\n");
  66
  67     *plcid = GetUserDefaultLCID();
  68     return S_OK;
  69 }
  70
  71 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface,
  72         LPCOLESTR pstrName, DWORD dwReturnMask, IUnknown **ppunkItem, ITypeInfo **ppti)
  73 {
  74     WINE_FIXME("(%s %x %p %p)\n", wine_dbgstr_w(pstrName), dwReturnMask, ppunkItem, ppti);
  75     return E_NOTIMPL;
  76 }
  77
  78 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface,
  79         BSTR *pbstrVersion)
  80 {
  81     WINE_FIXME("()\n");
  82     return E_NOTIMPL;
  83 }
  84
  85 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
  86         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
  87 {
  88     WINE_FIXME("()\n");
  89     return E_NOTIMPL;
  90 }
  91
  92 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface,
  93         SCRIPTSTATE ssScriptState)
  94 {
  95     WINE_TRACE("(%x)\n", ssScriptState);
  96     return S_OK;
  97 }
  98
  99 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface,
 100         IActiveScriptError *pscripterror)
 101 {
 102     WINE_FIXME("()\n");
 103     return E_NOTIMPL;
 104 }
 105
 106 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 107 {
 108     WINE_TRACE("()\n");
 109     return S_OK;
 110 }
 111
 112 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 113 {
 114     WINE_TRACE("()\n");
 115     return S_OK;
 116 }
 117
 118 static IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
 119     ActiveScriptSite_QueryInterface,
 120     ActiveScriptSite_AddRef,
 121     ActiveScriptSite_Release,
 122     ActiveScriptSite_GetLCID,
 123     ActiveScriptSite_GetItemInfo,
 124     ActiveScriptSite_GetDocVersionString,
 125     ActiveScriptSite_OnScriptTerminate,
 126     ActiveScriptSite_OnStateChange,
 127     ActiveScriptSite_OnScriptError,
 128     ActiveScriptSite_OnEnterScript,
 129     ActiveScriptSite_OnLeaveScript
 130 };
 131
 132 IActiveScriptSite script_site = { &ActiveScriptSiteVtbl };
 133
 134 static BOOL get_engine_clsid(const WCHAR *ext, CLSID *clsid)
 135 {
 136     WCHAR fileid[64], progid[64];
 137     DWORD res;
 138     LONG size;
 139     HKEY hkey;
 140     HRESULT hres;
 141
 142     static const WCHAR script_engineW[] =
 143         {'\\','S','c','r','i','p','t','E','n','g','i','n','e',0};
 144
 145     res = RegOpenKeyW(HKEY_CLASSES_ROOT, ext, &hkey);
 146     if(res != ERROR_SUCCESS)
 147         return FALSE;
 148
 149     size = sizeof(fileid)/sizeof(WCHAR);
 150     res = RegQueryValueW(hkey, NULL, fileid, &size);
 151     RegCloseKey(hkey);
 152     if(res != ERROR_SUCCESS)
 153         return FALSE;
 154
 155     WINE_TRACE("fileid is %s\n", wine_dbgstr_w(fileid));
 156
 157     strcatW(fileid, script_engineW);
 158     res = RegOpenKeyW(HKEY_CLASSES_ROOT, fileid, &hkey);
 159     if(res != ERROR_SUCCESS)
 160         return FALSE;
 161
 162     size = sizeof(progid)/sizeof(WCHAR);
 163     res = RegQueryValueW(hkey, NULL, progid, &size);
 164     RegCloseKey(hkey);
 165     if(res != ERROR_SUCCESS)
 166         return FALSE;
 167
 168     WINE_TRACE("ProgID is %s\n", wine_dbgstr_w(progid));
 169
 170     hres = CLSIDFromProgID(progid, clsid);
 171     return SUCCEEDED(hres);
 172 }
 173
 174 static HRESULT create_engine(CLSID *clsid, IActiveScript **script_ret,
 175         IActiveScriptParse **parser)
 176 {
 177     IActiveScript *script;
 178     IUnknown *unk;
 179     HRESULT hres;
 180
 181     hres = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 182             &IID_IUnknown, (void**)&unk);
 183     if(FAILED(hres))
 184         return FALSE;
 185
 186     hres = IUnknown_QueryInterface(unk, &IID_IActiveScript, (void**)&script);
 187     IUnknown_Release(unk);
 188     if(FAILED(hres))
 189         return FALSE;
 190
 191     hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)parser);
 192     if(FAILED(hres)) {
 193         IActiveScript_Release(script);
 194         return FALSE;
 195     }
 196
 197     *script_ret = script;
 198     return TRUE;
 199 }
 200
 201 static HRESULT init_engine(IActiveScript *script, IActiveScriptParse *parser)
 202 {
 203     HRESULT hres;
 204
 205     static const WCHAR wscriptW[] = {'W','S','c','r','i','p','t',0};
 206     static const WCHAR wshW[] = {'W','S','H',0};
 207
 208     hres = IActiveScript_SetScriptSite(script, &script_site);
 209     if(FAILED(hres))
 210         return FALSE;
 211
 212     hres = IActiveScriptParse64_InitNew(parser);
 213     if(FAILED(hres))
 214         return FALSE;
 215
 216     hres = IActiveScript_AddNamedItem(script, wscriptW, SCRIPTITEM_ISVISIBLE);
 217     if(FAILED(hres))
 218         return FALSE;
 219
 220     hres = IActiveScript_AddNamedItem(script, wshW, SCRIPTITEM_ISVISIBLE);
 221     if(FAILED(hres))
 222         return FALSE;
 223
 224     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_INITIALIZED);
 225     return SUCCEEDED(hres);
 226 }
 227
 228 static BSTR get_script_str(const WCHAR *filename)
 229 {
 230     const char *file_map;
 231     HANDLE file, map;
 232     DWORD size, len;
 233     BSTR ret;
 234
 235     file = CreateFileW(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
 236     if(file == INVALID_HANDLE_VALUE)
 237         return NULL;
 238
 239     size = GetFileSize(file, NULL);
 240     map = CreateFileMappingW(file, NULL, PAGE_READONLY, 0, 0, NULL);
 241     CloseHandle(file);
 242     if(map == INVALID_HANDLE_VALUE)
 243         return NULL;
 244
 245     file_map = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0);
 246     CloseHandle(map);
 247     if(!file_map)
 248         return NULL;
 249
 250     len = MultiByteToWideChar(CP_ACP, 0, file_map, size, NULL, 0);
 251     ret = SysAllocStringLen(NULL, len);
 252     MultiByteToWideChar(CP_ACP, 0, file_map, size, ret, len);
 253
 254     UnmapViewOfFile(file_map);
 255     return ret;
 256 }
 257
 258 static void run_script(const WCHAR *filename, IActiveScript *script, IActiveScriptParse *parser)
 259 {
 260     BSTR text;
 261     HRESULT hres;
 262
 263     text = get_script_str(filename);
 264     if(!text) {
 265         WINE_FIXME("Could not get script text\n");
 266         return;
 267     }
 268
 269     hres = IActiveScriptParse64_ParseScriptText(parser, text, NULL, NULL, NULL, 1, 1,
 270             SCRIPTTEXT_HOSTMANAGESSOURCE|SCRIPTITEM_ISVISIBLE, NULL, NULL);
 271     SysFreeString(text);
 272     if(FAILED(hres)) {
 273         WINE_FIXME("ParseScriptText failed: %08x\n", hres);
 274         return;
 275     }
 276
 277     hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_STARTED);
 278     if(FAILED(hres))
 279         WINE_FIXME("SetScriptState failed: %08x\n", hres);
 280 }
 281
 282 int WINAPI wWinMain(HINSTANCE hInst, HINSTANCE hPrevInst, LPWSTR cmdline, int cmdshow)
 283 {
 284     IActiveScriptParse *parser;
 285     IActiveScript *script;
 286     const WCHAR *ext;
 287     CLSID clsid;
 288
 289     WINE_TRACE("(%p %p %s %x)\n", hInst, hPrevInst, wine_dbgstr_w(cmdline), cmdshow);
 290
 291     if(!*cmdline)
 292         return 1;
 293
 294     ext = strchrW(cmdline, '.');
 295     if(!ext)
 296         ext = cmdline;
 297     if(!get_engine_clsid(ext, &clsid)) {
 298         WINE_FIXME("Could not fine engine for %s\n", wine_dbgstr_w(ext));
 299         return 1;
 300     }
 301
 302     CoInitialize(NULL);
 303
 304     if(!create_engine(&clsid, &script, &parser)) {
 305         WINE_FIXME("Could not create script engine\n");
 306         CoUninitialize();
 307         return 1;
 308     }
 309
 310     if(init_engine(script, parser)) {
 311         run_script(cmdline, script, parser);
 312         IActiveScript_Close(script);
 313     }else {
 314         WINE_FIXME("Script initialization failed\n");
 315     }
 316
 317     IActiveScript_Release(script);
 318     IUnknown_Release(parser);
 319
 320     CoUninitialize();
 321
 322     return 0;
 323 }