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 | num: 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(mode: &rr_mode); |
62 | } |
63 | |
64 | static void __exit rr_cleanup_module(void) |
65 | { |
66 | team_mode_unregister(mode: &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 | |