EFI_FILE_PROTOCOL spec conformance bug fix.

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

/*++\r
\r
Copyright (c) 2005 - 2014, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials are licensed and made available\r
under the terms and conditions of the BSD License which accompanies this\r
distribution. The full text of the license may be found at\r
http://opensource.org/licenses/bsd-license.php\r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
\r
Module Name:\r
\r
  Fat.c\r
\r
Abstract:\r
\r
  Fat File System driver routines that support EFI driver model\r
\r
--*/\r
\r
#include "Fat.h"\r
\r
EFI_STATUS\r
EFIAPI\r
FatEntryPoint (\r
  IN EFI_HANDLE         ImageHandle,\r
  IN EFI_SYSTEM_TABLE   *SystemTable\r
  );\r
\r
EFI_STATUS\r
EFIAPI\r
FatUnload (\r
  IN EFI_HANDLE         ImageHandle\r
  );\r
\r
EFI_STATUS\r
EFIAPI\r
FatDriverBindingSupported (\r
  IN EFI_DRIVER_BINDING_PROTOCOL  *This,\r
  IN EFI_HANDLE                   Controller,\r
  IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath\r
  );\r
\r
EFI_STATUS\r
EFIAPI\r
FatDriverBindingStart (\r
  IN EFI_DRIVER_BINDING_PROTOCOL  *This,\r
  IN EFI_HANDLE                   Controller,\r
  IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath\r
  );\r
\r
EFI_STATUS\r
EFIAPI\r
FatDriverBindingStop (\r
  IN  EFI_DRIVER_BINDING_PROTOCOL  *This,\r
  IN  EFI_HANDLE                   Controller,\r
  IN  UINTN                        NumberOfChildren,\r
  IN  EFI_HANDLE                   *ChildHandleBuffer\r
  );\r
