Category

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

Windows Kernel

Development Platform:

Visual C++

ftpmi.cpp：Code Content
  1. /*****************************************************************************
  2.  *
  3.  * ftpmi.cpp - IMalloc interface for allocing pidls
  4.  *
  5.  *****************************************************************************/
  7. #include "priv.h"
  8. #include "ftpmi.h"
  11. /*****************************************************************************
  12.  * IMalloc::Alloc
  13.  *****************************************************************************/
  15. LPVOID CMallocItem::Alloc(ULONG cbSize)
  16. {
  17.     WORD cbActualSize = sizeof(DELEGATEITEMID) - 1 + cbSize;
  18.     PDELEGATEITEMID pidl = (PDELEGATEITEMID)SHAlloc(cbActualSize + 2);
  20.     if (pidl)
  21.     {
  22.         pidl->cbSize = cbActualSize;
  23.         pidl->wOuter = 0x6646;          // "Ff"
  24.         pidl->cbInner = (WORD)cbSize;
  25.         *(WORD *)&(((BYTE *)pidl)[cbActualSize]) = 0;
  26.     }
  28.     return pidl;
  29. }
  31. /*****************************************************************************
  32.  * IMalloc::Realloc
  33.  *****************************************************************************/
  35. LPVOID CMallocItem::Realloc(LPVOID pv, ULONG cb)
  36. {
  37.     return NULL;
  38. }
  40. /*****************************************************************************
  41.  * IMalloc::Free
  42.  *****************************************************************************/
  44. void CMallocItem::Free(LPVOID pv)
  45. {
  46.     SHFree(pv);
  47. }
  49. /*****************************************************************************
  50.  * IMalloc::GetSize
  51.  *****************************************************************************/
  53. ULONG CMallocItem::GetSize(LPVOID pv)
  54. {
  55.     return (ULONG)-1;
  56. }
  58. /*****************************************************************************
  59.  * IMalloc::DidAlloc
  60.  *****************************************************************************/
  62. int CMallocItem::DidAlloc(LPVOID pv)
  63. {
  64.     return -1;
  65. }
  67. /*****************************************************************************
  68.  * IMalloc::HeapMinimize
  69.  *****************************************************************************/
  71. void CMallocItem::HeapMinimize(void)
  72. {
  73.     NULL;
  74. }
  77. /*****************************************************************************
  78.  * CMallocItem_Create
  79.  *****************************************************************************/
  81. HRESULT CMallocItem_Create(IMalloc ** ppm)
  82. {
  83.     HRESULT hres = E_OUTOFMEMORY;
  84.     CMallocItem * pmi = new CMallocItem();
  86.     if (pmi)
  87.     {
  88.         hres = pmi->QueryInterface(IID_IMalloc, (LPVOID *) ppm);
  89.         pmi->Release();
  90.     }
  92.     return hres;
  93. }
  96. /****************************************************
  97.     Constructor
  98. ****************************************************/
  99. CMallocItem::CMallocItem() : m_cRef(1)
  100. {
  101.     DllAddRef();
  103.     // This needs to be allocated in Zero Inited Memory.
  104.     // Assert that all Member Variables are inited to Zero.
  105.     LEAK_ADDREF(LEAK_CMallocItem);
  106. }
  109. /****************************************************
  110.     Destructor
  111. ****************************************************/
  112. CMallocItem::~CMallocItem()
  113. {
  114.     DllRelease();
  115.     LEAK_DELREF(LEAK_CMallocItem);
  116. }
  119. //===========================
  120. // *** IUnknown Interface ***
  121. //===========================
  123. ULONG CMallocItem::AddRef()
  124. {
  125.     m_cRef++;
  126.     return m_cRef;
  127. }
  129. ULONG CMallocItem::Release()
  130. {
  131.     ASSERT(m_cRef > 0);
  132.     m_cRef--;
  134.     if (m_cRef > 0)
  135.         return m_cRef;
  137.     delete this;
  138.     return 0;
  139. }
  141. HRESULT CMallocItem::QueryInterface(REFIID riid, void **ppvObj)
  142. {
  143.     if (IsEqualIID(riid, IID_IUnknown) || IsEqualIID(riid, IID_IMalloc))
  144.     {
  145.         *ppvObj = SAFECAST(this, IMalloc *);
  146.     }
  147.     else
  148.     {
  149.         TraceMsg(TF_FTPQI, "CMallocItem::QueryInterface() failed.");
  150.         *ppvObj = NULL;
  151.         return E_NOINTERFACE;
  152.     }
  154.     AddRef();
  155.     return S_OK;
  156. }