Jiri Pirko | 753f993 | 2013-03-06 01:31:13 +0000 | [diff] [blame] | 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> |
Jiri Pirko | 753f993 | 2013-03-06 01:31:13 +0000 | [diff] [blame] | 16 | #include <linux/if_team.h> |
| 17 | |
Jiri Pirko | 753f993 | 2013-03-06 01:31:13 +0000 | [diff] [blame] | 18 | static bool rnd_transmit(struct team *team, struct sk_buff *skb) |
| 19 | { |
| 20 | struct team_port *port; |
| 21 | int port_index; |
| 22 | |
Daniel Borkmann | f337db6 | 2014-01-22 02:29:39 +0100 | [diff] [blame] | 23 | port_index = prandom_u32_max(team->en_port_count); |
Jiri Pirko | 753f993 | 2013-03-06 01:31:13 +0000 | [diff] [blame] | 24 | port = team_get_port_by_index_rcu(team, port_index); |
Jiri Pirko | 76c455d | 2013-06-08 15:00:53 +0200 | [diff] [blame] | 25 | if (unlikely(!port)) |
| 26 | goto drop; |
Jiri Pirko | 753f993 | 2013-03-06 01:31:13 +0000 | [diff] [blame] | 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, |
Jiri Pirko | 8fd7285 | 2015-12-03 12:12:13 +0100 | [diff] [blame] | 49 | .lag_tx_type = NETDEV_LAG_TX_TYPE_RANDOM, |
Jiri Pirko | 753f993 | 2013-03-06 01:31:13 +0000 | [diff] [blame] | 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"); |