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

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

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
-      UpdateIndex                   = mFvbCount++;;\r
+      UpdateIndex                   = mFvbCount++;\r
+                       ASSERT (UpdateIndex < MAX_FVB_COUNT);      \r
       mFvbEntry[UpdateIndex].Handle = Handle;\r
     }\r
     //\r