摘 要 介绍了Windows磁盘清理工具二次开发的扩展接口对其COM接口加以分解并运用ATL库具体实现了清理*tmp临时文件的功能
关键词 磁盘清理工具ATL库COM接口
引言
Windows磁盘清理工具（Disk CleanUp）是一个实用快捷并拥有简单易用界面的系统清理软件更值得系统开发管理人员注意的是此系统清理软件是建立在以COM技术为基础发展的支持第三方插件并且可以根据需要自制定义功能二次开发的平台在这里我们对于Windows磁盘清理工具的开发接口做深入地研究在此基础上举例示范添加一个查找*tmp临时文件的功能
技术讨论
微软的COM技术广泛地运用在Windows的模块化设计中致使支持二次开发关于COM技术基础与应用可参考在此我们只为Windows磁盘清理工具简称清理工具的扩展接口加以分解清理工具首次出现在Windows 操作系统中并在后来推出的Windows版本中予以改进添加了新的功能比如说在NTFS的文件系统下自动压缩不经常访问的文件这些新功能通过COM模块实现在清理工具中作为插件调用早期的版本是通过IEmptyVolumeCache接口调用在Windows 以后的版本中还加入了IEmptyVolumeCache接口加入了较小的更新
IEmptyVolumeCache接口由五个函数组成根据呼叫的顺序分别是
virtual /* [local] */ HRESULT STDMETHODCALLTYPE initialize(
/* [in] */ HKEY hkRegKey
/* [in] */ LPCWSTR pcwszVolume
/* [out] */ LPWSTR *ppwszDisplayName
/* [out] */ LPWSTR *ppwszDescription
/* [out] */ DWORD *pdwFlags) = ;
virtual HRESULT STDMETHODCALLTYPE getspaceused(
/* [out] */ DWORDLONG *pdwlSpaceUsed
/* [in] */ IEmptyVolumeCacheCallBack *picb) = ;
virtual HRESULT STDMETHODCALLTYPE showproperties(
/* [in] */ HWND hwnd) = ;
virtual HRESULT STDMETHODCALLTYPE purge(
/* [in] */ DWORDLONG dwlSpaceToFree
/* [in] */ IEmptyVolumeCacheCallBack *picb) = ;
virtual HRESULT STDMETHODCALLTYPE deactivate(
/* [out] */ DWORD *pdwFlags) = ;
清理工具在正常执行时首先调用Initialize初始化插件随后执行GetSpaceUsed来扫描可清除的文件大小扫描完毕后清理工具的主界面就出现了如图所示在此我们加入了清理TMP文件的功能可以浏览不同的清理文件种类列表中的每一个文件种类均由一个COM插件实现除了阅览可清理文件大小以外用户在可以点击一个可自定义的按钮调用插件的ShowProperties功能以显示更详细的资料如用户选择OK清理工具就调用Purge函数清理扫描出来的文件最后Deactivate函数被调用终止插件的应用
运用于Windows 以后的清理工具的插件也应该支持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) = ;
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表示可定期执行的无人控制模式在此模式下GetSpaceUsedPurge及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从IEmptyVolumeCache继承并且我们添加了下列变量
// 储存扫描出文件的大小
DWORDLONG m_dwlFileSize;
// 储存根目录
WCHAR m_strRootDir[MAX_PATH];
// 储存扫描出文件列表
std::vector<WCHAR *> m_lstFilesToDel;
然后我们一一实现IEmptyVolumeCache及IEmptyVolumeCache接口的函数在下面的代码列表中没有包括严格的检查错误返回值这是为了简短代码的长度提高可读性在实际应用中检查错误返回值是不可少的为了不同版本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
ppwszDescriptionpdwFlags);
*ppwszBtnText = (LPWSTR) CoTaskMemAlloc ( * sizeof (WCHAR));
StrCpyW(*ppwszBtnText LView files);
return hr;
}
HRESULT CCleanSimpleHandler::Initialize (HKEY hKey LPCWSTR pcwszVolume
LPWSTR *ppwszDisplayName LPWSTR *ppwszDescription DWORD *pdwFlags)
{
StrCpyW(m_strRootDir pcwszVolume);
*ppwszDisplayName = (LPWSTR) CoTaskMemAlloc( * sizeof (WCHAR));
StrCpyW(*ppwszDisplayName L*TMP files);
*ppwszDescription = (LPWSTR) CoTaskMemAlloc ( * sizeof (WCHAR));
StrCpyW(*ppwszDescription LTemporary files *TMP);
*pdwFlags = EVCF_HASSETTINGS | EVCF_ENABLEBYDEFAULT;
m_dwlFileSize = ;
return S_OK;
}
在GetSpaceUsed中我们调用ScanDir来扫描*TMP文件储存在m_lstFilesToDel中GetSpaceUsed的第二个参数是IEmptyVolumeCacheCallBack接口的指针用于调用其ScanProgress函数以报告扫描的进展情况ScanProgress函数定义是
HRESULT ScanProgress（DWORDLONG dwlSpaceUsed DWORD dwFlags LPCWSTR pwszReserved）
其中dwFlags正常应设为零在结束时改为EVCCBF_LASTNOTIFICATIONScanProgress函数的返回值很重要因为用户可以在任何时候中断在进行中的清理任务如ScanProgress返回E_ABORTGetSpaceUsed应最快终端扫描函数返回因此我们在递归的目录扫描函数ScanDir中加入了如中断立即退出的功能
HRESULT CCleanSimpleHandler::GetSpaceUsed (DWORDLONG *pdwSpaceUsed
IEmptyVolumeCacheCallBack *picb)
{
m_dwlFileSize = ;
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;
WIN_FIND_DATAW fd;
HANDLE hFind;
if (cancelled = FAILED(pcib>ScanProgress(m_dwlFileSize NULL NULL)))
return false;
StrCpyW(strPathszDir);
PathAppendW(strPath L*);
pchPathFileName = strPath+lstrlenW(strPath);
hFind = FindFirstFileW(strPath &fd);
if (hFind == INVALID_HANDLE_VALUE) // Eg Due to security issues
return true;
do {
StrCpyW(pchPathFileName fdcFileName);
if ((fddwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
if (fdcFileName[] != ) {
if (cancelled = !ScanDir(strPath pcib)) break;
}
} else {
WCHAR* pchExt = PathFindExtensionW(strPath);
if ( StrCmpIW(pchExt Ltmp) == ) {
m_dwlFileSize += ((DWORDLONG)fdnFileSizeHigh)*+
fdnFileSizeLow;
WCHAR* filename = (WCHAR *)CoTaskMemAlloc((lstrlenW(strPath)+)*
sizeof(WCHAR));
StrCpyW(filename strPath);
m_lstFilesToDelpush_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=; i < m_lstFilesToDelsize(); ++i)
DeleteFileW(m_lstFilesToDel[i]);
return S_OK;
}
HRESULT CCleanSimpleHandler::ShowProperties (HWND hWnd)
{
for (unsigned int i=; i < m_lstFilesToDelsize(); ++i)
if (MessageBoxW(hWnd m_lstFilesToDel[i] LView files
MB_OKCANCEL|MB_ICONINFORMATION)==IDCANCEL) break;
return S_OK;
}
HRESULT CCleanSimpleHandler::Deactivate (LPDWORD pdwFlags)
{
for (unsigned int i=; i < m_lstFilesToDelsize(); ++i)
CoTaskMemFree(m_lstFilesToDel[i]);
m_lstFilesToDelclear();
*pdwFlags = ;
return S_OK;
结论和建议
通过实例分解我们对Windows磁盘清理工具的基于COM技术的开发接口做了深入地研究Windows外壳中有较多的开发接口本文介绍的开发思想也可以运用在其它扩展插件中
