From a6ccb05bb107bb91a2eeeb95b00c384d1d627968 Mon Sep 17 00:00:00 2001
From: Andre Puschmann <andre@softwareradiosystems.com>
Date: Fri, 16 Feb 2018 21:32:53 +0100
Subject: [PATCH] fix mem leak in RLC UM test and RLC AM stress tester

---
 lib/test/upper/rlc_am_stress_test.cc | 1 +
 lib/test/upper/rlc_um_test.cc        | 8 ++++++++
 2 files changed, 9 insertions(+)

diff --git a/lib/test/upper/rlc_am_stress_test.cc b/lib/test/upper/rlc_am_stress_test.cc
index 086a41bd8..137236b4a 100644
--- a/lib/test/upper/rlc_am_stress_test.cc
+++ b/lib/test/upper/rlc_am_stress_test.cc
@@ -85,6 +85,7 @@ private:
       usleep(100);
     }
     running = false;
+    byte_buffer_pool::get_instance()->deallocate(pdu);
   }
 
   rlc_interface_mac *rlc1;
diff --git a/lib/test/upper/rlc_um_test.cc b/lib/test/upper/rlc_um_test.cc
index 6ce39ef7f..3755c1174 100644
--- a/lib/test/upper/rlc_um_test.cc
+++ b/lib/test/upper/rlc_um_test.cc
@@ -62,6 +62,14 @@ public:
     n_sdus = 0;
   }
 
+  ~rlc_um_tester(){
+    for (uint32_t i = 0; i < NBUFS; i++) {
+      if (sdus[i] != NULL) {
+        byte_buffer_pool::get_instance()->deallocate(sdus[i]);
+      }
+    }
+  }
+
   // PDCP interface
   void write_pdu(uint32_t lcid, byte_buffer_t *sdu)
   {