drivers/net/team/team.c

   1 /*
   2  * drivers/net/team/team.c - Network team device driver
   3  * Copyright (c) 2011 Jiri Pirko <jpirko@redhat.com>
   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/slab.h>
  16 #include <linux/rcupdate.h>
  17 #include <linux/errno.h>
  18 #include <linux/ctype.h>
  19 #include <linux/notifier.h>
  20 #include <linux/netdevice.h>
  21 #include <linux/netpoll.h>
  22 #include <linux/if_vlan.h>
  23 #include <linux/if_arp.h>
  24 #include <linux/socket.h>
  25 #include <linux/etherdevice.h>
  26 #include <linux/rtnetlink.h>
  27 #include <net/rtnetlink.h>
  28 #include <net/genetlink.h>
  29 #include <net/netlink.h>
  30 #include <net/sch_generic.h>
  31 #include <generated/utsrelease.h>
  32 #include <linux/if_team.h>
  33
  34 #define DRV_NAME "team"
  35
  36
  37 /**********
  38  * Helpers
  39  **********/
  40
  41 #define team_port_exists(dev) (dev->priv_flags & IFF_TEAM_PORT)
  42
  43 static struct team_port *team_port_get_rcu(const struct net_device *dev)
  44 {
  45         struct team_port *port = rcu_dereference(dev->rx_handler_data);
  46
  47         return team_port_exists(dev) ? port : NULL;
  48 }
  49
  50 static struct team_port *team_port_get_rtnl(const struct net_device *dev)
  51 {
  52         struct team_port *port = rtnl_dereference(dev->rx_handler_data);
  53
  54         return team_port_exists(dev) ? port : NULL;
  55 }
  56
  57 /*
  58  * Since the ability to change device address for open port device is tested in
  59  * team_port_add, this function can be called without control of return value
  60  */
  61 static int __set_port_dev_addr(struct net_device *port_dev,
  62                                const unsigned char *dev_addr)
  63 {
  64         struct sockaddr addr;
  65
  66         memcpy(addr.sa_data, dev_addr, port_dev->addr_len);
  67         addr.sa_family = port_dev->type;
  68         return dev_set_mac_address(port_dev, &addr);
  69 }
  70
  71 static int team_port_set_orig_dev_addr(struct team_port *port)
  72 {
  73         return __set_port_dev_addr(port->dev, port->orig.dev_addr);
  74 }
  75
  76 static int team_port_set_team_dev_addr(struct team *team,
  77                                        struct team_port *port)
  78 {
  79         return __set_port_dev_addr(port->dev, team->dev->dev_addr);
  80 }
  81
  82 int team_modeop_port_enter(struct team *team, struct team_port *port)
  83 {
  84         return team_port_set_team_dev_addr(team, port);
  85 }
  86 EXPORT_SYMBOL(team_modeop_port_enter);
  87
  88 void team_modeop_port_change_dev_addr(struct team *team,
  89                                       struct team_port *port)
  90 {
  91         team_port_set_team_dev_addr(team, port);
  92 }
  93 EXPORT_SYMBOL(team_modeop_port_change_dev_addr);
  94
  95 static void team_refresh_port_linkup(struct team_port *port)
  96 {
  97         port->linkup = port->user.linkup_enabled ? port->user.linkup :
  98                                                    port->state.linkup;
  99 }
 100
 101
 102 /*******************
 103  * Options handling
 104  *******************/
 105
 106 struct team_option_inst { /* One for each option instance */
 107         struct list_head list;
 108         struct list_head tmp_list;
 109         struct team_option *option;
 110         struct team_option_inst_info info;
 111         bool changed;
 112         bool removed;
 113 };
 114
 115 static struct team_option *__team_find_option(struct team *team,
 116                                               const char *opt_name)
 117 {
 118         struct team_option *option;
 119
 120         list_for_each_entry(option, &team->option_list, list) {
 121                 if (strcmp(option->name, opt_name) == 0)
 122                         return option;
 123         }
 124         return NULL;
 125 }
 126
 127 static void __team_option_inst_del(struct team_option_inst *opt_inst)
 128 {
 129         list_del(&opt_inst->list);
 130         kfree(opt_inst);
 131 }
 132
 133 static void __team_option_inst_del_option(struct team *team,
 134                                           struct team_option *option)
 135 {
 136         struct team_option_inst *opt_inst, *tmp;
 137
 138         list_for_each_entry_safe(opt_inst, tmp, &team->option_inst_list, list) {
 139                 if (opt_inst->option == option)
 140                         __team_option_inst_del(opt_inst);
 141         }
 142 }
 143
 144 static int __team_option_inst_add(struct team *team, struct team_option *option,
 145                                   struct team_port *port)
 146 {
 147         struct team_option_inst *opt_inst;
 148         unsigned int array_size;
 149         unsigned int i;
 150         int err;
 151
 152         array_size = option->array_size;
 153         if (!array_size)
 154                 array_size = 1; /* No array but still need one instance */
 155
 156         for (i = 0; i < array_size; i++) {
 157                 opt_inst = kmalloc(sizeof(*opt_inst), GFP_KERNEL);
 158                 if (!opt_inst)
 159                         return -ENOMEM;
 160                 opt_inst->option = option;
 161                 opt_inst->info.port = port;
 162                 opt_inst->info.array_index = i;
 163                 opt_inst->changed = true;
 164                 opt_inst->removed = false;
 165                 list_add_tail(&opt_inst->list, &team->option_inst_list);
 166                 if (option->init) {
 167                         err = option->init(team, &opt_inst->info);
 168                         if (err)
 169                                 return err;
 170                 }
 171
 172         }
 173         return 0;
 174 }
 175
 176 static int __team_option_inst_add_option(struct team *team,
 177                                          struct team_option *option)
 178 {
 179         struct team_port *port;
 180         int err;
 181
 182         if (!option->per_port) {
 183                 err = __team_option_inst_add(team, option, NULL);
 184                 if (err)
 185                         goto inst_del_option;
 186         }
 187
 188         list_for_each_entry(port, &team->port_list, list) {
 189                 err = __team_option_inst_add(team, option, port);
 190                 if (err)
 191                         goto inst_del_option;
 192         }
 193         return 0;
 194
 195 inst_del_option:
 196         __team_option_inst_del_option(team, option);
 197         return err;
 198 }
 199
 200 static void __team_option_inst_mark_removed_option(struct team *team,
 201                                                    struct team_option *option)
 202 {
 203         struct team_option_inst *opt_inst;
 204
 205         list_for_each_entry(opt_inst, &team->option_inst_list, list) {
 206                 if (opt_inst->option == option) {
 207                         opt_inst->changed = true;
 208                         opt_inst->removed = true;
 209                 }
 210         }
 211 }
 212
 213 static void __team_option_inst_del_port(struct team *team,
 214                                         struct team_port *port)
 215 {
 216         struct team_option_inst *opt_inst, *tmp;
 217
 218         list_for_each_entry_safe(opt_inst, tmp, &team->option_inst_list, list) {
 219                 if (opt_inst->option->per_port &&
 220                     opt_inst->info.port == port)
 221                         __team_option_inst_del(opt_inst);
 222         }
 223 }
 224
 225 static int __team_option_inst_add_port(struct team *team,
 226                                        struct team_port *port)
 227 {
 228         struct team_option *option;
 229         int err;
 230
 231         list_for_each_entry(option, &team->option_list, list) {
 232                 if (!option->per_port)
 233                         continue;
 234                 err = __team_option_inst_add(team, option, port);
 235                 if (err)
 236                         goto inst_del_port;
 237         }
 238         return 0;
 239
 240 inst_del_port:
 241         __team_option_inst_del_port(team, port);
 242         return err;
 243 }
 244
 245 static void __team_option_inst_mark_removed_port(struct team *team,
 246                                                  struct team_port *port)
 247 {
 248         struct team_option_inst *opt_inst;
 249
 250         list_for_each_entry(opt_inst, &team->option_inst_list, list) {
 251                 if (opt_inst->info.port == port) {
 252                         opt_inst->changed = true;
 253                         opt_inst->removed = true;
 254                 }
 255         }
 256 }
 257
 258 static int __team_options_register(struct team *team,
 259                                    const struct team_option *option,
 260                                    size_t option_count)
 261 {
 262         int i;
 263         struct team_option **dst_opts;
 264         int err;
 265
 266         dst_opts = kzalloc(sizeof(struct team_option *) * option_count,
 267                            GFP_KERNEL);
 268         if (!dst_opts)
 269                 return -ENOMEM;
 270         for (i = 0; i < option_count; i++, option++) {
 271                 if (__team_find_option(team, option->name)) {
 272                         err = -EEXIST;
 273                         goto alloc_rollback;
 274                 }
 275                 dst_opts[i] = kmemdup(option, sizeof(*option), GFP_KERNEL);
 276                 if (!dst_opts[i]) {
 277                         err = -ENOMEM;
 278                         goto alloc_rollback;
 279                 }
 280         }
 281
 282         for (i = 0; i < option_count; i++) {
 283                 err = __team_option_inst_add_option(team, dst_opts[i]);
 284                 if (err)
 285                         goto inst_rollback;
 286                 list_add_tail(&dst_opts[i]->list, &team->option_list);
 287         }
 288
 289         kfree(dst_opts);
 290         return 0;
 291
 292 inst_rollback:
 293         for (i--; i >= 0; i--)
 294                 __team_option_inst_del_option(team, dst_opts[i]);
 295
 296         i = option_count - 1;
 297 alloc_rollback:
 298         for (i--; i >= 0; i--)
 299                 kfree(dst_opts[i]);
 300
 301         kfree(dst_opts);
 302         return err;
 303 }
 304
 305 static void __team_options_mark_removed(struct team *team,
 306                                         const struct team_option *option,
 307                                         size_t option_count)
 308 {
 309         int i;
 310
 311         for (i = 0; i < option_count; i++, option++) {
 312                 struct team_option *del_opt;
 313
 314                 del_opt = __team_find_option(team, option->name);
 315                 if (del_opt)
 316                         __team_option_inst_mark_removed_option(team, del_opt);
 317         }
 318 }
 319
 320 static void __team_options_unregister(struct team *team,
 321                                       const struct team_option *option,
 322                                       size_t option_count)
 323 {
 324         int i;
 325
 326         for (i = 0; i < option_count; i++, option++) {
 327                 struct team_option *del_opt;
 328
 329                 del_opt = __team_find_option(team, option->name);
 330                 if (del_opt) {
 331                         __team_option_inst_del_option(team, del_opt);
 332                         list_del(&del_opt->list);
 333                         kfree(del_opt);
 334                 }
 335         }
 336 }
 337
 338 static void __team_options_change_check(struct team *team);
 339
 340 int team_options_register(struct team *team,
 341                           const struct team_option *option,
 342                           size_t option_count)
 343 {
 344         int err;
 345
 346         err = __team_options_register(team, option, option_count);
 347         if (err)
 348                 return err;
 349         __team_options_change_check(team);
 350         return 0;
 351 }
 352 EXPORT_SYMBOL(team_options_register);
 353
 354 void team_options_unregister(struct team *team,
 355                              const struct team_option *option,
 356                              size_t option_count)
 357 {
 358         __team_options_mark_removed(team, option, option_count);
 359         __team_options_change_check(team);
 360         __team_options_unregister(team, option, option_count);
 361 }
 362 EXPORT_SYMBOL(team_options_unregister);
 363
 364 static int team_option_get(struct team *team,
 365                            struct team_option_inst *opt_inst,
 366                            struct team_gsetter_ctx *ctx)
 367 {
 368         if (!opt_inst->option->getter)
 369                 return -EOPNOTSUPP;
 370         return opt_inst->option->getter(team, ctx);
 371 }
 372
 373 static int team_option_set(struct team *team,
 374                            struct team_option_inst *opt_inst,
 375                            struct team_gsetter_ctx *ctx)
 376 {
 377         if (!opt_inst->option->setter)
 378                 return -EOPNOTSUPP;
 379         return opt_inst->option->setter(team, ctx);
 380 }
 381
 382 void team_option_inst_set_change(struct team_option_inst_info *opt_inst_info)
 383 {
 384         struct team_option_inst *opt_inst;
 385
 386         opt_inst = container_of(opt_inst_info, struct team_option_inst, info);
 387         opt_inst->changed = true;
 388 }
 389 EXPORT_SYMBOL(team_option_inst_set_change);
 390
 391 void team_options_change_check(struct team *team)
 392 {
 393         __team_options_change_check(team);
 394 }
 395 EXPORT_SYMBOL(team_options_change_check);
 396
 397
 398 /****************
 399  * Mode handling
 400  ****************/
 401
 402 static LIST_HEAD(mode_list);
 403 static DEFINE_SPINLOCK(mode_list_lock);
 404
 405 struct team_mode_item {
 406         struct list_head list;
 407         const struct team_mode *mode;
 408 };
 409
 410 static struct team_mode_item *__find_mode(const char *kind)
 411 {
 412         struct team_mode_item *mitem;
 413
 414         list_for_each_entry(mitem, &mode_list, list) {
 415                 if (strcmp(mitem->mode->kind, kind) == 0)
 416                         return mitem;
 417         }
 418         return NULL;
 419 }
 420
 421 static bool is_good_mode_name(const char *name)
 422 {
 423         while (*name != '\0') {
 424                 if (!isalpha(*name) && !isdigit(*name) && *name != '_')
 425                         return false;
 426                 name++;
 427         }
 428         return true;
 429 }
 430
 431 int team_mode_register(const struct team_mode *mode)
 432 {
 433         int err = 0;
 434         struct team_mode_item *mitem;
 435
 436         if (!is_good_mode_name(mode->kind) ||
 437             mode->priv_size > TEAM_MODE_PRIV_SIZE)
 438                 return -EINVAL;
 439
 440         mitem = kmalloc(sizeof(*mitem), GFP_KERNEL);
 441         if (!mitem)
 442                 return -ENOMEM;
 443
 444         spin_lock(&mode_list_lock);
 445         if (__find_mode(mode->kind)) {
 446                 err = -EEXIST;
 447                 kfree(mitem);
 448                 goto unlock;
 449         }
 450         mitem->mode = mode;
 451         list_add_tail(&mitem->list, &mode_list);
 452 unlock:
 453         spin_unlock(&mode_list_lock);
 454         return err;
 455 }
 456 EXPORT_SYMBOL(team_mode_register);
 457
 458 void team_mode_unregister(const struct team_mode *mode)
 459 {
 460         struct team_mode_item *mitem;
 461
 462         spin_lock(&mode_list_lock);
 463         mitem = __find_mode(mode->kind);
 464         if (mitem) {
 465                 list_del_init(&mitem->list);
 466                 kfree(mitem);
 467         }
 468         spin_unlock(&mode_list_lock);
 469 }
 470 EXPORT_SYMBOL(team_mode_unregister);
 471
 472 static const struct team_mode *team_mode_get(const char *kind)
 473 {
 474         struct team_mode_item *mitem;
 475         const struct team_mode *mode = NULL;
 476
 477         spin_lock(&mode_list_lock);
 478         mitem = __find_mode(kind);
 479         if (!mitem) {
 480                 spin_unlock(&mode_list_lock);
 481                 request_module("team-mode-%s", kind);
 482                 spin_lock(&mode_list_lock);
 483                 mitem = __find_mode(kind);
 484         }
 485         if (mitem) {
 486                 mode = mitem->mode;
 487                 if (!try_module_get(mode->owner))
 488                         mode = NULL;
 489         }
 490
 491         spin_unlock(&mode_list_lock);
 492         return mode;
 493 }
 494
 495 static void team_mode_put(const struct team_mode *mode)
 496 {
 497         module_put(mode->owner);
 498 }
 499
 500 static bool team_dummy_transmit(struct team *team, struct sk_buff *skb)
 501 {
 502         dev_kfree_skb_any(skb);
 503         return false;
 504 }
 505
 506 static rx_handler_result_t team_dummy_receive(struct team *team,
 507                                               struct team_port *port,
 508                                               struct sk_buff *skb)
 509 {
 510         return RX_HANDLER_ANOTHER;
 511 }
 512
 513 static const struct team_mode __team_no_mode = {
 514         .kind           = "*NOMODE*",
 515 };
 516
 517 static bool team_is_mode_set(struct team *team)
 518 {
 519         return team->mode != &__team_no_mode;
 520 }
 521
 522 static void team_set_no_mode(struct team *team)
 523 {
 524         team->user_carrier_enabled = false;
 525         team->mode = &__team_no_mode;
 526 }
 527
 528 static void __team_adjust_ops(struct team *team, int en_port_count)
 529 {
 530         /*
 531          * To avoid checks in rx/tx skb paths, ensure here that non-null and
 532          * correct ops are always set.
 533          */
 534
 535         if (!en_port_count || !team_is_mode_set(team) ||
 536             !team->mode->ops->transmit)
 537                 team->ops.transmit = team_dummy_transmit;
 538         else
 539                 team->ops.transmit = team->mode->ops->transmit;
 540
 541         if (!en_port_count || !team_is_mode_set(team) ||
 542             !team->mode->ops->receive)
 543                 team->ops.receive = team_dummy_receive;
 544         else
 545                 team->ops.receive = team->mode->ops->receive;
 546 }
 547
 548 static void team_adjust_ops(struct team *team)
 549 {
 550         __team_adjust_ops(team, team->en_port_count);
 551 }
 552
 553 /*
 554  * We can benefit from the fact that it's ensured no port is present
 555  * at the time of mode change. Therefore no packets are in fly so there's no
 556  * need to set mode operations in any special way.
 557  */
 558 static int __team_change_mode(struct team *team,
 559                               const struct team_mode *new_mode)
 560 {
 561         /* Check if mode was previously set and do cleanup if so */
 562         if (team_is_mode_set(team)) {
 563                 void (*exit_op)(struct team *team) = team->ops.exit;
 564
 565                 /* Clear ops area so no callback is called any longer */
 566                 memset(&team->ops, 0, sizeof(struct team_mode_ops));
 567                 team_adjust_ops(team);
 568
 569                 if (exit_op)
 570                         exit_op(team);
 571                 team_mode_put(team->mode);
 572                 team_set_no_mode(team);
 573                 /* zero private data area */
 574                 memset(&team->mode_priv, 0,
 575                        sizeof(struct team) - offsetof(struct team, mode_priv));
 576         }
 577
 578         if (!new_mode)
 579                 return 0;
 580
 581         if (new_mode->ops->init) {
 582                 int err;
 583
 584                 err = new_mode->ops->init(team);
 585                 if (err)
 586                         return err;
 587         }
 588
 589         team->mode = new_mode;
 590         memcpy(&team->ops, new_mode->ops, sizeof(struct team_mode_ops));
 591         team_adjust_ops(team);
 592
 593         return 0;
 594 }
 595
 596 static int team_change_mode(struct team *team, const char *kind)
 597 {
 598         const struct team_mode *new_mode;
 599         struct net_device *dev = team->dev;
 600         int err;
 601
 602         if (!list_empty(&team->port_list)) {
 603                 netdev_err(dev, "No ports can be present during mode change\n");
 604                 return -EBUSY;
 605         }
 606
 607         if (team_is_mode_set(team) && strcmp(team->mode->kind, kind) == 0) {
 608                 netdev_err(dev, "Unable to change to the same mode the team is in\n");
 609                 return -EINVAL;
 610         }
 611
 612         new_mode = team_mode_get(kind);
 613         if (!new_mode) {
 614                 netdev_err(dev, "Mode \"%s\" not found\n", kind);
 615                 return -EINVAL;
 616         }
 617
 618         err = __team_change_mode(team, new_mode);
 619         if (err) {
 620                 netdev_err(dev, "Failed to change to mode \"%s\"\n", kind);
 621                 team_mode_put(new_mode);
 622                 return err;
 623         }
 624
 625         netdev_info(dev, "Mode changed to \"%s\"\n", kind);
 626         return 0;
 627 }
 628
 629
 630 /************************
 631  * Rx path frame handler
 632  ************************/
 633
 634 /* note: already called with rcu_read_lock */
 635 static rx_handler_result_t team_handle_frame(struct sk_buff **pskb)
 636 {
 637         struct sk_buff *skb = *pskb;
 638         struct team_port *port;
 639         struct team *team;
 640         rx_handler_result_t res;
 641
 642         skb = skb_share_check(skb, GFP_ATOMIC);
 643         if (!skb)
 644                 return RX_HANDLER_CONSUMED;
 645
 646         *pskb = skb;
 647
 648         port = team_port_get_rcu(skb->dev);
 649         team = port->team;
 650         if (!team_port_enabled(port)) {
 651                 /* allow exact match delivery for disabled ports */
 652                 res = RX_HANDLER_EXACT;
 653         } else {
 654                 res = team->ops.receive(team, port, skb);
 655         }
 656         if (res == RX_HANDLER_ANOTHER) {
 657                 struct team_pcpu_stats *pcpu_stats;
 658
 659                 pcpu_stats = this_cpu_ptr(team->pcpu_stats);
 660                 u64_stats_update_begin(&pcpu_stats->syncp);
 661                 pcpu_stats->rx_packets++;
 662                 pcpu_stats->rx_bytes += skb->len;
 663                 if (skb->pkt_type == PACKET_MULTICAST)
 664                         pcpu_stats->rx_multicast++;
 665                 u64_stats_update_end(&pcpu_stats->syncp);
 666
 667                 skb->dev = team->dev;
 668         } else {
 669                 this_cpu_inc(team->pcpu_stats->rx_dropped);
 670         }
 671
 672         return res;
 673 }
 674
 675
 676 /*************************************
 677  * Multiqueue Tx port select override
 678  *************************************/
 679
 680 static int team_queue_override_init(struct team *team)
 681 {
 682         struct list_head *listarr;
 683         unsigned int queue_cnt = team->dev->num_tx_queues - 1;
 684         unsigned int i;
 685
 686         if (!queue_cnt)
 687                 return 0;
 688         listarr = kmalloc(sizeof(struct list_head) * queue_cnt, GFP_KERNEL);
 689         if (!listarr)
 690                 return -ENOMEM;
 691         team->qom_lists = listarr;
 692         for (i = 0; i < queue_cnt; i++)
 693                 INIT_LIST_HEAD(listarr++);
 694         return 0;
 695 }
 696
 697 static void team_queue_override_fini(struct team *team)
 698 {
 699         kfree(team->qom_lists);
 700 }
 701
 702 static struct list_head *__team_get_qom_list(struct team *team, u16 queue_id)
 703 {
 704         return &team->qom_lists[queue_id - 1];
 705 }
 706
 707 /*
 708  * note: already called with rcu_read_lock
 709  */
 710 static bool team_queue_override_transmit(struct team *team, struct sk_buff *skb)
 711 {
 712         struct list_head *qom_list;
 713         struct team_port *port;
 714
 715         if (!team->queue_override_enabled || !skb->queue_mapping)
 716                 return false;
 717         qom_list = __team_get_qom_list(team, skb->queue_mapping);
 718         list_for_each_entry_rcu(port, qom_list, qom_list) {
 719                 if (!team_dev_queue_xmit(team, port, skb))
 720                         return true;
 721         }
 722         return false;
 723 }
 724
 725 static void __team_queue_override_port_del(struct team *team,
 726                                            struct team_port *port)
 727 {
 728         if (!port->queue_id)
 729                 return;
 730         list_del_rcu(&port->qom_list);
 731 }
 732
 733 static bool team_queue_override_port_has_gt_prio_than(struct team_port *port,
 734                                                       struct team_port *cur)
 735 {
 736         if (port->priority < cur->priority)
 737                 return true;
 738         if (port->priority > cur->priority)
 739                 return false;
 740         if (port->index < cur->index)
 741                 return true;
 742         return false;
 743 }
 744
 745 static void __team_queue_override_port_add(struct team *team,
 746                                            struct team_port *port)
 747 {
 748         struct team_port *cur;
 749         struct list_head *qom_list;
 750         struct list_head *node;
 751
 752         if (!port->queue_id)
 753                 return;
 754         qom_list = __team_get_qom_list(team, port->queue_id);
 755         node = qom_list;
 756         list_for_each_entry(cur, qom_list, qom_list) {
 757                 if (team_queue_override_port_has_gt_prio_than(port, cur))
 758                         break;
 759                 node = &cur->qom_list;
 760         }
 761         list_add_tail_rcu(&port->qom_list, node);
 762 }
 763
 764 static void __team_queue_override_enabled_check(struct team *team)
 765 {
 766         struct team_port *port;
 767         bool enabled = false;
 768
 769         list_for_each_entry(port, &team->port_list, list) {
 770                 if (port->queue_id) {
 771                         enabled = true;
 772                         break;
 773                 }
 774         }
 775         if (enabled == team->queue_override_enabled)
 776                 return;
 777         netdev_dbg(team->dev, "%s queue override\n",
 778                    enabled ? "Enabling" : "Disabling");
 779         team->queue_override_enabled = enabled;
 780 }
 781
 782 static void team_queue_override_port_prio_changed(struct team *team,
 783                                                   struct team_port *port)
 784 {
 785         if (!port->queue_id || team_port_enabled(port))
 786                 return;
 787         __team_queue_override_port_del(team, port);
 788         __team_queue_override_port_add(team, port);
 789         __team_queue_override_enabled_check(team);
 790 }
 791
 792 static void team_queue_override_port_change_queue_id(struct team *team,
 793                                                      struct team_port *port,
 794                                                      u16 new_queue_id)
 795 {
 796         if (team_port_enabled(port)) {
 797                 __team_queue_override_port_del(team, port);
 798                 port->queue_id = new_queue_id;
 799                 __team_queue_override_port_add(team, port);
 800                 __team_queue_override_enabled_check(team);
 801         } else {
 802                 port->queue_id = new_queue_id;
 803         }
 804 }
 805
 806 static void team_queue_override_port_add(struct team *team,
 807                                          struct team_port *port)
 808 {
 809         __team_queue_override_port_add(team, port);
 810         __team_queue_override_enabled_check(team);
 811 }
 812
 813 static void team_queue_override_port_del(struct team *team,
 814                                          struct team_port *port)
 815 {
 816         __team_queue_override_port_del(team, port);
 817         __team_queue_override_enabled_check(team);
 818 }
 819
 820
 821 /****************
 822  * Port handling
 823  ****************/
 824
 825 static bool team_port_find(const struct team *team,
 826                            const struct team_port *port)
 827 {
 828         struct team_port *cur;
 829
 830         list_for_each_entry(cur, &team->port_list, list)
 831                 if (cur == port)
 832                         return true;
 833         return false;
 834 }
 835
 836 /*
 837  * Enable/disable port by adding to enabled port hashlist and setting
 838  * port->index (Might be racy so reader could see incorrect ifindex when
 839  * processing a flying packet, but that is not a problem). Write guarded
 840  * by team->lock.
 841  */
 842 static void team_port_enable(struct team *team,
 843                              struct team_port *port)
 844 {
 845         if (team_port_enabled(port))
 846                 return;
 847         port->index = team->en_port_count++;
 848         hlist_add_head_rcu(&port->hlist,
 849                            team_port_index_hash(team, port->index));
 850         team_adjust_ops(team);
 851         team_queue_override_port_add(team, port);
 852         if (team->ops.port_enabled)
 853                 team->ops.port_enabled(team, port);
 854 }
 855
 856 static void __reconstruct_port_hlist(struct team *team, int rm_index)
 857 {
 858         int i;
 859         struct team_port *port;
 860
 861         for (i = rm_index + 1; i < team->en_port_count; i++) {
 862                 port = team_get_port_by_index(team, i);
 863                 hlist_del_rcu(&port->hlist);
 864                 port->index--;
 865                 hlist_add_head_rcu(&port->hlist,
 866                                    team_port_index_hash(team, port->index));
 867         }
 868 }
 869
 870 static void team_port_disable(struct team *team,
 871                               struct team_port *port)
 872 {
 873         if (!team_port_enabled(port))
 874                 return;
 875         if (team->ops.port_disabled)
 876                 team->ops.port_disabled(team, port);
 877         hlist_del_rcu(&port->hlist);
 878         __reconstruct_port_hlist(team, port->index);
 879         port->index = -1;
 880         team_queue_override_port_del(team, port);
 881         __team_adjust_ops(team, team->en_port_count - 1);
 882         /*
 883          * Wait until readers see adjusted ops. This ensures that
 884          * readers never see team->en_port_count == 0
 885          */
 886         synchronize_rcu();
 887         team->en_port_count--;
 888 }
 889
 890 #define TEAM_VLAN_FEATURES (NETIF_F_ALL_CSUM | NETIF_F_SG | \
 891                             NETIF_F_FRAGLIST | NETIF_F_ALL_TSO | \
 892                             NETIF_F_HIGHDMA | NETIF_F_LRO)
 893
 894 static void __team_compute_features(struct team *team)
 895 {
 896         struct team_port *port;
 897         u32 vlan_features = TEAM_VLAN_FEATURES;
 898         unsigned short max_hard_header_len = ETH_HLEN;
 899         unsigned int flags, dst_release_flag = IFF_XMIT_DST_RELEASE;
 900
 901         list_for_each_entry(port, &team->port_list, list) {
 902                 vlan_features = netdev_increment_features(vlan_features,
 903                                         port->dev->vlan_features,
 904                                         TEAM_VLAN_FEATURES);
 905
 906                 dst_release_flag &= port->dev->priv_flags;
 907                 if (port->dev->hard_header_len > max_hard_header_len)
 908                         max_hard_header_len = port->dev->hard_header_len;
 909         }
 910
 911         team->dev->vlan_features = vlan_features;
 912         team->dev->hard_header_len = max_hard_header_len;
 913
 914         flags = team->dev->priv_flags & ~IFF_XMIT_DST_RELEASE;
 915         team->dev->priv_flags = flags | dst_release_flag;
 916
 917         netdev_change_features(team->dev);
 918 }
 919
 920 static void team_compute_features(struct team *team)
 921 {
 922         mutex_lock(&team->lock);
 923         __team_compute_features(team);
 924         mutex_unlock(&team->lock);
 925 }
 926
 927 static int team_port_enter(struct team *team, struct team_port *port)
 928 {
 929         int err = 0;
 930
 931         dev_hold(team->dev);
 932         port->dev->priv_flags |= IFF_TEAM_PORT;
 933         if (team->ops.port_enter) {
 934                 err = team->ops.port_enter(team, port);
 935                 if (err) {
 936                         netdev_err(team->dev, "Device %s failed to enter team mode\n",
 937                                    port->dev->name);
 938                         goto err_port_enter;
 939                 }
 940         }
 941
 942         return 0;
 943
 944 err_port_enter:
 945         port->dev->priv_flags &= ~IFF_TEAM_PORT;
 946         dev_put(team->dev);
 947
 948         return err;
 949 }
 950
 951 static void team_port_leave(struct team *team, struct team_port *port)
 952 {
 953         if (team->ops.port_leave)
 954                 team->ops.port_leave(team, port);
 955         port->dev->priv_flags &= ~IFF_TEAM_PORT;
 956         dev_put(team->dev);
 957 }
 958
 959 #ifdef CONFIG_NET_POLL_CONTROLLER
 960 static int team_port_enable_netpoll(struct team *team, struct team_port *port,
 961                                     gfp_t gfp)
 962 {
 963         struct netpoll *np;
 964         int err;
 965
 966         np = kzalloc(sizeof(*np), gfp);
 967         if (!np)
 968                 return -ENOMEM;
 969
 970         err = __netpoll_setup(np, port->dev, gfp);
 971         if (err) {
 972                 kfree(np);
 973                 return err;
 974         }
 975         port->np = np;
 976         return err;
 977 }
 978
 979 static void team_port_disable_netpoll(struct team_port *port)
 980 {
 981         struct netpoll *np = port->np;
 982
 983         if (!np)
 984                 return;
 985         port->np = NULL;
 986
 987         /* Wait for transmitting packets to finish before freeing. */
 988         synchronize_rcu_bh();
 989         __netpoll_cleanup(np);
 990         kfree(np);
 991 }
 992
 993 static struct netpoll_info *team_netpoll_info(struct team *team)
 994 {
 995         return team->dev->npinfo;
 996 }
 997
 998 #else
 999 static int team_port_enable_netpoll(struct team *team, struct team_port *port,
1000                                     gfp_t gfp)
1001 {
1002         return 0;
1003 }
1004 static void team_port_disable_netpoll(struct team_port *port)
1005 {
1006 }
1007 static struct netpoll_info *team_netpoll_info(struct team *team)
1008 {
1009         return NULL;
1010 }
1011 #endif
1012
1013 static void __team_port_change_port_added(struct team_port *port, bool linkup);
1014 static int team_dev_type_check_change(struct net_device *dev,
1015                                       struct net_device *port_dev);
1016
1017 static int team_port_add(struct team *team, struct net_device *port_dev)
1018 {
1019         struct net_device *dev = team->dev;
1020         struct team_port *port;
1021         char *portname = port_dev->name;
1022         int err;
1023
1024         if (port_dev->flags & IFF_LOOPBACK) {
1025                 netdev_err(dev, "Device %s is loopback device. Loopback devices can't be added as a team port\n",
1026                            portname);
1027                 return -EINVAL;
1028         }
1029
1030         if (team_port_exists(port_dev)) {
1031                 netdev_err(dev, "Device %s is already a port "
1032                                 "of a team device\n", portname);
1033                 return -EBUSY;
1034         }
1035
1036         if (port_dev->features & NETIF_F_VLAN_CHALLENGED &&
1037             vlan_uses_dev(dev)) {
1038                 netdev_err(dev, "Device %s is VLAN challenged and team device has VLAN set up\n",
1039                            portname);
1040                 return -EPERM;
1041         }
1042
1043         err = team_dev_type_check_change(dev, port_dev);
1044         if (err)
1045                 return err;
1046
1047         if (port_dev->flags & IFF_UP) {
1048                 netdev_err(dev, "Device %s is up. Set it down before adding it as a team port\n",
1049                            portname);
1050                 return -EBUSY;
1051         }
1052
1053         port = kzalloc(sizeof(struct team_port) + team->mode->port_priv_size,
1054                        GFP_KERNEL);
1055         if (!port)
1056                 return -ENOMEM;
1057
1058         port->dev = port_dev;
1059         port->team = team;
1060         INIT_LIST_HEAD(&port->qom_list);
1061
1062         port->orig.mtu = port_dev->mtu;
1063         err = dev_set_mtu(port_dev, dev->mtu);
1064         if (err) {
1065                 netdev_dbg(dev, "Error %d calling dev_set_mtu\n", err);
1066                 goto err_set_mtu;
1067         }
1068
1069         memcpy(port->orig.dev_addr, port_dev->dev_addr, port_dev->addr_len);
1070
1071         err = team_port_enter(team, port);
1072         if (err) {
1073                 netdev_err(dev, "Device %s failed to enter team mode\n",
1074                            portname);
1075                 goto err_port_enter;
1076         }
1077
1078         err = dev_open(port_dev);
1079         if (err) {
1080                 netdev_dbg(dev, "Device %s opening failed\n",
1081                            portname);
1082                 goto err_dev_open;
1083         }
1084
1085         err = vlan_vids_add_by_dev(port_dev, dev);
1086         if (err) {
1087                 netdev_err(dev, "Failed to add vlan ids to device %s\n",
1088                                 portname);
1089                 goto err_vids_add;
1090         }
1091
1092         if (team_netpoll_info(team)) {
1093                 err = team_port_enable_netpoll(team, port, GFP_KERNEL);
1094                 if (err) {
1095                         netdev_err(dev, "Failed to enable netpoll on device %s\n",
1096                                    portname);
1097                         goto err_enable_netpoll;
1098                 }
1099         }
1100
1101         err = netdev_master_upper_dev_link(port_dev, dev);
1102         if (err) {
1103                 netdev_err(dev, "Device %s failed to set upper link\n",
1104                            portname);
1105                 goto err_set_upper_link;
1106         }
1107
1108         err = netdev_rx_handler_register(port_dev, team_handle_frame,
1109                                          port);
1110         if (err) {
1111                 netdev_err(dev, "Device %s failed to register rx_handler\n",
1112                            portname);
1113                 goto err_handler_register;
1114         }
1115
1116         err = __team_option_inst_add_port(team, port);
1117         if (err) {
1118                 netdev_err(dev, "Device %s failed to add per-port options\n",
1119                            portname);
1120                 goto err_option_port_add;
1121         }
1122
1123         port->index = -1;
1124         team_port_enable(team, port);
1125         list_add_tail_rcu(&port->list, &team->port_list);
1126         __team_compute_features(team);
1127         __team_port_change_port_added(port, !!netif_carrier_ok(port_dev));
1128         __team_options_change_check(team);
1129
1130         netdev_info(dev, "Port device %s added\n", portname);
1131
1132         return 0;
1133
1134 err_option_port_add:
1135         netdev_rx_handler_unregister(port_dev);
1136
1137 err_handler_register:
1138         netdev_upper_dev_unlink(port_dev, dev);
1139
1140 err_set_upper_link:
1141         team_port_disable_netpoll(port);
1142
1143 err_enable_netpoll:
1144         vlan_vids_del_by_dev(port_dev, dev);
1145
1146 err_vids_add:
1147         dev_close(port_dev);
1148
1149 err_dev_open:
1150         team_port_leave(team, port);
1151         team_port_set_orig_dev_addr(port);
1152
1153 err_port_enter:
1154         dev_set_mtu(port_dev, port->orig.mtu);
1155
1156 err_set_mtu:
1157         kfree(port);
1158
1159         return err;
1160 }
1161
1162 static void __team_port_change_port_removed(struct team_port *port);
1163
1164 static int team_port_del(struct team *team, struct net_device *port_dev)
1165 {
1166         struct net_device *dev = team->dev;
1167         struct team_port *port;
1168         char *portname = port_dev->name;
1169
1170         port = team_port_get_rtnl(port_dev);
1171         if (!port || !team_port_find(team, port)) {
1172                 netdev_err(dev, "Device %s does not act as a port of this team\n",
1173                            portname);
1174                 return -ENOENT;
1175         }
1176
1177         team_port_disable(team, port);
1178         list_del_rcu(&port->list);
1179         netdev_rx_handler_unregister(port_dev);
1180         netdev_upper_dev_unlink(port_dev, dev);
1181         team_port_disable_netpoll(port);
1182         vlan_vids_del_by_dev(port_dev, dev);
1183         dev_uc_unsync(port_dev, dev);
1184         dev_mc_unsync(port_dev, dev);
1185         dev_close(port_dev);
1186         team_port_leave(team, port);
1187
1188         __team_option_inst_mark_removed_port(team, port);
1189         __team_options_change_check(team);
1190         __team_option_inst_del_port(team, port);
1191         __team_port_change_port_removed(port);
1192
1193         team_port_set_orig_dev_addr(port);
1194         dev_set_mtu(port_dev, port->orig.mtu);
1195         synchronize_rcu();
1196         kfree(port);
1197         netdev_info(dev, "Port device %s removed\n", portname);
1198         __team_compute_features(team);
1199
1200         return 0;
1201 }
1202
1203
1204 /*****************
1205  * Net device ops
1206  *****************/
1207
1208 static int team_mode_option_get(struct team *team, struct team_gsetter_ctx *ctx)
1209 {
1210         ctx->data.str_val = team->mode->kind;
1211         return 0;
1212 }
1213
1214 static int team_mode_option_set(struct team *team, struct team_gsetter_ctx *ctx)
1215 {
1216         return team_change_mode(team, ctx->data.str_val);
1217 }
1218
1219 static int team_port_en_option_get(struct team *team,
1220                                    struct team_gsetter_ctx *ctx)
1221 {
1222         struct team_port *port = ctx->info->port;
1223
1224         ctx->data.bool_val = team_port_enabled(port);
1225         return 0;
1226 }
1227
1228 static int team_port_en_option_set(struct team *team,
1229                                    struct team_gsetter_ctx *ctx)
1230 {
1231         struct team_port *port = ctx->info->port;
1232
1233         if (ctx->data.bool_val)
1234                 team_port_enable(team, port);
1235         else
1236                 team_port_disable(team, port);
1237         return 0;
1238 }
1239
1240 static int team_user_linkup_option_get(struct team *team,
1241                                        struct team_gsetter_ctx *ctx)
1242 {
1243         struct team_port *port = ctx->info->port;
1244
1245         ctx->data.bool_val = port->user.linkup;
1246         return 0;
1247 }
1248
1249 static int team_user_linkup_option_set(struct team *team,
1250                                        struct team_gsetter_ctx *ctx)
1251 {
1252         struct team_port *port = ctx->info->port;
1253
1254         port->user.linkup = ctx->data.bool_val;
1255         team_refresh_port_linkup(port);
1256         return 0;
1257 }
1258
1259 static int team_user_linkup_en_option_get(struct team *team,
1260                                           struct team_gsetter_ctx *ctx)
1261 {
1262         struct team_port *port = ctx->info->port;
1263
1264         ctx->data.bool_val = port->user.linkup_enabled;
1265         return 0;
1266 }
1267
1268 static int team_user_linkup_en_option_set(struct team *team,
1269                                           struct team_gsetter_ctx *ctx)
1270 {
1271         struct team_port *port = ctx->info->port;
1272
1273         port->user.linkup_enabled = ctx->data.bool_val;
1274         team_refresh_port_linkup(port);
1275         return 0;
1276 }
1277
1278 static int team_priority_option_get(struct team *team,
1279                                     struct team_gsetter_ctx *ctx)
1280 {
1281         struct team_port *port = ctx->info->port;
1282
1283         ctx->data.s32_val = port->priority;
1284         return 0;
1285 }
1286
1287 static int team_priority_option_set(struct team *team,
1288                                     struct team_gsetter_ctx *ctx)
1289 {
1290         struct team_port *port = ctx->info->port;
1291         s32 priority = ctx->data.s32_val;
1292
1293         if (port->priority == priority)
1294                 return 0;
1295         port->priority = priority;
1296         team_queue_override_port_prio_changed(team, port);
1297         return 0;
1298 }
1299
1300 static int team_queue_id_option_get(struct team *team,
1301                                     struct team_gsetter_ctx *ctx)
1302 {
1303         struct team_port *port = ctx->info->port;
1304
1305         ctx->data.u32_val = port->queue_id;
1306         return 0;
1307 }
1308
1309 static int team_queue_id_option_set(struct team *team,
1310                                     struct team_gsetter_ctx *ctx)
1311 {
1312         struct team_port *port = ctx->info->port;
1313         u16 new_queue_id = ctx->data.u32_val;
1314
1315         if (port->queue_id == new_queue_id)
1316                 return 0;
1317         if (new_queue_id >= team->dev->real_num_tx_queues)
1318                 return -EINVAL;
1319         team_queue_override_port_change_queue_id(team, port, new_queue_id);
1320         return 0;
1321 }
1322
1323 static const struct team_option team_options[] = {
1324         {
1325                 .name = "mode",
1326                 .type = TEAM_OPTION_TYPE_STRING,
1327                 .getter = team_mode_option_get,
1328                 .setter = team_mode_option_set,
1329         },
1330         {
1331                 .name = "enabled",
1332                 .type = TEAM_OPTION_TYPE_BOOL,
1333                 .per_port = true,
1334                 .getter = team_port_en_option_get,
1335                 .setter = team_port_en_option_set,
1336         },
1337         {
1338                 .name = "user_linkup",
1339                 .type = TEAM_OPTION_TYPE_BOOL,
1340                 .per_port = true,
1341                 .getter = team_user_linkup_option_get,
1342                 .setter = team_user_linkup_option_set,
1343         },
1344         {
1345                 .name = "user_linkup_enabled",
1346                 .type = TEAM_OPTION_TYPE_BOOL,
1347                 .per_port = true,
1348                 .getter = team_user_linkup_en_option_get,
1349                 .setter = team_user_linkup_en_option_set,
1350         },
1351         {
1352                 .name = "priority",
1353                 .type = TEAM_OPTION_TYPE_S32,
1354                 .per_port = true,
1355                 .getter = team_priority_option_get,
1356                 .setter = team_priority_option_set,
1357         },
1358         {
1359                 .name = "queue_id",
1360                 .type = TEAM_OPTION_TYPE_U32,
1361                 .per_port = true,
1362                 .getter = team_queue_id_option_get,
1363                 .setter = team_queue_id_option_set,
1364         },
1365 };
1366
1367 static struct lock_class_key team_netdev_xmit_lock_key;
1368 static struct lock_class_key team_netdev_addr_lock_key;
1369 static struct lock_class_key team_tx_busylock_key;
1370
1371 static void team_set_lockdep_class_one(struct net_device *dev,
1372                                        struct netdev_queue *txq,
1373                                        void *unused)
1374 {
1375         lockdep_set_class(&txq->_xmit_lock, &team_netdev_xmit_lock_key);
1376 }
1377
1378 static void team_set_lockdep_class(struct net_device *dev)
1379 {
1380         lockdep_set_class(&dev->addr_list_lock, &team_netdev_addr_lock_key);
1381         netdev_for_each_tx_queue(dev, team_set_lockdep_class_one, NULL);
1382         dev->qdisc_tx_busylock = &team_tx_busylock_key;
1383 }
1384
1385 static int team_init(struct net_device *dev)
1386 {
1387         struct team *team = netdev_priv(dev);
1388         int i;
1389         int err;
1390
1391         team->dev = dev;
1392         mutex_init(&team->lock);
1393         team_set_no_mode(team);
1394
1395         team->pcpu_stats = alloc_percpu(struct team_pcpu_stats);
1396         if (!team->pcpu_stats)
1397                 return -ENOMEM;
1398
1399         for (i = 0; i < TEAM_PORT_HASHENTRIES; i++)
1400                 INIT_HLIST_HEAD(&team->en_port_hlist[i]);
1401         INIT_LIST_HEAD(&team->port_list);
1402         err = team_queue_override_init(team);
1403         if (err)
1404                 goto err_team_queue_override_init;
1405
1406         team_adjust_ops(team);
1407
1408         INIT_LIST_HEAD(&team->option_list);
1409         INIT_LIST_HEAD(&team->option_inst_list);
1410         err = team_options_register(team, team_options, ARRAY_SIZE(team_options));
1411         if (err)
1412                 goto err_options_register;
1413         netif_carrier_off(dev);
1414
1415         team_set_lockdep_class(dev);
1416
1417         return 0;
1418
1419 err_options_register:
1420         team_queue_override_fini(team);
1421 err_team_queue_override_init:
1422         free_percpu(team->pcpu_stats);
1423
1424         return err;
1425 }
1426
1427 static void team_uninit(struct net_device *dev)
1428 {
1429         struct team *team = netdev_priv(dev);
1430         struct team_port *port;
1431         struct team_port *tmp;
1432
1433         mutex_lock(&team->lock);
1434         list_for_each_entry_safe(port, tmp, &team->port_list, list)
1435                 team_port_del(team, port->dev);
1436
1437         __team_change_mode(team, NULL); /* cleanup */
1438         __team_options_unregister(team, team_options, ARRAY_SIZE(team_options));
1439         team_queue_override_fini(team);
1440         mutex_unlock(&team->lock);
1441 }
1442
1443 static void team_destructor(struct net_device *dev)
1444 {
1445         struct team *team = netdev_priv(dev);
1446
1447         free_percpu(team->pcpu_stats);
1448         free_netdev(dev);
1449 }
1450
1451 static int team_open(struct net_device *dev)
1452 {
1453         return 0;
1454 }
1455
1456 static int team_close(struct net_device *dev)
1457 {
1458         return 0;
1459 }
1460
1461 /*
1462  * note: already called with rcu_read_lock
1463  */
1464 static netdev_tx_t team_xmit(struct sk_buff *skb, struct net_device *dev)
1465 {
1466         struct team *team = netdev_priv(dev);
1467         bool tx_success;
1468         unsigned int len = skb->len;
1469
1470         tx_success = team_queue_override_transmit(team, skb);
1471         if (!tx_success)
1472                 tx_success = team->ops.transmit(team, skb);
1473         if (tx_success) {
1474                 struct team_pcpu_stats *pcpu_stats;
1475
1476                 pcpu_stats = this_cpu_ptr(team->pcpu_stats);
1477                 u64_stats_update_begin(&pcpu_stats->syncp);
1478                 pcpu_stats->tx_packets++;
1479                 pcpu_stats->tx_bytes += len;
1480                 u64_stats_update_end(&pcpu_stats->syncp);
1481         } else {
1482                 this_cpu_inc(team->pcpu_stats->tx_dropped);
1483         }
1484
1485         return NETDEV_TX_OK;
1486 }
1487
1488 static u16 team_select_queue(struct net_device *dev, struct sk_buff *skb)
1489 {
1490         /*
1491          * This helper function exists to help dev_pick_tx get the correct
1492          * destination queue.  Using a helper function skips a call to
1493          * skb_tx_hash and will put the skbs in the queue we expect on their
1494          * way down to the team driver.
1495          */
1496         u16 txq = skb_rx_queue_recorded(skb) ? skb_get_rx_queue(skb) : 0;
1497
1498         /*
1499          * Save the original txq to restore before passing to the driver
1500          */
1501         qdisc_skb_cb(skb)->slave_dev_queue_mapping = skb->queue_mapping;
1502
1503         if (unlikely(txq >= dev->real_num_tx_queues)) {
1504                 do {
1505                         txq -= dev->real_num_tx_queues;
1506                 } while (txq >= dev->real_num_tx_queues);
1507         }
1508         return txq;
1509 }
1510
1511 static void team_change_rx_flags(struct net_device *dev, int change)
1512 {
1513         struct team *team = netdev_priv(dev);
1514         struct team_port *port;
1515         int inc;
1516
1517         rcu_read_lock();
1518         list_for_each_entry_rcu(port, &team->port_list, list) {
1519                 if (change & IFF_PROMISC) {
1520                         inc = dev->flags & IFF_PROMISC ? 1 : -1;
1521                         dev_set_promiscuity(port->dev, inc);
1522                 }
1523                 if (change & IFF_ALLMULTI) {
1524                         inc = dev->flags & IFF_ALLMULTI ? 1 : -1;
1525                         dev_set_allmulti(port->dev, inc);
1526                 }
1527         }
1528         rcu_read_unlock();
1529 }
1530
1531 static void team_set_rx_mode(struct net_device *dev)
1532 {
1533         struct team *team = netdev_priv(dev);
1534         struct team_port *port;
1535
1536         rcu_read_lock();
1537         list_for_each_entry_rcu(port, &team->port_list, list) {
1538                 dev_uc_sync_multiple(port->dev, dev);
1539                 dev_mc_sync_multiple(port->dev, dev);
1540         }
1541         rcu_read_unlock();
1542 }
1543
1544 static int team_set_mac_address(struct net_device *dev, void *p)
1545 {
1546         struct sockaddr *addr = p;
1547         struct team *team = netdev_priv(dev);
1548         struct team_port *port;
1549
1550         if (dev->type == ARPHRD_ETHER && !is_valid_ether_addr(addr->sa_data))
1551                 return -EADDRNOTAVAIL;
1552         memcpy(dev->dev_addr, addr->sa_data, dev->addr_len);
1553         rcu_read_lock();
1554         list_for_each_entry_rcu(port, &team->port_list, list)
1555                 if (team->ops.port_change_dev_addr)
1556                         team->ops.port_change_dev_addr(team, port);
1557         rcu_read_unlock();
1558         return 0;
1559 }
1560
1561 static int team_change_mtu(struct net_device *dev, int new_mtu)
1562 {
1563         struct team *team = netdev_priv(dev);
1564         struct team_port *port;
1565         int err;
1566
1567         /*
1568          * Alhough this is reader, it's guarded by team lock. It's not possible
1569          * to traverse list in reverse under rcu_read_lock
1570          */
1571         mutex_lock(&team->lock);
1572         list_for_each_entry(port, &team->port_list, list) {
1573                 err = dev_set_mtu(port->dev, new_mtu);
1574                 if (err) {
1575                         netdev_err(dev, "Device %s failed to change mtu",
1576                                    port->dev->name);
1577                         goto unwind;
1578                 }
1579         }
1580         mutex_unlock(&team->lock);
1581
1582         dev->mtu = new_mtu;
1583
1584         return 0;
1585
1586 unwind:
1587         list_for_each_entry_continue_reverse(port, &team->port_list, list)
1588                 dev_set_mtu(port->dev, dev->mtu);
1589         mutex_unlock(&team->lock);
1590
1591         return err;
1592 }
1593
1594 static struct rtnl_link_stats64 *
1595 team_get_stats64(struct net_device *dev, struct rtnl_link_stats64 *stats)
1596 {
1597         struct team *team = netdev_priv(dev);
1598         struct team_pcpu_stats *p;
1599         u64 rx_packets, rx_bytes, rx_multicast, tx_packets, tx_bytes;
1600         u32 rx_dropped = 0, tx_dropped = 0;
1601         unsigned int start;
1602         int i;
1603
1604         for_each_possible_cpu(i) {
1605                 p = per_cpu_ptr(team->pcpu_stats, i);
1606                 do {
1607                         start = u64_stats_fetch_begin_bh(&p->syncp);
1608                         rx_packets      = p->rx_packets;
1609                         rx_bytes        = p->rx_bytes;
1610                         rx_multicast    = p->rx_multicast;
1611                         tx_packets      = p->tx_packets;
1612                         tx_bytes        = p->tx_bytes;
1613                 } while (u64_stats_fetch_retry_bh(&p->syncp, start));
1614
1615                 stats->rx_packets       += rx_packets;
1616                 stats->rx_bytes         += rx_bytes;
1617                 stats->multicast        += rx_multicast;
1618                 stats->tx_packets       += tx_packets;
1619                 stats->tx_bytes         += tx_bytes;
1620                 /*
1621                  * rx_dropped & tx_dropped are u32, updated
1622                  * without syncp protection.
1623                  */
1624                 rx_dropped      += p->rx_dropped;
1625                 tx_dropped      += p->tx_dropped;
1626         }
1627         stats->rx_dropped       = rx_dropped;
1628         stats->tx_dropped       = tx_dropped;
1629         return stats;
1630 }
1631
1632 static int team_vlan_rx_add_vid(struct net_device *dev, __be16 proto, u16 vid)
1633 {
1634         struct team *team = netdev_priv(dev);
1635         struct team_port *port;
1636         int err;
1637
1638         /*
1639          * Alhough this is reader, it's guarded by team lock. It's not possible
1640          * to traverse list in reverse under rcu_read_lock
1641          */
1642         mutex_lock(&team->lock);
1643         list_for_each_entry(port, &team->port_list, list) {
1644                 err = vlan_vid_add(port->dev, proto, vid);
1645                 if (err)
1646                         goto unwind;
1647         }
1648         mutex_unlock(&team->lock);
1649
1650         return 0;
1651
1652 unwind:
1653         list_for_each_entry_continue_reverse(port, &team->port_list, list)
1654                 vlan_vid_del(port->dev, proto, vid);
1655         mutex_unlock(&team->lock);
1656
1657         return err;
1658 }
1659
1660 static int team_vlan_rx_kill_vid(struct net_device *dev, __be16 proto, u16 vid)
1661 {
1662         struct team *team = netdev_priv(dev);
1663         struct team_port *port;
1664
1665         rcu_read_lock();
1666         list_for_each_entry_rcu(port, &team->port_list, list)
1667                 vlan_vid_del(port->dev, proto, vid);
1668         rcu_read_unlock();
1669
1670         return 0;
1671 }
1672
1673 #ifdef CONFIG_NET_POLL_CONTROLLER
1674 static void team_poll_controller(struct net_device *dev)
1675 {
1676 }
1677
1678 static void __team_netpoll_cleanup(struct team *team)
1679 {
1680         struct team_port *port;
1681
1682         list_for_each_entry(port, &team->port_list, list)
1683                 team_port_disable_netpoll(port);
1684 }
1685
1686 static void team_netpoll_cleanup(struct net_device *dev)
1687 {
1688         struct team *team = netdev_priv(dev);
1689
1690         mutex_lock(&team->lock);
1691         __team_netpoll_cleanup(team);
1692         mutex_unlock(&team->lock);
1693 }
1694
1695 static int team_netpoll_setup(struct net_device *dev,
1696                               struct netpoll_info *npifo, gfp_t gfp)
1697 {
1698         struct team *team = netdev_priv(dev);
1699         struct team_port *port;
1700         int err = 0;
1701
1702         mutex_lock(&team->lock);
1703         list_for_each_entry(port, &team->port_list, list) {
1704                 err = team_port_enable_netpoll(team, port, gfp);
1705                 if (err) {
1706                         __team_netpoll_cleanup(team);
1707                         break;
1708                 }
1709         }
1710         mutex_unlock(&team->lock);
1711         return err;
1712 }
1713 #endif
1714
1715 static int team_add_slave(struct net_device *dev, struct net_device *port_dev)
1716 {
1717         struct team *team = netdev_priv(dev);
1718         int err;
1719
1720         mutex_lock(&team->lock);
1721         err = team_port_add(team, port_dev);
1722         mutex_unlock(&team->lock);
1723         return err;
1724 }
1725
1726 static int team_del_slave(struct net_device *dev, struct net_device *port_dev)
1727 {
1728         struct team *team = netdev_priv(dev);
1729         int err;
1730
1731         mutex_lock(&team->lock);
1732         err = team_port_del(team, port_dev);
1733         mutex_unlock(&team->lock);
1734         return err;
1735 }
1736
1737 static netdev_features_t team_fix_features(struct net_device *dev,
1738                                            netdev_features_t features)
1739 {
1740         struct team_port *port;
1741         struct team *team = netdev_priv(dev);
1742         netdev_features_t mask;
1743
1744         mask = features;
1745         features &= ~NETIF_F_ONE_FOR_ALL;
1746         features |= NETIF_F_ALL_FOR_ALL;
1747
1748         rcu_read_lock();
1749         list_for_each_entry_rcu(port, &team->port_list, list) {
1750                 features = netdev_increment_features(features,
1751                                                      port->dev->features,
1752                                                      mask);
1753         }
1754         rcu_read_unlock();
1755         return features;
1756 }
1757
1758 static int team_change_carrier(struct net_device *dev, bool new_carrier)
1759 {
1760         struct team *team = netdev_priv(dev);
1761
1762         team->user_carrier_enabled = true;
1763
1764         if (new_carrier)
1765                 netif_carrier_on(dev);
1766         else
1767                 netif_carrier_off(dev);
1768         return 0;
1769 }
1770
1771 static const struct net_device_ops team_netdev_ops = {
1772         .ndo_init               = team_init,
1773         .ndo_uninit             = team_uninit,
1774         .ndo_open               = team_open,
1775         .ndo_stop               = team_close,
1776         .ndo_start_xmit         = team_xmit,
1777         .ndo_select_queue       = team_select_queue,
1778         .ndo_change_rx_flags    = team_change_rx_flags,
1779         .ndo_set_rx_mode        = team_set_rx_mode,
1780         .ndo_set_mac_address    = team_set_mac_address,
1781         .ndo_change_mtu         = team_change_mtu,
1782         .ndo_get_stats64        = team_get_stats64,
1783         .ndo_vlan_rx_add_vid    = team_vlan_rx_add_vid,
1784         .ndo_vlan_rx_kill_vid   = team_vlan_rx_kill_vid,
1785 #ifdef CONFIG_NET_POLL_CONTROLLER
1786         .ndo_poll_controller    = team_poll_controller,
1787         .ndo_netpoll_setup      = team_netpoll_setup,
1788         .ndo_netpoll_cleanup    = team_netpoll_cleanup,
1789 #endif
1790         .ndo_add_slave          = team_add_slave,
1791         .ndo_del_slave          = team_del_slave,
1792         .ndo_fix_features       = team_fix_features,
1793         .ndo_change_carrier     = team_change_carrier,
1794 };
1795
1796 /***********************
1797  * ethtool interface
1798  ***********************/
1799
1800 static void team_ethtool_get_drvinfo(struct net_device *dev,
1801                                      struct ethtool_drvinfo *drvinfo)
1802 {
1803         strlcpy(drvinfo->driver, DRV_NAME, sizeof(drvinfo->driver));
1804         strlcpy(drvinfo->version, UTS_RELEASE, sizeof(drvinfo->version));
1805 }
1806
1807 static const struct ethtool_ops team_ethtool_ops = {
1808         .get_drvinfo            = team_ethtool_get_drvinfo,
1809         .get_link               = ethtool_op_get_link,
1810 };
1811
1812 /***********************
1813  * rt netlink interface
1814  ***********************/
1815
1816 static void team_setup_by_port(struct net_device *dev,
1817                                struct net_device *port_dev)
1818 {
1819         dev->header_ops = port_dev->header_ops;
1820         dev->type = port_dev->type;
1821         dev->hard_header_len = port_dev->hard_header_len;
1822         dev->addr_len = port_dev->addr_len;
1823         dev->mtu = port_dev->mtu;
1824         memcpy(dev->broadcast, port_dev->broadcast, port_dev->addr_len);
1825         memcpy(dev->dev_addr, port_dev->dev_addr, port_dev->addr_len);
1826 }
1827
1828 static int team_dev_type_check_change(struct net_device *dev,
1829                                       struct net_device *port_dev)
1830 {
1831         struct team *team = netdev_priv(dev);
1832         char *portname = port_dev->name;
1833         int err;
1834
1835         if (dev->type == port_dev->type)
1836                 return 0;
1837         if (!list_empty(&team->port_list)) {
1838                 netdev_err(dev, "Device %s is of different type\n", portname);
1839                 return -EBUSY;
1840         }
1841         err = call_netdevice_notifiers(NETDEV_PRE_TYPE_CHANGE, dev);
1842         err = notifier_to_errno(err);
1843         if (err) {
1844                 netdev_err(dev, "Refused to change device type\n");
1845                 return err;
1846         }
1847         dev_uc_flush(dev);
1848         dev_mc_flush(dev);
1849         team_setup_by_port(dev, port_dev);
1850         call_netdevice_notifiers(NETDEV_POST_TYPE_CHANGE, dev);
1851         return 0;
1852 }
1853
1854 static void team_setup(struct net_device *dev)
1855 {
1856         ether_setup(dev);
1857
1858         dev->netdev_ops = &team_netdev_ops;
1859         dev->ethtool_ops = &team_ethtool_ops;
1860         dev->destructor = team_destructor;
1861         dev->tx_queue_len = 0;
1862         dev->flags |= IFF_MULTICAST;
1863         dev->priv_flags &= ~(IFF_XMIT_DST_RELEASE | IFF_TX_SKB_SHARING);
1864
1865         /*
1866          * Indicate we support unicast address filtering. That way core won't
1867          * bring us to promisc mode in case a unicast addr is added.
1868          * Let this up to underlay drivers.
1869          */
1870         dev->priv_flags |= IFF_UNICAST_FLT | IFF_LIVE_ADDR_CHANGE;
1871
1872         dev->features |= NETIF_F_LLTX;
1873         dev->features |= NETIF_F_GRO;
1874         dev->hw_features = TEAM_VLAN_FEATURES |
1875                            NETIF_F_HW_VLAN_CTAG_TX |
1876                            NETIF_F_HW_VLAN_CTAG_RX |
1877                            NETIF_F_HW_VLAN_CTAG_FILTER;
1878
1879         dev->hw_features &= ~(NETIF_F_ALL_CSUM & ~NETIF_F_HW_CSUM);
1880         dev->features |= dev->hw_features;
1881 }
1882
1883 static int team_newlink(struct net *src_net, struct net_device *dev,
1884                         struct nlattr *tb[], struct nlattr *data[])
1885 {
1886         int err;
1887
1888         if (tb[IFLA_ADDRESS] == NULL)
1889                 eth_hw_addr_random(dev);
1890
1891         err = register_netdevice(dev);
1892         if (err)
1893                 return err;
1894
1895         return 0;
1896 }
1897
1898 static int team_validate(struct nlattr *tb[], struct nlattr *data[])
1899 {
1900         if (tb[IFLA_ADDRESS]) {
1901                 if (nla_len(tb[IFLA_ADDRESS]) != ETH_ALEN)
1902                         return -EINVAL;
1903                 if (!is_valid_ether_addr(nla_data(tb[IFLA_ADDRESS])))
1904                         return -EADDRNOTAVAIL;
1905         }
1906         return 0;
1907 }
1908
1909 static unsigned int team_get_num_tx_queues(void)
1910 {
1911         return TEAM_DEFAULT_NUM_TX_QUEUES;
1912 }
1913
1914 static unsigned int team_get_num_rx_queues(void)
1915 {
1916         return TEAM_DEFAULT_NUM_RX_QUEUES;
1917 }
1918
1919 static struct rtnl_link_ops team_link_ops __read_mostly = {
1920         .kind                   = DRV_NAME,
1921         .priv_size              = sizeof(struct team),
1922         .setup                  = team_setup,
1923         .newlink                = team_newlink,
1924         .validate               = team_validate,
1925         .get_num_tx_queues      = team_get_num_tx_queues,
1926         .get_num_rx_queues      = team_get_num_rx_queues,
1927 };
1928
1929
1930 /***********************************
1931  * Generic netlink custom interface
1932  ***********************************/
1933
1934 static struct genl_family team_nl_family = {
1935         .id             = GENL_ID_GENERATE,
1936         .name           = TEAM_GENL_NAME,
1937         .version        = TEAM_GENL_VERSION,
1938         .maxattr        = TEAM_ATTR_MAX,
1939         .netnsok        = true,
1940 };
1941
1942 static const struct nla_policy team_nl_policy[TEAM_ATTR_MAX + 1] = {
1943         [TEAM_ATTR_UNSPEC]                      = { .type = NLA_UNSPEC, },
1944         [TEAM_ATTR_TEAM_IFINDEX]                = { .type = NLA_U32 },
1945         [TEAM_ATTR_LIST_OPTION]                 = { .type = NLA_NESTED },
1946         [TEAM_ATTR_LIST_PORT]                   = { .type = NLA_NESTED },
1947 };
1948
1949 static const struct nla_policy
1950 team_nl_option_policy[TEAM_ATTR_OPTION_MAX + 1] = {
1951         [TEAM_ATTR_OPTION_UNSPEC]               = { .type = NLA_UNSPEC, },
1952         [TEAM_ATTR_OPTION_NAME] = {
1953                 .type = NLA_STRING,
1954                 .len = TEAM_STRING_MAX_LEN,
1955         },
1956         [TEAM_ATTR_OPTION_CHANGED]              = { .type = NLA_FLAG },
1957         [TEAM_ATTR_OPTION_TYPE]                 = { .type = NLA_U8 },
1958         [TEAM_ATTR_OPTION_DATA]                 = { .type = NLA_BINARY },
1959 };
1960
1961 static int team_nl_cmd_noop(struct sk_buff *skb, struct genl_info *info)
1962 {
1963         struct sk_buff *msg;
1964         void *hdr;
1965         int err;
1966
1967         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1968         if (!msg)
1969                 return -ENOMEM;
1970
1971         hdr = genlmsg_put(msg, info->snd_portid, info->snd_seq,
1972                           &team_nl_family, 0, TEAM_CMD_NOOP);
1973         if (!hdr) {
1974                 err = -EMSGSIZE;
1975                 goto err_msg_put;
1976         }
1977
1978         genlmsg_end(msg, hdr);
1979
1980         return genlmsg_unicast(genl_info_net(info), msg, info->snd_portid);
1981
1982 err_msg_put:
1983         nlmsg_free(msg);
1984
1985         return err;
1986 }
1987
1988 /*
1989  * Netlink cmd functions should be locked by following two functions.
1990  * Since dev gets held here, that ensures dev won't disappear in between.
1991  */
1992 static struct team *team_nl_team_get(struct genl_info *info)
1993 {
1994         struct net *net = genl_info_net(info);
1995         int ifindex;
1996         struct net_device *dev;
1997         struct team *team;
1998
1999         if (!info->attrs[TEAM_ATTR_TEAM_IFINDEX])
2000                 return NULL;
2001
2002         ifindex = nla_get_u32(info->attrs[TEAM_ATTR_TEAM_IFINDEX]);
2003         dev = dev_get_by_index(net, ifindex);
2004         if (!dev || dev->netdev_ops != &team_netdev_ops) {
2005                 if (dev)
2006                         dev_put(dev);
2007                 return NULL;
2008         }
2009
2010         team = netdev_priv(dev);
2011         mutex_lock(&team->lock);
2012         return team;
2013 }
2014
2015 static void team_nl_team_put(struct team *team)
2016 {
2017         mutex_unlock(&team->lock);
2018         dev_put(team->dev);
2019 }
2020
2021 typedef int team_nl_send_func_t(struct sk_buff *skb,
2022                                 struct team *team, u32 portid);
2023
2024 static int team_nl_send_unicast(struct sk_buff *skb, struct team *team, u32 portid)
2025 {
2026         return genlmsg_unicast(dev_net(team->dev), skb, portid);
2027 }
2028
2029 static int team_nl_fill_one_option_get(struct sk_buff *skb, struct team *team,
2030                                        struct team_option_inst *opt_inst)
2031 {
2032         struct nlattr *option_item;
2033         struct team_option *option = opt_inst->option;
2034         struct team_option_inst_info *opt_inst_info = &opt_inst->info;
2035         struct team_gsetter_ctx ctx;
2036         int err;
2037
2038         ctx.info = opt_inst_info;
2039         err = team_option_get(team, opt_inst, &ctx);
2040         if (err)
2041                 return err;
2042
2043         option_item = nla_nest_start(skb, TEAM_ATTR_ITEM_OPTION);
2044         if (!option_item)
2045                 return -EMSGSIZE;
2046
2047         if (nla_put_string(skb, TEAM_ATTR_OPTION_NAME, option->name))
2048                 goto nest_cancel;
2049         if (opt_inst_info->port &&
2050             nla_put_u32(skb, TEAM_ATTR_OPTION_PORT_IFINDEX,
2051                         opt_inst_info->port->dev->ifindex))
2052                 goto nest_cancel;
2053         if (opt_inst->option->array_size &&
2054             nla_put_u32(skb, TEAM_ATTR_OPTION_ARRAY_INDEX,
2055                         opt_inst_info->array_index))
2056                 goto nest_cancel;
2057
2058         switch (option->type) {
2059         case TEAM_OPTION_TYPE_U32:
2060                 if (nla_put_u8(skb, TEAM_ATTR_OPTION_TYPE, NLA_U32))
2061                         goto nest_cancel;
2062                 if (nla_put_u32(skb, TEAM_ATTR_OPTION_DATA, ctx.data.u32_val))
2063                         goto nest_cancel;
2064                 break;
2065         case TEAM_OPTION_TYPE_STRING:
2066                 if (nla_put_u8(skb, TEAM_ATTR_OPTION_TYPE, NLA_STRING))
2067                         goto nest_cancel;
2068                 if (nla_put_string(skb, TEAM_ATTR_OPTION_DATA,
2069                                    ctx.data.str_val))
2070                         goto nest_cancel;
2071                 break;
2072         case TEAM_OPTION_TYPE_BINARY:
2073                 if (nla_put_u8(skb, TEAM_ATTR_OPTION_TYPE, NLA_BINARY))
2074                         goto nest_cancel;
2075                 if (nla_put(skb, TEAM_ATTR_OPTION_DATA, ctx.data.bin_val.len,
2076                             ctx.data.bin_val.ptr))
2077                         goto nest_cancel;
2078                 break;
2079         case TEAM_OPTION_TYPE_BOOL:
2080                 if (nla_put_u8(skb, TEAM_ATTR_OPTION_TYPE, NLA_FLAG))
2081                         goto nest_cancel;
2082                 if (ctx.data.bool_val &&
2083                     nla_put_flag(skb, TEAM_ATTR_OPTION_DATA))
2084                         goto nest_cancel;
2085                 break;
2086         case TEAM_OPTION_TYPE_S32:
2087                 if (nla_put_u8(skb, TEAM_ATTR_OPTION_TYPE, NLA_S32))
2088                         goto nest_cancel;
2089                 if (nla_put_s32(skb, TEAM_ATTR_OPTION_DATA, ctx.data.s32_val))
2090                         goto nest_cancel;
2091                 break;
2092         default:
2093                 BUG();
2094         }
2095         if (opt_inst->removed && nla_put_flag(skb, TEAM_ATTR_OPTION_REMOVED))
2096                 goto nest_cancel;
2097         if (opt_inst->changed) {
2098                 if (nla_put_flag(skb, TEAM_ATTR_OPTION_CHANGED))
2099                         goto nest_cancel;
2100                 opt_inst->changed = false;
2101         }
2102         nla_nest_end(skb, option_item);
2103         return 0;
2104
2105 nest_cancel:
2106         nla_nest_cancel(skb, option_item);
2107         return -EMSGSIZE;
2108 }
2109
2110 static int __send_and_alloc_skb(struct sk_buff **pskb,
2111                                 struct team *team, u32 portid,
2112                                 team_nl_send_func_t *send_func)
2113 {
2114         int err;
2115
2116         if (*pskb) {
2117                 err = send_func(*pskb, team, portid);
2118                 if (err)
2119                         return err;
2120         }
2121         *pskb = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
2122         if (!*pskb)
2123                 return -ENOMEM;
2124         return 0;
2125 }
2126
2127 static int team_nl_send_options_get(struct team *team, u32 portid, u32 seq,
2128                                     int flags, team_nl_send_func_t *send_func,
2129                                     struct list_head *sel_opt_inst_list)
2130 {
2131         struct nlattr *option_list;
2132         struct nlmsghdr *nlh;
2133         void *hdr;
2134         struct team_option_inst *opt_inst;
2135         int err;
2136         struct sk_buff *skb = NULL;
2137         bool incomplete;
2138         int i;
2139
2140         opt_inst = list_first_entry(sel_opt_inst_list,
2141                                     struct team_option_inst, tmp_list);
2142
2143 start_again:
2144         err = __send_and_alloc_skb(&skb, team, portid, send_func);
2145         if (err)
2146                 return err;
2147
2148         hdr = genlmsg_put(skb, portid, seq, &team_nl_family, flags | NLM_F_MULTI,
2149                           TEAM_CMD_OPTIONS_GET);
2150         if (!hdr)
2151                 return -EMSGSIZE;
2152
2153         if (nla_put_u32(skb, TEAM_ATTR_TEAM_IFINDEX, team->dev->ifindex))
2154                 goto nla_put_failure;
2155         option_list = nla_nest_start(skb, TEAM_ATTR_LIST_OPTION);
2156         if (!option_list)
2157                 goto nla_put_failure;
2158
2159         i = 0;
2160         incomplete = false;
2161         list_for_each_entry_from(opt_inst, sel_opt_inst_list, tmp_list) {
2162                 err = team_nl_fill_one_option_get(skb, team, opt_inst);
2163                 if (err) {
2164                         if (err == -EMSGSIZE) {
2165                                 if (!i)
2166                                         goto errout;
2167                                 incomplete = true;
2168                                 break;
2169                         }
2170                         goto errout;
2171                 }
2172                 i++;
2173         }
2174
2175         nla_nest_end(skb, option_list);
2176         genlmsg_end(skb, hdr);
2177         if (incomplete)
2178                 goto start_again;
2179
2180 send_done:
2181         nlh = nlmsg_put(skb, portid, seq, NLMSG_DONE, 0, flags | NLM_F_MULTI);
2182         if (!nlh) {
2183                 err = __send_and_alloc_skb(&skb, team, portid, send_func);
2184                 if (err)
2185                         goto errout;
2186                 goto send_done;
2187         }
2188
2189         return send_func(skb, team, portid);
2190
2191 nla_put_failure:
2192         err = -EMSGSIZE;
2193 errout:
2194         genlmsg_cancel(skb, hdr);
2195         nlmsg_free(skb);
2196         return err;
2197 }
2198
2199 static int team_nl_cmd_options_get(struct sk_buff *skb, struct genl_info *info)
2200 {
2201         struct team *team;
2202         struct team_option_inst *opt_inst;
2203         int err;
2204         LIST_HEAD(sel_opt_inst_list);
2205
2206         team = team_nl_team_get(info);
2207         if (!team)
2208                 return -EINVAL;
2209
2210         list_for_each_entry(opt_inst, &team->option_inst_list, list)
2211                 list_add_tail(&opt_inst->tmp_list, &sel_opt_inst_list);
2212         err = team_nl_send_options_get(team, info->snd_portid, info->snd_seq,
2213                                        NLM_F_ACK, team_nl_send_unicast,
2214                                        &sel_opt_inst_list);
2215
2216         team_nl_team_put(team);
2217
2218         return err;
2219 }
2220
2221 static int team_nl_send_event_options_get(struct team *team,
2222                                           struct list_head *sel_opt_inst_list);
2223
2224 static int team_nl_cmd_options_set(struct sk_buff *skb, struct genl_info *info)
2225 {
2226         struct team *team;
2227         int err = 0;
2228         int i;
2229         struct nlattr *nl_option;
2230         LIST_HEAD(opt_inst_list);
2231
2232         team = team_nl_team_get(info);
2233         if (!team)
2234                 return -EINVAL;
2235
2236         err = -EINVAL;
2237         if (!info->attrs[TEAM_ATTR_LIST_OPTION]) {
2238                 err = -EINVAL;
2239                 goto team_put;
2240         }
2241
2242         nla_for_each_nested(nl_option, info->attrs[TEAM_ATTR_LIST_OPTION], i) {
2243                 struct nlattr *opt_attrs[TEAM_ATTR_OPTION_MAX + 1];
2244                 struct nlattr *attr;
2245                 struct nlattr *attr_data;
2246                 enum team_option_type opt_type;
2247                 int opt_port_ifindex = 0; /* != 0 for per-port options */
2248                 u32 opt_array_index = 0;
2249                 bool opt_is_array = false;
2250                 struct team_option_inst *opt_inst;
2251                 char *opt_name;
2252                 bool opt_found = false;
2253
2254                 if (nla_type(nl_option) != TEAM_ATTR_ITEM_OPTION) {
2255                         err = -EINVAL;
2256                         goto team_put;
2257                 }
2258                 err = nla_parse_nested(opt_attrs, TEAM_ATTR_OPTION_MAX,
2259                                        nl_option, team_nl_option_policy);
2260                 if (err)
2261                         goto team_put;
2262                 if (!opt_attrs[TEAM_ATTR_OPTION_NAME] ||
2263                     !opt_attrs[TEAM_ATTR_OPTION_TYPE]) {
2264                         err = -EINVAL;
2265                         goto team_put;
2266                 }
2267                 switch (nla_get_u8(opt_attrs[TEAM_ATTR_OPTION_TYPE])) {
2268                 case NLA_U32:
2269                         opt_type = TEAM_OPTION_TYPE_U32;
2270                         break;
2271                 case NLA_STRING:
2272                         opt_type = TEAM_OPTION_TYPE_STRING;
2273                         break;
2274                 case NLA_BINARY:
2275                         opt_type = TEAM_OPTION_TYPE_BINARY;
2276                         break;
2277                 case NLA_FLAG:
2278                         opt_type = TEAM_OPTION_TYPE_BOOL;
2279                         break;
2280                 case NLA_S32:
2281                         opt_type = TEAM_OPTION_TYPE_S32;
2282                         break;
2283                 default:
2284                         goto team_put;
2285                 }
2286
2287                 attr_data = opt_attrs[TEAM_ATTR_OPTION_DATA];
2288                 if (opt_type != TEAM_OPTION_TYPE_BOOL && !attr_data) {
2289                         err = -EINVAL;
2290                         goto team_put;
2291                 }
2292
2293                 opt_name = nla_data(opt_attrs[TEAM_ATTR_OPTION_NAME]);
2294                 attr = opt_attrs[TEAM_ATTR_OPTION_PORT_IFINDEX];
2295                 if (attr)
2296                         opt_port_ifindex = nla_get_u32(attr);
2297
2298                 attr = opt_attrs[TEAM_ATTR_OPTION_ARRAY_INDEX];
2299                 if (attr) {
2300                         opt_is_array = true;
2301                         opt_array_index = nla_get_u32(attr);
2302                 }
2303
2304                 list_for_each_entry(opt_inst, &team->option_inst_list, list) {
2305                         struct team_option *option = opt_inst->option;
2306                         struct team_gsetter_ctx ctx;
2307                         struct team_option_inst_info *opt_inst_info;
2308                         int tmp_ifindex;
2309
2310                         opt_inst_info = &opt_inst->info;
2311                         tmp_ifindex = opt_inst_info->port ?
2312                                       opt_inst_info->port->dev->ifindex : 0;
2313                         if (option->type != opt_type ||
2314                             strcmp(option->name, opt_name) ||
2315                             tmp_ifindex != opt_port_ifindex ||
2316                             (option->array_size && !opt_is_array) ||
2317                             opt_inst_info->array_index != opt_array_index)
2318                                 continue;
2319                         opt_found = true;
2320                         ctx.info = opt_inst_info;
2321                         switch (opt_type) {
2322                         case TEAM_OPTION_TYPE_U32:
2323                                 ctx.data.u32_val = nla_get_u32(attr_data);
2324                                 break;
2325                         case TEAM_OPTION_TYPE_STRING:
2326                                 if (nla_len(attr_data) > TEAM_STRING_MAX_LEN) {
2327                                         err = -EINVAL;
2328                                         goto team_put;
2329                                 }
2330                                 ctx.data.str_val = nla_data(attr_data);
2331                                 break;
2332                         case TEAM_OPTION_TYPE_BINARY:
2333                                 ctx.data.bin_val.len = nla_len(attr_data);
2334                                 ctx.data.bin_val.ptr = nla_data(attr_data);
2335                                 break;
2336                         case TEAM_OPTION_TYPE_BOOL:
2337                                 ctx.data.bool_val = attr_data ? true : false;
2338                                 break;
2339                         case TEAM_OPTION_TYPE_S32:
2340                                 ctx.data.s32_val = nla_get_s32(attr_data);
2341                                 break;
2342                         default:
2343                                 BUG();
2344                         }
2345                         err = team_option_set(team, opt_inst, &ctx);
2346                         if (err)
2347                                 goto team_put;
2348                         opt_inst->changed = true;
2349                         list_add(&opt_inst->tmp_list, &opt_inst_list);
2350                 }
2351                 if (!opt_found) {
2352                         err = -ENOENT;
2353                         goto team_put;
2354                 }
2355         }
2356
2357         err = team_nl_send_event_options_get(team, &opt_inst_list);
2358
2359 team_put:
2360         team_nl_team_put(team);
2361
2362         return err;
2363 }
2364
2365 static int team_nl_fill_one_port_get(struct sk_buff *skb,
2366                                      struct team_port *port)
2367 {
2368         struct nlattr *port_item;
2369
2370         port_item = nla_nest_start(skb, TEAM_ATTR_ITEM_PORT);
2371         if (!port_item)
2372                 goto nest_cancel;
2373         if (nla_put_u32(skb, TEAM_ATTR_PORT_IFINDEX, port->dev->ifindex))
2374                 goto nest_cancel;
2375         if (port->changed) {
2376                 if (nla_put_flag(skb, TEAM_ATTR_PORT_CHANGED))
2377                         goto nest_cancel;
2378                 port->changed = false;
2379         }
2380         if ((port->removed &&
2381              nla_put_flag(skb, TEAM_ATTR_PORT_REMOVED)) ||
2382             (port->state.linkup &&
2383              nla_put_flag(skb, TEAM_ATTR_PORT_LINKUP)) ||
2384             nla_put_u32(skb, TEAM_ATTR_PORT_SPEED, port->state.speed) ||
2385             nla_put_u8(skb, TEAM_ATTR_PORT_DUPLEX, port->state.duplex))
2386                 goto nest_cancel;
2387         nla_nest_end(skb, port_item);
2388         return 0;
2389
2390 nest_cancel:
2391         nla_nest_cancel(skb, port_item);
2392         return -EMSGSIZE;
2393 }
2394
2395 static int team_nl_send_port_list_get(struct team *team, u32 portid, u32 seq,
2396                                       int flags, team_nl_send_func_t *send_func,
2397                                       struct team_port *one_port)
2398 {
2399         struct nlattr *port_list;
2400         struct nlmsghdr *nlh;
2401         void *hdr;
2402         struct team_port *port;
2403         int err;
2404         struct sk_buff *skb = NULL;
2405         bool incomplete;
2406         int i;
2407
2408         port = list_first_entry_or_null(&team->port_list,
2409                                         struct team_port, list);
2410
2411 start_again:
2412         err = __send_and_alloc_skb(&skb, team, portid, send_func);
2413         if (err)
2414                 return err;
2415
2416         hdr = genlmsg_put(skb, portid, seq, &team_nl_family, flags | NLM_F_MULTI,
2417                           TEAM_CMD_PORT_LIST_GET);
2418         if (!hdr)
2419                 return -EMSGSIZE;
2420
2421         if (nla_put_u32(skb, TEAM_ATTR_TEAM_IFINDEX, team->dev->ifindex))
2422                 goto nla_put_failure;
2423         port_list = nla_nest_start(skb, TEAM_ATTR_LIST_PORT);
2424         if (!port_list)
2425                 goto nla_put_failure;
2426
2427         i = 0;
2428         incomplete = false;
2429
2430         /* If one port is selected, called wants to send port list containing
2431          * only this port. Otherwise go through all listed ports and send all
2432          */
2433         if (one_port) {
2434                 err = team_nl_fill_one_port_get(skb, one_port);
2435                 if (err)
2436                         goto errout;
2437         } else if (port) {
2438                 list_for_each_entry_from(port, &team->port_list, list) {
2439                         err = team_nl_fill_one_port_get(skb, port);
2440                         if (err) {
2441                                 if (err == -EMSGSIZE) {
2442                                         if (!i)
2443                                                 goto errout;
2444                                         incomplete = true;
2445                                         break;
2446                                 }
2447                                 goto errout;
2448                         }
2449                         i++;
2450                 }
2451         }
2452
2453         nla_nest_end(skb, port_list);
2454         genlmsg_end(skb, hdr);
2455         if (incomplete)
2456                 goto start_again;
2457
2458 send_done:
2459         nlh = nlmsg_put(skb, portid, seq, NLMSG_DONE, 0, flags | NLM_F_MULTI);
2460         if (!nlh) {
2461                 err = __send_and_alloc_skb(&skb, team, portid, send_func);
2462                 if (err)
2463                         goto errout;
2464                 goto send_done;
2465         }
2466
2467         return send_func(skb, team, portid);
2468
2469 nla_put_failure:
2470         err = -EMSGSIZE;
2471 errout:
2472         genlmsg_cancel(skb, hdr);
2473         nlmsg_free(skb);
2474         return err;
2475 }
2476
2477 static int team_nl_cmd_port_list_get(struct sk_buff *skb,
2478                                      struct genl_info *info)
2479 {
2480         struct team *team;
2481         int err;
2482
2483         team = team_nl_team_get(info);
2484         if (!team)
2485                 return -EINVAL;
2486
2487         err = team_nl_send_port_list_get(team, info->snd_portid, info->snd_seq,
2488                                          NLM_F_ACK, team_nl_send_unicast, NULL);
2489
2490         team_nl_team_put(team);
2491
2492         return err;
2493 }
2494
2495 static struct genl_ops team_nl_ops[] = {
2496         {
2497                 .cmd = TEAM_CMD_NOOP,
2498                 .doit = team_nl_cmd_noop,
2499                 .policy = team_nl_policy,
2500         },
2501         {
2502                 .cmd = TEAM_CMD_OPTIONS_SET,
2503                 .doit = team_nl_cmd_options_set,
2504                 .policy = team_nl_policy,
2505                 .flags = GENL_ADMIN_PERM,
2506         },
2507         {
2508                 .cmd = TEAM_CMD_OPTIONS_GET,
2509                 .doit = team_nl_cmd_options_get,
2510                 .policy = team_nl_policy,
2511                 .flags = GENL_ADMIN_PERM,
2512         },
2513         {
2514                 .cmd = TEAM_CMD_PORT_LIST_GET,
2515                 .doit = team_nl_cmd_port_list_get,
2516                 .policy = team_nl_policy,
2517                 .flags = GENL_ADMIN_PERM,
2518         },
2519 };
2520
2521 static struct genl_multicast_group team_change_event_mcgrp = {
2522         .name = TEAM_GENL_CHANGE_EVENT_MC_GRP_NAME,
2523 };
2524
2525 static int team_nl_send_multicast(struct sk_buff *skb,
2526                                   struct team *team, u32 portid)
2527 {
2528         return genlmsg_multicast_netns(dev_net(team->dev), skb, 0,
2529                                        team_change_event_mcgrp.id, GFP_KERNEL);
2530 }
2531
2532 static int team_nl_send_event_options_get(struct team *team,
2533                                           struct list_head *sel_opt_inst_list)
2534 {
2535         return team_nl_send_options_get(team, 0, 0, 0, team_nl_send_multicast,
2536                                         sel_opt_inst_list);
2537 }
2538
2539 static int team_nl_send_event_port_get(struct team *team,
2540                                        struct team_port *port)
2541 {
2542         return team_nl_send_port_list_get(team, 0, 0, 0, team_nl_send_multicast,
2543                                           port);
2544 }
2545
2546 static int team_nl_init(void)
2547 {
2548         int err;
2549
2550         err = genl_register_family_with_ops(&team_nl_family, team_nl_ops,
2551                                             ARRAY_SIZE(team_nl_ops));
2552         if (err)
2553                 return err;
2554
2555         err = genl_register_mc_group(&team_nl_family, &team_change_event_mcgrp);
2556         if (err)
2557                 goto err_change_event_grp_reg;
2558
2559         return 0;
2560
2561 err_change_event_grp_reg:
2562         genl_unregister_family(&team_nl_family);
2563
2564         return err;
2565 }
2566
2567 static void team_nl_fini(void)
2568 {
2569         genl_unregister_family(&team_nl_family);
2570 }
2571
2572
2573 /******************
2574  * Change checkers
2575  ******************/
2576
2577 static void __team_options_change_check(struct team *team)
2578 {
2579         int err;
2580         struct team_option_inst *opt_inst;
2581         LIST_HEAD(sel_opt_inst_list);
2582
2583         list_for_each_entry(opt_inst, &team->option_inst_list, list) {
2584                 if (opt_inst->changed)
2585                         list_add_tail(&opt_inst->tmp_list, &sel_opt_inst_list);
2586         }
2587         err = team_nl_send_event_options_get(team, &sel_opt_inst_list);
2588         if (err && err != -ESRCH)
2589                 netdev_warn(team->dev, "Failed to send options change via netlink (err %d)\n",
2590                             err);
2591 }
2592
2593 /* rtnl lock is held */
2594
2595 static void __team_port_change_send(struct team_port *port, bool linkup)
2596 {
2597         int err;
2598
2599         port->changed = true;
2600         port->state.linkup = linkup;
2601         team_refresh_port_linkup(port);
2602         if (linkup) {
2603                 struct ethtool_cmd ecmd;
2604
2605                 err = __ethtool_get_settings(port->dev, &ecmd);
2606                 if (!err) {
2607                         port->state.speed = ethtool_cmd_speed(&ecmd);
2608                         port->state.duplex = ecmd.duplex;
2609                         goto send_event;
2610                 }
2611         }
2612         port->state.speed = 0;
2613         port->state.duplex = 0;
2614
2615 send_event:
2616         err = team_nl_send_event_port_get(port->team, port);
2617         if (err && err != -ESRCH)
2618                 netdev_warn(port->team->dev, "Failed to send port change of device %s via netlink (err %d)\n",
2619                             port->dev->name, err);
2620
2621 }
2622
2623 static void __team_carrier_check(struct team *team)
2624 {
2625         struct team_port *port;
2626         bool team_linkup;
2627
2628         if (team->user_carrier_enabled)
2629                 return;
2630
2631         team_linkup = false;
2632         list_for_each_entry(port, &team->port_list, list) {
2633                 if (port->linkup) {
2634                         team_linkup = true;
2635                         break;
2636                 }
2637         }
2638
2639         if (team_linkup)
2640                 netif_carrier_on(team->dev);
2641         else
2642                 netif_carrier_off(team->dev);
2643 }
2644
2645 static void __team_port_change_check(struct team_port *port, bool linkup)
2646 {
2647         if (port->state.linkup != linkup)
2648                 __team_port_change_send(port, linkup);
2649         __team_carrier_check(port->team);
2650 }
2651
2652 static void __team_port_change_port_added(struct team_port *port, bool linkup)
2653 {
2654         __team_port_change_send(port, linkup);
2655         __team_carrier_check(port->team);
2656 }
2657
2658 static void __team_port_change_port_removed(struct team_port *port)
2659 {
2660         port->removed = true;
2661         __team_port_change_send(port, false);
2662         __team_carrier_check(port->team);
2663 }
2664
2665 static void team_port_change_check(struct team_port *port, bool linkup)
2666 {
2667         struct team *team = port->team;
2668
2669         mutex_lock(&team->lock);
2670         __team_port_change_check(port, linkup);
2671         mutex_unlock(&team->lock);
2672 }
2673
2674
2675 /************************************
2676  * Net device notifier event handler
2677  ************************************/
2678
2679 static int team_device_event(struct notifier_block *unused,
2680                              unsigned long event, void *ptr)
2681 {
2682         struct net_device *dev = netdev_notifier_info_to_dev(ptr);
2683         struct team_port *port;
2684
2685         port = team_port_get_rtnl(dev);
2686         if (!port)
2687                 return NOTIFY_DONE;
2688
2689         switch (event) {
2690         case NETDEV_UP:
2691                 if (netif_carrier_ok(dev))
2692                         team_port_change_check(port, true);
2693         case NETDEV_DOWN:
2694                 team_port_change_check(port, false);
2695         case NETDEV_CHANGE:
2696                 if (netif_running(port->dev))
2697                         team_port_change_check(port,
2698                                                !!netif_carrier_ok(port->dev));
2699                 break;
2700         case NETDEV_UNREGISTER:
2701                 team_del_slave(port->team->dev, dev);
2702                 break;
2703         case NETDEV_FEAT_CHANGE:
2704                 team_compute_features(port->team);
2705                 break;
2706         case NETDEV_CHANGEMTU:
2707                 /* Forbid to change mtu of underlaying device */
2708                 return NOTIFY_BAD;
2709         case NETDEV_PRE_TYPE_CHANGE:
2710                 /* Forbid to change type of underlaying device */
2711                 return NOTIFY_BAD;
2712         }
2713         return NOTIFY_DONE;
2714 }
2715
2716 static struct notifier_block team_notifier_block __read_mostly = {
2717         .notifier_call = team_device_event,
2718 };
2719
2720
2721 /***********************
2722  * Module init and exit
2723  ***********************/
2724
2725 static int __init team_module_init(void)
2726 {
2727         int err;
2728
2729         register_netdevice_notifier(&team_notifier_block);
2730
2731         err = rtnl_link_register(&team_link_ops);
2732         if (err)
2733                 goto err_rtnl_reg;
2734
2735         err = team_nl_init();
2736         if (err)
2737                 goto err_nl_init;
2738
2739         return 0;
2740
2741 err_nl_init:
2742         rtnl_link_unregister(&team_link_ops);
2743
2744 err_rtnl_reg:
2745         unregister_netdevice_notifier(&team_notifier_block);
2746
2747         return err;
2748 }
2749
2750 static void __exit team_module_exit(void)
2751 {
2752         team_nl_fini();
2753         rtnl_link_unregister(&team_link_ops);
2754         unregister_netdevice_notifier(&team_notifier_block);
2755 }
2756
2757 module_init(team_module_init);
2758 module_exit(team_module_exit);
2759
2760 MODULE_LICENSE("GPL v2");
2761 MODULE_AUTHOR("Jiri Pirko <jpirko@redhat.com>");
2762 MODULE_DESCRIPTION("Ethernet team device driver");
2763 MODULE_ALIAS_RTNL_LINK(DRV_NAME);