#!/usr/bin/env node

var http = require('http');
var net = require('net');
var url = require('url');
var path = require('path');
var fs = require('fs');
var events = require('events');
var child_process = require('child_process');
// Dependencies
var posix = require("posix");
var pty = require("pty.js");
var WebSocketServer = require("websocket").server;

/* Configuration variables */
// The first file is NOT in the chroot.
var ctlsocket = "/var/local/rlgwebd/ctl";
var httpPort = 8080;
var chrootDir = "/var/dgl/";
var dropToUser = "rodney";
var serveStaticRoot = "/var/www/"; // inside the chroot

/* Data on the games available. */
var games = {
  "rogue3": {
    "name": "Rogue V3",
    "uname": "rogue3",
    "suffix": ".r3sav",
    "path": "/usr/bin/rogue3",
    "clear": new Buffer([27, 91, 72, 27, 91, 50, 74]) // CSI H CSI 2J
  },
  "rogue4": {
    "name": "Rogue V4",
    "uname": "rogue4",
    "suffix": ".r4sav",
    "path": "/usr/bin/rogue4",
    "clear": new Buffer([27, 91, 72, 27, 91, 50, 74]) // CSI H CSI 2J
  },
  "rogue5": {
    "name": "Rogue V5",
    "uname": "rogue5",
    "suffix": ".r5sav",
    "path": "/usr/bin/rogue5",
    "clear": new Buffer([27, 91, 72, 27, 91, 50, 74]) // CSI H CSI 2J
  },
  "srogue": {
    "name": "Super-Rogue",
    "uname": "srogue",
    "suffix": ".srsav",
    "path": "/usr/bin/srogue",
    "clear": new Buffer([27, 91, 72, 27, 91, 50, 74]) // CSI H CSI 2J
  },
  "arogue5": {
    "name": "Advanced Rogue 5",
    "uname": "arogue5",
    "suffix": ".ar5sav",
    "path": "/usr/bin/arogue5",
    "clear": new Buffer([27, 91, 72, 27, 91, 50, 74]) // CSI H CSI 2J
  }
};

/* Global state */
var logins = {};
var sessions = {};
var dglgames = {};
var allowlogin = true;
var gamemux = new events.EventEmitter();

/* Constructor.  A TermSession handles a pty and the game running on it.
 *   gname: (String) Name of the game to launch.
 *   pname: (String) The player's name.
 *   wsReq: (WebSocketRequest) The request from the client.
 *
 *  Events:
 *   "data": Data generated by child. Parameters: buf (Buffer)
 *   "exit": Child terminated. Parameters: none
 */
