Category

Development Platform
Home > SourceCode/Document > Windows Develop > Windows Kernel > View Code
idmk.c
Package: shell.rar　[view]
Upload User: xhy777
Upload Date: 2007-02-14
Package Size: 24088k
Code Size: 3k
Category:

Windows Kernel

Development Platform:

Visual C++

idmk.c：Code Content
  1. #include "shellprv.h"
  2. #pragma  hdrstop
  4. typedef struct _DefEnum         // deunk
  5. {
  6.     IEnumIDList         eunk;
  7.     int                 cRef;
  8.     int                 iCur;
  9.     LPITEMIDLIST        pidlReturn;
  10.     LPVOID              lpData;
  11.     LPFNENUMCALLBACK    lpfn;
  12. } CDefEnum;
  14. extern IEnumIDListVtbl c_DefEnumVtbl;   // forward
  16. STDAPI SHCreateEnumObjects(HWND hwndOwner, LPVOID lpData, LPFNENUMCALLBACK lpfn, IEnumIDList **ppeunk)
  17. {
  18.     CDefEnum * pdeunk = (CDefEnum *)LocalAlloc( LPTR, SIZEOF(CDefEnum));
  19.     if (pdeunk)
  20.     {
  21.         pdeunk->eunk.lpVtbl = &c_DefEnumVtbl;
  22.         pdeunk->cRef = 1;
  24.         pdeunk->lpData = lpData;
  25.         pdeunk->lpfn   = lpfn;
  27.         ASSERT(pdeunk->iCur == 0);
  28.         ASSERT(pdeunk->pidlReturn == NULL);
  30.         *ppeunk = &pdeunk->eunk;
  32.         return NOERROR;
  33.     }
  34.     *ppeunk = NULL;
  35.     return E_OUTOFMEMORY;
  36. }
  39. void CDefEnum_SetReturn(LPARAM lParam, LPITEMIDLIST pidl)
  40. {
  41.     CDefEnum *pdeunk = (CDefEnum *)lParam;
  42.     pdeunk->pidlReturn = pidl;
  43. }
  45. STDMETHODIMP CDefEnum_QueryInterface(IEnumIDList *peunk, REFIID riid, LPVOID *ppvObj)
  46. {
  47.     CDefEnum * this = IToClass(CDefEnum, eunk, peunk);
  49.     if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IEnumIDList))
  50.     {
  51.         *ppvObj = &this->eunk;
  52.         InterlockedIncrement(&this->cRef);
  53.         return NOERROR;
  54.     }
  55.     *ppvObj = NULL;
  56.     return E_NOINTERFACE;
  57. }
  59. STDMETHODIMP_(ULONG) CDefEnum_AddRef(IEnumIDList *peunk)
  60. {
  61.     CDefEnum * this = IToClass(CDefEnum, eunk, peunk);
  62.     return InterlockedIncrement(&this->cRef);
  63. }
  65. STDMETHODIMP_(ULONG) CDefEnum_Release(IEnumIDList *peunk)
  66. {
  67.     CDefEnum * this = IToClass(CDefEnum, eunk, peunk);
  69.     if (InterlockedDecrement(&this->cRef))
  70.         return this->cRef;
  72.     this->lpfn((LPARAM)this, this->lpData, ECID_RELEASE, 0);
  73.     LocalFree((HLOCAL)this);
  75.     return 0;
  76. }
  78. STDMETHODIMP CDefEnum_Next(IEnumIDList *peunk, ULONG celt, LPITEMIDLIST *ppidlOut, ULONG *pceltFetched)
  79. {
  80.     CDefEnum * this = IToClass(CDefEnum, eunk, peunk);
  81.     HRESULT hres;
  83.     this->pidlReturn = NULL;
  85.     hres = this->lpfn((LPARAM)this, this->lpData, ECID_SETNEXTID, this->iCur);
  86.     if (hres == NOERROR)
  87.     {
  88.         this->iCur++;
  90.         ASSERT(this->pidlReturn);
  92.         *ppidlOut = this->pidlReturn;
  94.         if (pceltFetched)
  95.             *pceltFetched = 1;
  96.     }
  97.     else
  98.     {
  99.         ASSERT(this->pidlReturn == NULL);
  101.         *ppidlOut = NULL;
  102.         if (pceltFetched)
  103.             *pceltFetched = 0;
  104.     }
  106.     return hres;
  107. }
  110. ///// These will be used by all enums that aren't yet implemented
  112. STDMETHODIMP CDefEnum_Skip(IEnumIDList *peunk, ULONG celt)
  113. {
  114.     // REVIEW: Implement it later.
  115.     return E_NOTIMPL;
  116. }
  118. STDMETHODIMP CDefEnum_Reset(IEnumIDList *peunk)
  119. {
  120.     // REVIEW: Implement it later.
  121.     return E_NOTIMPL;
  122. }
  124. STDMETHODIMP CDefEnum_Clone(IEnumIDList *peunk, IEnumIDList ** ppenm)
  125. {
  126.     // We'll never support this function.
  127.     return E_FAIL;
  128. }
  130. IEnumIDListVtbl c_DefEnumVtbl =
  131. {
  132.     CDefEnum_QueryInterface, CDefEnum_AddRef, CDefEnum_Release,
  133.     CDefEnum_Next,
  134.     CDefEnum_Skip,
  135.     CDefEnum_Reset,
  136.     CDefEnum_Clone
  137. };