VC.NET扩展Windows磁盘清理工具的功能-技术开发专区

VC.NET扩展Windows磁盘清理工具的功能

作者：刘桂英编辑：李博（泡泡） 2006-04-12 00:00

【IT168 技术文档】

　　引言

　　Windows磁盘清理工具（Disk CleanUp）是一个实用快捷并拥有简单易用界面的系统清理软件，更值得系统开发管理人员注意的是，此系统清理软件是建立在以COM技术为基础发展的，支持第三方插件，并且可以根据需要自制定义功能二次开发的平台。在这里，我们对于Windows磁盘清理工具的开发接口做深入地研究，在此基础上举例示范添加一个查找“*.tmp”临时文件的功能。

　　技术讨论

　　微软的COM技术广泛地运用在Windows的模块化设计中，致使支持二次开发。关于COM技术基础与应用，可参考。在此，我们只为Windows磁盘清理工具，简称清理工具的扩展接口加以分解。清理工具首次出现在Windows 98操作系统中，并在后来推出的Windows版本中予以改进，添加了新的功能。比如说，在NTFS的文件系统下，自动压缩不经常访问的文件。这些新功能通过COM模块实现，在清理工具中作为插件调用。早期的版本是通过IEmptyVolumeCache接口调用，在Windows 2000以后的版本中，还加入了IEmptyVolumeCache2接口，加入了较小的更新。

　　IEmptyVolumeCache接口由五个函数组成，根据呼叫的顺序，分别是：

virtual /* [local] */ HRESULT STDMETHODCALLTYPE Initialize(
/* [in] */ HKEY hkRegKey,
/* [in] */ LPCWSTR pcwszVolume,
/* [out] */ LPWSTR *ppwszDisplayName,
/* [out] */ LPWSTR *ppwszDescription,
/* [out] */ DWORD *pdwFlags) = 0;
virtual HRESULT STDMETHODCALLTYPE GetSpaceUsed(
/* [out] */ DWORDLONG *pdwlSpaceUsed,
/* [in] */ IEmptyVolumeCacheCallBack *picb) = 0;
virtual HRESULT STDMETHODCALLTYPE ShowProperties(
/* [in] */ HWND hwnd) = 0;
virtual HRESULT STDMETHODCALLTYPE Purge(
/* [in] */ DWORDLONG dwlSpaceToFree,
/* [in] */ IEmptyVolumeCacheCallBack *picb) = 0;
virtual HRESULT STDMETHODCALLTYPE Deactivate(
/* [out] */ DWORD *pdwFlags) = 0;

　　清理工具在正常执行时，首先调用Initialize初始化插件，随后执行GetSpaceUsed来扫描可清除的文件大小。扫描完毕后，清理工具的主界面就出现了，在此，我们加入了清理TMP文件的功能可以浏览不同的清理文件种类。列表中的每一个文件种类均由一个COM插件实现。除了阅览可清理文件大小以外，用户在可以点击一个可自定义的按钮，调用插件的ShowProperties功能，以显示更详细的资料。如用户选择OK，清理工具就调用Purge函数，清理扫描出来的文件。最后，Deactivate函数被调用，终止插件的应用。

　　运用于Windows 2000以后的清理工具的插件也应该支持IEmptyVolumeCache的接口。IEmptyVolumeCache只由一个函数组成：

virtual /* [local] */ HRESULT STDMETHODCALLTYPE InitializeEx(
/* [in] */ HKEY hkRegKey,
/* [in] */ LPCWSTR pcwszVolume,
/* [in] */ LPCWSTR pcwszKeyName,
/* [out] */ LPWSTR *ppwszDisplayName,
/* [out] */ LPWSTR *ppwszDescription,
/* [out] */ LPWSTR *ppwszBtnText,
/* [out] */ DWORD *pdwFlags) = 0;

　　InitializeEx增加了更严格的本地化语言要求，加强了国际化的支持，并且允许自定义按钮的显示文字。pdwFlags变量用于在工具与插件间传递信息，支持下列旗标：

　　EVCF_OUTOFDISKSPACE

　　EVCF_SETTINGSMODE

　　EVCF_DONTSHOWIFZERO

　　EVCF_ENABLEBYDEFAULT

