//  Microsoft Windows
//  Copyright (C) Microsoft Corporation, 1992 - 1995.

//  File:       transact.cxx

//  Contents:   Class that performs the download of a particular request.

//  Classes:

//  Functions:

//  History:    12-04-95   JohannP (Johann Posch)   Created


#include <trans.h>
#include <shlwapi.h>

#include "oinet.hxx"

PerfDbgTag(tagCTransaction,    "Urlmon", "Log CTransaction",        DEB_TRANS);
    DbgTag(tagCTransactionErr, "Urlmon", "Log CTransaction Errors", DEB_TRANS|DEB_ERROR);

HRESULT GetClassDocFileBuffer(LPVOID pbuffer, DWORD dwSize, CLSID *pclsid);
HRESULT GetCodeBaseFromDocFile(LPBYTE pBuffer, ULONG ulSize, LPWSTR *pwzClassStr,
                               LPWSTR pwzBaseUrl, DWORD *lpdwVersionMS, DWORD *lpdwVersionLS);
HRESULT IsHandlerAvailable(LPWSTR pwzUrl, LPWSTR pwzMime, CLSID *pclsid,
                                LPBYTE pBuffer, ULONG cbSize);

#if 0


//  Method:     COInetProtSnk::QueryInterface

//  Synopsis:

//  Arguments:  [riid] --
//              [ppvObj] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProtSnk::QueryInterface(REFIID riid, void **ppvObj)
{
    VDATEPTROUT(ppvObj, void *);
    VDATETHIS(this);
    HRESULT hr = NOERROR;

    PerfDbgLog(tagCTransaction, this, "+COInetProtSnk::QueryInterface");

    *ppvObj = NULL;
    if (_pUnk)
    {
        hr = _pUnk->QueryInterface(riid, ppvObj);
    }
    else
    {
        if (   (riid == IID_IUnknown)
            || (riid == IID_IOInetProtocol))
        {
            *ppvObj = (IOInetProtocol *) this;
            AddRef();
        }
        else if (riid == IID_IOInetProtocolSink)
        {
            *ppvObj = (IOInetProtocolSink *) this;
            AddRef();
        }
        else
        {
            hr = E_NOINTERFACE;
        }
    }

    PerfDbgLog1(tagCTransaction, this, "-COInetProtSnk::QueryInterface (hr:%lx)", hr);
    return hr;
}



//  Function:   COInetProtSnk::AddRef

//  Synopsis:

//  Arguments:  [ULONG] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP_(ULONG) COInetProtSnk::AddRef(void)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProtSnk::AddRef");

    LONG lRet = _pProtSnk->AddRef();

    PerfDbgLog1(tagCTransaction, this, "-COInetProtSnk::AddRef (cRefs:%ld)", lRet);
    return lRet;
}



//  Function:   COInetProtSnk::Release

//  Synopsis:

//  Arguments:  [ULONG] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP_(ULONG) COInetProtSnk::Release(void)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProtSnk::Release");

    LONG lRet = _pProtSnk->Release();

    PerfDbgLog1(tagCTransaction, this, "-COInetProtSnk::Release (cRefs:%ld)", lRet);
    return lRet;
}




//  Method:     COInetProtSnk::Switch

//  Synopsis:

//  Arguments:  [pStateInfo] --

//  Returns:

//  History:    11-07-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProtSnk::Switch(PROTOCOLDATA *pStateInfo)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProtSnk::Switch");
    HRESULT hr = NOERROR;

    hr = _pProtSnk->Switch(pStateInfo);

    PerfDbgLog1(tagCTransaction, this, "-COInetProtSnk::Switch (hr:%lx)", hr);
    return hr;
}



//  Method:     COInetProtSnk::ReportProgress

//  Synopsis:

//  Arguments:  [NotMsg] --
//              [szStatusText] --

//  Returns:

//  History:    11-07-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProtSnk::ReportProgress(ULONG NotMsg, LPCWSTR pwzStatusText)
{
    PerfDbgLog1(tagCTransaction, this, "+COInetProtSnk::ReportProgress pwzStatusText:%ws", pwzStatusText);
    HRESULT hr = NOERROR;

    hr = _pProtSnk->ReportProgress(NotMsg, pwzStatusText);

    PerfDbgLog2(tagCTransaction, this, "-COInetProtSnk::ReportProgress (pwzStatusText:%ws, hr:%lx)", pwzStatusText, hr);
    return hr;
}



