net/dsa/dsa2.c

   1 /*
   2  * net/dsa/dsa2.c - Hardware switch handling, binding version 2
   3  * Copyright (c) 2008-2009 Marvell Semiconductor
   4  * Copyright (c) 2013 Florian Fainelli <florian@openwrt.org>
   5  * Copyright (c) 2016 Andrew Lunn <andrew@lunn.ch>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  */
  12
  13 #include <linux/device.h>
  14 #include <linux/err.h>
  15 #include <linux/list.h>
  16 #include <linux/netdevice.h>
  17 #include <linux/slab.h>
  18 #include <linux/rtnetlink.h>
  19 #include <linux/of.h>
  20 #include <linux/of_net.h>
  21 #include <net/dsa.h>
  22 #include "dsa_priv.h"
  23
  24 static LIST_HEAD(dsa_switch_trees);
  25 static DEFINE_MUTEX(dsa2_mutex);
  26
  27 static const struct devlink_ops dsa_devlink_ops = {
  28 };
  29
  30 static struct dsa_switch_tree *dsa_get_dst(u32 tree)
  31 {
  32         struct dsa_switch_tree *dst;
  33
  34         list_for_each_entry(dst, &dsa_switch_trees, list)
  35                 if (dst->tree == tree) {
  36                         kref_get(&dst->refcount);
  37                         return dst;
  38                 }
  39         return NULL;
  40 }
  41
  42 static void dsa_free_dst(struct kref *ref)
  43 {
  44         struct dsa_switch_tree *dst = container_of(ref, struct dsa_switch_tree,
  45                                                    refcount);
  46
  47         list_del(&dst->list);
  48         kfree(dst);
  49 }
  50
  51 static void dsa_put_dst(struct dsa_switch_tree *dst)
  52 {
  53         kref_put(&dst->refcount, dsa_free_dst);
  54 }
  55
  56 static struct dsa_switch_tree *dsa_add_dst(u32 tree)
  57 {
  58         struct dsa_switch_tree *dst;
  59
  60         dst = kzalloc(sizeof(*dst), GFP_KERNEL);
  61         if (!dst)
  62                 return NULL;
  63         dst->tree = tree;
  64         INIT_LIST_HEAD(&dst->list);
  65         list_add_tail(&dsa_switch_trees, &dst->list);
  66         kref_init(&dst->refcount);
  67
  68         return dst;
  69 }
  70
  71 static void dsa_dst_add_ds(struct dsa_switch_tree *dst,
  72                            struct dsa_switch *ds, u32 index)
  73 {
  74         kref_get(&dst->refcount);
  75         dst->ds[index] = ds;
  76 }
  77
  78 static void dsa_dst_del_ds(struct dsa_switch_tree *dst,
  79                            struct dsa_switch *ds, u32 index)
  80 {
  81         dst->ds[index] = NULL;
  82         kref_put(&dst->refcount, dsa_free_dst);
  83 }
  84
  85 /* For platform data configurations, we need to have a valid name argument to
  86  * differentiate a disabled port from an enabled one
  87  */
  88 static bool dsa_port_is_valid(struct dsa_port *port)
  89 {
  90         return !!(port->dn || port->name);
  91 }
  92
  93 static bool dsa_port_is_dsa(struct dsa_port *port)
  94 {
  95         if (port->name && !strcmp(port->name, "dsa"))
  96                 return true;
  97         else
  98                 return !!of_parse_phandle(port->dn, "link", 0);
  99 }
 100
 101 static bool dsa_port_is_cpu(struct dsa_port *port)
 102 {
 103         if (port->name && !strcmp(port->name, "cpu"))
 104                 return true;
 105         else
 106                 return !!of_parse_phandle(port->dn, "ethernet", 0);
 107 }
 108
 109 static bool dsa_ds_find_port_dn(struct dsa_switch *ds,
 110                                 struct device_node *port)
 111 {
 112         u32 index;
 113
 114         for (index = 0; index < ds->num_ports; index++)
 115                 if (ds->ports[index].dn == port)
 116                         return true;
 117         return false;
 118 }
 119
 120 static struct dsa_switch *dsa_dst_find_port_dn(struct dsa_switch_tree *dst,
 121                                                struct device_node *port)
 122 {
 123         struct dsa_switch *ds;
 124         u32 index;
 125
 126         for (index = 0; index < DSA_MAX_SWITCHES; index++) {
 127                 ds = dst->ds[index];
 128                 if (!ds)
 129                         continue;
 130
 131                 if (dsa_ds_find_port_dn(ds, port))
 132                         return ds;
 133         }
 134
 135         return NULL;
 136 }
 137
 138 static int dsa_port_complete(struct dsa_switch_tree *dst,
 139                              struct dsa_switch *src_ds,
 140                              struct dsa_port *port,
 141                              u32 src_port)
 142 {
 143         struct device_node *link;
 144         int index;
 145         struct dsa_switch *dst_ds;
 146
 147         for (index = 0;; index++) {
 148                 link = of_parse_phandle(port->dn, "link", index);
 149                 if (!link)
 150                         break;
 151
 152                 dst_ds = dsa_dst_find_port_dn(dst, link);
 153                 of_node_put(link);
 154
 155                 if (!dst_ds)
 156                         return 1;
 157
 158                 src_ds->rtable[dst_ds->index] = src_port;
 159         }
 160
 161         return 0;
 162 }
 163
 164 /* A switch is complete if all the DSA ports phandles point to ports
 165  * known in the tree. A return value of 1 means the tree is not
 166  * complete. This is not an error condition. A value of 0 is
 167  * success.
 168  */
 169 static int dsa_ds_complete(struct dsa_switch_tree *dst, struct dsa_switch *ds)
 170 {
 171         struct dsa_port *port;
 172         u32 index;
 173         int err;
 174
 175         for (index = 0; index < ds->num_ports; index++) {
 176                 port = &ds->ports[index];
 177                 if (!dsa_port_is_valid(port))
 178                         continue;
 179
 180                 if (!dsa_port_is_dsa(port))
 181                         continue;
 182
 183                 err = dsa_port_complete(dst, ds, port, index);
 184                 if (err != 0)
 185                         return err;
 186
 187                 ds->dsa_port_mask |= BIT(index);
 188         }
 189
 190         return 0;
 191 }
 192
 193 /* A tree is complete if all the DSA ports phandles point to ports
 194  * known in the tree. A return value of 1 means the tree is not
 195  * complete. This is not an error condition. A value of 0 is
 196  * success.
 197  */
 198 static int dsa_dst_complete(struct dsa_switch_tree *dst)
 199 {
 200         struct dsa_switch *ds;
 201         u32 index;
 202         int err;
 203
 204         for (index = 0; index < DSA_MAX_SWITCHES; index++) {
 205                 ds = dst->ds[index];
 206                 if (!ds)
 207                         continue;
 208
 209                 err = dsa_ds_complete(dst, ds);
 210                 if (err != 0)
 211                         return err;
 212         }
 213
 214         return 0;
 215 }
 216
 217 static int dsa_dsa_port_apply(struct dsa_port *port, u32 index,
 218                               struct dsa_switch *ds)
 219 {
 220         int err;
 221
 222         err = dsa_cpu_dsa_setup(ds, ds->dev, port, index);
 223         if (err) {
 224                 dev_warn(ds->dev, "Failed to setup dsa port %d: %d\n",
 225                          index, err);
 226                 return err;
 227         }
 228
 229         memset(&ds->ports[index].devlink_port, 0,
 230                sizeof(ds->ports[index].devlink_port));
 231
 232         return devlink_port_register(ds->devlink,
 233                                      &ds->ports[index].devlink_port,
 234                                      index);
 235 }
 236
 237 static void dsa_dsa_port_unapply(struct dsa_port *port, u32 index,
 238                                  struct dsa_switch *ds)
 239 {
 240         devlink_port_unregister(&ds->ports[index].devlink_port);
 241         dsa_cpu_dsa_destroy(port);
 242 }
 243
 244 static int dsa_cpu_port_apply(struct dsa_port *port, u32 index,
 245                               struct dsa_switch *ds)
 246 {
 247         int err;
 248
 249         err = dsa_cpu_dsa_setup(ds, ds->dev, port, index);
 250         if (err) {
 251                 dev_warn(ds->dev, "Failed to setup cpu port %d: %d\n",
 252                          index, err);
 253                 return err;
 254         }
 255
 256         ds->cpu_port_mask |= BIT(index);
 257
 258         memset(&ds->ports[index].devlink_port, 0,
 259                sizeof(ds->ports[index].devlink_port));
 260         err = devlink_port_register(ds->devlink, &ds->ports[index].devlink_port,
 261                                     index);
 262         return err;
 263 }
 264
 265 static void dsa_cpu_port_unapply(struct dsa_port *port, u32 index,
 266                                  struct dsa_switch *ds)
 267 {
 268         devlink_port_unregister(&ds->ports[index].devlink_port);
 269         dsa_cpu_dsa_destroy(port);
 270         ds->cpu_port_mask &= ~BIT(index);
 271
 272 }
 273
 274 static int dsa_user_port_apply(struct dsa_port *port, u32 index,
 275                                struct dsa_switch *ds)
 276 {
 277         const char *name = port->name;
 278         int err;
 279
 280         if (port->dn)
 281                 name = of_get_property(port->dn, "label", NULL);
 282         if (!name)
 283                 name = "eth%d";
 284
 285         err = dsa_slave_create(ds, ds->dev, index, name);
 286         if (err) {
 287                 dev_warn(ds->dev, "Failed to create slave %d: %d\n",
 288                          index, err);
 289                 ds->ports[index].netdev = NULL;
 290                 return err;
 291         }
 292
 293         memset(&ds->ports[index].devlink_port, 0,
 294                sizeof(ds->ports[index].devlink_port));
 295         err = devlink_port_register(ds->devlink, &ds->ports[index].devlink_port,
 296                                     index);
 297         if (err)
 298                 return err;
 299
 300         devlink_port_type_eth_set(&ds->ports[index].devlink_port,
 301                                   ds->ports[index].netdev);
 302
 303         return 0;
 304 }
 305
 306 static void dsa_user_port_unapply(struct dsa_port *port, u32 index,
 307                                   struct dsa_switch *ds)
 308 {
 309         devlink_port_unregister(&ds->ports[index].devlink_port);
 310         if (ds->ports[index].netdev) {
 311                 dsa_slave_destroy(ds->ports[index].netdev);
 312                 ds->ports[index].netdev = NULL;
 313                 ds->enabled_port_mask &= ~(1 << index);
 314         }
 315 }
 316
 317 static int dsa_ds_apply(struct dsa_switch_tree *dst, struct dsa_switch *ds)
 318 {
 319         struct dsa_port *port;
 320         u32 index;
 321         int err;
 322
 323         /* Initialize ds->phys_mii_mask before registering the slave MDIO bus
 324          * driver and before ops->setup() has run, since the switch drivers and
 325          * the slave MDIO bus driver rely on these values for probing PHY
 326          * devices or not
 327          */
 328         ds->phys_mii_mask = ds->enabled_port_mask;
 329
 330         /* Add the switch to devlink before calling setup, so that setup can
 331          * add dpipe tables
 332          */
 333         ds->devlink = devlink_alloc(&dsa_devlink_ops, 0);
 334         if (!ds->devlink)
 335                 return -ENOMEM;
 336
 337         err = devlink_register(ds->devlink, ds->dev);
 338         if (err)
 339                 return err;
 340
 341         err = ds->ops->setup(ds);
 342         if (err < 0)
 343                 return err;
 344
 345         err = dsa_switch_register_notifier(ds);
 346         if (err)
 347                 return err;
 348
 349         if (ds->ops->set_addr) {
 350                 err = ds->ops->set_addr(ds, dst->master_netdev->dev_addr);
 351                 if (err < 0)
 352                         return err;
 353         }
 354
 355         if (!ds->slave_mii_bus && ds->ops->phy_read) {
 356                 ds->slave_mii_bus = devm_mdiobus_alloc(ds->dev);
 357                 if (!ds->slave_mii_bus)
 358                         return -ENOMEM;
 359
 360                 dsa_slave_mii_bus_init(ds);
 361
 362                 err = mdiobus_register(ds->slave_mii_bus);
 363                 if (err < 0)
 364                         return err;
 365         }
 366
 367         for (index = 0; index < ds->num_ports; index++) {
 368                 port = &ds->ports[index];
 369                 if (!dsa_port_is_valid(port))
 370                         continue;
 371
 372                 if (dsa_port_is_dsa(port)) {
 373                         err = dsa_dsa_port_apply(port, index, ds);
 374                         if (err)
 375                                 return err;
 376                         continue;
 377                 }
 378
 379                 if (dsa_port_is_cpu(port)) {
 380                         err = dsa_cpu_port_apply(port, index, ds);
 381                         if (err)
 382                                 return err;
 383                         continue;
 384                 }
 385
 386                 err = dsa_user_port_apply(port, index, ds);
 387                 if (err)
 388                         continue;
 389         }
 390
 391         return 0;
 392 }
 393
 394 static void dsa_ds_unapply(struct dsa_switch_tree *dst, struct dsa_switch *ds)
 395 {
 396         struct dsa_port *port;
 397         u32 index;
 398
 399         for (index = 0; index < ds->num_ports; index++) {
 400                 port = &ds->ports[index];
 401                 if (!dsa_port_is_valid(port))
 402                         continue;
 403
 404                 if (dsa_port_is_dsa(port)) {
 405                         dsa_dsa_port_unapply(port, index, ds);
 406                         continue;
 407                 }
 408
 409                 if (dsa_port_is_cpu(port)) {
 410                         dsa_cpu_port_unapply(port, index, ds);
 411                         continue;
 412                 }
 413
 414                 dsa_user_port_unapply(port, index, ds);
 415         }
 416
 417         if (ds->slave_mii_bus && ds->ops->phy_read)
 418                 mdiobus_unregister(ds->slave_mii_bus);
 419
 420         dsa_switch_unregister_notifier(ds);
 421
 422         if (ds->devlink) {
 423                 devlink_unregister(ds->devlink);
 424                 devlink_free(ds->devlink);
 425                 ds->devlink = NULL;
 426         }
 427
 428 }
 429
 430 static int dsa_dst_apply(struct dsa_switch_tree *dst)
 431 {
 432         struct dsa_switch *ds;
 433         u32 index;
 434         int err;
 435
 436         for (index = 0; index < DSA_MAX_SWITCHES; index++) {
 437                 ds = dst->ds[index];
 438                 if (!ds)
 439                         continue;
 440
 441                 err = dsa_ds_apply(dst, ds);
 442                 if (err)
 443                         return err;
 444         }
 445
 446         if (dst->cpu_switch) {
 447                 err = dsa_cpu_port_ethtool_setup(dst->cpu_switch);
 448                 if (err)
 449                         return err;
 450         }
 451
 452         /* If we use a tagging format that doesn't have an ethertype
 453          * field, make sure that all packets from this point on get
 454          * sent to the tag format's receive function.
 455          */
 456         wmb();
 457         dst->master_netdev->dsa_ptr = (void *)dst;
 458         dst->applied = true;
 459
 460         return 0;
 461 }
 462
 463 static void dsa_dst_unapply(struct dsa_switch_tree *dst)
 464 {
 465         struct dsa_switch *ds;
 466         u32 index;
 467
 468         if (!dst->applied)
 469                 return;
 470
 471         dst->master_netdev->dsa_ptr = NULL;
 472
 473         /* If we used a tagging format that doesn't have an ethertype
 474          * field, make sure that all packets from this point get sent
 475          * without the tag and go through the regular receive path.
 476          */
 477         wmb();
 478
 479         for (index = 0; index < DSA_MAX_SWITCHES; index++) {
 480                 ds = dst->ds[index];
 481                 if (!ds)
 482                         continue;
 483
 484                 dsa_ds_unapply(dst, ds);
 485         }
 486
 487         if (dst->cpu_switch) {
 488                 dsa_cpu_port_ethtool_restore(dst->cpu_switch);
 489                 dst->cpu_switch = NULL;
 490         }
 491
 492         pr_info("DSA: tree %d unapplied\n", dst->tree);
 493         dst->applied = false;
 494 }
 495
 496 static int dsa_cpu_parse(struct dsa_port *port, u32 index,
 497                          struct dsa_switch_tree *dst,
 498                          struct dsa_switch *ds)
 499 {
 500         enum dsa_tag_protocol tag_protocol;
 501         struct net_device *ethernet_dev;
 502         struct device_node *ethernet;
 503
 504         if (port->dn) {
 505                 ethernet = of_parse_phandle(port->dn, "ethernet", 0);
 506                 if (!ethernet)
 507                         return -EINVAL;
 508                 ethernet_dev = of_find_net_device_by_node(ethernet);
 509         } else {
 510                 ethernet_dev = dsa_dev_to_net_device(ds->cd->netdev[index]);
 511                 dev_put(ethernet_dev);
 512         }
 513
 514         if (!ethernet_dev)
 515                 return -EPROBE_DEFER;
 516
 517         if (!ds->master_netdev)
 518                 ds->master_netdev = ethernet_dev;
 519
 520         if (!dst->master_netdev)
 521                 dst->master_netdev = ethernet_dev;
 522
 523         if (!dst->cpu_switch) {
 524                 dst->cpu_switch = ds;
 525                 dst->cpu_port = index;
 526         }
 527
 528         tag_protocol = ds->ops->get_tag_protocol(ds);
 529         dst->tag_ops = dsa_resolve_tag_protocol(tag_protocol);
 530         if (IS_ERR(dst->tag_ops)) {
 531                 dev_warn(ds->dev, "No tagger for this switch\n");
 532                 return PTR_ERR(dst->tag_ops);
 533         }
 534
 535         dst->rcv = dst->tag_ops->rcv;
 536
 537         return 0;
 538 }
 539
 540 static int dsa_ds_parse(struct dsa_switch_tree *dst, struct dsa_switch *ds)
 541 {
 542         struct dsa_port *port;
 543         u32 index;
 544         int err;
 545
 546         for (index = 0; index < ds->num_ports; index++) {
 547                 port = &ds->ports[index];
 548                 if (!dsa_port_is_valid(port))
 549                         continue;
 550
 551                 if (dsa_port_is_cpu(port)) {
 552                         err = dsa_cpu_parse(port, index, dst, ds);
 553                         if (err)
 554                                 return err;
 555                 }
 556         }
 557
 558         pr_info("DSA: switch %d %d parsed\n", dst->tree, ds->index);
 559
 560         return 0;
 561 }
 562
 563 static int dsa_dst_parse(struct dsa_switch_tree *dst)
 564 {
 565         struct dsa_switch *ds;
 566         u32 index;
 567         int err;
 568
 569         for (index = 0; index < DSA_MAX_SWITCHES; index++) {
 570                 ds = dst->ds[index];
 571                 if (!ds)
 572                         continue;
 573
 574                 err = dsa_ds_parse(dst, ds);
 575                 if (err)
 576                         return err;
 577         }
 578
 579         if (!dst->master_netdev) {
 580                 pr_warn("Tree has no master device\n");
 581                 return -EINVAL;
 582         }
 583
 584         pr_info("DSA: tree %d parsed\n", dst->tree);
 585
 586         return 0;
 587 }
 588
 589 static int dsa_parse_ports_dn(struct device_node *ports, struct dsa_switch *ds)
 590 {
 591         struct device_node *port;
 592         int err;
 593         u32 reg;
 594
 595         for_each_available_child_of_node(ports, port) {
 596                 err = of_property_read_u32(port, "reg", &reg);
 597                 if (err)
 598                         return err;
 599
 600                 if (reg >= ds->num_ports)
 601                         return -EINVAL;
 602
 603                 ds->ports[reg].dn = port;
 604
 605                 /* Initialize enabled_port_mask now for ops->setup()
 606                  * to have access to a correct value, just like what
 607                  * net/dsa/dsa.c::dsa_switch_setup_one does.
 608                  */
 609                 if (!dsa_port_is_cpu(&ds->ports[reg]))
 610                         ds->enabled_port_mask |= 1 << reg;
 611         }
 612
 613         return 0;
 614 }
 615
 616 static int dsa_parse_ports(struct dsa_chip_data *cd, struct dsa_switch *ds)
 617 {
 618         bool valid_name_found = false;
 619         unsigned int i;
 620
 621         for (i = 0; i < DSA_MAX_PORTS; i++) {
 622                 if (!cd->port_names[i])
 623                         continue;
 624
 625                 ds->ports[i].name = cd->port_names[i];
 626
 627                 /* Initialize enabled_port_mask now for drv->setup()
 628                  * to have access to a correct value, just like what
 629                  * net/dsa/dsa.c::dsa_switch_setup_one does.
 630                  */
 631                 if (!dsa_port_is_cpu(&ds->ports[i]))
 632                         ds->enabled_port_mask |= 1 << i;
 633
 634                 valid_name_found = true;
 635         }
 636
 637         if (!valid_name_found && i == DSA_MAX_PORTS)
 638                 return -EINVAL;
 639
 640         return 0;
 641 }
 642
 643 static int dsa_parse_member_dn(struct device_node *np, u32 *tree, u32 *index)
 644 {
 645         int err;
 646
 647         *tree = *index = 0;
 648
 649         err = of_property_read_u32_index(np, "dsa,member", 0, tree);
 650         if (err) {
 651                 /* Does not exist, but it is optional */
 652                 if (err == -EINVAL)
 653                         return 0;
 654                 return err;
 655         }
 656
 657         err = of_property_read_u32_index(np, "dsa,member", 1, index);
 658         if (err)
 659                 return err;
 660
 661         if (*index >= DSA_MAX_SWITCHES)
 662                 return -EINVAL;
 663
 664         return 0;
 665 }
 666
 667 static int dsa_parse_member(struct dsa_chip_data *pd, u32 *tree, u32 *index)
 668 {
 669         if (!pd)
 670                 return -ENODEV;
 671
 672         /* We do not support complex trees with dsa_chip_data */
 673         *tree = 0;
 674         *index = 0;
 675
 676         return 0;
 677 }
 678
 679 static struct device_node *dsa_get_ports(struct dsa_switch *ds,
 680                                          struct device_node *np)
 681 {
 682         struct device_node *ports;
 683
 684         ports = of_get_child_by_name(np, "ports");
 685         if (!ports) {
 686                 dev_err(ds->dev, "no ports child node found\n");
 687                 return ERR_PTR(-EINVAL);
 688         }
 689
 690         return ports;
 691 }
 692
 693 static int _dsa_register_switch(struct dsa_switch *ds, struct device *dev)
 694 {
 695         struct dsa_chip_data *pdata = dev->platform_data;
 696         struct device_node *np = dev->of_node;
 697         struct dsa_switch_tree *dst;
 698         struct device_node *ports;
 699         u32 tree, index;
 700         int i, err;
 701
 702         if (np) {
 703                 err = dsa_parse_member_dn(np, &tree, &index);
 704                 if (err)
 705                         return err;
 706
 707                 ports = dsa_get_ports(ds, np);
 708                 if (IS_ERR(ports))
 709                         return PTR_ERR(ports);
 710
 711                 err = dsa_parse_ports_dn(ports, ds);
 712                 if (err)
 713                         return err;
 714         } else {
 715                 err = dsa_parse_member(pdata, &tree, &index);
 716                 if (err)
 717                         return err;
 718
 719                 err = dsa_parse_ports(pdata, ds);
 720                 if (err)
 721                         return err;
 722         }
 723
 724         dst = dsa_get_dst(tree);
 725         if (!dst) {
 726                 dst = dsa_add_dst(tree);
 727                 if (!dst)
 728                         return -ENOMEM;
 729         }
 730
 731         if (dst->ds[index]) {
 732                 err = -EBUSY;
 733                 goto out;
 734         }
 735
 736         ds->dst = dst;
 737         ds->index = index;
 738         ds->cd = pdata;
 739
 740         /* Initialize the routing table */
 741         for (i = 0; i < DSA_MAX_SWITCHES; ++i)
 742                 ds->rtable[i] = DSA_RTABLE_NONE;
 743
 744         dsa_dst_add_ds(dst, ds, index);
 745
 746         err = dsa_dst_complete(dst);
 747         if (err < 0)
 748                 goto out_del_dst;
 749
 750         if (err == 1) {
 751                 /* Not all switches registered yet */
 752                 err = 0;
 753                 goto out;
 754         }
 755
 756         if (dst->applied) {
 757                 pr_info("DSA: Disjoint trees?\n");
 758                 return -EINVAL;
 759         }
 760
 761         err = dsa_dst_parse(dst);
 762         if (err) {
 763                 if (err == -EPROBE_DEFER) {
 764                         dsa_dst_del_ds(dst, ds, ds->index);
 765                         return err;
 766                 }
 767
 768                 goto out_del_dst;
 769         }
 770
 771         err = dsa_dst_apply(dst);
 772         if (err) {
 773                 dsa_dst_unapply(dst);
 774                 goto out_del_dst;
 775         }
 776
 777         dsa_put_dst(dst);
 778         return 0;
 779
 780 out_del_dst:
 781         dsa_dst_del_ds(dst, ds, ds->index);
 782 out:
 783         dsa_put_dst(dst);
 784
 785         return err;
 786 }
 787
 788 struct dsa_switch *dsa_switch_alloc(struct device *dev, size_t n)
 789 {
 790         size_t size = sizeof(struct dsa_switch) + n * sizeof(struct dsa_port);
 791         struct dsa_switch *ds;
 792         int i;
 793
 794         ds = devm_kzalloc(dev, size, GFP_KERNEL);
 795         if (!ds)
 796                 return NULL;
 797
 798         ds->dev = dev;
 799         ds->num_ports = n;
 800
 801         for (i = 0; i < ds->num_ports; ++i) {
 802                 ds->ports[i].index = i;
 803                 ds->ports[i].ds = ds;
 804         }
 805
 806         return ds;
 807 }
 808 EXPORT_SYMBOL_GPL(dsa_switch_alloc);
 809
 810 int dsa_register_switch(struct dsa_switch *ds, struct device *dev)
 811 {
 812         int err;
 813
 814         mutex_lock(&dsa2_mutex);
 815         err = _dsa_register_switch(ds, dev);
 816         mutex_unlock(&dsa2_mutex);
 817
 818         return err;
 819 }
 820 EXPORT_SYMBOL_GPL(dsa_register_switch);
 821
 822 static void _dsa_unregister_switch(struct dsa_switch *ds)
 823 {
 824         struct dsa_switch_tree *dst = ds->dst;
 825
 826         dsa_dst_unapply(dst);
 827
 828         dsa_dst_del_ds(dst, ds, ds->index);
 829 }
 830
 831 void dsa_unregister_switch(struct dsa_switch *ds)
 832 {
 833         mutex_lock(&dsa2_mutex);
 834         _dsa_unregister_switch(ds);
 835         mutex_unlock(&dsa2_mutex);
 836 }
 837 EXPORT_SYMBOL_GPL(dsa_unregister_switch);