Fabio/webssh2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: light refactoring

Browse source
This commit is contained in:
Bill Church 2022-05-20 15:12:01 -04:00
parent 0dcaa6e150
commit e75392b879
2 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
app/server/app.js
View file

@ -20,7 +20,7 @@ const session = require('express-session')(config.express);
const appSocket = require('./socket'); const appSocket = require('./socket');
const { setDefaultCredentials, basicAuth } = require('./util'); const { setDefaultCredentials, basicAuth } = require('./util');
const { webssh2debug, auditLog, logError } = require('./logging'); const { webssh2debug } = require('./logging');
const { reauth, connect, notfound, handleErrors } = require('./routes'); const { reauth, connect, notfound, handleErrors } = require('./routes');
setDefaultCredentials(config); setDefaultCredentials(config);

14
app/server/routes.js
View file

@ -11,10 +11,12 @@ const config = require('./config');
exports.reauth = function reauth(req, res) { exports.reauth = function reauth(req, res) {
let { referer } = req.headers; let { referer } = req.headers;
console.log(`referer: ${referer}`); if (!validator.isURL(referer, { host_whitelist: ['localhost'] })) {
if (!validator.isURL(referer, { host_whitelist: ['localhost'] })) referer = '/'; console.error(
console.log(`referer: ${referer}`); `WebSSH2 (${req.sessionID}) ERROR: Referrer '${referer}' for '/reauth' invalid. Setting to '/' which will probably fail.`
);
referer = '/';
}
res res
.status(401) .status(401)
.send( .send(
@ -87,11 +89,11 @@ exports.connect = function connect(req, res) {
if (req.session.ssh.header.background) validator.escape(req.session.ssh.header.background); if (req.session.ssh.header.background) validator.escape(req.session.ssh.header.background);
}; };
exports.notfound = function notfound(req, res) { exports.notfound = function notfound(_req, res) {
res.status(404).send("Sorry, can't find that!"); res.status(404).send("Sorry, can't find that!");
}; };
exports.handleErrors = function handleErrors(err, req, res) { exports.handleErrors = function handleErrors(err, _req, res) {
console.error(err.stack); console.error(err.stack);
res.status(500).send('Something broke!'); res.status(500).send('Something broke!');
}; };