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