recyclebin_generic_enumerator.c File Reference

#include "recyclebin_private.h"

Include dependency graph for recyclebin_generic_enumerator.c:

Go to the source code of this file.

Classes
struct	RecycleBinGenericEnum

Functions
static HRESULT STDMETHODCALLTYPE	RecycleBinGenericEnum_RecycleBinEnumList_QueryInterface (IN IRecycleBinEnumList *This, IN REFIID riid, OUT void **ppvObject)
static ULONG STDMETHODCALLTYPE	RecycleBinGenericEnum_RecycleBinEnumList_AddRef (IN IRecycleBinEnumList *This)
static VOID	RecycleBinGenericEnum_Destructor (struct RecycleBinGenericEnum *s)
static ULONG STDMETHODCALLTYPE	RecycleBinGenericEnum_RecycleBinEnumList_Release (IN IRecycleBinEnumList *This)
static HRESULT STDMETHODCALLTYPE	RecycleBinGenericEnum_RecycleBinEnumList_Next (IN IRecycleBinEnumList *This, IN DWORD celt, IN OUT IRecycleBinFile **rgelt, OUT DWORD *pceltFetched)
static HRESULT STDMETHODCALLTYPE	RecycleBinGenericEnum_RecycleBinEnumList_Skip (IN IRecycleBinEnumList *This, IN DWORD celt)
static HRESULT STDMETHODCALLTYPE	RecycleBinGenericEnum_RecycleBinEnumList_Reset (IN IRecycleBinEnumList *This)
HRESULT	RecycleBinGenericEnum_Constructor (OUT IRecycleBinEnumList **pprbel)

Variables
CONST_VTBL struct IRecycleBinEnumListVtbl	RecycleBinGenericEnumVtbl

Function Documentation

RecycleBinGenericEnum_Constructor()

HRESULT RecycleBinGenericEnum_Constructor

(

OUT IRecycleBinEnumList **

pprbel

)

Definition at line 219 of file recyclebin_generic_enumerator.c.

{
    struct RecycleBinGenericEnum *s;
 
    s = CoTaskMemAlloc(sizeof(struct RecycleBinGenericEnum));
    if (!s)
        return E_OUTOFMEMORY;
    ZeroMemory(s, sizeof(struct RecycleBinGenericEnum));
    s->ref = 1;
    s->recycleBinEnumImpl.lpVtbl = &RecycleBinGenericEnumVtbl;
 
    *pprbel = &s->recycleBinEnumImpl;
    return IRecycleBinEnumList_Reset(*pprbel);
}

Referenced by RecycleBinGeneric_RecycleBin_EnumObjects().

RecycleBinGenericEnum_Destructor()

static VOID RecycleBinGenericEnum_Destructor ( struct RecycleBinGenericEnum *  s )

static

Definition at line 58 of file recyclebin_generic_enumerator.c.

{
    TRACE("(%p)\n", s);
 
    if (s->current)
        IRecycleBinEnumList_Release(s->current);
    CoTaskMemFree(s);
}

Referenced by RecycleBinGenericEnum_RecycleBinEnumList_Release().

RecycleBinGenericEnum_RecycleBinEnumList_AddRef()

static ULONG STDMETHODCALLTYPE RecycleBinGenericEnum_RecycleBinEnumList_AddRef ( IN IRecycleBinEnumList *  This )

static

Definition at line 48 of file recyclebin_generic_enumerator.c.

{
    struct RecycleBinGenericEnum *s = CONTAINING_RECORD(This, struct RecycleBinGenericEnum, recycleBinEnumImpl);
    ULONG refCount = InterlockedIncrement((PLONG)&s->ref);
    TRACE("(%p)\n", This);
    return refCount;
}

RecycleBinGenericEnum_RecycleBinEnumList_Next()

static HRESULT STDMETHODCALLTYPE RecycleBinGenericEnum_RecycleBinEnumList_Next ( IN IRecycleBinEnumList *  This, IN DWORD  celt, IN OUT IRecycleBinFile **  rgelt, OUT DWORD *  pceltFetched  )

static

Definition at line 86 of file recyclebin_generic_enumerator.c.

