Protect token bucket against overflow.

This could happen when the clock is stepped.

Protect token bucket against overflow.
This could happen when the clock is stepped.
4ee703d7 · Juliusz Chroboczek · 1fef22a2 · 4ee703d7
Commit 4ee703d7 authored Mar 31, 2008 by Juliusz Chroboczek
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

message.c message.c +3 -1

No files found.
--- a/message.c
+++ b/message.c
@@ -331,7 +331,9 @@ check_bucket(struct network *net)
 {
    if(net->bucket > 0 && now.tv_sec > net->bucket_time) {
        net->bucket =
-            MAX(0, (int)net->bucket - 40 * (now.tv_sec - net->bucket_time));
+            MIN(MAX((int)net->bucket - 40 * (now.tv_sec - net->bucket_time),
+                    0),
+                400);
    }

    net->bucket_time = now.tv_sec;