From: Remi Gacogne <rgacogne-github@coredump.fr>
Date: Tue, 1 Dec 2015 17:43:56 +0000 (+0100)
Subject: Make getServer() return a dummy one in client mode. Closes #2976.
X-Git-Tag: dnsdist-1.0.0-alpha1~149^2
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=refs%2Fpull%2F2977%2Fhead;p=thirdparty%2Fpdns.git

Make getServer() return a dummy one in client mode. Closes #2976.
---

diff --git a/pdns/dnsdist-lua.cc b/pdns/dnsdist-lua.cc
index cfbb251d3a..544838eae6 100644
--- a/pdns/dnsdist-lua.cc
+++ b/pdns/dnsdist-lua.cc
@@ -514,7 +514,11 @@ vector<std::function<void(void)>> setupLua(bool client, const std::string& confi
       return getDownstreamCandidates(g_dstates.getCopy(), pool);
     });
 
-  g_lua.writeFunction("getServer", [](int i) { return g_dstates.getCopy().at(i); });
+  g_lua.writeFunction("getServer", [client](int i) {
+      if (client)
+        return std::make_shared<DownstreamState>(ComboAddress());
+      return g_dstates.getCopy().at(i);
+    });
 
   g_lua.registerFunction<void(DownstreamState::*)(int)>("setQPS", [](DownstreamState& s, int lim) { s.qps = lim ? QPSLimiter(lim, lim) : QPSLimiter(); });
   g_lua.registerFunction<void(DownstreamState::*)(string)>("addPool", [](DownstreamState& s, string pool) { s.pools.insert(pool);});