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/Mtftp4Dxe/Mtftp4Support.c | 2 ++ 1 file changed, 2 insertions(+) (limited to 'MdeModulePkg/Universal/Network/Mtftp4Dxe') diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c index 26860e52bb..90207a2a9d 100644 --- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c +++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c @@ -465,6 +465,7 @@ Mtftp4SendPacket ( // to the connected port // Value = *((UINT16 *) NetbufGetByte (Packet, 0, NULL)); + ASSERT (Value != NULL); OpCode = NTOHS (Value); if ((OpCode == EFI_MTFTP4_OPCODE_RRQ) || @@ -522,6 +523,7 @@ Mtftp4Retransmit ( // Set the requests to the listening port, other packets to the connected port // Value = *(UINT16 *) NetbufGetByte (Instance->LastPacket, 0, NULL); + ASSERT (Value != NULL); OpCode = NTOHS (Value); if ((OpCode == EFI_MTFTP4_OPCODE_RRQ) || (OpCode == EFI_MTFTP4_OPCODE_DIR) || -- cgit v1.2.3