CapsulePei coalesce need add more sanity check for each Capsule Fragment.

[mirror_edk2.git] / MdeModulePkg / Universal / CapsulePei / Common / CommonHeader.h

/** @file\r
  Common header file.\r
\r
Copyright (c) 2011 - 2014, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this 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
\r
#ifndef _CAPSULE_COMMON_HEADER_\r
#define _CAPSULE_COMMON_HEADER_\r
\r
//\r
// This capsule PEIM puts its private data at the start of the\r
// coalesced capsule. Here's the structure definition.\r
//\r
#define EFI_CAPSULE_PEIM_PRIVATE_DATA_SIGNATURE SIGNATURE_32 ('C', 'a', 'p', 'P')\r
\r
#pragma pack(1)\r
typedef struct {\r
  UINT64  Signature;\r
  UINT64  CapsuleAllImageSize;\r
  UINT64  CapsuleNumber;\r
  UINT64  CapsuleOffset[1];\r
} EFI_CAPSULE_PEIM_PRIVATE_DATA;\r
#pragma pack()\r
\r
#define CAPSULE_TEST_SIGNATURE SIGNATURE_32('T', 'E', 'S', 'T')\r
\r
#if defined (MDE_CPU_IA32) || defined (MDE_CPU_X64)\r
typedef struct {\r
  EFI_PHYSICAL_ADDRESS  EntryPoint;\r
  EFI_PHYSICAL_ADDRESS  StackBufferBase;\r
  UINT64                StackBufferLength;\r
  EFI_PHYSICAL_ADDRESS  JumpBuffer;\r
  EFI_PHYSICAL_ADDRESS  BlockListAddr;\r
  EFI_PHYSICAL_ADDRESS  MemoryBase64Ptr;\r
  EFI_PHYSICAL_ADDRESS  MemorySize64Ptr;\r
} SWITCH_32_TO_64_CONTEXT;\r
\r
typedef struct {\r
  UINT16                ReturnCs;\r
  EFI_PHYSICAL_ADDRESS  ReturnEntryPoint;\r
  UINT64                ReturnStatus;\r
  IA32_DESCRIPTOR       Gdtr;\r
} SWITCH_64_TO_32_CONTEXT;\r
#endif\r
\r
/**\r
  The function to coalesce a fragmented capsule in memory.\r
\r
  @param PeiServices        General purpose services available to every PEIM.\r
  @param BlockListBuffer    Point to the buffer of Capsule Descriptor Variables.\r
  @param MemoryBase         Pointer to the base of a block of memory that we can walk\r
                            all over while trying to coalesce our buffers.\r
                            On output, this variable will hold the base address of\r
                            a coalesced capsule.\r
  @param MemorySize         Size of the memory region pointed to by MemoryBase.\r
                            On output, this variable will contain the size of the\r
                            coalesced capsule.\r
\r
  @retval EFI_NOT_FOUND     if we can't determine the boot mode\r
                            if the boot mode is not flash-update\r
                            if we could not find the capsule descriptors\r
\r
  @retval EFI_BUFFER_TOO_SMALL\r
                            if we could not coalesce the capsule in the memory\r
                            region provided to us\r
\r
  @retval EFI_SUCCESS       if there's no capsule, or if we processed the\r
                            capsule successfully.\r
**/\r
EFI_STATUS\r
EFIAPI\r
CapsuleDataCoalesce (\r
  IN EFI_PEI_SERVICES                **PeiServices,\r
  IN IN EFI_PHYSICAL_ADDRESS         *BlockListBuffer,\r
  IN OUT VOID                        **MemoryBase,\r
  IN OUT UINTN                       *MemorySize\r
  );\r
\r
#endif\r