all files / src/middleware/ ratelimit.js

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
 
 
1×
 
1×
 
1×
1×
 
1×
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1×
 
 
 
'use strict';
var winston = require('winston');
 
var ratelimit = {};
 
var allowedCalls = 100;
var timeframe = 10000;
 
ratelimit.isFlooding = function(socket) {
	socket.callsPerSecond = socket.callsPerSecond || 0;
	socket.elapsedTime = socket.elapsedTime || 0;
	socket.lastCallTime = socket.lastCallTime || Date.now();
 
	++socket.callsPerSecond;
 
	var now = Date.now();
	socket.elapsedTime += now - socket.lastCallTime;
 
	if (socket.callsPerSecond > allowedCalls && socket.elapsedTime < timeframe) {
		winston.warn('Flooding detected! Calls : ' + socket.callsPerSecond + ', Duration : ' + socket.elapsedTime);
		return true;
	}
 
	if (socket.elapsedTime >= timeframe) {
		socket.elapsedTime = 0;
		socket.callsPerSecond = 0;
	}
 
	socket.lastCallTime = now;
	return false;
};
 
module.exports = ratelimit;