[PATCH net-next v5 5/7] ice: make use of DEFINE_FLEX() for struct ice_aqc_add_tx_qgrp

[Przemek Kitszel]

Use DEFINE_FLEX() macro for 1-elem flex array use case
of struct ice_aqc_add_tx_qgrp.

Signed-off-by: Przemek Kitszel <przemyslaw.kits...@intel.com>
---
add/remove: 0/2 grow/shrink: 2/2 up/down: 220/-255 (-35)
---
 drivers/net/ethernet/intel/ice/ice_lib.c | 23 +++++------------------
 drivers/net/ethernet/intel/ice/ice_xsk.c | 22 ++++++++--------------
 2 files changed, 13 insertions(+), 32 deletions(-)

diff --git a/drivers/net/ethernet/intel/ice/ice_lib.c 
b/drivers/net/ethernet/intel/ice/ice_lib.c
index 201570cd2e0b..b47d1055802e 100644
--- a/drivers/net/ethernet/intel/ice/ice_lib.c
+++ b/drivers/net/ethernet/intel/ice/ice_lib.c
@@ -1832,21 +1832,14 @@ int ice_vsi_cfg_single_rxq(struct ice_vsi *vsi, u16 
q_idx)
 
 int ice_vsi_cfg_single_txq(struct ice_vsi *vsi, struct ice_tx_ring **tx_rings, 
u16 q_idx)
 {
-       struct ice_aqc_add_tx_qgrp *qg_buf;
-       int err;
+       DEFINE_FLEX(struct ice_aqc_add_tx_qgrp, qg_buf, txqs, 1);
 
        if (q_idx >= vsi->alloc_txq || !tx_rings || !tx_rings[q_idx])
                return -EINVAL;
 
-       qg_buf = kzalloc(struct_size(qg_buf, txqs, 1), GFP_KERNEL);
-       if (!qg_buf)
-               return -ENOMEM;
-
        qg_buf->num_txqs = 1;
 
-       err = ice_vsi_cfg_txq(vsi, tx_rings[q_idx], qg_buf);
-       kfree(qg_buf);
-       return err;
+       return ice_vsi_cfg_txq(vsi, tx_rings[q_idx], qg_buf);
 }
 
 /**
@@ -1888,24 +1881,18 @@ int ice_vsi_cfg_rxqs(struct ice_vsi *vsi)
 static int
 ice_vsi_cfg_txqs(struct ice_vsi *vsi, struct ice_tx_ring **rings, u16 count)
 {
-       struct ice_aqc_add_tx_qgrp *qg_buf;
-       u16 q_idx = 0;
+       DEFINE_FLEX(struct ice_aqc_add_tx_qgrp, qg_buf, txqs, 1);
        int err = 0;
-
-       qg_buf = kzalloc(struct_size(qg_buf, txqs, 1), GFP_KERNEL);
-       if (!qg_buf)
-               return -ENOMEM;
+       u16 q_idx;
 
        qg_buf->num_txqs = 1;
 
        for (q_idx = 0; q_idx < count; q_idx++) {
                err = ice_vsi_cfg_txq(vsi, rings[q_idx], qg_buf);
                if (err)
-                       goto err_cfg_txqs;
+                       break;
        }
 
-err_cfg_txqs:
-       kfree(qg_buf);
        return err;
 }
 
diff --git a/drivers/net/ethernet/intel/ice/ice_xsk.c 
b/drivers/net/ethernet/intel/ice/ice_xsk.c
index 2a3f0834e139..99954508184f 100644
--- a/drivers/net/ethernet/intel/ice/ice_xsk.c
+++ b/drivers/net/ethernet/intel/ice/ice_xsk.c
@@ -217,61 +217,55 @@ static int ice_qp_dis(struct ice_vsi *vsi, u16 q_idx)
  */
 static int ice_qp_ena(struct ice_vsi *vsi, u16 q_idx)
 {
-       struct ice_aqc_add_tx_qgrp *qg_buf;
+       DEFINE_FLEX(struct ice_aqc_add_tx_qgrp, qg_buf, txqs, 1);
+       u16 size = __struct_size(qg_buf);
        struct ice_q_vector *q_vector;
        struct ice_tx_ring *tx_ring;
        struct ice_rx_ring *rx_ring;
-       u16 size;
        int err;
 
        if (q_idx >= vsi->num_rxq || q_idx >= vsi->num_txq)
                return -EINVAL;
 
-       size = struct_size(qg_buf, txqs, 1);
-       qg_buf = kzalloc(size, GFP_KERNEL);
-       if (!qg_buf)
-               return -ENOMEM;
-
        qg_buf->num_txqs = 1;
 
        tx_ring = vsi->tx_rings[q_idx];
        rx_ring = vsi->rx_rings[q_idx];
        q_vector = rx_ring->q_vector;
 
        err = ice_vsi_cfg_txq(vsi, tx_ring, qg_buf);
        if (err)
-               goto free_buf;
+               return err;
 
        if (ice_is_xdp_ena_vsi(vsi)) {
                struct ice_tx_ring *xdp_ring = vsi->xdp_rings[q_idx];
 
                memset(qg_buf, 0, size);
                qg_buf->num_txqs = 1;
                err = ice_vsi_cfg_txq(vsi, xdp_ring, qg_buf);
                if (err)
-                       goto free_buf;
+                       return err;
                ice_set_ring_xdp(xdp_ring);
                ice_tx_xsk_pool(vsi, q_idx);
        }
 
        err = ice_vsi_cfg_rxq(rx_ring);
        if (err)
-               goto free_buf;
+               return err;
 
        ice_qvec_cfg_msix(vsi, q_vector);
 
        err = ice_vsi_ctrl_one_rx_ring(vsi, true, q_idx, true);
        if (err)
-               goto free_buf;
+               return err;
 
        clear_bit(ICE_CFG_BUSY, vsi->state);
        ice_qvec_toggle_napi(vsi, q_vector, true);
        ice_qvec_ena_irq(vsi, q_vector);
 
        netif_tx_start_queue(netdev_get_tx_queue(vsi->netdev, q_idx));
-free_buf:
-       kfree(qg_buf);
-       return err;
+
+       return 0;
 }
 
 /**
-- 
2.40.1