MdeModulePkg/FaultTolerantWriteSmmDxe: switch to MM communicate 2

[mirror_edk2.git] / MdeModulePkg / Universal / FaultTolerantWriteDxe / FaultTolerantWriteSmmDxe.c

diff --git a/MdeModulePkg/Universal/FaultTolerantWriteDxe/FaultTolerantWriteSmmDxe.c b/MdeModulePkg/Universal/FaultTolerantWriteDxe/FaultTolerantWriteSmmDxe.c
index 6e3eb3b58b0901b50135e4c279f1e97236858aef..24c20950296dfdff2df5d961f3450155274f3cf3 100644 (file)
--- a/MdeModulePkg/Universal/FaultTolerantWriteDxe/FaultTolerantWriteSmmDxe.c
+++ b/MdeModulePkg/Universal/FaultTolerantWriteDxe/FaultTolerantWriteSmmDxe.c
@@ -11,7 +11,7 @@ SPDX-License-Identifier: BSD-2-Clause-Patent
 #include "FaultTolerantWriteSmmDxe.h"\r
 \r
 EFI_HANDLE                         mHandle                   = NULL;\r
-EFI_SMM_COMMUNICATION_PROTOCOL     *mSmmCommunication        = NULL;\r
+EFI_MM_COMMUNICATION2_PROTOCOL     *mMmCommunication2        = NULL;\r
 UINTN                              mPrivateDataSize          = 0;\r
 \r
 EFI_FAULT_TOLERANT_WRITE_PROTOCOL  mFaultTolerantWriteDriver = {\r
@@ -40,7 +40,7 @@ InitCommunicateBuffer (
   IN      UINTN                             Function\r
   )\r
 {\r
-  EFI_SMM_COMMUNICATE_HEADER                *SmmCommunicateHeader;\r
+  EFI_MM_COMMUNICATE_HEADER                 *SmmCommunicateHeader;\r
   SMM_FTW_COMMUNICATE_FUNCTION_HEADER       *SmmFtwFunctionHeader;\r
 \r
   //\r
@@ -74,7 +74,7 @@ InitCommunicateBuffer (
 **/\r
 EFI_STATUS\r
 SendCommunicateBuffer (\r
-  IN OUT  EFI_SMM_COMMUNICATE_HEADER        *SmmCommunicateHeader,\r
+  IN OUT  EFI_MM_COMMUNICATE_HEADER         *SmmCommunicateHeader,\r
   IN      UINTN                             DataSize\r
   )\r
 {\r
@@ -83,7 +83,10 @@ SendCommunicateBuffer (
   SMM_FTW_COMMUNICATE_FUNCTION_HEADER       *SmmFtwFunctionHeader;\r
 \r
   CommSize = DataSize + SMM_COMMUNICATE_HEADER_SIZE + SMM_FTW_COMMUNICATE_HEADER_SIZE;\r
-  Status = mSmmCommunication->Communicate (mSmmCommunication, SmmCommunicateHeader, &CommSize);\r
+  Status = mMmCommunication2->Communicate (mMmCommunication2,\r
+                                           SmmCommunicateHeader,\r
+                                           SmmCommunicateHeader,\r
+                                           &CommSize);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   SmmFtwFunctionHeader = (SMM_FTW_COMMUNICATE_FUNCTION_HEADER *) SmmCommunicateHeader->Data;\r
@@ -148,7 +151,7 @@ FtwGetMaxBlockSize (
 {\r
   EFI_STATUS                                Status;\r
   UINTN                                     PayloadSize;\r
-  EFI_SMM_COMMUNICATE_HEADER                *SmmCommunicateHeader;\r
+  EFI_MM_COMMUNICATE_HEADER                 *SmmCommunicateHeader;\r
   SMM_FTW_GET_MAX_BLOCK_SIZE_HEADER         *SmmFtwBlockSizeHeader;\r
 \r
   //\r
@@ -204,7 +207,7 @@ FtwAllocate (
 {\r
   EFI_STATUS                                Status;\r
   UINTN                                     PayloadSize;\r
-  EFI_SMM_COMMUNICATE_HEADER                *SmmCommunicateHeader;\r
+  EFI_MM_COMMUNICATE_HEADER                 *SmmCommunicateHeader;\r
   SMM_FTW_ALLOCATE_HEADER                   *SmmFtwAllocateHeader;\r
 \r
   //\r
@@ -270,7 +273,7 @@ FtwWrite (
 {\r
   EFI_STATUS                                Status;\r
   UINTN                                     PayloadSize;\r
-  EFI_SMM_COMMUNICATE_HEADER                *SmmCommunicateHeader;\r
+  EFI_MM_COMMUNICATE_HEADER                 *SmmCommunicateHeader;\r
   SMM_FTW_WRITE_HEADER                      *SmmFtwWriteHeader;\r
 \r
   //\r
@@ -336,7 +339,7 @@ FtwRestart (
 {\r
   EFI_STATUS                                Status;\r
   UINTN                                     PayloadSize;\r
-  EFI_SMM_COMMUNICATE_HEADER                *SmmCommunicateHeader;\r
+  EFI_MM_COMMUNICATE_HEADER                 *SmmCommunicateHeader;\r
   SMM_FTW_RESTART_HEADER                    *SmmFtwRestartHeader;\r
 \r
   //\r
@@ -381,7 +384,7 @@ FtwAbort (
   )\r
 {\r
   EFI_STATUS                                Status;\r
-  EFI_SMM_COMMUNICATE_HEADER                *SmmCommunicateHeader;\r
+  EFI_MM_COMMUNICATE_HEADER                 *SmmCommunicateHeader;\r
 \r
   //\r
   // Initialize the communicate buffer.\r
@@ -438,7 +441,7 @@ FtwGetLastWrite (
 {\r
   EFI_STATUS                                Status;\r
   UINTN                                     PayloadSize;\r
-  EFI_SMM_COMMUNICATE_HEADER                *SmmCommunicateHeader;\r
+  EFI_MM_COMMUNICATE_HEADER                 *SmmCommunicateHeader;\r
   SMM_FTW_GET_LAST_WRITE_HEADER             *SmmFtwGetLastWriteHeader;\r
 \r
   //\r
@@ -501,7 +504,7 @@ SmmFtwReady (
     return;\r
   }\r
 \r
-  Status = gBS->LocateProtocol (&gEfiSmmCommunicationProtocolGuid, NULL, (VOID **) &mSmmCommunication);\r
+  Status = gBS->LocateProtocol (&gEfiMmCommunication2ProtocolGuid, NULL, (VOID **) &mMmCommunication2);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   //\r