CEnumNTDirectory Class Reference

Inheritance diagram for CEnumNTDirectory:

Collaboration diagram for CEnumNTDirectory:

Public Member Functions
	CEnumNTDirectory ()
	~CEnumNTDirectory ()
HRESULT	Initialize (PCWSTR path)
HRESULT	EnumerateNext (LPITEMIDLIST *ppidl)
virtual HRESULT STDMETHODCALLTYPE	Next (ULONG celt, LPITEMIDLIST *rgelt, ULONG *pceltFetched)
virtual HRESULT STDMETHODCALLTYPE	Skip (ULONG celt)
virtual HRESULT STDMETHODCALLTYPE	Reset ()
virtual HRESULT STDMETHODCALLTYPE	Clone (IEnumIDList **ppenum)
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)
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 561 of file ntobjenum.cpp.

Constructor & Destructor Documentation

CEnumNTDirectory()

CEnumNTDirectory::CEnumNTDirectory ( )

inline

Definition at line 572 of file ntobjenum.cpp.

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

~CEnumNTDirectory()

CEnumNTDirectory::~CEnumNTDirectory ( )

inline

Definition at line 577 of file ntobjenum.cpp.

    {
        NtClose(m_directory);
    }

Member Function Documentation

Clone()

virtual HRESULT STDMETHODCALLTYPE CEnumNTDirectory::Clone ( IEnumIDList **  ppenum )

inlinevirtual

Implements IEnumIDList.

Definition at line 725 of file ntobjenum.cpp.

    {
        return E_NOTIMPL;
    }

EnumerateNext()

HRESULT CEnumNTDirectory::EnumerateNext ( LPITEMIDLIST *  ppidl )

inline

Definition at line 599 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 582 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()

virtual HRESULT STDMETHODCALLTYPE CEnumNTDirectory::Next ( ULONG  celt, LPITEMIDLIST *  rgelt, ULONG *  pceltFetched  )

inlinevirtual

Definition at line 686 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()

virtual HRESULT STDMETHODCALLTYPE CEnumNTDirectory::Reset ( void  )

inlinevirtual

Implements IEnumIDList.

Definition at line 720 of file ntobjenum.cpp.

    {
        return E_NOTIMPL;
    }

Skip()

virtual HRESULT STDMETHODCALLTYPE CEnumNTDirectory::Skip ( ULONG  celt )

inlinevirtual

Implements IEnumIDList.

Definition at line 706 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 565 of file ntobjenum.cpp.

m_directory

HANDLE CEnumNTDirectory::m_directory

private

Definition at line 566 of file ntobjenum.cpp.

m_enumContext

ULONG CEnumNTDirectory::m_enumContext

private

Definition at line 568 of file ntobjenum.cpp.

m_first

BOOL CEnumNTDirectory::m_first

private

Definition at line 567 of file ntobjenum.cpp.

m_pend

PWSTR CEnumNTDirectory::m_pend

private

Definition at line 569 of file ntobjenum.cpp.

---

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

• dll/shellext/ntobjshex/ntobjenum.cpp