#include <wininetp.h>
#include <cscpsite.h>
#include <objsafe.h>
#include "jsproxy.h"
#include "utils.h"


/*******************************************************************************
*    CScriptSite Functions
********************************************************************************/
CScriptSite::CScriptSite(CAutoProxy * pAutoProxy)
{
    m_refCount = 1;
    m_pios = NULL;
    m_pasp = NULL; 
    m_pScriptDispatch = NULL;
    m_Scriptdispid = -1;
    m_punkJSProxy = NULL;
    m_fInitialized = FALSE;
    m_dwScriptStartTime = 0;
    m_pAutoProxy = pAutoProxy;
}
CScriptSite::~CScriptSite()
{
    if (m_fInitialized)
        DeInit();
}

STDMETHODIMP CScriptSite::QueryInterface(REFIID riid, PVOID *ppvObject)
{
    if (riid == IID_IUnknown)
    {            
        *ppvObject = (LPVOID)(LPUNKNOWN)static_cast<IActiveScriptSite *>(this);
    }
    else if (riid == IID_IActiveScriptSite)
    {
        *ppvObject = (LPVOID)static_cast<IActiveScriptSite *>(this);
    }
    else if (riid == IID_IServiceProvider)
    {
        *ppvObject = (LPVOID)static_cast<IServiceProvider *>(this);
    }
    else if (riid == IID_IInternetHostSecurityManager)
    {
        *ppvObject = (LPVOID)static_cast<IInternetHostSecurityManager *>(this);
    }
    else if (riid == IID_IActiveScriptSiteInterruptPoll)
    {
        *ppvObject = (LPVOID)static_cast<IActiveScriptSiteInterruptPoll *>(this);
    }
    else
    {
        *ppvObject = 0;
        return E_NOINTERFACE;
    }

    AddRef();
    return S_OK;
}

