UAC 사용여부와 관리자 권한 획득 유무

User mode

UAC 사용여부와 관리자 권한 획득 유무

hswang 2014. 1. 14. 15:32

현재 프로세스가 실행되었을때,

UAC가 사용중인지 아닌지,

UAC를 사용중이라면 관리자 권한으로 실행이 된것인지 아닌지,

와 같은 동작이 필요할때가 있어 관련 코드를 찾다가 아래 코드를 발견하여 조금 손보아봤다.

// Token Elevation Type
typedef enum _TOKEN_ELEVATION_TYPE
{
	TokenElevationTypeDefault = 1,
	TokenElevationTypeFull,
	TokenElevationTypeLimited,
} TOKEN_ELEVATION_TYPE, *PTOKEN_ELEVATION_TYPE;

BOOL CheckUAC()
{
	BOOL bRet = FALSE;
	HANDLE hToken = NULL;
	TOKEN_ELEVATION_TYPE elevationType;
	DWORD dwSize = 0;

bRet = OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken);
	if( !bRet )
	{
		printf("OpenProcessToken fail : 0x%X\n", GetLastError());
		goto $cleanup;
	}

//GetTokenInformation(hToken, TokenElevationType, &elevationType, sizeof(elevationType), &dwSize);
	bRet = GetTokenInformation(hToken, MaxTokenInfoClass, &elevationType, sizeof(elevationType), &dwSize);
	if( !bRet )
	{
		// XP : ERROR_INVALID_PARAMETER (0x57)
		printf("GetTokenInformation fail : 0x%X\n", GetLastError());
		goto $cleanup;

}

switch(elevationType)
	{
	case TokenElevationTypeDefault :
		// UAC 미사용
		bRet = TRUE;
		break;

case TokenElevationTypeFull :
		// UAC 사용 : 관리자 권한으로 실행되었을때
		bRet = TRUE;
		break;

case TokenElevationTypeLimited :
		// UAC 사용 : 관리자 권한으로 실행되지 않았을때
		bRet = FALSE;
		break;

default:
		bRet = FALSE;
		break;
	}

$cleanup:

if( hToken )
	{
		CloseHandle(hToken);
		hToken = NULL;
	}

return bRet;
}

[참고 사이트]

http://blogs.msdn.com/b/cjacks/archive/2006/10/09/how-to-determine-if-a-user-is-a-member-of-the-administrators-group-with-uac-enabled-on-windows-vista.aspx