tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
audk/EmulatorPkg/Win/Host/WinFileSystem.c

2446 lines
64 KiB
C
Raw Blame History

/*++ @file
Support OS native directory access.
Copyright (c) 2006 - 2019, Intel Corporation. All rights reserved.<BR>
SPDX-License-Identifier: BSD-2-Clause-Patent
**/
#include "WinHost.h"
#define WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE_SIGNATURE SIGNATURE_32 ('N', 'T', 'f', 's')
typedef struct {
UINTN Signature;
EMU_IO_THUNK_PROTOCOL *Thunk;
EFI_SIMPLE_FILE_SYSTEM_PROTOCOL SimpleFileSystem;
CHAR16 *FilePath;
CHAR16 *VolumeLabel;
} WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE;
#define WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE_DATA_FROM_THIS(a) \
CR (a, \
WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE, \
SimpleFileSystem, \
WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE_SIGNATURE \
)
#define WIN_NT_EFI_FILE_PRIVATE_SIGNATURE SIGNATURE_32 ('l', 'o', 'f', 's')
typedef struct {
UINTN Signature;
EMU_IO_THUNK_PROTOCOL *Thunk;
EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *SimpleFileSystem;
EFI_FILE_PROTOCOL EfiFile;
HANDLE LHandle;
HANDLE DirHandle;
BOOLEAN IsRootDirectory;
BOOLEAN IsDirectoryPath;
BOOLEAN IsOpenedByRead;
CHAR16 *FilePath;
WCHAR *FileName;
BOOLEAN IsValidFindBuf;
WIN32_FIND_DATA FindBuf;
} WIN_NT_EFI_FILE_PRIVATE;
#define WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS(a) \
CR (a, \
WIN_NT_EFI_FILE_PRIVATE, \
EfiFile, \
WIN_NT_EFI_FILE_PRIVATE_SIGNATURE \
)
extern EFI_FILE_PROTOCOL gWinNtFileProtocol;
extern EFI_SIMPLE_FILE_SYSTEM_PROTOCOL gWinNtFileSystemProtocol;
EFI_STATUS
WinNtFileGetInfo (
IN EFI_FILE_PROTOCOL *This,
IN EFI_GUID *InformationType,
IN OUT UINTN *BufferSize,
OUT VOID *Buffer
);
EFI_STATUS
WinNtFileSetInfo (
IN EFI_FILE_PROTOCOL *This,
IN EFI_GUID *InformationType,
IN UINTN BufferSize,
IN VOID *Buffer
);
CHAR16 *
EfiStrChr (
IN CHAR16 *Str,
IN CHAR16 Chr
)
/*++
Routine Description:
Locate the first occurance of a character in a string.
Arguments:
Str - Pointer to NULL terminated unicode string.
Chr - Character to locate.
Returns:
If Str is NULL, then NULL is returned.
If Chr is not contained in Str, then NULL is returned.
If Chr is contained in Str, then a pointer to the first occurance of Chr in Str is returned.
--*/
{
if (Str == NULL) {
return Str;
}
while (*Str != '\0' && *Str != Chr) {
++Str;
}
return (*Str == Chr) ? Str : NULL;
}
BOOLEAN
IsZero (
IN VOID *Buffer,
IN UINTN Length
)
{
if (Buffer == NULL || Length == 0) {
return FALSE;
}
if (*(UINT8 *) Buffer != 0) {
return FALSE;
}
if (Length > 1) {
if (!CompareMem (Buffer, (UINT8 *) Buffer + 1, Length - 1)) {
return FALSE;
}
}
return TRUE;
}
VOID
CutPrefix (
IN CHAR16 *Str,
IN UINTN Count
)
{
CHAR16 *Pointer;
if (StrLen (Str) < Count) {
ASSERT (0);
}
if (Count != 0) {
for (Pointer = Str; *(Pointer + Count); Pointer++) {
*Pointer = *(Pointer + Count);
}
*Pointer = *(Pointer + Count);
}
}
/**
Open the root directory on a volume.
@param This Protocol instance pointer.
@param Root Returns an Open file handle for the root directory
@retval EFI_SUCCESS The device was opened.
@retval EFI_UNSUPPORTED This volume does not support the file system.
@retval EFI_NO_MEDIA The device has no media.
@retval EFI_DEVICE_ERROR The device reported an error.
@retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
@retval EFI_ACCESS_DENIED The service denied access to the file.
@retval EFI_OUT_OF_RESOURCES The volume was not opened due to lack of resources.
**/
EFI_STATUS
WinNtOpenVolume (
IN EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *This,
OUT EFI_FILE_PROTOCOL **Root
)
{
EFI_STATUS Status;
WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE *Private;
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
CHAR16 *TempFileName;
UINTN Size;
if (This == NULL || Root == NULL) {
return EFI_INVALID_PARAMETER;
}
Private = WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE_DATA_FROM_THIS (This);
PrivateFile = AllocatePool (sizeof (WIN_NT_EFI_FILE_PRIVATE));
if (PrivateFile == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
PrivateFile->FileName = AllocatePool (StrSize (Private->FilePath));
if (PrivateFile->FileName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
PrivateFile->FilePath = AllocatePool (StrSize (Private->FilePath));
if (PrivateFile->FilePath == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
StrCpyS (PrivateFile->FilePath,
StrSize (Private->FilePath) / sizeof (CHAR16),
Private->FilePath
);
StrCpyS (PrivateFile->FileName,
StrSize (Private->FilePath) / sizeof (CHAR16),
PrivateFile->FilePath
);
PrivateFile->Signature = WIN_NT_EFI_FILE_PRIVATE_SIGNATURE;
PrivateFile->Thunk = Private->Thunk;
PrivateFile->SimpleFileSystem = This;
PrivateFile->IsRootDirectory = TRUE;
PrivateFile->IsDirectoryPath = TRUE;
PrivateFile->IsOpenedByRead = TRUE;
CopyMem (&PrivateFile->EfiFile, &gWinNtFileProtocol, sizeof (gWinNtFileProtocol));
PrivateFile->IsValidFindBuf = FALSE;
//
// Set DirHandle
//
PrivateFile->DirHandle = CreateFile (
PrivateFile->FilePath,
GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_FLAG_BACKUP_SEMANTICS,
NULL
);
if (PrivateFile->DirHandle == INVALID_HANDLE_VALUE) {
Status = EFI_NOT_FOUND;
goto Done;
}
//
// Find the first file under it
//
Size = StrSize (PrivateFile->FilePath);
Size += StrSize (L"\\*");
TempFileName = AllocatePool (Size);
if (TempFileName == NULL) {
goto Done;
}
StrCpyS (TempFileName, Size / sizeof (CHAR16), PrivateFile->FilePath);
StrCatS (TempFileName, Size / sizeof (CHAR16), L"\\*");
PrivateFile->LHandle = FindFirstFile (TempFileName, &PrivateFile->FindBuf);
FreePool (TempFileName);
if (PrivateFile->LHandle == INVALID_HANDLE_VALUE) {
PrivateFile->IsValidFindBuf = FALSE;
} else {
PrivateFile->IsValidFindBuf = TRUE;
}
*Root = &PrivateFile->EfiFile;
Status = EFI_SUCCESS;
Done:
if (EFI_ERROR (Status)) {
if (PrivateFile) {
if (PrivateFile->FileName) {
FreePool (PrivateFile->FileName);
}
if (PrivateFile->FilePath) {
FreePool (PrivateFile->FilePath);
}
FreePool (PrivateFile);
}
}
return Status;
}
/**
Count the number of Leading Dot in FileNameToken.
@param FileNameToken A string representing a token in the path name.
@return UINTN The number of leading dot in the name.
**/
UINTN
CountLeadingDots (
IN CONST CHAR16 * FileNameToken
)
{
UINTN Num;
Num = 0;
while (*FileNameToken == L'.') {
Num++;
FileNameToken++;
}
return Num;
}
BOOLEAN
IsFileNameTokenValid (
IN CONST CHAR16 * FileNameToken
)
{
UINTN Num;
if (StrStr (FileNameToken, L"/") != NULL) {
//
// No L'/' in file name.
//
return FALSE;
} else {
//
// If Token has all dot, the number should not exceed 2
//
Num = CountLeadingDots (FileNameToken);
if (Num == StrLen (FileNameToken)) {
//
// If the FileNameToken only contains a number of L'.'.
//
if (Num > 2) {
return FALSE;
}
}
}
return TRUE;
}
/**
Return the first string token found in the indirect pointer a String named by FileName.
On input, FileName is a indirect pointer pointing to a String.
On output, FileName is a updated to point to the next character after the first
found L"\" or NULL if there is no L"\" found.
@param FileName A indirect pointer pointing to a FileName.
@return Token The first string token found before a L"\".
**/
CHAR16 *
GetNextFileNameToken (
IN OUT CONST CHAR16 ** FileName
)
{
CHAR16 *SlashPos;
CHAR16 *Token;
UINTN Offset;
ASSERT (**FileName != L'\\');
ASSERT (**FileName != L'\0');
SlashPos = StrStr (*FileName, L"\\");
if (SlashPos == NULL) {
Token = AllocateCopyPool (StrSize (*FileName), *FileName);
*FileName = NULL;
} else {
Offset = SlashPos - *FileName;
Token = AllocateZeroPool ((Offset + 1) * sizeof (CHAR16));
StrnCpyS (Token, Offset + 1, *FileName, Offset);
//
// Point *FileName to the next character after L'\'.
//
*FileName = *FileName + Offset + 1;
//
// If *FileName is an empty string, then set *FileName to NULL
//
if (**FileName == L'\0') {
*FileName = NULL;
}
}
return Token;
}
/**
Check if a FileName contains only Valid Characters.
If FileName contains only a single L'\', return TRUE.
If FileName contains two adjacent L'\', return FALSE.
If FileName conatins L'/' , return FALSE.
If FileName contains more than two dots separated with other FileName characters
by L'\', return FALSE. For example, L'.\...\filename.txt' is invalid path name. But L'..TwoDots\filename.txt' is valid path name.
@param FileName The File Name String to check.
@return TRUE FileName only contains valid characters.
@return FALSE FileName contains at least one invalid character.
**/
BOOLEAN
IsFileNameValid (
IN CONST CHAR16 *FileName
)
{
CHAR16 *Token;
BOOLEAN Valid;
//
// If FileName is just L'\', then it is a valid pathname.
//
if (StrCmp (FileName, L"\\") == 0) {
return TRUE;
}
//
// We don't support two or more adjacent L'\'.
//
if (StrStr (FileName, L"\\\\") != NULL) {
return FALSE;
}
//
// Is FileName has a leading L"\", skip to next character.
//
if (FileName [0] == L'\\') {
FileName++;
}
do {
Token = GetNextFileNameToken (&FileName);
Valid = IsFileNameTokenValid (Token);
FreePool (Token);
if (!Valid)
return FALSE;
} while (FileName != NULL);
return TRUE;
}
/**
Opens a new file relative to the source file's location.
@param This The protocol instance pointer.
@param NewHandle Returns File Handle for FileName.
@param FileName Null terminated string. "\", ".", and ".." are supported.
@param OpenMode Open mode for file.
@param Attributes Only used for EFI_FILE_MODE_CREATE.
@retval EFI_SUCCESS The device was opened.
@retval EFI_NOT_FOUND The specified file could not be found on the device.
@retval EFI_NO_MEDIA The device has no media.
@retval EFI_MEDIA_CHANGED The media has changed.
@retval EFI_DEVICE_ERROR The device reported an error.
@retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
@retval EFI_ACCESS_DENIED The service denied access to the file.
@retval EFI_OUT_OF_RESOURCES The volume was not opened due to lack of resources.
@retval EFI_VOLUME_FULL The volume is full.
**/
EFI_STATUS
WinNtFileOpen (
IN EFI_FILE_PROTOCOL *This,
OUT EFI_FILE_PROTOCOL **NewHandle,
IN CHAR16 *FileName,
IN UINT64 OpenMode,
IN UINT64 Attributes
)
{
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
WIN_NT_EFI_FILE_PRIVATE *NewPrivateFile;
WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE *PrivateRoot;
EFI_STATUS Status;
CHAR16 *RealFileName;
CHAR16 *TempFileName;
CHAR16 *ParseFileName;
CHAR16 *GuardPointer;
CHAR16 TempChar;
DWORD LastError;
UINTN Count;
BOOLEAN LoopFinish;
UINTN InfoSize;
EFI_FILE_INFO *Info;
UINTN Size;
//
// Init local variables
//
PrivateFile = WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS (This);
PrivateRoot = WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE_DATA_FROM_THIS (PrivateFile->SimpleFileSystem);
NewPrivateFile = NULL;
//
// Allocate buffer for FileName as the passed in FileName may be read only
//
TempFileName = AllocatePool (StrSize (FileName));
if (TempFileName == NULL) {
return EFI_OUT_OF_RESOURCES;
}
StrCpyS (TempFileName, StrSize (FileName) / sizeof (CHAR16), FileName);
FileName = TempFileName;
if (FileName[StrLen (FileName) - 1] == L'\\') {
FileName[StrLen (FileName) - 1] = 0;
}
//
// If file name does not equal to "." or ".." and not trailed with "\..",
// then we trim the leading/trailing blanks and trailing dots
//
if (StrCmp (FileName, L".") != 0 && StrCmp (FileName, L"..") != 0 &&
((StrLen (FileName) >= 3) ? (StrCmp (&FileName[StrLen (FileName) - 3], L"\\..") != 0) : TRUE)) {
//
// Trim leading blanks
//
Count = 0;
for (TempFileName = FileName;
*TempFileName != 0 && *TempFileName == L' ';
TempFileName++) {
Count++;
}
CutPrefix (FileName, Count);
//
// Trim trailing blanks
//
for (TempFileName = FileName + StrLen (FileName) - 1;
TempFileName >= FileName && (*TempFileName == L' ');
TempFileName--) {
;
}
*(TempFileName + 1) = 0;
}
//
// Attempt to open the file
//
NewPrivateFile = AllocatePool (sizeof (WIN_NT_EFI_FILE_PRIVATE));
if (NewPrivateFile == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
CopyMem (NewPrivateFile, PrivateFile, sizeof (WIN_NT_EFI_FILE_PRIVATE));
NewPrivateFile->FilePath = AllocatePool (StrSize (PrivateFile->FileName));
if (NewPrivateFile->FilePath == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
if (PrivateFile->IsDirectoryPath) {
StrCpyS (
NewPrivateFile->FilePath,
StrSize (PrivateFile->FileName) / sizeof (CHAR16),
PrivateFile->FileName
);
} else {
StrCpyS (
NewPrivateFile->FilePath,
StrSize (PrivateFile->FileName) / sizeof (CHAR16),
PrivateFile->FilePath
);
}
Size = StrSize (NewPrivateFile->FilePath);
Size += StrSize (L"\\");
Size += StrSize (FileName);
NewPrivateFile->FileName = AllocatePool (Size);
if (NewPrivateFile->FileName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
if (*FileName == L'\\') {
StrCpyS (NewPrivateFile->FileName, Size / sizeof (CHAR16), PrivateRoot->FilePath);
StrCatS (NewPrivateFile->FileName, Size / sizeof (CHAR16), L"\\");
StrCatS (NewPrivateFile->FileName, Size / sizeof (CHAR16), FileName + 1);
} else {
StrCpyS (NewPrivateFile->FileName, Size / sizeof (CHAR16), NewPrivateFile->FilePath);
if (StrCmp (FileName, L"") != 0) {
//
// In case the filename becomes empty, especially after trimming dots and blanks
//
StrCatS (NewPrivateFile->FileName, Size / sizeof (CHAR16), L"\\");
StrCatS (NewPrivateFile->FileName, Size / sizeof (CHAR16), FileName);
}
}
if (!IsFileNameValid (NewPrivateFile->FileName)) {
Status = EFI_NOT_FOUND;
goto Done;
}
//
// Get rid of . and .., except leading . or ..
//
//
// GuardPointer protect simplefilesystem root path not be destroyed
//
GuardPointer = NewPrivateFile->FileName + StrLen (PrivateRoot->FilePath);
LoopFinish = FALSE;
while (!LoopFinish) {
LoopFinish = TRUE;
for (ParseFileName = GuardPointer; *ParseFileName; ParseFileName++) {
if (*ParseFileName == L'.' &&
(*(ParseFileName + 1) == 0 || *(ParseFileName + 1) == L'\\') &&
*(ParseFileName - 1) == L'\\'
) {
//
// cut \.
//
CutPrefix (ParseFileName - 1, 2);
LoopFinish = FALSE;
break;
}
if (*ParseFileName == L'.' &&
*(ParseFileName + 1) == L'.' &&
(*(ParseFileName + 2) == 0 || *(ParseFileName + 2) == L'\\') &&
*(ParseFileName - 1) == L'\\'
) {
ParseFileName--;
Count = 3;
while (ParseFileName != GuardPointer) {
ParseFileName--;
Count++;
if (*ParseFileName == L'\\') {
break;
}
}
//
// cut \.. and its left directory
//
CutPrefix (ParseFileName, Count);
LoopFinish = FALSE;
break;
}
}
}
RealFileName = NewPrivateFile->FileName;
while (EfiStrChr (RealFileName, L'\\') != NULL) {
RealFileName = EfiStrChr (RealFileName, L'\\') + 1;
}
TempChar = 0;
if (RealFileName != NewPrivateFile->FileName) {
TempChar = *(RealFileName - 1);
*(RealFileName - 1) = 0;
}
FreePool (NewPrivateFile->FilePath);
NewPrivateFile->FilePath = NULL;
NewPrivateFile->FilePath = AllocatePool (StrSize (NewPrivateFile->FileName));
if (NewPrivateFile->FilePath == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
StrCpyS (
NewPrivateFile->FilePath,
StrSize (NewPrivateFile->FileName) / sizeof (CHAR16),
NewPrivateFile->FileName
);
if (TempChar != 0) {
*(RealFileName - 1) = TempChar;
}
NewPrivateFile->IsRootDirectory = FALSE;
//
// Test whether file or directory
//
if (OpenMode & EFI_FILE_MODE_CREATE) {
if (Attributes & EFI_FILE_DIRECTORY) {
NewPrivateFile->IsDirectoryPath = TRUE;
} else {
NewPrivateFile->IsDirectoryPath = FALSE;
}
} else {
NewPrivateFile->LHandle = CreateFile (
NewPrivateFile->FileName,
GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
0,
NULL
);
if (NewPrivateFile->LHandle != INVALID_HANDLE_VALUE) {
NewPrivateFile->IsDirectoryPath = FALSE;
CloseHandle (NewPrivateFile->LHandle);
} else {
NewPrivateFile->IsDirectoryPath = TRUE;
}
NewPrivateFile->LHandle = INVALID_HANDLE_VALUE;
}
if (OpenMode & EFI_FILE_MODE_WRITE) {
NewPrivateFile->IsOpenedByRead = FALSE;
} else {
NewPrivateFile->IsOpenedByRead = TRUE;
}
Status = EFI_SUCCESS;
//
// deal with directory
//
if (NewPrivateFile->IsDirectoryPath) {
Size = StrSize (NewPrivateFile->FileName);
Size += StrSize (L"\\*");
TempFileName = AllocatePool (Size);
if (TempFileName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
StrCpyS (TempFileName, Size / sizeof (CHAR16), NewPrivateFile->FileName);
if ((OpenMode & EFI_FILE_MODE_CREATE)) {
//
// Create a directory
//
if (!CreateDirectory (TempFileName, NULL)) {
LastError = GetLastError ();
if (LastError != ERROR_ALREADY_EXISTS) {
FreePool (TempFileName);
Status = EFI_ACCESS_DENIED;
goto Done;
}
}
}
NewPrivateFile->DirHandle = CreateFile (
TempFileName,
NewPrivateFile->IsOpenedByRead ? GENERIC_READ : (GENERIC_READ | GENERIC_WRITE),
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_FLAG_BACKUP_SEMANTICS,
NULL
);
if (NewPrivateFile->DirHandle == INVALID_HANDLE_VALUE) {
NewPrivateFile->DirHandle = CreateFile (
TempFileName,
GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_FLAG_BACKUP_SEMANTICS,
NULL
);
if (NewPrivateFile->DirHandle != INVALID_HANDLE_VALUE) {
CloseHandle (NewPrivateFile->DirHandle);
NewPrivateFile->DirHandle = INVALID_HANDLE_VALUE;
Status = EFI_ACCESS_DENIED;
} else {
Status = EFI_NOT_FOUND;
}
FreePool (TempFileName);
goto Done;
}
//
// Find the first file under it
//
StrCatS (TempFileName, Size / sizeof (CHAR16), L"\\*");
NewPrivateFile->LHandle = FindFirstFile (TempFileName, &NewPrivateFile->FindBuf);
FreePool (TempFileName);
if (NewPrivateFile->LHandle == INVALID_HANDLE_VALUE) {
NewPrivateFile->IsValidFindBuf = FALSE;
} else {
NewPrivateFile->IsValidFindBuf = TRUE;
}
} else {
//
// deal with file
//
if (!NewPrivateFile->IsOpenedByRead) {
NewPrivateFile->LHandle = CreateFile (
NewPrivateFile->FileName,
GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
(OpenMode & EFI_FILE_MODE_CREATE) ? OPEN_ALWAYS : OPEN_EXISTING,
0,
NULL
);
if (NewPrivateFile->LHandle == INVALID_HANDLE_VALUE) {
NewPrivateFile->LHandle = CreateFile (
NewPrivateFile->FileName,
GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
0,
NULL
);
if (NewPrivateFile->LHandle == INVALID_HANDLE_VALUE) {
Status = EFI_NOT_FOUND;
} else {
Status = EFI_ACCESS_DENIED;
CloseHandle (NewPrivateFile->LHandle);
NewPrivateFile->LHandle = INVALID_HANDLE_VALUE;
}
}
} else {
NewPrivateFile->LHandle = CreateFile (
NewPrivateFile->FileName,
GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
0,
NULL
);
if (NewPrivateFile->LHandle == INVALID_HANDLE_VALUE) {
Status = EFI_NOT_FOUND;
}
}
}
if ((OpenMode & EFI_FILE_MODE_CREATE) && Status == EFI_SUCCESS) {
//
// Set the attribute
//
InfoSize = 0;
Info = NULL;
Status = WinNtFileGetInfo (&NewPrivateFile->EfiFile, &gEfiFileInfoGuid, &InfoSize, Info);
if (Status != EFI_BUFFER_TOO_SMALL) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
Info = AllocatePool (InfoSize);
if (Info == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
Status = WinNtFileGetInfo (&NewPrivateFile->EfiFile, &gEfiFileInfoGuid, &InfoSize, Info);
if (EFI_ERROR (Status)) {
FreePool (Info);
goto Done;
}
Info->Attribute = Attributes;
WinNtFileSetInfo (&NewPrivateFile->EfiFile, &gEfiFileInfoGuid, InfoSize, Info);
FreePool (Info);
}
Done:
FreePool (FileName);
if (EFI_ERROR (Status)) {
if (NewPrivateFile) {
if (NewPrivateFile->FileName) {
FreePool (NewPrivateFile->FileName);
}
if (NewPrivateFile->FilePath) {
FreePool (NewPrivateFile->FilePath);
}
FreePool (NewPrivateFile);
}
} else {
*NewHandle = &NewPrivateFile->EfiFile;
if (StrCmp (NewPrivateFile->FileName, PrivateRoot->FilePath) == 0) {
NewPrivateFile->IsRootDirectory = TRUE;
}
}
return Status;
}
/**
Close the file handle
@param This Protocol instance pointer.
@retval EFI_SUCCESS The device was opened.
**/
EFI_STATUS
WinNtFileClose (
IN EFI_FILE_PROTOCOL *This
)
{
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
PrivateFile = WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS (This);
if (PrivateFile->LHandle != INVALID_HANDLE_VALUE) {
if (PrivateFile->IsDirectoryPath) {
FindClose (PrivateFile->LHandle);
} else {
CloseHandle (PrivateFile->LHandle);
}
PrivateFile->LHandle = INVALID_HANDLE_VALUE;
}
if (PrivateFile->IsDirectoryPath && PrivateFile->DirHandle != INVALID_HANDLE_VALUE) {
CloseHandle (PrivateFile->DirHandle);
PrivateFile->DirHandle = INVALID_HANDLE_VALUE;
}
if (PrivateFile->FileName) {
FreePool (PrivateFile->FileName);
}
if (PrivateFile->FilePath) {
FreePool (PrivateFile->FilePath);
}
FreePool (PrivateFile);
return EFI_SUCCESS;
}
/**
Close and delete the file handle.
@param This Protocol instance pointer.
@retval EFI_SUCCESS The device was opened.
@retval EFI_WARN_DELETE_FAILURE The handle was closed but the file was not deleted.
**/
EFI_STATUS
WinNtFileDelete (
IN EFI_FILE_PROTOCOL *This
)
{
EFI_STATUS Status;
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
PrivateFile = WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS (This);
Status = EFI_WARN_DELETE_FAILURE;
if (PrivateFile->IsDirectoryPath) {
if (PrivateFile->LHandle != INVALID_HANDLE_VALUE) {
FindClose (PrivateFile->LHandle);
}
if (PrivateFile->DirHandle != INVALID_HANDLE_VALUE) {
CloseHandle (PrivateFile->DirHandle);
PrivateFile->DirHandle = INVALID_HANDLE_VALUE;
}
if (RemoveDirectory (PrivateFile->FileName)) {
Status = EFI_SUCCESS;
}
} else {
CloseHandle (PrivateFile->LHandle);
PrivateFile->LHandle = INVALID_HANDLE_VALUE;
if (!PrivateFile->IsOpenedByRead) {
if (DeleteFile (PrivateFile->FileName)) {
Status = EFI_SUCCESS;
}
}
}
FreePool (PrivateFile->FileName);
FreePool (PrivateFile->FilePath);
FreePool (PrivateFile);
return Status;
}
VOID
WinNtSystemTimeToEfiTime (
IN SYSTEMTIME *SystemTime,
IN TIME_ZONE_INFORMATION *TimeZone,
OUT EFI_TIME *Time
)
/*++
Routine Description:
TODO: Add function description
Arguments:
SystemTime - TODO: add argument description
TimeZone - TODO: add argument description
Time - TODO: add argument description
Returns:
TODO: add return values
--*/
{
Time->Year = (UINT16)SystemTime->wYear;
Time->Month = (UINT8)SystemTime->wMonth;
Time->Day = (UINT8)SystemTime->wDay;
Time->Hour = (UINT8)SystemTime->wHour;
Time->Minute = (UINT8)SystemTime->wMinute;
Time->Second = (UINT8)SystemTime->wSecond;
Time->Nanosecond = (UINT32)SystemTime->wMilliseconds * 1000000;
Time->TimeZone = (INT16)TimeZone->Bias;
if (TimeZone->StandardDate.wMonth) {
Time->Daylight = EFI_TIME_ADJUST_DAYLIGHT;
}
}
/**
Convert the FileTime to EfiTime.
@param PrivateFile Pointer to WIN_NT_EFI_FILE_PRIVATE.
@param TimeZone Pointer to the current time zone.
@param FileTime Pointer to file time.
@param EfiTime Pointer to EFI time.
**/
VOID
WinNtFileTimeToEfiTime (
IN CONST WIN_NT_EFI_FILE_PRIVATE *PrivateFile,
IN TIME_ZONE_INFORMATION *TimeZone,
IN CONST FILETIME *FileTime,
OUT EFI_TIME *EfiTime
)
{
FILETIME TempFileTime;
SYSTEMTIME SystemTime;
FileTimeToLocalFileTime (FileTime, &TempFileTime);
FileTimeToSystemTime (&TempFileTime, &SystemTime);
WinNtSystemTimeToEfiTime (&SystemTime, TimeZone, EfiTime);
}
/**
Read data from the file.
@param This Protocol instance pointer.
@param BufferSize On input size of buffer, on output amount of data in buffer.
@param Buffer The buffer in which data is read.
@retval EFI_SUCCESS Data was read.
@retval EFI_NO_MEDIA The device has no media.
@retval EFI_DEVICE_ERROR The device reported an error.
@retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
@retval EFI_BUFFER_TO_SMALL BufferSize is too small. BufferSize contains required size.
**/
EFI_STATUS
WinNtFileRead (
IN EFI_FILE_PROTOCOL *This,
IN OUT UINTN *BufferSize,
OUT VOID *Buffer
)
{
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
EFI_STATUS Status;
UINTN Size;
UINTN NameSize;
UINTN ResultSize;
UINTN Index;
EFI_FILE_INFO *Info;
WCHAR *pw;
TIME_ZONE_INFORMATION TimeZone;
EFI_FILE_INFO *FileInfo;
UINT64 Pos;
UINT64 FileSize;
UINTN FileInfoSize;
PrivateFile = WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS (This);
if (PrivateFile->LHandle == INVALID_HANDLE_VALUE) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
if (!PrivateFile->IsDirectoryPath) {
if (This->GetPosition (This, &Pos) != EFI_SUCCESS) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
FileInfoSize = SIZE_OF_EFI_FILE_SYSTEM_INFO;
FileInfo = AllocatePool (FileInfoSize);
Status = This->GetInfo (
This,
&gEfiFileInfoGuid,
&FileInfoSize,
FileInfo
);
if (Status == EFI_BUFFER_TOO_SMALL) {
FreePool (FileInfo);
FileInfo = AllocatePool (FileInfoSize);
Status = This->GetInfo (
This,
&gEfiFileInfoGuid,
&FileInfoSize,
FileInfo
);
}
if (EFI_ERROR (Status)) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
FileSize = FileInfo->FileSize;
FreePool (FileInfo);
if (Pos >= FileSize) {
*BufferSize = 0;
if (Pos == FileSize) {
Status = EFI_SUCCESS;
goto Done;
} else {
Status = EFI_DEVICE_ERROR;
goto Done;
}
}
Status = ReadFile (
PrivateFile->LHandle,
Buffer,
(DWORD)*BufferSize,
(LPDWORD)BufferSize,
NULL
) ? EFI_SUCCESS : EFI_DEVICE_ERROR;
goto Done;
}
//
// Read on a directory. Perform a find next
//
if (!PrivateFile->IsValidFindBuf) {
*BufferSize = 0;
Status = EFI_SUCCESS;
goto Done;
}
Size = SIZE_OF_EFI_FILE_INFO;
NameSize = StrSize (PrivateFile->FindBuf.cFileName);
ResultSize = Size + NameSize;
Status = EFI_BUFFER_TOO_SMALL;
if (*BufferSize >= ResultSize) {
Status = EFI_SUCCESS;
Info = Buffer;
ZeroMem (Info, ResultSize);
Info->Size = ResultSize;
GetTimeZoneInformation (&TimeZone);
WinNtFileTimeToEfiTime (PrivateFile, &TimeZone, &PrivateFile->FindBuf.ftCreationTime, &Info->CreateTime);
WinNtFileTimeToEfiTime (PrivateFile, &TimeZone, &PrivateFile->FindBuf.ftLastAccessTime, &Info->LastAccessTime);
WinNtFileTimeToEfiTime (PrivateFile, &TimeZone, &PrivateFile->FindBuf.ftLastWriteTime, &Info->ModificationTime);
Info->FileSize = PrivateFile->FindBuf.nFileSizeLow;
Info->PhysicalSize = PrivateFile->FindBuf.nFileSizeLow;
if (PrivateFile->FindBuf.dwFileAttributes & FILE_ATTRIBUTE_ARCHIVE) {
Info->Attribute |= EFI_FILE_ARCHIVE;
}
if (PrivateFile->FindBuf.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) {
Info->Attribute |= EFI_FILE_HIDDEN;
}
if (PrivateFile->FindBuf.dwFileAttributes & FILE_ATTRIBUTE_SYSTEM) {
Info->Attribute |= EFI_FILE_SYSTEM;
}
if (PrivateFile->FindBuf.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
Info->Attribute |= EFI_FILE_READ_ONLY;
}
if (PrivateFile->FindBuf.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
Info->Attribute |= EFI_FILE_DIRECTORY;
}
NameSize = NameSize / sizeof (WCHAR);
pw = (WCHAR *)(((CHAR8 *)Buffer) + Size);
for (Index = 0; Index < NameSize; Index++) {
pw[Index] = PrivateFile->FindBuf.cFileName[Index];
}
if (FindNextFile (PrivateFile->LHandle, &PrivateFile->FindBuf)) {
PrivateFile->IsValidFindBuf = TRUE;
} else {
PrivateFile->IsValidFindBuf = FALSE;
}
}
*BufferSize = ResultSize;
Done:
return Status;
}
/**
Write data to a file.
@param This Protocol instance pointer.
@param BufferSize On input size of buffer, on output amount of data in buffer.
@param Buffer The buffer in which data to write.
@retval EFI_SUCCESS Data was written.
@retval EFI_UNSUPPORTED Writes to Open directory are not supported.
@retval EFI_NO_MEDIA The device has no media.
@retval EFI_DEVICE_ERROR The device reported an error.
@retval EFI_DEVICE_ERROR An attempt was made to write to a deleted file.
@retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
@retval EFI_WRITE_PROTECTED The device is write protected.
@retval EFI_ACCESS_DENIED The file was open for read only.
@retval EFI_VOLUME_FULL The volume is full.
**/
EFI_STATUS
WinNtFileWrite (
IN EFI_FILE_PROTOCOL *This,
IN OUT UINTN *BufferSize,
IN VOID *Buffer
)
{
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
EFI_STATUS Status;
PrivateFile = WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS (This);
if (PrivateFile->LHandle == INVALID_HANDLE_VALUE) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
if (PrivateFile->IsDirectoryPath) {
Status = EFI_UNSUPPORTED;
goto Done;
}
if (PrivateFile->IsOpenedByRead) {
Status = EFI_ACCESS_DENIED;
goto Done;
}
Status = WriteFile (
PrivateFile->LHandle,
Buffer,
(DWORD)*BufferSize,
(LPDWORD)BufferSize,
NULL
) ? EFI_SUCCESS : EFI_DEVICE_ERROR;
Done:
return Status;
//
// bugbug: need to access windows error reporting
//
}
/**
Set a files current position
@param This Protocol instance pointer.
@param Position Byte position from the start of the file.
@retval EFI_SUCCESS Data was written.
@retval EFI_UNSUPPORTED Seek request for non-zero is not valid on open.
**/
EFI_STATUS
WinNtFileSetPossition (
IN EFI_FILE_PROTOCOL *This,
IN UINT64 Position
)
{
EFI_STATUS Status;
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
UINT32 PosLow;
UINT32 PosHigh;
CHAR16 *FileName;
UINTN Size;
PrivateFile = WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS (This);
if (PrivateFile->IsDirectoryPath) {
if (Position != 0) {
Status = EFI_UNSUPPORTED;
goto Done;
}
Size = StrSize (PrivateFile->FileName);
Size += StrSize (L"\\*");
FileName = AllocatePool (Size);
if (FileName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
StrCpyS (FileName, Size / sizeof (CHAR16), PrivateFile->FileName);
StrCatS (FileName, Size / sizeof (CHAR16), L"\\*");
if (PrivateFile->LHandle != INVALID_HANDLE_VALUE) {
FindClose (PrivateFile->LHandle);
}
PrivateFile->LHandle = FindFirstFile (FileName, &PrivateFile->FindBuf);
if (PrivateFile->LHandle == INVALID_HANDLE_VALUE) {
PrivateFile->IsValidFindBuf = FALSE;
} else {
PrivateFile->IsValidFindBuf = TRUE;
}
FreePool (FileName);
Status = (PrivateFile->LHandle == INVALID_HANDLE_VALUE) ? EFI_DEVICE_ERROR : EFI_SUCCESS;
} else {
if (Position == (UINT64)-1) {
PosLow = SetFilePointer (PrivateFile->LHandle, (ULONG)0, NULL, FILE_END);
} else {
PosHigh = (UINT32)RShiftU64 (Position, 32);
PosLow = SetFilePointer (PrivateFile->LHandle, (ULONG)Position, (PLONG)&PosHigh, FILE_BEGIN);
}
Status = (PosLow == 0xFFFFFFFF) ? EFI_DEVICE_ERROR : EFI_SUCCESS;
}
Done:
return Status;
}
/**
Get a file's current position
@param This Protocol instance pointer.
@param Position Byte position from the start of the file.
@retval EFI_SUCCESS Data was written.
@retval EFI_UNSUPPORTED Seek request for non-zero is not valid on open..
**/
EFI_STATUS
WinNtFileGetPossition (
IN EFI_FILE_PROTOCOL *This,
OUT UINT64 *Position
)
{
EFI_STATUS Status;
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
INT32 PositionHigh;
UINT64 PosHigh64;
PrivateFile = WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS (This);
PositionHigh = 0;
PosHigh64 = 0;
if (PrivateFile->IsDirectoryPath) {
Status = EFI_UNSUPPORTED;
goto Done;
} else {
PositionHigh = 0;
*Position = SetFilePointer (
PrivateFile->LHandle,
0,
(PLONG)&PositionHigh,
FILE_CURRENT
);
Status = *Position == 0xffffffff ? EFI_DEVICE_ERROR : EFI_SUCCESS;
if (EFI_ERROR (Status)) {
goto Done;
}
PosHigh64 = PositionHigh;
*Position += LShiftU64 (PosHigh64, 32);
}
Done:
return Status;
}
EFI_STATUS
WinNtSimpleFileSystemFileInfo (
IN WIN_NT_EFI_FILE_PRIVATE *PrivateFile,
IN OUT UINTN *BufferSize,
OUT VOID *Buffer
)
/*++
Routine Description:
TODO: Add function description
Arguments:
PrivateFile - TODO: add argument description
BufferSize - TODO: add argument description
Buffer - TODO: add argument description
Returns:
TODO: add return values
--*/
{
EFI_STATUS Status;
UINTN Size;
UINTN NameSize;
UINTN ResultSize;
EFI_FILE_INFO *Info;
BY_HANDLE_FILE_INFORMATION FileInfo;
CHAR16 *RealFileName;
CHAR16 *TempPointer;
TIME_ZONE_INFORMATION TimeZone;
Size = SIZE_OF_EFI_FILE_INFO;
RealFileName = PrivateFile->FileName;
TempPointer = RealFileName;
while (*TempPointer) {
if (*TempPointer == '\\') {
RealFileName = TempPointer + 1;
}
TempPointer++;
}
NameSize = StrSize (RealFileName);
ResultSize = Size + NameSize;
Status = EFI_BUFFER_TOO_SMALL;
if (*BufferSize >= ResultSize) {
Status = EFI_SUCCESS;
Info = Buffer;
ZeroMem (Info, ResultSize);
Info->Size = ResultSize;
GetFileInformationByHandle (
PrivateFile->IsDirectoryPath ? PrivateFile->DirHandle : PrivateFile->LHandle,
&FileInfo
);
Info->FileSize = FileInfo.nFileSizeLow;
Info->PhysicalSize = Info->FileSize;
GetTimeZoneInformation (&TimeZone);
WinNtFileTimeToEfiTime (PrivateFile, &TimeZone, &FileInfo.ftCreationTime, &Info->CreateTime);
WinNtFileTimeToEfiTime (PrivateFile, &TimeZone, &FileInfo.ftLastAccessTime, &Info->LastAccessTime);
WinNtFileTimeToEfiTime (PrivateFile, &TimeZone, &FileInfo.ftLastWriteTime, &Info->ModificationTime);
if (FileInfo.dwFileAttributes & FILE_ATTRIBUTE_ARCHIVE) {
Info->Attribute |= EFI_FILE_ARCHIVE;
}
if (FileInfo.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) {
Info->Attribute |= EFI_FILE_HIDDEN;
}
if (FileInfo.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
Info->Attribute |= EFI_FILE_READ_ONLY;
}
if (FileInfo.dwFileAttributes & FILE_ATTRIBUTE_SYSTEM) {
Info->Attribute |= EFI_FILE_SYSTEM;
}
if (FileInfo.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
Info->Attribute |= EFI_FILE_DIRECTORY;
}
if (PrivateFile->IsDirectoryPath) {
Info->Attribute |= EFI_FILE_DIRECTORY;
}
if (PrivateFile->IsRootDirectory) {
*((CHAR8 *)Buffer + Size) = 0;
} else {
CopyMem ((CHAR8 *)Buffer + Size, RealFileName, NameSize);
}
}
*BufferSize = ResultSize;
return Status;
}
/**
Get information about a file.
@param This Protocol instance pointer.
@param InformationType Type of information to return in Buffer.
@param BufferSize On input size of buffer, on output amount of data in buffer.
@param Buffer The buffer to return data.
@retval EFI_SUCCESS Data was returned.
@retval EFI_UNSUPPORTED InformationType is not supported.
@retval EFI_NO_MEDIA The device has no media.
@retval EFI_DEVICE_ERROR The device reported an error.
@retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
@retval EFI_WRITE_PROTECTED The device is write protected.
@retval EFI_ACCESS_DENIED The file was open for read only.
@retval EFI_BUFFER_TOO_SMALL Buffer was too small; required size returned in BufferSize.
**/
EFI_STATUS
WinNtFileGetInfo (
IN EFI_FILE_PROTOCOL *This,
IN EFI_GUID *InformationType,
IN OUT UINTN *BufferSize,
OUT VOID *Buffer
)
{
EFI_STATUS Status;
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
EFI_FILE_SYSTEM_INFO *FileSystemInfoBuffer;
UINT32 SectorsPerCluster;
UINT32 BytesPerSector;
UINT32 FreeClusters;
UINT32 TotalClusters;
UINT32 BytesPerCluster;
CHAR16 *DriveName;
BOOLEAN DriveNameFound;
BOOL NtStatus;
UINTN Index;
WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE *PrivateRoot;
if (This == NULL || InformationType == NULL || BufferSize == NULL) {
return EFI_INVALID_PARAMETER;
}
PrivateFile = WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS (This);
PrivateRoot = WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE_DATA_FROM_THIS (PrivateFile->SimpleFileSystem);
Status = EFI_UNSUPPORTED;
if (CompareGuid (InformationType, &gEfiFileInfoGuid)) {
Status = WinNtSimpleFileSystemFileInfo (PrivateFile, BufferSize, Buffer);
}
if (CompareGuid (InformationType, &gEfiFileSystemInfoGuid)) {
if (*BufferSize < SIZE_OF_EFI_FILE_SYSTEM_INFO + StrSize (PrivateRoot->VolumeLabel)) {
*BufferSize = SIZE_OF_EFI_FILE_SYSTEM_INFO + StrSize (PrivateRoot->VolumeLabel);
Status = EFI_BUFFER_TOO_SMALL;
goto Done;
}
FileSystemInfoBuffer = (EFI_FILE_SYSTEM_INFO *)Buffer;
FileSystemInfoBuffer->Size = SIZE_OF_EFI_FILE_SYSTEM_INFO + StrSize (PrivateRoot->VolumeLabel);
FileSystemInfoBuffer->ReadOnly = FALSE;
//
// Try to get the drive name
//
DriveNameFound = FALSE;
DriveName = AllocatePool (StrSize (PrivateFile->FilePath) + 1);
if (DriveName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
StrCpyS (
DriveName,
(StrSize (PrivateFile->FilePath) + 1) / sizeof (CHAR16),
PrivateFile->FilePath
);
for (Index = 0; DriveName[Index] != 0 && DriveName[Index] != ':'; Index++) {
;
}
if (DriveName[Index] == ':') {
DriveName[Index + 1] = '\\';
DriveName[Index + 2] = 0;
DriveNameFound = TRUE;
} else if (DriveName[0] == '\\' && DriveName[1] == '\\') {
for (Index = 2; DriveName[Index] != 0 && DriveName[Index] != '\\'; Index++) {
;
}
if (DriveName[Index] == '\\') {
DriveNameFound = TRUE;
for (Index++; DriveName[Index] != 0 && DriveName[Index] != '\\'; Index++) {
;
}
DriveName[Index] = '\\';
DriveName[Index + 1] = 0;
}
}
//
// Try GetDiskFreeSpace first
//
NtStatus = GetDiskFreeSpace (
DriveNameFound ? DriveName : NULL,
(LPDWORD)&SectorsPerCluster,
(LPDWORD)&BytesPerSector,
(LPDWORD)&FreeClusters,
(LPDWORD)&TotalClusters
);
if (DriveName) {
FreePool (DriveName);
}
if (NtStatus) {
//
// Succeeded
//
BytesPerCluster = BytesPerSector * SectorsPerCluster;
FileSystemInfoBuffer->VolumeSize = MultU64x32 (TotalClusters, BytesPerCluster);
FileSystemInfoBuffer->FreeSpace = MultU64x32 (FreeClusters, BytesPerCluster);
FileSystemInfoBuffer->BlockSize = BytesPerCluster;
} else {
//
// try GetDiskFreeSpaceEx then
//
FileSystemInfoBuffer->BlockSize = 0;
NtStatus = GetDiskFreeSpaceEx (
PrivateFile->FilePath,
(PULARGE_INTEGER)(&FileSystemInfoBuffer->FreeSpace),
(PULARGE_INTEGER)(&FileSystemInfoBuffer->VolumeSize),
NULL
);
if (!NtStatus) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
}
StrCpyS (
(CHAR16 *)FileSystemInfoBuffer->VolumeLabel,
(*BufferSize - SIZE_OF_EFI_FILE_SYSTEM_INFO) / sizeof (CHAR16),
PrivateRoot->VolumeLabel
);
*BufferSize = SIZE_OF_EFI_FILE_SYSTEM_INFO + StrSize (PrivateRoot->VolumeLabel);
Status = EFI_SUCCESS;
}
if (CompareGuid (InformationType, &gEfiFileSystemVolumeLabelInfoIdGuid)) {
if (*BufferSize < StrSize (PrivateRoot->VolumeLabel)) {
*BufferSize = StrSize (PrivateRoot->VolumeLabel);
Status = EFI_BUFFER_TOO_SMALL;
goto Done;
}
StrCpyS (
(CHAR16 *)Buffer,
*BufferSize / sizeof (CHAR16),
PrivateRoot->VolumeLabel
);
*BufferSize = StrSize (PrivateRoot->VolumeLabel);
Status = EFI_SUCCESS;
}
Done:
return Status;
}
/**
Set information about a file
@param File Protocol instance pointer.
@param InformationType Type of information in Buffer.
@param BufferSize Size of buffer.
@param Buffer The data to write.
@retval EFI_SUCCESS Data was returned.
@retval EFI_UNSUPPORTED InformationType is not supported.
@retval EFI_NO_MEDIA The device has no media.
@retval EFI_DEVICE_ERROR The device reported an error.
@retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
@retval EFI_WRITE_PROTECTED The device is write protected.
@retval EFI_ACCESS_DENIED The file was open for read only.
**/
EFI_STATUS
WinNtFileSetInfo (
IN EFI_FILE_PROTOCOL *This,
IN EFI_GUID *InformationType,
IN UINTN BufferSize,
IN VOID *Buffer
)
{
WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE *PrivateRoot;
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
EFI_FILE_INFO *OldFileInfo;
EFI_FILE_INFO *NewFileInfo;
EFI_STATUS Status;
UINTN OldInfoSize;
INTN NtStatus;
UINT32 NewAttr;
UINT32 OldAttr;
CHAR16 *OldFileName;
CHAR16 *NewFileName;
CHAR16 *TempFileName;
CHAR16 *CharPointer;
BOOLEAN AttrChangeFlag;
BOOLEAN NameChangeFlag;
BOOLEAN SizeChangeFlag;
BOOLEAN TimeChangeFlag;
UINT64 CurPos;
SYSTEMTIME NewCreationSystemTime;
SYSTEMTIME NewLastAccessSystemTime;
SYSTEMTIME NewLastWriteSystemTime;
FILETIME NewCreationFileTime;
FILETIME NewLastAccessFileTime;
FILETIME NewLastWriteFileTime;
WIN32_FIND_DATA FindBuf;
EFI_FILE_SYSTEM_INFO *NewFileSystemInfo;
UINTN Size;
//
// Initialise locals.
//
PrivateFile = WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS (This);
PrivateRoot = WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE_DATA_FROM_THIS (PrivateFile->SimpleFileSystem);
Status = EFI_UNSUPPORTED;
OldFileInfo = NewFileInfo = NULL;
OldFileName = NewFileName = NULL;
AttrChangeFlag = NameChangeFlag = SizeChangeFlag = TimeChangeFlag = FALSE;
//
// Set file system information.
//
if (CompareGuid (InformationType, &gEfiFileSystemInfoGuid)) {
NewFileSystemInfo = (EFI_FILE_SYSTEM_INFO *)Buffer;
if (BufferSize < SIZE_OF_EFI_FILE_SYSTEM_INFO + StrSize (NewFileSystemInfo->VolumeLabel)) {
Status = EFI_BAD_BUFFER_SIZE;
goto Done;
}
FreePool (PrivateRoot->VolumeLabel);
PrivateRoot->VolumeLabel = AllocatePool (StrSize (NewFileSystemInfo->VolumeLabel));
if (PrivateRoot->VolumeLabel == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
StrCpyS (
PrivateRoot->VolumeLabel,
StrSize (NewFileSystemInfo->VolumeLabel) / sizeof (CHAR16),
NewFileSystemInfo->VolumeLabel
);
Status = EFI_SUCCESS;
goto Done;
}
//
// Set volume label information.
//
if (CompareGuid (InformationType, &gEfiFileSystemVolumeLabelInfoIdGuid)) {
if (BufferSize < StrSize (PrivateRoot->VolumeLabel)) {
Status = EFI_BAD_BUFFER_SIZE;
goto Done;
}
StrCpyS (
PrivateRoot->VolumeLabel,
StrSize (PrivateRoot->VolumeLabel) / sizeof (CHAR16),
(CHAR16 *)Buffer
);
Status = EFI_SUCCESS;
goto Done;
}
if (!CompareGuid (InformationType, &gEfiFileInfoGuid)) {
Status = EFI_UNSUPPORTED;
goto Done;
}
if (BufferSize < SIZE_OF_EFI_FILE_INFO) {
Status = EFI_BAD_BUFFER_SIZE;
goto Done;
}
//
// Set file/directory information.
//
//
// Check for invalid set file information parameters.
//
NewFileInfo = (EFI_FILE_INFO *)Buffer;
if ((NewFileInfo->Size <= SIZE_OF_EFI_FILE_INFO) ||
(NewFileInfo->Attribute &~(EFI_FILE_VALID_ATTR)) ||
(sizeof (UINTN) == 4 && NewFileInfo->Size > 0xFFFFFFFF)
) {
Status = EFI_INVALID_PARAMETER;
goto Done;
}
//
// bugbug: - This is not safe. We need something like EfiStrMaxSize()
// that would have an additional parameter that would be the size
// of the string array just in case there are no NULL characters in
// the string array.
//
//
// Get current file information so we can determine what kind
// of change request this is.
//
OldInfoSize = 0;
Status = WinNtSimpleFileSystemFileInfo (PrivateFile, &OldInfoSize, NULL);
if (Status != EFI_BUFFER_TOO_SMALL) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
OldFileInfo = AllocatePool (OldInfoSize);
if (OldFileInfo == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
Status = WinNtSimpleFileSystemFileInfo (PrivateFile, &OldInfoSize, OldFileInfo);
if (EFI_ERROR (Status)) {
goto Done;
}
OldFileName = AllocatePool (StrSize (PrivateFile->FileName));
if (OldFileName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
StrCpyS (
OldFileName,
StrSize (PrivateFile->FileName) / sizeof (CHAR16),
PrivateFile->FileName
);
//
// Make full pathname from new filename and rootpath.
//
if (NewFileInfo->FileName[0] == '\\') {
Size = StrSize (PrivateRoot->FilePath);
Size += StrSize (L"\\");
Size += StrSize (NewFileInfo->FileName);
NewFileName = AllocatePool (Size);
if (NewFileName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
StrCpyS (NewFileName, Size / sizeof (CHAR16), PrivateRoot->FilePath);
StrCatS (NewFileName, Size / sizeof (CHAR16), L"\\");
StrCatS (NewFileName, Size / sizeof (CHAR16), NewFileInfo->FileName + 1);
} else {
Size = StrSize (PrivateFile->FilePath);
Size += StrSize (L"\\");
Size += StrSize (NewFileInfo->FileName);
NewFileName = AllocatePool (Size);
if (NewFileName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
StrCpyS (NewFileName, Size / sizeof (CHAR16), PrivateFile->FilePath);
StrCatS (NewFileName, Size / sizeof (CHAR16), L"\\");
StrCatS (NewFileName, Size / sizeof (CHAR16), NewFileInfo->FileName);
}
//
// Is there an attribute change request?
//
if (NewFileInfo->Attribute != OldFileInfo->Attribute) {
if ((NewFileInfo->Attribute & EFI_FILE_DIRECTORY) != (OldFileInfo->Attribute & EFI_FILE_DIRECTORY)) {
Status = EFI_INVALID_PARAMETER;
goto Done;
}
AttrChangeFlag = TRUE;
}
//
// Is there a name change request?
// bugbug: - Need EfiStrCaseCmp()
//
if (StrCmp (NewFileInfo->FileName, OldFileInfo->FileName)) {
NameChangeFlag = TRUE;
}
//
// Is there a size change request?
//
if (NewFileInfo->FileSize != OldFileInfo->FileSize) {
SizeChangeFlag = TRUE;
}
//
// Is there a time stamp change request?
//
if (!IsZero (&NewFileInfo->CreateTime, sizeof (EFI_TIME)) &&
CompareMem (&NewFileInfo->CreateTime, &OldFileInfo->CreateTime, sizeof (EFI_TIME))
) {
TimeChangeFlag = TRUE;
} else if (!IsZero (&NewFileInfo->LastAccessTime, sizeof (EFI_TIME)) &&
CompareMem (&NewFileInfo->LastAccessTime, &OldFileInfo->LastAccessTime, sizeof (EFI_TIME))
) {
TimeChangeFlag = TRUE;
} else if (!IsZero (&NewFileInfo->ModificationTime, sizeof (EFI_TIME)) &&
CompareMem (&NewFileInfo->ModificationTime, &OldFileInfo->ModificationTime, sizeof (EFI_TIME))
) {
TimeChangeFlag = TRUE;
}
//
// All done if there are no change requests being made.
//
if (!(AttrChangeFlag || NameChangeFlag || SizeChangeFlag || TimeChangeFlag)) {
Status = EFI_SUCCESS;
goto Done;
}
//
// Set file or directory information.
//
OldAttr = GetFileAttributes (OldFileName);
//
// Name change.
//
if (NameChangeFlag) {
//
// Close the handles first
//
if (PrivateFile->IsOpenedByRead) {
Status = EFI_ACCESS_DENIED;
goto Done;
}
for (CharPointer = NewFileName; *CharPointer != 0 && *CharPointer != L'/'; CharPointer++) {
}
if (*CharPointer != 0) {
Status = EFI_ACCESS_DENIED;
goto Done;
}
if (PrivateFile->LHandle != INVALID_HANDLE_VALUE) {
if (PrivateFile->IsDirectoryPath) {
FindClose (PrivateFile->LHandle);
} else {
CloseHandle (PrivateFile->LHandle);
PrivateFile->LHandle = INVALID_HANDLE_VALUE;
}
}
if (PrivateFile->IsDirectoryPath && PrivateFile->DirHandle != INVALID_HANDLE_VALUE) {
CloseHandle (PrivateFile->DirHandle);
PrivateFile->DirHandle = INVALID_HANDLE_VALUE;
}
NtStatus = MoveFile (OldFileName, NewFileName);
if (NtStatus) {
//
// modify file name
//
FreePool (PrivateFile->FileName);
PrivateFile->FileName = AllocatePool (StrSize (NewFileName));
if (PrivateFile->FileName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto Done;
}
StrCpyS (PrivateFile->FileName, StrSize (NewFileName) / sizeof (CHAR16), NewFileName);
Size = StrSize (NewFileName);
Size += StrSize (L"\\*");
TempFileName = AllocatePool (Size);
StrCpyS (TempFileName, Size / sizeof (CHAR16), NewFileName);
if (!PrivateFile->IsDirectoryPath) {
PrivateFile->LHandle = CreateFile (
TempFileName,
PrivateFile->IsOpenedByRead ? GENERIC_READ : GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
0,
NULL
);
FreePool (TempFileName);
//
// Flush buffers just in case
//
if (FlushFileBuffers (PrivateFile->LHandle) == 0) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
} else {
PrivateFile->DirHandle = CreateFile (
TempFileName,
PrivateFile->IsOpenedByRead ? GENERIC_READ : GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_FLAG_BACKUP_SEMANTICS,
NULL
);
StrCatS (TempFileName, Size / sizeof (CHAR16), L"\\*");
PrivateFile->LHandle = FindFirstFile (TempFileName, &FindBuf);
FreePool (TempFileName);
}
} else {
Status = EFI_ACCESS_DENIED;
Reopen:;
NtStatus = SetFileAttributes (OldFileName, OldAttr);
if (!NtStatus) {
goto Done;
}
Size = StrSize (OldFileName);
Size += StrSize (L"\\*");
TempFileName = AllocatePool (Size);
StrCpyS (TempFileName, Size / sizeof (CHAR16), OldFileName);
if (!PrivateFile->IsDirectoryPath) {
PrivateFile->LHandle = CreateFile (
TempFileName,
PrivateFile->IsOpenedByRead ? GENERIC_READ : GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
0,
NULL
);
} else {
PrivateFile->DirHandle = CreateFile (
TempFileName,
PrivateFile->IsOpenedByRead ? GENERIC_READ : GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_FLAG_BACKUP_SEMANTICS,
NULL
);
StrCatS (TempFileName, Size / sizeof (CHAR16), L"\\*");
PrivateFile->LHandle = FindFirstFile (TempFileName, &FindBuf);
}
FreePool (TempFileName);
goto Done;
}
}
//
// Size change
//
if (SizeChangeFlag) {
if (PrivateFile->IsDirectoryPath) {
Status = EFI_UNSUPPORTED;
goto Done;
}
if (PrivateFile->IsOpenedByRead || OldFileInfo->Attribute & EFI_FILE_READ_ONLY) {
Status = EFI_ACCESS_DENIED;
goto Done;
}
Status = This->GetPosition (This, &CurPos);
if (EFI_ERROR (Status)) {
goto Done;
}
Status = This->SetPosition (This, NewFileInfo->FileSize);
if (EFI_ERROR (Status)) {
goto Done;
}
if (SetEndOfFile (PrivateFile->LHandle) == 0) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
Status = This->SetPosition (This, CurPos);
if (EFI_ERROR (Status)) {
goto Done;
}
}
//
// Time change
//
if (TimeChangeFlag) {
NewCreationSystemTime.wYear = NewFileInfo->CreateTime.Year;
NewCreationSystemTime.wMonth = NewFileInfo->CreateTime.Month;
NewCreationSystemTime.wDay = NewFileInfo->CreateTime.Day;
NewCreationSystemTime.wHour = NewFileInfo->CreateTime.Hour;
NewCreationSystemTime.wMinute = NewFileInfo->CreateTime.Minute;
NewCreationSystemTime.wSecond = NewFileInfo->CreateTime.Second;
NewCreationSystemTime.wMilliseconds = 0;
if (!SystemTimeToFileTime (
&NewCreationSystemTime,
&NewCreationFileTime
)) {
goto Done;
}
if (!LocalFileTimeToFileTime (
&NewCreationFileTime,
&NewCreationFileTime
)) {
goto Done;
}
NewLastAccessSystemTime.wYear = NewFileInfo->LastAccessTime.Year;
NewLastAccessSystemTime.wMonth = NewFileInfo->LastAccessTime.Month;
NewLastAccessSystemTime.wDay = NewFileInfo->LastAccessTime.Day;
NewLastAccessSystemTime.wHour = NewFileInfo->LastAccessTime.Hour;
NewLastAccessSystemTime.wMinute = NewFileInfo->LastAccessTime.Minute;
NewLastAccessSystemTime.wSecond = NewFileInfo->LastAccessTime.Second;
NewLastAccessSystemTime.wMilliseconds = 0;
if (!SystemTimeToFileTime (
&NewLastAccessSystemTime,
&NewLastAccessFileTime
)) {
goto Done;
}
if (!LocalFileTimeToFileTime (
&NewLastAccessFileTime,
&NewLastAccessFileTime
)) {
goto Done;
}
NewLastWriteSystemTime.wYear = NewFileInfo->ModificationTime.Year;
NewLastWriteSystemTime.wMonth = NewFileInfo->ModificationTime.Month;
NewLastWriteSystemTime.wDay = NewFileInfo->ModificationTime.Day;
NewLastWriteSystemTime.wHour = NewFileInfo->ModificationTime.Hour;
NewLastWriteSystemTime.wMinute = NewFileInfo->ModificationTime.Minute;
NewLastWriteSystemTime.wSecond = NewFileInfo->ModificationTime.Second;
NewLastWriteSystemTime.wMilliseconds = 0;
if (!SystemTimeToFileTime (
&NewLastWriteSystemTime,
&NewLastWriteFileTime
)) {
goto Done;
}
if (!LocalFileTimeToFileTime (
&NewLastWriteFileTime,
&NewLastWriteFileTime
)) {
goto Done;
}
if (!SetFileTime (
PrivateFile->IsDirectoryPath ? PrivateFile->DirHandle : PrivateFile->LHandle,
&NewCreationFileTime,
&NewLastAccessFileTime,
&NewLastWriteFileTime
)) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
}
//
// No matter about AttrChangeFlag, Attribute must be set.
// Because operation before may cause attribute change.
//
NewAttr = OldAttr;
if (NewFileInfo->Attribute & EFI_FILE_ARCHIVE) {
NewAttr |= FILE_ATTRIBUTE_ARCHIVE;
} else {
NewAttr &= ~FILE_ATTRIBUTE_ARCHIVE;
}
if (NewFileInfo->Attribute & EFI_FILE_HIDDEN) {
NewAttr |= FILE_ATTRIBUTE_HIDDEN;
} else {
NewAttr &= ~FILE_ATTRIBUTE_HIDDEN;
}
if (NewFileInfo->Attribute & EFI_FILE_SYSTEM) {
NewAttr |= FILE_ATTRIBUTE_SYSTEM;
} else {
NewAttr &= ~FILE_ATTRIBUTE_SYSTEM;
}
if (NewFileInfo->Attribute & EFI_FILE_READ_ONLY) {
NewAttr |= FILE_ATTRIBUTE_READONLY;
} else {
NewAttr &= ~FILE_ATTRIBUTE_READONLY;
}
NtStatus = SetFileAttributes (NewFileName, NewAttr);
if (!NtStatus) {
Status = EFI_DEVICE_ERROR;
goto Reopen;
}
Done:
if (OldFileInfo != NULL) {
FreePool (OldFileInfo);
}
if (OldFileName != NULL) {
FreePool (OldFileName);
}
if (NewFileName != NULL) {
FreePool (NewFileName);
}
return Status;
}
/**
Flush data back for the file handle.
@param This Protocol instance pointer.
@retval EFI_SUCCESS Data was written.
@retval EFI_UNSUPPORTED Writes to Open directory are not supported.
@retval EFI_NO_MEDIA The device has no media.
@retval EFI_DEVICE_ERROR The device reported an error.
@retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
@retval EFI_WRITE_PROTECTED The device is write protected.
@retval EFI_ACCESS_DENIED The file was open for read only.
@retval EFI_VOLUME_FULL The volume is full.
**/
EFI_STATUS
WinNtFileFlush (
IN EFI_FILE_PROTOCOL *This
)
{
BY_HANDLE_FILE_INFORMATION FileInfo;
WIN_NT_EFI_FILE_PRIVATE *PrivateFile;
EFI_STATUS Status;
PrivateFile = WIN_NT_EFI_FILE_PRIVATE_DATA_FROM_THIS (This);
if (PrivateFile->LHandle == INVALID_HANDLE_VALUE) {
Status = EFI_DEVICE_ERROR;
goto Done;
}
if (PrivateFile->IsDirectoryPath) {
Status = EFI_SUCCESS;
goto Done;
}
if (PrivateFile->IsOpenedByRead) {
Status = EFI_ACCESS_DENIED;
goto Done;
}
GetFileInformationByHandle (PrivateFile->LHandle, &FileInfo);
if (FileInfo.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
Status = EFI_ACCESS_DENIED;
goto Done;
}
Status = FlushFileBuffers (PrivateFile->LHandle) ? EFI_SUCCESS : EFI_DEVICE_ERROR;
Done:
return Status;
//
// bugbug: - Use Windows error reporting.
//
}
EFI_STATUS
WinNtFileSystmeThunkOpen (
IN EMU_IO_THUNK_PROTOCOL *This
)
{
WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE *Private;
Private = AllocateZeroPool (sizeof (*Private));
if (Private == NULL) {
return EFI_OUT_OF_RESOURCES;
}
Private->FilePath = AllocateCopyPool (StrSize (This->ConfigString), This->ConfigString);
if (Private->FilePath == NULL) {
FreePool (Private);
return EFI_OUT_OF_RESOURCES;
}
Private->VolumeLabel = AllocateCopyPool (StrSize (L"EFI_EMULATED"), L"EFI_EMULATED");
if (Private->VolumeLabel == NULL) {
FreePool (Private->FilePath);
FreePool (Private);
return EFI_OUT_OF_RESOURCES;
}
Private->Signature = WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE_SIGNATURE;
Private->Thunk = This;
CopyMem (&Private->SimpleFileSystem, &gWinNtFileSystemProtocol, sizeof (Private->SimpleFileSystem));
This->Interface = &Private->SimpleFileSystem;
This->Private = Private;
return EFI_SUCCESS;
}
EFI_STATUS
WinNtFileSystmeThunkClose (
IN EMU_IO_THUNK_PROTOCOL *This
)
{
WIN_NT_SIMPLE_FILE_SYSTEM_PRIVATE *Private;
Private = This->Private;
ASSERT (Private != NULL);
if (Private->VolumeLabel != NULL) {
FreePool (Private->VolumeLabel);
}
if (Private->FilePath != NULL) {
FreePool (Private->FilePath);
}
FreePool (Private);
return EFI_SUCCESS;
}
EFI_FILE_PROTOCOL gWinNtFileProtocol = {
EFI_FILE_REVISION,
WinNtFileOpen,
WinNtFileClose,
WinNtFileDelete,
WinNtFileRead,
WinNtFileWrite,
WinNtFileGetPossition,
WinNtFileSetPossition,
WinNtFileGetInfo,
WinNtFileSetInfo,
WinNtFileFlush
};
EFI_SIMPLE_FILE_SYSTEM_PROTOCOL gWinNtFileSystemProtocol = {
EFI_SIMPLE_FILE_SYSTEM_PROTOCOL_REVISION,
WinNtOpenVolume
};
EMU_IO_THUNK_PROTOCOL mWinNtFileSystemThunkIo = {
&gEfiSimpleFileSystemProtocolGuid,
NULL,
NULL,
0,
WinNtFileSystmeThunkOpen,
WinNtFileSystmeThunkClose,
NULL
};
Reference in New Issue View Git Blame Copy Permalink