handle bad packet more correctly

lib/tcpd.js

@@ -27,9 +27,10 @@ module.exports.create = function (cli, dnsd) {
 
         // 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; }
+
+          var lenbuf = Buffer.from([ newAb.length >> 8, newAb.length & 255 ]);
           console.log('TCP ' + dbgmsg);
 
           c.write(lenbuf);
@@ -53,10 +54,11 @@ module.exports.create = function (cli, dnsd) {
         process.exit(0);
       }
       console.error("TCP Server Error:");
-      console.error(err);
+      console.error(err.stack);
       tcpServer.close(function () {
         setTimeout(runTcp, 1000);
       });
+      //throw new Error(err);
     });
 
     tcpServer.listen(cli.port, function () {

lib/udpd.js

@@ -1,6 +1,7 @@
 'use strict';
 
 module.exports.create = function (cli, dnsd) {
+  function runUdp() {
     var server = require('dgram').createSocket({
       type: cli.udp6 ? 'udp6' : 'udp4'
     , reuseAddr: true
@@ -19,8 +20,12 @@ module.exports.create = function (cli, dnsd) {
         process.exit(123);
         return;
       }
-    console.error("error:", err.stack);
+      console.error("UDP Server Error:");
-    server.close();
+      console.error(err.stack);
+      server.close(function () {
+        setTimeout(runUdp, 1000);
+      });
+      //throw new Error(err);
     };
 
     handlers.onMessage = function (nb, rinfo) {
@@ -29,9 +34,9 @@ module.exports.create = function (cli, dnsd) {
 
       dnsd.onMessage(nb, function (err, newAb, dbgmsg) {
         // TODO send legit error message
-      if (err) { server.send(Buffer.from([0x00])); return; }
+        if (err) { server.send(Buffer.from([0x00]), rinfo.port, rinfo.address); return; }
         server.send(newAb, rinfo.port, rinfo.address, function () {
-        console.log(dbgmsg, rinfo.port, rinfo.address);
+          console.log('[dnsd.onMessage] ' + dbgmsg, rinfo.port, rinfo.address);
         });
       });
     };
@@ -55,4 +60,7 @@ module.exports.create = function (cli, dnsd) {
     server.on('listening', handlers.onListening);
 
     return server;
+  }
+
+  return runUdp();
 };

package.json

@@ -1,6 +1,6 @@
 {
   "name": "digd.js",
-  "version": "1.2.1",
+  "version": "1.2.2",
   "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",
   "homepage": "https://git.coolaj86.com/coolaj86/digd.js",