89 lines
2.1 KiB
C
89 lines
2.1 KiB
C
|
// Enum.h -- Definition of class CEnumStorage
|
||
|
|
|
||
|
|
#ifndef __ENUM_H__
|
||
|
|
|
||
|
|
#define __ENUM_H__
|
||
|
|
|
||
|
|
class CEnumStorage : public CITUnknown
|
||
|
|
{
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
static HRESULT NewEnumStorage
|
||
|
|
(IUnknown *pUnkOuter,
|
||
|
|
IITFileSystem *pITFS, PathInfo *pPI,
|
||
|
|
IEnumSTATSTG **ppEnumSTATSTG
|
||
|
|
);
|
||
|
|
|
||
|
|
~CEnumStorage(void);
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
CEnumStorage(IUnknown *pUnkOuter);
|
||
|
|
|
||
|
|
class CImpIEnumStorage : public IITEnumSTATSTG
|
||
|
|
{
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
CImpIEnumStorage(CEnumStorage *pBackObj, IUnknown *punkOuter);
|
||
|
|
~CImpIEnumStorage(void);
|
||
|
|
|
||
|
|
HRESULT Initial(IITFileSystem *pITFS, PathInfo *pPI);
|
||
|
|
HRESULT InitClone(CImpIEnumStorage *pEnum);
|
||
|
|
|
||
|
|
// IEnumSTATSTG methods:
|
||
|
|
HRESULT STDMETHODCALLTYPE GetNextEntryInSeq(ULONG celt, PathInfo *rgelt, ULONG *pceltFetched);
|
||
|
|
HRESULT STDMETHODCALLTYPE GetFirstEntryInSeq(PathInfo *rgelt);
|
||
|
|
|
||
|
|
/* [local] */ HRESULT __stdcall Next(
|
||
|
|
/* [in] */ ULONG celt,
|
||
|
|
/* [in] */ STATSTG __RPC_FAR *rgelt,
|
||
|
|
/* [out] */ ULONG __RPC_FAR *pceltFetched);
|
||
|
|
|
||
|
|
HRESULT __stdcall Skip(
|
||
|
|
/* [in] */ ULONG celt);
|
||
|
|
|
||
|
|
HRESULT __stdcall Reset( void);
|
||
|
|
|
||
|
|
HRESULT __stdcall Clone(
|
||
|
|
/* [out] */ IEnumSTATSTG __RPC_FAR *__RPC_FAR *ppenum);
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
HRESULT __stdcall NextPathEntry(STATSTG *pStatStg);
|
||
|
|
|
||
|
|
enum EnumState { Before, During, After };
|
||
|
|
|
||
|
|
IEnumSTATSTG *m_pEnumPaths;
|
||
|
|
UINT m_cwcBasePath;
|
||
|
|
WCHAR m_awszBasePath[MAX_PATH];
|
||
|
|
WCHAR m_awcKeyBuffer[MAX_PATH];
|
||
|
|
|
||
|
|
enum EnumState m_State;
|
||
|
|
};
|
||
|
|
|
||
|
|
friend CImpIEnumStorage;
|
||
|
|
|
||
|
|
CImpIEnumStorage m_ImpIEnumStorage;
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
static HRESULT NewClone(IUnknown *pUnkOuter, CImpIEnumStorage *pEnum,
|
||
|
|
IEnumSTATSTG **ppEnumSTATSTG);
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef CEnumStorage *PCEnumStorage;
|
||
|
|
|
||
|
|
inline CEnumStorage::CEnumStorage(IUnknown *pUnkOuter)
|
||
|
|
: m_ImpIEnumStorage(this, pUnkOuter),
|
||
|
|
CITUnknown(&IID_IEnumSTATSTG, 1, &m_ImpIEnumStorage)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
inline CEnumStorage::~CEnumStorage(void)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif // __ENUM_H__
|