25787327741

Committed 13 May 2026 08:22AM UTC coverage: 67.025% (-4.0%) from 71.073%

Build # 25787327741

Build Type

Pull #17358

github

Committed by

web-flow

Commit Message

Merge a0e927df0 into 391c7f8bc

Pull Request Pull Request #17358: auth: some views traces

Coverage Stats

42899 of 81086 branches covered (52.91%)

Branch coverage included in aggregate %.

14 of 24 new or added lines in 4 files covered. (58.33%)

9438 existing lines in 125 files now uncovered.

125326 of 169904 relevant lines covered (73.76%)

7758157.79 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

49.54

/pdns/lua-base4.cc

#include "config.h"
#include <cassert>
#include <fstream>
#include <unordered_set>
#include <unordered_map>
#include <typeinfo>
#include <sys/stat.h>
#include "logger.hh"
#include "logging.hh"
#include "iputils.hh"
#include "dnsname.hh"
#include "dnsparser.hh"
#include "dnspacket.hh"
#include "namespaces.hh"
#include "ednssubnet.hh"
#include "lua-base4.hh"
#include "ext/luawrapper/include/LuaContext.hpp"
#include "dns_random.hh"

void BaseLua4::loadFile(const std::string& fname, bool doPostLoad)
{
  std::ifstream ifs(fname);
  if (!ifs) {
    auto ret = errno;
    auto msg = stringerror(ret);
    SLOG(g_log << Logger::Error << "Unable to read configuration file from '" << fname << "': " << msg << endl,
         g_slog->withName("lua")->error(Logr::Error, msg, "Unable to read configuration file", "file", Logging::Loggable(fname)));
    throw std::runtime_error(msg);
  }
  loadStream(ifs, doPostLoad);
};

void BaseLua4::loadString(const std::string &script) {
  std::istringstream iss(script);
  loadStream(iss, true);
};

void BaseLua4::includePath(const std::string& directory) {
  std::vector<std::string> vec;
  const std::string& suffix = "lua";
  auto directoryError = pdns::visit_directory(directory, [this, &directory, &suffix, &vec]([[maybe_unused]] ino_t inodeNumber, const std::string_view& name) {
    (void)this;
    if (boost::starts_with(name, ".")) {
      return true; // skip any dots
    }
    if (boost::ends_with(name, suffix)) {
      // build name
      string fullName = directory + "/" + std::string(name);
      // ensure it's readable file
      struct stat statInfo
      {
      };
      if (stat(fullName.c_str(), &statInfo) != 0 || !S_ISREG(statInfo.st_mode)) {
        string msg = fullName + " is not a regular file";
        SLOG(g_log << Logger::Error << msg << std::endl,
             g_slog->withName("lua")->info(Logr::Error, "include file is not a regular file", "file", Logging::Loggable(fullName)));
        throw PDNSException(std::move(msg));
      }
      vec.emplace_back(fullName);
    }
    return true;
  });

  if (directoryError) {
    int err = errno;
    string msg = directory + " is not accessible: " + stringerror(err);
    SLOG(g_log << Logger::Error << msg << std::endl,
         g_slog->withName("lua")->error(Logr::Error, err, "Error trying to walk directory", "directory", Logging::Loggable(directory)));
    throw PDNSException(std::move(msg));
  }

  std::sort(vec.begin(), vec.end(), CIStringComparePOSIX());

  for(const auto& file: vec) {
    loadFile(file, false);
  }
};

//  By default no features
void BaseLua4::getFeatures(Features &) { }

