joshua
/
websockify
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add recording feature to websockify.js. Closes GH-249

This commit is contained in:
Aaron C. de Bruyn 2017-10-21 11:01:55 -07:00
parent 6b1d42e643
commit 46406c76e7
1 changed files with 52 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
other/js/websockify.js
View File

@ -29,6 +29,8 @@ var argv = require('optimist').argv,
// Handle new WebSocket client
new_client = function(client, req) {
var clientAddr = client._socket.remoteAddress, log;
var start_time = new Date().getTime();
console.log(req ? req.url : client.upgradeReq.url);
log = function (msg) {
console.log(' ' + clientAddr + ': '+ msg);
@ -36,11 +38,25 @@ new_client = function(client, req) {
log('WebSocket connection');
log('Version ' + client.protocolVersion + ', subprotocol: ' + client.protocol);
if (argv.record) {
var rs = fs.createWriteStream(argv.record + '/' + new Date().toISOString());
rs.write('var VNC_frame_data = [\n');
} else {
var rs = null;
}
var target = net.createConnection(target_port,target_host, function() {
log('connected to target');
});
target.on('data', function(data) {
//log("sending message: " + data);
if (rs) {
var tdelta = Math.floor(new Date().getTime()) - start_time;
var rsdata = '\'{' + tdelta + '{' + decodeBuffer(data) + '\',\n';
rs.write(rsdata);
}
try {
client.send(data);
} catch(e) {
@ -51,15 +67,28 @@ new_client = function(client, req) {
target.on('end', function() {
log('target disconnected');
client.close();
if (rs) {
rs.end('\'EOF\'];\n');
}
});
target.on('error', function() {
log('target connection error');
target.end();
client.close();
if (rs) {
rs.end('\'EOF\'];\n');
}
});
client.on('message', function(msg) {
//log('got message: ' + msg);
if (rs) {
var rdelta = Math.floor(new Date().getTime()) - start_time;
var rsdata = ('\'}' + rdelta + '}' + decodeBuffer(msg) + '\',\n');
~ rs.write(rsdata);
}
target.write(msg);
});
client.on('close', function(code, reason) {
@ -72,6 +101,28 @@ new_client = function(client, req) {
});
};
function decodeBuffer(buf) {
var returnString = '';
for (var i = 0; i < buf.length; i++) {
if (buf[i] >= 48 && buf[i] <= 90) {
returnString += String.fromCharCode(buf[i]);
} else if (buf[i] === 95) {
returnString += String.fromCharCode(buf[i]);
} else if (buf[i] >= 97 && buf[i] <= 122) {
returnString += String.fromCharCode(buf[i]);
} else {
var charToConvert = buf[i].toString(16);
if (charToConvert.length === 0) {
returnString += '\\x00';
} else if (charToConvert.length === 1) {
returnString += '\\x0' + charToConvert;
} else {
returnString += '\\x' + charToConvert;
}
}
}
return returnString;
}
// Send an HTTP error response
http_error = function (response, code, msg) {
@ -147,7 +198,7 @@ try {
throw("illegal port");
}
} catch(e) {
console.error("websockify.js [--web web_dir] [--cert cert.pem [--key key.pem]] [source_addr:]source_port target_addr:target_port");
console.error("websockify.js [--web web_dir] [--cert cert.pem [--key key.pem]] [--record dir] [source_addr:]source_port target_addr:target_port");
process.exit(2);
}