CEnumNTDirectory Class Reference

Inheritance diagram for CEnumNTDirectory:

Collaboration diagram for CEnumNTDirectory:

Public Member Functions
	CEnumNTDirectory ()
	~CEnumNTDirectory ()
HRESULT	Initialize (PCWSTR path)
HRESULT	EnumerateNext (LPITEMIDLIST *ppidl)
STDMETHODIMP	Next (ULONG celt, LPITEMIDLIST *rgelt, ULONG *pceltFetched) override
STDMETHODIMP	Skip (ULONG celt) override
STDMETHODIMP	Reset () override
STDMETHODIMP	Clone (IEnumIDList **ppenum) 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 IEnumIDList
HRESULT	Next ([in] ULONG celt, [out, size_is(celt), length_is(*pceltFetched)] PITEMID_CHILD *rgelt, [out] ULONG *pceltFetched)
HRESULT	Skip ([in] ULONG celt)
HRESULT	Reset ()
HRESULT	Clone ([out] IEnumIDList **ppenum)
Public Member Functions inherited from IUnknown
HRESULT	QueryInterface ([in] REFIID riid, [out, iid_is(riid)] void **ppvObject)
ULONG	AddRef ()
ULONG	Release ()

Private Attributes
WCHAR	buffer [MAX_PATH]
HANDLE	m_directory
BOOL	m_first
ULONG	m_enumContext
PWSTR	m_pend

Additional Inherited Members
Public Types inherited from IEnumIDList
typedef IEnumIDList *	LPENUMIDLIST
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 550 of file ntobjenum.cpp.

Constructor & Destructor Documentation

CEnumNTDirectory()

CEnumNTDirectory::CEnumNTDirectory ( )

inline

Definition at line 561 of file ntobjenum.cpp.

        : m_directory(NULL), m_first(TRUE), m_enumContext(0), m_pend(NULL)
    {
    }

~CEnumNTDirectory()

CEnumNTDirectory::~CEnumNTDirectory ( )

inline

Definition at line 566 of file ntobjenum.cpp.

    {
        NtClose(m_directory);
    }

Member Function Documentation

Clone()

STDMETHODIMP CEnumNTDirectory::Clone ( IEnumIDList **  ppenum )

inlineoverride

Implements IEnumIDList.

Definition at line 714 of file ntobjenum.cpp.

    {
        return E_NOTIMPL;
    }

EnumerateNext()

HRESULT CEnumNTDirectory::EnumerateNext ( LPITEMIDLIST *  ppidl )

inline

Definition at line 588 of file ntobjenum.cpp.

    {
        BYTE dirbuffer[2048];
        if (!NT_SUCCESS(NtQueryDirectoryObject(m_directory, dirbuffer, 2048, TRUE, m_first, &m_enumContext, NULL)))
            return S_FALSE;
 
        m_first = FALSE;
 
        // if ppidl is NULL, assume the caller was Skip(),
        // so we don't care about the info
        if (!ppidl)
            return S_OK;
 
        POBJECT_DIRECTORY_INFORMATION info = (POBJECT_DIRECTORY_INFORMATION) dirbuffer;
 
        if (info->Name.Buffer)
        {
            StringCbCopyNW(m_pend, sizeof(buffer), info->Name.Buffer, info->Name.Length);
        }
 
        OBJECT_TYPE otype = MapTypeNameToType(info->TypeName.Buffer, info->TypeName.Length);
 
        DWORD entryBufferLength = FIELD_OFFSET(NtPidlEntry, entryName) + sizeof(WCHAR);
        if (info->Name.Buffer)
            entryBufferLength += info->Name.Length;
 
        if (otype < 0)
        {
            entryBufferLength += FIELD_OFFSET(NtPidlTypeData, typeName) + sizeof(WCHAR);
 
            if (info->TypeName.Buffer)
            {
                entryBufferLength += info->TypeName.Length;
            }
        }
 
        // allocate space for the terminator
        entryBufferLength += FIELD_OFFSET(SHITEMID, abID);
 
        NtPidlEntry* entry = (NtPidlEntry*) CoTaskMemAlloc(entryBufferLength);
        if (!entry)
            return E_OUTOFMEMORY;
 
        memset(entry, 0, entryBufferLength);
 
        entry->cb = FIELD_OFFSET(NtPidlEntry, entryName);
        entry->magic = NT_OBJECT_PIDL_MAGIC;
        entry->objectType = otype;
 
        if (info->Name.Buffer)
        {
            entry->entryNameLength = info->Name.Length;
            StringCbCopyNW(entry->entryName, entryBufferLength, info->Name.Buffer, info->Name.Length);
            entry->cb += entry->entryNameLength + sizeof(WCHAR);
        }
        else
        {
            entry->entryNameLength = 0;
            entry->entryName[0] = 0;
            entry->cb += sizeof(WCHAR);
        }
 
        if (otype < 0)
        {
            NtPidlTypeData * typedata = (NtPidlTypeData*) ((PBYTE) entry + entry->cb);
            DWORD remainingSpace = entryBufferLength - ((PBYTE) (typedata->typeName) - (PBYTE) entry);
 
            if (info->TypeName.Buffer)
            {
                typedata->typeNameLength = info->TypeName.Length;
                StringCbCopyNW(typedata->typeName, remainingSpace, info->TypeName.Buffer, info->TypeName.Length);
 
                entry->cb += typedata->typeNameLength + sizeof(WCHAR);
            }
            else
            {
                typedata->typeNameLength = 0;
                typedata->typeName[0] = 0;
                entry->cb += typedata->typeNameLength + sizeof(WCHAR);
            }
        }
 
        *ppidl = (LPITEMIDLIST) entry;
 
        return S_OK;
    }

