mshtml: Fixed crash in tests if Gecko is not available.
[wine] / 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     LIST_FOR_EACH_ENTRY(iter, &doc->script_hosts, ScriptHost, entry) {
308         iter->doc = NULL;
309         IActiveScriptSite_Release(ACTSCPSITE(iter));
310     }
311 }