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;