Initialize()

HRESULT CEnumNTDirectory::Initialize ( PCWSTR  path )

inline

Definition at line 571 of file ntobjenum.cpp.

    {
        StringCbCopyExW(buffer, sizeof(buffer), path, &m_pend, NULL, 0);
 
        DWORD err = NtOpenObject(DIRECTORY_OBJECT, &m_directory, FILE_LIST_DIRECTORY, buffer);
        if (!NT_SUCCESS(err))
        {
            ERR("NtOpenDirectoryObject failed for path %S with status=%x\n", buffer, err);
            return HRESULT_FROM_NT(err);
        }
 
        if (m_pend[-1] != '\\')
            *m_pend++ = '\\';
 
        return S_OK;
    }

Next()

STDMETHODIMP CEnumNTDirectory::Next ( ULONG  celt, LPITEMIDLIST *  rgelt, ULONG *  pceltFetched  )

inlineoverride

Definition at line 675 of file ntobjenum.cpp.

    {
        if (pceltFetched)
            *pceltFetched = 0;
 
        while (celt-- > 0)
        {
            HRESULT hr = EnumerateNext(rgelt);
            if (hr != S_OK)
                return hr;
 
            if (pceltFetched)
                (*pceltFetched)++;
            if (rgelt)
                rgelt++;
        }
 
        return S_OK;
    }

Reset()

STDMETHODIMP CEnumNTDirectory::Reset ( )

inlineoverride

Implements IEnumIDList.

Definition at line 709 of file ntobjenum.cpp.

    {
        return E_NOTIMPL;
    }

Skip()

STDMETHODIMP CEnumNTDirectory::Skip ( ULONG  celt )

inlineoverride

Implements IEnumIDList.

Definition at line 695 of file ntobjenum.cpp.

    {
        while (celt > 0)
        {
            HRESULT hr = EnumerateNext(NULL);
            if (FAILED(hr))
                return hr;
            if (hr != S_OK)
                break;
        }
 
        return S_OK;
    }

Member Data Documentation

buffer

WCHAR CEnumNTDirectory::buffer[MAX_PATH]

private

Definition at line 554 of file ntobjenum.cpp.

m_directory

HANDLE CEnumNTDirectory::m_directory

private

Definition at line 555 of file ntobjenum.cpp.

m_enumContext

ULONG CEnumNTDirectory::m_enumContext

private

Definition at line 557 of file ntobjenum.cpp.

m_first

BOOL CEnumNTDirectory::m_first

private

Definition at line 556 of file ntobjenum.cpp.

m_pend

PWSTR CEnumNTDirectory::m_pend

private

Definition at line 558 of file ntobjenum.cpp.

---

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

• dll/shellext/ntobjshex/ntobjenum.cpp