bts: Add methods to BTS to track octets sent over the downlink

2 files changed, 21 insertions, 0 deletions

diff --git a/src/bts.cpp b/src/bts.cpp
index 69e5ad40..d2417cc1 100644
--- a/src/bts.cpp
+++ b/src/bts.cpp
@@ -100,6 +100,7 @@ BTS::BTS()
 	: m_cur_fn(0)
 	, m_pollController(*this)
 	, m_sba(*this)
+	, m_dl_octets_sent(0)
 {
 	memset(&m_bts, 0, sizeof(m_bts));
 	INIT_LLIST_HEAD(&m_bts.ul_tbfs);
@@ -512,6 +513,22 @@ void BTS::snd_dl_ass(gprs_rlcmac_tbf *tbf, uint8_t poll, const char *imsi)
 	bitvec_free(immediate_assignment);
 }
 
+/* Counts the number of octets sent over the downlink */
+void BTS::dl_octets_sent(size_t octets)
+{
+	m_dl_octets_sent += octets;
+}
+
+/* Counts the number of octets sent over the downlink */
+size_t BTS::dl_octets_sent_reset()
+{
+	size_t octets;
+
+	octets = m_dl_octets_sent;
+	m_dl_octets_sent = 0;
+
+	return octets;
+}
 
 /*
  * PDCH code below. TODO: move to a separate file
diff --git a/src/bts.h b/src/bts.h
index 924e2e1e..923a6ed2 100644
--- a/src/bts.h
+++ b/src/bts.h
@@ -223,6 +223,9 @@ public:
 	void llc_frame_sched();
 	void rach_frame();
 
+	void dl_octets_sent(size_t octets);
+	size_t dl_octets_sent_reset();
+
 	/*
 	 * Below for C interface for the VTY
 	 */
@@ -235,6 +238,7 @@ private:
 	SBAController m_sba;
 	TimingAdvance m_ta;
 	struct rate_ctr_group *m_ratectrs;
+	size_t m_dl_octets_sent;
 
 private:
 	/* disable copying to avoid slicing */