Fixes answer count in headers generated by DnsStaticLookupClient.

alanedwardes · Aug 4, 2023 · e43c51f · e43c51f
1 parent 30c4a92
commit e43c51f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/Ae.Dns.Client/Lookup/DnsStaticLookupClient.cs b/src/Ae.Dns.Client/Lookup/DnsStaticLookupClient.cs
@@ -77,7 +77,7 @@ private DnsMessage ReturnAddresses(DnsMessage query, IEnumerable<IPAddress> addr
         {
             return new DnsMessage
             {
-                Header = CreateAnswer(query, lookupSource),
+                Header = CreateAnswer(query, lookupSource, (short)addresses.Count()),
                 Answers = addresses.Select(address => new DnsResourceRecord
                 {
                     Class = DnsQueryClass.IN,
@@ -93,7 +93,7 @@ private DnsMessage ReturnPointer(DnsMessage query, IEnumerable<string> foundHost
         {
             return new DnsMessage
             {
-                Header = CreateAnswer(query, lookupSource),
+                Header = CreateAnswer(query, lookupSource, (short)foundHosts.Count()),
                 Answers = foundHosts.Select(foundHost => new DnsResourceRecord
                 {
                     Class = DnsQueryClass.IN,
@@ -105,13 +105,13 @@ private DnsMessage ReturnPointer(DnsMessage query, IEnumerable<string> foundHost
             };
         }
 
-        private DnsHeader CreateAnswer(DnsMessage query, IDnsLookupSource lookupSource) => new DnsHeader
+        private DnsHeader CreateAnswer(DnsMessage query, IDnsLookupSource lookupSource, short answers) => new DnsHeader
         {
             Id = query.Header.Id,
             ResponseCode = DnsResponseCode.NoError,
             IsQueryResponse = true,
             RecursionAvailable = true,
-            AnswerRecordCount = 1,
+            AnswerRecordCount = answers,
             AuthoritativeAnswer = true,
             RecursionDesired = query.Header.RecursionDesired,
             Host = query.Header.Host,