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();