提升进程权限-OpenProcessToken等函数的用法(转载)

GetCurrentProcessID 得到当前进程的ID OpenProcessToken 得到进程的令牌句柄LookupPrivilegeValue 查询进程的权限 AdjustTokenPrivileges 判断令牌权限
要对一个任意进程（包括系统安全进程和服务进程）进行指定了写相关的访问权的OpenProcess操作，只要当前进程具有SeDeDebug权限就可以了。要是一个用户是Administrator或是被给予了相应的权限，就可以具有该权限。可是，就算我们用Administrator帐号对一个系统安全进程执行OpenProcess(PROCESS_ALL_ACCESS,FALSE, dwProcessID)还是会遇到“访问拒绝”的错误。什么原因呢？原来在默认的情况下进程的一些访问权限是没有被使能（Enabled）的，所以我们要做的首先是使能这些权限。与此相关的一些API函数有OpenProcessToken、LookupPrivilegevalue、AdjustTokenPrivileges。我们要修改一个进程的访问令牌，首先要获得进程访问令牌的句柄，这可以通过OpenProcessToken得到，函数的原型如下：
BOOL OpenProcessToken(
HANDLE ProcessHandle,
DWORD DesiredAccess,
PHANDLE TokenHandle
)；
第一参数是要修改访问权限的进程句柄；第三个参数就是返回的访问令牌指针；第二个参数指定你要进行的操作类型，如要修改令牌我们要指定第二个参数为TOKEN_ADJUST_PRIVILEGES（其它一些参数可参考Platform SDK）。通过这个函数我们就可以得到当前进程的访问令牌的句柄（指定函数的第一个参数为GetCurrentProcess()就可以了）。接着我们可以调用AdjustTokenPrivileges对这个访问令牌进行修改。AdjustTokenPrivileges的原型如下：
BOOL AdjustTokenPrivileges(
HANDLE TokenHandle, // handle to token
BOOL DisableAllPrivileges, // disabling option
PTOKEN_PRIVILEGES NewState, // privilege information
DWORD BufferLength, // size of buffer
PTOKEN_PRIVILEGES PreviousState, // original state buffer
PDWORD ReturnLength // required buffer size
);
第一个参数是访问令牌的句柄；第二个参数决定是进行权限修改还是除能（Disable）所有权限；第三个参数指明要修改的权限，是一个指向TOKEN_PRIVILEGES结构的指针，该结构包含一个数组，数据组的每个项指明了权限的类型和要进行的操作; 第四个参数是结构PreviousState的长度，如果PreviousState为空，该参数应为NULL；第五个参数也是一个指向TOKEN_PRIVILEGES结构的指针，存放修改前的访问权限的信息，可空；最后一个参数为实际PreviousState结构返回的大小。在使用这个函数前再看一下TOKEN_PRIVILEGES这个结构，其声明如下：
typedef struct _TOKEN_PRIVILEGES {
DWORD PrivilegeCount;
LUID_AND_ATTRIBUTES Privileges[];
} TOKEN_PRIVILEGES, *PTOKEN_PRIVILEGES;
PrivilegeCount指的数组原素的个数，接着是一个LUID_AND_ATTRIBUTES类型的数组，再来看一下LUID_AND_ATTRIBUTES这个结构的内容，声明如下：
typedef struct _LUID_AND_ATTRIBUTES {
LUID Luid;
DWORD Attributes;
} LUID_AND_ATTRIBUTES, *PLUID_AND_ATTRIBUTES
第二个参数就指明了我们要进行的操作类型，有三个可选项： SE_PRIVILEGE_ENABLED、SE_PRIVILEGE_ENABLED_BY_DEFAULT、SE_PRIVILEGE_USED_FOR_ACCESS。要使能一个权限就指定Attributes为SE_PRIVILEGE_ENABLED。第一个参数就是指权限的类型，是一个LUID的值，LUID就是指locally unique identifier，我想GUID大家是比较熟悉的，和GUID的要求保证全局唯一不同，LUID只要保证局部唯一，就是指在系统的每一次运行期间保证是唯一的就可以了。另外和GUID相同的一点，LUID也是一个64位的值，相信大家都看过GUID那一大串的值，我们要怎么样才能知道一个权限对应的LUID值是多少呢？这就要用到另外一个API函数LookupPrivilegevalue，其原形如下：
BOOL LookupPrivilegevalue(
LPCTSTR lpSystemName, // system name
LPCTSTR lpName, // privilege name
PLUID lpLuid // locally unique identifier
);
第一个参数是系统的名称，如果是本地系统只要指明为NULL就可以了，第三个参数就是返回LUID的指针，第二个参数就是指明了权限的名称，如“SeDebugPrivilege”。在Winnt.h中还定义了一些权限名称的宏，如：
#define SE_BACKUP_NAME TEXT("SeBackupPrivilege")
#define SE_RESTORE_NAME TEXT("SeRestorePrivilege")
#define SE_SHUTDOWN_NAME TEXT("SeShutdownPrivilege")
#define SE_DEBUG_NAME TEXT("SeDebugPrivilege")
这样通过这三个函数的调用，我们就可以用OpenProcess(PROCESS_ALL_ACCESS,FALSE, dwProcessID)来打获得任意进程的句柄，并且指定了所有的访问权。
==============================================================================

定时关机程序代码：

