[PATCH V8 mlx5-next 05/16] bonding: Add helper function to get the xmit slave based on hash

[Maor Gottlieb]

Both xor and 802.3ad modes use bond_xmit_hash to get the xmit slave.
Export the logic to helper function so it could be used in the
following patches by the .ndo to get the xmit slave.

Signed-off-by: Maor Gottlieb <ma...@mellanox.com>
Reviewed-by: Jiri Pirko <j...@mellanox.com>
Reviewed-by: Jay Vosburgh <jay.vosbu...@canonical.com>
---
 drivers/net/bonding/bond_main.c | 28 +++++++++++++++++++++-------
 1 file changed, 21 insertions(+), 7 deletions(-)

diff --git a/drivers/net/bonding/bond_main.c b/drivers/net/bonding/bond_main.c
index 2cb41d480ae2..8e6305955c75 100644
--- a/drivers/net/bonding/bond_main.c
+++ b/drivers/net/bonding/bond_main.c
@@ -4185,6 +4185,23 @@ int bond_update_slave_arr(struct bonding *bond, struct 
slave *skipslave)
        return ret;
 }
 
+static struct slave *bond_xmit_3ad_xor_slave_get(struct bonding *bond,
+                                                struct sk_buff *skb,
+                                                struct bond_up_slave *slaves)
+{
+       struct slave *slave;
+       unsigned int count;
+       u32 hash;
+
+       hash = bond_xmit_hash(bond, skb);
+       count = slaves ? READ_ONCE(slaves->count) : 0;
+       if (unlikely(!count))
+               return NULL;
+
+       slave = slaves->arr[hash % count];
+       return slave;
+}
+
 /* Use this Xmit function for 3AD as well as XOR modes. The current
  * usable slave array is formed in the control path. The xmit function
  * just calculates hash and sends the packet out.
@@ -4193,18 +4210,15 @@ static netdev_tx_t bond_3ad_xor_xmit(struct sk_buff 
*skb,
                                     struct net_device *dev)
 {
        struct bonding *bond = netdev_priv(dev);
-       struct slave *slave;
        struct bond_up_slave *slaves;
-       unsigned int count;
+       struct slave *slave;
 
        slaves = rcu_dereference(bond->usable_slaves);
-       count = slaves ? READ_ONCE(slaves->count) : 0;
-       if (likely(count)) {
-               slave = slaves->arr[bond_xmit_hash(bond, skb) % count];
+       slave = bond_xmit_3ad_xor_slave_get(bond, skb, slaves);
+       if (likely(slave))
                bond_dev_queue_xmit(bond, skb, slave->dev);
-       } else {
+       else
                bond_tx_drop(dev, skb);
-       }
 
        return NETDEV_TX_OK;
 }
-- 
2.17.2