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