works all-at-once and byte-by-byte

This commit is contained in:
AJ ONeal 2016-09-22 11:06:34 -06:00
parent 2608f2e98d
commit ead5a713da
3 changed files with 130 additions and 25 deletions

View File

@ -19,18 +19,26 @@ function debug(chunk, i, len) {
} }
machine.fns.version = function (chunk) { machine.fns.version = function (chunk) {
console.log('');
console.log('[version]');
if ((255 - machine._version) !== chunk[machine.chunkIndex]) { if ((255 - machine._version) !== chunk[machine.chunkIndex]) {
console.error("not v" + machine._version + " (or data is corrupt)"); console.error("not v" + machine._version + " (or data is corrupt)");
// no idea how to fix this yet // no idea how to fix this yet
} }
machine.chunkIndex += 1; machine.chunkIndex += 1;
return true;
}; };
machine.headerLen = 0; machine.headerLen = 0;
machine.fns.headerLength = function (chunk) { machine.fns.headerLength = function (chunk) {
console.log('');
console.log('[headerLength]');
machine.headerLen = chunk[machine.chunkIndex]; machine.headerLen = chunk[machine.chunkIndex];
machine.chunkIndex += 1; machine.chunkIndex += 1;
return true;
}; };
@ -38,9 +46,12 @@ machine.buf = null;
machine.bufIndex = 0; machine.bufIndex = 0;
//var buf = Buffer.alloc(4096); //var buf = Buffer.alloc(4096);
machine.fns.header = function (chunk) { machine.fns.header = function (chunk) {
console.log('');
console.log('[header]');
var curSize = machine.bufIndex + (chunk.length - machine.chunkIndex); var curSize = machine.bufIndex + (chunk.length - machine.chunkIndex);
var partLen = 0; var partLen = 0;
var str = ''; var str = '';
var part;
if (curSize < machine.headerLen) { if (curSize < machine.headerLen) {
// I still don't have the whole header, // I still don't have the whole header,
@ -48,66 +59,97 @@ machine.fns.header = function (chunk) {
// write these bits, and wait for the next chunk. // write these bits, and wait for the next chunk.
if (!machine.buf) { if (!machine.buf) {
machine.buf = Buffer.alloc(machine.headerLen); machine.buf = Buffer.alloc(machine.headerLen);
console.log('[1a] machine.headerLen:', machine.headerLen);
} }
partLen = machine.headerLen - machine.bufIndex; // WHERE I STOPPED LAST NIGHT:
machine.buf.write(chunk.slice(machine.chunkIndex, machine.chunkIndex + partLen), machine.bufIndex); // partLen should be no more than the available size
partLen = Math.min(machine.headerLen - machine.bufIndex, chunk.length - machine.chunkIndex);
console.log('[1a] chunkIndex:', machine.chunkIndex);
console.log('[1a] chunk.length:', chunk.length);
console.log('[1a] partLen:', partLen);
console.log('[1a] bufIndex:', machine.bufIndex, typeof machine.bufIndex);
part = chunk.slice(machine.chunkIndex, machine.chunkIndex + partLen);
console.log('[1a] part:', part, typeof part);
//machine.buf.write(part.toString('binary'), machine.bufIndex, 'binary');
//part.copy(machine.buf, machine.bufIndex, machine.bufIndex, machine.bufIndex + part.length);
chunk.copy(machine.buf, machine.bufIndex, machine.chunkIndex, machine.chunkIndex + partLen);
machine.chunkIndex += partLen; // this MUST be chunk.length machine.chunkIndex += partLen; // this MUST be chunk.length
machine.bufIndex += partLen; machine.bufIndex += partLen;
console.log('[1a] chunkIndex:', machine.chunkIndex);
console.log('[1a] bufIndex:', machine.bufIndex);
return false;
} }
else { else {
// it's now ready to discover the whole header // it's now ready to discover the whole header
if (machine.buf) { if (machine.buf) {
str += machine.buf.slice(0, machine.bufIndex).toString(); str += machine.buf.slice(0, machine.bufIndex).toString();
} }
console.log('machine.headerLen:', machine.headerLen);
console.log('machine.chunkIndex:', machine.chunkIndex);
partLen = machine.headerLen - str.length; partLen = machine.headerLen - str.length;
str += chunk.slice(machine.chunkIndex, machine.chunkIndex + partLen).toString(); console.log('[1b] partLen:', partLen);
part = chunk.slice(machine.chunkIndex, machine.chunkIndex + partLen);
str += part.toString();
machine.chunkIndex += partLen; machine.chunkIndex += partLen;
console.log('machine.chunkIndex:', machine.chunkIndex);
// TODO maybe just use maxlen of buffer // TODO maybe just use maxlen of buffer
machine.buf = null; // back to null machine.buf = null; // back to null
machine.bufIndex = 0; // back to 0 machine.bufIndex = 0; // back to 0
machine._headers = str.split(/,/g); machine._headers = str.split(/,/g);
console.log(partLen); //console.log(chunk.toString());
console.log(''); console.log('str:', str);
console.log(chunk.toString()); console.log('headers:', machine._headers);
console.log('');
console.log(str);
console.log(machine._headers);
machine.family = machine._headers[0]; machine.family = machine._headers[0];
machine.address = machine._headers[1]; machine.address = machine._headers[1];
machine.port = machine._headers[2]; machine.port = machine._headers[2];
machine.bodyLen = machine._headers[3]; machine.bodyLen = parseInt(machine._headers[3], 10) || -1;
return true;
} }
}; };
machine.fns.data = function (chunk) { machine.fns.data = function (chunk) {
console.log('');
console.log('[data]');
var curSize = machine.bufIndex + (chunk.length - machine.chunkIndex); var curSize = machine.bufIndex + (chunk.length - machine.chunkIndex);
console.log('curSize:', curSize);
console.log('bodyLen:', machine.bodyLen, typeof machine.bodyLen);
var partLen = 0; var partLen = 0;
var buf; var buf;
partLen = Math.min(machine.bodyLen - machine.bufIndex, chunk.length - machine.chunkIndex);
if (curSize < machine.bodyLen) { if (curSize < machine.bodyLen) {
console.log('curSize < bodyLen');
// I still don't have the whole header, // I still don't have the whole header,
// so just create a large enough buffer, // so just create a large enough buffer,
// write these bits, and wait for the next chunk. // write these bits, and wait for the next chunk.
if (!machine.buf) { if (!machine.buf) {
machine.buf = Buffer.alloc(machine.bodyLen); machine.buf = Buffer.alloc(machine.bodyLen);
} }
partLen = machine.bodyLen - machine.bufIndex; console.log('bufIndex:', machine.bufIndex);
machine.buf.write(chunk.slice(machine.chunkIndex, machine.chunkIndex + partLen), machine.bufIndex); console.log('chunkIndex:', machine.chunkIndex);
//machine.chunkIndex += partLen; // this MUST be chunk.length console.log('chunk.length:', chunk.length);
console.log('partLen:', partLen);
//machine.buf.write(chunk.slice(machine.chunkIndex, machine.chunkIndex + partLen), machine.bufIndex);
chunk.copy(machine.buf, machine.bufIndex, machine.chunkIndex, machine.chunkIndex + partLen);
machine.chunkIndex += partLen; // this MUST be chunk.length
machine.bufIndex += partLen; machine.bufIndex += partLen;
return; return false;
} }
// it's now ready to discover the whole header // it's now ready to discover the whole body
if (!machine.buf) { if (!machine.buf) {
buf = chunk; buf = chunk.slice(machine.chunkIndex, machine.chunkIndex + partLen);
machine.chunkIndex = chunk.length; machine.chunkIndex += partLen;
} }
else { else {
partLen = machine.bodyLen - machine.bufIndex; //machine.buf.write(chunk.slice(machine.chunkIndex, machine.chunkIndex + partLen), machine.bufIndex);
machine.buf.write(chunk.slice(machine.chunkIndex, machine.chunkIndex + partLen), machine.bufIndex); chunk.copy(machine.buf, machine.bufIndex, machine.chunkIndex, machine.chunkIndex + partLen);
machine.chunkIndex += partLen; machine.chunkIndex += partLen;
buf = machine.buf;
} }
machine.onMessage({ machine.onMessage({
@ -120,19 +162,26 @@ machine.fns.data = function (chunk) {
// TODO maybe just use maxlen of buffer // TODO maybe just use maxlen of buffer
machine.buf = null; // back to null machine.buf = null; // back to null
machine.bufIndex = 0; // back to 0 machine.bufIndex = 0; // back to 0
return true;
}; };
machine.fns.addChunk = function (chunk) { machine.fns.addChunk = function (chunk) {
console.log('');
console.log('[addChunk]');
machine.chunkIndex = 0; machine.chunkIndex = 0;
while (machine.chunkIndex < chunk.length) { while (machine.chunkIndex < chunk.length) {
console.log(machine.state); console.log('chunkIndex:', machine.chunkIndex, 'state:', machine.state);
machine.fns[machine.states[machine.state]](chunk);
if (true === machine.fns[machine.states[machine.state]](chunk)) {
machine.state += 1; machine.state += 1;
machine.state %= machine.states_length; machine.state %= machine.states_length;
} }
}
}; };
module.exports = machine; module.exports = machine;
process.on('uncaughtException', function () { process.on('uncaughtException', function (err) {
debug(Buffer.from('0')); console.error(err);
debug(Buffer.from([0]));
}); });

BIN
sni.hello.bin Normal file

Binary file not shown.

56
test-machine.js Normal file
View File

@ -0,0 +1,56 @@
'use strict';
var sni = require('sni');
var machine = require('./machine.js');
var hello = require('fs').readFileSync('./sni.hello.bin');
var version = 1;
var header = 'IPv4,127.0.0.1,443,' + hello.byteLength;
var buf = Buffer.concat([
Buffer.from([ 255 - version, header.length ])
, Buffer.from(header)
, hello
]);
console.log('sni(hello):', sni(hello));
console.log('hello.byteLength:', hello.byteLength);
var services = { 'ssh': 22, 'http': 4080, 'https': 8443 };
//var clients = {};
machine.onMessage = function (opts) {
var id = opts.family + ',' + opts.address + ',' + opts.port;
var service = 'https';
var port = services[service];
/*
if (clients[id]) {
clients[id].write(opts.data);
return;
}
*/
var servername = sni(opts.data);
console.log('');
console.log('');
console.log('[onMessage]');
console.log('equal:', opts.data.equals(hello));
console.log('byteLength:', opts.data.byteLength);
console.log('port:', port);
if (!servername) {
console.warn("no servername found for '" + id + "'");
return;
}
console.log("servername: '" + servername + "'");
};
// full message in one go
machine.fns.addChunk(buf);
console.log('');
console.log('');
// messages one byte at a time
buf.forEach(function (byte) {
machine.fns.addChunk(Buffer.from([ byte ]));
});