Add more check to make code more safely.

[mirror_edk2.git] / MdeModulePkg / Universal / Network / Ip4Dxe / Ip4Input.c

diff --git a/MdeModulePkg/Universal/Network/Ip4Dxe/Ip4Input.c b/MdeModulePkg/Universal/Network/Ip4Dxe/Ip4Input.c
index bd15fc2e176c4939579a5877dfc4828370eac21c..d985b1aeb693fc33d60c71b5671ba26bb6ea1b34 100644 (file)
--- a/MdeModulePkg/Universal/Network/Ip4Dxe/Ip4Input.c
+++ b/MdeModulePkg/Universal/Network/Ip4Dxe/Ip4Input.c
@@ -849,7 +849,8 @@ Ip4AccpetFrame (
     goto DROP;\r
   }\r
 \r
-  Head      = (IP4_HEAD *) NetbufGetByte (Packet, 0, NULL); \r
+  Head      = (IP4_HEAD *) NetbufGetByte (Packet, 0, NULL);\r
+  ASSERT (Head != NULL);\r
   OptionLen = (Head->HeadLen << 2) - IP4_MIN_HEADLEN;\r
   if (OptionLen > 0) {\r
     Option = (UINT8 *) (Head + 1);\r
@@ -899,6 +900,7 @@ Ip4AccpetFrame (
   // is transfered to the packet process logic.\r
   //\r
     Head = (IP4_HEAD *) NetbufGetByte (Packet, 0, NULL);\r
+    ASSERT (Head != NULL);\r
     Status = Ip4PreProcessPacket (\r
                IpSb,\r
                &Packet,\r