Skip to content

Commit

Permalink
add rate limiting
Browse files Browse the repository at this point in the history
  • Loading branch information
djschilling committed Jan 12, 2023
1 parent d830560 commit f2409ba
Showing 1 changed file with 31 additions and 0 deletions.
31 changes: 31 additions & 0 deletions ctldap.js
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@ var helpers = require('ldap-filter/lib/helpers');

var config = ini.parse(fs.readFileSync(path.resolve(__dirname, 'ctldap.config'), 'utf-8'));

var rateLimitStore = {};

function logDebug(site, msg) {
if (config.debug) {
console.log("[DEBUG] "+site.sitename+" - "+msg);
Expand Down Expand Up @@ -244,6 +246,33 @@ function apiLogin(site) {
return site.loginPromise;
}

function checkRateLimit(site, windowInSeconds, maxRequests) {
sitename = site.sitename;
var now = new Date();
if (!rateLimitStore[windowInSeconds]) {
rateLimitStore[windowInSeconds] = {};
}
if (!rateLimitStore[windowInSeconds][sitename]) {
rateLimitStore[windowInSeconds][sitename] = {
windowStartTime: now,
requestCount: 0
};
}

var secondsBetweenStartOfWindowAndNow = (now.getTime() - rateLimitStore[windowInSeconds][sitename].windowStartTime.getTime()) / 1000;
if (secondsBetweenStartOfWindowAndNow > windowInSeconds) {
rateLimitStore[windowInSeconds][sitename] = {
windowStartTime: now,
requestCount: 0
};
}
if (rateLimitStore[windowInSeconds][sitename].requestCount > maxRequests) {
logError('Rate Limit reached');
throw new Error('Rate Limit reached');
}
rateLimitStore[windowInSeconds][sitename].requestCount++;
}

/**
* Retrieves data from the PHP API via a POST call.
* @param {object} site - The current site
Expand All @@ -253,6 +282,8 @@ function apiLogin(site) {
*/
function apiPost(site, func, data, triedLogin, triedCSRFUpdate) {
logDebug(site, "Performing request to API function "+func);
checkRateLimit(site, 60 * 10, site.requests10Minutes? site.requests10Minutes: 100);
checkRateLimit(site, 60 * 60, site.requests60Minutes? site.requests60Minutes: 200);
return rp({
"method": "POST",
"jar": site.cookieJar,
Expand Down

0 comments on commit f2409ba

Please sign in to comment.