record operations on mempool objects when it is allocated
and released inside the mlx5 PMD.

Signed-off-by: Shani Peretz <shper...@nvidia.com>
---
 drivers/net/mlx5/mlx5_rx.c       | 9 +++++++++
 drivers/net/mlx5/mlx5_rx.h       | 2 ++
 drivers/net/mlx5/mlx5_rxq.c      | 9 +++++++--
 drivers/net/mlx5/mlx5_rxtx_vec.c | 6 ++++++
 drivers/net/mlx5/mlx5_tx.h       | 7 +++++++
 drivers/net/mlx5/mlx5_txq.c      | 1 +
 6 files changed, 32 insertions(+), 2 deletions(-)

diff --git a/drivers/net/mlx5/mlx5_rx.c b/drivers/net/mlx5/mlx5_rx.c
index 5f4a93fe8c..a86ed2180e 100644
--- a/drivers/net/mlx5/mlx5_rx.c
+++ b/drivers/net/mlx5/mlx5_rx.c
@@ -560,12 +560,15 @@ mlx5_rx_err_handle(struct mlx5_rxq_data *rxq, uint8_t vec,
                                        elt_idx = (elts_ci + i) & e_mask;
                                        elt = &(*rxq->elts)[elt_idx];
                                        *elt = rte_mbuf_raw_alloc(rxq->mp);
+                                       rte_mempool_history_mark(*elt, 
RTE_MEMPOOL_PMD_ALLOC);
                                        if (!*elt) {
                                                for (i--; i >= 0; --i) {
                                                        elt_idx = (elts_ci +
                                                                   i) & elts_n;
                                                        elt = &(*rxq->elts)
                                                                [elt_idx];
+                                                       
rte_mempool_history_mark(*elt,
+                                                                       
RTE_MEMPOOL_PMD_FREE);
                                                        rte_pktmbuf_free_seg
                                                                (*elt);
                                                }
@@ -952,6 +955,7 @@ mlx5_rx_burst(void *dpdk_rxq, struct rte_mbuf **pkts, 
uint16_t pkts_n)
                rte_prefetch0(wqe);
                /* Allocate the buf from the same pool. */
                rep = rte_mbuf_raw_alloc(seg->pool);
+               rte_mempool_history_mark(rep, RTE_MEMPOOL_PMD_ALLOC);
                if (unlikely(rep == NULL)) {
                        ++rxq->stats.rx_nombuf;
                        if (!pkt) {
@@ -966,6 +970,7 @@ mlx5_rx_burst(void *dpdk_rxq, struct rte_mbuf **pkts, 
uint16_t pkts_n)
                                rep = NEXT(pkt);
                                NEXT(pkt) = NULL;
                                NB_SEGS(pkt) = 1;
+                               rte_mempool_history_mark(pkt, 
RTE_MEMPOOL_PMD_FREE);
                                rte_mbuf_raw_free(pkt);
                                pkt = rep;
                        }
@@ -979,6 +984,7 @@ mlx5_rx_burst(void *dpdk_rxq, struct rte_mbuf **pkts, 
uint16_t pkts_n)
                        len = mlx5_rx_poll_len(rxq, cqe, cqe_n, cqe_mask, 
&mcqe, &skip_cnt, false);
                        if (unlikely(len & MLX5_ERROR_CQE_MASK)) {
                                /* We drop packets with non-critical errors */
+                               rte_mempool_history_mark(rep, 
RTE_MEMPOOL_PMD_FREE);
                                rte_mbuf_raw_free(rep);
                                if (len == MLX5_CRITICAL_ERROR_CQE_RET) {
                                        rq_ci = rxq->rq_ci << sges_n;
@@ -992,6 +998,7 @@ mlx5_rx_burst(void *dpdk_rxq, struct rte_mbuf **pkts, 
uint16_t pkts_n)
                                continue;
                        }
                        if (len == 0) {
+                               rte_mempool_history_mark(rep, 
RTE_MEMPOOL_PMD_FREE);
                                rte_mbuf_raw_free(rep);
                                break;
                        }
@@ -1268,6 +1275,7 @@ mlx5_rx_burst_mprq(void *dpdk_rxq, struct rte_mbuf 
**pkts, uint16_t pkts_n)
                        ++rxq->stats.rx_nombuf;
                        break;
                }
+               rte_mempool_history_mark(pkt, RTE_MEMPOOL_PMD_ALLOC);
                len = (byte_cnt & MLX5_MPRQ_LEN_MASK) >> MLX5_MPRQ_LEN_SHIFT;
                MLX5_ASSERT((int)len >= (rxq->crc_present << 2));
                if (rxq->crc_present)
@@ -1275,6 +1283,7 @@ mlx5_rx_burst_mprq(void *dpdk_rxq, struct rte_mbuf 
**pkts, uint16_t pkts_n)
                rxq_code = mprq_buf_to_pkt(rxq, pkt, len, buf,
                                           strd_idx, strd_cnt);
                if (unlikely(rxq_code != MLX5_RXQ_CODE_EXIT)) {
+                       rte_mempool_history_mark(pkt, RTE_MEMPOOL_PMD_FREE);
                        rte_pktmbuf_free_seg(pkt);
                        if (rxq_code == MLX5_RXQ_CODE_DROPPED) {
                                ++rxq->stats.idropped;
diff --git a/drivers/net/mlx5/mlx5_rx.h b/drivers/net/mlx5/mlx5_rx.h
index 6380895502..db4ef10ca1 100644
--- a/drivers/net/mlx5/mlx5_rx.h
+++ b/drivers/net/mlx5/mlx5_rx.h
@@ -516,6 +516,7 @@ mprq_buf_to_pkt(struct mlx5_rxq_data *rxq, struct rte_mbuf 
*pkt, uint32_t len,
 
                                if (unlikely(next == NULL))
                                        return MLX5_RXQ_CODE_NOMBUF;
+                               rte_mempool_history_mark(next, 
RTE_MEMPOOL_PMD_ALLOC);
                                NEXT(prev) = next;
                                SET_DATA_OFF(next, 0);
                                addr = RTE_PTR_ADD(addr, seg_len);
@@ -579,6 +580,7 @@ mprq_buf_to_pkt(struct mlx5_rxq_data *rxq, struct rte_mbuf 
*pkt, uint32_t len,
 
                        if (unlikely(seg == NULL))
                                return MLX5_RXQ_CODE_NOMBUF;
+                       rte_mempool_history_mark(seg, RTE_MEMPOOL_PMD_ALLOC);
                        SET_DATA_OFF(seg, 0);
                        rte_memcpy(rte_pktmbuf_mtod(seg, void *),
                                RTE_PTR_ADD(addr, len - hdrm_overlap),
diff --git a/drivers/net/mlx5/mlx5_rxq.c b/drivers/net/mlx5/mlx5_rxq.c
index f5df451a32..e95bef9d55 100644
--- a/drivers/net/mlx5/mlx5_rxq.c
+++ b/drivers/net/mlx5/mlx5_rxq.c
@@ -164,6 +164,7 @@ rxq_alloc_elts_sprq(struct mlx5_rxq_ctrl *rxq_ctrl)
                        rte_errno = ENOMEM;
                        goto error;
                }
+               rte_mempool_history_mark(buf, RTE_MEMPOOL_PMD_ALLOC);
                /* Only vectored Rx routines rely on headroom size. */
                MLX5_ASSERT(!has_vec_support ||
                            DATA_OFF(buf) >= RTE_PKTMBUF_HEADROOM);
@@ -221,8 +222,10 @@ rxq_alloc_elts_sprq(struct mlx5_rxq_ctrl *rxq_ctrl)
        err = rte_errno; /* Save rte_errno before cleanup. */
        elts_n = i;
        for (i = 0; (i != elts_n); ++i) {
-               if ((*rxq_ctrl->rxq.elts)[i] != NULL)
+               if ((*rxq_ctrl->rxq.elts)[i] != NULL) {
+                       rte_mempool_history_mark((*rxq_ctrl->rxq.elts)[i], 
RTE_MEMPOOL_PMD_FREE);
                        rte_pktmbuf_free_seg((*rxq_ctrl->rxq.elts)[i]);
+               }
                (*rxq_ctrl->rxq.elts)[i] = NULL;
        }
        if (rxq_ctrl->share_group == 0)
@@ -324,8 +327,10 @@ rxq_free_elts_sprq(struct mlx5_rxq_ctrl *rxq_ctrl)
                rxq->rq_pi = elts_ci;
        }
        for (i = 0; i != q_n; ++i) {
-               if ((*rxq->elts)[i] != NULL)
+               if ((*rxq->elts)[i] != NULL) {
+                       rte_mempool_history_mark((*rxq->elts)[i], 
RTE_MEMPOOL_PMD_FREE);
                        rte_pktmbuf_free_seg((*rxq->elts)[i]);
+               }
                (*rxq->elts)[i] = NULL;
        }
 }
diff --git a/drivers/net/mlx5/mlx5_rxtx_vec.c b/drivers/net/mlx5/mlx5_rxtx_vec.c
index 1b701801c5..ffaa10c547 100644
--- a/drivers/net/mlx5/mlx5_rxtx_vec.c
+++ b/drivers/net/mlx5/mlx5_rxtx_vec.c
@@ -64,6 +64,7 @@ rxq_handle_pending_error(struct mlx5_rxq_data *rxq, struct 
rte_mbuf **pkts,
 #ifdef MLX5_PMD_SOFT_COUNTERS
                        err_bytes += PKT_LEN(pkt);
 #endif
+                       rte_mempool_history_mark(pkt, RTE_MEMPOOL_PMD_FREE);
                        rte_pktmbuf_free_seg(pkt);
                } else {
                        pkts[n++] = pkt;
@@ -107,6 +108,7 @@ mlx5_rx_replenish_bulk_mbuf(struct mlx5_rxq_data *rxq)
                        rxq->stats.rx_nombuf += n;
                        return;
                }
+               rte_mempool_history_bulk((void *)elts, n, 
RTE_MEMPOOL_PMD_ALLOC);
                if (unlikely(mlx5_mr_btree_len(&rxq->mr_ctrl.cache_bh) > 1)) {
                        for (i = 0; i < n; ++i) {
                                /*
@@ -171,6 +173,7 @@ mlx5_rx_mprq_replenish_bulk_mbuf(struct mlx5_rxq_data *rxq)
                        rxq->stats.rx_nombuf += n;
                        return;
                }
+               rte_mempool_history_bulk((void *)elts, n, 
RTE_MEMPOOL_PMD_ALLOC);
                rxq->elts_ci += n;
                /* Prevent overflowing into consumed mbufs. */
                elts_idx = rxq->elts_ci & wqe_mask;
@@ -224,6 +227,7 @@ rxq_copy_mprq_mbuf_v(struct mlx5_rxq_data *rxq,
 
                if (!elts[i]->pkt_len) {
                        rxq->consumed_strd = strd_n;
+                       rte_mempool_history_mark(elts[i], RTE_MEMPOOL_PMD_FREE);
                        rte_pktmbuf_free_seg(elts[i]);
 #ifdef MLX5_PMD_SOFT_COUNTERS
                        rxq->stats.ipackets -= 1;
@@ -236,6 +240,7 @@ rxq_copy_mprq_mbuf_v(struct mlx5_rxq_data *rxq,
                                           buf, rxq->consumed_strd, strd_cnt);
                rxq->consumed_strd += strd_cnt;
                if (unlikely(rxq_code != MLX5_RXQ_CODE_EXIT)) {
+                       rte_mempool_history_mark(elts[i], RTE_MEMPOOL_PMD_FREE);
                        rte_pktmbuf_free_seg(elts[i]);
 #ifdef MLX5_PMD_SOFT_COUNTERS
                        rxq->stats.ipackets -= 1;
@@ -586,6 +591,7 @@ mlx5_rx_burst_mprq_vec(void *dpdk_rxq, struct rte_mbuf 
**pkts, uint16_t pkts_n)
                rte_io_wmb();
                *rxq->cq_db = rte_cpu_to_be_32(rxq->cq_ci);
        } while (tn != pkts_n);
+
        return tn;
 }
 
diff --git a/drivers/net/mlx5/mlx5_tx.h b/drivers/net/mlx5/mlx5_tx.h
index 55568c41b1..7b61d87120 100644
--- a/drivers/net/mlx5/mlx5_tx.h
+++ b/drivers/net/mlx5/mlx5_tx.h
@@ -553,6 +553,7 @@ mlx5_tx_free_mbuf(struct mlx5_txq_data *__rte_restrict txq,
        if (!MLX5_TXOFF_CONFIG(MULTI) && txq->fast_free) {
                mbuf = *pkts;
                pool = mbuf->pool;
+               rte_mempool_history_bulk((void *)pkts, pkts_n, 
RTE_MEMPOOL_PMD_FREE);
                rte_mempool_put_bulk(pool, (void *)pkts, pkts_n);
                return;
        }
@@ -608,6 +609,7 @@ mlx5_tx_free_mbuf(struct mlx5_txq_data *__rte_restrict txq,
                         * Free the array of pre-freed mbufs
                         * belonging to the same memory pool.
                         */
+                       rte_mempool_history_bulk((void *)p_free, n_free, 
RTE_MEMPOOL_PMD_FREE);
                        rte_mempool_put_bulk(pool, (void *)p_free, n_free);
                        if (unlikely(mbuf != NULL)) {
                                /* There is the request to start new scan. */
@@ -1223,6 +1225,7 @@ mlx5_tx_mseg_memcpy(uint8_t *pdst,
                        /* Exhausted packet, just free. */
                        mbuf = loc->mbuf;
                        loc->mbuf = mbuf->next;
+                       rte_mempool_history_mark(mbuf, RTE_MEMPOOL_PMD_FREE);
                        rte_pktmbuf_free_seg(mbuf);
                        loc->mbuf_off = 0;
                        MLX5_ASSERT(loc->mbuf_nseg > 1);
@@ -1265,6 +1268,7 @@ mlx5_tx_mseg_memcpy(uint8_t *pdst,
                                /* Exhausted packet, just free. */
                                mbuf = loc->mbuf;
                                loc->mbuf = mbuf->next;
+                               rte_mempool_history_mark(mbuf, 
RTE_MEMPOOL_PMD_FREE);
                                rte_pktmbuf_free_seg(mbuf);
                                loc->mbuf_off = 0;
                                MLX5_ASSERT(loc->mbuf_nseg >= 1);
@@ -1715,6 +1719,7 @@ mlx5_tx_mseg_build(struct mlx5_txq_data *__rte_restrict 
txq,
                        /* Zero length segment found, just skip. */
                        mbuf = loc->mbuf;
                        loc->mbuf = loc->mbuf->next;
+                       rte_mempool_history_mark(mbuf, RTE_MEMPOOL_PMD_FREE);
                        rte_pktmbuf_free_seg(mbuf);
                        if (--loc->mbuf_nseg == 0)
                                break;
@@ -2018,6 +2023,7 @@ mlx5_tx_packet_multi_send(struct mlx5_txq_data 
*__rte_restrict txq,
                        wqe->cseg.sq_ds -= RTE_BE32(1);
                        mbuf = loc->mbuf;
                        loc->mbuf = mbuf->next;
+                       rte_mempool_history_mark(mbuf, RTE_MEMPOOL_PMD_FREE);
                        rte_pktmbuf_free_seg(mbuf);
                        if (--nseg == 0)
                                break;
@@ -3317,6 +3323,7 @@ mlx5_tx_burst_single_send(struct mlx5_txq_data 
*__rte_restrict txq,
                                 * Packet data are completely inlined,
                                 * free the packet immediately.
                                 */
+                               rte_mempool_history_mark(loc->mbuf, 
RTE_MEMPOOL_PMD_FREE);
                                rte_pktmbuf_free_seg(loc->mbuf);
                        } else if ((!MLX5_TXOFF_CONFIG(EMPW) ||
                                     MLX5_TXOFF_CONFIG(MPW)) &&
diff --git a/drivers/net/mlx5/mlx5_txq.c b/drivers/net/mlx5/mlx5_txq.c
index 5fee5bc4e8..156f8c2ef8 100644
--- a/drivers/net/mlx5/mlx5_txq.c
+++ b/drivers/net/mlx5/mlx5_txq.c
@@ -78,6 +78,7 @@ txq_free_elts(struct mlx5_txq_ctrl *txq_ctrl)
                struct rte_mbuf *elt = (*elts)[elts_tail & elts_m];
 
                MLX5_ASSERT(elt != NULL);
+               rte_mempool_history_mark(elt, RTE_MEMPOOL_PMD_FREE);
                rte_pktmbuf_free_seg(elt);
 #ifdef RTE_LIBRTE_MLX5_DEBUG
                /* Poisoning. */
-- 
2.34.1

Reply via email to