1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0-or-later
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun * drivers/net/team/team_mode_random.c - Random mode for team
4*4882a593Smuzhiyun * Copyright (c) 2013 Jiri Pirko <jiri@resnulli.us>
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/skbuff.h>
12*4882a593Smuzhiyun #include <linux/if_team.h>
13*4882a593Smuzhiyun
rnd_transmit(struct team * team,struct sk_buff * skb)14*4882a593Smuzhiyun static bool rnd_transmit(struct team *team, struct sk_buff *skb)
15*4882a593Smuzhiyun {
16*4882a593Smuzhiyun struct team_port *port;
17*4882a593Smuzhiyun int port_index;
18*4882a593Smuzhiyun
19*4882a593Smuzhiyun port_index = prandom_u32_max(team->en_port_count);
20*4882a593Smuzhiyun port = team_get_port_by_index_rcu(team, port_index);
21*4882a593Smuzhiyun if (unlikely(!port))
22*4882a593Smuzhiyun goto drop;
23*4882a593Smuzhiyun port = team_get_first_port_txable_rcu(team, port);
24*4882a593Smuzhiyun if (unlikely(!port))
25*4882a593Smuzhiyun goto drop;
26*4882a593Smuzhiyun if (team_dev_queue_xmit(team, port, skb))
27*4882a593Smuzhiyun return false;
28*4882a593Smuzhiyun return true;
29*4882a593Smuzhiyun
30*4882a593Smuzhiyun drop:
31*4882a593Smuzhiyun dev_kfree_skb_any(skb);
32*4882a593Smuzhiyun return false;
33*4882a593Smuzhiyun }
34*4882a593Smuzhiyun
35*4882a593Smuzhiyun static const struct team_mode_ops rnd_mode_ops = {
36*4882a593Smuzhiyun .transmit = rnd_transmit,
37*4882a593Smuzhiyun .port_enter = team_modeop_port_enter,
38*4882a593Smuzhiyun .port_change_dev_addr = team_modeop_port_change_dev_addr,
39*4882a593Smuzhiyun };
40*4882a593Smuzhiyun
41*4882a593Smuzhiyun static const struct team_mode rnd_mode = {
42*4882a593Smuzhiyun .kind = "random",
43*4882a593Smuzhiyun .owner = THIS_MODULE,
44*4882a593Smuzhiyun .ops = &rnd_mode_ops,
45*4882a593Smuzhiyun .lag_tx_type = NETDEV_LAG_TX_TYPE_RANDOM,
46*4882a593Smuzhiyun };
47*4882a593Smuzhiyun
rnd_init_module(void)48*4882a593Smuzhiyun static int __init rnd_init_module(void)
49*4882a593Smuzhiyun {
50*4882a593Smuzhiyun return team_mode_register(&rnd_mode);
51*4882a593Smuzhiyun }
52*4882a593Smuzhiyun
rnd_cleanup_module(void)53*4882a593Smuzhiyun static void __exit rnd_cleanup_module(void)
54*4882a593Smuzhiyun {
55*4882a593Smuzhiyun team_mode_unregister(&rnd_mode);
56*4882a593Smuzhiyun }
57*4882a593Smuzhiyun
58*4882a593Smuzhiyun module_init(rnd_init_module);
59*4882a593Smuzhiyun module_exit(rnd_cleanup_module);
60*4882a593Smuzhiyun
61*4882a593Smuzhiyun MODULE_LICENSE("GPL v2");
62*4882a593Smuzhiyun MODULE_AUTHOR("Jiri Pirko <jiri@resnulli.us>");
63*4882a593Smuzhiyun MODULE_DESCRIPTION("Random mode for team");
64*4882a593Smuzhiyun MODULE_ALIAS_TEAM_MODE("random");
65