add rate limiting feature

This commit is contained in:
teddit 2021-03-27 21:09:18 +01:00
parent c30a191d78
commit 71f99b44fc
2 changed files with 45 additions and 0 deletions

2
app.js
View File

@ -3,6 +3,8 @@ const config = require('./config')
global.client_id_b64 = Buffer.from(`${config.reddit_app_id}:`).toString('base64') global.client_id_b64 = Buffer.from(`${config.reddit_app_id}:`).toString('base64')
global.reddit_access_token = null global.reddit_access_token = null
global.reddit_refresh_token = null global.reddit_refresh_token = null
global.ratelimit_counts = {}
global.ratelimit_timestamps = {}
const pug = require('pug') const pug = require('pug')
const path = require('path') const path = require('path')

View File

@ -11,6 +11,49 @@ module.exports = (app, redis, fetch, RedditAPI) => {
let tedditApiSubreddit = require('./inc/teddit_api/handleSubreddit.js')(); let tedditApiSubreddit = require('./inc/teddit_api/handleSubreddit.js')();
let tedditApiUser = require('./inc/teddit_api/handleUser.js')(); let tedditApiUser = require('./inc/teddit_api/handleUser.js')();
let processSubredditsExplore = require('./inc/processSubredditsExplore.js')(); let processSubredditsExplore = require('./inc/processSubredditsExplore.js')();
app.all('*', (req, res, next) => {
if(config.rate_limiting.enabled) {
/**
* This route enforces request limits based on an IP address if
* config.rate_limiting.enabled is true. By default it's false.
*/
let ip = String(req.headers['x-forwarded-for'] || req.connection.remoteAddress || 'unknown')
if(ip === 'unknown') {
next()
}
if(ratelimit_counts[ip] == undefined) {
ratelimit_counts[ip] = 0
}
if(ratelimit_timestamps[ip] == undefined) {
ratelimit_timestamps[ip] = Date.now()
}
let diff = Date.now() - ratelimit_timestamps[ip]
let credit = (diff / 60000) * config.rate_limiting.limit_after_limited
ratelimit_counts[ip] -= credit
if(ratelimit_counts[ip] < 0) {
ratelimit_counts[ip] = 0
}
ratelimit_counts[ip]++
ratelimit_timestamps[ip] = Date.now()
if(ratelimit_counts[ip] > config.rate_limiting.initial_limit) {
console.log(`RATE LIMITED IP ADDRESS: ${ip}`)
return res.send(`Hold your horses! You have hit the request limit. You should be able to refresh this page in a couple of seconds. If you think you are wrongfully limited, create an issue at https://codeberg.org/teddit/teddit. Rate limiting is highly experimental feature.`)
} else {
next()
}
} else {
next()
}
})
app.get('/about', (req, res, next) => { app.get('/about', (req, res, next) => {
return res.render('about', { user_preferences: req.cookies }) return res.render('about', { user_preferences: req.cookies })