首页 > 代码库 > 提升进程权限-OpenProcessToken等函数的用法(转载)

提升进程权限-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);}

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