proxy-packer.js/test/parse.js

'use strict';

var sni = require('sni');
var hello = require('fs').readFileSync(__dirname + '/sni.hello.bin');
var version = 1;
function getAddress() {
  return {
    family: 'IPv4'
  , address: '127.0.1.1'
  , port: 4321
  , service: 'foo-https'
  , serviceport: 443
  , name: 'foo-pokemap.hellabit.com'
  };
}
var addr = getAddress();
var connectionHeader = addr.family + ',' + addr.address + ',' + addr.port
  + ',0,connection,'
  + (addr.serviceport || '') + ',' + (addr.name || '') + ',' + (addr.service || '')
  ;
var           header = addr.family + ',' + addr.address + ',' + addr.port
  + ',' + hello.byteLength + ',' + (addr.service || '') + ','
  + (addr.serviceport || '') + ',' + (addr.name || '')
  ;
var        endHeader = addr.family + ',' + addr.address + ',' + addr.port
  + ',0,end,'
  + (addr.serviceport || '') + ',' + (addr.name || '')
  ;
var buf = Buffer.concat([
  Buffer.from([ 255 - version, connectionHeader.length ])
, Buffer.from(connectionHeader)
, Buffer.from([ 255 - version, header.length ])
, Buffer.from(header)
, hello
, Buffer.from([ 255 - version, endHeader.length ])
, Buffer.from(endHeader)
]);
var services = { 'ssh': 22, 'http': 4080, 'https': 8443 };
var clients = {};
var count = 0;
var packer = require('../');
var machine = packer.create({
  onconnection: function (tun) {
    console.info('');
    if (!tun.service || 'connection' === tun.service) {
      throw new Error("missing service: " + JSON.stringify(tun));
    }
    console.info('[onConnection]');
    count += 1;
  }
, onmessage: function (tun) {
    //console.log('onmessage', tun);
    var id = tun.family + ',' + tun.address + ',' + tun.port;
    var service = 'https';
    var port = services[service];
    var servername = sni(tun.data);

    console.info('[onMessage]', service, port, servername, tun.data.byteLength);
    if (!tun.data.equals(hello)) {
      throw new Error("'data' packet is not equal to original 'hello' packet");
    }
    //console.log('all', tun.data.byteLength, 'bytes are equal');
    //console.log('src:', tun.family, tun.address + ':' + tun.port + ':' + tun.serviceport);
    //console.log('dst:', 'IPv4 127.0.0.1:' + port);

    if (!clients[id]) {
      clients[id] = true;
      if (!servername) {
        throw new Error("no servername found for '" + id + "'");
      }
      //console.log("servername: '" + servername + "'", tun.name);
    }

    count += 1;
  }
, onerror: function () {
    throw new Error("Did not expect onerror");
  }
, onend: function () {
    console.info('[onEnd]');
    count += 1;
  }
});

var packts, packed;

packts = [];
packts.push(packer.packHeader(getAddress(), null, 'connection'));
//packts.push(packer.pack(address, hello));
packts.push(packer.packHeader(getAddress(), hello));
packts.push(hello);
packts.push(packer.packHeader(getAddress(), null, 'end'));
packed = Buffer.concat(packts);

if (!packed.equals(buf)) {
  console.error("");
  console.error(buf.toString('hex') === packed.toString('hex'));
  console.error("");
  console.error("auto-packed:");
  console.error(packed.toString('hex'), packed.byteLength);
  console.error("");
  console.error("hand-packed:");
  console.error(buf.toString('hex'), buf.byteLength);
  console.error("");
  throw new Error("packer (new) did not pack as expected");
}

packts = [];
packts.push(packer.pack(getAddress(), null, 'connection'));
packts.push(packer.pack(getAddress(), hello));
//packts.push(packer.packHeader(getAddress(), hello));
//packts.push(hello);
packts.push(packer.pack(getAddress(), null, 'end'));
packed = Buffer.concat(packts);

// XXX TODO REMOVE
//
// Nasty fix for short-term backwards-compat
//
// In the old way of doing things we always have at least one byte
// of data (due to a parser bug which has now been fixed) and so
// there are two strings padded with a space which gives the
// data a length of 1 rather than 0
//
// Here all four of those instances are replaced, but it requires
// maching a few things on either side.
//
// Only 6 bytes are changed - two 1 => 0, four ' ' => ''
var hex = packed.toString('hex')
  //.replace(/2c313939/, '2c30')
  .replace(/32312c312c636f/, '32312c302c636f')
  .replace(/3332312c312c656e64/, '3332312c302c656e64')
  .replace(/7320/, '73')
  .replace(/20$/, '')
  ;
if (hex !== buf.toString('hex')) {
  console.error("");
  console.error(buf.toString('hex') === hex);
  console.error("");
  console.error("auto-packed:");
  console.error(hex, packed.byteLength);
  console.error("");
  console.error("hand-packed:");
  console.error(buf.toString('hex'), buf.byteLength);
  console.error("");
  throw new Error("packer (old) did not pack as expected");
}


console.info('');

// full message in one go
// 223 = 2 + 22 + 199
console.info('[WHOLE BUFFER]', 2, header.length, hello.length, buf.byteLength);
clients = {};
machine.fns.addChunk(buf);
console.info('');


// messages one byte at a time
console.info('[BYTE-BY-BYTE BUFFER]', 1);
clients = {};
buf.forEach(function (byte) {
  machine.fns.addChunk(Buffer.from([ byte ]));
});
console.info('');


// split messages in overlapping thirds
// 0-2      (2)
// 2-24     (22)
// 24-223   (199)
// 223-225  (2)
// 225-247  (22)
// 247-446  (199)
buf = Buffer.concat([ buf, buf ]);
console.info('[OVERLAPPING BUFFERS]', buf.length);
clients = {};
[ buf.slice(0, 7)                 // version + header
, buf.slice(7, 14)                // header
, buf.slice(14, 21)               // header
, buf.slice(21, 28)               // header + body
, buf.slice(28, 217)              // body
, buf.slice(217, 224)             // body + version
, buf.slice(224, 238)             // version + header
, buf.slice(238, buf.byteLength)  // header + body
].forEach(function (buf) {
  machine.fns.addChunk(Buffer.from(buf));
});
console.info('');

process.on('exit', function () {
  if (count !== 12) {
    throw new Error("should have delivered 12 messages, not " + count);
  }
  console.info('TESTS PASS');
  console.info('');
});