NetworkPkg: Code logic optimization for DnsDxe and HttpDxe driver

[mirror_edk2.git] / NetworkPkg / DnsDxe / DnsImpl.c

diff --git a/NetworkPkg/DnsDxe/DnsImpl.c b/NetworkPkg/DnsDxe/DnsImpl.c
index c227093986ba1831e240c21faa530d82a559b710..b196d181cae492a5866f412715b01d48c3f4420f 100644 (file)
--- a/NetworkPkg/DnsDxe/DnsImpl.c
+++ b/NetworkPkg/DnsDxe/DnsImpl.c
@@ -1049,6 +1049,7 @@ IsValidDnsResponse (
       continue;\r
     } else {\r
       TxString = NetbufGetByte (Packet, 0, NULL);\r
+      ASSERT (TxString != NULL);\r
       DnsHeader = (DNS_HEADER *) TxString;\r
       QueryName = (CHAR8 *) (TxString + sizeof (*DnsHeader));\r
       QuerySection = (DNS_QUERY_SECTION *) (QueryName + AsciiStrLen (QueryName) + 1);\r
@@ -1426,7 +1427,8 @@ DnsOnPacketReceived (
   ASSERT (Packet != NULL);\r
   \r
   RcvString = NetbufGetByte (Packet, 0, NULL);\r
-\r
+  ASSERT (RcvString != NULL);\r
+  \r
   //\r
   // Parse Dns Response\r
   //\r