STDMETHODIMP CScriptSite::Init(AUTO_PROXY_HELPER_APIS* pAPHA, LPCSTR szScript)
{
    CHAR szClassId[64];    
    CLSID clsid;
    HRESULT hr = S_OK;
    BSTR    bstrClsID = NULL;
    BSTR    bstrScriptText = NULL;
    OLECHAR *    rgwszNames[1] = {L"FindProxyForURL"};
    EXCEPINFO    exceptinfo;
    IObjectSafety * pIObjSafety = NULL;

    // pAPHA can be null - it is checked in the autoproxy object!
    if (!szScript)
        return E_POINTER;

    if (m_fInitialized)
        return hr;
    // CoCreateInstance the JScript engine.

    if(!DelayLoad(&g_moduleOle32)
       || !DelayLoad(&g_moduleOleAut32))
    {
        hr = E_OUTOFMEMORY;
        goto exit;
    }

    // Get the class id of the desired language engine
    hr = GetScriptEngineClassIDFromName(
        "JavaScript",
        szClassId,
        sizeof(szClassId)
        );
    if (FAILED(hr)) {
        return E_FAIL;
    }
    //convert CLSID string to clsid

    bstrClsID = BSTRFROMANSI(szClassId);
    if (!bstrClsID)
        goto exit;
    hr = DL(CLSIDFromString)(bstrClsID, &clsid);
    DL(SysFreeString)(bstrClsID);
    if (FAILED(hr))
        goto exit;

    // Instantiate the script engine
    hr = DL(CoCreateInstance)(clsid, NULL, CLSCTX_INPROC_SERVER, IID_IActiveScript, (void**)&m_pios);
    if (FAILED(hr))
        goto exit;

    // Get the IActiveScriptParse interface, if any
    hr = m_pios->QueryInterface(IID_IActiveScriptParse, (void**) &m_pasp);
    if (FAILED(hr))
        goto exit;

    hr = m_pasp->InitNew();
    if (FAILED(hr))
        goto exit;

    // SetScriptSite to this
    hr = m_pios->SetScriptSite((IActiveScriptSite *)this);
    if (FAILED(hr))
        goto exit;
    hr = m_pios->SetScriptState(SCRIPTSTATE_INITIALIZED);

    //
    // Inform the script engine that this host implements
    // the IInternetHostSecurityManager interface, which
    // is used to prevent the script code from using any
    // ActiveX objects.
    //
    hr = m_pios->QueryInterface(IID_IObjectSafety, (void **)&pIObjSafety);

    if (SUCCEEDED(hr) && (pIObjSafety != NULL))
    {
        pIObjSafety->SetInterfaceSafetyOptions(IID_NULL, 
                INTERFACE_USES_SECURITY_MANAGER,
                INTERFACE_USES_SECURITY_MANAGER);

        pIObjSafety->Release();
        pIObjSafety = NULL;
    }

    // AddNamedItem for pUnk and set m_punkJSProxy to pUnk.
    // If we added JSProxy to the name space the store away the JSProxy objects punk.
    m_punkJSProxy = new CJSProxy;
    if( !m_punkJSProxy )
    {
        hr = E_OUTOFMEMORY;
        goto exit;
    }
    m_punkJSProxy->Init(pAPHA);
    hr = m_pios->AddNamedItem(L"JSProxy",SCRIPTITEM_ISVISIBLE | SCRIPTITEM_GLOBALMEMBERS);
    if (FAILED(hr))
        goto exit;
    
    // Convert the ANSI script text to a bstr.
    bstrScriptText = BSTRFROMANSI(szScript);
    if (!bstrScriptText)
    {
        hr = E_OUTOFMEMORY;
        goto exit;
    }
    // Add the script text to the parser
    hr = m_pasp->ParseScriptText(
             bstrScriptText,
             NULL,
             NULL,
             NULL,
             0,
             0,
             SCRIPTTEXT_ISEXPRESSION|SCRIPTTEXT_ISVISIBLE,
             NULL,
             &exceptinfo);
    
    DL(SysFreeString)(bstrScriptText);
    if (FAILED(hr))
        goto exit;

    m_dwScriptStartTime = GetTickCount();

    hr = m_pios->SetScriptState(SCRIPTSTATE_STARTED);
    if (FAILED(hr))
        goto exit;
    // Now get the script dispatch and find the DISPID for the method just added.  since this is a single use dll
    // I can do this otherwise this would be bad.
    hr = m_pios->GetScriptDispatch(NULL,&m_pScriptDispatch);
    if (FAILED(hr))
        goto exit;
    hr = m_pScriptDispatch->GetIDsOfNames(IID_NULL,rgwszNames,1,LOCALE_SYSTEM_DEFAULT,&m_Scriptdispid);
    if (FAILED(hr))
        goto exit;

    m_fInitialized = TRUE;

    return hr;

exit: // we come here if something fails  -  release everything and set to null.
    if (m_pasp)
        m_pasp->Release();
    if (m_pScriptDispatch)
        m_pScriptDispatch->Release();
    if (m_pios)
    {
        m_pios->Close();
        m_pios->Release();
    }
    if (m_punkJSProxy)
        m_punkJSProxy->Release();
    m_pios = NULL;
    m_pasp = NULL;
    m_pScriptDispatch = NULL;
    m_Scriptdispid = -1;
    m_punkJSProxy = NULL;
    
    return hr;
}

STDMETHODIMP CScriptSite::DeInit()
{
    HRESULT hr = S_OK;

    if (m_pasp)
        m_pasp->Release();

    if (m_pScriptDispatch)
        m_pScriptDispatch->Release();

    if (m_pios)
    {
        hr = m_pios->Close();
        m_pios->Release();
    }

    if (m_punkJSProxy)
        m_punkJSProxy->Release();

    m_pios = NULL;
    m_pasp = NULL;
    m_pScriptDispatch = NULL;
    m_Scriptdispid = -1;
    m_fInitialized = FALSE;

    return hr;
}

