Windows下的开始菜单程序组和桌面快捷方式对普通用户来讲可以说是整个系统的入口,因此一般的安装程序在运行时都会创建这两种链接。那么我们普通的绿色程序(我说的是那种没有依靠打包向导生成的程序)有没有机会创建呢,当然是有的。
在VC下创建这两类链接有几个要点,第一是获取链接所在目录,比如桌面的目录通常是:C:/Documents and Settings/UserName/桌面。 程序组的目录通常是:C:/Documents and Settings/UserName/「开始」菜单/程序。
获取的方法是访问特定的注册表项。
第二个要点是使用COM接口写入lnk文件到上述目录,原理并不深奥,请直接看下面代码
bool CreateProgramGroup(const char * szGroupName)
{
CRegKey cKey;
if(ERROR_SUCCESS != cKey.Open(HKEY_CURRENT_USER,
"Software//MicroSoft//Windows//CurrentVersion//Explorer//Shell Folders"))
{
return false;
}
char szRootDir[MAX_PATH];
DWORD dwSize = MAX_PATH;
if(ERROR_SUCCESS != cKey.QueryValue(szRootDir, "Programs", &dwSize))
{
cKey.Close();
return false;
}
cKey.Close();
char szFullDir[MAX_PATH];
sprintf(szFullDir, "%s//%s", szRootDir, szGroupName);
CreateDirectory(szFullDir, NULL);
return true;
}
bool CreateLink(const char * szLinkName, const char * szProgPath,
bool bDesktop, const char * szGroupName /* = 0 */)
{
CComPtr<IShellLink> spIShellLink;
HRESULT hr = ::CoCreateInstance(CLSID_ShellLink, NULL, 1,
IID_IShellLink, (LPVOID *)&spIShellLink);
if(FAILED(hr))
{
return false;
}
spIShellLink->SetPath(szProgPath);
spIShellLink->SetShowCmd(SW_SHOWNORMAL);
//获取程序所在目录,设置为工作目录
char * szSep = strrchr(szProgPath, '//');
if(!szSep)
return false;
//获取程序目录
char szProgFolder[MAX_PATH];
memset(szProgFolder, 0, MAX_PATH);
strncpy(szProgFolder, szProgPath, szSep - szProgPath + 1);
spIShellLink->SetWorkingDirectory(szProgFolder);
CComPtr<IPersistFile> spIPersistFile;
hr = spIShellLink->QueryInterface(IID_IPersistFile, (void **)&spIPersistFile);
if(FAILED(hr))
{
return false;
}
CRegKey cKey;
if(ERROR_SUCCESS != cKey.Open(HKEY_CURRENT_USER,
"Software//MicroSoft//Windows//CurrentVersion//Explorer//Shell Folders"))
{
return false;
}
char szRootDir[MAX_PATH];
DWORD dwSize = MAX_PATH;
if(ERROR_SUCCESS != cKey.QueryValue(szRootDir, bDesktop ? "Desktop" : "Programs", &dwSize))
{
cKey.Close();
return false;
}
cKey.Close();
char szFullDir[MAX_PATH];
if(szGroupName)
sprintf(szFullDir, "%s//%s//%s.lnk", szRootDir, szGroupName, szLinkName);
else
sprintf(szFullDir, "%s//%s.lnk", szRootDir, szLinkName);
WCHAR wszFullDir[MAX_PATH];
::MultiByteToWideChar(0, 0, szFullDir, -1, wszFullDir, MAX_PATH);
hr = spIPersistFile->Save(wszFullDir, FALSE);
if(FAILED(hr))
{
return false;
}
::SHChangeNotify(SHCNE_ALLEVENTS, SHCNF_PATH|SHCNF_FLUSH, szFullDir, 0);
return true;
}
使用示例:
CoInitialize(NULL);
char szDemoExeDir[MAX_PATH];
sprintf(szDemoExeDir, "c://program files//demo//demo.exe");
char szUnInstallDir[MAX_PATH];
sprintf(szUnInstallDir, "c://program files//demo//uninstall.exe");
CreateLink("Demo", szDemoExeDir, TRUE);
CreateProgramGroup("Demo程序组");
CreateLink("Demo", szDemoExeDir, FALSE, "Demo程序组");
CreateLink("卸载Demo", szUnInstallDir, FALSE, "Demo程序组");
CoUninitialize();