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