Fix potential buffer overflow issue.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@8627 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
klu2 2009-06-23 08:57:31 +00:00
parent f0a3b1a221
commit 84fa6c177e
2 changed files with 2 additions and 1 deletions

View File

@ -755,6 +755,7 @@ GetConsoleMenu (
NewMenuEntry->OptionNumber = Index2; NewMenuEntry->OptionNumber = Index2;
NewConsoleContext->DevicePath = DuplicateDevicePath (DevicePathInst); NewConsoleContext->DevicePath = DuplicateDevicePath (DevicePathInst);
ASSERT (NewConsoleContext->DevicePath != NULL);
NewMenuEntry->DisplayString = EfiLibStrFromDatahub (NewConsoleContext->DevicePath); NewMenuEntry->DisplayString = EfiLibStrFromDatahub (NewConsoleContext->DevicePath);
if (NULL == NewMenuEntry->DisplayString) { if (NULL == NewMenuEntry->DisplayString) {
NewMenuEntry->DisplayString = DevicePathToStr (NewConsoleContext->DevicePath); NewMenuEntry->DisplayString = DevicePathToStr (NewConsoleContext->DevicePath);

View File

@ -469,7 +469,7 @@ HotkeyRegisterNotify (
return Status; return Status;
} }
Index ++; Index ++;
} while (Index < Hotkey->CodeCount); } while ((Index < Hotkey->CodeCount) && (Index < (sizeof (Hotkey->KeyData) / sizeof (EFI_KEY_DATA))));
Link = GetNextNode (&mHotkeyList, Link); Link = GetNextNode (&mHotkeyList, Link);
} }