//  Method:     COInetProtSnk::ReportData

//  Synopsis:

//  Arguments:  [grfBSCF] --
//              [ULONG] --
//              [ulProgressMax] --

//  Returns:

//  History:    11-07-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProtSnk::ReportData(DWORD grfBSCF, ULONG ulProgress,ULONG ulProgressMax)
{
    PerfDbgLog3(tagCTransaction, this, "+COInetProtSnk::ReportData(grfBSCF:%lx, ulProgress:%ld, ulProgressMax:%ld)",
                                       grfBSCF, ulProgress, ulProgressMax);
    HRESULT hr = NOERROR;

    hr = _pProtSnk->ReportData( grfBSCF, ulProgress, ulProgressMax);

    PerfDbgLog1(tagCTransaction, this, "-COInetProtSnk::ReportData (hr:%lx)", hr);
    return hr;
}



//  Method:     COInetProtSnk::ReportResult

//  Synopsis:

//  Arguments:  [DWORD] --
//              [dwError] --
//              [wzResult] --

//  Returns:

//  History:    11-07-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProtSnk::ReportResult(HRESULT hrResult, DWORD dwError, LPCWSTR wzResult)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProtSnk::ReportResult");
    HRESULT hr = NOERROR;

    hr = _pProtSnk->ReportResult(hrResult, dwError, wzResult);

    PerfDbgLog1(tagCTransaction, this, "-COInetProtSnk::ReportResult (hr:%lx)", hr);
    return hr;
}
#endif // 0


//  Method:     COInetProt::QueryInterface

//  Synopsis:

//  Arguments:  [riid] --
//              [ppvObj] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::QueryInterface(REFIID riid, void **ppvObj)
{
    VDATEPTROUT(ppvObj, void *);
    VDATETHIS(this);
    HRESULT hr = NOERROR;

    PerfDbgLog(tagCTransaction, this, "+COInetProt::QueryInterface");

    *ppvObj = NULL;

    if (_pUnk)
    {
        hr = _pUnk->QueryInterface(riid, ppvObj);
    }
    else
    {
        if (   (riid == IID_IUnknown)
            || (riid == IID_IOInetProtocol))
        {
            *ppvObj = (IOInetProtocol *) this;
            AddRef();
        }
        else if (riid == IID_IOInetProtocolSink)
        {
            *ppvObj = (IOInetProtocolSink *) this;
            AddRef();
        }
        else if (riid == IID_IServiceProvider)
        {
            *ppvObj = (IServiceProvider *) this;
            AddRef();
        }
        else if (riid == IID_IOInetPriority)
        {
            *ppvObj = (IOInetPriority *) this;
            AddRef();
        }
        else
        {
            hr = E_NOINTERFACE;
        }
    }


    PerfDbgLog1(tagCTransaction, this, "-COInetProt::QueryInterface (hr:%lx)", hr);
    return hr;
}



//  Function:   COInetProt::AddRef

//  Synopsis:

//  Arguments:  [ULONG] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP_(ULONG) COInetProt::AddRef(void)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::AddRef");

    LONG lRet;

    if (_pUnk)
    {
        lRet = _pUnk->AddRef();
    }
    else
    {
        lRet = ++_CRefs;
    }

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::AddRef (cRefs:%ld)", lRet);
    return lRet;
}



//  Function:   COInetProt::Release

//  Synopsis:

//  Arguments:  [ULONG] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP_(ULONG) COInetProt::Release(void)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::Release");

    LONG lRet;
    if (_pUnk)
    {
        lRet = _pUnk->Release();
    }
    else
    {
        lRet = --_CRefs;
        if (_CRefs == 0)
        {

            // release all objects
            if (_pProtSnk)
            {
                _pProtSnk->Release();
            }
            if (_pProt)
            {
                _pProt->Release();
            }

            if (_dwMode & PP_DELETE)
            {
                delete this;
            }
        }
    }

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::Release (cRefs:%ld)", lRet);
    return lRet;
}



//  Method:     COInetProt::Start

//  Synopsis:

