coolaj86
/
digd.js
2
1
Fork 1
Code Issues 7 Pull Requests 2 Releases Activity

making room for API

This commit is contained in:
AJ ONeal 2017-11-05 09:16:27 -07:00
parent b5ec1f7982
commit 59980dfa60
3 changed files with 170 additions and 134 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
bin/digd.js
View File

@ -373,7 +373,7 @@ cli.main(function (args, cli) {
return;
}
require('../lib/dns-store').query(path.resolve(cli.input), query, function (err, resp) {
function respondWithResults(err, resp) {
if (err) { console.log('[DEV] answer not found in local db, recursing'); console.error(err); recurse(); return; }
@ -383,7 +383,16 @@ cli.main(function (args, cli) {
if (!cli.norecurse && query.header.rd) { resp.header.ra = 1; }
sendResponse(resp);
});
}
var engine;
try {
engine = require('../lib/store.json.js').create({ filepath: path.resolve(cli.input) });
} catch(e) {
respondWithResults(e);
return;
}
require('../lib/digd.js').query(engine, query, respondWithResults);
};

62
lib/dns-store.js → lib/digd.js
View File

@ -10,23 +10,14 @@ var NOERROR = 0;
var NXDOMAIN = 3;
var REFUSED = 5;
function getRecords(db, qname, cb) {
function getRecords(engine, qname, cb) {
var delMe = {};
var dns = require('dns');
// SECURITY XXX TODO var dig = require('dig.js/dns-request');
var count;
var myRecords = db.records.slice(0).filter(function (r) {
if ('string' !== typeof r.name) {
return false;
}
// TODO use IN in masterquest (or implement OR)
// Only return single-level wildcard?
if (qname === r.name || ('*.' + qname.split('.').slice(1).join('.')) === r.name) {
return true;
}
});
return engine.getRecords({ name: qname }, function (err, myRecords) {
if (err) { cb(err); return; }
function checkCount() {
var ready;
@ -108,12 +99,13 @@ function getRecords(db, qname, cb) {
if (!myRecords.length) {
checkCount();
}
});
}
function dbToResourceRecord(r) {
return {
name: r.name
, typeName: r.type // NS
, typeName: r.typeName || r.type // NS
, className: 'IN'
, ttl: r.ttl || 300
@ -147,7 +139,7 @@ function dbToResourceRecord(r) {
};
}
function getNs(db, ds, results, cb) {
function getNs(engine, ds, results, cb) {
console.log('[DEV] getNs entered with domains', ds);
var d = ds.shift();
@ -161,7 +153,7 @@ function getNs(db, ds, results, cb) {
var qn = d.id.toLowerCase();
return getRecords(db, qn, function (err, records) {
return getRecords(engine, qn, function (err, records) {
if (err) { cb(err); return; }
records.forEach(function (r) {
@ -187,16 +179,16 @@ function getNs(db, ds, results, cb) {
});
if (!results.authority.length) {
return getNs(db, ds, results, cb);
return getNs(engine, ds, results, cb);
}
// d.vanityNs should only be vanity nameservers (pointing to this same server)
if (d.vanityNs || results.authority.some(function (ns) {
console.log('[debug] ns', ns);
return -1 !== db.primaryNameservers.indexOf(ns.data.toLowerCase());
return -1 !== engine.primaryNameservers.indexOf(ns.data.toLowerCase());
})) {
results.authority.length = 0;
results.authority.push(domainToSoa(db, d));
results.authority.push(domainToSoa(engine.primaryNameservers, d));
results.header.rcode = NXDOMAIN;
}
cb(null, results);
@ -204,8 +196,8 @@ function getNs(db, ds, results, cb) {
});
}
function domainToSoa(db, domain) {
var nameservers = domain.vanityNs || db.primaryNameservers;
function domainToSoa(primaryNameservers, domain) {
var nameservers = domain.vanityNs || primaryNameservers;
var index = Math.floor(Math.random() * nameservers.length) % nameservers.length;
var nameserver = nameservers[index];
@ -245,20 +237,20 @@ function domainToSoa(db, domain) {
};
}
function getSoa(db, domain, results, cb, answerSoa) {
function getSoa(primaryNameservers, domain, results, cb, answerSoa) {
console.log('[DEV] getSoa entered');
if (!answerSoa) {
results.authority.push(domainToSoa(db, domain));
results.authority.push(domainToSoa(primaryNameservers, domain));
} else {
results.answer.push(domainToSoa(db, domain));
results.answer.push(domainToSoa(primaryNameservers, domain));
}
cb(null, results);
return;
}
module.exports.query = function (input, query, cb) {
module.exports.query = function (engine, query, cb) {
/*
var fs = require('fs');
@ -271,11 +263,7 @@ module.exports.query = function (input, query, cb) {
});
*/
var db;
var qname;
try {
db = require(input);
} catch(e) { cb(e); return; }
if (!Array.isArray(query.question) || query.question.length < 1) {
cb(new Error("query is missing question section"));
@ -337,9 +325,10 @@ module.exports.query = function (input, query, cb) {
console.log('[DEV] answerSoa?', answerSoa);
console.log('[DEV] qnames');
console.log(qnames);
var myDomains = db.domains.filter(function (d) {
return -1 !== qnames.indexOf(d.id.toLowerCase());
});
return engine.getSoas({ names: qnames}, function (err, myDomains) {
console.log('[SOA] looking for', qnames, 'and proudly serving', err, myDomains);
if (err) { cb(err); return; }
// this should result in a REFUSED status
if (!myDomains.length) {
@ -361,10 +350,10 @@ module.exports.query = function (input, query, cb) {
//console.log('sorted domains', myDomains);
if (!getNsAlso) {
return getSoa(db, myDomains[0], results, cb, answerSoa);
return getSoa(engine.primaryNameservers, myDomains[0], results, cb, answerSoa);
}
return getNs(db, /*myDomains.slice(0)*/qnames.map(function (qn) { return { id: qn }; }), results, function (err, results) {
return getNs(engine, /*myDomains.slice(0)*/qnames.map(function (qn) { return { id: qn }; }), results, function (err, results) {
//console.log('[DEV] getNs complete');
if (err) { cb(err, results); return; }
@ -375,9 +364,10 @@ module.exports.query = function (input, query, cb) {
}
// myDomains was sorted such that the longest was first
return getSoa(db, myDomains[0], results, cb);
return getSoa(engine.primaryNameservers, myDomains[0], results, cb);
});
});
}
if ('SOA' === query.question[0].typeName) {
@ -385,7 +375,7 @@ module.exports.query = function (input, query, cb) {
}
//console.log('[DEV] QUERY NAME', qname);
return getRecords(db, qname, function (err, someRecords) {
return getRecords(engine, qname, function (err, someRecords) {
var myRecords;
var nsRecords = [];
@ -410,7 +400,7 @@ module.exports.query = function (input, query, cb) {
// NOTE: I think that the issue here is EXTERNAL vs INTERNAL vanity NS
// We _should_ reply for EXTERNAL vanity NS... but not when it's listed on the SOA internally?
// It's surrounding the problem of what if I do sub domain delegation to the same server.
if (-1 === db.primaryNameservers.indexOf(r.data.toLowerCase())) {
if (-1 === engine.primaryNameservers.indexOf(r.data.toLowerCase())) {
console.log("It's a vanity NS");
return false;
}

37
lib/store.json.js Normal file
View File

@ -0,0 +1,37 @@
'use strict';
module.exports.create = function (opts) {
// opts = { filepath };
var engine = { db: null };
var db = require(opts.filepath);
engine.primaryNameservers = db.primaryNameservers;
engine.getSoas = function (query, cb) {
var myDomains = db.domains.filter(function (d) {
return -1 !== query.names.indexOf(d.id.toLowerCase());
});
process.nextTick(function () {
cb(null, myDomains);
});
};
engine.getRecords = function (query, cb) {
var myRecords = db.records.slice(0).filter(function (r) {
if ('string' !== typeof r.name) {
return false;
}
// TODO use IN in masterquest (or implement OR)
// Only return single-level wildcard?
if (query.name === r.name || ('*.' + query.name.split('.').slice(1).join('.')) === r.name) {
return true;
}
});
process.nextTick(function () {
cb(null, myRecords);
});
};
return engine;
};