diff options
| author | Lukáš Ondráček <lukas.ondracek@nic.cz> | 2024-12-12 15:39:06 +0100 |
|---|---|---|
| committer | Lukáš Ondráček <lukas.ondracek@nic.cz> | 2024-12-12 15:53:29 +0100 |
| commit | 5ec4d4e8461a47cd7e69bb7529d7bd12329d7b93 (patch) | |
| tree | e9593c0ec914229b158b229df72bcecc20231b92 /daemon/defer.c | |
| parent | daemon/defer: change KRU configuration (diff) | |
| download | knot-resolver-5ec4d4e8461a47cd7e69bb7529d7bd12329d7b93.tar.xz knot-resolver-5ec4d4e8461a47cd7e69bb7529d7bd12329d7b93.zip | |
daemon/defer: fix price overflow for too long operations
Diffstat (limited to 'daemon/defer.c')
| -rw-r--r-- | daemon/defer.c | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/daemon/defer.c b/daemon/defer.c index fabc76c8..0734fd0e 100644 --- a/daemon/defer.c +++ b/daemon/defer.c @@ -223,14 +223,15 @@ void defer_charge(uint64_t nsec, union kr_sockaddr *addr, bool stream) _Alignas(16) uint8_t key[16] = {0, }; uint16_t max_load = 0; uint8_t prefix = 0; - kru_price_t base_price = BASE_PRICE(nsec); + uint64_t base_price = BASE_PRICE(nsec); if (addr->ip.sa_family == AF_INET6) { memcpy(key, &addr->ip6.sin6_addr, 16); kru_price_t prices[V6_PREFIXES_CNT]; for (size_t i = 0; i < V6_PREFIXES_CNT; i++) { - prices[i] = base_price / V6_RATE_MULT[i]; + uint64_t price = base_price / V6_RATE_MULT[i]; + prices[i] = price > (kru_price_t)-1 ? -1 : price; } max_load = KRU.load_multi_prefix_max((struct kru *)defer->kru, kr_now(), @@ -240,7 +241,8 @@ void defer_charge(uint64_t nsec, union kr_sockaddr *addr, bool stream) kru_price_t prices[V4_PREFIXES_CNT]; for (size_t i = 0; i < V4_PREFIXES_CNT; i++) { - prices[i] = base_price / V4_RATE_MULT[i]; + uint64_t price = base_price / V4_RATE_MULT[i]; + prices[i] = price > (kru_price_t)-1 ? -1 : price; } max_load = KRU.load_multi_prefix_max((struct kru *)defer->kru, kr_now(), |