Refine code to make it more safely.
Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Eric Dong <eric.dong@intel.com> Reviewed-by: Feng Tian <feng.tian@intel.com> git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@15598 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
@@ -227,6 +227,7 @@ IScsiUpdateDeviceList (
|
||||
);
|
||||
if (Status == EFI_BUFFER_TOO_SMALL) {
|
||||
DeviceList = (ISCSI_DEVICE_LIST *) AllocatePool (DataSize);
|
||||
ASSERT (DeviceList != NULL);
|
||||
|
||||
gRT->GetVariable (
|
||||
L"iSCSIDeviceList",
|
||||
@@ -291,6 +292,7 @@ IScsiUpdateDeviceList (
|
||||
//
|
||||
DeviceListSize = sizeof (ISCSI_DEVICE_LIST) + (NumHandles - 1) * sizeof (ISCSI_MAC_INFO);
|
||||
DeviceList = (ISCSI_DEVICE_LIST *) AllocatePool (DeviceListSize);
|
||||
ASSERT (DeviceList != NULL);
|
||||
DeviceList->NumDevice = (UINT8) NumHandles;
|
||||
|
||||
for (Index = 0; Index < NumHandles; Index++) {
|
||||
|
Reference in New Issue
Block a user