dnsdist: Do not waste CPU cycles handling EDNS twice on self-generated answers

Signed-off-by: Remi Gacogne <remi.gacogne@powerdns.com>

diff --git a/pdns/dnsdistdist/dnsdist-self-answers.cc b/pdns/dnsdistdist/dnsdist-self-answers.cc
index 4e0b66084e52c3f82e835e3e33be497b09cec30c..be608fbd6e7405f17444eeb68ea2e41900705eaa 100644 (file)
--- a/pdns/dnsdistdist/dnsdist-self-answers.cc
+++ b/pdns/dnsdistdist/dnsdist-self-answers.cc
@@ -94,6 +94,7 @@ bool generateAnswerFromCNAME(DNSQuestion& dnsQuestion, const DNSName& cname, con
     addEDNS(dnsQuestion.getMutableData(), dnsQuestion.getMaximumSize(), dnssecOK, dnsdist::configuration::getCurrentRuntimeConfiguration().d_payloadSizeSelfGenAnswers, 0);
   }
 
+  dnsQuestion.d_selfGeneratedHandledEDNS = true;
   return true;
 }
 
@@ -164,6 +165,7 @@ bool generateAnswerFromIPAddresses(DNSQuestion& dnsQuestion, const std::vector<C
     addEDNS(dnsQuestion.getMutableData(), dnsQuestion.getMaximumSize(), dnssecOK, dnsdist::configuration::getCurrentRuntimeConfiguration().d_payloadSizeSelfGenAnswers, 0);
   }
 
+  dnsQuestion.d_selfGeneratedHandledEDNS = true;
   return true;
 }
 
@@ -226,6 +228,7 @@ bool generateAnswerFromRDataEntries(DNSQuestion& dnsQuestion, const std::vector<
     addEDNS(dnsQuestion.getMutableData(), dnsQuestion.getMaximumSize(), dnssecOK, dnsdist::configuration::getCurrentRuntimeConfiguration().d_payloadSizeSelfGenAnswers, 0);
   }
 
+  dnsQuestion.d_selfGeneratedHandledEDNS = true;
   return true;
 }
 
@@ -233,7 +236,7 @@ bool generateAnswerFromRawPacket(DNSQuestion& dnsQuestion, const PacketBuffer& p
 {
   auto questionId = dnsQuestion.getHeader()->id;
   dnsQuestion.getMutableData() = packet;
-  dnsQuestion.d_selfGeneratedFromPacket = true;
+  dnsQuestion.d_selfGeneratedHandledEDNS = true;
   dnsdist::PacketMangling::editDNSHeaderFromPacket(dnsQuestion.getMutableData(), [questionId](dnsheader& header) {
     header.id = questionId;
     return true;

diff --git a/pdns/dnsdistdist/dnsdist.cc b/pdns/dnsdistdist/dnsdist.cc
index b96617e0a570cced41ec7e8928e09d8ff63c7e54..ff05ffdc2151af96eb8aa1695da4c8a882e1aba3 100644 (file)
--- a/pdns/dnsdistdist/dnsdist.cc
+++ b/pdns/dnsdistdist/dnsdist.cc
@@ -326,7 +326,7 @@ static bool fixUpQueryTurnedResponse(DNSQuestion& dnsQuestion, const uint16_t or
     return true;
   });
 
-  if (dnsQuestion.d_selfGeneratedFromPacket) {
+  if (dnsQuestion.d_selfGeneratedHandledEDNS) {
     return true;
   }
   return addEDNSToQueryTurnedResponse(dnsQuestion);

diff --git a/pdns/dnsdistdist/dnsdist.hh b/pdns/dnsdistdist/dnsdist.hh
index 06b32ade6fec14f4cdd8efcd6e44ceefadba622f..2d10591eab6989cdae12b6f819f53c45981d1028 100644 (file)
--- a/pdns/dnsdistdist/dnsdist.hh
+++ b/pdns/dnsdistdist/dnsdist.hh
@@ -184,7 +184,7 @@ public:
   bool ecsOverride;
   bool useECS{true};
   bool asynchronous{false};
-  bool d_selfGeneratedFromPacket{false};
+  bool d_selfGeneratedHandledEDNS{false};
 };
 
 struct DownstreamState;