From d9a779013ce911160125d13e20a395495d30c723 Mon Sep 17 00:00:00 2001
From: Liu Zixing <liuzixing@hygon.cn>
Date: Fri, 25 Feb 2022 16:12:38 +0800
Subject: [PATCH 05/11] OvmfPkg/PlatformPei: Initialize CSV VM's memory

For CSV VM, the Secure Processor builds a temporary nested
page table to help the guest to run into the PEI phase.

In PEI phase, CSV VM detects the start address and size of the
guest physical memory.

The CSV VM sends the memory information to the Secure Processor
to build the permanent nested page table.

Signed-off-by: Xin Jiang <jiangxin@hygon.cn>
---
 OvmfPkg/Include/Library/PlatformInitLib.h   |  5 ++
 OvmfPkg/Library/PlatformInitLib/MemDetect.c |  2 +-
 OvmfPkg/PlatformPei/Csv.c                   | 82 +++++++++++++++++++++
 OvmfPkg/PlatformPei/Platform.c              |  2 +
 OvmfPkg/PlatformPei/Platform.h              | 10 +++
 OvmfPkg/PlatformPei/PlatformPei.inf         |  4 +
 6 files changed, 104 insertions(+), 1 deletion(-)
 create mode 100644 OvmfPkg/PlatformPei/Csv.c

diff --git a/OvmfPkg/Include/Library/PlatformInitLib.h b/OvmfPkg/Include/Library/PlatformInitLib.h
index 57b18b94..6c28c7fb 100644
--- a/OvmfPkg/Include/Library/PlatformInitLib.h
+++ b/OvmfPkg/Include/Library/PlatformInitLib.h
@@ -151,6 +151,11 @@ PlatformGetSystemMemorySizeBelow4gb (
   IN EFI_HOB_PLATFORM_INFO  *PlatformInfoHob
   );
 
+UINT64
+EFIAPI
+PlatformGetSystemMemorySizeAbove4gb (
+  );
+
 /**
   Initialize the PhysMemAddressWidth field in PlatformInfoHob based on guest RAM size.
 **/
diff --git a/OvmfPkg/Library/PlatformInitLib/MemDetect.c b/OvmfPkg/Library/PlatformInitLib/MemDetect.c
index 662e7e85..3c9f01cf 100644
--- a/OvmfPkg/Library/PlatformInitLib/MemDetect.c
+++ b/OvmfPkg/Library/PlatformInitLib/MemDetect.c
@@ -402,8 +402,8 @@ PlatformGetSystemMemorySizeBelow4gb (
   PlatformInfoHob->LowMemory = (UINT32)(((UINTN)((Cmos0x35 << 8) + Cmos0x34) << 16) + SIZE_16MB);
 }
 
-STATIC
 UINT64
