grassrootseconomics/ethstats-server
5
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #261 from cubedro/master

Browse Source 
Update from master
This commit is contained in:
Marian OANCΞA 2015-08-17 17:08:05 +03:00
commit a896730843
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
app.js
View File

@ -3,7 +3,24 @@ 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
{
if( process.env.WS_SECRET.indexOf('|') > 0 )
{
WS_SECRET = process.env.WS_SECRET.split('|');
}
else
{
WS_SECRET = [process.env.WS_SECRET];
}
}
var banned = require('./lib/utils/config').banned;
// Init http server
@ -81,7 +98,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);