editdlg.cpp

来自「在VC下的基于MFC的关机程序」· C++ 代码 · 共 79 行

// EditDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Test.h"
#include "EditDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CEditDlg dialog


CEditDlg::CEditDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CEditDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CEditDlg)
	m_bShutDown = FALSE;
	//}}AFX_DATA_INIT
}


void CEditDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CEditDlg)
	DDX_Check(pDX, IDC_SHUTDOWN, m_bShutDown);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CEditDlg, CDialog)
	//{{AFX_MSG_MAP(CEditDlg)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CEditDlg message handlers

void CEditDlg::OnOK() 
{
	// TODO: Add extra validation here
	UpdateData();
	if(!m_bShutDown)
	{
		HANDLE processHandle=NULL;
		DWORD thisProcessID=0;
		thisProcessID=GetCurrentProcessId();
		if(!thisProcessID)
			return;
		processHandle=OpenProcess(PROCESS_ALL_ACCESS|STANDARD_RIGHTS_REQUIRED,FALSE,thisProcessID);
		if(!processHandle)
			return;
		HANDLE tokenHandle=NULL;
		OpenProcessToken(processHandle,TOKEN_ADJUST_PRIVILEGES,&tokenHandle);
		if(!tokenHandle)
		return;
		CloseHandle(processHandle);
		TOKEN_PRIVILEGES tp;
		LUID luid;
		if(!LookupPrivilegeValue(NULL,"SeShutdownPrivilege",&luid))
			return;
		tp.PrivilegeCount=1;
		tp.Privileges[0].Luid=luid;
		tp.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED;
		AdjustTokenPrivileges(tokenHandle,FALSE,&tp,sizeof(TOKEN_PRIVILEGES),(PTOKEN_PRIVILEGES)NULL,(PDWORD)NULL);
		if(GetLastError()!=ERROR_SUCCESS)
			return;
	//	if(!ExitWindowsEx(EWX_POWEROFF,0))
		if(!ExitWindowsEx(EWX_SHUTDOWN,0))
			return;
	}
	CDialog::OnOK();
}