void BaseLua4::prepareContext() {
  d_lw = std::make_unique<LuaContext>();

  // lua features available
  Features features;
  getFeatures(features);
  d_lw->writeVariable("pdns_features", features);

  // dnsheader
  d_lw->registerFunction<int(dnsheader::*)()>("getID", [](dnsheader& dh) { return ntohs(dh.id); });
  d_lw->registerFunction<bool(dnsheader::*)()>("getCD", [](dnsheader& dh) { return dh.cd; });
  d_lw->registerFunction<bool(dnsheader::*)()>("getTC", [](dnsheader& dh) { return dh.tc; });
  d_lw->registerFunction<bool(dnsheader::*)()>("getRA", [](dnsheader& dh) { return dh.ra; });
  d_lw->registerFunction<bool(dnsheader::*)()>("getAD", [](dnsheader& dh) { return dh.ad; });
  d_lw->registerFunction<bool(dnsheader::*)()>("getAA", [](dnsheader& dh) { return dh.aa; });
  d_lw->registerFunction<bool(dnsheader::*)()>("getRD", [](dnsheader& dh) { return dh.rd; });
  d_lw->registerFunction<int(dnsheader::*)()>("getRCODE", [](dnsheader& dh) { return dh.rcode; });
  d_lw->registerFunction<int(dnsheader::*)()>("getOPCODE", [](dnsheader& dh) { return dh.opcode; });
  d_lw->registerFunction<int(dnsheader::*)()>("getQDCOUNT", [](dnsheader& dh) { return ntohs(dh.qdcount); });
  d_lw->registerFunction<int(dnsheader::*)()>("getANCOUNT", [](dnsheader& dh) { return ntohs(dh.ancount); });
  d_lw->registerFunction<int(dnsheader::*)()>("getNSCOUNT", [](dnsheader& dh) { return ntohs(dh.nscount); });
  d_lw->registerFunction<int(dnsheader::*)()>("getARCOUNT", [](dnsheader& dh) { return ntohs(dh.arcount); });

  // DNSName
  d_lw->writeFunction("newDN", [](const std::string& dom){ return DNSName(dom); });
  d_lw->registerFunction("__lt", &DNSName::operator<);
  d_lw->registerFunction<bool(DNSName::*)(const DNSName&)>("canonCompare", [](const DNSName& name, const DNSName& rhs) { return name.canonCompare(rhs); });
  d_lw->registerFunction<DNSName(DNSName::*)(const DNSName&)>("makeRelative", [](const DNSName& name, const DNSName& zone) { return name.makeRelative(zone); });
  d_lw->registerFunction<bool(DNSName::*)(const DNSName&)>("isPartOf", [](const DNSName& name, const DNSName& rhs) { return name.isPartOf(rhs); });
  d_lw->registerFunction("getRawLabels", &DNSName::getRawLabels);
  d_lw->registerFunction<unsigned int(DNSName::*)()>("countLabels", [](const DNSName& name) { return name.countLabels(); });
  d_lw->registerFunction<size_t(DNSName::*)()>("wireLength", [](const DNSName& name) { return name.wirelength(); });
  d_lw->registerFunction<size_t(DNSName::*)()>("wirelength", [](const DNSName& name) { return name.wirelength(); });
  d_lw->registerFunction<bool(DNSName::*)(const std::string&)>("equal", [](const DNSName& lhs, const std::string& rhs) { return lhs==DNSName(rhs); });
  d_lw->registerEqFunction(&DNSName::operator==);
  d_lw->registerToStringFunction<string(DNSName::*)()>([](const DNSName&dn ) { return dn.toString(); });
  d_lw->registerFunction<string(DNSName::*)()>("toString", [](const DNSName&dn ) { return dn.toString(); });
  d_lw->registerFunction<string(DNSName::*)()>("toStringNoDot", [](const DNSName&dn ) { return dn.toStringNoDot(); });
  d_lw->registerFunction<bool(DNSName::*)()>("chopOff", [](DNSName&dn ) { return dn.chopOff(); });

  // DNSResourceRecord
  d_lw->writeFunction("newDRR", [](const DNSName& qname, const string& qtype, const unsigned int ttl, const string& content, boost::optional<int> domain_id, boost::optional<int> auth){
    auto drr = DNSResourceRecord();
    drr.qname = qname;
    drr.qtype = qtype;
    drr.ttl = ttl;
    drr.setContent(content);
    if (domain_id)
      drr.domain_id = *domain_id;
    if (auth)
      drr.auth = *auth;
     return drr;
  });
  d_lw->registerEqFunction(&DNSResourceRecord::operator==);
  d_lw->registerFunction("__lt", &DNSResourceRecord::operator<);
  d_lw->registerToStringFunction<string(DNSResourceRecord::*)()>([](const DNSResourceRecord& rec) { return rec.getZoneRepresentation(); });
  d_lw->registerFunction<string(DNSResourceRecord::*)()>("toString", [](const DNSResourceRecord& rec) { return rec.getZoneRepresentation();} );
  d_lw->registerFunction<DNSName(DNSResourceRecord::*)()>("qname", [](DNSResourceRecord& rec) { return rec.qname; });
  d_lw->registerFunction<DNSName(DNSResourceRecord::*)()>("wildcardName", [](DNSResourceRecord& rec) { return rec.wildcardname; });
  d_lw->registerFunction<string(DNSResourceRecord::*)()>("content", [](DNSResourceRecord& rec) { return rec.content; });
  d_lw->registerFunction<time_t(DNSResourceRecord::*)()>("lastModified", [](DNSResourceRecord& rec) { return rec.last_modified; });
  d_lw->registerFunction<uint32_t(DNSResourceRecord::*)()>("ttl", [](DNSResourceRecord& rec) { return rec.ttl; });
  d_lw->registerFunction<uint32_t(DNSResourceRecord::*)()>("signttl", [](DNSResourceRecord& rec) { return rec.signttl; });
  d_lw->registerFunction<int(DNSResourceRecord::*)()>("domainId", [](DNSResourceRecord& rec) { return rec.domain_id; });
  d_lw->registerFunction<uint16_t(DNSResourceRecord::*)()>("qtype", [](DNSResourceRecord& rec) { return rec.qtype.getCode(); });
  d_lw->registerFunction<uint16_t(DNSResourceRecord::*)()>("qclass", [](DNSResourceRecord& rec) { return rec.qclass; });
  d_lw->registerFunction<uint8_t(DNSResourceRecord::*)()>("scopeMask", [](DNSResourceRecord& rec) { return rec.scopeMask; });
  d_lw->registerFunction<bool(DNSResourceRecord::*)()>("auth", [](DNSResourceRecord& rec) { return rec.auth; });
  d_lw->registerFunction<bool(DNSResourceRecord::*)()>("disabled", [](DNSResourceRecord& rec) { return rec.disabled; });

  // ComboAddress
  d_lw->registerFunction<bool(ComboAddress::*)()>("isIPv4", [](const ComboAddress& addr) { return addr.sin4.sin_family == AF_INET; });
  d_lw->registerFunction<bool(ComboAddress::*)()>("isIPv6", [](const ComboAddress& addr) { return addr.sin4.sin_family == AF_INET6; });
  d_lw->registerFunction<uint16_t(ComboAddress::*)()>("getPort", [](const ComboAddress& addr) { return ntohs(addr.sin4.sin_port); } );
  d_lw->registerFunction<bool(ComboAddress::*)()>("isMappedIPv4", [](const ComboAddress& addr) { return addr.isMappedIPv4(); });
  d_lw->registerFunction<ComboAddress(ComboAddress::*)()>("mapToIPv4", [](const ComboAddress& addr) { return addr.mapToIPv4(); });
  d_lw->registerFunction<void(ComboAddress::*)(unsigned int)>("truncate", [](ComboAddress& addr, unsigned int bits) { addr.truncate(bits); });
  d_lw->registerFunction<string(ComboAddress::*)()>("toString", [](const ComboAddress& addr) { return addr.toString(); });
  d_lw->registerToStringFunction<string(ComboAddress::*)()>([](const ComboAddress& addr) { return addr.toString(); });
  d_lw->registerFunction<string(ComboAddress::*)()>("toStringWithPort", [](const ComboAddress& addr) { return addr.toStringWithPort(); });
  d_lw->registerFunction<string(ComboAddress::*)()>("getRaw", [](const ComboAddress& addr) { return addr.toByteString(); });

  d_lw->writeFunction("newCA", [](const std::string& a) { return ComboAddress(a); });
  d_lw->writeFunction("newCAFromRaw", [](const std::string& raw, boost::optional<uint16_t> port) {
                                        if (raw.size() == 4) {
                                          struct sockaddr_in sin4;
                                          memset(&sin4, 0, sizeof(sin4));
                                          sin4.sin_family = AF_INET;
                                          memcpy(&sin4.sin_addr.s_addr, raw.c_str(), raw.size());
                                          if (port) {
                                            sin4.sin_port = htons(*port);
                                          }
                                          return ComboAddress(&sin4);
                                        }
                                        else if (raw.size() == 16) {
                                          struct sockaddr_in6 sin6;
                                          memset(&sin6, 0, sizeof(sin6));
                                          sin6.sin6_family = AF_INET6;
                                          memcpy(&sin6.sin6_addr.s6_addr, raw.c_str(), raw.size());
                                          if (port) {
                                            sin6.sin6_port = htons(*port);
                                          }
                                          return ComboAddress(&sin6);
                                        }
                                        return ComboAddress();
                                      });
  typedef std::unordered_set<ComboAddress,ComboAddress::addressOnlyHash,ComboAddress::addressOnlyEqual> cas_t;
  d_lw->registerFunction<bool(ComboAddress::*)(const ComboAddress&)>("equal", [](const ComboAddress& lhs, const ComboAddress& rhs) { return ComboAddress::addressOnlyEqual()(lhs, rhs); });

  // cas_t
  d_lw->writeFunction("newCAS", []{ return cas_t(); });
  d_lw->registerFunction<void(cas_t::*)(boost::variant<string,ComboAddress, vector<pair<unsigned int,string> > >)>("add",
    [](cas_t& cas, const boost::variant<string,ComboAddress,vector<pair<unsigned int,string> > >& in)
    {
      try {
      if(auto s = boost::get<string>(&in)) {
        cas.insert(ComboAddress(*s));
      }
      else if(auto v = boost::get<vector<pair<unsigned int, string> > >(&in)) {
        for(const auto& str : *v)
          cas.insert(ComboAddress(str.second));
      }
      else
        cas.insert(boost::get<ComboAddress>(in));
      }
      catch(std::exception& e) {
        SLOG(g_log <<Logger::Error<<e.what()<<endl,
             g_slog->withName("lua")->error(Logr::Error, e.what(), "Exception in newCAS", "exception", Logging::Loggable("std::exception")));
      }
    });
  d_lw->registerFunction<bool(cas_t::*)(const ComboAddress&)>("check",[](const cas_t& cas, const ComboAddress&ca) { return cas.count(ca)>0; });

  // QType
  d_lw->writeFunction("newQType", [](const string& s) { QType q; q = s; return q; });
  d_lw->registerFunction("getCode", &QType::getCode);
  d_lw->registerFunction("getName", &QType::toString);
  d_lw->registerEqFunction<bool(QType::*)(const QType&)>([](const QType& a, const QType& b){ return a == b;}); // operator overloading confuses LuaContext
  d_lw->registerToStringFunction(&QType::toString);

  // Netmask
  d_lw->writeFunction("newNetmask", [](const string& s) { return Netmask(s); });
  d_lw->registerFunction<ComboAddress(Netmask::*)()>("getNetwork", [](const Netmask& nm) { return nm.getNetwork(); } ); // const reference makes this necessary
  d_lw->registerFunction<ComboAddress(Netmask::*)()>("getMaskedNetwork", [](const Netmask& nm) { return nm.getMaskedNetwork(); } );
  d_lw->registerFunction("isIpv4", &Netmask::isIPv4);
  d_lw->registerFunction("isIPv4", &Netmask::isIPv4);
  d_lw->registerFunction("isIpv6", &Netmask::isIPv6);
  d_lw->registerFunction("isIPv6", &Netmask::isIPv6);
  d_lw->registerFunction("getBits", &Netmask::getBits);
  d_lw->registerFunction("toString", &Netmask::toString);
  d_lw->registerFunction("empty", &Netmask::empty);
  d_lw->registerFunction("match", (bool (Netmask::*)(const string&) const)&Netmask::match);
  d_lw->registerEqFunction(&Netmask::operator==);
  d_lw->registerToStringFunction(&Netmask::toString);

  // NetmaskGroup
  d_lw->writeFunction("newNMG", [](boost::optional<vector<pair<unsigned int, std::string>>> masks) {
    auto nmg = NetmaskGroup();

    if (masks) {
      for(const auto& mask: *masks) {
        nmg.addMask(mask.second);
      }
    }

    return nmg;
  });
  // d_lw->writeFunction("newNMG", []() { return NetmaskGroup(); });
  d_lw->registerFunction<void(NetmaskGroup::*)(const std::string&mask)>("addMask", [](NetmaskGroup&nmg, const std::string& mask) { nmg.addMask(mask); });
  d_lw->registerFunction<void(NetmaskGroup::*)(const vector<pair<unsigned int, std::string>>&)>("addMasks", [](NetmaskGroup&nmg, const vector<pair<unsigned int, std::string>>& masks) { for(const auto& mask: masks) { nmg.addMask(mask.second); } });
  d_lw->registerFunction("match", (bool (NetmaskGroup::*)(const ComboAddress&) const)&NetmaskGroup::match);

  // DNSRecord
  d_lw->writeFunction("newDR", [](const DNSName& name, const std::string& type, unsigned int ttl, const std::string& content, int place) { QType qtype; qtype = type; auto dr = DNSRecord(); dr.d_name = name; dr.d_type = qtype.getCode(); dr.d_ttl = ttl; dr.setContent(shared_ptr<DNSRecordContent>(DNSRecordContent::make(dr.d_type, QClass::IN, content))); dr.d_place = static_cast<DNSResourceRecord::Place>(place); return dr; });
  d_lw->registerMember("name", &DNSRecord::d_name);
  d_lw->registerMember("type", &DNSRecord::d_type);
  d_lw->registerMember("ttl", &DNSRecord::d_ttl);
  d_lw->registerMember("place", &DNSRecord::d_place);
  d_lw->registerFunction<string(DNSRecord::*)()>("getContent", [](const DNSRecord& dr) { return dr.getContent()->getZoneRepresentation(); });
  d_lw->registerFunction<boost::optional<ComboAddress>(DNSRecord::*)()>("getCA", [](const DNSRecord& dr) {
      boost::optional<ComboAddress> ret;

      if(auto arec = getRR<ARecordContent>(dr))
        ret=arec->getCA(53);
      else if(auto aaaarec = getRR<AAAARecordContent>(dr))
        ret=aaaarec->getCA(53);
      return ret;
    });
  d_lw->registerFunction<void (DNSRecord::*)(const std::string&)>("changeContent", [](DNSRecord& dr, const std::string& newContent) { dr.setContent(shared_ptr<DNSRecordContent>(DNSRecordContent::make(dr.d_type, 1, newContent))); });

  // pdnslog
#if defined(PDNS_AUTH)
  d_lw->writeFunction("pdnslog", [](const std::string& msg, boost::optional<int> loglevel) {
    if (g_slogStructured) {
      auto log = g_slog->withName("lua");
      log->info(static_cast<Logr::Priority>(loglevel.get_value_or(Logr::Warning)), msg);
    }
    else {
      g_log << (Logger::Urgency)loglevel.get_value_or(Logger::Warning)<<msg<<endl;
    }
  });
#elif defined(RECURSOR)
  d_lw->writeFunction("pdnslog", [](const std::string& msg, boost::optional<int> loglevel, boost::optional<std::map<std::string, std::string>> values) {
    auto log = g_slog->withName("lua");
    if (values) {
      for (const auto& [key, value] : *values) {
        log = log->withValues(key, Logging::Loggable(value));
      }
    }
    log->info(static_cast<Logr::Priority>(loglevel.get_value_or(Logr::Warning)), msg);
  });
#else /* DNSDIST */
  d_lw->writeFunction("pdnslog", [](const std::string& msg, boost::optional<int> loglevel) {
    g_log << (Logger::Urgency)loglevel.get_value_or(Logger::Warning)<<msg<<endl;
  });
#endif

  d_lw->writeFunction("pdnsrandom", [](boost::optional<uint32_t> maximum) {
    return maximum ? dns_random(*maximum) : dns_random_uint32();
  });

  // certain constants

  vector<pair<string, int> > rcodes = {{"NOERROR",  RCode::NoError  },
                                       {"FORMERR",  RCode::FormErr  },
                                       {"SERVFAIL", RCode::ServFail },
                                       {"NXDOMAIN", RCode::NXDomain },
                                       {"NOTIMP",   RCode::NotImp   },
                                       {"REFUSED",  RCode::Refused  },
                                       {"YXDOMAIN", RCode::YXDomain },
                                       {"YXRRSET",  RCode::YXRRSet  },
                                       {"NXRRSET",  RCode::NXRRSet  },
                                       {"NOTAUTH",  RCode::NotAuth  },
                                       {"NOTZONE",  RCode::NotZone  },
                                       {"DROP",    -2               }}; // To give backport-incompatibility warning
  for(const auto& rcode : rcodes)
    d_pd.push_back({rcode.first, rcode.second});

  d_pd.push_back({"place", in_t{
    {"QUESTION", 0},
    {"ANSWER", 1},
    {"AUTHORITY", 2},
    {"ADDITIONAL", 3}
  }});

  d_pd.push_back({"loglevels", in_t{
        {"Alert", LOG_ALERT},
        {"Critical", LOG_CRIT},
        {"Debug", LOG_DEBUG},
        {"Emergency", LOG_EMERG},
        {"Info", LOG_INFO},
        {"Notice", LOG_NOTICE},
        {"Warning", LOG_WARNING},
        {"Error", LOG_ERR}
          }});

  for(const auto& n : QType::names)
    d_pd.push_back({n.first, n.second});

  d_lw->registerMember("tv_sec", &timeval::tv_sec);
  d_lw->registerMember("tv_usec", &timeval::tv_usec);

  postPrepareContext();

  // so we can let postprepare do changes to this
  d_lw->writeVariable("pdns", d_pd);
}

