dlls/mshtml/script.c

   1 /*
   2  * Copyright 2008 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 "config.h"
  20
  21 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "winuser.h"
  28 #include "ole2.h"
  29 #include "activscp.h"
  30
  31 #include "wine/debug.h"
  32
  33 #include "mshtml_private.h"
  34
  35 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  36
  37 static const CLSID CLSID_JScript =
  38     {0xf414c260,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};
  39
  40 typedef struct {
  41     const IActiveScriptSiteVtbl  *lpActiveScriptSiteVtbl;
  42
  43     LONG ref;
  44
  45     IActiveScript *script;
  46
  47     SCRIPTSTATE script_state;
  48
  49     HTMLDocument *doc;
  50
  51     GUID guid;
  52     struct list entry;
  53 } ScriptHost;
  54
  55 #define ACTSCPSITE(x)  ((IActiveScriptSite*)               &(x)->lpActiveScriptSiteVtbl)
  56
  57 #define ACTSCPSITE_THIS(iface) DEFINE_THIS(ScriptHost, ActiveScriptSite, iface)
  58
  59 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
  60 {
  61     ScriptHost *This = ACTSCPSITE_THIS(iface);
  62
  63     *ppv = NULL;
  64
  65     if(IsEqualGUID(&IID_IUnknown, riid)) {
  66         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
  67         *ppv = ACTSCPSITE(This);
  68     }else if(IsEqualGUID(&IID_IActiveScriptSite, riid)) {
  69         TRACE("(%p)->(IID_IActiveScriptSite %p)\n", This, ppv);
  70         *ppv = ACTSCPSITE(This);
  71     }else {
  72         FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
  73         return E_NOINTERFACE;
  74     }
  75
  76     IUnknown_AddRef((IUnknown*)*ppv);
  77     return S_OK;
  78 }
  79
  80 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
  81 {
  82     ScriptHost *This = ACTSCPSITE_THIS(iface);
  83     LONG ref = InterlockedIncrement(&This->ref);
  84
  85     TRACE("(%p) ref=%d\n", This, ref);
  86
  87     return ref;
  88 }
  89
  90 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
  91 {
  92     ScriptHost *This = ACTSCPSITE_THIS(iface);
  93     LONG ref = InterlockedDecrement(&This->ref);
  94
  95     TRACE("(%p) ref=%d\n", This, ref);
  96
  97     if(!ref) {
  98         if(This->doc)
  99             list_remove(&This->entry);
 100         heap_free(This);
 101     }
 102
 103     return ref;
 104 }
 105
 106 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
 107 {
 108     ScriptHost *This = ACTSCPSITE_THIS(iface);
 109     FIXME("(%p)->(%p)\n", This, plcid);
 110     return E_NOTIMPL;
 111 }
 112
 113 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
 114         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
 115 {
 116     ScriptHost *This = ACTSCPSITE_THIS(iface);
 117     FIXME("(%p)->(%s %x %p %p)\n", This, debugstr_w(pstrName), dwReturnMask, ppiunkItem, ppti);
 118     return E_NOTIMPL;
 119 }
 120
 121 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
 122 {
 123     ScriptHost *This = ACTSCPSITE_THIS(iface);
 124     FIXME("(%p)->(%p)\n", This, pbstrVersion);
 125     return E_NOTIMPL;
 126 }
 127
 128 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
 129         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
 130 {
 131     ScriptHost *This = ACTSCPSITE_THIS(iface);
 132     FIXME("(%p)->(%p %p)\n", This, pvarResult, pexcepinfo);
 133     return E_NOTIMPL;
 134 }
 135
 136 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
 137 {
 138     ScriptHost *This = ACTSCPSITE_THIS(iface);
 139
 140     TRACE("(%p)->(%x)\n", This, ssScriptState);
 141
 142     This->script_state = ssScriptState;
 143     return S_OK;
 144 }
 145
 146 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
 147 {
 148     ScriptHost *This = ACTSCPSITE_THIS(iface);
 149     FIXME("(%p)->(%p)\n", This, pscripterror);
 150     return E_NOTIMPL;
 151 }
 152
 153 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 154 {
 155     ScriptHost *This = ACTSCPSITE_THIS(iface);
 156     FIXME("(%p)->()\n", This);
 157     return E_NOTIMPL;
 158 }
 159
 160 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 161 {
 162     ScriptHost *This = ACTSCPSITE_THIS(iface);
 163     FIXME("(%p)->()\n", This);
 164     return E_NOTIMPL;
 165 }
 166
 167 #undef ACTSCPSITE_THIS
 168
 169 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
 170     ActiveScriptSite_QueryInterface,
 171     ActiveScriptSite_AddRef,
 172     ActiveScriptSite_Release,
 173     ActiveScriptSite_GetLCID,
 174     ActiveScriptSite_GetItemInfo,
 175     ActiveScriptSite_GetDocVersionString,
 176     ActiveScriptSite_OnScriptTerminate,
 177     ActiveScriptSite_OnStateChange,
 178     ActiveScriptSite_OnScriptError,
 179     ActiveScriptSite_OnEnterScript,
 180     ActiveScriptSite_OnLeaveScript
 181 };
 182
 183 static ScriptHost *create_script_host(HTMLDocument *doc, GUID *guid)
 184 {
 185     ScriptHost *ret;
 186     HRESULT hres;
 187
 188     ret = heap_alloc_zero(sizeof(*ret));
 189     ret->lpActiveScriptSiteVtbl = &ActiveScriptSiteVtbl;
 190     ret->ref = 1;
 191     ret->doc = doc;
 192
 193     ret->guid = *guid;
 194     list_add_tail(&doc->script_hosts, &ret->entry);
 195
 196     hres = CoCreateInstance(&ret->guid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 197             &IID_IActiveScript, (void**)&ret->script);
 198    if(FAILED(hres))
 199         WARN("Could not load script engine: %08x\n", hres);
 200
 201     return ret;
 202 }
 203
 204 static BOOL get_guid_from_type(LPCWSTR type, GUID *guid)
 205 {
 206     const WCHAR text_javascriptW[] =
 207         {'t','e','x','t','/','j','a','v','a','s','c','r','i','p','t',0};
 208
 209     /* FIXME: Handle more types */
 210     if(!strcmpW(type, text_javascriptW)) {
 211         *guid = CLSID_JScript;
 212     }else {
 213         FIXME("Unknown type %s\n", debugstr_w(type));
 214         return FALSE;
 215     }
 216
 217     return TRUE;
 218 }
 219
 220 static BOOL get_guid_from_language(LPCWSTR type, GUID *guid)
 221 {
 222     HRESULT hres;
 223
 224     hres = CLSIDFromProgID(type, guid);
 225     if(FAILED(hres))
 226         return FALSE;
 227
 228     /* FIXME: Check CATID_ActiveScriptParse */
 229
 230     return TRUE;
 231 }
 232
 233 static BOOL get_script_guid(nsIDOMHTMLScriptElement *nsscript, GUID *guid)
 234 {
 235     nsAString attr_str, val_str;
 236     BOOL ret = FALSE;
 237     nsresult nsres;
 238
 239     static const PRUnichar languageW[] = {'l','a','n','g','u','a','g','e',0};
 240
 241     nsAString_Init(&val_str, NULL);
 242
 243     nsres = nsIDOMHTMLScriptElement_GetType(nsscript, &val_str);
 244     if(NS_SUCCEEDED(nsres)) {
 245         const PRUnichar *type;
 246
 247         nsAString_GetData(&val_str, &type);
 248         if(*type) {
 249             ret = get_guid_from_type(type, guid);
 250             nsAString_Finish(&val_str);
 251             return ret;
 252         }
 253     }else {
 254         ERR("GetType failed: %08x\n", nsres);
 255     }
 256
 257     nsAString_Init(&attr_str, languageW);
 258
 259     nsres = nsIDOMHTMLScriptElement_GetAttribute(nsscript, &attr_str, &val_str);
 260     if(NS_SUCCEEDED(nsres)) {
 261         const PRUnichar *language;
 262
 263         nsAString_GetData(&val_str, &language);
 264
 265         if(*language) {
 266             ret = get_guid_from_language(language, guid);
 267         }else {
 268             *guid = CLSID_JScript;
 269             ret = TRUE;
 270         }
 271     }else {
 272         ERR("GetAttribute(language) failed: %08x\n", nsres);
 273     }
 274
 275     nsAString_Finish(&attr_str);
 276     nsAString_Finish(&val_str);
 277
 278     return ret;
 279 }
 280
 281 static ScriptHost *get_script_host(HTMLDocument *doc, nsIDOMHTMLScriptElement *nsscript)
 282 {
 283     ScriptHost *iter;
 284     GUID guid;
 285
 286     if(!get_script_guid(nsscript, &guid)) {
 287         WARN("Could not find script GUID\n");
 288         return NULL;
 289     }
 290
 291     if(IsEqualGUID(&CLSID_JScript, &guid)) {
 292         FIXME("Ignoring JScript\n");
 293         return NULL;
 294     }
 295
 296     LIST_FOR_EACH_ENTRY(iter, &doc->script_hosts, ScriptHost, entry) {
 297         if(IsEqualGUID(&guid, &iter->guid))
 298             return iter;
 299     }
 300
 301     return create_script_host(doc, &guid);
 302 }
 303
 304 void doc_insert_script(HTMLDocument *doc, nsIDOMHTMLScriptElement *nsscript)
 305 {
 306     get_script_host(doc, nsscript);
 307 }
 308
 309 void release_script_hosts(HTMLDocument *doc)
 310 {
 311     ScriptHost *iter;
 312
 313     while(!list_empty(&doc->script_hosts)) {
 314         iter = LIST_ENTRY(list_head(&doc->script_hosts), ScriptHost, entry);
 315
 316         list_remove(&iter->entry);
 317         iter->doc = NULL;
 318         IActiveScript_Release(ACTSCPSITE(iter));
 319     }
 320 }