modules/rostests/apitests/ntdll/NtOpenThreadToken.c

   1 /*
   2  * PROJECT:         ReactOS api tests
   3  * LICENSE:         LGPLv2.1+ - See COPYING.LIB in the top level directory
   4  * PURPOSE:         Test for NtOpenThreadToken[Ex]
   5  * PROGRAMMER:      Thomas Faber <thomas.faber@reactos.org>
   6  */
   7
   8 #include "precomp.h"
   9
  10 START_TEST(NtOpenThreadToken)
  11 {
  12     NTSTATUS Status;
  13     HANDLE TokenHandle;
  14     BOOLEAN OpenAsSelf;
  15     ULONG HandleAttributes[] = { 0, OBJ_KERNEL_HANDLE };
  16     ULONG i;
  17
  18     TokenHandle = (HANDLE)0x55555555;
  19     Status = NtOpenThreadToken(NtCurrentThread(),
  20                                TOKEN_READ,
  21                                TRUE,
  22                                &TokenHandle);
  23     ok(Status == STATUS_NO_TOKEN, "Status = %lx\n", Status);
  24     ok(TokenHandle == (HANDLE)0x55555555 /* 2003 */ ||
  25        TokenHandle == NULL /* Win7 */, "TokenHandle = %p\n", TokenHandle);
  26
  27     Status = RtlImpersonateSelf(SecurityImpersonation);
  28     ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
  29     for (OpenAsSelf = FALSE; OpenAsSelf <= TRUE; OpenAsSelf++)
  30     {
  31         Status = NtOpenThreadToken(NtCurrentThread(),
  32                                    TOKEN_READ,
  33                                    OpenAsSelf,
  34                                    &TokenHandle);
  35         ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
  36         if (NT_SUCCESS(Status))
  37         {
  38             ok((LONG_PTR)TokenHandle > 0, "TokenHandle = %p\n", TokenHandle);
  39             Status = NtClose(TokenHandle);
  40             ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
  41         }
  42
  43         for (i = 0; i < RTL_NUMBER_OF(HandleAttributes); i++)
  44         {
  45             Status = NtOpenThreadTokenEx(NtCurrentThread(),
  46                                          TOKEN_READ,
  47                                          OpenAsSelf,
  48                                          HandleAttributes[i],
  49                                          &TokenHandle);
  50             ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
  51             if (NT_SUCCESS(Status))
  52             {
  53                 ok((LONG_PTR)TokenHandle > 0, "TokenHandle = %p\n", TokenHandle);
  54                 Status = NtClose(TokenHandle);
  55                 ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
  56             }
  57         }
  58     }
  59
  60     TokenHandle = NULL;
  61     Status = NtSetInformationThread(NtCurrentThread(),
  62                                     ThreadImpersonationToken,
  63                                     &TokenHandle,
  64                                     sizeof(TokenHandle));
  65     ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
  66 }