Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353/* * PROJECT: ReactOS API tests * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory * PURPOSE: Test for NtWriteFile * 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;}
staticULONGSizeOfSector(VOID){ BOOL Ret; ULONG SectorSize;
/* FIXME: Would be better to actually open systemroot */ Ret = GetDiskFreeSpaceW(NULL, NULL, &SectorSize, NULL, NULL); ok(Ret != FALSE, "GetDiskFreeSpaceW failed: %lx\n", GetLastError()); if (!Ret) { SectorSize = 4096; /* On failure, assume max size */ }
return SectorSize;}
START_TEST(NtWriteFile){ NTSTATUS Status; HANDLE FileHandle; UNICODE_STRING FileName = RTL_CONSTANT_STRING(L"\\SystemRoot\\ntdll-apitest-NtWriteFile-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_READONLY); 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_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);
/* non-cached, max size -- succeeds */ Status = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, LargeMdlMaxDataSize - PAGE_SIZE, &ByteOffset, NULL); ok_hex(Status, STATUS_SUCCESS);
/* non-cached, max size -- succeeds */ Status = NtWriteFile(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 = NtWriteFile(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 = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, LongToPtr(-1), PAGE_SIZE, &ByteOffset, NULL); ok_hex(Status, IsWow64() ? STATUS_INVALID_PARAMETER : STATUS_ACCESS_VIOLATION); // Different to NtReadFile
/* Buffer probing fails */ Status = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, 2 * LargeMdlMaxDataSize, &ByteOffset, NULL); ok_hex(Status, Is64BitSystem() ? STATUS_INVALID_USER_BUFFER: STATUS_INSUFFICIENT_RESOURCES); // Different to NtReadFile
/* non-cached, unaligned -- fails with invalid parameter */ Status = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, LargeMdlMaxDataSize + 1, &ByteOffset, NULL); ok_hex(Status, STATUS_INVALID_PARAMETER);
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_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);
/* cached: succeeds with arbitrary length */ Status = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, LargeMdlMaxDataSize, &ByteOffset, NULL); ok_hex(Status, STATUS_SUCCESS);
Status = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, LargeMdlMaxDataSize + 1, &ByteOffset, NULL); ok_hex(Status, STATUS_SUCCESS);
Status = NtWriteFile(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);
/* Now, testing aligned/non aligned writes */
BufferSize = SizeOfSector(); trace("Sector is %ld bytes\n", BufferSize);
Status = NtAllocateVirtualMemory(NtCurrentProcess(), &Buffer, 0, &BufferSize, MEM_RESERVE | MEM_COMMIT, PAGE_READONLY); if (!NT_SUCCESS(Status)) { skip("Failed to allocate memory, status %lx\n", Status); return; }
Status = NtCreateFile(&FileHandle, FILE_WRITE_DATA | DELETE | SYNCHRONIZE, &ObjectAttributes, &IoStatus, NULL, 0, 0, FILE_SUPERSEDE, FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT | FILE_NO_INTERMEDIATE_BUFFERING | FILE_WRITE_THROUGH, NULL, 0); ok_hex(Status, STATUS_SUCCESS);
/* non-cached, broken length -- fails with invalid parameter */ ByteOffset.QuadPart = 0; Status = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, 4, &ByteOffset, NULL); ok_hex(Status, STATUS_INVALID_PARAMETER);
/* non-cached, broken offset -- fails with invalid parameter */ ByteOffset.QuadPart = 4; Status = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, BufferSize, &ByteOffset, NULL); ok_hex(Status, STATUS_INVALID_PARAMETER);
/* non-cached, good length and offset -- succeeds */ ByteOffset.QuadPart = 0; Status = NtWriteFile(FileHandle, NULL, NULL, NULL, &IoStatus, Buffer, BufferSize, &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);}