//  Arguments:  [pwzUrl] --
//              [pTrans] --
//              [pOIBindInfo] --
//              [grfSTI] --
//              [dwReserved] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::Start(LPCWSTR pwzUrl, IOInetProtocolSink *pOInetProtSnk, IOInetBindInfo *pOIBindInfo,
                          DWORD grfSTI, DWORD_PTR dwReserved)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::Start\n");
    HRESULT hr = NOERROR;

    TransAssert((pOIBindInfo && pOInetProtSnk && pwzUrl));

    // Just before starting the transaction give it the priority.

    IOInetPriority * pOInetPriority = NULL;
    if (_pProt->QueryInterface(IID_IOInetPriority, (void **) &pOInetPriority) == S_OK)
    {
        pOInetPriority->SetPriority(_nPriority);
        pOInetPriority->Release();
    }

    delete [] _pwzUrl;
    _pwzUrl = OLESTRDuplicate(pwzUrl);

    hr = _pProt->Start(pwzUrl, pOInetProtSnk, pOIBindInfo, grfSTI, dwReserved);

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::Start (hr:%lx)\n", hr);
    return hr;
}



//  Method:     COInetProt::Continue

//  Synopsis:

//  Arguments:  [pStateInfoIn] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::Continue(PROTOCOLDATA *pStateInfoIn)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::Continue\n");

    HRESULT hr = _pProt->Continue(pStateInfoIn);

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::Continue (hr:%lx)\n",hr);
    return hr;
}




//  Method:     COInetProt::Abort

//  Synopsis:

//  Arguments:  [hrReason] --
//              [dwOptions] --

//  Returns:

//  History:    11-09-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::Abort(HRESULT hrReason, DWORD dwOptions)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::Abort\n");
    HRESULT hr = NOERROR;

    hr = _pProt->Abort(hrReason, dwOptions);

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::Abort (hr:%lx)\n", hr);
    return hr;
}



//  Method:     COInetProt::Terminate

//  Synopsis:

//  Arguments:  [dwOptions] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::Terminate(DWORD dwOptions)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::Terminate\n");
    HRESULT hr = NOERROR;

    TransAssert((_pProt));

    hr = _pProt->Terminate(dwOptions);
    SetProtocolSink(0);
    SetServiceProvider(0);

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::Terminate (hr:%lx)\n", hr);
    return hr;
}



//  Method:     COInetProt::Suspend

//  Synopsis:

//  Arguments:  (none)

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::Suspend()
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::Suspend\n");

    HRESULT hr = _pProt->Suspend();

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::Suspend (hr:%lx)\n", hr);
    return hr;
}



//  Method:     COInetProt::Resume

//  Synopsis:

//  Arguments:  (none)

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::Resume()
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::Resume\n");

    HRESULT hr = _pProt->Resume();

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::Resume (hr:%lx)\n", hr);
    return hr;
}



//  Method:     COInetProt::Read

//  Synopsis:

//  Arguments:  [ULONG] --
//              [ULONG] --
//              [pcbRead] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::Read(void *pBuffer, ULONG cbBuffer,ULONG *pcbRead)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::Read\n");
    HRESULT     hr = E_FAIL;

    BOOL fRead = TRUE;
    DWORD dwCopy = 0;
    DWORD dwCopyNew = 0;

    if (   (_dwOInetBdgFlags & (PI_MIMEVERIFICATION | PI_DOCFILECLSIDLOOKUP))
        && (_cbBufferFilled))
    {
        fRead = FALSE;

        // copy data form the local buffer to the provide buffer
        if (cbBuffer < _cbBufferFilled)
        {
            dwCopy = cbBuffer;
            memcpy(pBuffer, _pBuffer, cbBuffer);
            // move the memory to the front
            memcpy(_pBuffer, _pBuffer + cbBuffer, _cbBufferFilled - cbBuffer);
            _cbBufferFilled -= cbBuffer;
            hr = S_OK;
        }
        else if (cbBuffer == _cbBufferFilled)
        {
            dwCopy = _cbBufferFilled;
            memcpy(pBuffer, _pBuffer, _cbBufferFilled);
            _cbBufferFilled = 0;
            hr = S_OK;
        }
        else
        {

            // user buffer is greater than what is available in

            dwCopy = _cbBufferFilled;
            memcpy(pBuffer, _pBuffer, _cbBufferFilled);
            _cbBufferFilled = 0;
            fRead = TRUE;
            hr = E_PENDING;
        }
        if (_cbBufferFilled == 0)
        {
            // delete the buffer since it is not needed any more!
            delete [] _pBuffer;
            _pBuffer = 0;
        }
    }

    if (fRead)
    {
        if (_pProt)
        {
            hr = _pProt->Read( ((LPBYTE)pBuffer) + dwCopy, cbBuffer - dwCopy, &dwCopyNew);
            _cbTotalBytesRead += dwCopyNew;
        }
        else
        {
            hr = S_FALSE;
        }
    }

    if (pcbRead)
    {
        *pcbRead = dwCopy + dwCopyNew;
    }


    PerfDbgLog1(tagCTransaction, this, "-COInetProt::Read (hr:%lx)\n",hr);
    return hr;
}



