From a56b6e03e22c4023fdf5b026b0fcb096d6a0f677 Mon Sep 17 00:00:00 2001 From: ydong10 Date: Thu, 1 Nov 2012 02:10:31 +0000 Subject: Add more check to make code more safely. Signed-off-by: Eric Dong Reviewed-by: Ouyang Qian git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@13903 6f19259b-4bc3-4df7-8a09-765794883524 --- MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Impl.c | 2 ++ 1 file changed, 2 insertions(+) (limited to 'MdeModulePkg/Universal/Network/Udp4Dxe') diff --git a/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Impl.c b/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Impl.c index ffcbfed2c5..9326f3d344 100644 --- a/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Impl.c +++ b/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Impl.c @@ -1612,6 +1612,7 @@ Udp4Demultiplex ( // Get the datagram header from the packet buffer. // Udp4Header = (EFI_UDP_HEADER *) NetbufGetByte (Packet, 0, NULL); + ASSERT (Udp4Header != NULL); if (Udp4Header->Checksum != 0) { // @@ -1799,6 +1800,7 @@ Udp4IcmpHandler ( UDP4_INSTANCE_DATA *Instance; Udp4Header = (EFI_UDP_HEADER *) NetbufGetByte (Packet, 0, NULL); + ASSERT (Udp4Header != NULL); CopyMem (&Udp4Session.SourceAddress, &NetSession->Source, sizeof (EFI_IPv4_ADDRESS)); CopyMem (&Udp4Session.DestinationAddress, &NetSession->Dest, sizeof (EFI_IPv4_ADDRESS)); -- cgit v1.2.3