dlls/mshtml/script.c - wine - Git at Google

 /*
  * Copyright 2008 Jacek Caban for CodeWeavers
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
  * version 2.1 of the License, or (at your option) any later version.
  *
  * This library is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
  * License along with this library; if not, write to the Free Software
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  */

 #include "config.h"

 #include <stdarg.h>

 #define COBJMACROS

 #include "windef.h"
 #include "winbase.h"
 #include "winuser.h"
 #include "ole2.h"
 #include "activscp.h"
 #include "objsafe.h"

 #include "wine/debug.h"

 #include "mshtml_private.h"

 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);

 static const WCHAR windowW[] = {'w','i','n','d','o','w',0};

 static const CLSID CLSID_JScript =
     {0xf414c260,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};

 typedef struct {
     const IActiveScriptSiteVtbl               *lpIActiveScriptSiteVtbl;
     const IActiveScriptSiteInterruptPollVtbl  *lpIActiveScriptSiteInterruptPollVtbl;
     const IActiveScriptSiteWindowVtbl         *lpIActiveScriptSiteWindowVtbl;

     LONG ref;

     IActiveScript *script;
     IActiveScriptParse *parse;

     SCRIPTSTATE script_state;

     HTMLDocument *doc;

     GUID guid;
     struct list entry;
 } ScriptHost;

 #define ACTSCPSITE(x)  ((IActiveScriptSite*)               &(x)->lpIActiveScriptSiteVtbl)
 #define ACTSCPPOLL(x)  ((IActiveScriptSiteInterruptPoll*)  &(x)->lpIActiveScriptSiteInterruptPollVtbl)
 #define ACTSCPWIN(x)   ((IActiveScriptSiteWindow*)         &(x)->lpIActiveScriptSiteWindowVtbl)

 static BOOL init_script_engine(ScriptHost *script_host)
 {
     IActiveScriptProperty *property;
     IObjectSafety *safety;
     SCRIPTSTATE state;
     DWORD supported_opts=0, enabled_opts=0;
     HRESULT hres;

     hres = IActiveScript_QueryInterface(script_host->script, &IID_IActiveScriptParse, (void**)&script_host->parse);
     if(FAILED(hres)) {
         WARN("Could not get IActiveScriptHost: %08x\n", hres);
         return FALSE;
     }

     hres = IActiveScript_QueryInterface(script_host->script, &IID_IObjectSafety, (void**)&safety);
     if(FAILED(hres)) {
         FIXME("Could not get IObjectSafety: %08x\n", hres);
         return FALSE;
     }

     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported_opts, &enabled_opts);
     if(FAILED(hres)) {
         FIXME("GetInterfaceSafetyOptions failed: %08x\n", hres);
     }else if(!(supported_opts & INTERFACE_USES_DISPEX)) {
         FIXME("INTERFACE_USES_DISPEX is not supported\n");
     }else {
         hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
                 INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER,
                 INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
         if(FAILED(hres))
             FIXME("SetInterfaceSafetyOptions failed: %08x\n", hres);
     }

     IObjectSafety_Release(safety);
     if(FAILED(hres))
         return FALSE;

     hres = IActiveScript_QueryInterface(script_host->script, &IID_IActiveScriptProperty, (void**)&property);
     if(SUCCEEDED(hres)) {
         VARIANT var;

         V_VT(&var) = VT_BOOL;
         V_BOOL(&var) = VARIANT_TRUE;
         hres = IActiveScriptProperty_SetProperty(property, SCRIPTPROP_HACK_TRIDENTEVENTSINK, NULL, &var);
         if(FAILED(hres))
             WARN("SetProperty failed: %08x\n", hres);

         IActiveScriptProperty_Release(property);
     }else {
         WARN("Could not get IActiveScriptProperty: %08x\n", hres);
     }

     hres = IActiveScriptParse_InitNew(script_host->parse);
     if(FAILED(hres)) {
         WARN("InitNew failed: %08x\n", hres);
         return FALSE;
     }

     hres = IActiveScript_SetScriptSite(script_host->script, ACTSCPSITE(script_host));
     if(FAILED(hres)) {
         WARN("SetScriptSite failed: %08x\n", hres);
         IActiveScript_Close(script_host->script);
         return FALSE;
     }

     hres = IActiveScript_GetScriptState(script_host->script, &state);
     if(FAILED(hres))
         WARN("GetScriptState failed: %08x\n", hres);
     else if(state != SCRIPTSTATE_INITIALIZED)
         FIXME("state = %x\n", state);

     hres = IActiveScript_SetScriptState(script_host->script, SCRIPTSTATE_STARTED);
     if(FAILED(hres)) {
         WARN("Starting script failed: %08x\n", hres);
         return FALSE;
     }

     hres = IActiveScript_AddNamedItem(script_host->script, windowW,
             SCRIPTITEM_ISVISIBLE|SCRIPTITEM_ISSOURCE|SCRIPTITEM_GLOBALMEMBERS);
     if(FAILED(hres))
        WARN("AddNamedItem failed: %08x\n", hres);

     /* FIXME: QI for IActiveScriptParseProcedure2 and IActiveScriptParseProcedure */

     return TRUE;
 }

 static void release_script_engine(ScriptHost *This)
 {
     if(!This->script)
         return;

     switch(This->script_state) {
     case SCRIPTSTATE_CONNECTED:
         IActiveScript_SetScriptState(This->script, SCRIPTSTATE_DISCONNECTED);

     case SCRIPTSTATE_STARTED:
     case SCRIPTSTATE_DISCONNECTED:
     case SCRIPTSTATE_INITIALIZED:
         IActiveScript_Close(This->script);

     default:
         if(This->parse) {
             IActiveScriptParse_Release(This->parse);
             This->parse = NULL;
         }
     }

     IActiveScript_Release(This->script);
     This->script = NULL;
     This->script_state = SCRIPTSTATE_UNINITIALIZED;
 }

 void connect_scripts(HTMLDocument *doc)
 {
     ScriptHost *iter;

     LIST_FOR_EACH_ENTRY(iter, &doc->script_hosts, ScriptHost, entry) {
         if(iter->script_state == SCRIPTSTATE_STARTED)
             IActiveScript_SetScriptState(iter->script, SCRIPTSTATE_CONNECTED);
     }
 }

 #define ACTSCPSITE_THIS(iface) DEFINE_THIS(ScriptHost, IActiveScriptSite, iface)

 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);

     *ppv = NULL;

     if(IsEqualGUID(&IID_IUnknown, riid)) {
         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
         *ppv = ACTSCPSITE(This);
     }else if(IsEqualGUID(&IID_IActiveScriptSite, riid)) {
         TRACE("(%p)->(IID_IActiveScriptSite %p)\n", This, ppv);
         *ppv = ACTSCPSITE(This);
     }else if(IsEqualGUID(&IID_IActiveScriptSiteInterruptPoll, riid)) {
         TRACE("(%p)->(IID_IActiveScriptSiteInterruprtPoll %p)\n", This, ppv);
         *ppv = ACTSCPPOLL(This);
     }else if(IsEqualGUID(&IID_IActiveScriptSiteWindow, riid)) {
         TRACE("(%p)->(IID_IActiveScriptSiteWindow %p)\n", This, ppv);
         *ppv = ACTSCPWIN(This);
     }else {
         FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
         return E_NOINTERFACE;
     }

     IUnknown_AddRef((IUnknown*)*ppv);
     return S_OK;
 }

 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);
     LONG ref = InterlockedIncrement(&This->ref);

     TRACE("(%p) ref=%d\n", This, ref);

     return ref;
 }

 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);
     LONG ref = InterlockedDecrement(&This->ref);

     TRACE("(%p) ref=%d\n", This, ref);

     if(!ref) {
         release_script_engine(This);
         if(This->doc)
             list_remove(&This->entry);
         heap_free(This);
     }

     return ref;
 }

 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);

     TRACE("(%p)->(%p)\n", This, plcid);

     *plcid = GetUserDefaultLCID();
     return S_OK;
 }

 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);

     TRACE("(%p)->(%s %x %p %p)\n", This, debugstr_w(pstrName), dwReturnMask, ppiunkItem, ppti);

     if(dwReturnMask != SCRIPTINFO_IUNKNOWN) {
         FIXME("Unsupported mask %x\n", dwReturnMask);
         return E_NOTIMPL;
     }

     *ppiunkItem = NULL;

     if(strcmpW(pstrName, windowW))
         return DISP_E_MEMBERNOTFOUND;

     if(!This->doc)
         return E_FAIL;

     /* FIXME: Return proxy object */
     *ppiunkItem = (IUnknown*)HTMLWINDOW2(This->doc->window);
     IUnknown_AddRef(*ppiunkItem);

     return S_OK;
 }

 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);
     FIXME("(%p)->(%p)\n", This, pbstrVersion);
     return E_NOTIMPL;
 }

 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);
     FIXME("(%p)->(%p %p)\n", This, pvarResult, pexcepinfo);
     return E_NOTIMPL;
 }

 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);

     TRACE("(%p)->(%x)\n", This, ssScriptState);

     This->script_state = ssScriptState;
     return S_OK;
 }

 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);
     FIXME("(%p)->(%p)\n", This, pscripterror);
     return E_NOTIMPL;
 }

 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);

     TRACE("(%p)->()\n", This);

     return S_OK;
 }

 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 {
     ScriptHost *This = ACTSCPSITE_THIS(iface);

     TRACE("(%p)->()\n", This);

     return S_OK;
 }

 #undef ACTSCPSITE_THIS

 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
     ActiveScriptSite_QueryInterface,
     ActiveScriptSite_AddRef,
     ActiveScriptSite_Release,
     ActiveScriptSite_GetLCID,
     ActiveScriptSite_GetItemInfo,
     ActiveScriptSite_GetDocVersionString,
     ActiveScriptSite_OnScriptTerminate,
     ActiveScriptSite_OnStateChange,
     ActiveScriptSite_OnScriptError,
     ActiveScriptSite_OnEnterScript,
     ActiveScriptSite_OnLeaveScript
 };

 #define ACTSCPPOLL_THIS(iface) DEFINE_THIS(ScriptHost, IActiveScriptSiteInterruptPoll, iface)

 static HRESULT WINAPI ActiveScriptSiteInterruptPoll_QueryInterface(IActiveScriptSiteInterruptPoll *iface,
         REFIID riid, void **ppv)
 {
     ScriptHost *This = ACTSCPPOLL_THIS(iface);
     return IActiveScriptSite_QueryInterface(ACTSCPSITE(This), riid, ppv);
 }

 static ULONG WINAPI ActiveScriptSiteInterruptPoll_AddRef(IActiveScriptSiteInterruptPoll *iface)
 {
     ScriptHost *This = ACTSCPPOLL_THIS(iface);
     return IActiveScriptSite_AddRef(ACTSCPSITE(This));
 }

 static ULONG WINAPI ActiveScriptSiteInterruptPoll_Release(IActiveScriptSiteInterruptPoll *iface)
 {
     ScriptHost *This = ACTSCPPOLL_THIS(iface);
     return IActiveScriptSite_Release(ACTSCPSITE(This));
 }

 static HRESULT WINAPI ActiveScriptSiteInterruptPoll_QueryContinue(IActiveScriptSiteInterruptPoll *iface)
 {
     ScriptHost *This = ACTSCPPOLL_THIS(iface);
     FIXME("(%p)\n", This);
     return E_NOTIMPL;
 }

 #undef ACTSCPPOLL_THIS

 static const IActiveScriptSiteInterruptPollVtbl ActiveScriptSiteInterruptPollVtbl = {
     ActiveScriptSiteInterruptPoll_QueryInterface,
     ActiveScriptSiteInterruptPoll_AddRef,
     ActiveScriptSiteInterruptPoll_Release,
     ActiveScriptSiteInterruptPoll_QueryContinue
 };

 #define ACTSCPWIN_THIS(iface) DEFINE_THIS(ScriptHost, IActiveScriptSiteWindow, iface)

 static HRESULT WINAPI ActiveScriptSiteWindow_QueryInterface(IActiveScriptSiteWindow *iface,
         REFIID riid, void **ppv)
 {
     ScriptHost *This = ACTSCPWIN_THIS(iface);
     return IActiveScriptSite_QueryInterface(ACTSCPSITE(This), riid, ppv);
 }

 static ULONG WINAPI ActiveScriptSiteWindow_AddRef(IActiveScriptSiteWindow *iface)
 {
     ScriptHost *This = ACTSCPWIN_THIS(iface);
     return IActiveScriptSite_AddRef(ACTSCPSITE(This));
 }

 static ULONG WINAPI ActiveScriptSiteWindow_Release(IActiveScriptSiteWindow *iface)
 {
     ScriptHost *This = ACTSCPWIN_THIS(iface);
     return IActiveScriptSite_Release(ACTSCPSITE(This));
 }

 static HRESULT WINAPI ActiveScriptSiteWindow_GetWindow(IActiveScriptSiteWindow *iface, HWND *phwnd)
 {
     ScriptHost *This = ACTSCPWIN_THIS(iface);
     FIXME("(%p)->(%p)\n", This, phwnd);
     return E_NOTIMPL;
 }

 static HRESULT WINAPI ActiveScriptSiteWindow_EnableModeless(IActiveScriptSiteWindow *iface, BOOL fEnable)
 {
     ScriptHost *This = ACTSCPWIN_THIS(iface);
     FIXME("(%p)->(%x)\n", This, fEnable);
     return E_NOTIMPL;
 }

 #undef ACTSCPWIN_THIS

 static const IActiveScriptSiteWindowVtbl ActiveScriptSiteWindowVtbl = {
     ActiveScriptSiteWindow_QueryInterface,
     ActiveScriptSiteWindow_AddRef,
     ActiveScriptSiteWindow_Release,
     ActiveScriptSiteWindow_GetWindow,
     ActiveScriptSiteWindow_EnableModeless
 };

 static ScriptHost *create_script_host(HTMLDocument *doc, GUID *guid)
 {
     ScriptHost *ret;
     HRESULT hres;

     ret = heap_alloc_zero(sizeof(*ret));
     ret->lpIActiveScriptSiteVtbl               = &ActiveScriptSiteVtbl;
     ret->lpIActiveScriptSiteInterruptPollVtbl  = &ActiveScriptSiteInterruptPollVtbl;
     ret->lpIActiveScriptSiteWindowVtbl         = &ActiveScriptSiteWindowVtbl;
     ret->ref = 1;
     ret->doc = doc;
     ret->script_state = SCRIPTSTATE_UNINITIALIZED;

     ret->guid = *guid;
     list_add_tail(&doc->script_hosts, &ret->entry);

     hres = CoCreateInstance(&ret->guid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
             &IID_IActiveScript, (void**)&ret->script);
     if(FAILED(hres))
         WARN("Could not load script engine: %08x\n", hres);
     else if(!init_script_engine(ret))
         release_script_engine(ret);

     return ret;
 }

 static void parse_text(ScriptHost *script_host, LPCWSTR text)
 {
     EXCEPINFO excepinfo;
     VARIANT var;
     HRESULT hres;

     static const WCHAR script_endW[] = {'<','/','S','C','R','I','P','T','>',0};

     TRACE("%s\n", debugstr_w(text));

     VariantInit(&var);
     memset(&excepinfo, 0, sizeof(excepinfo));
     hres = IActiveScriptParse_ParseScriptText(script_host->parse, text, windowW, NULL, script_endW,
                                               0, 0, SCRIPTTEXT_ISVISIBLE|SCRIPTTEXT_HOSTMANAGESSOURCE,
                                               &var, &excepinfo);
     if(FAILED(hres))
         WARN("ParseScriptText failed: %08x\n", hres);

 }

 static void parse_extern_script(ScriptHost *script_host, LPCWSTR src)
 {
     IMoniker *mon;
     char *buf;
     WCHAR *text;
     HRESULT hres;

     static const WCHAR wine_schemaW[] = {'w','i','n','e',':'};

     if(strlenW(src) > sizeof(wine_schemaW)/sizeof(WCHAR) && !memcmp(src, wine_schemaW, sizeof(wine_schemaW)))
         src += sizeof(wine_schemaW)/sizeof(WCHAR);

     hres = CreateURLMoniker(NULL, src, &mon);
     if(FAILED(hres))
         return;

     hres = bind_mon_to_buffer(script_host->doc, mon, (void**)&buf);
     IMoniker_Release(mon);
     if(FAILED(hres))
         return;

     text = heap_strdupAtoW(buf);
     heap_free(buf);

     parse_text(script_host, text);

     heap_free(text);
 }

 static void parse_inline_script(ScriptHost *script_host, nsIDOMHTMLScriptElement *nsscript)
 {
     const PRUnichar *text;
     nsAString text_str;
     nsresult nsres;

     nsAString_Init(&text_str, NULL);

     nsres = nsIDOMHTMLScriptElement_GetText(nsscript, &text_str);

     if(NS_SUCCEEDED(nsres)) {
         nsAString_GetData(&text_str, &text);
         parse_text(script_host, text);
     }else {
         ERR("GetText failed: %08x\n", nsres);
     }

     nsAString_Finish(&text_str);
 }

 static void parse_script_elem(ScriptHost *script_host, nsIDOMHTMLScriptElement *nsscript)
 {
     const PRUnichar *src;
     nsAString src_str;
     nsresult nsres;

     nsAString_Init(&src_str, NULL);

     nsres = nsIDOMHTMLScriptElement_GetSrc(nsscript, &src_str);
     nsAString_GetData(&src_str, &src);

     if(NS_FAILED(nsres))
         ERR("GetSrc failed: %08x\n", nsres);
     else if(*src)
         parse_extern_script(script_host, src);
     else
         parse_inline_script(script_host, nsscript);

     nsAString_Finish(&src_str);
 }

 static BOOL get_guid_from_type(LPCWSTR type, GUID *guid)
 {
     const WCHAR text_javascriptW[] =
         {'t','e','x','t','/','j','a','v','a','s','c','r','i','p','t',0};

     /* FIXME: Handle more types */
     if(!strcmpW(type, text_javascriptW)) {
         *guid = CLSID_JScript;
     }else {
         FIXME("Unknown type %s\n", debugstr_w(type));
         return FALSE;
     }

     return TRUE;
 }

 static BOOL get_guid_from_language(LPCWSTR type, GUID *guid)
 {
     HRESULT hres;

     hres = CLSIDFromProgID(type, guid);
     if(FAILED(hres))
         return FALSE;

     /* FIXME: Check CATID_ActiveScriptParse */

     return TRUE;
 }

 static BOOL get_script_guid(nsIDOMHTMLScriptElement *nsscript, GUID *guid)
 {
     nsAString attr_str, val_str;
     BOOL ret = FALSE;
     nsresult nsres;

     static const PRUnichar languageW[] = {'l','a','n','g','u','a','g','e',0};

     nsAString_Init(&val_str, NULL);

     nsres = nsIDOMHTMLScriptElement_GetType(nsscript, &val_str);
     if(NS_SUCCEEDED(nsres)) {
         const PRUnichar *type;

         nsAString_GetData(&val_str, &type);
         if(*type) {
             ret = get_guid_from_type(type, guid);
             nsAString_Finish(&val_str);
             return ret;
         }
     }else {
         ERR("GetType failed: %08x\n", nsres);
     }

     nsAString_Init(&attr_str, languageW);

     nsres = nsIDOMHTMLScriptElement_GetAttribute(nsscript, &attr_str, &val_str);
     if(NS_SUCCEEDED(nsres)) {
         const PRUnichar *language;

         nsAString_GetData(&val_str, &language);

         if(*language) {
             ret = get_guid_from_language(language, guid);
         }else {
             *guid = CLSID_JScript;
             ret = TRUE;
         }
     }else {
         ERR("GetAttribute(language) failed: %08x\n", nsres);
     }

     nsAString_Finish(&attr_str);
     nsAString_Finish(&val_str);

     return ret;
 }

 static ScriptHost *get_script_host(HTMLDocument *doc, nsIDOMHTMLScriptElement *nsscript)
 {
     ScriptHost *iter;
     GUID guid;

     if(!get_script_guid(nsscript, &guid)) {
         WARN("Could not find script GUID\n");
         return NULL;
     }

     if(IsEqualGUID(&CLSID_JScript, &guid)) {
         FIXME("Ignoring JScript\n");
         return NULL;
     }

     LIST_FOR_EACH_ENTRY(iter, &doc->script_hosts, ScriptHost, entry) {
         if(IsEqualGUID(&guid, &iter->guid))
             return iter;
     }

     return create_script_host(doc, &guid);
 }

 void doc_insert_script(HTMLDocument *doc, nsIDOMHTMLScriptElement *nsscript)
 {
     ScriptHost *script_host;

     script_host = get_script_host(doc, nsscript);
     if(!script_host)
         return;

     if(script_host->parse)
         parse_script_elem(script_host, nsscript);
 }

 void release_script_hosts(HTMLDocument *doc)
 {
     ScriptHost *iter;

     while(!list_empty(&doc->script_hosts)) {
         iter = LIST_ENTRY(list_head(&doc->script_hosts), ScriptHost, entry);

         release_script_engine(iter);
         list_remove(&iter->entry);
         iter->doc = NULL;
         IActiveScript_Release(ACTSCPSITE(iter));
     }
 }
	/*
	* Copyright 2008 Jacek Caban for CodeWeavers
	*
	* This library is free software; you can redistribute it and/or
	* modify it under the terms of the GNU Lesser General Public
	* License as published by the Free Software Foundation; either
	* version 2.1 of the License, or (at your option) any later version.
	*
	* This library is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	* Lesser General Public License for more details.
	*
	* You should have received a copy of the GNU Lesser General Public
	* License along with this library; if not, write to the Free Software
	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
	*/

	#include "config.h"

	#include <stdarg.h>

	#define COBJMACROS

	#include "windef.h"
	#include "winbase.h"
	#include "winuser.h"
	#include "ole2.h"
	#include "activscp.h"
	#include "objsafe.h"

	#include "wine/debug.h"

	#include "mshtml_private.h"

	WINE_DEFAULT_DEBUG_CHANNEL(mshtml);

	static const WCHAR windowW[] = {'w','i','n','d','o','w',0};

	static const CLSID CLSID_JScript =
	{0xf414c260,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};

	typedef struct {
	const IActiveScriptSiteVtbl *lpIActiveScriptSiteVtbl;
	const IActiveScriptSiteInterruptPollVtbl *lpIActiveScriptSiteInterruptPollVtbl;
	const IActiveScriptSiteWindowVtbl *lpIActiveScriptSiteWindowVtbl;

	LONG ref;

	IActiveScript *script;
	IActiveScriptParse *parse;

	SCRIPTSTATE script_state;

	HTMLDocument *doc;

	GUID guid;
	struct list entry;
	} ScriptHost;

	#define ACTSCPSITE(x) ((IActiveScriptSite*) &(x)->lpIActiveScriptSiteVtbl)
	#define ACTSCPPOLL(x) ((IActiveScriptSiteInterruptPoll*) &(x)->lpIActiveScriptSiteInterruptPollVtbl)
	#define ACTSCPWIN(x) ((IActiveScriptSiteWindow*) &(x)->lpIActiveScriptSiteWindowVtbl)

	static BOOL init_script_engine(ScriptHost *script_host)
	{
	IActiveScriptProperty *property;
	IObjectSafety *safety;
	SCRIPTSTATE state;
	DWORD supported_opts=0, enabled_opts=0;
	HRESULT hres;

	hres = IActiveScript_QueryInterface(script_host->script, &IID_IActiveScriptParse, (void**)&script_host->parse);
	if(FAILED(hres)) {
	WARN("Could not get IActiveScriptHost: %08x\n", hres);
	return FALSE;
	}

	hres = IActiveScript_QueryInterface(script_host->script, &IID_IObjectSafety, (void**)&safety);
	if(FAILED(hres)) {
	FIXME("Could not get IObjectSafety: %08x\n", hres);
	return FALSE;
	}

	hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported_opts, &enabled_opts);
	if(FAILED(hres)) {
	FIXME("GetInterfaceSafetyOptions failed: %08x\n", hres);
	}else if(!(supported_opts & INTERFACE_USES_DISPEX)) {
	FIXME("INTERFACE_USES_DISPEX is not supported\n");
	}else {
	hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
	INTERFACESAFE_FOR_UNTRUSTED_DATA\|INTERFACE_USES_DISPEX\|INTERFACE_USES_SECURITY_MANAGER,
	INTERFACESAFE_FOR_UNTRUSTED_DATA\|INTERFACE_USES_DISPEX\|INTERFACE_USES_SECURITY_MANAGER);
	if(FAILED(hres))
	FIXME("SetInterfaceSafetyOptions failed: %08x\n", hres);
	}

	IObjectSafety_Release(safety);
	if(FAILED(hres))
	return FALSE;

	hres = IActiveScript_QueryInterface(script_host->script, &IID_IActiveScriptProperty, (void**)&property);
	if(SUCCEEDED(hres)) {
	VARIANT var;

	V_VT(&var) = VT_BOOL;
	V_BOOL(&var) = VARIANT_TRUE;
	hres = IActiveScriptProperty_SetProperty(property, SCRIPTPROP_HACK_TRIDENTEVENTSINK, NULL, &var);
	if(FAILED(hres))
	WARN("SetProperty failed: %08x\n", hres);

	IActiveScriptProperty_Release(property);
	}else {
	WARN("Could not get IActiveScriptProperty: %08x\n", hres);
	}

	hres = IActiveScriptParse_InitNew(script_host->parse);
	if(FAILED(hres)) {
	WARN("InitNew failed: %08x\n", hres);
	return FALSE;
	}

	hres = IActiveScript_SetScriptSite(script_host->script, ACTSCPSITE(script_host));
	if(FAILED(hres)) {
	WARN("SetScriptSite failed: %08x\n", hres);
	IActiveScript_Close(script_host->script);
	return FALSE;
	}

	hres = IActiveScript_GetScriptState(script_host->script, &state);
	if(FAILED(hres))
	WARN("GetScriptState failed: %08x\n", hres);
	else if(state != SCRIPTSTATE_INITIALIZED)
	FIXME("state = %x\n", state);

	hres = IActiveScript_SetScriptState(script_host->script, SCRIPTSTATE_STARTED);
	if(FAILED(hres)) {
	WARN("Starting script failed: %08x\n", hres);
	return FALSE;
	}

	hres = IActiveScript_AddNamedItem(script_host->script, windowW,
	SCRIPTITEM_ISVISIBLE\|SCRIPTITEM_ISSOURCE\|SCRIPTITEM_GLOBALMEMBERS);
	if(FAILED(hres))
	WARN("AddNamedItem failed: %08x\n", hres);

	/* FIXME: QI for IActiveScriptParseProcedure2 and IActiveScriptParseProcedure */

	return TRUE;
	}

	static void release_script_engine(ScriptHost *This)
	{
	if(!This->script)
	return;

	switch(This->script_state) {
	case SCRIPTSTATE_CONNECTED:
	IActiveScript_SetScriptState(This->script, SCRIPTSTATE_DISCONNECTED);

	case SCRIPTSTATE_STARTED:
	case SCRIPTSTATE_DISCONNECTED:
	case SCRIPTSTATE_INITIALIZED:
	IActiveScript_Close(This->script);

	default:
	if(This->parse) {
	IActiveScriptParse_Release(This->parse);
	This->parse = NULL;
	}
	}

	IActiveScript_Release(This->script);
	This->script = NULL;
	This->script_state = SCRIPTSTATE_UNINITIALIZED;
	}

	void connect_scripts(HTMLDocument *doc)
	{
	ScriptHost *iter;

	LIST_FOR_EACH_ENTRY(iter, &doc->script_hosts, ScriptHost, entry) {
	if(iter->script_state == SCRIPTSTATE_STARTED)
	IActiveScript_SetScriptState(iter->script, SCRIPTSTATE_CONNECTED);
	}
	}

	#define ACTSCPSITE_THIS(iface) DEFINE_THIS(ScriptHost, IActiveScriptSite, iface)

	static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite iface, REFIID riid, void *ppv)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);

	*ppv = NULL;

	if(IsEqualGUID(&IID_IUnknown, riid)) {
	TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
	*ppv = ACTSCPSITE(This);
	}else if(IsEqualGUID(&IID_IActiveScriptSite, riid)) {
	TRACE("(%p)->(IID_IActiveScriptSite %p)\n", This, ppv);
	*ppv = ACTSCPSITE(This);
	}else if(IsEqualGUID(&IID_IActiveScriptSiteInterruptPoll, riid)) {
	TRACE("(%p)->(IID_IActiveScriptSiteInterruprtPoll %p)\n", This, ppv);
	*ppv = ACTSCPPOLL(This);
	}else if(IsEqualGUID(&IID_IActiveScriptSiteWindow, riid)) {
	TRACE("(%p)->(IID_IActiveScriptSiteWindow %p)\n", This, ppv);
	*ppv = ACTSCPWIN(This);
	}else {
	FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
	return E_NOINTERFACE;
	}

	IUnknown_AddRef((IUnknown)ppv);
	return S_OK;
	}

	static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);
	LONG ref = InterlockedIncrement(&This->ref);

	TRACE("(%p) ref=%d\n", This, ref);

	return ref;
	}

	static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);
	LONG ref = InterlockedDecrement(&This->ref);

	TRACE("(%p) ref=%d\n", This, ref);

	if(!ref) {
	release_script_engine(This);
	if(This->doc)
	list_remove(&This->entry);
	heap_free(This);
	}

	return ref;
	}

	static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite iface, LCID plcid)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);

	TRACE("(%p)->(%p)\n", This, plcid);

	*plcid = GetUserDefaultLCID();
	return S_OK;
	}

	static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
	DWORD dwReturnMask, IUnknown ppiunkItem, ITypeInfo ppti)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);

	TRACE("(%p)->(%s %x %p %p)\n", This, debugstr_w(pstrName), dwReturnMask, ppiunkItem, ppti);

	if(dwReturnMask != SCRIPTINFO_IUNKNOWN) {
	FIXME("Unsupported mask %x\n", dwReturnMask);
	return E_NOTIMPL;
	}

	*ppiunkItem = NULL;

	if(strcmpW(pstrName, windowW))
	return DISP_E_MEMBERNOTFOUND;

	if(!This->doc)
	return E_FAIL;

	/* FIXME: Return proxy object */
	ppiunkItem = (IUnknown)HTMLWINDOW2(This->doc->window);
	IUnknown_AddRef(*ppiunkItem);

	return S_OK;
	}

	static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite iface, BSTR pbstrVersion)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);
	FIXME("(%p)->(%p)\n", This, pbstrVersion);
	return E_NOTIMPL;
	}

	static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
	const VARIANT pvarResult, const EXCEPINFO pexcepinfo)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);
	FIXME("(%p)->(%p %p)\n", This, pvarResult, pexcepinfo);
	return E_NOTIMPL;
	}

	static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);

	TRACE("(%p)->(%x)\n", This, ssScriptState);

	This->script_state = ssScriptState;
	return S_OK;
	}

	static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite iface, IActiveScriptError pscripterror)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);
	FIXME("(%p)->(%p)\n", This, pscripterror);
	return E_NOTIMPL;
	}

	static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);

	TRACE("(%p)->()\n", This);

	return S_OK;
	}

	static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
	{
	ScriptHost *This = ACTSCPSITE_THIS(iface);

	TRACE("(%p)->()\n", This);

	return S_OK;
	}

	#undef ACTSCPSITE_THIS

	static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
	ActiveScriptSite_QueryInterface,
	ActiveScriptSite_AddRef,
	ActiveScriptSite_Release,
	ActiveScriptSite_GetLCID,
	ActiveScriptSite_GetItemInfo,
	ActiveScriptSite_GetDocVersionString,
	ActiveScriptSite_OnScriptTerminate,
	ActiveScriptSite_OnStateChange,
	ActiveScriptSite_OnScriptError,
	ActiveScriptSite_OnEnterScript,
	ActiveScriptSite_OnLeaveScript
	};

	#define ACTSCPPOLL_THIS(iface) DEFINE_THIS(ScriptHost, IActiveScriptSiteInterruptPoll, iface)

	static HRESULT WINAPI ActiveScriptSiteInterruptPoll_QueryInterface(IActiveScriptSiteInterruptPoll *iface,
	REFIID riid, void **ppv)
	{
	ScriptHost *This = ACTSCPPOLL_THIS(iface);
	return IActiveScriptSite_QueryInterface(ACTSCPSITE(This), riid, ppv);
	}

	static ULONG WINAPI ActiveScriptSiteInterruptPoll_AddRef(IActiveScriptSiteInterruptPoll *iface)
	{
	ScriptHost *This = ACTSCPPOLL_THIS(iface);
	return IActiveScriptSite_AddRef(ACTSCPSITE(This));
	}

	static ULONG WINAPI ActiveScriptSiteInterruptPoll_Release(IActiveScriptSiteInterruptPoll *iface)
	{
	ScriptHost *This = ACTSCPPOLL_THIS(iface);
	return IActiveScriptSite_Release(ACTSCPSITE(This));
	}

	static HRESULT WINAPI ActiveScriptSiteInterruptPoll_QueryContinue(IActiveScriptSiteInterruptPoll *iface)
	{
	ScriptHost *This = ACTSCPPOLL_THIS(iface);
	FIXME("(%p)\n", This);
	return E_NOTIMPL;
	}

	#undef ACTSCPPOLL_THIS

	static const IActiveScriptSiteInterruptPollVtbl ActiveScriptSiteInterruptPollVtbl = {
	ActiveScriptSiteInterruptPoll_QueryInterface,
	ActiveScriptSiteInterruptPoll_AddRef,
	ActiveScriptSiteInterruptPoll_Release,
	ActiveScriptSiteInterruptPoll_QueryContinue
	};

	#define ACTSCPWIN_THIS(iface) DEFINE_THIS(ScriptHost, IActiveScriptSiteWindow, iface)

	static HRESULT WINAPI ActiveScriptSiteWindow_QueryInterface(IActiveScriptSiteWindow *iface,
	REFIID riid, void **ppv)
	{
	ScriptHost *This = ACTSCPWIN_THIS(iface);
	return IActiveScriptSite_QueryInterface(ACTSCPSITE(This), riid, ppv);
	}

	static ULONG WINAPI ActiveScriptSiteWindow_AddRef(IActiveScriptSiteWindow *iface)
	{
	ScriptHost *This = ACTSCPWIN_THIS(iface);
	return IActiveScriptSite_AddRef(ACTSCPSITE(This));
	}

	static ULONG WINAPI ActiveScriptSiteWindow_Release(IActiveScriptSiteWindow *iface)
	{
	ScriptHost *This = ACTSCPWIN_THIS(iface);
	return IActiveScriptSite_Release(ACTSCPSITE(This));
	}

	static HRESULT WINAPI ActiveScriptSiteWindow_GetWindow(IActiveScriptSiteWindow iface, HWND phwnd)
	{
	ScriptHost *This = ACTSCPWIN_THIS(iface);
	FIXME("(%p)->(%p)\n", This, phwnd);
	return E_NOTIMPL;
	}

	static HRESULT WINAPI ActiveScriptSiteWindow_EnableModeless(IActiveScriptSiteWindow *iface, BOOL fEnable)
	{
	ScriptHost *This = ACTSCPWIN_THIS(iface);
	FIXME("(%p)->(%x)\n", This, fEnable);
	return E_NOTIMPL;
	}

	#undef ACTSCPWIN_THIS

	static const IActiveScriptSiteWindowVtbl ActiveScriptSiteWindowVtbl = {
	ActiveScriptSiteWindow_QueryInterface,
	ActiveScriptSiteWindow_AddRef,
	ActiveScriptSiteWindow_Release,
	ActiveScriptSiteWindow_GetWindow,
	ActiveScriptSiteWindow_EnableModeless
	};

	static ScriptHost create_script_host(HTMLDocument doc, GUID *guid)
	{
	ScriptHost *ret;
	HRESULT hres;

	ret = heap_alloc_zero(sizeof(*ret));
	ret->lpIActiveScriptSiteVtbl = &ActiveScriptSiteVtbl;
	ret->lpIActiveScriptSiteInterruptPollVtbl = &ActiveScriptSiteInterruptPollVtbl;
	ret->lpIActiveScriptSiteWindowVtbl = &ActiveScriptSiteWindowVtbl;
	ret->ref = 1;
	ret->doc = doc;
	ret->script_state = SCRIPTSTATE_UNINITIALIZED;

	ret->guid = *guid;
	list_add_tail(&doc->script_hosts, &ret->entry);

	hres = CoCreateInstance(&ret->guid, NULL, CLSCTX_INPROC_SERVER\|CLSCTX_INPROC_HANDLER,
	&IID_IActiveScript, (void**)&ret->script);
	if(FAILED(hres))
	WARN("Could not load script engine: %08x\n", hres);
	else if(!init_script_engine(ret))
	release_script_engine(ret);

	return ret;
	}

	static void parse_text(ScriptHost *script_host, LPCWSTR text)
	{
	EXCEPINFO excepinfo;
	VARIANT var;
	HRESULT hres;

	static const WCHAR script_endW[] = {'<','/','S','C','R','I','P','T','>',0};

	TRACE("%s\n", debugstr_w(text));

	VariantInit(&var);
	memset(&excepinfo, 0, sizeof(excepinfo));
	hres = IActiveScriptParse_ParseScriptText(script_host->parse, text, windowW, NULL, script_endW,
	0, 0, SCRIPTTEXT_ISVISIBLE\|SCRIPTTEXT_HOSTMANAGESSOURCE,
	&var, &excepinfo);
	if(FAILED(hres))
	WARN("ParseScriptText failed: %08x\n", hres);

	}

	static void parse_extern_script(ScriptHost *script_host, LPCWSTR src)
	{
	IMoniker *mon;
	char *buf;
	WCHAR *text;
	HRESULT hres;

	static const WCHAR wine_schemaW[] = {'w','i','n','e',':'};

	if(strlenW(src) > sizeof(wine_schemaW)/sizeof(WCHAR) && !memcmp(src, wine_schemaW, sizeof(wine_schemaW)))
	src += sizeof(wine_schemaW)/sizeof(WCHAR);

	hres = CreateURLMoniker(NULL, src, &mon);
	if(FAILED(hres))
	return;

	hres = bind_mon_to_buffer(script_host->doc, mon, (void**)&buf);
	IMoniker_Release(mon);
	if(FAILED(hres))
	return;

	text = heap_strdupAtoW(buf);
	heap_free(buf);

	parse_text(script_host, text);

	heap_free(text);
	}

	static void parse_inline_script(ScriptHost script_host, nsIDOMHTMLScriptElement nsscript)
	{
	const PRUnichar *text;
	nsAString text_str;
	nsresult nsres;

	nsAString_Init(&text_str, NULL);

	nsres = nsIDOMHTMLScriptElement_GetText(nsscript, &text_str);

	if(NS_SUCCEEDED(nsres)) {
	nsAString_GetData(&text_str, &text);
	parse_text(script_host, text);
	}else {
	ERR("GetText failed: %08x\n", nsres);
	}

	nsAString_Finish(&text_str);
	}

	static void parse_script_elem(ScriptHost script_host, nsIDOMHTMLScriptElement nsscript)
	{
	const PRUnichar *src;
	nsAString src_str;
	nsresult nsres;

	nsAString_Init(&src_str, NULL);

	nsres = nsIDOMHTMLScriptElement_GetSrc(nsscript, &src_str);
	nsAString_GetData(&src_str, &src);

	if(NS_FAILED(nsres))
	ERR("GetSrc failed: %08x\n", nsres);
	else if(*src)
	parse_extern_script(script_host, src);
	else
	parse_inline_script(script_host, nsscript);

	nsAString_Finish(&src_str);
	}

	static BOOL get_guid_from_type(LPCWSTR type, GUID *guid)
	{
	const WCHAR text_javascriptW[] =
	{'t','e','x','t','/','j','a','v','a','s','c','r','i','p','t',0};

	/* FIXME: Handle more types */
	if(!strcmpW(type, text_javascriptW)) {
	*guid = CLSID_JScript;
	}else {
	FIXME("Unknown type %s\n", debugstr_w(type));
	return FALSE;
	}

	return TRUE;
	}

	static BOOL get_guid_from_language(LPCWSTR type, GUID *guid)
	{
	HRESULT hres;

	hres = CLSIDFromProgID(type, guid);
	if(FAILED(hres))
	return FALSE;

	/* FIXME: Check CATID_ActiveScriptParse */

	return TRUE;
	}

	static BOOL get_script_guid(nsIDOMHTMLScriptElement nsscript, GUID guid)
	{
	nsAString attr_str, val_str;
	BOOL ret = FALSE;
	nsresult nsres;

	static const PRUnichar languageW[] = {'l','a','n','g','u','a','g','e',0};

	nsAString_Init(&val_str, NULL);

	nsres = nsIDOMHTMLScriptElement_GetType(nsscript, &val_str);
	if(NS_SUCCEEDED(nsres)) {
	const PRUnichar *type;

	nsAString_GetData(&val_str, &type);
	if(*type) {
	ret = get_guid_from_type(type, guid);
	nsAString_Finish(&val_str);
	return ret;
	}
	}else {
	ERR("GetType failed: %08x\n", nsres);
	}

	nsAString_Init(&attr_str, languageW);

	nsres = nsIDOMHTMLScriptElement_GetAttribute(nsscript, &attr_str, &val_str);
	if(NS_SUCCEEDED(nsres)) {
	const PRUnichar *language;

	nsAString_GetData(&val_str, &language);

	if(*language) {
	ret = get_guid_from_language(language, guid);
	}else {
	*guid = CLSID_JScript;
	ret = TRUE;
	}
	}else {
	ERR("GetAttribute(language) failed: %08x\n", nsres);
	}

	nsAString_Finish(&attr_str);
	nsAString_Finish(&val_str);

	return ret;
	}

	static ScriptHost get_script_host(HTMLDocument doc, nsIDOMHTMLScriptElement *nsscript)
	{
	ScriptHost *iter;
	GUID guid;

	if(!get_script_guid(nsscript, &guid)) {
	WARN("Could not find script GUID\n");
	return NULL;
	}

	if(IsEqualGUID(&CLSID_JScript, &guid)) {
	FIXME("Ignoring JScript\n");
	return NULL;
	}

	LIST_FOR_EACH_ENTRY(iter, &doc->script_hosts, ScriptHost, entry) {
	if(IsEqualGUID(&guid, &iter->guid))
	return iter;
	}

	return create_script_host(doc, &guid);
	}

	void doc_insert_script(HTMLDocument doc, nsIDOMHTMLScriptElement nsscript)
	{
	ScriptHost *script_host;

	script_host = get_script_host(doc, nsscript);
	if(!script_host)
	return;

	if(script_host->parse)
	parse_script_elem(script_host, nsscript);
	}

	void release_script_hosts(HTMLDocument *doc)
	{
	ScriptHost *iter;

	while(!list_empty(&doc->script_hosts)) {
	iter = LIST_ENTRY(list_head(&doc->script_hosts), ScriptHost, entry);

	release_script_engine(iter);
	list_remove(&iter->entry);
	iter->doc = NULL;
	IActiveScript_Release(ACTSCPSITE(iter));
	}
	}