mirror of https://github.com/tongzx/nt5src
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.
1513 lines
30 KiB
1513 lines
30 KiB
/*++
|
|
|
|
Copyright (c) 1997 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
basefile.c
|
|
|
|
Abstract:
|
|
|
|
Contains simple wrappers for commonly used file i/o functions.
|
|
|
|
Author:
|
|
|
|
Marc R. Whitten (marcw) 02-Sep-1999
|
|
|
|
Revision History:
|
|
|
|
<alias> <date> <comments>
|
|
|
|
--*/
|
|
|
|
#include "pch.h"
|
|
|
|
//
|
|
// Includes
|
|
//
|
|
|
|
// None
|
|
|
|
#define DBG_BASEFILE "File Utils"
|
|
|
|
//
|
|
// Strings
|
|
//
|
|
|
|
// None
|
|
|
|
//
|
|
// Constants
|
|
//
|
|
|
|
// None
|
|
|
|
//
|
|
// Macros
|
|
//
|
|
|
|
// None
|
|
|
|
//
|
|
// Types
|
|
//
|
|
|
|
// None
|
|
|
|
//
|
|
// Globals
|
|
//
|
|
|
|
// None
|
|
|
|
//
|
|
// Macro expansion list
|
|
//
|
|
|
|
// None
|
|
|
|
//
|
|
// Private function prototypes
|
|
//
|
|
|
|
// None
|
|
|
|
//
|
|
// Macro expansion definition
|
|
//
|
|
|
|
// None
|
|
|
|
//
|
|
// Code
|
|
//
|
|
|
|
|
|
BOOL
|
|
WriteFileStringA (
|
|
IN HANDLE File,
|
|
IN PCSTR String
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Writes a DBCS string to the specified file.
|
|
|
|
Arguments:
|
|
|
|
File - Specifies the file handle that was opened with write access.
|
|
|
|
String - Specifies the nul-terminated string to write to the file.
|
|
|
|
Return Value:
|
|
|
|
TRUE if successful, FALSE if an error occurred. Call GetLastError
|
|
for error condition.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD DontCare;
|
|
|
|
return WriteFile (File, String, ByteCountA (String), &DontCare, NULL);
|
|
}
|
|
|
|
|
|
BOOL
|
|
WriteFileStringW (
|
|
IN HANDLE File,
|
|
IN PCWSTR String
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Converts a UNICODE string to DBCS, then Writes it to the specified file.
|
|
|
|
Arguments:
|
|
|
|
File - Specifies the file handle that was opened with write access.
|
|
|
|
String - Specifies the UNICODE nul-terminated string to convert and
|
|
write to the file.
|
|
|
|
Return Value:
|
|
|
|
TRUE if successful, FALSE if an error occurred. Call GetLastError for
|
|
error condition.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD DontCare;
|
|
PCSTR AnsiVersion;
|
|
BOOL b;
|
|
|
|
AnsiVersion = ConvertWtoA (String);
|
|
if (!AnsiVersion) {
|
|
return FALSE;
|
|
}
|
|
|
|
b = WriteFile (File, AnsiVersion, ByteCountA (AnsiVersion), &DontCare, NULL);
|
|
|
|
FreeConvertedStr (AnsiVersion);
|
|
|
|
return b;
|
|
}
|
|
|
|
BOOL
|
|
DoesFileExistExA(
|
|
IN PCSTR FileName,
|
|
OUT PWIN32_FIND_DATAA FindData OPTIONAL
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Determine if a file exists and is accessible.
|
|
Errormode is set (and then restored) so the user will not see
|
|
any pop-ups.
|
|
|
|
Arguments:
|
|
|
|
FileName - supplies full path of file to check for existance.
|
|
|
|
FindData - if specified, receives find data for the file.
|
|
|
|
Return Value:
|
|
|
|
TRUE if the file exists and is accessible.
|
|
FALSE if not. GetLastError() returns extended error info.
|
|
|
|
--*/
|
|
|
|
{
|
|
WIN32_FIND_DATAA ourFindData;
|
|
HANDLE FindHandle;
|
|
UINT OldMode;
|
|
DWORD Error;
|
|
|
|
if (!FindData) {
|
|
// Win95 GetFileAttributes does not return a failure if FileName is NULL
|
|
if (FileName == NULL) {
|
|
return FALSE;
|
|
} else {
|
|
return GetFileAttributesA (FileName) != 0xffffffff;
|
|
}
|
|
}
|
|
|
|
OldMode = SetErrorMode(SEM_FAILCRITICALERRORS);
|
|
|
|
FindHandle = FindFirstFileA(FileName, &ourFindData);
|
|
|
|
if (FindHandle == INVALID_HANDLE_VALUE) {
|
|
Error = GetLastError();
|
|
} else {
|
|
FindClose(FindHandle);
|
|
*FindData = ourFindData;
|
|
Error = NO_ERROR;
|
|
}
|
|
|
|
SetErrorMode(OldMode);
|
|
|
|
SetLastError(Error);
|
|
return (Error == NO_ERROR);
|
|
}
|
|
|
|
|
|
BOOL
|
|
DoesFileExistExW (
|
|
IN PCWSTR FileName,
|
|
OUT PWIN32_FIND_DATAW FindData OPTIONAL
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Determine if a file exists and is accessible.
|
|
Errormode is set (and then restored) so the user will not see
|
|
any pop-ups.
|
|
|
|
Arguments:
|
|
|
|
FileName - supplies full path of file to check for existance.
|
|
|
|
FindData - if specified, receives find data for the file.
|
|
|
|
Return Value:
|
|
|
|
TRUE if the file exists and is accessible.
|
|
FALSE if not. GetLastError() returns extended error info.
|
|
|
|
--*/
|
|
|
|
{
|
|
WIN32_FIND_DATAW ourFindData;
|
|
HANDLE FindHandle;
|
|
UINT OldMode;
|
|
DWORD Error;
|
|
|
|
if (!FindData) {
|
|
// Win95 GetFileAttributes does not return a failure if FileName is NULL
|
|
if (FileName == NULL) {
|
|
return FALSE;
|
|
} else {
|
|
return GetFileAttributesW (FileName) != 0xffffffff;
|
|
}
|
|
}
|
|
|
|
OldMode = SetErrorMode(SEM_FAILCRITICALERRORS);
|
|
|
|
FindHandle = FindFirstFileW(FileName,&ourFindData);
|
|
|
|
if (FindHandle == INVALID_HANDLE_VALUE) {
|
|
Error = GetLastError();
|
|
} else {
|
|
FindClose(FindHandle);
|
|
*FindData = ourFindData;
|
|
Error = NO_ERROR;
|
|
}
|
|
|
|
SetErrorMode(OldMode);
|
|
|
|
SetLastError(Error);
|
|
return (Error == NO_ERROR);
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
PathIsDirectory determines if a path identifies an accessible directory.
|
|
|
|
Arguments:
|
|
|
|
PathSpec - Specifies the full path.
|
|
|
|
Return Value:
|
|
|
|
TRUE if the path identifies a directory.
|
|
FALSE if not. GetLastError() returns extended error info.
|
|
|
|
--*/
|
|
|
|
BOOL
|
|
BfPathIsDirectoryA (
|
|
IN PCSTR PathSpec
|
|
)
|
|
{
|
|
DWORD attribs;
|
|
|
|
MYASSERT (PathSpec);
|
|
if (!PathSpec) {
|
|
return FALSE;
|
|
}
|
|
attribs = GetFileAttributesA (PathSpec);
|
|
return attribs != (DWORD)-1 && (attribs & FILE_ATTRIBUTE_DIRECTORY);
|
|
}
|
|
|
|
BOOL
|
|
BfPathIsDirectoryW (
|
|
IN PCWSTR PathSpec
|
|
)
|
|
{
|
|
DWORD attribs;
|
|
|
|
MYASSERT (PathSpec);
|
|
if (!PathSpec) {
|
|
return FALSE;
|
|
}
|
|
attribs = GetFileAttributesW (PathSpec);
|
|
return attribs != (DWORD)-1 && (attribs & FILE_ATTRIBUTE_DIRECTORY);
|
|
}
|
|
|
|
|
|
PVOID
|
|
MapFileIntoMemoryExA (
|
|
IN PCSTR FileName,
|
|
OUT PHANDLE FileHandle,
|
|
OUT PHANDLE MapHandle,
|
|
IN BOOL WriteAccess
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
MapFileIntoMemoryA and MapFileIntoMemoryW map a file into memory. It does that
|
|
by opening the file, creating a mapping object and mapping opened file into
|
|
created mapping object. It returnes the address where the file is mapped and
|
|
also sets FileHandle and MapHandle variables to be used in order to unmap the
|
|
file when work is done.
|
|
|
|
Arguments:
|
|
|
|
FileName - the name of the file to be mapped into memory
|
|
FileHandle - will end keeping the file handle if the file was opened successfully
|
|
MapHandle - will end keeping the mapping object handle if this object was created successfully
|
|
|
|
Return Value:
|
|
|
|
NULL if function fails, a valid memory address if successfull
|
|
|
|
Comments:
|
|
|
|
If the return value is NULL you should call UnmapFile to release all allocated resources
|
|
|
|
--*/
|
|
|
|
{
|
|
PVOID fileImage = NULL;
|
|
|
|
//verify function parameters
|
|
if ((FileHandle == NULL) || (MapHandle == NULL)) {
|
|
return NULL;
|
|
}
|
|
|
|
//first thing. Try to open the file, read-only
|
|
*FileHandle = CreateFileA (
|
|
FileName,
|
|
WriteAccess?GENERIC_READ|GENERIC_WRITE:GENERIC_READ,
|
|
FILE_SHARE_READ,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL
|
|
);
|
|
|
|
if (*FileHandle == INVALID_HANDLE_VALUE) {
|
|
*FileHandle = NULL;
|
|
return NULL;
|
|
}
|
|
|
|
//now try to create a mapping object, read-only
|
|
*MapHandle = CreateFileMappingA (*FileHandle, NULL, WriteAccess?PAGE_READWRITE:PAGE_READONLY, 0, 0, NULL);
|
|
|
|
if (*MapHandle == NULL) {
|
|
CloseHandle (*FileHandle);
|
|
*FileHandle = NULL;
|
|
return NULL;
|
|
}
|
|
|
|
//one more thing to do: map view of file
|
|
fileImage = MapViewOfFile (*MapHandle, WriteAccess?FILE_MAP_WRITE:FILE_MAP_READ, 0, 0, 0);
|
|
|
|
return fileImage;
|
|
}
|
|
|
|
|
|
PVOID
|
|
MapFileIntoMemoryExW (
|
|
IN PCWSTR FileName,
|
|
OUT PHANDLE FileHandle,
|
|
OUT PHANDLE MapHandle,
|
|
IN BOOL WriteAccess
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
MapFileIntoMemoryA and MapFileIntoMemoryW map a file into memory. It does that
|
|
by opening the file, creating a mapping object and mapping opened file into
|
|
created mapping object. It returnes the address where the file is mapped and
|
|
also sets FileHandle and MapHandle variables to be used in order to unmap the
|
|
file when work is done.
|
|
|
|
Arguments:
|
|
|
|
FileName - the name of the file to be mapped into memory
|
|
FileHandle - will end keeping the file handle if the file was opened successfully
|
|
MapHandle - will end keeping the mapping object handle if this object was created successfully
|
|
|
|
Return Value:
|
|
|
|
NULL if function fails, a valid memory address if successfull
|
|
|
|
Comments:
|
|
|
|
If the return value is NULL you should call UnmapFile to release all allocated resources
|
|
|
|
--*/
|
|
|
|
{
|
|
PVOID fileImage = NULL;
|
|
|
|
//verify function parameters
|
|
if ((FileHandle == NULL) || (MapHandle == NULL)) {
|
|
return NULL;
|
|
}
|
|
|
|
//first thing. Try to open the file, read-only
|
|
*FileHandle = CreateFileW (
|
|
FileName,
|
|
WriteAccess?GENERIC_READ|GENERIC_WRITE:GENERIC_READ,
|
|
FILE_SHARE_READ,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL
|
|
);
|
|
|
|
if (*FileHandle == INVALID_HANDLE_VALUE) {
|
|
*FileHandle = NULL;
|
|
return NULL;
|
|
}
|
|
|
|
//now try to create a mapping object, read-only
|
|
*MapHandle = CreateFileMappingW (*FileHandle, NULL, WriteAccess?PAGE_READWRITE:PAGE_READONLY, 0, 0, NULL);
|
|
|
|
if (*MapHandle == NULL) {
|
|
CloseHandle (*FileHandle);
|
|
*FileHandle = NULL;
|
|
return NULL;
|
|
}
|
|
|
|
//one more thing to do: map view of file
|
|
fileImage = MapViewOfFile (*MapHandle, WriteAccess?FILE_MAP_WRITE:FILE_MAP_READ, 0, 0, 0);
|
|
|
|
return fileImage;
|
|
}
|
|
|
|
|
|
BOOL
|
|
UnmapFile (
|
|
IN PCVOID FileImage,
|
|
IN HANDLE MapHandle,
|
|
IN HANDLE FileHandle
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
UnmapFile is used to release all resources allocated by MapFileIntoMemory.
|
|
|
|
Arguments:
|
|
|
|
FileImage - image of the mapped file as returned by MapFileIntoMemory
|
|
MapHandle - handle of the mapping object as returned by MapFileIntoMemory
|
|
FileHandle - handle of the file as returned by MapFileIntoMemory
|
|
|
|
Return Value:
|
|
|
|
TRUE if successfull, FALSE if not
|
|
|
|
--*/
|
|
|
|
{
|
|
BOOL result = TRUE;
|
|
|
|
//if FileImage is a valid pointer then try to unmap file
|
|
if (FileImage != NULL) {
|
|
if (UnmapViewOfFile (FileImage) == 0) {
|
|
result = FALSE;
|
|
}
|
|
}
|
|
|
|
//if mapping object is valid then try to delete it
|
|
if (MapHandle != NULL) {
|
|
if (CloseHandle (MapHandle) == 0) {
|
|
result = FALSE;
|
|
}
|
|
}
|
|
|
|
//if file handle is valid then try to close the file
|
|
if (FileHandle != INVALID_HANDLE_VALUE) {
|
|
if (CloseHandle (FileHandle) == 0) {
|
|
result = FALSE;
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfGetTempFileNameExA (
|
|
OUT PSTR Buffer,
|
|
IN UINT BufferTchars,
|
|
IN PCSTR Prefix
|
|
)
|
|
{
|
|
CHAR tempPath[MAX_MBCHAR_PATH];
|
|
CHAR tempFile[MAX_MBCHAR_PATH];
|
|
UINT tchars;
|
|
PSTR p;
|
|
|
|
if (!GetTempPathA (ARRAYSIZE(tempPath), tempPath)) {
|
|
return FALSE;
|
|
}
|
|
|
|
p = _mbsrchr (tempPath, '\\');
|
|
if (p && !p[1]) {
|
|
*p = 0;
|
|
}
|
|
|
|
if (!DoesFileExistA (tempPath)) {
|
|
BfCreateDirectoryA (tempPath);
|
|
}
|
|
|
|
if (BufferTchars >= MAX_PATH) {
|
|
if (!GetTempFileNameA (tempPath, Prefix, 0, Buffer)) {
|
|
return FALSE;
|
|
}
|
|
} else {
|
|
if (!GetTempFileNameA (tempPath, Prefix, 0, tempFile)) {
|
|
DWORD err = GetLastError ();
|
|
return FALSE;
|
|
}
|
|
|
|
tchars = TcharCountA (tempFile) + 1;
|
|
|
|
if (tchars > BufferTchars) {
|
|
DEBUGMSG ((DBG_ERROR, "Can't get temp file name -- buffer too small"));
|
|
return FALSE;
|
|
}
|
|
|
|
CopyMemory (Buffer, tempFile, tchars * sizeof (CHAR));
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfGetTempFileNameExW (
|
|
OUT PWSTR Buffer,
|
|
IN UINT BufferTchars,
|
|
IN PCWSTR Prefix
|
|
)
|
|
{
|
|
WCHAR tempPath[MAX_WCHAR_PATH];
|
|
WCHAR tempFile[MAX_WCHAR_PATH];
|
|
UINT tchars;
|
|
PWSTR p;
|
|
|
|
if (!GetTempPathW (ARRAYSIZE(tempPath), tempPath)) {
|
|
return FALSE;
|
|
}
|
|
|
|
p = wcsrchr (tempPath, '\\');
|
|
if (p && !p[1]) {
|
|
*p = 0;
|
|
}
|
|
|
|
if (BufferTchars >= MAX_PATH) {
|
|
if (!GetTempFileNameW (tempPath, Prefix, 0, Buffer)) {
|
|
return FALSE;
|
|
}
|
|
} else {
|
|
if (!GetTempFileNameW (tempPath, Prefix, 0, tempFile)) {
|
|
return FALSE;
|
|
}
|
|
|
|
tchars = TcharCountW (tempFile);
|
|
|
|
if (tchars > BufferTchars) {
|
|
DEBUGMSG ((DBG_ERROR, "Can't get temp file name -- buffer too small"));
|
|
return FALSE;
|
|
}
|
|
|
|
CopyMemory (Buffer, tempFile, tchars * sizeof (WCHAR));
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfGetTempDirectoryExA (
|
|
OUT PSTR Buffer,
|
|
IN UINT BufferTchars,
|
|
IN PCSTR Prefix
|
|
)
|
|
{
|
|
BOOL result = FALSE;
|
|
|
|
result = BfGetTempFileNameExA (Buffer, BufferTchars, Prefix);
|
|
|
|
if (result) {
|
|
if (!DeleteFileA (Buffer)) {
|
|
return FALSE;
|
|
}
|
|
if (!CreateDirectoryA (Buffer, NULL)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfGetTempDirectoryExW (
|
|
OUT PWSTR Buffer,
|
|
IN UINT BufferTchars,
|
|
IN PCWSTR Prefix
|
|
)
|
|
{
|
|
BOOL result = FALSE;
|
|
|
|
result = BfGetTempFileNameExW (Buffer, BufferTchars, Prefix);
|
|
|
|
if (result) {
|
|
if (!DeleteFileW (Buffer)) {
|
|
return FALSE;
|
|
}
|
|
if (!CreateDirectoryW (Buffer, NULL)) {
|
|
return FALSE;
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
|
|
HANDLE
|
|
BfGetTempFile (
|
|
VOID
|
|
)
|
|
{
|
|
CHAR tempFile[MAX_MBCHAR_PATH];
|
|
HANDLE file;
|
|
|
|
if (!BfGetTempFileNameA (tempFile, ARRAYSIZE(tempFile))) {
|
|
return NULL;
|
|
}
|
|
|
|
file = CreateFileA (
|
|
tempFile,
|
|
GENERIC_READ|GENERIC_WRITE,
|
|
0,
|
|
NULL,
|
|
CREATE_ALWAYS,
|
|
FILE_ATTRIBUTE_NORMAL|FILE_FLAG_DELETE_ON_CLOSE,
|
|
NULL
|
|
);
|
|
|
|
if (file == INVALID_HANDLE_VALUE) {
|
|
file = NULL;
|
|
}
|
|
|
|
return file;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfSetFilePointer (
|
|
IN HANDLE File,
|
|
IN LONGLONG Offset
|
|
)
|
|
{
|
|
LARGE_INTEGER li;
|
|
|
|
li.QuadPart = Offset;
|
|
|
|
li.LowPart = SetFilePointer (File, li.LowPart, &li.HighPart, FILE_BEGIN);
|
|
|
|
if (li.LowPart == 0xFFFFFFFF && GetLastError() != NO_ERROR) {
|
|
li.QuadPart = -1;
|
|
}
|
|
|
|
return li.QuadPart != -1;
|
|
}
|
|
|
|
|
|
HANDLE
|
|
BfOpenReadFileA (
|
|
IN PCSTR FileName
|
|
)
|
|
{
|
|
HANDLE handle;
|
|
|
|
handle = CreateFileA (
|
|
FileName,
|
|
GENERIC_READ,
|
|
FILE_SHARE_READ|FILE_SHARE_WRITE,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL
|
|
);
|
|
|
|
if (handle == INVALID_HANDLE_VALUE) {
|
|
handle = NULL;
|
|
}
|
|
|
|
return handle;
|
|
}
|
|
|
|
|
|
HANDLE
|
|
BfOpenReadFileW (
|
|
IN PCWSTR FileName
|
|
)
|
|
{
|
|
HANDLE handle;
|
|
|
|
handle = CreateFileW (
|
|
FileName,
|
|
GENERIC_READ,
|
|
FILE_SHARE_READ|FILE_SHARE_WRITE,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL
|
|
);
|
|
|
|
if (handle == INVALID_HANDLE_VALUE) {
|
|
handle = NULL;
|
|
}
|
|
|
|
return handle;
|
|
}
|
|
|
|
HANDLE
|
|
BfOpenFileA (
|
|
IN PCSTR FileName
|
|
)
|
|
{
|
|
HANDLE handle;
|
|
|
|
handle = CreateFileA (
|
|
FileName,
|
|
GENERIC_READ|GENERIC_WRITE,
|
|
0,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL
|
|
);
|
|
|
|
if (handle == INVALID_HANDLE_VALUE) {
|
|
handle = NULL;
|
|
}
|
|
|
|
return handle;
|
|
}
|
|
|
|
|
|
HANDLE
|
|
BfOpenFileW (
|
|
IN PCWSTR FileName
|
|
)
|
|
{
|
|
HANDLE handle;
|
|
|
|
handle = CreateFileW (
|
|
FileName,
|
|
GENERIC_READ|GENERIC_WRITE,
|
|
0,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL
|
|
);
|
|
|
|
if (handle == INVALID_HANDLE_VALUE) {
|
|
handle = NULL;
|
|
}
|
|
|
|
return handle;
|
|
}
|
|
|
|
HANDLE
|
|
BfCreateFileA (
|
|
IN PCSTR FileName
|
|
)
|
|
{
|
|
HANDLE handle;
|
|
|
|
handle = CreateFileA (
|
|
FileName,
|
|
GENERIC_READ|GENERIC_WRITE,
|
|
0,
|
|
NULL,
|
|
CREATE_ALWAYS,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL
|
|
);
|
|
|
|
if (handle == INVALID_HANDLE_VALUE) {
|
|
handle = NULL;
|
|
}
|
|
|
|
return handle;
|
|
|
|
}
|
|
|
|
HANDLE
|
|
BfCreateFileW (
|
|
IN PCWSTR FileName
|
|
)
|
|
{
|
|
HANDLE handle;
|
|
|
|
handle = CreateFileW (
|
|
FileName,
|
|
GENERIC_READ|GENERIC_WRITE,
|
|
0,
|
|
NULL,
|
|
CREATE_ALWAYS,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL
|
|
);
|
|
|
|
if (handle == INVALID_HANDLE_VALUE) {
|
|
handle = NULL;
|
|
}
|
|
|
|
return handle;
|
|
}
|
|
|
|
HANDLE
|
|
BfCreateSharedFileA (
|
|
IN PCSTR FileName
|
|
)
|
|
{
|
|
HANDLE handle;
|
|
|
|
handle = CreateFileA (
|
|
FileName,
|
|
GENERIC_READ|GENERIC_WRITE,
|
|
FILE_SHARE_READ|FILE_SHARE_WRITE,
|
|
NULL,
|
|
CREATE_ALWAYS,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL
|
|
);
|
|
|
|
if (handle == INVALID_HANDLE_VALUE) {
|
|
handle = NULL;
|
|
}
|
|
|
|
return handle;
|
|
}
|
|
|
|
HANDLE
|
|
BfCreateSharedFileW (
|
|
IN PCWSTR FileName
|
|
)
|
|
{
|
|
HANDLE handle;
|
|
|
|
handle = CreateFileW (
|
|
FileName,
|
|
GENERIC_READ|GENERIC_WRITE,
|
|
FILE_SHARE_READ|FILE_SHARE_WRITE,
|
|
NULL,
|
|
CREATE_ALWAYS,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL
|
|
);
|
|
|
|
if (handle == INVALID_HANDLE_VALUE) {
|
|
handle = NULL;
|
|
}
|
|
|
|
return handle;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfSetSizeOfFile (
|
|
HANDLE File,
|
|
LONGLONG Size
|
|
)
|
|
{
|
|
if (!BfSetFilePointer (File, Size)) {
|
|
return FALSE;
|
|
}
|
|
|
|
return SetEndOfFile (File);
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfGoToEndOfFile (
|
|
IN HANDLE File,
|
|
OUT PLONGLONG FileSize OPTIONAL
|
|
)
|
|
{
|
|
LARGE_INTEGER li;
|
|
|
|
li.HighPart = 0;
|
|
li.LowPart = SetFilePointer (File, 0, &li.HighPart, FILE_END);
|
|
|
|
if (li.LowPart == 0xFFFFFFFF && GetLastError() != NO_ERROR) {
|
|
li.QuadPart = -1;
|
|
} else if (FileSize) {
|
|
*FileSize = li.QuadPart;
|
|
}
|
|
|
|
return li.QuadPart != -1;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfGetFilePointer (
|
|
IN HANDLE File,
|
|
OUT PLONGLONG FilePointer OPTIONAL
|
|
)
|
|
{
|
|
LARGE_INTEGER li;
|
|
|
|
li.HighPart = 0;
|
|
li.LowPart = SetFilePointer (File, 0, &li.HighPart, FILE_CURRENT);
|
|
|
|
if (li.LowPart == 0xFFFFFFFF && GetLastError() != NO_ERROR) {
|
|
li.QuadPart = -1;
|
|
} else if (FilePointer) {
|
|
*FilePointer = li.QuadPart;
|
|
}
|
|
|
|
return li.QuadPart != -1;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfReadFile (
|
|
IN HANDLE File,
|
|
OUT PBYTE Buffer,
|
|
IN UINT BytesToRead
|
|
)
|
|
{
|
|
DWORD bytesRead;
|
|
|
|
if (!ReadFile (File, Buffer, BytesToRead, &bytesRead, NULL)) {
|
|
return FALSE;
|
|
}
|
|
|
|
return bytesRead == BytesToRead;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfWriteFile (
|
|
IN HANDLE File,
|
|
OUT PCBYTE Buffer,
|
|
IN UINT BytesToWrite
|
|
)
|
|
{
|
|
DWORD bytesWritten;
|
|
|
|
if (!WriteFile (File, Buffer, BytesToWrite, &bytesWritten, NULL)) {
|
|
return FALSE;
|
|
}
|
|
|
|
return bytesWritten == BytesToWrite;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfCreateDirectoryExA (
|
|
IN PCSTR FullPath,
|
|
IN BOOL CreateLastSegment
|
|
)
|
|
{
|
|
PSTR pathCopy;
|
|
PSTR p;
|
|
BOOL b = TRUE;
|
|
|
|
pathCopy = DuplicatePathStringA (FullPath, 0);
|
|
|
|
//
|
|
// Advance past first directory
|
|
//
|
|
|
|
if (pathCopy[1] == ':' && pathCopy[2] == '\\') {
|
|
//
|
|
// <drive>:\ case
|
|
//
|
|
|
|
p = _mbschr (&pathCopy[3], '\\');
|
|
|
|
} else if (pathCopy[0] == '\\' && pathCopy[1] == '\\') {
|
|
|
|
//
|
|
// UNC case
|
|
//
|
|
|
|
p = _mbschr (pathCopy + 2, '\\');
|
|
if (p) {
|
|
p = _mbschr (p + 1, '\\');
|
|
if (p) {
|
|
p = _mbschr (p + 1, '\\');
|
|
}
|
|
}
|
|
|
|
} else {
|
|
|
|
//
|
|
// Relative dir case
|
|
//
|
|
|
|
p = _mbschr (pathCopy, '\\');
|
|
}
|
|
|
|
//
|
|
// Make all directories along the path
|
|
//
|
|
|
|
while (p) {
|
|
|
|
*p = 0;
|
|
b = CreateDirectoryA (pathCopy, NULL);
|
|
|
|
if (!b && GetLastError() == ERROR_ALREADY_EXISTS) {
|
|
b = TRUE;
|
|
}
|
|
|
|
if (!b) {
|
|
LOG ((LOG_ERROR, "Can't create %s", pathCopy));
|
|
break;
|
|
}
|
|
|
|
*p = '\\';
|
|
p = _mbschr (p + 1, '\\');
|
|
}
|
|
|
|
//
|
|
// At last, make the FullPath directory
|
|
//
|
|
|
|
if (b && CreateLastSegment) {
|
|
b = CreateDirectoryA (pathCopy, NULL);
|
|
|
|
if (!b && GetLastError() == ERROR_ALREADY_EXISTS) {
|
|
b = TRUE;
|
|
}
|
|
}
|
|
|
|
FreePathStringA (pathCopy);
|
|
|
|
return b;
|
|
}
|
|
|
|
|
|
BOOL
|
|
BfCreateDirectoryExW (
|
|
IN PCWSTR FullPath,
|
|
IN BOOL CreateLastSegment
|
|
)
|
|
{
|
|
PWSTR pathCopy;
|
|
PWSTR p;
|
|
BOOL b = TRUE;
|
|
|
|
pathCopy = DuplicatePathStringW (FullPath, 0);
|
|
|
|
//
|
|
// Advance past first directory
|
|
//
|
|
|
|
if (pathCopy[1] == L':' && pathCopy[2] == L'\\') {
|
|
//
|
|
// <drive>:\ case
|
|
//
|
|
|
|
p = wcschr (&pathCopy[3], L'\\');
|
|
|
|
} else if (pathCopy[0] == L'\\' && pathCopy[1] == L'\\') {
|
|
|
|
//
|
|
// UNC case
|
|
//
|
|
|
|
p = wcschr (pathCopy + 2, L'\\');
|
|
if (p) {
|
|
p = wcschr (p + 1, L'\\');
|
|
if (p) {
|
|
p = wcschr (p + 1, L'\\');
|
|
}
|
|
}
|
|
|
|
} else {
|
|
|
|
//
|
|
// Relative dir case
|
|
//
|
|
|
|
p = wcschr (pathCopy, L'\\');
|
|
}
|
|
|
|
//
|
|
// Make all directories along the path
|
|
//
|
|
|
|
while (p) {
|
|
|
|
*p = 0;
|
|
b = CreateDirectoryW (pathCopy, NULL);
|
|
|
|
if (!b && GetLastError() == ERROR_ALREADY_EXISTS) {
|
|
b = TRUE;
|
|
}
|
|
|
|
if (!b) {
|
|
break;
|
|
}
|
|
|
|
*p = L'\\';
|
|
p = wcschr (p + 1, L'\\');
|
|
}
|
|
|
|
//
|
|
// At last, make the FullPath directory
|
|
//
|
|
|
|
if (b && CreateLastSegment) {
|
|
b = CreateDirectoryW (pathCopy, NULL);
|
|
|
|
if (!b && GetLastError() == ERROR_ALREADY_EXISTS) {
|
|
b = TRUE;
|
|
}
|
|
}
|
|
|
|
FreePathStringW (pathCopy);
|
|
|
|
return b;
|
|
}
|
|
|
|
|
|
LONGLONG
|
|
BfGetFileSizeA (
|
|
IN PCSTR FileName
|
|
)
|
|
{
|
|
WIN32_FIND_DATAA fd;
|
|
LONGLONG l;
|
|
|
|
if (!DoesFileExistExA (FileName, &fd)) {
|
|
return 0;
|
|
}
|
|
|
|
l = ((LONGLONG) fd.nFileSizeHigh << 32) | fd.nFileSizeLow;
|
|
|
|
return l;
|
|
}
|
|
|
|
LONGLONG
|
|
BfGetFileSizeW (
|
|
IN PCWSTR FileName
|
|
)
|
|
{
|
|
WIN32_FIND_DATAW fd;
|
|
LONGLONG l;
|
|
|
|
if (!DoesFileExistExW (FileName, &fd)) {
|
|
return 0;
|
|
}
|
|
|
|
l = ((LONGLONG) fd.nFileSizeHigh << 32) | fd.nFileSizeLow;
|
|
|
|
return l;
|
|
}
|
|
|
|
PSTR
|
|
pGetFirstSegA (
|
|
IN PCSTR SrcFileName
|
|
)
|
|
{
|
|
if (SrcFileName [0] == '\\') {
|
|
SrcFileName ++;
|
|
if (SrcFileName [0] == '\\') {
|
|
SrcFileName ++;
|
|
}
|
|
return (_mbschr (SrcFileName, '\\'));
|
|
} else {
|
|
return (_mbschr (SrcFileName, '\\'));
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
pGetLongFileNameWorkerA (
|
|
IN PCSTR SrcFileName,
|
|
IN PGROWBUFFER GrowBuf
|
|
)
|
|
{
|
|
PSTR beginSegPtr;
|
|
PSTR endSegPtr;
|
|
WIN32_FIND_DATAA findData;
|
|
CHAR savedChar;
|
|
|
|
beginSegPtr = pGetFirstSegA (SrcFileName);
|
|
|
|
if (!beginSegPtr) {
|
|
GbAppendStringA (GrowBuf, SrcFileName);
|
|
return TRUE;
|
|
}
|
|
beginSegPtr = _mbsinc (beginSegPtr);
|
|
|
|
GbAppendStringABA (GrowBuf, SrcFileName, beginSegPtr);
|
|
|
|
while (beginSegPtr) {
|
|
endSegPtr = _mbschr (beginSegPtr, '\\');
|
|
if (!endSegPtr) {
|
|
endSegPtr = GetEndOfStringA (beginSegPtr);
|
|
MYASSERT (endSegPtr);
|
|
}
|
|
savedChar = *endSegPtr;
|
|
*endSegPtr = 0;
|
|
if (DoesFileExistExA (SrcFileName, &findData)) {
|
|
if (StringIMatchA (findData.cFileName, ".")) {
|
|
GbAppendStringA (GrowBuf, beginSegPtr);
|
|
} else {
|
|
GbAppendStringA (GrowBuf, findData.cFileName);
|
|
}
|
|
} else {
|
|
GbAppendStringABA (GrowBuf, beginSegPtr, endSegPtr);
|
|
}
|
|
*endSegPtr = savedChar;
|
|
if (savedChar) {
|
|
beginSegPtr = _mbsinc (endSegPtr);
|
|
GbAppendStringABA (GrowBuf, endSegPtr, beginSegPtr);
|
|
} else {
|
|
beginSegPtr = NULL;
|
|
}
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
PCSTR
|
|
BfGetLongFileNameA (
|
|
IN PCSTR SrcFileName
|
|
)
|
|
{
|
|
GROWBUFFER growBuf = INIT_GROWBUFFER;
|
|
PSTR srcFileName;
|
|
PCSTR result = NULL;
|
|
|
|
srcFileName = (PSTR)SanitizePathA (SrcFileName);
|
|
if (pGetLongFileNameWorkerA (srcFileName, &growBuf)) {
|
|
result = DuplicatePathStringA (growBuf.Buf, 0);
|
|
GbFree (&growBuf);
|
|
}
|
|
FreePathStringA (srcFileName);
|
|
return result;
|
|
}
|
|
|
|
BOOL
|
|
BfGetLongFileNameExA (
|
|
IN PCSTR SrcFileName,
|
|
IN PGROWBUFFER GrowBuff
|
|
)
|
|
{
|
|
PSTR srcFileName;
|
|
BOOL result;
|
|
|
|
srcFileName = (PSTR)SanitizePathA (SrcFileName);
|
|
result = pGetLongFileNameWorkerA (srcFileName, GrowBuff);
|
|
FreePathStringA (srcFileName);
|
|
|
|
return result;
|
|
}
|
|
|
|
PWSTR
|
|
pGetFirstSegW (
|
|
IN PCWSTR SrcFileName
|
|
)
|
|
{
|
|
if (SrcFileName [0] == L'\\') {
|
|
SrcFileName ++;
|
|
if (SrcFileName [0] == L'\\') {
|
|
SrcFileName ++;
|
|
}
|
|
return (wcschr (SrcFileName, L'\\'));
|
|
} else {
|
|
return (wcschr (SrcFileName, L'\\'));
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
pGetLongFileNameWorkerW (
|
|
IN PCWSTR SrcFileName,
|
|
IN PGROWBUFFER GrowBuf
|
|
)
|
|
{
|
|
PWSTR beginSegPtr;
|
|
PWSTR endSegPtr;
|
|
WIN32_FIND_DATAW findData;
|
|
WCHAR savedChar;
|
|
|
|
beginSegPtr = pGetFirstSegW (SrcFileName);
|
|
|
|
if (!beginSegPtr) {
|
|
GbAppendStringW (GrowBuf, SrcFileName);
|
|
return TRUE;
|
|
}
|
|
beginSegPtr ++;
|
|
|
|
GbAppendStringABW (GrowBuf, SrcFileName, beginSegPtr);
|
|
|
|
while (beginSegPtr) {
|
|
endSegPtr = wcschr (beginSegPtr, L'\\');
|
|
if (!endSegPtr) {
|
|
endSegPtr = GetEndOfStringW (beginSegPtr);
|
|
MYASSERT (endSegPtr);
|
|
}
|
|
savedChar = *endSegPtr;
|
|
*endSegPtr = 0;
|
|
if (DoesFileExistExW (SrcFileName, &findData)) {
|
|
GbAppendStringW (GrowBuf, findData.cFileName);
|
|
} else {
|
|
GbAppendStringABW (GrowBuf, beginSegPtr, endSegPtr);
|
|
}
|
|
*endSegPtr = savedChar;
|
|
if (savedChar) {
|
|
beginSegPtr = endSegPtr + 1;
|
|
GbAppendStringABW (GrowBuf, endSegPtr, beginSegPtr);
|
|
} else {
|
|
beginSegPtr = NULL;
|
|
}
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
PCWSTR
|
|
BfGetLongFileNameW (
|
|
IN PCWSTR SrcFileName
|
|
)
|
|
{
|
|
GROWBUFFER growBuf = INIT_GROWBUFFER;
|
|
PWSTR srcFileName;
|
|
PCWSTR result = NULL;
|
|
|
|
srcFileName = (PWSTR)SanitizePathW (SrcFileName);
|
|
if (pGetLongFileNameWorkerW (srcFileName, &growBuf)) {
|
|
result = DuplicatePathStringW ((PCWSTR)growBuf.Buf, 0);
|
|
GbFree (&growBuf);
|
|
}
|
|
FreePathStringW (srcFileName);
|
|
return result;
|
|
}
|
|
|
|
BOOL
|
|
BfGetLongFileNameExW (
|
|
IN PCWSTR SrcFileName,
|
|
IN PGROWBUFFER GrowBuff
|
|
)
|
|
{
|
|
PWSTR srcFileName;
|
|
BOOL result;
|
|
|
|
srcFileName = (PWSTR)SanitizePathW (SrcFileName);
|
|
result = pGetLongFileNameWorkerW (srcFileName, GrowBuff);
|
|
FreePathStringW (srcFileName);
|
|
|
|
return result;
|
|
}
|
|
|
|
BOOL
|
|
BfCopyAndFlushFileA (
|
|
IN PCSTR SrcFileName,
|
|
IN PCSTR DestFileName,
|
|
IN BOOL FailIfExists
|
|
)
|
|
{
|
|
BYTE buffer[4096];
|
|
HANDLE srcHandle;
|
|
HANDLE destHandle;
|
|
DWORD bytesRead = 4096;
|
|
DWORD bytesWritten;
|
|
BOOL error = FALSE;
|
|
BOOL result = FALSE;
|
|
|
|
srcHandle = BfOpenReadFileA (SrcFileName);
|
|
if (srcHandle) {
|
|
if (FailIfExists && DoesFileExistA (DestFileName)) {
|
|
SetLastError (ERROR_ALREADY_EXISTS);
|
|
} else {
|
|
destHandle = BfCreateFileA (DestFileName);
|
|
if (destHandle) {
|
|
while (bytesRead == 4096) {
|
|
if (!ReadFile (srcHandle, buffer, 4096, &bytesRead, NULL)) {
|
|
error = TRUE;
|
|
break;
|
|
}
|
|
if (bytesRead == 0) {
|
|
break;
|
|
}
|
|
if (!WriteFile (destHandle, buffer, bytesRead, &bytesWritten, NULL)) {
|
|
error = TRUE;
|
|
break;
|
|
}
|
|
if (bytesRead != bytesWritten) {
|
|
error = TRUE;
|
|
break;
|
|
}
|
|
}
|
|
if (!error) {
|
|
result = TRUE;
|
|
}
|
|
if (result) {
|
|
FlushFileBuffers (destHandle);
|
|
}
|
|
CloseHandle (destHandle);
|
|
}
|
|
}
|
|
CloseHandle (srcHandle);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
BOOL
|
|
BfCopyAndFlushFileW (
|
|
IN PCWSTR SrcFileName,
|
|
IN PCWSTR DestFileName,
|
|
IN BOOL FailIfExists
|
|
)
|
|
{
|
|
BYTE buffer[4096];
|
|
HANDLE srcHandle;
|
|
HANDLE destHandle;
|
|
DWORD bytesRead = 4096;
|
|
DWORD bytesWritten;
|
|
BOOL error = FALSE;
|
|
BOOL result = FALSE;
|
|
|
|
srcHandle = BfOpenReadFileW (SrcFileName);
|
|
if (srcHandle) {
|
|
if (FailIfExists && DoesFileExistW (DestFileName)) {
|
|
SetLastError (ERROR_ALREADY_EXISTS);
|
|
} else {
|
|
destHandle = BfCreateFileW (DestFileName);
|
|
if (destHandle) {
|
|
while (bytesRead == 4096) {
|
|
if (!ReadFile (srcHandle, buffer, 4096, &bytesRead, NULL)) {
|
|
error = TRUE;
|
|
break;
|
|
}
|
|
if (bytesRead == 0) {
|
|
break;
|
|
}
|
|
if (!WriteFile (destHandle, buffer, bytesRead, &bytesWritten, NULL)) {
|
|
error = TRUE;
|
|
break;
|
|
}
|
|
if (bytesRead != bytesWritten) {
|
|
error = TRUE;
|
|
break;
|
|
}
|
|
}
|
|
if (!error) {
|
|
result = TRUE;
|
|
}
|
|
if (result) {
|
|
FlushFileBuffers (destHandle);
|
|
}
|
|
CloseHandle (destHandle);
|
|
}
|
|
}
|
|
CloseHandle (srcHandle);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|