[PATCH 02/13] selftests/bpf: test_xsk: Fix memory leaks

[Bastien Curutchet (eBPF Foundation)]

Some tests introduce memory leaks by not freeing all the pkt_stream
objects they're creating.

Fix these memory leaks.

Signed-off-by: Bastien Curutchet (eBPF Foundation) 
<bastien.curutc...@bootlin.com>
---
 tools/testing/selftests/bpf/xskxceiver.c | 32 +++++++++++++++++++++++++++++++-
 1 file changed, 31 insertions(+), 1 deletion(-)

diff --git a/tools/testing/selftests/bpf/xskxceiver.c 
b/tools/testing/selftests/bpf/xskxceiver.c
index 
3585c5afe9842b59fbd50256e5bc83a58a0ee38f..f43a6ab524f4758d0e155c254af3838b4079df13
 100644
--- a/tools/testing/selftests/bpf/xskxceiver.c
+++ b/tools/testing/selftests/bpf/xskxceiver.c
@@ -791,6 +791,13 @@ static void pkt_stream_receive_half(struct test_spec *test)
        struct pkt_stream *pkt_stream = test->ifobj_tx->xsk->pkt_stream;
        u32 i;
 
+       if (test->ifobj_rx->xsk->pkt_stream != test->rx_pkt_stream_default)
+               /* Packet stream has already been replaced so we have to 
release this one.
+                * The newly created one will be freed by the restore_default() 
at the
+                * end of the test
+                */
+               pkt_stream_delete(test->ifobj_rx->xsk->pkt_stream);
+
        test->ifobj_rx->xsk->pkt_stream = 
pkt_stream_generate(pkt_stream->nb_pkts,
                                                              
pkt_stream->pkts[0].len);
        pkt_stream = test->ifobj_rx->xsk->pkt_stream;
@@ -818,6 +825,22 @@ static void pkt_stream_even_odd_sequence(struct test_spec 
*test)
        }
 }
 
+static void release_even_odd_sequence(struct test_spec *test)
+{
+       struct pkt_stream *later_free_tx = test->ifobj_tx->xsk->pkt_stream;
+       struct pkt_stream *later_free_rx = test->ifobj_rx->xsk->pkt_stream;
+       int i;
+
+       for (i = 0; i < test->nb_sockets; i++) {
+               /* later_free_{rx/tx} will be freed by restore_default() */
+               if (test->ifobj_tx->xsk_arr[i].pkt_stream != later_free_tx)
+                       
pkt_stream_delete(test->ifobj_tx->xsk_arr[i].pkt_stream);
+               if (test->ifobj_rx->xsk_arr[i].pkt_stream != later_free_rx)
+                       
pkt_stream_delete(test->ifobj_rx->xsk_arr[i].pkt_stream);
+       }
+
+}
+
 static u64 pkt_get_addr(struct pkt *pkt, struct xsk_umem_info *umem)
 {
        if (!pkt->valid)
@@ -2057,6 +2080,7 @@ static int testapp_stats_tx_invalid_descs(struct 
test_spec *test)
 static int testapp_stats_rx_full(struct test_spec *test)
 {
        pkt_stream_replace(test, DEFAULT_UMEM_BUFFERS + DEFAULT_UMEM_BUFFERS / 
2, MIN_PKT_SIZE);
+       pkt_stream_delete(test->ifobj_rx->xsk->pkt_stream);
        test->ifobj_rx->xsk->pkt_stream = 
pkt_stream_generate(DEFAULT_UMEM_BUFFERS, MIN_PKT_SIZE);
 
        test->ifobj_rx->xsk->rxqsize = DEFAULT_UMEM_BUFFERS;
@@ -2068,6 +2092,7 @@ static int testapp_stats_rx_full(struct test_spec *test)
 static int testapp_stats_fill_empty(struct test_spec *test)
 {
        pkt_stream_replace(test, DEFAULT_UMEM_BUFFERS + DEFAULT_UMEM_BUFFERS / 
2, MIN_PKT_SIZE);
+       pkt_stream_delete(test->ifobj_rx->xsk->pkt_stream);
        test->ifobj_rx->xsk->pkt_stream = 
pkt_stream_generate(DEFAULT_UMEM_BUFFERS, MIN_PKT_SIZE);
 
        test->ifobj_rx->use_fill_ring = false;
@@ -2238,6 +2263,7 @@ static int testapp_xdp_shared_umem(struct test_spec *test)
 {
        struct xsk_xdp_progs *skel_rx = test->ifobj_rx->xdp_progs;
        struct xsk_xdp_progs *skel_tx = test->ifobj_tx->xdp_progs;
+       int ret;
 
        test->total_steps = 1;
        test->nb_sockets = 2;
@@ -2248,7 +2274,11 @@ static int testapp_xdp_shared_umem(struct test_spec 
*test)
 
        pkt_stream_even_odd_sequence(test);
 
-       return testapp_validate_traffic(test);
+       ret = testapp_validate_traffic(test);
+
+       release_even_odd_sequence(test);
+
+       return ret;
 }
 
 static int testapp_poll_txq_tmout(struct test_spec *test)

-- 
2.48.1