ReactOS 0.4.16-dev-336-gb667d82
NtWriteFile.c
Go to the documentation of this file.
1/*
2 * PROJECT: ReactOS API tests
3 * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
4 * PURPOSE: Test for NtWriteFile
5 * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
6 */
7
8#include "precomp.h"
9
10static
11BOOL
12Is64BitSystem(VOID)
13{
14#ifdef _WIN64
15 return TRUE;
16#else
17 NTSTATUS Status;
18 ULONG_PTR IsWow64;
19
20 Status = NtQueryInformationProcess(NtCurrentProcess(),
21 ProcessWow64Information,
22 &IsWow64,
23 sizeof(IsWow64),
24 NULL);
25 if (NT_SUCCESS(Status))
26 {
27 return IsWow64 != 0;
28 }
29
30 return FALSE;
31#endif
32}
33
34static
35ULONG
36SizeOfMdl(VOID)
37{
38 return Is64BitSystem() ? 48 : 28;
39}
40
41static
42ULONG
43SizeOfSector(VOID)
44{
45 BOOL Ret;
46 ULONG SectorSize;
47
48 /* FIXME: Would be better to actually open systemroot */
49 Ret = GetDiskFreeSpaceW(NULL, NULL, &SectorSize, NULL, NULL);
50 ok(Ret != FALSE, "GetDiskFreeSpaceW failed: %lx\n", GetLastError());
51 if (!Ret)
52 {
53 SectorSize = 4096; /* On failure, assume max size */
54 }
55
56 return SectorSize;
57}
58
59START_TEST(NtWriteFile)
60{
61 NTSTATUS Status;
62 HANDLE FileHandle;
63 UNICODE_STRING FileName = RTL_CONSTANT_STRING(L"\\SystemRoot\\ntdll-apitest-NtWriteFile-test.bin");
64 PVOID Buffer;
65 SIZE_T BufferSize;
66 LARGE_INTEGER ByteOffset;
67 OBJECT_ATTRIBUTES ObjectAttributes;
68 IO_STATUS_BLOCK IoStatus;
69 FILE_DISPOSITION_INFORMATION DispositionInfo;
70 ULONG TooLargeDataSize = (MAXUSHORT + 1 - SizeOfMdl()) / sizeof(ULONG_PTR) * PAGE_SIZE; // 0x3FF9000 on x86
71 ULONG LargeMdlMaxDataSize = TooLargeDataSize - PAGE_SIZE;
72
73 trace("System is %d bits, Size of MDL: %lu\n", Is64BitSystem() ? 64 : 32, SizeOfMdl());
74 trace("Max MDL data size: 0x%lx bytes\n", LargeMdlMaxDataSize);
75
76 ByteOffset.QuadPart = 0;
77
78 Buffer = NULL;
79 BufferSize = TooLargeDataSize;
80 Status = NtAllocateVirtualMemory(NtCurrentProcess(),
81 &Buffer,
82 0,
83 &BufferSize,
84 MEM_RESERVE | MEM_COMMIT,
85 PAGE_READONLY);
86 if (!NT_SUCCESS(Status))
87 {
88 skip("Failed to allocate memory, status %lx\n", Status);
89 return;
90 }
91
92 InitializeObjectAttributes(&ObjectAttributes,
93 &FileName,
94 OBJ_CASE_INSENSITIVE,
95 NULL,
96 NULL);
97 Status = NtCreateFile(&FileHandle,
98 FILE_WRITE_DATA | DELETE | SYNCHRONIZE,
99 &ObjectAttributes,
100 &IoStatus,
101 NULL,
102 0,
103 0,
104 FILE_SUPERSEDE,
105 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT |
106 FILE_NO_INTERMEDIATE_BUFFERING,
107 NULL,
108 0);
109 ok_hex(Status, STATUS_SUCCESS);
110
111 /* non-cached, max size -- succeeds */
112 Status = NtWriteFile(FileHandle,
113 NULL,
114 NULL,
115 NULL,
116 &IoStatus,
117 Buffer,
118 LargeMdlMaxDataSize - PAGE_SIZE,
119 &ByteOffset,
120 NULL);
121 ok_hex(Status, STATUS_SUCCESS);
122
123 /* non-cached, max size -- succeeds */
124 Status = NtWriteFile(FileHandle,
125 NULL,
126 NULL,
127 NULL,
128 &IoStatus,
129 Buffer,
130 LargeMdlMaxDataSize,
131 &ByteOffset,
132 NULL);
133 ok_hex(Status, STATUS_SUCCESS);
134
135 /* non-cached, too large -- fails to allocate MDL
136 * Note: this returns STATUS_SUCCESS on Win7 -- higher MDL size limit */
137 Status = NtWriteFile(FileHandle,
138 NULL,
139 NULL,
140 NULL,
141 &IoStatus,
142 Buffer,
143 LargeMdlMaxDataSize + PAGE_SIZE,
144 &ByteOffset,
145 NULL);
146 ok_hex(Status, STATUS_INSUFFICIENT_RESOURCES);
147
148 /* non-cached, unaligned -- fails with invalid parameter */
149 Status = NtWriteFile(FileHandle,
150 NULL,
151 NULL,
152 NULL,
153 &IoStatus,
154 Buffer,
155 LargeMdlMaxDataSize + 1,
156 &ByteOffset,
157 NULL);
158 ok_hex(Status, STATUS_INVALID_PARAMETER);
159
160 DispositionInfo.DeleteFile = TRUE;
161 Status = NtSetInformationFile(FileHandle,
162 &IoStatus,
163 &DispositionInfo,
164 sizeof(DispositionInfo),
165 FileDispositionInformation);
166 ok_hex(Status, STATUS_SUCCESS);
167 Status = NtClose(FileHandle);
168 ok_hex(Status, STATUS_SUCCESS);
169
170 Status = NtCreateFile(&FileHandle,
171 FILE_WRITE_DATA | DELETE | SYNCHRONIZE,
172 &ObjectAttributes,
173 &IoStatus,
174 NULL,
175 0,
176 0,
177 FILE_SUPERSEDE,
178 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT,
179 NULL,
180 0);
181 ok_hex(Status, STATUS_SUCCESS);
182
183 /* cached: succeeds with arbitrary length */
184 Status = NtWriteFile(FileHandle,
185 NULL,
186 NULL,
187 NULL,
188 &IoStatus,
189 Buffer,
190 LargeMdlMaxDataSize,
191 &ByteOffset,
192 NULL);
193 ok_hex(Status, STATUS_SUCCESS);
194
195 Status = NtWriteFile(FileHandle,
196 NULL,
197 NULL,
198 NULL,
199 &IoStatus,
200 Buffer,
201 LargeMdlMaxDataSize + 1,
202 &ByteOffset,
203 NULL);
204 ok_hex(Status, STATUS_SUCCESS);
205
206 Status = NtWriteFile(FileHandle,
207 NULL,
208 NULL,
209 NULL,
210 &IoStatus,
211 Buffer,
212 TooLargeDataSize,
213 &ByteOffset,
214 NULL);
215 ok_hex(Status, STATUS_SUCCESS);
216
217 DispositionInfo.DeleteFile = TRUE;
218 Status = NtSetInformationFile(FileHandle,
219 &IoStatus,
220 &DispositionInfo,
221 sizeof(DispositionInfo),
222 FileDispositionInformation);
223 ok_hex(Status, STATUS_SUCCESS);
224 Status = NtClose(FileHandle);
225 ok_hex(Status, STATUS_SUCCESS);
226
227 Status = NtFreeVirtualMemory(NtCurrentProcess(),
228 &Buffer,
229 &BufferSize,
230 MEM_RELEASE);
231 ok_hex(Status, STATUS_SUCCESS);
232
233 /* Now, testing aligned/non aligned writes */
234
235 BufferSize = SizeOfSector();
236 trace("Sector is %ld bytes\n", BufferSize);
237
238 Status = NtAllocateVirtualMemory(NtCurrentProcess(),
239 &Buffer,
240 0,
241 &BufferSize,
242 MEM_RESERVE | MEM_COMMIT,
243 PAGE_READONLY);
244 if (!NT_SUCCESS(Status))
245 {
246 skip("Failed to allocate memory, status %lx\n", Status);
247 return;
248 }
249
250 Status = NtCreateFile(&FileHandle,
251 FILE_WRITE_DATA | DELETE | SYNCHRONIZE,
252 &ObjectAttributes,
253 &IoStatus,
254 NULL,
255 0,
256 0,
257 FILE_SUPERSEDE,
258 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT |
259 FILE_NO_INTERMEDIATE_BUFFERING |
260 FILE_WRITE_THROUGH,
261 NULL,
262 0);
263 ok_hex(Status, STATUS_SUCCESS);
264
265 /* non-cached, broken length -- fails with invalid parameter */
266 ByteOffset.QuadPart = 0;
267 Status = NtWriteFile(FileHandle,
268 NULL,
269 NULL,
270 NULL,
271 &IoStatus,
272 Buffer,
273 4,
274 &ByteOffset,
275 NULL);
276 ok_hex(Status, STATUS_INVALID_PARAMETER);
277
278 /* non-cached, broken offset -- fails with invalid parameter */
279 ByteOffset.QuadPart = 4;
280 Status = NtWriteFile(FileHandle,
281 NULL,
282 NULL,
283 NULL,
284 &IoStatus,
285 Buffer,
286 BufferSize,
287 &ByteOffset,
288 NULL);
289 ok_hex(Status, STATUS_INVALID_PARAMETER);
290
291 /* non-cached, good length and offset -- succeeds */
292 ByteOffset.QuadPart = 0;
293 Status = NtWriteFile(FileHandle,
294 NULL,
295 NULL,
296 NULL,
297 &IoStatus,
298 Buffer,
299 BufferSize,
300 &ByteOffset,
301 NULL);
302 ok_hex(Status, STATUS_SUCCESS);
303
304 DispositionInfo.DeleteFile = TRUE;
305 Status = NtSetInformationFile(FileHandle,
306 &IoStatus,
307 &DispositionInfo,
308 sizeof(DispositionInfo),
309 FileDispositionInformation);
310 ok_hex(Status, STATUS_SUCCESS);
311 Status = NtClose(FileHandle);
312 ok_hex(Status, STATUS_SUCCESS);
313
314 Status = NtFreeVirtualMemory(NtCurrentProcess(),
315 &Buffer,
316 &BufferSize,
317 MEM_RELEASE);
318 ok_hex(Status, STATUS_SUCCESS);
319}
SizeOfSector
static ULONG SizeOfSector(VOID)
Definition: NtWriteFile.c:43
Is64BitSystem
static BOOL Is64BitSystem(VOID)
Definition: NtWriteFile.c:12
SizeOfMdl
static ULONG SizeOfMdl(VOID)
Definition: NtWriteFile.c:36
ok_hex
#define ok_hex(expression, result)
Definition: atltest.h:94
trace
#define trace
Definition: atltest.h:70
ok
#define ok(value,...)
Definition: atltest.h:57
skip
#define skip(...)
Definition: atltest.h:64
START_TEST
#define START_TEST(x)
Definition: atltest.h:75
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
FILE_NON_DIRECTORY_FILE
#define FILE_NON_DIRECTORY_FILE
Definition: constants.h:492
Buffer
Definition: bufpool.h:45
ObjectAttributes
IN PUNICODE_STRING IN POBJECT_ATTRIBUTES ObjectAttributes
Definition: conport.c:36
BufferSize
#define BufferSize
Definition: mmc.h:75
NULL
#define NULL
Definition: types.h:112
TRUE
#define TRUE
Definition: types.h:120
FALSE
#define FALSE
Definition: types.h:117
NT_SUCCESS
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:33
PAGE_READONLY
#define PAGE_READONLY
Definition: compat.h:138
GetDiskFreeSpaceW
BOOL WINAPI GetDiskFreeSpaceW(IN LPCWSTR lpRootPathName, OUT LPDWORD lpSectorsPerCluster, OUT LPDWORD lpBytesPerSector, OUT LPDWORD lpNumberOfFreeClusters, OUT LPDWORD lpTotalNumberOfClusters)
Definition: disk.c:173
PAGE_SIZE
#define PAGE_SIZE
Definition: env_spec_w32.h:49
ByteOffset
IN PDCB IN PCCB IN VBO IN OUT PULONG OUT PDIRENT OUT PBCB OUT PVBO ByteOffset
Definition: fatprocs.h:732
BOOL
unsigned int BOOL
Definition: ntddk_ex.h:94
FileHandle
_Must_inspect_result_ _In_opt_ PFLT_INSTANCE _Out_ PHANDLE FileHandle
Definition: fltkernel.h:1231
FileDispositionInformation
@ FileDispositionInformation
Definition: from_kernel.h:74
FILE_SYNCHRONOUS_IO_NONALERT
#define FILE_SYNCHRONOUS_IO_NONALERT
Definition: from_kernel.h:31
FILE_NO_INTERMEDIATE_BUFFERING
#define FILE_NO_INTERMEDIATE_BUFFERING
Definition: from_kernel.h:28
FILE_WRITE_THROUGH
#define FILE_WRITE_THROUGH
Definition: from_kernel.h:26
FILE_SUPERSEDE
#define FILE_SUPERSEDE
Definition: from_kernel.h:53
Status
Status
Definition: gdiplustypes.h:25
OBJ_CASE_INSENSITIVE
#define OBJ_CASE_INSENSITIVE
Definition: winternl.h:228
ProcessWow64Information
@ ProcessWow64Information
Definition: winternl.h:396
InitializeObjectAttributes
#define InitializeObjectAttributes(p, n, a, r, s)
Definition: reg.c:106
IoStatus
__in UCHAR __in POWER_STATE __in_opt PVOID __in PIO_STATUS_BLOCK IoStatus
Definition: mxum.h:159
SYNCHRONIZE
#define SYNCHRONIZE
Definition: nt_native.h:61
FILE_WRITE_DATA
#define FILE_WRITE_DATA
Definition: nt_native.h:631
NtWriteFile
NTSYSAPI NTSTATUS NTAPI NtWriteFile(IN HANDLE hFile, IN HANDLE hEvent OPTIONAL, IN PIO_APC_ROUTINE IoApcRoutine OPTIONAL, IN PVOID IoApcContext OPTIONAL, OUT PIO_STATUS_BLOCK pIoStatusBlock, IN PVOID WriteBuffer, IN ULONG WriteBufferLength, IN PLARGE_INTEGER FileOffset OPTIONAL, IN PULONG LockOperationKey OPTIONAL)
NtCurrentProcess
#define NtCurrentProcess()
Definition: nt_native.h:1657
NtSetInformationFile
NTSYSAPI NTSTATUS NTAPI NtSetInformationFile(IN HANDLE hFile, OUT PIO_STATUS_BLOCK pIoStatusBlock, IN PVOID FileInformationBuffer, IN ULONG FileInformationBufferLength, IN FILE_INFORMATION_CLASS FileInfoClass)
Definition: iofunc.c:3096
NtClose
NTSTATUS NTAPI NtClose(IN HANDLE Handle)
Definition: obhandle.c:3402
DELETE
#define DELETE
Definition: nt_native.h:57
MEM_RESERVE
#define MEM_RESERVE
Definition: nt_native.h:1314
MEM_RELEASE
#define MEM_RELEASE
Definition: nt_native.h:1316
NtCreateFile
NTSTATUS NTAPI NtCreateFile(OUT PHANDLE FileHandle, IN ACCESS_MASK DesiredAccess, IN POBJECT_ATTRIBUTES ObjectAttributes, OUT PIO_STATUS_BLOCK IoStatusBlock, IN PLARGE_INTEGER AllocationSize OPTIONAL, IN ULONG FileAttributes, IN ULONG ShareAccess, IN ULONG CreateDisposition, IN ULONG CreateOptions, IN PVOID EaBuffer OPTIONAL, IN ULONG EaLength)
MEM_COMMIT
#define MEM_COMMIT
Definition: nt_native.h:1313
NtFreeVirtualMemory
NTSTATUS NTAPI NtFreeVirtualMemory(IN HANDLE ProcessHandle, IN PVOID *UBaseAddress, IN PSIZE_T URegionSize, IN ULONG FreeType)
Definition: virtual.c:5230
NtAllocateVirtualMemory
NTSTATUS NTAPI NtAllocateVirtualMemory(IN HANDLE ProcessHandle, IN OUT PVOID *UBaseAddress, IN ULONG_PTR ZeroBits, IN OUT PSIZE_T URegionSize, IN ULONG AllocationType, IN ULONG Protect)
Definition: virtual.c:4492
NtQueryInformationProcess
NTSTATUS NTAPI NtQueryInformationProcess(_In_ HANDLE ProcessHandle, _In_ PROCESSINFOCLASS ProcessInformationClass, _Out_ PVOID ProcessInformation, _In_ ULONG ProcessInformationLength, _Out_opt_ PULONG ReturnLength)
Definition: query.c:59
L
#define L(x)
Definition: ntvdm.h:50
STATUS_SUCCESS
#define STATUS_SUCCESS
Definition: shellext.h:65
_FILE_DISPOSITION_INFORMATION
Definition: nt_native.h:972
_FILE_DISPOSITION_INFORMATION::DeleteFile
BOOLEAN DeleteFile
Definition: nt_native.h:973
_FileName
Definition: filecomp.c:348
_IO_STATUS_BLOCK
Definition: env_spec_w32.h:870
_OBJECT_ATTRIBUTES
Definition: umtypes.h:182
_UNICODE_STRING
Definition: env_spec_w32.h:368
RTL_CONSTANT_STRING
#define RTL_CONSTANT_STRING(s)
Definition: tunneltest.c:14
SIZE_T
ULONG_PTR SIZE_T
Definition: typedefs.h:80
MAXUSHORT
#define MAXUSHORT
Definition: typedefs.h:83
ULONG_PTR
uint32_t ULONG_PTR
Definition: typedefs.h:65
ULONG
uint32_t ULONG
Definition: typedefs.h:59
STATUS_INVALID_PARAMETER
#define STATUS_INVALID_PARAMETER
Definition: udferr_usr.h:135
STATUS_INSUFFICIENT_RESOURCES
#define STATUS_INSUFFICIENT_RESOURCES
Definition: udferr_usr.h:158
_LARGE_INTEGER
Definition: typedefs.h:103
IsWow64
BOOL IsWow64(VOID)
Definition: user_lib.cpp:579
BufferSize
_In_ WDFMEMORY _Out_opt_ size_t * BufferSize
Definition: wdfmemory.h:254
GetLastError
DWORD WINAPI GetLastError(void)
Definition: except.c:1042
SectorSize
_In_ ULONG SectorSize
Definition: halfuncs.h:291