WebTTY: remove polling.

Communication now uses WebSockets only.
This commit is contained in:
John "Elwin" Edwards 2014-01-27 16:02:27 -08:00
parent de69f47fc1
commit 64d9d8cd54
2 changed files with 19 additions and 390 deletions

197
webtty.js
View file

@ -11,7 +11,7 @@ var webSocketServer = require(path.join(localModules, "websocket")).server;
var serveStaticRoot = fs.realpathSync(".");
var sessions = {};
var sessionsWS = {};
var nsessid = 0;
var env_dontuse = {"TMUX": true, "TMUX_PANE": true};
@ -79,94 +79,11 @@ function TermSessionWS(conn, h, w) {
ss.conn.sendUTF(JSON.stringify({"t": "q"}));
}
};
sessions[nsessid++] = this;
this.conn.sendUTF(JSON.stringify({"t": "l", "w": w, "h": h}));
console.log("New WebSocket connection.");
}
function TermSession(sessid, h, w) {
/* Set up the sizes. */
w = Math.floor(Number(w));
if (!(w > 0 && w < 256))
w = 80;
this.w = w;
h = Math.floor(Number(h));
if (!(h > 0 && h < 256))
h = 25;
this.h = h;
/* Customize the environment. */
var childenv = {};
for (var key in process.env) {
if (!(key in env_dontuse))
childenv[key] = process.env[key];
}
var spawnopts = {"env": childenv, "cwd": process.env["HOME"],
"rows": this.h, "cols": this.w};
this.term = pty.spawn("bash", [], spawnopts);
var ss = this;
/* Eventually we'll need to make sure the sessid isn't in use yet. */
this.sessid = sessid;
this.alive = true;
this.data = []; // Buffer for the process' output.
this.nsend = 0; // Number to use for the next message sent.
this.nrecv = 0; // Number expected on the next message received.
this.msgQ = []; // Queue for messages that arrived out of order.
this.term.on("data", function (buf) {
ss.data.push(buf);
});
this.term.on("exit", function () {
ss.alive = false;
/* Wait for all the data to get collected */
setTimeout(ss.cleanup, 1000);
});
this.write = function (data, n) {
if (!this.alive) {
/* Throw some kind of exception? */
return;
}
if (n !== this.nrecv) {
console.log("Session " + this.sessid + ": Expected message " + this.nrecv + ", got " + n);
}
this.nrecv = n + 1;
this.term.write(data);
};
this.read = function () {
if (this.data.length == 0)
return null;
var pos = 0;
var i = 0;
for (i = 0; i < this.data.length; i++)
pos += Buffer.byteLength(this.data[i]);
var nbuf = new Buffer(pos);
var tptr;
pos = 0;
while (this.data.length > 0) {
tptr = new Buffer(this.data.shift());
tptr.copy(nbuf, pos);
pos += tptr.length;
}
return nbuf;
};
this.close = function () {
if (this.alive)
this.term.kill('SIGHUP');
};
this.cleanup = function () {
/* Call this when the child is dead. */
if (this.alive)
return;
/* Give the client a chance to read any leftover data. */
if (ss.data.length > 0)
setTimeout(ss.remove, 8000);
else
ss.remove();
};
this.remove = function () {
delete sessions[ss.sessid];
console.log("Session " + this.sessid + " removed.");
};
sessions[sessid] = this;
}
function randkey() {
rnum = Math.floor(Math.random() * 65536 * 65536);
hexstr = rnum.toString(16);
@ -226,45 +143,6 @@ function getFormValues(formtext) {
return jsonobj;
}
function login(req, res, formdata) {
var resheaders = {'Content-Type': 'text/plain'};
var sessid = randkey();
/* The TermSession constructor will check these thoroughly too, but
* you can't be too suspicious of client-supplied data. */
var w = 80;
var h = 25;
var t;
if ("w" in formdata) {
t = Math.floor(Number(formdata["w"]));
if (t > 0 && t < 256)
w = t;
}
if ("h" in formdata) {
t = Math.floor(Number(formdata["h"]));
if (t > 0 && t < 256)
h = t;
}
var nsession = new TermSession(sessid, h, w);
resheaders["Set-Cookie"] = "ID=" + sessid;
res.writeHead(200, resheaders);
var logindict = {"login": true, "id": sessid, "w": w, "h": h};
res.write(JSON.stringify(logindict));
res.end();
console.log("Started new session with key " + sessid + ", pid " + nsession.term.pid);
return;
}
function findTermSession(req) {
var cookies = getCookies(req);
if ("id" in cookies) {
var sessid = cookies["id"];
if (sessid in sessions) {
return sessions[sessid];
}
}
return null;
}
function serveStatic(req, res, fname) {
var nname = path.normalize(fname);
if (nname == "" || nname == "/")
@ -306,27 +184,6 @@ function serveStatic(req, res, fname) {
return;
}
function readFeed(res, term) {
res.writeHead(200, { "Content-Type": "text/plain" });
if (term) {
var answer = {};
var result = term.read();
if (result == null) {
answer["t"] = "n";
}
else {
answer["t"] = "d";
answer["d"] = result.toString("hex");
answer["n"] = term.nsend++;
}
res.write(JSON.stringify(answer));
res.end();
}
else {
sendError(res, 1);
}
}
var errorcodes = [ "Generic Error", "Not logged in", "Invalid data" ];
function sendError(res, ecode) {
@ -353,55 +210,14 @@ function handler(req, res) {
/* This will send the response once the whole request is here. */
function respond() {
var target = url.parse(req.url).pathname;
var cterm = findTermSession(req);
/* First figure out if the client is POSTing to a command interface. */
/* Currently only static files and WebSockets are needed. */
if (req.method == 'POST') {
formdata = getFormValues(reqbody);
if (target == '/feed') {
if (!cterm) {
sendError(res, 1);
return;
}
if (formdata["t"] == "q") {
/* The client wants to quit. */
// FIXME need to send a message back to the client
cterm.close();
}
else if (formdata["t"] == "d" && typeof(formdata["d"]) == "string") {
/* process the keys */
hexstr = formdata["d"].replace(/[^0-9a-f]/gi, "");
if (hexstr.length % 2 != 0) {
sendError(res, 2);
return;
}
keybuf = new Buffer(hexstr, "hex");
cterm.write(keybuf, formdata["n"]);
}
readFeed(res, cterm);
}
else if (target == "/login") {
login(req, res, formdata);
}
else {
res.writeHead(405, resheaders);
res.end();
}
res.writeHead(405, resheaders);
res.end();
}
else if (req.method == 'GET' || req.method == 'HEAD') {
if (target == '/feed') {
if (!cterm) {
sendError(res, 1);
return;
}
readFeed(res, cterm);
}
/* Default page, create a new term */
/* FIXME New term not created anymore, is a special case still needed? */
else if (target == '/') {
serveStatic(req, res, "/");
}
else /* Go look for it in the filesystem */
serveStatic(req, res, target);
serveStatic(req, res, target);
}
else { /* Some other method */
res.writeHead(501, resheaders);
@ -411,7 +227,6 @@ function handler(req, res) {
return;
}
req.on('end', respond);
}
process.on("exit", function () {