MdeModulePkg: Fix use-after-free error in InstallConfigurationTable()

[mirror_edk2.git] / Vlv2TbltDevicePkg / FvbRuntimeDxe / FvbServiceSmm.c

/** @file\r
  SMM Firmware Volume Block Driver for Lakeport Platform.\r
\r
  Firmware volume block driver for FWH or SPI device.\r
  It depends on which Flash Device Library to be linked with this driver.\r
\r
Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>\r
                                                                                   \r\r
  This program and the accompanying materials are licensed and made available under\r\r
  the terms and conditions of the BSD License that accompanies this distribution.  \r\r
  The full text of the license may be found at                                     \r\r
  http://opensource.org/licenses/bsd-license.php.                                  \r\r
                                                                                   \r\r
  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,            \r\r
  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.    \r\r
                                                                                   \r\r
\r
**/\r
\r
#include <PiSmm.h>\r
#include <Library/SmmServicesTableLib.h>\r
#include "FvbSmmCommon.h"\r
#include "FvbService.h"\r
\r
/**\r
  The function installs EFI_SMM_FIRMWARE_VOLUME_BLOCK protocol\r
  for each FV in the system.\r
\r
  @param[in]  FwhInstance   The pointer to a FW volume instance structure,\r
                            which contains the information about one FV.\r
  @param[in]  InstanceNum   The instance number which can be used as a ID\r
                            to locate this FwhInstance in other functions.\r
\r
  @retval     VOID\r
\r
**/\r
VOID\r
InstallFvbProtocol (\r
  IN  EFI_FW_VOL_INSTANCE               *FwhInstance,\r
  IN  UINTN                             InstanceNum\r
  )\r
{\r
  EFI_FW_VOL_BLOCK_DEVICE               *FvbDevice;\r
  EFI_FIRMWARE_VOLUME_HEADER            *FwVolHeader;\r
  EFI_STATUS                            Status;\r
  EFI_HANDLE                            FvbHandle;\r
\r
  FvbDevice = (EFI_FW_VOL_BLOCK_DEVICE *) AllocateRuntimeCopyPool (\r
                                            sizeof (EFI_FW_VOL_BLOCK_DEVICE),\r
                                            &mFvbDeviceTemplate\r
                                            );\r
  ASSERT (FvbDevice != NULL);\r
\r
  FvbDevice->Instance = InstanceNum;\r
  FwVolHeader         = &FwhInstance->VolumeHeader;\r
\r
  //\r
  // Set up the devicepath.\r
  //\r
  if (FwVolHeader->ExtHeaderOffset == 0) {\r
    //\r
    // FV does not contains extension header, then produce MEMMAP_DEVICE_PATH.\r
    //\r
    FvbDevice->DevicePath = (EFI_DEVICE_PATH_PROTOCOL *) AllocateRuntimeCopyPool (sizeof (FV_MEMMAP_DEVICE_PATH), &mFvMemmapDevicePathTemplate);\r
    ((FV_MEMMAP_DEVICE_PATH *) FvbDevice->DevicePath)->MemMapDevPath.StartingAddress = FwhInstance->FvBase;\r
    ((FV_MEMMAP_DEVICE_PATH *) FvbDevice->DevicePath)->MemMapDevPath.EndingAddress   = FwhInstance->FvBase + FwVolHeader->FvLength - 1;\r
  } else {\r
    FvbDevice->DevicePath = (EFI_DEVICE_PATH_PROTOCOL *) AllocateRuntimeCopyPool (sizeof (FV_PIWG_DEVICE_PATH), &mFvPIWGDevicePathTemplate);\r
    CopyGuid (\r
      &((FV_PIWG_DEVICE_PATH *)FvbDevice->DevicePath)->FvDevPath.FvName,\r
      (GUID *)(UINTN)(FwhInstance->FvBase + FwVolHeader->ExtHeaderOffset)\r
      );\r
  }\r
\r
  //\r
  // Install the SMM Firmware Volume Block Protocol and Device Path Protocol.\r
  //\r
  FvbHandle = NULL;\r
  Status = gSmst->SmmInstallProtocolInterface (\r
                    &FvbHandle,\r
                    &gEfiSmmFirmwareVolumeBlockProtocolGuid,\r
                    EFI_NATIVE_INTERFACE,\r
                    &FvbDevice->FwVolBlockInstance\r
                    );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  Status = gSmst->SmmInstallProtocolInterface (\r
                    &FvbHandle,\r
                    &gEfiDevicePathProtocolGuid,\r
                    EFI_NATIVE_INTERFACE,\r
                    FvbDevice->DevicePath\r
                    );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  //\r
  // Notify the Fvb wrapper driver SMM fvb is ready.\r
  //\r
  FvbHandle = NULL;\r
  Status = gBS->InstallProtocolInterface (\r
                  &FvbHandle,\r
                  &gEfiSmmFirmwareVolumeBlockProtocolGuid,\r
                  EFI_NATIVE_INTERFACE,\r
                  &FvbDevice->FwVolBlockInstance\r
                  );\r
}\r
\r
\r
/**\r
  The driver entry point for SMM Firmware Volume Block Driver.\r
\r
  The function does the necessary initialization work\r
  Firmware Volume Block Driver.\r
\r
  @param[in]  ImageHandle       The firmware allocated handle for the UEFI image.\r
  @param[in]  SystemTable       A pointer to the EFI system table.\r
\r
  @retval     EFI_SUCCESS       This funtion always return EFI_SUCCESS.\r
                                It will ASSERT on errors.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FvbSmmInitialize (\r
  IN EFI_HANDLE         ImageHandle,\r
  IN EFI_SYSTEM_TABLE   *SystemTable\r
  )\r
{\r
  FvbInitialize ();\r
\r
  return EFI_SUCCESS;\r
}\r
\r