function TermSession(gname, pname, wsReq) {
  var ss = this;
  /* Subclass EventEmitter to do the hard work. */
  events.EventEmitter.call(this);
  /* Don't launch anything that's not a real game. */
  if (gname in games) {
    this.game = games[gname];
  }
  else {
    this.failed = true;
    wsReq.reject(404, errorcodes[2], "No such game");
    tslog("Game %s is not available", game);
    return;
  }
  this.pname = pname;
  /* Set up the sizes. */
  this.w = Math.floor(Number(wsReq.resourceURL.query.w));
  if (!(this.w > 0 && this.w < 256))
    this.w = 80;
  this.h = Math.floor(Number(wsReq.resourceURL.query.h));
  if (!(this.h > 0 && this.h < 256))
    this.h = 24;
  /* Environment. */
  var childenv = {};
  for (var key in process.env) {
    childenv[key] = process.env[key];
  }
  var args = ["-n", this.pname];
  var spawnopts = {"env": childenv, "cwd": "/", "rows": this.h, "cols": this.w,
                   "name": "xterm-256color"};
  this.term = pty.spawn(this.game.path, args, spawnopts);
  tslog("%s playing %s (pid %d)", this.pname, this.game.uname, this.term.pid);
  this.failed = false;
  sessions[this.game.uname + "/" + this.pname] = this;
  gamemux.emit('begin', this.game.uname, this.pname);
  /* Set up the lockfile and ttyrec */
  this.lasttime = new Date();
  var ts = timestamp(this.lasttime);
  var progressdir = path.join("/dgldir/inprogress", this.game.uname);
  this.lock = path.join(progressdir, this.pname + ":node:" + ts + ".ttyrec");
  var lmsg = this.term.pid.toString() + '\n' + this.h + '\n' + this.w + '\n'; 
  fs.writeFile(this.lock, lmsg, "utf8"); 
  var ttyrec = path.join("/dgldir/ttyrec", this.pname, this.game.uname, 
               ts + ".ttyrec");
  this.record = fs.createWriteStream(ttyrec, { mode: 0664 });
  /* Holds the output since the last screen clear, so watchers can begin
   * with a complete screen. */
  this.framebuf = new Buffer(1024);
  this.frameoff = 0;
  this.playerconn = wsReq.accept(null, wsReq.origin);
  /* Array for watcher connections. */
  this.watchers = [];
  /* END setup */
  function ttyrec_chunk(datastr) {
    ss.lasttime = new Date();
    var buf = new Buffer(datastr);
    var chunk = new Buffer(buf.length + 12);
    /* TTYREC headers */
    chunk.writeUInt32LE(Math.floor(ss.lasttime.getTime() / 1000), 0);
    chunk.writeUInt32LE(1000 * (ss.lasttime.getTime() % 1000), 4);
    chunk.writeUInt32LE(buf.length, 8);
    buf.copy(chunk, 12);
    ss.record.write(chunk);
    ss.framepush(buf);
    /* Send to the player. */
    var msg = JSON.stringify({"t": "d", "d": buf.toString("hex")});
    ss.playerconn.sendUTF(msg);
    /* Send to any watchers. */
    for (var i = 0; i < ss.watchers.length; i++) {
      if (ss.watchers[i].connected)
        ss.watchers[i].sendUTF(msg);
    }
    ss.emit('data', buf);
  }
  this.term.on("data", ttyrec_chunk);
  this.framepush = function(chunk) {
    /* If this chunk resets the screen, discard what preceded it. */
    if (bufncmp(chunk, this.game.clear, this.game.clear.length)) {
      this.framebuf = new Buffer(1024);
      this.frameoff = 0;
    }
    /* Make sure there's space. */
    while (this.framebuf.length < chunk.length + this.frameoff) {
      var nbuf = new Buffer(this.framebuf.length * 2);
      this.framebuf.copy(nbuf, 0, 0, this.frameoff);
      this.framebuf = nbuf;
      if (this.framebuf.length > 65536) {
        tslog("Warning: Game %s frame buffer at %d bytes", this.tag(), 
                this.framebuf.length);
      }
    }
    chunk.copy(this.framebuf, this.frameoff);
    this.frameoff += chunk.length;
  };
  this.write = function(data) {
    this.term.write(data);
  };
  this.tag = function() {
    return this.game.uname + "/" + this.pname;
  };
  // Teardown.
  this.term.on("exit", function () {
    var tag = ss.tag();
    fs.unlink(ss.lock);
    ss.record.end();
    var watchsocks = ss.watchers;
    ss.watchers = [];
    for (var i = 0; i < watchsocks.length; i++) {
      if (watchsocks[i].connected)
        watchsocks[i].close();
    }
    if (ss.playerconn.connected) {
      ss.playerconn.sendUTF(JSON.stringify({"t": "q"}));
      ss.playerconn.close();
    }
    ss.emit('exit');
    gamemux.emit('end', ss.game.uname, ss.pname);
    delete sessions[tag];
    tslog("Game %s ended.", tag);
  });
  this.close = function () {
    if (ss.tag() in sessions)
      ss.term.kill('SIGHUP');
  };
  /* Send initial data. */
  this.playerconn.sendUTF(JSON.stringify({"t": "s", "w": this.w, "h": this.h, 
               "p": this.pname, "g": this.game.uname}));
  /* Attach handlers. */
  function messageH(message) {
    var parsedMsg = getMsgWS(message);
    if (parsedMsg.t == 'q') {
      ss.close();
    }
    else if (parsedMsg.t == 'd') {
      var hexstr = parsedMsg.d.replace(/[^0-9a-f]/gi, "");
      if (hexstr.length % 2 != 0) {
        hexstr = hexstr.slice(0, -1);
      }
      var keybuf = new Buffer(hexstr, "hex");
      ss.write(keybuf);
    }
  }
  this.playerconn.on('message', messageH);
  this.playerconn.on('close', this.close);
  /* To attach a watcher. */
  this.attach = function (wsReq) {
    var conn = wsReq.accept(null, wsReq.origin);
    conn.sendUTF(JSON.stringify({
            "t": "w", "w": this.w, "h": this.h, "p": this.pname, 
            "g": this.game.uname
    }));
    conn.sendUTF(JSON.stringify({"t": "d",
        "d": this.framebuf.toString("hex", 0, this.frameoff)}));
    conn.on('close', function () {
      /* 'this' is the connection when triggered */
      var n = ss.watchers.indexOf(this);
      if (n >= 0) {
        ss.watchers.splice(n, 1);
        tslog("A WebSocket watcher has left game %s", ss.tag());
      }
    });
    this.watchers.push(conn);
  };
}
TermSession.prototype = new events.EventEmitter();

