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