\r
//\r
// DriverBinding protocol instance\r
//\r
EFI_DRIVER_BINDING_PROTOCOL gFatDriverBinding = {\r
  FatDriverBindingSupported,\r
  FatDriverBindingStart,\r
  FatDriverBindingStop,\r
  0xa,\r
  NULL,\r
  NULL\r
};\r
\r
EFI_STATUS\r
EFIAPI\r
FatEntryPoint (\r
  IN EFI_HANDLE         ImageHandle,\r
  IN EFI_SYSTEM_TABLE   *SystemTable\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Register Driver Binding protocol for this driver.\r
\r
Arguments:\r
\r
  ImageHandle           - Handle for the image of this driver.\r
  SystemTable           - Pointer to the EFI System Table.\r
\r
Returns:\r
\r
  EFI_SUCCESS           - Driver loaded.\r
  other                 - Driver not loaded.\r
\r
--*/\r
{\r
  EFI_STATUS                Status;\r
\r
  //\r
  // Initialize the EFI Driver Library\r
  //\r
  Status = EfiLibInstallDriverBindingComponentName2 (\r
             ImageHandle,\r
             SystemTable,\r
             &gFatDriverBinding,\r
             ImageHandle,\r
             &gFatComponentName,\r
             &gFatComponentName2\r
             );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  return Status;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
FatUnload (\r
  IN EFI_HANDLE  ImageHandle\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Unload function for this image. Uninstall DriverBinding protocol.\r
\r
Arguments:\r
\r
  ImageHandle           - Handle for the image of this driver.\r
\r
Returns:\r
\r
  EFI_SUCCESS           - Driver unloaded successfully.\r
  other                 - Driver can not unloaded.\r
\r
--*/\r
{\r
  EFI_STATUS  Status;\r
  EFI_HANDLE  *DeviceHandleBuffer;\r
  UINTN       DeviceHandleCount;\r
  UINTN       Index;\r
  VOID        *ComponentName;\r
  VOID        *ComponentName2;\r
\r
  Status = gBS->LocateHandleBuffer (\r
                  AllHandles,\r
                  NULL,\r
                  NULL,\r
                  &DeviceHandleCount,\r
                  &DeviceHandleBuffer\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  for (Index = 0; Index < DeviceHandleCount; Index++) {\r
    Status = EfiTestManagedDevice (DeviceHandleBuffer[Index], ImageHandle, &gEfiDiskIoProtocolGuid);\r
    if (!EFI_ERROR (Status)) {\r
      Status = gBS->DisconnectController (\r
                      DeviceHandleBuffer[Index],\r
                      ImageHandle,\r
                      NULL\r
                      );\r
      if (EFI_ERROR (Status)) {\r
        break;\r
      }\r
    }\r
  }\r
\r
  if (Index == DeviceHandleCount) {\r
    //\r
    // Driver is stopped successfully.\r
    //\r
    Status = gBS->HandleProtocol (ImageHandle, &gEfiComponentNameProtocolGuid, &ComponentName);\r
    if (EFI_ERROR (Status)) {\r
      ComponentName = NULL;\r
    }\r
\r
    Status = gBS->HandleProtocol (ImageHandle, &gEfiComponentName2ProtocolGuid, &ComponentName2);\r
    if (EFI_ERROR (Status)) {\r
      ComponentName2 = NULL;\r
    }\r
\r
    if (ComponentName == NULL) {\r
      if (ComponentName2 == NULL) {\r
        Status = gBS->UninstallMultipleProtocolInterfaces (\r
                        ImageHandle,\r
                        &gEfiDriverBindingProtocolGuid,  &gFatDriverBinding,\r
                        NULL\r
                        );\r
      } else {\r
        Status = gBS->UninstallMultipleProtocolInterfaces (\r
                        ImageHandle,\r
                        &gEfiDriverBindingProtocolGuid,  &gFatDriverBinding,\r
                        &gEfiComponentName2ProtocolGuid, ComponentName2,\r
                        NULL\r
                        );\r
      }\r
    } else {\r
      if (ComponentName2 == NULL) {\r
        Status = gBS->UninstallMultipleProtocolInterfaces (\r
                        ImageHandle,\r
                        &gEfiDriverBindingProtocolGuid,  &gFatDriverBinding,\r
                        &gEfiComponentNameProtocolGuid,  ComponentName,\r
                        NULL\r
                        );\r
      } else {\r
        Status = gBS->UninstallMultipleProtocolInterfaces (\r
                        ImageHandle,\r
                        &gEfiDriverBindingProtocolGuid,  &gFatDriverBinding,\r
                        &gEfiComponentNameProtocolGuid,  ComponentName,\r
                        &gEfiComponentName2ProtocolGuid, ComponentName2,\r
                        NULL\r
                        );\r
      }\r
    }\r
  }\r
\r
  if (DeviceHandleBuffer != NULL) {\r
    FreePool (DeviceHandleBuffer);\r
  }\r
\r
  return Status;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
FatDriverBindingSupported (\r
  IN EFI_DRIVER_BINDING_PROTOCOL  *This,\r
  IN EFI_HANDLE                   ControllerHandle,\r
  IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Test to see if this driver can add a file system to ControllerHandle.\r
  ControllerHandle must support both Disk IO and Block IO protocols.\r
\r
Arguments:\r
\r
  This                  - Protocol instance pointer.\r
  ControllerHandle      - Handle of device to test.\r
  RemainingDevicePath   - Not used.\r
\r
Returns:\r
\r
  EFI_SUCCESS           - This driver supports this device.\r
  EFI_ALREADY_STARTED   - This driver is already running on this device.\r
  other                 - This driver does not support this device.\r
\r
--*/\r
{\r
  EFI_STATUS            Status;\r
  EFI_DISK_IO_PROTOCOL  *DiskIo;\r
\r
  //\r
  // Open the IO Abstraction(s) needed to perform the supported test\r
  //\r
  Status = gBS->OpenProtocol (\r
                  ControllerHandle,\r
                  &gEfiDiskIoProtocolGuid,\r
                  (VOID **) &DiskIo,\r
                  This->DriverBindingHandle,\r
                  ControllerHandle,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  //\r
  // Close the I/O Abstraction(s) used to perform the supported test\r
  //\r
  gBS->CloseProtocol (\r
         ControllerHandle,\r
         &gEfiDiskIoProtocolGuid,\r
         This->DriverBindingHandle,\r
         ControllerHandle\r
         );\r
\r
  //\r
  // Open the IO Abstraction(s) needed to perform the supported test\r
  //\r
  Status = gBS->OpenProtocol (\r
                  ControllerHandle,\r
                  &gEfiBlockIoProtocolGuid,\r
                  NULL,\r
                  This->DriverBindingHandle,\r
                  ControllerHandle,\r
                  EFI_OPEN_PROTOCOL_TEST_PROTOCOL\r
                  );\r
\r
  return Status;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
FatDriverBindingStart (\r
  IN EFI_DRIVER_BINDING_PROTOCOL  *This,\r
  IN EFI_HANDLE                   ControllerHandle,\r
  IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Start this driver on ControllerHandle by opening a Block IO and Disk IO\r
  protocol, reading Device Path. Add a Simple File System protocol to\r
  ControllerHandle if the media contains a valid file system.\r
\r
Arguments:\r
\r
  This                  - Protocol instance pointer.\r
  ControllerHandle      - Handle of device to bind driver to.\r
  RemainingDevicePath   - Not used.\r
\r
Returns:\r
\r
  EFI_SUCCESS           - This driver is added to DeviceHandle.\r
  EFI_ALREADY_STARTED   - This driver is already running on DeviceHandle.\r
  EFI_OUT_OF_RESOURCES  - Can not allocate the memory.\r
  other                 - This driver does not support this device.\r
\r
--*/\r
{\r
  EFI_STATUS            Status;\r
  EFI_BLOCK_IO_PROTOCOL *BlockIo;\r
  EFI_DISK_IO_PROTOCOL  *DiskIo;\r
  EFI_DISK_IO2_PROTOCOL *DiskIo2;\r
  BOOLEAN               LockedByMe;\r
\r
  LockedByMe = FALSE;\r
  //\r
  // Acquire the lock.\r
  // If caller has already acquired the lock, cannot lock it again.\r
  //\r
  Status = FatAcquireLockOrFail ();\r
  if (!EFI_ERROR (Status)) {\r
    LockedByMe = TRUE;\r
  }\r
\r
  Status = InitializeUnicodeCollationSupport (This->DriverBindingHandle);\r
  if (EFI_ERROR (Status)) {\r
    goto Exit;\r
  }\r
  //\r
  // Open our required BlockIo and DiskIo\r
  //\r
  Status = gBS->OpenProtocol (\r
                  ControllerHandle,\r
                  &gEfiBlockIoProtocolGuid,\r
                  (VOID **) &BlockIo,\r
                  This->DriverBindingHandle,\r
                  ControllerHandle,\r
                  EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    goto Exit;\r
  }\r
\r
  Status = gBS->OpenProtocol (\r
                  ControllerHandle,\r
                  &gEfiDiskIoProtocolGuid,\r
                  (VOID **) &DiskIo,\r
                  This->DriverBindingHandle,\r
                  ControllerHandle,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    goto Exit;\r
  }\r
\r
  Status = gBS->OpenProtocol (\r
                  ControllerHandle,\r
                  &gEfiDiskIo2ProtocolGuid,\r
                  (VOID **) &DiskIo2,\r
                  This->DriverBindingHandle,\r
                  ControllerHandle,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    DiskIo2 = NULL;\r
  }\r
\r
  //\r
  // Allocate Volume structure. In FatAllocateVolume(), Resources\r
  // are allocated with protocol installed and cached initialized\r
  //\r
  Status = FatAllocateVolume (ControllerHandle, DiskIo, DiskIo2, BlockIo);\r
\r
  //\r
  // When the media changes on a device it will Reinstall the BlockIo interaface.\r
  // This will cause a call to our Stop(), and a subsequent reentrant call to our\r
  // Start() successfully. We should leave the device open when this happen.\r
  //\r
  if (EFI_ERROR (Status)) {\r
    Status = gBS->OpenProtocol (\r
                    ControllerHandle,\r
                    &gEfiSimpleFileSystemProtocolGuid,\r
                    NULL,\r
                    This->DriverBindingHandle,\r
                    ControllerHandle,\r
                    EFI_OPEN_PROTOCOL_TEST_PROTOCOL\r
                    );\r
    if (EFI_ERROR (Status)) {\r
      gBS->CloseProtocol (\r
             ControllerHandle,\r
             &gEfiDiskIoProtocolGuid,\r
             This->DriverBindingHandle,\r
             ControllerHandle\r
             );\r
      gBS->CloseProtocol (\r
             ControllerHandle,\r
             &gEfiDiskIo2ProtocolGuid,\r
             This->DriverBindingHandle,\r
             ControllerHandle\r
             );\r
    }\r
  }\r
\r
Exit:\r
  //\r
  // Unlock if locked by myself.\r
  //\r
  if (LockedByMe) {\r
    FatReleaseLock ();\r
  }\r
  return Status;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
FatDriverBindingStop (\r
  IN  EFI_DRIVER_BINDING_PROTOCOL   *This,\r
  IN  EFI_HANDLE                    ControllerHandle,\r
  IN  UINTN                         NumberOfChildren,\r
  IN  EFI_HANDLE                    *ChildHandleBuffer\r
  )\r
/*++\r
\r
Routine Description:\r
  Stop this driver on ControllerHandle.\r
\r
Arguments:\r
  This                  - Protocol instance pointer.\r
  ControllerHandle      - Handle of device to stop driver on.\r
  NumberOfChildren      - Not used.\r
  ChildHandleBuffer     - Not used.\r
\r
Returns:\r
  EFI_SUCCESS           - This driver is removed DeviceHandle.\r
  other                 - This driver was not removed from this device.\r
\r
--*/\r
{\r
  EFI_STATUS                      Status;\r
  EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *FileSystem;\r
  FAT_VOLUME                      *Volume;\r
  EFI_DISK_IO2_PROTOCOL           *DiskIo2;\r
\r
  DiskIo2 = NULL;\r
  //\r
  // Get our context back\r
  //\r
  Status = gBS->OpenProtocol (\r
                  ControllerHandle,\r
                  &gEfiSimpleFileSystemProtocolGuid,\r
                  (VOID **) &FileSystem,\r
                  This->DriverBindingHandle,\r
                  ControllerHandle,\r
                  EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
                  );\r
\r
  if (!EFI_ERROR (Status)) {\r
    Volume  = VOLUME_FROM_VOL_INTERFACE (FileSystem);\r
    DiskIo2 = Volume->DiskIo2;\r
    Status  = FatAbandonVolume (Volume);\r
  }\r
\r
  if (!EFI_ERROR (Status)) {\r
    if (DiskIo2 != NULL) {\r
      Status = gBS->CloseProtocol (\r
        ControllerHandle,\r
        &gEfiDiskIo2ProtocolGuid,\r
        This->DriverBindingHandle,\r
        ControllerHandle\r
        );\r
      ASSERT_EFI_ERROR (Status);\r
    }\r
    Status = gBS->CloseProtocol (\r
      ControllerHandle,\r
      &gEfiDiskIoProtocolGuid,\r
      This->DriverBindingHandle,\r
      ControllerHandle\r
      );\r
    ASSERT_EFI_ERROR (Status);\r
  }\r
\r
  return Status;\r
}\r