function DglSession(filename) {
  var ss = this;
  events.EventEmitter.call(this);
  var pathcoms = filename.split('/');
  this.gname = pathcoms[pathcoms.length - 2];
  if (!(this.gname in games)) {
    ss.emit('open', false);
    return;
  }
  var basename = pathcoms[pathcoms.length - 1];
  var firstsep = basename.indexOf(':');
  this.pname = basename.slice(0, firstsep);
  var fname = basename.slice(firstsep + 1);
  this.ttyrec = path.join("/dgldir/ttyrec", this.pname, this.gname, fname);
  /* Flag to prevent multiple handlers from reading simultaneously and
   * getting into a race. */
  this.reading = false;
  this.rpos = 0;
  this.framebuf = new Buffer(1024);
  this.frameoff = 0;
  this.framepush = function(chunk) {
    /* If this chunk resets the screen, discard what preceded it. */
    var cgame = games[this.gname];
    if (bufncmp(chunk, cgame.clear, cgame.clear.length)) {
      tslog("DGL %s: clearing frame", ss.tag());
      this.framebuf = new Buffer(1024);
      this.frameoff = 0;
    }
    /* Make sure there's space. */
    while (this.framebuf.length < chunk.length + this.frameoff) {
      var nbuf = new Buffer(this.framebuf.length * 2);
      this.framebuf.copy(nbuf, 0, 0, this.frameoff);
      this.framebuf = nbuf;
      if (this.framebuf.length > 65536) {
        tslog("Warning: DGL %s frame buffer at %d bytes", this.tag(), 
                this.framebuf.length);
      }
    }
    chunk.copy(this.framebuf, this.frameoff);
    this.frameoff += chunk.length;
  };
  this.readchunk = function () {
    if (this.reading)
      return;
    this.reading = true;
    var header = new Buffer(12);
    fs.read(ss.fd, header, 0, 12, ss.rpos, function (err, n, buf) {
      /* Stop recursion if end of file has been reached. */
      if (err || n < 12) {
        if (!err && n > 0) {
          tslog("DGL %s: expected 12-byte header, got %d", ss.tag(), n);
        }
        ss.reading = false;
        return;
      }
      ss.rpos += 12;
      var datalen = buf.readUInt32LE(8);
      //tslog("Allocating %d bytes", datalen);
      if (datalen > 16384) {
        tslog("DGL %s: looking for %d bytes", ss.tag(), datalen);
      }
      var databuf = new Buffer(datalen);
      fs.read(ss.fd, databuf, 0, datalen, ss.rpos, function (err, n, buf) {
        if (err || n < datalen) {
          /* Next time, read the header again. */
          ss.rpos -= 12;
          ss.reading = false;
          tslog("DGL %s: expected %d bytes, got %d", ss.tag(), datalen, n);
          return;
        }
        ss.rpos += n;
        ss.reading = false;
        /* Process the data */
        ss.framepush(buf);
        ss.emit("data", buf);
        //tslog("DGL %s: %d bytes", ss.tag(), buf.length);
        /* Recurse. */
        ss.readchunk();
      });
    });
  };
  fs.readFile(filename, {encoding: "utf8"}, function (err, data) {
    if (err) {
      ss.emit('open', false);
      return;
    }
    var lines = data.split('\n');
    ss.h = Number(lines[1]);
    ss.w = Number(lines[2]);
    fs.open(ss.ttyrec, "r", function(err, fd) {
      if (err) {
        ss.emit('open', false);
      }
      else {
        ss.fd = fd;
        ss.emit('open', true);
        tslog("DGL %s: open", ss.tag());
        ss.readchunk();
        ss.watcher = fs.watch(ss.ttyrec, function (ev, finame) {
          if (ev == "change")
            ss.readchunk();
        });
      }
    });
  });
  this.tag = function () {
    return this.gname + "/" + this.pname;
  };
  this.close = function () {
    this.watcher.close()
    /* Ensure all data is handled before quitting. */
    this.readchunk();
    fs.close(this.fd);
    this.emit("close");
    tslog("DGL %s: closed", ss.tag());
  };
}
DglSession.prototype = new events.EventEmitter();

