+++ /dev/null
-/* Copyright (C) 2015 CZ.NIC, z.s.p.o. <knot-dns@labs.nic.cz>
-
- This program is free software: you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published by
- the Free Software Foundation, either version 3 of the License, or
- (at your option) any later version.
-
- 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, see <http://www.gnu.org/licenses/>.
- */
-
-#include <libknot/descriptor.h>
-#include "daemon/bindings/kres.h"
-#include "daemon/bindings.h"
-
-#define WRAP_NUMBER(L, name, val) \
- lua_pushnumber((L), (val)); \
- lua_setfield((L), -2, (name))
-
-#define WRAP_CONST(L, name, prefix...) \
- WRAP_NUMBER(L, #name, prefix ## name)
-
-#define WRAP_LUT(L, prefix, table) \
- lua_newtable(L); \
- for (const lookup_table_t *elm = (table); elm->name; ++elm) { \
- WRAP_NUMBER((L), elm->name, elm->id); \
- } \
- lua_setfield((L), -2, (prefix))
-
-#define LUA_ERRSTR(L, errstr) \
- lua_pushliteral(L, errstr); \
- lua_error(L)
-
-#define CHECK_UDATA(udata, L) \
- lua_touserdata(L, 1); if (!udata) return 0
-
-/** @internal Register metatable. */
-static void lua_register_meta(lua_State *L, const luaL_Reg *funcs, const char *name)
-{
- luaL_newmetatable(L, name); \
- luaL_setfuncs(L, funcs, 0); \
- lua_pushvalue(L, -1); \
- lua_setfield(L, -2, "__index"); \
- lua_pop(L, 1);
-}
-
-/** @internal Shortcut for dname conversion. */
-static inline void lua_pushdname(lua_State *L, const knot_dname_t *name)
-{
- char dname_str[KNOT_DNAME_MAXLEN];
- knot_dname_to_str(dname_str, name, sizeof(dname_str));
- lua_pushstring(L, dname_str);
-}
-
-/*
- * Record types, since the libknot doesn't export them.
- */
-#define RECORD_TYPES(X) \
- X(A) X(NS) X(CNAME) X(SOA) X(PTR) X(HINFO) X(MINFO) X(MX) \
- X(TXT) X(RP) X(AFSDB) X(RT) X(SIG) X(KEY) X(AAAA) X(LOC) \
- X(SRV) X(NAPTR) X(KX) X(CERT) X(DNAME) X(OPT) X(APL) X(DS) \
- X(SSHFP) X(IPSECKEY) X(RRSIG) X(NSEC) X(DNSKEY) X(DHCID) \
- X(NSEC3) X(NSEC3PARAM) X(TLSA) X(CDS) X(CDNSKEY) X(SPF) \
- X(NID) X(L32) X(L64) X(LP) X(EUI48) X(EUI64) X(TKEY) \
- X(TSIG) X(IXFR) X(AXFR) X(ANY)
-
-static lookup_table_t rrtype_names[] = {
- #define X(rc) { KNOT_RRTYPE_ ## rc, #rc },
- RECORD_TYPES(X)
- #undef X
- { 0, NULL }
-};
-
-/*
- * Record class names.
- */
-#define RECORD_CLASS(X) X(IN) X(CH) X(NONE) X(ANY)
-static lookup_table_t rrclass_names[] = {
- #define X(rc) { KNOT_CLASS_ ## rc, #rc },
- RECORD_CLASS(X)
- #undef X
- { 0, NULL }
-};
-
-/*
- * Packet interface
- * @note Packets are always light userdata, use single pointers.
- */
-
-#define WIRE_FLAGS(X) \
- X(AA,aa) X(AD,ad) X(CD,cd) X(RD,rd) X(QR,qr) X(RA,ra) X(TC,tc)
-enum {
- #define X(flag, _) WIRE_ ## flag,
- WIRE_FLAGS(X)
- #undef X
-};
-static lookup_table_t wire_flag_names[] = {
- #define X(flag, _) { WIRE_ ## flag, #flag },
- WIRE_FLAGS(X)
- #undef X
- { 0, NULL }
-};
-
-#define PKT_UDATA_CHECK(L) \
- if (!lua_touserdata(L, 1)) { \
- LUA_ERRSTR(L, "bad parameters, expected (pkt[, newvalue])"); \
- }
-
-static int pkt_flag(lua_State *L)
-{
- knot_pkt_t *pkt = CHECK_UDATA(pkt, L);
- if (lua_gettop(L) > 1 && lua_isnumber(L, 2)) {
- int flag_id = lua_tonumber(L, 2);
- switch(flag_id) {
- #define X(flag, code) case WIRE_ ## flag: knot_wire_set_ ## code (pkt->wire); break;
- WIRE_FLAGS(X)
- #undef X
- }
- }
- return 0;
-}
-
-static int pkt_opcode(lua_State *L)
-{
- knot_pkt_t *pkt = CHECK_UDATA(pkt, L);
- if (lua_gettop(L) > 1 && lua_isnumber(L, 2)) {
- knot_wire_set_opcode(pkt->wire, lua_tonumber(L, 2));
- }
- lua_pushnumber(L, knot_wire_get_opcode(pkt->wire));
- return 1;
-}
-
-static int pkt_rcode(lua_State *L)
-{
- knot_pkt_t *pkt = CHECK_UDATA(pkt, L);
- if (lua_gettop(L) > 1 && lua_isnumber(L, 2)) {
- knot_wire_set_rcode(pkt->wire, lua_tonumber(L, 2));
- }
- lua_pushnumber(L, knot_wire_get_rcode(pkt->wire));
- return 1;
-}
-
-static int pkt_qtype(lua_State *L)
-{
- knot_pkt_t *pkt = CHECK_UDATA(pkt, L);
- lua_pushnumber(L, knot_pkt_qtype(pkt));
- return 1;
-}
-
-static int pkt_qclass(lua_State *L)
-{
- knot_pkt_t *pkt = CHECK_UDATA(pkt, L);
- lua_pushnumber(L, knot_pkt_qclass(pkt));
- return 1;
-}
-
-static int pkt_qname(lua_State *L)
-{
- knot_pkt_t *pkt = CHECK_UDATA(pkt, L);
- lua_pushdname(L, knot_pkt_qname(pkt));
- return 1;
-}
-
-static int pkt_question(lua_State *L)
-{
- knot_pkt_t *pkt = CHECK_UDATA(pkt, L);
- if (lua_gettop(L) < 3) {
- return 0;
- }
- /* Check parameters */
- uint8_t dname[KNOT_DNAME_MAXLEN];
- knot_dname_from_str(dname, lua_tostring(L, 2), sizeof(dname));
- uint16_t rrtype = lua_tointeger(L, 3);
- uint16_t rrclass = lua_tointeger(L, 4);
- if (!lua_isnumber(L, 3)) {
- LUA_ERRSTR(L, "invalid RR type");
- }
- if (!lua_isnumber(L, 4)) { /* Default class is IN */
- rrclass = KNOT_CLASS_IN;
- }
- const knot_dname_t *qname = knot_pkt_qname(pkt);
- if (pkt->rrset_count > 0 || (!qname || !knot_dname_is_equal(qname, dname))) {
- KR_PKT_RECYCLE(pkt);
- knot_pkt_put_question(pkt, dname, rrclass, rrtype);
- pkt->parsed = pkt->size;
- }
- return 0;
-}
-
-static int pkt_begin(lua_State *L)
-{
- knot_pkt_t *pkt = CHECK_UDATA(pkt, L);
- if (!lua_isnumber(L, 2) || lua_tonumber(L, 2) < pkt->current) {
- LUA_ERRSTR(L, "bad parameters, expected packet section >= current");
- }
- knot_pkt_begin(pkt, lua_tointeger(L, 2));
- return 0;
-}
-
-static int pkt_add(lua_State *L)
-{
- knot_pkt_t *pkt = CHECK_UDATA(pkt, L);
- if (lua_gettop(L) < 6) {
- return 0;
- }
- /* Get parameters */
- uint8_t dname[KNOT_DNAME_MAXLEN];
- knot_dname_from_str(dname, lua_tostring(L, 2), sizeof(dname));
- uint16_t rrclass = lua_tointeger(L, 3);
- uint16_t rrtype = lua_tointeger(L, 4);
- uint32_t ttl = lua_tointeger(L, 5);
- size_t rdlen = 0;
- const char *raw_data = lua_tolstring(L, 6, &rdlen);
- /* Create empty RR */
- knot_rrset_t rr;
- knot_rrset_init(&rr, knot_dname_copy(dname, &pkt->mm), rrtype, rrclass);
- /* Create RDATA */
- knot_rdata_t rdata[knot_rdata_array_size(rdlen)];
- knot_rdata_init(rdata, rdlen, (const uint8_t *)raw_data, ttl);
- knot_rdataset_add(&rr.rrs, rdata, &pkt->mm);
- /* Append RR */
- int ret = knot_pkt_put(pkt, 0, &rr, KNOT_PF_FREE);
- lua_pushboolean(L, ret == 0);
- pkt->parsed = pkt->size;
- return 1;
-}
-
-static int pkt_get(lua_State *L)
-{
- knot_pkt_t *pkt = CHECK_UDATA(pkt, L);
- if (lua_gettop(L) < 3) {
- return 0;
- }
- /* Get parameters */
- uint16_t section_id = lua_tointeger(L, 2);
- uint16_t index = lua_tointeger(L, 3);
- /* Get RR */
- const knot_pktsection_t *sec = knot_pkt_section(pkt, section_id);
- if (!sec || sec->count <= index) {
- return 0;
- }
- const knot_rrset_t *rr = knot_pkt_rr(sec, index);
- lua_newtable(L);
- lua_pushdname(L, rr->owner);
- lua_setfield(L, -2, "owner");
- lua_pushnumber(L, rr->rclass);
- lua_setfield(L, -2, "class");
- lua_pushnumber(L, rr->type);
- lua_setfield(L, -2, "type");
- lua_pushnumber(L, knot_rrset_ttl(rr));
- lua_setfield(L, -2, "ttl");
- lua_pushlightuserdata(L, (void *)&rr->rrs);
- lua_setfield(L, -2, "rdata");
- return 1;
-}
-
-static int pkt_meta_register(lua_State *L)
-{
- static const luaL_Reg wrap[] = {
- { "flag", pkt_flag },
- { "rcode", pkt_rcode },
- { "opcode", pkt_opcode },
- { "qtype", pkt_qtype },
- { "qclass", pkt_qclass },
- { "qname", pkt_qname },
- { "question", pkt_question },
- { "begin", pkt_begin },
- { "add", pkt_add },
- { "get", pkt_get },
- { NULL, NULL }
- };
- lua_register_meta(L, wrap, META_PKT);
- return 0;
-}
-
-/**
- * Query interface.
- * @note Query is a full userdata, use double pointers.
- */
-
-static int query_qtype(lua_State *L)
-{
- struct kr_query *qry = CHECK_UDATA(qry, L);
- lua_pushnumber(L, qry->stype);
- return 1;
-}
-
-static int query_qclass(lua_State *L)
-{
- struct kr_query *qry = CHECK_UDATA(qry, L);
- lua_pushnumber(L, qry->sclass);
- return 1;
-}
-
-static int query_qname(lua_State *L)
-{
- struct kr_query *qry = CHECK_UDATA(qry, L);
- lua_pushdname(L, qry->sname);
- return 1;
-}
-
-static int query_flag(lua_State *L)
-{
- struct kr_query *qry = CHECK_UDATA(qry, L);
- if (lua_gettop(L) < 2 || !lua_isnumber(L, 2)) {
- return 0;
- }
- qry->flags |= lua_tointeger(L, 2);
- return 0;
-}
-
-static int query_clear_flag(lua_State *L)
-{
- struct kr_query *qry = CHECK_UDATA(qry, L);
- if (lua_gettop(L) < 2 || !lua_isnumber(L, 2)) {
- return 0;
- }
- qry->flags &= ~lua_tointeger(L, 2);
- return 0;
-}
-
-static int query_has_flag(lua_State *L)
-{
- struct kr_query *qry = CHECK_UDATA(qry, L);
- if (lua_gettop(L) < 2 || !lua_isnumber(L, 2)) {
- return 0;
- }
- lua_pushboolean(L, qry->flags & lua_tointeger(L, 2));
- return 1;
-}
-
-static int query_current(lua_State *L)
-{
- struct kr_request *req = CHECK_UDATA(req, L);
- lua_pushlightuserdata(L, kr_rplan_current(&req->rplan));
- return 1;
-}
-
-static int query_resolved(lua_State *L)
-{
- struct kr_request *req = CHECK_UDATA(req, L);
- lua_pushlightuserdata(L, TAIL(req->rplan.resolved));
- return 1;
-}
-
-static int qry_meta_register(lua_State *L)
-{
- static const luaL_Reg wrap[] = {
- { "qtype", query_qtype },
- { "qclass", query_qclass },
- { "qname", query_qname },
- { "flag", query_flag },
- { "clear_flag", query_clear_flag },
- { "has_flag", query_has_flag },
- { NULL, NULL }
- };
- lua_getfield(L, -1, "query");
- for (const luaL_Reg *reg = wrap; reg->name; ++reg) {
- lua_pushcfunction(L, reg->func);
- lua_setfield(L, -2, reg->name);
- }
- lua_pop(L, 1);
- return 0;
-}
-
-int lib_kres(lua_State *L)
-{
- static const luaL_Reg lib[] = {
- { "query_current", query_current },
- { "query_resolved", query_resolved },
- { NULL, NULL }
- };
- /* Create module and register functions */
- register_lib(L, "kres", lib);
- /* Register states */
- WRAP_CONST(L, NOOP, KNOT_STATE_);
- WRAP_CONST(L, CONSUME, KNOT_STATE_);
- WRAP_CONST(L, PRODUCE, KNOT_STATE_);
- WRAP_CONST(L, DONE, KNOT_STATE_);
- WRAP_CONST(L, FAIL, KNOT_STATE_);
- /* Register packet sections */
- WRAP_CONST(L, ANSWER, KNOT_);
- WRAP_CONST(L, AUTHORITY, KNOT_);
- WRAP_CONST(L, ADDITIONAL, KNOT_);
- /* Register RCODE, OPCODE */
- WRAP_LUT(L, "rcode", knot_rcode_names);
- WRAP_LUT(L, "type", rrtype_names);
- WRAP_LUT(L, "class", rrclass_names);
- WRAP_LUT(L, "opcode", knot_opcode_names);
- WRAP_LUT(L, "wire", wire_flag_names);
- WRAP_LUT(L, "query", query_flag_names);
- /* Register metatables */
- pkt_meta_register(L);
- qry_meta_register(L);
- return 1;
-}
--- /dev/null
+-- LuaJIT ffi bindings for libkres, a DNS resolver library.
+-- @note Since it's statically compiled, it expects to find the symbols in the C namespace.
+
+local ffi = require('ffi')
+local bit = require('bit')
+local csym = ffi.C
+local knot = ffi.load('knot')
+ffi.cdef[[
+
+/*
+ * Record types and classes.
+ */
+struct rr_class {
+ static const int IN = 1;
+ static const int CH = 3;
+ static const int NONE = 254;
+ static const int ANY = 255;
+};
+struct rr_type {
+ static const int A = 1;
+ static const int NS = 2;
+ static const int CNAME = 5;
+ static const int SOA = 6;
+ static const int PTR = 12;
+ static const int HINFO = 13;
+ static const int MINFO = 14;
+ static const int MX = 15;
+ static const int TXT = 16;
+ static const int RP = 17;
+ static const int AFSDB = 18;
+ static const int RT = 21;
+ static const int SIG = 24;
+ static const int KEY = 25;
+ static const int AAAA = 28;
+ static const int LOC = 29;
+ static const int SRV = 33;
+ static const int NAPTR = 35;
+ static const int KX = 36;
+ static const int CERT = 37;
+ static const int DNAME = 39;
+ static const int OPT = 41;
+ static const int APL = 42;
+ static const int DS = 43;
+ static const int SSHFP = 44;
+ static const int IPSECKEY = 45;
+ static const int RRSIG = 46;
+ static const int NSEC = 47;
+ static const int DNSKEY = 48;
+ static const int DHCID = 49;
+ static const int NSEC3 = 50;
+ static const int NSEC3PARAM = 51;
+ static const int TLSA = 52;
+ static const int CDS = 59;
+ static const int CDNSKEY = 60;
+ static const int SPF = 99;
+ static const int NID = 104;
+ static const int L32 = 105;
+ static const int L64 = 106;
+ static const int LP = 107;
+ static const int EUI48 = 108;
+ static const int EUI64 = 109;
+ static const int TKEY = 249;
+ static const int TSIG = 250;
+ static const int IXFR = 251;
+ static const int AXFR = 252;
+ static const int ANY = 255;
+};
+struct pkt_section {
+ static const int ANSWER = 0;
+ static const int AUTHORITY = 1;
+ static const int ADDITIONAL = 2;
+};
+struct pkt_rcode {
+ static const int NOERROR = 0;
+ static const int FORMERR = 1;
+ static const int SERVFAIL = 2;
+ static const int NXDOMAIN = 3;
+ static const int NOTIMPL = 4;
+ static const int REFUSED = 5;
+ static const int YXDOMAIN = 6;
+ static const int YXRRSET = 7;
+ static const int NXRRSET = 8;
+ static const int NOTAUTH = 9;
+ static const int NOTZONE = 10;
+ static const int BADVERS = 16;
+};
+/*
+ * Data structures
+ */
+
+/* libknot */
+typedef struct node {
+ struct node *next, *prev;
+} node_t;
+typedef uint8_t knot_dname_t;
+typedef struct {
+ uint8_t *wire;
+ size_t size;
+ size_t max_size;
+ size_t parsed;
+ uint16_t reserved;
+ uint16_t qname_size;
+ uint16_t rrset_count;
+ uint16_t flags;
+ uint8_t _stub[]; /* Do not touch */
+} knot_pkt_t;
+
+/* libkres */
+struct kr_query {
+ node_t _node;
+ struct kr_query *parent;
+ knot_dname_t *sname;
+ uint16_t type;
+ uint16_t class;
+ uint16_t id;
+ uint16_t flags;
+ unsigned secret;
+ uint8_t _stub[]; /* Do not touch */
+};
+struct kr_rplan {
+ uint8_t _stub[]; /* Do not touch */
+};
+struct kr_request {
+ struct kr_context *_ctx;
+ knot_pkt_t *answer;
+ uint32_t options;
+ int state;
+ uint8_t _stub[]; /* Do not touch */
+};
+
+/* libknot API
+ */
+
+/* Domain names */
+/* Resource records */
+/* Packet */
+const knot_dname_t *knot_pkt_qname(const knot_pkt_t *pkt);
+uint16_t knot_pkt_qtype(const knot_pkt_t *pkt);
+uint16_t knot_pkt_qclass(const knot_pkt_t *pkt);
+int knot_pkt_begin(knot_pkt_t *pkt, int section_id);
+int knot_pkt_put_question(knot_pkt_t *pkt, const knot_dname_t *qname, uint16_t qclass, uint16_t qtype);
+
+/* libkres API
+ */
+
+/* Resolution request */
+struct kr_rplan *kr_resolve_plan(struct kr_request *request);
+/* Resolution plan */
+struct kr_query *kr_rplan_current(struct kr_rplan *rplan);
+/* Query */
+/* Utils */
+unsigned kr_rand_uint(unsigned max);
+int kr_pkt_put(knot_pkt_t *pkt, const knot_dname_t *name, uint32_t ttl,
+ uint16_t rclass, uint16_t rtype, const uint8_t *rdata, uint16_t rdlen);
+]]
+
+-- Metatype for packet
+local knot_pkt_t = ffi.typeof('knot_pkt_t')
+ffi.metatype( knot_pkt_t, {
+ __index = {
+ qname = function(pkt) return ffi.string(knot.knot_pkt_qname(pkt)) end,
+ qclass = function(pkt) return knot.knot_pkt_qclass(pkt) end,
+ qtype = function(pkt) return knot.knot_pkt_qtype(pkt) end,
+ rcode = function (pkt, val)
+ if val then
+ pkt.wire[3] = bit.bor(bit.band(pkt.wire[3], 0xf0), val)
+ end
+ return bit.band(pkt.wire[3], 0x0f)
+ end,
+ begin = function (pkt, section) return knot.knot_pkt_begin(pkt, section) end,
+ put = function (pkt, owner, ttl, rclass, rtype, rdata)
+ return csym.kr_pkt_put(pkt, owner, ttl, rclass, rtype, rdata, string.len(rdata))
+ end
+ },
+})
+-- Metatype for query
+local kr_query_t = ffi.typeof('struct kr_query')
+ffi.metatype( kr_query_t, {
+ __index = {
+ name = function(qry) return ffi.string(qry.sname) end,
+ },
+})
+-- Metatype for request
+local kr_request_t = ffi.typeof('struct kr_request')
+ffi.metatype( kr_request_t, {
+ __index = {
+ current = function(req)
+ assert(req)
+ return csym.kr_rplan_current(csym.kr_resolve_plan(req))
+ end,
+ },
+})
+
+-- Module API
+local kres = {
+ -- Constants
+ class = ffi.new('struct rr_class'),
+ type = ffi.new('struct rr_type'),
+ section = ffi.new('struct pkt_section'),
+ rcode = ffi.new('struct pkt_rcode'),
+ NOOP = 0, CONSUME = 1, PRODUCE = 2, DONE = 4, FAIL = 8,
+ -- Metatypes
+ pkt_t = function (udata) return ffi.cast('knot_pkt_t *', udata) end,
+ request_t = function (udata) return ffi.cast('struct kr_request *', udata) end,
+ -- Global API functions
+}
+
+return kres
\ No newline at end of file
projects / thirdparty / knot-resolver.git / commitdiff