//  Method:     COInetProt::Seek

//  Synopsis:

//  Arguments:  [DWORD] --
//              [ULARGE_INTEGER] --
//              [plibNewPosition] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::Seek(LARGE_INTEGER dlibMove,DWORD dwOrigin,ULARGE_INTEGER *plibNewPosition)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::Seek\n");

    HRESULT hr = _pProt->Seek(dlibMove, dwOrigin, plibNewPosition);

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::Seek (hr:%lx)\n", hr);
    return hr;
}



//  Method:     COInetProt::LockRequest

//  Synopsis:

//  Arguments:  [dwOptions] --

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::LockRequest(DWORD dwOptions)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::LockRequest\n");

    HRESULT hr = hr = _pProt->LockRequest(dwOptions);

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::LockRequest (hr:%lx)\n",hr);
    return hr;
}



//  Method:     COInetProt::UnlockRequest

//  Synopsis:

//  Arguments:  (none)

//  Returns:

//  History:    10-29-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::UnlockRequest()
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::UnlockRequest\n");
    HRESULT hr = NOERROR;

    hr = _pProt->UnlockRequest();

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::UnlockRequest (hr:%lx)\n", hr);
    return hr;
}



//  Method:     CTransaction::OnDataReceived

//  Synopsis:

//  Arguments:  [grfBSC] --
//              [cbBytesAvailable] --
//              [dwTotalSize] --
//              [pcbNewAvailable] --

//  Returns:

//  History:    4-15-1997   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::OnDataReceived(DWORD *pgrfBSC, DWORD *pcbBytesAvailable, DWORD *pdwTotalSize)
{
    PerfDbgLog3(tagCTransaction, this, "+COInetProt::OnDataReceived (grfBSC:%lx,  *pcbBytesAvailable:%ld, _cbTotalBytesRead:%ld)",
                                    *pgrfBSC, *pcbBytesAvailable, _cbTotalBytesRead);
    HRESULT hr = NOERROR;
    DWORD grfBSC = *pgrfBSC;

    if (_fWaitOnHandler)
    {
        hr = S_NEEDMOREDATA;
    }
    else if (   (_dwOInetBdgFlags & (PI_MIMEVERIFICATION | PI_DOCFILECLSIDLOOKUP | PI_CLASSINSTALL))
        && (!_fMimeVerified || _fNeedMoreData))
    {
        DWORD dwNewData = 0;
        TransAssert((_pProt && _cbDataSniffMin));
        //TransAssert((pcbBytesAvailable && *pcbBytesAvailable));
        //TransAssert((pdwTotalSize));

        // _cbTotalBytesRead = # of bytes read so far
        if (_cbTotalBytesRead < _cbDataSniffMin)
        {
            // no bytes read so far
            TransAssert((_cbTotalBytesRead < _cbDataSniffMin));
            // read data into buffer and report progess
            do
            {
                PProtAssert((_pBuffer && (_pBuffer + _cbBufferFilled) ));
                hr = _pProt->Read(_pBuffer + _cbBufferFilled, _cbBufferSize - _cbBufferFilled, &dwNewData);
                _cbTotalBytesRead += dwNewData;
                _cbBufferFilled += dwNewData;
            } while ((hr == S_OK) && (_cbTotalBytesRead < _cbDataSniffMin));

            // now check if this is docfile
            // if so download at least 2k
            if (!_fDocFile && _cbBufferFilled && (IsDocFile(_pBuffer, _cbBufferFilled) == S_OK))
            {
                _fDocFile = TRUE;

                // we may need to sniff to maximum to find handler
                if (!(_dwOInetBdgFlags & PI_CLASSINSTALL))
                {
                    _cbDataSniffMin =  (*pdwTotalSize && *pdwTotalSize < DATASNIFSIZEDOCFILE_MIN) ? *pdwTotalSize : DATASNIFSIZEDOCFILE_MIN;
                }
            }

            if ((hr == E_PENDING) && (_cbTotalBytesRead < _cbDataSniffMin))
            {
                // do not report anything - wait until we get more data
                // a request is pending at this time
                // need more data to sniff properly
                hr  = S_NEEDMOREDATA;
            }
            else if (hr == NOERROR || hr == E_PENDING)
            {
                TransAssert((_cbTotalBytesRead != 0));

                // report the data we have in the buffer or
                // the available #
                DWORD cbBytesReport =  (*pcbBytesAvailable > _cbTotalBytesRead) ? *pcbBytesAvailable : _cbTotalBytesRead + 1;

                if (*pdwTotalSize && ((cbBytesReport > *pdwTotalSize)))
                {
                    cbBytesReport = *pdwTotalSize;
                }
                *pcbBytesAvailable = cbBytesReport;
            }
            else if (hr == S_FALSE)
            {
                // end of stream
                *pgrfBSC |=  (BSCF_LASTDATANOTIFICATION & BSCF_DATAFULLYAVAILABLE);
                *pcbBytesAvailable = *pdwTotalSize =  _cbTotalBytesRead;
            }

            if (   (!_fMimeVerified)
                && (   (*pcbBytesAvailable >= _cbDataSniffMin)
                    || (hr == S_FALSE)) )
            {
                // enough data or end of stream
                _fMimeVerified = TRUE;
                LPWSTR  pwzStr = 0;
                DWORD dwMimeFlags = FMFD_DEFAULT;
                if( !_pwzFileName )
                {
                    dwMimeFlags = FMFD_URLASFILENAME;
                }
                hr = FindMimeFromData(NULL, (_pwzFileName) ? _pwzFileName : _pwzUrl, _pBuffer, _cbBufferFilled, _pwzMimeSuggested, dwMimeFlags, &pwzStr, 0);

                TransAssert(pwzStr);

                // note: _pwzUrl & _pwzFileName may be used later for composing URL
                //       in code base attribute of active document (deleted in destructor)

                if (pwzStr)
                {
                    _fMimeReported = 1;
                    _pProtSnk->ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE, pwzStr);
                }
                else
                {
                    TransAssert((!_pwzMimeSuggested));
                }

                if (pwzStr)
                {
                    delete [] _pwzMimeSuggested;
                    _pwzMimeSuggested = pwzStr;
                    pwzStr = 0;
                }

                if (   _fDocFile
                    && (_dwOInetBdgFlags & PI_DOCFILECLSIDLOOKUP))
                {
                    // find the class id and send it on
                    CLSID clsid;

                    HRESULT hr1 = GetClassDocFileBuffer(_pBuffer, _cbBufferFilled, &clsid);
                    if (hr1 == NOERROR)
                    {
                        LPOLESTR pwzStrClsId;
                        StringFromCLSID(clsid, &pwzStrClsId);
                        if (pwzStrClsId)
                        {
                            _pProtSnk->ReportProgress(BINDSTATUS_CLASSIDAVAILABLE, pwzStrClsId);
                        }
                        delete [] pwzStrClsId;
                    }
                }
            }

            if (   (_dwOInetBdgFlags & PI_CLASSINSTALL)
                && !_fGotHandler )
            {
                BOOL fIgnoreMimeClsid = FALSE;
                DWORD dwVersionMS = 0, dwVersionLS = 0;
                LPWSTR pwzCodeBase = 0, pwzVerInfo = 0;
                CLSID clsid;
                HRESULT hr1;

                if (IsHandlerAvailable((_pwzFileName) ? _pwzFileName : _pwzUrl, _pwzMimeSuggested, &clsid, _pBuffer, _cbBufferFilled) == S_OK)
                {
                    _fGotHandler = TRUE;
                }
                else if (_fDocFile)
                {
                    // try get code base + version information
                    // even if handler is installed, a newer version may be required for this doc file

                    hr1 = GetCodeBaseFromDocFile(_pBuffer, _cbBufferFilled, &pwzCodeBase, _pwzUrl, &dwVersionMS, &dwVersionLS);

                    // convert version info. to string
                    if (SUCCEEDED(hr1) && (dwVersionMS || dwVersionLS))
                    {
                        CHAR szVerInfo[MAX_PATH];

                        wsprintfA(szVerInfo,"%ld,%ld", dwVersionMS, dwVersionLS);
                        pwzVerInfo = DupA2W(szVerInfo);
                    }

                    // if failed to get codebase, keep sniffing
                    // if we are at end of bits then go with out a code base

                    _fNeedMoreData = FAILED(hr1) && (*pcbBytesAvailable < _cbDataSniffMin);
                }

                //BUGBUG #51944: This currently requires a DocFile and CodeBase property

                if (!_fNeedMoreData && !_fGotHandler && _fDocFile && pwzCodeBase)
                {

                    {
                        LPOLESTR pwzClsId = 0;

                        if (!IsEqualCLSID(clsid, CLSID_NULL))
                        {
                            StringFromCLSID(clsid, &pwzClsId);
                        }
                        else if (_pwzMimeSuggested)
                        {
                            // this is an optimization, if no CLSID and only this as mime type
                            // then we won't find anything in ObjectStore
                            if (!StrCmpNIW(_pwzMimeSuggested, L"application/octet-stream", 24))
                            {
                                _fGotHandler = TRUE;
                            }
                        }

                        if (!_fGotHandler && (pwzClsId || _pwzMimeSuggested))
                        {
                            LPWSTR pwzClassStr = 0;
                            int cbClassStr;

                            cbClassStr = (pwzCodeBase ? lstrlenW(pwzCodeBase) : 0)
                                + (pwzClsId ? lstrlenW(pwzClsId) : 0)
                                + (_pwzMimeSuggested ? lstrlenW(_pwzMimeSuggested) : 0)
                                + (pwzVerInfo ? (lstrlenW(pwzVerInfo) + 1) : 0)
                                + 3;

                            pwzClassStr = new WCHAR[cbClassStr];

                            if (pwzClassStr)
                            {
                                //BUGBUG: This is a bit of a hack, we have a collection of
                                //info to pass into filter, we concatenate it into one long
                                //string and send it in.

                                *pwzClassStr = '\0';
                                if (pwzCodeBase && *pwzCodeBase)
                                {
                                    StrCpyW(pwzClassStr, pwzCodeBase);
                                }

                                StrCatW(pwzClassStr, L"?");

                                if (pwzClsId)
                                {
                                    StrCatW(pwzClassStr, pwzClsId);
                                }
                                else if (_pwzMimeSuggested)
                                {
                                    StrCatW(pwzClassStr, _pwzMimeSuggested);
                                }

                                if (pwzVerInfo)
                                {
                                    StrCatW(pwzClassStr, L"?");
                                    StrCatW(pwzClassStr, pwzVerInfo);
                                }

                                _pCTrans->ReportProgress(BINDSTATUS_CLASSINSTALLLOCATION,
                                                pwzClassStr);

                                delete [] pwzClassStr;

                                // don't process this code branch again
                                _fGotHandler = TRUE;

                                // wait on more data now
                                hr = S_NEEDMOREDATA;

                               // wait on more data for future calls
                                _fWaitOnHandler = TRUE;

                                // wait for more data to sniff
                                _fNeedMoreData = TRUE;

                            }

                            delete [] pwzClsId;

                        } // pwzClsId || _pwzMimeSuggested

                    } // IsFileHandlerInstalled

                } // !_fNeedMoreData

                if (pwzCodeBase)
                {
                    CoTaskMemFree(pwzCodeBase);
                }

                if (pwzVerInfo)
                {
                    delete [] pwzVerInfo;
                }

                // if we don't need to sniff any more, curb _cbDataSniffMin
                if (_fDocFile && _fGotHandler)
                {
                    _cbDataSniffMin =  (*pdwTotalSize && *pdwTotalSize < DATASNIFSIZEDOCFILE_MIN) ? *pdwTotalSize : DATASNIFSIZEDOCFILE_MIN;
                }

                // once we have a handler we can release pwzUrl & pwzFilename
                if (_fGotHandler)
                {
                    delete [] _pwzUrl;
                    _pwzUrl = 0;
                    delete [] _pwzFileName;
                    _pwzFileName = 0;
                }

            }  // PI_CLASSINSTALL && !_fGotHandler

        }

        if (*pdwTotalSize && (*pdwTotalSize < *pcbBytesAvailable))
        {
            *pcbBytesAvailable = *pdwTotalSize;
        }

        if (hr == S_FALSE)
        {
            hr = NOERROR;
        }
    }

    {
        CLock lck(_mxs);
        _cbBytesReported = *pcbBytesAvailable;
    }
    //TransAssert((pcbBytesAvailable && *pcbBytesAvailable));
    //TransAssert((hr == NOERROR || hr == S_NEEDMOREDATA));

    PerfDbgLog2(tagCTransaction, this, "-COInetProt::OnDataReceived (hr:%lx, _cbBufferFilled:%ld)", hr, _cbBufferFilled);
    return hr;
}



//  Method:     COInetProt::Switch

//  Synopsis:

//  Arguments:  [pStateInfo] --

//  Returns:

//  History:    11-07-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::Switch(PROTOCOLDATA *pStateInfo)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::Switch");
    HRESULT hr = NOERROR;

    hr = _pProtSnk->Switch(pStateInfo);

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::Switch (hr:%lx)", hr);
    return hr;
}




//  Method:     COInetProt::ReportProgress

//  Synopsis:

//  Arguments:  [NotMsg] --
//              [szStatusText] --

//  Returns:

//  History:    11-07-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::ReportProgress(ULONG NotMsg, LPCWSTR pwzStatusText)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::ReportProgress");
    HRESULT hr = NOERROR;

    switch (NotMsg)
    {
    case BINDSTATUS_FILTERREPORTMIMETYPE:
    {
        if( _pCTrans && pwzStatusText )
        {
            // mime filter sending signal to tell us the true mime type
            _pCTrans->UpdateVerifiedMimeType(pwzStatusText);
        }

    }
    break;


    case BINDSTATUS_MIMETYPEAVAILABLE:
        if (_dwOInetBdgFlags & (PI_MIMEVERIFICATION | PI_DOCFILECLSIDLOOKUP | PI_CLASSINSTALL))
        {
            // report the mime later after sniffing data
            _pwzMimeSuggested = OLESTRDuplicate(pwzStatusText);
        }
        else
        {
            hr = _pProtSnk->ReportProgress(NotMsg, pwzStatusText);
        }

    break;

    case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE:
    {
        // disable mime sniffing
        _dwOInetBdgFlags &= (~PI_MIMEVERIFICATION &~PI_DOCFILECLSIDLOOKUP &~PI_CLASSINSTALL);
        hr = _pProtSnk->ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE, pwzStatusText);

    }
    break;

    case BINDSTATUS_CACHEFILENAMEAVAILABLE :
        _pwzFileName = OLESTRDuplicate(pwzStatusText);
        hr = _pProtSnk->ReportProgress(NotMsg, pwzStatusText);
    break;

    case BINDSTATUS_DIRECTBIND:
        _fMimeVerified = TRUE;
    break;

    case BINDSTATUS_ENDDOWNLOADCOMPONENTS :
        if (_fWaitOnHandler)
        {
            // don't stall ReportData any more
            _fWaitOnHandler = FALSE;

            // we're done with our stuff, skip sniffing
            _fNeedMoreData = FALSE;

        }


    default:
    {
        hr = _pProtSnk->ReportProgress(NotMsg, pwzStatusText);
    }

    } // end switch

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::ReportProgress (hr:%lx)", hr);
    return hr;
}



