1 files changed, 84 insertions, 0 deletions
diff --git a/MdeModulePkg/Universal/CapsulePei/Common/CommonHeader.h b/MdeModulePkg/Universal/CapsulePei/Common/CommonHeader.h
new file mode 100644
index 0000000000..5c0aa3068c
--- /dev/null
+++ b/MdeModulePkg/Universal/CapsulePei/Common/CommonHeader.h
@@ -0,0 +1,84 @@
+/** @file
+  Common header file.
+
+Copyright (c) 2011, Intel Corporation. All rights reserved.<BR>
+This program and the accompanying materials
+are licensed and made available under the terms and conditions of the BSD License
+which accompanies this distribution.  The full text of the license may be found at
+http://opensource.org/licenses/bsd-license.php
+
+THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
+WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
+
+**/
+
+#ifndef _CAPSULE_THUNK_32_TO_64_
+#define _CAPSULE_THUNK_32_TO_64_
+
+#include <Uefi.h>
+#include "PiPei.h"
+
+//
+// This capsule PEIM puts its private data at the start of the
+// coalesced capsule. Here's the structure definition.
+//
+#define EFI_CAPSULE_PEIM_PRIVATE_DATA_SIGNATURE SIGNATURE_32 ('C', 'a', 'p', 'D')
+
+typedef struct {
+  UINT32  Signature;
+  UINTN   CapsuleSize;
+} EFI_CAPSULE_PEIM_PRIVATE_DATA;
+
+#define CAPSULE_TEST_SIGNATURE SIGNATURE_32('T', 'E', 'S', 'T')
+
+typedef struct {
+  EFI_PHYSICAL_ADDRESS  EntryPoint;
+  EFI_PHYSICAL_ADDRESS  StackBufferBase;
+  UINT64                StackBufferLength;
+  EFI_PHYSICAL_ADDRESS  JumpBuffer;
+  EFI_PHYSICAL_ADDRESS  BlockListAddr;
+  EFI_PHYSICAL_ADDRESS  MemoryBase64Ptr;
+  EFI_PHYSICAL_ADDRESS  MemorySize64Ptr;
+} SWITCH_32_TO_64_CONTEXT;
+
+typedef struct {
+  UINT16                ReturnCs;
+  EFI_PHYSICAL_ADDRESS  ReturnEntryPoint;
+  UINT64                ReturnStatus;
+  IA32_DESCRIPTOR       Gdtr;
+} SWITCH_64_TO_32_CONTEXT;
+
+/**
+  The function to coalesce a fragmented capsule in memory.
+
+  @param PeiServices        General purpose services available to every PEIM.
+  @param BlockListBuffer    Point to the buffer of Capsule Descriptor Variables.
+  @param MemoryBase         Pointer to the base of a block of memory that we can walk
+                            all over while trying to coalesce our buffers.
+                            On output, this variable will hold the base address of
+                            a coalesced capsule.
+  @param MemorySize         Size of the memory region pointed to by MemoryBase.
+                            On output, this variable will contain the size of the
+                            coalesced capsule.
+
+  @retval EFI_NOT_FOUND     if we can't determine the boot mode
+                            if the boot mode is not flash-update
+                            if we could not find the capsule descriptors
+
+  @retval EFI_BUFFER_TOO_SMALL
+                            if we could not coalesce the capsule in the memory
+                            region provided to us
+
+  @retval EFI_SUCCESS       if there's no capsule, or if we processed the
+                            capsule successfully.
+**/
+EFI_STATUS
+EFIAPI
+CapsuleDataCoalesce (
+  IN EFI_PEI_SERVICES                **PeiServices,
+  IN IN EFI_PHYSICAL_ADDRESS         *BlockListBuffer,
+  IN OUT VOID                        **MemoryBase,
+  IN OUT UINTN                       *MemorySize
+  );
+
+#endif