void CClosepcDlg::OnTimer(UINT nIDEvent)
{
	// TODO: Add your message handler code here and/or call default
	CTime t=CTime::GetCurrentTime();
	int h=t.GetHour ();
	int min=t.GetMinute ();
	CString hh,mins;
	hh.Format ("%d",h);
	mins.Format ("%d",min);
	CString str=t.Format("%H:%M:%S");
	GetDlgItem(IDC_SHOW)->SetWindowText(str);
	/////////////////准备关机//////////////////////
	if(start && close_hour.CompareNoCase(hh)==0 && close_min.CompareNoCase(mins)==0)
	{
		KillTimer(1);
		/////////首先声明3个全局变量///////////BOOL fResult;TOKEN_PRIVILEGES tkp;HANDLE hToken;
		if (!OpenProcessToken(GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken)) //得到进程的令牌句柄；
		{
			MessageBox("OpenProcessToken failed!");
		}

		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)
		{
			MessageBox("AdjustTokenPrivileges enable failed!");
		}

		fResult =InitiateSystemShutdown(
			NULL,                                  // 要关的计算机用户名
			"你设置的自动关机时间以到，请保存好你的文件，系统马上就要关闭！", // 显示的消息
			20,                                    // 关机所需的时间
			TRUE,                                 // ask user to close apps
			//        TRUE);                               //设为TRUE为重起，设为FALSE为关机
			FALSE);
		if(!fResult)
		{
			MessageBox("InitiateSystemShutdown failed.");
		}

		tkp.Privileges[0].Attributes = 0;
		AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,(PTOKEN_PRIVILEGES) NULL, 0);

		if (GetLastError() != ERROR_SUCCESS)
		{
			MessageBox("AdjustTokenPrivileges disable failed.");
		}

		ExitWindowsEx(EWX_SHUTDOWN,0);
	}
	CDialog::OnTimer(nIDEvent);
}

void CClosepcDlg::OnStar()
{
	// TODO: Add your control notification handler code here
	start=true;
	ShowWindow(SW_HIDE);
	//////////////////定义系统托盘//////////////////////////
	m_nid.cbSize =sizeof(NOTIFYICONDATA);//设置结构大小
	m_nid.hWnd =this->m_hWnd ;//设置图标对应的窗口
	m_nid.uFlags =NIF_MESSAGE|NIF_ICON|NIF_TIP;//图标属性
	m_nid.uCallbackMessage =MYWM_NOTIFYICON;//应用程序定义的回调消息ID
	////////////设置NOTIFYICONDATA结构///////////
	CString szToolTip;
	szToolTip=_T("自动关机程序");
	_tcscpy(m_nid.szTip ,szToolTip);//帮助信息
	m_nid.uID =IDR_MAINFRAME;//应用程序图标
	HICON hIcon;
	hIcon=AfxGetApp()->LoadIcon (IDR_MAINFRAME);
	m_nid.hIcon =hIcon;//图标句柄
	// PNOTIFYICONDATA m_pnid=&m_nid;
	::Shell_NotifyIcon (NIM_ADD,&m_nid);//增加图标到系统盘
	if(hIcon)
		::DestroyIcon (hIcon);
}

LRESULT CClosepcDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)
{
	switch(message)
	{
	case MYWM_NOTIFYICON:
		if(lParam==WM_LBUTTONDBLCLK)
		{
			AfxGetApp()->m_pMainWnd->ShowWindow(SW_SHOW);
		}
		else if(lParam==WM_RBUTTONDOWN)
		{
			CMenu menu;
			menu.LoadMenu(IDR_MENU1);
			CMenu *pMenu=menu.GetSubMenu(0);
			CPoint pos;
			GetCursorPos(&pos);
			pMenu->TrackPopupMenu(TPM_LEFTALIGN|TPM_RIGHTBUTTON,pos.x,pos.y,AfxGetMainWnd());
		}
		break;
	case WM_SYSCOMMAND:
		if(wParam==SC_MINIMIZE)
		{
			//////////////////定义系统托盘//////////////////////////
			m_nid.cbSize =sizeof(NOTIFYICONDATA);//设置结构大小
			m_nid.hWnd =this->m_hWnd ;//设置图标对应的窗口
			m_nid.uFlags =NIF_MESSAGE|NIF_ICON|NIF_TIP;//图标属性
			m_nid.uCallbackMessage =MYWM_NOTIFYICON;//应用程序定义的回调消息ID
			////////////设置NOTIFYICONDATA结构///////////
			CString szToolTip;
			szToolTip=_T("自动关机程序");
			_tcscpy(m_nid.szTip ,szToolTip);//帮助信息
			m_nid.uID =IDR_MAINFRAME;//应用程序图标
			HICON hIcon;
			hIcon=AfxGetApp()->LoadIcon (IDR_MAINFRAME);
			m_nid.hIcon =hIcon;//图标句柄
			//    PNOTIFYICONDATA m_pnid=&m_nid;
			::Shell_NotifyIcon (NIM_ADD,&m_nid);//增加图标到系统盘
			if(hIcon)
				::DestroyIcon (hIcon);
			AfxGetApp()->m_pMainWnd->ShowWindow(SW_HIDE);
			return 0;
		}
		break;
	}
	return CDialog::WindowProc(message, wParam, lParam);
}

void CClosepcDlg::OnQuit()
{
	// TODO: Add your command handler code here
	PostQuitMessage(0);
}

void CClosepcDlg::OnShow()
{
	// TODO: Add your command handler code here
	ShowWindow(1);
	::Shell_NotifyIcon(NIM_DELETE,&m_nid);
}

void CClosepcDlg::OnStop()
{
	// TODO: Add your command handler code here
	start=false;
	ShowWindow(1);
	::Shell_NotifyIcon(NIM_DELETE,&m_nid);
}

void CClosepcDlg::OnSelchangeHour()
{
	m_hour.GetWindowText(close_hour);
}

void CClosepcDlg::OnSelchangeMin()
{
	m_min.GetWindowText (close_min);
}