{
    struct RecycleBinGenericEnum *s = CONTAINING_RECORD(This, struct RecycleBinGenericEnum, recycleBinEnumImpl);
    IRecycleBin *prb;
    DWORD i;
    DWORD fetched = 0, newFetched;
    HRESULT hr;
 
    TRACE("(%p, %u, %p, %p)\n", This, celt, rgelt, pceltFetched);
 
    if (!rgelt)
        return E_POINTER;
    if (!pceltFetched && celt > 1)
        return E_INVALIDARG;
 
    while (TRUE)
    {
        /* Get enumerator implementation */
        if (!s->current && s->dwLogicalDrives)
        {
            for (i = 0; i < 26; i++)
                if (s->dwLogicalDrives & (1 << i))
                {
                    WCHAR szVolumeName[4];
                    szVolumeName[0] = (WCHAR)('A' + i);
                    szVolumeName[1] = ':';
                    szVolumeName[2] = '\\';
                    szVolumeName[3] = UNICODE_NULL;
                    if (GetDriveTypeW(szVolumeName) != DRIVE_FIXED)
                    {
                        s->dwLogicalDrives &= ~(1 << i);
                        continue;
                    }
                    hr = GetDefaultRecycleBin(szVolumeName, &prb);
                    if (!SUCCEEDED(hr))
                        return hr;
                    hr = IRecycleBin_EnumObjects(prb, &s->current);
                    IRecycleBin_Release(prb);
                    if (!SUCCEEDED(hr))
                        return hr;
                    s->dwLogicalDrives &= ~(1 << i);
                    break;
                }
        }
        if (!s->current)
        {
            /* Nothing more to enumerate */
            if (pceltFetched)
                *pceltFetched = fetched;
            return S_FALSE;
        }
 
        /* Skip some elements */
        while (s->skip > 0)
        {
            IRecycleBinFile *rbf;
            hr = IRecycleBinEnumList_Next(s->current, 1, &rbf, NULL);
            if (hr == S_OK)
                hr = IRecycleBinFile_Release(rbf);
            else if (hr == S_FALSE)
                break;
            else if (!SUCCEEDED(hr))
                return hr;
        }
        if (s->skip > 0)
            continue;
 
        /* Fill area */
        hr = IRecycleBinEnumList_Next(s->current, celt - fetched, &rgelt[fetched], &newFetched);
        if (SUCCEEDED(hr))
            fetched += newFetched;
        if (hr == S_FALSE || newFetched == 0)
        {
            hr = IRecycleBinEnumList_Release(s->current);
            s->current = NULL;
        }
        else if (!SUCCEEDED(hr))
            return hr;
        if (fetched == celt)
        {
            if (pceltFetched)
                *pceltFetched = fetched;
            return S_OK;
        }
    }
 
    /* Never go here */
}

RecycleBinGenericEnum_RecycleBinEnumList_QueryInterface()

static HRESULT STDMETHODCALLTYPE RecycleBinGenericEnum_RecycleBinEnumList_QueryInterface ( IN IRecycleBinEnumList *  This, IN REFIID  riid, OUT void **  ppvObject  )

static

Definition at line 21 of file recyclebin_generic_enumerator.c.

{
    struct RecycleBinGenericEnum *s = CONTAINING_RECORD(This, struct RecycleBinGenericEnum, recycleBinEnumImpl);
 
    TRACE("(%p, %s, %p)\n", This, debugstr_guid(riid), ppvObject);
 
    if (!ppvObject)
        return E_POINTER;
 
    if (IsEqualIID(riid, &IID_IUnknown))
        *ppvObject = &s->recycleBinEnumImpl;
    else if (IsEqualIID(riid, &IID_IRecycleBinEnumList))
        *ppvObject = &s->recycleBinEnumImpl;
    else
    {
        *ppvObject = NULL;
        return E_NOINTERFACE;
    }
 
    IUnknown_AddRef(This);
    return S_OK;
}

RecycleBinGenericEnum_RecycleBinEnumList_Release()

static ULONG STDMETHODCALLTYPE RecycleBinGenericEnum_RecycleBinEnumList_Release ( IN IRecycleBinEnumList *  This )

static

Definition at line 69 of file recyclebin_generic_enumerator.c.

{
    struct RecycleBinGenericEnum *s = CONTAINING_RECORD(This, struct RecycleBinGenericEnum, recycleBinEnumImpl);
    ULONG refCount;
 
    TRACE("(%p)\n", This);
 
    refCount = InterlockedDecrement((PLONG)&s->ref);
 
    if (refCount == 0)
        RecycleBinGenericEnum_Destructor(s);
 
    return refCount;
}

RecycleBinGenericEnum_RecycleBinEnumList_Reset()

static HRESULT STDMETHODCALLTYPE RecycleBinGenericEnum_RecycleBinEnumList_Reset ( IN IRecycleBinEnumList *  This )

static

Definition at line 191 of file recyclebin_generic_enumerator.c.

{
    struct RecycleBinGenericEnum *s = CONTAINING_RECORD(This, struct RecycleBinGenericEnum, recycleBinEnumImpl);
 
    TRACE("(%p)\n", This);
 
    if (s->current)
    {
        IRecycleBinEnumList_Release(s->current);
        s->current = NULL;
        s->skip = 0;
    }
    s->dwLogicalDrives = GetLogicalDrives();
    return S_OK;
}

RecycleBinGenericEnum_RecycleBinEnumList_Skip()

static HRESULT STDMETHODCALLTYPE RecycleBinGenericEnum_RecycleBinEnumList_Skip ( IN IRecycleBinEnumList *  This, IN DWORD  celt  )

static

Definition at line 180 of file recyclebin_generic_enumerator.c.

{
    struct RecycleBinGenericEnum *s = CONTAINING_RECORD(This, struct RecycleBinGenericEnum, recycleBinEnumImpl);
    TRACE("(%p, %u)\n", This, celt);
    s->skip += celt;
    return S_OK;
}

Variable Documentation

RecycleBinGenericEnumVtbl

CONST_VTBL struct IRecycleBinEnumListVtbl RecycleBinGenericEnumVtbl

Initial value:

=
{
    RecycleBinGenericEnum_RecycleBinEnumList_QueryInterface,
    RecycleBinGenericEnum_RecycleBinEnumList_AddRef,
    RecycleBinGenericEnum_RecycleBinEnumList_Release,
    RecycleBinGenericEnum_RecycleBinEnumList_Next,
    RecycleBinGenericEnum_RecycleBinEnumList_Skip,
    RecycleBinGenericEnum_RecycleBinEnumList_Reset,
}

Definition at line 208 of file recyclebin_generic_enumerator.c.

Referenced by RecycleBinGenericEnum_Constructor().