Add more check for index whether exceed maximum value of fv count.

[mirror_edk2.git] / EdkModulePkg / Library / EdkFvbServiceLib / Ia32 / Fvb.c

diff --git a/EdkModulePkg/Library/EdkFvbServiceLib/Ia32/Fvb.c b/EdkModulePkg/Library/EdkFvbServiceLib/Ia32/Fvb.c
index 222fb9bd4d5d04a273d14e19748037f088a22c15..04693eb6322b184c3d448c2ee2275cd89144be6c 100644 (file)
--- a/EdkModulePkg/Library/EdkFvbServiceLib/Ia32/Fvb.c
+++ b/EdkModulePkg/Library/EdkFvbServiceLib/Ia32/Fvb.c
@@ -100,7 +100,11 @@ Returns:
       //\r
       // Use the next free slot for a new entry\r
       //\r
-      UpdateIndex                   = mFvbCount++;;\r
+      UpdateIndex                   = mFvbCount++;\r
+      //\r
+      // Check the UpdateIndex whether exceed the maximum value.\r
+      //\r
+      ASSERT (UpdateIndex < MAX_FVB_COUNT);\r
       mFvbEntry[UpdateIndex].Handle = Handle;\r
     }\r
     //\r