diff --git a/app.js b/app.js index acc047f..6a4a32f 100644 --- a/app.js +++ b/app.js @@ -39,6 +39,10 @@ var opts = require('optimist') alias: 'p', description: 'wetty listen port' }, + whitelist: { + demand: false, + description: 'whitelist of username/hosts, you can connect to' + } }).boolean('allow_discovery').argv; var runhttps = false; @@ -47,6 +51,7 @@ var globalsshhost = 'localhost'; var sshhost = globalsshhost; var sshauth = 'password'; var globalsshuser = ''; +var whitelist = ['^.*@localhost$']; if (opts.sshport) { sshport = opts.sshport; @@ -71,6 +76,19 @@ if (opts.sslkey && opts.sslcert) { opts.ssl.cert = fs.readFileSync(path.resolve(opts.sslcert)); } +if (opts.whitelist) { + whitelist = opts.whitelist.split(','); +} + +function checkWhitelist(target){ + for (var idx = 0; idx