]> git.ipfire.org Git - thirdparty/pdns.git/blob - pdns/dnsdist-lua-bindings.cc
Merge pull request #8505 from rgacogne/dnsdist-lua-ffi
[thirdparty/pdns.git] / pdns / dnsdist-lua-bindings.cc
1 /*
2 * This file is part of PowerDNS or dnsdist.
3 * Copyright -- PowerDNS.COM B.V. and its contributors
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of version 2 of the GNU General Public License as
7 * published by the Free Software Foundation.
8 *
9 * In addition, for the avoidance of any doubt, permission is granted to
10 * link this program with OpenSSL and to (re)distribute the binaries
11 * produced as the result of such linking.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
21 */
22 #include "config.h"
23 #include "dnsdist.hh"
24 #include "dnsdist-lua.hh"
25
26 #include "dolog.hh"
27
28 void setupLuaBindings(bool client)
29 {
30 g_lua.writeFunction("infolog", [](const string& arg) {
31 infolog("%s", arg);
32 });
33 g_lua.writeFunction("errlog", [](const string& arg) {
34 errlog("%s", arg);
35 });
36 g_lua.writeFunction("warnlog", [](const string& arg) {
37 warnlog("%s", arg);
38 });
39 g_lua.writeFunction("show", [](const string& arg) {
40 g_outputBuffer+=arg;
41 g_outputBuffer+="\n";
42 });
43
44 /* Exceptions */
45 g_lua.registerFunction<string(std::exception_ptr::*)()>("__tostring", [](const std::exception_ptr& eptr) {
46 try {
47 if (eptr) {
48 std::rethrow_exception(eptr);
49 }
50 } catch(const std::exception& e) {
51 return string(e.what());
52 } catch(const PDNSException& e) {
53 return e.reason;
54 } catch(...) {
55 return string("Unknown exception");
56 }
57 return string("No exception");
58 });
59 /* ServerPolicy */
60 g_lua.writeFunction("newServerPolicy", [](string name, ServerPolicy::policyfunc_t policy) { return std::make_shared<ServerPolicy>(name, policy, true);});
61 g_lua.registerMember("name", &ServerPolicy::name);
62 g_lua.registerMember("policy", &ServerPolicy::policy);
63 g_lua.registerMember("ffipolicy", &ServerPolicy::ffipolicy);
64 g_lua.registerMember("isLua", &ServerPolicy::isLua);
65 g_lua.registerMember("isFFI", &ServerPolicy::isFFI);
66 g_lua.registerFunction("toString", &ServerPolicy::toString);
67
68 g_lua.writeVariable("firstAvailable", ServerPolicy{"firstAvailable", firstAvailable, false});
69 g_lua.writeVariable("roundrobin", ServerPolicy{"roundrobin", roundrobin, false});
70 g_lua.writeVariable("wrandom", ServerPolicy{"wrandom", wrandom, false});
71 g_lua.writeVariable("whashed", ServerPolicy{"whashed", whashed, false});
72 g_lua.writeVariable("chashed", ServerPolicy{"chashed", chashed, false});
73 g_lua.writeVariable("leastOutstanding", ServerPolicy{"leastOutstanding", leastOutstanding, false});
74
75 /* ServerPool */
76 g_lua.registerFunction<void(std::shared_ptr<ServerPool>::*)(std::shared_ptr<DNSDistPacketCache>)>("setCache", [](std::shared_ptr<ServerPool> pool, std::shared_ptr<DNSDistPacketCache> cache) {
77 if (pool) {
78 pool->packetCache = cache;
79 }
80 });
81 g_lua.registerFunction("getCache", &ServerPool::getCache);
82 g_lua.registerFunction<void(std::shared_ptr<ServerPool>::*)()>("unsetCache", [](std::shared_ptr<ServerPool> pool) {
83 if (pool) {
84 pool->packetCache = nullptr;
85 }
86 });
87 g_lua.registerFunction("getECS", &ServerPool::getECS);
88 g_lua.registerFunction("setECS", &ServerPool::setECS);
89
90 /* DownstreamState */
91 g_lua.registerFunction<void(DownstreamState::*)(int)>("setQPS", [](DownstreamState& s, int lim) { s.qps = lim ? QPSLimiter(lim, lim) : QPSLimiter(); });
92 g_lua.registerFunction<void(std::shared_ptr<DownstreamState>::*)(string)>("addPool", [](std::shared_ptr<DownstreamState> s, string pool) {
93 auto localPools = g_pools.getCopy();
94 addServerToPool(localPools, pool, s);
95 g_pools.setState(localPools);
96 s->pools.insert(pool);
97 });
98 g_lua.registerFunction<void(std::shared_ptr<DownstreamState>::*)(string)>("rmPool", [](std::shared_ptr<DownstreamState> s, string pool) {
99 auto localPools = g_pools.getCopy();
100 removeServerFromPool(localPools, pool, s);
101 g_pools.setState(localPools);
102 s->pools.erase(pool);
103 });
104 g_lua.registerFunction<uint64_t(DownstreamState::*)()>("getOutstanding", [](const DownstreamState& s) { return s.outstanding.load(); });
105 g_lua.registerFunction("isUp", &DownstreamState::isUp);
106 g_lua.registerFunction("setDown", &DownstreamState::setDown);
107 g_lua.registerFunction("setUp", &DownstreamState::setUp);
108 g_lua.registerFunction<void(DownstreamState::*)(boost::optional<bool> newStatus)>("setAuto", [](DownstreamState& s, boost::optional<bool> newStatus) {
109 if (newStatus) {
110 s.upStatus = *newStatus;
111 }
112 s.setAuto();
113 });
114 g_lua.registerFunction("getName", &DownstreamState::getName);
115 g_lua.registerFunction("getNameWithAddr", &DownstreamState::getNameWithAddr);
116 g_lua.registerMember("upStatus", &DownstreamState::upStatus);
117 g_lua.registerMember<int (DownstreamState::*)>("weight",
118 [](const DownstreamState& s) -> int {return s.weight;},
119 [](DownstreamState& s, int newWeight) {s.setWeight(newWeight);}
120 );
121 g_lua.registerMember("order", &DownstreamState::order);
122 g_lua.registerMember<const std::string(DownstreamState::*)>("name", [](const DownstreamState& backend) -> const std::string { return backend.getName(); }, [](DownstreamState& backend, const std::string& newName) { backend.setName(newName); });
123 g_lua.registerFunction<std::string(DownstreamState::*)()>("getID", [](const DownstreamState& s) { return boost::uuids::to_string(s.id); });
124
125 /* dnsheader */
126 g_lua.registerFunction<void(dnsheader::*)(bool)>("setRD", [](dnsheader& dh, bool v) {
127 dh.rd=v;
128 });
129
130 g_lua.registerFunction<bool(dnsheader::*)()>("getRD", [](dnsheader& dh) {
131 return (bool)dh.rd;
132 });
133
134 g_lua.registerFunction<void(dnsheader::*)(bool)>("setRA", [](dnsheader& dh, bool v) {
135 dh.ra=v;
136 });
137
138 g_lua.registerFunction<bool(dnsheader::*)()>("getRA", [](dnsheader& dh) {
139 return (bool)dh.ra;
140 });
141
142 g_lua.registerFunction<void(dnsheader::*)(bool)>("setAD", [](dnsheader& dh, bool v) {
143 dh.ad=v;
144 });
145
146 g_lua.registerFunction<bool(dnsheader::*)()>("getAD", [](dnsheader& dh) {
147 return (bool)dh.ad;
148 });
149
150 g_lua.registerFunction<void(dnsheader::*)(bool)>("setAA", [](dnsheader& dh, bool v) {
151 dh.aa=v;
152 });
153
154 g_lua.registerFunction<bool(dnsheader::*)()>("getAA", [](dnsheader& dh) {
155 return (bool)dh.aa;
156 });
157
158 g_lua.registerFunction<void(dnsheader::*)(bool)>("setCD", [](dnsheader& dh, bool v) {
159 dh.cd=v;
160 });
161
162 g_lua.registerFunction<bool(dnsheader::*)()>("getCD", [](dnsheader& dh) {
163 return (bool)dh.cd;
164 });
165
166 g_lua.registerFunction<void(dnsheader::*)(bool)>("setTC", [](dnsheader& dh, bool v) {
167 dh.tc=v;
168 if(v) dh.ra = dh.rd; // you'll always need this, otherwise TC=1 gets ignored
169 });
170
171 g_lua.registerFunction<void(dnsheader::*)(bool)>("setQR", [](dnsheader& dh, bool v) {
172 dh.qr=v;
173 });
174
175 /* ComboAddress */
176 g_lua.writeFunction("newCA", [](const std::string& name) { return ComboAddress(name); });
177 g_lua.writeFunction("newCAFromRaw", [](const std::string& raw, boost::optional<uint16_t> port) {
178 if (raw.size() == 4) {
179 struct sockaddr_in sin4;
180 memset(&sin4, 0, sizeof(sin4));
181 sin4.sin_family = AF_INET;
182 memcpy(&sin4.sin_addr.s_addr, raw.c_str(), raw.size());
183 if (port) {
184 sin4.sin_port = htons(*port);
185 }
186 return ComboAddress(&sin4);
187 }
188 else if (raw.size() == 16) {
189 struct sockaddr_in6 sin6;
190 memset(&sin6, 0, sizeof(sin6));
191 sin6.sin6_family = AF_INET6;
192 memcpy(&sin6.sin6_addr.s6_addr, raw.c_str(), raw.size());
193 if (port) {
194 sin6.sin6_port = htons(*port);
195 }
196 return ComboAddress(&sin6);
197 }
198 return ComboAddress();
199 });
200 g_lua.registerFunction<string(ComboAddress::*)()>("tostring", [](const ComboAddress& ca) { return ca.toString(); });
201 g_lua.registerFunction<string(ComboAddress::*)()>("tostringWithPort", [](const ComboAddress& ca) { return ca.toStringWithPort(); });
202 g_lua.registerFunction<string(ComboAddress::*)()>("toString", [](const ComboAddress& ca) { return ca.toString(); });
203 g_lua.registerFunction<string(ComboAddress::*)()>("toStringWithPort", [](const ComboAddress& ca) { return ca.toStringWithPort(); });
204 g_lua.registerFunction<uint16_t(ComboAddress::*)()>("getPort", [](const ComboAddress& ca) { return ntohs(ca.sin4.sin_port); } );
205 g_lua.registerFunction<void(ComboAddress::*)(unsigned int)>("truncate", [](ComboAddress& ca, unsigned int bits) { ca.truncate(bits); });
206 g_lua.registerFunction<bool(ComboAddress::*)()>("isIPv4", [](const ComboAddress& ca) { return ca.sin4.sin_family == AF_INET; });
207 g_lua.registerFunction<bool(ComboAddress::*)()>("isIPv6", [](const ComboAddress& ca) { return ca.sin4.sin_family == AF_INET6; });
208 g_lua.registerFunction<bool(ComboAddress::*)()>("isMappedIPv4", [](const ComboAddress& ca) { return ca.isMappedIPv4(); });
209 g_lua.registerFunction<ComboAddress(ComboAddress::*)()>("mapToIPv4", [](const ComboAddress& ca) { return ca.mapToIPv4(); });
210 g_lua.registerFunction<bool(nmts_t::*)(const ComboAddress&)>("match", [](nmts_t& s, const ComboAddress& ca) { return s.match(ca); });
211
212 /* DNSName */
213 g_lua.registerFunction("isPartOf", &DNSName::isPartOf);
214 g_lua.registerFunction<bool(DNSName::*)()>("chopOff", [](DNSName&dn ) { return dn.chopOff(); });
215 g_lua.registerFunction<unsigned int(DNSName::*)()>("countLabels", [](const DNSName& name) { return name.countLabels(); });
216 g_lua.registerFunction<size_t(DNSName::*)()>("hash", [](const DNSName& name) { return name.hash(); });
217 g_lua.registerFunction<size_t(DNSName::*)()>("wirelength", [](const DNSName& name) { return name.wirelength(); });
218 g_lua.registerFunction<string(DNSName::*)()>("tostring", [](const DNSName&dn ) { return dn.toString(); });
219 g_lua.registerFunction<string(DNSName::*)()>("toString", [](const DNSName&dn ) { return dn.toString(); });
220 g_lua.writeFunction("newDNSName", [](const std::string& name) { return DNSName(name); });
221 g_lua.writeFunction("newDNSNameFromRaw", [](const std::string& name) { return DNSName(name.c_str(), name.size(), 0, false); });
222 g_lua.writeFunction("newSuffixMatchNode", []() { return SuffixMatchNode(); });
223 g_lua.writeFunction("newDNSNameSet", []() { return DNSNameSet(); });
224
225 /* DNSNameSet */
226 g_lua.registerFunction<string(DNSNameSet::*)()>("toString", [](const DNSNameSet&dns ) { return dns.toString(); });
227 g_lua.registerFunction<void(DNSNameSet::*)(DNSName&)>("add", [](DNSNameSet& dns, DNSName& dn) { dns.insert(dn); });
228 g_lua.registerFunction<bool(DNSNameSet::*)(DNSName&)>("check", [](DNSNameSet& dns, DNSName& dn) { return dns.find(dn) != dns.end(); });
229 g_lua.registerFunction("delete",(size_t (DNSNameSet::*)(const DNSName&)) &DNSNameSet::erase);
230 g_lua.registerFunction("size",(size_t (DNSNameSet::*)() const) &DNSNameSet::size);
231 g_lua.registerFunction("clear",(void (DNSNameSet::*)()) &DNSNameSet::clear);
232 g_lua.registerFunction("empty",(bool (DNSNameSet::*)()) &DNSNameSet::empty);
233
234 /* SuffixMatchNode */
235 g_lua.registerFunction<void (SuffixMatchNode::*)(const boost::variant<DNSName, string, vector<pair<int, DNSName>>, vector<pair<int, string>>> &name)>("add", [](SuffixMatchNode &smn, const boost::variant<DNSName, string, vector<pair<int, DNSName>>, vector<pair<int, string>>> &name) {
236 if (name.type() == typeid(DNSName)) {
237 auto n = boost::get<DNSName>(name);
238 smn.add(n);
239 return;
240 }
241 if (name.type() == typeid(string)) {
242 auto n = boost::get<string>(name);
243 smn.add(n);
244 return;
245 }
246 if (name.type() == typeid(vector<pair<int, DNSName>>)) {
247 auto names = boost::get<vector<pair<int, DNSName>>>(name);
248 for (auto const n : names) {
249 smn.add(n.second);
250 }
251 return;
252 }
253 if (name.type() == typeid(vector<pair<int, string>>)) {
254 auto names = boost::get<vector<pair<int, string>>>(name);
255 for (auto const n : names) {
256 smn.add(n.second);
257 }
258 return;
259 }
260 });
261 g_lua.registerFunction("check",(bool (SuffixMatchNode::*)(const DNSName&) const) &SuffixMatchNode::check);
262
263 /* Netmask */
264 g_lua.writeFunction("newNetmask", [](boost::variant<std::string,ComboAddress> s, boost::optional<uint8_t> bits) {
265 if (s.type() == typeid(ComboAddress)) {
266 auto ca = boost::get<ComboAddress>(s);
267 if (bits) {
268 return Netmask(ca, *bits);
269 }
270 return Netmask(ca);
271 }
272 else if (s.type() == typeid(std::string)) {
273 auto str = boost::get<std::string>(s);
274 return Netmask(str);
275 }
276 throw std::runtime_error("Invalid parameter passed to 'newNetmask()'");
277 });
278 g_lua.registerFunction("empty", &Netmask::empty);
279 g_lua.registerFunction("getBits", &Netmask::getBits);
280 g_lua.registerFunction<ComboAddress(Netmask::*)()>("getNetwork", [](const Netmask& nm) { return nm.getNetwork(); } ); // const reference makes this necessary
281 g_lua.registerFunction<ComboAddress(Netmask::*)()>("getMaskedNetwork", [](const Netmask& nm) { return nm.getMaskedNetwork(); } );
282 g_lua.registerFunction("isIpv4", &Netmask::isIPv4);
283 g_lua.registerFunction("isIPv4", &Netmask::isIPv4);
284 g_lua.registerFunction("isIpv6", &Netmask::isIPv6);
285 g_lua.registerFunction("isIPv6", &Netmask::isIPv6);
286 g_lua.registerFunction("match", (bool (Netmask::*)(const string&) const)&Netmask::match);
287 g_lua.registerFunction("toString", &Netmask::toString);
288 g_lua.registerEqFunction(&Netmask::operator==);
289 g_lua.registerToStringFunction(&Netmask::toString);
290
291 /* NetmaskGroup */
292 g_lua.writeFunction("newNMG", []() { return NetmaskGroup(); });
293 g_lua.registerFunction<void(NetmaskGroup::*)(const std::string&mask)>("addMask", [](NetmaskGroup&nmg, const std::string& mask)
294 {
295 nmg.addMask(mask);
296 });
297 g_lua.registerFunction<void(NetmaskGroup::*)(const std::map<ComboAddress,int>& map)>("addMasks", [](NetmaskGroup&nmg, const std::map<ComboAddress,int>& map)
298 {
299 for (const auto& entry : map) {
300 nmg.addMask(Netmask(entry.first));
301 }
302 });
303
304 g_lua.registerFunction("match", (bool (NetmaskGroup::*)(const ComboAddress&) const)&NetmaskGroup::match);
305 g_lua.registerFunction("size", &NetmaskGroup::size);
306 g_lua.registerFunction("clear", &NetmaskGroup::clear);
307 g_lua.registerFunction<string(NetmaskGroup::*)()>("toString", [](const NetmaskGroup& nmg ) { return "NetmaskGroup " + nmg.toString(); });
308
309 /* QPSLimiter */
310 g_lua.writeFunction("newQPSLimiter", [](int rate, int burst) { return QPSLimiter(rate, burst); });
311 g_lua.registerFunction("check", &QPSLimiter::check);
312
313 /* ClientState */
314 g_lua.registerFunction<std::string(ClientState::*)()>("toString", [](const ClientState& fe) {
315 setLuaNoSideEffect();
316 return fe.local.toStringWithPort();
317 });
318 g_lua.registerMember("muted", &ClientState::muted);
319 #ifdef HAVE_EBPF
320 g_lua.registerFunction<void(ClientState::*)(std::shared_ptr<BPFFilter>)>("attachFilter", [](ClientState& frontend, std::shared_ptr<BPFFilter> bpf) {
321 if (bpf) {
322 frontend.attachFilter(bpf);
323 }
324 });
325 g_lua.registerFunction<void(ClientState::*)()>("detachFilter", [](ClientState& frontend) {
326 frontend.detachFilter();
327 });
328 #endif /* HAVE_EBPF */
329
330 /* BPF Filter */
331 #ifdef HAVE_EBPF
332 g_lua.writeFunction("newBPFFilter", [client](uint32_t maxV4, uint32_t maxV6, uint32_t maxQNames) {
333 if (client) {
334 return std::shared_ptr<BPFFilter>(nullptr);
335 }
336 return std::make_shared<BPFFilter>(maxV4, maxV6, maxQNames);
337 });
338
339 g_lua.registerFunction<void(std::shared_ptr<BPFFilter>::*)(const ComboAddress& ca)>("block", [](std::shared_ptr<BPFFilter> bpf, const ComboAddress& ca) {
340 if (bpf) {
341 return bpf->block(ca);
342 }
343 });
344
345 g_lua.registerFunction<void(std::shared_ptr<BPFFilter>::*)(const DNSName& qname, boost::optional<uint16_t> qtype)>("blockQName", [](std::shared_ptr<BPFFilter> bpf, const DNSName& qname, boost::optional<uint16_t> qtype) {
346 if (bpf) {
347 return bpf->block(qname, qtype ? *qtype : 255);
348 }
349 });
350
351 g_lua.registerFunction<void(std::shared_ptr<BPFFilter>::*)(const ComboAddress& ca)>("unblock", [](std::shared_ptr<BPFFilter> bpf, const ComboAddress& ca) {
352 if (bpf) {
353 return bpf->unblock(ca);
354 }
355 });
356
357 g_lua.registerFunction<void(std::shared_ptr<BPFFilter>::*)(const DNSName& qname, boost::optional<uint16_t> qtype)>("unblockQName", [](std::shared_ptr<BPFFilter> bpf, const DNSName& qname, boost::optional<uint16_t> qtype) {
358 if (bpf) {
359 return bpf->unblock(qname, qtype ? *qtype : 255);
360 }
361 });
362
363 g_lua.registerFunction<std::string(std::shared_ptr<BPFFilter>::*)()>("getStats", [](const std::shared_ptr<BPFFilter> bpf) {
364 setLuaNoSideEffect();
365 std::string res;
366 if (bpf) {
367 std::vector<std::pair<ComboAddress, uint64_t> > stats = bpf->getAddrStats();
368 for (const auto& value : stats) {
369 if (value.first.sin4.sin_family == AF_INET) {
370 res += value.first.toString() + ": " + std::to_string(value.second) + "\n";
371 }
372 else if (value.first.sin4.sin_family == AF_INET6) {
373 res += "[" + value.first.toString() + "]: " + std::to_string(value.second) + "\n";
374 }
375 }
376 std::vector<std::tuple<DNSName, uint16_t, uint64_t> > qstats = bpf->getQNameStats();
377 for (const auto& value : qstats) {
378 res += std::get<0>(value).toString() + " " + std::to_string(std::get<1>(value)) + ": " + std::to_string(std::get<2>(value)) + "\n";
379 }
380 }
381 return res;
382 });
383
384 g_lua.registerFunction<void(std::shared_ptr<BPFFilter>::*)()>("attachToAllBinds", [](std::shared_ptr<BPFFilter> bpf) {
385 std::string res;
386 if (bpf) {
387 for (const auto& frontend : g_frontends) {
388 frontend->attachFilter(bpf);
389 }
390 }
391 });
392
393 g_lua.writeFunction("newDynBPFFilter", [client](std::shared_ptr<BPFFilter> bpf) {
394 if (client) {
395 return std::shared_ptr<DynBPFFilter>(nullptr);
396 }
397 return std::make_shared<DynBPFFilter>(bpf);
398 });
399
400 g_lua.registerFunction<void(std::shared_ptr<DynBPFFilter>::*)(const ComboAddress& addr, boost::optional<int> seconds)>("block", [](std::shared_ptr<DynBPFFilter> dbpf, const ComboAddress& addr, boost::optional<int> seconds) {
401 if (dbpf) {
402 struct timespec until;
403 clock_gettime(CLOCK_MONOTONIC, &until);
404 until.tv_sec += seconds ? *seconds : 10;
405 dbpf->block(addr, until);
406 }
407 });
408
409 g_lua.registerFunction<void(std::shared_ptr<DynBPFFilter>::*)()>("purgeExpired", [](std::shared_ptr<DynBPFFilter> dbpf) {
410 if (dbpf) {
411 struct timespec now;
412 clock_gettime(CLOCK_MONOTONIC, &now);
413 dbpf->purgeExpired(now);
414 }
415 });
416
417 g_lua.registerFunction<void(std::shared_ptr<DynBPFFilter>::*)(boost::variant<std::string, std::vector<std::pair<int, std::string>>>)>("excludeRange", [](std::shared_ptr<DynBPFFilter> dbpf, boost::variant<std::string, std::vector<std::pair<int, std::string>>> ranges) {
418 if (ranges.type() == typeid(std::vector<std::pair<int, std::string>>)) {
419 for (const auto& range : *boost::get<std::vector<std::pair<int, std::string>>>(&ranges)) {
420 dbpf->excludeRange(Netmask(range.second));
421 }
422 }
423 else {
424 dbpf->excludeRange(Netmask(*boost::get<std::string>(&ranges)));
425 }
426 });
427
428 g_lua.registerFunction<void(std::shared_ptr<DynBPFFilter>::*)(boost::variant<std::string, std::vector<std::pair<int, std::string>>>)>("includeRange", [](std::shared_ptr<DynBPFFilter> dbpf, boost::variant<std::string, std::vector<std::pair<int, std::string>>> ranges) {
429 if (ranges.type() == typeid(std::vector<std::pair<int, std::string>>)) {
430 for (const auto& range : *boost::get<std::vector<std::pair<int, std::string>>>(&ranges)) {
431 dbpf->includeRange(Netmask(range.second));
432 }
433 }
434 else {
435 dbpf->includeRange(Netmask(*boost::get<std::string>(&ranges)));
436 }
437 });
438 #endif /* HAVE_EBPF */
439
440 /* EDNSOptionView */
441 g_lua.registerFunction<size_t(EDNSOptionView::*)()>("count", [](const EDNSOptionView& option) {
442 return option.values.size();
443 });
444 g_lua.registerFunction<std::vector<string>(EDNSOptionView::*)()>("getValues", [] (const EDNSOptionView& option) {
445 std::vector<string> values;
446 for (const auto& value : option.values) {
447 values.push_back(std::string(value.content, value.size));
448 }
449 return values;
450 });
451
452 g_lua.writeFunction("newDOHResponseMapEntry", [](const std::string& regex, uint16_t status, const std::string& content, boost::optional<std::map<std::string, std::string>> customHeaders) {
453 boost::optional<std::vector<std::pair<std::string, std::string>>> headers{boost::none};
454 if (customHeaders) {
455 headers = std::vector<std::pair<std::string, std::string>>();
456 for (const auto& header : *customHeaders) {
457 headers->push_back({ boost::to_lower_copy(header.first), header.second });
458 }
459 }
460 return std::make_shared<DOHResponseMapEntry>(regex, status, content, headers);
461 });
462 }