1 // SPDX-License-Identifier: GPL-2.0-or-later 2 /* 3 * drivers/net/team/team_mode_roundrobin.c - Round-robin mode for team 4 * Copyright (c) 2011 Jiri Pirko <jpirko@redhat.com> 5 */ 6 7 #include <linux/kernel.h> 8 #include <linux/types.h> 9 #include <linux/module.h> 10 #include <linux/init.h> 11 #include <linux/netdevice.h> 12 #include <linux/if_team.h> 13 14 struct rr_priv { 15 unsigned int sent_packets; 16 }; 17 18 static struct rr_priv *rr_priv(struct team *team) 19 { 20 return (struct rr_priv *) &team->mode_priv; 21 } 22 23 static bool rr_transmit(struct team *team, struct sk_buff *skb) 24 { 25 struct team_port *port; 26 int port_index; 27 28 port_index = team_num_to_port_index(team, 29 rr_priv(team)->sent_packets++); 30 port = team_get_port_by_index_rcu(team, port_index); 31 if (unlikely(!port)) 32 goto drop; 33 port = team_get_first_port_txable_rcu(team, port); 34 if (unlikely(!port)) 35 goto drop; 36 if (team_dev_queue_xmit(team, port, skb)) 37 return false; 38 return true; 39 40 drop: 41 dev_kfree_skb_any(skb); 42 return false; 43 } 44 45 static const struct team_mode_ops rr_mode_ops = { 46 .transmit = rr_transmit, 47 .port_enter = team_modeop_port_enter, 48 .port_change_dev_addr = team_modeop_port_change_dev_addr, 49 }; 50 51 static const struct team_mode rr_mode = { 52 .kind = "roundrobin", 53 .owner = THIS_MODULE, 54 .priv_size = sizeof(struct rr_priv), 55 .ops = &rr_mode_ops, 56 .lag_tx_type = NETDEV_LAG_TX_TYPE_ROUNDROBIN, 57 }; 58 59 static int __init rr_init_module(void) 60 { 61 return team_mode_register(&rr_mode); 62 } 63 64 static void __exit rr_cleanup_module(void) 65 { 66 team_mode_unregister(&rr_mode); 67 } 68 69 module_init(rr_init_module); 70 module_exit(rr_cleanup_module); 71 72 MODULE_LICENSE("GPL v2"); 73 MODULE_AUTHOR("Jiri Pirko <jpirko@redhat.com>"); 74 MODULE_DESCRIPTION("Round-robin mode for team"); 75 MODULE_ALIAS_TEAM_MODE("roundrobin"); 76