//  Method:     COInetProt::ReportData

//  Synopsis:

//  Arguments:  [grfBSCF] --
//              [ULONG] --
//              [ulProgressMax] --

//  Returns:

//  History:    11-07-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::ReportData(DWORD grfBSCF, ULONG ulProgress,ULONG ulProgressMax)
{
    PerfDbgLog3(tagCTransaction, this, "+COInetProt::ReportData(grfBSCF:%lx, ulProgress:%ld, ulProgressMax:%ld)",
                                       grfBSCF, ulProgress, ulProgressMax);
    HRESULT hr = NOERROR, hr2;

    TransAssert((ulProgress));

    if (   (_dwOInetBdgFlags & (PI_MIMEVERIFICATION | PI_DOCFILECLSIDLOOKUP | PI_CLASSINSTALL))
        && (!_fMimeVerified || _fNeedMoreData))
    {
        if (   (OnDataReceived(&grfBSCF, &ulProgress, &ulProgressMax) == NOERROR)
            && _pProtSnk
            && (ulProgress || (!ulProgress && !ulProgressMax)) )
        {
            // OnDataReceived sniffs data and calls Read - EOF with ReportResult might occure
            //TransAssert((ulProgress));
            TransAssert((_fMimeReported));
            hr = _pProtSnk->ReportData( grfBSCF, ulProgress, ulProgressMax);
        }
    }
    else
    {
        TransAssert((ulProgress));
        hr = _pProtSnk->ReportData( grfBSCF, ulProgress, ulProgressMax);
    }

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::ReportData (hr:%lx)", hr);
    return hr;
}



//  Method:     COInetProt::ReportResult

//  Synopsis:

//  Arguments:  [DWORD] --
//              [dwError] --
//              [wzResult] --

//  Returns:

