问题:
将exe程序设置为开机自启动,启动后最小化至托盘。
思路:
1、程序开机自启动:修改注册表RegOpenKeyEx
2、程序最小化至托盘:Shell_NotifyIcon创建托盘对象
3、开机自启动和双击打开的区分:增加启动命令参数,使用GetCommandLine获取命令内容
解决:
1、设置程序开机自启动,修改注册表信息。
bool ModifyRegedit(bool bAutoRun)
{
charpFileName[MAX_PATH] = { 0 };
DWORD dwRet = GetModuleFileName(NULL, (LPWSTR)pFileName, MAX_PATH);
HKEY hKey;
LPCTSTR lpRun = _T("Software\\Microsoft\\Windows\\CurrentVersion\\Run");
long lRet = RegOpenKeyEx(HKEY_LOCAL_MACHINE, lpRun, 0, KEY_WRITE, &hKey);
if(lRet != ERROR_SUCCESS)
return false;
if (bAutoRun)
RegSetValueEx(hKey, _T("AutoRun"), 0, REG_SZ, (const BYTE*)(LPCSTR)pFileName, MAX_PATH);
else
RegDeleteValueA(hKey, "AutoRun");
RegCloseKey(hKey);
return true;
}
2、 程序最小化至托盘。
//.h
NOTIFYICONDATA m_trayIcon; //!< 定义托盘图标对象
//资源中添加程序托盘图标bmp文件 IDR_TRAYICON
//.cpp
void CreateTrayIcon() {
CreateSemaphore(NULL, 1, 1, _T("TestExe"));
if (GetLastError() == ERROR_ALREADY_EXISTS)
{
ShowWindow(SW_HIDE);
return;
}
memset(&m_trayIcon, 0, sizeof(NOTIFYICONDATA));
m_trayIcon.cbSize = sizeof(NOTIFYICONDATA);
m_trayIcon.hIcon = ::LoadIcon(GetModuleHandle(NULL), MAKEINTRESOURCE(IDR_TRAYICON));
m_trayIcon.hWnd = m_hWnd;
lstrcpy(m_trayIcon.szTip, _T("自定义程序"));
m_trayIcon.uCallbackMessage = WM_SHOWTASK;
m_trayIcon.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
Shell_NotifyIcon(NIM_ADD, &m_trayIcon);
ShowWindow(SW_HIDE);
}
3 、程序开机自启动和正常启动的区分
为了区分程序双击启动和开机自启动,在设置开机自启动修改注册表的时候增加命令参数,参数可以自定义,例:-a -s -o。
然后对开机自启动进行相应的修改:
void ModifyRegedit(bool bAutoRun)
{
//...
CString strFileName;
strFileName = pFileName;
strFileName = _T("\"") + strFileName + _T("\"") + _T(" -s");
//...
RegSetValueEx(hKey, _T("AutoRun"), 0, REG_SZ, (const BYTE*)strFileName.GetString(), MAX_PATH);
//...
}
在程序初始化函数中进行相应的区分判断,MFC程序的话在OnInitDialog中进行设置,DUILib程序在InitWindow中或者处理消息DUI_MSGTYPE_WINDOWINIT的时候进行设置。
CString strTest = GetCommandLine();
if (strTest.Find(_T("-s")) > 0)
CreateTrayIcon();
OK,搞定!