CACLCustomMRU Class Reference

#include <ACLCustomMRU.h>

Inheritance diagram for CACLCustomMRU:

Collaboration diagram for CACLCustomMRU:

Public Member Functions
	CACLCustomMRU ()
	~CACLCustomMRU ()
STDMETHODIMP	Next (ULONG celt, LPWSTR *rgelt, ULONG *pceltFetched) override
STDMETHODIMP	Skip (ULONG celt) override
STDMETHODIMP	Reset () override
STDMETHODIMP	Clone (IEnumString **ppenum) override
STDMETHODIMP	Expand (LPCOLESTR pszExpand) override
STDMETHODIMP	Initialize (LPCWSTR pwszMRURegKey, DWORD dwMax) override
STDMETHODIMP	AddMRUString (LPCWSTR pwszEntry) override
Public Member Functions inherited from ATL::CComObjectRootEx< CComMultiThreadModelNoCS >
	~CComObjectRootEx ()
ULONG	InternalAddRef ()
ULONG	InternalRelease ()
void	Lock ()
void	Unlock ()
HRESULT	_AtlInitialConstruct ()
Public Member Functions inherited from ATL::CComObjectRootBase
	CComObjectRootBase ()
	~CComObjectRootBase ()
void	SetVoid (void *)
HRESULT	_AtlFinalConstruct ()
HRESULT	FinalConstruct ()
void	InternalFinalConstructAddRef ()
void	InternalFinalConstructRelease ()
void	FinalRelease ()
Public Member Functions inherited from IEnumString
HRESULT	Next ([in] ULONG celt, [out, size_is(celt), length_is(*pceltFetched)] LPOLESTR *rgelt, [out] ULONG *pceltFetched)
HRESULT	RemoteNext ([in] ULONG celt, [out, size_is(celt), length_is(*pceltFetched)] LPOLESTR *rgelt, [out] ULONG *pceltFetched)
Public Member Functions inherited from IUnknown
HRESULT	QueryInterface ([in] REFIID riid, [out, iid_is(riid)] void **ppvObject)
ULONG	AddRef ()
ULONG	Release ()

Private Member Functions
void	PersistMRU ()
HRESULT	LoadTypedURLs (DWORD dwMax)
HRESULT	LoadMRUList (DWORD dwMax)

Private Attributes
CRegKey	m_Key
CStringW	m_MRUList
CSimpleArray< CStringW >	m_MRUData
bool	m_bDirty
BOOL	m_bTypedURLs
ULONG	m_ielt

Additional Inherited Members
Public Types inherited from IEnumString
typedef IEnumString *	LPENUMSTRING
Public Types inherited from IUnknown
typedef IUnknown *	LPUNKNOWN
Static Public Member Functions inherited from ATL::CComObjectRootBase
static void WINAPI	ObjectMain (bool)
static const struct _ATL_CATMAP_ENTRY *	GetCategoryMap ()
static HRESULT WINAPI	InternalQueryInterface (void *pThis, const _ATL_INTMAP_ENTRY *pEntries, REFIID iid, void **ppvObject)
Public Attributes inherited from ATL::CComObjectRootBase
LONG	m_dwRef

Detailed Description

Definition at line 11 of file ACLCustomMRU.h.

Constructor & Destructor Documentation

CACLCustomMRU()

CACLCustomMRU::CACLCustomMRU

(

)

Definition at line 13 of file ACLCustomMRU.cpp.

    : m_bDirty(false), m_bTypedURLs(FALSE), m_ielt(0)
{
}

~CACLCustomMRU()

CACLCustomMRU::~CACLCustomMRU

(

)

Definition at line 18 of file ACLCustomMRU.cpp.

{
    PersistMRU();
}

Member Function Documentation

AddMRUString()

HRESULT STDMETHODCALLTYPE CACLCustomMRU::AddMRUString ( LPCWSTR  pwszEntry )

override

Definition at line 215 of file ACLCustomMRU.cpp.

