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

图1 清理工具的主界面

　　实现方法

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

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

// 储存扫描出文件的大小 DWORDLONG m_dwlFileSize; // 储存根目录 WCHAR m_strRootDir[MAX_PATH]; // 储存扫描出文件列表 std::vector 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外壳中有较多的开发接口，本文介绍的开发思想也可以运用在其它扩展插件中。

查看本文来源