一个定时关机的小程序

一：需求描述
最近经常在需要让计算机在特定时间关闭或者重启，所以写了这个程序。当然满足这一需求的方法很多，比如用计划任务等等，但是既然在学VC，就自己动手做一个吧。
二、解题思路
定时关机无非是设定一个关机时间，然后跟当前系统时间比较，取得这个差，然后设置一个定时器，到时间就执行相应的操作，比如shutdown、reboot、logoff等等。
三、详细过程
程序最终的画面如下：



第一步：利用Wizard生成一个对话框框架。然后按上图所示放置控件。其中“倒计时”的ID设为IDC_STATIC_NAME,“00:00:00”的ID设为IDC_STATIC_TIME（这两个在程序里要用）,另外为日期、时间、ComboBox添加控件变量分别为m_Data1,m_Data2,m_ComboBox.ComboBox的样式为Drop list，即不接收输入。好了，下面就添加代码。
第二步：添加代码：
我要在按下“设定”按钮时，，把“设定”改为“重新设定”，让能设定的选项均不能再修改，除非再按一次这个按钮。所以要用一个bool型的变量记录按钮的状态，所以在对话框的头文件添加变量IsChanged,类型为bool.另外再添加3个变量m_Hour,m_Min,m_Sec,类型为int，一个long型的m_remain。好了，变量先添加这些，接下来要重载OnOK、OnCancel函数，直接用向导添加就行了；为按钮添加默认的点击处理函数；添加最小化到系统托盘的函数toTray()，消息映射 afx_msg LRESULT onShowTask(WPARAM wParam,LPARAM lParam);以及afx_msg void OnTimer(UINT_PTR nIDEvent);记得还要在对话框cpp文件中添加相应的内容，此处不说了。在对话框cpp中添加#define WM_SHOWTASK WM_USER+1为自定义的消息。好了，现在就为函数写代码吧。
首先是在OnInitDialog()中添加
m_ComboBox.InsertString(0,_T("注销"));
m_ComboBox.InsertString(1,_T("关闭计算机"));
m_ComboBox.InsertString(2,_T("重新启动"));
m_ComboBox.SetCurSel(1);
// TODO: 在此添加额外的初始化代码
IsChanged =true;
然后是OnOK函数中添加一行代码OnBnClickedButton1();
OnCancel函数中添加如下代码：
NOTIFYICONDATA nid1;
nid1.hWnd = this->m_hWnd;
nid1.uID = IDR_MAINFRAME;
Shell_NotifyIcon(NIM_DELETE,&nid1);
CDialog::OnCancel();
目的是把系统托盘中的图标移走。
OnTimer函数中添加如下代码
m_Hour = m_remain/3600;
m_Min = (m_remain%3600)/60;
m_Sec = (m_remain%3600)%60;
CString m_time;
if(m_Hour>=10)
m_time.Format("%d:",m_Hour);
else
m_time.Format("0%d:",m_Hour);
if(m_Min>=10)
m_time.Format(m_time+"%d:",m_Min);
else
m_time.Format(m_time+"0%d:",m_Min);
if(m_Sec>=10)
m_time.Format(m_time+"%d",m_Sec);
else
m_time.Format(m_time+"0%d",m_Sec);
SetDlgItemText(IDC_STATIC_TIME,m_time);
if(m_remain<=60)
{
if(!IsTop)//只设定一次
{
CWnd::SetWindowPos(&wndTopMost,0,0,0,0,SWP_NOREDRAW|SWP_NOSIZE);
IsTop = true;
}
MessageBeep(-1);
}
if(m_remain==0)
{
//shutdowncomputer adjust privalige
HANDLE hToken;
TOKEN_PRIVILEGES tkp;
if(!OpenProcessToken(::GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY,&hToken))
return;
LookupPrivilegeValue(NULL,SE_SHUTDOWN_NAME,&tkp.Privileges[0].Luid);
tkp.PrivilegeCount = 1;
tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
AdjustTokenPrivileges(hToken, false, &tkp, 0, (PTOKEN_PRIVILEGES)NULL, 0);
if(GetLastError()!=ERROR_SUCCESS)
return;
switch(m_ComboBox.GetCurSel())
{
case 0://logoff
ExitWindowsEx(EWX_LOGOFF|EWX_FORCE,0);
break;
case 1://shutdown
ExitWindowsEx(EWX_SHUTDOWN | EWX_POWEROFF | EWX_FORCE, 0);
break;
default://reboot
ExitWindowsEx(EWX_REBOOT | EWX_FORCE,0);
}
PostMessage(WM_QUIT);
}
m_remain -=1;
实现了两点：1是显示剩余时间，2是时间到达后执行操作。
OnBnClickedButton1中添加如下代码
if(IsChanged)//设定关机
{
//取得设定时间
SYSTEMTIME m_data,m_time,m_systime;
__int64 m_Start,m_End;
FILETIME m_filetime,m_sysfile;
PULARGE_INTEGER m_inttime,m_sysint;
GetSystemTime(&m_systime);
m_systime.wHour +=8;
m_Data1.GetTime(&m_data);
m_Data2.GetTime(&m_time);
m_data.wHour = m_time.wHour;
m_data.wMinute = m_time.wMinute;
m_data.wSecond = m_time.wSecond;
m_data.wMilliseconds = m_time.wMilliseconds;
SystemTimeToFileTime(&m_data,&m_filetime);
SystemTimeToFileTime(&m_systime,&m_sysfile);
m_inttime = (PULARGE_INTEGER)&m_filetime;
m_sysint = (PULARGE_INTEGER)&m_sysfile;
m_Start = *(__int64 *)m_sysint;
m_End = *(__int64 *)m_inttime;
if(m_Start>=m_End)
{
MessageBox(_T("不能小于当前时间！"),_T("时间设定出错"),MB_ICONERROR|MB_OK);
return;
}
m_remain = (long)((m_End-m_Start)/10000/1000);
CWnd::SetTimer(1,1000,0);
SetDlgItemText(IDC_BUTTON1,_T("重新设定"));
m_Data1.EnableWindow(false);
m_Data2.EnableWindow(false);
m_ComboBox.EnableWindow(false);
IsChanged = false;
switch(m_ComboBox.GetCurSel())
{
case 0:
SetDlgItemText(IDC_STATIC_NAME,_T("注销倒计时："));
break;
case 1:
SetDlgItemText(IDC_STATIC_NAME,_T("关机倒计时："));
break;
default:
SetDlgItemText(IDC_STATIC_NAME,_T("重启倒计时："));
}
}
else//重新设定
{
IsChanged = true;
m_Data1.EnableWindow();
m_Data2.EnableWindow();
m_ComboBox.EnableWindow();
SetDlgItemText(IDC_BUTTON1,_T("设定"));
// SYSTEMTIME m_time;
// GetSystemTime(&m_time);
m_Data1.SetTime();
m_Data2.SetTime();
CWnd::KillTimer(1);
}

