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 #include <assert.h>
  23
  24 #define COBJMACROS
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winuser.h"
  29 #include "ole2.h"
  30 #include "activscp.h"
  31 #include "activdbg.h"
  32 #include "shlwapi.h"
  33
  34 #include "wine/debug.h"
  35
  36 #include "mshtml_private.h"
  37 #include "htmlscript.h"
  38 #include "pluginhost.h"
  39 #include "htmlevent.h"
  40 #include "binding.h"
  41
  42 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  43
  44 #ifdef _WIN64
  45
  46 #define CTXARG_T DWORDLONG
  47 #define IActiveScriptSiteDebugVtbl IActiveScriptSiteDebug64Vtbl
  48
  49 #define IActiveScriptParse_Release IActiveScriptParse64_Release
  50 #define IActiveScriptParse_InitNew IActiveScriptParse64_InitNew
  51 #define IActiveScriptParse_ParseScriptText IActiveScriptParse64_ParseScriptText
  52 #define IActiveScriptParseProcedure2_Release IActiveScriptParseProcedure2_64_Release
  53 #define IActiveScriptParseProcedure2_ParseProcedureText IActiveScriptParseProcedure2_64_ParseProcedureText
  54
  55 #else
  56
  57 #define CTXARG_T DWORD
  58 #define IActiveScriptSiteDebugVtbl IActiveScriptSiteDebug32Vtbl
  59
  60 #define IActiveScriptParse_Release IActiveScriptParse32_Release
  61 #define IActiveScriptParse_InitNew IActiveScriptParse32_InitNew
  62 #define IActiveScriptParse_ParseScriptText IActiveScriptParse32_ParseScriptText
  63 #define IActiveScriptParseProcedure2_Release IActiveScriptParseProcedure2_32_Release
  64 #define IActiveScriptParseProcedure2_ParseProcedureText IActiveScriptParseProcedure2_32_ParseProcedureText
  65
  66 #endif
  67
  68 static const WCHAR documentW[] = {'d','o','c','u','m','e','n','t',0};
  69 static const WCHAR windowW[] = {'w','i','n','d','o','w',0};
  70 static const WCHAR script_endW[] = {'<','/','S','C','R','I','P','T','>',0};
  71 static const WCHAR emptyW[] = {0};
  72
  73 struct ScriptHost {
  74     IActiveScriptSite              IActiveScriptSite_iface;
  75     IActiveScriptSiteInterruptPoll IActiveScriptSiteInterruptPoll_iface;
  76     IActiveScriptSiteWindow        IActiveScriptSiteWindow_iface;
  77     IActiveScriptSiteUIControl     IActiveScriptSiteUIControl_iface;
  78     IActiveScriptSiteDebug         IActiveScriptSiteDebug_iface;
  79     IServiceProvider               IServiceProvider_iface;
  80
  81     LONG ref;
  82
  83     IActiveScript *script;
  84     IActiveScriptParse *parse;
  85     IActiveScriptParseProcedure2 *parse_proc;
  86
  87     SCRIPTSTATE script_state;
  88
  89     HTMLInnerWindow *window;
  90
  91     GUID guid;
  92     struct list entry;
  93 };
  94
  95 static ScriptHost *get_elem_script_host(HTMLInnerWindow*,HTMLScriptElement*);
  96
  97 static void set_script_prop(ScriptHost *script_host, DWORD property, VARIANT *val)
  98 {
  99     IActiveScriptProperty *script_prop;
 100     HRESULT hres;
 101
 102     hres = IActiveScript_QueryInterface(script_host->script, &IID_IActiveScriptProperty,
 103             (void**)&script_prop);
 104     if(FAILED(hres)) {
 105         WARN("Could not get IActiveScriptProperty iface: %08x\n", hres);
 106         return;
 107     }
 108
 109     hres = IActiveScriptProperty_SetProperty(script_prop, property, NULL, val);
 110     IActiveScriptProperty_Release(script_prop);
 111     if(FAILED(hres))
 112         WARN("SetProperty(%x) failed: %08x\n", property, hres);
 113 }
 114
 115 static BOOL init_script_engine(ScriptHost *script_host)
 116 {
 117     IObjectSafety *safety;
 118     SCRIPTSTATE state;
 119     DWORD supported_opts=0, enabled_opts=0;
 120     VARIANT var;
 121     HRESULT hres;
 122
 123     hres = IActiveScript_QueryInterface(script_host->script, &IID_IActiveScriptParse, (void**)&script_host->parse);
 124     if(FAILED(hres)) {
 125         WARN("Could not get IActiveScriptParse: %08x\n", hres);
 126         return FALSE;
 127     }
 128
 129     hres = IActiveScript_QueryInterface(script_host->script, &IID_IObjectSafety, (void**)&safety);
 130     if(FAILED(hres)) {
 131         FIXME("Could not get IObjectSafety: %08x\n", hres);
 132         return FALSE;
 133     }
 134
 135     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported_opts, &enabled_opts);
 136     if(FAILED(hres)) {
 137         FIXME("GetInterfaceSafetyOptions failed: %08x\n", hres);
 138     }else if(!(supported_opts & INTERFACE_USES_DISPEX)) {
 139         FIXME("INTERFACE_USES_DISPEX is not supported\n");
 140     }else {
 141         hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 142                 INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER,
 143                 INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
 144         if(FAILED(hres))
 145             FIXME("SetInterfaceSafetyOptions failed: %08x\n", hres);
 146     }
 147
 148     IObjectSafety_Release(safety);
 149     if(FAILED(hres))
 150         return FALSE;
 151
 152     V_VT(&var) = VT_I4;
 153     V_I4(&var) = lock_document_mode(script_host->window->doc) == COMPAT_MODE_QUIRKS ? 1 : 2;
 154     set_script_prop(script_host, SCRIPTPROP_INVOKEVERSIONING, &var);
 155
 156     V_VT(&var) = VT_BOOL;
 157     V_BOOL(&var) = VARIANT_TRUE;
 158     set_script_prop(script_host, SCRIPTPROP_HACK_TRIDENTEVENTSINK, &var);
 159
 160     hres = IActiveScriptParse_InitNew(script_host->parse);
 161     if(FAILED(hres)) {
 162         WARN("InitNew failed: %08x\n", hres);
 163         return FALSE;
 164     }
 165
 166     hres = IActiveScript_SetScriptSite(script_host->script, &script_host->IActiveScriptSite_iface);
 167     if(FAILED(hres)) {
 168         WARN("SetScriptSite failed: %08x\n", hres);
 169         IActiveScript_Close(script_host->script);
 170         return FALSE;
 171     }
 172
 173     hres = IActiveScript_GetScriptState(script_host->script, &state);
 174     if(FAILED(hres))
 175         WARN("GetScriptState failed: %08x\n", hres);
 176     else if(state != SCRIPTSTATE_INITIALIZED)
 177         FIXME("state = %x\n", state);
 178
 179     hres = IActiveScript_SetScriptState(script_host->script, SCRIPTSTATE_STARTED);
 180     if(FAILED(hres)) {
 181         WARN("Starting script failed: %08x\n", hres);
 182         return FALSE;
 183     }
 184
 185     hres = IActiveScript_AddNamedItem(script_host->script, windowW,
 186             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
 187     if(SUCCEEDED(hres)) {
 188         V_VT(&var) = VT_BOOL;
 189         V_BOOL(&var) = VARIANT_TRUE;
 190         set_script_prop(script_host, SCRIPTPROP_ABBREVIATE_GLOBALNAME_RESOLUTION, &var);
 191     }else {
 192        WARN("AddNamedItem failed: %08x\n", hres);
 193     }
 194
 195     hres = IActiveScript_QueryInterface(script_host->script, &IID_IActiveScriptParseProcedure2,
 196                                         (void**)&script_host->parse_proc);
 197     if(FAILED(hres)) {
 198         /* FIXME: QI for IActiveScriptParseProcedure */
 199         WARN("Could not get IActiveScriptParseProcedure iface: %08x\n", hres);
 200     }
 201
 202     return TRUE;
 203 }
 204
 205 static void release_script_engine(ScriptHost *This)
 206 {
 207     if(!This->script)
 208         return;
 209
 210     switch(This->script_state) {
 211     case SCRIPTSTATE_CONNECTED:
 212         IActiveScript_SetScriptState(This->script, SCRIPTSTATE_DISCONNECTED);
 213
 214     case SCRIPTSTATE_STARTED:
 215     case SCRIPTSTATE_DISCONNECTED:
 216     case SCRIPTSTATE_INITIALIZED:
 217         IActiveScript_Close(This->script);
 218
 219     default:
 220         if(This->parse_proc) {
 221             IActiveScriptParseProcedure2_Release(This->parse_proc);
 222             This->parse_proc = NULL;
 223         }
 224
 225         if(This->parse) {
 226             IActiveScriptParse_Release(This->parse);
 227             This->parse = NULL;
 228         }
 229     }
 230
 231     IActiveScript_Release(This->script);
 232     This->script = NULL;
 233     This->script_state = SCRIPTSTATE_UNINITIALIZED;
 234 }
 235
 236 void connect_scripts(HTMLInnerWindow *window)
 237 {
 238     ScriptHost *iter;
 239
 240     LIST_FOR_EACH_ENTRY(iter, &window->script_hosts, ScriptHost, entry) {
 241         if(iter->script_state == SCRIPTSTATE_STARTED)
 242             IActiveScript_SetScriptState(iter->script, SCRIPTSTATE_CONNECTED);
 243     }
 244 }
 245
 246 static inline ScriptHost *impl_from_IActiveScriptSite(IActiveScriptSite *iface)
 247 {
 248     return CONTAINING_RECORD(iface, ScriptHost, IActiveScriptSite_iface);
 249 }
 250
 251 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
 252 {
 253     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 254
 255     *ppv = NULL;
 256
 257     if(IsEqualGUID(&IID_IUnknown, riid)) {
 258         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
 259         *ppv = &This->IActiveScriptSite_iface;
 260     }else if(IsEqualGUID(&IID_IActiveScriptSite, riid)) {
 261         TRACE("(%p)->(IID_IActiveScriptSite %p)\n", This, ppv);
 262         *ppv = &This->IActiveScriptSite_iface;
 263     }else if(IsEqualGUID(&IID_IActiveScriptSiteInterruptPoll, riid)) {
 264         TRACE("(%p)->(IID_IActiveScriptSiteInterruprtPoll %p)\n", This, ppv);
 265         *ppv = &This->IActiveScriptSiteInterruptPoll_iface;
 266     }else if(IsEqualGUID(&IID_IActiveScriptSiteWindow, riid)) {
 267         TRACE("(%p)->(IID_IActiveScriptSiteWindow %p)\n", This, ppv);
 268         *ppv = &This->IActiveScriptSiteWindow_iface;
 269     }else if(IsEqualGUID(&IID_IActiveScriptSiteUIControl, riid)) {
 270         TRACE("(%p)->(IID_IActiveScriptSiteUIControl %p)\n", This, ppv);
 271         *ppv = &This->IActiveScriptSiteUIControl_iface;
 272     }else if(IsEqualGUID(&IID_IActiveScriptSiteDebug, riid)) {
 273         TRACE("(%p)->(IID_IActiveScriptSiteDebug %p)\n", This, ppv);
 274         *ppv = &This->IActiveScriptSiteDebug_iface;
 275     }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
 276         TRACE("(%p)->(IID_IServiceProvider %p)\n", This, ppv);
 277         *ppv = &This->IServiceProvider_iface;
 278     }else if(IsEqualGUID(&IID_ICanHandleException, riid)) {
 279         TRACE("(%p)->(IID_ICanHandleException not supported %p)\n", This, ppv);
 280         return E_NOINTERFACE;
 281     }else {
 282         FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
 283         return E_NOINTERFACE;
 284     }
 285
 286     IUnknown_AddRef((IUnknown*)*ppv);
 287     return S_OK;
 288 }
 289
 290 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
 291 {
 292     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 293     LONG ref = InterlockedIncrement(&This->ref);
 294
 295     TRACE("(%p) ref=%d\n", This, ref);
 296
 297     return ref;
 298 }
 299
 300 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
 301 {
 302     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 303     LONG ref = InterlockedDecrement(&This->ref);
 304
 305     TRACE("(%p) ref=%d\n", This, ref);
 306
 307     if(!ref) {
 308         release_script_engine(This);
 309         if(This->window)
 310             list_remove(&This->entry);
 311         heap_free(This);
 312     }
 313
 314     return ref;
 315 }
 316
 317 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
 318 {
 319     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 320
 321     TRACE("(%p)->(%p)\n", This, plcid);
 322
 323     *plcid = GetUserDefaultLCID();
 324     return S_OK;
 325 }
 326
 327 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
 328         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
 329 {
 330     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 331
 332     TRACE("(%p)->(%s %x %p %p)\n", This, debugstr_w(pstrName), dwReturnMask, ppiunkItem, ppti);
 333
 334     if(dwReturnMask != SCRIPTINFO_IUNKNOWN) {
 335         FIXME("Unsupported mask %x\n", dwReturnMask);
 336         return E_NOTIMPL;
 337     }
 338
 339     *ppiunkItem = NULL;
 340
 341     if(strcmpW(pstrName, windowW))
 342         return DISP_E_MEMBERNOTFOUND;
 343
 344     if(!This->window)
 345         return E_FAIL;
 346
 347     /* FIXME: Return proxy object */
 348     *ppiunkItem = (IUnknown*)&This->window->base.IHTMLWindow2_iface;
 349     IUnknown_AddRef(*ppiunkItem);
 350
 351     return S_OK;
 352 }
 353
 354 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
 355 {
 356     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 357     FIXME("(%p)->(%p)\n", This, pbstrVersion);
 358     return E_NOTIMPL;
 359 }
 360
 361 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
 362         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
 363 {
 364     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 365     FIXME("(%p)->(%p %p)\n", This, pvarResult, pexcepinfo);
 366     return E_NOTIMPL;
 367 }
 368
 369 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
 370 {
 371     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 372
 373     TRACE("(%p)->(%x)\n", This, ssScriptState);
 374
 375     This->script_state = ssScriptState;
 376     return S_OK;
 377 }
 378
 379 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
 380 {
 381     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 382     FIXME("(%p)->(%p)\n", This, pscripterror);
 383     return E_NOTIMPL;
 384 }
 385
 386 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 387 {
 388     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 389
 390     TRACE("(%p)->()\n", This);
 391
 392     return S_OK;
 393 }
 394
 395 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 396 {
 397     ScriptHost *This = impl_from_IActiveScriptSite(iface);
 398
 399     TRACE("(%p)->()\n", This);
 400
 401     return S_OK;
 402 }
 403
 404 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
 405     ActiveScriptSite_QueryInterface,
 406     ActiveScriptSite_AddRef,
 407     ActiveScriptSite_Release,
 408     ActiveScriptSite_GetLCID,
 409     ActiveScriptSite_GetItemInfo,
 410     ActiveScriptSite_GetDocVersionString,
 411     ActiveScriptSite_OnScriptTerminate,
 412     ActiveScriptSite_OnStateChange,
 413     ActiveScriptSite_OnScriptError,
 414     ActiveScriptSite_OnEnterScript,
 415     ActiveScriptSite_OnLeaveScript
 416 };
 417
 418 static inline ScriptHost *impl_from_IActiveScriptSiteInterruptPoll(IActiveScriptSiteInterruptPoll *iface)
 419 {
 420     return CONTAINING_RECORD(iface, ScriptHost, IActiveScriptSiteInterruptPoll_iface);
 421 }
 422
 423 static HRESULT WINAPI ActiveScriptSiteInterruptPoll_QueryInterface(IActiveScriptSiteInterruptPoll *iface,
 424         REFIID riid, void **ppv)
 425 {
 426     ScriptHost *This = impl_from_IActiveScriptSiteInterruptPoll(iface);
 427     return IActiveScriptSite_QueryInterface(&This->IActiveScriptSite_iface, riid, ppv);
 428 }
 429
 430 static ULONG WINAPI ActiveScriptSiteInterruptPoll_AddRef(IActiveScriptSiteInterruptPoll *iface)
 431 {
 432     ScriptHost *This = impl_from_IActiveScriptSiteInterruptPoll(iface);
 433     return IActiveScriptSite_AddRef(&This->IActiveScriptSite_iface);
 434 }
 435
 436 static ULONG WINAPI ActiveScriptSiteInterruptPoll_Release(IActiveScriptSiteInterruptPoll *iface)
 437 {
 438     ScriptHost *This = impl_from_IActiveScriptSiteInterruptPoll(iface);
 439     return IActiveScriptSite_Release(&This->IActiveScriptSite_iface);
 440 }
 441
 442 static HRESULT WINAPI ActiveScriptSiteInterruptPoll_QueryContinue(IActiveScriptSiteInterruptPoll *iface)
 443 {
 444     ScriptHost *This = impl_from_IActiveScriptSiteInterruptPoll(iface);
 445
 446     TRACE("(%p)\n", This);
 447
 448     return S_OK;
 449 }
 450
 451 static const IActiveScriptSiteInterruptPollVtbl ActiveScriptSiteInterruptPollVtbl = {
 452     ActiveScriptSiteInterruptPoll_QueryInterface,
 453     ActiveScriptSiteInterruptPoll_AddRef,
 454     ActiveScriptSiteInterruptPoll_Release,
 455     ActiveScriptSiteInterruptPoll_QueryContinue
 456 };
 457
 458 static inline ScriptHost *impl_from_IActiveScriptSiteWindow(IActiveScriptSiteWindow *iface)
 459 {
 460     return CONTAINING_RECORD(iface, ScriptHost, IActiveScriptSiteWindow_iface);
 461 }
 462
 463 static HRESULT WINAPI ActiveScriptSiteWindow_QueryInterface(IActiveScriptSiteWindow *iface,
 464         REFIID riid, void **ppv)
 465 {
 466     ScriptHost *This = impl_from_IActiveScriptSiteWindow(iface);
 467     return IActiveScriptSite_QueryInterface(&This->IActiveScriptSite_iface, riid, ppv);
 468 }
 469
 470 static ULONG WINAPI ActiveScriptSiteWindow_AddRef(IActiveScriptSiteWindow *iface)
 471 {
 472     ScriptHost *This = impl_from_IActiveScriptSiteWindow(iface);
 473     return IActiveScriptSite_AddRef(&This->IActiveScriptSite_iface);
 474 }
 475
 476 static ULONG WINAPI ActiveScriptSiteWindow_Release(IActiveScriptSiteWindow *iface)
 477 {
 478     ScriptHost *This = impl_from_IActiveScriptSiteWindow(iface);
 479     return IActiveScriptSite_Release(&This->IActiveScriptSite_iface);
 480 }
 481
 482 static HRESULT WINAPI ActiveScriptSiteWindow_GetWindow(IActiveScriptSiteWindow *iface, HWND *phwnd)
 483 {
 484     ScriptHost *This = impl_from_IActiveScriptSiteWindow(iface);
 485
 486     TRACE("(%p)->(%p)\n", This, phwnd);
 487
 488     if(!This->window || !This->window->base.outer_window || !This->window->base.outer_window->doc_obj)
 489         return E_UNEXPECTED;
 490
 491     *phwnd = This->window->base.outer_window->doc_obj->hwnd;
 492     return S_OK;
 493 }
 494
 495 static HRESULT WINAPI ActiveScriptSiteWindow_EnableModeless(IActiveScriptSiteWindow *iface, BOOL fEnable)
 496 {
 497     ScriptHost *This = impl_from_IActiveScriptSiteWindow(iface);
 498     FIXME("(%p)->(%x)\n", This, fEnable);
 499     return S_OK;
 500 }
 501
 502 static const IActiveScriptSiteWindowVtbl ActiveScriptSiteWindowVtbl = {
 503     ActiveScriptSiteWindow_QueryInterface,
 504     ActiveScriptSiteWindow_AddRef,
 505     ActiveScriptSiteWindow_Release,
 506     ActiveScriptSiteWindow_GetWindow,
 507     ActiveScriptSiteWindow_EnableModeless
 508 };
 509
 510 static inline ScriptHost *impl_from_IActiveScriptSiteUIControl(IActiveScriptSiteUIControl *iface)
 511 {
 512     return CONTAINING_RECORD(iface, ScriptHost, IActiveScriptSiteUIControl_iface);
 513 }
 514
 515 static HRESULT WINAPI ActiveScriptSiteUIControl_QueryInterface(IActiveScriptSiteUIControl *iface, REFIID riid, void **ppv)
 516 {
 517     ScriptHost *This = impl_from_IActiveScriptSiteUIControl(iface);
 518     return IActiveScriptSite_QueryInterface(&This->IActiveScriptSite_iface, riid, ppv);
 519 }
 520
 521 static ULONG WINAPI ActiveScriptSiteUIControl_AddRef(IActiveScriptSiteUIControl *iface)
 522 {
 523     ScriptHost *This = impl_from_IActiveScriptSiteUIControl(iface);
 524     return IActiveScriptSite_AddRef(&This->IActiveScriptSite_iface);
 525 }
 526
 527 static ULONG WINAPI ActiveScriptSiteUIControl_Release(IActiveScriptSiteUIControl *iface)
 528 {
 529     ScriptHost *This = impl_from_IActiveScriptSiteUIControl(iface);
 530     return IActiveScriptSite_Release(&This->IActiveScriptSite_iface);
 531 }
 532
 533 static HRESULT WINAPI ActiveScriptSiteUIControl_GetUIBehavior(IActiveScriptSiteUIControl *iface, SCRIPTUICITEM UicItem,
 534         SCRIPTUICHANDLING *pUicHandling)
 535 {
 536     ScriptHost *This = impl_from_IActiveScriptSiteUIControl(iface);
 537
 538     WARN("(%p)->(%d %p) semi-stub\n", This, UicItem, pUicHandling);
 539
 540     *pUicHandling = SCRIPTUICHANDLING_ALLOW;
 541     return S_OK;
 542 }
 543
 544 static const IActiveScriptSiteUIControlVtbl ActiveScriptSiteUIControlVtbl = {
 545     ActiveScriptSiteUIControl_QueryInterface,
 546     ActiveScriptSiteUIControl_AddRef,
 547     ActiveScriptSiteUIControl_Release,
 548     ActiveScriptSiteUIControl_GetUIBehavior
 549 };
 550
 551 static inline ScriptHost *impl_from_IActiveScriptSiteDebug(IActiveScriptSiteDebug *iface)
 552 {
 553     return CONTAINING_RECORD(iface, ScriptHost, IActiveScriptSiteDebug_iface);
 554 }
 555
 556 static HRESULT WINAPI ActiveScriptSiteDebug_QueryInterface(IActiveScriptSiteDebug *iface,
 557         REFIID riid, void **ppv)
 558 {
 559     ScriptHost *This = impl_from_IActiveScriptSiteDebug(iface);
 560     return IActiveScriptSite_QueryInterface(&This->IActiveScriptSite_iface, riid, ppv);
 561 }
 562
 563 static ULONG WINAPI ActiveScriptSiteDebug_AddRef(IActiveScriptSiteDebug *iface)
 564 {
 565     ScriptHost *This = impl_from_IActiveScriptSiteDebug(iface);
 566     return IActiveScriptSite_AddRef(&This->IActiveScriptSite_iface);
 567 }
 568
 569 static ULONG WINAPI ActiveScriptSiteDebug_Release(IActiveScriptSiteDebug *iface)
 570 {
 571     ScriptHost *This = impl_from_IActiveScriptSiteDebug(iface);
 572     return IActiveScriptSite_Release(&This->IActiveScriptSite_iface);
 573 }
 574
 575 static HRESULT WINAPI ActiveScriptSiteDebug_GetDocumentContextFromPosition(IActiveScriptSiteDebug *iface,
 576             CTXARG_T dwSourceContext, ULONG uCharacterOffset, ULONG uNumChars, IDebugDocumentContext **ppsc)
 577 {
 578     ScriptHost *This = impl_from_IActiveScriptSiteDebug(iface);
 579     FIXME("(%p)->(%s %u %u %p)\n", This, wine_dbgstr_longlong(dwSourceContext), uCharacterOffset,
 580           uNumChars, ppsc);
 581     return E_NOTIMPL;
 582 }
 583
 584 static HRESULT WINAPI ActiveScriptSiteDebug_GetApplication(IActiveScriptSiteDebug *iface, IDebugApplication **ppda)
 585 {
 586     ScriptHost *This = impl_from_IActiveScriptSiteDebug(iface);
 587     FIXME("(%p)->(%p)\n", This, ppda);
 588     return E_NOTIMPL;
 589 }
 590
 591 static HRESULT WINAPI ActiveScriptSiteDebug_GetRootApplicationNode(IActiveScriptSiteDebug *iface,
 592             IDebugApplicationNode **ppdanRoot)
 593 {
 594     ScriptHost *This = impl_from_IActiveScriptSiteDebug(iface);
 595     FIXME("(%p)->(%p)\n", This, ppdanRoot);
 596     return E_NOTIMPL;
 597 }
 598
 599 static HRESULT WINAPI ActiveScriptSiteDebug_OnScriptErrorDebug(IActiveScriptSiteDebug *iface,
 600             IActiveScriptErrorDebug *pErrorDebug, BOOL *pfEnterDebugger, BOOL *pfCallOnScriptErrorWhenContinuing)
 601 {
 602     ScriptHost *This = impl_from_IActiveScriptSiteDebug(iface);
 603     FIXME("(%p)->(%p %p %p)\n", This, pErrorDebug, pfEnterDebugger, pfCallOnScriptErrorWhenContinuing);
 604     return E_NOTIMPL;
 605 }
 606
 607 static const IActiveScriptSiteDebugVtbl ActiveScriptSiteDebugVtbl = {
 608     ActiveScriptSiteDebug_QueryInterface,
 609     ActiveScriptSiteDebug_AddRef,
 610     ActiveScriptSiteDebug_Release,
 611     ActiveScriptSiteDebug_GetDocumentContextFromPosition,
 612     ActiveScriptSiteDebug_GetApplication,
 613     ActiveScriptSiteDebug_GetRootApplicationNode,
 614     ActiveScriptSiteDebug_OnScriptErrorDebug
 615 };
 616
 617 static inline ScriptHost *impl_from_IServiceProvider(IServiceProvider *iface)
 618 {
 619     return CONTAINING_RECORD(iface, ScriptHost, IServiceProvider_iface);
 620 }
 621
 622 static HRESULT WINAPI ASServiceProvider_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv)
 623 {
 624     ScriptHost *This = impl_from_IServiceProvider(iface);
 625     return IActiveScriptSite_QueryInterface(&This->IActiveScriptSite_iface, riid, ppv);
 626 }
 627
 628 static ULONG WINAPI ASServiceProvider_AddRef(IServiceProvider *iface)
 629 {
 630     ScriptHost *This = impl_from_IServiceProvider(iface);
 631     return IActiveScriptSite_AddRef(&This->IActiveScriptSite_iface);
 632 }
 633
 634 static ULONG WINAPI ASServiceProvider_Release(IServiceProvider *iface)
 635 {
 636     ScriptHost *This = impl_from_IServiceProvider(iface);
 637     return IActiveScriptSite_Release(&This->IActiveScriptSite_iface);
 638 }
 639
 640 static HRESULT WINAPI ASServiceProvider_QueryService(IServiceProvider *iface, REFGUID guidService,
 641         REFIID riid, void **ppv)
 642 {
 643     ScriptHost *This = impl_from_IServiceProvider(iface);
 644
 645     if(IsEqualGUID(&SID_SInternetHostSecurityManager, guidService)) {
 646         TRACE("(%p)->(SID_SInternetHostSecurityManager)\n", This);
 647
 648         if(!This->window || !This->window->doc)
 649             return E_NOINTERFACE;
 650
 651         return IInternetHostSecurityManager_QueryInterface(&This->window->doc->IInternetHostSecurityManager_iface,
 652                 riid, ppv);
 653     }
 654
 655     if(IsEqualGUID(&SID_SContainerDispatch, guidService)) {
 656         TRACE("(%p)->(SID_SContainerDispatch)\n", This);
 657
 658         if(!This->window || !This->window->doc)
 659             return E_NOINTERFACE;
 660
 661         return IHTMLDocument2_QueryInterface(&This->window->doc->basedoc.IHTMLDocument2_iface, riid, ppv);
 662     }
 663
 664     FIXME("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
 665     return E_NOINTERFACE;
 666 }
 667
 668 static const IServiceProviderVtbl ASServiceProviderVtbl = {
 669     ASServiceProvider_QueryInterface,
 670     ASServiceProvider_AddRef,
 671     ASServiceProvider_Release,
 672     ASServiceProvider_QueryService
 673 };
 674
 675 static ScriptHost *create_script_host(HTMLInnerWindow *window, const GUID *guid)
 676 {
 677     ScriptHost *ret;
 678     HRESULT hres;
 679
 680     ret = heap_alloc_zero(sizeof(*ret));
 681     if(!ret)
 682         return NULL;
 683
 684     ret->IActiveScriptSite_iface.lpVtbl = &ActiveScriptSiteVtbl;
 685     ret->IActiveScriptSiteInterruptPoll_iface.lpVtbl = &ActiveScriptSiteInterruptPollVtbl;
 686     ret->IActiveScriptSiteWindow_iface.lpVtbl = &ActiveScriptSiteWindowVtbl;
 687     ret->IActiveScriptSiteUIControl_iface.lpVtbl = &ActiveScriptSiteUIControlVtbl;
 688     ret->IActiveScriptSiteDebug_iface.lpVtbl = &ActiveScriptSiteDebugVtbl;
 689     ret->IServiceProvider_iface.lpVtbl = &ASServiceProviderVtbl;
 690     ret->ref = 1;
 691     ret->window = window;
 692     ret->script_state = SCRIPTSTATE_UNINITIALIZED;
 693
 694     ret->guid = *guid;
 695     list_add_tail(&window->script_hosts, &ret->entry);
 696
 697     hres = CoCreateInstance(&ret->guid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 698             &IID_IActiveScript, (void**)&ret->script);
 699     if(FAILED(hres))
 700         WARN("Could not load script engine: %08x\n", hres);
 701     else if(!init_script_engine(ret))
 702         release_script_engine(ret);
 703
 704     return ret;
 705 }
 706
 707 static void dispatch_script_readystatechange_event(HTMLScriptElement *script)
 708 {
 709     DOMEvent *event;
 710     HRESULT hres;
 711
 712     hres = create_document_event(script->element.node.doc, EVENTID_READYSTATECHANGE, &event);
 713     if(FAILED(hres))
 714         return;
 715
 716     dispatch_event(&script->element.node.event_target, event);
 717     IDOMEvent_Release(&event->IDOMEvent_iface);
 718 }
 719
 720 typedef struct {
 721     task_t header;
 722     HTMLScriptElement *elem;
 723 } fire_readystatechange_task_t;
 724
 725 static void fire_readystatechange_proc(task_t *_task)
 726 {
 727     fire_readystatechange_task_t *task = (fire_readystatechange_task_t*)_task;
 728
 729     if(!task->elem->pending_readystatechange_event)
 730         return;
 731
 732     task->elem->pending_readystatechange_event = FALSE;
 733     dispatch_script_readystatechange_event(task->elem);
 734 }
 735
 736 static void fire_readystatechange_task_destr(task_t *_task)
 737 {
 738     fire_readystatechange_task_t *task = (fire_readystatechange_task_t*)_task;
 739
 740     IHTMLScriptElement_Release(&task->elem->IHTMLScriptElement_iface);
 741 }
 742
 743 static void set_script_elem_readystate(HTMLScriptElement *script_elem, READYSTATE readystate)
 744 {
 745     script_elem->readystate = readystate;
 746
 747     if(readystate != READYSTATE_INTERACTIVE) {
 748         if(!script_elem->element.node.doc->window->parser_callback_cnt) {
 749             fire_readystatechange_task_t *task;
 750             HRESULT hres;
 751
 752             if(script_elem->pending_readystatechange_event)
 753                 return;
 754
 755             task = heap_alloc(sizeof(*task));
 756             if(!task)
 757                 return;
 758
 759             IHTMLScriptElement_AddRef(&script_elem->IHTMLScriptElement_iface);
 760             task->elem = script_elem;
 761
 762             hres = push_task(&task->header, fire_readystatechange_proc, fire_readystatechange_task_destr,
 763                     script_elem->element.node.doc->window->task_magic);
 764             if(SUCCEEDED(hres))
 765                 script_elem->pending_readystatechange_event = TRUE;
 766         }else {
 767             script_elem->pending_readystatechange_event = FALSE;
 768             dispatch_script_readystatechange_event(script_elem);
 769         }
 770     }
 771 }
 772
 773 static void parse_elem_text(ScriptHost *script_host, HTMLScriptElement *script_elem, LPCWSTR text)
 774 {
 775     EXCEPINFO excepinfo;
 776     VARIANT var;
 777     HRESULT hres;
 778
 779     TRACE("%s\n", debugstr_w(text));
 780
 781     VariantInit(&var);
 782     memset(&excepinfo, 0, sizeof(excepinfo));
 783     TRACE(">>>\n");
 784     hres = IActiveScriptParse_ParseScriptText(script_host->parse, text, windowW, NULL, script_endW,
 785                                               0, 0, SCRIPTTEXT_ISVISIBLE|SCRIPTTEXT_HOSTMANAGESSOURCE,
 786                                               &var, &excepinfo);
 787     if(SUCCEEDED(hres))
 788         TRACE("<<<\n");
 789     else
 790         WARN("<<< %08x\n", hres);
 791 }
 792
 793 typedef struct {
 794     BSCallback bsc;
 795
 796     HTMLScriptElement *script_elem;
 797     DWORD scheme;
 798
 799     DWORD size;
 800     char *buf;
 801 } ScriptBSC;
 802
 803 static HRESULT get_binding_text(ScriptBSC *bsc, WCHAR **ret)
 804 {
 805     UINT cp = CP_UTF8;
 806     WCHAR *text;
 807
 808     if(!bsc->bsc.readed) {
 809         text = heap_alloc(sizeof(WCHAR));
 810         if(!text)
 811             return E_OUTOFMEMORY;
 812         *text = 0;
 813         *ret = text;
 814         return S_OK;
 815     }
 816
 817     switch(bsc->bsc.bom) {
 818     case BOM_UTF16:
 819         if(bsc->bsc.readed % sizeof(WCHAR)) {
 820             FIXME("The buffer is not a valid utf16 string\n");
 821             return E_FAIL;
 822         }
 823
 824         text = heap_alloc(bsc->bsc.readed+sizeof(WCHAR));
 825         if(!text)
 826             return E_OUTOFMEMORY;
 827
 828         memcpy(text, bsc->buf, bsc->bsc.readed);
 829         text[bsc->bsc.readed/sizeof(WCHAR)] = 0;
 830         break;
 831
 832     default:
 833         /* FIXME: Try to use charset from HTTP headers first */
 834         cp = get_document_charset(bsc->script_elem->element.node.doc);
 835         /* fall through */
 836     case BOM_UTF8: {
 837         DWORD len;
 838
 839         len = MultiByteToWideChar(cp, 0, bsc->buf, bsc->bsc.readed, NULL, 0);
 840         text = heap_alloc((len+1)*sizeof(WCHAR));
 841         if(!text)
 842             return E_OUTOFMEMORY;
 843
 844         MultiByteToWideChar(cp, 0, bsc->buf, bsc->bsc.readed, text, len);
 845         text[len] = 0;
 846     }
 847     }
 848
 849     *ret = text;
 850     return S_OK;
 851 }
 852
 853 static void script_file_available(ScriptBSC *bsc)
 854 {
 855     HTMLScriptElement *script_elem = bsc->script_elem;
 856     HTMLInnerWindow *window = bsc->bsc.window;
 857     ScriptHost *script_host;
 858     nsIDOMNode *parent;
 859     nsresult nsres;
 860     HRESULT hres;
 861
 862     assert(window != NULL);
 863
 864     hres = get_binding_text(bsc, &script_elem->src_text);
 865     if(FAILED(hres))
 866         return;
 867
 868     script_host = get_elem_script_host(window, script_elem);
 869     if(!script_host)
 870         return;
 871
 872     if(window->parser_callback_cnt) {
 873         script_queue_entry_t *queue;
 874
 875         TRACE("Adding to queue\n");
 876
 877         queue = heap_alloc(sizeof(*queue));
 878         if(!queue)
 879             return;
 880
 881         IHTMLScriptElement_AddRef(&script_elem->IHTMLScriptElement_iface);
 882         queue->script = script_elem;
 883
 884         list_add_tail(&window->script_queue, &queue->entry);
 885         return;
 886     }
 887
 888     nsres = nsIDOMElement_GetParentNode(script_elem->element.dom_element, &parent);
 889     if(NS_FAILED(nsres) || !parent) {
 890         TRACE("No parent, not executing\n");
 891         script_elem->parse_on_bind = TRUE;
 892         return;
 893     }
 894
 895     nsIDOMNode_Release(parent);
 896
 897     script_host = get_elem_script_host(window, script_elem);
 898     if(!script_host)
 899         return;
 900
 901     script_elem->parsed = TRUE;
 902     if(script_host->parse)
 903         parse_elem_text(script_host, script_elem, script_elem->src_text);
 904 }
 905
 906 static inline ScriptBSC *impl_from_BSCallback(BSCallback *iface)
 907 {
 908     return CONTAINING_RECORD(iface, ScriptBSC, bsc);
 909 }
 910
 911 static void ScriptBSC_destroy(BSCallback *bsc)
 912 {
 913     ScriptBSC *This = impl_from_BSCallback(bsc);
 914
 915     if(This->script_elem) {
 916         IHTMLScriptElement_Release(&This->script_elem->IHTMLScriptElement_iface);
 917         This->script_elem = NULL;
 918     }
 919
 920     heap_free(This->buf);
 921     heap_free(This);
 922 }
 923
 924 static HRESULT ScriptBSC_init_bindinfo(BSCallback *bsc)
 925 {
 926     return S_OK;
 927 }
 928
 929 static HRESULT ScriptBSC_start_binding(BSCallback *bsc)
 930 {
 931     ScriptBSC *This = impl_from_BSCallback(bsc);
 932
 933     This->script_elem->binding = &This->bsc;
 934
 935     /* FIXME: We should find a better to decide if 'loading' state is supposed to be used by the protocol. */
 936     if(This->scheme == URL_SCHEME_HTTPS || This->scheme == URL_SCHEME_HTTP)
 937         set_script_elem_readystate(This->script_elem, READYSTATE_LOADING);
 938
 939     return S_OK;
 940 }
 941
 942 static HRESULT ScriptBSC_stop_binding(BSCallback *bsc, HRESULT result)
 943 {
 944     ScriptBSC *This = impl_from_BSCallback(bsc);
 945
 946     if(SUCCEEDED(result) && !This->script_elem)
 947         result = E_UNEXPECTED;
 948
 949     assert(FAILED(result) || This->script_elem->binding == &This->bsc);
 950     This->script_elem->binding = NULL;
 951
 952     if(This->script_elem->readystate == READYSTATE_LOADING)
 953         set_script_elem_readystate(This->script_elem, READYSTATE_LOADED);
 954
 955     if(SUCCEEDED(result)) {
 956         script_file_available(This);
 957     }else {
 958         FIXME("binding failed %08x\n", result);
 959         heap_free(This->buf);
 960         This->buf = NULL;
 961         This->size = 0;
 962     }
 963
 964     IHTMLScriptElement_Release(&This->script_elem->IHTMLScriptElement_iface);
 965     This->script_elem = NULL;
 966     return S_OK;
 967 }
 968
 969 static HRESULT ScriptBSC_read_data(BSCallback *bsc, IStream *stream)
 970 {
 971     ScriptBSC *This = impl_from_BSCallback(bsc);
 972     DWORD readed;
 973     HRESULT hres;
 974
 975     if(!This->buf) {
 976         This->buf = heap_alloc(128);
 977         if(!This->buf)
 978             return E_OUTOFMEMORY;
 979         This->size = 128;
 980     }
 981
 982     do {
 983         if(This->bsc.readed >= This->size) {
 984             void *new_buf;
 985             new_buf = heap_realloc(This->buf, This->size << 1);
 986             if(!new_buf)
 987                 return E_OUTOFMEMORY;
 988             This->size <<= 1;
 989             This->buf = new_buf;
 990         }
 991
 992         hres = read_stream(&This->bsc, stream, This->buf+This->bsc.readed, This->size-This->bsc.readed, &readed);
 993     }while(hres == S_OK);
 994
 995     return S_OK;
 996 }
 997
 998 static HRESULT ScriptBSC_on_progress(BSCallback *bsc, ULONG status_code, LPCWSTR status_text)
 999 {
1000     return S_OK;
1001 }
1002
1003 static HRESULT ScriptBSC_on_response(BSCallback *bsc, DWORD response_code,
1004         LPCWSTR response_headers)
1005 {
1006     return S_OK;
1007 }
1008
1009 static HRESULT ScriptBSC_beginning_transaction(BSCallback *bsc, WCHAR **additional_headers)
1010 {
1011     return S_FALSE;
1012 }
1013
1014 static const BSCallbackVtbl ScriptBSCVtbl = {
1015     ScriptBSC_destroy,
1016     ScriptBSC_init_bindinfo,
1017     ScriptBSC_start_binding,
1018     ScriptBSC_stop_binding,
1019     ScriptBSC_read_data,
1020     ScriptBSC_on_progress,
1021     ScriptBSC_on_response,
1022     ScriptBSC_beginning_transaction
1023 };
1024
1025
1026 HRESULT load_script(HTMLScriptElement *script_elem, const WCHAR *src, BOOL async)
1027 {
1028     HTMLInnerWindow *window;
1029     ScriptBSC *bsc;
1030     IMoniker *mon;
1031     IUri *uri;
1032     HRESULT hres;
1033
1034     static const WCHAR wine_schemaW[] = {'w','i','n','e',':'};
1035
1036     if(strlenW(src) > sizeof(wine_schemaW)/sizeof(WCHAR) && !memcmp(src, wine_schemaW, sizeof(wine_schemaW)))
1037         src += sizeof(wine_schemaW)/sizeof(WCHAR);
1038
1039     TRACE("(%p %s %x)\n", script_elem, debugstr_w(src), async);
1040
1041     if(!script_elem->element.node.doc || !(window = script_elem->element.node.doc->window)) {
1042         ERR("no window\n");
1043         return E_UNEXPECTED;
1044     }
1045
1046     hres = create_uri(src, 0, &uri);
1047     if(FAILED(hres))
1048         return hres;
1049
1050     hres = CreateURLMonikerEx2(NULL, uri, &mon, URL_MK_UNIFORM);
1051     if(FAILED(hres)) {
1052         IUri_Release(uri);
1053         return hres;
1054     }
1055
1056     bsc = heap_alloc_zero(sizeof(*bsc));
1057     if(!bsc) {
1058         IMoniker_Release(mon);
1059         IUri_Release(uri);
1060         return E_OUTOFMEMORY;
1061     }
1062
1063     init_bscallback(&bsc->bsc, &ScriptBSCVtbl, mon, async ? BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA : 0);
1064     IMoniker_Release(mon);
1065
1066     hres = IUri_GetScheme(uri, &bsc->scheme);
1067     IUri_Release(uri);
1068     if(FAILED(hres))
1069         bsc->scheme = URL_SCHEME_UNKNOWN;
1070
1071     IHTMLScriptElement_AddRef(&script_elem->IHTMLScriptElement_iface);
1072     bsc->script_elem = script_elem;
1073
1074     hres = start_binding(window, &bsc->bsc, NULL);
1075
1076     IBindStatusCallback_Release(&bsc->bsc.IBindStatusCallback_iface);
1077     return hres;
1078 }
1079
1080 static void parse_inline_script(ScriptHost *script_host, HTMLScriptElement *script_elem)
1081 {
1082     const PRUnichar *text;
1083     nsAString text_str;
1084     nsresult nsres;
1085
1086     nsAString_Init(&text_str, NULL);
1087     nsres = nsIDOMHTMLScriptElement_GetText(script_elem->nsscript, &text_str);
1088     nsAString_GetData(&text_str, &text);
1089
1090     set_script_elem_readystate(script_elem, READYSTATE_INTERACTIVE);
1091
1092     if(NS_FAILED(nsres)) {
1093         ERR("GetText failed: %08x\n", nsres);
1094     }else if(*text) {
1095         parse_elem_text(script_host, script_elem, text);
1096     }
1097
1098     nsAString_Finish(&text_str);
1099 }
1100
1101 static BOOL parse_script_elem(ScriptHost *script_host, HTMLScriptElement *script_elem)
1102 {
1103     nsAString src_str, event_str;
1104     BOOL is_complete = FALSE;
1105     const PRUnichar *src;
1106     nsresult nsres;
1107
1108     nsAString_Init(&event_str, NULL);
1109     nsres = nsIDOMHTMLScriptElement_GetEvent(script_elem->nsscript, &event_str);
1110     if(NS_SUCCEEDED(nsres)) {
1111         const PRUnichar *event;
1112
1113         nsAString_GetData(&event_str, &event);
1114         if(*event) {
1115             TRACE("deferring event %s script evaluation\n", debugstr_w(event));
1116             nsAString_Finish(&event_str);
1117             return FALSE;
1118         }
1119     }else {
1120         ERR("GetEvent failed: %08x\n", nsres);
1121     }
1122     nsAString_Finish(&event_str);
1123
1124     nsAString_Init(&src_str, NULL);
1125     nsres = nsIDOMHTMLScriptElement_GetSrc(script_elem->nsscript, &src_str);
1126     nsAString_GetData(&src_str, &src);
1127
1128     if(NS_FAILED(nsres)) {
1129         ERR("GetSrc failed: %08x\n", nsres);
1130     }else if(*src) {
1131         load_script(script_elem, src, FALSE);
1132         is_complete = script_elem->parsed;
1133     }else {
1134         parse_inline_script(script_host, script_elem);
1135         is_complete = TRUE;
1136     }
1137
1138     nsAString_Finish(&src_str);
1139
1140     return is_complete;
1141 }
1142
1143 static GUID get_default_script_guid(HTMLInnerWindow *window)
1144 {
1145     /* If not specified, we should use very first script host that was created for the page (or JScript if none) */
1146     return list_empty(&window->script_hosts)
1147         ? CLSID_JScript
1148         : LIST_ENTRY(list_head(&window->script_hosts), ScriptHost, entry)->guid;
1149 }
1150
1151 static BOOL get_guid_from_type(LPCWSTR type, GUID *guid)
1152 {
1153     const WCHAR text_javascriptW[] =
1154         {'t','e','x','t','/','j','a','v','a','s','c','r','i','p','t',0};
1155     const WCHAR text_jscriptW[] =
1156         {'t','e','x','t','/','j','s','c','r','i','p','t',0};
1157     const WCHAR text_vbscriptW[] =
1158         {'t','e','x','t','/','v','b','s','c','r','i','p','t',0};
1159
1160     /* FIXME: Handle more types */
1161     if(!strcmpiW(type, text_javascriptW) || !strcmpiW(type, text_jscriptW)) {
1162         *guid = CLSID_JScript;
1163     }else if(!strcmpiW(type, text_vbscriptW)) {
1164         *guid = CLSID_VBScript;
1165     }else {
1166         FIXME("Unknown type %s\n", debugstr_w(type));
1167         return FALSE;
1168     }
1169
1170     return TRUE;
1171 }
1172
1173 static BOOL get_guid_from_language(LPCWSTR type, GUID *guid)
1174 {
1175     HRESULT hres;
1176
1177     hres = CLSIDFromProgID(type, guid);
1178     if(FAILED(hres))
1179         return FALSE;
1180
1181     /* FIXME: Check CATID_ActiveScriptParse */
1182
1183     return TRUE;
1184 }
1185
1186 static BOOL get_script_guid(HTMLInnerWindow *window, nsIDOMHTMLScriptElement *nsscript, GUID *guid)
1187 {
1188     nsIDOMElement *nselem;
1189     const PRUnichar *language;
1190     nsAString val_str;
1191     BOOL ret = FALSE;
1192     nsresult nsres;
1193
1194     static const PRUnichar languageW[] = {'l','a','n','g','u','a','g','e',0};
1195
1196     nsAString_Init(&val_str, NULL);
1197
1198     nsres = nsIDOMHTMLScriptElement_GetType(nsscript, &val_str);
1199     if(NS_SUCCEEDED(nsres)) {
1200         const PRUnichar *type;
1201
1202         nsAString_GetData(&val_str, &type);
1203         if(*type) {
1204             ret = get_guid_from_type(type, guid);
1205             nsAString_Finish(&val_str);
1206             return ret;
1207         }
1208     }else {
1209         ERR("GetType failed: %08x\n", nsres);
1210     }
1211
1212     nsres = nsIDOMHTMLScriptElement_QueryInterface(nsscript, &IID_nsIDOMElement, (void**)&nselem);
1213     assert(nsres == NS_OK);
1214
1215     nsres = get_elem_attr_value(nselem, languageW, &val_str, &language);
1216     nsIDOMElement_Release(nselem);
1217     if(NS_SUCCEEDED(nsres)) {
1218         if(*language) {
1219             ret = get_guid_from_language(language, guid);
1220         }else {
1221             *guid = get_default_script_guid(window);
1222             ret = TRUE;
1223         }
1224         nsAString_Finish(&val_str);
1225     }
1226
1227     return ret;
1228 }
1229
1230 static ScriptHost *get_script_host(HTMLInnerWindow *window, const GUID *guid)
1231 {
1232     ScriptHost *iter;
1233
1234     LIST_FOR_EACH_ENTRY(iter, &window->script_hosts, ScriptHost, entry) {
1235         if(IsEqualGUID(guid, &iter->guid))
1236             return iter;
1237     }
1238
1239     return create_script_host(window, guid);
1240 }
1241
1242 static ScriptHost *get_elem_script_host(HTMLInnerWindow *window, HTMLScriptElement *script_elem)
1243 {
1244     GUID guid;
1245
1246     if(!get_script_guid(window, script_elem->nsscript, &guid)) {
1247         WARN("Could not find script GUID\n");
1248         return NULL;
1249     }
1250
1251     if(IsEqualGUID(&CLSID_JScript, &guid)
1252        && (!window->base.outer_window || window->base.outer_window->scriptmode != SCRIPTMODE_ACTIVESCRIPT)) {
1253         TRACE("Ignoring JScript\n");
1254         return NULL;
1255     }
1256
1257     return get_script_host(window, &guid);
1258 }
1259
1260 void doc_insert_script(HTMLInnerWindow *window, HTMLScriptElement *script_elem, BOOL from_parser)
1261 {
1262     ScriptHost *script_host;
1263     BOOL is_complete = FALSE;
1264
1265     script_host = get_elem_script_host(window, script_elem);
1266     if(!script_host)
1267         return;
1268
1269     if(script_host->parse) {
1270         if(script_elem->src_text) {
1271             if(from_parser)
1272                 set_script_elem_readystate(script_elem, READYSTATE_INTERACTIVE);
1273             script_elem->parsed = TRUE;
1274             parse_elem_text(script_host, script_elem, script_elem->src_text);
1275             is_complete = TRUE;
1276         }else if(!script_elem->binding) {
1277             is_complete = parse_script_elem(script_host, script_elem);
1278         }
1279     }
1280
1281     if(is_complete)
1282         set_script_elem_readystate(script_elem, READYSTATE_COMPLETE);
1283 }
1284
1285 IDispatch *script_parse_event(HTMLInnerWindow *window, LPCWSTR text)
1286 {
1287     ScriptHost *script_host;
1288     GUID guid;
1289     const WCHAR *ptr;
1290     IDispatch *disp;
1291     HRESULT hres;
1292
1293     static const WCHAR delimiterW[] = {'\"',0};
1294
1295     TRACE("%s\n", debugstr_w(text));
1296
1297     for(ptr = text; isalnumW(*ptr); ptr++);
1298     if(*ptr == ':') {
1299         LPWSTR language;
1300         BOOL b;
1301
1302         language = heap_alloc((ptr-text+1)*sizeof(WCHAR));
1303         if(!language)
1304             return NULL;
1305
1306         memcpy(language, text, (ptr-text)*sizeof(WCHAR));
1307         language[ptr-text] = 0;
1308
1309         b = get_guid_from_language(language, &guid);
1310
1311         heap_free(language);
1312
1313         if(!b) {
1314             WARN("Could not find language\n");
1315             return NULL;
1316         }
1317
1318         ptr++;
1319     }else {
1320         ptr = text;
1321         guid = get_default_script_guid(window);
1322     }
1323
1324     if(IsEqualGUID(&CLSID_JScript, &guid)
1325        && (!window->base.outer_window || window->base.outer_window->scriptmode != SCRIPTMODE_ACTIVESCRIPT)) {
1326         TRACE("Ignoring JScript\n");
1327         return NULL;
1328     }
1329
1330     script_host = get_script_host(window, &guid);
1331     if(!script_host || !script_host->parse_proc)
1332         return NULL;
1333
1334     hres = IActiveScriptParseProcedure2_ParseProcedureText(script_host->parse_proc, ptr, NULL, emptyW,
1335             NULL, NULL, delimiterW, 0 /* FIXME */, 0,
1336             SCRIPTPROC_HOSTMANAGESSOURCE|SCRIPTPROC_IMPLICIT_THIS|SCRIPTPROC_IMPLICIT_PARENTS, &disp);
1337     if(FAILED(hres)) {
1338         WARN("ParseProcedureText failed: %08x\n", hres);
1339         return NULL;
1340     }
1341
1342     TRACE("ret %p\n", disp);
1343     return disp;
1344 }
1345
1346 HRESULT exec_script(HTMLInnerWindow *window, const WCHAR *code, const WCHAR *lang, VARIANT *ret)
1347 {
1348     ScriptHost *script_host;
1349     EXCEPINFO ei;
1350     GUID guid;
1351     HRESULT hres;
1352
1353     static const WCHAR delimW[] = {'"',0};
1354
1355     if(!get_guid_from_language(lang, &guid)) {
1356         WARN("Could not find script GUID\n");
1357         return CO_E_CLASSSTRING;
1358     }
1359
1360     script_host = get_script_host(window, &guid);
1361     if(!script_host) {
1362         FIXME("No script host\n");
1363         return E_FAIL;
1364     }
1365
1366     if(!script_host->parse) {
1367         FIXME("script_host->parse == NULL\n");
1368         return E_FAIL;
1369     }
1370
1371     memset(&ei, 0, sizeof(ei));
1372     TRACE(">>>\n");
1373     hres = IActiveScriptParse_ParseScriptText(script_host->parse, code, NULL, NULL, delimW, 0, 0, SCRIPTTEXT_ISVISIBLE, ret, &ei);
1374     if(SUCCEEDED(hres))
1375         TRACE("<<<\n");
1376     else
1377         WARN("<<< %08x\n", hres);
1378
1379     return hres;
1380 }
1381
1382 IDispatch *get_script_disp(ScriptHost *script_host)
1383 {
1384     IDispatch *disp;
1385     HRESULT hres;
1386
1387     if(!script_host->script)
1388         return NULL;
1389
1390     hres = IActiveScript_GetScriptDispatch(script_host->script, windowW, &disp);
1391     if(FAILED(hres))
1392         return NULL;
1393
1394     return disp;
1395 }
1396
1397 static EventTarget *find_event_target(HTMLDocumentNode *doc, HTMLScriptElement *script_elem)
1398 {
1399     EventTarget *event_target = NULL;
1400     const PRUnichar *target_id;
1401     nsAString target_id_str;
1402     nsresult nsres;
1403     HRESULT hres;
1404
1405     nsAString_Init(&target_id_str, NULL);
1406     nsres = nsIDOMHTMLScriptElement_GetHtmlFor(script_elem->nsscript, &target_id_str);
1407     if(NS_FAILED(nsres)) {
1408         ERR("GetScriptFor failed: %08x\n", nsres);
1409         nsAString_Finish(&target_id_str);
1410         return NULL;
1411     }
1412
1413     nsAString_GetData(&target_id_str, &target_id);
1414     if(!*target_id) {
1415         FIXME("Empty for attribute\n");
1416     }else if(!strcmpW(target_id, documentW)) {
1417         event_target = &doc->node.event_target;
1418         htmldoc_addref(&doc->basedoc);
1419     }else if(!strcmpW(target_id, windowW)) {
1420         if(doc->window) {
1421             event_target = &doc->window->event_target;
1422             IDispatchEx_AddRef(&event_target->dispex.IDispatchEx_iface);
1423         }
1424     }else {
1425         HTMLElement *target_elem;
1426
1427         hres = get_doc_elem_by_id(doc, target_id, &target_elem);
1428         if(SUCCEEDED(hres) && target_elem) {
1429             event_target = &target_elem->node.event_target;
1430         }
1431     }
1432
1433     nsAString_Finish(&target_id_str);
1434     return event_target;
1435 }
1436
1437 static BOOL parse_event_str(WCHAR *event, const WCHAR **args)
1438 {
1439     WCHAR *ptr;
1440
1441     TRACE("%s\n", debugstr_w(event));
1442
1443     for(ptr = event; isalnumW(*ptr); ptr++);
1444     if(!*ptr) {
1445         *args = NULL;
1446         return TRUE;
1447     }
1448
1449     if(*ptr != '(')
1450         return FALSE;
1451
1452     *ptr++ = 0;
1453     *args = ptr;
1454     while(isalnumW(*ptr) || isspaceW(*ptr) || *ptr == ',')
1455         ptr++;
1456
1457     if(*ptr != ')')
1458         return FALSE;
1459
1460     *ptr++ = 0;
1461     return !*ptr;
1462 }
1463
1464 static IDispatch *parse_event_elem(HTMLDocumentNode *doc, HTMLScriptElement *script_elem, WCHAR **ret_event)
1465 {
1466     ScriptHost *script_host;
1467     WCHAR *event = NULL;
1468     const WCHAR *args;
1469     nsAString nsstr;
1470     IDispatch *disp;
1471     nsresult nsres;
1472     HRESULT hres;
1473
1474     if(script_elem->parsed)
1475         return NULL;
1476
1477     script_host = get_elem_script_host(doc->window, script_elem);
1478     if(!script_host || !script_host->parse_proc)
1479         return NULL;
1480
1481     nsAString_Init(&nsstr, NULL);
1482     nsres = nsIDOMHTMLScriptElement_GetEvent(script_elem->nsscript, &nsstr);
1483     if(NS_SUCCEEDED(nsres)) {
1484         const PRUnichar *event_val;
1485
1486         nsAString_GetData(&nsstr, &event_val);
1487         event = heap_strdupW(event_val);
1488     }
1489     nsAString_Finish(&nsstr);
1490     if(!event)
1491         return NULL;
1492
1493     if(!parse_event_str(event, &args)) {
1494         WARN("parsing %s failed\n", debugstr_w(event));
1495         heap_free(event);
1496         return NULL;
1497     }
1498
1499     nsAString_Init(&nsstr, NULL);
1500     nsres = nsIDOMHTMLScriptElement_GetText(script_elem->nsscript, &nsstr);
1501     if(NS_SUCCEEDED(nsres)) {
1502         const PRUnichar *text;
1503
1504         nsAString_GetData(&nsstr, &text);
1505         hres = IActiveScriptParseProcedure2_ParseProcedureText(script_host->parse_proc, text, args,
1506                 emptyW, NULL, NULL, script_endW, 0, 0,
1507                 SCRIPTPROC_HOSTMANAGESSOURCE|SCRIPTPROC_IMPLICIT_THIS|SCRIPTPROC_IMPLICIT_PARENTS, &disp);
1508         if(FAILED(hres))
1509             disp = NULL;
1510     }else {
1511         ERR("GetText failed: %08x\n", nsres);
1512         disp = NULL;
1513     }
1514     nsAString_Finish(&nsstr);
1515     if(!disp) {
1516         heap_free(event);
1517         return NULL;
1518     }
1519
1520     *ret_event = event;
1521     return disp;
1522 }
1523
1524 void bind_event_scripts(HTMLDocumentNode *doc)
1525 {
1526     HTMLPluginContainer *plugin_container;
1527     nsIDOMHTMLScriptElement *nsscript;
1528     HTMLScriptElement *script_elem;
1529     EventTarget *event_target;
1530     nsIDOMNodeList *node_list;
1531     nsIDOMNode *script_node;
1532     nsAString selector_str;
1533     IDispatch *event_disp;
1534     UINT32 length, i;
1535     WCHAR *event;
1536     nsresult nsres;
1537     HRESULT hres;
1538
1539     static const PRUnichar selectorW[] = {'s','c','r','i','p','t','[','e','v','e','n','t',']',0};
1540
1541     TRACE("%p\n", doc);
1542
1543     if(!doc->nsdoc)
1544         return;
1545
1546     nsAString_InitDepend(&selector_str, selectorW);
1547     nsres = nsIDOMHTMLDocument_QuerySelectorAll(doc->nsdoc, &selector_str, &node_list);
1548     nsAString_Finish(&selector_str);
1549     if(NS_FAILED(nsres)) {
1550         ERR("QuerySelectorAll failed: %08x\n", nsres);
1551         return;
1552     }
1553
1554     if(!node_list)
1555         return;
1556
1557     nsres = nsIDOMNodeList_GetLength(node_list, &length);
1558     assert(nsres == NS_OK);
1559
1560     for(i=0; i < length; i++) {
1561         nsres = nsIDOMNodeList_Item(node_list, i, &script_node);
1562         if(NS_FAILED(nsres) || !script_node) {
1563             ERR("Item(%d) failed: %08x\n", i, nsres);
1564             continue;
1565         }
1566
1567         nsres = nsIDOMNode_QueryInterface(script_node, &IID_nsIDOMHTMLScriptElement, (void**)&nsscript);
1568         assert(nsres == NS_OK);
1569         nsIDOMNode_Release(script_node);
1570
1571         hres = script_elem_from_nsscript(nsscript, &script_elem);
1572         if(FAILED(hres))
1573             continue;
1574
1575         event_disp = parse_event_elem(doc, script_elem, &event);
1576         if(event_disp) {
1577             event_target = find_event_target(doc, script_elem);
1578             if(event_target) {
1579                 hres = IDispatchEx_QueryInterface(&event_target->dispex.IDispatchEx_iface, &IID_HTMLPluginContainer,
1580                         (void**)&plugin_container);
1581                 if(SUCCEEDED(hres))
1582                     bind_activex_event(doc, plugin_container, event, event_disp);
1583                 else
1584                     bind_target_event(doc, event_target, event, event_disp);
1585
1586                 IDispatchEx_Release(&event_target->dispex.IDispatchEx_iface);
1587                 if(plugin_container)
1588                     node_release(&plugin_container->element.node);
1589             }
1590
1591             heap_free(event);
1592             IDispatch_Release(event_disp);
1593         }
1594
1595         IHTMLScriptElement_Release(&script_elem->IHTMLScriptElement_iface);
1596     }
1597
1598     nsIDOMNodeList_Release(node_list);
1599 }
1600
1601 BOOL find_global_prop(HTMLInnerWindow *window, BSTR name, DWORD flags, ScriptHost **ret_host, DISPID *ret_id)
1602 {
1603     IDispatchEx *dispex;
1604     IDispatch *disp;
1605     ScriptHost *iter;
1606     HRESULT hres;
1607
1608     LIST_FOR_EACH_ENTRY(iter, &window->script_hosts, ScriptHost, entry) {
1609         disp = get_script_disp(iter);
1610         if(!disp)
1611             continue;
1612
1613         hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1614         if(SUCCEEDED(hres)) {
1615             hres = IDispatchEx_GetDispID(dispex, name, flags & (~fdexNameEnsure), ret_id);
1616             IDispatchEx_Release(dispex);
1617         }else {
1618             FIXME("No IDispatchEx\n");
1619             hres = E_NOTIMPL;
1620         }
1621
1622         IDispatch_Release(disp);
1623         if(SUCCEEDED(hres)) {
1624             *ret_host = iter;
1625             return TRUE;
1626         }
1627     }
1628
1629     return FALSE;
1630 }
1631
1632 static BOOL is_jscript_available(void)
1633 {
1634     static BOOL available, checked;
1635
1636     if(!checked) {
1637         IUnknown *unk;
1638         HRESULT hres = CoGetClassObject(&CLSID_JScript, CLSCTX_INPROC_SERVER, NULL, &IID_IUnknown, (void**)&unk);
1639
1640         if(SUCCEEDED(hres)) {
1641             available = TRUE;
1642             IUnknown_Release(unk);
1643         }else {
1644             available = FALSE;
1645         }
1646         checked = TRUE;
1647     }
1648
1649     return available;
1650 }
1651
1652 void set_script_mode(HTMLOuterWindow *window, SCRIPTMODE mode)
1653 {
1654     nsIWebBrowserSetup *setup;
1655     nsresult nsres;
1656
1657     if(mode == SCRIPTMODE_ACTIVESCRIPT && !is_jscript_available()) {
1658         TRACE("jscript.dll not available\n");
1659         window->scriptmode = SCRIPTMODE_GECKO;
1660         return;
1661     }
1662
1663     window->scriptmode = mode;
1664
1665     if(!window->doc_obj->nscontainer || !window->doc_obj->nscontainer->webbrowser)
1666         return;
1667
1668     nsres = nsIWebBrowser_QueryInterface(window->doc_obj->nscontainer->webbrowser,
1669             &IID_nsIWebBrowserSetup, (void**)&setup);
1670     if(NS_SUCCEEDED(nsres)) {
1671         nsres = nsIWebBrowserSetup_SetProperty(setup, SETUP_ALLOW_JAVASCRIPT,
1672                 window->scriptmode == SCRIPTMODE_GECKO);
1673
1674         if(NS_SUCCEEDED(nsres))
1675             nsres = nsIWebBrowserSetup_SetProperty(setup, SETUP_DISABLE_NOSCRIPT, TRUE);
1676
1677         nsIWebBrowserSetup_Release(setup);
1678     }
1679
1680     if(NS_FAILED(nsres))
1681         ERR("JavaScript setup failed: %08x\n", nsres);
1682 }
1683
1684 void release_script_hosts(HTMLInnerWindow *window)
1685 {
1686     script_queue_entry_t *queue_iter;
1687     ScriptHost *iter;
1688
1689     while(!list_empty(&window->script_queue)) {
1690         queue_iter = LIST_ENTRY(list_head(&window->script_queue), script_queue_entry_t, entry);
1691
1692         list_remove(&queue_iter->entry);
1693         IHTMLScriptElement_Release(&queue_iter->script->IHTMLScriptElement_iface);
1694         heap_free(queue_iter);
1695     }
1696
1697     while(!list_empty(&window->script_hosts)) {
1698         iter = LIST_ENTRY(list_head(&window->script_hosts), ScriptHost, entry);
1699
1700         release_script_engine(iter);
1701         list_remove(&iter->entry);
1702         iter->window = NULL;
1703         IActiveScriptSite_Release(&iter->IActiveScriptSite_iface);
1704     }
1705 }