[英]Impersonate and run any method using a logged in user context from SYSTEM level process (Windows Service):
我已經創建了一個模板類,它將使用登錄的用戶令牌創建一個線程並運行任何方法。 我借助std :: function將其傳遞給模板類,並在用戶級別的上下文中運行它。
我目前的用法:
1.該類在Windows服務進程(以SYSTEM級別運行)中使用。
2.線程在掛起狀態下創建,並且SetThreadToken()用於應用從“ explorer.exe”創建的模擬令牌。 (我知道在多個已登錄用戶的情況下這不是可行的方法,但暫時我需要使用此工具解決問題)。
3.之后,我將繼續線程以在用戶級別上下文中執行傳遞的util方法。
引用的線程模擬來自: 創建具有特定特權的線程c ++
面對問題:
成功模擬了已登錄的用戶,並且創建的線程也在用戶級上下文中運行(已使用GetUserName() API進行了驗證),但是從我傳遞給模板類的方法中執行的任何API均與預期不符。 (示例:我試圖讀取模擬用戶的HKCU條目,但它總是失敗,並顯示錯誤: ERROR_FILE_NOT_FOUND。2(0x2) )。 PS:交叉檢查了注冊表樹,並且那里有相應的注冊表項。
注意:上面的示例( REG read )是我嘗試傳遞給Userimpersonator類並嘗試從服務在用戶上下文中運行的util方法之一,但是在運行時,我將將此方法與任何util方法一起使用。
UserImpersonator.h
class UserImpersonator
{
public:
UserImpersonator();
UserImpersonator(ImpersonationType typeVal,bool b_ImpersonateAndRunAsThreadNeeded = false);
~UserImpersonator();
T1 ImpersonateAndRun(T2 callback_function);
T1 ImpersonateAndRunAsThread(T2 callback_function);
private:
ImpersonationType ImpersonationTypeVal;
CString m_processname;
HANDLE hToken;
HANDLE impToken;
bool m_ImpersonateAndRunAsThreadNeeded;
T1 return_value;
T2 callable_function_object;
HANDLE hThread;
BOOL InitializeImpersonation();
BOOL RevertImpersonation();
static DWORD WINAPI SpawnImpersonatedThread ( LPVOID lpParam );
};
方法定義:
template<typename T1,typename T2>
UserImpersonator<T1,T2>::UserImpersonator()
{
ImpersonationTypeVal = ImpersonationType::IMPERSONATION_USING_WINLOGON;
m_processname = _T("winlogon.exe");
hToken = NULL;
m_ImpersonateAndRunAsThreadNeeded = false;
hThread = NULL;
impToken = NULL;
InitializeImpersonation();
}
template<typename T1,typename T2>
UserImpersonator<T1,T2>::UserImpersonator(ImpersonationType typeVal,bool b_ImpersonateAndRunAsThreadNeeded)
{
ImpersonationTypeVal = typeVal;
m_processname = (typeVal == ImpersonationType::IMPERSONATION_USING_WINLOGON) ? _T("winlogon.exe") : _T("explorer.exe");
hToken = NULL;
m_ImpersonateAndRunAsThreadNeeded = b_ImpersonateAndRunAsThreadNeeded;
hThread = NULL;
impToken = NULL;
InitializeImpersonation();
}
template<typename T1,typename T2>
DWORD WINAPI UserImpersonator<T1,T2> :: SpawnImpersonatedThread ( LPVOID lpParam )
{
TRY
{
UserImpersonator* ImpersonatorObject = (UserImpersonator*) lpParam;
TCHAR UserName[200] = _T("");
DWORD size = 200 ; //sizeof ( UserName ) ;
GetUserName ( UserName , &size ) ;
CString name = CString(UserName);
ImpersonatorObject->return_value = ImpersonatorObject->ImpersonateAndRun(ImpersonatorObject->callable_function_object);
}
CATCH_ALL( e )
{
LogDebug ( _T("Exception occurs:%s"),__FUNCTIONW__ ) ;
return FALSE ;
}
END_CATCH_ALL
return 0;
}
template<typename T1,typename T2>
BOOL UserImpersonator<T1,T2>::InitializeImpersonation()
{
BOOL res = TRUE;
try
{
TCHAR UserName[200] = _T("");
DWORD size = 200 ; //sizeof ( UserName ) ;
GetUserName ( UserName , &size ) ;
CString name = CString(UserName);
HANDLE process_handle = GetProcessHandleByName(m_processname);
if ( OpenProcessToken(process_handle, TOKEN_ALL_ACCESS, &hToken) == 0 )
{
res = FALSE;
CloseHandle(process_handle);
LogCritical(_T("%s : OpenProcessToken Failed with error-%d"), __FUNCTIONW__, GetLastError());
}
}
catch(...)
{
LogDebug(_T("%s::Exception occurred"),__FUNCTIONW__);
}
return res;
}
template<typename T1,typename T2>
T1 UserImpersonator<T1,T2>::ImpersonateAndRunAsThread(T2 callback_function)
{
try
{
callable_function_object = callback_function;
hThread = ::CreateThread(0,0,SpawnImpersonatedThread,this,CREATE_SUSPENDED,0); //without using sb
BOOL b = DuplicateTokenEx(hToken,MAXIMUM_ALLOWED,NULL,SecurityImpersonation,TokenImpersonation,&impToken);
if( hThread )
{
if(SetThreadToken(&hThread,impToken))
{
DWORD thread_suspended_count = ResumeThread(hThread);
if( thread_suspended_count == (DWORD) 0 || thread_suspended_count == (DWORD) 1 )
{
DWORD thread_return_status = WaitForSingleObject(hThread,INFINITE);
if( thread_return_status == WAIT_OBJECT_0 )
{
LogDebug(_T("%s::SpawnImpersonatedThread successfully executed the callback function"),__FUNCTIONW__);
}
else
LogDebug(_T("%s::WaitForSingleObject failed with error=%d"),__FUNCTIONW__,GetLastError());
}
}
else
LogDebug(_T("%s::SetThreadToken failed with error=%d"),__FUNCTIONW__,GetLastError());
}
else
LogDebug(_T("%s::CreateThread failed with error=%d"),__FUNCTIONW__,GetLastError());
}
catch(...)
{
LogDebug(_T("%s::Exception occurred"),__FUNCTIONW__);
}
return return_value;
}
template<typename T1,typename T2>
T1 UserImpersonator<T1,T2>:: ImpersonateAndRun(T2 callback_function)
{
try
{
return_value = callback_function();
}
catch(...)
{
LogDebug(_T("%s::Exception occurred"),__FUNCTIONW__);
}
return return_value;
}
template<typename T1,typename T2>
BOOL UserImpersonator<T1,T2>::RevertImpersonation()
{
if(hToken)
CloseHandle(hToken) ;
if(impToken)
CloseHandle(impToken) ;
return RevertToSelf();
}
template<typename T1,typename T2>
UserImpersonator<T1,T2>::~UserImpersonator()
{
RevertImpersonation();
}
用法示例:
UserImpersonator< bool,std::function<bool()> > ImpersonatedObj(ImpersonationType::IMPERSONATION_USING_EXPLORER,true);
auto f = std::bind(&IsRegKeyExists);
BOOL res = ImpersonatedObj.ImpersonateAndRunAsThread(f);
使用方法:
bool IsRegKeyExists()
{
HKEY phKey = NULL;
bool res = false;
is64bit = Is64BitConfiguration();
CString subkey = _T("Volatile Environment\\USERPROFILE");
if(is64bit)
{
lRes = RegOpenKeyEx( HKEY_CURRENT_USER, subkey.GetBuffer(), 0, KEY_ALL_ACCESS | KEY_WOW64_64KEY, &phKey );
}
else
{
lRes = RegOpenKeyEx( HKEY_CURRENT_USER, subkey.GetBuffer(), 0, KEY_ALL_ACCESS, &phKey );
}
if(lRes == ERROR_SUCCESS)
{
res = true;
}
else
{
LogDebug ( _T("Key open failure! %d"), GetLastError() );
}
return res;
}
HKEY_CURRENT_USER
這是預定義的密鑰 -此句柄已緩存:
HKEY_CURRENT_USER
鍵映射到HKEY_USERS
鍵中當前用戶分支的根。 它為進程中的所有線程緩存 。 因此,加載另一個用戶的配置文件時,此值不會更改。RegOpenCurrentUser
使用線程的令牌訪問相應的密鑰,如果未加載配置文件,則使用默認值。
因此,您首先需要通過RegOpenCurrentUser
打開用戶根密鑰,然后使用此句柄HKEY_CURRENT_USER
。 如果您間接訪問用戶注冊表配置單元,也可以使用RegDisablePredefinedCache
和RegDisablePredefinedCacheEx
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.