1 files changed, 68 insertions, 23 deletions
diff --git a/SecurityPkg/Tcg/MemoryOverwriteControl/TcgMor.c b/SecurityPkg/Tcg/MemoryOverwriteControl/TcgMor.c
index ce53112dc5..def965f138 100644
--- a/SecurityPkg/Tcg/MemoryOverwriteControl/TcgMor.c
+++ b/SecurityPkg/Tcg/MemoryOverwriteControl/TcgMor.c
@@ -4,7 +4,7 @@
   This driver initilize MemoryOverwriteRequestControl variable. It 
   will clear MOR_CLEAR_MEMORY_BIT bit if it is set.
 
-Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>
+Copyright (c) 2009 - 2012, 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 
@@ -17,6 +17,50 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 
 #include "TcgMor.h"
 
+UINT8    mMorControl;
+
+/**
+  Ready to Boot Event notification handler.
+
+  Sequence of OS boot events is measured in this event notification handler.
+
+  @param[in]  Event     Event whose notification function is being invoked
+  @param[in]  Context   Pointer to the notification function's context
+
+**/
+VOID
+EFIAPI
+OnReadyToBoot (
+  IN      EFI_EVENT                 Event,
+  IN      VOID                      *Context
+  )
+{
+  UINTN       DataSize;
+  
+  if (MOR_CLEAR_MEMORY_VALUE (mMorControl) == 0x0) {
+    //
+    // MorControl is expected, directly return to avoid unnecessary variable operation
+    //
+    return ;
+  }
+  //
+  // Clear MOR_CLEAR_MEMORY_BIT
+  //
+  DEBUG ((EFI_D_INFO, "TcgMor: Clear MorClearMemory bit\n"));
+  mMorControl &= 0xFE;  
+
+  DataSize = sizeof (mMorControl);
+  gRT->SetVariable (
+         MEMORY_OVERWRITE_REQUEST_VARIABLE_NAME, 
+         &gEfiMemoryOverwriteControlDataGuid, 
+         EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS,
+         DataSize, 
+         &mMorControl
+         );
+
+}
+
+
 /**
   Entry Point for TCG MOR Control driver.
 
@@ -34,48 +78,49 @@ MorDriverEntryPoint (
   )
 {
   EFI_STATUS  Status;
-  UINT8       MorControl;
   UINTN       DataSize;
+  EFI_EVENT   Event;
 
   ///
   /// The firmware is required to create the MemoryOverwriteRequestControl UEFI variable.
   ///
 
-  DataSize = sizeof (MorControl);
+  DataSize = sizeof (mMorControl);
   Status = gRT->GetVariable (
                   MEMORY_OVERWRITE_REQUEST_VARIABLE_NAME, 
                   &gEfiMemoryOverwriteControlDataGuid, 
                   NULL, 
                   &DataSize, 
-                  &MorControl
+                  &mMorControl
                   );
   if (EFI_ERROR (Status)) {
     //
     // Set default value to 0
     //
-    MorControl = 0;
+    mMorControl = 0;
+    DEBUG ((EFI_D_INFO, "TcgMor: Create gEfiMemoryOverwriteControlDataGuid!\n"));
+    Status = gRT->SetVariable (
+                    MEMORY_OVERWRITE_REQUEST_VARIABLE_NAME, 
+                    &gEfiMemoryOverwriteControlDataGuid, 
+                    EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS,
+                    DataSize, 
+                    &mMorControl
+                    );
+    ASSERT_EFI_ERROR (Status);
+    
   } else {
-    if (MOR_CLEAR_MEMORY_VALUE (MorControl) == 0x0) {
-      //
-      // MorControl is expected, directly return to avoid unnecessary variable operation
-      //
-      return EFI_SUCCESS;
-    }
     //
-    // Clear MOR_CLEAR_MEMORY_BIT
+    // Create a Ready To Boot Event and Clear the MorControl bit in the call back function.
     //
-    DEBUG ((EFI_D_INFO, "TcgMor: Clear MorClearMemory bit\n"));
-    MorControl &= 0xFE;
-  }
+    DEBUG ((EFI_D_INFO, "TcgMor: Create ReadyToBoot Event for MorControl Bit cleanning!\n"));
+    Status = EfiCreateEventReadyToBootEx (
+               TPL_CALLBACK,
+               OnReadyToBoot,
+               NULL,
+               &Event
+               );
+  }    
   
-  Status = gRT->SetVariable (
-                  MEMORY_OVERWRITE_REQUEST_VARIABLE_NAME, 
-                  &gEfiMemoryOverwriteControlDataGuid, 
-                  EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS,
-                  DataSize, 
-                  &MorControl
-                  );
-  ASSERT_EFI_ERROR (Status);
   return Status;
 }