Add more check to avoid code enter into wrong state.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10068 6f19259b-4bc3-4df7-8a09-765794883524
2010-02-25 10:08:54 +00:00
parent bade9bf5b2
commit 234980f6e9
7 changed files with 46 additions and 14 deletions
--- a/EdkCompatibilityPkg/Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk/ConvLib.c
+++ b/EdkCompatibilityPkg/Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk/ConvLib.c
@@ -377,7 +377,7 @@ SmbiosFindHandle (
    }
  }

-  if (Link == &mStructureList) {
+  if (Link == &mStructureList || StructureNode == NULL) {
    return EFI_NOT_FOUND;
  } else {
    *HandleNum = StructureNode->Structure->Handle;
--- a/EdkCompatibilityPkg/Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk/Translate.c
+++ b/EdkCompatibilityPkg/Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk/Translate.c
@@ -161,7 +161,7 @@ SmbiosProcessDataRecord (
      }
    }

-    if (Link == &mStructureList) {
+    if (Link == &mStructureList || StructureNode == NULL) {

      //
      // Not found, create a new structure