turnpoweroffdlg.cpp

本程序提供一种编写定时关机的一个很好例子。

// TurnPowerOffDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "TurnPowerOff.h"
#include "TurnPowerOffDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

#define  MYWM_NOTIFYICON  WM_USER+1
#define  MAX_LENGTH		200

// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// CTurnPowerOffDlg 对话框




CTurnPowerOffDlg::CTurnPowerOffDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CTurnPowerOffDlg::IDD, pParent)
	, timeOff(0)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	timeOff=m_Time1 = m_Time2 =CTime::GetCurrentTime();
	bAppend=false;

}

void CTurnPowerOffDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_DateTimeCtrl(pDX, IDC_DATETIMEPICKER1, m_Time1);
	DDX_DateTimeCtrl(pDX, IDC_DATETIMEPICKER2, m_Time2);
	DDX_Control(pDX, IDC_STATIC_INFO, m_cInfo);
}

BEGIN_MESSAGE_MAP(CTurnPowerOffDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_SAVEBUTTON, &CTurnPowerOffDlg::OnBnClickedSavebutton)
	ON_WM_TIMER()
	ON_BN_CLICKED(IDC_STARTBUTTON, &CTurnPowerOffDlg::OnBnClickedStartbutton)
	ON_BN_CLICKED(ID_EXIT, &CTurnPowerOffDlg::OnBnClickedExit)
END_MESSAGE_MAP()


// CTurnPowerOffDlg 消息处理程序

BOOL CTurnPowerOffDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 将“关于...”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码
    m_tnid.cbSize=sizeof(NOTIFYICONDATA);        //定义结构的成员变量
    m_tnid.hWnd=this->m_hWnd;
    m_tnid.uFlags=NIF_MESSAGE|NIF_ICON|NIF_TIP;
    m_tnid.uCallbackMessage=MYWM_NOTIFYICON;     //用户定义的回调消息
    _tcscpy(m_tnid.szTip, _T("定时关机程序"));
	m_tnid.uID=IDR_MAINFRAME;
//	hIcon=AfxGetApp()->LoadIcon(m_hIcon);        //图标化的图标
	hIcon=m_hIcon;
	m_tnid.hIcon=hIcon;
	if(hIcon)::DestroyIcon(hIcon);

	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

void CTurnPowerOffDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CTurnPowerOffDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
//
HCURSOR CTurnPowerOffDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}


void CTurnPowerOffDlg::OnBnClickedSavebutton()
{
	// TODO: 在此添加控件通知处理程序代码
	UpdateData(true);
	CString str1,str2,str3;
	str1=m_Time1.Format("%Y-%m-%d");
	str2=m_Time2.Format("%H:%M:%S");
	str3=str1+_T(" ")+str2;
	CString strPath=_T(".\\powertime.ini");
	WritePrivateProfileString(_T("time"), _T("time"), str3, strPath);
	   
	m_Time1=m_Time2=GetOffTime();
	UpdateData(false);
}

CTime CTurnPowerOffDlg::GetOffTime(void)
{
	CString strPath=_T(".\\powertime.ini");
	CString strSqlInfo;

	GetPrivateProfileString(_T("time"), _T("time"), _T("time"), strSqlInfo.GetBuffer(MAX_LENGTH), MAX_LENGTH, strPath);
	CString strTime=strSqlInfo;

	if(strTime=="" || strTime=="time") 
		return 0;

	return StrToTime(strTime);
}

void CTurnPowerOffDlg::PowerOff(void)
{
	OnCancel(); 
	TOKEN_PRIVILEGES tp;
	HANDLE hToken; 
	LUID luid; 
	LPTSTR MachineName=NULL; 

	if(!OpenProcessToken(GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES,&hToken ))
	{
		// PERR("OpenProcessToken",GetLastError());
		return ;
	}

	if(!LookupPrivilegeValue(MachineName, SE_SHUTDOWN_NAME, &luid))
	{
		//  PERR("LookupPrivilegeValue", GetLastError());
		return ; 
	}

	tp.PrivilegeCount = 1; 
	tp.Privileges[0].Luid = luid; 
	tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED; 
	AdjustTokenPrivileges(hToken,FALSE,&tp,sizeof(TOKEN_PRIVILEGES),NULL, NULL ); //到这里,是取得权限///
	ExitWindowsEx(EWX_POWEROFF,EWX_FORCE);	
}

