dlls/mshtml/script.c

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