NtAdjustGroupsToken.c File Reference

#include "precomp.h"

Include dependency graph for NtAdjustGroupsToken.c:

Go to the source code of this file.

Functions
static HANDLE	GetProcessToken (_In_ DWORD Access)
	START_TEST (NtAdjustGroupsToken)

Function Documentation

GetProcessToken()

static HANDLE GetProcessToken ( _In_ DWORD  Access )

static

Definition at line 12 of file NtAdjustGroupsToken.c.

{
    BOOL Success;
    HANDLE Token;
 
    Success = OpenProcessToken(GetCurrentProcess(), Access, &Token);
    if (!Success)
    {
        skip("Failed to open the process' token (error code: %lu)!\n", GetLastError());
        return NULL;
    }
 
    return Token;
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

NtAdjustGroupsToken

)

Definition at line 28 of file NtAdjustGroupsToken.c.

{
    HANDLE TokenHandle;
    NTSTATUS Status;
 
    /* Get the token from current process but with incorrect rights */
    TokenHandle = GetProcessToken(TOKEN_DUPLICATE);
 
    /* We give an invalid handle */
    Status = NtAdjustGroupsToken(NULL,
                                 TRUE,
                                 NULL,
                                 0,
                                 NULL,
                                 NULL);
    ok_hex(Status, STATUS_INVALID_HANDLE);
 
    /* We're trying to adjust the token's groups with wrong rights */
    Status = NtAdjustGroupsToken(TokenHandle,
                                 TRUE,
                                 NULL,
                                 0,
                                 NULL,
                                 NULL);
    ok_hex(Status, STATUS_ACCESS_DENIED);
 
    /* Close our handle and open a new one with right access right */
    CloseHandle(TokenHandle);
    TokenHandle = GetProcessToken(TOKEN_ADJUST_GROUPS);
 
    /* We don't give a list of groups to be adjusted in token */
    Status = NtAdjustGroupsToken(TokenHandle,
                                 FALSE,
                                 NULL,
                                 0,
                                 NULL,
                                 NULL);
    ok_hex(Status, STATUS_INVALID_PARAMETER);
 
    /* Reset the groups of an access token to default */
    Status = NtAdjustGroupsToken(TokenHandle,
                                 TRUE,
                                 NULL,
                                 0,
                                 NULL,
                                 NULL);
    ok_hex(Status, STATUS_SUCCESS);
 
    CloseHandle(TokenHandle);
}