Refine code to make code run safely.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10880 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
ydong10
2010-09-16 04:51:25 +00:00
parent 7bee5a761f
commit cd7bfc2c63
12 changed files with 29 additions and 26 deletions

View File

@@ -611,7 +611,7 @@ DhcpParseOption (
// Allocate a buffer to hold the DHCP options, and after that, a
// continuous buffer to put all the options' data.
//
Options = AllocateZeroPool (OptNum * sizeof (DHCP_OPTION) + TotalLen);
Options = AllocateZeroPool ((UINTN) (OptNum * sizeof (DHCP_OPTION)) + TotalLen);
if (Options == NULL) {
Status = EFI_OUT_OF_RESOURCES;
@@ -754,7 +754,7 @@ DhcpAppendOption (
*(Buf++) = Tag;
*(Buf++) = (UINT8) Len;
CopyMem (Buf, Data + Index * 255, Len);
CopyMem (Buf, Data + Index * 255, (UINTN) Len);
Buf += Len;
}