Compare commits

..

7 Commits
v1.1 ... master

Author SHA1 Message Date
c0eb4225dc Update 'README.md' 2019-03-05 23:43:14 +00:00
6ef61a8674 don't let it die, duh!! 2018-05-04 16:05:59 +00:00
AJ ONeal
d8ead3181d update installer with v1.2 2017-12-14 20:33:32 -07:00
AJ ONeal
bc93b942ee v1.2.1 2017-12-14 20:29:02 -07:00
AJ ONeal
c3934acb30 add tcp support 2017-12-14 20:26:28 -07:00
AJ ONeal
bd8b57efd1 bump version 2017-11-05 09:17:55 -07:00
AJ ONeal
59980dfa60 making room for API 2017-11-05 09:16:27 -07:00
9 changed files with 353 additions and 201 deletions

View File

@ -5,7 +5,7 @@ digd.js
| [dig.js](https://git.coolaj86.com/coolaj86/dig.js) | [dig.js](https://git.coolaj86.com/coolaj86/dig.js)
| [mdig.js](https://git.coolaj86.com/coolaj86/mdig.js) | [mdig.js](https://git.coolaj86.com/coolaj86/mdig.js)
| **digd.js** | **digd.js**
| Sponsored by [Daplie](https://daplie.com). | A [Root project](https://rootprojects.org).
A lightweight DNS / mDNS daemon (server) in node.js. A lightweight DNS / mDNS daemon (server) in node.js.
@ -22,7 +22,7 @@ Install
### systemd service ### systemd service
```bash ```bash
curl -L https://git.coolaj86.com/coolaj86/digd.js/raw/v1.1/install.sh | bash curl -L https://git.coolaj86.com/coolaj86/digd.js/raw/v1.2/install.sh | bash
``` ```
### with git ### with git
@ -33,14 +33,13 @@ npm install -g 'git+https://git.coolaj86.com/coolaj86/digd.js.git#v1'
``` ```
```bash ```bash
# Install exactly v1.1.9 # Install exactly v1.2.0
npm install -g 'git+https://git.coolaj86.com/coolaj86/digd.js.git#v1.1.9' npm install -g 'git+https://git.coolaj86.com/coolaj86/digd.js.git#v1.2.0'
``` ```
### without git ### without git
Don't have git? Well, you can also bow down to the gods of the centralized, monopolized, concentrated, *dictator*net Don't have git? You can use npm's centralized repository:
(as we like to call it here at Daplie Labs), if that's how you roll:
```bash ```bash
npm install -g digd.js npm install -g digd.js
@ -89,6 +88,8 @@ Options
+time=<seconds> Sets the timeout for a query in seconds. +time=<seconds> Sets the timeout for a query in seconds.
+norecurse Set `ra` flag to 0. Do not perform recursion. +norecurse Set `ra` flag to 0. Do not perform recursion.
+aaonly Set `aa` flag to 1. Do not respond with non-authoritative responses. +aaonly Set `aa` flag to 1. Do not respond with non-authoritative responses.
+notcp Disable TCP server (default in v1.2)
+tcp Enable TCP server (default in v1.3)
--debug verbose output --debug verbose output
``` ```

View File

@ -5,9 +5,7 @@
var cli = require('cli'); var cli = require('cli');
var pkg = require('../package.json'); var pkg = require('../package.json');
var dig = require('dig.js/dns-request'); var dig = require('dig.js/dns-request');
var dgram = require('dgram');
var dnsjs = require('dns-suite'); var dnsjs = require('dns-suite');
var crypto = require('crypto');
var common = require('dig.js/common'); var common = require('dig.js/common');
var defaultNameservers = require('dns').getServers(); var defaultNameservers = require('dns').getServers();
var hexdump; var hexdump;
@ -49,8 +47,32 @@ cli.main(function (args, cli) {
cli.norecurse = true; cli.norecurse = true;
return; return;
} }
if (arg === '+notcp') {
if (cli.notcp) {
console.error("'+notcp' was specified more than once");
process.exit(1);
return;
}
cli.notcp = true;
return;
}
if (arg === '+tcp') {
if (cli.tcp) {
console.error("'+tcp' was specified more than once");
process.exit(1);
return;
}
cli.tcp = true;
return;
}
}); });
if (!cli.tcp) {
if (!cli.notcp) {
console.info("[WARNING] Set '+notcp' to disable tcp connections. The default behavior changes to +tcp in v1.3");
}
}
if (cli.mdns) { if (cli.mdns) {
if (!cli.type) { if (!cli.type) {
cli.type = cli.t = 'PTR'; cli.type = cli.t = 'PTR';
@ -73,32 +95,10 @@ cli.main(function (args, cli) {
} }
} }
var handlers = {}; var dnsd = {};
var server = dgram.createSocket({ dnsd.onMessage = function (nb, cb) {
type: cli.udp6 ? 'udp6' : 'udp4' var byteOffset = nb._dnsByteOffset || nb.byteOffset;
, reuseAddr: true var queryAb = nb.buffer.slice(byteOffset, byteOffset + nb.byteLength);
});
server.bind({
port: cli.port
, address: cli.address
});
handlers.onError = function (err) {
if ('EACCES' === err.code) {
console.error("");
console.error("EACCES: Couldn't bind to port. You probably need to use sudo, authbind, or setcap.");
console.error("");
process.exit(123);
return;
}
console.error("error:", err.stack);
server.close();
};
handlers.onMessage = function (nb, rinfo) {
console.log('[DEBUG] got a message');
var queryAb = nb.buffer.slice(nb.byteOffset, nb.byteOffset + nb.byteLength);
var query; var query;
var count; var count;
@ -215,12 +215,11 @@ cli.main(function (args, cli) {
console.error("Could not write empty DNS response"); console.error("Could not write empty DNS response");
console.error(e); console.error(e);
console.error(emptyResp); console.error(emptyResp);
cb(e, null, '[DEV] response sent (empty)');
return; return;
} }
server.send(newAb, rinfo.port, rinfo.address, function () { cb(null, newAb, '[DEV] response sent (empty)');
console.log('[DEV] response sent (empty)', rinfo.port, rinfo.address);
});
} }
function sendResponse(newPacket) { function sendResponse(newPacket) {
@ -232,12 +231,11 @@ cli.main(function (args, cli) {
console.error("Could not write DNS response from local"); console.error("Could not write DNS response from local");
console.error(e); console.error(e);
console.error(newPacket); console.error(newPacket);
cb(e, null, '[DEV] response sent (local query)');
return; return;
} }
server.send(newAb, rinfo.port, rinfo.address, function () { cb(null, newAb, '[DEV] response sent (local query)');
console.log('[DEV] response sent (local query)', rinfo.port, rinfo.address);
});
} }
function recurse() { function recurse() {
@ -290,12 +288,11 @@ cli.main(function (args, cli) {
} catch(e) { } catch(e) {
console.error("Could not write DNS response"); console.error("Could not write DNS response");
console.error(newResponse); console.error(newResponse);
cb(e, null, '[DEV] response sent');
return; return;
} }
server.send(newAb, rinfo.port, rinfo.address, function () { cb(null, newAb, '[DEV] response sent');
console.log('[DEV] response sent', rinfo.port, rinfo.address);
});
} }
} }
@ -335,7 +332,8 @@ cli.main(function (args, cli) {
console.log('request sent to', res.nameserver); console.log('request sent to', res.nameserver);
} }
*/ */
console.log('[DEV] query sent (recurse)', rinfo.port, rinfo.address); //console.log('[DEV] query sent (recurse)', rinfo.port, rinfo.address);
//dnsd.onSent('[DEV] query sent (recurse)');
} }
, onTimeout: function (res) { , onTimeout: function (res) {
console.log(";; [" + q.name + "] connection timed out; no servers could be reached"); console.log(";; [" + q.name + "] connection timed out; no servers could be reached");
@ -373,7 +371,7 @@ cli.main(function (args, cli) {
return; 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; } if (err) { console.log('[DEV] answer not found in local db, recursing'); console.error(err); recurse(); return; }
@ -383,33 +381,34 @@ cli.main(function (args, cli) {
if (!cli.norecurse && query.header.rd) { resp.header.ra = 1; } if (!cli.norecurse && query.header.rd) { resp.header.ra = 1; }
sendResponse(resp); 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);
}; };
handlers.onListening = function () { cli.defaultNameservers = defaultNameservers;
/*jshint validthis:true*/ require('../lib/udpd.js').create(cli, dnsd).on('listening', function () {
var server = this;
cli.chosenNameserver = cli.nameserver; cli.chosenNameserver = cli.nameserver;
var index; var index;
if (!cli.chosenNameserver) { if (!cli.chosenNameserver) {
index = crypto.randomBytes(2).readUInt16BE(0) % defaultNameservers.length; index = require('crypto').randomBytes(2).readUInt16BE(0) % cli.defaultNameservers.length;
cli.chosenNameserver = defaultNameservers[index]; cli.chosenNameserver = cli.defaultNameservers[index];
if (cli.debug) { if (cli.debug) {
console.log('index, defaultNameservers', index, defaultNameservers); console.log('index, defaultNameservers', index, cli.defaultNameservers);
} }
} }
});
if (cli.mdns || '224.0.0.251' === cli.nameserver) { if (cli.tcp /* TODO v1.3 !cli.notcp */) {
server.setBroadcast(true); require('../lib/tcpd.js').create(cli, dnsd);
server.addMembership(cli.nameserver); }
}
console.log('');
console.log('Bound and Listening:');
console.log(server.address().address + '#' + server.address().port + ' (' + server.type + ')');
};
console.log(''); console.log('');
if (!cli.nocmd) { if (!cli.nocmd) {
@ -417,7 +416,4 @@ cli.main(function (args, cli) {
console.log(';; global options: +cmd'); console.log(';; global options: +cmd');
} }
server.on('error', handlers.onError);
server.on('message', handlers.onMessage);
server.on('listening', handlers.onListening);
}); });

