windows-nt/Source/XPSP1/NT/shell/shdocvw/tlog.cpp
2020-09-26 16:20:57 +08:00

2933 lines
75 KiB
C++

#include "priv.h"
#include "dspsprt.h"
#include <hlink.h>
#include "iface.h"
#include "resource.h"
#include <mluisupp.h>
#include "shdocfl.h"
class CTravelLog;
class CEnumEntry : public IEnumTravelLogEntry
{
public:
// *** IUnknown
STDMETHODIMP QueryInterface(REFIID riid, void **ppvObj);
STDMETHODIMP_(ULONG) AddRef();
STDMETHODIMP_(ULONG) Release();
// *** IEnumTravelLogEntry specific methods
STDMETHODIMP Next(ULONG cElt, ITravelLogEntry **rgElt, ULONG *pcEltFetched);
STDMETHODIMP Skip(ULONG cElt);
STDMETHODIMP Reset();
STDMETHODIMP Clone(IEnumTravelLogEntry **ppEnum);
CEnumEntry();
void Init(CTravelLog *ptl, IUnknown *punk, DWORD dwOffset, DWORD dwFlags);
void SetBase();
protected:
~CEnumEntry();
LONG _cRef;
DWORD _dwFlags;
DWORD _dwOffset;
LONG _lStart;
CTravelLog *_ptl;
IUnknown *_punk;
};
class CTravelEntry : public ITravelEntry,
public ITravelLogEntry,
public IPropertyBag
{
public:
CTravelEntry(BOOL fIsLocalAnchor);
// *** IUnknown
STDMETHODIMP QueryInterface(REFIID riid, void **ppvObj);
STDMETHODIMP_(ULONG) AddRef();
STDMETHODIMP_(ULONG) Release();
// *** ITravelEntry specific methods
STDMETHODIMP Update(IUnknown *punk, BOOL fIsLocalAnchor);
STDMETHODIMP Invoke(IUnknown *punk);
STDMETHODIMP GetPidl(LPITEMIDLIST *ppidl);
// *** ITravelLogEntry specific methods
STDMETHODIMP GetTitle(LPOLESTR *ppszTitle);
STDMETHODIMP GetURL(LPOLESTR *ppszURL);
// *** IPropertyBag specific methods
STDMETHODIMP Read(LPCOLESTR pszPropName, VARIANT *pVar, IErrorLog *pErrorLog);
STDMETHODIMP Write(LPCOLESTR pszPropName, VARIANT *pVar);
static HRESULT CreateTravelEntry(IBrowserService *pbs, BOOL fIsLocalAnchor, CTravelEntry **ppte);
void SetPrev(CTravelEntry *ptePrev);
void SetNext(CTravelEntry *pteNext);
CTravelEntry *GetPrev() {return _ptePrev;}
CTravelEntry *GetNext() {return _pteNext;}
void RemoveSelf();
BOOL CanInvoke(IUnknown *punk, BOOL fAllowLocalAnchor);
HRESULT GetIndexBrowser(IUnknown *punkIn, IUnknown ** ppsbOut) const;
DWORD Size();
DWORD ListSize();
HRESULT Clone(CTravelEntry **ppte);
HRESULT UpdateExternal(IUnknown *punk, IUnknown *punkHLBrowseContext);
HRESULT UpdateSelf(IUnknown *punk)
{return Update(punk, (_type == TET_LOCALANCHOR));}
BOOL IsExternal(void)
{ return (_type==TET_EXTERNALNAV); }
HRESULT GetDisplayName(LPTSTR psz, DWORD cch, DWORD dwFlags);
BOOL IsEqual(LPCITEMIDLIST pidl)
{return ILIsEqual(pidl, _pidl);}
BOOL IsLocalAnchor(void)
{ return (_type==TET_LOCALANCHOR);}
protected:
CTravelEntry(void);
HRESULT _InvokeExternal(IUnknown *punk);
HRESULT _UpdateTravelLog(IUnknown *punk, BOOL fIsLocalAnchor);
HRESULT _UpdateFromTLClient(IUnknown * punk, IStream ** ppStream);
LONG _cRef;
~CTravelEntry();
void _Reset(void);
enum {
TET_EMPTY = 0,
TET_DEFAULT = 1,
TET_LOCALANCHOR,
TET_EXTERNALNAV
#ifdef FEATURE_STARTPAGE
, TET_CLOSEDOWN
#endif
};
DWORD _type; // flags for our own sake...
LPITEMIDLIST _pidl; // pidl of the entry
HGLOBAL _hGlobalData; // the stream data saved by the entry
DWORD _bid; // the BrowserIndex for frame specific navigation
DWORD _dwCookie; // if _hGlobalData is NULL the cookie should be set
WCHAR * _pwzTitle;
WCHAR * _pwzUrlLocation;
IHlink *_phl;
IHlinkBrowseContext *_phlbc;
IPropertyBag *_ppb;
CTravelEntry *_ptePrev;
CTravelEntry *_pteNext;
};
CTravelEntry::CTravelEntry(BOOL fIsLocalAnchor) : _cRef(1)
{
//these should always be allocated
// thus they will always start 0
if (fIsLocalAnchor)
_type = TET_LOCALANCHOR;
else
ASSERT(!_type);
ASSERT(!_pwzTitle);
ASSERT(!_pwzUrlLocation);
ASSERT(!_pidl);
ASSERT(!_hGlobalData);
ASSERT(!_bid);
ASSERT(!_dwCookie);
ASSERT(!_ptePrev);
ASSERT(!_pteNext);
ASSERT(!_phl);
ASSERT(!_ppb);
ASSERT(!_phlbc);
TraceMsg(TF_TRAVELLOG, "TE[%X] created _type = %x", this, _type);
#ifdef FEATURE_STARTPAGE
if (fIsLocalAnchor == 42) // HACK ALERT! Special value for CloseDown
{
_type = TET_CLOSEDOWN;
_bid = -1; // BID_TOPFRAMEBROWSER;
_pwzTitle = StrDup(TEXT("Close"));
}
#endif
}
CTravelEntry::CTravelEntry(void) : _cRef(1)
{
ASSERT(!_type);
ASSERT(!_pwzTitle);
ASSERT(!_pwzUrlLocation);
ASSERT(!_pidl);
ASSERT(!_hGlobalData);
ASSERT(!_bid);
ASSERT(!_dwCookie);
ASSERT(!_ptePrev);
ASSERT(!_pteNext);
ASSERT(!_phl);
ASSERT(!_ppb);
ASSERT(!_phlbc);
TraceMsg(TF_TRAVELLOG, "TE[%X] created", this, _type);
}
HGLOBAL CloneHGlobal(HGLOBAL hGlobalIn)
{
DWORD dwSize = (DWORD)GlobalSize(hGlobalIn);
HGLOBAL hGlobalOut = GlobalAlloc(GlobalFlags(hGlobalIn), dwSize);
HGLOBAL hGlobalResult = NULL;
if (NULL != hGlobalOut)
{
LPVOID pIn= GlobalLock(hGlobalIn);
if (NULL != pIn)
{
LPVOID pOut= GlobalLock(hGlobalOut);
if (NULL != pOut)
{
memcpy(pOut, pIn, dwSize);
GlobalUnlock(hGlobalOut);
hGlobalResult = hGlobalOut;
}
GlobalUnlock(hGlobalIn);
}
if (!hGlobalResult)
{
GlobalFree(hGlobalOut);
hGlobalOut = NULL;
}
}
return hGlobalResult;
}
HRESULT
CTravelEntry::Clone(CTravelEntry **ppte)
{
// dont ever clone an external entry
if (_type == TET_EXTERNALNAV)
return E_FAIL;
CTravelEntry *pte = new CTravelEntry();
HRESULT hr = S_OK;
if (pte)
{
pte->_type = _type;
pte->_bid = _bid;
pte->_dwCookie = _dwCookie;
if (_pwzTitle)
{
pte->_pwzTitle = StrDup(_pwzTitle);
if (!pte->_pwzTitle)
{
hr = E_OUTOFMEMORY;
}
}
if (_pwzUrlLocation)
{
pte->_pwzUrlLocation = StrDup(_pwzUrlLocation);
if (!pte->_pwzUrlLocation)
{
hr = E_OUTOFMEMORY;
}
}
if (_pidl)
{
pte->_pidl = ILClone(_pidl);
if (!pte->_pidl)
hr = E_OUTOFMEMORY;
}
else
pte->_pidl = NULL;
if (_hGlobalData)
{
pte->_hGlobalData = CloneHGlobal(_hGlobalData);
if (NULL == pte->_hGlobalData)
{
hr = E_OUTOFMEMORY;
}
}
else
{
ASSERT(NULL == pte->_hGlobalData);
}
}
else
hr = E_OUTOFMEMORY;
if (FAILED(hr) && pte)
{
pte->Release();
*ppte = NULL;
}
else
*ppte = pte;
TraceMsg(TF_TRAVELLOG, "TE[%X] Clone hr = %x", this, hr);
return hr;
}
CTravelEntry::~CTravelEntry()
{
ILFree(_pidl);
if (_hGlobalData)
{
GlobalFree(_hGlobalData);
_hGlobalData = NULL;
}
if (_pwzTitle)
{
LocalFree(_pwzTitle);
_pwzTitle = NULL;
}
if (_pwzUrlLocation)
{
LocalFree(_pwzUrlLocation);
_pwzUrlLocation = NULL;
}
if (_pteNext)
{
_pteNext->Release();
}
// Don't need to release _ptePrev because TravelEntry only addref's pteNext
ATOMICRELEASE(_ppb);
ATOMICRELEASE(_phl);
ATOMICRELEASE(_phlbc);
TraceMsg(TF_TRAVELLOG, "TE[%X] destroyed ", this);
}
HRESULT CTravelEntry::QueryInterface(REFIID riid, void **ppvObj)
{
static const QITAB qit[] = {
QITABENT(CTravelEntry, ITravelEntry), // IID_ITravelEntry
QITABENT(CTravelEntry, ITravelLogEntry),
QITABENT(CTravelEntry, IPropertyBag),
{ 0 },
};
return QISearch(this, qit, riid, ppvObj);
}
ULONG CTravelEntry::AddRef()
{
return InterlockedIncrement(&_cRef);
}
ULONG CTravelEntry::Release()
{
if (InterlockedDecrement(&_cRef))
return _cRef;
delete this;
return 0;
}
//+-------------------------------------------------------------------------
//
// Method : CTravelEntry::GetIndexBrowser
//
// Synopsis : This method finds and returns the IUnknown of the browser
// with the index in _bid. This method first checks to see
// if the passed in punk supports ITravelLogClient. If it
// doesn't, it checks for IBrowserService.
//
//--------------------------------------------------------------------------
HRESULT
CTravelEntry::GetIndexBrowser(IUnknown * punk, IUnknown ** ppunkBrowser) const
{
HRESULT hr = E_FAIL;
ITravelLogClient * ptlcTop = NULL;
ASSERT(ppunkBrowser);
hr = punk->QueryInterface(IID_PPV_ARG(ITravelLogClient, &ptlcTop));
if (SUCCEEDED(hr))
{
hr = ptlcTop->FindWindowByIndex(_bid, ppunkBrowser);
}
TraceMsg(TF_TRAVELLOG, "TE[%X]::GetIndexBrowser _bid = %X, hr = %X", this, _bid, hr);
SAFERELEASE(ptlcTop);
return hr;
}
//+-------------------------------------------------------------------------
//
// Method : CTravelEntry::CanInvoke
//
// Synopsis : This method determines if the current travel entry can
// be invoked. There are two criteria that determine if
// this entry can be invoked.
// 1) If the entry is a local anchor, fAllowLocalAnchor must
// be TRUE.
// 2) A browser with the index in _bid must exist.
//
//--------------------------------------------------------------------------
BOOL CTravelEntry::CanInvoke(IUnknown *punk, BOOL fAllowLocalAnchor)
{
IUnknown * punkBrowser = NULL;
BOOL fRet = IsLocalAnchor() ? fAllowLocalAnchor : TRUE;
if (fRet)
fRet = fRet && SUCCEEDED(GetIndexBrowser(punk, &punkBrowser));
SAFERELEASE(punkBrowser);
return fRet;
}
DWORD CTravelEntry::Size()
{
DWORD cbSize = SIZEOF(*this);
if (_pidl)
cbSize += ILGetSize(_pidl);
if (_hGlobalData)
{
cbSize += (DWORD)GlobalSize(_hGlobalData);
}
if (_pwzTitle)
{
cbSize += (DWORD)LocalSize(_pwzTitle);
}
if (_pwzUrlLocation)
{
cbSize += (DWORD)LocalSize(_pwzUrlLocation);
}
return cbSize;
}
DWORD CTravelEntry::ListSize()
{
CTravelEntry *pte = GetNext();
DWORD cb = Size();
while (pte)
{
cb += pte->Size();
pte = pte->GetNext();
}
return cb;
}
void CTravelEntry::_Reset()
{
Pidl_Set(&_pidl, NULL);
if (NULL != _hGlobalData)
{
GlobalFree(_hGlobalData);
_hGlobalData = NULL;
}
ATOMICRELEASE(_phl);
ATOMICRELEASE(_phlbc);
_bid = 0;
_type = TET_EMPTY;
_dwCookie = 0;
if (_pwzTitle)
{
LocalFree(_pwzTitle);
_pwzTitle = NULL;
}
if (_pwzUrlLocation)
{
LocalFree(_pwzUrlLocation);
_pwzUrlLocation = NULL;
}
TraceMsg(TF_TRAVELLOG, "TE[%X]::_Reset", this);
}
HRESULT CTravelEntry::_UpdateTravelLog(IUnknown *punk, BOOL fIsLocalAnchor)
{
IBrowserService *pbs;
HRESULT hr = E_FAIL;
// we need to update here
if (SUCCEEDED(punk->QueryInterface(IID_PPV_ARG(IBrowserService, &pbs))))
{
ITravelLog *ptl;
if (SUCCEEDED(pbs->GetTravelLog(&ptl)))
{
hr = ptl->UpdateEntry(punk, fIsLocalAnchor);
ptl->Release();
}
pbs->Release();
}
return hr;
}
HRESULT CTravelEntry::_InvokeExternal(IUnknown *punk)
{
HRESULT hr = E_FAIL;
ASSERT(_phl);
ASSERT(_phlbc);
TraceMsg(TF_TRAVELLOG, "TE[%X]::InvokeExternal entered on _bid = %X, _phl = %X, _phlbc = %X", this, _bid, _phl, _phlbc);
// set the size and position of the browser frame window, so that the
// external target can sync up its frame window to those coordinates
HLBWINFO hlbwi;
hlbwi.cbSize = sizeof(hlbwi);
hlbwi.grfHLBWIF = 0;
IOleWindow *pow;
HWND hwnd = NULL;
if (SUCCEEDED(punk->QueryInterface(IID_PPV_ARG(IOleWindow, &pow))))
{
pow->GetWindow(&hwnd);
pow->Release();
}
if (hwnd)
{
WINDOWPLACEMENT wp = {0};
wp.length = sizeof(WINDOWPLACEMENT);
GetWindowPlacement(hwnd, &wp);
hlbwi.grfHLBWIF = HLBWIF_HASFRAMEWNDINFO;
hlbwi.rcFramePos = wp.rcNormalPosition;
if (wp.showCmd == SW_SHOWMAXIMIZED)
hlbwi.grfHLBWIF |= HLBWIF_FRAMEWNDMAXIMIZED;
}
_phlbc->SetBrowseWindowInfo(&hlbwi);
//
// right now we always now we are going back, but later on
// maybe we should ask the browser whether this is back or forward
//
hr = _phl->Navigate(HLNF_NAVIGATINGBACK, NULL, NULL, _phlbc);
IServiceProvider *psp;
if (SUCCEEDED(punk->QueryInterface(IID_PPV_ARG(IServiceProvider, &psp))))
{
IWebBrowser2 *pwb;
ASSERT(psp);
if (SUCCEEDED(psp->QueryService(SID_SWebBrowserApp, IID_IWebBrowser2, (void **)&pwb)))
{
ASSERT(pwb);
pwb->put_Visible(FALSE);
pwb->Release();
}
psp->Release();
}
_UpdateTravelLog(punk, FALSE);
TraceMsg(TF_TRAVELLOG, "TE[%X]::InvokeExternal exited hr = %X", this, hr);
return hr;
}
HRESULT CTravelEntry::Invoke(IUnknown *punk)
{
IPersistHistory *pph = NULL;
HRESULT hr = E_FAIL;
IUnknown * punkBrowser = NULL;
IHTMLWindow2 * pWindow = NULL;
TraceMsg(TF_TRAVELLOG, "TE[%X]::Invoke entered on _bid = %X", this, _bid);
TraceMsgW(TF_TRAVELLOG, "TE[%X]::Invoke title '%s'", this, _pwzTitle);
if (_type == TET_EXTERNALNAV)
{
hr = _InvokeExternal(punk);
goto Quit;
}
// Get the window/browser with the index. If that
// fails, punk may be a IHTMLWindow2. If so,
// get its IPersistHistory so the travel entry
// can be loaded directly. (This is needed by Trident
// in order to navigate in frames when traveling
// backwards or forwards.
//
hr = GetIndexBrowser(punk, &punkBrowser);
if (!hr)
{
hr = punkBrowser->QueryInterface(IID_PPV_ARG(IPersistHistory, &pph));
}
else
{
hr = punk->QueryInterface(IID_PPV_ARG(IHTMLWindow2, &pWindow));
if (!hr)
{
hr = pWindow->QueryInterface(IID_PPV_ARG(IPersistHistory, &pph));
}
}
if (SUCCEEDED(hr))
{
ASSERT(pph);
#ifdef FEATURE_STARTPAGE
if (_type == TET_CLOSEDOWN)
{
HWND hwnd;
hr = IUnknown_GetWindow(punkBrowser, &hwnd);
if (SUCCEEDED(hr))
PostMessage(hwnd, WM_CLOSE, 0, 0);
goto Quit;
}
#endif
if (_type == TET_LOCALANCHOR)
{
ITravelLogClient * pTLClient;
hr = pph->QueryInterface(IID_PPV_ARG(ITravelLogClient, &pTLClient));
if (SUCCEEDED(hr))
{
hr = pTLClient->LoadHistoryPosition(_pwzUrlLocation, _dwCookie);
pTLClient->Release();
}
else
{
hr = pph->SetPositionCookie(_dwCookie);
}
}
else
{
// we need to clone it
ASSERT(_hGlobalData);
HGLOBAL hGlobal = CloneHGlobal(_hGlobalData);
if (NULL != hGlobal)
{
IStream *pstm;
hr = CreateStreamOnHGlobal(hGlobal, TRUE, &pstm);
if (SUCCEEDED(hr))
{
hr = pph->LoadHistory(pstm, NULL);
pstm->Release();
}
else
{
GlobalFree(hGlobal);
hGlobal = NULL;
}
}
else
{
hr = E_OUTOFMEMORY;
}
}
pph->Release();
}
Quit:
SAFERELEASE(punkBrowser);
SAFERELEASE(pWindow);
TraceMsg(TF_TRAVELLOG, "TE[%X]::Invoke exited on _bid = %X, hr = %X", this, _bid, hr);
return hr;
}
HRESULT CTravelEntry::UpdateExternal(IUnknown *punk, IUnknown *punkHLBrowseContext)
{
TraceMsg(TF_TRAVELLOG, "TE[%X]::UpdateExternal entered on punk = %X, punkhlbc = %X", this, punk, punkHLBrowseContext);
_Reset();
ASSERT(punkHLBrowseContext);
punkHLBrowseContext->QueryInterface(IID_PPV_ARG(IHlinkBrowseContext, &_phlbc));
ASSERT(_phlbc);
_type = TET_EXTERNALNAV;
HRESULT hr = E_FAIL;
//
// right now we only support externals being previous. we never actually navigate
// to another app. we handle everything in pane ourselves.
// so theoretically we never need to worry about HLID_NEXT
_phlbc->GetHlink((ULONG) HLID_PREVIOUS, &_phl);
IBrowserService *pbs;
punk->QueryInterface(IID_PPV_ARG(IBrowserService, &pbs));
if (pbs && _phl)
{
_bid = pbs->GetBrowserIndex();
WCHAR *pwszTarget;
hr = _phl->GetStringReference(HLINKGETREF_ABSOLUTE, &pwszTarget, NULL);
if (SUCCEEDED(hr))
{
TCHAR szName[MAX_URL_STRING];
StrCpyN(szName, pwszTarget, ARRAYSIZE(szName));
OleFree(pwszTarget);
// create pidl
hr = IECreateFromPath(szName, &_pidl);
}
}
ATOMICRELEASE(pbs);
TraceMsg(TF_TRAVELLOG, "TE[%X]::UpdateExternal exited _bid = %X, hr = %X", this, _bid, hr);
return hr;
}
HRESULT CTravelEntry::Update(IUnknown *punk, BOOL fIsLocalAnchor)
{
HRESULT hr, hrStat;
STATSTG stg;
IStream * pstm = NULL;
IPersistHistory * pph = NULL;
ASSERT(punk);
// this means that we went back to an external app,
// and now we are going forward again. we dont persist
// any state info about them that would be different.
if (_type == TET_EXTERNALNAV)
{
TraceMsg(TF_TRAVELLOG, "TE[%X]::Update NOOP on external entry", this);
return S_OK;
}
_Reset();
// Try ITravelLogClient first. If that fails, revert to IBrowserService.
//
hr = _UpdateFromTLClient(punk, &pstm);
if (hr)
goto Cleanup;
hr = punk->QueryInterface(IID_PPV_ARG(IPersistHistory, &pph));
ASSERT(SUCCEEDED(hr));
if (hr)
goto Cleanup;
if (fIsLocalAnchor)
{
// persist a cookie
//
_type = TET_LOCALANCHOR;
hr = pph->GetPositionCookie(&_dwCookie);
}
else
{
_type = TET_DEFAULT;
// persist a stream
//
ASSERT(!_hGlobalData);
if (!pstm)
{
hr = CreateStreamOnHGlobal(NULL, FALSE, &pstm);
if (hr)
goto Cleanup;
pph->SaveHistory(pstm);
}
hrStat = pstm->Stat(&stg, STATFLAG_NONAME);
hr = GetHGlobalFromStream(pstm, &_hGlobalData);
// This little exercise here is to shrink the memory block we get from
// the OLE API which allocates blocks in chunks of 8KB. Typical stream
// sizes are only a few hundred bytes.
if (hrStat)
goto Cleanup;
HGLOBAL hGlobalTemp = GlobalReAlloc(_hGlobalData, stg.cbSize.LowPart, GMEM_MOVEABLE);
if (NULL != hGlobalTemp)
{
_hGlobalData = hGlobalTemp;
}
}
Cleanup:
if (FAILED(hr))
_Reset();
SAFERELEASE(pstm);
SAFERELEASE(pph);
TraceMsg(TF_TRAVELLOG, "TE[%X]::Update exited on _bid = %X, hr = %X", this, _bid, hr);
return hr;
}
//+-----------------------------------------------------------------------------
//
// Method : CTravelEntry::_UpdateFromTLClient
//
// Synopsis : Updates the travel entry using the ITravelLogClient interface
//
//------------------------------------------------------------------------------
HRESULT
CTravelEntry::_UpdateFromTLClient(IUnknown * punk, IStream ** ppStream)
{
HRESULT hr;
WINDOWDATA windata = {0};
ITravelLogClient * ptlc = NULL;
hr = punk->QueryInterface(IID_PPV_ARG(ITravelLogClient, &ptlc));
if (hr)
goto Cleanup;
hr = ptlc->GetWindowData(&windata);
if (hr)
goto Cleanup;
_bid = windata.dwWindowID;
ILFree(_pidl);
if (windata.pidl)
{
_pidl = ILClone(windata.pidl);
}
else
{
hr = IEParseDisplayNameWithBCW(windata.uiCP, windata.lpszUrl, NULL, &_pidl);
if (hr)
goto Cleanup;
}
ASSERT(_pidl);
// If there is an url location, append it to the end of the url
//
if (_pwzUrlLocation)
{
LocalFree(_pwzUrlLocation);
_pwzUrlLocation = NULL;
}
if (windata.lpszUrlLocation && *windata.lpszUrlLocation)
{
_pwzUrlLocation = StrDup(windata.lpszUrlLocation);
}
// Pick up the title as a display name for menus and such.
//
if (_pwzTitle)
{
LocalFree(_pwzTitle);
_pwzTitle = NULL;
}
if (windata.lpszTitle)
_pwzTitle = StrDup(windata.lpszTitle);
*ppStream = windata.pStream;
TraceMsgW(TF_TRAVELLOG, "TE[%X]::_UpdateFromTLClient - ptlc:[0x%X] _bid:[%ld] Url:[%ws] Title:[%ws] UrlLocation:[%ws] ppStream:[0x%X]",
this, ptlc, _bid, windata.lpszUrl, _pwzTitle, _pwzUrlLocation, *ppStream);
Cleanup:
ILFree(windata.pidl);
CoTaskMemFree(windata.lpszUrl);
CoTaskMemFree(windata.lpszUrlLocation);
CoTaskMemFree(windata.lpszTitle);
SAFERELEASE(ptlc);
// Don't release windata.pStream. It will
// be released when ppStream is released.
return hr;
}
HRESULT CTravelEntry::GetPidl(LPITEMIDLIST * ppidl)
{
if (EVAL(ppidl))
{
*ppidl = ILClone(_pidl);
if (*ppidl)
return S_OK;
}
return E_FAIL;
}
void CTravelEntry::SetNext(CTravelEntry *pteNext)
{
if (_pteNext)
_pteNext->Release();
_pteNext = pteNext;
if (_pteNext)
{
_pteNext->_ptePrev = this;
}
}
void CTravelEntry::SetPrev(CTravelEntry *ptePrev)
{
_ptePrev = ptePrev;
if (_ptePrev)
_ptePrev->SetNext(this);
}
//
// this is for removing from the middle of the list...
//
void CTravelEntry::RemoveSelf()
{
if (_pteNext)
_pteNext->_ptePrev = _ptePrev;
// remove yourself from the list
if (_ptePrev)
{
// after this point, we may be destroyed so can't touch any more member vars
_ptePrev->_pteNext = _pteNext;
}
_ptePrev = NULL;
_pteNext = NULL;
// we lose a reference now because we're gone from _ptePrev's _pteNext
// (or if we were the top of the list, we're also nuked)
Release();
}
HRESULT GetUnescapedUrlIfAppropriate(LPCITEMIDLIST pidl, LPTSTR pszUrl, DWORD cch)
{
TCHAR szUrl[MAX_URL_STRING];
// The SHGDN_NORMAL display name will be the pretty name (Web Page title) unless
// it's an FTP URL or the web page didn't set a title.
if (SUCCEEDED(IEGetDisplayName(pidl, szUrl, SHGDN_NORMAL)) &&
UrlIs(szUrl, URLIS_URL))
{
// NT #279192, If an URL is escaped, it normally contains three types of
// escaped chars.
// 1) Seperating type chars ('#' for frag, '?' for params, etc.)
// 2) DBCS chars,
// 3) Data (a bitmap in the url by escaping the binary bytes)
// Since #2 is very common, we want to try to unescape it so it has meaning
// to the user. UnEscaping isn't safe if the user can copy or modify the data
// because they could loose data when it's reparsed. One thing we need to
// do for #2 to work is for it to be in ANSI when unescaped. This is needed
// or the DBCS lead and trail bytes will be in unicode as [0x<LeadByte> 0x00]
// [0x<TrailByte> 0x00]. Being in ANSI could cause a problem if the the string normally
// crosses code pages, but that is uncommon or non-existent in the IsURLChild()
// case.
CHAR szUrlAnsi[MAX_URL_STRING];
SHTCharToAnsi(szUrl, szUrlAnsi, ARRAYSIZE(szUrlAnsi));
UrlUnescapeA(szUrlAnsi, NULL, NULL, URL_UNESCAPE_INPLACE|URL_UNESCAPE_HIGH_ANSI_ONLY);
SHAnsiToTChar(szUrlAnsi, pszUrl, cch);
}
else
{
StrCpyN(pszUrl, szUrl, cch); // Truncate if needed
}
return S_OK;
}
#define TEGDN_FORSYSTEM 0x00000001
HRESULT CTravelEntry::GetDisplayName(LPTSTR psz, DWORD cch, DWORD dwFlags)
{
if (!psz || !cch)
return E_INVALIDARG;
psz[0] = 0;
if ((NULL != _pwzTitle) && (*_pwzTitle != 0))
{
StrCpyNW(psz, _pwzTitle, cch);
}
else if (_pidl)
{
GetUnescapedUrlIfAppropriate(_pidl, psz, cch);
}
if (dwFlags & TEGDN_FORSYSTEM)
{
if (!SHIsDisplayable(psz, g_fRunOnFE, g_bRunOnNT5))
{
// Display name isn't system-displayable. Just use the path/url instead.
SHTitleFromPidl(_pidl, psz, cch, FALSE);
}
}
SHCleanupUrlForDisplay(psz);
return psz[0] ? S_OK : E_FAIL;
}
HRESULT CTravelEntry::GetTitle(LPOLESTR *ppszTitle)
{
HRESULT hres = S_OK;
TCHAR szTitle[MAX_BROWSER_WINDOW_TITLE];
ASSERT(IS_VALID_WRITE_PTR(ppszTitle, LPOLESTR));
hres = GetDisplayName(szTitle, MAX_BROWSER_WINDOW_TITLE, TEGDN_FORSYSTEM);
if(SUCCEEDED(hres))
{
ASSERT(*szTitle);
hres = SHStrDup(szTitle, ppszTitle);
}
return hres;
}
HRESULT CTravelEntry::GetURL(LPOLESTR *ppszUrl)
{
HRESULT hres = E_FAIL;
LPITEMIDLIST pidl = NULL;
WCHAR wszURL[MAX_URL_STRING];
if(_pidl)
hres = ::IEGetDisplayName(_pidl, wszURL, SHGDN_FORPARSING);
if(SUCCEEDED(hres))
hres = SHStrDup(wszURL, ppszUrl);
return hres;
}
HRESULT CTravelEntry::Read(LPCOLESTR pszPropName, VARIANT *pVar, IErrorLog *pErrorLog)
{
if(!_ppb)
{
return E_INVALIDARG;
}
return _ppb->Read(pszPropName, pVar, pErrorLog);
}
HRESULT CTravelEntry::Write(LPCOLESTR pszPropName, VARIANT *pVar)
{
HRESULT hres = S_OK;
if(!_ppb)
{
hres = SHCreatePropertyBagOnMemory(STGM_READWRITE, IID_IPropertyBag, (void **) &_ppb);
}
if(SUCCEEDED(hres))
{
ASSERT(_ppb);
hres = _ppb->Write(pszPropName, pVar);
}
return hres;
}
class CTravelLog : public ITravelLog,
public ITravelLogEx
{
public:
// *** IUnknown
STDMETHODIMP QueryInterface(REFIID riid, void **ppvObj);
STDMETHODIMP_(ULONG) AddRef() ;
STDMETHODIMP_(ULONG) Release();
// *** ITravelLog specific methods
STDMETHODIMP AddEntry(IUnknown *punk, BOOL fIsLocalAnchor);
STDMETHODIMP UpdateEntry(IUnknown *punk, BOOL fIsLocalAnchor);
STDMETHODIMP UpdateExternal(IUnknown *punk, IUnknown *punkHLBrowseContext);
STDMETHODIMP Travel(IUnknown *punk, int iOffset);
STDMETHODIMP GetTravelEntry(IUnknown *punk, int iOffset, ITravelEntry **ppte);
STDMETHODIMP FindTravelEntry(IUnknown *punk, LPCITEMIDLIST pidl, ITravelEntry **ppte);
STDMETHODIMP GetToolTipText(IUnknown *punk, int iOffset, int idsTemplate, LPWSTR pwzText, DWORD cchText);
STDMETHODIMP InsertMenuEntries(IUnknown *punk, HMENU hmenu, int nPos, int idFirst, int idLast, DWORD dwFlags);
STDMETHODIMP Clone(ITravelLog **pptl);
STDMETHODIMP_(DWORD) CountEntries(IUnknown *punk);
STDMETHODIMP Revert(void);
// *** ITravelLogEx specific methods
STDMETHODIMP FindTravelEntryWithUrl(IUnknown * punk, UINT uiCP, LPOLESTR lpszUrl, ITravelEntry ** ppte);
STDMETHODIMP TravelToUrl(IUnknown * punk, UINT uiCP, LPOLESTR lpszUrl);
STDMETHOD(DeleteIndexEntry)(IUnknown *punk, int index);
STDMETHOD(DeleteUrlEntry)(IUnknown *punk, UINT uiCP, LPOLESTR pszUrl);
STDMETHOD(CountEntryNodes)(IUnknown *punk, DWORD dwFlags, DWORD *pdwCount);
STDMETHOD(CreateEnumEntry)(IUnknown *punk, IEnumTravelLogEntry **ppEnum, DWORD dwFlags);
STDMETHOD(DeleteEntry)(IUnknown *punk, ITravelLogEntry *pte);
STDMETHOD(InsertEntry)(IUnknown *punkBrowser, ITravelLogEntry *pteRelativeTo, BOOL fPrepend,
IUnknown* punkTLClient, ITravelLogEntry **ppEntry);
STDMETHOD(TravelToEntry)(IUnknown *punkBrowser, ITravelLogEntry *pteDestination);
CTravelLog();
protected:
~CTravelLog();
HRESULT _FindEntryByOffset(IUnknown *punk, int iOffset, CTravelEntry **ppte);
HRESULT _FindEntryByPidl(IUnknown * punk, LPCITEMIDLIST pidl, CTravelEntry ** ppte);
HRESULT _FindEntryByPunk(IUnknown * punk, ITravelLogEntry *pteSearch, CTravelEntry ** ppte);
void _DeleteFrameSetEntry(IUnknown *punk, CTravelEntry *pte);
void _Prune(void);
LONG _cRef;
DWORD _cbMaxSize;
DWORD _cbTotalSize;
CTravelEntry *_pteCurrent; //pteCurrent
CTravelEntry *_pteUpdate;
CTravelEntry *_pteRoot;
};
CTravelLog::CTravelLog() : _cRef(1)
{
ASSERT(!_pteCurrent);
ASSERT(!_pteUpdate);
ASSERT(!_pteRoot);
DWORD dwType, dwSize = SIZEOF(_cbMaxSize), dwDefault = 1024 * 1024;
SHRegGetUSValue(TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\TravelLog"), TEXT("MaxSize"), &dwType, (LPVOID)&_cbMaxSize, &dwSize, FALSE, (void *)&dwDefault, SIZEOF(dwDefault));
TraceMsg(TF_TRAVELLOG, "TL[%X] created", this);
}
CTravelLog::~CTravelLog()
{
//DestroyList by releasing the root
SAFERELEASE(_pteRoot);
TraceMsg(TF_TRAVELLOG, "TL[%X] destroyed ", this);
}
HRESULT CTravelLog::QueryInterface(REFIID riid, void **ppvObj)
{
static const QITAB qit[] = {
QITABENT(CTravelLog, ITravelLog), // IID_ITravelLog
QITABENT(CTravelLog, ITravelLogEx), // IID_ITravelLogEx
{ 0 },
};
return QISearch(this, qit, riid, ppvObj);
}
ULONG CTravelLog::AddRef()
{
return InterlockedIncrement(&_cRef);
}
ULONG CTravelLog::Release()
{
if (InterlockedDecrement(&_cRef))
return _cRef;
delete this;
return 0;
}
HRESULT CTravelLog::AddEntry(IUnknown *punk, BOOL fIsLocalAnchor)
{
ASSERT(punk);
if(SHRestricted2W(REST_NoNavButtons, NULL, 0))
{
return S_FALSE;
}
TraceMsg(TF_TRAVELLOG, "TL[%X]::AddEntry punk = %X, IsLocal = %s", this, punk, fIsLocalAnchor ? "TRUE" : "FALSE");
#ifdef FEATURE_STARTPAGE
if (!_pteCurrent)
{
if (SHRegGetBoolUSValue(REGSTR_PATH_EXPLORER, TEXT("FaultID"), FALSE, FALSE))
{
CTravelEntry *pteClose = new CTravelEntry(42);
if (pteClose)
{
_pteRoot = pteClose;
_pteCurrent = pteClose;
_cbTotalSize += pteClose->Size();
}
}
}
#endif
CTravelEntry *pte = new CTravelEntry(fIsLocalAnchor);
if (pte)
{
//replace the current with the new
if (_pteCurrent)
{
CTravelEntry *pteNext = _pteCurrent->GetNext();
if (pteNext)
{
_cbTotalSize -= pteNext->ListSize();
}
// the list keeps its own ref count, and only needs
// to be modified when passed outside of the list
// setnext will release the current next if necessary
// this will also set pte->prev = pteCurrent
_pteCurrent->SetNext(pte);
}
else
_pteRoot = pte;
_cbTotalSize += pte->Size();
_pteCurrent = pte;
ASSERT(_cbTotalSize == _pteRoot->ListSize());
}
TraceMsg(TF_TRAVELLOG, "TL[%X]::AddEntry punk = %X, IsLocal = %d, pte = %X", this, punk, fIsLocalAnchor, pte);
return pte ? S_OK : E_OUTOFMEMORY;
}
void CTravelLog::_Prune(void)
{
// FEATURE: need an increment or something
ASSERT(_cbTotalSize == _pteRoot->ListSize());
while (_cbTotalSize > _cbMaxSize && _pteRoot != _pteCurrent)
{
CTravelEntry *pte = _pteRoot;
_pteRoot = _pteRoot->GetNext();
_cbTotalSize -= pte->Size();
pte->RemoveSelf();
ASSERT(_cbTotalSize == _pteRoot->ListSize());
}
}
HRESULT CTravelLog::UpdateEntry(IUnknown *punk, BOOL fIsLocalAnchor)
{
CTravelEntry *pte = _pteUpdate ? _pteUpdate : _pteCurrent;
// this can happen under weird stress conditions, evidently
if (!pte)
return E_FAIL;
_cbTotalSize -= pte->Size();
HRESULT hr = pte->Update(punk, fIsLocalAnchor);
_cbTotalSize += pte->Size();
ASSERT(_cbTotalSize == _pteRoot->ListSize());
// Debug prints need to be before _Prune() since pte can get freed by _Prune() resulting
// in a crash if pte->Size() is called
TraceMsg(TF_TRAVELLOG, "TL[%X]::UpdateEntry pte->Size() = %d", this, pte->Size());
TraceMsg(TF_TRAVELLOG, "TL[%X]::UpdateEntry punk = %X, IsLocal = %d, hr = %X", this, punk, fIsLocalAnchor, hr);
_Prune();
_pteUpdate = NULL;
return hr;
}
HRESULT CTravelLog::UpdateExternal(IUnknown *punk, IUnknown *punkHLBrowseContext)
{
CTravelEntry *pte = _pteUpdate ? _pteUpdate : _pteCurrent;
ASSERT(punk);
ASSERT(pte);
ASSERT(punkHLBrowseContext);
if (pte)
return pte->UpdateExternal(punk, punkHLBrowseContext);
return E_FAIL;
}
HRESULT CTravelLog::Travel(IUnknown *punk, int iOffset)
{
ASSERT(punk);
HRESULT hr = E_FAIL;
CTravelEntry *pte;
TraceMsg(TF_TRAVELLOG, "TL[%X]::Travel entered with punk = %X, iOffset = %d", this, punk, iOffset);
if (SUCCEEDED(_FindEntryByOffset(punk, iOffset, &pte)))
{
#ifdef DEBUG
TCHAR szPath[MAX_PATH];
LPITEMIDLIST pidl;
pte->GetPidl(&pidl);
SHGetNameAndFlags(pidl, SHGDN_FORPARSING, szPath, ARRAYSIZE(szPath), NULL);
ILFree(pidl);
TraceMsgW(TF_TRAVELLOG, "TL[%X]::URL %s", this, szPath);
#endif
// we will update where we are before we move away...
// but external navigates dont go through the normal activation
// so we dont want to setup the external to be updated
// _pteUpdate is also what allows us to Revert().
if (!_pteCurrent->IsExternal() && !_pteUpdate)
_pteUpdate = _pteCurrent;
_pteCurrent = pte;
hr = _pteCurrent->Invoke(punk);
//
// if the entry bails with an error, then we need to reset ourself
// to what we were. right now, the only place this should happen
// is if an Abort was returned from SetPositionCookie
// because somebody aborted during before navigate.
// but i think that any error means that we can legitimately Revert().
//
if (FAILED(hr))
{
Revert();
}
}
TraceMsg(TF_TRAVELLOG, "TL[%X]::Travel exited with hr = %X", this, hr);
return hr;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::TravelToUrl
//
// Interface : ITravelLogEx
//
// Synopsis : Travels to the specified URL in the travel log.
//
//----------------------------------------------------------------------------
HRESULT
CTravelLog::TravelToUrl(IUnknown * punk, UINT uiCP, LPOLESTR lpszUrl)
{
ASSERT(punk);
ASSERT(lpszUrl);
HRESULT hr;
LPITEMIDLIST pidl;
CTravelEntry * pte = NULL;
DWORD cchOut = INTERNET_MAX_URL_LENGTH;
TCHAR szUrl[INTERNET_MAX_URL_LENGTH];
hr = UrlCanonicalize(lpszUrl, szUrl, &cchOut, URL_ESCAPE_SPACES_ONLY);
if (SUCCEEDED(hr))
{
hr = IEParseDisplayName(uiCP, szUrl, &pidl);
if (SUCCEEDED(hr))
{
hr = _FindEntryByPidl(punk, pidl, &pte);
ILFree(pidl);
if (SUCCEEDED(hr))
{
// We will update where we are before we move away...
// but external navigates don't go through the normal activation
// so we dont want to setup the external to be updated
// _pteUpdate is also what allows us to Revert().
//
if (!_pteCurrent->IsExternal() && !_pteUpdate)
{
_pteUpdate = _pteCurrent;
}
_pteCurrent = pte;
hr = _pteCurrent->Invoke(punk);
// If the entry bails with an error, then we need to reset ourself
// to what we were. Right now, the only place this should happen
// is if an Abort was returned from SetPositionCookie
// because somebody aborted during before navigate.
// But i think that any error means that we can legitimately Revert().
//
if (FAILED(hr))
{
Revert();
}
}
}
}
TraceMsg(TF_TRAVELLOG, "TL[%X]::TravelToUrl exited with hr = %X", this, hr);
return hr;
}
HRESULT CTravelLog::_FindEntryByOffset(IUnknown *punk, int iOffset, CTravelEntry **ppte)
{
CTravelEntry *pte = _pteCurrent;
BOOL fAllowLocalAnchor = TRUE;
if (iOffset < 0)
{
while (iOffset && pte)
{
pte = pte->GetPrev();
if (pte && pte->CanInvoke(punk, fAllowLocalAnchor))
{
iOffset++;
fAllowLocalAnchor = fAllowLocalAnchor && pte->IsLocalAnchor();
}
}
}
else if (iOffset > 0)
{
while (iOffset && pte)
{
pte = pte->GetNext();
if (pte && pte->CanInvoke(punk, fAllowLocalAnchor))
{
iOffset--;
fAllowLocalAnchor = fAllowLocalAnchor && pte->IsLocalAnchor();
}
}
}
if (pte)
{
*ppte = pte;
return S_OK;
}
return E_FAIL;
}
HRESULT CTravelLog::GetTravelEntry(IUnknown *punk, int iOffset, ITravelEntry **ppte)
{
HRESULT hr;
BOOL fCheckExternal = FALSE;
if (iOffset == TLOG_BACKEXTERNAL)
{
iOffset = TLOG_BACK;
fCheckExternal = TRUE;
}
if (iOffset == 0)
{
// APPCOMPAT - going back and fore between external apps is dangerous - zekel 24-JUN-97
// we always fail if the current is external
// this is because word will attempt to navigate us to
// the same url instead of FORE when the user selects
// it from the drop down.
if (_pteCurrent && _pteCurrent->IsExternal())
{
hr = E_FAIL;
ASSERT(!_pteCurrent->GetPrev());
TraceMsg(TF_TRAVELLOG, "TL[%X]::GetTravelEntry current is External", this);
goto Quit;
}
}
CTravelEntry *pte;
hr = _FindEntryByOffset(punk, iOffset, &pte);
//
// If TLOG_BACKEXTERNAL is specified, we return S_OK only if the previous
// entry is external.
//
if (fCheckExternal && SUCCEEDED(hr)) {
if (!pte->IsExternal()) {
hr = E_FAIL;
}
TraceMsg(TF_TRAVELLOG, "TL[%X]::GetTravelEntry(BACKEX)", this);
}
if (ppte && SUCCEEDED(hr)) {
hr = pte->QueryInterface(IID_PPV_ARG(ITravelEntry, ppte));
}
Quit:
TraceMsg(TF_TRAVELLOG, "TL[%X]::GetTravelEntry iOffset = %d, hr = %X", this, iOffset, hr);
return hr;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::FindTravelEntry
//
// Synopsis : Finds the travel entry with the specified PIDL and returns
// the ITravelEntry interface of the entry.
//
//----------------------------------------------------------------------------
HRESULT CTravelLog::FindTravelEntry(IUnknown *punk, LPCITEMIDLIST pidl, ITravelEntry **ppte)
{
CTravelEntry * pte = _pteRoot;
_FindEntryByPidl(punk, pidl, &pte);
if (pte)
{
return pte->QueryInterface(IID_PPV_ARG(ITravelEntry, ppte));
}
*ppte = NULL;
return E_FAIL;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::_FindEntryByPidl
//
// Synopsis : Finds and returns the travel entry with the specified PIDL.
// This private method returns a CTravelEntry instead of
// an ITravelEntry.
//
//----------------------------------------------------------------------------
HRESULT
CTravelLog::_FindEntryByPidl(IUnknown * punk, LPCITEMIDLIST pidl, CTravelEntry ** ppte)
{
CTravelEntry * pte = _pteRoot;
BOOL fAllowLocalAnchor = TRUE;
ASSERT(punk);
ASSERT(pidl);
ASSERT(ppte);
while (pte)
{
if (pte->CanInvoke(punk, fAllowLocalAnchor) && pte->IsEqual(pidl))
{
*ppte = pte;
return S_OK;
}
fAllowLocalAnchor = fAllowLocalAnchor && pte->IsLocalAnchor();
pte = pte->GetNext();
}
*ppte = NULL;
return E_FAIL;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::FindEntryByPunk
//
// Interface : ITravelLogEx
//
// Synopsis : Find the entry object given its punk.
//
//----------------------------------------------------------------------------
HRESULT
CTravelLog::_FindEntryByPunk(IUnknown * punk, ITravelLogEntry *pteSearch, CTravelEntry ** ppte)
{
CTravelEntry *pte = _pteRoot;
ITravelEntry *pteCur;
BOOL fAllowLocalAnchor = TRUE;
ASSERT(ppte);
// check for the current entry.
// often the current entry will fail CanInvoke because it's incomplete at this time.
if (IsSameObject(pteSearch, SAFECAST(_pteCurrent, ITravelEntry*)))
{
*ppte = _pteCurrent;
return S_OK;
}
while (pte)
{
pteCur = SAFECAST(pte, ITravelEntry*);
if ((pte->CanInvoke(punk, fAllowLocalAnchor)) && IsSameObject(pteCur, pteSearch))
{
*ppte = pte;
return S_OK;
}
fAllowLocalAnchor = fAllowLocalAnchor && pte->IsLocalAnchor();
pte = pte->GetNext();
}
*ppte = NULL;
return E_FAIL;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::FindTravelEntryWithUrl
//
// Interface : ITravelLogEx
//
// Synopsis : Finds and returns the travel entry with the specified URL.
//
//----------------------------------------------------------------------------
HRESULT
CTravelLog::FindTravelEntryWithUrl(IUnknown * punk, UINT uiCP, LPOLESTR lpszUrl, ITravelEntry ** ppte)
{
LPITEMIDLIST pidl;
HRESULT hr = E_FAIL;
ASSERT(punk);
ASSERT(lpszUrl);
ASSERT(ppte);
if (SUCCEEDED(IEParseDisplayNameWithBCW(uiCP, lpszUrl, NULL, &pidl)))
{
hr = FindTravelEntry(punk, pidl, ppte);
ILFree(pidl);
}
return hr;
}
HRESULT CTravelLog::Clone(ITravelLog **pptl)
{
CTravelLog *ptl = new CTravelLog();
HRESULT hr = S_OK;
if (ptl && _pteCurrent)
{
// first set the current pointer
hr = _pteCurrent->Clone(&ptl->_pteCurrent);
if (SUCCEEDED(hr))
{
ptl->_cbTotalSize = _cbTotalSize;
CTravelEntry *pteSrc;
CTravelEntry *pteClone, *pteDst = ptl->_pteCurrent;
// then we need to loop forward and set each
for (pteSrc = _pteCurrent->GetNext(), pteDst = ptl->_pteCurrent;
pteSrc; pteSrc = pteSrc->GetNext())
{
ASSERT(pteDst);
if (FAILED(pteSrc->Clone(&pteClone)))
break;
ASSERT(pteClone);
pteDst->SetNext(pteClone);
pteDst = pteClone;
}
//then loop back and set them all
for (pteSrc = _pteCurrent->GetPrev(), pteDst = ptl->_pteCurrent;
pteSrc; pteSrc = pteSrc->GetPrev())
{
ASSERT(pteDst);
if (FAILED(pteSrc->Clone(&pteClone)))
break;
ASSERT(pteClone);
pteDst->SetPrev(pteClone);
pteDst = pteClone;
}
// the root is the furthest back we could go
ptl->_pteRoot = pteDst;
}
}
else
hr = E_OUTOFMEMORY;
if (SUCCEEDED(hr))
{
ptl->QueryInterface(IID_PPV_ARG(ITravelLog, pptl));
}
else
{
*pptl = NULL;
}
if (ptl)
ptl->Release();
TraceMsg(TF_TRAVELLOG, "TL[%X]::Clone hr = %x, ptlClone = %X", this, hr, ptl);
return hr;
}
// HACKHACK: 3rd parameter used to be idsTemplate, which we would use to grab the
// string template. However, since there's no way the caller can specify the hinst
// of the module in which to look for this resource, this broke in the shdocvw /
// browseui split (callers would pass offsets into browseui.dll; we'd look for them in
// shdocvw.dll). My solution is is to ignore this parameter entirely and assume that:
//
// if iOffset is negative, the caller wants the "back to" text
// else, the caller wants the "forward to" text
//
// tjgreen 14-july-98.
//
HRESULT CTravelLog::GetToolTipText(IUnknown *punk, int iOffset, int, LPWSTR pwzText, DWORD cchText)
{
TCHAR szName[MAX_URL_STRING];
TCHAR szTemplate[80];
TraceMsg(TF_TRAVELLOG, "TL[%X]::ToolTip entering iOffset = %d, ptlClone = %X", this, iOffset);
ASSERT(pwzText);
ASSERT(cchText);
*pwzText = 0;
CTravelEntry *pte;
HRESULT hr = _FindEntryByOffset(punk, iOffset, &pte);
if (SUCCEEDED(hr))
{
ASSERT(pte);
pte->GetDisplayName(szName, SIZECHARS(szName), 0);
int idsTemplate = (iOffset < 0) ? IDS_NAVIGATEBACKTO : IDS_NAVIGATEFORWARDTO;
if (MLLoadString(idsTemplate, szTemplate, ARRAYSIZE(szTemplate))) {
DWORD cchTemplateLen = lstrlen(szTemplate);
DWORD cchLen = cchTemplateLen + lstrlen(szName);
if (cchLen > cchText) {
// so that we don't overflow the pwzText buffer
szName[cchText - cchTemplateLen - 1] = 0;
}
wnsprintf(pwzText, cchText, szTemplate, szName);
}
else
hr = E_UNEXPECTED;
}
TraceMsg(TF_TRAVELLOG, "TL[%X]::ToolTip exiting hr = %X, pwzText = %ls", this, hr, pwzText);
return hr;
}
HRESULT CTravelLog::InsertMenuEntries(IUnknown *punk, HMENU hmenu, int iIns, int idFirst, int idLast, DWORD dwFlags)
{
ASSERT(idLast >= idFirst);
ASSERT(hmenu);
ASSERT(punk);
int cItemsBack = idLast - idFirst + 1;
int cItemsFore = 0;
CTravelEntry *pte;
LONG cAdded = 0;
TCHAR szName[40];
DWORD cchName = SIZECHARS(szName);
UINT uFlags = MF_STRING | MF_ENABLED | MF_BYPOSITION;
TraceMsg(TF_TRAVELLOG, "TL[%X]::InsertMenuEntries entered on punk = %X, hmenu = %X, iIns = %d, idRange = %d-%d, flags = %X", this, punk, hmenu, iIns, idFirst, idLast, dwFlags);
ASSERT(cItemsFore >= 0);
ASSERT(cItemsBack >= 0);
if (IsFlagSet(dwFlags, TLMENUF_INCLUDECURRENT))
cItemsBack--;
if (IsFlagSet(dwFlags, TLMENUF_BACKANDFORTH))
{
cItemsFore = cItemsBack / 2;
cItemsBack = cItemsBack - cItemsFore;
}
else if (IsFlagSet(dwFlags, TLMENUF_FORE))
{
cItemsFore = cItemsBack;
cItemsBack = 0;
}
while (cItemsFore)
{
if (SUCCEEDED(_FindEntryByOffset(punk, cItemsFore, &pte)))
{
pte->GetDisplayName(szName, cchName, TEGDN_FORSYSTEM);
ASSERT(*szName);
FixAmpersands(szName, ARRAYSIZE(szName));
InsertMenu(hmenu, iIns, uFlags, idLast, szName);
cAdded++;
TraceMsg(TF_TRAVELLOG, "TL[%X]::IME Fore id = %d, szName = %s", this, idLast, szName);
}
cItemsFore--;
idLast--;
}
if (IsFlagSet(dwFlags, TLMENUF_INCLUDECURRENT))
{
// clear the name
*szName = 0;
//have to get the title from the actual pbs
IBrowserService *pbs ;
WCHAR wzTitle[MAX_PATH];
LPITEMIDLIST pidl = NULL;
if (SUCCEEDED(punk->QueryInterface(IID_PPV_ARG(IBrowserService, &pbs))))
{
pbs->GetPidl(&pidl);
if (SUCCEEDED(pbs->GetTitle(NULL, wzTitle, SIZECHARS(wzTitle))))
{
StrCpyN(szName, wzTitle, cchName);
}
else if (pidl)
{
GetUnescapedUrlIfAppropriate(pidl, szName, ARRAYSIZE(szName));
}
pbs->Release();
}
if (!SHIsDisplayable(szName, g_fRunOnFE, g_bRunOnNT5) && pidl)
{
// Display name isn't system-displayable. Just use the path/url instead.
SHTitleFromPidl(pidl, szName, ARRAYSIZE(szName), FALSE);
}
if (!(*szName))
TraceMsg(TF_ERROR, "CTravelLog::InsertMenuEntries -- failed to find title for current entry");
ILFree(pidl);
FixAmpersands(szName, ARRAYSIZE(szName));
InsertMenu(hmenu, iIns, uFlags | (IsFlagSet(dwFlags, TLMENUF_CHECKCURRENT) ? MF_CHECKED : 0), idLast, szName);
cAdded++;
TraceMsg(TF_TRAVELLOG, "TL[%X]::IME Current id = %d, szName = %s", this, idLast, szName);
idLast--;
}
if (IsFlagSet(dwFlags, TLMENUF_BACKANDFORTH))
{
// we need to reverse the order of insertion for back
// when both directions are displayed
int i;
for (i = 1; i <= cItemsBack; i++, idLast--)
{
if (SUCCEEDED(_FindEntryByOffset(punk, -i, &pte)))
{
pte->GetDisplayName(szName, cchName, TEGDN_FORSYSTEM);
ASSERT(*szName);
FixAmpersands(szName, ARRAYSIZE(szName));
InsertMenu(hmenu, iIns, uFlags, idLast, szName);
cAdded++;
TraceMsg(TF_TRAVELLOG, "TL[%X]::IME Back id = %d, szName = %s", this, idLast, szName);
}
}
}
else while (cItemsBack)
{
if (SUCCEEDED(_FindEntryByOffset(punk, -cItemsBack, &pte)))
{
pte->GetDisplayName(szName, cchName, TEGDN_FORSYSTEM);
ASSERT(*szName);
FixAmpersands(szName, ARRAYSIZE(szName));
InsertMenu(hmenu, iIns, uFlags, idLast, szName);
cAdded++;
TraceMsg(TF_TRAVELLOG, "TL[%X]::IME Back id = %d, szName = %s", this, idLast, szName);
}
cItemsBack--;
idLast--;
}
TraceMsg(TF_TRAVELLOG, "TL[%X]::InsertMenuEntries exiting added = %d", this, cAdded);
return cAdded ? S_OK : S_FALSE;
}
DWORD CTravelLog::CountEntries(IUnknown *punk)
{
CTravelEntry *pte = _pteRoot;
DWORD dw = 0;
BOOL fAllowLocalAnchor = TRUE;
while (pte)
{
if (pte->CanInvoke(punk, fAllowLocalAnchor))
dw++;
fAllowLocalAnchor = fAllowLocalAnchor && pte->IsLocalAnchor();
pte = pte->GetNext();
}
TraceMsg(TF_TRAVELLOG, "TL[%X]::CountEntries count = %d", this, dw);
return dw;
}
HRESULT CTravelLog::Revert(void)
{
// this function should only be called when
// we have travelled, and we stop the travel before finishing
if (_pteUpdate)
{
// trade them back
_pteCurrent = _pteUpdate;
_pteUpdate = NULL;
return S_OK;
}
return E_FAIL;
}
//
// delete nodes belonging to the frameset pte
//
void CTravelLog::_DeleteFrameSetEntry(IUnknown *punk, CTravelEntry *pte)
{
ASSERT(pte);
CTravelEntry *ptetmp = pte;
BOOL fAllowLocalAnchor = TRUE;
while(ptetmp && ptetmp != _pteCurrent)
ptetmp = ptetmp->GetNext();
if(ptetmp)
// entry on left of _pteCurrent , delete on left
do {
if(pte == _pteRoot)
_pteRoot = pte->GetNext();
ptetmp = pte;
pte = pte->GetPrev();
fAllowLocalAnchor = fAllowLocalAnchor && ptetmp->IsLocalAnchor();
_cbTotalSize -= ptetmp->Size();
ptetmp->RemoveSelf();
} while (pte && !(pte->CanInvoke(punk, fAllowLocalAnchor)));
else
if (pte)
{
do {
ptetmp = pte;
pte = pte->GetNext();
fAllowLocalAnchor = fAllowLocalAnchor && ptetmp->IsLocalAnchor();
_cbTotalSize -= ptetmp->Size();
ptetmp->RemoveSelf();
} while(pte && !(pte->CanInvoke(punk, fAllowLocalAnchor)));
}
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::DeleteIndexEntry
//
// Interface : ITravelLogEx
//
// Synopsis : Delete the entry given by index.
//
//----------------------------------------------------------------------------
HRESULT CTravelLog::DeleteIndexEntry(IUnknown *punk, int index)
{
HRESULT hres = E_FAIL;
CTravelEntry *pte;
IBrowserService *pbs;
BOOL fAllowLocalAnchor = TRUE;
ASSERT(punk);
if (index == 0) // don't delete current entry
return hres;
hres = _FindEntryByOffset(punk, index, &pte);
if(SUCCEEDED(hres))
{
_DeleteFrameSetEntry(punk, pte);
ASSERT(_cbTotalSize == _pteRoot->ListSize());
if (SUCCEEDED(punk->QueryInterface(IID_PPV_ARG(IBrowserService, &pbs))))
{
pbs->UpdateBackForwardState();
pbs->Release();
}
}
return hres;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::DeleteUrlEntry
//
// Interface : ITravelLogEx
//
// Synopsis : Delete all the entries given by URL. Fails for current entry.
//
//----------------------------------------------------------------------------
HRESULT CTravelLog::DeleteUrlEntry(IUnknown *punk, UINT uiCP, LPOLESTR lpszUrl)
{
HRESULT hres = E_FAIL;
CTravelEntry *pte;
IBrowserService *pbs;
LPITEMIDLIST pidl;
BOOL fAllowLocalAnchor = TRUE;
int count = 0;
ASSERT(punk);
if (SUCCEEDED(IEParseDisplayNameWithBCW(uiCP, lpszUrl, NULL, &pidl)))
{
// delete only if different from current
if(!_pteCurrent->IsEqual(pidl))
{
hres = S_OK;
while(SUCCEEDED(_FindEntryByPidl(punk, pidl, &pte)))
{
_DeleteFrameSetEntry(punk, pte);
count++;
ASSERT(_cbTotalSize == _pteRoot->ListSize());
}
}
ILFree(pidl);
if (count && SUCCEEDED(punk->QueryInterface(IID_PPV_ARG(IBrowserService, &pbs))))
{
pbs->UpdateBackForwardState();
pbs->Release();
}
}
return hres;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::DeleteEntry
//
// Interface : ITravelLogEx
//
// Synopsis : Delete the entries given by punk. Fails for current entry.
//
//----------------------------------------------------------------------------
HRESULT CTravelLog::DeleteEntry(IUnknown *punk, ITravelLogEntry *ptleDelete)
{
HRESULT hres;
CTravelEntry *pte;
BOOL fAllowLocalAnchor = TRUE;
IBrowserService *pbs;
ASSERT(punk);
hres = _FindEntryByPunk(punk, ptleDelete, &pte);
if(SUCCEEDED(hres) && pte != _pteCurrent) // don't remove current
{
_DeleteFrameSetEntry(punk, pte);
ASSERT(_cbTotalSize == _pteRoot->ListSize());
if (SUCCEEDED(punk->QueryInterface(IID_PPV_ARG(IBrowserService, &pbs))))
{
pbs->UpdateBackForwardState();
pbs->Release();
}
} else
hres = E_FAIL;
return hres;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::CountEntryNodes
//
// Synopsis : Counts Back/Forward entries including the current one
// as given by dwFlags
//
//----------------------------------------------------------------------------
HRESULT CTravelLog::CountEntryNodes(IUnknown *punk, DWORD dwFlags, DWORD *pdwCount)
{
CTravelEntry *pte;
BOOL fAllowLocalAnchor = TRUE;
ASSERT(punk);
DWORD dwCount = 0;
if(!_pteCurrent)
{
*pdwCount = 0;
return S_OK;
}
if(IsFlagSet(dwFlags, TLMENUF_BACK))
{
pte = _pteRoot;
while (pte != _pteCurrent)
{
if(pte->CanInvoke(punk, fAllowLocalAnchor))
{
dwCount++;
fAllowLocalAnchor = fAllowLocalAnchor && pte->IsLocalAnchor();
}
pte = pte->GetNext();
}
}
if(IsFlagSet(dwFlags, TLMENUF_INCLUDECURRENT))
{
if(_pteCurrent->CanInvoke(punk, fAllowLocalAnchor))
{
dwCount++;
fAllowLocalAnchor = fAllowLocalAnchor && _pteCurrent->IsLocalAnchor();
}
}
if(IsFlagSet(dwFlags, TLMENUF_FORE))
{
pte = _pteCurrent->GetNext();
while (pte)
{
if(pte->CanInvoke(punk, fAllowLocalAnchor))
{
dwCount++;
fAllowLocalAnchor = fAllowLocalAnchor && pte->IsLocalAnchor();
}
pte = pte->GetNext();
}
}
*pdwCount = dwCount;
TraceMsg(TF_TRAVELLOG, "TL[%X]::CountEntryNodes count = %d", this, *pdwCount);
return S_OK;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::CreateEnumEntry
//
// Synopsis : Returns an enumerator object for the back/fore travel entries
// as selected by the dwFlags option
//
//----------------------------------------------------------------------------
HRESULT
CTravelLog::CreateEnumEntry(IUnknown *punk, IEnumTravelLogEntry **ppEnum, DWORD dwFlags)
{
HRESULT hres = E_OUTOFMEMORY;
CEnumEntry *penum;
ASSERT(punk);
ASSERT(ppEnum);
*ppEnum = 0;
penum = new CEnumEntry();
if(penum)
{
penum->Init(this, punk, 0, dwFlags);
*ppEnum = SAFECAST(penum, IEnumTravelLogEntry *);
hres = S_OK;
}
return hres;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLogEx::InsertEntry
//
// Synopsis : Inserts an entry into the specified position in the
// travellog and calls Update with the given IUnknown.
//
//----------------------------------------------------------------------------
HRESULT
CTravelLog::InsertEntry(IUnknown *punkBrowser, ITravelLogEntry *pteRelativeTo, BOOL fPrepend,
IUnknown* punkTLClient, ITravelLogEntry **ppEntry)
{
HRESULT hres = E_FAIL;
CTravelEntry * pteRelative;
CTravelEntry * pte;
IBrowserService *pbs;
TraceMsg(TF_TRAVELLOG, "TL[%X]::InsertEntry", this);
ASSERT(punkBrowser);
_FindEntryByPunk(punkBrowser, pteRelativeTo, &pteRelative);
if (!pteRelative)
pteRelative = _pteCurrent;
pte = new CTravelEntry(FALSE);
if (!pte)
return E_OUTOFMEMORY;
if (fPrepend)
{
// keep relative alive as it's reconnected
pteRelative->AddRef();
pte->SetPrev(pteRelative->GetPrev());
pteRelative->SetPrev(pte);
if (pteRelative == _pteRoot)
{
_pteRoot = pte;
}
}
else
{
CTravelEntry * pteNext = pteRelative->GetNext();
if (pteNext)
pteNext->AddRef();
pte->SetNext(pteNext);
pteRelative->SetNext(pte);
}
// update will fill in all the data from the passed in TL Client
hres = pte->Update(punkTLClient, FALSE);
_cbTotalSize += pte->Size();
ASSERT(_cbTotalSize == _pteRoot->ListSize());
if (SUCCEEDED(punkBrowser->QueryInterface(IID_PPV_ARG(IBrowserService, &pbs))))
{
pbs->UpdateBackForwardState();
pbs->Release();
hres = S_OK;
}
// return the ITLEntry for the new entry
if (SUCCEEDED(hres) && ppEntry)
{
hres = pte->QueryInterface(IID_PPV_ARG(ITravelLogEntry, ppEntry));
}
return hres;
}
//+---------------------------------------------------------------------------
//
// Method : CTravelLog::TravelToEntry
//
// Synopsis : Travels directly to the specified entry.
// Invoke cannot be called directly due to update strangeness.
//
//----------------------------------------------------------------------------
HRESULT CTravelLog::TravelToEntry(
IUnknown *punkBrowser,
ITravelLogEntry *pteDestination)
{
HRESULT hr = E_FAIL;
CTravelEntry *pte = NULL;
ASSERT(punkBrowser);
ASSERT(pteDestination);
_FindEntryByPunk(punkBrowser, pteDestination , &pte);
if (pte)
{
if (!_pteCurrent->IsExternal() && !_pteUpdate)
_pteUpdate = _pteCurrent;
_pteCurrent = pte;
hr = pte->Invoke(punkBrowser);
if (FAILED(hr))
{
Revert();
}
}
return hr;
}
#ifdef TRAVELDOCS
GetNewDocument()
{
new CTraveledDocument();
ptd->Init();
DPA_Add(ptd);
DPA_Sort();
}
// really need to use ILIsEqual() instead of this
int ILCompareFastButWrong(LPITEMIDLIST pidl1, LPITEMIDLIST pidl2)
{
int iret;
DWORD cb1 = ILGetSize(ptd1->_pidl);
DWORD cb2 = ILGetSize(ptd2->_pidl);
iret = cb1 - cb2;
if (0 == iret)
iret = memcmp(pidl1, cb1, pidl2, cb2);
return iret;
}
static int CTraveledDocument::Compare(PTRAVELEDDOCUMENT ptd1, PTRAVELEDDOCUMENT ptd2)
{
int iret;
iret = ptd1->_type - ptd2->_type;
if (0 = iret)
{
iret = ptd1->_hash - ptd2->_hash;
if (0 == iret)
{
switch (ptd1->_type)
{
case TDOC_PIDL:
iret = ILCompareFastButWrong(ptd1->_pidl, ptd2->_pidl);
break;
case TDOC_URL:
iret = UrlCompare((LPTSTR)ptd1->_strUrl, (LPTSTR)ptd2->_strUrl, FALSE);
break;
default:
ASSERT(FALSE);
}
}
}
return iret;
}
#endif //TRAVELDOCS
HRESULT CreateTravelLog(ITravelLog **pptl)
{
HRESULT hres;
CTravelLog *ptl = new CTravelLog();
if (ptl)
{
hres = ptl->QueryInterface(IID_PPV_ARG(ITravelLog, pptl));
ptl->Release();
}
else
{
*pptl = NULL;
hres = E_OUTOFMEMORY;
}
return hres;
}
CEnumEntry::CEnumEntry() :_cRef(1)
{
ASSERT(!_ptl);
ASSERT(!_punk);
TraceMsg(TF_TRAVELLOG, "EET[%X] created ", this);
}
CEnumEntry::~CEnumEntry()
{
SAFERELEASE(_ptl);
SAFERELEASE(_punk);
TraceMsg(TF_TRAVELLOG, "EET[%X] destroyed ", this);
}
HRESULT CEnumEntry::QueryInterface(REFIID riid, void **ppvObj)
{
static const QITAB qit[] = {
QITABENT(CEnumEntry, IEnumTravelLogEntry),
{ 0 },
};
return QISearch(this, qit, riid, ppvObj);
}
ULONG CEnumEntry::AddRef()
{
return InterlockedIncrement(&_cRef);
}
ULONG CEnumEntry::Release()
{
if (InterlockedDecrement(&_cRef))
return _cRef;
delete this;
return 0;
}
void CEnumEntry::Init(CTravelLog *ptl, IUnknown *punk, DWORD dwOffset, DWORD dwFlags)
{
ASSERT(ptl);
ASSERT(punk);
_ptl = ptl;
_dwFlags = dwFlags;
_punk = punk;
_dwOffset = dwOffset;
_lStart = 0;
_ptl->AddRef();
_punk->AddRef();
SetBase();
}
void CEnumEntry::SetBase()
{
ITravelEntry *ptetmp;
// the start is always computed relative to the current entry
if(IsFlagSet(_dwFlags, TLEF_RELATIVE_FORE|TLEF_RELATIVE_BACK))
{
_lStart = -1;
while(SUCCEEDED(_ptl->GetTravelEntry(_punk, _lStart, &ptetmp)))
{
_lStart--;
ptetmp->Release();
}
_lStart++;
}
else if(!IsFlagSet(_dwFlags, TLEF_RELATIVE_INCLUDE_CURRENT))
_lStart = IsFlagSet(_dwFlags, TLEF_RELATIVE_BACK) ? -1: 1;
}
HRESULT CEnumEntry::Reset()
{
_dwOffset = 0;
// base changes when add/delete entries
SetBase();
return S_OK;
}
HRESULT CEnumEntry::Skip(ULONG cElt)
{
HRESULT hres;
ITravelEntry *pte;
ULONG uCount;
LONG lIndex;
BOOL fToRight;
BOOL fIncludeCurrent;
fToRight = IsFlagSet(_dwFlags, TLEF_RELATIVE_FORE);
fIncludeCurrent = IsFlagSet(_dwFlags, TLEF_RELATIVE_INCLUDE_CURRENT);
for (uCount = 0; uCount < cElt; uCount++)
{
lIndex = fToRight ? _lStart + _dwOffset : _lStart - _dwOffset;
if(lIndex || fIncludeCurrent)
{
if(SUCCEEDED(hres = _ptl->GetTravelEntry(_punk, lIndex, &pte)))
{
_dwOffset++;
pte->Release();
}
else
break;
}
else
{
_dwOffset++;
uCount--;
}
}
if ( uCount != cElt )
hres = S_FALSE;
return hres;
}
HRESULT CEnumEntry::Next(ULONG cElt, ITravelLogEntry **rgpte2, ULONG *pcEltFetched)
{
HRESULT hres = S_OK;
ULONG uCount = 0;
ITravelEntry *pte;
LONG lIndex;
BOOL fToRight;
BOOL fIncludeCurrent;
fToRight = IsFlagSet(_dwFlags, TLEF_RELATIVE_FORE);
fIncludeCurrent = IsFlagSet(_dwFlags, TLEF_RELATIVE_INCLUDE_CURRENT);
if(pcEltFetched)
*pcEltFetched = 0;
for (uCount = 0; uCount < cElt; uCount++)
{
lIndex = fToRight ? _lStart + _dwOffset : _lStart - _dwOffset;
if(lIndex || fIncludeCurrent)
{
hres = _ptl->GetTravelEntry(_punk, lIndex, &pte);
if(SUCCEEDED(hres))
{
_dwOffset++;
pte->QueryInterface(IID_PPV_ARG(ITravelLogEntry, &rgpte2[uCount]));
pte->Release();
}
else
break;
}
else
{
_dwOffset++;
uCount--;
}
}
if (pcEltFetched )
*pcEltFetched = uCount;
if ( uCount != cElt )
{
hres = S_FALSE;
for(;uCount < cElt; uCount++)
rgpte2[uCount] = 0;
}
return hres;
}
STDMETHODIMP CEnumEntry::Clone(IEnumTravelLogEntry **ppEnum)
{
HRESULT hres = E_OUTOFMEMORY;
CEnumEntry *penum;
ASSERT(ppEnum);
*ppEnum = 0;
penum = new CEnumEntry();
if(penum)
{
penum->Init(_ptl, _punk, _dwOffset, _dwFlags);
*ppEnum = SAFECAST(penum, IEnumTravelLogEntry*);
hres = S_OK;
}
return hres;
}
// Helper object for creating new travel entries
class CPublicTravelLogCreateHelper : public ITravelLogClient, IPersistHistory
{
public:
// *** IUnknown
STDMETHODIMP QueryInterface(REFIID riid, void **ppvObj);
STDMETHODIMP_(ULONG) AddRef() ;
STDMETHODIMP_(ULONG) Release();
// ITravelLogClient methods
STDMETHODIMP FindWindowByIndex(DWORD dwID, IUnknown **ppunk);
STDMETHODIMP GetWindowData(WINDOWDATA *pwindata);
STDMETHODIMP LoadHistoryPosition(LPOLESTR pszUrlLocation, DWORD dwCookie)
{ return SetPositionCookie(dwCookie); }
// IPersist methods. (dummy)
STDMETHODIMP GetClassID(CLSID *pClassID)
{ ASSERT(FALSE); return E_NOTIMPL; }
// IPersistHistory methods. (dummy)
// These should never be called, but Update QI's the client to see if it supports IPH.
STDMETHODIMP LoadHistory(IStream *pStream, IBindCtx *pbc)
{ return E_NOTIMPL; }
STDMETHODIMP SaveHistory(IStream *pStream)
{ ASSERT(FALSE); return S_OK; }
STDMETHODIMP SetPositionCookie(DWORD dwPositioncookie)
{ return E_NOTIMPL; }
STDMETHODIMP GetPositionCookie(DWORD *pdwPositioncookie)
{ return E_NOTIMPL; }
CPublicTravelLogCreateHelper(IBrowserService *pbs, LPCOLESTR pszUrl, LPCOLESTR pszTitle);
protected:
~CPublicTravelLogCreateHelper();
LONG _cRef;
IBrowserService *_pbs;
LPOLESTR _pszUrl;
LPOLESTR _pszTitle;
};
CPublicTravelLogCreateHelper::CPublicTravelLogCreateHelper(IBrowserService *pbs, LPCOLESTR pszUrl, LPCOLESTR pszTitle) : _pbs(pbs), _cRef(1)
{
ASSERT(_pbs);
ASSERT(!_pszUrl);
ASSERT(!_pszTitle);
ASSERT(pszUrl);
ASSERT(pszTitle);
if(_pbs)
_pbs->AddRef();
if(pszUrl)
{
SHStrDup(pszUrl, &_pszUrl);
}
if(pszTitle)
{
SHStrDup(pszTitle, &_pszTitle);
}
TraceMsg(TF_TRAVELLOG, "TPLCH[%X] created", this);
}
CPublicTravelLogCreateHelper::~CPublicTravelLogCreateHelper()
{
SAFERELEASE(_pbs);
CoTaskMemFree(_pszUrl);
CoTaskMemFree(_pszTitle);
TraceMsg(TF_TRAVELLOG, "TPLCH[%X] destroyed ", this);
}
HRESULT CPublicTravelLogCreateHelper ::QueryInterface(REFIID riid, void **ppvObj)
{
static const QITAB qit[] = {
QITABENT(CPublicTravelLogCreateHelper , ITravelLogClient),
QITABENT(CPublicTravelLogCreateHelper , IPersistHistory),
QITABENT(CPublicTravelLogCreateHelper , IPersist),
{ 0 },
};
return QISearch(this, qit, riid, ppvObj);
}
ULONG CPublicTravelLogCreateHelper ::AddRef()
{
return InterlockedIncrement(&_cRef);
}
ULONG CPublicTravelLogCreateHelper ::Release()
{
if (InterlockedDecrement(&_cRef))
return _cRef;
delete this;
return 0;
}
// ITravelLogClient::FindWindowByIndex
HRESULT
CPublicTravelLogCreateHelper::FindWindowByIndex(DWORD dwID, IUnknown **ppunk)
{
*ppunk = NULL;
return E_NOTIMPL;
}
// ITravelLogClient::GetWindowData
// turns around and talks to the browser
HRESULT
CPublicTravelLogCreateHelper::GetWindowData(WINDOWDATA *pwindata)
{
HRESULT hres;
ITravelLogClient2 *pcli;
hres = _pbs->QueryInterface(IID_PPV_ARG(ITravelLogClient2, &pcli));
if (SUCCEEDED(hres))
hres = pcli->GetDummyWindowData(_pszUrl, _pszTitle, pwindata);
if (pcli)
pcli->Release();
return SUCCEEDED(hres) ? S_OK : E_FAIL;
}
// Implements the publicly exposed interface ITravelLogStg (can be QS'd for from top browser)
class CPublicTravelLog : public ITravelLogStg
{
public:
// *** IUnknown
STDMETHODIMP QueryInterface(REFIID riid, void **ppvObj);
STDMETHODIMP_(ULONG) AddRef() ;
STDMETHODIMP_(ULONG) Release();
// *** ITravelLogStg specific methods
STDMETHODIMP CreateEntry(LPCOLESTR pszUrl, LPCOLESTR pszTitle, ITravelLogEntry *ptleRelativeTo,
BOOL fPrepend, ITravelLogEntry **pptle);
STDMETHODIMP TravelTo(ITravelLogEntry *ptle);
STDMETHODIMP EnumEntries(TLENUMF flags, IEnumTravelLogEntry **ppenum);
STDMETHODIMP FindEntries(TLENUMF flags, LPCOLESTR pszUrl, IEnumTravelLogEntry **ppenum);
STDMETHODIMP GetCount(TLENUMF flags, DWORD *pcEntries);
STDMETHODIMP RemoveEntry(ITravelLogEntry *ptle);
STDMETHODIMP GetRelativeEntry(int iOffset, ITravelLogEntry **ptle);
CPublicTravelLog(IBrowserService *pbs, ITravelLogEx *ptlx);
protected:
~CPublicTravelLog();
LONG _cRef;
IBrowserService *_pbs;
ITravelLogEx *_ptlx;
};
CPublicTravelLog::CPublicTravelLog(IBrowserService *pbs, ITravelLogEx *ptlx) : _pbs(pbs), _ptlx(ptlx), _cRef(1)
{
ASSERT(pbs);
ASSERT(ptlx);
// We don't addref _pbs because we are always contained within the browser serivce,
// so avoid the circular ref.
if(_ptlx)
_ptlx->AddRef();
TraceMsg(TF_TRAVELLOG, "TLP[%X] created", this);
}
CPublicTravelLog::~CPublicTravelLog()
{
// We don't need to release _pbs because we are always contained within the browser serivce,
// so we didn't addref to avoid the circular ref so don't release
SAFERELEASE(_ptlx);
TraceMsg(TF_TRAVELLOG, "TLP[%X] destroyed ", this);
}
HRESULT CPublicTravelLog::QueryInterface(REFIID riid, void **ppvObj)
{
static const QITAB qit[] = {
QITABENT(CPublicTravelLog, ITravelLogStg), // IID_ITravelLogStg
{ 0 },
};
return QISearch(this, qit, riid, ppvObj);
}
ULONG CPublicTravelLog::AddRef()
{
return InterlockedIncrement(&_cRef);
}
ULONG CPublicTravelLog::Release()
{
if (InterlockedDecrement(&_cRef))
return _cRef;
delete this;
return 0;
}
//+---------------------------------------------------------------------------
//
// Method : CPublicTravelLog::CreateEntry
//
// Interface : ITravelLogStg
//
// Synopsis : Insert a new dummy entry.
// Creates an entry in the travel log and passes CPTHCEHelper
// as travel log client; that gets called back and fills in the
// data from the browser.
//
//----------------------------------------------------------------------------
HRESULT CPublicTravelLog::CreateEntry(LPCOLESTR pszUrl, LPCOLESTR pszTitle, ITravelLogEntry *ptleRelativeTo,
BOOL fPrepend, ITravelLogEntry **pptle)
{
HRESULT hres = E_FAIL;
CPublicTravelLogCreateHelper * ptlch;
ITravelLogClient *ptlc;
ptlch = new CPublicTravelLogCreateHelper(_pbs, pszUrl, pszTitle);
if (!ptlch)
return E_OUTOFMEMORY;
ptlc = SAFECAST(ptlch, ITravelLogClient *);
if (ptlc)
{
// Create TLogEntry and have it get its data from the helper.
hres = _ptlx->InsertEntry(_pbs, ptleRelativeTo, fPrepend, ptlc, pptle);
}
ptlc->Release();
return hres;
}
HRESULT CPublicTravelLog::TravelTo(ITravelLogEntry *ptle)
{
if (ptle)
return _ptlx->TravelToEntry(_pbs, ptle);
else
return E_POINTER;
}
//+---------------------------------------------------------------------------
//
// Method : CPublicTravelLog::EnumEntries
//
// Interface : ITravelLogStg
//
// Synopsis : Get an enumerators for specific entries given by flags.
// Flags should match with those used by ITravelLogEx!
//
//----------------------------------------------------------------------------
HRESULT CPublicTravelLog::EnumEntries(TLENUMF flags, IEnumTravelLogEntry **ppenum)
{
return _ptlx->CreateEnumEntry(_pbs, ppenum, flags);
}
//+---------------------------------------------------------------------------
//
// Method : CPublicTravelLog::FindEntries
//
// Interface : ITravelLogStg
//
// Synopsis : Allow to retrieve duplicate entries.
// Flags should match with those used by ITravelLogEx!
//
//----------------------------------------------------------------------------
HRESULT CPublicTravelLog::FindEntries(TLENUMF flags, LPCOLESTR pszUrl, IEnumTravelLogEntry **ppenum)
{
return E_NOTIMPL;
}
//+---------------------------------------------------------------------------
//
// Method : CPublicTravelLog::GetCount
//
// Interface : ITravelLogStg
//
// Synopsis : Public methods to get ITravelLogEx count.
// Flags should match with those used by ITravelLogEx!
//
//----------------------------------------------------------------------------
HRESULT CPublicTravelLog::GetCount(TLENUMF flags, DWORD *pcEntries)
{
return _ptlx->CountEntryNodes(_pbs, flags, pcEntries);
}
//+---------------------------------------------------------------------------
//
// Method : CPublicTravelLog::RemoveEntry
//
// Interface : ITravelLogStg
//
// Synopsis : Delete the entry ant its frameset.
//
//----------------------------------------------------------------------------
HRESULT CPublicTravelLog::RemoveEntry(ITravelLogEntry *ptle)
{
HRESULT hr = E_FAIL;
if(ptle)
hr = _ptlx->DeleteEntry(_pbs, ptle);
return hr;
}
HRESULT CPublicTravelLog::GetRelativeEntry(int iOffset, ITravelLogEntry **pptle)
{
HRESULT hr = E_FAIL;
ITravelEntry* pte;
ITravelLog* ptl;
if(SUCCEEDED(_ptlx->QueryInterface(IID_PPV_ARG(ITravelLog, &ptl))))
{
hr = ptl->GetTravelEntry(_pbs, iOffset, &pte);
if(SUCCEEDED(hr) && pte)
{
hr = pte->QueryInterface(IID_PPV_ARG(ITravelLogEntry, pptle));
pte->Release();
}
ptl->Release();
}
return hr;
}
// public method used by the browser to create us
HRESULT CreatePublicTravelLog(IBrowserService *pbs, ITravelLogEx* ptlx, ITravelLogStg **pptlstg)
{
HRESULT hres;
CPublicTravelLog *ptlp = new CPublicTravelLog(pbs, ptlx);
if (ptlp)
{
hres = ptlp->QueryInterface(IID_PPV_ARG(ITravelLogStg, pptlstg));
ptlp->Release();
}
else
{
*pptlstg = NULL;
hres = E_OUTOFMEMORY;
}
return hres;
}