void BaseLua4::loadStream(std::istream &stream, bool doPostLoad) {
  d_lw->executeCode(stream);

  if (doPostLoad) {
    postLoad();
  }
}

BaseLua4::~BaseLua4() = default;

1	#include "config.h"
2	#include <cassert>
3	#include <fstream>
4	#include <unordered_set>
5	#include <unordered_map>
6	#include <typeinfo>
7	#include <sys/stat.h>
8	#include "logger.hh"
9	#include "logging.hh"
10	#include "iputils.hh"
11	#include "dnsname.hh"
12	#include "dnsparser.hh"
13	#include "dnspacket.hh"
14	#include "namespaces.hh"
15	#include "ednssubnet.hh"
16	#include "lua-base4.hh"
17	#include "ext/luawrapper/include/LuaContext.hpp"
18	#include "dns_random.hh"
19
20	void BaseLua4::loadFile(const std::string& fname, bool doPostLoad)
21	{	451✔
22	std::ifstream ifs(fname);	451✔
23	if (!ifs) {	451!
24	auto ret = errno;	×
25	auto msg = stringerror(ret);	×
26	SLOG(g_log << Logger::Error << "Unable to read configuration file from '" << fname << "': " << msg << endl,	×
27	g_slog->withName("lua")->error(Logr::Error, msg, "Unable to read configuration file", "file", Logging::Loggable(fname)));	×
28	throw std::runtime_error(msg);	×
29	}	×
30	loadStream(ifs, doPostLoad);	451✔
31	};	451✔
32
33	void BaseLua4::loadString(const std::string &script) {	2✔
34	std::istringstream iss(script);	2✔
35	loadStream(iss, true);	2✔
36	};	2✔
37
38	void BaseLua4::includePath(const std::string& directory) {	×
39	std::vector<std::string> vec;	×
40	const std::string& suffix = "lua";	×
41	auto directoryError = pdns::visit_directory(directory, [this, &directory, &suffix, &vec]([[maybe_unused]] ino_t inodeNumber, const std::string_view& name) {	×
42	(void)this;	×
43	if (boost::starts_with(name, ".")) {	×
44	return true; // skip any dots	×
45	}	×
46	if (boost::ends_with(name, suffix)) {	×
47	// build name
48	string fullName = directory + "/" + std::string(name);	×
49	// ensure it's readable file
50	struct stat statInfo	×
51	{	×
52	};	×
53	if (stat(fullName.c_str(), &statInfo) != 0 \|\| !S_ISREG(statInfo.st_mode)) {	×
54	string msg = fullName + " is not a regular file";	×
55	SLOG(g_log << Logger::Error << msg << std::endl,	×
56	g_slog->withName("lua")->info(Logr::Error, "include file is not a regular file", "file", Logging::Loggable(fullName)));	×
57	throw PDNSException(std::move(msg));	×
58	}	×
59	vec.emplace_back(fullName);	×
60	}	×
61	return true;	×
62	});	×
63
64	if (directoryError) {	×
65	int err = errno;	×
66	string msg = directory + " is not accessible: " + stringerror(err);	×
67	SLOG(g_log << Logger::Error << msg << std::endl,	×
68	g_slog->withName("lua")->error(Logr::Error, err, "Error trying to walk directory", "directory", Logging::Loggable(directory)));	×
69	throw PDNSException(std::move(msg));	×
70	}	×
71
72	std::sort(vec.begin(), vec.end(), CIStringComparePOSIX());	×
73
74	for(const auto& file: vec) {	×
75	loadFile(file, false);	×
76	}	×
77	};	×
78
79	// By default no features
80	void BaseLua4::getFeatures(Features &) { }	474✔
81
82	void BaseLua4::prepareContext() {	474✔
83	d_lw = std::make_unique<LuaContext>();	474✔
84
85	// lua features available
86	Features features;	474✔
87	getFeatures(features);	474✔
88	d_lw->writeVariable("pdns_features", features);	474✔
89
90	// dnsheader
91	d_lw->registerFunction<int(dnsheader::*)()>("getID", [](dnsheader& dh) { return ntohs(dh.id); });	474✔
92	d_lw->registerFunction<bool(dnsheader::*)()>("getCD", [](dnsheader& dh) { return dh.cd; });	474✔
93	d_lw->registerFunction<bool(dnsheader::*)()>("getTC", [](dnsheader& dh) { return dh.tc; });	474✔
94	d_lw->registerFunction<bool(dnsheader::*)()>("getRA", [](dnsheader& dh) { return dh.ra; });	474✔
95	d_lw->registerFunction<bool(dnsheader::*)()>("getAD", [](dnsheader& dh) { return dh.ad; });	474✔
96	d_lw->registerFunction<bool(dnsheader::*)()>("getAA", [](dnsheader& dh) { return dh.aa; });	474✔
97	d_lw->registerFunction<bool(dnsheader::*)()>("getRD", [](dnsheader& dh) { return dh.rd; });	474✔
98	d_lw->registerFunction<int(dnsheader::*)()>("getRCODE", [](dnsheader& dh) { return dh.rcode; });	474✔
99	d_lw->registerFunction<int(dnsheader::*)()>("getOPCODE", [](dnsheader& dh) { return dh.opcode; });	474✔
100	d_lw->registerFunction<int(dnsheader::*)()>("getQDCOUNT", [](dnsheader& dh) { return ntohs(dh.qdcount); });	474✔
101	d_lw->registerFunction<int(dnsheader::*)()>("getANCOUNT", [](dnsheader& dh) { return ntohs(dh.ancount); });	474✔
102	d_lw->registerFunction<int(dnsheader::*)()>("getNSCOUNT", [](dnsheader& dh) { return ntohs(dh.nscount); });	474✔
103	d_lw->registerFunction<int(dnsheader::*)()>("getARCOUNT", [](dnsheader& dh) { return ntohs(dh.arcount); });	474✔
104
105	// DNSName
106	d_lw->writeFunction("newDN", [](const std::string& dom){ return DNSName(dom); });	97,631✔
107	d_lw->registerFunction("__lt", &DNSName::operator<);	474✔
108	d_lw->registerFunction<bool(DNSName::*)(const DNSName&)>("canonCompare", [](const DNSName& name, const DNSName& rhs) { return name.canonCompare(rhs); });	696✔
109	d_lw->registerFunction<DNSName(DNSName::*)(const DNSName&)>("makeRelative", [](const DNSName& name, const DNSName& zone) { return name.makeRelative(zone); });	504✔
110	d_lw->registerFunction<bool(DNSName::*)(const DNSName&)>("isPartOf", [](const DNSName& name, const DNSName& rhs) { return name.isPartOf(rhs); });	565✔
111	d_lw->registerFunction("getRawLabels", &DNSName::getRawLabels);	474✔
112	d_lw->registerFunction<unsigned int(DNSName::*)()>("countLabels", [](const DNSName& name) { return name.countLabels(); });	474✔
113	d_lw->registerFunction<size_t(DNSName::*)()>("wireLength", [](const DNSName& name) { return name.wirelength(); });	474✔
114	d_lw->registerFunction<size_t(DNSName::*)()>("wirelength", [](const DNSName& name) { return name.wirelength(); });	474✔
115	d_lw->registerFunction<bool(DNSName::*)(const std::string&)>("equal", [](const DNSName& lhs, const std::string& rhs) { return lhs==DNSName(rhs); });	474✔
116	d_lw->registerEqFunction(&DNSName::operator==);	474✔
117	d_lw->registerToStringFunction<string(DNSName::*)()>([](const DNSName&dn ) { return dn.toString(); });	474✔
118	d_lw->registerFunction<string(DNSName::*)()>("toString", [](const DNSName&dn ) { return dn.toString(); });	474✔
119	d_lw->registerFunction<string(DNSName::*)()>("toStringNoDot", [](const DNSName&dn ) { return dn.toStringNoDot(); });	474✔
120	d_lw->registerFunction<bool(DNSName::*)()>("chopOff", [](DNSName&dn ) { return dn.chopOff(); });	474✔
121
122	// DNSResourceRecord
123	d_lw->writeFunction("newDRR", [](const DNSName& qname, const string& qtype, const unsigned int ttl, const string& content, boost::optional<int> domain_id, boost::optional<int> auth){	474✔
124	auto drr = DNSResourceRecord();	×
125	drr.qname = qname;	×
126	drr.qtype = qtype;	×
127	drr.ttl = ttl;	×
128	drr.setContent(content);	×
129	if (domain_id)	×
130	drr.domain_id = *domain_id;	×
131	if (auth)	×
132	drr.auth = *auth;	×
133	return drr;	×
134	});	×
135	d_lw->registerEqFunction(&DNSResourceRecord::operator==);	474✔
136	d_lw->registerFunction("__lt", &DNSResourceRecord::operator<);	474✔
137	d_lw->registerToStringFunction<string(DNSResourceRecord::*)()>([](const DNSResourceRecord& rec) { return rec.getZoneRepresentation(); });	474✔
138	d_lw->registerFunction<string(DNSResourceRecord::*)()>("toString", [](const DNSResourceRecord& rec) { return rec.getZoneRepresentation();} );	474✔
139	d_lw->registerFunction<DNSName(DNSResourceRecord::*)()>("qname", [](DNSResourceRecord& rec) { return rec.qname; });	474✔
140	d_lw->registerFunction<DNSName(DNSResourceRecord::*)()>("wildcardName", [](DNSResourceRecord& rec) { return rec.wildcardname; });	474✔
141	d_lw->registerFunction<string(DNSResourceRecord::*)()>("content", [](DNSResourceRecord& rec) { return rec.content; });	474✔
142	d_lw->registerFunction<time_t(DNSResourceRecord::*)()>("lastModified", [](DNSResourceRecord& rec) { return rec.last_modified; });	474✔
143	d_lw->registerFunction<uint32_t(DNSResourceRecord::*)()>("ttl", [](DNSResourceRecord& rec) { return rec.ttl; });	474✔
144	d_lw->registerFunction<uint32_t(DNSResourceRecord::*)()>("signttl", [](DNSResourceRecord& rec) { return rec.signttl; });	474✔
145	d_lw->registerFunction<int(DNSResourceRecord::*)()>("domainId", [](DNSResourceRecord& rec) { return rec.domain_id; });	474✔
146	d_lw->registerFunction<uint16_t(DNSResourceRecord::*)()>("qtype", [](DNSResourceRecord& rec) { return rec.qtype.getCode(); });	474✔
147	d_lw->registerFunction<uint16_t(DNSResourceRecord::*)()>("qclass", [](DNSResourceRecord& rec) { return rec.qclass; });	474✔
148	d_lw->registerFunction<uint8_t(DNSResourceRecord::*)()>("scopeMask", [](DNSResourceRecord& rec) { return rec.scopeMask; });	474✔
149	d_lw->registerFunction<bool(DNSResourceRecord::*)()>("auth", [](DNSResourceRecord& rec) { return rec.auth; });	474✔
150	d_lw->registerFunction<bool(DNSResourceRecord::*)()>("disabled", [](DNSResourceRecord& rec) { return rec.disabled; });	474✔
151
152	// ComboAddress
153	d_lw->registerFunction<bool(ComboAddress::*)()>("isIPv4", [](const ComboAddress& addr) { return addr.sin4.sin_family == AF_INET; });	474✔
154	d_lw->registerFunction<bool(ComboAddress::*)()>("isIPv6", [](const ComboAddress& addr) { return addr.sin4.sin_family == AF_INET6; });	474✔
155	d_lw->registerFunction<uint16_t(ComboAddress::*)()>("getPort", [](const ComboAddress& addr) { return ntohs(addr.sin4.sin_port); } );	474✔
156	d_lw->registerFunction<bool(ComboAddress::*)()>("isMappedIPv4", [](const ComboAddress& addr) { return addr.isMappedIPv4(); });	474✔
157	d_lw->registerFunction<ComboAddress(ComboAddress::*)()>("mapToIPv4", [](const ComboAddress& addr) { return addr.mapToIPv4(); });	474✔
158	d_lw->registerFunction<void(ComboAddress::*)(unsigned int)>("truncate", [](ComboAddress& addr, unsigned int bits) { addr.truncate(bits); });	474✔
159	d_lw->registerFunction<string(ComboAddress::*)()>("toString", [](const ComboAddress& addr) { return addr.toString(); });	474✔
160	d_lw->registerToStringFunction<string(ComboAddress::*)()>([](const ComboAddress& addr) { return addr.toString(); });	97,382✔
161	d_lw->registerFunction<string(ComboAddress::*)()>("toStringWithPort", [](const ComboAddress& addr) { return addr.toStringWithPort(); });	474✔
162	d_lw->registerFunction<string(ComboAddress::*)()>("getRaw", [](const ComboAddress& addr) { return addr.toByteString(); });	474✔
163
164	d_lw->writeFunction("newCA", [](const std::string& a) { return ComboAddress(a); });	474✔
165	d_lw->writeFunction("newCAFromRaw", [](const std::string& raw, boost::optional<uint16_t> port) {	474✔
166	if (raw.size() == 4) {	×
167	struct sockaddr_in sin4;	×
168	memset(&sin4, 0, sizeof(sin4));	×
169	sin4.sin_family = AF_INET;	×
170	memcpy(&sin4.sin_addr.s_addr, raw.c_str(), raw.size());	×
171	if (port) {	×
172	sin4.sin_port = htons(*port);	×
173	}	×
174	return ComboAddress(&sin4);	×
175	}	×
176	else if (raw.size() == 16) {	×
177	struct sockaddr_in6 sin6;	×
178	memset(&sin6, 0, sizeof(sin6));	×
179	sin6.sin6_family = AF_INET6;	×
180	memcpy(&sin6.sin6_addr.s6_addr, raw.c_str(), raw.size());	×
181	if (port) {	×
182	sin6.sin6_port = htons(*port);	×
183	}	×
184	return ComboAddress(&sin6);	×
185	}	×
186	return ComboAddress();	×
187	});	×
188	typedef std::unordered_set<ComboAddress,ComboAddress::addressOnlyHash,ComboAddress::addressOnlyEqual> cas_t;	474✔
189	d_lw->registerFunction<bool(ComboAddress::*)(const ComboAddress&)>("equal", [](const ComboAddress& lhs, const ComboAddress& rhs) { return ComboAddress::addressOnlyEqual()(lhs, rhs); });	474✔
190
191	// cas_t
192	d_lw->writeFunction("newCAS", []{ return cas_t(); });	474✔
193	d_lw->registerFunction<void(cas_t::*)(boost::variant<string,ComboAddress, vector<pair<unsigned int,string> > >)>("add",	474✔
194	[](cas_t& cas, const boost::variant<string,ComboAddress,vector<pair<unsigned int,string> > >& in)	474✔
195	{	474✔
196	try {	×
197	if(auto s = boost::get<string>(&in)) {	×
198	cas.insert(ComboAddress(*s));	×
199	}	×
200	else if(auto v = boost::get<vector<pair<unsigned int, string> > >(&in)) {	×
201	for(const auto& str : *v)	×
202	cas.insert(ComboAddress(str.second));	×
203	}	×
204	else	×
205	cas.insert(boost::get<ComboAddress>(in));	×
206	}	×
207	catch(std::exception& e) {	×
208	SLOG(g_log <<Logger::Error<<e.what()<<endl,	×
209	g_slog->withName("lua")->error(Logr::Error, e.what(), "Exception in newCAS", "exception", Logging::Loggable("std::exception")));	×
210	}	×
211	});	×
212	d_lw->registerFunction<bool(cas_t::*)(const ComboAddress&)>("check",[](const cas_t& cas, const ComboAddress&ca) { return cas.count(ca)>0; });	474✔
213
214	// QType
215	d_lw->writeFunction("newQType", [](const string& s) { QType q; q = s; return q; });	520✔
216	d_lw->registerFunction("getCode", &QType::getCode);	474✔
217	d_lw->registerFunction("getName", &QType::toString);	474✔
218	d_lw->registerEqFunction<bool(QType::*)(const QType&)>([](const QType& a, const QType& b){ return a == b;}); // operator overloading confuses LuaContext	474✔
219	d_lw->registerToStringFunction(&QType::toString);	474✔
220
221	// Netmask
222	d_lw->writeFunction("newNetmask", [](const string& s) { return Netmask(s); });	474✔
223	d_lw->registerFunction<ComboAddress(Netmask::*)()>("getNetwork", [](const Netmask& nm) { return nm.getNetwork(); } ); // const reference makes this necessary	474✔
224	d_lw->registerFunction<ComboAddress(Netmask::*)()>("getMaskedNetwork", [](const Netmask& nm) { return nm.getMaskedNetwork(); } );	474✔
225	d_lw->registerFunction("isIpv4", &Netmask::isIPv4);	474✔
226	d_lw->registerFunction("isIPv4", &Netmask::isIPv4);	474✔
227	d_lw->registerFunction("isIpv6", &Netmask::isIPv6);	474✔
228	d_lw->registerFunction("isIPv6", &Netmask::isIPv6);	474✔
229	d_lw->registerFunction("getBits", &Netmask::getBits);	474✔
230	d_lw->registerFunction("toString", &Netmask::toString);	474✔
231	d_lw->registerFunction("empty", &Netmask::empty);	474✔
232	d_lw->registerFunction("match", (bool (Netmask::*)(const string&) const)&Netmask::match);	474✔
233	d_lw->registerEqFunction(&Netmask::operator==);	474✔
234	d_lw->registerToStringFunction(&Netmask::toString);	474✔
235
236	// NetmaskGroup
237	d_lw->writeFunction("newNMG", [](boost::optional<vector<pair<unsigned int, std::string>>> masks) {	474✔
UNCOV 238	auto nmg = NetmaskGroup();	×
239
UNCOV 240	if (masks) {	×
241	for(const auto& mask: *masks) {	×
242	nmg.addMask(mask.second);	×
243	}	×
244	}	×
245
UNCOV 246	return nmg;	×
UNCOV 247	});	×
248	// d_lw->writeFunction("newNMG", []() { return NetmaskGroup(); });
249	d_lw->registerFunction<void(NetmaskGroup::*)(const std::string&mask)>("addMask", [](NetmaskGroup&nmg, const std::string& mask) { nmg.addMask(mask); });	474✔
250	d_lw->registerFunction<void(NetmaskGroup::*)(const vector<pair<unsigned int, std::string>>&)>("addMasks", [](NetmaskGroup&nmg, const vector<pair<unsigned int, std::string>>& masks) { for(const auto& mask: masks) { nmg.addMask(mask.second); } });	474!
251	d_lw->registerFunction("match", (bool (NetmaskGroup::*)(const ComboAddress&) const)&NetmaskGroup::match);	474✔
252
253	// DNSRecord
254	d_lw->writeFunction("newDR", [](const DNSName& name, const std::string& type, unsigned int ttl, const std::string& content, int place) { QType qtype; qtype = type; auto dr = DNSRecord(); dr.d_name = name; dr.d_type = qtype.getCode(); dr.d_ttl = ttl; dr.setContent(shared_ptr<DNSRecordContent>(DNSRecordContent::make(dr.d_type, QClass::IN, content))); dr.d_place = static_cast<DNSResourceRecord::Place>(place); return dr; });	474✔
255	d_lw->registerMember("name", &DNSRecord::d_name);	474✔
256	d_lw->registerMember("type", &DNSRecord::d_type);	474✔
257	d_lw->registerMember("ttl", &DNSRecord::d_ttl);	474✔
258	d_lw->registerMember("place", &DNSRecord::d_place);	474✔
259	d_lw->registerFunction<string(DNSRecord::*)()>("getContent", [](const DNSRecord& dr) { return dr.getContent()->getZoneRepresentation(); });	474✔
260	d_lw->registerFunction<boost::optional<ComboAddress>(DNSRecord::*)()>("getCA", [](const DNSRecord& dr) {	474✔
261	boost::optional<ComboAddress> ret;	×
262
263	if(auto arec = getRR<ARecordContent>(dr))	×
264	ret=arec->getCA(53);	×
265	else if(auto aaaarec = getRR<AAAARecordContent>(dr))	×
266	ret=aaaarec->getCA(53);	×
267	return ret;	×
268	});	×
269	d_lw->registerFunction<void (DNSRecord::*)(const std::string&)>("changeContent", [](DNSRecord& dr, const std::string& newContent) { dr.setContent(shared_ptr<DNSRecordContent>(DNSRecordContent::make(dr.d_type, 1, newContent))); });	474✔
270
271	// pdnslog
272	#if defined(PDNS_AUTH)	474✔
273	d_lw->writeFunction("pdnslog", [](const std::string& msg, boost::optional<int> loglevel) {	474✔
UNCOV 274	if (g_slogStructured) {	×
UNCOV 275	auto log = g_slog->withName("lua");
UNCOV 276	log->info(static_cast<Logr::Priority>(loglevel.get_value_or(Logr::Warning)), msg);
UNCOV 277	}
UNCOV 278	else {
UNCOV 279	g_log << (Logger::Urgency)loglevel.get_value_or(Logger::Warning)<<msg<<endl;
UNCOV 280	}
UNCOV 281	});
282	#elif defined(RECURSOR)
283	d_lw->writeFunction("pdnslog", [](const std::string& msg, boost::optional<int> loglevel, boost::optional<std::map<std::string, std::string>> values) {
284	auto log = g_slog->withName("lua");
285	if (values) {	×
286	for (const auto& [key, value] : *values) {	×
287	log = log->withValues(key, Logging::Loggable(value));
288	}
289	}
290	log->info(static_cast<Logr::Priority>(loglevel.get_value_or(Logr::Warning)), msg);
291	});
292	#else /* DNSDIST */
293	d_lw->writeFunction("pdnslog", [](const std::string& msg, boost::optional<int> loglevel) {
294	g_log << (Logger::Urgency)loglevel.get_value_or(Logger::Warning)<<msg<<endl;
295	});
296	#endif
297
298	d_lw->writeFunction("pdnsrandom", [](boost::optional<uint32_t> maximum) {	474✔
UNCOV 299	return maximum ? dns_random(*maximum) : dns_random_uint32();	×
UNCOV 300	});	×
301
302	// certain constants
303
304	vector<pair<string, int> > rcodes = {{"NOERROR", RCode::NoError },	474✔
305	{"FORMERR", RCode::FormErr },	474✔
306	{"SERVFAIL", RCode::ServFail },	474✔
307	{"NXDOMAIN", RCode::NXDomain },	474✔
308	{"NOTIMP", RCode::NotImp },	474✔
309	{"REFUSED", RCode::Refused },	474✔
310	{"YXDOMAIN", RCode::YXDomain },	474✔
311	{"YXRRSET", RCode::YXRRSet },	474✔
312	{"NXRRSET", RCode::NXRRSet },	474✔
313	{"NOTAUTH", RCode::NotAuth },	474✔
314	{"NOTZONE", RCode::NotZone },	474✔
315	{"DROP", -2 }}; // To give backport-incompatibility warning	474✔
316	for(const auto& rcode : rcodes)	474✔
317	d_pd.push_back({rcode.first, rcode.second});	5,679✔
318
319	d_pd.push_back({"place", in_t{	474✔
320	{"QUESTION", 0},	474✔
321	{"ANSWER", 1},	474✔
322	{"AUTHORITY", 2},	474✔
323	{"ADDITIONAL", 3}	474✔
324	}});	474✔
325
326	d_pd.push_back({"loglevels", in_t{	474✔
327	{"Alert", LOG_ALERT},	474✔
328	{"Critical", LOG_CRIT},	474✔
329	{"Debug", LOG_DEBUG},	474✔
330	{"Emergency", LOG_EMERG},	474✔
331	{"Info", LOG_INFO},	474✔
332	{"Notice", LOG_NOTICE},	474✔
333	{"Warning", LOG_WARNING},	474✔
334	{"Error", LOG_ERR}	474✔
335	}});	474✔
336
337	for(const auto& n : QType::names)	474✔
338	d_pd.push_back({n.first, n.second});	32,664✔
339
340	d_lw->registerMember("tv_sec", &timeval::tv_sec);	474✔
341	d_lw->registerMember("tv_usec", &timeval::tv_usec);	474✔
342
343	postPrepareContext();	474✔
344
345	// so we can let postprepare do changes to this
346	d_lw->writeVariable("pdns", d_pd);	474✔
347	}	474✔
348
349	void BaseLua4::loadStream(std::istream &stream, bool doPostLoad) {	453✔
350	d_lw->executeCode(stream);	453✔
351
352	if (doPostLoad) {	453!
353	postLoad();	453✔
354	}	453✔
355	}	453✔
356
357	BaseLua4::~BaseLua4() = default;	19✔

PowerDNS / pdns / 25787327741

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous