From 234980f6e935a51fdafecb9930b4a2ecbf47b727 Mon Sep 17 00:00:00 2001
From: lgao4 <lgao4@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Thu, 25 Feb 2010 10:08:54 +0000
Subject: 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
---
 .../Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk/ConvLib.c    | 2 +-
 .../Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk/Translate.c  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

(limited to 'EdkCompatibilityPkg/Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk')

diff --git a/EdkCompatibilityPkg/Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk/ConvLib.c b/EdkCompatibilityPkg/Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk/ConvLib.c
index fac1fb7b58..224f621e9d 100644
--- 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;
diff --git a/EdkCompatibilityPkg/Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk/Translate.c b/EdkCompatibilityPkg/Compatibility/PiSmbiosRecordOnDataHubSmbiosRecordThunk/Translate.c
index 3230d96576..ad41ec0783 100644
--- 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
-- 
cgit v1.2.3