MdePkg,NetwokrPkg: Do not use AIP InformationBlock if its size is 0

Signed-off-by: Michał Żygowski <michal.zygowski@3mdeb.com>

MdePkg/Library/DxeHstiLib/HstiDxe.c

@@ -105,7 +105,7 @@ InternalHstiFindAip (
                     &InformationBlock,
                     &InformationBlockSize
                     );
-    if (EFI_ERROR (Status)) {
+    if (EFI_ERROR (Status) || InformationBlockSize == 0) {
       continue;
     }
 

NetworkPkg/Library/DxeNetLib/DxeNetLib.c

@@ -2742,6 +2742,13 @@ NetLibDetectMediaWaitTimeout (
                   &DataSize
                   );
   if (!EFI_ERROR (Status)) {
+    if (DataSize == 0) {
+      if (MediaInfo != NULL) {
+        FreePool (MediaInfo);
+      }
+      return EFI_NOT_FOUND;
+    }
+
     *MediaState = MediaInfo->MediaState;
     FreePool (MediaInfo);
     if ((*MediaState != EFI_NOT_READY) || (Timeout < MEDIA_STATE_DETECT_TIME_INTERVAL)) {
@@ -2805,6 +2812,12 @@ NetLibDetectMediaWaitTimeout (
                                &DataSize
                                );
         if (!EFI_ERROR (Status)) {
+          if (DataSize == 0) {
+            if (MediaInfo != NULL) {
+              FreePool (MediaInfo);
+            }
+            return EFI_NOT_FOUND;
+          }
           *MediaState = MediaInfo->MediaState;
           FreePool (MediaInfo);
         } else {

NetworkPkg/WifiConnectionManagerDxe/WifiConnectionMgrImpl.c

@@ -830,8 +830,12 @@ WifiMgrGetLinkState (
 
   gBS->RestoreTPL (OldTpl);
 
-  CopyMem (LinkState, UndiState, sizeof (EFI_ADAPTER_INFO_MEDIA_STATE));
-  FreePool (UndiState);
+  if (DataSize != 0) {
+    CopyMem (LinkState, UndiState, sizeof (EFI_ADAPTER_INFO_MEDIA_STATE));
+    FreePool (UndiState);
+  } else {
+    return EFI_NOT_FOUND;
+  }
   return EFI_SUCCESS;
 }