implement getLocal() for our undocumented and unsupported Lua prequer…

…y handler in auth
ahupowerdns · Sep 15, 2015 · 995dd33 · 995dd33
1 parent 1dc8f4d
commit 995dd33
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pdns/dnspacket.hh b/pdns/dnspacket.hh
@@ -75,13 +75,13 @@ public:
   void setRemote(const ComboAddress*);
   string getRemote() const;
   Netmask getRealRemote() const;
-  string getLocal() const
+  ComboAddress getLocal() const
   {
     ComboAddress ca;
     socklen_t len=sizeof(ca);
     getsockname(d_socket, (sockaddr*)&ca, &len);
-    return ca.toString();
-  }
+    return ca;
+  } 
   uint16_t getRemotePort() const;
 
   boost::optional<ComboAddress> d_anyLocal;

diff --git a/pdns/lua-auth.cc b/pdns/lua-auth.cc
@@ -256,6 +256,7 @@ void AuthLua::registerLuaDNSPacket(void) {
 DNSPacket* AuthLua::prequery(DNSPacket *p)
 {
   lua_getglobal(d_lua,"prequery");
+  d_local = p->getLocal();
   if(!lua_isfunction(d_lua, -1)) {
     // cerr<<"No such function 'prequery'\n";
     lua_pop(d_lua, 1);