View File

@ -8,7 +8,7 @@ Wants=network-online.target systemd-networkd-wait-online.service
# Restart on crash (bad signal), but not on 'clean' failure (error exit code) # Restart on crash (bad signal), but not on 'clean' failure (error exit code)
# Allow up to 3 restarts within 10 seconds # Allow up to 3 restarts within 10 seconds
# (it's unlikely that a user or properly-running script will do this) # (it's unlikely that a user or properly-running script will do this)
Restart=on-abnormal Restart=always
StartLimitInterval=10 StartLimitInterval=10
StartLimitBurst=3 StartLimitBurst=3

View File

@ -6,7 +6,7 @@ sudo mkdir -p /opt/digd.js /srv/digd.js
#chown -R $(whoami):$(whoami) /opt/digd.js /srv/digd.js #chown -R $(whoami):$(whoami) /opt/digd.js /srv/digd.js
chown -R digd:digd /opt/digd.js /srv/digd.js chown -R digd:digd /opt/digd.js /srv/digd.js
echo "v8.9.0" > /tmp/NODEJS_VER echo "v8.9.3" > /tmp/NODEJS_VER
export NODE_PATH=/opt/digd.js/lib/node_modules export NODE_PATH=/opt/digd.js/lib/node_modules
export NPM_CONFIG_PREFIX=/opt/digd.js export NPM_CONFIG_PREFIX=/opt/digd.js
curl -fsSL https://git.coolaj86.com/coolaj86/node-installer.sh/raw/master/install.sh -o ./node-installer.sh.tmp curl -fsSL https://git.coolaj86.com/coolaj86/node-installer.sh/raw/master/install.sh -o ./node-installer.sh.tmp
@ -16,7 +16,7 @@ rm ./node-installer.sh.tmp
git clone https://git.coolaj86.com/coolaj86/digd.js /opt/digd.js/lib/node_modules/digd.js git clone https://git.coolaj86.com/coolaj86/digd.js /opt/digd.js/lib/node_modules/digd.js
pushd /opt/digd.js/lib/node_modules/digd.js pushd /opt/digd.js/lib/node_modules/digd.js
git checkout v1.1 git checkout v1.2
/opt/digd.js/bin/node /opt/digd.js/bin/npm install /opt/digd.js/bin/node /opt/digd.js/bin/npm install
popd popd

