EdbSupportFile.c

Go to the documentation of this file.

 
#include "Edb.h"
 
EFI_STATUS
EFIAPI
ReadFileFromVol (
  IN  EFI_SIMPLE_FILE_SYSTEM_PROTOCOL  *Vol,
  IN  CHAR16                           *FileName,
  OUT UINTN                            *BufferSize,
  OUT VOID                             **Buffer
  )
{
  EFI_STATUS       Status;
  EFI_FILE_HANDLE  RootDir;
  EFI_FILE_HANDLE  Handle;
  UINTN            FileInfoSize;
  EFI_FILE_INFO    *FileInfo;
  UINTN            TempBufferSize;
  VOID             *TempBuffer;
 
  //
  // Open the root directory
  //
  Status = Vol->OpenVolume (Vol, &RootDir);
  if (EFI_ERROR (Status)) {
    return Status;
  }
 
  //
  // Open the file
  //
  Status = RootDir->Open (
                      RootDir,
                      &Handle,
                      FileName,
                      EFI_FILE_MODE_READ,
                      0
                      );
  if (EFI_ERROR (Status)) {
    RootDir->Close (RootDir);
    return Status;
  }
 
  RootDir->Close (RootDir);
 
  //
  // Get the file information
  //
  FileInfoSize = sizeof (EFI_FILE_INFO) + 1024;
 
  FileInfo = AllocateZeroPool (FileInfoSize);
  if (FileInfo == NULL) {
    Handle->Close (Handle);
    return Status;
  }
 
  Status = Handle->GetInfo (
                     Handle,
                     &gEfiFileInfoGuid,
                     &FileInfoSize,
                     FileInfo
                     );
  if (EFI_ERROR (Status)) {
    Handle->Close (Handle);
    gBS->FreePool (FileInfo);
    return Status;
  }
 
  //
  // Allocate buffer for the file data. The last CHAR16 is for L'\0'
  //
  TempBufferSize = (UINTN)FileInfo->FileSize + sizeof (CHAR16);
  TempBuffer     = AllocateZeroPool (TempBufferSize);
  if (TempBuffer == NULL) {
    Handle->Close (Handle);
    gBS->FreePool (FileInfo);
    return Status;
  }
 
  gBS->FreePool (FileInfo);
 
  //
  // Read the file data to the buffer
  //
  Status = Handle->Read (
                     Handle,
                     &TempBufferSize,
                     TempBuffer
                     );
  if (EFI_ERROR (Status)) {
    Handle->Close (Handle);
    gBS->FreePool (TempBuffer);
    return Status;
  }
 
  Handle->Close (Handle);
 
  *BufferSize = TempBufferSize;
  *Buffer     = TempBuffer;
  return EFI_SUCCESS;
}
 
EFI_STATUS
EFIAPI
ReadFileToBuffer (
  IN  EFI_DEBUGGER_PRIVATE_DATA  *DebuggerPrivate,
  IN  CHAR16                     *FileName,
  OUT UINTN                      *BufferSize,
  OUT VOID                       **Buffer,
  IN  BOOLEAN                    ScanFs
  )
{
  EFI_STATUS                       Status;
  EFI_SIMPLE_FILE_SYSTEM_PROTOCOL  *Vol;
  UINTN                            TempBufferSize;
  VOID                             *TempBuffer;
  UINTN                            NoHandles;
  EFI_HANDLE                       *HandleBuffer;
  UINTN                            Index;
 
  //
  // Check parameters
  //
  if ((FileName == NULL) || (Buffer == NULL)) {
    return EFI_INVALID_PARAMETER;
  }
 
  //
  // not scan fs
  //
  if (!ScanFs) {
    if (DebuggerPrivate->Vol == NULL) {
      return EFI_INVALID_PARAMETER;
    }
 
    //
    // Read file directly from Vol
    //
    return ReadFileFromVol (DebuggerPrivate->Vol, FileName, BufferSize, Buffer);
  }
 
  //
  // need scan fs
  //
 
  //
  // Get all Vol handle
  //
  Status = gBS->LocateHandleBuffer (
                  ByProtocol,
                  &gEfiSimpleFileSystemProtocolGuid,
                  NULL,
                  &NoHandles,
                  &HandleBuffer
                  );
  if (EFI_ERROR (Status) && (NoHandles == 0)) {
    return EFI_NOT_FOUND;
  }
 
  //
  // Walk through each Vol
  //
  DebuggerPrivate->Vol = NULL;
  *BufferSize          = 0;
  *Buffer              = NULL;
  for (Index = 0; Index < NoHandles; Index++) {
    Status = gBS->HandleProtocol (
                    HandleBuffer[Index],
                    &gEfiSimpleFileSystemProtocolGuid,
                    (VOID **)&Vol
                    );
    if (EFI_ERROR (Status)) {
      continue;
    }
 
    Status = ReadFileFromVol (Vol, FileName, &TempBufferSize, &TempBuffer);
    if (!EFI_ERROR (Status)) {
      //
      // Read file OK, check duplication
      //
      if (DebuggerPrivate->Vol != NULL) {
        //
        // Find the duplicated file
        //
        gBS->FreePool (TempBuffer);
        gBS->FreePool (*Buffer);
        EDBPrint (L"Duplicated FileName found!\n");
        return EFI_NO_MAPPING;
      } else {
        //
        // Record value
        //
        DebuggerPrivate->Vol = Vol;
        *BufferSize          = TempBufferSize;
        *Buffer              = TempBuffer;
      }
    }
  }
 
  //
  // Scan Fs done
  //
  if (DebuggerPrivate->Vol == NULL) {
    return EFI_NOT_FOUND;
  }
 
  //
  // Done
  //
  return EFI_SUCCESS;
}
 
