Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274/* * PROJECT: ReactOS API tests * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory * PURPOSE: Test for NtReadFile * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org> */
#include "precomp.h"
staticBOOLIs64BitSystem(VOID){#ifdef _WIN64 return TRUE;#else NTSTATUS Status; ULONG_PTR IsWow64;
Status = NtQueryInformationProcess(NtCurrentProcess(), ProcessWow64Information, &IsWow64, sizeof(IsWow64), NULL); if (NT_SUCCESS(Status)) { return IsWow64 != 0; }
return FALSE;#endif}
#ifdef _WIN64#define IsWow64() FALSE#else#define IsWow64() Is64BitSystem()#endif
staticULONGSizeOfMdl(VOID){ return Is64BitSystem() ? 48 : 28;}
START_TEST(NtReadFile){ NTSTATUS Status; HANDLE FileHandle; UNICODE_STRING FileName = RTL_CONSTANT_STRING(L"\\SystemRoot\\ntdll-apitest-NtReadFile-test.bin"); PVOID Buffer; SIZE_T BufferSize; LARGE_INTEGER ByteOffset; OBJECT_ATTRIBUTES ObjectAttributes; IO_STATUS_BLOCK IoStatus; FILE_DISPOSITION_INFORMATION DispositionInfo; ULONG TooLargeDataSize = (MAXUSHORT + 1 - SizeOfMdl()) / sizeof(ULONG_PTR) * PAGE_SIZE; // 0x3FF9000 on x86 ULONG LargeMdlMaxDataSize = TooLargeDataSize - PAGE_SIZE;
trace("System is %d bits, Size of MDL: %lu\n", Is64BitSystem() ? 64 : 32, SizeOfMdl()); trace("Max MDL data size: 0x%lx bytes\n", LargeMdlMaxDataSize);
ByteOffset.QuadPart = 0;
Buffer = NULL; BufferSize = TooLargeDataSize; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &Buffer, 0, &BufferSize, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE); if (!NT_SUCCESS(Status)) { skip("Failed to allocate memory, status %lx\n", Status); return; }
InitializeObjectAttributes(&ObjectAttributes, &FileName, OBJ_CASE_INSENSITIVE, NULL, NULL); Status = NtCreateFile(&FileHandle, FILE_READ_DATA | FILE_WRITE_DATA | DELETE | SYNCHRONIZE, &ObjectAttributes, &IoStatus, NULL, 0, 0, FILE_SUPERSEDE, FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT | FILE_NO_INTERMEDIATE_BUFFERING, NULL, 0); ok_hex(Status, STATUS_SUCCESS);
ByteOffset.QuadPart = 0x10000; Status = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, BufferSize - 0x10000, &ByteOffset, NULL); ok_hex(Status, STATUS_SUCCESS); ByteOffset.QuadPart = 0;
/* non-cached, max size -- succeeds */ Status = NtReadFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, LargeMdlMaxDataSize - PAGE_SIZE, &ByteOffset, NULL); ok_hex(Status, STATUS_SUCCESS);
/* non-cached, max size -- succeeds */ Status = NtReadFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, LargeMdlMaxDataSize, &ByteOffset, NULL); ok_hex(Status, STATUS_SUCCESS);
/* non-cached, too large -- fails to allocate MDL * Note: this returns STATUS_SUCCESS on Vista+ -- higher MDL size limit */ Status = NtReadFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, LargeMdlMaxDataSize + PAGE_SIZE, &ByteOffset, NULL); if (GetNTVersion() >= _WIN32_WINNT_VISTA) ok_hex(Status, STATUS_SUCCESS); else ok_hex(Status, STATUS_INSUFFICIENT_RESOURCES);
/* Invalid buffer address */ Status = NtReadFile(FileHandle, NULL, NULL, NULL, &IoStatus, LongToPtr(-1), PAGE_SIZE, &ByteOffset, NULL); ok_hex(Status, STATUS_ACCESS_VIOLATION);
/* Buffer probing fails */ Status = NtReadFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, 2 * LargeMdlMaxDataSize, &ByteOffset, NULL); ok_hex(Status, STATUS_ACCESS_VIOLATION); // Different to NtWriteFile
/* non-cached, unaligned -- fails with invalid parameter */ Status = NtReadFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, LargeMdlMaxDataSize + 1, &ByteOffset, NULL); ok_hex(Status, STATUS_INVALID_PARAMETER); // Different to NtWriteFile
DispositionInfo.DeleteFile = TRUE; Status = NtSetInformationFile(FileHandle, &IoStatus, &DispositionInfo, sizeof(DispositionInfo), FileDispositionInformation); ok_hex(Status, STATUS_SUCCESS); Status = NtClose(FileHandle); ok_hex(Status, STATUS_SUCCESS);
Status = NtCreateFile(&FileHandle, FILE_READ_DATA | FILE_WRITE_DATA | DELETE | SYNCHRONIZE, &ObjectAttributes, &IoStatus, NULL, 0, 0, FILE_SUPERSEDE, FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0); ok_hex(Status, STATUS_SUCCESS);
ByteOffset.QuadPart = 0x10000; Status = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, BufferSize - 0x10000, &ByteOffset, NULL); ok_hex(Status, STATUS_SUCCESS); ByteOffset.QuadPart = 0;
/* cached: succeeds with arbitrary length */ Status = NtReadFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, LargeMdlMaxDataSize, &ByteOffset, NULL); ok_hex(Status, STATUS_SUCCESS);
Status = NtReadFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, LargeMdlMaxDataSize + 1, &ByteOffset, NULL); ok_hex(Status, STATUS_SUCCESS);
Status = NtReadFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, TooLargeDataSize, &ByteOffset, NULL); ok_hex(Status, STATUS_SUCCESS);
DispositionInfo.DeleteFile = TRUE; Status = NtSetInformationFile(FileHandle, &IoStatus, &DispositionInfo, sizeof(DispositionInfo), FileDispositionInformation); ok_hex(Status, STATUS_SUCCESS); Status = NtClose(FileHandle); ok_hex(Status, STATUS_SUCCESS);
Status = NtFreeVirtualMemory(NtCurrentProcess(), &Buffer, &BufferSize, MEM_RELEASE); ok_hex(Status, STATUS_SUCCESS);}