Adjust code to fix potential array out-bound issues.

[mirror_edk2.git] / MdeModulePkg / Universal / Variable / Pei / Ipf / VariableWorker.c

diff --git a/MdeModulePkg/Universal/Variable/Pei/Ipf/VariableWorker.c b/MdeModulePkg/Universal/Variable/Pei/Ipf/VariableWorker.c
index a5e0e550e3f4a758a2d5a687f3a577dae9c44cda..17f152a0f1317ba4a00b55fe8aca05186558f22a 100644 (file)
--- a/MdeModulePkg/Universal/Variable/Pei/Ipf/VariableWorker.c
+++ b/MdeModulePkg/Universal/Variable/Pei/Ipf/VariableWorker.c
@@ -49,7 +49,9 @@ VariableIndexTableUpdate (
   IN      VARIABLE_HEADER        *Variable\r
   )\r
 {\r
-  IndexTable->Index[IndexTable->Length++] = (UINT16) (((UINT32)(UINTN) Variable) >> 2);\r
+  if (IndexTable->Length < VARIABLE_INDEX_TABLE_VOLUME) {\r
+    IndexTable->Index[IndexTable->Length++] = (UINT16) (((UINT32)(UINTN) Variable) >> 2);\r
+  }\r
 \r
   return;\r
 }\r