Mercurial > hg > rlgwebd
view webtty.js @ 0:bd412f63ce0d
Put this project under version control, finally.
author | John "Elwin" Edwards <elwin@sdf.org> |
---|---|
date | Sun, 06 May 2012 08:45:40 -0700 |
parents | |
children | 98bf7c94c954 |
line wrap: on
line source
#!/usr/bin/env node var http = require('http'); var url = require('url'); var path = require('path'); var fs = require('fs'); //var tty = require("tty"); var child_process = require("child_process"); var serveStaticRoot = "/home/elwin/hk/nodejs/rlg/s/"; var sessions = {}; /* Constructor for TermSessions. Note that it opens the terminal and * adds itself to the sessions dict. */ function TermSession(sessid) { //var pterm = tty.open("/bin/bash"); this.child = child_process.spawn("./ptywreck/ptyhelperC", ["bash"]); var ss = this; /* Eventually we'll need to make sure the sessid isn't in use yet. */ this.sessid = sessid; //this.ptmx = pterm[0]; //this.child = pterm[1]; this.alive = true; this.data = []; this.child.stdout.on("data", function (buf) { ss.data.push(buf); }); this.child.stderr.on("data", function (buf) { ss.data.push(buf); }); this.child.on("exit", function (code, signal) { ss.exitcode = (code != null ? code : 255); ss.alive = false; /* Wait for all the data to get collected */ setTimeout(ss.cleanup, 1000); }); this.write = function (data) { if (this.alive) this.child.stdin.write(data); /* Otherwise, throw some kind of exception? */ }; 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 += this.data[i].length; var nbuf = new Buffer(pos); var tptr; pos = 0; while (this.data.length > 0) { tptr = this.data.shift(); tptr.copy(nbuf, pos); pos += tptr.length; } return nbuf; }; this.close = function () { if (this.alive) this.child.kill('SIGHUP'); }; this.cleanup = function () { /* Call this when the child is dead. */ if (this.alive) return; //ss.ptmx.destroy(); /* 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); while (hexstr.length < 8) hexstr = "0" + hexstr; return hexstr; } /* Returns a list of the cookies in the request, obviously. */ function getCookies(req) { cookies = []; if ("cookie" in req.headers) { cookstrs = req.headers["cookie"].split("; "); for (var i = 0; i < cookstrs.length; i++) { eqsign = cookstrs[i].indexOf("="); if (eqsign > 0) { name = cookstrs[i].slice(0, eqsign).toLowerCase(); val = cookstrs[i].slice(eqsign + 1); cookies[name] = val; } else if (eqsign < 0) cookies[cookstrs[i]] = null; } } return cookies; } function urlDec(encstr) { var decstr = ""; var tnum; for (var i = 0; i < encstr.length; i++) { if (encstr.charAt(i) == "+") decstr += " "; else if (encstr.charAt(i) == "%") { tnum = Number("0x" + encstr.slice(i + 1, 2)); if (!isNaN(tnum) && tnum >= 0) decstr += String.fromCharCode(tnum); i += 2; } else decstr += encstr.charAt(i); } return decstr; } /* Returns the contents of a form */ function getFormValues(formtext) { var pairstrs = formtext.split("&"); var data = {}; for (var i = 0; i < pairstrs.length; i++) { var eqsign = pairstrs[i].indexOf("="); if (eqsign > 0) { rawname = pairstrs[i].slice(0, eqsign); rawval = pairstrs[i].slice(eqsign + 1); name = urlDec(rawname); val = urlDec(rawval); if (!(name in data)) data[name] = []; data[name].push(val); } } return data; } function login(req, res, formdata) { var resheaders = {'Content-Type': 'text/plain'}; var sessid = randkey(); var nsession = new TermSession(sessid); resheaders["Set-Cookie"] = "ID=" + sessid; res.writeHead(200, resheaders); res.write("l1\n" + sessid + "\n"); res.end(); console.log("Started new session with key " + sessid + ", pid " + nsession.child.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 == "/") nname = "index.html"; if (nname.match(/\/$/)) path.join(nname, "index.html"); /* it was a directory */ var realname = path.join(serveStaticRoot, nname); var extension = path.extname(realname); path.exists(realname, function (exists) { var resheaders = {}; if (!exists || !extension || extension == ".html") resheaders["Content-Type"] = "text/html"; else if (extension == ".png") resheaders["Content-Type"] = "image/png"; else if (extension == ".css") resheaders["Content-Type"] = "text/css"; else if (extension == ".js") resheaders["Content-Type"] = "text/javascript"; else if (extension == ".svg") resheaders["Content-Type"] = "image/svg+xml"; else resheaders["Content-Type"] = "application/octet-stream"; if (exists) { /* Not nice, not sensible. First see if it's readable, then respond * 200 or 500. Don't throw nasty errors. */ res.writeHead(200, resheaders); fs.readFile(realname, function (error, data) { if (error) throw error; res.write(data); res.end(); }); } else { res.writeHead(404, resheaders); res.write("<html><head><title>" + nname + "</title></head>\n<body><h1>" + nname + " Not Found</h1></body></html>\n"); res.end(); } }); return; } function readFeed(res, term) { res.writeHead(200, { "Content-Type": "text/plain" }); if (term) { var result = term.read(); if (result == null) resultstr = ""; else resultstr = result.toString("hex"); res.write("d" + resultstr.length.toString() + "\n" + resultstr + "\n"); } else { //console.log("Where's the term?"); res.write("d0\n\n"); } } var errorcodes = [ "Generic Error", "Not logged in", "Invalid data" ]; function sendError(res, ecode) { res.writeHead(200, { "Content-Type": "text/plain" }); if (ecode < errorcodes.length && ecode > 0) res.write("E" + ecode + '\n' + errorcodes[ecode] + '\n'); else res.write("E0\nGeneric Error\n"); res.end(); } function handler(req, res) { /* default headers for the response */ var resheaders = {'Content-Type': 'text/html'}; /* The request body will be added to this as it arrives. */ var reqbody = ""; var formdata; /* Register a listener to get the body. */ function moredata(chunk) { reqbody += chunk; } req.on('data', moredata); /* 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. */ if (req.method == 'POST') { formdata = getFormValues(reqbody); if (target == '/feed') { if (!cterm) { sendError(res, 1); return; } if (formdata["quit"] == "quit") { /* The client wants to quit. */ // FIXME need to send a message back to the client cterm.close(); } else if (formdata["keys"]) { /* process the keys */ hexstr = formdata["keys"][0].replace(/[^0-9a-f]/gi, ""); if (hexstr.length % 2 != 0) { sendError(res, 2); return; } keybuf = new Buffer(hexstr, "hex"); cterm.write(keybuf); } readFeed(res, cterm); res.end(); } else if (target == "/login") { login(req, res, formdata); } else { 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); res.end(); } /* 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); } else { /* Some other method */ res.writeHead(501, resheaders); res.write("<html><head><title>501</title></head>\n<body><h1>501 Not Implemented</h1></body></html>\n"); res.end(); } return; } req.on('end', respond); } process.on("exit", function () { for (var sessid in sessions) { if (sessions[sessid].alive) sessions[sessid].child.kill('SIGHUP'); } console.log("Quitting..."); return; }); process.env["TERM"] = "xterm-256color"; http.createServer(handler).listen(8080, "127.0.0.1"); console.log('Server running at http://127.0.0.1:8080/');