From: Peter van Dijk <peter.van.dijk@powerdns.com>
Date: Mon, 8 Aug 2016 19:06:15 +0000 (+0200)
Subject: allow newDN to take a DNSName in
X-Git-Tag: rec-4.0.2~41^2~1
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=0bfc199afca5a6f492691676eca38318fc3e7b46;p=thirdparty%2Fpdns.git

allow newDN to take a DNSName in
---

diff --git a/pdns/lua-recursor4.cc b/pdns/lua-recursor4.cc
index 0db7ef432f..4e7bdf0c9b 100644
--- a/pdns/lua-recursor4.cc
+++ b/pdns/lua-recursor4.cc
@@ -229,7 +229,12 @@ RecursorLua4::RecursorLua4(const std::string& fname)
   d_lw->registerFunction<int(dnsheader::*)()>("getNSCOUNT", [](dnsheader& dh) { return ntohs(dh.nscount); });
   d_lw->registerFunction<int(dnsheader::*)()>("getARCOUNT", [](dnsheader& dh) { return ntohs(dh.arcount); });
 
-  d_lw->writeFunction("newDN", [](const std::string& dom){ return DNSName(dom); });  
+  d_lw->writeFunction("newDN", [](boost::variant<const std::string, const DNSName> dom){
+    if(dom.which() == 0)
+      return DNSName(boost::get<const std::string>(dom));
+    else
+      return DNSName(boost::get<const DNSName>(dom));
+  });
   d_lw->registerFunction("isPartOf", &DNSName::isPartOf);
   d_lw->registerFunction<bool(DNSName::*)(const std::string&)>("equal",
 							      [](const DNSName& lhs, const std::string& rhs) { return lhs==DNSName(rhs); });