tbf: Add ul and dl TBF types and allocate them in tbf_alloc()

Ticket: SYS#389
Sponsored-by: On-Waves ehf

2 files changed, 11 insertions, 1 deletions

diff --git a/src/tbf.cpp b/src/tbf.cpp
index f913f970..39549f1e 100644
--- a/src/tbf.cpp
+++ b/src/tbf.cpp
@@ -495,7 +495,11 @@ struct gprs_rlcmac_tbf *tbf_alloc(struct gprs_rlcmac_bts *bts,
 	if (trx >= 8 || tfi >= 32)
 		return NULL;
 
-	tbf = talloc_zero(tall_pcu_ctx, struct gprs_rlcmac_tbf);
+	if (dir == GPRS_RLCMAC_UL_TBF)
+		tbf = talloc_zero(tall_pcu_ctx, struct gprs_rlcmac_ul_tbf);
+	else
+		tbf = talloc_zero(tall_pcu_ctx, struct gprs_rlcmac_dl_tbf);
+
 	if (!tbf)
 		return NULL;
 
diff --git a/src/tbf.h b/src/tbf.h
index 80e2068f..8464e197 100644
--- a/src/tbf.h
+++ b/src/tbf.h
@@ -326,6 +326,12 @@ inline time_t gprs_rlcmac_tbf::created_ts() const
 	return m_created_ts;
 }
 
+struct gprs_rlcmac_dl_tbf : public gprs_rlcmac_tbf {
+};
+
+struct gprs_rlcmac_ul_tbf : public gprs_rlcmac_tbf {
+};
+
 #endif
 
 #ifdef __cplusplus