From 8a852397462e39cfbc21e3cea20ddd39be40598d Mon Sep 17 00:00:00 2001 From: Nick Mathewson Date: Tue, 10 Apr 2018 12:33:30 -0400 Subject: Add a helper function to decrement read and write at the same time --- src/common/token_bucket.c | 11 +++++++++++ src/common/token_bucket.h | 3 +++ 2 files changed, 14 insertions(+) (limited to 'src') diff --git a/src/common/token_bucket.c b/src/common/token_bucket.c index f4d2cccffd..abb0501833 100644 --- a/src/common/token_bucket.c +++ b/src/common/token_bucket.c @@ -178,3 +178,14 @@ token_bucket_dec_write(token_bucket_t *bucket, return decrement_single_bucket(&bucket->write_bucket, n); } +/** + * As token_bucket_dec_read and token_bucket_dec_write, in a single operation. + */ +void +token_bucket_dec(token_bucket_t *bucket, + ssize_t n_read, ssize_t n_written) +{ + token_bucket_dec_read(bucket, n_read); + token_bucket_dec_read(bucket, n_written); +} + diff --git a/src/common/token_bucket.h b/src/common/token_bucket.h index ef0735219e..2d1ccd5cf3 100644 --- a/src/common/token_bucket.h +++ b/src/common/token_bucket.h @@ -43,6 +43,9 @@ int token_bucket_dec_read(token_bucket_t *bucket, int token_bucket_dec_write(token_bucket_t *bucket, ssize_t n); +void token_bucket_dec(token_bucket_t *bucket, + ssize_t n_read, ssize_t n_written); + static inline size_t token_bucket_get_read(const token_bucket_t *bucket); static inline size_t token_bucket_get_read(const token_bucket_t *bucket) -- cgit v1.2.3-54-g00ecf