1 files changed, 82 insertions, 0 deletions
diff --git a/EmbeddedPkg/EmbeddedMonotonicCounter/EmbeddedMonotonicCounter.c b/EmbeddedPkg/EmbeddedMonotonicCounter/EmbeddedMonotonicCounter.c
new file mode 100644
index 0000000000..66ebe67a92
--- /dev/null
+++ b/EmbeddedPkg/EmbeddedMonotonicCounter/EmbeddedMonotonicCounter.c
@@ -0,0 +1,82 @@
+/** @file
+
+  Copyright (c) 2008-2009, Apple Inc. All rights reserved.
+  
+  All rights reserved. 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.
+
+**/
+
+#include <Uefi.h>
+
+#include <Library/BaseLib.h>
+#include <Library/DebugLib.h>
+#include <Library/UefiBootServicesTableLib.h>
+#include <Library/UefiRuntimeServicesTableLib.h>
+
+#include <Protocol/MonotonicCounter.h>
+
+UINT64 gCurrentMonotonicCount = 0;
+
+EFI_STATUS
+EFIAPI
+GetNextMonotonicCount (
+  OUT UINT64  *Count
+  )
+{
+  if (Count == NULL) {
+    return EFI_INVALID_PARAMETER;
+  }
+  
+  *Count = gCurrentMonotonicCount++;
+  return EFI_SUCCESS;
+}
+
+EFI_STATUS
+EFIAPI
+GetNextHighMonotonicCount (
+  OUT UINT32  *HighCount
+  )
+{
+  if (HighCount == NULL) {
+    return EFI_INVALID_PARAMETER;
+  }
+  
+  gCurrentMonotonicCount += 0x0000000100000000ULL;
+  
+  *HighCount = RShiftU64 (gCurrentMonotonicCount, 32) & 0xFFFFFFFF;
+
+  return EFI_SUCCESS;
+}
+
+
+EFI_STATUS
+EFIAPI
+MonotonicCounterDriverInitialize (
+  IN EFI_HANDLE        ImageHandle,
+  IN EFI_SYSTEM_TABLE  *SystemTable
+  )
+{
+  EFI_STATUS  Status;
+  EFI_HANDLE  Handle = NULL;
+
+  // Make sure the Monotonic Counter Architectural Protocol is not already installed in the system
+  ASSERT_PROTOCOL_ALREADY_INSTALLED(NULL, &gEfiMonotonicCounterArchProtocolGuid);
+
+  // Fill in the EFI Boot Services and EFI Runtime Services Monotonic Counter Fields
+  gBS->GetNextMonotonicCount     = GetNextMonotonicCount;
+  gRT->GetNextHighMonotonicCount = GetNextHighMonotonicCount;
+
+  // Install the Monotonic Counter Architctural Protocol onto a new handle
+  Status = gBS->InstallMultipleProtocolInterfaces (
+                  &Handle,
+                  &gEfiMonotonicCounterArchProtocolGuid,    NULL,
+                  NULL
+                  );
+  return Status;
+}