diff options
| author | Nick Mathewson <nickm@torproject.org> | 2018-04-10 12:33:30 -0400 |
|---|---|---|
| committer | Nick Mathewson <nickm@torproject.org> | 2018-04-13 10:41:14 -0400 |
| commit | 8a852397462e39cfbc21e3cea20ddd39be40598d (patch) | |
| tree | d2f55e9b9ae4f573f1d29b6b2aade39e93d58262 /src | |
| parent | c376200f6a77b2509928bc08d2aa1245028cec30 (diff) | |
| download | tor-8a852397462e39cfbc21e3cea20ddd39be40598d.tar.gz tor-8a852397462e39cfbc21e3cea20ddd39be40598d.zip | |
Add a helper function to decrement read and write at the same time
Diffstat (limited to 'src')
| -rw-r--r-- | src/common/token_bucket.c | 11 | ||||
| -rw-r--r-- | src/common/token_bucket.h | 3 |
2 files changed, 14 insertions, 0 deletions
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) |