mirror of
https://github.com/reactos/reactos.git
synced 2024-10-08 02:13:56 +00:00
70 lines
2.5 KiB
C
70 lines
2.5 KiB
C
|
/*
|
||
|
* PROJECT: ReactOS api tests
|
||
|
* LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
|
||
|
* PURPOSE: Test for NtOpenThreadToken[Ex]
|
||
|
* PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
|
||
|
*/
|
||
|
|
||
|
#include <apitest.h>
|
||
|
|
||
|
#include <ndk/obfuncs.h>
|
||
|
#include <ndk/psfuncs.h>
|
||
|
#include <ndk/rtlfuncs.h>
|
||
|
|
||
|
START_TEST(NtOpenThreadToken)
|
||
|
{
|
||
|
NTSTATUS Status;
|
||
|
HANDLE TokenHandle;
|
||
|
BOOLEAN OpenAsSelf;
|
||
|
ULONG HandleAttributes[] = { 0, OBJ_KERNEL_HANDLE };
|
||
|
ULONG i;
|
||
|
|
||
|
TokenHandle = (HANDLE)0x55555555;
|
||
|
Status = NtOpenThreadToken(NtCurrentThread(),
|
||
|
TOKEN_READ,
|
||
|
TRUE,
|
||
|
&TokenHandle);
|
||
|
ok(Status == STATUS_NO_TOKEN, "Status = %lx\n", Status);
|
||
|
ok(TokenHandle == NULL, "TokenHandle = %p\n", TokenHandle);
|
||
|
|
||
|
Status = RtlImpersonateSelf(SecurityImpersonation);
|
||
|
ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
|
||
|
for (OpenAsSelf = FALSE; OpenAsSelf <= TRUE; OpenAsSelf++)
|
||
|
{
|
||
|
Status = NtOpenThreadToken(NtCurrentThread(),
|
||
|
TOKEN_READ,
|
||
|
OpenAsSelf,
|
||
|
&TokenHandle);
|
||
|
ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
|
||
|
if (NT_SUCCESS(Status))
|
||
|
{
|
||
|
ok((LONG_PTR)TokenHandle > 0, "TokenHandle = %p\n", TokenHandle);
|
||
|
Status = NtClose(TokenHandle);
|
||
|
ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
|
||
|
}
|
||
|
|
||
|
for (i = 0; i < RTL_NUMBER_OF(HandleAttributes); i++)
|
||
|
{
|
||
|
Status = NtOpenThreadTokenEx(NtCurrentThread(),
|
||
|
TOKEN_READ,
|
||
|
OpenAsSelf,
|
||
|
HandleAttributes[i],
|
||
|
&TokenHandle);
|
||
|
ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
|
||
|
if (NT_SUCCESS(Status))
|
||
|
{
|
||
|
ok((LONG_PTR)TokenHandle > 0, "TokenHandle = %p\n", TokenHandle);
|
||
|
Status = NtClose(TokenHandle);
|
||
|
ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
TokenHandle = NULL;
|
||
|
Status = NtSetInformationThread(NtCurrentThread(),
|
||
|
ThreadImpersonationToken,
|
||
|
&TokenHandle,
|
||
|
sizeof(TokenHandle));
|
||
|
ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
|
||
|
}
|