{
    if (m_bTypedURLs)
        return E_FAIL;

    ATLASSERT(m_MRUData.GetSize() <= m_MRUList.GetLength());
    m_bDirty = true;

    CStringW NewElement = pwszEntry;
    WCHAR Key[2] = { 0, 0 };
    int Index = m_MRUData.Find(NewElement);
    if (Index >= 0)
    {
        /* Move the key to the front */
        Key[0] = Index + 'a';
        m_MRUList.Replace(Key, L"");
        m_MRUList = Key + m_MRUList;
        return S_OK;
    }

    int TotalLen = m_MRUList.GetLength();
    if (m_MRUData.GetSize() == TotalLen)
    {
        /* Find oldest element, move that to the front */
        Key[0] = m_MRUList[TotalLen-1];
        m_MRUList = Key + m_MRUList.Left(TotalLen-1);
        Index = Key[0] - 'a';
    }
    else
    {
        /* Find the first empty entry */
        for (Index = 0; Index < m_MRUData.GetSize(); ++Index)
        {
            if (m_MRUData[Index].IsEmpty())
                break;
        }
        Key[0] = Index + 'a';
        m_MRUList = Key + m_MRUList;
    }
    m_MRUData[Index] = NewElement;

    PersistMRU();
    return S_OK;
}

Clone()

STDMETHODIMP CACLCustomMRU::Clone ( IEnumString **  ppenum )

override

Implements IEnumString.

Definition at line 69 of file ACLCustomMRU.cpp.

{
    *ppenum = NULL;
    return E_NOTIMPL;
}

Expand()

STDMETHODIMP CACLCustomMRU::Expand ( LPCOLESTR  pszExpand )

override

Definition at line 75 of file ACLCustomMRU.cpp.

{
    return E_NOTIMPL;
}

Initialize()

HRESULT STDMETHODCALLTYPE CACLCustomMRU::Initialize ( LPCWSTR  pwszMRURegKey, DWORD  dwMax  )

override

Definition at line 149 of file ACLCustomMRU.cpp.

{
    m_ielt = 0;

    LSTATUS Status = m_Key.Create(HKEY_CURRENT_USER, pwszMRURegKey);
    if (Status != ERROR_SUCCESS)
        return HRESULT_FROM_WIN32(Status);

    m_MRUData.RemoveAll();
    if (lstrcmpiW(pwszMRURegKey, TYPED_URLS_KEY) == 0)
    {
        m_bTypedURLs = TRUE;
        return LoadTypedURLs(dwMax);
    }
    else
    {
        m_bTypedURLs = FALSE;
        return LoadMRUList(dwMax);
    }
}

LoadMRUList()

HRESULT CACLCustomMRU::LoadMRUList ( DWORD  dwMax )

private

Definition at line 170 of file ACLCustomMRU.cpp.

{
    dwMax = max(0, dwMax);
    dwMax = min(29, dwMax);
    while (dwMax--)
        m_MRUData.Add(CStringW());

    WCHAR MRUList[40];
    ULONG nChars = _countof(MRUList);

    LSTATUS Status = m_Key.QueryStringValue(L"MRUList", MRUList, &nChars);
    if (Status != ERROR_SUCCESS)
        return S_OK;

    if (nChars > 0 && MRUList[nChars-1] == '\0')
        nChars--;

    if (nChars > (ULONG)m_MRUData.GetSize())
        return S_OK;

    for (ULONG n = 0; n < nChars; ++n)
    {
        if (MRUList[n] >= 'a' && MRUList[n] <= '}' && m_MRUList.Find(MRUList[n]) < 0)
        {
            WCHAR Key[2] = { MRUList[n], NULL };
            WCHAR Value[MAX_PATH * 2];
            ULONG nValueChars = _countof(Value);

            m_MRUList += MRUList[n];
            int Index = MRUList[n] - 'a';

            if (Index < m_MRUData.GetSize())
            {
                Status = m_Key.QueryStringValue(Key, Value, &nValueChars);
                if (Status == ERROR_SUCCESS)
                {
                    m_MRUData[Index] = CStringW(Value, nValueChars);
                }
            }
        }
    }

    return S_OK;
}

Referenced by Initialize().

LoadTypedURLs()

HRESULT CACLCustomMRU::LoadTypedURLs ( DWORD  dwMax )

private

Definition at line 124 of file ACLCustomMRU.cpp.

