ReactOS 0.4.16-dev-2224-g3637fd8
Macros | Functions
NtWriteFile.c File Reference
#include "precomp.h"
Include dependency graph for NtWriteFile.c:

Go to the source code of this file.

Macros

#define IsWow64()   Is64BitSystem()
 

Functions

static BOOL Is64BitSystem (VOID)
 
static ULONG SizeOfMdl (VOID)
 
static ULONG SizeOfSector (VOID)
 
 START_TEST (NtWriteFile)
 

Macro Definition Documentation

◆ IsWow64

#define IsWow64 ( )    Is64BitSystem()

Definition at line 37 of file NtWriteFile.c.

Function Documentation

◆ Is64BitSystem()

static BOOL Is64BitSystem ( VOID  )
static

Definition at line 12 of file NtWriteFile.c.

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}
IsWow64
#define IsWow64()
Definition: NtWriteFile.c:37
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
ProcessWow64Information
@ ProcessWow64Information
Definition: cicbase.cpp:65
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
Status
Status
Definition: gdiplustypes.h:25
NtCurrentProcess
#define NtCurrentProcess()
Definition: nt_native.h:1660
NtQueryInformationProcess
NTSTATUS NTAPI NtQueryInformationProcess(_In_ HANDLE ProcessHandle, _In_ PROCESSINFOCLASS ProcessInformationClass, _Out_writes_bytes_to_opt_(ProcessInformationLength, *ReturnLength) PVOID ProcessInformation, _In_ ULONG ProcessInformationLength, _Out_opt_ PULONG ReturnLength)
Definition: query.c:211
ULONG_PTR
uint32_t ULONG_PTR
Definition: typedefs.h:65

Referenced by SizeOfMdl(), and START_TEST().

◆ SizeOfMdl()

static ULONG SizeOfMdl ( VOID  )
static

Definition at line 42 of file NtWriteFile.c.

43{
44 return Is64BitSystem() ? 48 : 28;
45}
Is64BitSystem
static BOOL Is64BitSystem(VOID)
Definition: NtWriteFile.c:12

Referenced by START_TEST().

◆ SizeOfSector()

static ULONG SizeOfSector ( VOID  )
static

Definition at line 49 of file NtWriteFile.c.

50{
51 BOOL Ret;
52 ULONG SectorSize;
53
54 /* FIXME: Would be better to actually open systemroot */
55 Ret = GetDiskFreeSpaceW(NULL, NULL, &SectorSize, NULL, NULL);
56 ok(Ret != FALSE, "GetDiskFreeSpaceW failed: %lx\n", GetLastError());
57 if (!Ret)
58 {
59 SectorSize = 4096; /* On failure, assume max size */
60 }
61
62 return SectorSize;
63}
ok
#define ok(value,...)
Definition: atltest.h:57
GetDiskFreeSpaceW
BOOL WINAPI GetDiskFreeSpaceW(IN LPCWSTR lpRootPathName, OUT LPDWORD lpSectorsPerCluster, OUT LPDWORD lpBytesPerSector, OUT LPDWORD lpNumberOfFreeClusters, OUT LPDWORD lpTotalNumberOfClusters)
Definition: disk.c:173
BOOL
unsigned int BOOL
Definition: ntddk_ex.h:94
ULONG
uint32_t ULONG
Definition: typedefs.h:59
GetLastError
DWORD WINAPI GetLastError(void)
Definition: except.c:1042
SectorSize
_In_ ULONG SectorSize
Definition: halfuncs.h:291

Referenced by START_TEST().

◆ START_TEST()

START_TEST ( NtWriteFile  )

Definition at line 65 of file NtWriteFile.c.

