proxy-packer.js/test/parse.js

229 lines
5.6 KiB
JavaScript
Raw Normal View History

2016-09-22 19:11:54 +00:00
'use strict';
var sni = require('sni');
2018-05-31 04:48:32 +00:00
var hello = require('fs').readFileSync(__dirname + '/sni.hello.bin');
2016-09-22 19:11:54 +00:00
var version = 1;
function getAddress() {
2019-09-03 15:21:08 +00:00
return {
family: 'IPv4',
address: '127.0.1.1',
port: 4321,
service: 'foo-https',
serviceport: 443,
name: 'foo-pokemap.hellabit.com'
};
}
var addr = getAddress();
2019-09-03 15:21:08 +00:00
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 || '');
2016-09-22 19:11:54 +00:00
var buf = Buffer.concat([
2019-09-03 15:21:08 +00:00
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)
2016-09-22 19:11:54 +00:00
]);
2019-09-03 15:21:08 +00:00
var services = { ssh: 22, http: 4080, https: 8443 };
2016-09-22 19:11:54 +00:00
var clients = {};
var count = 0;
2018-05-31 04:48:32 +00:00
var packer = require('../');
2016-09-22 19:31:31 +00:00
var machine = packer.create({
2019-09-03 15:21:08 +00:00
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;
}
2016-09-22 19:11:54 +00:00
});
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);
2016-09-22 19:11:54 +00:00
2016-09-22 19:31:31 +00:00
if (!packed.equals(buf)) {
2019-09-03 15:21:08 +00:00
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 ' ' => ''
2019-09-03 15:21:08 +00:00
var hex = packed
.toString('hex')
//.replace(/2c313939/, '2c30')
.replace(/32312c312c636f/, '32312c302c636f')
.replace(/3332312c312c656e64/, '3332312c302c656e64')
.replace(/7320/, '73')
.replace(/20$/, '');
if (hex !== buf.toString('hex')) {
2019-09-03 15:21:08 +00:00
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');
2016-09-22 19:31:31 +00:00
}
2016-09-22 19:11:54 +00:00
console.info('');
2016-09-22 19:11:54 +00:00
// full message in one go
// 223 = 2 + 22 + 199
console.info('[WHOLE BUFFER]', 2, header.length, hello.length, buf.byteLength);
2016-09-22 19:11:54 +00:00
clients = {};
machine.fns.addChunk(buf);
console.info('');
2016-09-22 19:11:54 +00:00
// messages one byte at a time
console.info('[BYTE-BY-BYTE BUFFER]', 1);
2016-09-22 19:11:54 +00:00
clients = {};
2019-09-03 15:21:08 +00:00
buf.forEach(function(byte) {
machine.fns.addChunk(Buffer.from([byte]));
2016-09-22 19:11:54 +00:00
});
console.info('');
2016-09-22 19:11:54 +00:00
// split messages in overlapping thirds
// 0-2 (2)
// 2-24 (22)
// 24-223 (199)
// 223-225 (2)
// 225-247 (22)
// 247-446 (199)
2019-09-03 15:21:08 +00:00
buf = Buffer.concat([buf, buf]);
console.info('[OVERLAPPING BUFFERS]', buf.length);
2016-09-22 19:11:54 +00:00
clients = {};
2019-09-03 15:21:08 +00:00
[
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));
2016-09-22 19:11:54 +00:00
});
console.info('');
2016-09-22 19:11:54 +00:00
2019-09-03 15:21:08 +00:00
process.on('exit', function() {
if (count !== 12) {
throw new Error('should have delivered 12 messages, not ' + count);
}
console.info('TESTS PASS');
console.info('');
2016-09-22 19:11:54 +00:00
});