function wsStartGame(wsReq) {
  var playmatch = wsReq.resourceURL.pathname.match(/^\/play\/([^\/]*)$/);
  if (!playmatch[1] || !(playmatch[1] in games)) {
    wsReq.reject(404, errorcodes[2]);
    return;
  }
  var gname = playmatch[1];
  if (!allowlogin) {
    wsReq.reject(404, errorcodes[6]);
    return;
  }
  if (!("key" in wsReq.resourceURL.query)) {
    wsReq.reject(404, "No key given.");
    return;
  }
  var lkey = wsReq.resourceURL.query["key"];
  if (!(lkey in logins)) {
    wsReq.reject(404, errorcodes[1]);
    return;
  }
  var pname = logins[lkey].name;
  function progcallback(err, fname) {
    if (fname) {
      wsReq.reject(404, errorcodes[4]);
      tslog("%s is already playing %s", pname, gname);
    }
    else {
      new TermSession(gname, pname, wsReq);
    }
  };
  checkprogress(pname, games[gname], progcallback, []);
}

/* Some functions which check whether a player is currently playing or 
 * has a saved game.  Maybe someday they will provide information on 
 * the game. */
function checkprogress(user, game, callback, args) {
  var progressdir = path.join("/dgldir/inprogress", game.uname);
  fs.readdir(progressdir, function(err, files) {
    if (err) {
      args.unshift(err, null);
      callback.apply(null, args);
      return;
    }
    var fre = RegExp("^" + user + ":");
    for (var i = 0; i < files.length; i++) {
      if (files[i].match(fre)) {
        args.unshift(null, files[i]);
        callback.apply(null, args);
        return;
      }
    }
    args.unshift(null, false);
    callback.apply(null, args);
  });
}

function checksaved(user, game, callback, args) {
  var savedirc = game.uname + "save";
  var basename = String(pwent.uid) + "-" + user + game.suffix;
  var savefile = path.join("/var/games/roguelike", savedirc, basename);
  fs.exists(savefile, function (exist) {
    args.unshift(exist);
    callback.apply(null, args);
  });
}

function playerstatus(user, callback) {
  var sdata = {};
  function finishp() {
    for (var gname in games) {
      if (!(gname in sdata))
        return;
    }
    callback(sdata);
  }
  function regsaved(exists, game) {
    if (exists)
      sdata[game.uname] = "s";
    else
      sdata[game.uname] = "0";
    finishp();
  }
  function regactive(err, filename, game) {
    if (!err && filename) {
      if (filename.match(/^[^:]*:node:/))
        sdata[game.uname] = "p";
      else
        sdata[game.uname] = "d";
      finishp();
    }
    else
      checksaved(user, game, regsaved, [game]);
  }
  for (var gname in games) {
    checkprogress(user, games[gname], regactive, [games[gname]]);
  }
}

/* A few utility functions */
function timestamp(dd) {
  if (!(dd instanceof Date)) {
    dd = new Date();
  }
  sd = dd.toISOString();
  sd = sd.slice(0, sd.indexOf("."));
  return sd.replace("T", ".");
}

function randkey(words) {
  if (!words || !(words > 0))
    words = 1;
  function rand32() {
    rnum = Math.floor(Math.random() * 65536 * 65536);
    hexstr = rnum.toString(16);
    while (hexstr.length < 8)
      hexstr = "0" + hexstr;
    return hexstr;
  }
  var key = "";
  for (var i = 0; i < words; i++)
    key += rand32();
  return key;
}

/* Compares two buffers, returns true for equality up to index n */
function bufncmp(buf1, buf2, n) {
  if (!Buffer.isBuffer(buf1) || !Buffer.isBuffer(buf2))
    return false;
  for (var i = 0; i < n; i++) {
    if (i == buf1.length && i == buf2.length)
      return true;
    if (i == buf1.length || i == buf2.length)
      return false;
    if (buf1[i] != buf2[i])
      return false;
  }
  return true;
}

function tslog() {
  arguments[0] = new Date().toISOString() + ": " + String(arguments[0]);
  console.log.apply(console, arguments);
}

