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> 16 #include <linux/if_team.h> 17 18 static bool rnd_transmit(struct team *team, struct sk_buff *skb) 19 { 20 struct team_port *port; 21 int port_index; 22 23 port_index = prandom_u32_max(team->en_port_count); 24 port = team_get_port_by_index_rcu(team, port_index); 25 if (unlikely(!port)) 26 goto drop; 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, 49 .lag_tx_type = NETDEV_LAG_TX_TYPE_RANDOM, 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"); 69