BaseTools: Library hashing fix and optimization for --hash feature

[mirror_edk2.git] / FatPkg / EnhancedFatDxe / Flush.c

/** @file\r
  Routines that check references and flush OFiles\r
\r
Copyright (c) 2005 - 2013, Intel Corporation. All rights reserved.<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
\r
**/\r
\r
#include "Fat.h"\r
\r
/**\r
\r
  Flushes all data associated with the file handle.\r
\r
  @param  FHand                 - Handle to file to flush.\r
  @param  Token                 - A pointer to the token associated with the transaction.\r
\r
  @retval EFI_SUCCESS           - Flushed the file successfully.\r
  @retval EFI_WRITE_PROTECTED   - The volume is read only.\r
  @retval EFI_ACCESS_DENIED     - The file is read only.\r
  @return Others                - Flushing of the file failed.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FatFlushEx (\r
  IN EFI_FILE_PROTOCOL  *FHand,\r
  IN EFI_FILE_IO_TOKEN  *Token\r
  )\r
{\r
  FAT_IFILE   *IFile;\r
  FAT_OFILE   *OFile;\r
  FAT_VOLUME  *Volume;\r
  EFI_STATUS  Status;\r
  FAT_TASK    *Task;\r
\r
  IFile   = IFILE_FROM_FHAND (FHand);\r
  OFile   = IFile->OFile;\r
  Volume  = OFile->Volume;\r
  Task    = NULL;\r
\r
  //\r
  // If the file has a permanent error, return it\r
  //\r
  if (EFI_ERROR (OFile->Error)) {\r
    return OFile->Error;\r
  }\r
\r
  if (Volume->ReadOnly) {\r
    return EFI_WRITE_PROTECTED;\r
  }\r
  //\r
  // If read only, return error\r
  //\r
  if (IFile->ReadOnly) {\r
    return EFI_ACCESS_DENIED;\r
  }\r
\r
  if (Token == NULL) {\r
    FatWaitNonblockingTask (IFile);\r
  } else {\r
    //\r
    // Caller shouldn't call the non-blocking interfaces if the low layer doesn't support DiskIo2.\r
    // But if it calls, the below check can avoid crash.\r
    //\r
    if (FHand->Revision < EFI_FILE_PROTOCOL_REVISION2) {\r
      return EFI_UNSUPPORTED;\r
    }\r
    Task = FatCreateTask (IFile, Token);\r
    if (Task == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
  }\r
\r
  //\r
  // Flush the OFile\r
  //\r
  FatAcquireLock ();\r
  Status  = FatOFileFlush (OFile);\r
  Status  = FatCleanupVolume (OFile->Volume, OFile, Status, Task);\r
  FatReleaseLock ();\r
\r
  if (Token != NULL) {\r
    if (!EFI_ERROR (Status)) {\r
      Status = FatQueueTask (IFile, Task);\r
    } else {\r
      FatDestroyTask (Task);\r
    }\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
\r
  Flushes all data associated with the file handle.\r
\r
  @param  FHand                 - Handle to file to flush.\r
\r
  @retval EFI_SUCCESS           - Flushed the file successfully.\r
  @retval EFI_WRITE_PROTECTED   - The volume is read only.\r
  @retval EFI_ACCESS_DENIED     - The file is read only.\r
  @return Others                - Flushing of the file failed.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FatFlush (\r
  IN EFI_FILE_PROTOCOL  *FHand\r
  )\r
{\r
  return FatFlushEx (FHand, NULL);\r
}\r
\r
/**\r
\r
  Flushes & Closes the file handle.\r
\r
  @param  FHand                 - Handle to the file to delete.\r
\r
  @retval EFI_SUCCESS           - Closed the file successfully.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FatClose (\r
  IN EFI_FILE_PROTOCOL  *FHand\r
  )\r
{\r
  FAT_IFILE   *IFile;\r
  FAT_OFILE   *OFile;\r
  FAT_VOLUME  *Volume;\r
\r
  IFile   = IFILE_FROM_FHAND (FHand);\r
  OFile   = IFile->OFile;\r
  Volume  = OFile->Volume;\r
\r
  //\r
  // Lock the volume\r
  //\r
  FatAcquireLock ();\r
\r
  //\r
  // Close the file instance handle\r
  //\r
  FatIFileClose (IFile);\r
\r
  //\r
  // Done. Unlock the volume\r
  //\r
  FatCleanupVolume (Volume, OFile, EFI_SUCCESS, NULL);\r
  FatReleaseLock ();\r
\r
  //\r
  // Close always succeed\r
  //\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
\r
  Close the open file instance.\r
\r
  @param  IFile                 - Open file instance.\r
\r
  @retval EFI_SUCCESS           - Closed the file successfully.\r
\r
**/\r
EFI_STATUS\r
FatIFileClose (\r
  FAT_IFILE           *IFile\r
  )\r
{\r
  FAT_OFILE   *OFile;\r
  FAT_VOLUME  *Volume;\r
\r
  OFile   = IFile->OFile;\r
  Volume  = OFile->Volume;\r
\r
  ASSERT_VOLUME_LOCKED (Volume);\r
\r
  FatWaitNonblockingTask (IFile);\r
\r
  //\r
  // Remove the IFile struct\r
  //\r
  RemoveEntryList (&IFile->Link);\r
\r
  //\r
  // Add the OFile to the check reference list\r
  //\r
  if (OFile->CheckLink.ForwardLink == NULL) {\r
    InsertHeadList (&Volume->CheckRef, &OFile->CheckLink);\r
  }\r
  //\r
  // Done. Free the open instance structure\r
  //\r
  FreePool (IFile);\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
\r
  Flush the data associated with an open file.\r
  In this implementation, only last Mod/Access time is updated.\r
\r
  @param  OFile                 - The open file.\r
\r
  @retval EFI_SUCCESS           - The OFile is flushed successfully.\r
  @return Others                - An error occurred when flushing this OFile.\r
\r
**/\r
EFI_STATUS\r
FatOFileFlush (\r
  IN FAT_OFILE    *OFile\r
  )\r
{\r
  EFI_STATUS    Status;\r
  FAT_OFILE     *Parent;\r
  FAT_DIRENT    *DirEnt;\r
  FAT_DATE_TIME FatNow;\r
\r
  //\r
  // Flush each entry up the tree while dirty\r
  //\r
  do {\r
    //\r
    // If the file has a permanant error, then don't write any\r
    // of its data to the device (may be from different media)\r
    //\r
    if (EFI_ERROR (OFile->Error)) {\r
      return OFile->Error;\r
    }\r
\r
    Parent  = OFile->Parent;\r
    DirEnt  = OFile->DirEnt;\r
    if (OFile->Dirty) {\r
      //\r
      // Update the last modification time\r
      //\r
      FatGetCurrentFatTime (&FatNow);\r
      CopyMem (&DirEnt->Entry.FileLastAccess, &FatNow.Date, sizeof (FAT_DATE));\r
      if (!OFile->PreserveLastModification) {\r
        FatGetCurrentFatTime (&DirEnt->Entry.FileModificationTime);\r
      }\r
\r
      OFile->PreserveLastModification = FALSE;\r
      if (OFile->Archive) {\r
        DirEnt->Entry.Attributes |= FAT_ATTRIBUTE_ARCHIVE;\r
        OFile->Archive = FALSE;\r
      }\r
      //\r
      // Write the directory entry\r
      //\r
      if (Parent != NULL && !DirEnt->Invalid) {\r
        //\r
        // Write the OFile's directory entry\r
        //\r
        Status = FatStoreDirEnt (Parent, DirEnt);\r
        if (EFI_ERROR (Status)) {\r
          return Status;\r
        }\r
      }\r
\r
      OFile->Dirty = FALSE;\r
    }\r
    //\r
    // Check the parent\r
    //\r
    OFile = Parent;\r
  } while (OFile != NULL);\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
\r
  Check the references of the OFile.\r
  If the OFile (that is checked) is no longer\r
  referenced, then it is freed.\r
\r
  @param  OFile                 - The OFile to be checked.\r
\r
  @retval TRUE                  - The OFile is not referenced and freed.\r
  @retval FALSE                 - The OFile is kept.\r
\r
**/\r
BOOLEAN\r
FatCheckOFileRef (\r
  IN FAT_OFILE   *OFile\r
  )\r
{\r
  //\r
  // If the OFile is on the check ref list, remove it\r
  //\r
  if (OFile->CheckLink.ForwardLink != NULL) {\r
    RemoveEntryList (&OFile->CheckLink);\r
    OFile->CheckLink.ForwardLink = NULL;\r
  }\r
\r
  FatOFileFlush (OFile);\r
  //\r
  // Are there any references to this OFile?\r
  //\r
  if (!IsListEmpty (&OFile->Opens) || !IsListEmpty (&OFile->ChildHead)) {\r
    //\r
    // The OFile cannot be freed\r
    //\r
    return FALSE;\r
  }\r
  //\r
  // Free the Ofile\r
  //\r
  FatCloseDirEnt (OFile->DirEnt);\r
  return TRUE;\r
}\r
\r
/**\r
\r
  Check the references of all open files on the volume.\r
  Any open file (that is checked) that is no longer\r
  referenced, is freed - and it's parent open file\r
  is then referenced checked.\r
\r
  @param  Volume                - The volume to check the pending open file list.\r
\r
**/\r
STATIC\r
VOID\r
FatCheckVolumeRef (\r
  IN FAT_VOLUME   *Volume\r
  )\r
{\r
  FAT_OFILE *OFile;\r
  FAT_OFILE *Parent;\r
\r
  //\r
  // Check all files on the pending check list\r
  //\r
  while (!IsListEmpty (&Volume->CheckRef)) {\r
    //\r
    // Start with the first file listed\r
    //\r
    Parent = OFILE_FROM_CHECKLINK (Volume->CheckRef.ForwardLink);\r
    //\r
    // Go up the tree cleaning up any un-referenced OFiles\r
    //\r
    while (Parent != NULL) {\r
      OFile   = Parent;\r
      Parent  = OFile->Parent;\r
      if (!FatCheckOFileRef (OFile)) {\r
        break;\r
      }\r
    }\r
  }\r
}\r
\r
/**\r
\r
  Set error status for a specific OFile, reference checking the volume.\r
  If volume is already marked as invalid, and all resources are freed\r
  after reference checking, the file system protocol is uninstalled and\r
  the volume structure is freed.\r
\r
  @param  Volume                - the Volume that is to be reference checked and unlocked.\r
  @param  OFile                 - the OFile whose permanent error code is to be set.\r
  @param  EfiStatus             - error code to be set.\r
  @param  Task                    point to task instance.\r
\r
  @retval EFI_SUCCESS           - Clean up the volume successfully.\r
  @return Others                - Cleaning up of the volume is failed.\r
\r
**/\r
EFI_STATUS\r
FatCleanupVolume (\r
  IN FAT_VOLUME       *Volume,\r
  IN FAT_OFILE        *OFile,\r
  IN EFI_STATUS       EfiStatus,\r
  IN FAT_TASK         *Task\r
  )\r
{\r
  EFI_STATUS  Status;\r
  //\r
  // Flag the OFile\r
  //\r
  if (OFile != NULL) {\r
    FatSetVolumeError (OFile, EfiStatus);\r
  }\r
  //\r
  // Clean up any dangling OFiles that don't have IFiles\r
  // we don't check return status here because we want the\r
  // volume be cleaned up even the volume is invalid.\r
  //\r
  FatCheckVolumeRef (Volume);\r
  if (Volume->Valid) {\r
    //\r
    // Update the free hint info. Volume->FreeInfoPos != 0\r
    // indicates this a FAT32 volume\r
    //\r
    if (Volume->FreeInfoValid && Volume->FatDirty && Volume->FreeInfoPos) {\r
      Status = FatDiskIo (Volume, WriteDisk, Volume->FreeInfoPos, sizeof (FAT_INFO_SECTOR), &Volume->FatInfoSector, Task);\r
      if (EFI_ERROR (Status)) {\r
        return Status;\r
      }\r
    }\r
    //\r
    // Update that the volume is not dirty\r
    //\r
    if (Volume->FatDirty && Volume->FatType != Fat12) {\r
      Volume->FatDirty  = FALSE;\r
      Status            = FatAccessVolumeDirty (Volume, WriteFat, &Volume->NotDirtyValue);\r
      if (EFI_ERROR (Status)) {\r
        return Status;\r
      }\r
    }\r
    //\r
    // Flush all dirty cache entries to disk\r
    //\r
    Status = FatVolumeFlushCache (Volume, Task);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
  }\r
  //\r
  // If the volume is cleared , remove it.\r
  // The only time volume be invalidated is in DriverBindingStop.\r
  //\r
  if (Volume->Root == NULL && !Volume->Valid) {\r
    //\r
    // Free the volume structure\r
    //\r
    FatFreeVolume (Volume);\r
  }\r
\r
  return EfiStatus;\r
}\r
\r
/**\r
\r
  Set the OFile and its child OFile with the error Status\r
\r
  @param  OFile                 - The OFile whose permanent error code is to be set.\r
  @param  Status                - Error code to be set.\r
\r
**/\r
VOID\r
FatSetVolumeError (\r
  IN FAT_OFILE            *OFile,\r
  IN EFI_STATUS           Status\r
  )\r
{\r
  LIST_ENTRY      *Link;\r
  FAT_OFILE       *ChildOFile;\r
\r
  //\r
  // If this OFile doesn't already have an error, set one\r
  //\r
  if (!EFI_ERROR (OFile->Error)) {\r
    OFile->Error = Status;\r
  }\r
  //\r
  // Set the error on each child OFile\r
  //\r
  for (Link = OFile->ChildHead.ForwardLink; Link != &OFile->ChildHead; Link = Link->ForwardLink) {\r
    ChildOFile = OFILE_FROM_CHILDLINK (Link);\r
    FatSetVolumeError (ChildOFile, Status);\r
  }\r
}\r