You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
nodebb/src/middleware/ratelimit.js

33 lines
795 B
JavaScript

'use strict';
const winston = require('winston');
const ratelimit = module.exports;
const allowedCalls = 100;
const timeframe = 10000;
ratelimit.isFlooding = function (socket) {
socket.callsPerSecond = socket.callsPerSecond || 0;
socket.elapsedTime = socket.elapsedTime || 0;
socket.lastCallTime = socket.lastCallTime || Date.now();
socket.callsPerSecond += 1;
const 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;
};