View File

@ -10,110 +10,102 @@ var NOERROR = 0;
var NXDOMAIN = 3; var NXDOMAIN = 3;
var REFUSED = 5; var REFUSED = 5;
function getRecords(db, qname, cb) { function getRecords(engine, qname, cb) {
var delMe = {}; var delMe = {};
var dns = require('dns'); var dns = require('dns');
// SECURITY XXX TODO var dig = require('dig.js/dns-request'); // SECURITY XXX TODO var dig = require('dig.js/dns-request');
var count; var count;
var myRecords = db.records.slice(0).filter(function (r) {
if ('string' !== typeof r.name) { return engine.getRecords({ name: qname }, function (err, myRecords) {
return false; if (err) { cb(err); return; }
}
// TODO use IN in masterquest (or implement OR) function checkCount() {
// Only return single-level wildcard? var ready;
if (qname === r.name || ('*.' + qname.split('.').slice(1).join('.')) === r.name) {
return true;
}
});
function checkCount() { count -= 1;
var ready; ready = count <= 0;
count -= 1; if (!ready) {
ready = count <= 0; return;
if (!ready) {
return;
}
myRecords = myRecords.filter(function (r) {
return !delMe[r.id];
});
// There are a number of ways to interpret the wildcard rules
var hasWild = false;
var hasMatch = false;
myRecords.some(function (r) {
if (qname === r.name) {
hasMatch = true;
return true;
} }
if ('*' === r.name[0]) {
hasWild = true;
}
});
if (hasMatch) {
myRecords = myRecords.filter(function (r) { myRecords = myRecords.filter(function (r) {
if ('*' !== r.name[0]) { return true; } return !delMe[r.id];
}); });
}
/* // There are a number of ways to interpret the wildcard rules
// no need to filter out records if wildcard is used var hasWild = false;
else { var hasMatch = false;
records = records.filter(function (r) { myRecords.some(function (r) {
if ('*' === r.name[0]) { return true; } if (qname === r.name) {
hasMatch = true;
return true;
}
if ('*' === r.name[0]) {
hasWild = true;
}
}); });
}
*/
cb(null, myRecords); if (hasMatch) {
} myRecords = myRecords.filter(function (r) {
if ('*' !== r.name[0]) { return true; }
function getRecord(r) { });
// TODO allow multiple records to be returned(?)
return function (err, addresses) {
if (err || !addresses.length) {
r.id = r.id || Math.random();
delMe[r.id] = true;
} else if (addresses.length > 1) {
r._address = addresses[Math.floor(Math.random() * addresses.length)];
} else {
r._address = addresses[0];
} }
/*
// no need to filter out records if wildcard is used
else {
records = records.filter(function (r) {
if ('*' === r.name[0]) { return true; }
});
}
*/
cb(null, myRecords);
}
function getRecord(r) {
// TODO allow multiple records to be returned(?)
return function (err, addresses) {
if (err || !addresses.length) {
r.id = r.id || Math.random();
delMe[r.id] = true;
} else if (addresses.length > 1) {
r._address = addresses[Math.floor(Math.random() * addresses.length)];
} else {
r._address = addresses[0];
}
checkCount();
};
}
count = myRecords.length;
myRecords.forEach(function (r) {
if (r.aname && !r.address) {
if ('A' === r.type) {
// SECURITY XXX TODO dig.resolveJson(query, opts);
dns.resolve4(r.aname, getRecord(r));
return;
}
if ('AAAA' === r.type) {
// SECURITY XXX TODO dig.resolveJson(query, opts);
dns.resolve6(r.aname, getRecord(r));
return;
}
}
checkCount(); checkCount();
}; });
}
count = myRecords.length; if (!myRecords.length) {
myRecords.forEach(function (r) { checkCount();
if (r.aname && !r.address) {
if ('A' === r.type) {
// SECURITY XXX TODO dig.resolveJson(query, opts);
dns.resolve4(r.aname, getRecord(r));
return;
}
if ('AAAA' === r.type) {
// SECURITY XXX TODO dig.resolveJson(query, opts);
dns.resolve6(r.aname, getRecord(r));
return;
}
} }
checkCount();
}); });
if (!myRecords.length) {
checkCount();
}
} }
function dbToResourceRecord(r) { function dbToResourceRecord(r) {
return { return {
name: r.name name: r.name
, typeName: r.type // NS , typeName: r.typeName || r.type // NS
, className: 'IN' , className: 'IN'
, ttl: r.ttl || 300 , 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); console.log('[DEV] getNs entered with domains', ds);
var d = ds.shift(); var d = ds.shift();
@ -161,7 +153,7 @@ function getNs(db, ds, results, cb) {
var qn = d.id.toLowerCase(); var qn = d.id.toLowerCase();
return getRecords(db, qn, function (err, records) { return getRecords(engine, qn, function (err, records) {
if (err) { cb(err); return; } if (err) { cb(err); return; }
records.forEach(function (r) { records.forEach(function (r) {
@ -187,16 +179,16 @@ function getNs(db, ds, results, cb) {
}); });
if (!results.authority.length) { 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) // d.vanityNs should only be vanity nameservers (pointing to this same server)
if (d.vanityNs || results.authority.some(function (ns) { if (d.vanityNs || results.authority.some(function (ns) {
console.log('[debug] ns', 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.length = 0;
results.authority.push(domainToSoa(db, d)); results.authority.push(domainToSoa(engine.primaryNameservers, d));
results.header.rcode = NXDOMAIN; results.header.rcode = NXDOMAIN;
} }
cb(null, results); cb(null, results);
@ -204,8 +196,8 @@ function getNs(db, ds, results, cb) {
}); });
} }
function domainToSoa(db, domain) { function domainToSoa(primaryNameservers, domain) {
var nameservers = domain.vanityNs || db.primaryNameservers; var nameservers = domain.vanityNs || primaryNameservers;
var index = Math.floor(Math.random() * nameservers.length) % nameservers.length; var index = Math.floor(Math.random() * nameservers.length) % nameservers.length;
var nameserver = nameservers[index]; 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'); console.log('[DEV] getSoa entered');
if (!answerSoa) { if (!answerSoa) {
results.authority.push(domainToSoa(db, domain)); results.authority.push(domainToSoa(primaryNameservers, domain));
} else { } else {
results.answer.push(domainToSoa(db, domain)); results.answer.push(domainToSoa(primaryNameservers, domain));
} }
cb(null, results); cb(null, results);
return; return;
} }
module.exports.query = function (input, query, cb) { module.exports.query = function (engine, query, cb) {
/* /*
var fs = require('fs'); var fs = require('fs');
@ -271,11 +263,7 @@ module.exports.query = function (input, query, cb) {
}); });
*/ */
var db;
var qname; var qname;
try {
db = require(input);
} catch(e) { cb(e); return; }
if (!Array.isArray(query.question) || query.question.length < 1) { if (!Array.isArray(query.question) || query.question.length < 1) {
cb(new Error("query is missing question section")); cb(new Error("query is missing question section"));
@ -337,46 +325,48 @@ module.exports.query = function (input, query, cb) {
console.log('[DEV] answerSoa?', answerSoa); console.log('[DEV] answerSoa?', answerSoa);
console.log('[DEV] qnames'); console.log('[DEV] qnames');
console.log(qnames); console.log(qnames);
var myDomains = db.domains.filter(function (d) {
return -1 !== qnames.indexOf(d.id.toLowerCase());
});
// this should result in a REFUSED status return engine.getSoas({ names: qnames}, function (err, myDomains) {
if (!myDomains.length) { console.log('[SOA] looking for', qnames, 'and proudly serving', err, myDomains);
// REFUSED will have no records, so we could still recursion, if enabled if (err) { cb(err); return; }
results.header.rcode = REFUSED;
cb(null, results);
return;
}
myDomains.sort(function (d1, d2) { // this should result in a REFUSED status
if (d1.id.length > d2.id.length) { if (!myDomains.length) {
return -1; // REFUSED will have no records, so we could still recursion, if enabled
} results.header.rcode = REFUSED;
if (d1.id.length < d2.id.length) { cb(null, results);
return 1; return;
}
return 0;
});
//console.log('sorted domains', myDomains);
if (!getNsAlso) {
return getSoa(db, myDomains[0], results, cb, answerSoa);
}
return getNs(db, /*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; }
// has NS records (or SOA record if NS records match the server itself)
if (results.authority.length) {
console.log(results); cb(null, results); return;
} }
// myDomains was sorted such that the longest was first myDomains.sort(function (d1, d2) {
return getSoa(db, myDomains[0], results, cb); if (d1.id.length > d2.id.length) {
return -1;
}
if (d1.id.length < d2.id.length) {
return 1;
}
return 0;
});
//console.log('sorted domains', myDomains);
if (!getNsAlso) {
return getSoa(engine.primaryNameservers, myDomains[0], results, cb, answerSoa);
}
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; }
// has NS records (or SOA record if NS records match the server itself)
if (results.authority.length) {
console.log(results); cb(null, results); return;
}
// myDomains was sorted such that the longest was first
return getSoa(engine.primaryNameservers, myDomains[0], results, cb);
});
}); });
} }
@ -385,7 +375,7 @@ module.exports.query = function (input, query, cb) {
} }
//console.log('[DEV] QUERY NAME', qname); //console.log('[DEV] QUERY NAME', qname);
return getRecords(db, qname, function (err, someRecords) { return getRecords(engine, qname, function (err, someRecords) {
var myRecords; var myRecords;
var nsRecords = []; 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 // 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? // 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. // 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"); console.log("It's a vanity NS");
return false; 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;
};

70
lib/tcpd.js Normal file
View File

@ -0,0 +1,70 @@
'use strict';
module.exports.create = function (cli, dnsd) {
function runTcp() {
var tcpServer = require('net').createServer({ }, function (c) {
c.on('error', function (err) {
console.warn("TCP Connection Error:");
console.warn(err);
});
c.on('data', function (nb) {
//console.log('TCP data.length:', nb.length);
//console.log(nb.toString('hex'));
// DNS packets include a 2-byte length header
var count = nb.length;
var length = nb[0] << 8;
length = length | nb[1];
count -= 2;
// TODO slice?
nb._dnsByteOffset = nb.byteOffset + 2;
if (length !== count) {
console.error("Handling TCP packets > 512 bytes not implemented.");
c.end();
return;
}
// TODO pad two bytes for lengths
dnsd.onMessage(nb, function (err, newAb, dbgmsg) {
var lenbuf = Buffer.from([ newAb.length >> 8, newAb.length & 255 ]);
// TODO XXX generate legit error packet
if (err) { console.error("Error", err); c.end(); return; }
console.log('TCP ' + dbgmsg);
c.write(lenbuf);
c.end(newAb);
});
});
c.on('end', function () {
console.log('TCP client disconnected from server');
});
});
tcpServer.on('error', function (err) {
if ('EADDRINUSE' === err.code) {
console.error("Port '" + cli.port + "' is already in use.");
tcpServer.close();
process.exit(0);
}
if ('EACCES' === err.code) {
console.error("Could not bind on port '" + cli.port + "': EACCESS (you probably need root permissions)");
tcpServer.close();
process.exit(0);
}
console.error("TCP Server Error:");
console.error(err);
tcpServer.close(function () {
setTimeout(runTcp, 1000);
});
});
tcpServer.listen(cli.port, function () {
console.log('TCP Server bound');
});
return tcpServer;
}
return runTcp();
};

58
lib/udpd.js Normal file
View File

@ -0,0 +1,58 @@
'use strict';
module.exports.create = function (cli, dnsd) {
var server = require('dgram').createSocket({
type: cli.udp6 ? 'udp6' : 'udp4'
, reuseAddr: true
});
server.bind({
port: cli.port
, address: cli.address
});
var handlers = {};
handlers.onError = function (err) {
if ('EACCES' === err.code) {
console.error("");
console.error("EACCES: Couldn't bind to port. You probably need to use sudo, authbind, or setcap.");
console.error("");
process.exit(123);
return;
}
console.error("error:", err.stack);
server.close();
};
handlers.onMessage = function (nb, rinfo) {
//console.log('[DEBUG] got a UDP message', nb.length);
//console.log(nb.toString('hex'));
dnsd.onMessage(nb, function (err, newAb, dbgmsg) {
// TODO send legit error message
if (err) { server.send(Buffer.from([0x00])); return; }
server.send(newAb, rinfo.port, rinfo.address, function () {
console.log(dbgmsg, rinfo.port, rinfo.address);
});
});
};
handlers.onListening = function () {
/*jshint validthis:true*/
var server = this;
if (cli.mdns || '224.0.0.251' === cli.nameserver) {
server.setBroadcast(true);
server.addMembership(cli.nameserver);
}
console.log('');
console.log('Bound and Listening:');
console.log(server.address().address + '#' + server.address().port + ' (' + server.type + ')');
};
server.on('error', handlers.onError);
server.on('message', handlers.onMessage);
server.on('listening', handlers.onListening);
return server;
};

View File

@ -1,6 +1,6 @@
{ {
"name": "digd.js", "name": "digd.js",
"version": "1.1.9", "version": "1.2.1",
"description": "A lightweight DNS / mDNS daemon (server) for creating and capturing DNS and mDNS query and response packets to disk as binary and/or JSON. Options are similar to the Unix dig command.", "description": "A lightweight DNS / mDNS daemon (server) for creating and capturing DNS and mDNS query and response packets to disk as binary and/or JSON. Options are similar to the Unix dig command.",
"main": "bin/digd.js", "main": "bin/digd.js",
"homepage": "https://git.coolaj86.com/coolaj86/digd.js", "homepage": "https://git.coolaj86.com/coolaj86/digd.js",