CTime CTurnPowerOffDlg::StrToTime(CString str)//函数功能：STRTOTIME
{
	// 时间串格式 "%Y-%m-%d %H:%M:%S",如"1999-01-01 11:11:11"
	int nYear,nMonth,nDay,nHour,nMinute,nSecond;
str="2008-05-07 01:07:00";
	nYear=_ttoi(str.Left(4));		// 得到年
	nMonth=_ttoi(str.Mid(5,2));		// 得到月
	nDay=_ttoi(str.Mid(8,2));		// 得到日
	nHour=_ttoi(str.Mid(11,2));		// 得到时
	nMinute=_ttoi(str.Mid(14,2));	// 得到分
	nSecond=_ttoi(str.Mid(17,2));	// 得到秒


	// 构造CTime变量
	CTime result(nYear,nMonth,nDay,nHour,nMinute,nSecond);

	return result;
}
void CTurnPowerOffDlg::OnTimer(UINT_PTR nIDEvent)
{
	// TODO: 在此添加消息处理程序代码和/或调用默认值
	if(nIDEvent)
	{
		CTime CurTime=CTime::GetCurrentTime();
		CTimeSpan tpUse=timeOff-CurTime;
		long iMins=tpUse.GetTotalSeconds();
		m_cInfo.SetWindowText(_T("离关机时间还有:")+tpUse.Format("%H时%M分%S秒"));

		if(iMins<-30)
		{
			m_cInfo.SetWindowTextW(_T("无效时间"));
			return;
		}

		if(timeOff<=0) 
		{  
			m_cInfo.SetWindowTextW(_T("无效时间"));
			return;
		}

		if(CurTime>timeOff)
		{
			PowerOff();
		}
	}

	CDialog::OnTimer(nIDEvent);
}

void CTurnPowerOffDlg::OnBnClickedStartbutton()
{
	// TODO: 在此添加控件通知处理程序代码
	timeOff=GetOffTime(); 
	bAppend=!bAppend;

	if(bAppend)
	{
		SetDlgItemText(IDC_STARTBUTTON,_T("停止计时器"));
		SetTimer(1,1000,NULL);
	}
	else
	{
		SetDlgItemText(IDC_STARTBUTTON,_T("启动计时器"));
		KillTimer(1);
	}

	OnTimer(1); 
}

void CTurnPowerOffDlg::OnBnClickedExi()
{
	// TODO: 在此添加控件通知处理程序代码
}

void CTurnPowerOffDlg::OnBnClickedExit()
{
	// TODO: 在此添加控件通知处理程序代码
	OnOK();
}

LRESULT CTurnPowerOffDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam) 
{ 
	if((message==WM_SYSCOMMAND)&&(wParam==SC_MINIMIZE))         //如果是系统消息
	{
		this->ShowWindow(SW_HIDE);
		Shell_NotifyIcon(NIM_ADD,&m_tnid);

		return true;
	}
	else if((message==MYWM_NOTIFYICON)&&(lParam==WM_LBUTTONDBLCLK))  //如果是用户定义的消息
	{
		this->ShowWindow(SW_SHOW);
		this->SetActiveWindow();   //双击并且出现在最上面
		Shell_NotifyIcon(NIM_DELETE,&m_tnid);

		return true;
	}
	else if((message==MYWM_NOTIFYICON) && (lParam==WM_RBUTTONDOWN))
	{
		UINT uID;//发出该消息的图标的ID

		UINT uMouseMsg;//鼠标动作

		POINT pt;

		uID=(UINT) wParam;

		uMouseMsg=(UINT) lParam;

		CMenu dummyMenu;

		if(!dummyMenu.LoadMenu(IDR_POPUP))
			AfxThrowResourceException();

		CMenu *pPopupMenu=dummyMenu.GetSubMenu(0);
		ASSERT(pPopupMenu!=NULL);

		switch(uID)
		{
		case IDR_MAINFRAME:		//如果是我们的图标
			GetCursorPos(&pt);	//取得鼠标位置
			pt.y-=90;
			//		ClientToScreen(&pt);
			pPopupMenu->TrackPopupMenu(TPM_CENTERALIGN|TPM_LEFTBUTTON,pt.x,pt.y,AfxGetMainWnd());
			break;
		default:
			break;
		}
	}

	return CDialog::WindowProc(message, wParam, lParam);
}

 CTurnPowerOffDlg::~CTurnPowerOffDlg(void) 
{
	// TODO: Add your command handler code here

	m_tnid.cbSize=sizeof(NOTIFYICONDATA);

	m_tnid.hWnd=this->m_hWnd;

	m_tnid.uID=IDR_MAINFRAME;//保证删除的是我们的图标

	Shell_NotifyIcon(NIM_DELETE,&m_tnid);
}