STDMETHODIMP CScriptSite::RunScript(LPCSTR szURL, LPCSTR szHost, LPSTR* result)
{
    HRESULT        hr = S_OK;
    UINT        puArgErr = 0;
    EXCEPINFO    excep;
    VARIANT        varresult;
    DISPPARAMS    dispparams;
    VARIANT        args[2]; // We always call with 2 args!

    
    if (!szURL || !szHost || !result)
        return E_POINTER;

    if (!m_fInitialized)
        return E_UNEXPECTED;
        
    DL(VariantInit)(&varresult);

    *result = NULL;

    dispparams.cArgs = 2;
    DL(VariantInit)(&args[0]);
    DL(VariantInit)(&args[1]);

    args[0].vt = VT_BSTR;
    args[1].vt = VT_BSTR;

    args[0].bstrVal = BSTRFROMANSI(szHost);
    args[1].bstrVal = BSTRFROMANSI(szURL);

    if (args[0].bstrVal==NULL || args[1].bstrVal==NULL)
    {
        hr = E_OUTOFMEMORY;
        goto Cleanup;
    }

    dispparams.rgvarg = args;

    dispparams.cNamedArgs = 0;    
    dispparams.rgdispidNamedArgs = NULL; 

    // Call invoke on the stored dispid
    hr = m_pScriptDispatch->Invoke(m_Scriptdispid,
                    IID_NULL,LOCALE_SYSTEM_DEFAULT,
                    DISPATCH_METHOD,
                    &dispparams,
                    &varresult,
                    &excep,
                    &puArgErr);

    if (FAILED(hr))
        goto Cleanup;

    // convert result into bstr and return ansi version of the string!
    if (varresult.vt == VT_BSTR)
    {
        MAKE_ANSIPTR_FROMWIDE(rescpy, varresult.bstrVal);
        *result = (LPSTR) GlobalAlloc(GPTR|GMEM_ZEROINIT,lstrlen(rescpy)+1);
        if (!*result)
        {
            hr = E_OUTOFMEMORY;
            goto Cleanup;
        }
        lstrcpy(*result,rescpy);

    }
    else
    {
        VARIANT    resvar;

        DL(VariantInit)(&resvar);
        hr = DL(VariantChangeType)(&resvar,&varresult,NULL,VT_BSTR);
        if (SUCCEEDED(hr))
        {
            MAKE_ANSIPTR_FROMWIDE(rescpy, resvar.bstrVal);
            *result = (LPSTR) GlobalAlloc(GPTR|GMEM_ZEROINIT,lstrlen(rescpy)+1);
            if (!*result)
            {
                hr = E_OUTOFMEMORY;
                DL(VariantClear)(&resvar);
                goto Cleanup;
            }
            lstrcpy(*result,rescpy);
        }
        else
            *result = NULL;
        DL(VariantClear)(&resvar);
    }

Cleanup:
    DL(VariantClear)(&varresult);
    DL(VariantClear)(&args[0]);
    DL(VariantClear)(&args[1]);
    
    return hr;
}

STDMETHODIMP CScriptSite::GetLCID(LCID *plcid)
{
    UNREFERENCED_PARAMETER(plcid);
    return E_NOTIMPL;
}
STDMETHODIMP CScriptSite::GetItemInfo(LPCOLESTR pstrName, DWORD dwReturnMask, IUnknown **ppunkItem, ITypeInfo **ppTypeInfo)
{
    UNREFERENCED_PARAMETER(ppTypeInfo);
    if (!pstrName || !ppunkItem)
        return E_POINTER;

    if ((StrCmpW(L"JSProxy",pstrName) == 0) && (dwReturnMask == SCRIPTINFO_IUNKNOWN))
    {
        *ppunkItem = (LPUNKNOWN)(IDispatch*)(CJSProxy*)m_punkJSProxy;
        (*ppunkItem)->AddRef();
        return S_OK;
    }
    else
        return TYPE_E_ELEMENTNOTFOUND;
}
STDMETHODIMP CScriptSite::GetDocVersionString(BSTR *pstrVersionString)
{
    UNREFERENCED_PARAMETER(pstrVersionString);
    return E_NOTIMPL;
}

