You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
3288 lines
99 KiB
3288 lines
99 KiB
/*++
|
|
|
|
Copyright (c) 1990 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
filehops.c
|
|
|
|
Abstract:
|
|
|
|
This module implements Win32 file handle APIs
|
|
|
|
Author:
|
|
|
|
Mark Lucovsky (markl) 25-Sep-1990
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
#include "basedll.h"
|
|
#include "winioctl.h"
|
|
|
|
HANDLE
|
|
WINAPI
|
|
GetStdHandle(
|
|
DWORD nStdHandle
|
|
)
|
|
{
|
|
PPEB Peb;
|
|
HANDLE rv;
|
|
|
|
|
|
Peb = NtCurrentPeb();
|
|
switch( nStdHandle ) {
|
|
case STD_INPUT_HANDLE:
|
|
rv = Peb->ProcessParameters->StandardInput;
|
|
break;
|
|
|
|
case STD_OUTPUT_HANDLE:
|
|
rv = Peb->ProcessParameters->StandardOutput;
|
|
break;
|
|
|
|
case STD_ERROR_HANDLE:
|
|
rv = Peb->ProcessParameters->StandardError;
|
|
break;
|
|
default:
|
|
rv = INVALID_HANDLE_VALUE;
|
|
break;
|
|
}
|
|
if ( rv == INVALID_HANDLE_VALUE ) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
SetStdHandle(
|
|
DWORD nStdHandle,
|
|
HANDLE hHandle
|
|
)
|
|
{
|
|
PPEB Peb;
|
|
|
|
Peb = NtCurrentPeb();
|
|
switch( nStdHandle ) {
|
|
case STD_INPUT_HANDLE:
|
|
Peb->ProcessParameters->StandardInput = hHandle;
|
|
break;
|
|
|
|
case STD_OUTPUT_HANDLE:
|
|
Peb->ProcessParameters->StandardOutput = hHandle;
|
|
break;
|
|
|
|
case STD_ERROR_HANDLE:
|
|
Peb->ProcessParameters->StandardError = hHandle;
|
|
break;
|
|
|
|
default:
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
return( TRUE );
|
|
}
|
|
|
|
DWORD
|
|
WINAPI
|
|
GetFileType(
|
|
HANDLE hFile
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
GetFileType is used to determine the file type of the specified file.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose type is to be
|
|
determined
|
|
|
|
Return Value:
|
|
|
|
FILE_TYPE_UNKNOWN - The type of the specified file is unknown.
|
|
|
|
FILE_TYPE_DISK - The specified file is a disk file.
|
|
|
|
FILE_TYPE_CHAR - The specified file is a character file (LPT,
|
|
console...)
|
|
|
|
FILE_TYPE_PIPE - The specified file is a pipe (either a named pipe or
|
|
a pipe created by CreatePipe).
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
FILE_FS_DEVICE_INFORMATION DeviceInformation;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
PPEB Peb;
|
|
|
|
Peb = NtCurrentPeb();
|
|
|
|
switch( HandleToUlong(hFile) ) {
|
|
case STD_INPUT_HANDLE:
|
|
hFile = Peb->ProcessParameters->StandardInput;
|
|
break;
|
|
case STD_OUTPUT_HANDLE:
|
|
hFile = Peb->ProcessParameters->StandardOutput;
|
|
break;
|
|
case STD_ERROR_HANDLE:
|
|
hFile = Peb->ProcessParameters->StandardError;
|
|
break;
|
|
}
|
|
|
|
if (CONSOLE_HANDLE(hFile) && VerifyConsoleIoHandle(hFile)) {
|
|
return( FILE_TYPE_CHAR );
|
|
}
|
|
|
|
if (hFile == NULL) {
|
|
BaseSetLastNTError( STATUS_INVALID_HANDLE );
|
|
return( FILE_TYPE_UNKNOWN );
|
|
}
|
|
|
|
//
|
|
// If handle cannot be a real kernel handle we will fail
|
|
// the call instead of calling with a bogus value NtQuery.
|
|
//
|
|
|
|
if (((ULONG_PTR)hFile & 0x01)) {
|
|
|
|
BaseSetLastNTError( STATUS_INVALID_HANDLE );
|
|
return( FILE_TYPE_UNKNOWN );
|
|
}
|
|
|
|
Status = NtQueryVolumeInformationFile( hFile,
|
|
&IoStatusBlock,
|
|
&DeviceInformation,
|
|
sizeof( DeviceInformation ),
|
|
FileFsDeviceInformation
|
|
);
|
|
|
|
if (!NT_SUCCESS( Status )) {
|
|
BaseSetLastNTError( Status );
|
|
return( FILE_TYPE_UNKNOWN );
|
|
}
|
|
|
|
switch( DeviceInformation.DeviceType ) {
|
|
|
|
case FILE_DEVICE_SCREEN:
|
|
case FILE_DEVICE_KEYBOARD:
|
|
case FILE_DEVICE_MOUSE:
|
|
case FILE_DEVICE_PARALLEL_PORT:
|
|
case FILE_DEVICE_PRINTER:
|
|
case FILE_DEVICE_SERIAL_PORT:
|
|
case FILE_DEVICE_MODEM:
|
|
case FILE_DEVICE_SOUND:
|
|
case FILE_DEVICE_NULL:
|
|
return( FILE_TYPE_CHAR );
|
|
|
|
case FILE_DEVICE_CD_ROM:
|
|
case FILE_DEVICE_CD_ROM_FILE_SYSTEM:
|
|
case FILE_DEVICE_CONTROLLER:
|
|
case FILE_DEVICE_DATALINK:
|
|
case FILE_DEVICE_DFS:
|
|
case FILE_DEVICE_DISK:
|
|
case FILE_DEVICE_DISK_FILE_SYSTEM:
|
|
case FILE_DEVICE_VIRTUAL_DISK:
|
|
return( FILE_TYPE_DISK );
|
|
|
|
case FILE_DEVICE_NAMED_PIPE:
|
|
return( FILE_TYPE_PIPE );
|
|
|
|
case FILE_DEVICE_NETWORK:
|
|
case FILE_DEVICE_NETWORK_FILE_SYSTEM:
|
|
case FILE_DEVICE_PHYSICAL_NETCARD:
|
|
case FILE_DEVICE_TAPE:
|
|
case FILE_DEVICE_TAPE_FILE_SYSTEM:
|
|
case FILE_DEVICE_TRANSPORT:
|
|
// NTRAID#589623-2002/03/28-earhart - changed to NTRAID marker
|
|
// how should we handle tapes, network devices, etc...?
|
|
|
|
case FILE_DEVICE_UNKNOWN:
|
|
default:
|
|
SetLastError( NO_ERROR );
|
|
return( FILE_TYPE_UNKNOWN );
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
ReadFile(
|
|
HANDLE hFile,
|
|
LPVOID lpBuffer,
|
|
DWORD nNumberOfBytesToRead,
|
|
LPDWORD lpNumberOfBytesRead,
|
|
LPOVERLAPPED lpOverlapped
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Data can be read from a file using ReadFile.
|
|
|
|
This API is used to read data from a file. Data is read from the
|
|
file from the position indicated by the file pointer. After the
|
|
read completes, the file pointer is adjusted by the number of bytes
|
|
actually read. A return value of TRUE coupled with a bytes read of
|
|
0 indicates that the file pointer was beyond the current end of the
|
|
file at the time of the read.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to be read. The
|
|
file handle must have been created with GENERIC_READ access to
|
|
the file.
|
|
|
|
lpBuffer - Supplies the address of a buffer to receive the data read
|
|
from the file.
|
|
|
|
nNumberOfBytesToRead - Supplies the number of bytes to read from the
|
|
file.
|
|
|
|
lpNumberOfBytesRead - Returns the number of bytes read by this call.
|
|
This parameter is always set to 0 before doing any IO or error
|
|
checking.
|
|
|
|
lpOverlapped - Optionally points to an OVERLAPPED structure to be used with the
|
|
request. If NULL then the transfer starts at the current file position
|
|
and ReadFile will not return until the operation completes.
|
|
|
|
If the handle hFile was created without specifying FILE_FLAG_OVERLAPPED
|
|
the file pointer is moved to the specified offset plus
|
|
lpNumberOfBytesRead before ReadFile returns. ReadFile will wait for the
|
|
request to complete before returning (it will not return
|
|
ERROR_IO_PENDING).
|
|
|
|
When FILE_FLAG_OVERLAPPED is specified, ReadFile may return
|
|
ERROR_IO_PENDING to allow the calling function to continue processing
|
|
while the operation completes. The event (or hFile if hEvent is NULL) will
|
|
be set to the signalled state upon completion of the request.
|
|
|
|
When the handle is created with FILE_FLAG_OVERLAPPED and lpOverlapped
|
|
is set to NULL, ReadFile will return ERROR_INVALID_PARAMTER because
|
|
the file offset is required.
|
|
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successul.
|
|
|
|
FALSE - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
PPEB Peb;
|
|
DWORD InputMode;
|
|
|
|
if ( ARGUMENT_PRESENT(lpNumberOfBytesRead) ) {
|
|
*lpNumberOfBytesRead = 0;
|
|
}
|
|
|
|
Peb = NtCurrentPeb();
|
|
|
|
switch( HandleToUlong(hFile) ) {
|
|
case STD_INPUT_HANDLE: hFile = Peb->ProcessParameters->StandardInput;
|
|
break;
|
|
case STD_OUTPUT_HANDLE: hFile = Peb->ProcessParameters->StandardOutput;
|
|
break;
|
|
case STD_ERROR_HANDLE: hFile = Peb->ProcessParameters->StandardError;
|
|
break;
|
|
}
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
if (ReadConsoleA(hFile,
|
|
lpBuffer,
|
|
nNumberOfBytesToRead,
|
|
lpNumberOfBytesRead,
|
|
lpOverlapped
|
|
)
|
|
) {
|
|
Status = STATUS_SUCCESS;
|
|
if (!GetConsoleMode( hFile, &InputMode )) {
|
|
InputMode = 0;
|
|
}
|
|
|
|
if (InputMode & ENABLE_PROCESSED_INPUT) {
|
|
try {
|
|
if (*(PCHAR)lpBuffer == 0x1A) {
|
|
*lpNumberOfBytesRead = 0;
|
|
}
|
|
}
|
|
except( EXCEPTION_EXECUTE_HANDLER ) {
|
|
Status = GetExceptionCode();
|
|
}
|
|
}
|
|
|
|
if (NT_SUCCESS(Status)) {
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
else {
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
if ( ARGUMENT_PRESENT( lpOverlapped ) ) {
|
|
LARGE_INTEGER Li;
|
|
|
|
lpOverlapped->Internal = (DWORD)STATUS_PENDING;
|
|
Li.LowPart = lpOverlapped->Offset;
|
|
Li.HighPart = lpOverlapped->OffsetHigh;
|
|
Status = NtReadFile(
|
|
hFile,
|
|
lpOverlapped->hEvent,
|
|
NULL,
|
|
(ULONG_PTR)lpOverlapped->hEvent & 1 ? NULL : lpOverlapped,
|
|
(PIO_STATUS_BLOCK)&lpOverlapped->Internal,
|
|
lpBuffer,
|
|
nNumberOfBytesToRead,
|
|
&Li,
|
|
NULL
|
|
);
|
|
|
|
|
|
if ( NT_SUCCESS(Status) && Status != STATUS_PENDING) {
|
|
if ( ARGUMENT_PRESENT(lpNumberOfBytesRead) ) {
|
|
try {
|
|
*lpNumberOfBytesRead = (DWORD)lpOverlapped->InternalHigh;
|
|
}
|
|
except(EXCEPTION_EXECUTE_HANDLER) {
|
|
*lpNumberOfBytesRead = 0;
|
|
}
|
|
}
|
|
return TRUE;
|
|
}
|
|
else
|
|
if (Status == STATUS_END_OF_FILE) {
|
|
if ( ARGUMENT_PRESENT(lpNumberOfBytesRead) ) {
|
|
*lpNumberOfBytesRead = 0;
|
|
}
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Status = NtReadFile(
|
|
hFile,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&IoStatusBlock,
|
|
lpBuffer,
|
|
nNumberOfBytesToRead,
|
|
NULL,
|
|
NULL
|
|
);
|
|
|
|
if ( Status == STATUS_PENDING) {
|
|
// Operation must complete before return & IoStatusBlock destroyed
|
|
Status = NtWaitForSingleObject( hFile, FALSE, NULL );
|
|
if ( NT_SUCCESS(Status)) {
|
|
Status = IoStatusBlock.Status;
|
|
}
|
|
}
|
|
|
|
if ( NT_SUCCESS(Status) ) {
|
|
*lpNumberOfBytesRead = (DWORD)IoStatusBlock.Information;
|
|
return TRUE;
|
|
}
|
|
else
|
|
if (Status == STATUS_END_OF_FILE) {
|
|
*lpNumberOfBytesRead = 0;
|
|
return TRUE;
|
|
}
|
|
else {
|
|
if ( NT_WARNING(Status) ) {
|
|
*lpNumberOfBytesRead = (DWORD)IoStatusBlock.Information;
|
|
}
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
BOOL
|
|
WINAPI
|
|
WriteFile(
|
|
HANDLE hFile,
|
|
LPCVOID lpBuffer,
|
|
DWORD nNumberOfBytesToWrite,
|
|
LPDWORD lpNumberOfBytesWritten,
|
|
LPOVERLAPPED lpOverlapped
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Data can be written to a file using WriteFile.
|
|
|
|
This API is used to write data to a file. Data is written to the
|
|
file from the position indicated by the file pointer. After the
|
|
write completes, the file pointer is adjusted by the number of bytes
|
|
actually written.
|
|
|
|
Unlike DOS, a NumberOfBytesToWrite value of zero does not truncate
|
|
or extend the file. If this function is required, SetEndOfFile
|
|
should be used.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to be written. The
|
|
file handle must have been created with GENERIC_WRITE access to
|
|
the file.
|
|
|
|
lpBuffer - Supplies the address of the data that is to be written to
|
|
the file.
|
|
|
|
nNumberOfBytesToWrite - Supplies the number of bytes to write to the
|
|
file. Unlike DOS, a value of zero is interpreted a null write.
|
|
|
|
lpNumberOfBytesWritten - Returns the number of bytes written by this
|
|
call. Before doing any work or error processing, the API sets this
|
|
to zero.
|
|
|
|
|
|
lpOverlapped - Optionally points to an OVERLAPPED structure to be
|
|
used with the request. If NULL then the transfer starts at the
|
|
current file position and WriteFile will not return until the
|
|
operation completes.
|
|
|
|
If the handle <hFile> was created without specifying
|
|
FILE_FLAG_OVERLAPPED the file pointer is moved to the specified
|
|
offset plus lpNumberOfBytesWritten before WriteFile returns.
|
|
WriteFile will wait for the request to complete before returning
|
|
(it will not set ERROR_IO_PENDING).
|
|
|
|
When FILE_FLAG_OVERLAPPED is specified, WriteFile may return
|
|
ERROR_IO_PENDING to allow the calling function to continue processing
|
|
while the operation completes. The event (or hFile if hEvent is NULL) will
|
|
be set to the signalled state upon completion of the request.
|
|
|
|
When the handle is created with FILE_FLAG_OVERLAPPED and lpOverlapped
|
|
is set to NULL, WriteFile will return ERROR_INVALID_PARAMTER because
|
|
the file offset is required.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was a success.
|
|
|
|
FALSE - The operation failed. Extended error status is
|
|
available using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
PPEB Peb;
|
|
|
|
if ( ARGUMENT_PRESENT(lpNumberOfBytesWritten) ) {
|
|
*lpNumberOfBytesWritten = 0;
|
|
}
|
|
|
|
Peb = NtCurrentPeb();
|
|
switch( HandleToUlong(hFile) ) {
|
|
case STD_INPUT_HANDLE: hFile = Peb->ProcessParameters->StandardInput;
|
|
break;
|
|
case STD_OUTPUT_HANDLE: hFile = Peb->ProcessParameters->StandardOutput;
|
|
break;
|
|
case STD_ERROR_HANDLE: hFile = Peb->ProcessParameters->StandardError;
|
|
break;
|
|
}
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
return WriteConsoleA(hFile,
|
|
(LPVOID)lpBuffer,
|
|
nNumberOfBytesToWrite,
|
|
lpNumberOfBytesWritten,
|
|
lpOverlapped
|
|
);
|
|
}
|
|
|
|
if ( ARGUMENT_PRESENT( lpOverlapped ) ) {
|
|
LARGE_INTEGER Li;
|
|
|
|
lpOverlapped->Internal = (DWORD)STATUS_PENDING;
|
|
Li.LowPart = lpOverlapped->Offset;
|
|
Li.HighPart = lpOverlapped->OffsetHigh;
|
|
Status = NtWriteFile(
|
|
hFile,
|
|
lpOverlapped->hEvent,
|
|
NULL,
|
|
(ULONG_PTR)lpOverlapped->hEvent & 1 ? NULL : lpOverlapped,
|
|
(PIO_STATUS_BLOCK)&lpOverlapped->Internal,
|
|
(PVOID)lpBuffer,
|
|
nNumberOfBytesToWrite,
|
|
&Li,
|
|
NULL
|
|
);
|
|
|
|
if ( !NT_ERROR(Status) && Status != STATUS_PENDING) {
|
|
if ( ARGUMENT_PRESENT(lpNumberOfBytesWritten) ) {
|
|
try {
|
|
*lpNumberOfBytesWritten = (DWORD)lpOverlapped->InternalHigh;
|
|
}
|
|
except(EXCEPTION_EXECUTE_HANDLER) {
|
|
*lpNumberOfBytesWritten = 0;
|
|
}
|
|
}
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
else {
|
|
Status = NtWriteFile(
|
|
hFile,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&IoStatusBlock,
|
|
(PVOID)lpBuffer,
|
|
nNumberOfBytesToWrite,
|
|
NULL,
|
|
NULL
|
|
);
|
|
|
|
if ( Status == STATUS_PENDING) {
|
|
// Operation must complete before return & IoStatusBlock destroyed
|
|
Status = NtWaitForSingleObject( hFile, FALSE, NULL );
|
|
if ( NT_SUCCESS(Status)) {
|
|
Status = IoStatusBlock.Status;
|
|
}
|
|
}
|
|
|
|
if ( NT_SUCCESS(Status)) {
|
|
*lpNumberOfBytesWritten = (DWORD)IoStatusBlock.Information;
|
|
return TRUE;
|
|
}
|
|
else {
|
|
if ( NT_WARNING(Status) ) {
|
|
*lpNumberOfBytesWritten = (DWORD)IoStatusBlock.Information;
|
|
}
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
SetEndOfFile(
|
|
HANDLE hFile
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
The end of file position of an open file can be set to the current
|
|
file pointer using SetEndOfFile.
|
|
|
|
This API is used to set the end of file position of a file to the
|
|
same value as the current file pointer. This has the effect of
|
|
truncating or extending a file. This functionality is similar to
|
|
DOS (int 21h, function 40H with CX=0).
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to be extended or
|
|
truncated. The file handle must have been created with
|
|
GENERIC_WRITE access to the file.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful.
|
|
|
|
FALSE/NULL - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
FILE_POSITION_INFORMATION CurrentPosition;
|
|
FILE_END_OF_FILE_INFORMATION EndOfFile;
|
|
FILE_ALLOCATION_INFORMATION Allocation;
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
//
|
|
// Get the current position of the file pointer
|
|
//
|
|
|
|
Status = NtQueryInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&CurrentPosition,
|
|
sizeof(CurrentPosition),
|
|
FilePositionInformation
|
|
);
|
|
if ( !NT_SUCCESS(Status) ) {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
//
|
|
// Set the end of file based on the current file position
|
|
//
|
|
|
|
EndOfFile.EndOfFile = CurrentPosition.CurrentByteOffset;
|
|
|
|
Status = NtSetInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&EndOfFile,
|
|
sizeof(EndOfFile),
|
|
FileEndOfFileInformation
|
|
);
|
|
if ( !NT_SUCCESS(Status) ) {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
//
|
|
// Set the allocation based on the current file size
|
|
//
|
|
|
|
Allocation.AllocationSize = CurrentPosition.CurrentByteOffset;
|
|
|
|
Status = NtSetInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&Allocation,
|
|
sizeof(Allocation),
|
|
FileAllocationInformation
|
|
);
|
|
if ( NT_SUCCESS(Status) ) {
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
DWORD
|
|
WINAPI
|
|
SetFilePointer(
|
|
HANDLE hFile,
|
|
LONG lDistanceToMove,
|
|
PLONG lpDistanceToMoveHigh,
|
|
DWORD dwMoveMethod
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
An open file's file pointer can be set using SetFilePointer.
|
|
|
|
The purpose of this function is to update the current value of a
|
|
file's file pointer. Care should be taken in multi-threaded
|
|
applications that have multiple threads sharing a file handle with
|
|
each thread updating the file pointer and then doing a read. This
|
|
sequence should be treated as a critical section of code and should
|
|
be protected using either a critical section object or a mutex
|
|
object.
|
|
|
|
This API provides the same functionality as DOS (int 21h, function
|
|
42h) and OS/2's DosSetFilePtr.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose file pointer is to be
|
|
moved. The file handle must have been created with
|
|
GENERIC_READ or GENERIC_WRITE access to the file.
|
|
|
|
lDistanceToMove - Supplies the number of bytes to move the file
|
|
pointer. A positive value moves the pointer forward in the file
|
|
and a negative value moves backwards in the file.
|
|
|
|
lpDistanceToMoveHigh - An optional parameter that if specified
|
|
supplies the high order 32-bits of the 64-bit distance to move.
|
|
If the value of this parameter is NULL, this API can only
|
|
operate on files whose maximum size is (2**32)-2. If this
|
|
parameter is specified, than the maximum file size is (2**64)-2.
|
|
This value also returns the high order 32-bits of the new value
|
|
of the file pointer. If this value, and the return value
|
|
are 0xffffffff, then an error is indicated.
|
|
|
|
dwMoveMethod - Supplies a value that specifies the starting point
|
|
for the file pointer move.
|
|
|
|
FILE_BEGIN - The starting point is zero or the beginning of the
|
|
file. If FILE_BEGIN is specified, then DistanceToMove is
|
|
interpreted as an unsigned location for the new
|
|
file pointer.
|
|
|
|
FILE_CURRENT - The current value of the file pointer is used as
|
|
the starting point.
|
|
|
|
FILE_END - The current end of file position is used as the
|
|
starting point.
|
|
|
|
|
|
Return Value:
|
|
|
|
Not -1 - Returns the low order 32-bits of the new value of the file
|
|
pointer.
|
|
|
|
0xffffffff - If the value of lpDistanceToMoveHigh was NULL, then The
|
|
operation failed. Extended error status is available using
|
|
GetLastError. Otherwise, this is the low order 32-bits of the
|
|
new value of the file pointer.
|
|
|
|
--*/
|
|
|
|
{
|
|
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
FILE_POSITION_INFORMATION CurrentPosition;
|
|
FILE_STANDARD_INFORMATION StandardInfo;
|
|
LARGE_INTEGER Large;
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return (DWORD)-1;
|
|
}
|
|
|
|
if (ARGUMENT_PRESENT(lpDistanceToMoveHigh)) {
|
|
Large.HighPart = *lpDistanceToMoveHigh;
|
|
Large.LowPart = lDistanceToMove;
|
|
}
|
|
else {
|
|
Large.QuadPart = lDistanceToMove;
|
|
}
|
|
switch (dwMoveMethod) {
|
|
case FILE_BEGIN :
|
|
CurrentPosition.CurrentByteOffset = Large;
|
|
break;
|
|
|
|
case FILE_CURRENT :
|
|
|
|
//
|
|
// Get the current position of the file pointer
|
|
//
|
|
|
|
Status = NtQueryInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&CurrentPosition,
|
|
sizeof(CurrentPosition),
|
|
FilePositionInformation
|
|
);
|
|
if ( !NT_SUCCESS(Status) ) {
|
|
BaseSetLastNTError(Status);
|
|
return (DWORD)-1;
|
|
}
|
|
CurrentPosition.CurrentByteOffset.QuadPart += Large.QuadPart;
|
|
break;
|
|
|
|
case FILE_END :
|
|
Status = NtQueryInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&StandardInfo,
|
|
sizeof(StandardInfo),
|
|
FileStandardInformation
|
|
);
|
|
if ( !NT_SUCCESS(Status) ) {
|
|
BaseSetLastNTError(Status);
|
|
return (DWORD)-1;
|
|
}
|
|
CurrentPosition.CurrentByteOffset.QuadPart =
|
|
StandardInfo.EndOfFile.QuadPart + Large.QuadPart;
|
|
break;
|
|
|
|
default:
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return (DWORD)-1;
|
|
break;
|
|
}
|
|
|
|
//
|
|
// If the resulting file position is negative, or if the app is not
|
|
// prepared for greater than
|
|
// then 32 bits than fail
|
|
//
|
|
|
|
if ( CurrentPosition.CurrentByteOffset.QuadPart < 0 ) {
|
|
SetLastError(ERROR_NEGATIVE_SEEK);
|
|
return (DWORD)-1;
|
|
}
|
|
if ( !ARGUMENT_PRESENT(lpDistanceToMoveHigh) &&
|
|
(CurrentPosition.CurrentByteOffset.HighPart & MAXLONG) ) {
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return (DWORD)-1;
|
|
}
|
|
|
|
|
|
//
|
|
// Set the current file position
|
|
//
|
|
|
|
Status = NtSetInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&CurrentPosition,
|
|
sizeof(CurrentPosition),
|
|
FilePositionInformation
|
|
);
|
|
if ( NT_SUCCESS(Status) ) {
|
|
if (ARGUMENT_PRESENT(lpDistanceToMoveHigh)){
|
|
*lpDistanceToMoveHigh = CurrentPosition.CurrentByteOffset.HighPart;
|
|
}
|
|
if ( CurrentPosition.CurrentByteOffset.LowPart == -1 ) {
|
|
SetLastError(0);
|
|
}
|
|
return CurrentPosition.CurrentByteOffset.LowPart;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
if (ARGUMENT_PRESENT(lpDistanceToMoveHigh)){
|
|
*lpDistanceToMoveHigh = -1;
|
|
}
|
|
return (DWORD)-1;
|
|
}
|
|
}
|
|
|
|
|
|
BOOL
|
|
WINAPI
|
|
SetFilePointerEx(
|
|
HANDLE hFile,
|
|
LARGE_INTEGER liDistanceToMove,
|
|
PLARGE_INTEGER lpNewFilePointer,
|
|
DWORD dwMoveMethod
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
An open file's file pointer can be set using SetFilePointer.
|
|
|
|
The purpose of this function is to update the current value of a
|
|
file's file pointer. Care should be taken in multi-threaded
|
|
applications that have multiple threads sharing a file handle with
|
|
each thread updating the file pointer and then doing a read. This
|
|
sequence should be treated as a critical section of code and should
|
|
be protected using either a critical section object or a mutex
|
|
object.
|
|
|
|
This API provides the same functionality as DOS (int 21h, function
|
|
42h) and OS/2's DosSetFilePtr.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose file pointer is to be
|
|
moved. The file handle must have been created with
|
|
GENERIC_READ or GENERIC_WRITE access to the file.
|
|
|
|
liDistanceToMove - Supplies the number of bytes to move the file
|
|
pointer. A positive value moves the pointer forward in the file
|
|
and a negative value moves backwards in the file.
|
|
|
|
lpNewFilePointer - An optional parameter that if specified returns
|
|
the new file pointer
|
|
|
|
dwMoveMethod - Supplies a value that specifies the starting point
|
|
for the file pointer move.
|
|
|
|
FILE_BEGIN - The starting point is zero or the beginning of the
|
|
file. If FILE_BEGIN is specified, then DistanceToMove is
|
|
interpreted as an unsigned location for the new
|
|
file pointer.
|
|
|
|
FILE_CURRENT - The current value of the file pointer is used as
|
|
the starting point.
|
|
|
|
FILE_END - The current end of file position is used as the
|
|
starting point.
|
|
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful
|
|
|
|
FALSE - The operation failed. Extended error status is available using
|
|
GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
FILE_POSITION_INFORMATION CurrentPosition;
|
|
FILE_STANDARD_INFORMATION StandardInfo;
|
|
LARGE_INTEGER Large;
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
Large = liDistanceToMove;
|
|
|
|
switch (dwMoveMethod) {
|
|
case FILE_BEGIN :
|
|
CurrentPosition.CurrentByteOffset = Large;
|
|
break;
|
|
|
|
case FILE_CURRENT :
|
|
|
|
//
|
|
// Get the current position of the file pointer
|
|
//
|
|
|
|
Status = NtQueryInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&CurrentPosition,
|
|
sizeof(CurrentPosition),
|
|
FilePositionInformation
|
|
);
|
|
if ( !NT_SUCCESS(Status) ) {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
CurrentPosition.CurrentByteOffset.QuadPart += Large.QuadPart;
|
|
break;
|
|
|
|
case FILE_END :
|
|
Status = NtQueryInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&StandardInfo,
|
|
sizeof(StandardInfo),
|
|
FileStandardInformation
|
|
);
|
|
if ( !NT_SUCCESS(Status) ) {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
CurrentPosition.CurrentByteOffset.QuadPart =
|
|
StandardInfo.EndOfFile.QuadPart + Large.QuadPart;
|
|
break;
|
|
|
|
default:
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
break;
|
|
}
|
|
|
|
//
|
|
// If the resulting file position is negative fail
|
|
//
|
|
|
|
if ( CurrentPosition.CurrentByteOffset.QuadPart < 0 ) {
|
|
SetLastError(ERROR_NEGATIVE_SEEK);
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
//
|
|
// Set the current file position
|
|
//
|
|
|
|
Status = NtSetInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&CurrentPosition,
|
|
sizeof(CurrentPosition),
|
|
FilePositionInformation
|
|
);
|
|
if ( NT_SUCCESS(Status) ) {
|
|
if (ARGUMENT_PRESENT(lpNewFilePointer)){
|
|
*lpNewFilePointer = CurrentPosition.CurrentByteOffset;
|
|
}
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
BOOL
|
|
WINAPI
|
|
GetFileInformationByHandle(
|
|
HANDLE hFile,
|
|
LPBY_HANDLE_FILE_INFORMATION lpFileInformation
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose modification date and
|
|
times are to be read. The file handle must have been created with
|
|
GENERIC_READ access to the file.
|
|
|
|
lpCreationTime - An optional parameter that if specified points to
|
|
the location to return the date and time the file was created.
|
|
A returned time of all zero indicates that the file system
|
|
containing the file does not support this time value.
|
|
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful.
|
|
|
|
FALSE/NULL - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
BY_HANDLE_FILE_INFORMATION LocalFileInformation;
|
|
FILE_ALL_INFORMATION FileInformation;
|
|
FILE_FS_VOLUME_INFORMATION VolumeInfo;
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
Status = NtQueryVolumeInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&VolumeInfo,
|
|
sizeof(VolumeInfo),
|
|
FileFsVolumeInformation
|
|
);
|
|
if ( !NT_ERROR(Status) ) {
|
|
LocalFileInformation.dwVolumeSerialNumber = VolumeInfo.VolumeSerialNumber;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
Status = NtQueryInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&FileInformation,
|
|
sizeof(FileInformation),
|
|
FileAllInformation
|
|
);
|
|
|
|
//
|
|
// we really plan for buffer overflow
|
|
//
|
|
|
|
if ( !NT_ERROR(Status) ) {
|
|
LocalFileInformation.dwFileAttributes = FileInformation.BasicInformation.FileAttributes;
|
|
LocalFileInformation.ftCreationTime = *(LPFILETIME)&FileInformation.BasicInformation.CreationTime;
|
|
LocalFileInformation.ftLastAccessTime = *(LPFILETIME)&FileInformation.BasicInformation.LastAccessTime;
|
|
LocalFileInformation.ftLastWriteTime = *(LPFILETIME)&FileInformation.BasicInformation.LastWriteTime;
|
|
LocalFileInformation.nFileSizeHigh = FileInformation.StandardInformation.EndOfFile.HighPart;
|
|
LocalFileInformation.nFileSizeLow = FileInformation.StandardInformation.EndOfFile.LowPart;
|
|
LocalFileInformation.nNumberOfLinks = FileInformation.StandardInformation.NumberOfLinks;
|
|
LocalFileInformation.nFileIndexHigh = FileInformation.InternalInformation.IndexNumber.HighPart;
|
|
LocalFileInformation.nFileIndexLow = FileInformation.InternalInformation.IndexNumber.LowPart;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
*lpFileInformation = LocalFileInformation;
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
APIENTRY
|
|
GetFileTime(
|
|
HANDLE hFile,
|
|
LPFILETIME lpCreationTime,
|
|
LPFILETIME lpLastAccessTime,
|
|
LPFILETIME lpLastWriteTime
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
The date and time that a file was created, last accessed or last
|
|
modified can be read using GetFileTime. File time stamps are
|
|
returned as 64-bit values, that represent the number of 100
|
|
nanoseconds since January 1st, 1601. This date was chosen because
|
|
it is the start of a new quadricentury. At 100ns resolution 32 bits
|
|
is good for about 429 seconds (or 7 minutes) and a 63-bit integer is
|
|
good for about 29,247 years, or around 10,682,247 days.
|
|
|
|
This API provides the same functionality as DOS (int 21h, function
|
|
47H with AL=0), and provides a subset of OS/2's DosQueryFileInfo.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose modification date and
|
|
times are to be read. The file handle must have been created with
|
|
GENERIC_READ access to the file.
|
|
|
|
lpCreationTime - An optional parameter that if specified points to
|
|
the location to return the date and time the file was created.
|
|
A returned time of all zero indicates that the file system
|
|
containing the file does not support this time value.
|
|
|
|
lpLastAccessTime - An optional parameter that if specified points to
|
|
the location to return the date and time the file was last accessed.
|
|
A returned time of all zero indicates that the file system
|
|
containing the file does not support this time value.
|
|
|
|
lpLastWriteTime - An optional parameter that if specified points to
|
|
the location to return the date and time the file was last written.
|
|
A file system must support this time and thus a valid value will
|
|
always be returned for this time value.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful.
|
|
|
|
FALSE/NULL - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
FILE_BASIC_INFORMATION BasicInfo;
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
//
|
|
// Get the attributes
|
|
//
|
|
|
|
Status = NtQueryInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&BasicInfo,
|
|
sizeof(BasicInfo),
|
|
FileBasicInformation
|
|
);
|
|
|
|
if ( NT_SUCCESS(Status) ) {
|
|
if (ARGUMENT_PRESENT( lpCreationTime )) {
|
|
*lpCreationTime = *(LPFILETIME)&BasicInfo.CreationTime;
|
|
}
|
|
|
|
if (ARGUMENT_PRESENT( lpLastAccessTime )) {
|
|
*lpLastAccessTime = *(LPFILETIME)&BasicInfo.LastAccessTime;
|
|
}
|
|
|
|
if (ARGUMENT_PRESENT( lpLastWriteTime )) {
|
|
*lpLastWriteTime = *(LPFILETIME)&BasicInfo.LastWriteTime;
|
|
}
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
SetFileTime(
|
|
HANDLE hFile,
|
|
CONST FILETIME *lpCreationTime,
|
|
CONST FILETIME *lpLastAccessTime,
|
|
CONST FILETIME *lpLastWriteTime
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
The date and time that a file was created, last accessed or last
|
|
modified can be modified using SetFileTime. File time stamps are
|
|
returned as 64-bit values, that represent the number of 100
|
|
nanoseconds since January 1st, 1601. This date was chosen because
|
|
it is the start of a new quadricentury. At 100ns resolution 32 bits
|
|
is good for about 429 seconds (or 7 minutes) and a 63-bit integer is
|
|
good for about 29,247 years, or around 10,682,247 days.
|
|
|
|
This API provides the same functionality as DOS (int 21h, function
|
|
47H with AL=1), and provides a subset of OS/2's DosSetFileInfo.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose modification date and
|
|
times are to be written. The file handle must have been created
|
|
with GENERIC_WRITE access to the file.
|
|
|
|
lpCreationTime - An optional parameter, that if specified supplies
|
|
the new creation time for the file. Some file system's do not
|
|
support this time value, so this parameter may be ignored.
|
|
|
|
lpLastAccessTime - An optional parameter, that if specified supplies
|
|
the new last access time for the file. Some file system's do
|
|
not support this time value, so this parameter may be ignored.
|
|
|
|
lpLastWriteTime - An optional parameter, that if specified supplies
|
|
the new last write time for the file. A file system must support
|
|
this time value.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful.
|
|
|
|
FALSE/NULL - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
FILE_BASIC_INFORMATION BasicInfo;
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
//
|
|
// Zero all the time values we can set.
|
|
//
|
|
|
|
RtlZeroMemory(&BasicInfo,sizeof(BasicInfo));
|
|
|
|
//
|
|
// For each time value that is specified, copy it to the I/O system
|
|
// record.
|
|
//
|
|
if (ARGUMENT_PRESENT( lpCreationTime )) {
|
|
BasicInfo.CreationTime.LowPart = lpCreationTime->dwLowDateTime;
|
|
BasicInfo.CreationTime.HighPart = lpCreationTime->dwHighDateTime;
|
|
}
|
|
|
|
if (ARGUMENT_PRESENT( lpLastAccessTime )) {
|
|
BasicInfo.LastAccessTime.LowPart = lpLastAccessTime->dwLowDateTime;
|
|
BasicInfo.LastAccessTime.HighPart = lpLastAccessTime->dwHighDateTime;
|
|
}
|
|
|
|
if (ARGUMENT_PRESENT( lpLastWriteTime )) {
|
|
BasicInfo.LastWriteTime.LowPart = lpLastWriteTime->dwLowDateTime;
|
|
BasicInfo.LastWriteTime.HighPart = lpLastWriteTime->dwHighDateTime;
|
|
}
|
|
|
|
//
|
|
// Set the requested times.
|
|
//
|
|
|
|
Status = NtSetInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&BasicInfo,
|
|
sizeof(BasicInfo),
|
|
FileBasicInformation
|
|
);
|
|
|
|
if ( NT_SUCCESS(Status) ) {
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
FlushFileBuffers(
|
|
HANDLE hFile
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Buffered data may be flushed out to the file using the
|
|
FlushFileBuffers service.
|
|
|
|
The FlushFileBuffers service causes all buffered data to be written
|
|
to the specified file.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose buffers are to be
|
|
flushed. The file handle must have been created with
|
|
GENERIC_WRITE access to the file.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful.
|
|
|
|
FALSE/NULL - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
PPEB Peb;
|
|
|
|
Peb = NtCurrentPeb();
|
|
|
|
switch( HandleToUlong(hFile) ) {
|
|
case STD_INPUT_HANDLE: hFile = Peb->ProcessParameters->StandardInput;
|
|
break;
|
|
case STD_OUTPUT_HANDLE: hFile = Peb->ProcessParameters->StandardOutput;
|
|
break;
|
|
case STD_ERROR_HANDLE: hFile = Peb->ProcessParameters->StandardError;
|
|
break;
|
|
}
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
return( FlushConsoleInputBuffer( hFile ) );
|
|
}
|
|
|
|
Status = NtFlushBuffersFile(hFile,&IoStatusBlock);
|
|
|
|
if ( NT_SUCCESS(Status) ) {
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
LockFile(
|
|
HANDLE hFile,
|
|
DWORD dwFileOffsetLow,
|
|
DWORD dwFileOffsetHigh,
|
|
DWORD nNumberOfBytesToLockLow,
|
|
DWORD nNumberOfBytesToLockHigh
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
A byte range within an open file may be locked for exclusive access
|
|
using LockFile.
|
|
|
|
Locking a region of a file is used to aquire exclusive access to the
|
|
specified region of the file. File locks are not inherited by the
|
|
new process during process creation.
|
|
|
|
Locking a portion of a file denies all other processes both read and
|
|
write access to the specified region of the file. Locking a region
|
|
that goes beyond the current end-of-file position is not an error.
|
|
|
|
Locks may not overlap an existing locked region of the file.
|
|
|
|
For DOS based systems running share.exe the lock semantics work as
|
|
described above. Without share.exe, all attempts to lock or unlock
|
|
a file will fail.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to have a range of
|
|
bytes locked for exclusive access. The handle must have been
|
|
created with either GENERIC_READ or GENERIC_WRITE access to the
|
|
file.
|
|
|
|
dwFileOffsetLow - Supplies the low order 32-bits of the starting
|
|
byte offset of the file where the lock should begin.
|
|
|
|
dwFileOffsetHigh - Supplies the high order 32-bits of the starting
|
|
byte offset of the file where the lock should begin.
|
|
|
|
nNumberOfBytesToLockLow - Supplies the low order 32-bits of the length
|
|
of the byte range to be locked.
|
|
|
|
nNumberOfBytesToLockHigh - Supplies the high order 32-bits of the length
|
|
of the byte range to be locked.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful.
|
|
|
|
FALSE/NULL - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
LARGE_INTEGER ByteOffset;
|
|
LARGE_INTEGER Length;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
ByteOffset.LowPart = dwFileOffsetLow;
|
|
ByteOffset.HighPart = dwFileOffsetHigh;
|
|
|
|
Length.LowPart = nNumberOfBytesToLockLow;
|
|
Length.HighPart = nNumberOfBytesToLockHigh;
|
|
|
|
Status = NtLockFile( hFile,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&IoStatusBlock,
|
|
&ByteOffset,
|
|
&Length,
|
|
0,
|
|
TRUE,
|
|
TRUE
|
|
);
|
|
|
|
if (Status == STATUS_PENDING) {
|
|
|
|
Status = NtWaitForSingleObject( hFile, FALSE, NULL );
|
|
if (NT_SUCCESS( Status )) {
|
|
Status = IoStatusBlock.Status;
|
|
}
|
|
}
|
|
|
|
if ( NT_SUCCESS(Status) ) {
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
|
|
BOOL
|
|
WINAPI
|
|
LockFileEx(
|
|
HANDLE hFile,
|
|
DWORD dwFlags,
|
|
DWORD dwReserved,
|
|
DWORD nNumberOfBytesToLockLow,
|
|
DWORD nNumberOfBytesToLockHigh,
|
|
LPOVERLAPPED lpOverlapped
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
A byte range within an open file may be locked for shared or
|
|
exclusive access using LockFileEx.
|
|
|
|
Locking a region of a file is used to aquire shared or exclusive
|
|
access to the specified region of the file. File locks are not
|
|
inherited by the new process during process creation.
|
|
|
|
Locking a portion of a file for exclusive access denies all other
|
|
processes both read and write access to the specified region of the
|
|
file. Locking a region that goes beyond the current end-of-file
|
|
position is not an error.
|
|
|
|
Locking a portion of a file for shared access denies all other
|
|
processes write access to the specified region of the file, but
|
|
allows other processes to read the locked region.
|
|
|
|
If requesting an exclusive lock for a file that is already locked
|
|
shared or exclusively by other threads, then this call will wait
|
|
until the lock is granted unless the LOCKFILE_FAIL_IMMEDIATELY
|
|
flag is specified.
|
|
|
|
Locks may not overlap an existing locked region of the file.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to have a range of
|
|
bytes locked for exclusive access. The handle must have been
|
|
created with either GENERIC_READ or GENERIC_WRITE access to the
|
|
file.
|
|
|
|
dwFlags - Supplies flag bits that modify the behavior of this function.
|
|
|
|
LOCKFILE_FAIL_IMMEDIATELY - if set, then this function will return
|
|
immediately if it is unable to acquire the requested lock.
|
|
Otherwise it will wait.
|
|
|
|
LOCKFILE_EXCLUSIVE_LOCK - if set, then this function requests an
|
|
exclusive lock, otherwise it requested a shared lock.
|
|
|
|
dwReserved - Reserved parameter that must be zero.
|
|
|
|
nNumberOfBytesToLockLow - Supplies the low order 32-bits of the length
|
|
of the byte range to be locked.
|
|
|
|
nNumberOfBytesToLockHigh - Supplies the high order 32-bits of the length
|
|
of the byte range to be locked.
|
|
|
|
lpOverlapped - Required pointer to an OVERLAPPED structure to be
|
|
used with the request. It contains the file offset of the
|
|
beginning of the lock range.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful.
|
|
|
|
FALSE/NULL - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
LARGE_INTEGER ByteOffset;
|
|
LARGE_INTEGER Length;
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
if (dwReserved != 0) {
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
}
|
|
|
|
ByteOffset.LowPart = lpOverlapped->Offset;
|
|
ByteOffset.HighPart = lpOverlapped->OffsetHigh;
|
|
|
|
Length.LowPart = nNumberOfBytesToLockLow;
|
|
Length.HighPart = nNumberOfBytesToLockHigh;
|
|
lpOverlapped->Internal = (DWORD)STATUS_PENDING;
|
|
|
|
Status = NtLockFile( hFile,
|
|
lpOverlapped->hEvent,
|
|
NULL,
|
|
(ULONG_PTR)lpOverlapped->hEvent & 1 ? NULL : lpOverlapped,
|
|
(PIO_STATUS_BLOCK)&lpOverlapped->Internal,
|
|
&ByteOffset,
|
|
&Length,
|
|
0,
|
|
(BOOLEAN)((dwFlags & LOCKFILE_FAIL_IMMEDIATELY) ? TRUE : FALSE),
|
|
(BOOLEAN)((dwFlags & LOCKFILE_EXCLUSIVE_LOCK) ? TRUE : FALSE)
|
|
);
|
|
|
|
if ( NT_SUCCESS(Status) && Status != STATUS_PENDING) {
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
|
|
BOOL
|
|
WINAPI
|
|
UnlockFile(
|
|
HANDLE hFile,
|
|
DWORD dwFileOffsetLow,
|
|
DWORD dwFileOffsetHigh,
|
|
DWORD nNumberOfBytesToUnlockLow,
|
|
DWORD nNumberOfBytesToUnlockHigh
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
A previously locked byte range within an open file may be Unlocked
|
|
using UnlockFile.
|
|
|
|
Unlocking a region of a file is used release a previously aquired
|
|
lock on a file. The region to unlock must exactly correspond to an
|
|
existing locked region. Two adjacent regions of a file can not be
|
|
locked seperately and then be unlocked using a single region that
|
|
spans both locked regions.
|
|
|
|
If a process terminates with a portion of a file locked, or closes a
|
|
file that has outstanding locks, the behavior is not specified.
|
|
|
|
For DOS based systems running share.exe the lock semantics work as
|
|
described above. Without share.exe, all attempts to lock or unlock
|
|
a file will fail.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to have an
|
|
existing locked region unlocked. The handle must have been
|
|
created with either GENERIC_READ or GENERIC_WRITE access to the
|
|
file.
|
|
|
|
dwFileOffsetLow - Supplies the low order 32-bits of an existing
|
|
locked region to be unlocked.
|
|
|
|
dwFileOffsetHigh - Supplies the high order 32-bits of an existing
|
|
locked region to be unlocked.
|
|
|
|
nNumberOfBytesToUnlockLow - Supplies the low order 32-bits of the
|
|
length of the byte range to be unlocked.
|
|
|
|
nNumberOfBytesToUnlockHigh - Supplies the high order 32-bits of the
|
|
length of the byte range to be unlocked.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful.
|
|
|
|
FALSE/NULL - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
BOOL bResult;
|
|
OVERLAPPED Overlapped;
|
|
NTSTATUS Status;
|
|
|
|
Overlapped.Offset = dwFileOffsetLow;
|
|
Overlapped.OffsetHigh = dwFileOffsetHigh;
|
|
bResult = UnlockFileEx( hFile,
|
|
0,
|
|
nNumberOfBytesToUnlockLow,
|
|
nNumberOfBytesToUnlockHigh,
|
|
&Overlapped
|
|
);
|
|
if (!bResult && GetLastError() == ERROR_IO_PENDING) {
|
|
Status = NtWaitForSingleObject( hFile, FALSE, NULL );
|
|
if (NT_SUCCESS( Status )) {
|
|
Status = (NTSTATUS)Overlapped.Internal;
|
|
}
|
|
|
|
if ( NT_SUCCESS(Status) ) {
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
else {
|
|
return bResult;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
BOOL
|
|
WINAPI
|
|
UnlockFileEx(
|
|
HANDLE hFile,
|
|
DWORD dwReserved,
|
|
DWORD nNumberOfBytesToUnlockLow,
|
|
DWORD nNumberOfBytesToUnlockHigh,
|
|
LPOVERLAPPED lpOverlapped
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
A previously locked byte range within an open file may be Unlocked
|
|
using UnlockFile.
|
|
|
|
Unlocking a region of a file is used release a previously aquired
|
|
lock on a file. The region to unlock must exactly correspond to an
|
|
existing locked region. Two adjacent regions of a file can not be
|
|
locked seperately and then be unlocked using a single region that
|
|
spans both locked regions.
|
|
|
|
If a process terminates with a portion of a file locked, or closes a
|
|
file that has outstanding locks, the behavior is not specified.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to have an
|
|
existing locked region unlocked. The handle must have been
|
|
created with either GENERIC_READ or GENERIC_WRITE access to the
|
|
file.
|
|
|
|
dwReserved - Reserved parameter that must be zero.
|
|
|
|
nNumberOfBytesToUnlockLow - Supplies the low order 32-bits of the
|
|
length of the byte range to be unlocked.
|
|
|
|
nNumberOfBytesToUnlockHigh - Supplies the high order 32-bits of the
|
|
length of the byte range to be unlocked.
|
|
|
|
lpOverlapped - Required pointer to an OVERLAPPED structure to be
|
|
used with the request. It contains the file offset of the
|
|
beginning of the lock range.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful.
|
|
|
|
FALSE/NULL - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
LARGE_INTEGER ByteOffset;
|
|
LARGE_INTEGER Length;
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
if (dwReserved != 0) {
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
}
|
|
|
|
ByteOffset.LowPart = lpOverlapped->Offset;
|
|
ByteOffset.HighPart = lpOverlapped->OffsetHigh;
|
|
|
|
Length.LowPart = nNumberOfBytesToUnlockLow;
|
|
Length.HighPart = nNumberOfBytesToUnlockHigh;
|
|
|
|
Status = NtUnlockFile(
|
|
hFile,
|
|
(PIO_STATUS_BLOCK)&lpOverlapped->Internal,
|
|
&ByteOffset,
|
|
&Length,
|
|
0
|
|
);
|
|
|
|
if ( NT_SUCCESS(Status) ) {
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
UINT
|
|
WINAPI
|
|
SetHandleCount(
|
|
UINT uNumber
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function changes the number of file handles available to a
|
|
process. For DOS based Win32, the default maximum number of file
|
|
handles available to a process is 20. For NT/Win32 systems, this
|
|
API has no effect.
|
|
|
|
Arguments:
|
|
|
|
uNumber - Specifies the number of file handles needed by the
|
|
application. The maximum is 255.
|
|
|
|
Return Value:
|
|
|
|
The return value specifies the number of file handles actually
|
|
available to the application. It may be less than the number
|
|
specified by the wNumber parameter.
|
|
|
|
--*/
|
|
|
|
{
|
|
return uNumber;
|
|
}
|
|
|
|
DWORD
|
|
WINAPI
|
|
GetFileSize(
|
|
HANDLE hFile,
|
|
LPDWORD lpFileSizeHigh
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function returns the size of the file specified by
|
|
hFile. It is capable of returning 64-bits worth of file size.
|
|
|
|
The return value contains the low order 32-bits of the file's size.
|
|
The optional lpFileSizeHigh returns the high order 32-bits of the
|
|
file's size.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose size is to be
|
|
returned. The handle must have been created with either
|
|
GENERIC_READ or GENERIC_WRITE access to the file.
|
|
|
|
lpFileSizeHigh - An optional parameter, that if specified, returns
|
|
the high order 64-bits of the file's size.
|
|
|
|
|
|
Return Value:
|
|
|
|
Not -1 - Returns the low order 32-bits of the specified file's size.
|
|
|
|
|
|
0xffffffff - If the value of size of the file cannot be determined,
|
|
or an invalid handle or handle with inappropriate access, or a
|
|
handle to a non-file is specified, this error is returned. If
|
|
the file's size (low 32-bits) is -1, then this value is
|
|
returned, and GetLastError() will return 0. Extended error
|
|
status is available using GetLastError.
|
|
|
|
|
|
--*/
|
|
|
|
{
|
|
BOOL b;
|
|
LARGE_INTEGER Li;
|
|
|
|
b = GetFileSizeEx(hFile,&Li);
|
|
|
|
if ( b ) {
|
|
|
|
if ( ARGUMENT_PRESENT(lpFileSizeHigh) ) {
|
|
*lpFileSizeHigh = (DWORD)Li.HighPart;
|
|
}
|
|
if (Li.LowPart == -1 ) {
|
|
SetLastError(0);
|
|
}
|
|
}
|
|
else {
|
|
Li.LowPart = -1;
|
|
}
|
|
|
|
return Li.LowPart;
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
GetFileSizeEx(
|
|
HANDLE hFile,
|
|
PLARGE_INTEGER lpFileSize
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function returns the size of the file specified by
|
|
hFile. It is capable of returning 64-bits worth of file size.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose size is to be
|
|
returned. The handle must have been created with either
|
|
GENERIC_READ or GENERIC_WRITE access to the file.
|
|
|
|
lpFileSize - Returns the files size
|
|
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successful
|
|
|
|
|
|
FALSE - The operation failed. Extended error
|
|
status is available using GetLastError.
|
|
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
FILE_STANDARD_INFORMATION StandardInfo;
|
|
|
|
Status = NtQueryInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&StandardInfo,
|
|
sizeof(StandardInfo),
|
|
FileStandardInformation
|
|
);
|
|
if ( !NT_SUCCESS(Status) ) {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
else {
|
|
*lpFileSize = StandardInfo.EndOfFile;
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
VOID
|
|
WINAPI
|
|
BasepIoCompletion(
|
|
PVOID ApcContext,
|
|
PIO_STATUS_BLOCK IoStatusBlock,
|
|
DWORD Reserved
|
|
)
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This procedure is called to complete ReadFileEx and WriteFileEx
|
|
asynchronous I/O. Its primary function is to extract the
|
|
appropriate information from the passed IoStatusBlock and call the
|
|
users completion routine.
|
|
|
|
The users completion routine is called as:
|
|
|
|
Routine Description:
|
|
|
|
When an outstanding I/O completes with a callback, this
|
|
function is called. This function is only called while the
|
|
thread is in an alertable wait (SleepEx,
|
|
WaitForSingleObjectEx, or WaitForMultipleObjectsEx with the
|
|
bAlertable flag set to TRUE). Returning from this function
|
|
allows another pendiong I/O completion callback to be
|
|
processed. If this is the case, this callback is entered
|
|
before the termination of the thread's wait with a return
|
|
code of WAIT_IO_COMPLETION.
|
|
|
|
Note that each time your completion routine is called, the
|
|
system uses some of your stack. If you code your completion
|
|
logic to do additional ReadFileEx's and WriteFileEx's within
|
|
your completion routine, AND you do alertable waits in your
|
|
completion routine, you may grow your stack without ever
|
|
trimming it back.
|
|
|
|
Arguments:
|
|
|
|
dwErrorCode - Supplies the I/O completion status for the
|
|
related I/O. A value of 0 indicates that the I/O was
|
|
successful. Note that end of file is indicated by a
|
|
non-zero dwErrorCode value of ERROR_HANDLE_EOF.
|
|
|
|
dwNumberOfBytesTransfered - Supplies the number of bytes
|
|
transfered during the associated I/O. If an error
|
|
occured, a value of 0 is supplied.
|
|
|
|
lpOverlapped - Supplies the address of the OVERLAPPED
|
|
structure used to initiate the associated I/O. The
|
|
hEvent field of this structure is not used by the system
|
|
and may be used by the application to provide additional
|
|
I/O context. Once a completion routine is called, the
|
|
system will not use the OVERLAPPED structure. The
|
|
completion routine is free to deallocate the overlapped
|
|
structure.
|
|
|
|
Arguments:
|
|
|
|
ApcContext - Supplies the users completion routine. The format of
|
|
this routine is an LPOVERLAPPED_COMPLETION_ROUTINE.
|
|
|
|
IoStatusBlock - Supplies the address of the IoStatusBlock that
|
|
contains the I/O completion status. The IoStatusBlock is
|
|
contained within the OVERLAPPED structure.
|
|
|
|
Reserved - Not used; reserved for future use.
|
|
|
|
Return Value:
|
|
|
|
None.
|
|
|
|
--*/
|
|
|
|
{
|
|
PBASE_ACTIVATION_CONTEXT_ACTIVATION_BLOCK ActivationBlock;
|
|
LPOVERLAPPED_COMPLETION_ROUTINE CompletionRoutine;
|
|
DWORD dwErrorCode;
|
|
DWORD dwNumberOfBytesTransfered;
|
|
LPOVERLAPPED lpOverlapped;
|
|
NTSTATUS Status;
|
|
RTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME ActivationFrame = { sizeof(ActivationFrame), RTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME_FORMAT_WHISTLER };
|
|
PACTIVATION_CONTEXT ActivationContext = NULL;
|
|
|
|
if ( NT_ERROR(IoStatusBlock->Status) ) {
|
|
dwErrorCode = RtlNtStatusToDosError(IoStatusBlock->Status);
|
|
dwNumberOfBytesTransfered = 0;
|
|
} else {
|
|
dwErrorCode = 0;
|
|
dwNumberOfBytesTransfered = (DWORD)IoStatusBlock->Information;
|
|
}
|
|
|
|
ActivationBlock = (PBASE_ACTIVATION_CONTEXT_ACTIVATION_BLOCK) ApcContext;
|
|
ActivationContext = ActivationBlock->ActivationContext;
|
|
CompletionRoutine = (LPOVERLAPPED_COMPLETION_ROUTINE) ActivationBlock->CallbackFunction;
|
|
lpOverlapped = (LPOVERLAPPED) CONTAINING_RECORD(IoStatusBlock, OVERLAPPED, Internal);
|
|
|
|
if (!(ActivationBlock->Flags & BASE_ACTIVATION_CONTEXT_ACTIVATION_BLOCK_FLAG_DO_NOT_FREE_AFTER_CALLBACK))
|
|
BasepFreeActivationContextActivationBlock(ActivationBlock);
|
|
|
|
RtlActivateActivationContextUnsafeFast(&ActivationFrame, ActivationContext);
|
|
__try {
|
|
(*CompletionRoutine)(dwErrorCode, dwNumberOfBytesTransfered, lpOverlapped);
|
|
} __finally {
|
|
RtlDeactivateActivationContextUnsafeFast(&ActivationFrame);
|
|
}
|
|
|
|
Reserved;
|
|
}
|
|
|
|
VOID
|
|
WINAPI
|
|
BasepIoCompletionSimple(
|
|
PVOID ApcContext,
|
|
PIO_STATUS_BLOCK IoStatusBlock,
|
|
DWORD Reserved
|
|
)
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This procedure is called to complete ReadFileEx and WriteFileEx
|
|
asynchronous I/O. Its primary function is to extract the
|
|
appropriate information from the passed IoStatusBlock and call the
|
|
users completion routine.
|
|
|
|
The users completion routine is called as:
|
|
|
|
Routine Description:
|
|
|
|
When an outstanding I/O completes with a callback, this
|
|
function is called. This function is only called while the
|
|
thread is in an alertable wait (SleepEx,
|
|
WaitForSingleObjectEx, or WaitForMultipleObjectsEx with the
|
|
bAlertable flag set to TRUE). Returning from this function
|
|
allows another pendiong I/O completion callback to be
|
|
processed. If this is the case, this callback is entered
|
|
before the termination of the thread's wait with a return
|
|
code of WAIT_IO_COMPLETION.
|
|
|
|
Note that each time your completion routine is called, the
|
|
system uses some of your stack. If you code your completion
|
|
logic to do additional ReadFileEx's and WriteFileEx's within
|
|
your completion routine, AND you do alertable waits in your
|
|
completion routine, you may grow your stack without ever
|
|
trimming it back.
|
|
|
|
Arguments:
|
|
|
|
dwErrorCode - Supplies the I/O completion status for the
|
|
related I/O. A value of 0 indicates that the I/O was
|
|
successful. Note that end of file is indicated by a
|
|
non-zero dwErrorCode value of ERROR_HANDLE_EOF.
|
|
|
|
dwNumberOfBytesTransfered - Supplies the number of bytes
|
|
transfered during the associated I/O. If an error
|
|
occured, a value of 0 is supplied.
|
|
|
|
lpOverlapped - Supplies the address of the OVERLAPPED
|
|
structure used to initiate the associated I/O. The
|
|
hEvent field of this structure is not used by the system
|
|
and may be used by the application to provide additional
|
|
I/O context. Once a completion routine is called, the
|
|
system will not use the OVERLAPPED structure. The
|
|
completion routine is free to deallocate the overlapped
|
|
structure.
|
|
|
|
Arguments:
|
|
|
|
ApcContext - Supplies the users completion routine. The format of
|
|
this routine is an LPOVERLAPPED_COMPLETION_ROUTINE.
|
|
|
|
IoStatusBlock - Supplies the address of the IoStatusBlock that
|
|
contains the I/O completion status. The IoStatusBlock is
|
|
contained within the OVERLAPPED structure.
|
|
|
|
Reserved - Not used; reserved for future use.
|
|
|
|
Return Value:
|
|
|
|
None.
|
|
|
|
--*/
|
|
|
|
{
|
|
LPOVERLAPPED_COMPLETION_ROUTINE CompletionRoutine;
|
|
DWORD dwErrorCode;
|
|
DWORD dwNumberOfBytesTransfered;
|
|
LPOVERLAPPED lpOverlapped;
|
|
|
|
dwErrorCode = 0;
|
|
|
|
if ( NT_ERROR(IoStatusBlock->Status) ) {
|
|
dwErrorCode = RtlNtStatusToDosError(IoStatusBlock->Status);
|
|
dwNumberOfBytesTransfered = 0;
|
|
}
|
|
else {
|
|
dwErrorCode = 0;
|
|
dwNumberOfBytesTransfered = (DWORD)IoStatusBlock->Information;
|
|
}
|
|
|
|
CompletionRoutine = (LPOVERLAPPED_COMPLETION_ROUTINE)ApcContext;
|
|
lpOverlapped = (LPOVERLAPPED)CONTAINING_RECORD(IoStatusBlock,OVERLAPPED,Internal);
|
|
|
|
(CompletionRoutine)(dwErrorCode,dwNumberOfBytesTransfered,lpOverlapped);
|
|
|
|
Reserved;
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
ReadFileEx(
|
|
HANDLE hFile,
|
|
LPVOID lpBuffer,
|
|
DWORD nNumberOfBytesToRead,
|
|
LPOVERLAPPED lpOverlapped,
|
|
LPOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Data can be read from a file using ReadFileEx.
|
|
|
|
This API reports its completion status asynchronously by calling the
|
|
specified lpCompletionRoutine.
|
|
|
|
The caller of this routine uses the lpOverlappedStructure to specify
|
|
the byte offset within the file where the read is to begin from.
|
|
For files that do not support this concept (pipes...), the starting
|
|
file offset is ignored.
|
|
|
|
Upon successful completion of this API (return value of TRUE), the
|
|
calling thread has an I/O outstanding. When the I/O completes, and
|
|
the thread is blocked in an alertable wait, the lpCompletionRoutine
|
|
will be called and the wait will return with a return code of
|
|
WAIT_IO_COMPLETION. If the I/O completes, but the thread issuing
|
|
the I/O is not in an alertable wait, the call to the completion
|
|
routine is queued until the thread executes an alertable wait.
|
|
|
|
If this API fails (by returning FALSE), GetLastError can be used to
|
|
get additional error information. If this call fails because the
|
|
thread issued a read beyond the end of file, GetLastError will
|
|
return a value of ERROR_HANDLE_EOF.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to be read. The
|
|
file handle must have been created with GENERIC_READ access to
|
|
the file. The file must have been created with the
|
|
FILE_FLAG_OVERLAPPED flag.
|
|
|
|
lpBuffer - Supplies the address of a buffer to receive the data read
|
|
from the file.
|
|
|
|
nNumberOfBytesToRead - Supplies the number of bytes to read from the
|
|
file.
|
|
|
|
lpOverlapped - Supplies the address of an OVERLAPPED structure to be
|
|
used with the request. The caller of this function must specify
|
|
a starting byte offset within the file to start the read from.
|
|
It does this using the Offset and OffsetHigh fields of the
|
|
overlapped structure. This call does not use or modify the
|
|
hEvent field of the overlapped structure. The caller may use
|
|
this field for any purpose. This API does use the Internal and
|
|
InternalHigh fields of the overlapped structure, the thread
|
|
should not manipulate this. The lpOverlapped structure must
|
|
remain valid for the duration of the I/O. It is not a good idea
|
|
to make it a local variable and then possibly returning from the
|
|
routine with the I/O that is using this structure still pending.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successul. Completion status will be
|
|
propagated to the caller using the completion callback
|
|
mechanism. Note that this information is only made available to
|
|
the thread that issued the I/O, and only when the I/O completes,
|
|
and the thread is executing in an alertable wait.
|
|
|
|
FALSE - The operation failed. Extended error status is available
|
|
using GetLastError. Note that end of file is treated as a failure
|
|
with an error code of ERROR_HANDLE_EOF.
|
|
|
|
--*/
|
|
{
|
|
NTSTATUS Status;
|
|
LARGE_INTEGER Li;
|
|
PBASE_ACTIVATION_CONTEXT_ACTIVATION_BLOCK ActivationBlock = NULL;
|
|
PIO_APC_ROUTINE IoApcRoutine = &BasepIoCompletionSimple;
|
|
PVOID ApcContext = lpCompletionRoutine;
|
|
|
|
Li.LowPart = lpOverlapped->Offset;
|
|
Li.HighPart = lpOverlapped->OffsetHigh;
|
|
|
|
// If there's an APC routine to call we need to allocate a little chunk of heap
|
|
// to pass the activation context to the APC callback.
|
|
if (lpCompletionRoutine != NULL) {
|
|
Status = BasepAllocateActivationContextActivationBlock(
|
|
BASEP_ALLOCATE_ACTIVATION_CONTEXT_ACTIVATION_BLOCK_FLAG_DO_NOT_ALLOCATE_IF_PROCESS_DEFAULT,
|
|
lpCompletionRoutine,
|
|
lpOverlapped,
|
|
&ActivationBlock);
|
|
if (!NT_SUCCESS(Status)) {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
// If there's nothing to do, call the simpler one that doesn't try to do activation context stuff
|
|
if (ActivationBlock != NULL) {
|
|
IoApcRoutine = &BasepIoCompletion;
|
|
ApcContext = ActivationBlock;
|
|
}
|
|
}
|
|
|
|
Status = NtReadFile(
|
|
hFile,
|
|
NULL,
|
|
IoApcRoutine,
|
|
ApcContext,
|
|
(PIO_STATUS_BLOCK) &lpOverlapped->Internal,
|
|
lpBuffer,
|
|
nNumberOfBytesToRead,
|
|
&Li,
|
|
NULL
|
|
);
|
|
if ( NT_ERROR(Status) ) {
|
|
if (ActivationBlock != NULL)
|
|
BasepFreeActivationContextActivationBlock(ActivationBlock);
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
WriteFileEx(
|
|
HANDLE hFile,
|
|
LPCVOID lpBuffer,
|
|
DWORD nNumberOfBytesToWrite,
|
|
LPOVERLAPPED lpOverlapped,
|
|
LPOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Data can be written to a file using WriteFileEx.
|
|
|
|
This API reports its completion status asynchronously by calling the
|
|
specified lpCompletionRoutine.
|
|
|
|
The caller of this routine uses the lpOverlappedStructure to specify
|
|
the byte offset within the file where the write is to begin.
|
|
For files that do not support this concept (pipes...), the starting
|
|
file offset is ignored.
|
|
|
|
Upon successful completion of this API (return value of TRUE), the
|
|
calling thread has an I/O outstanding. When the I/O completes, and
|
|
the thread is blocked in an alertable wait, the lpCompletionRoutine
|
|
will be called and the wait will return with a return code of
|
|
WAIT_IO_COMPLETION. If the I/O completes, but the thread issuing
|
|
the I/O is not in an alertable wait, the call to the completion
|
|
routine is queued until the thread executes an alertable wait.
|
|
|
|
If this API fails (by returning FALSE), GetLastError can be used to
|
|
get additional error information.
|
|
|
|
Unlike DOS, a NumberOfBytesToWrite value of zero does not truncate
|
|
or extend the file. If this function is required, SetEndOfFile
|
|
should be used.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to be written. The
|
|
file handle must have been created with GENERIC_WRITE access to
|
|
the file.
|
|
|
|
lpBuffer - Supplies the address of the data that is to be written to
|
|
the file.
|
|
|
|
nNumberOfBytesToWrite - Supplies the number of bytes to write to the
|
|
file. Unlike DOS, a value of zero is interpreted a null write.
|
|
|
|
lpOverlapped - Supplies the address of an OVERLAPPED structure to be
|
|
used with the request. The caller of this function must specify
|
|
a starting byte offset within the file to start the write to.
|
|
It does this using the Offset and OffsetHigh fields of the
|
|
overlapped structure. This call does not use or modify the
|
|
hEvent field of the overlapped structure. The caller may use
|
|
this field for any purpose. This API does use the Internal and
|
|
InternalHigh fields of the overlapped structure, the thread
|
|
should not manipulate this. The lpOverlapped structure must
|
|
remain valid for the duration of the I/O. It is not a good idea
|
|
to make it a local variable and then possibly returning from the
|
|
routine with the I/O that is using this structure still pending.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successul. Completion status will be
|
|
propagated to the caller using the completion callback
|
|
mechanism. Note that this information is only made available to
|
|
the thread that issued the I/O, and only when the I/O completes,
|
|
and the thread is executing in an alertable wait.
|
|
|
|
FALSE - The operation failed. Extended error status is available
|
|
using GetLastError. Note that end of file is treated as a failure
|
|
with an error code of ERROR_HANDLE_EOF.
|
|
|
|
--*/
|
|
{
|
|
NTSTATUS Status;
|
|
LARGE_INTEGER Li;
|
|
PBASE_ACTIVATION_CONTEXT_ACTIVATION_BLOCK ActivationBlock = NULL;
|
|
PIO_APC_ROUTINE IoApcRoutine = &BasepIoCompletionSimple;
|
|
PVOID ApcContext = lpCompletionRoutine;
|
|
|
|
Li.LowPart = lpOverlapped->Offset;
|
|
Li.HighPart = lpOverlapped->OffsetHigh;
|
|
|
|
// If there's an APC routine to call we may need to allocate a little chunk of heap
|
|
// to pass to the APC callback.
|
|
//
|
|
// we'll replace the parameters to the common NtWriteFile call below so that
|
|
// the control flow is obvious.
|
|
//
|
|
if (lpCompletionRoutine != NULL) {
|
|
Status = BasepAllocateActivationContextActivationBlock(
|
|
BASEP_ALLOCATE_ACTIVATION_CONTEXT_ACTIVATION_BLOCK_FLAG_DO_NOT_ALLOCATE_IF_PROCESS_DEFAULT,
|
|
lpCompletionRoutine,
|
|
lpOverlapped,
|
|
&ActivationBlock);
|
|
if (!NT_SUCCESS(Status)) {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
// If there's nothing to do, call the simpler one that doesn't try to do activation context stuff
|
|
if (ActivationBlock != NULL) {
|
|
IoApcRoutine = &BasepIoCompletion;
|
|
ApcContext = ActivationBlock;
|
|
}
|
|
}
|
|
|
|
Status = NtWriteFile(
|
|
hFile,
|
|
NULL,
|
|
IoApcRoutine,
|
|
ApcContext,
|
|
(PIO_STATUS_BLOCK)&lpOverlapped->Internal,
|
|
(LPVOID)lpBuffer,
|
|
nNumberOfBytesToWrite,
|
|
&Li,
|
|
NULL
|
|
);
|
|
if ( NT_ERROR(Status) ) {
|
|
if (ActivationBlock != NULL) {
|
|
BasepFreeActivationContextActivationBlock(ActivationBlock);
|
|
}
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
DeviceIoControl(
|
|
HANDLE hDevice,
|
|
DWORD dwIoControlCode,
|
|
LPVOID lpInBuffer,
|
|
DWORD nInBufferSize,
|
|
LPVOID lpOutBuffer,
|
|
DWORD nOutBufferSize,
|
|
LPDWORD lpBytesReturned,
|
|
LPOVERLAPPED lpOverlapped
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
An operation on a device may be performed by calling the device driver
|
|
directly using the DeviceIoContrl function.
|
|
|
|
The device driver must first be opened to get a valid handle.
|
|
|
|
Arguments:
|
|
|
|
hDevice - Supplies an open handle a device on which the operation is to
|
|
be performed.
|
|
|
|
dwIoControlCode - Supplies the control code for the operation. This
|
|
control code determines on which type of device the operation must
|
|
be performed and determines exactly what operation is to be
|
|
performed.
|
|
|
|
lpInBuffer - Suplies an optional pointer to an input buffer that contains
|
|
the data required to perform the operation. Whether or not the
|
|
buffer is actually optional is dependent on the IoControlCode.
|
|
|
|
nInBufferSize - Supplies the length of the input buffer in bytes.
|
|
|
|
lpOutBuffer - Suplies an optional pointer to an output buffer into which
|
|
the output data will be copied. Whether or not the buffer is actually
|
|
optional is dependent on the IoControlCode.
|
|
|
|
nOutBufferSize - Supplies the length of the output buffer in bytes.
|
|
|
|
lpBytesReturned - Supplies a pointer to a dword which will receive the
|
|
actual length of the data returned in the output buffer.
|
|
|
|
lpOverlapped - An optional parameter that supplies an overlap structure to
|
|
be used with the request. If NULL or the handle was created without
|
|
FILE_FLAG_OVERLAPPED then the DeviceIoControl will not return until
|
|
the operation completes.
|
|
|
|
When lpOverlapped is supplied and FILE_FLAG_OVERLAPPED was specified
|
|
when the handle was created, DeviceIoControl may return
|
|
ERROR_IO_PENDING to allow the caller to continue processing while the
|
|
operation completes. The event (or File handle if hEvent == NULL) will
|
|
be set to the not signalled state before ERROR_IO_PENDING is
|
|
returned. The event will be set to the signalled state upon completion
|
|
of the request. GetOverlappedResult is used to determine the result
|
|
when ERROR_IO_PENDING is returned.
|
|
|
|
Return Value:
|
|
|
|
TRUE -- The operation was successful.
|
|
|
|
FALSE -- The operation failed. Extended error status is available using
|
|
GetLastError.
|
|
|
|
--*/
|
|
{
|
|
NTSTATUS Status;
|
|
BOOLEAN DevIoCtl;
|
|
|
|
// On terminal Services TS-app-server, only allow admins to eject media while connected remotely
|
|
if ( ( dwIoControlCode == IOCTL_STORAGE_EJECT_MEDIA ) ||
|
|
( dwIoControlCode == IOCTL_DISK_EJECT_MEDIA ) ||
|
|
( dwIoControlCode == FSCTL_DISMOUNT_VOLUME ) )
|
|
{
|
|
if ( (NtCurrentPeb()->SessionId != USER_SHARED_DATA->ActiveConsoleId))
|
|
{
|
|
BOOL TSAppCompatEnabled;
|
|
NTSTATUS Status;
|
|
|
|
Status = IsTSAppCompatEnabled(&TSAppCompatEnabled);
|
|
if (!NT_SUCCESS(Status)) {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
if (TSAppCompatEnabled && !IsCallerAdminOrSystem()) {
|
|
BaseSetLastNTError(STATUS_ACCESS_DENIED);
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
if ( dwIoControlCode >> 16 == FILE_DEVICE_FILE_SYSTEM ) {
|
|
DevIoCtl = FALSE;
|
|
}
|
|
else {
|
|
DevIoCtl = TRUE;
|
|
}
|
|
|
|
if ( ARGUMENT_PRESENT( lpOverlapped ) ) {
|
|
lpOverlapped->Internal = (DWORD)STATUS_PENDING;
|
|
|
|
if ( DevIoCtl ) {
|
|
|
|
Status = NtDeviceIoControlFile(
|
|
hDevice,
|
|
lpOverlapped->hEvent,
|
|
NULL, // APC routine
|
|
(ULONG_PTR)lpOverlapped->hEvent & 1 ? NULL : lpOverlapped,
|
|
(PIO_STATUS_BLOCK)&lpOverlapped->Internal,
|
|
dwIoControlCode, // IoControlCode
|
|
lpInBuffer, // Buffer for data to the FS
|
|
nInBufferSize,
|
|
lpOutBuffer, // OutputBuffer for data from the FS
|
|
nOutBufferSize // OutputBuffer Length
|
|
);
|
|
}
|
|
else {
|
|
|
|
Status = NtFsControlFile(
|
|
hDevice,
|
|
lpOverlapped->hEvent,
|
|
NULL, // APC routine
|
|
(ULONG_PTR)lpOverlapped->hEvent & 1 ? NULL : lpOverlapped,
|
|
(PIO_STATUS_BLOCK)&lpOverlapped->Internal,
|
|
dwIoControlCode, // IoControlCode
|
|
lpInBuffer, // Buffer for data to the FS
|
|
nInBufferSize,
|
|
lpOutBuffer, // OutputBuffer for data from the FS
|
|
nOutBufferSize // OutputBuffer Length
|
|
);
|
|
|
|
}
|
|
|
|
// handle warning value STATUS_BUFFER_OVERFLOW somewhat correctly
|
|
if ( !NT_ERROR(Status) && ARGUMENT_PRESENT(lpBytesReturned) ) {
|
|
try {
|
|
*lpBytesReturned = (DWORD)lpOverlapped->InternalHigh;
|
|
}
|
|
except(EXCEPTION_EXECUTE_HANDLER) {
|
|
*lpBytesReturned = 0;
|
|
}
|
|
}
|
|
if ( NT_SUCCESS(Status) && Status != STATUS_PENDING) {
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
IO_STATUS_BLOCK Iosb;
|
|
|
|
if ( DevIoCtl ) {
|
|
Status = NtDeviceIoControlFile(
|
|
hDevice,
|
|
NULL,
|
|
NULL, // APC routine
|
|
NULL, // APC Context
|
|
&Iosb,
|
|
dwIoControlCode, // IoControlCode
|
|
lpInBuffer, // Buffer for data to the FS
|
|
nInBufferSize,
|
|
lpOutBuffer, // OutputBuffer for data from the FS
|
|
nOutBufferSize // OutputBuffer Length
|
|
);
|
|
}
|
|
else {
|
|
Status = NtFsControlFile(
|
|
hDevice,
|
|
NULL,
|
|
NULL, // APC routine
|
|
NULL, // APC Context
|
|
&Iosb,
|
|
dwIoControlCode, // IoControlCode
|
|
lpInBuffer, // Buffer for data to the FS
|
|
nInBufferSize,
|
|
lpOutBuffer, // OutputBuffer for data from the FS
|
|
nOutBufferSize // OutputBuffer Length
|
|
);
|
|
}
|
|
|
|
if ( Status == STATUS_PENDING) {
|
|
// Operation must complete before return & Iosb destroyed
|
|
Status = NtWaitForSingleObject( hDevice, FALSE, NULL );
|
|
if ( NT_SUCCESS(Status)) {
|
|
Status = Iosb.Status;
|
|
}
|
|
}
|
|
|
|
if ( NT_SUCCESS(Status) ) {
|
|
*lpBytesReturned = (DWORD)Iosb.Information;
|
|
return TRUE;
|
|
}
|
|
else {
|
|
// handle warning value STATUS_BUFFER_OVERFLOW somewhat correctly
|
|
if ( !NT_ERROR(Status) ) {
|
|
*lpBytesReturned = (DWORD)Iosb.Information;
|
|
}
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
CancelIo(
|
|
HANDLE hFile
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This routine cancels all of the outstanding I/O for the specified handle
|
|
for the specified file.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies the handle to the file whose pending I/O is to be
|
|
canceled.
|
|
|
|
Return Value:
|
|
|
|
TRUE -- The operation was successful.
|
|
|
|
FALSE -- The operation failed. Extended error status is available using
|
|
GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
|
|
//
|
|
// Simply cancel the I/O for the specified file.
|
|
//
|
|
|
|
Status = NtCancelIoFile(hFile, &IoStatusBlock);
|
|
|
|
if ( NT_SUCCESS(Status) ) {
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
ReadFileScatter(
|
|
HANDLE hFile,
|
|
FILE_SEGMENT_ELEMENT aSegementArray[],
|
|
DWORD nNumberOfBytesToRead,
|
|
LPDWORD lpReserved,
|
|
LPOVERLAPPED lpOverlapped
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Data can be read from a file using ReadFileScatter. The data
|
|
is then scatter to specified buffer segements.
|
|
|
|
This API is used to read data from a file. Data is read from the
|
|
file from the position indicated by the file pointer. After the
|
|
read completes, the file pointer is adjusted by the number of bytes
|
|
actually read. A return value of TRUE coupled with a bytes read of
|
|
0 indicates that the file pointer was beyond the current end of the
|
|
file at the time of the read.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to be read. The
|
|
file handle must have been created with GENERIC_READ access to
|
|
the file.
|
|
|
|
aSegementArray - Supplies a pointer an array of virtual segments.
|
|
A virtual segment is a memory buffer where part of the transferred data
|
|
should be placed. Segments are have a fix size of PAGE_SIZE
|
|
and must be aligned on a PAGE_SIZE boundary.
|
|
|
|
nNumberOfBytesToRead - Supplies the number of bytes to read from the file.
|
|
|
|
lpReserved - Reserved for now.
|
|
|
|
lpOverlapped - Optionally points to an OVERLAPPED structure to be used with the
|
|
request. If NULL then the transfer starts at the current file position
|
|
and ReadFile will not return until the operation completes.
|
|
|
|
If the handle hFile was created without specifying FILE_FLAG_OVERLAPPED
|
|
the file pointer is moved to the specified offset plus
|
|
lpNumberOfBytesRead before ReadFile returns. ReadFile will wait for the
|
|
request to complete before returning (it will not return
|
|
ERROR_IO_PENDING).
|
|
|
|
When FILE_FLAG_OVERLAPPED is specified, ReadFile may return
|
|
ERROR_IO_PENDING to allow the calling function to continue processing
|
|
while the operation completes. The event (or hFile if hEvent is NULL) will
|
|
be set to the signalled state upon completion of the request.
|
|
|
|
When the handle is created with FILE_FLAG_OVERLAPPED and lpOverlapped
|
|
is set to NULL, ReadFile will return ERROR_INVALID_PARAMTER because
|
|
the file offset is required.
|
|
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successul.
|
|
|
|
FALSE - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
LPDWORD lpNumberOfBytesRead = NULL;
|
|
|
|
if ( ARGUMENT_PRESENT(lpReserved) ||
|
|
!ARGUMENT_PRESENT( lpOverlapped )) {
|
|
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
if ( ARGUMENT_PRESENT(lpNumberOfBytesRead) ) {
|
|
*lpNumberOfBytesRead = 0;
|
|
}
|
|
|
|
if ( ARGUMENT_PRESENT( lpOverlapped ) ) {
|
|
LARGE_INTEGER Li;
|
|
|
|
lpOverlapped->Internal = (DWORD)STATUS_PENDING;
|
|
Li.LowPart = lpOverlapped->Offset;
|
|
Li.HighPart = lpOverlapped->OffsetHigh;
|
|
Status = NtReadFileScatter(
|
|
hFile,
|
|
lpOverlapped->hEvent,
|
|
NULL,
|
|
(ULONG_PTR)lpOverlapped->hEvent & 1 ? NULL : lpOverlapped,
|
|
(PIO_STATUS_BLOCK)&lpOverlapped->Internal,
|
|
aSegementArray,
|
|
nNumberOfBytesToRead,
|
|
&Li,
|
|
NULL
|
|
);
|
|
|
|
|
|
if ( NT_SUCCESS(Status) && Status != STATUS_PENDING) {
|
|
if ( ARGUMENT_PRESENT(lpNumberOfBytesRead) ) {
|
|
try {
|
|
*lpNumberOfBytesRead = (DWORD)lpOverlapped->InternalHigh;
|
|
}
|
|
except(EXCEPTION_EXECUTE_HANDLER) {
|
|
*lpNumberOfBytesRead = 0;
|
|
}
|
|
}
|
|
return TRUE;
|
|
}
|
|
else
|
|
if (Status == STATUS_END_OF_FILE) {
|
|
if ( ARGUMENT_PRESENT(lpNumberOfBytesRead) ) {
|
|
*lpNumberOfBytesRead = 0;
|
|
}
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Status = NtReadFileScatter(
|
|
hFile,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&IoStatusBlock,
|
|
aSegementArray,
|
|
nNumberOfBytesToRead,
|
|
NULL,
|
|
NULL
|
|
);
|
|
|
|
if ( Status == STATUS_PENDING) {
|
|
// Operation must complete before return & IoStatusBlock destroyed
|
|
Status = NtWaitForSingleObject( hFile, FALSE, NULL );
|
|
if ( NT_SUCCESS(Status)) {
|
|
Status = IoStatusBlock.Status;
|
|
}
|
|
}
|
|
|
|
if ( NT_SUCCESS(Status) ) {
|
|
*lpNumberOfBytesRead = (DWORD)IoStatusBlock.Information;
|
|
return TRUE;
|
|
}
|
|
else
|
|
if (Status == STATUS_END_OF_FILE) {
|
|
*lpNumberOfBytesRead = 0;
|
|
return TRUE;
|
|
}
|
|
else {
|
|
if ( NT_WARNING(Status) ) {
|
|
*lpNumberOfBytesRead = (DWORD)IoStatusBlock.Information;
|
|
}
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
BOOL
|
|
WINAPI
|
|
WriteFileGather(
|
|
HANDLE hFile,
|
|
FILE_SEGMENT_ELEMENT aSegementArray[],
|
|
DWORD nNumberOfBytesToWrite,
|
|
LPDWORD lpReserved,
|
|
LPOVERLAPPED lpOverlapped
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Data can be written to a file using WriteFileGather. The data can
|
|
be in multple file segement buffers.
|
|
|
|
This API is used to write data to a file. Data is written to the
|
|
file from the position indicated by the file pointer. After the
|
|
write completes, the file pointer is adjusted by the number of bytes
|
|
actually written.
|
|
|
|
Unlike DOS, a NumberOfBytesToWrite value of zero does not truncate
|
|
or extend the file. If this function is required, SetEndOfFile
|
|
should be used.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file that is to be written. The
|
|
file handle must have been created with GENERIC_WRITE access to
|
|
the file.
|
|
|
|
aSegementArray - Supplies a pointer an array of virtual segments.
|
|
A virtual segment is a memory buffer where part of the transferred data
|
|
should be placed. Segments are have a fix size of PAGE_SIZE
|
|
and must be aligned on a PAGE_SIZE boundary. The number of
|
|
entries in the array must be equal to nNumberOfBytesToRead /
|
|
PAGE_SIZE.
|
|
|
|
nNumberOfBytesToWrite - Supplies the number of bytes to write to the
|
|
file. Unlike DOS, a value of zero is interpreted a null write.
|
|
|
|
lpReserved - Unused for now.
|
|
|
|
lpOverlapped - Optionally points to an OVERLAPPED structure to be
|
|
used with the request. If NULL then the transfer starts at the
|
|
current file position and WriteFileGather will not return until the
|
|
operation completes.
|
|
|
|
If the handle <hFile> was created without specifying
|
|
FILE_FLAG_OVERLAPPED the file pointer is moved to the specified
|
|
offset plus lpNumberOfBytesWritten before WriteFile returns.
|
|
WriteFile will wait for the request to complete before returning
|
|
(it will not set ERROR_IO_PENDING).
|
|
|
|
When FILE_FLAG_OVERLAPPED is specified, WriteFile may return
|
|
ERROR_IO_PENDING to allow the calling function to continue processing
|
|
while the operation completes. The event (or hFile if hEvent is NULL) will
|
|
be set to the signalled state upon completion of the request.
|
|
|
|
When the handle is created with FILE_FLAG_OVERLAPPED and lpOverlapped
|
|
is set to NULL, WriteFile will return ERROR_INVALID_PARAMTER because
|
|
the file offset is required.
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was a success.
|
|
|
|
FALSE - The operation failed. Extended error status is
|
|
available using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
LPDWORD lpNumberOfBytesWritten = NULL;
|
|
|
|
if ( ARGUMENT_PRESENT(lpReserved) ||
|
|
!ARGUMENT_PRESENT( lpOverlapped )) {
|
|
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
if ( ARGUMENT_PRESENT(lpNumberOfBytesWritten) ) {
|
|
*lpNumberOfBytesWritten = 0;
|
|
}
|
|
|
|
if ( ARGUMENT_PRESENT( lpOverlapped ) ) {
|
|
LARGE_INTEGER Li;
|
|
|
|
lpOverlapped->Internal = (DWORD)STATUS_PENDING;
|
|
Li.LowPart = lpOverlapped->Offset;
|
|
Li.HighPart = lpOverlapped->OffsetHigh;
|
|
Status = NtWriteFileGather(
|
|
hFile,
|
|
lpOverlapped->hEvent,
|
|
NULL,
|
|
(ULONG_PTR)lpOverlapped->hEvent & 1 ? NULL : lpOverlapped,
|
|
(PIO_STATUS_BLOCK)&lpOverlapped->Internal,
|
|
aSegementArray,
|
|
nNumberOfBytesToWrite,
|
|
&Li,
|
|
NULL
|
|
);
|
|
|
|
if ( !NT_ERROR(Status) && Status != STATUS_PENDING) {
|
|
if ( ARGUMENT_PRESENT(lpNumberOfBytesWritten) ) {
|
|
try {
|
|
*lpNumberOfBytesWritten = (DWORD)lpOverlapped->InternalHigh;
|
|
}
|
|
except(EXCEPTION_EXECUTE_HANDLER) {
|
|
*lpNumberOfBytesWritten = 0;
|
|
}
|
|
}
|
|
return TRUE;
|
|
}
|
|
else {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
else {
|
|
Status = NtWriteFileGather(
|
|
hFile,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&IoStatusBlock,
|
|
aSegementArray,
|
|
nNumberOfBytesToWrite,
|
|
NULL,
|
|
NULL
|
|
);
|
|
|
|
if ( Status == STATUS_PENDING) {
|
|
// Operation must complete before return & IoStatusBlock destroyed
|
|
Status = NtWaitForSingleObject( hFile, FALSE, NULL );
|
|
if ( NT_SUCCESS(Status)) {
|
|
Status = IoStatusBlock.Status;
|
|
}
|
|
}
|
|
|
|
if ( NT_SUCCESS(Status)) {
|
|
*lpNumberOfBytesWritten = (DWORD)IoStatusBlock.Information;
|
|
return TRUE;
|
|
}
|
|
else {
|
|
if ( NT_WARNING(Status) ) {
|
|
*lpNumberOfBytesWritten = (DWORD)IoStatusBlock.Information;
|
|
}
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
BOOL
|
|
APIENTRY
|
|
SetFileValidData(
|
|
IN HANDLE hFile,
|
|
IN LONGLONG ValidDataLength
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
SetFileValidData is used to set the valid data length for the given file.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose type valid data
|
|
length is to be set
|
|
|
|
ValidDataLength - Supplies the desired valid data length
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was a success.
|
|
|
|
FALSE - The operation failed. Extended error status is
|
|
available using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
FILE_VALID_DATA_LENGTH_INFORMATION ValidDataInfo;
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
ValidDataInfo.ValidDataLength.QuadPart = ValidDataLength;
|
|
|
|
Status = NtSetInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
&ValidDataInfo,
|
|
sizeof(FILE_VALID_DATA_LENGTH_INFORMATION),
|
|
FileValidDataLengthInformation
|
|
);
|
|
if (!NT_SUCCESS(Status)) {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
BOOL
|
|
APIENTRY
|
|
SetFileShortNameW(
|
|
IN HANDLE hFile,
|
|
IN LPCWSTR lpShortName
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
SetFileShortNameW is used to set the short name for the given file.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose short name is to be changed
|
|
|
|
lpShortName - Supplies the desired short name
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was a success.
|
|
|
|
FALSE - The operation failed. Extended error status is
|
|
available using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
PFILE_NAME_INFORMATION FileNameInfo;
|
|
DWORD FileNameInfoSize;
|
|
DWORD FileInformationClass;
|
|
|
|
|
|
if (CONSOLE_HANDLE(hFile)) {
|
|
BaseSetLastNTError(STATUS_INVALID_HANDLE);
|
|
return FALSE;
|
|
}
|
|
|
|
if (!ARGUMENT_PRESENT(lpShortName)) {
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
}
|
|
|
|
FileNameInfoSize = FIELD_OFFSET(FILE_NAME_INFORMATION, FileName) + ((wcslen(lpShortName)+1)*sizeof(WCHAR));
|
|
FileNameInfo = RtlAllocateHeap( RtlProcessHeap(), MAKE_TAG(TMP_TAG), FileNameInfoSize );
|
|
if (!FileNameInfo) {
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return FALSE;
|
|
}
|
|
|
|
FileNameInfo->FileNameLength = wcslen(lpShortName) * sizeof(WCHAR);
|
|
wcscpy( FileNameInfo->FileName, lpShortName );
|
|
|
|
Status = NtSetInformationFile(
|
|
hFile,
|
|
&IoStatusBlock,
|
|
FileNameInfo,
|
|
FileNameInfoSize,
|
|
FileShortNameInformation
|
|
);
|
|
|
|
RtlFreeHeap( RtlProcessHeap(), 0, FileNameInfo );
|
|
|
|
if (!NT_SUCCESS(Status)) {
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
FileInformationClass = FileShortNameInformation;
|
|
|
|
if ((FileInformationClass == FileEndOfFileInformation) ||
|
|
(FileInformationClass == FileAllocationInformation) ||
|
|
(FileInformationClass == FilePositionInformation))
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
BOOL
|
|
APIENTRY
|
|
SetFileShortNameA(
|
|
IN HANDLE hFile,
|
|
IN LPCSTR lpShortName
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
SetFileShortNameW is used to set the short name for the given file.
|
|
|
|
Arguments:
|
|
|
|
hFile - Supplies an open handle to a file whose short name is to be changed
|
|
|
|
lpShortName - Supplies the desired short name
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was a success.
|
|
|
|
FALSE - The operation failed. Extended error status is
|
|
available using GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
PUNICODE_STRING Unicode;
|
|
|
|
Unicode = Basep8BitStringToStaticUnicodeString( lpShortName );
|
|
if (Unicode == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
return ( SetFileShortNameW(
|
|
hFile,
|
|
(LPCWSTR)Unicode->Buffer
|
|
)
|
|
);
|
|
}
|
|
|
|
|
|
BOOLEAN
|
|
Wow64EnableWow64FsRedirection (
|
|
IN BOOLEAN Wow64FsEnableRedirection
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function enables/disables Wow64 file system redirection.
|
|
|
|
Wow64 redirects all accesses to %windir%\system32 to %windir%\syswow64.
|
|
This API is useful for 32-bit applications which want to gain access to the
|
|
native system32 directory. By default, Wow64 file system redirection is enabled.
|
|
|
|
File redirection is only affected for the thread calling this API.
|
|
|
|
Note : You must enable file system redirection after disabling it. Once you have
|
|
a file handle, you must enable file system redirection back.
|
|
|
|
Example:
|
|
|
|
BOOL bRet = Wow64EnableWow64FsRedirection (FALSE);
|
|
if (bRet == TRUE) {
|
|
|
|
//
|
|
// Open the file handle
|
|
//
|
|
|
|
CreateFile (..."c:\\windows\\system32\\notepad.exe"...)
|
|
|
|
//
|
|
// Enable Wow64 file system redirection.
|
|
//
|
|
|
|
Wow64EnableWow64FsRedirection (TRUE);
|
|
}
|
|
|
|
//
|
|
// Use the file handle
|
|
//
|
|
|
|
|
|
Arguments:
|
|
|
|
Wow64FsEnableRedirection - Boolean to indicate whether to enable Wow64 file system
|
|
redirection. Specify FALSE if you want to disable Wow64 file system redirection,
|
|
otherwise TRUE to enable it.
|
|
|
|
|
|
Return Value:
|
|
|
|
TRUE - The operation was successul.
|
|
|
|
FALSE - The operation failed. Extended error status is available
|
|
using GetLastError.
|
|
--*/
|
|
|
|
{
|
|
NTSTATUS NtStatus;
|
|
|
|
NtStatus = RtlWow64EnableFsRedirection (Wow64FsEnableRedirection);
|
|
|
|
if (!NT_SUCCESS (NtStatus)) {
|
|
|
|
BaseSetLastNTError (NtStatus);
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|