开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > Shell编程 > 查看源码
AutoCloseDlg.cpp
资源名称：autoclose　[点击查看]
上传用户：zhuoli121
上传日期：2007-01-13
资源大小：43k
文件大小：11k
源码类别：

Shell编程

开发平台：

WINDOWS

AutoCloseDlg.cpp：源码内容
  1. // AutoCloseDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "AutoClose.h"
  6. #include "AutoCloseDlg.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CAboutDlg dialog used for App About
  17. class CAboutDlg : public CDialog
  18. {
  19. public:
  20. CAboutDlg();
  22. // Dialog Data
  23. //{{AFX_DATA(CAboutDlg)
  24. enum { IDD = IDD_ABOUTBOX };
  25. //}}AFX_DATA
  27. // ClassWizard generated virtual function overrides
  28. //{{AFX_VIRTUAL(CAboutDlg)
  29. protected:
  30. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  31. //}}AFX_VIRTUAL
  33. // Implementation
  34. protected:
  35. //{{AFX_MSG(CAboutDlg)
  36. //}}AFX_MSG
  37. DECLARE_MESSAGE_MAP()
  38. };
  40. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  41. {
  42. //{{AFX_DATA_INIT(CAboutDlg)
  43. //}}AFX_DATA_INIT
  44. }
  46. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  47. {
  48. CDialog::DoDataExchange(pDX);
  49. //{{AFX_DATA_MAP(CAboutDlg)
  50. //}}AFX_DATA_MAP
  51. }
  53. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  54. //{{AFX_MSG_MAP(CAboutDlg)
  55. // No message handlers
  56. //}}AFX_MSG_MAP
  57. END_MESSAGE_MAP()
  59. /////////////////////////////////////////////////////////////////////////////
  60. // CAutoCloseDlg dialog
  62. CAutoCloseDlg::CAutoCloseDlg(CWnd* pParent /*=NULL*/)
  63. : CDialog(CAutoCloseDlg::IDD, pParent)
  64. {
  65. //{{AFX_DATA_INIT(CAutoCloseDlg)
  66. // NOTE: the ClassWizard will add member initialization here
  67. //}}AFX_DATA_INIT
  68. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  69. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  70. }
  72. void CAutoCloseDlg::DoDataExchange(CDataExchange* pDX)
  73. {
  74. CDialog::DoDataExchange(pDX);
  75. //{{AFX_DATA_MAP(CAutoCloseDlg)
  77. //}}AFX_DATA_MAP
  78. }
  80. BEGIN_MESSAGE_MAP(CAutoCloseDlg, CDialog)
  81. //{{AFX_MSG_MAP(CAutoCloseDlg)
  82. ON_WM_SYSCOMMAND()
  83. ON_WM_PAINT()
  84. ON_WM_QUERYDRAGICON()
  85. ON_WM_TIMER()
  86. ON_CBN_SELCHANGE(IDC_HOUR, OnSelchangeHour)
  87. ON_CBN_SELCHANGE(IDC_MIN, OnSelchangeMin)
  88. ON_BN_CLICKED(IDC_BUTTON2, OnStart)
  89. ON_WM_DESTROY()
  90. ON_COMMAND(ID_ABOUT, OnAbout)
  91. ON_COMMAND(ID_STOP, OnStop)
  92. ON_COMMAND(ID_DISPLAY, OnDisplay)
  93. ON_COMMAND(ID_QUIT, OnQuit)
  94. ON_WM_LBUTTONDOWN()
  95. //}}AFX_MSG_MAP
  96. END_MESSAGE_MAP()
  98. /////////////////////////////////////////////////////////////////////////////
  99. // CAutoCloseDlg message handlers
  101. BOOL CAutoCloseDlg::OnInitDialog()
  102. {
  103.  CDialog::OnInitDialog();
  105. // Add "About..." menu item to system menu.
  107. // IDM_ABOUTBOX must be in the system command range.
  108. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  109. ASSERT(IDM_ABOUTBOX < 0xF000);
  111. CMenu* pSysMenu = GetSystemMenu(FALSE);
  112. if (pSysMenu != NULL)
  113. {
  114. CString strAboutMenu;
  115. strAboutMenu.LoadString(IDS_ABOUTBOX);
  116. if (!strAboutMenu.IsEmpty())
  117. {
  118. pSysMenu->AppendMenu(MF_SEPARATOR);
  119. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  120. }
  121. }
  123. // Set the icon for this dialog.  The framework does this automatically
  124. //  when the application's main window is not a dialog
  125. SetIcon(m_hIcon, TRUE); // Set big icon
  126. SetIcon(m_hIcon, FALSE); // Set small icon
  128. // TODO: Add extra initialization here
  129. SetWindowPos(&CWnd::wndTopMost,0,0,0,0,SWP_NOMOVE|SWP_NOSIZE);
  130. CenterWindow(GetDesktopWindow());    //让窗口处于屏幕中间
  131. SetWindowText("MFC自动关机程序!");
  132.     SetTimer(10,100,NULL);              //调用OnTimer函数，并使之没隔一秒发送一次 WM_TIMER 消息
  134.     CComboBox *hour;
  135. CComboBox *min;
  136. hour=(CComboBox*)GetDlgItem(IDC_HOUR);
  137. min=(CComboBox*)GetDlgItem(IDC_MIN);
  139. ////////////添加24小时///////////////////
  140. for(int i=0;i<=23;i++)         
  141. {
  142. if(i<10)
  143. {
  144.     CString s;
  145.     s.Format("0%d",i);
  146. hour->AddString(s);
  147. }
  148. else
  149. {
  150. CString ss;
  151. ss.Format("%d",i);
  152.     hour->AddString(ss);
  153. }
  154. }
  155.     /////////////添加60分钟////////////////
  156. for(i=0;i<=59;i++)
  157. {
  158. if(i<10)
  159. {
  160.     CString s;
  161.     s.Format("0%d",i);
  162. min->AddString(s);
  163. }
  164. else
  165. {
  166. CString ss;
  167. ss.Format("%d",i);
  168.     min->AddString(ss);
  169. }
  170. }
  171. return TRUE;  // return TRUE  unless you set the focus to a control
  172. }
  174. void CAutoCloseDlg::OnSysCommand(UINT nID, LPARAM lParam)
  175. {
  176. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  177. {
  178. CAboutDlg dlgAbout;
  179. dlgAbout.DoModal();
  180. }
  181. else
  182. {
  183. CDialog::OnSysCommand(nID, lParam);
  184. }
  185. }
  187. // If you add a minimize button to your dialog, you will need the code below
  188. //  to draw the icon.  For MFC applications using the document/view model,
  189. //  this is automatically done for you by the framework.
  191. void CAutoCloseDlg::OnPaint() 
  192. {
  193. if (IsIconic())
  194. {
  195. CPaintDC dc(this); // device context for painting
  197. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  199. // Center icon in client rectangle
  200. int cxIcon = GetSystemMetrics(SM_CXICON);
  201. int cyIcon = GetSystemMetrics(SM_CYICON);
  202. CRect rect;
  203. GetClientRect(&rect);
  204. int x = (rect.Width() - cxIcon + 1) / 2;
  205. int y = (rect.Height() - cyIcon + 1) / 2;
  207. // Draw the icon
  208. dc.DrawIcon(x, y, m_hIcon);
  209. }
  210. else
  211. {
  212. CDialog::OnPaint();
  213. }
  214. }
  216. // The system calls this to obtain the cursor to display while the user drags
  217. //  the minimized window.
  218. HCURSOR CAutoCloseDlg::OnQueryDragIcon()
  219. {
  220. return (HCURSOR) m_hIcon;
  221. }
  223. void CAutoCloseDlg::OnTimer(UINT nIDEvent) 
  224. {
  225. CTime t=CTime::GetCurrentTime();   //取出当前时间
  226.     int y=t.GetYear();
  227. int d=t.GetDay();
  228. int m=t.GetMonth();
  229. int h=t.GetHour();
  230. int min=t.GetMinute();
  231. int s=t.GetSecond();
  233. CString yy,dd,mm,hh,mins,ss,total;
  234. yy.Format("%d",y);
  235. dd.Format("%d",d);
  236. mm.Format("%d",m);
  237. if(h<10)
  238.     hh.Format("0%d",h);
  239. else
  240. hh.Format("%d",h);
  241. if(min<10)
  242.         mins.Format("0%d",min);
  243. else
  244.     mins.Format("%d",min);
  245. if(s<10)
  246.     ss.Format("0%d",s);
  247. else
  248. ss.Format("%d",s);
  250.     total=yy+"-"+mm+"-"+dd+"  "+hh+":"+mins+":"+ss;
  252. CStatic *displaytime;
  253. displaytime=(CStatic*)GetDlgItem(IDC_STATIC);
  254. displaytime->SetWindowText(total);                     //在静态文本框中显示出当前时间
  256. /////////////////准备关机//////////////////////
  257. if(begin && close_hour.CompareNoCase(hh)==0 && close_min.CompareNoCase(mins)==0)
  258. {
  259. KillTimer(10);
  260. /////////首先声明3个全局变量///////////BOOL fResult;TOKEN_PRIVILEGES tkp;HANDLE hToken;
  261.         if (!OpenProcessToken(GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken)) 
  262. {
  263.     MessageBox("OpenProcessToken failed!");
  264. }
  266.     LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME,&tkp.Privileges[0].Luid); //获得本地机唯一的标识
  267.     tkp.PrivilegeCount = 1;  
  268.     tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED; 
  269.         AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,(PTOKEN_PRIVILEGES) NULL, 0); //调整获得的权限
  270.     
  271.     if (GetLastError() != ERROR_SUCCESS) 
  272. {
  273.             MessageBox("AdjustTokenPrivileges enable failed!");
  274. }
  276.     fResult =InitiateSystemShutdown( 
  277.              NULL,                                  // 要关的计算机用户名
  278.              "由于系统不稳定，WINDOWS将在上面的时间内关机，请做好保存工作!",  // 显示的消息
  279.              10,                                    // 关机所需的时间
  280.              TRUE,                                 // ask user to close apps 
  281.              TRUE);                               //设为TRUE为重起，设为FALSE为关机
  282.     if(!fResult) 
  283. { 
  284.              MessageBox("InitiateSystemShutdown failed."); 
  285. } 
  287.     tkp.Privileges[0].Attributes = 0; 
  288.         AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,(PTOKEN_PRIVILEGES) NULL, 0); 
  290.     if (GetLastError() != ERROR_SUCCESS) 
  291. {
  292.              MessageBox("AdjustTokenPrivileges disable failed."); 
  293. } 
  295.     ExitWindowsEx(EWX_SHUTDOWN,0);
  296. }
  297.     CDialog::OnTimer(nIDEvent);
  298. }
  299. /////////////////准备关机//////////////////////
  301. //////////////取出 Combo Box 中的值////////////////////
  302. void CAutoCloseDlg::OnSelchangeHour() 
  303. {
  304.     CComboBox *hour;
  305. hour=(CComboBox*)GetDlgItem(IDC_HOUR);
  306. hour->GetWindowText(close_hour);        
  307. }
  309. void CAutoCloseDlg::OnSelchangeMin() 
  310. {
  311. CComboBox *min;
  312. min=(CComboBox*)GetDlgItem(IDC_MIN);
  313. min->GetWindowText(close_min);
  314. }
  315. //////////////取出 Combo Box 中的值////////////////////
  317. void CAutoCloseDlg::OnStart() 
  318. {
  319.     begin=true;
  320.     ShowWindow(SW_HIDE);
  321. //////////////////定义系统托盘///////////////////////////
  322. m_tnid.cbSize=sizeof(NOTIFYICONDATA);//设置结构大小//
  323. m_tnid.hWnd=this->m_hWnd;//设置图标对应的窗口
  324.     m_tnid.uFlags=NIF_MESSAGE|NIF_ICON|NIF_TIP;//图标属性
  325.     m_tnid.uCallbackMessage=MYWM_NOTIFYICON;//应用程序定义的回调消息ID
  326. ////////////设置NOTIFYICONDATA结构///////////
  327. CString szToolTip;
  328.     szToolTip=_T("MFC自动关机程序");
  329.     _tcscpy(m_tnid.szTip, szToolTip);//帮助信息
  330.     m_tnid.uID=IDR_MAINFRAME;//应用程序图标
  331.     HICON hIcon;
  332.     hIcon=AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  333.     m_tnid.hIcon=hIcon;//图标句柄
  334. PNOTIFYICONDATA m_ptnid=&m_tnid;
  335. ::Shell_NotifyIcon(NIM_ADD,m_ptnid);//增加图标到系统盘
  336.     if(hIcon)::DestroyIcon(hIcon);
  337. //////////////////定义系统托盘///////////////////////////
  338. }
  339. ///////////////////处理系统托盘的消息///////////////////////
  340. LRESULT CAutoCloseDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam) 
  341. {
  342. switch(message)
  343. {
  344.         case MYWM_NOTIFYICON:
  345.     //如果是用户定义的消息
  346.         if(lParam==WM_LBUTTONDBLCLK)
  347. {
  348.         //鼠标双击时主窗口出现
  349.     AfxGetApp()->m_pMainWnd->ShowWindow(SW_SHOW);
  350. }
  351.         else if(lParam==WM_RBUTTONDOWN)    //鼠标右键单击弹出选单
  352. { 
  353.             CMenu menu;
  354.             menu.LoadMenu(IDR_MENU1); //载入事先定义的选单
  355.             CMenu *pMenu=menu.GetSubMenu(0);
  356.             CPoint pos;
  357.         GetCursorPos(&pos);
  358.             pMenu->TrackPopupMenu(TPM_LEFTALIGN|TPM_RIGHTBUTTON,pos.x,pos.y,AfxGetMainWnd());
  359. }
  360.         break;
  361.         case WM_SYSCOMMAND:
  362.         //如果是系统消息
  363.         if(wParam==SC_MINIMIZE)
  364. {
  365.             //接收到最小化消息时主窗口隐藏
  366.             AfxGetApp()->m_pMainWnd->ShowWindow(SW_HIDE);
  367.             return 0;
  368. }
  369.         break;
  370. }
  371. return CDialog::WindowProc(message, wParam, lParam);
  372. }
  373. ///////////////////处理系统托盘的消息///////////////////////
  375. void CAutoCloseDlg::OnDestroy() 
  376. {
  377. CDialog::OnDestroy();
  378. ::Shell_NotifyIcon(NIM_DELETE,&m_tnid);
  379. }
  381. void CAutoCloseDlg::OnAbout() 
  382. {
  383. MessageBox("闵 尖 :nminjian_ken@21cn.com","MFC自动关机程序");
  384. }
  386. void CAutoCloseDlg::OnStop() 
  387. {
  388. begin=false;
  389. ShowWindow(1);
  390. ::Shell_NotifyIcon(NIM_DELETE,&m_tnid);
  391. }
  393. void CAutoCloseDlg::OnDisplay() 
  394. {
  395. ShowWindow(1);
  396. ::Shell_NotifyIcon(NIM_DELETE,&m_tnid);
  397. }
  399. void CAutoCloseDlg::OnQuit() 
  400. {
  401. PostQuitMessage(0);
  402. }
  404. void CAutoCloseDlg::OnLButtonDown(UINT nFlags, CPoint point) 
  405. {
  406.     CDialog::OnLButtonDown(nFlags, point); 
  407.     // 发送WM_NCLBUTTONDOWN消息 
  408.     // 使Windows认为鼠标在标题条上 
  409.     PostMessage(WM_NCLBUTTONDOWN,HTCAPTION,MAKELPARAM(point.x, point.y)); 
  410. CDialog::OnLButtonDown(nFlags, point);
  411. }