cleaner_screencap

Author: Mikej81

app.js

@@ -4,7 +4,7 @@
 var path = require('path')
 var config = require('read-config')(path.join(__dirname, 'config.json'))
 //  var logger = require('morgan')
-//  var validator = require('validator')
+var validator = require('validator')
 var myutil = require('./util')
 
 var session = require('express-session')({
@@ -33,6 +33,28 @@ app.get('/', function (req, res, next) {
 app.get('/rdp/host/:host?', function (req, res, next) {
   req.session.host = req.params.host
   res.sendFile(__dirname + '/client/html/client.html')
+  req.session.rdp = {
+    host: (validator.isIP(req.params.host + '') && req.params.host) ||
+    (validator.isFQDN(req.params.host) && req.params.host) ||
+    (/^(([a-z]|[A-Z]|[0-9]|[!^(){}\-_~])+)?\w$/.test(req.params.host) &&
+      req.params.host) || config.rdp.host,
+    port: (validator.isInt(req.query.port + '', {min: 1, max: 65535}) &&
+      req.query.port) || config.rdp.port,
+    header: {
+      name: req.query.header || config.header.text,
+      background: req.query.headerBackground || config.header.background
+    },
+    algorithms: config.algorithms,
+    allowreplay: validator.isBoolean(req.headers.allowreplay + '') || false,
+    log: {
+      screencapture: config.logging.logScreenOnMouseclicks || false
+    },
+    readyTimeout: (validator.isInt(req.query.readyTimeout + '', {min: 1, max: 300000}) &&
+      req.query.readyTimeout) || config.rdp.readyTimeout
+  }
+  req.session.rdp.header.name && validator.escape(req.session.rdp.header.name)
+  req.session.rdp.header.background &&
+  validator.escape(req.session.rdp.header.background)
 })
 
 // Express error handling

client/js/client.js

@@ -153,6 +153,8 @@
       }).on('rdp-bitmap', function (bitmap) {
         console.log('[WebRDP] bitmap update bpp : ' + bitmap.bitsPerPixel)
         self.render.update(bitmap)
+      }).on('title', function (data) {
+        document.title = data
       }).on('headerBackground', function (data) {
         document.getElementById('header').style.backgroundColor = data
       }).on('header', function (data) {

config.json

@@ -12,6 +12,10 @@
     "port": 3389,
     "readyTimeout": 20000
   },
+  "logging": {
+    "logScreenOnMouseclicks": true,
+    "logKeypress": false
+  },
   "useminified": false,
   "header": {
     "text": null,

socket/index.js

@@ -79,6 +79,11 @@ module.exports = function (socket) {
         // clean older connection
       rdpClient.close()
     }
+    console.log('WebRDP Login: user=' + socket.request.session.username + ' from=' + socket.handshake.address + ' host=' + socket.request.session.rdp.host + ' port=' + socket.request.session.rdp.port + ' sessionID=' + socket.request.sessionID + '/' + socket.id + ' allowreplay=' + socket.request.session.rdp.allowreplay)
+    socket.emit('title', 'rdp://' + socket.request.session.rdp.host)
+    socket.request.session.rdp.header.background && socket.emit('headerBackground', socket.request.session.rdp.header.background)
+    socket.request.session.rdp.header.name && socket.emit('header', socket.request.session.rdp.header.name)
+
     socket.emit('headerBackground', 'green')
     socket.emit('header', '//HEADER//')