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

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2115 6f19259b-4bc3-4df7-8a09-765794883524

diff --git a/EdkModulePkg/Library/EdkFvbServiceLib/x64/Fvb.c b/EdkModulePkg/Library/EdkFvbServiceLib/x64/Fvb.c
index e9d7500261a2183f22e1c84b3803a353acdc3478..4dcb81d2161c4b8b3a9776604db2d4c21abafcb4 100644 (file)
--- a/EdkModulePkg/Library/EdkFvbServiceLib/x64/Fvb.c
+++ b/EdkModulePkg/Library/EdkFvbServiceLib/x64/Fvb.c
@@ -102,7 +102,8 @@ Returns:
       //\r
       // Use the next free slot for a new entry\r
       //\r
       //\r
       // Use the next free slot for a new entry\r
       //\r

-      UpdateIndex                   = mFvbCount++;;\r
+      UpdateIndex                   = mFvbCount++;\r
+                       ASSERT (UpdateIndex < MAX_FVB_COUNT);      \r

       mFvbEntry[UpdateIndex].Handle = Handle;\r
     }\r
     //\r
       mFvbEntry[UpdateIndex].Handle = Handle;\r
     }\r
     //\r