UefiCpuPkg/MpInitLib: Move two functions location

Just move BackupAndPrepareWakeupBuffer() and RestoreWakeupBuffer() from
PeiMpLib.c to MpLib.c.

Cc: Michael Kinney <michael.d.kinney@intel.com>
Cc: Feng Tian <feng.tian@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Jeff Fan <jeff.fan@intel.com>
Reviewed-by: Feng Tian <feng.tian@intel.com>
diff --git a/UefiCpuPkg/Library/MpInitLib/MpLib.c b/UefiCpuPkg/Library/MpInitLib/MpLib.c
index f0e9e46..c3fe721 100644
--- a/UefiCpuPkg/Library/MpInitLib/MpLib.c
+++ b/UefiCpuPkg/Library/MpInitLib/MpLib.c
@@ -2012,3 +2012,42 @@
   }

   return CpuMpData;

 }

+

+/**

+  Get available system memory below 1MB by specified size.

+

+  @param[in]  CpuMpData  The pointer to CPU MP Data structure.

+**/

+VOID

+BackupAndPrepareWakeupBuffer(

+  IN CPU_MP_DATA              *CpuMpData

+  )

+{

+  CopyMem (

+    (VOID *) CpuMpData->BackupBuffer,

+    (VOID *) CpuMpData->WakeupBuffer,

+    CpuMpData->BackupBufferSize

+    );

+  CopyMem (

+    (VOID *) CpuMpData->WakeupBuffer,

+    (VOID *) CpuMpData->AddressMap.RendezvousFunnelAddress,

+    CpuMpData->AddressMap.RendezvousFunnelSize

+    );

+}

+

+/**

+  Restore wakeup buffer data.

+

+  @param[in]  CpuMpData  The pointer to CPU MP Data structure.

+**/

+VOID

+RestoreWakeupBuffer(

+  IN CPU_MP_DATA              *CpuMpData

+  )

+{

+  CopyMem (

+    (VOID *) CpuMpData->WakeupBuffer,

+    (VOID *) CpuMpData->BackupBuffer,

+    CpuMpData->BackupBufferSize

+    );

+}

diff --git a/UefiCpuPkg/Library/MpInitLib/MpLib.h b/UefiCpuPkg/Library/MpInitLib/MpLib.h
index 165853d..ae78c59 100644
--- a/UefiCpuPkg/Library/MpInitLib/MpLib.h
+++ b/UefiCpuPkg/Library/MpInitLib/MpLib.h
@@ -550,5 +550,25 @@
   IN VOID                         *Ppi

   );

 

+/**

+  Get available system memory below 1MB by specified size.

+

+  @param[in]  CpuMpData  The pointer to CPU MP Data structure.

+**/

+VOID

+BackupAndPrepareWakeupBuffer(

+  IN CPU_MP_DATA              *CpuMpData

+  );

+

+/**

+  Restore wakeup buffer data.

+

+  @param[in]  CpuMpData  The pointer to CPU MP Data structure.

+**/

+VOID

+RestoreWakeupBuffer(

+  IN CPU_MP_DATA              *CpuMpData

+  );

+

 #endif

 

diff --git a/UefiCpuPkg/Library/MpInitLib/PeiMpLib.c b/UefiCpuPkg/Library/MpInitLib/PeiMpLib.c
index 590f963..e242d37 100644
--- a/UefiCpuPkg/Library/MpInitLib/PeiMpLib.c
+++ b/UefiCpuPkg/Library/MpInitLib/PeiMpLib.c
@@ -65,45 +65,6 @@
 }

 

 /**

-  Get available system memory below 1MB by specified size.

-

-  @param[in]  CpuMpData  The pointer to CPU MP Data structure.

-**/

-VOID

-BackupAndPrepareWakeupBuffer(

-  IN CPU_MP_DATA              *CpuMpData

-  )

-{

-  CopyMem (

-    (VOID *) CpuMpData->BackupBuffer,

-    (VOID *) CpuMpData->WakeupBuffer,

-    CpuMpData->BackupBufferSize

-    );

-  CopyMem (

-    (VOID *) CpuMpData->WakeupBuffer,

-    (VOID *) CpuMpData->AddressMap.RendezvousFunnelAddress,

-    CpuMpData->AddressMap.RendezvousFunnelSize

-    );

-}

-

-/**

-  Restore wakeup buffer data.

-

-  @param[in]  CpuMpData  The pointer to CPU MP Data structure.

-**/

-VOID

-RestoreWakeupBuffer(

-  IN CPU_MP_DATA              *CpuMpData

-  )

-{

-  CopyMem (

-    (VOID *) CpuMpData->WakeupBuffer,

-    (VOID *) CpuMpData->BackupBuffer,

-    CpuMpData->BackupBufferSize

-    );

-}

-

-/**

   Notify function on End Of PEI PPI.

 

   On S3 boot, this function will restore wakeup buffer data.