NetworkPkg/DnsDxe: [CVE-2018-12178] Check the received packet size before parsing...

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

diff --git a/NetworkPkg/DnsDxe/DnsImpl.h b/NetworkPkg/DnsDxe/DnsImpl.h
index 90dc0549038428859190500088263a6f1d92affa..45feca21602c08e953fd956fd9b1db0cfb251610 100644 (file)
--- a/NetworkPkg/DnsDxe/DnsImpl.h
+++ b/NetworkPkg/DnsDxe/DnsImpl.h
@@ -583,6 +583,7 @@ IsValidDnsResponse (
 \r
   @param  Instance              The DNS instance\r
   @param  RxString              Received buffer.\r
+  @param  Length                Received buffer length.\r
   @param  Completed             Flag to indicate that Dns response is valid.\r
 \r
   @retval EFI_SUCCESS           Parse Dns Response successfully.\r
@@ -593,6 +594,7 @@ EFI_STATUS
 ParseDnsResponse (\r
   IN OUT DNS_INSTANCE              *Instance,\r
   IN     UINT8                     *RxString,\r
+  IN     UINT32                    Length,\r
      OUT BOOLEAN                   *Completed\r
   );\r
 \r