CHAR16 *
EFIAPI
GetFileNameUnderDir (
  IN  EFI_DEBUGGER_PRIVATE_DATA  *DebuggerPrivate,
  IN  CHAR16                     *DirName,
  IN  CHAR16                     *FileName,
  IN OUT UINTN                   *Index
  )
{
  EFI_STATUS                       Status;
  EFI_FILE_HANDLE                  RootDir;
  EFI_FILE_HANDLE                  Handle;
  UINTN                            FileInfoSize;
  EFI_FILE_INFO                    *FileInfo;
  EFI_SIMPLE_FILE_SYSTEM_PROTOCOL  *Vol;
  VOID                             *TempName;
  UINTN                            FileIndex;
 
  if (DebuggerPrivate->Vol == NULL) {
    Status = gBS->LocateProtocol (
                    &gEfiSimpleFileSystemProtocolGuid,
                    NULL,
                    (VOID **)&DebuggerPrivate->Vol
                    );
    if (EFI_ERROR (Status)) {
      return NULL;
    }
  }
 
  Vol = DebuggerPrivate->Vol;
 
  //
  // Open the root directory
  //
  Status = Vol->OpenVolume (Vol, &RootDir);
  if (EFI_ERROR (Status)) {
    return NULL;
  }
 
  //
  // Open the file
  //
  Status = RootDir->Open (
                      RootDir,
                      &Handle,
                      DirName,
                      EFI_FILE_MODE_READ,
                      EFI_FILE_DIRECTORY
                      );
  if (EFI_ERROR (Status)) {
    RootDir->Close (RootDir);
    return NULL;
  }
 
  RootDir->Close (RootDir);
 
  //
  // Set Dir Position
  //
  Status = Handle->SetPosition (Handle, 0);
  if (EFI_ERROR (Status)) {
    Handle->Close (Handle);
    return NULL;
  }
 
  //
  // Get the file information
  //
  FileInfoSize = sizeof (EFI_FILE_INFO) + 1024;
 
  FileInfo = AllocateZeroPool (FileInfoSize);
  if (FileInfo == NULL) {
    Handle->Close (Handle);
    return NULL;
  }
 
  //
  // Walk through each file in the directory
  //
  FileIndex = 0;
  TempName  = NULL;
  while (TRUE) {
    //
    // Read a file entry
    //
    FileInfoSize = sizeof (EFI_FILE_INFO) + 1024;
 
    Status = Handle->Read (
                       Handle,
                       &FileInfoSize,
                       FileInfo
                       );
    if (EFI_ERROR (Status) || (FileInfoSize == 0)) {
      break;
    }
 
    if ((FileInfo->Attribute & EFI_FILE_DIRECTORY) == 0) {
      //
      // This is a file
      //
 
      //
      // Only deal with the EFI key file
      //
      if (!StrEndWith (FileInfo->FileName, FileName)) {
        continue;
      }
 
      if (FileIndex == *Index) {
        TempName = StrDuplicate (FileInfo->FileName);
        *Index   = *Index + 1;
        break;
      }
 
      FileIndex++;
    }
  }
 
  //
  // Free resources
  //
  gBS->FreePool (FileInfo);
  Handle->Close (Handle);
 
  return TempName;
}