1 files changed, 114 insertions, 0 deletions
diff --git a/dnsdist-lua-bindings-network.cc b/dnsdist-lua-bindings-network.cc
new file mode 100644
index 0000000..62dce3b
--- /dev/null
+++ b/dnsdist-lua-bindings-network.cc
@@ -0,0 +1,114 @@
+/*
+ * This file is part of PowerDNS or dnsdist.
+ * Copyright -- PowerDNS.COM B.V. and its contributors
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of version 2 of the GNU General Public License as
+ * published by the Free Software Foundation.
+ *
+ * In addition, for the avoidance of any doubt, permission is granted to
+ * link this program with OpenSSL and to (re)distribute the binaries
+ * produced as the result of such linking.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+#include "dnsdist.hh"
+#include "dnsdist-async.hh"
+#include "dnsdist-lua.hh"
+#include "dnsdist-lua-ffi.hh"
+#include "dnsdist-lua-network.hh"
+#include "dolog.hh"
+
+void setupLuaBindingsNetwork(LuaContext& luaCtx, bool client)
+{
+  luaCtx.writeFunction("newNetworkEndpoint", [client](const std::string& path) {
+    if (client) {
+      return std::shared_ptr<dnsdist::NetworkEndpoint>(nullptr);
+    }
+
+    try {
+      return std::make_shared<dnsdist::NetworkEndpoint>(path);
+    }
+    catch (const std::exception& e) {
+      warnlog("Error connecting to network endpoint: %s", e.what());
+    }
+    return std::shared_ptr<dnsdist::NetworkEndpoint>(nullptr);
+  });
+
+  luaCtx.registerFunction<bool (std::shared_ptr<dnsdist::NetworkEndpoint>::*)() const>("isValid", [](const std::shared_ptr<dnsdist::NetworkEndpoint>& endpoint) {
+    return endpoint != nullptr;
+  });
+
+  luaCtx.registerFunction<bool (std::shared_ptr<dnsdist::NetworkEndpoint>::*)(const std::string&) const>("send", [client](const std::shared_ptr<dnsdist::NetworkEndpoint>& endpoint, const std::string& payload) {
+    if (client || !endpoint || payload.empty()) {
+      return false;
+    }
+
+    return endpoint->send(payload);
+  });
+
+  luaCtx.writeFunction("newNetworkListener", [client]() {
+    if (client) {
+      return std::shared_ptr<dnsdist::NetworkListener>(nullptr);
+    }
+
+    return std::make_shared<dnsdist::NetworkListener>();
+  });
+
+  luaCtx.registerFunction<bool (std::shared_ptr<dnsdist::NetworkListener>::*)(const std::string&, uint16_t, std::function<void(uint16_t, std::string & dgram, const std::string& from)>)>("addUnixListeningEndpoint", [client](std::shared_ptr<dnsdist::NetworkListener>& listener, const std::string& path, uint16_t endpointID, std::function<void(uint16_t endpoint, std::string & dgram, const std::string& from)> cb) {
+    if (client || !cb) {
+      return false;
+    }
+
+    return listener->addUnixListeningEndpoint(path, endpointID, [cb](dnsdist::NetworkListener::EndpointID endpoint, std::string&& dgram, const std::string& from) {
+      {
+        auto lock = g_lua.lock();
+        cb(endpoint, dgram, from);
+      }
+      dnsdist::handleQueuedAsynchronousEvents();
+    });
+  });
+
+  // if you make the dnsdist_ffi_network_message_t* in the function prototype const, LuaWrapper will stop treating it like a lightuserdata, messing everything up!!
+  luaCtx.registerFunction<bool (std::shared_ptr<dnsdist::NetworkListener>::*)(const std::string&, uint16_t, std::function<void(dnsdist_ffi_network_message_t*)>)>("addUnixListeningEndpointFFI", [client](std::shared_ptr<dnsdist::NetworkListener>& listener, const std::string& path, uint16_t endpointID, std::function<void(dnsdist_ffi_network_message_t*)> cb) {
+    if (client) {
+      return false;
+    }
+
+    return listener->addUnixListeningEndpoint(path, endpointID, [cb](dnsdist::NetworkListener::EndpointID endpoint, std::string&& dgram, const std::string& from) {
+      {
+        auto lock = g_lua.lock();
+        dnsdist_ffi_network_message_t msg(dgram, from, endpoint);
+        cb(&msg);
+      }
+      dnsdist::handleQueuedAsynchronousEvents();
+    });
+  });
+
+  luaCtx.registerFunction<void (std::shared_ptr<dnsdist::NetworkListener>::*)()>("start", [client](std::shared_ptr<dnsdist::NetworkListener>& listener) {
+    if (client) {
+      return;
+    }
+
+    listener->start();
+  });
+
+  luaCtx.writeFunction("getResolvers", [](const std::string& resolvConfPath) -> LuaArray<std::string> {
+    auto resolvers = getResolvers(resolvConfPath);
+    LuaArray<std::string> result;
+    result.reserve(resolvers.size());
+    int counter = 1;
+    for (const auto& resolver : resolvers) {
+      result.emplace_back(counter, resolver.toString());
+      counter++;
+    }
+    return result;
+  });
+};