tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix 3 K8 issues for PXE module 

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10489 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
hhuan13 2010-05-14 07:47:29 +00:00
parent d60957b856
commit 29a4f92d81
2 changed files with 31 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcDhcp.c
View File

@ -1832,6 +1832,8 @@ PxeBcSelectBootMenu (
gST->ConOut->SetCursorPosition (gST->ConOut, 0, TopRow + MenuNum);
} while (!Finish);
ASSERT (Select < PXEBC_MAX_MENU_NUM);
//
// Swap the byte order
//

31
MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcImpl.c
View File

@ -749,6 +749,7 @@ EfiPxeBcDiscover (
PXEBC_PRIVATE_DATA *Private;
EFI_PXE_BASE_CODE_MODE *Mode;
EFI_PXE_BASE_CODE_DISCOVER_INFO DefaultInfo;
EFI_PXE_BASE_CODE_DISCOVER_INFO *CreatedInfo;
EFI_PXE_BASE_CODE_SRVLIST *SrvList;
EFI_PXE_BASE_CODE_SRVLIST DefaultSrvList;
PXEBC_CACHED_DHCP4_PACKET *Packet;
@ -765,6 +766,7 @@ EfiPxeBcDiscover (
Mode = Private->PxeBc.Mode;
BootSvrEntry = NULL;
SrvList = NULL;
CreatedInfo = NULL;
Status = EFI_DEVICE_ERROR;
Private->Function = EFI_PXE_BASE_CODE_FUNCTION_DISCOVER;
@ -831,6 +833,8 @@ EfiPxeBcDiscover (
}
DefaultInfo.IpCnt = 0;
Info = &DefaultInfo;
SrvList = Info->SrvList;
if (DefaultInfo.MustUseList) {
BootSvrEntry = VendorOpt->BootSvr;
@ -851,9 +855,25 @@ EfiPxeBcDiscover (
}
DefaultInfo.IpCnt = BootSvrEntry->IpCnt;
if (DefaultInfo.IpCnt >= 1) {
CreatedInfo = AllocatePool (sizeof (DefaultInfo) + (DefaultInfo.IpCnt - 1) * sizeof (*SrvList));
if (CreatedInfo == NULL) {
return EFI_OUT_OF_RESOURCES;
}
CopyMem (CreatedInfo, &DefaultInfo, sizeof (DefaultInfo));
Info = CreatedInfo;
SrvList = Info->SrvList;
}
for (Index = 0; Index < DefaultInfo.IpCnt; Index++) {
CopyMem (&SrvList[Index].IpAddr, &BootSvrEntry->IpAddr[Index], sizeof (EFI_IPv4_ADDRESS));
SrvList[Index].AcceptAnyResponse = FALSE;
SrvList[Index].Type = BootSvrEntry->Type;
}
}
Info = &DefaultInfo;
} else {
SrvList = Info->SrvList;
@ -904,6 +924,9 @@ EfiPxeBcDiscover (
TRUE,
&Private->PxeReply.Packet.Ack
);
if (!EFI_ERROR (Status)) {
break;
}
}
} else if (Info->UseMCast) {
@ -953,6 +976,10 @@ EfiPxeBcDiscover (
);
}
if (CreatedInfo != NULL) {
FreePool (CreatedInfo);
}
return Status;
}
@ -1524,7 +1551,7 @@ EfiPxeBcUdpRead (
return EFI_INVALID_PARAMETER;
}
if ((BufferSize == NULL) || ((BufferPtr == NULL) && (*BufferSize != 0))) {
if ((BufferSize == NULL) || (BufferPtr == NULL)) {
return EFI_INVALID_PARAMETER;
}