//  History:    11-07-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::ReportResult(HRESULT hrResult, DWORD dwError, LPCWSTR wzResult)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::ReportResult");
    HRESULT hr = NOERROR;

    hr = _pProtSnk->ReportResult(hrResult, dwError, wzResult);

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::ReportResult (hr:%lx)", hr);
    return hr;
}



//  Method:     COInetProt::Initialize

//  Synopsis:

//  Arguments:  [pCTrans] --
//              [dwMode] --
//              [dwOptions] --
//              [pUnk] --
//              [pProt] --
//              [pProtSnk] --

//  Returns:

//  History:    11-07-1996   JohannP (Johann Posch)   Created

//  Notes:


STDMETHODIMP COInetProt::Initialize(CTransaction *pCTrans,IServiceProvider *pSrvPrv, DWORD dwMode, DWORD dwOptions, IUnknown *pUnk, IOInetProtocol *pProt, IOInetProtocolSink *pProtSnk, LPWSTR pwzUrl)
{
    HRESULT hr = NOERROR;
    _dwMode = dwMode;
    _pUnk = pUnk;
    //_pProt = pProt;
    //_pProtSnk = pProtSnk;
    _pCTrans = pCTrans;
    _pSrvPrv = pSrvPrv;
    if (_pSrvPrv)
    {
        _pSrvPrv->AddRef();
    }
    SetProtocolSink(pProtSnk);
    SetProtocol(pProt);

    _dwOInetBdgFlags = dwOptions;
    if (_dwOInetBdgFlags & (PI_MIMEVERIFICATION | PI_DOCFILECLSIDLOOKUP))
    {
        TransAssert(( DATASNIFSIZE_MIN <= DATASNIFSIZEDOCFILE_MIN));
        if (_dwOInetBdgFlags & PI_CLASSINSTALL)
        {
            _cbBufferSize = DATASNIFSIZEDOCFILE_MAX;
        }
        else
        {
            _cbBufferSize = DATASNIFSIZEDOCFILE_MIN; //DATASNIFSIZE_MIN;
        }

        _pBuffer = (LPBYTE) new BYTE[_cbBufferSize];

        if (!_pBuffer)
        {
            _cbBufferSize = 0;
            hr = E_OUTOFMEMORY;
        }
        else
        {
            _cbDataSniffMin = DATASNIFSIZE_MIN;
        }

        if (pwzUrl)
        {
            _pwzUrl = OLESTRDuplicate(pwzUrl);
        }

    }
    TransAssert((_pProt && _pProtSnk));
    return hr;
}



//  Method:     COInetProt::QueryService

//  Synopsis:

//  Arguments:  [rsid] --
//              [riid] --
//              [ppvObj] --

//  Returns:

//  History:    11-07-1996   JohannP (Johann Posch)   Created

//  Notes:


HRESULT IUnknown_QueryService(IUnknown* punk, REFGUID rsid, REFIID riid, void ** ppvObj);

HRESULT COInetProt::QueryService(REFGUID rsid, REFIID riid, void ** ppvObj)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::QueryService");
    HRESULT     hr = NOERROR;
    VDATETHIS(this);
    TransAssert((ppvObj));

    if (_pSrvPrv)
    {
        hr = _pSrvPrv->QueryService(rsid,riid, ppvObj);
    }
    else
    {
        hr = IUnknown_QueryService(_pProtSnk, rsid, riid, ppvObj);
    }

    TransAssert(( ((hr == E_NOINTERFACE) && !*ppvObj)  || ((hr == NOERROR) && *ppvObj) ));

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::QueryService (hr:%lx)", hr);
    return hr;
}

STDMETHODIMP COInetProt::SetPriority(LONG nPriority)
{
    PerfDbgLog1(tagCTransaction, this, "+COInetProt::SetPriority (%ld)", nPriority);

    HRESULT hr = S_OK;

    _nPriority = nPriority;

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::SetPriority (hr:%lx)", hr);
    return hr;
}

STDMETHODIMP COInetProt::GetPriority(LONG * pnPriority)
{
    PerfDbgLog(tagCTransaction, this, "+COInetProt::GetPriority");

    HRESULT hr;

    if (!pnPriority)
    {
        hr = E_INVALIDARG;
    }
    else
    {
        *pnPriority = _nPriority;
        hr = S_OK;
    }

    PerfDbgLog1(tagCTransaction, this, "-COInetProt::GetPriority (hr:%lx)", hr);
    return hr;
}