Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077/* * PROJECT: ReactOS Setup Library * LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+) * PURPOSE: File support functions. * COPYRIGHT: Casper S. Hornstrup (chorns@users.sourceforge.net) * Copyright 2017-2018 Hermes Belusca-Maito */
/* INCLUDES *****************************************************************/
#include "precomp.h"#include "filesup.h"#include <pseh/pseh2.h>
#define NDEBUG#include <debug.h>
/* FUNCTIONS ****************************************************************/
staticNTSTATUSSetupCreateSingleDirectory( _In_ PCUNICODE_STRING DirectoryName){ NTSTATUS Status; UNICODE_STRING PathName = *DirectoryName; OBJECT_ATTRIBUTES ObjectAttributes; IO_STATUS_BLOCK IoStatusBlock; HANDLE DirectoryHandle;
/* Remove the trailing separator if needed */ if (PathName.Length >= 2 * sizeof(WCHAR) && PathName.Buffer[PathName.Length / sizeof(WCHAR) - 1] == OBJ_NAME_PATH_SEPARATOR) { PathName.Length -= sizeof(WCHAR); }
InitializeObjectAttributes(&ObjectAttributes, &PathName, OBJ_CASE_INSENSITIVE, NULL, NULL);
Status = NtCreateFile(&DirectoryHandle, FILE_LIST_DIRECTORY | SYNCHRONIZE, &ObjectAttributes, &IoStatusBlock, NULL, FILE_ATTRIBUTE_DIRECTORY, FILE_SHARE_READ | FILE_SHARE_WRITE, FILE_OPEN_IF, FILE_OPEN_FOR_BACKUP_INTENT | FILE_DIRECTORY_FILE, NULL, 0); if (NT_SUCCESS(Status)) NtClose(DirectoryHandle);
return Status;}
/** * @brief * Create a new directory, specified by the given path. * Any intermediate non-existing directory is created as well. * * @param[in] PathName * The path of the directory to be created. * * @return An NTSTATUS code indicating success or failure. **/NTSTATUSSetupCreateDirectory( _In_ PCWSTR PathName){ NTSTATUS Status = STATUS_SUCCESS; UNICODE_STRING PathNameU; PCWSTR Buffer; PCWCH Ptr, End;
RtlInitUnicodeString(&PathNameU, PathName); Buffer = PathNameU.Buffer; End = Buffer + (PathNameU.Length / sizeof(WCHAR));
/* Find the deepest existing sub-directory: start from the * end and go back, verifying each sub-directory in turn */ for (Ptr = End; Ptr > Buffer;) { BOOLEAN bExists;
/* If we are on a separator, truncate at the next character. * The trailing separator is kept for the existence check. */ if ((Ptr < End) && (*Ptr == OBJ_NAME_PATH_SEPARATOR)) PathNameU.Length = (ULONG_PTR)(Ptr+1) - (ULONG_PTR)Buffer;
/* Check if the sub-directory exists and stop * if so: this is the deepest existing one */ DPRINT("PathName: %wZ\n", &PathNameU); bExists = DoesPathExist_UStr(NULL, &PathNameU, TRUE); if (bExists) break;
/* Skip back any consecutive path separators */ while ((Ptr > Buffer) && (*Ptr == OBJ_NAME_PATH_SEPARATOR)) --Ptr; /* Go to the beginning of the path component, stop at the separator */ while ((Ptr > Buffer) && (*Ptr != OBJ_NAME_PATH_SEPARATOR)) --Ptr; }
/* Skip any consecutive path separators */ while ((Ptr < End) && (*Ptr == OBJ_NAME_PATH_SEPARATOR)) ++Ptr;
/* Create all the remaining sub-directories */ for (; Ptr < End; ++Ptr) { /* Go to the end of the current path component, stop at * the separator or terminating NUL and truncate it */ while ((Ptr < End) && (*Ptr != OBJ_NAME_PATH_SEPARATOR)) ++Ptr; PathNameU.Length = (ULONG_PTR)Ptr - (ULONG_PTR)Buffer;
DPRINT("Create: %wZ\n", &PathNameU); Status = SetupCreateSingleDirectory(&PathNameU); if (!NT_SUCCESS(Status)) break; }
DPRINT("Done.\n"); return Status;}
NTSTATUSSetupDeleteFile( IN PCWSTR FileName, IN BOOLEAN ForceDelete) // ForceDelete can be used to delete read-only files{ NTSTATUS Status; UNICODE_STRING NtPathU; OBJECT_ATTRIBUTES ObjectAttributes; IO_STATUS_BLOCK IoStatusBlock; HANDLE FileHandle; FILE_DISPOSITION_INFORMATION FileDispInfo; BOOLEAN RetryOnce = FALSE;
/* Open the directory name that was passed in */ RtlInitUnicodeString(&NtPathU, FileName); InitializeObjectAttributes(&ObjectAttributes, &NtPathU, OBJ_CASE_INSENSITIVE, NULL, NULL);
Retry: // We go back there once if RetryOnce == TRUE Status = NtOpenFile(&FileHandle, DELETE | FILE_READ_ATTRIBUTES | (RetryOnce ? FILE_WRITE_ATTRIBUTES : 0), &ObjectAttributes, &IoStatusBlock, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_NON_DIRECTORY_FILE | FILE_OPEN_FOR_BACKUP_INTENT); if (!NT_SUCCESS(Status)) { DPRINT1("NtOpenFile failed with Status 0x%08lx\n", Status); return Status; }
if (RetryOnce) { FILE_BASIC_INFORMATION FileInformation;
Status = NtQueryInformationFile(FileHandle, &IoStatusBlock, &FileInformation, sizeof(FILE_BASIC_INFORMATION), FileBasicInformation); if (!NT_SUCCESS(Status)) { DPRINT1("NtQueryInformationFile failed with Status 0x%08lx\n", Status); NtClose(FileHandle); return Status; }
FileInformation.FileAttributes = FILE_ATTRIBUTE_NORMAL; Status = NtSetInformationFile(FileHandle, &IoStatusBlock, &FileInformation, sizeof(FILE_BASIC_INFORMATION), FileBasicInformation); NtClose(FileHandle); if (!NT_SUCCESS(Status)) { DPRINT1("NtSetInformationFile failed with Status 0x%08lx\n", Status); return Status; } }
/* Ask for the file to be deleted */ FileDispInfo.DeleteFile = TRUE; Status = NtSetInformationFile(FileHandle, &IoStatusBlock, &FileDispInfo, sizeof(FILE_DISPOSITION_INFORMATION), FileDispositionInformation); NtClose(FileHandle);
if (!NT_SUCCESS(Status)) DPRINT1("Deletion of file '%S' failed, Status 0x%08lx\n", FileName, Status);
// FIXME: Check the precise value of Status! if (!NT_SUCCESS(Status) && ForceDelete && !RetryOnce) { /* Retry once */ RetryOnce = TRUE; goto Retry; }
/* Return result to the caller */ return Status;}
NTSTATUSSetupCopyFile( IN PCWSTR SourceFileName, IN PCWSTR DestinationFileName, IN BOOLEAN FailIfExists){ NTSTATUS Status; UNICODE_STRING FileName; OBJECT_ATTRIBUTES ObjectAttributes; HANDLE FileHandleSource; HANDLE FileHandleDest; IO_STATUS_BLOCK IoStatusBlock; FILE_STANDARD_INFORMATION FileStandard; FILE_BASIC_INFORMATION FileBasic; ULONG RegionSize; HANDLE SourceFileSection; PVOID SourceFileMap = NULL; SIZE_T SourceSectionSize = 0; LARGE_INTEGER ByteOffset;
RtlInitUnicodeString(&FileName, SourceFileName); InitializeObjectAttributes(&ObjectAttributes, &FileName, OBJ_CASE_INSENSITIVE, NULL, NULL);
Status = NtOpenFile(&FileHandleSource, GENERIC_READ, &ObjectAttributes, &IoStatusBlock, FILE_SHARE_READ, FILE_SEQUENTIAL_ONLY); if (!NT_SUCCESS(Status)) { DPRINT1("NtOpenFile failed: %x, %wZ\n", Status, &FileName); goto done; }
Status = NtQueryInformationFile(FileHandleSource, &IoStatusBlock, &FileStandard, sizeof(FILE_STANDARD_INFORMATION), FileStandardInformation); if (!NT_SUCCESS(Status)) { DPRINT1("NtQueryInformationFile failed: %x\n", Status); goto closesrc; }
Status = NtQueryInformationFile(FileHandleSource, &IoStatusBlock, &FileBasic, sizeof(FILE_BASIC_INFORMATION), FileBasicInformation); if (!NT_SUCCESS(Status)) { DPRINT1("NtQueryInformationFile failed: %x\n", Status); goto closesrc; }
Status = NtCreateSection(&SourceFileSection, SECTION_MAP_READ, NULL, NULL, PAGE_READONLY, SEC_COMMIT, FileHandleSource); if (!NT_SUCCESS(Status)) { DPRINT1("NtCreateSection failed: %x, %S\n", Status, SourceFileName); goto closesrc; }
Status = NtMapViewOfSection(SourceFileSection, NtCurrentProcess(), &SourceFileMap, 0, 0, NULL, &SourceSectionSize, ViewUnmap, 0, PAGE_READONLY); if (!NT_SUCCESS(Status)) { DPRINT1("NtMapViewOfSection failed: %x, %S\n", Status, SourceFileName); goto closesrcsec; }
RtlInitUnicodeString(&FileName, DestinationFileName); InitializeObjectAttributes(&ObjectAttributes, &FileName, OBJ_CASE_INSENSITIVE, NULL, NULL);
Status = NtCreateFile(&FileHandleDest, GENERIC_WRITE | SYNCHRONIZE, &ObjectAttributes, &IoStatusBlock, NULL, FileBasic.FileAttributes, // FILE_ATTRIBUTE_NORMAL, 0, FailIfExists ? FILE_CREATE : FILE_OVERWRITE_IF, FILE_NO_INTERMEDIATE_BUFFERING | FILE_SEQUENTIAL_ONLY | FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0); if (!NT_SUCCESS(Status)) { /* * Open may have failed because the file to overwrite * is in readonly mode. */ if (Status == STATUS_ACCESS_DENIED) { FILE_BASIC_INFORMATION FileBasicInfo;
/* Reattempt to open it with limited access */ Status = NtCreateFile(&FileHandleDest, FILE_WRITE_ATTRIBUTES | SYNCHRONIZE, &ObjectAttributes, &IoStatusBlock, NULL, FILE_ATTRIBUTE_NORMAL, 0, FILE_OPEN, FILE_NO_INTERMEDIATE_BUFFERING | FILE_SEQUENTIAL_ONLY | FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0); /* Fail for real if we cannot open it that way */ if (!NT_SUCCESS(Status)) { DPRINT1("NtCreateFile failed: %x, %wZ\n", Status, &FileName); goto unmapsrcsec; }
/* Zero our basic info, just to set attributes */ RtlZeroMemory(&FileBasicInfo, sizeof(FileBasicInfo)); /* Reset attributes to normal, no read-only */ FileBasicInfo.FileAttributes = FILE_ATTRIBUTE_NORMAL; /* * We basically don't care about whether it succeed: * if it didn't, later open will fail. */ NtSetInformationFile(FileHandleDest, &IoStatusBlock, &FileBasicInfo, sizeof(FileBasicInfo), FileBasicInformation);
/* Close file */ NtClose(FileHandleDest);
/* And re-attempt overwrite */ Status = NtCreateFile(&FileHandleDest, GENERIC_WRITE | SYNCHRONIZE, &ObjectAttributes, &IoStatusBlock, NULL, FILE_ATTRIBUTE_NORMAL, 0, FILE_OVERWRITE_IF, FILE_NO_INTERMEDIATE_BUFFERING | FILE_SEQUENTIAL_ONLY | FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0); }
/* We failed */ if (!NT_SUCCESS(Status)) { DPRINT1("NtCreateFile failed: %x, %wZ\n", Status, &FileName); goto unmapsrcsec; } }
RegionSize = (ULONG)PAGE_ROUND_UP(FileStandard.EndOfFile.u.LowPart); IoStatusBlock.Status = 0; ByteOffset.QuadPart = 0ULL; Status = NtWriteFile(FileHandleDest, NULL, NULL, NULL, &IoStatusBlock, SourceFileMap, RegionSize, &ByteOffset, NULL); if (!NT_SUCCESS(Status)) { DPRINT1("NtWriteFile failed: %x:%x, iosb: %p src: %p, size: %x\n", Status, IoStatusBlock.Status, &IoStatusBlock, SourceFileMap, RegionSize); goto closedest; }
/* Copy file date/time from source file */ Status = NtSetInformationFile(FileHandleDest, &IoStatusBlock, &FileBasic, sizeof(FILE_BASIC_INFORMATION), FileBasicInformation); if (!NT_SUCCESS(Status)) { DPRINT1("NtSetInformationFile failed: %x\n", Status); goto closedest; }
/* Shorten the file back to its real size after completing the write */ Status = NtSetInformationFile(FileHandleDest, &IoStatusBlock, &FileStandard.EndOfFile, sizeof(FILE_END_OF_FILE_INFORMATION), FileEndOfFileInformation); if (!NT_SUCCESS(Status)) { DPRINT1("NtSetInformationFile failed: %x\n", Status); }
closedest: NtClose(FileHandleDest);
unmapsrcsec: NtUnmapViewOfSection(NtCurrentProcess(), SourceFileMap);
closesrcsec: NtClose(SourceFileSection);
closesrc: NtClose(FileHandleSource);
done: return Status;}
/* * Synchronized with its kernel32 counterpart, but we don't manage reparse points here. */NTSTATUSSetupMoveFile( IN PCWSTR ExistingFileName, IN PCWSTR NewFileName, IN ULONG Flags){ NTSTATUS Status; IO_STATUS_BLOCK IoStatusBlock; OBJECT_ATTRIBUTES ObjectAttributes; PFILE_RENAME_INFORMATION RenameInfo; UNICODE_STRING NewPathU, ExistingPathU; HANDLE SourceHandle = NULL; BOOLEAN ReplaceIfExists;
RtlInitUnicodeString(&ExistingPathU, ExistingFileName); RtlInitUnicodeString(&NewPathU, NewFileName);
_SEH2_TRY { ReplaceIfExists = !!(Flags & MOVEFILE_REPLACE_EXISTING);
/* Unless we manage a proper opening, we'll attempt to reopen without reparse support */ InitializeObjectAttributes(&ObjectAttributes, &ExistingPathU, OBJ_CASE_INSENSITIVE, NULL, NULL); /* Attempt to open source file */ Status = NtOpenFile(&SourceHandle, FILE_READ_ATTRIBUTES | DELETE | SYNCHRONIZE, &ObjectAttributes, &IoStatusBlock, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_OPEN_FOR_BACKUP_INTENT | ((Flags & MOVEFILE_WRITE_THROUGH) ? FILE_WRITE_THROUGH : 0)); if (!NT_SUCCESS(Status)) { if (Status != STATUS_INVALID_PARAMETER) { _SEH2_LEAVE; } }
/* At that point, we MUST have a source handle */ ASSERT(SourceHandle);
/* Allocate renaming buffer and fill it */ RenameInfo = RtlAllocateHeap(RtlGetProcessHeap(), 0, NewPathU.Length + sizeof(FILE_RENAME_INFORMATION)); if (RenameInfo == NULL) { Status = STATUS_NO_MEMORY; _SEH2_LEAVE; }
RtlCopyMemory(&RenameInfo->FileName, NewPathU.Buffer, NewPathU.Length); RenameInfo->ReplaceIfExists = ReplaceIfExists; RenameInfo->RootDirectory = NULL; RenameInfo->FileNameLength = NewPathU.Length;
/* Attempt to rename the file */ Status = NtSetInformationFile(SourceHandle, &IoStatusBlock, RenameInfo, NewPathU.Length + sizeof(FILE_RENAME_INFORMATION), FileRenameInformation); RtlFreeHeap(RtlGetProcessHeap(), 0, RenameInfo); if (NT_SUCCESS(Status)) { /* If it succeeded, all fine, quit */ _SEH2_LEAVE; } /* * If we failed for any other reason than not the same device, fail. * If we failed because of different devices, only allow renaming * if user allowed copy. */ if (Status != STATUS_NOT_SAME_DEVICE || !(Flags & MOVEFILE_COPY_ALLOWED)) { /* ReactOS hack! To be removed once all FSD have proper renaming support * Just leave status to error and leave */ if (Status == STATUS_NOT_IMPLEMENTED) { DPRINT1("Forcing copy, renaming not supported by FSD\n"); } else { _SEH2_LEAVE; } }
/* Close the source file */ NtClose(SourceHandle); SourceHandle = NULL;
/* Perform the file copy */ Status = SetupCopyFile(ExistingFileName, NewFileName, !ReplaceIfExists);
/* If it succeeded, delete the source file */ if (NT_SUCCESS(Status)) { /* Force-delete files even if read-only */ SetupDeleteFile(ExistingFileName, TRUE); } } _SEH2_FINALLY { if (SourceHandle) NtClose(SourceHandle); } _SEH2_END;
return Status;}
NTSTATUSConcatPathsV( IN OUT PWSTR PathBuffer, IN SIZE_T cchPathSize, IN ULONG NumberOfPathComponents, IN va_list PathComponentsList){ NTSTATUS Status = STATUS_SUCCESS; SIZE_T cchPathLen; PCWSTR PathComponent;
if (cchPathSize < 1) return STATUS_SUCCESS;
while (NumberOfPathComponents--) { PathComponent = va_arg(PathComponentsList, PCWSTR); if (!PathComponent) continue;
cchPathLen = min(cchPathSize, wcslen(PathBuffer)); if (cchPathLen >= cchPathSize) return STATUS_BUFFER_OVERFLOW;
if (PathComponent[0] != OBJ_NAME_PATH_SEPARATOR && cchPathLen > 0 && PathBuffer[cchPathLen-1] != OBJ_NAME_PATH_SEPARATOR) { /* PathComponent does not start with '\' and PathBuffer does not end with '\' */ Status = RtlStringCchCatW(PathBuffer, cchPathSize, L"\\"); if (!NT_SUCCESS(Status)) return Status; } else if (PathComponent[0] == OBJ_NAME_PATH_SEPARATOR && cchPathLen > 0 && PathBuffer[cchPathLen-1] == OBJ_NAME_PATH_SEPARATOR) { /* PathComponent starts with '\' and PathBuffer ends with '\' */ while (*PathComponent == OBJ_NAME_PATH_SEPARATOR) ++PathComponent; // Skip any backslash } Status = RtlStringCchCatW(PathBuffer, cchPathSize, PathComponent); if (!NT_SUCCESS(Status)) return Status; }
return Status;}
NTSTATUSCombinePathsV( OUT PWSTR PathBuffer, IN SIZE_T cchPathSize, IN ULONG NumberOfPathComponents, IN va_list PathComponentsList){ if (cchPathSize < 1) return STATUS_SUCCESS;
*PathBuffer = UNICODE_NULL; return ConcatPathsV(PathBuffer, cchPathSize, NumberOfPathComponents, PathComponentsList);}
NTSTATUSConcatPaths( IN OUT PWSTR PathBuffer, IN SIZE_T cchPathSize, IN ULONG NumberOfPathComponents, IN /* PCWSTR */ ...){ NTSTATUS Status; va_list PathComponentsList;
if (cchPathSize < 1) return STATUS_SUCCESS;
va_start(PathComponentsList, NumberOfPathComponents); Status = ConcatPathsV(PathBuffer, cchPathSize, NumberOfPathComponents, PathComponentsList); va_end(PathComponentsList);
return Status;}
NTSTATUSCombinePaths( OUT PWSTR PathBuffer, IN SIZE_T cchPathSize, IN ULONG NumberOfPathComponents, IN /* PCWSTR */ ...){ NTSTATUS Status; va_list PathComponentsList;
if (cchPathSize < 1) return STATUS_SUCCESS;
*PathBuffer = UNICODE_NULL;
va_start(PathComponentsList, NumberOfPathComponents); Status = CombinePathsV(PathBuffer, cchPathSize, NumberOfPathComponents, PathComponentsList); va_end(PathComponentsList);
return Status;}
BOOLEANDoesPathExist_UStr( _In_opt_ HANDLE RootDirectory, _In_ PCUNICODE_STRING PathName, _In_ BOOLEAN IsDirectory){ NTSTATUS Status; HANDLE FileHandle; OBJECT_ATTRIBUTES ObjectAttributes; IO_STATUS_BLOCK IoStatusBlock;
InitializeObjectAttributes(&ObjectAttributes, (PUNICODE_STRING)PathName, OBJ_CASE_INSENSITIVE, RootDirectory, NULL);
Status = NtOpenFile(&FileHandle, IsDirectory ? (FILE_LIST_DIRECTORY | SYNCHRONIZE) : FILE_GENERIC_READ, // Contains SYNCHRONIZE &ObjectAttributes, &IoStatusBlock, FILE_SHARE_READ | FILE_SHARE_WRITE, FILE_SYNCHRONOUS_IO_NONALERT | (IsDirectory ? FILE_DIRECTORY_FILE : FILE_NON_DIRECTORY_FILE)); if (NT_SUCCESS(Status)) { NtClose(FileHandle); } else { DPRINT("Failed to open %s '%wZ', Status 0x%08lx\n", IsDirectory ? "directory" : "file", PathName, Status); }
return NT_SUCCESS(Status);}
BOOLEANDoesPathExist( _In_opt_ HANDLE RootDirectory, _In_ PCWSTR PathName, _In_ BOOLEAN IsDirectory){ UNICODE_STRING PathNameU; RtlInitUnicodeString(&PathNameU, PathName); return DoesPathExist_UStr(RootDirectory, &PathNameU, IsDirectory);}
// FIXME: DEPRECATED! HACKish function that needs to be deprecated!BOOLEANDoesFileExist_2( IN PCWSTR PathName OPTIONAL, IN PCWSTR FileName){ WCHAR FullName[MAX_PATH]; CombinePaths(FullName, ARRAYSIZE(FullName), 2, PathName, FileName); return DoesFileExist(NULL, FullName);}
/* * The format of NtPath should be: * \Device\HarddiskXXX\PartitionYYY[\path] , * where XXX and YYY respectively represent the hard disk and partition numbers, * and [\path] represent an optional path (separated by '\\'). * * If a NT path of such a form is correctly parsed, the function returns respectively: * - in pDiskNumber: the hard disk number XXX, * - in pPartNumber: the partition number YYY, * - in PathComponent: pointer value (inside NtPath) to the beginning of \path. * * NOTE: The function does not accept leading whitespace. */BOOLEANNtPathToDiskPartComponents( IN PCWSTR NtPath, OUT PULONG pDiskNumber, OUT PULONG pPartNumber, OUT PCWSTR* PathComponent OPTIONAL){ ULONG DiskNumber, PartNumber; PCWSTR Path;
*pDiskNumber = 0; *pPartNumber = 0; if (PathComponent) *PathComponent = NULL;
Path = NtPath;
if (_wcsnicmp(Path, L"\\Device\\Harddisk", 16) != 0) { /* The NT path doesn't start with the prefix string, thus it cannot be a hard disk device path */ DPRINT1("'%S' : Not a possible hard disk device.\n", NtPath); return FALSE; }
Path += 16;
/* A number must be present now */ if (!iswdigit(*Path)) { DPRINT1("'%S' : expected a number! Not a regular hard disk device.\n", Path); return FALSE; } DiskNumber = wcstoul(Path, (PWSTR*)&Path, 10);
/* Either NULL termination, or a path separator must be present now */ if (*Path && *Path != OBJ_NAME_PATH_SEPARATOR) { DPRINT1("'%S' : expected a path separator!\n", Path); return FALSE; }
if (!*Path) { DPRINT1("The path only specified a hard disk (and nothing else, like a partition...), so we stop there.\n"); goto Quit; }
/* Here, *Path == L'\\' */
if (_wcsnicmp(Path, L"\\Partition", 10) != 0) { /* Actually, \Partition is optional so, if we don't have it, we still return success. Or should we? */ DPRINT1("'%S' : unexpected format!\n", NtPath); goto Quit; }
Path += 10;
/* A number must be present now */ if (!iswdigit(*Path)) { /* If we don't have a number it means this part of path is actually not a partition specifier, so we shouldn't fail either. Or should we? */ DPRINT1("'%S' : expected a number!\n", Path); goto Quit; } PartNumber = wcstoul(Path, (PWSTR*)&Path, 10);
/* Either NULL termination, or a path separator must be present now */ if (*Path && *Path != OBJ_NAME_PATH_SEPARATOR) { /* We shouldn't fail here because it just means this part of path is actually not a partition specifier. Or should we? */ DPRINT1("'%S' : expected a path separator!\n", Path); goto Quit; }
/* OK, here we really have a partition specifier: return its number */ *pPartNumber = PartNumber;
Quit: /* Return the disk number */ *pDiskNumber = DiskNumber;
/* Return the path component also, if the user wants it */ if (PathComponent) *PathComponent = Path;
return TRUE;}
/** * @brief * Opens and maps a file in memory. * * @param[in] RootDirectory * @param[in] PathNameToFile * Path to the file, either in absolute form, or relative to the opened * root directory given by the RootDirectory handle. * * @param[out] FileHandle * An optional pointer to a variable receiving a handle to the opened file. * If NULL, the underlying file handle is closed. * * @param[out] FileSize * An optional pointer to a variable receiving the size of the opened file. * * @param[out] SectionHandle * A pointer to a variable receiving a handle to a section mapping the file. * * @param[out] BaseAddress * A pointer to a variable receiving the address where the file is mapped. * * @param[in] ReadWriteAccess * A boolean variable specifying whether to map the file for read and write * access (TRUE), or read-only access (FALSE). * * @return * STATUS_SUCCESS in case of success, or a status code in case of error. **/NTSTATUSOpenAndMapFile( _In_opt_ HANDLE RootDirectory, _In_ PCWSTR PathNameToFile, _Out_opt_ PHANDLE FileHandle, _Out_opt_ PULONG FileSize, _Out_ PHANDLE SectionHandle, _Out_ PVOID* BaseAddress, _In_ BOOLEAN ReadWriteAccess){ NTSTATUS Status; UNICODE_STRING FileName; OBJECT_ATTRIBUTES ObjectAttributes; IO_STATUS_BLOCK IoStatusBlock; HANDLE LocalFileHandle;
/* Open the file */ RtlInitUnicodeString(&FileName, PathNameToFile); InitializeObjectAttributes(&ObjectAttributes, &FileName, OBJ_CASE_INSENSITIVE, RootDirectory, NULL);
if (FileHandle) *FileHandle = NULL; Status = NtOpenFile(&LocalFileHandle, FILE_GENERIC_READ | // Contains SYNCHRONIZE (ReadWriteAccess ? FILE_GENERIC_WRITE : 0), &ObjectAttributes, &IoStatusBlock, FILE_SHARE_READ, FILE_SYNCHRONOUS_IO_NONALERT | FILE_NON_DIRECTORY_FILE); if (!NT_SUCCESS(Status)) { DPRINT1("Failed to open file '%wZ' (Status 0x%08lx)\n", &FileName, Status); return Status; }
if (FileSize) { /* Query the file size */ FILE_STANDARD_INFORMATION FileInfo; Status = NtQueryInformationFile(LocalFileHandle, &IoStatusBlock, &FileInfo, sizeof(FileInfo), FileStandardInformation); if (!NT_SUCCESS(Status)) { DPRINT("NtQueryInformationFile() failed (Status 0x%08lx)\n", Status); goto Quit; }
if (FileInfo.EndOfFile.HighPart != 0) DPRINT1("WARNING!! The file '%wZ' is too large!\n", &FileName);
*FileSize = FileInfo.EndOfFile.LowPart; DPRINT("File size: %lu\n", *FileSize); }
/* Map the whole file into memory */ Status = MapFile(LocalFileHandle, SectionHandle, BaseAddress, ReadWriteAccess); if (!NT_SUCCESS(Status)) { DPRINT1("Failed to map file '%wZ' (Status 0x%08lx)\n", &FileName, Status); goto Quit; }
Quit: /* If we succeeded, return the opened file handle if needed. * If we failed or the caller does not need the handle, close it now. */ if (NT_SUCCESS(Status) && FileHandle) *FileHandle = LocalFileHandle; else NtClose(LocalFileHandle);
return Status;}
/** * @brief * Maps an opened file in memory. * * @param[in] FileHandle * A handle to an opened file to map. * * @param[out] SectionHandle * A pointer to a variable receiving a handle to a section mapping the file. * * @param[out] BaseAddress * A pointer to a variable receiving the address where the file is mapped. * * @param[in] ReadWriteAccess * A boolean variable specifying whether to map the file for read and write * access (TRUE), or read-only access (FALSE). * * @return * STATUS_SUCCESS in case of success, or a status code in case of error. **/NTSTATUSMapFile( _In_ HANDLE FileHandle, _Out_ PHANDLE SectionHandle, _Out_ PVOID* BaseAddress, _In_ BOOLEAN ReadWriteAccess){ NTSTATUS Status; ULONG SectionPageProtection; SIZE_T ViewSize; PVOID ViewBase;
SectionPageProtection = (ReadWriteAccess ? PAGE_READWRITE : PAGE_READONLY);
/* Create the section */ *SectionHandle = NULL; Status = NtCreateSection(SectionHandle, STANDARD_RIGHTS_REQUIRED | SECTION_QUERY | SECTION_MAP_READ | (ReadWriteAccess ? SECTION_MAP_WRITE : 0), NULL, NULL, SectionPageProtection, SEC_COMMIT /* | SEC_IMAGE (_NO_EXECUTE) */, FileHandle); if (!NT_SUCCESS(Status)) { DPRINT1("Failed to create a memory section for file 0x%p (Status 0x%08lx)\n", FileHandle, Status); return Status; }
/* Map the section */ ViewSize = 0; ViewBase = NULL; Status = NtMapViewOfSection(*SectionHandle, NtCurrentProcess(), &ViewBase, 0, 0, NULL, &ViewSize, ViewShare, 0, SectionPageProtection); if (!NT_SUCCESS(Status)) { DPRINT1("Failed to map a view for file 0x%p (Status 0x%08lx)\n", FileHandle, Status); NtClose(*SectionHandle); *SectionHandle = NULL; return Status; }
*BaseAddress = ViewBase; return STATUS_SUCCESS;}
/** * @brief * Unmaps a mapped file by section. * * @param[in] SectionHandle * The handle to the section mapping the file. * * @param[in] BaseAddress * The base address where the file is mapped. * * @return * TRUE if the file was successfully unmapped; FALSE if an error occurred. **/BOOLEANUnMapFile( _In_ HANDLE SectionHandle, _In_ PVOID BaseAddress){ NTSTATUS Status; BOOLEAN Success = TRUE;
Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress); if (!NT_SUCCESS(Status)) { DPRINT1("NtUnmapViewOfSection(0x%p) failed (Status 0x%08lx)\n", BaseAddress, Status); Success = FALSE; } Status = NtClose(SectionHandle); if (!NT_SUCCESS(Status)) { DPRINT1("NtClose(0x%p) failed (Status 0x%08lx)\n", SectionHandle, Status); Success = FALSE; }
return Success;}
/* EOF */