+EFIAPI
 PlatformGetSystemMemorySizeAbove4gb (
   )
 {
diff --git a/OvmfPkg/PlatformPei/Csv.c b/OvmfPkg/PlatformPei/Csv.c
new file mode 100644
index 00000000..5ab83312
--- /dev/null
+++ b/OvmfPkg/PlatformPei/Csv.c
@@ -0,0 +1,82 @@
+/** @file
+
+  CSV initialization in PEI
+
+  Copyright (c) 2022, HYGON. 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
+  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 <Library/BaseLib.h>
+#include <Uefi/UefiBaseType.h>
+#include <Library/DebugLib.h>
+#include <Library/BaseMemoryLib.h>
+#include <Library/HobLib.h>
+#include <Library/CsvLib.h>
+#include <Library/MemEncryptSevLib.h>
+#include <Library/PlatformInitLib.h>
+
+#include "Platform.h"
+
+VOID
+CsvInitializeMemInfo (
+  IN EFI_HOB_PLATFORM_INFO  *PlatformInfoHob
+  )
+{
+  UINT64                      LowerMemorySize;
+  UINT64                      UpperMemorySize;
+
+  if (!CsvIsEnabled ()) {
+    return ;
+  }
+
+  LowerMemorySize = PlatformInfoHob->LowMemory;
+  UpperMemorySize = PlatformGetSystemMemorySizeAbove4gb ();
+
+  CsvUpdateMapLowerMemory (
+    0,
+    LowerMemorySize >> EFI_PAGE_SHIFT
+    );
+
+  if (UpperMemorySize > 0) {
+    CsvUpdateMapUpperMemory (
+      BASE_4GB,
+      UpperMemorySize >> EFI_PAGE_SHIFT
+      );
+  }
+
+  BuildMemoryAllocationHob (
+    (EFI_PHYSICAL_ADDRESS)(UINTN) FixedPcdGet32 (PcdCsvDefaultSecureCallBase),
+    (UINT64)(UINTN) FixedPcdGet32 (PcdCsvDefaultSecureCallSize),
+    EfiReservedMemoryType
+    );
+}
+
+VOID
+CsvInitializeGhcb (
+  VOID
+  )
+{
+  RETURN_STATUS        EncryptStatus;
+
+  if (!CsvIsEnabled ()) {
+    return ;
+  }
+
+  //
+  // Encrypt the SecGhcb as it's not a Ghcb any more
+  //
+  EncryptStatus = MemEncryptSevSetPageEncMask(
+                    0,
+                    PcdGet32 (PcdOvmfSecGhcbBase),
+                    1
+                    );
+  ASSERT_RETURN_ERROR (EncryptStatus);
+}
diff --git a/OvmfPkg/PlatformPei/Platform.c b/OvmfPkg/PlatformPei/Platform.c
index f5dc41c3..34d764e4 100644
--- a/OvmfPkg/PlatformPei/Platform.c
+++ b/OvmfPkg/PlatformPei/Platform.c
@@ -345,6 +345,7 @@ InitializePlatform (
   PlatformQemuUc32BaseInitialization (PlatformInfoHob);
 
   InitializeRamRegions (PlatformInfoHob);
+  CsvInitializeMemInfo (PlatformInfoHob);
 
   if (PlatformInfoHob->BootMode != BOOT_ON_S3_RESUME) {
     if (!PlatformInfoHob->SmmSmramRequire) {
@@ -364,6 +365,7 @@ InitializePlatform (
   } else {
     MiscInitialization (PlatformInfoHob);
   }
+  CsvInitializeGhcb();
 
   IntelTdxInitialize ();
   InstallFeatureControlCallback (PlatformInfoHob);
diff --git a/OvmfPkg/PlatformPei/Platform.h b/OvmfPkg/PlatformPei/Platform.h
index 1cf44844..1893f3fe 100644
--- a/OvmfPkg/PlatformPei/Platform.h
+++ b/OvmfPkg/PlatformPei/Platform.h
@@ -106,4 +106,14 @@ SevInitializeRam (
   VOID
   );
 
+VOID
+CsvInitializeMemInfo (
+  IN EFI_HOB_PLATFORM_INFO  *PlatformInfoHob
+  );
+
+VOID
+CsvInitializeGhcb (
+  VOID
+  );
+
 #endif // _PLATFORM_PEI_H_INCLUDED_
diff --git a/OvmfPkg/PlatformPei/PlatformPei.inf b/OvmfPkg/PlatformPei/PlatformPei.inf
index 3934aeed..45d16889 100644
--- a/OvmfPkg/PlatformPei/PlatformPei.inf
+++ b/OvmfPkg/PlatformPei/PlatformPei.inf
@@ -32,6 +32,7 @@
   Platform.c
   Platform.h
   IntelTdx.c
+  Csv.c
 
 [Packages]
   EmbeddedPkg/EmbeddedPkg.dec
@@ -65,6 +66,7 @@
   PcdLib
   CcExitLib
   PlatformInitLib
+  CsvLib
 
 [Pcd]
   gUefiOvmfPkgTokenSpaceGuid.PcdOvmfPeiMemFvBase
@@ -131,6 +133,8 @@
   gUefiOvmfPkgTokenSpaceGuid.PcdOvmfWorkAreaSize
   gUefiOvmfPkgTokenSpaceGuid.PcdOvmfSnpSecretsBase
   gUefiOvmfPkgTokenSpaceGuid.PcdOvmfSnpSecretsSize
+  gUefiOvmfPkgTokenSpaceGuid.PcdCsvDefaultSecureCallBase
+  gUefiOvmfPkgTokenSpaceGuid.PcdCsvDefaultSecureCallSize
 
 [FeaturePcd]
   gUefiOvmfPkgTokenSpaceGuid.PcdCsmEnable
-- 
2.43.0