RecycleBinGenericEnum Class Reference

Inheritance diagram for RecycleBinGenericEnum:

Collaboration diagram for RecycleBinGenericEnum:

Public Member Functions
	RecycleBinGenericEnum ()
virtual	~RecycleBinGenericEnum ()
STDMETHODIMP	QueryInterface (REFIID riid, void **ppvObject) override
	STDMETHODIMP_ (ULONG) AddRef() override
	STDMETHODIMP_ (ULONG) Release() override
STDMETHODIMP	Next (DWORD celt, IRecycleBinFile **rgelt, DWORD *pceltFetched) override
STDMETHODIMP	Skip (DWORD celt) override
STDMETHODIMP	Reset () override

Protected Attributes
LONG	m_ref
IRecycleBinEnumList *	m_current
DWORD	m_dwLogicalDrives
SIZE_T	m_skip

Detailed Description

Definition at line 11 of file recyclebin_generic_enumerator.cpp.

Constructor & Destructor Documentation

RecycleBinGenericEnum()

RecycleBinGenericEnum::RecycleBinGenericEnum

(

)

Definition at line 200 of file recyclebin_generic_enumerator.cpp.

    : m_ref(1)
    , m_current(NULL)
    , m_dwLogicalDrives(0)
    , m_skip(0)
{
}

~RecycleBinGenericEnum()

RecycleBinGenericEnum::~RecycleBinGenericEnum ( )

virtual

Definition at line 62 of file recyclebin_generic_enumerator.cpp.

{
    TRACE("(%p)\n", this);
 
    if (m_current)
        m_current->Release();
}

Member Function Documentation

Next()

STDMETHODIMP RecycleBinGenericEnum::Next ( DWORD  celt, IRecycleBinFile **  rgelt, DWORD *  pceltFetched  )

override

Definition at line 82 of file recyclebin_generic_enumerator.cpp.

{
    TRACE("(%p, %u, %p, %p)\n", this, celt, rgelt, pceltFetched);
 
    if (!rgelt)
        return E_POINTER;
    if (!pceltFetched && celt > 1)
        return E_INVALIDARG;
 
    HRESULT hr;
    DWORD fetched = 0;
    while (TRUE)
    {
        /* Get enumerator implementation */
        if (!m_current && m_dwLogicalDrives)
        {
            for (DWORD i = 0; i < L'Z' - L'A' + 1; ++i)
            {
                if (m_dwLogicalDrives & (1 << i))
                {
                    WCHAR szVolumeName[4];
                    szVolumeName[0] = (WCHAR)(L'A' + i);
                    szVolumeName[1] = L':';
                    szVolumeName[2] = L'\\';
                    szVolumeName[3] = UNICODE_NULL;
                    if (GetDriveTypeW(szVolumeName) != DRIVE_FIXED)
                    {
                        m_dwLogicalDrives &= ~(1 << i);
                        continue;
                    }
 
                    IRecycleBin *prb;
                    hr = GetDefaultRecycleBin(szVolumeName, &prb);
                    if (!SUCCEEDED(hr))
                        return hr;
                    hr = prb->EnumObjects(&m_current);
                    prb->Release();
 
                    if (!SUCCEEDED(hr))
                        return hr;
 
                    m_dwLogicalDrives &= ~(1 << i);
                    break;
                }
            }
        }
 
        if (!m_current)
        {
            /* Nothing more to enumerate */
            if (pceltFetched)
                *pceltFetched = fetched;
            return S_FALSE;
        }
 
        /* Skip some elements */
        while (m_skip > 0)
        {
            IRecycleBinFile *rbf;
            hr = m_current->Next(1, &rbf, NULL);
            if (hr == S_OK)
                rbf->Release();
            else if (hr == S_FALSE)
                break;
            else if (!SUCCEEDED(hr))
                return hr;
        }
 
        if (m_skip > 0)
            continue;
 
        /* Fill area */
        DWORD newFetched;
        hr = m_current->Next(celt - fetched, &rgelt[fetched], &newFetched);
        if (SUCCEEDED(hr))
            fetched += newFetched;
 
        if (hr == S_FALSE || newFetched == 0)
        {
            m_current->Release();
            m_current = NULL;
        }
        else if (!SUCCEEDED(hr))
            return hr;
 
        if (fetched == celt)
        {
            if (pceltFetched)
                *pceltFetched = fetched;
            return S_OK;
        }
    }
 
    /* Never go here */
    UNREACHABLE;
}

QueryInterface()

STDMETHODIMP RecycleBinGenericEnum::QueryInterface ( REFIID  riid, void **  ppvObject  )

override

Definition at line 35 of file recyclebin_generic_enumerator.cpp.

{
    TRACE("(%p, %s, %p)\n", this, debugstr_guid(&riid), ppvObject);
 
    if (!ppvObject)
        return E_POINTER;
 
    if (IsEqualIID(riid, IID_IUnknown) || IsEqualIID(riid, IID_IRecycleBinEnumList))
        *ppvObject = static_cast<IRecycleBinEnumList *>(this);
    else
    {
        *ppvObject = NULL;
        return E_NOINTERFACE;
    }
 
    AddRef();
    return S_OK;
}

Reset()

STDMETHODIMP RecycleBinGenericEnum::Reset ( )

override

Definition at line 186 of file recyclebin_generic_enumerator.cpp.

{
    TRACE("(%p)\n", this);
 
    if (m_current)
    {
        m_current->Release();
        m_current = NULL;
    }
    m_skip = 0;
    m_dwLogicalDrives = ::GetLogicalDrives();
    return S_OK;
}

Skip()

STDMETHODIMP RecycleBinGenericEnum::Skip ( DWORD  celt )

override

Definition at line 179 of file recyclebin_generic_enumerator.cpp.

{
    TRACE("(%p, %u)\n", this, celt);
    m_skip += celt;
    return S_OK;
}

STDMETHODIMP_() [1/2]

RecycleBinGenericEnum::STDMETHODIMP_ ( ULONG  )

override

STDMETHODIMP_() [2/2]

RecycleBinGenericEnum::STDMETHODIMP_ ( ULONG  )

override

Member Data Documentation

m_current

IRecycleBinEnumList* RecycleBinGenericEnum::m_current

protected

Definition at line 29 of file recyclebin_generic_enumerator.cpp.

Referenced by Next(), Reset(), and ~RecycleBinGenericEnum().

m_dwLogicalDrives

DWORD RecycleBinGenericEnum::m_dwLogicalDrives

protected

Definition at line 30 of file recyclebin_generic_enumerator.cpp.

Referenced by Next(), and Reset().

m_ref

LONG RecycleBinGenericEnum::m_ref

protected

Definition at line 28 of file recyclebin_generic_enumerator.cpp.

m_skip

SIZE_T RecycleBinGenericEnum::m_skip

protected

Definition at line 31 of file recyclebin_generic_enumerator.cpp.

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

---

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

• dll/win32/shell32/shellrecyclebin/recyclebin_generic_enumerator.cpp