Add DNSPacket::setRealRemote: accepts an EDNS Subnet Options Netmask

diff --git a/pdns/dnspacket.cc b/pdns/dnspacket.cc
index 471af2a46ef00b3a659a464c3ef07fe634933d2d..499fa999f1ddd9160341d245ccb54a7a0ca036d0 100644 (file)
--- a/pdns/dnspacket.cc
+++ b/pdns/dnspacket.cc
@@ -704,6 +704,11 @@ bool DNSPacket::hasValidEDNSCookie() const
   return d_ednscookievalid;
 }
 
+void DNSPacket::setRealRemote(const Netmask& netmask) {
+  d_eso.source = netmask;
+  d_haveednssubnet = true;
+}
+
 Netmask DNSPacket::getRealRemote() const
 {
   return d_haveednssubnet ? d_eso.source : Netmask{getInnerRemote()};

diff --git a/pdns/dnspacket.hh b/pdns/dnspacket.hh
index 4d2271b80ac07a3d8cba74f28ba0054730eb427b..a9bd01004831298cfbeddfe75e0085dbcf066a0a 100644 (file)
--- a/pdns/dnspacket.hh
+++ b/pdns/dnspacket.hh
@@ -65,6 +65,7 @@ public:
   ComboAddress getRemote() const;
   ComboAddress getInnerRemote() const; // for proxy protocol
   Netmask getRealRemote() const;
+  void setRealRemote(const Netmask& netmask);
   ComboAddress getLocal() const
   {
     ComboAddress ca;