Add pointer check for NULL before dereference it.

Signed-off-by: ydong10
Reviewed-by: rsun3, lgao4

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12472 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
ydong10
2011-09-29 06:33:23 +00:00
parent 57ad9d43b1
commit 04eb20aa85
7 changed files with 28 additions and 5 deletions

View File

@@ -419,7 +419,15 @@ Ip4Reassemble (
}
NewPacket->Ip.Ip4 = Assemble->Head;
CopyMem (IP4_GET_CLIP_INFO (NewPacket), Assemble->Info, sizeof (*IP4_GET_CLIP_INFO (NewPacket)));
ASSERT (Assemble->Info != NULL);
CopyMem (
IP4_GET_CLIP_INFO (NewPacket),
Assemble->Info,
sizeof (*IP4_GET_CLIP_INFO (NewPacket))
);
return NewPacket;
}
@@ -1272,7 +1280,9 @@ Ip4InstanceDeliverPacket (
// headless. Trim the head off after copy. The IP head
// may be not continuous before the data.
//
Head = NetbufAllocSpace (Dup, IP4_MAX_HEADLEN, NET_BUF_HEAD);
Head = NetbufAllocSpace (Dup, IP4_MAX_HEADLEN, NET_BUF_HEAD);
ASSERT (Head != NULL);
Dup->Ip.Ip4 = (IP4_HEAD *) Head;
CopyMem (Head, Packet->Ip.Ip4, Packet->Ip.Ip4->HeadLen << 2);