AmpereAltraPkg: Fix OS kernel crash if not call SetVirtualAddressMap()

OS kernel avoided SetVirtualAddressMap() on arm64 due to commit d3549a93
("efi/arm64: libstub: avoid SetVirtualAddressMap() when possible").
Currently, RuntimeFlashLib in AmpereAltraPkg relies on function call
SetVirtualAddressMap() to determine whether it is in boot time or
run-time. At boot time, the code optimizes to avoid the use of temporary
buffer because the virtual address and physical address are identical.
If SetVirtualAddressMap() isn't called, it does not know when to switch.
As such, don't do that optimization and the physical address is always
known.

Signed-off-by: Chuong Tran <[email protected]>

Silicon/Ampere/AmpereAltraPkg/Library/FlashLib/FlashLibCommon.c

@@ -17,7 +17,6 @@
 
 #define UEFI_MISC_SIZE        0x1000
 
-BOOLEAN                       gFlashLibRuntime = FALSE;
 UINT8                         *gFlashLibPhysicalBuffer;
 UINT8                         *gFlashLibVirtualBuffer;
 
@@ -36,13 +35,9 @@ ConvertToPhysicalBuffer (
   IN UINT32 Size
   )
 {
-  if (gFlashLibRuntime) {
-    ASSERT (VirtualPtr != NULL);
-    CopyMem (gFlashLibVirtualBuffer, VirtualPtr, Size);
-    return gFlashLibPhysicalBuffer;
-  }
-
-  return VirtualPtr;
+  ASSERT (VirtualPtr != NULL);
+  CopyMem (gFlashLibVirtualBuffer, VirtualPtr, Size);
+  return gFlashLibPhysicalBuffer;
 }
 
 /**

Silicon/Ampere/AmpereAltraPkg/Library/FlashLib/RuntimeFlashLib.c

@@ -35,8 +35,6 @@ FlashLibAddressChangeEvent (
 {
   gRT->ConvertPointer (0x0, (VOID **)&gFlashLibVirtualBuffer);
   gRT->ConvertPointer (0x0, (VOID **)&mMmCommunicationProtocol);
-
-  gFlashLibRuntime = TRUE;
 }
 
 /**