Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

632 lines
15 KiB

//+---------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1992 - 1995.
//
// File: resprot.cxx
//
// Contents:
//
// Classes:
//
// Functions:
//
// History: 11-07-1996 JohannP (Johann Posch) Created
//
//----------------------------------------------------------------------------
#include <eapp.h>
#include <tchar.h>
#define MAX_ID 10000
#define WITH_TAGS
//+---------------------------------------------------------------------------
//
// Method: CResProtocol::Start
//
// Synopsis:
//
// Arguments: [pwzUrl] --
// [pTrans] --
// [pOIBindInfo] --
// [grfSTI] --
// [dwReserved] --
//
// Returns:
//
// History: 10-29-1996 JohannP (Johann Posch) Created
//
// Notes:
//
//----------------------------------------------------------------------------
STDMETHODIMP CResProtocol::Start(LPCWSTR pwzUrl, IOInetProtocolSink *pTrans, IOInetBindInfo *pOIBindInfo,
DWORD grfSTI, DWORD dwReserved)
{
EProtDebugOut((DEB_PLUGPROT, "%p _IN CResProtocol::Start\n", this));
HRESULT hr = NOERROR;
WCHAR wzURL[MAX_URL_SIZE];
EProtAssert((!_pProtSink && pOIBindInfo && pTrans));
EProtAssert((_pwzUrl == NULL));
hr = CBaseProtocol::Start(pwzUrl,pTrans, pOIBindInfo, grfSTI, dwReserved);
if ( (grfSTI & PI_PARSE_URL) )
{
hr = ParseAndBind(FALSE);
}
else if (hr == NOERROR)
{
// asked to go async as soon as possible
// use the switch mechanism which will \
// call back later on ::Continue
if (grfSTI & PI_FORCE_ASYNC)
{
hr = E_PENDING;
PROTOCOLDATA protdata;
protdata.grfFlags = PI_FORCE_ASYNC;
protdata.dwState = RES_STATE_BIND;
protdata.pData = 0;
protdata.cbData = 0;
_pProtSink->Switch(&protdata);
}
else
{
hr = ParseAndBind(TRUE);
}
}
EProtDebugOut((DEB_PLUGPROT, "%p OUT CResProtocol::Start (hr:%lx)\n",this, hr));
return hr;
}
//+---------------------------------------------------------------------------
//
// Method: CResProtocol::Continue
//
// Synopsis:
//
// Arguments: [pStateInfoIn] --
//
// Returns:
//
// History: 10-29-1996 JohannP (Johann Posch) Created
//
// Notes:
//
//----------------------------------------------------------------------------
STDMETHODIMP CResProtocol::Continue(PROTOCOLDATA *pStateInfoIn)
{
EProtDebugOut((DEB_PLUGPROT, "%p _IN CResProtocol::Continue\n", this));
HRESULT hr = E_FAIL;
EProtAssert((!pStateInfoIn->pData && pStateInfoIn->cbData && (pStateInfoIn->dwState == RES_STATE_BIND)));
if (pStateInfoIn->dwState == RES_STATE_BIND)
{
hr = ParseAndBind();
}
EProtDebugOut((DEB_PLUGPROT, "%p OUT CResProtocol::Continue (hr:%lx)\n",this, hr));
return hr;
}
//+---------------------------------------------------------------------------
//
// Method: CResProtocol::Read
//
// Synopsis:
//
// Arguments: [ULONG] --
// [ULONG] --
// [pcbRead] --
//
// Returns:
//
// History: 10-29-1996 JohannP (Johann Posch) Created
//
// Notes:
//
//----------------------------------------------------------------------------
STDMETHODIMP CResProtocol::Read(void *pv,ULONG cb,ULONG *pcbRead)
{
EProtDebugOut((DEB_PLUGPROT, "%p _IN CResProtocol::Read (cb:%ld)\n", this,cb));
HRESULT hr = NOERROR;
if (_cbBuffer > _cbPos)
{
ULONG cbCopy = (cb < (_cbBuffer - _cbPos)) ? cb : _cbBuffer - _cbPos;
memcpy((LPVOID)pv, ((LPBYTE)_pBuffer) + _cbPos, cbCopy);
_cbPos += cbCopy;
*pcbRead = cbCopy;
hr = (_cbPos < _cbBuffer) ? S_OK : S_FALSE;
}
else
{
hr = S_FALSE;
}
EProtDebugOut((DEB_PLUGPROT, "%p OUT CResProtocol::Read (pcbRead:%ld, hr:%lx)\n",this,*pcbRead, hr));
return hr;
}
//+---------------------------------------------------------------------------
//
// Method: CResProtocol::Seek
//
// Synopsis:
//
// Arguments: [DWORD] --
// [ULARGE_INTEGER] --
// [plibNewPosition] --
//
// Returns:
//
// History: 10-29-1996 JohannP (Johann Posch) Created
//
// Notes: WORK: not done
//
//----------------------------------------------------------------------------
STDMETHODIMP CResProtocol::Seek(LARGE_INTEGER dlibMove,DWORD dwOrigin,ULARGE_INTEGER *plibNewPosition)
{
EProtDebugOut((DEB_PLUGPROT, "%p _IN CResProtocol::Seek\n", this));
HRESULT hr = NOERROR;
if (dwOrigin == STREAM_SEEK_SET)
{
if (dlibMove.LowPart >= 0)
{
_cbPos = dlibMove.LowPart;
if (plibNewPosition)
{
plibNewPosition->HighPart = 0;
plibNewPosition->LowPart = _cbPos;
}
}
else
{
hr = STG_E_INVALIDPOINTER;
}
}
else
{
hr = E_NOTIMPL;
}
EProtDebugOut((DEB_PLUGPROT, "%p OUT CResProtocol::Seek (hr:%lx)\n",this, hr));
return hr;
}
//+---------------------------------------------------------------------------
//
// Method: CResProtocol::LockRequest
//
// Synopsis:
//
// Arguments: [dwOptions] --
//
// Returns:
//
// History: 10-29-1996 JohannP (Johann Posch) Created
//
// Notes:
//
//----------------------------------------------------------------------------
STDMETHODIMP CResProtocol::LockRequest(DWORD dwOptions)
{
EProtDebugOut((DEB_PLUGPROT, "%p _IN CResProtocol::LockRequest\n", this));
HRESULT hr = NOERROR;
if (OpenTempFile())
{
DWORD dwWrite;
if (!WriteFile(_hFile, _pBuffer, _cbBuffer, &dwWrite,NULL))
{
hr = E_FAIL;
}
CloseTempFile();
}
EProtDebugOut((DEB_PLUGPROT, "%p OUT CResProtocol::LockRequest (hr:%lx)\n",this, hr));
return hr;
}
//+---------------------------------------------------------------------------
//
// Method: CResProtocol::UnlockRequest
//
// Synopsis:
//
// Arguments: (none)
//
// Returns:
//
// History: 10-29-1996 JohannP (Johann Posch) Created
//
// Notes:
//
//----------------------------------------------------------------------------
STDMETHODIMP CResProtocol::UnlockRequest()
{
EProtDebugOut((DEB_PLUGPROT, "%p _IN CResProtocol::UnlockRequest\n", this));
HRESULT hr = NOERROR;
//CloseTempFile();
EProtDebugOut((DEB_PLUGPROT, "%p OUT CResProtocol::UnlockRequest (hr:%lx)\n",this, hr));
return hr;
}
//+---------------------------------------------------------------------------
//
// Method: CResProtocol::CResProtocol
//
// Synopsis:
//
// Arguments: (none)
//
// Returns:
//
// History: 1-27-96 JohannP (Johann Posch) Created
//
// Notes:
//
//----------------------------------------------------------------------------
CResProtocol::CResProtocol(REFCLSID rclsid, IUnknown *pUnkOuter, IUnknown **ppUnkInner) : CBaseProtocol(rclsid, pUnkOuter, ppUnkInner)
{
EProtDebugOut((DEB_PLUGPROT, "%p _IN CResProtocol::CResProtocol \n", this));
_hInst = NULL;
_cbBuffer = 0;
_pBuffer = 0;
_hgbl = 0;
_cbPos = 0;
EProtDebugOut((DEB_PLUGPROT, "%p OUT CResProtocol::CResProtocol \n", this));
}
//+---------------------------------------------------------------------------
//
// Method: CResProtocol::~CResProtocol
//
// Synopsis:
//
// Arguments: (none)
//
// Returns:
//
// History: 11-09-1996 JohannP (Johann Posch) Created
//
// Notes:
//
//----------------------------------------------------------------------------
CResProtocol::~CResProtocol()
{
if (_hgbl)
{
UnlockResource(_hgbl);
}
if (_hInst)
{
FreeLibrary(_hInst);
}
EProtDebugOut((DEB_PLUGPROT, "%p _IN/OUT CResProtocol::~CResProtocol \n", this));
}
//+---------------------------------------------------------------------------
//
// Method: CResProtocol::GetResource
//
// Synopsis:
//
// Arguments: [pwzFileName] --
// [pwzResName] --
// [pwzResType] --
//
// Returns:
//
// History: 11-09-1996 JohannP (Johann Posch) Created
//
// Notes:
//
//----------------------------------------------------------------------------
STDMETHODIMP CResProtocol::GetResource(LPCWSTR pwzFileName, LPCWSTR pwzResName, LPCWSTR pwzResType, LPCWSTR pwzMime)
{
EProtDebugOut((DEB_PLUGPROT, "CResProtocol::GetResource\n)"));
HRESULT hr = NOERROR;
DWORD dwError = 0;
HRSRC hrsrc;
LPSTR pszFileName = DupW2A(pwzFileName );
LPSTR pszResName = DupW2A(pwzResName );
LPSTR pszResType = DupW2A(pwzResType );
if (!pszFileName || !pszResName || !pszResType)
{
hr = E_OUTOFMEMORY;
}
else do
{
_pProtSink->ReportProgress(BINDSTATUS_SENDINGREQUEST, pwzResName);
_hInst = LoadLibraryEx(pszFileName, NULL, DONT_RESOLVE_DLL_REFERENCES);
if (!_hInst)
{
hr = INET_E_RESOURCE_NOT_FOUND;
dwError = GetLastError();
break;
}
if (!wcscmp(pwzResName, L"?"))
{
for (int i = 0 ; i < MAX_ID; i++)
{
hrsrc = FindResource(_hInst, (LPSTR)MAKEINTRESOURCE(i), pszResType);
if (hrsrc)
{
EProtDebugOut((DEB_PLUGPROT, "CResProtocol::GetResource (szResName:#%ld, wzRestype:%s\n)",i,pszResType));
i = MAX_ID;
}
}
}
else
{
hrsrc = FindResource(_hInst, pszResName, pszResType);
}
if (!hrsrc)
{
hr = INET_E_OBJECT_NOT_FOUND;
dwError = GetLastError();
break;
}
_hgbl = LoadResource(_hInst, hrsrc);
if (!_hgbl)
{
hr = INET_E_DATA_NOT_AVAILABLE;
dwError = GetLastError();
break;
}
_pBuffer = LockResource(_hgbl);
if (!_pBuffer)
{
hr = INET_E_DATA_NOT_AVAILABLE;
dwError = GetLastError();
break;
}
_cbBuffer = SizeofResource(_hInst, hrsrc);
if (_grfBindF & INTERNET_FLAG_NEED_FILE)
{
LockRequest(0);
}
#ifdef UNUSED
if (OpenTempFile())
{
DWORD dwWrite;
if (!WriteFile(_hFile, _pBuffer, _cbBuffer, &dwWrite,NULL))
{
dwError = GetLastError();
break;
}
CloseTempFile();
}
#endif //UNUSED
hr = NOERROR;
break;
} while (1);
if (dwError || (hr != NOERROR))
{
_pProtSink->ReportResult(hr, dwError, 0);
}
else
{
if (pwzMime)
{
_pProtSink->ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE, pwzMime);
}
_bscf |= BSCF_LASTDATANOTIFICATION | BSCF_DATAFULLYAVAILABLE;
_pProtSink->ReportData(_bscf, _cbBuffer, _cbBuffer);
}
if (pszFileName)
{
delete pszFileName;
}
if (pszResName)
{
delete pszResName;
}
if (pszResType)
{
delete pszResType;
}
EProtDebugOut((DEB_PLUGPROT, "%p OUT CResProtocol::GetResource (hr:%lx)\n",this, hr));
return hr;
}
//+---------------------------------------------------------------------------
//
// Method: CResProtocol::ParseAndBind
//
// Synopsis:
//
// Arguments: (none)
//
// Returns:
//
// History: 11-09-1996 JohannP (Johann Posch) Created
//
// Notes:
//
//----------------------------------------------------------------------------
STDMETHODIMP CResProtocol::ParseAndBind(BOOL fBind)
{
EProtDebugOut((DEB_PLUGPROT, "%p _IN CResProtocol::ParseAndBind\n", this));
HRESULT hr = MK_E_SYNTAX;
WCHAR wzlURL[MAX_URL_SIZE];
wcscpy(wzlURL, _wzFullURL);
do
{
// check if protocol part
LPWSTR pwz = wcschr(wzlURL, ':');
pwz++;
if (wcsnicmp(pwz, L"//", 2) && wcsnicmp(pwz, L"\\\\", 2))
{
break;
}
// find the file name and path
LPWSTR pwzFileName = pwz + 2;
EProtAssert((pwzFileName));
// the file is
LPWSTR pwz1 = wcsrchr(wzlURL, '/');
if (!pwz1)
{
break;
}
*pwz1 = '\0';
pwz1++;
if (!*pwz1)
{
break;
}
#ifdef WITH_TAGS
LPWSTR pwzResDelimiter = wcschr(pwz1, '?');
if (!pwzResDelimiter)
{
break;
}
LPWSTR pwzResTag = wcsstr(pwz1, L"name:");
if (!pwzResTag)
{
break;
}
LPWSTR pwzTypeTag = wcsstr(pwz1, L"type:");
if (!pwzTypeTag)
{
break;
}
LPWSTR pwzMimeTag = wcsstr(pwz1, L"mime:");
// get the resource name
LPWSTR pwzResName = wcschr(pwzResTag, ':');
pwzResName++;
//find the end of the resource name
LPWSTR pwzResType = wcschr(pwzTypeTag, ':');
pwzResType++;
LPWSTR pwzMime = 0;
if (pwzMimeTag)
{
pwzMime = wcschr(pwzMimeTag, ':');
pwzMime++;
*pwzMimeTag = 0;
}
*pwzResTag = 0;
*pwzTypeTag = 0;
#else
// find the delimiter for the private part
LPWSTR pwzResName = wcschr(pwz1, '?');
if (!pwzResName)
{
break;
}
// get the resource name
pwzResName++;
//find the end of the resource name
LPWSTR pwzResType = wcschr(pwzResName, ' ');
if (!pwzResType)
{
break;
}
*pwzResType = '\0';
pwzResType++;
EProtDebugOut((DEB_PLUGPROT, "CResProtocol::GetResource (wzResName:%ws,pwzResType:%ws\n)",pwzResName,pwzResType));
/*
{
pwzResType = (LPWSTR)MAKEINTRESOURCE(RT_ICON);
EProtDebugOut((DEB_PLUGPROT, "CResProtocol::GetResource (wzResName:%ws,pwzResType:RT_ICON\n)",pwzResName));
}
*/
#endif //WITH_TAGS
EProtAssert(((WCHAR *)pwzFileName && (WCHAR *)pwzResName));
if (fBind && pwzFileName && pwzResName && pwzResType)
{
hr = GetResource(pwzFileName, pwzResName, pwzResType, pwzMime);
}
break;
} while (1);
if (hr == MK_E_SYNTAX)
{
_pProtSink->ReportResult(hr, 0, 0);
}
EProtDebugOut((DEB_PLUGPROT, "%p OUT CResProtocol::ParseAndBind (hr:%lx)\n", this,hr));
return hr;
}