mirror of https://github.com/tongzx/nt5src
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.
545 lines
15 KiB
545 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 <urlint.h>
|
|
#include <stdio.h>
|
|
#include <sem.hxx>
|
|
#include <wininet.h>
|
|
#include "urlcf.hxx"
|
|
#include "protbase.hxx"
|
|
#include "resprot.hxx"
|
|
#include <tchar.h>
|
|
|
|
#define MAX_ID 10000
|
|
|
|
|
|
HRESULT LookupProtocolClsIDFromReg(LPCTSTR pszUrl, CLSID *pclsid);
|
|
|
|
#define SZPROTOCOLROOT "PROTOCOLS\\Handler\\"
|
|
#define SZCLASS "CLSID"
|
|
|
|
#define SZMYPROTOCOL "search"
|
|
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// 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)
|
|
{
|
|
TransDebugOut((DEB_PROT, "%p _IN CResProtocol::Start\n", this));
|
|
HRESULT hr = NOERROR;
|
|
CLSID clsid;
|
|
|
|
TransAssert((!_pProtSink && pOIBindInfo && pTrans));
|
|
TransAssert((_pszUrl == NULL));
|
|
|
|
// have to start the base class to get the bindinfo and the full URL.
|
|
|
|
hr = CBaseProtocol::Start(pwzUrl,pTrans, pOIBindInfo, grfSTI, dwReserved);
|
|
|
|
if (hr == NOERROR)
|
|
{
|
|
// first, check if this is hookable URL.
|
|
// return E_USEDEFAULTPROTOCAL if not.
|
|
|
|
if ((hr = Bind()) == NOERROR)
|
|
{
|
|
// We need to use the new (cooked) URL if Bind() succeeded.
|
|
|
|
if ((hr = LookupProtocolClsIDFromReg(_szNewUrl, &clsid)) == NOERROR)
|
|
{
|
|
IClassFactory *pCF = 0;
|
|
hr = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER,IID_IClassFactory, (void**)&pCF);
|
|
if (hr == NOERROR)
|
|
{
|
|
// Perf:
|
|
// we might want to move Create/Release of the HTTP protocl handler out of
|
|
// Start/Terminate to Create/Release.
|
|
|
|
hr = pCF->CreateInstance(NULL, IID_IOInetProtocol, (void **)&_pProt);
|
|
if (hr == NOERROR)
|
|
{
|
|
// ???
|
|
// We also may need to implement our own IOInetBindInfo so that we
|
|
// can give it the new (cooked) BindInfo.
|
|
|
|
// We need to pass down the new (cooked) URL if Bind() succeeded.
|
|
|
|
LPWSTR pwzNewUrl = DupA2W(_szNewUrl);
|
|
|
|
if (pwzNewUrl)
|
|
{
|
|
hr = _pProt->Start(pwzNewUrl, pTrans, pOIBindInfo, grfSTI, dwReserved);
|
|
|
|
delete pwzNewUrl;
|
|
}
|
|
}
|
|
|
|
pCF->Release();
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
TransDebugOut((DEB_PROT, "%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)
|
|
{
|
|
TransDebugOut((DEB_PROT, "%p _IN CResProtocol::Continue\n", this));
|
|
HRESULT hr = E_FAIL;
|
|
|
|
if (_pProt)
|
|
hr = _pProt->Continue(pStateInfoIn);
|
|
|
|
TransDebugOut((DEB_PROT, "%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)
|
|
{
|
|
TransDebugOut((DEB_PROT, "%p _IN CResProtocol::Read (cb:%ld)\n", this,cb));
|
|
HRESULT hr = NOERROR;
|
|
|
|
if (_pProt)
|
|
hr = _pProt->Read(pv, cb, pcbRead);
|
|
|
|
TransDebugOut((DEB_PROT, "%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)
|
|
{
|
|
TransDebugOut((DEB_PROT, "%p _IN CResProtocol::Seek\n", this));
|
|
HRESULT hr = NOERROR;
|
|
|
|
if (_pProt)
|
|
hr = _pProt->Seek(dlibMove, dwOrigin, plibNewPosition);
|
|
|
|
TransDebugOut((DEB_PROT, "%p OUT CResProtocol::Seek (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)
|
|
{
|
|
TransDebugOut((DEB_PROT, "%p _IN/OUT CResProtocol::CResProtocol \n", this));
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Method: CResProtocol::~CResProtocol
|
|
//
|
|
// Synopsis:
|
|
//
|
|
// Arguments: (none)
|
|
//
|
|
// Returns:
|
|
//
|
|
// History: 11-09-1996 JohannP (Johann Posch) Created
|
|
//
|
|
// Notes:
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
CResProtocol::~CResProtocol()
|
|
{
|
|
TransDebugOut((DEB_PROT, "%p _IN/OUT CResProtocol::~CResProtocol \n", this));
|
|
}
|
|
|
|
|
|
// SUPER HACK FUNCTION because InternetCrackUrl is not very good.
|
|
// THIS FUNCTION DOES NOT CHECK THE INPUT BUFFER SIZE.
|
|
// AND CALLER MUST PROVIDE lpszUrlPath buffer for us to parse.
|
|
|
|
BOOL
|
|
MyCrackUrl(
|
|
LPCSTR lpszUrl,
|
|
DWORD dwUrlLength,
|
|
DWORD dwFlags,
|
|
LPURL_COMPONENTSA lpUC
|
|
)
|
|
{
|
|
LPSTR pszBuf = lpUC->lpszUrlPath;
|
|
DWORD dwSize = lpUC->dwUrlPathLength;
|
|
BOOL ret = FALSE;
|
|
|
|
if (pszBuf)
|
|
{
|
|
if (InternetCanonicalizeUrl(lpszUrl, pszBuf, &dwSize, ICU_DECODE | ICU_NO_ENCODE))
|
|
{
|
|
// find protocol
|
|
|
|
LPSTR pTmp = StrChr(pszBuf, ':');
|
|
|
|
if (pTmp)
|
|
{
|
|
*pTmp = '\0';
|
|
|
|
if (lpUC->lpszScheme)
|
|
{
|
|
lstrcpy(lpUC->lpszScheme, pszBuf);
|
|
lpUC->dwSchemeLength = pTmp - pszBuf;
|
|
}
|
|
|
|
pszBuf = ++pTmp;
|
|
|
|
// skip '/'s
|
|
|
|
while (*pszBuf && (*pszBuf == '/'))
|
|
pszBuf++;
|
|
|
|
// find host name
|
|
|
|
pTmp = StrChr(pszBuf, '/');
|
|
|
|
if (lpUC->lpszHostName)
|
|
{
|
|
if (pTmp)
|
|
lpUC->dwHostNameLength = pTmp - pszBuf;
|
|
else
|
|
lpUC->dwHostNameLength = lstrlen(pszBuf);
|
|
|
|
// + 1 for the NULL terminator
|
|
|
|
lstrcpyn(lpUC->lpszHostName, pszBuf, lpUC->dwHostNameLength + 1);
|
|
}
|
|
|
|
// if a '/' was found, the rest is URL path
|
|
|
|
if (pTmp)
|
|
lpUC->dwUrlPathLength = lstrlen(pTmp);
|
|
else
|
|
lpUC->dwUrlPathLength = 0;
|
|
|
|
lpUC->lpszUrlPath = pTmp;
|
|
|
|
ret = TRUE;
|
|
}
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Method: CResProtocol::Bind
|
|
//
|
|
// Synopsis:
|
|
//
|
|
// Arguments: (none)
|
|
//
|
|
// Returns:
|
|
//
|
|
// History: 11-09-1996 JohannP (Johann Posch) Created
|
|
//
|
|
// Notes:
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
STDMETHODIMP CResProtocol::Bind()
|
|
{
|
|
TransDebugOut((DEB_PROT, "%p _IN CResProtocol::Bind (szUrl >%s< )\n", this, _pszUrl));
|
|
|
|
// HRESULT hr = MK_E_SYNTAX;
|
|
// HRESULT hr = INET_E_USE_DEFAULT_PROTOCOLHANDLER;
|
|
HRESULT hr = MK_E_NOPREFIX;
|
|
// HRESULT hr = INET_E_UNKNOWN_PROTOCOL;
|
|
URL_COMPONENTS uc;
|
|
TCHAR szScheme[INTERNET_MAX_SCHEME_LENGTH];
|
|
TCHAR szHost[INTERNET_MAX_HOST_NAME_LENGTH];
|
|
TCHAR szURL[MAX_URL_SIZE];
|
|
DWORD dwNeeded;
|
|
LPSTR lpSearchName;
|
|
LPSTR lpNewName = NULL;
|
|
|
|
ZeroMemory(&uc, sizeof(uc));
|
|
uc.dwStructSize = sizeof(uc);
|
|
uc.lpszScheme = szScheme;
|
|
uc.dwSchemeLength = ARRAYSIZE(szScheme);
|
|
uc.lpszHostName = szHost;
|
|
uc.dwHostNameLength = ARRAYSIZE(szHost);
|
|
uc.lpszUrlPath = szURL;
|
|
uc.dwUrlPathLength = ARRAYSIZE(szURL);
|
|
|
|
// uc.dwExtraInfoLength ???
|
|
|
|
// BUGBUG ???
|
|
// InternetCrackUrl doesn't work with "search:\\..."
|
|
|
|
if (MyCrackUrl(_pszUrl, 0, ICU_DECODE, &uc))
|
|
{
|
|
// TODO:
|
|
// process the URL string
|
|
|
|
if (!lstrcmpi(szScheme, SZMYPROTOCOL))
|
|
{
|
|
// if this is our "search:" protocol
|
|
|
|
// BUGBUG
|
|
// we need to look up the registry for the default protocol to use.
|
|
|
|
uc.lpszScheme = NULL;
|
|
uc.nScheme = INTERNET_SCHEME_HTTP;
|
|
uc.nPort = INTERNET_DEFAULT_HTTP_PORT;
|
|
|
|
if (uc.dwHostNameLength)
|
|
{
|
|
lpSearchName = uc.lpszHostName;
|
|
uc.dwHostNameLength = 0;
|
|
|
|
// ???
|
|
// should we clear the UrlPath and ExtraInfo
|
|
}
|
|
else if (uc.dwUrlPathLength)
|
|
{
|
|
lpSearchName = uc.lpszUrlPath;
|
|
uc.lpszUrlPath = NULL;
|
|
}
|
|
else
|
|
lpSearchName = NULL;
|
|
|
|
if (lpSearchName)
|
|
{
|
|
// apply the search.
|
|
// ==========================================================
|
|
if (!lstrcmpi(lpSearchName, "united airline"))
|
|
lpNewName = "www.ual.com";
|
|
else if (!lstrcmpi(lpSearchName, "foo bar"))
|
|
lpNewName = "msw";
|
|
// ==========================================================
|
|
|
|
if (lpNewName)
|
|
{
|
|
// this is a searchable string
|
|
|
|
uc.lpszHostName = lpNewName;
|
|
|
|
dwNeeded = ARRAYSIZE(_szNewUrl);
|
|
|
|
if (InternetCreateUrl(&uc, 0, _szNewUrl, &dwNeeded))
|
|
hr = NOERROR;
|
|
}
|
|
}
|
|
}
|
|
else if (uc.nScheme == INTERNET_SCHEME_HTTP)
|
|
{
|
|
lpSearchName = uc.lpszHostName;
|
|
|
|
// apply search
|
|
// ==========================================================
|
|
if (!lstrcmpi(lpSearchName, "united airline"))
|
|
lpNewName = "www.ual.com";
|
|
else if (!lstrcmpi(lpSearchName, "foo bar"))
|
|
lpNewName = "msw";
|
|
// ==========================================================
|
|
|
|
if (lpNewName)
|
|
{
|
|
// if search succeeded
|
|
uc.lpszHostName = lpNewName;
|
|
uc.dwHostNameLength = 0;
|
|
|
|
dwNeeded = ARRAYSIZE(_szNewUrl);
|
|
|
|
if (InternetCreateUrl(&uc, 0, _szNewUrl, &dwNeeded))
|
|
hr = NOERROR;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
DebugBreak();
|
|
|
|
DWORD dwError = GetLastError();
|
|
}
|
|
|
|
if (hr != NOERROR)
|
|
{
|
|
_pProtSink->ReportResult(hr, 0, 0);
|
|
}
|
|
|
|
TransDebugOut((DEB_PROT, "%p OUT CResProtocol::Bind (hr:%lx)\n", this,hr));
|
|
return hr;
|
|
}
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: LookupProtocolClsIDFromReg
|
|
//
|
|
// Synopsis: finds a protocol handler class for a given URL
|
|
//
|
|
// Arguments: [pwzUrl] --
|
|
// [pclsid] --
|
|
//
|
|
// Returns:
|
|
//
|
|
// History: 11-01-1996 JohannP (Johann Posch) Created
|
|
//
|
|
// Notes:
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
HRESULT LookupProtocolClsIDFromReg(LPCTSTR pszUrl, CLSID *pclsid)
|
|
{
|
|
TransDebugOut((DEB_PROT, "API _IN LookupProtocolClsIDFromReg (szUrl >%s< )\n", pszUrl));
|
|
HRESULT hr = INET_E_UNKNOWN_PROTOCOL;
|
|
DWORD dwType;
|
|
TCHAR pszProt[MAX_URL_SIZE + 1];
|
|
|
|
TransAssert((pszUrl && pclsid));
|
|
|
|
if (pszUrl)
|
|
{
|
|
char szDelimiter = ':';
|
|
|
|
lstrcpy(pszProt, pszUrl);
|
|
|
|
LPSTR pszDel = StrChr(pszProt, szDelimiter);
|
|
|
|
if (pszDel)
|
|
{
|
|
*pszDel = '\0';
|
|
|
|
// fail if the protocol is "search" so we don't get call recursively.
|
|
if (lstrcmpi(pszProt, SZMYPROTOCOL))
|
|
{
|
|
HKEY hProtocolKey = NULL;
|
|
DWORD dwLen = 256;
|
|
char szProtocolKey[256];
|
|
|
|
lstrcpy(szProtocolKey, SZPROTOCOLROOT);
|
|
lstrcat(szProtocolKey, pszProt);
|
|
|
|
if (RegOpenKeyEx(HKEY_CLASSES_ROOT, szProtocolKey, 0, KEY_QUERY_VALUE, &hProtocolKey) == ERROR_SUCCESS)
|
|
{
|
|
if (RegQueryValueEx(hProtocolKey, SZCLASS, NULL, &dwType, (LPBYTE)szProtocolKey, &dwLen) == ERROR_SUCCESS)
|
|
{
|
|
LPWSTR pwzClsId = DupA2W(szProtocolKey);
|
|
|
|
if (pwzClsId)
|
|
{
|
|
hr = CLSIDFromString(pwzClsId, pclsid);
|
|
TransDebugOut((DEB_PROT, "API FOUND LookupProtocolClsIDFromReg(hr:%lx, ClsId:%ws)\n", hr,pwzClsId));
|
|
delete pwzClsId;
|
|
}
|
|
else
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
}
|
|
|
|
RegCloseKey(hProtocolKey);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// look up the registry
|
|
hr = MK_E_SYNTAX;
|
|
}
|
|
}
|
|
|
|
TransDebugOut((DEB_PROT, "API OUT LookupProtocolClsIDFromReg(hr:%lx)\n", hr));
|
|
return hr;
|
|
}
|
|
|
|
|