66{
67 NTSTATUS Status;
68 HANDLE FileHandle;
69 UNICODE_STRING FileName = RTL_CONSTANT_STRING(L"\\SystemRoot\\ntdll-apitest-NtWriteFile-test.bin");
70 PVOID Buffer;
71 SIZE_T BufferSize;
72 LARGE_INTEGER ByteOffset;
73 OBJECT_ATTRIBUTES ObjectAttributes;
74 IO_STATUS_BLOCK IoStatus;
75 FILE_DISPOSITION_INFORMATION DispositionInfo;
76 ULONG TooLargeDataSize = (MAXUSHORT + 1 - SizeOfMdl()) / sizeof(ULONG_PTR) * PAGE_SIZE; // 0x3FF9000 on x86
77 ULONG LargeMdlMaxDataSize = TooLargeDataSize - PAGE_SIZE;
78
79 trace("System is %d bits, Size of MDL: %lu\n", Is64BitSystem() ? 64 : 32, SizeOfMdl());
80 trace("Max MDL data size: 0x%lx bytes\n", LargeMdlMaxDataSize);
81
82 ByteOffset.QuadPart = 0;
83
84 Buffer = NULL;
85 BufferSize = TooLargeDataSize;
86 Status = NtAllocateVirtualMemory(NtCurrentProcess(),
87 &Buffer,
88 0,
89 &BufferSize,
90 MEM_RESERVE | MEM_COMMIT,
91 PAGE_READONLY);
92 if (!NT_SUCCESS(Status))
93 {
94 skip("Failed to allocate memory, status %lx\n", Status);
95 return;
96 }
97
98 InitializeObjectAttributes(&ObjectAttributes,
99 &FileName,
100 OBJ_CASE_INSENSITIVE,
101 NULL,
102 NULL);
103 Status = NtCreateFile(&FileHandle,
104 FILE_WRITE_DATA | DELETE | SYNCHRONIZE,
105 &ObjectAttributes,
106 &IoStatus,
107 NULL,
108 0,
109 0,
110 FILE_SUPERSEDE,
111 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT |
112 FILE_NO_INTERMEDIATE_BUFFERING,
113 NULL,
114 0);
115 ok_hex(Status, STATUS_SUCCESS);
116
117 /* non-cached, max size -- succeeds */
118 Status = NtWriteFile(FileHandle,
119 NULL,
120 NULL,
121 NULL,
122 &IoStatus,
123 Buffer,
124 LargeMdlMaxDataSize - PAGE_SIZE,
125 &ByteOffset,
126 NULL);
127 ok_hex(Status, STATUS_SUCCESS);
128
129 /* non-cached, max size -- succeeds */
130 Status = NtWriteFile(FileHandle,
131 NULL,
132 NULL,
133 NULL,
134 &IoStatus,
135 Buffer,
136 LargeMdlMaxDataSize,
137 &ByteOffset,
138 NULL);
139 ok_hex(Status, STATUS_SUCCESS);
140
141 /* non-cached, too large -- fails to allocate MDL
142 * Note: this returns STATUS_SUCCESS on Vista+ -- higher MDL size limit */
143 Status = NtWriteFile(FileHandle,
144 NULL,
145 NULL,
146 NULL,
147 &IoStatus,
148 Buffer,
149 LargeMdlMaxDataSize + PAGE_SIZE,
150 &ByteOffset,
151 NULL);
152 if (GetNTVersion() >= _WIN32_WINNT_VISTA)
153 ok_hex(Status, STATUS_SUCCESS);
154 else
155 ok_hex(Status, STATUS_INSUFFICIENT_RESOURCES);
156
157 /* Invalid buffer address */
158 Status = NtWriteFile(FileHandle,
159 NULL,
160 NULL,
161 NULL,
162 &IoStatus,
163 LongToPtr(-1),
164 PAGE_SIZE,
165 &ByteOffset,
166 NULL);
167 ok_hex(Status, IsWow64() ? STATUS_INVALID_PARAMETER : STATUS_ACCESS_VIOLATION); // Different to NtReadFile
168
169 /* Buffer probing fails */
170 Status = NtWriteFile(FileHandle,
171 NULL,
172 NULL,
173 NULL,
174 &IoStatus,
175 Buffer,
176 2 * LargeMdlMaxDataSize,
177 &ByteOffset,
178 NULL);
179 ok_hex(Status, Is64BitSystem() ? STATUS_INVALID_USER_BUFFER: STATUS_INSUFFICIENT_RESOURCES); // Different to NtReadFile
180
181 /* non-cached, unaligned -- fails with invalid parameter */
182 Status = NtWriteFile(FileHandle,
183 NULL,
184 NULL,
185 NULL,
186 &IoStatus,
187 Buffer,
188 LargeMdlMaxDataSize + 1,
189 &ByteOffset,
190 NULL);
191 ok_hex(Status, STATUS_INVALID_PARAMETER);
192
193 DispositionInfo.DeleteFile = TRUE;
194 Status = NtSetInformationFile(FileHandle,
195 &IoStatus,
196 &DispositionInfo,
197 sizeof(DispositionInfo),
198 FileDispositionInformation);
199 ok_hex(Status, STATUS_SUCCESS);
200 Status = NtClose(FileHandle);
201 ok_hex(Status, STATUS_SUCCESS);
202
203 Status = NtCreateFile(&FileHandle,
204 FILE_WRITE_DATA | DELETE | SYNCHRONIZE,
205 &ObjectAttributes,
206 &IoStatus,
207 NULL,
208 0,
209 0,
210 FILE_SUPERSEDE,
211 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT,
212 NULL,
213 0);
214 ok_hex(Status, STATUS_SUCCESS);
215
216 /* cached: succeeds with arbitrary length */
217 Status = NtWriteFile(FileHandle,
218 NULL,
219 NULL,
220 NULL,
221 &IoStatus,
222 Buffer,
223 LargeMdlMaxDataSize,
224 &ByteOffset,
225 NULL);
226 ok_hex(Status, STATUS_SUCCESS);
227
228 Status = NtWriteFile(FileHandle,
229 NULL,
230 NULL,
231 NULL,
232 &IoStatus,
233 Buffer,
234 LargeMdlMaxDataSize + 1,
235 &ByteOffset,
236 NULL);
237 ok_hex(Status, STATUS_SUCCESS);
238
239 Status = NtWriteFile(FileHandle,
240 NULL,
241 NULL,
242 NULL,
243 &IoStatus,
244 Buffer,
245 TooLargeDataSize,
246 &ByteOffset,
247 NULL);
248 ok_hex(Status, STATUS_SUCCESS);
249
250 DispositionInfo.DeleteFile = TRUE;
251 Status = NtSetInformationFile(FileHandle,
252 &IoStatus,
253 &DispositionInfo,
254 sizeof(DispositionInfo),
255 FileDispositionInformation);
256 ok_hex(Status, STATUS_SUCCESS);
257 Status = NtClose(FileHandle);
258 ok_hex(Status, STATUS_SUCCESS);
259
260 Status = NtFreeVirtualMemory(NtCurrentProcess(),
261 &Buffer,
262 &BufferSize,
263 MEM_RELEASE);
264 ok_hex(Status, STATUS_SUCCESS);
265
266 /* Now, testing aligned/non aligned writes */
267
268 BufferSize = SizeOfSector();
269 trace("Sector is %ld bytes\n", BufferSize);
270
271 Status = NtAllocateVirtualMemory(NtCurrentProcess(),
272 &Buffer,
273 0,
274 &BufferSize,
275 MEM_RESERVE | MEM_COMMIT,
276 PAGE_READONLY);
277 if (!NT_SUCCESS(Status))
278 {
279 skip("Failed to allocate memory, status %lx\n", Status);
280 return;
281 }
282
283 Status = NtCreateFile(&FileHandle,
284 FILE_WRITE_DATA | DELETE | SYNCHRONIZE,
285 &ObjectAttributes,
286 &IoStatus,
287 NULL,
288 0,
289 0,
290 FILE_SUPERSEDE,
291 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT |
292 FILE_NO_INTERMEDIATE_BUFFERING |
293 FILE_WRITE_THROUGH,
294 NULL,
295 0);
296 ok_hex(Status, STATUS_SUCCESS);
297
298 /* non-cached, broken length -- fails with invalid parameter */
299 ByteOffset.QuadPart = 0;
300 Status = NtWriteFile(FileHandle,
301 NULL,
302 NULL,
303 NULL,
304 &IoStatus,
305 Buffer,
306 4,
307 &ByteOffset,
308 NULL);
309 ok_hex(Status, STATUS_INVALID_PARAMETER);
310
311 /* non-cached, broken offset -- fails with invalid parameter */
312 ByteOffset.QuadPart = 4;
313 Status = NtWriteFile(FileHandle,
314 NULL,
315 NULL,
316 NULL,
317 &IoStatus,
318 Buffer,
319 BufferSize,
320 &ByteOffset,
321 NULL);
322 ok_hex(Status, STATUS_INVALID_PARAMETER);
323
324 /* non-cached, good length and offset -- succeeds */
325 ByteOffset.QuadPart = 0;
326 Status = NtWriteFile(FileHandle,
327 NULL,
328 NULL,
329 NULL,
330 &IoStatus,
331 Buffer,
332 BufferSize,
333 &ByteOffset,
334 NULL);
335 ok_hex(Status, STATUS_SUCCESS);
336
337 DispositionInfo.DeleteFile = TRUE;
338 Status = NtSetInformationFile(FileHandle,
339 &IoStatus,
340 &DispositionInfo,
341 sizeof(DispositionInfo),
342 FileDispositionInformation);
343 ok_hex(Status, STATUS_SUCCESS);
344 Status = NtClose(FileHandle);
345 ok_hex(Status, STATUS_SUCCESS);
346
347 Status = NtFreeVirtualMemory(NtCurrentProcess(),
348 &Buffer,
349 &BufferSize,
350 MEM_RELEASE);
351 ok_hex(Status, STATUS_SUCCESS);
352}
SizeOfSector
static ULONG SizeOfSector(VOID)
Definition: NtWriteFile.c:49
SizeOfMdl
static ULONG SizeOfMdl(VOID)
Definition: NtWriteFile.c:42
GetNTVersion
#define GetNTVersion()
Definition: apitest.h:17
ok_hex
#define ok_hex(expression, result)
Definition: atltest.h:94
trace
#define trace
Definition: atltest.h:70
skip
#define skip(...)
Definition: atltest.h:64
FILE_NON_DIRECTORY_FILE
#define FILE_NON_DIRECTORY_FILE
Definition: constants.h:492
LongToPtr
#define LongToPtr(l)
Definition: basetsd.h:85
Buffer
Definition: bufpool.h:45
ObjectAttributes
IN PUNICODE_STRING IN POBJECT_ATTRIBUTES ObjectAttributes
Definition: conport.c:36
BufferSize
#define BufferSize
Definition: mmc.h:75
PAGE_READONLY
#define PAGE_READONLY
Definition: compat.h:138
L
#define L(x)
Definition: resources.c:13
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
FileHandle
_Must_inspect_result_ _In_opt_ PFLT_INSTANCE _Out_ PHANDLE FileHandle
Definition: fltkernel.h:1231
STATUS_ACCESS_VIOLATION
#define STATUS_ACCESS_VIOLATION
Definition: fpEnumPrinters.c:23
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
OBJ_CASE_INSENSITIVE
#define OBJ_CASE_INSENSITIVE
Definition: winternl.h:228
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)
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:1317
MEM_RELEASE
#define MEM_RELEASE
Definition: nt_native.h:1319
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:1316
NtFreeVirtualMemory
NTSTATUS NTAPI NtFreeVirtualMemory(IN HANDLE ProcessHandle, IN PVOID *UBaseAddress, IN PSIZE_T URegionSize, IN ULONG FreeType)
Definition: virtual.c:5192
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:4457
_WIN32_WINNT_VISTA
#define _WIN32_WINNT_VISTA
Definition: sdkddkver.h:25
STATUS_SUCCESS
#define STATUS_SUCCESS
Definition: shellext.h:65
_FILE_DISPOSITION_INFORMATION
Definition: nt_native.h:975
_FILE_DISPOSITION_INFORMATION::DeleteFile
BOOLEAN DeleteFile
Definition: nt_native.h:976
_FileName
Definition: filecomp.c:348
_IO_STATUS_BLOCK
Definition: env_spec_w32.h:870
_OBJECT_ATTRIBUTES
Definition: umtypes.h:184
_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
STATUS_INVALID_PARAMETER
#define STATUS_INVALID_PARAMETER
Definition: udferr_usr.h:135
STATUS_INVALID_USER_BUFFER
#define STATUS_INVALID_USER_BUFFER
Definition: udferr_usr.h:166
STATUS_INSUFFICIENT_RESOURCES
#define STATUS_INSUFFICIENT_RESOURCES
Definition: udferr_usr.h:158
_LARGE_INTEGER
Definition: typedefs.h:103
BufferSize
_In_ WDFMEMORY _Out_opt_ size_t * BufferSize
Definition: wdfmemory.h:254