1 files changed, 149 insertions, 0 deletions
diff --git a/Platform/Intel/MinPlatformPkg/Test/Library/TestPointCheckLib/DxeCheckSmmInfo.c b/Platform/Intel/MinPlatformPkg/Test/Library/TestPointCheckLib/DxeCheckSmmInfo.c
new file mode 100644
index 0000000000..59a59921e4
--- /dev/null
+++ b/Platform/Intel/MinPlatformPkg/Test/Library/TestPointCheckLib/DxeCheckSmmInfo.c
@@ -0,0 +1,149 @@
+/** @file
+
+Copyright (c) 2017, 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 that 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 <PiDxe.h>
+#include <Library/TestPointCheckLib.h>
+#include <Library/TestPointLib.h>
+#include <Library/DebugLib.h>
+#include <Library/UefiBootServicesTableLib.h>
+#include <Library/MemoryAllocationLib.h>
+#include <Protocol/SmmAccess2.h>
+
+CHAR8 *mSmramStateName[] = {
+  "Open",
+  "Closed",
+  "Locked",
+  "Cacheable",
+  "Allocated",
+  "NeedTest",
+  "NeedEccInit",
+};
+
+VOID
+DumpSmramDescriptor (
+  IN UINTN                 NumberOfSmmReservedRegions,
+  IN EFI_SMRAM_DESCRIPTOR  *Descriptor
+  )
+{
+  UINTN                                 Index;
+  UINTN                                 BitIndex;
+
+  for (Index = 0; Index < NumberOfSmmReservedRegions; Index++) {
+    DEBUG ((DEBUG_INFO,
+      "  BA=%016lx (A=%016lx) L=%016lx  State=%016lx",
+      Descriptor[Index].PhysicalStart,
+      Descriptor[Index].CpuStart,
+      Descriptor[Index].PhysicalSize,
+      Descriptor[Index].RegionState
+      ));
+    DEBUG ((DEBUG_INFO, "  ("));
+    for (BitIndex = 0; BitIndex < sizeof(mSmramStateName)/sizeof(mSmramStateName[0]); BitIndex++) {
+      if ((Descriptor[Index].RegionState & LShiftU64 (1, BitIndex)) != 0) {
+        DEBUG ((DEBUG_INFO, mSmramStateName[BitIndex]));
+        DEBUG ((DEBUG_INFO, ","));
+      }
+    }
+    DEBUG ((DEBUG_INFO, ")\n"));
+  }
+}
+
+EFI_STATUS
+CheckSmramDescriptor (
+  IN UINTN                 NumberOfSmmReservedRegions,
+  IN EFI_SMRAM_DESCRIPTOR  *Descriptor
+  )
+{
+  UINTN   Index;
+  UINT64  Base;
+  UINT64  Length;
+
+  Base = 0;
+  Length = 0;
+  for (Index = 0; Index < NumberOfSmmReservedRegions; Index++) {
+    if (Base == 0) {
+      Base   = Descriptor[Index].PhysicalStart;
+      Length = Descriptor[Index].PhysicalSize;
+    } else {
+      if (Base + Length == Descriptor[Index].PhysicalStart) {
+        Length = Length + Descriptor[Index].PhysicalSize;
+      } else if (Descriptor[Index].PhysicalStart + Descriptor[Index].PhysicalSize == Base) {
+        Base = Descriptor[Index].PhysicalStart;
+        Length = Length + Descriptor[Index].PhysicalSize;
+      } else {
+        DEBUG ((DEBUG_ERROR, "Smram is not adjacent\n"));
+        TestPointLibAppendErrorString (
+          PLATFORM_TEST_POINT_ROLE_PLATFORM_IBV,
+          NULL,
+          TEST_POINT_BYTE2_DXE_SMM_READY_TO_LOCK_SMRAM_ALIGNED_ERROR_CODE \
+            TEST_POINT_DXE_SMM_READY_TO_LOCK 
+            TEST_POINT_BYTE2_DXE_SMM_READY_TO_LOCK_SMRAM_ALIGNED_ERROR_STRING
+          );
+        return EFI_INVALID_PARAMETER;
+      }
+    }
+  }        
+
+  if (Length != GetPowerOfTwo64 (Length)) {
+    DEBUG ((DEBUG_ERROR, "Smram is not aligned\n"));
+    TestPointLibAppendErrorString (
+      PLATFORM_TEST_POINT_ROLE_PLATFORM_IBV,
+      NULL,
+      TEST_POINT_BYTE2_DXE_SMM_READY_TO_LOCK_SMRAM_ALIGNED_ERROR_CODE \
+        TEST_POINT_DXE_SMM_READY_TO_LOCK 
+        TEST_POINT_BYTE2_DXE_SMM_READY_TO_LOCK_SMRAM_ALIGNED_ERROR_STRING
+      );
+    return EFI_INVALID_PARAMETER;
+  }
+  return EFI_SUCCESS;
+}
+
+EFI_STATUS
+TestPointCheckSmmInfo (
+  VOID
+  )
+{
+  EFI_STATUS               Status;
+  EFI_SMM_ACCESS2_PROTOCOL *SmmAccess;
+  UINTN                    Size;
+  EFI_SMRAM_DESCRIPTOR     *SmramRanges;
+  
+  DEBUG ((DEBUG_INFO, "==== TestPointCheckSmmInfo - Enter\n"));
+  
+  Status = gBS->LocateProtocol (&gEfiSmmAccess2ProtocolGuid, NULL, (VOID **)&SmmAccess);
+  if (EFI_ERROR (Status)) {
+    goto Done ;
+  }
+  
+  Size = 0;
+  Status = SmmAccess->GetCapabilities (SmmAccess, &Size, NULL);
+  ASSERT (Status == EFI_BUFFER_TOO_SMALL);
+
+  SmramRanges = (EFI_SMRAM_DESCRIPTOR *) AllocateZeroPool (Size);
+  ASSERT (SmramRanges != NULL);
+
+  Status = SmmAccess->GetCapabilities (SmmAccess, &Size, SmramRanges);
+  ASSERT_EFI_ERROR (Status);
+  
+  DEBUG ((DEBUG_INFO, "SMRAM Info\n"));
+  DumpSmramDescriptor (Size / sizeof (EFI_SMRAM_DESCRIPTOR), SmramRanges);
+
+  Status = CheckSmramDescriptor (Size / sizeof (EFI_SMRAM_DESCRIPTOR), SmramRanges);
+
+  FreePool (SmramRanges);
+
+Done:
+  DEBUG ((DEBUG_INFO, "==== TestPointCheckSmmInfo - Exit\n"));
+
+  return Status;
+}
+\ No newline at end of file