Modify AdvancedLoggerDxelib

microsoft · Aug 13, 2024 · 29bd4ef · 29bd4ef
1 parent cf4b663
commit 29bd4ef
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/AdvLoggerPkg/Library/AdvancedLoggerLib/Dxe/AdvancedLoggerLib.c b/AdvLoggerPkg/Library/AdvancedLoggerLib/Dxe/AdvancedLoggerLib.c
@@ -66,6 +66,17 @@ AdvancedLoggerWrite (
                     NULL,
                     (VOID **)&mLoggerProtocol
                     );
+
+    if (!EFI_ERROR (Status)) {
+      if ((mLoggerProtocol->Signature != ADVANCED_LOGGER_PROTOCOL_SIGNATURE) ||
+          (mLoggerProtocol->Version != ADVANCED_LOGGER_PROTOCOL_VERSION))
+      {
+        ASSERT (mLoggerProtocol->Signature == ADVANCED_LOGGER_PROTOCOL_SIGNATURE);
+        ASSERT (mLoggerProtocol->Version == ADVANCED_LOGGER_PROTOCOL_VERSION);
+        Status = EFI_NOT_FOUND;
+      }
+    }
+
     if (EFI_ERROR (Status)) {
       mLoggerProtocol = NULL;
       Status          = gBS->LocateProtocol (
@@ -76,9 +87,6 @@ AdvancedLoggerWrite (
       if (EFI_ERROR (Status)) {
         mDebugPortProtocol = NULL;
       }
-    } else {
-      ASSERT (mLoggerProtocol->Signature == ADVANCED_LOGGER_PROTOCOL_SIGNATURE);
-      ASSERT (mLoggerProtocol->Version == ADVANCED_LOGGER_PROTOCOL_VERSION);
     }
   }