// I am not interested it the transitioning of state or the status of where we are in
// the executing of the script.
STDMETHODIMP CScriptSite::OnScriptTerminate(const VARIANT *pvarResult,const EXCEPINFO *pexcepinfo)
{
    UNREFERENCED_PARAMETER(pvarResult);
    UNREFERENCED_PARAMETER(pexcepinfo);
    return S_OK;
}
STDMETHODIMP CScriptSite::OnStateChange(SCRIPTSTATE ssScriptState)
{
     UNREFERENCED_PARAMETER(ssScriptState);
     return S_OK;
}
STDMETHODIMP CScriptSite::OnScriptError(IActiveScriptError *pase)
{
     UNREFERENCED_PARAMETER(pase);
     return S_OK;
}
STDMETHODIMP CScriptSite::OnEnterScript()
{
    return S_OK;
}
STDMETHODIMP CScriptSite::OnLeaveScript()
{
    return S_OK;
}

//
// IServiceProvider
//
//      Implemented to help wire up the script engine with our
//      IInternetHostSecurityManager interface.
//

STDMETHODIMP CScriptSite::QueryService(
    REFGUID guidService,
    REFIID  riid,
    void ** ppvObject)
{
    if (guidService == SID_SInternetHostSecurityManager)
    {
        return QueryInterface(riid, ppvObject);
    }
    else
    {
        return E_NOINTERFACE;
    }
}


//
// IActiveScriptSiteInterruptPoll
//
STDMETHODIMP CScriptSite::QueryContinue(void)
{
    if (m_pAutoProxy->IsSessionAborted())
        return E_ABORT;
    
    if (GetTickCount() > (m_dwScriptStartTime + MAX_PAC_SCRIPT_RUN_TIMEOUT))
        return E_ABORT;

    return NOERROR;
}

//
// IInternetHostSecurityManager
// 
//      Implemented to prevent the script code from using ActiveX objects.
//

STDMETHODIMP CScriptSite::GetSecurityId( 
    BYTE *      pbSecurityId,
    DWORD *     pcbSecurityId,
    DWORD_PTR   dwReserved)
{
    UNREFERENCED_PARAMETER(pbSecurityId);
    UNREFERENCED_PARAMETER(pcbSecurityId);
    UNREFERENCED_PARAMETER(dwReserved);

    return E_NOTIMPL;
}


STDMETHODIMP CScriptSite::ProcessUrlAction( 
    DWORD   dwAction,
    BYTE *  pPolicy,
    DWORD   cbPolicy,
    BYTE *  pContext,
    DWORD   cbContext,
    DWORD   dwFlags,
    DWORD   dwReserved)
{
    UNREFERENCED_PARAMETER(dwAction);
    UNREFERENCED_PARAMETER(pContext);
    UNREFERENCED_PARAMETER(cbContext);
    UNREFERENCED_PARAMETER(dwFlags);
    UNREFERENCED_PARAMETER(dwReserved);

    //
    // Deny the script any capabilites. In particular, this
    // will disallow the script code from instantiating 
    // ActiveX objects.
    //

    if (cbPolicy == sizeof(DWORD))
    {
        *(DWORD *)pPolicy = URLPOLICY_DISALLOW;
    }

    return S_FALSE; // S_FALSE means the policy != URLPOLICY_ALLOW.
}


STDMETHODIMP CScriptSite::QueryCustomPolicy( 
    REFGUID guidKey,
    BYTE ** ppPolicy,
    DWORD * pcbPolicy,
    BYTE *  pContext,
    DWORD   cbContext,
    DWORD   dwReserved)
{
    UNREFERENCED_PARAMETER(guidKey);
    UNREFERENCED_PARAMETER(ppPolicy);
    UNREFERENCED_PARAMETER(pcbPolicy);
    UNREFERENCED_PARAMETER(pContext);
    UNREFERENCED_PARAMETER(cbContext);
    UNREFERENCED_PARAMETER(dwReserved);

    return E_NOTIMPL;
}