JavaScript: Add support for rate-limiter-flexible package.

This commit is contained in:
Max Schaefer
2019-09-18 12:13:23 +01:00
parent e7d8fa4251
commit 3970ead7ab
3 changed files with 50 additions and 0 deletions

View File

@@ -155,3 +155,44 @@ class RouteHandlerLimitedByExpressLimiter extends RateLimitedRouteHandlerExpr {
)
}
}
/**
* A rate-handler function implemented using one of the rate-limiting classes provided
* by the `rate-limiter-flexible` package.
*
* We look for functions that invoke the `consume` method of one of the `RateLimiter*`
* classes from the `rate-limiter-flexible` package on a property of their first argument,
* like the `rateLimiterMiddleware` function in this example:
*
* ```
* import { RateLimiterRedis } from 'rate-limiter-flexible';
* const rateLimiter = new RateLimiterRedis(...);
* function rateLimiterMiddleware(req, res, next) {
* rateLimiter.consume(req.ip).then(next).catch(res.status(429).send('rate limited'));
* }
* ```
*/
class RateLimiterFlexibleRateLimiter extends DataFlow::FunctionNode {
RateLimiterFlexibleRateLimiter() {
exists(
string rateLimiterClassName, DataFlow::SourceNode rateLimiterClass,
DataFlow::SourceNode rateLimiterInstance
|
rateLimiterClassName.matches("RateLimiter%") and
rateLimiterClass = DataFlow::moduleMember("rate-limiter-flexible", rateLimiterClassName) and
rateLimiterInstance = rateLimiterClass.getAnInstantiation() and
getParameter(0).getAPropertyRead() = rateLimiterInstance
.getAMemberCall("consume")
.getAnArgument()
)
}
}
/**
* A route-handler expression that is rate-limited by the `rate-limiter-flexible` package.
*/
class RouteHandlerLimitedByRateLimiterFlexible extends RateLimiter {
RouteHandlerLimitedByRateLimiterFlexible() {
any(RateLimiterFlexibleRateLimiter rl).flowsToExpr(this)
}
}