2a155c53ab
This helper alone doesn't make too much sense, but it's preparatory work for #17274, and I guess it can't hurt to land it early, it does make the ratelimit code a tiny bit prettier after all.
39 lines
740 B
C
39 lines
740 B
C
/* SPDX-License-Identifier: LGPL-2.1+ */
|
|
|
|
#include <sys/time.h>
|
|
|
|
#include "macro.h"
|
|
#include "ratelimit.h"
|
|
|
|
/* Modelled after Linux' lib/ratelimit.c by Dave Young
|
|
* <hidave.darkstar@gmail.com>, which is licensed GPLv2. */
|
|
|
|
bool ratelimit_below(RateLimit *r) {
|
|
usec_t ts;
|
|
|
|
assert(r);
|
|
|
|
if (!ratelimit_configured(r))
|
|
return true;
|
|
|
|
ts = now(CLOCK_MONOTONIC);
|
|
|
|
if (r->begin <= 0 ||
|
|
r->begin + r->interval < ts) {
|
|
r->begin = ts;
|
|
|
|
/* Reset counter */
|
|
r->num = 0;
|
|
goto good;
|
|
}
|
|
|
|
if (r->num < r->burst)
|
|
goto good;
|
|
|
|
return false;
|
|
|
|
good:
|
|
r->num++;
|
|
return true;
|
|
}
|