public inbox for devel@edk2.groups.io
 help / color / mirror / Atom feed
From: "Sunny Wang" <Sunny.Wang@arm.com>
To: Dimitrije Pavlov <Dimitrije.Pavlov@arm.com>,
	"devel@edk2.groups.io" <devel@edk2.groups.io>
Cc: G Edhaya Chandran <Edhaya.Chandran@arm.com>,
	Jeff Booher-Kaeding <Jeff.Booher-Kaeding@arm.com>,
	Samer El-Haj-Mahmoud <Samer.El-Haj-Mahmoud@arm.com>,
	Sunny Wang <Sunny.Wang@arm.com>,
	Stuart Yoder <Stuart.Yoder@arm.com>
Subject: Re: [edk2-test][PATCH v1 1/1] uefi-sct/SctPkg: Check for memory below 4G
Date: Thu, 1 Dec 2022 17:11:35 +0000	[thread overview]
Message-ID: <AS8PR08MB77914F9E333AD147687A8FD585149@AS8PR08MB7791.eurprd08.prod.outlook.com> (raw)
In-Reply-To: <20221010020530.1261275-1-dimitrije.pavlov@arm.com>

Reviewed the patch with Stuart and Edhaya in edk2-test monthly meeting. The patch looks good to me. Thanks, Dimitrije.

Reviewed-by: Sunny Wang <sunny.wang@arm.com>

-----Original Message-----
From: Dimitrije Pavlov <dimitrije.pavlov@arm.com>
Sent: 10 October 2022 03:06
To: devel@edk2.groups.io
Cc: G Edhaya Chandran <Edhaya.Chandran@arm.com>; Jeff Booher-Kaeding <Jeff.Booher-Kaeding@arm.com>; Samer El-Haj-Mahmoud <Samer.El-Haj-Mahmoud@arm.com>; Sunny Wang <Sunny.Wang@arm.com>
Subject: [edk2-test][PATCH v1 1/1] uefi-sct/SctPkg: Check for memory below 4G

REF: https://bugzilla.tianocore.org/show_bug.cgi?id=4009

Check if there is usable memory below 4 GiB before testing for
allocation without the EFI_PCI_ATTRIBUTE_DUAL_ADDRESS_CYCLE attribute.

Cc: G Edhaya Chandran <Edhaya.Chandran@arm.com>
Cc: Jeff Booher-Kaeding <Jeff.Booher-Kaeding@arm.com>
Cc: Samer El-Haj-Mahmoud <Samer.El-Haj-Mahmoud@arm.com>
Cc: Sunny Wang <Sunny.Wang@arm.com>

Signed-off-by: Dimitrije Pavlov <Dimitrije.Pavlov@arm.com>
---
 uefi-sct/SctPkg/TestCase/UEFI/EFI/Protocol/PciRootBridgeIo/BlackBoxTest/PciRootBridgeIoBBTestFunction_2.c | 142 +++++++++++++++++++-
 1 file changed, 140 insertions(+), 2 deletions(-)

