From 8573431f9ff3e84aaf6ff8599f733b9baa5503c5 Mon Sep 17 00:00:00 2001 From: cubedro Date: Mon, 17 Aug 2015 16:39:36 +0300 Subject: [PATCH] added multiple secrets --- app.js | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/app.js b/app.js index 6c26222..e1e57f6 100644 --- a/app.js +++ b/app.js @@ -3,7 +3,17 @@ var logger = require('./lib/utils/logger'); var chalk = require('chalk'); var http = require('http'); -var WS_SECRET = process.env.WS_SECRET || "eth-net-stats-has-a-secret"; +var WS_SECRET; + +if( !_.isUndefined(process.env.WS_SECRET) && !_.isNull(process.env.WS_SECRET) ) +{ + WS_SECRET = ["eth-net-stats-has-a-secret"]; +} +else +{ + WS_SECRET = [process.env.WS_SECRET]; +} + var banned = require('./lib/utils/config').banned; // Init http server @@ -81,7 +91,7 @@ api.on('connection', function (spark) { console.info('API', 'CON', 'Hello', data['id']); - if( _.isUndefined(data.secret) || data.secret !== WS_SECRET || banned.indexOf(spark.address.ip) >= 0 ) + if( _.isUndefined(data.secret) || WS_SECRET.indexOf(data.secret) === -1 || banned.indexOf(spark.address.ip) >= 0 ) { spark.end(undefined, { reconnect: false }); console.error('API', 'CON', 'Closed - wrong auth', data);