Commit | Line | Data |
---|---|---|
753f9939 JP |
1 | /* |
2 | * drivers/net/team/team_mode_random.c - Random mode for team | |
3 | * Copyright (c) 2013 Jiri Pirko <jiri@resnulli.us> | |
4 | * | |
5 | * This program is free software; you can redistribute it and/or modify | |
6 | * it under the terms of the GNU General Public License as published by | |
7 | * the Free Software Foundation; either version 2 of the License, or | |
8 | * (at your option) any later version. | |
9 | */ | |
10 | ||
11 | #include <linux/kernel.h> | |
12 | #include <linux/types.h> | |
13 | #include <linux/module.h> | |
14 | #include <linux/init.h> | |
15 | #include <linux/skbuff.h> | |
753f9939 JP |
16 | #include <linux/if_team.h> |
17 | ||
753f9939 JP |
18 | static bool rnd_transmit(struct team *team, struct sk_buff *skb) |
19 | { | |
20 | struct team_port *port; | |
21 | int port_index; | |
22 | ||
f337db64 | 23 | port_index = prandom_u32_max(team->en_port_count); |
753f9939 | 24 | port = team_get_port_by_index_rcu(team, port_index); |
76c455de JP |
25 | if (unlikely(!port)) |
26 | goto drop; | |
753f9939 JP |
27 | port = team_get_first_port_txable_rcu(team, port); |
28 | if (unlikely(!port)) | |
29 | goto drop; | |
30 | if (team_dev_queue_xmit(team, port, skb)) | |
31 | return false; | |
32 | return true; | |
33 | ||
34 | drop: | |
35 | dev_kfree_skb_any(skb); | |
36 | return false; | |
37 | } | |
38 | ||
39 | static const struct team_mode_ops rnd_mode_ops = { | |
40 | .transmit = rnd_transmit, | |
41 | .port_enter = team_modeop_port_enter, | |
42 | .port_change_dev_addr = team_modeop_port_change_dev_addr, | |
43 | }; | |
44 | ||
45 | static const struct team_mode rnd_mode = { | |
46 | .kind = "random", | |
47 | .owner = THIS_MODULE, | |
48 | .ops = &rnd_mode_ops, | |
8fd72856 | 49 | .lag_tx_type = NETDEV_LAG_TX_TYPE_RANDOM, |
753f9939 JP |
50 | }; |
51 | ||
52 | static int __init rnd_init_module(void) | |
53 | { | |
54 | return team_mode_register(&rnd_mode); | |
55 | } | |
56 | ||
57 | static void __exit rnd_cleanup_module(void) | |
58 | { | |
59 | team_mode_unregister(&rnd_mode); | |
60 | } | |
61 | ||
62 | module_init(rnd_init_module); | |
63 | module_exit(rnd_cleanup_module); | |
64 | ||
65 | MODULE_LICENSE("GPL v2"); | |
66 | MODULE_AUTHOR("Jiri Pirko <jiri@resnulli.us>"); | |
67 | MODULE_DESCRIPTION("Random mode for team"); | |
68 | MODULE_ALIAS("team-mode-random"); |