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