dns-suite.js/dns.packer.js

(function (exports) {
'use strict';

var classes = exports.DNS_CLASSES || require('./dns.classes.js').DNS_CLASSES;
var types = exports.DNS_TYPES || require('./dns.types.js').DNS_TYPES;

var dnspack = exports.DNS_PACKER = {
  pack: function (packet) {

    // TODO update for edns payload size
    var ab = new ArrayBuffer(576);                  // http://serverfault.com/a/587626/93930
    var dv = new DataView(ab);
    var labelsMap = {};
    var total = 0;
    var id = packet.header.id;                      // 2 bytes
    var header = dnspack.packHeader(packet.header); // 2 bytes
    total = 12;                                     // 2+2+2+2 bytes ({qd,an,ns,ar}count)

    dv.setUint16(0, id, false);
    dv.setUint16(2, header, false);
    dv.setUint16(4, packet.question.length, false);
    dv.setUint16(6, packet.answer.length, false);
    dv.setUint16(8, packet.authority.length, false);
    // EDNS is added as an additional with TYPE 41 (OPT, 0x29)
    dv.setUint16(10, packet.additional.length + (packet.payload ? 1 : 0), false);

    function lint(r) {
      if (!r.name) {
        throw new Error("missing name");
      }

      if (!r.class) {
        if (!r.className) {
          throw new Error("no className");
        }
        r.class = classes[r.className];
      }

      if (!r.type) {
        if (!r.typeName) {
          throw new Error("no typeName");
        }
        r.type = types[r.typeName];
        if (!r.type) {
          console.warn("ignoring invalid type '" + r.type + "' for '" + r.name + "', ignoring");
        }
      }
    }

    function packLabelSequence(sequence, terminates) {
      console.log('sequence:', sequence);
      if (labelsMap[sequence]) {
        console.log('cached sequence:', sequence);
        // minimal compression pointer 0xc0 (192)
        dv.setUint8(total, 0xc0, false);
        total += 1;
        dv.setUint8(total, labelsMap[sequence].total, false);
        total += 1;
        return;
      }
      if (terminates) {
        // we don't want non-terminating rdata cached, just terminating names
        labelsMap[total] = { total: total, name: sequence };
        labelsMap[sequence] = labelsMap[total];
      }
      sequence.split('.').forEach(function (label) {
        dv.setUint8(total, label.length, false);
        total += 1;

        label.split('').forEach(function (ch) {
          dv.setUint8(total, ch.charCodeAt(0), false);
          total += 1;
        });
      });
      if (terminates) {
        // trailing 0 (null) as label sequence terminator
        dv.setUint8(total, 0, false);
        total += 1;
      }
    }

    function packQuestion(q) {
      lint(q);

      packLabelSequence(q.name, true);

      dv.setUint16(total, q.type, false);
      total += 2;
      dv.setUint16(total, q.class, false);
      total += 2;
    }

    function packAnswer(a) {
      packQuestion(a);

      if ('undefined' === typeof a.ttl) {
        throw new Error("missing ttl");
      }

      // 32-bit ttl
      dv.setUint32(total, a.ttl, false);
      total += 4;

      if (a.rdata) {
        // 16-bit
        dv.setUint16(total, a.rdata.byteLength, false);
        total += 2;
        (new Uint8Array(a.rdata)).forEach(function (b) {
          dv.setUint8(total, b, false);
          total += 1;
        });
      } else {
        total = dnspack.packRdata(ab, dv, total, a);
      }
    }

    //console.log(packet);
    // fixed size, other than labels

    packet.question.forEach(packQuestion);
    packet.answer.forEach(packAnswer);
    packet.authority.forEach(packAnswer);
    packet.additional.forEach(packAnswer);

    // TODO handle compression pointers
    // TODO handle edns

    return ab.slice(0, total);
  }

  /*
, packLabels: function(map, labels) {
  }
  */

, packHeader: function(h) {
    var val = 0;

    val += (h.qr     << 15) & 0x8000;
    val += (h.opcode << 11) & 0x7800;
    val += (h.aa     << 10) &  0x400;
    val += (h.tc     <<  9) &  0x200;
    val += (h.rd     <<  8) &  0x100;
    val += (h.ra     <<  7) &   0x80;
    val += (h.res1   <<  6) &   0x40;
    val += (h.res2   <<  5) &   0x20;
    val += (h.res3   <<  4) &   0x10;
    val += h.rcode          &    0xF;

    return val;
  }

};
dnspack.packRdata = exports.DNS_RDATA_PACK || require('./dns.rdata.pack.js').DNS_RDATA_PACK;

}('undefined' !== typeof window ? window : exports));