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/reciprocal_div.h> 17 #include <linux/if_team.h> 18 19 static u32 random_N(unsigned int N) 20 { 21 return reciprocal_divide(prandom_u32(), N); 22 } 23 24 static bool rnd_transmit(struct team *team, struct sk_buff *skb) 25 { 26 struct team_port *port; 27 int port_index; 28 29 port_index = random_N(team->en_port_count); 30 port = team_get_port_by_index_rcu(team, port_index); 31 port = team_get_first_port_txable_rcu(team, port); 32 if (unlikely(!port)) 33 goto drop; 34 if (team_dev_queue_xmit(team, port, skb)) 35 return false; 36 return true; 37 38 drop: 39 dev_kfree_skb_any(skb); 40 return false; 41 } 42 43 static const struct team_mode_ops rnd_mode_ops = { 44 .transmit = rnd_transmit, 45 .port_enter = team_modeop_port_enter, 46 .port_change_dev_addr = team_modeop_port_change_dev_addr, 47 }; 48 49 static const struct team_mode rnd_mode = { 50 .kind = "random", 51 .owner = THIS_MODULE, 52 .ops = &rnd_mode_ops, 53 }; 54 55 static int __init rnd_init_module(void) 56 { 57 return team_mode_register(&rnd_mode); 58 } 59 60 static void __exit rnd_cleanup_module(void) 61 { 62 team_mode_unregister(&rnd_mode); 63 } 64 65 module_init(rnd_init_module); 66 module_exit(rnd_cleanup_module); 67 68 MODULE_LICENSE("GPL v2"); 69 MODULE_AUTHOR("Jiri Pirko <jiri@resnulli.us>"); 70 MODULE_DESCRIPTION("Random mode for team"); 71 MODULE_ALIAS("team-mode-random"); 72