toTray中添加如下代码
NOTIFYICONDATA nid;
nid.cbSize = (DWORD)sizeof(NOTIFYICONDATA);
nid.hWnd = this->m_hWnd;
nid.uID = IDR_MAINFRAME;
nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
nid.uCallbackMessage = WM_SHOWTASK;
nid.hIcon = LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDR_MAINFRAME));
strcpy(nid.szTip,"定时关机");
Shell_NotifyIcon(NIM_ADD,&nid);
ShowWindow(SW_HIDE);
主要是实现了最小化到系统托盘

LRESULT CautoshutdownDlg::onShowTask(WPARAM wParam,LPARAM lParam)
{
if(wParam != IDR_MAINFRAME)
return -1;
switch(lParam)
{
case WM_RBUTTONUP:
{
LPPOINT lpoint = new tagPOINT;
::GetCursorPos(lpoint);
CMenu menu;
menu.CreatePopupMenu();
menu.AppendMenu(MF_STRING,WM_DESTROY,"关闭");
menu.TrackPopupMenu(TPM_LEFTALIGN,lpoint->x,lpoint->y,this);
HMENU hmenu=menu.Detach();
menu.DestroyMenu();
delete lpoint;
}
break;
case WM_LBUTTONUP:
{
this->ShowWindow(SW_SHOW);
}
break;
}
return 0;
}
这个就是相应的处理函数。
最后别忘了在OnSysCommand中添加 if(nID == SC_MINIMIZE)
{
toTray();

}
好了，工作到次结束了。
需要说明的几点是：取得系统时间后要把它转化成FILETIME，然后化成ULARGEINTERG，最后转化成__int64之后才比较，然后/10000/1000换成秒。