From: Nick Mathewson <nickm@torproject.org>
Date: Tue, 10 Apr 2018 16:33:30 +0000 (-0400)
Subject: Add a helper function to decrement read and write at the same time
X-Git-Tag: tor-0.3.4.1-alpha~159^2~7
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=8a852397462e39cfbc21e3cea20ddd39be40598d;p=thirdparty%2Ftor.git

Add a helper function to decrement read and write at the same time
---

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)