|
摘 要 介绍了Windows磁盘清理工具二次开发的扩展接口,对其COM接口加以分解,并运用ATL库具体实现了清理“*.tmp”临时文件的功能。
关键词 磁盘清理工具、ATL库、COM接口。
引言
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来扫描可清除的文件大小。扫描完毕后,清理工具的主界面就出现了如图1所示,在此,我们加入了清理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 |
其中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外壳中有较多的开发接口,本文介绍的开发思想也可以运用在其它扩展插件中。
|