Add NULL check for GetPciSegmentInfo()

Suggested-by: Michael A Kubacki <michael.a.kubacki@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Jiewen Yao <jiewen.yao@intel.com>

1 files changed, 7 insertions, 0 deletions

diff --git a/Platform/Intel/MinPlatformPkg/Pci/Library/PciSegmentInfoLibSimple/PciSegmentInfoLibSimple.c b/Platform/Intel/MinPlatformPkg/Pci/Library/PciSegmentInfoLibSimple/PciSegmentInfoLibSimple.c
index a7e09f16b8..1facc63d01 100644
--- a/Platform/Intel/MinPlatformPkg/Pci/Library/PciSegmentInfoLibSimple/PciSegmentInfoLibSimple.c
+++ b/Platform/Intel/MinPlatformPkg/Pci/Library/PciSegmentInfoLibSimple/PciSegmentInfoLibSimple.c
@@ -44,6 +44,10 @@ GetPciSegmentInfo (
   VOID              *Hob;

   PCI_SEGMENT_INFO  *PciSegmentInfo;

 

+  if (Count == NULL) {

+    return NULL;

+  }

+

   *Count = 1;

   Hob = GetFirstGuidHob (&mPciSegmentInfoHobGuid);

   if (Hob != NULL) {

@@ -51,6 +55,9 @@ GetPciSegmentInfo (
   } else {

     PciSegmentInfo = BuildGuidHob (&mPciSegmentInfoHobGuid, sizeof(PCI_SEGMENT_INFO));

     ASSERT(PciSegmentInfo != NULL);

+    if (PciSegmentInfo == NULL) {

+      return NULL;

+    }

     PciSegmentInfo->SegmentNumber = 0;

     PciSegmentInfo->BaseAddress = PcdGet64(PcdPciExpressBaseAddress);

     PciSegmentInfo->StartBusNumber = 0;