dlls/mshtml/htmlscript.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 <stdarg.h>
  20 #include <assert.h>
  21
  22 #define COBJMACROS
  23
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "winuser.h"
  27 #include "ole2.h"
  28
  29 #include "wine/debug.h"
  30
  31 #include "mshtml_private.h"
  32 #include "htmlscript.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  35
  36 static inline HTMLScriptElement *impl_from_IHTMLScriptElement(IHTMLScriptElement *iface)
  37 {
  38     return CONTAINING_RECORD(iface, HTMLScriptElement, IHTMLScriptElement_iface);
  39 }
  40
  41 static HRESULT WINAPI HTMLScriptElement_QueryInterface(IHTMLScriptElement *iface,
  42         REFIID riid, void **ppv)
  43 {
  44     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
  45
  46     return IHTMLDOMNode_QueryInterface(&This->element.node.IHTMLDOMNode_iface, riid, ppv);
  47 }
  48
  49 static ULONG WINAPI HTMLScriptElement_AddRef(IHTMLScriptElement *iface)
  50 {
  51     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
  52
  53     return IHTMLDOMNode_AddRef(&This->element.node.IHTMLDOMNode_iface);
  54 }
  55
  56 static ULONG WINAPI HTMLScriptElement_Release(IHTMLScriptElement *iface)
  57 {
  58     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
  59
  60     return IHTMLDOMNode_Release(&This->element.node.IHTMLDOMNode_iface);
  61 }
  62
  63 static HRESULT WINAPI HTMLScriptElement_GetTypeInfoCount(IHTMLScriptElement *iface, UINT *pctinfo)
  64 {
  65     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
  66     return IDispatchEx_GetTypeInfoCount(&This->element.node.dispex.IDispatchEx_iface, pctinfo);
  67 }
  68
  69 static HRESULT WINAPI HTMLScriptElement_GetTypeInfo(IHTMLScriptElement *iface, UINT iTInfo,
  70                                               LCID lcid, ITypeInfo **ppTInfo)
  71 {
  72     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
  73     return IDispatchEx_GetTypeInfo(&This->element.node.dispex.IDispatchEx_iface, iTInfo, lcid,
  74             ppTInfo);
  75 }
  76
  77 static HRESULT WINAPI HTMLScriptElement_GetIDsOfNames(IHTMLScriptElement *iface, REFIID riid,
  78                                                 LPOLESTR *rgszNames, UINT cNames,
  79                                                 LCID lcid, DISPID *rgDispId)
  80 {
  81     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
  82     return IDispatchEx_GetIDsOfNames(&This->element.node.dispex.IDispatchEx_iface, riid, rgszNames,
  83             cNames, lcid, rgDispId);
  84 }
  85
  86 static HRESULT WINAPI HTMLScriptElement_Invoke(IHTMLScriptElement *iface, DISPID dispIdMember,
  87                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
  88                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
  89 {
  90     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
  91     return IDispatchEx_Invoke(&This->element.node.dispex.IDispatchEx_iface, dispIdMember, riid,
  92             lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
  93 }
  94
  95 static HRESULT WINAPI HTMLScriptElement_put_src(IHTMLScriptElement *iface, BSTR v)
  96 {
  97     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
  98     HTMLInnerWindow *window;
  99     nsIDOMNode *parent;
 100     nsAString src_str;
 101     nsresult nsres;
 102
 103     TRACE("(%p)->(%s)\n", This, debugstr_w(v));
 104
 105     if(!This->element.node.doc || !This->element.node.doc->window) {
 106         WARN("no windoow\n");
 107         return E_UNEXPECTED;
 108     }
 109
 110     window = This->element.node.doc->window;
 111
 112     nsAString_InitDepend(&src_str, v);
 113     nsres = nsIDOMHTMLScriptElement_SetSrc(This->nsscript, &src_str);
 114     nsAString_Finish(&src_str);
 115     if(NS_FAILED(nsres)) {
 116         ERR("SetSrc failed: %08x\n", nsres);
 117         return E_FAIL;
 118     }
 119
 120     if(This->parsed) {
 121         WARN("already parsed\n");
 122         return S_OK;
 123     }
 124
 125     if(window->parser_callback_cnt) {
 126         script_queue_entry_t *queue;
 127
 128         queue = heap_alloc(sizeof(*queue));
 129         if(!queue)
 130             return E_OUTOFMEMORY;
 131
 132         IHTMLScriptElement_AddRef(&This->IHTMLScriptElement_iface);
 133         queue->script = This;
 134
 135         list_add_tail(&window->script_queue, &queue->entry);
 136         return S_OK;
 137     }
 138
 139     nsres = nsIDOMHTMLScriptElement_GetParentNode(This->nsscript, &parent);
 140     if(NS_FAILED(nsres) || !parent) {
 141         TRACE("No parent, not executing\n");
 142         This->parse_on_bind = TRUE;
 143         return S_OK;
 144     }
 145
 146     nsIDOMNode_Release(parent);
 147     doc_insert_script(window, This);
 148     return S_OK;
 149 }
 150
 151 static HRESULT WINAPI HTMLScriptElement_get_src(IHTMLScriptElement *iface, BSTR *p)
 152 {
 153     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 154     nsAString src_str;
 155     nsresult nsres;
 156
 157     TRACE("(%p)->(%p)\n", This, p);
 158
 159     nsAString_Init(&src_str, NULL);
 160     nsres = nsIDOMHTMLScriptElement_GetSrc(This->nsscript, &src_str);
 161     return return_nsstr(nsres, &src_str, p);
 162 }
 163
 164 static HRESULT WINAPI HTMLScriptElement_put_htmlFor(IHTMLScriptElement *iface, BSTR v)
 165 {
 166     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 167     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
 168     return E_NOTIMPL;
 169 }
 170
 171 static HRESULT WINAPI HTMLScriptElement_get_htmlFor(IHTMLScriptElement *iface, BSTR *p)
 172 {
 173     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 174     FIXME("(%p)->(%p)\n", This, p);
 175     return E_NOTIMPL;
 176 }
 177
 178 static HRESULT WINAPI HTMLScriptElement_put_event(IHTMLScriptElement *iface, BSTR v)
 179 {
 180     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 181     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
 182     return E_NOTIMPL;
 183 }
 184
 185 static HRESULT WINAPI HTMLScriptElement_get_event(IHTMLScriptElement *iface, BSTR *p)
 186 {
 187     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 188     FIXME("(%p)->(%p)\n", This, p);
 189     return E_NOTIMPL;
 190 }
 191
 192 static HRESULT WINAPI HTMLScriptElement_put_text(IHTMLScriptElement *iface, BSTR v)
 193 {
 194     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 195     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
 196     return E_NOTIMPL;
 197 }
 198
 199 static HRESULT WINAPI HTMLScriptElement_get_text(IHTMLScriptElement *iface, BSTR *p)
 200 {
 201     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 202     FIXME("(%p)->(%p)\n", This, p);
 203     return E_NOTIMPL;
 204 }
 205
 206 static HRESULT WINAPI HTMLScriptElement_put_defer(IHTMLScriptElement *iface, VARIANT_BOOL v)
 207 {
 208     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 209     HRESULT hr = S_OK;
 210     nsresult nsres;
 211
 212     TRACE("(%p)->(%x)\n", This, v);
 213
 214     nsres = nsIDOMHTMLScriptElement_SetDefer(This->nsscript, v != VARIANT_FALSE);
 215     if(NS_FAILED(nsres))
 216     {
 217         hr = E_FAIL;
 218     }
 219
 220     return hr;
 221 }
 222
 223 static HRESULT WINAPI HTMLScriptElement_get_defer(IHTMLScriptElement *iface, VARIANT_BOOL *p)
 224 {
 225     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 226     cpp_bool defer = FALSE;
 227     nsresult nsres;
 228
 229     TRACE("(%p)->(%p)\n", This, p);
 230
 231     if(!p)
 232         return E_INVALIDARG;
 233
 234     nsres = nsIDOMHTMLScriptElement_GetDefer(This->nsscript, &defer);
 235     if(NS_FAILED(nsres)) {
 236         ERR("GetSrc failed: %08x\n", nsres);
 237     }
 238
 239     *p = defer ? VARIANT_TRUE : VARIANT_FALSE;
 240
 241     TRACE("*p = %d\n", *p);
 242     return S_OK;
 243 }
 244
 245 static HRESULT WINAPI HTMLScriptElement_get_readyState(IHTMLScriptElement *iface, BSTR *p)
 246 {
 247     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 248     FIXME("(%p)->(%p)\n", This, p);
 249     return E_NOTIMPL;
 250 }
 251
 252 static HRESULT WINAPI HTMLScriptElement_put_onerror(IHTMLScriptElement *iface, VARIANT v)
 253 {
 254     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 255     FIXME("(%p)->(%s)\n", This, debugstr_variant(&v));
 256     return E_NOTIMPL;
 257 }
 258
 259 static HRESULT WINAPI HTMLScriptElement_get_onerror(IHTMLScriptElement *iface, VARIANT *p)
 260 {
 261     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 262     FIXME("(%p)->(%p)\n", This, p);
 263     return E_NOTIMPL;
 264 }
 265
 266 static HRESULT WINAPI HTMLScriptElement_put_type(IHTMLScriptElement *iface, BSTR v)
 267 {
 268     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 269     nsAString nstype_str;
 270     nsresult nsres;
 271
 272     TRACE("(%p)->(%s)\n", This, debugstr_w(v));
 273
 274     nsAString_Init(&nstype_str, v);
 275     nsres = nsIDOMHTMLScriptElement_SetType(This->nsscript, &nstype_str);
 276     if (NS_FAILED(nsres))
 277         ERR("SetType failed: %08x\n", nsres);
 278     nsAString_Finish (&nstype_str);
 279
 280     return S_OK;
 281 }
 282
 283 static HRESULT WINAPI HTMLScriptElement_get_type(IHTMLScriptElement *iface, BSTR *p)
 284 {
 285     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
 286     nsAString nstype_str;
 287     nsresult nsres;
 288
 289     TRACE("(%p)->(%p)\n", This, p);
 290
 291     nsAString_Init(&nstype_str, NULL);
 292     nsres = nsIDOMHTMLScriptElement_GetType(This->nsscript, &nstype_str);
 293     return return_nsstr(nsres, &nstype_str, p);
 294 }
 295
 296 static const IHTMLScriptElementVtbl HTMLScriptElementVtbl = {
 297     HTMLScriptElement_QueryInterface,
 298     HTMLScriptElement_AddRef,
 299     HTMLScriptElement_Release,
 300     HTMLScriptElement_GetTypeInfoCount,
 301     HTMLScriptElement_GetTypeInfo,
 302     HTMLScriptElement_GetIDsOfNames,
 303     HTMLScriptElement_Invoke,
 304     HTMLScriptElement_put_src,
 305     HTMLScriptElement_get_src,
 306     HTMLScriptElement_put_htmlFor,
 307     HTMLScriptElement_get_htmlFor,
 308     HTMLScriptElement_put_event,
 309     HTMLScriptElement_get_event,
 310     HTMLScriptElement_put_text,
 311     HTMLScriptElement_get_text,
 312     HTMLScriptElement_put_defer,
 313     HTMLScriptElement_get_defer,
 314     HTMLScriptElement_get_readyState,
 315     HTMLScriptElement_put_onerror,
 316     HTMLScriptElement_get_onerror,
 317     HTMLScriptElement_put_type,
 318     HTMLScriptElement_get_type
 319 };
 320
 321 static inline HTMLScriptElement *impl_from_HTMLDOMNode(HTMLDOMNode *iface)
 322 {
 323     return CONTAINING_RECORD(iface, HTMLScriptElement, element.node);
 324 }
 325
 326 static HRESULT HTMLScriptElement_QI(HTMLDOMNode *iface, REFIID riid, void **ppv)
 327 {
 328     HTMLScriptElement *This = impl_from_HTMLDOMNode(iface);
 329
 330     *ppv = NULL;
 331
 332     if(IsEqualGUID(&IID_IUnknown, riid)) {
 333         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
 334         *ppv = &This->IHTMLScriptElement_iface;
 335     }else if(IsEqualGUID(&IID_IDispatch, riid)) {
 336         TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
 337         *ppv = &This->IHTMLScriptElement_iface;
 338     }else if(IsEqualGUID(&IID_IHTMLScriptElement, riid)) {
 339         TRACE("(%p)->(IID_IHTMLScriptElement %p)\n", This, ppv);
 340         *ppv = &This->IHTMLScriptElement_iface;
 341     }
 342
 343     if(*ppv) {
 344         IUnknown_AddRef((IUnknown*)*ppv);
 345         return S_OK;
 346     }
 347
 348     return HTMLElement_QI(&This->element.node, riid, ppv);
 349 }
 350
 351 static HRESULT HTMLScriptElement_get_readystate(HTMLDOMNode *iface, BSTR *p)
 352 {
 353     HTMLScriptElement *This = impl_from_HTMLDOMNode(iface);
 354
 355     return IHTMLScriptElement_get_readyState(&This->IHTMLScriptElement_iface, p);
 356 }
 357
 358 static const NodeImplVtbl HTMLScriptElementImplVtbl = {
 359     HTMLScriptElement_QI,
 360     HTMLElement_destructor,
 361     HTMLElement_clone,
 362     HTMLElement_handle_event,
 363     HTMLElement_get_attr_col,
 364     NULL,
 365     NULL,
 366     NULL,
 367     NULL,
 368     NULL,
 369     HTMLScriptElement_get_readystate
 370 };
 371
 372 HRESULT script_elem_from_nsscript(HTMLDocumentNode *doc, nsIDOMHTMLScriptElement *nsscript, HTMLScriptElement **ret)
 373 {
 374     HTMLDOMNode *node;
 375     HRESULT hres;
 376
 377     hres = get_node(doc, (nsIDOMNode*)nsscript, TRUE, &node);
 378     if(FAILED(hres))
 379         return hres;
 380
 381     assert(node->vtbl == &HTMLScriptElementImplVtbl);
 382     *ret = impl_from_HTMLDOMNode(node);
 383     return S_OK;
 384 }
 385
 386 static const tid_t HTMLScriptElement_iface_tids[] = {
 387     HTMLELEMENT_TIDS,
 388     IHTMLScriptElement_tid,
 389     0
 390 };
 391
 392 static dispex_static_data_t HTMLScriptElement_dispex = {
 393     NULL,
 394     DispHTMLScriptElement_tid,
 395     NULL,
 396     HTMLScriptElement_iface_tids
 397 };
 398
 399 HRESULT HTMLScriptElement_Create(HTMLDocumentNode *doc, nsIDOMHTMLElement *nselem, HTMLElement **elem)
 400 {
 401     HTMLScriptElement *ret;
 402     nsresult nsres;
 403
 404     ret = heap_alloc_zero(sizeof(HTMLScriptElement));
 405     if(!ret)
 406         return E_OUTOFMEMORY;
 407
 408     ret->IHTMLScriptElement_iface.lpVtbl = &HTMLScriptElementVtbl;
 409     ret->element.node.vtbl = &HTMLScriptElementImplVtbl;
 410
 411     HTMLElement_Init(&ret->element, doc, nselem, &HTMLScriptElement_dispex);
 412
 413     nsres = nsIDOMHTMLElement_QueryInterface(nselem, &IID_nsIDOMHTMLScriptElement, (void**)&ret->nsscript);
 414
 415     /* Share nsscript reference with nsnode */
 416     assert(nsres == NS_OK && (nsIDOMNode*)ret->nsscript == ret->element.node.nsnode);
 417     nsIDOMNode_Release(ret->element.node.nsnode);
 418
 419     *elem = &ret->element;
 420     return S_OK;
 421 }