Windows2000/private/inet/urlmon/trans/cnetstrm.cxx
2020-09-30 17:12:32 +02:00

369 lines
6.7 KiB
C++

// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1992 - 1995.
// File: cnetstrm.cxx
// Contents: Implements the stream: protocol
// Classes: CINetStream
// Functions:
// History: 5/3/96 Created Craig Critchley [craigc]
#include <trans.h>
// Method: CINetStream::CINetStream
// Synopsis: Constructs a stream protcol object
// Arguments: rclsid
// Returns:
// History: 5/3/96 Created Craig Critchley [craigc]
// Notes:
CINetStream::CINetStream(REFCLSID rclsid, IUnknown *pUnkOuter) : CINet(rclsid,pUnkOuter)
{
_dwIsA = DLD_PROTOCOL_STREAM;
_pstm = NULL;
}
// Method: CINetStream::~CINetStream
// Synopsis: destroys a stream protocol object
// Arguments:
// Returns:
// History: 5/3/96 Created Craig Critchley [craigc]
// Notes:
CINetStream::~CINetStream()
{
if (_pstm)
{
_pstm->Release();
}
}
// Method: CINetStream::INetAsyncOpen
// Synopsis: opens and synchronously downloads data from a stream
// Arguments: rclsid
// Returns:
// History: 5/3/96 Created Craig Critchley [craigc]
// Notes:
HRESULT CINetStream::INetAsyncOpen()
{
IBindCtx * pbc = 0;
IMoniker * pmk = 0;
IStream * pstm = 0;
IParseDisplayName * pParser = 0;
STATSTG stat;
CLSID clsid;
CHAR szDisplayName[MAX_PATH];
WCHAR wzDisplayName[MAX_PATH];
WCHAR wzProgId[MAX_PATH];
LPSTR pa, psz;
LPWSTR pwz, pwzI;
int cch;
HRESULT hr = E_NOTIMPL;
ULONG cchE;
BOOL fGotMIMEType = FALSE;
ReportNotification(BINDSTATUS_SENDINGREQUEST);
// need one of these
if (FAILED(hr = CreateBindCtx(0,&pbc)))
goto End;
// form is mk:@progid:moniker
strcpy(szDisplayName,GetServerName());
strcat(szDisplayName,psz = GetObjectName());
// if the moniker has a file extension, try to
// determine the MIME type that way...
psz = FindFileExtension(psz);
if (psz)
{
char szMime[MAX_PATH];
DWORD cb = MAX_PATH;
if (SUCCEEDED(GetMimeFromExt(psz,szMime,&cb)))
{
ReportNotification(BINDSTATUS_MIMETYPEAVAILABLE, szMime);
fGotMIMEType = TRUE;
}
}
A2W(szDisplayName,wzDisplayName,MAX_PATH);
// find progid
for (pwz = wzDisplayName, pwzI = wzProgId; *pwz; pwz++)
{
if (*pwz == '@')
{
pwzI = wzProgId;
}
else if (*pwz == ':')
{
*pwzI = 0;
// the remainder may have a filename with a useful
// extension... just in case, set the filename...
LPSTR pszStr = DupW2A(pwz+1);
if (pszStr)
{
ReportNotification(BINDSTATUS_CACHEFILENAMEAVAILABLE, pszStr);
delete pszStr;
}
break;
}
else
{
*pwzI++ = *pwz;
}
}
#if 0
if (FAILED(hr = MkParseDisplayName(pbc,wzDisplayName,&cchE,&pmk)) && pmk)
goto End;
#else
// BUGBUG
// MkParseDisplayName was opening another instance of app
// force inproc server. ick.
if (FAILED(hr = CLSIDFromProgID(wzProgId,&clsid)))
{
goto End;
}
if (FAILED(hr = CoCreateInstance(clsid,0,CLSCTX_INPROC_SERVER, IID_IParseDisplayName,(void**)&pParser)))
{
goto End;
}
TransAssert((pParser));
if (FAILED(hr = pParser->ParseDisplayName(pbc,wzDisplayName,&cchE,&pmk)))
{
goto End;
}
TransAssert((pmk));
#endif
if (FAILED(hr = pmk->BindToStorage(pbc,0,IID_IStream,(void**)&pstm)))
{
hr = INET_E_RESOURCE_NOT_FOUND;
goto End;
}
TransAssert((hr != MK_S_ASYNCHRONOUS));
TransAssert((pstm));
_pstm = pstm;
_pstm->AddRef();
// now we have a stream - stuff it into the trans data
if (FAILED(hr = pstm->Stat(&stat,STATFLAG_NONAME)))
{
goto End;
}
_cbTotalBytesRead = stat.cbSize.LowPart;
_cbDataSize = stat.cbSize.LowPart;
hr = _pCTrans->ReportData(BSCF_FIRSTDATANOTIFICATION | BSCF_LASTDATANOTIFICATION, stat.cbSize.LowPart,stat.cbSize.LowPart);
_hrError = INET_E_DONE;
End:
if (FAILED(hr))
{
SetCNetBindResult(GetLastError());
_hrError = INET_E_RESOURCE_NOT_FOUND;
ReportResultAndStop(_hrError);
}
// play nice, no leaks
if (pParser)
{
pParser->Release();
}
if (pmk)
{
pmk->Release();
}
if (pbc)
{
pbc->Release();
}
if (pstm)
{
pstm->Release();
}
// make sure I set this any way I get out
TransAssert((hr != E_NOTIMPL));
return hr;
}
// Method: CINetStream::ReadDataHere
// Synopsis:
// Arguments: [pBuffer] --
// [cbBytes] --
// [pcbBytes] --
// Returns:
// History: 10-29-1996 JohannP (Johann Posch) Created
// Notes:
HRESULT CINetStream::ReadDataHere(BYTE *pBuffer, DWORD cbBytes, DWORD *pcbBytes)
{
TransDebugOut((DEB_PROT, "%p _IN CINetStream::ReadDataHere\n", this));
HRESULT hr = E_FAIL;
TransAssert((cbBytes && pcbBytes));
if (_pstm)
{
hr = _pstm->Read(pBuffer,cbBytes,pcbBytes);
}
TransDebugOut((DEB_PROT, "%p OUT CINetStream::ReadDataHere (_hrError:%lx, [hr:%lx,cbBytesAsked:%ld,cbBytesReturned:%ld]) \n",
this, _hrError, hr, cbBytes, *pcbBytes));
return hr;
}
// Method: CINetStream::INetSeek
// Synopsis:
// Arguments: [DWORD] --
// [ULARGE_INTEGER] --
// [plibNewPosition] --
// Returns:
// History: 10-29-1996 JohannP (Johann Posch) Created
// Notes:
HRESULT CINetStream::INetSeek(LARGE_INTEGER dlibMove,DWORD dwOrigin,ULARGE_INTEGER *plibNewPosition)
{
TransDebugOut((DEB_PROT, "%p _IN CINetStream::INetSeek\n", this));
HRESULT hr = E_FAIL;
if (_pstm)
{
hr = _pstm->Seek(dlibMove, dwOrigin, plibNewPosition);
}
TransDebugOut((DEB_PROT, "%p OUT CINetStream::INetSeek (hr:%lx) \n",this, hr));
return hr;
}
// Method: CINetStream::LockFile
// Synopsis:
// Arguments:
// Returns:
// History: 8-13-96 JohannP (Johann Posch) Created
// Notes:
HRESULT CINetStream::LockFile(BOOL fRetrieve)
{
TransDebugOut((DEB_PROT, "%p _IN CINetStream::LockFile\n", this));
HRESULT hr = NOERROR;
// nothing to do for now
TransDebugOut((DEB_PROT,"%p OUT CINetStream::LockFile (hr:%lx)\n",this, hr));
return hr;
}
// Method: CINetStream::UnlockFile
// Synopsis: unlocks the file
// Arguments: (none)
// Returns:
// History: 8-13-96 JohannP (Johann Posch) Created
// Notes:
HRESULT CINetStream::UnlockFile()
{
TransDebugOut((DEB_PROT, "%p IN CINetStream::UnlockFile\n", this));
HRESULT hr = NOERROR;
// nothing to do for now
TransDebugOut((DEB_PROT,"%p OUT CINetStream::UnlockFile (hr:%lx)\n", this, hr));
return hr;
}