/* 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 getMsg(posttext) {
  var jsonobj;
  if (!posttext)
    return {};
  try {
    jsonobj = JSON.parse(posttext);
  }
  catch (e) {
    if (e instanceof SyntaxError)
      return {};
  }
  if (typeof(jsonobj) != "object")
    return {};
  return jsonobj;
}

function getMsgWS(msgObj) {
  if (msgObj.type != "utf8")
    return {};
  return getMsg(msgObj.utf8Data);
}

function login(req, res, formdata) {
  if (!allowlogin) {
    sendError(res, 6, null, false);
    return;
  }
  if (!("name" in formdata)) {
    sendError(res, 2, "Username not given.", false);
    return;
  }
  else if (!("pw" in formdata)) {
    sendError(res, 2, "Password not given.", false);
    return;
  }
  var username = String(formdata["name"]);
  var password = String(formdata["pw"]);
  function checkit(code, signal) {
    /* Checks the exit status, see sqlickrypt.c for details. */
    if (code != 0) {
      sendError(res, 3);
      if (code == 1)
        tslog("Password check failed for user %s", username);
      else if (code == 2)
        tslog("Attempted login by nonexistent user %s", username);
      else
        tslog("Login failed: sqlickrypt error %d", code);
      return;
    }
    var lkey = randkey(2);
    while (lkey in logins)
      lkey = randkey(2);
    logins[lkey] = {"name": username, "ts": new Date()};
    res.writeHead(200, {'Content-Type': 'application/json'});
    var reply = {"t": "l", "k": lkey, "u": username};
    res.write(JSON.stringify(reply));
    res.end();
    tslog("%s has logged in (key %s)", username, lkey);
    return;
  }
  /* Launch the sqlickrypt utility to check the password. */
  var pwchecker = child_process.spawn("/bin/sqlickrypt", ["check"]);
  pwchecker.on("exit", checkit);
  pwchecker.stdin.end(username + '\n' + password + '\n', "utf8");
  return;
}

/* Sets things up for a new user, like dgamelaunch's commands[register] */
function regsetup(username) {
  function regsetup_l2(err) {
    for (var g in games) {
      fs.mkdir(path.join("/dgldir/ttyrec", username, games[g].uname), 0755);
    }
  }
  fs.mkdir(path.join("/dgldir/userdata", username), 0755);
  fs.mkdir(path.join("/dgldir/ttyrec/", username), 0755, regsetup_l2);
}

function register(req, res, formdata) {
  var uname, passwd, email;
  if (typeof (formdata.name) != "string" || formdata.name === "") {
    sendError(res, 2, "No name given.");
    return;
  }
  else
    uname = formdata["name"];
  if (typeof (formdata.pw) != "string" || formdata.pw === "") {
    sendError(res, 2, "No password given.");
    return;
  }
  else
    passwd = formdata["pw"];
  if (typeof (formdata.email) != "string" || formdata.email === "") {
    /* E-mail is optional */
    email = "nobody@nowhere.not";
  }
  else
    email = formdata["email"];
  function checkreg(code, signal) {
    if (code === 0) {
      var lkey = randkey(2);
      while (lkey in logins)
        lkey = randkey(2);
      logins[lkey] = {"name": uname, "ts": new Date()};
      var reply = {"t": "r", "k": lkey, "u": uname};
      res.writeHead(200, {'Content-Type': 'application/json'});
      res.write(JSON.stringify(reply));
      res.end();
      tslog("Added new user: %s", uname);
      regsetup(uname);
    }
    else if (code == 4) {
      sendError(res, 2, "Invalid characters in name or email.");
      tslog("Attempted registration: %s %s", uname, email);
    }
    else if (code == 1) {
      sendError(res, 2, "Username " + uname + " is already being used.");
      tslog("Attempted duplicate registration: %s", uname);
    }
    else {
      sendError(res, 0, null);
      tslog("sqlickrypt register failed with code %d", code);
    }
  }
  var child_adder = child_process.spawn("/bin/sqlickrypt", ["register"]);
  child_adder.on("exit", checkreg);
  child_adder.stdin.end(uname + '\n' + passwd + '\n' + email + '\n', "utf8");
  return;
}

/* Stops a running game if the request has the proper key. */
function stopgame(res, formdata) {
  if (!("key" in formdata) || !(formdata["key"] in logins)) {
    sendError(res, 1);
    return;
  }
  var pname = logins[formdata["key"]].name;
  if (!("g" in formdata) || !(formdata["g"] in games)) {
    sendError(res, 2, "No such game.");
    return;
  }
  var gname = formdata["g"];
  function checkback(err, fname) {
    if (!fname) {
      sendError(res, 7);
      return;
    }
    var fullfile = path.join("/dgldir/inprogress", gname, fname);
    fs.readFile(fullfile, "utf8", function(err, fdata) {
      if (err) {
        sendError(res, 7);
        return;
      }
      var pid = parseInt(fdata.split('\n')[0], 10);
      try {
        process.kill(pid, 'SIGHUP');
      }
      catch (err) {
        /* If the PID is invalid, the lockfile is stale. */
        if (err.code == "ESRCH") {
          var nodere = RegExp("^" + pname + ":node:");
          if (fname.match(nodere)) {
            fs.unlink(fullfile);
          }
        }
      }
      /* The response doesn't mean that the game is gone.  The only way
       * to make sure a dgamelaunch-supervised game is over would be to
       * poll fname until it disappears. */
      res.writeHead(200, {'Content-Type': 'application/json'});
      res.write(JSON.stringify({"t": "q"}));
      res.end();
    });
  }
  checkprogress(pname, games[gname], checkback, []);
}

function startProgressWatcher() {
  var watchdirs = [];
  for (var gname in games) {
    watchdirs.push(path.join("/dgldir/inprogress", gname));
  }
  var subproc = child_process.spawn("/bin/dglwatcher", watchdirs);
  subproc.stdout.setEncoding('utf8');
  subproc.stdout.on('data', function (chunk) {
    var fname = chunk.slice(2, -1);
    var filere = /.*\/([^\/]*)\/([^\/:]*):(node:)?(.*)/;
    var matchresult = fname.match(filere);
    if (!matchresult || matchresult[3])
      return;
    var gname = matchresult[1];
    var pname = matchresult[2];
    var tag = gname + "/" + pname;
    if (chunk[0] == "E") {
      tslog("DGL: %s is playing %s: %s", pname, gname, fname)
      dglgames[tag] = new DglSession(fname);
    }
    else if (chunk[0] == "C") {
      tslog("DGL: %s started playing %s: %s", pname, gname, fname)
      dglgames[tag] = new DglSession(fname);
    }
    else if (chunk[0] == "D") {
      tslog("DGL: %s finished playing %s: %s", pname, gname, fname)
      dglgames[tag].close();
      delete dglgames[tag];
    }
    else {
      tslog("Watcher says: %s", chunk)
    }
  });
  subproc.stdout.resume();
  return subproc;
}

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);
  fs.exists(realname, function (exists) {
    var resheaders = {};
    if (!exists || !extension || extension == ".html")
      resheaders["Content-Type"] = "text/html; charset=utf-8";
    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) {
      fs.readFile(realname, function (error, data) {
        if (error) {
          res.writeHead(500, {});
          res.end();
        }
        else {
          res.writeHead(200, resheaders);
          if (req.method != 'HEAD')
            res.write(data);
          res.end();
        }
      });
    }
    else {
      send404(res, nname, req.method == 'HEAD');
    }
  });
  return;
}

/* Currently, this doesn't do anything blocking, but keep the callback */
function getStatus(callback) {
  var now = new Date();
  var statusinfo = {"s": allowlogin, "g": []};
  for (var tag in sessions) {
    var gamedesc = {};
    gamedesc["p"] = sessions[tag].pname;
    gamedesc["g"] = sessions[tag].game.uname;
    gamedesc["i"] = now - sessions[tag].lasttime;
    statusinfo["g"].push(gamedesc);
  }
  statusinfo["dgl"] = [];
  for (var tag in dglgames) {
    var dglinfo = {};
    var slash = tag.search('/');
    dglinfo["g"] = tag.slice(0, slash);
    dglinfo["p"] = tag.slice(slash + 1);
    dglinfo["i"] = -1;
    statusinfo["dgl"].push(dglinfo);
  }
  callback(statusinfo);
}

function statusmsg(req, res) {
  function respond(info) {
    res.writeHead(200, { "Content-Type": "application/json" });
    if (req.method != 'HEAD')
      res.write(JSON.stringify(info));
    res.end();
  }
  getStatus(respond);
}

function pstatusmsg(req, res) {
  if (req.method == 'HEAD') {
    res.writeHead(200, { "Content-Type": "application/json" });
    res.end();
    return;
  }
  var target = url.parse(req.url).pathname;
  var pmatch = target.match(/^\/pstatus\/(.*)/);
  if (pmatch && pmatch[1])
    var pname = pmatch[1];
  else {
    sendError(res, 2, "No name given.");
    return;
  }
  var reply = {"name": pname};
  playerstatus(pname, function (pdata) {
    reply["stat"] = pdata;
    res.writeHead(200, { "Content-Type": "application/json" });
    res.write(JSON.stringify(reply));
    res.end();
  });
}

function getuinfo(req, res) {
  var urlobj = url.parse(req.url, true);
  var query = urlobj.query;
  if (!("key" in query) || !(query["key"] in logins)) {
    sendError(res, 1);
    return;
  }
  var match = urlobj.pathname.match(/^\/[^\/]*\/(.*)/);
  if (!match || !match[1]) {
    send404(res, urlobj.pathname, req.method == 'HEAD');
    return;
  }
  var which = match[1];
  var name = logins[query["key"]].name;
  var reply = { "u": name };
  function send() {
    res.writeHead(200, { "Content-Type": "application/json" });
    res.write(JSON.stringify(reply));
    res.end();
  }
  if (which == "pw") {
    /* Don't actually divulge passwords. */
    reply["pw"] = "";
    send();
  }
  else if (which == "email") {
    var address;
    function finish(code, signal) {
      if (code != 0) {
        tslog("sqlickrypt: %d with name %s", code, name);
        sendError(res, 2);
      }
      else {
        reply["email"] = address;
        send();
      }
    }
    var subproc = child_process.spawn("/bin/sqlickrypt", ["getmail"]);
    subproc.stdout.on("data", function (data) {
      address = data.toString().replace(/\n/g, "");
    });
    subproc.on("exit", finish);
    subproc.stdin.end(name + '\n', "utf8");
  }
  else {
    send404(res, urlobj.pathname, req.method == 'HEAD');
    return;
  }
}

function setuinfo(req, res, postdata) {
  var urlobj = url.parse(req.url, true);
  var query = urlobj.query;
  if (!("key" in query) || !(query["key"] in logins)) {
    sendError(res, 1);
    return;
  }
  var name = logins[query["key"]].name;
  var match = urlobj.pathname.match(/^\/[^\/]*\/(.*)/);
  if (!match || !match[1]) {
    send404(res, urlobj.pathname, true);
    return;
  }
  var which = match[1];
  if (!("v" in postdata)) {
    sendError(res, 2, "No value provided");
    return;
  }
  if (which == "email" || which == "pw") {
    var args;
    if (which == "email")
      args = ["setmail"];
    else
      args = ["setpw"];
    var child = child_process.execFile("/bin/sqlickrypt", args, 
                  function (err, stdout, stderr) {
      if (err) {
        tslog("Could not set %s: sqlickrypt error %d", which, err.code);
        sendError(res, 2);
      }
      else {
        tslog("User %s has changed %s", name, which);
        res.writeHead(200, { "Content-Type": "application/json" });
        res.end(JSON.stringify({"t": "t"}));
      }
    });
    child.stdin.end(name + "\n" + postdata.v + "\n", "utf8");
  }
  else {
    send404(res, urlobj.pathname, true);
  }
}

var errorcodes = [ "Generic Error", "Not logged in", "Invalid data", 
        "Login failed", "Already playing", "Game launch failed",
        "Server shutting down", "Game not in progress" ];

function sendError(res, ecode, msg, box) {
  res.writeHead(200, { "Content-Type": "application/json" });
  var edict = {"t": "E"};
  if (!(ecode < errorcodes.length && ecode > 0))
    ecode = 0;
  edict["c"] = ecode;
  edict["s"] = errorcodes[ecode];
  if (msg)
    edict["s"] += ": " + msg;
  if (box)
    res.write(JSON.stringify([edict]));
  else
    res.write(JSON.stringify(edict));
  res.end();
}

function send404(res, path, nopage) {
  res.writeHead(404, {"Content-Type": "text/html; charset=utf-8"});
  if (!nopage) {
    res.write("<html><head><title>" + path + "</title></head>\n<body><h1>"
              + path + " Not Found</h1></body></html>\n");
  }
  res.end();
}

function webHandler(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() {
    formdata = getMsg(reqbody);
    var target = url.parse(req.url).pathname;
    /* First figure out if the client is POSTing to a command interface. */
    if (req.method == 'POST') {
      if (target == "/login") {
        login(req, res, formdata);
      }
      else if (target == "/addacct") {
        register(req, res, formdata);
      }
      else if (target == "/quit") {
        stopgame(res, formdata);
      }
      else if (target.match(/^\/uinfo\//)) {
        setuinfo(req, res, formdata);
      }
      else {
        res.writeHead(405, resheaders);
        res.end();
      }
    }
    else if (req.method == 'GET' || req.method == 'HEAD') {
      if (target == '/status') {
        statusmsg(req, res);
      }
      else if (target.match(/^\/uinfo\//)) {
        getuinfo(req, res);
      }
      else if (target.match(/^\/pstatus\//)) {
        pstatusmsg(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);
}

function wsHandler(wsRequest) {
  var watchmatch = wsRequest.resource.match(/^\/watch\/(.*)$/);
  var playmatch = wsRequest.resource.match(/^\/play\//);
  if (watchmatch !== null) {
    if (!(watchmatch[1] in sessions)) {
      wsRequest.reject(404, errorcodes[7]);
      return;
    }
    var tsession = sessions[watchmatch[1]];
    tsession.attach(wsRequest);
    tslog("Game %s is being watched via WebSockets", tsession.tag());
  }
  else if (playmatch !== null) {
    wsStartGame(wsRequest);
  }
  else if (wsRequest.resourceURL.pathname == "/status") {
    var conn = wsRequest.accept(null, wsRequest.origin);
    var tell = function () {
      getStatus(function (info) {
        info["t"] = "t";
        conn.sendUTF(JSON.stringify(info));
      });
    }
    var beginH = function (gname, pname) {
      conn.sendUTF(JSON.stringify({"t": "b", "p": pname, "g": gname}));
    };
    var listH = function (list) {
      conn.sendUTF(JSON.stringify(list));
    };
    var endH = function (gname, pname) {
      conn.sendUTF(JSON.stringify({"t": "e", "p": pname, "g": gname}));
    };
    gamemux.on('begin', beginH);
    gamemux.on('list', listH);
    gamemux.on('end', endH);
    conn.on('message', tell);
    conn.on('close', function () {
      gamemux.removeListener('begin', beginH);
      gamemux.removeListener('list', listH);
      gamemux.removeListener('end', endH);
    });
    tell();
  }
  else
    wsRequest.reject(404, "No such resource.");
}

/* Only games with low idle time are included.  Use getStatus() for the
 * complete list. */
function pushStatus() {
  var now = new Date();
  var statusinfo = {"t": "p", "s": allowlogin, "g": [], "dgl": []};
  for (var tag in sessions) {
    var delta = now - sessions[tag].lasttime;
    if (delta < 60000) {
      var gamedesc = {};
      gamedesc["p"] = sessions[tag].pname;
      gamedesc["g"] = sessions[tag].game.uname;
      gamedesc["i"] = delta;
      statusinfo["g"].push(gamedesc);
    }
  }
  for (var tag in dglgames) {
    var dglinfo = {};
    var slash = tag.search('/');
    dglinfo["g"] = tag.slice(0, slash);
    dglinfo["p"] = tag.slice(slash + 1);
    dglinfo["i"] = -1;
    statusinfo["dgl"].push(dglinfo);
  }
  gamemux.emit('list', statusinfo);
}

function shutdown () {
  httpServer.close();
  httpServer.removeAllListeners('request');
  ctlServer.close();
  tslog("Shutting down...");
  process.exit();
}

function consoleHandler(chunk) {
  var msg = chunk.toString().split('\n')[0];
  if (msg == "quit") {
    allowlogin = false;
    tslog("Disconnecting...");
    for (var tag in sessions) {
      sessions[tag].close();
    }
    progressWatcher.stdin.end("\n");
    setTimeout(shutdown, 2000);
  }
}

process.on("exit", function () {
  for (var tag in sessions) {
    sessions[tag].term.kill('SIGHUP');
  }
  tslog("Quitting...");
  return;
});

/* Initialization STARTS HERE */
process.env["TERM"] = "xterm-256color";

if (process.getuid() != 0) {
  tslog("Not running as root, cannot chroot.");
  process.exit(1);
}

var httpServer; // declare here so shutdown() can find it
var wsServer;
var progressWatcher;

var pwent; 
try {
  pwent = posix.getpwnam(dropToUser);
}
catch (err) {
  tslog("Could not drop to user %s: user does not exist", dropToUser);
  process.exit(1);
}

/* This could be nonblocking, but nothing else can start yet anyway. */
if (fs.existsSync(ctlsocket)) {
  fs.unlinkSync(ctlsocket);
}

/* Open the control socket before chrooting where it can't be found */
var ctlServer = net.createServer(function (sock) {
  sock.on('data', consoleHandler);
});
ctlServer.listen(ctlsocket, function () {
  /* rlgwebd.js now assumes that it has been started by the rlgwebd shell
   * script, or some other method that detaches it and sets up stdio. */
  /* chroot and drop permissions.  posix.chroot() does chdir() itself. */
  try {
    posix.chroot(chrootDir);
  }
  catch (err) {
    tslog("chroot to %s failed: %s", chrootDir, err);
    process.exit(1);
  }
  try {
    // drop gid first, that requires UID=0
    process.setgid(pwent.gid);
    process.setuid(pwent.uid);
  }
  catch (err) {
    tslog("Could not drop permissions: %s", err);
    process.exit(1);
  }
  httpServer = http.createServer(webHandler);
  httpServer.listen(httpPort);
  tslog('rlgwebd running on port %d', httpPort); 
  wsServer = new WebSocketServer({"httpServer": httpServer});
  wsServer.on("request", wsHandler);
  tslog('WebSockets are online'); 
  progressWatcher = startProgressWatcher();
  setInterval(pushStatus, 40000);
});