{
    dwMax = max(0, dwMax);
    dwMax = min(29, dwMax);

    WCHAR szName[32];
    CStringW strData;
    LSTATUS status;
    for (DWORD i = 1; i <= dwMax; ++i)
    {
        // Build a registry value name
        StringCbPrintfW(szName, sizeof(szName), L"url%lu", i);

        // Read a registry value
        status = RegQueryCStringW(m_Key, szName, strData);
        if (status != ERROR_SUCCESS)
            break;

        m_MRUData.Add(strData);
    }

    return S_OK;
}

Referenced by Initialize().

Next()

STDMETHODIMP CACLCustomMRU::Next ( ULONG  celt, LPWSTR *  rgelt, ULONG *  pceltFetched  )

override

Definition at line 23 of file ACLCustomMRU.cpp.

{
    if (!pceltFetched || !rgelt)
        return E_POINTER;

    *pceltFetched = 0;
    if (celt == 0)
        return S_OK;

    *rgelt = NULL;
    if (INT(m_ielt) >= m_MRUData.GetSize())
        return S_FALSE;

    CStringW str = m_MRUData[m_ielt];

    if (!m_bTypedURLs)
    {
        // Erase the last "\\1" etc. (indicates SW_* value)
        INT ich = str.ReverseFind(L'\\');
        if (ich >= 0)
            str = str.Left(ich);
    }

    size_t cb = (str.GetLength() + 1) * sizeof(WCHAR);
    LPWSTR psz = (LPWSTR)CoTaskMemAlloc(cb);
    if (!psz)
        return S_FALSE;

    CopyMemory(psz, (LPCWSTR)str, cb);
    *rgelt = psz;
    *pceltFetched = 1;
    ++m_ielt;
    return S_OK;
}

PersistMRU()

void CACLCustomMRU::PersistMRU ( )

private

Definition at line 80 of file ACLCustomMRU.cpp.

{
    if (!m_bDirty || m_bTypedURLs)
        return;

    WCHAR Key[2] = { 0, 0 };

    m_bDirty = false;

    if (m_Key.m_hKey)
    {
        m_Key.SetStringValue(L"MRUList", m_MRUList);
        for (int Index = 0; Index < m_MRUList.GetLength(); ++Index)
        {
            Key[0] = Index + 'a';
            m_Key.SetStringValue(Key, m_MRUData[Index]);
        }
    }
}

Referenced by AddMRUString(), and ~CACLCustomMRU().

Reset()

STDMETHODIMP CACLCustomMRU::Reset ( void  )

override

Implements IEnumString.

Definition at line 63 of file ACLCustomMRU.cpp.

{
    m_ielt = 0;
    return S_OK;
}

Skip()

STDMETHODIMP CACLCustomMRU::Skip ( ULONG  celt )

override

Implements IEnumString.

Definition at line 58 of file ACLCustomMRU.cpp.

{
    return E_NOTIMPL;
}

Member Data Documentation

m_bDirty

bool CACLCustomMRU::m_bDirty

private

Definition at line 22 of file ACLCustomMRU.h.

Referenced by AddMRUString(), and PersistMRU().

m_bTypedURLs

BOOL CACLCustomMRU::m_bTypedURLs

private

Definition at line 23 of file ACLCustomMRU.h.

Referenced by AddMRUString(), Initialize(), Next(), and PersistMRU().

m_ielt

ULONG CACLCustomMRU::m_ielt

private

Definition at line 24 of file ACLCustomMRU.h.

Referenced by Initialize(), Next(), and Reset().

m_Key

CRegKey CACLCustomMRU::m_Key

private

Definition at line 19 of file ACLCustomMRU.h.

Referenced by Initialize(), LoadMRUList(), LoadTypedURLs(), and PersistMRU().

m_MRUData

CSimpleArray<CStringW> CACLCustomMRU::m_MRUData

private

Definition at line 21 of file ACLCustomMRU.h.

Referenced by AddMRUString(), Initialize(), LoadMRUList(), LoadTypedURLs(), Next(), and PersistMRU().

m_MRUList

CStringW CACLCustomMRU::m_MRUList

private

Definition at line 20 of file ACLCustomMRU.h.

Referenced by AddMRUString(), LoadMRUList(), and PersistMRU().

---

The documentation for this class was generated from the following files:

• dll/win32/browseui/ACLCustomMRU.h
• dll/win32/browseui/ACLCustomMRU.cpp