From e255b5421b5c7a4ebd111491b3424a41ef7597d2 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/20] 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 57b18b9..6c28c7f 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 f042517..01e01a7 100644
--- a/OvmfPkg/Library/PlatformInitLib/MemDetect.c
+++ b/OvmfPkg/Library/PlatformInitLib/MemDetect.c
@@ -437,8 +437,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 0000000..5ab8331
--- /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 ce9868d..16f17a6 100644
--- a/OvmfPkg/PlatformPei/Platform.c
+++ b/OvmfPkg/PlatformPei/Platform.c
@@ -357,6 +357,7 @@ InitializePlatform (
   PlatformQemuUc32BaseInitialization (PlatformInfoHob);
 
   InitializeRamRegions (PlatformInfoHob);
+  CsvInitializeMemInfo (PlatformInfoHob);
 
   if (PlatformInfoHob->BootMode != BOOT_ON_S3_RESUME) {
     if (!PlatformInfoHob->SmmSmramRequire) {
@@ -377,6 +378,7 @@ InitializePlatform (
   } else {
     MiscInitialization (PlatformInfoHob);
   }
+  CsvInitializeGhcb();
 
   IntelTdxInitialize ();
   InstallFeatureControlCallback (PlatformInfoHob);
diff --git a/OvmfPkg/PlatformPei/Platform.h b/OvmfPkg/PlatformPei/Platform.h
index 1cf4484..1893f3f 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 381a85c..f8f57c4 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
@@ -132,6 +134,8 @@
   gUefiOvmfPkgTokenSpaceGuid.PcdOvmfWorkAreaSize
   gUefiOvmfPkgTokenSpaceGuid.PcdOvmfSnpSecretsBase
   gUefiOvmfPkgTokenSpaceGuid.PcdOvmfSnpSecretsSize
+  gUefiOvmfPkgTokenSpaceGuid.PcdCsvDefaultSecureCallBase
+  gUefiOvmfPkgTokenSpaceGuid.PcdCsvDefaultSecureCallSize
 
 [FeaturePcd]
   gUefiOvmfPkgTokenSpaceGuid.PcdSmmSmramRequire
-- 
2.41.1