Category

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

Windows Kernel

Development Platform:

Visual C++

appsize.cpp：Code Content
  1. //---------------------------------------------------------------------------
  2. //
  3. // Copyright (c) Microsoft Corporation 
  4. //
  5. // File: appsize.cpp
  6. //
  7. // Compute the size of an application 
  8. // 
  9. // History:
  10. //         2-17-98  by dli implemented CAppFolderSize
  11. //------------------------------------------------------------------------
  12. #include "priv.h"
  14. // Do not build this file if on Win9X or NT4
  15. #ifndef DOWNLEVEL_PLATFORM
  17. #include "appsize.h"
  19. // NOTE: CAppFolderSize and CAppFolderFinder are very similar to C*TreeWalkCB in Shell32.
  21. CAppFolderSize::CAppFolderSize(ULONGLONG * puSize): _cRef(1), _puSize(puSize)
  22. {
  23.     CoInitialize(0);
  24. }
  26. CAppFolderSize::~CAppFolderSize()
  27. {
  28.     if (_pstw)
  29.         _pstw->Release();
  30.     CoUninitialize();
  31. }
  33. HRESULT CAppFolderSize::QueryInterface(REFIID riid, LPVOID * ppvOut)
  34. { 
  35.     static const QITAB qit[] = {
  36.         QITABENT(CAppFolderSize, IShellTreeWalkerCallBack),       // IID_IShellTreeWalkerCallBack
  37.         { 0 },
  38.     };
  40.     return QISearch(this, qit, riid, ppvOut);
  41. }
  44. ULONG CAppFolderSize::AddRef()
  45. {
  46.     _cRef++;
  47.     return _cRef;
  48. }
  50. ULONG CAppFolderSize::Release()
  51. {
  52.     _cRef--;
  53.     if (_cRef > 0)
  54.         return _cRef;
  56.     delete this;
  57.     return 0;
  58. }
  60. HRESULT CAppFolderSize::Initialize()
  61. {
  62.     return CoCreateInstance(CLSID_CShellTreeWalker, NULL, CLSCTX_INPROC_SERVER, IID_IShellTreeWalker, (LPVOID *)&_pstw);
  63. }   
  65. //
  66. // IShellTreeWalkerCallBack::FoundFile 
  67. //
  68. HRESULT CAppFolderSize::FoundFile(LPCWSTR pwszFolder, LPTREEWALKERSTATS ptws, WIN32_FIND_DATAW * pwfd)
  69. {
  70.     HRESULT hres = S_OK;
  71.     if (_puSize)
  72.         *_puSize = ptws->ulActualSize;
  73.     return hres;
  74. }
  76. HRESULT CAppFolderSize::EnterFolder(LPCWSTR pwszFolder, LPTREEWALKERSTATS ptws, WIN32_FIND_DATAW * pwfd)
  77. {
  78.     return E_NOTIMPL;
  79. }
  81. //
  82. // IShellTreeWalkerCallBack::LeaveFolder
  83. //
  84. HRESULT CAppFolderSize::LeaveFolder(LPCWSTR pwszFolder, LPTREEWALKERSTATS ptws)
  85. {
  86.     return E_NOTIMPL;
  87. }
  89. //
  90. // IShellTreeWalkerCallBack::HandleError 
  91. //
  92. HRESULT CAppFolderSize::HandleError(LPCWSTR pwszFolder, LPTREEWALKERSTATS ptws, HRESULT ErrorCode)
  93. {
  94.     return E_NOTIMPL;
  95. }
  97. #endif //DOWNLEVEL_PLATFORM