mirror of
https://github.com/NodeBB/NodeBB.git
synced 2025-11-18 03:31:03 +01:00
closes #2432
This commit is contained in:
35
src/middleware/ratelimit.js
Normal file
35
src/middleware/ratelimit.js
Normal file
@@ -0,0 +1,35 @@
|
||||
|
||||
|
||||
'use strict';
|
||||
var winston = require('winston');
|
||||
|
||||
var ratelimit = {};
|
||||
|
||||
var allowedCallsPerSecond = 10;
|
||||
|
||||
|
||||
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 > allowedCallsPerSecond && socket.elapsedTime < 1000) {
|
||||
winston.warn('Flooding detected! Calls : ' + socket.callsPerSecond + ', Duration : ' + socket.elapsedTime);
|
||||
return true;
|
||||
}
|
||||
|
||||
if (socket.elapsedTime >= 1000) {
|
||||
socket.elapsedTime = 0;
|
||||
socket.callsPerSecond = 0;
|
||||
}
|
||||
|
||||
socket.lastCallTime = now;
|
||||
return false;
|
||||
};
|
||||
|
||||
module.exports = ratelimit;
|
||||
Reference in New Issue
Block a user