　　EVCF_ENABLEBYDEFAULT_AUTO

　　EVCF_HASSETTINGS

　　EVCF_REMOVEFROMLIST

　　EVCF_OUTOFDISKSPACE与EVCF_SETTINGSMODE用于工具传递给插件的设定。EVCF_OUTOFDISKSPACE表示当前硬盘的空余空间非常有限，需要尽可能多地清理，即使是系统的性能会受到影响。EVCF_SETTINGSMODE表示可定期执行的无人控制模式。在此模式下，GetSpaceUsed，Purge，及ShowProperties都将不予调用，所有清理任务应予InitializeEx时执行。其它旗标用于插件传递给工具的不同运行模式。EVCF_DONTSHOWIFZERO表示在没有找到可删除文件时不显示此类型，EVCF_ENABLEBYDEFAULT表示此类型文件可以安全删除，EVCF_ENABLEBYDEFAULT_AUTO表示此类型文件可以非常安全的删除，EVCF_HASSETTINGS表示此插件支持ShowProperties功能，可以显示详细信息。EVCF_REMOVEFROMLIST表示是一次性清理任务，清理工具在执行后自动将插件关闭，以后不再执行。

　　实现方法

　　我们开发一个新的清理工具插件，扫描并清理*.TMP文件。COM的编程有多种方法，我们选择了ATL库。关于ATL库的运用。

　　我们在Visual Studio.NET中生成新的ATL的DLL Server项目，并使用Add Class加入新的ATL Simple Object控件类CCleanSimpleHandler。在定义中，我们让CCleanSimpleHandler从IEmptyVolumeCache2继承。并且，我们添加了下列变量：

// 储存扫描出文件的大小
DWORDLONG m_dwlFileSize;
// 储存根目录
WCHAR m_strRootDir[MAX_PATH];
// 储存扫描出文件列表
std::vector<WCHAR *> m_lstFilesToDel;

　　然后，我们一一实现IEmptyVolumeCache及IEmptyVolumeCache2接口的函数。在下面的代码列表中，没有包括严格的检查错误返回值。这是为了简短代码的长度，提高可读性。在实际应用中，检查错误返回值是不可少的。为了不同版本Windows兼容，我们在InitializeEx中调用Initialize。

HRESULT CCleanSimpleHandler::InitializeEx (HKEY hKey,_
LPCWSTR pcwszVolume, LPCWSTR pcwszKeyName, LPWSTR *ppwszDisplayName,_
LPWSTR *ppwszDescription, LPWSTR *ppwszBtnText, DWORD *pdwFlags)
{
　HRESULT hr = Initialize (hKey, pcwszVolume,_
ppwszDisplayName, ppwszDescription, pdwFlags);
　*ppwszBtnText = (LPWSTR) CoTaskMemAlloc (64 * sizeof (WCHAR));
　StrCpyW(*ppwszBtnText, L"View files");
　return hr;
}
HRESULT CCleanSimpleHandler::Initialize (HKEY hKey,_
LPCWSTR pcwszVolume, LPWSTR *ppwszDisplayName,_
LPWSTR *ppwszDescription, DWORD *pdwFlags)
{
　StrCpyW(m_strRootDir, pcwszVolume);
　*ppwszDisplayName = (LPWSTR) CoTaskMemAlloc(256 * sizeof (WCHAR));
　StrCpyW(*ppwszDisplayName, L"*.TMP files");
　*ppwszDescription = (LPWSTR) CoTaskMemAlloc (256 * sizeof (WCHAR));
　StrCpyW(*ppwszDescription, L"Temporary files - *.TMP");
　*pdwFlags = EVCF_HASSETTINGS | EVCF_ENABLEBYDEFAULT;
　m_dwlFileSize = 0;
　return S_OK;
}

　　在GetSpaceUsed中，我们调用ScanDir来扫描*.TMP文件，储存在m_lstFilesToDel中。GetSpaceUsed的第二个参数是IEmptyVolumeCacheCallBack接口的指针，用于调用其ScanProgress函数以报告扫描的进展情况。ScanProgress函数定义是：

HRESULT ScanProgress(DWORDLONG dwlSpaceUsed, DWORD dwFlags, LPCWSTR pwszReserved);

