From 3cd01b932e1c85394272ae64fae67ebeda92fb00 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sat, 13 Apr 2024 23:11:59 +0200 Subject: Adding upstream version 1.8.3. Signed-off-by: Daniel Baumann --- dnsdist-lua-bindings-dnsparser.cc | 66 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 66 insertions(+) create mode 100644 dnsdist-lua-bindings-dnsparser.cc (limited to 'dnsdist-lua-bindings-dnsparser.cc') diff --git a/dnsdist-lua-bindings-dnsparser.cc b/dnsdist-lua-bindings-dnsparser.cc new file mode 100644 index 0000000..606220e --- /dev/null +++ b/dnsdist-lua-bindings-dnsparser.cc @@ -0,0 +1,66 @@ +/* + * 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-dnsparser.hh" +#include "dnsdist-lua.hh" + +void setupLuaBindingsDNSParser(LuaContext& luaCtx) +{ +#ifndef DISABLE_DNSPACKET_BINDINGS + luaCtx.writeFunction("newDNSPacketOverlay", [](const std::string& packet) { + dnsdist::DNSPacketOverlay dpo(packet); + return dpo; + }); + + luaCtx.registerMember(std::string("qname"), [](const dnsdist::DNSPacketOverlay& overlay) { return overlay.d_qname; }); + luaCtx.registerMember(std::string("qtype"), [](const dnsdist::DNSPacketOverlay& overlay) { return overlay.d_qtype; }); + luaCtx.registerMember(std::string("qclass"), [](const dnsdist::DNSPacketOverlay& overlay) { return overlay.d_qclass; }); + luaCtx.registerMember(std::string("dh"), [](const dnsdist::DNSPacketOverlay& overlay) { return overlay.d_header; }); + + luaCtx.registerFunction("getRecordsCountInSection", [](const dnsdist::DNSPacketOverlay& overlay, uint8_t section) -> uint16_t { + if (section > DNSResourceRecord::ADDITIONAL) { + return 0; + } + uint16_t count = 0; + for (const auto& record : overlay.d_records) { + if (record.d_place == section) { + count++; + } + } + + return count; + }); + + luaCtx.registerFunction("getRecord", [](const dnsdist::DNSPacketOverlay& overlay, size_t idx) { + return overlay.d_records.at(idx); + }); + + luaCtx.registerMember(std::string("name"), [](const dnsdist::DNSPacketOverlay::Record& record) { return record.d_name; }); + luaCtx.registerMember(std::string("type"), [](const dnsdist::DNSPacketOverlay::Record& record) { return record.d_type; }); + luaCtx.registerMember(std::string("class"), [](const dnsdist::DNSPacketOverlay::Record& record) { return record.d_class; }); + luaCtx.registerMember(std::string("ttl"), [](const dnsdist::DNSPacketOverlay::Record& record) { return record.d_ttl; }); + luaCtx.registerMember(std::string("place"), [](const dnsdist::DNSPacketOverlay::Record& record) { return record.d_place; }); + luaCtx.registerMember(std::string("contentLength"), [](const dnsdist::DNSPacketOverlay::Record& record) { return record.d_contentLength; }); + luaCtx.registerMember(std::string("contentOffset"), [](const dnsdist::DNSPacketOverlay::Record& record) { return record.d_contentOffset; }); + +#endif /* DISABLE_DNSPACKET_BINDINGS */ +} -- cgit v1.2.3