drivers/net/team/team.c

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