diff --git a/uefi-sct/SctPkg/TestCase/UEFI/EFI/Protocol/PciRootBridgeIo/BlackBoxTest/PciRootBridgeIoBBTestFunction_2.c b/uefi-sct/SctPkg/TestCase/UEFI/EFI/Protocol/PciRootBridgeIo/BlackBoxTest/PciRootBridgeIoBBTestFunction_2.c
index 89adcba91e70..fafbf62f77c6 100644
--- a/uefi-sct/SctPkg/TestCase/UEFI/EFI/Protocol/PciRootBridgeIo/BlackBoxTest/PciRootBridgeIoBBTestFunction_2.c
+++ b/uefi-sct/SctPkg/TestCase/UEFI/EFI/Protocol/PciRootBridgeIo/BlackBoxTest/PciRootBridgeIoBBTestFunction_2.c
@@ -4118,7 +4118,69 @@ AllocateBuffer_Func (
   UINTN                                        AttributesNum;
   EFI_PCI_ROOT_BRIDGE_IO_DEVICE                *RBDev;
   EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL              *RootBridgeIo;
+  EFI_MEMORY_DESCRIPTOR                        *MemoryMap;
+  EFI_MEMORY_DESCRIPTOR                        *Entry;
+  UINTN                                        MemoryMapSize;
+  UINTN                                        MapKey;
+  UINTN                                        DescriptorSize;
+  UINT32                                       DescriptorVersion;
+  UINTN                                        Iterator;
+  BOOLEAN                                      UsableMemoryBelow4G;

+  //
+  // Obtain the memory map size
+  //
+  MemoryMapSize = 0;
+  Status = gBS->GetMemoryMap (
+                  &MemoryMapSize,
+                  NULL,
+                  &MapKey,
+                  &DescriptorSize,
+                  &DescriptorVersion
+                  );
+  ASSERT (Status == EFI_BUFFER_TOO_SMALL);
+  //
+  // Allocating a buffer for the memory map will change
+  // the memory map, so we increase the size here just in case
+  //
+  MemoryMapSize += EFI_PAGE_SIZE;
+  Status = gBS->AllocatePool (
+                  EfiLoaderData,
+                  MemoryMapSize,
+                  (VOID **)&MemoryMap
+                  );
+  if (EFI_ERROR (Status)) {
+    return Status;
+  }
+  //
+  // Get the actual memory map
+  //
+  Status = gBS->GetMemoryMap (
+                  &MemoryMapSize,
+                  MemoryMap,
+                  &MapKey,
+                  &DescriptorSize,
+                  &DescriptorVersion
+                  );
+  if (EFI_ERROR (Status)) {
+    gBS->FreePool (MemoryMap);
+  }
+
+  //
+  // Check each entry in the memory map for free memory below 4 GiB and set
+  // UsableMemoryBelow4G accordingly
+  //
+  UsableMemoryBelow4G = FALSE;
+  for (Iterator = 0; Iterator < MemoryMapSize; Iterator += DescriptorSize) {
+    Entry = (EFI_MEMORY_DESCRIPTOR *)((UINTN)MemoryMap + Iterator);
+    if ( Entry->PhysicalStart < (EFI_PHYSICAL_ADDRESS)SIZE_4GB
+      && ( Entry->Type == EfiConventionalMemory || Entry->Type == EfiPersistentMemory))
+    {
+      UsableMemoryBelow4G = TRUE;
+      break;
+    }
+  }
+  gBS->FreePool (MemoryMap);

   AllocateType = 0;

@@ -4188,7 +4250,14 @@ AllocateBuffer_Func (

   for (MemoryTypeNum = 0; MemoryTypeNum < 2; MemoryTypeNum++) {
     for (AttributesNum = 0; AttributesNum < 8; AttributesNum++) {
-
+      //
+      // If there is no usable memory below 4 GiB, skip 32-bit allocations
+      //
+      if ( !UsableMemoryBelow4G
+        && !(Attributes[AttributesNum] & EFI_PCI_ATTRIBUTE_DUAL_ADDRESS_CYCLE))
+      {
+        continue;
+      }
       Status = RootBridgeIo->AllocateBuffer (
                                RootBridgeIo,
                                AllocateType,
@@ -4298,7 +4367,69 @@ FreeBuffer_Func (
   UINTN                                        AttributesNum;
   EFI_PCI_ROOT_BRIDGE_IO_DEVICE                *RBDev;
   EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL              *RootBridgeIo;
+  EFI_MEMORY_DESCRIPTOR                        *MemoryMap;
+  EFI_MEMORY_DESCRIPTOR                        *Entry;
+  UINTN                                        MemoryMapSize;
+  UINTN                                        MapKey;
+  UINTN                                        DescriptorSize;
+  UINT32                                       DescriptorVersion;
+  UINTN                                        Iterator;
+  BOOLEAN                                      UsableMemoryBelow4G;

+  //
+  // Obtain the memory map size
+  //
+  MemoryMapSize = 0;
+  Status = gBS->GetMemoryMap (
+                  &MemoryMapSize,
+                  NULL,
+                  &MapKey,
+                  &DescriptorSize,
+                  &DescriptorVersion
+                  );
+  ASSERT (Status == EFI_BUFFER_TOO_SMALL);
+  //
+  // Allocating a buffer for the memory map will change
+  // the memory map, so we increase the size here just in case
+  //
+  MemoryMapSize += EFI_PAGE_SIZE;
+  Status = gBS->AllocatePool (
+                  EfiLoaderData,
+                  MemoryMapSize,
+                  (VOID **)&MemoryMap
+                  );
+  if (EFI_ERROR (Status)) {
+    return Status;
+  }
+  //
+  // Get the actual memory map
+  //
+  Status = gBS->GetMemoryMap (
+                  &MemoryMapSize,
+                  MemoryMap,
+                  &MapKey,
+                  &DescriptorSize,
+                  &DescriptorVersion
+                  );
+  if (EFI_ERROR (Status)) {
+    gBS->FreePool (MemoryMap);
+  }
+
+  //
+  // Check each entry in the memory map for free memory below 4 GiB and set
+  // UsableMemoryBelow4G accordingly
+  //
+  UsableMemoryBelow4G = FALSE;
+  for (Iterator = 0; Iterator < MemoryMapSize; Iterator += DescriptorSize) {
+    Entry = (EFI_MEMORY_DESCRIPTOR *)((UINTN)MemoryMap + Iterator);
+    if ( Entry->PhysicalStart < (EFI_PHYSICAL_ADDRESS)SIZE_4GB
+      && ( Entry->Type == EfiConventionalMemory || Entry->Type == EfiPersistentMemory))
+    {
+      UsableMemoryBelow4G = TRUE;
+      break;
+    }
+  }
+  gBS->FreePool (MemoryMap);

   AllocateType = 0 ;

@@ -4368,7 +4499,14 @@ FreeBuffer_Func (

   for (MemoryTypeNum = 0; MemoryTypeNum < 2; MemoryTypeNum++) {
     for (AttributesNum = 0; AttributesNum < 8; AttributesNum++) {
-
+      //
+      // If there is no usable memory below 4 GiB, skip 32-bit allocations
+      //
+      if ( !UsableMemoryBelow4G
+        && !(Attributes[AttributesNum] & EFI_PCI_ATTRIBUTE_DUAL_ADDRESS_CYCLE))
+      {
+        continue;
+      }
       Status = RootBridgeIo->AllocateBuffer (
                                RootBridgeIo,
                                AllocateType,
--
2.37.3

IMPORTANT NOTICE: The contents of this email and any attachments are confidential and may also be privileged. If you are not the intended recipient, please notify the sender immediately and do not disclose the contents to any other person, use it for any purpose, or store or copy the information in any medium. Thank you.

  parent reply	other threads:[~2022-12-01 17:11 UTC|newest]

Thread overview: 5+ messages / expand[flat|nested]  mbox.gz  Atom feed  top
2022-10-10  2:05 [edk2-test][PATCH v1 1/1] uefi-sct/SctPkg: Check for memory below 4G Dimitrije Pavlov
2022-12-01 15:19 ` [edk2-devel] " G Edhaya Chandran
2022-12-01 17:11 ` Sunny Wang [this message]
2023-07-20 14:29   ` G Edhaya Chandran
2023-07-20 15:40     ` Ard Biesheuvel

Reply instructions:

You may reply publicly to this message via plain-text email
using any one of the following methods:

* Save the following mbox file, import it into your mail client,
  and reply-to-list from there: mbox

  Avoid top-posting and favor interleaved quoting:
  https://en.wikipedia.org/wiki/Posting_style#Interleaved_style

* Reply using the --to, --cc, and --in-reply-to
  switches of git-send-email(1):

  git send-email \
    --in-reply-to=AS8PR08MB77914F9E333AD147687A8FD585149@AS8PR08MB7791.eurprd08.prod.outlook.com \
    --to=devel@edk2.groups.io \
    /path/to/YOUR_REPLY

  https://kernel.org/pub/software/scm/git/docs/git-send-email.html

* If your mail client supports setting the In-Reply-To header
  via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox