novnc.js

// noVNC port of Node.js
// Copyright (c) 2013 Tom Zhou<appnet.link@gmail.com>

var Connect = require('connect'),
    Net = require('net'),
    Buffer = require('buffer').Buffer,
    Fs = require('fs'),
    WSS = require('wspp').createWebSocketStream;

// authentication module
var httpauth = require('http-auth');

// file upload module
var multipart = require('connect-multiparty');


// Debug level
var Debug = 0;

// web server
var webServer = module.exports.webServer = function(options) {
    var app = Connect();
    var auth = (options && options.auth) || false;
    var upload = (options && options.upload) || __dirname + '/upload/';

    // rewrite req.url to remove vToken string
    var vtokenregex = /\/vtoken\/([0-9]|[a-f]){16}/gi;

    app.use(function(req, res, next){
        if (vtokenregex.test(req.url)) {
            res.writeHead(301, {'location': req.url.replace(vtokenregex, '')});
            res.end();
                ///req.url = req.url.replace(vtokenregex, '');
        } else next();
    });

    // basic-auth middleware
    if (auth) {
        var basic = httpauth.basic({
            realm: "51dese.com"
        }, function (username, password, callback) {
            callback(username === auth.username && password === auth.password);
        });

        app.use(httpauth.connect(basic));
    }

    // file upload middleware
    if (upload) {
        var upapp = Connect();
        var multipartMiddleware = multipart({uploadDir: upload});

        upapp.use(multipartMiddleware);
        upapp.use(function(req, res) {
            ///console.log(req.body, req.files);

            if (typeof req.files.file === 'object') {
                // rename file as timestamp_originalfilename
                var newname = '' + Date.now() + '__' + req.files.file.originalFilename;
                var newpath = upload + '/' + newname;
                Fs.rename(req.files.file.path, newpath, function(err) {
                    if (err) {
                        res.writeHeader(501);
                        res.end('upload rename failed');
                    } else {
                        res.writeHeader(200, {'content-type': 'application/json'});
                        req.files.file.newname = newname;
                        req.files.file.newpath = newpath;
                        res.end(JSON.stringify(req.files));
                    }
                });
            } else {
                res.writeHeader(200, {'content-type': 'application/json'});
                res.end(JSON.stringify(req.files));
            }
        });

        app.use('/upload', upapp);
    }

    ///app(Connect.staticCache({maxLength: 256*1024, maxObjects: 8}))
    app.use(Connect.static(__dirname+'/front'));

    app.use(function(req, res){
        res.writeHeader(200, {'content-type': 'text/html'});
        res.end(Fs.readFileSync(__dirname+'/front/novnc.html'));
    });

    return app;
};

// ws2tcp proxy
// vnc: {host: ..., port: ...}, VNC server info
var tcpProxy = module.exports.tcpProxy = function(vnc){
    vnc = vnc || {};

    vnc.host = vnc.host || 'localhost';
    vnc.port = vnc.port || 5900;

    if (Debug) console.log('connect to vnc %j ...', vnc);

    return function(ws){
        // create tcp connection to VNC server
        var ts = Net.connect(vnc, function(){
            if (Debug) console.log('tcp connection...');

            // wrap stream on ws
            var wss = WSS(ws);

            // pipe each other
            wss.pipe(ts);
            ts.pipe(wss);

            // check error
            wss.on('error', function(){
                if (Debug) console.log('ws.onerror...');
                ts.end();
            });
        });

        // check error
        ws.on('error', function(){
            if (Debug) console.log('ws.onerror...');
            ts.end();
        });
        ts.on('error', function(err){
            if (Debug) console.log('tcp connection error '+err);
            ws.close();
        });
    };
};

// simple test
/*var http = require('http'),
    WebSocket = require('wspp'),
    WebSocketServer = WebSocket.Server;

var srv = http.createServer(webServer());
srv.listen(5600);
console.log('noVNC proxy server listening on 5600');

var wss = new WebSocketServer({server: srv, path: '/peervnc'});
wss.on('connection', tcpProxy({host: '192.188.1.101', port: 5900}));
*/