　　其中dwFlags正常应设为零，在结束时改为EVCCBF_LASTNOTIFICATION。ScanProgress函数的返回值很重要，因为用户可以在任何时候中断在进行中的清理任务。如ScanProgress返回E_ABORT，GetSpaceUsed应最快终端扫描，函数返回。因此，我们在递归的目录扫描函数ScanDir中，加入了如中断立即退出的功能。

HRESULT CCleanSimpleHandler::GetSpaceUsed (DWORDLONG *pdwSpaceUsed,
 　IEmptyVolumeCacheCallBack *picb)
{
　m_dwlFileSize = 0;
　ScanDir(m_strRootDir, picb);
　picb->ScanProgress(m_dwlFileSize, EVCCBF_LASTNOTIFICATION ,NULL);
　*pdwSpaceUsed = m_dwlFileSize;
　return S_OK;
}
bool CCleanSimpleHandler::ScanDir(WCHAR * szDir,
IEmptyVolumeCacheCallBack *pcib)
{
　WCHAR strPath[MAX_PATH];
　WCHAR* pchPathFileName;
　bool cancelled = false;
　WIN32_FIND_DATAW fd;
　HANDLE hFind;
　if (cancelled = FAILED(pcib->ScanProgress(m_dwlFileSize,
NULL, NULL))) return false;
　StrCpyW(strPath,szDir);
　PathAppendW(strPath, L"*");
　pchPathFileName = strPath+lstrlenW(strPath)-1;
　hFind = FindFirstFileW(strPath, &fd);
　if (hFind == INVALID_HANDLE_VALUE) // E.g. Due to security issues
　　return true;
　　do {
　　　StrCpyW(pchPathFileName, fd.cFileName);
　　　if ((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
　　　　if (fd.cFileName[0] != '.') {
　　　　　if (cancelled = !ScanDir(strPath, pcib)) break;
　　　　}
　　　} else {
　　　　WCHAR* pchExt = PathFindExtensionW(strPath);
　　　　if ( StrCmpIW(pchExt, L".tmp") == 0 ) {
　　　　　m_dwlFileSize += ((DWORDLONG)fd.nFileSizeHigh)*4294967295+fd.nFileSizeLow;
　　　　　WCHAR* filename = (WCHAR *)CoTaskMemAlloc(
(lstrlenW(strPath)+1)*sizeof(WCHAR));
　　　　　StrCpyW(filename, strPath);
　　　　　m_lstFilesToDel.push_back(filename);
　　　　}
　　　}
　　} while (FindNextFileW(hFind, &fd) != NULL);
　　FindClose(hFind);
　　return !cancelled;
}

　　其他的函数很简单。Purge函数将扫描出的文件列表m_lstFilesToDel中的文件一一删除。ShowProperties中，我们显示扫描出来的文件。最后，Deactivate将分配的内存释放。

HRESULT CCleanSimpleHandler::Purge (DWORDLONG dwSpaceToFree,
　IEmptyVolumeCacheCallBack *picb)
{
　for (unsigned int i=0; i < m_lstFilesToDel.size(); ++i)
　　DeleteFileW(m_lstFilesToDel[i]);
　　return S_OK;
}
HRESULT CCleanSimpleHandler::ShowProperties (HWND hWnd)
{
　for (unsigned int i=0; i < m_lstFilesToDel.size(); ++i)
　　if (MessageBoxW(hWnd, m_lstFilesToDel[i], L"View files",
　　　MB_OKCANCEL|MB_ICONINFORMATION)==IDCANCEL) break;
　　　return S_OK;
}
HRESULT CCleanSimpleHandler::Deactivate (LPDWORD pdwFlags)
{
　for (unsigned int i=0; i < m_lstFilesToDel.size(); ++i)
　　CoTaskMemFree(m_lstFilesToDel[i]);
　　m_lstFilesToDel.clear();
　　*pdwFlags = 0;
　　return S_OK;

　　结论和建议

　　通过实例分解，我们对Windows磁盘清理工具的基于COM技术的开发接口做了深入地研究。Windows外壳中有较多的开发接口，本文介绍的开发思想也可以运用在其它扩展插件中。

关注我们