X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=DuetPkg%2FSataControllerDxe%2FSataController.c;fp=DuetPkg%2FSataControllerDxe%2FSataController.c;h=1807c88224ed355216f65874b0a018c06c1252e8;hp=d1e85cc6e4a4b3216976a353b3cf4fc2902fd01c;hb=86702b3fa9437301a6ae02df610ad3ca15fc860a;hpb=6d896ff1cb89aa29830a6da0678a10d777f13021

diff --git a/DuetPkg/SataControllerDxe/SataController.c b/DuetPkg/SataControllerDxe/SataController.c
index d1e85cc6e4..1807c88224 100644
--- a/DuetPkg/SataControllerDxe/SataController.c
+++ b/DuetPkg/SataControllerDxe/SataController.c
@@ -920,6 +920,7 @@ IdeInitCalculateMode (
 
   *SupportedModes = AllocateZeroPool (sizeof (EFI_ATA_COLLECTIVE_MODE));
   if (*SupportedModes == NULL) {
+    ASSERT (*SupportedModes != NULL);
     return EFI_OUT_OF_RESOURCES;
   }
 
@@ -931,6 +932,7 @@ IdeInitCalculateMode (
   // Make sure we've got the valid identify data of the device from SubmitData()
   //
   if (!IdentifyValid) {
+    FreePool (*SupportedModes);
     return EFI_NOT_READY;
   }