155 lines
4.0 KiB
C
155 lines
4.0 KiB
C
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 1989 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
copy.c
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
This module contains the routine to copy a file.
|
||
|
|
|
||
|
|
Author:
|
||
|
|
|
||
|
|
Dan Hinsley (DanHi) 24-Feb-1991
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
02-Feb-1994 Danl
|
||
|
|
Fixed memory leak where ioBuffer wasn't getting free'd when doing
|
||
|
|
an error exit from ElfpCopyFile.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
|
||
|
|
// INCLUDES
|
||
|
|
|
||
|
|
|
||
|
|
#include <eventp.h>
|
||
|
|
|
||
|
|
|
||
|
|
NTSTATUS
|
||
|
|
ElfpCopyFile(
|
||
|
|
IN HANDLE SourceHandle,
|
||
|
|
IN PUNICODE_STRING TargetFileName
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
Routine Description:
|
||
|
|
This routine copies or appends from the source file to the target file.
|
||
|
|
If the target file already exists, the copy fails.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
SourceHandle - An open handle to the source file.
|
||
|
|
TargetFileName - The name of the file to copy to.
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
NTSTATUS - STATUS_SUCCESS or error.
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
NTSTATUS Status;
|
||
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
||
|
|
FILE_STANDARD_INFORMATION sourceStandardInfo;
|
||
|
|
OBJECT_ATTRIBUTES ObjectAttributes;
|
||
|
|
HANDLE TargetHandle;
|
||
|
|
PCHAR ioBuffer;
|
||
|
|
ULONG ioBufferSize;
|
||
|
|
ULONG bytesRead;
|
||
|
|
|
||
|
|
// Get the size of the file so we can set the attributes of the target
|
||
|
|
// file.
|
||
|
|
Status = NtQueryInformationFile(
|
||
|
|
SourceHandle,
|
||
|
|
&IoStatusBlock,
|
||
|
|
&sourceStandardInfo,
|
||
|
|
sizeof(sourceStandardInfo),
|
||
|
|
FileStandardInformation
|
||
|
|
);
|
||
|
|
if (!NT_SUCCESS(Status)) {
|
||
|
|
return(Status);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Open the target file, fail if the file already exists.
|
||
|
|
InitializeObjectAttributes(
|
||
|
|
&ObjectAttributes,
|
||
|
|
TargetFileName,
|
||
|
|
OBJ_CASE_INSENSITIVE,
|
||
|
|
NULL,
|
||
|
|
NULL
|
||
|
|
);
|
||
|
|
Status = NtCreateFile(&TargetHandle,
|
||
|
|
GENERIC_WRITE | SYNCHRONIZE,
|
||
|
|
&ObjectAttributes,
|
||
|
|
&IoStatusBlock,
|
||
|
|
&(sourceStandardInfo.EndOfFile),
|
||
|
|
FILE_ATTRIBUTE_NORMAL,
|
||
|
|
0, // Share access
|
||
|
|
FILE_CREATE,
|
||
|
|
FILE_SYNCHRONOUS_IO_ALERT | FILE_SEQUENTIAL_ONLY,
|
||
|
|
NULL, // EA buffer
|
||
|
|
0 // EA length
|
||
|
|
);
|
||
|
|
if (!NT_SUCCESS(Status)) {
|
||
|
|
return(Status);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Allocate a buffer to use for the data copy.
|
||
|
|
ioBufferSize = 4096;
|
||
|
|
ioBuffer = ElfpAllocateBuffer(ioBufferSize);
|
||
|
|
if (ioBuffer == NULL) {
|
||
|
|
return (STATUS_NO_MEMORY);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Copy data--read from source, write to target. Do this until
|
||
|
|
// all the data is written or an error occurs.
|
||
|
|
while (TRUE) {
|
||
|
|
Status = NtReadFile(
|
||
|
|
SourceHandle,
|
||
|
|
NULL, // Event
|
||
|
|
NULL, // ApcRoutine
|
||
|
|
NULL, // ApcContext
|
||
|
|
&IoStatusBlock,
|
||
|
|
ioBuffer,
|
||
|
|
ioBufferSize,
|
||
|
|
NULL, // ByteOffset
|
||
|
|
NULL // Key
|
||
|
|
);
|
||
|
|
if (!NT_SUCCESS(Status) && Status != STATUS_END_OF_FILE) {
|
||
|
|
ElfDbgPrint(("[ELF] Copy failed reading source file - %X\n",
|
||
|
|
Status));
|
||
|
|
ElfpFreeBuffer(ioBuffer);
|
||
|
|
return (Status);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (IoStatusBlock.Information == 0 || Status == STATUS_END_OF_FILE) {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
bytesRead = (ULONG)IoStatusBlock.Information;
|
||
|
|
Status = NtWriteFile(
|
||
|
|
TargetHandle,
|
||
|
|
NULL, // Event
|
||
|
|
NULL, // ApcRoutine
|
||
|
|
NULL, // ApcContext
|
||
|
|
&IoStatusBlock,
|
||
|
|
ioBuffer,
|
||
|
|
bytesRead,
|
||
|
|
NULL, // ByteOffset
|
||
|
|
NULL // Key
|
||
|
|
);
|
||
|
|
if (!NT_SUCCESS(Status)) {
|
||
|
|
ElfDbgPrint(("[ELF] Copy failed writing target file - %X\n",
|
||
|
|
Status));
|
||
|
|
ElfpFreeBuffer(ioBuffer);
|
||
|
|
return (Status);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
ElfpFreeBuffer(ioBuffer);
|
||
|
|
|
||
|
|
Status = NtClose(TargetHandle);
|
||
|
|
ASSERT(NT_SUCCESS(Status));
|
||
|
|
|
||
|
|
return STATUS_SUCCESS;
|
||
|
|
} // ElfpCopyFile
|