/* This function will provide the caller with the specified block device's media information. If the media changes, calling this function will update the media information accordingly. */
EFI_STATUS EFIAPI BotGetMediaInfo2(IN EFI_PEI_SERVICES **PeiServices, IN EFI_PEI_RECOVERY_BLOCK_IO2_PPI *This, IN UINTN DeviceIndex, OUT EFI_PEI_BLOCK_IO2_MEDIA *MediaInfo)
/* This function will provide the caller with the specified block device's media information. If the media changes, calling this function will update the media information accordingly. */
EFI_STATUS EFIAPI BotGetMediaInfo2(IN EFI_PEI_SERVICES **PeiServices, IN EFI_PEI_RECOVERY_BLOCK_IO2_PPI *This, IN UINTN DeviceIndex, OUT EFI_PEI_BLOCK_IO2_MEDIA *MediaInfo)
{
  PEI_BOT_DEVICE  *PeiBotDev;
  EFI_STATUS      Status;
  PeiBotDev = PEI_BOT_DEVICE2_FROM_THIS (This);
  Status = BotGetMediaInfo (
             PeiServices,
             &PeiBotDev->BlkIoPpi,
             DeviceIndex,
             &PeiBotDev->Media
             );
  if (EFI_ERROR (Status)) {
    return Status;
  }
  CopyMem (
    MediaInfo,
    &(PeiBotDev->Media2),
    sizeof (EFI_PEI_BLOCK_IO2_MEDIA)
    );
  return EFI_SUCCESS;
}