NtAdjustPrivilegesToken.c File Reference

#include "precomp.h"

Include dependency graph for NtAdjustPrivilegesToken.c:

Go to the source code of this file.

Functions
static BOOL	IsPrivilegeEnabled (_In_ HANDLE TokenHandle, _In_ ULONG Privilege)
static VOID	AdjustEnableDefaultPriv (VOID)
	START_TEST (NtAdjustPrivilegesToken)

Function Documentation

AdjustEnableDefaultPriv()

static VOID AdjustEnableDefaultPriv ( VOID  )

static

Definition at line 36 of file NtAdjustPrivilegesToken.c.

{
    NTSTATUS Status;
    HANDLE Token;
    TOKEN_PRIVILEGES Priv;
    BOOL Success, IsEnabled;
    LUID PrivLuid;
 
    Success = OpenProcessToken(GetCurrentProcess(),
                               TOKEN_READ | TOKEN_ADJUST_PRIVILEGES,
                               &Token);
    if (!Success)
    {
        skip("OpenProcessToken() has failed to get the process' token (error code: %lu)!\n", GetLastError());
        return;
    }
 
    Success = LookupPrivilegeValueW(NULL, L"SeImpersonatePrivilege", &PrivLuid);
    if (!Success)
    {
        skip("LookupPrivilegeValueW() has failed to locate the privilege value (error code: %lu)!\n", GetLastError());
        return;
    }
 
    Priv.PrivilegeCount = 1;
    Priv.Privileges[0].Luid = PrivLuid;
    Priv.Privileges[0].Attributes = 0;
 
    IsEnabled = IsPrivilegeEnabled(Token, SE_IMPERSONATE_PRIVILEGE);
    trace("The privilege is %s!\n", IsEnabled ? "enabled" : "disabled");
 
    Status = NtAdjustPrivilegesToken(Token,
                                     FALSE,
                                     &Priv,
                                     0,
                                     NULL,
                                     NULL);
    ok_hex(Status, STATUS_SUCCESS);
 
    IsEnabled = IsPrivilegeEnabled(Token, SE_IMPERSONATE_PRIVILEGE);
    trace("The privilege is %s!\n", IsEnabled ? "enabled" : "disabled");
 
    CloseHandle(Token);
}

Referenced by START_TEST().

IsPrivilegeEnabled()

static BOOL IsPrivilegeEnabled ( _In_ HANDLE  TokenHandle, _In_ ULONG  Privilege  )

static

Definition at line 12 of file NtAdjustPrivilegesToken.c.

{
    PRIVILEGE_SET PrivSet;
    BOOL Result, Success;
 
    PrivSet.PrivilegeCount = 1;
    PrivSet.Control = PRIVILEGE_SET_ALL_NECESSARY;
    PrivSet.Privilege[0].Luid = RtlConvertUlongToLuid(Privilege);
    PrivSet.Privilege[0].Attributes = 0;
 
    Success = PrivilegeCheck(TokenHandle, &PrivSet, &Result);
    if (!Success)
    {
        skip("Failed to check the privilege (error code: %lu)\n", GetLastError());
        return FALSE;
    }
 
    return Result;
}

Referenced by AdjustEnableDefaultPriv().

START_TEST()

START_TEST

(

NtAdjustPrivilegesToken

)

Definition at line 81 of file NtAdjustPrivilegesToken.c.

{
    AdjustEnableDefaultPriv();
}