Use original query TSigOptions on cached response.

stackia · Jun 17, 2015 · 5c18eb7 · 5c18eb7
1 parent ea3505f
commit 5c18eb7
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/DNSAgent/DnsAgent.cs b/DNSAgent/DnsAgent.cs
@@ -201,7 +201,7 @@ await Task.Run(async () =>
                 {
                     DnsMessage message;
                     DnsQuestion question;
-                    var responseFromCache = false;
+                    var respondedFromCache = false;
 
                     try
                     {
@@ -241,9 +241,10 @@ await Task.Run(async () =>
                                 var cachedMessage = entry.Message;
                                 Logger.Info("-> #{0} served from cache.", message.TransactionID,
                                     cachedMessage.TransactionID);
-                                cachedMessage.TransactionID = message.TransactionID;
+                                cachedMessage.TransactionID = message.TransactionID; // Update transaction ID
+                                cachedMessage.TSigOptions = message.TSigOptions; // Update TSig options
                                 message = cachedMessage;
-                                responseFromCache = true;
+                                respondedFromCache = true;
                             }
                         }
                     }
@@ -334,7 +335,7 @@ await Task.Run(async () =>
                                 _udpListener.SendAsync(responseBuffer, responseBuffer.Length, udpMessage.RemoteEndPoint);
 
                             // Update cache
-                            if (Options.CacheResponse && !responseFromCache)
+                            if (Options.CacheResponse && !respondedFromCache)
                                 Cache.Update(question, message, Options.CacheAge);
                         }
                     }