2 files changed, 12 insertions, 0 deletions

diff --git a/include/litmus/reservation.h b/include/litmus/reservation.h
index 9c23e27165b2..4eecd3f088e8 100644
--- a/include/litmus/reservation.h
+++ b/include/litmus/reservation.h
@@ -117,6 +117,10 @@ struct reservation {
         lt_t cur_budget;
         lt_t next_replenishment;
 
+        /* budget stats */
+        lt_t budget_consumed; /* how much budget consumed in this allocation cycle? */
+        lt_t budget_consumed_total;
+
         /* interaction with framework */
         struct reservation_environment *env;
         struct reservation_ops *ops;
diff --git a/litmus/polling_reservations.c b/litmus/polling_reservations.c
index 5c9b1832d2d9..2c481b46ba39 100644
--- a/litmus/polling_reservations.c
+++ b/litmus/polling_reservations.c
@@ -84,6 +84,7 @@ static void periodic_polling_on_replenishment(
         /* replenish budget */
         res->cur_budget = pres->max_budget;
         res->next_replenishment += pres->period;
+        res->budget_consumed = 0;
 
         switch (res->state) {
                 case RESERVATION_DEPLETED:
@@ -129,6 +130,9 @@ static void common_drain_budget(
         else
                 res->cur_budget -= how_much;
 
+        res->budget_consumed += how_much;
+        res->budget_consumed_total += how_much;
+
         switch (res->state) {
                 case RESERVATION_DEPLETED:
                 case RESERVATION_INACTIVE:
@@ -384,6 +388,7 @@ static void td_replenish(
         /* replenish budget */
         tdres->cur_interval = tdres->intervals + tdres->next_interval;
         res->cur_budget = td_time_remaining_until_end(tdres);
+        res->budget_consumed = 0;
         TRACE("td_replenish(%u): %s budget=%llu\n", res->id,
                 res->cur_budget ? "" : "WARNING", res->cur_budget);
 
@@ -422,6 +427,9 @@ static void td_drain_budget(
         struct table_driven_reservation *tdres =
                 container_of(res, struct table_driven_reservation, res);
 
+        res->budget_consumed += how_much;
+        res->budget_consumed_total += how_much;
+
         /* Table-driven scheduling: instead of tracking the budget, we compute
          * how much time is left in this allocation interval. */