dsa2.c 16.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * net/dsa/dsa2.c - Hardware switch handling, binding version 2
 * Copyright (c) 2008-2009 Marvell Semiconductor
 * Copyright (c) 2013 Florian Fainelli <florian@openwrt.org>
 * Copyright (c) 2016 Andrew Lunn <andrew@lunn.ch>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 */

#include <linux/device.h>
#include <linux/err.h>
#include <linux/list.h>
16
#include <linux/netdevice.h>
17 18 19 20
#include <linux/slab.h>
#include <linux/rtnetlink.h>
#include <linux/of.h>
#include <linux/of_net.h>
21
#include <net/devlink.h>
22

23 24
#include "dsa_priv.h"

25
static LIST_HEAD(dsa_tree_list);
26 27
static DEFINE_MUTEX(dsa2_mutex);

28 29 30
static const struct devlink_ops dsa_devlink_ops = {
};

31
static struct dsa_switch_tree *dsa_tree_find(int index)
32 33 34
{
	struct dsa_switch_tree *dst;

35
	list_for_each_entry(dst, &dsa_tree_list, list)
36
		if (dst->index == index)
37
			return dst;
38

39 40 41
	return NULL;
}

42
static struct dsa_switch_tree *dsa_tree_alloc(int index)
43 44 45 46 47 48
{
	struct dsa_switch_tree *dst;

	dst = kzalloc(sizeof(*dst), GFP_KERNEL);
	if (!dst)
		return NULL;
49

50
	dst->index = index;
51

52
	INIT_LIST_HEAD(&dst->list);
53
	list_add_tail(&dsa_tree_list, &dst->list);
54

55 56 57 58 59
	kref_init(&dst->refcount);

	return dst;
}

60 61 62 63 64 65
static void dsa_tree_free(struct dsa_switch_tree *dst)
{
	list_del(&dst->list);
	kfree(dst);
}

66
static struct dsa_switch_tree *dsa_tree_get(struct dsa_switch_tree *dst)
67
{
68 69
	if (dst)
		kref_get(&dst->refcount);
70 71 72 73

	return dst;
}

74
static struct dsa_switch_tree *dsa_tree_touch(int index)
75
{
76 77 78 79 80 81 82
	struct dsa_switch_tree *dst;

	dst = dsa_tree_find(index);
	if (dst)
		return dsa_tree_get(dst);
	else
		return dsa_tree_alloc(index);
83 84 85 86 87 88 89 90 91 92 93 94 95
}

static void dsa_tree_release(struct kref *ref)
{
	struct dsa_switch_tree *dst;

	dst = container_of(ref, struct dsa_switch_tree, refcount);

	dsa_tree_free(dst);
}

static void dsa_tree_put(struct dsa_switch_tree *dst)
{
96 97
	if (dst)
		kref_put(&dst->refcount, dsa_tree_release);
98 99
}

100
static bool dsa_port_is_dsa(struct dsa_port *port)
101
{
102
	return port->type == DSA_PORT_TYPE_DSA;
103 104 105 106
}

static bool dsa_port_is_cpu(struct dsa_port *port)
{
107
	return port->type == DSA_PORT_TYPE_CPU;
108 109
}

110 111 112 113 114
static bool dsa_port_is_user(struct dsa_port *dp)
{
	return dp->type == DSA_PORT_TYPE_USER;
}

115 116
static struct dsa_port *dsa_tree_find_port_by_node(struct dsa_switch_tree *dst,
						   struct device_node *dn)
117 118
{
	struct dsa_switch *ds;
119 120
	struct dsa_port *dp;
	int device, port;
121

122 123
	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
124 125 126
		if (!ds)
			continue;

127 128 129 130 131 132
		for (port = 0; port < ds->num_ports; port++) {
			dp = &ds->ports[port];

			if (dp->dn == dn)
				return dp;
		}
133 134 135 136 137
	}

	return NULL;
}

V
Vivien Didelot 已提交
138
static bool dsa_port_setup_routing_table(struct dsa_port *dp)
139
{
V
Vivien Didelot 已提交
140 141 142
	struct dsa_switch *ds = dp->ds;
	struct dsa_switch_tree *dst = ds->dst;
	struct device_node *dn = dp->dn;
V
Vivien Didelot 已提交
143
	struct of_phandle_iterator it;
144
	struct dsa_port *link_dp;
V
Vivien Didelot 已提交
145
	int err;
146

V
Vivien Didelot 已提交
147 148 149 150
	of_for_each_phandle(&it, err, dn, "link", NULL, 0) {
		link_dp = dsa_tree_find_port_by_node(dst, it.node);
		if (!link_dp) {
			of_node_put(it.node);
V
Vivien Didelot 已提交
151
			return false;
V
Vivien Didelot 已提交
152
		}
153

V
Vivien Didelot 已提交
154
		ds->rtable[link_dp->ds->index] = dp->index;
155 156
	}

V
Vivien Didelot 已提交
157
	return true;
158 159
}

V
Vivien Didelot 已提交
160
static bool dsa_switch_setup_routing_table(struct dsa_switch *ds)
161
{
V
Vivien Didelot 已提交
162 163 164
	bool complete = true;
	struct dsa_port *dp;
	int i;
165

V
Vivien Didelot 已提交
166 167
	for (i = 0; i < DSA_MAX_SWITCHES; i++)
		ds->rtable[i] = DSA_RTABLE_NONE;
168

V
Vivien Didelot 已提交
169 170
	for (i = 0; i < ds->num_ports; i++) {
		dp = &ds->ports[i];
171

V
Vivien Didelot 已提交
172 173 174 175 176
		if (dsa_port_is_dsa(dp)) {
			complete = dsa_port_setup_routing_table(dp);
			if (!complete)
				break;
		}
177 178
	}

V
Vivien Didelot 已提交
179
	return complete;
180 181
}

V
Vivien Didelot 已提交
182
static bool dsa_tree_setup_routing_table(struct dsa_switch_tree *dst)
183 184
{
	struct dsa_switch *ds;
V
Vivien Didelot 已提交
185 186
	bool complete = true;
	int device;
187

V
Vivien Didelot 已提交
188 189
	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
190 191 192
		if (!ds)
			continue;

V
Vivien Didelot 已提交
193 194 195
		complete = dsa_switch_setup_routing_table(ds);
		if (!complete)
			break;
196 197
	}

V
Vivien Didelot 已提交
198
	return complete;
199 200
}

201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244
static struct dsa_port *dsa_tree_find_first_cpu(struct dsa_switch_tree *dst)
{
	struct dsa_switch *ds;
	struct dsa_port *dp;
	int device, port;

	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
		if (!ds)
			continue;

		for (port = 0; port < ds->num_ports; port++) {
			dp = &ds->ports[port];

			if (dsa_port_is_cpu(dp))
				return dp;
		}
	}

	return NULL;
}

static int dsa_tree_setup_default_cpu(struct dsa_switch_tree *dst)
{
	struct dsa_switch *ds;
	struct dsa_port *dp;
	int device, port;

	/* DSA currently only supports a single CPU port */
	dst->cpu_dp = dsa_tree_find_first_cpu(dst);
	if (!dst->cpu_dp) {
		pr_warn("Tree has no master device\n");
		return -EINVAL;
	}

	/* Assign the default CPU port to all ports of the fabric */
	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
		if (!ds)
			continue;

		for (port = 0; port < ds->num_ports; port++) {
			dp = &ds->ports[port];

245
			if (dsa_port_is_user(dp) || dsa_port_is_dsa(dp))
246 247 248 249 250 251 252 253 254 255 256 257 258
				dp->cpu_dp = dst->cpu_dp;
		}
	}

	return 0;
}

static void dsa_tree_teardown_default_cpu(struct dsa_switch_tree *dst)
{
	/* DSA currently only supports a single CPU port */
	dst->cpu_dp = NULL;
}

259
static int dsa_port_setup(struct dsa_port *dp)
260
{
261
	struct dsa_switch *ds = dp->ds;
262
	int err = 0;
263

264
	memset(&dp->devlink_port, 0, sizeof(dp->devlink_port));
265

266 267 268
	if (dp->type != DSA_PORT_TYPE_UNUSED)
		err = devlink_port_register(ds->devlink, &dp->devlink_port,
					    dp->index);
269
	if (err)
270 271
		return err;

272 273 274 275
	switch (dp->type) {
	case DSA_PORT_TYPE_UNUSED:
		break;
	case DSA_PORT_TYPE_CPU:
276 277 278 279 280 281 282 283 284 285 286 287 288 289
		/* dp->index is used now as port_number. However
		 * CPU ports should have separate numbering
		 * independent from front panel port numbers.
		 */
		devlink_port_attrs_set(&dp->devlink_port,
				       DEVLINK_PORT_FLAVOUR_CPU,
				       dp->index, false, 0);
		err = dsa_port_link_register_of(dp);
		if (err) {
			dev_err(ds->dev, "failed to setup link for port %d.%d\n",
				ds->index, dp->index);
			return err;
		}
		break;
290
	case DSA_PORT_TYPE_DSA:
291 292 293 294 295 296 297
		/* dp->index is used now as port_number. However
		 * DSA ports should have separate numbering
		 * independent from front panel port numbers.
		 */
		devlink_port_attrs_set(&dp->devlink_port,
				       DEVLINK_PORT_FLAVOUR_DSA,
				       dp->index, false, 0);
298
		err = dsa_port_link_register_of(dp);
299
		if (err) {
300
			dev_err(ds->dev, "failed to setup link for port %d.%d\n",
301 302 303 304 305
				ds->index, dp->index);
			return err;
		}
		break;
	case DSA_PORT_TYPE_USER:
306 307 308
		devlink_port_attrs_set(&dp->devlink_port,
				       DEVLINK_PORT_FLAVOUR_PHYSICAL,
				       dp->index, false, 0);
309 310 311 312 313 314 315
		err = dsa_slave_create(dp);
		if (err)
			dev_err(ds->dev, "failed to create slave for port %d.%d\n",
				ds->index, dp->index);
		else
			devlink_port_type_eth_set(&dp->devlink_port, dp->slave);
		break;
316 317 318 319 320
	}

	return 0;
}

321
static void dsa_port_teardown(struct dsa_port *dp)
322
{
323 324
	if (dp->type != DSA_PORT_TYPE_UNUSED)
		devlink_port_unregister(&dp->devlink_port);
325 326 327 328 329 330

	switch (dp->type) {
	case DSA_PORT_TYPE_UNUSED:
		break;
	case DSA_PORT_TYPE_CPU:
	case DSA_PORT_TYPE_DSA:
331
		dsa_port_link_unregister_of(dp);
332 333 334 335 336 337 338
		break;
	case DSA_PORT_TYPE_USER:
		if (dp->slave) {
			dsa_slave_destroy(dp->slave);
			dp->slave = NULL;
		}
		break;
339 340 341
	}
}

342
static int dsa_switch_setup(struct dsa_switch *ds)
343 344 345
{
	int err;

346
	/* Initialize ds->phys_mii_mask before registering the slave MDIO bus
347
	 * driver and before ops->setup() has run, since the switch drivers and
348 349 350
	 * the slave MDIO bus driver rely on these values for probing PHY
	 * devices or not
	 */
351
	ds->phys_mii_mask |= dsa_user_ports(ds);
352

353 354 355 356 357 358 359 360 361 362 363
	/* Add the switch to devlink before calling setup, so that setup can
	 * add dpipe tables
	 */
	ds->devlink = devlink_alloc(&dsa_devlink_ops, 0);
	if (!ds->devlink)
		return -ENOMEM;

	err = devlink_register(ds->devlink, ds->dev);
	if (err)
		return err;

364
	err = ds->ops->setup(ds);
365 366 367
	if (err < 0)
		return err;

V
Vivien Didelot 已提交
368 369 370 371
	err = dsa_switch_register_notifier(ds);
	if (err)
		return err;

372
	if (!ds->slave_mii_bus && ds->ops->phy_read) {
373 374 375 376 377 378 379 380 381 382 383
		ds->slave_mii_bus = devm_mdiobus_alloc(ds->dev);
		if (!ds->slave_mii_bus)
			return -ENOMEM;

		dsa_slave_mii_bus_init(ds);

		err = mdiobus_register(ds->slave_mii_bus);
		if (err < 0)
			return err;
	}

384 385 386
	return 0;
}

387
static void dsa_switch_teardown(struct dsa_switch *ds)
388
{
389
	if (ds->slave_mii_bus && ds->ops->phy_read)
390
		mdiobus_unregister(ds->slave_mii_bus);
V
Vivien Didelot 已提交
391 392

	dsa_switch_unregister_notifier(ds);
393 394 395 396 397 398 399

	if (ds->devlink) {
		devlink_unregister(ds->devlink);
		devlink_free(ds->devlink);
		ds->devlink = NULL;
	}

400 401
}

402 403 404
static int dsa_tree_setup_switches(struct dsa_switch_tree *dst)
{
	struct dsa_switch *ds;
405 406
	struct dsa_port *dp;
	int device, port;
407 408 409 410 411 412 413 414 415 416
	int err;

	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
		if (!ds)
			continue;

		err = dsa_switch_setup(ds);
		if (err)
			return err;
417 418 419 420 421 422 423 424

		for (port = 0; port < ds->num_ports; port++) {
			dp = &ds->ports[port];

			err = dsa_port_setup(dp);
			if (err)
				return err;
		}
425 426 427 428 429 430 431 432
	}

	return 0;
}

static void dsa_tree_teardown_switches(struct dsa_switch_tree *dst)
{
	struct dsa_switch *ds;
433 434
	struct dsa_port *dp;
	int device, port;
435 436 437 438 439 440

	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
		if (!ds)
			continue;

441 442 443 444 445 446
		for (port = 0; port < ds->num_ports; port++) {
			dp = &ds->ports[port];

			dsa_port_teardown(dp);
		}

447 448 449 450
		dsa_switch_teardown(ds);
	}
}

451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467
static int dsa_tree_setup_master(struct dsa_switch_tree *dst)
{
	struct dsa_port *cpu_dp = dst->cpu_dp;
	struct net_device *master = cpu_dp->master;

	/* DSA currently supports a single pair of CPU port and master device */
	return dsa_master_setup(master, cpu_dp);
}

static void dsa_tree_teardown_master(struct dsa_switch_tree *dst)
{
	struct dsa_port *cpu_dp = dst->cpu_dp;
	struct net_device *master = cpu_dp->master;

	return dsa_master_teardown(master);
}

V
Vivien Didelot 已提交
468
static int dsa_tree_setup(struct dsa_switch_tree *dst)
469
{
V
Vivien Didelot 已提交
470
	bool complete;
471 472
	int err;

V
Vivien Didelot 已提交
473 474 475 476 477 478
	if (dst->setup) {
		pr_err("DSA: tree %d already setup! Disjoint trees?\n",
		       dst->index);
		return -EEXIST;
	}

V
Vivien Didelot 已提交
479 480 481 482
	complete = dsa_tree_setup_routing_table(dst);
	if (!complete)
		return 0;

483 484 485 486
	err = dsa_tree_setup_default_cpu(dst);
	if (err)
		return err;

487 488 489
	err = dsa_tree_setup_switches(dst);
	if (err)
		return err;
490

491
	err = dsa_tree_setup_master(dst);
492 493 494
	if (err)
		return err;

V
Vivien Didelot 已提交
495 496 497
	dst->setup = true;

	pr_info("DSA: tree %d setup\n", dst->index);
498 499 500 501

	return 0;
}

V
Vivien Didelot 已提交
502
static void dsa_tree_teardown(struct dsa_switch_tree *dst)
503
{
V
Vivien Didelot 已提交
504
	if (!dst->setup)
505 506
		return;

507
	dsa_tree_teardown_master(dst);
508

509
	dsa_tree_teardown_switches(dst);
510

511
	dsa_tree_teardown_default_cpu(dst);
512

V
Vivien Didelot 已提交
513 514 515
	pr_info("DSA: tree %d torn down\n", dst->index);

	dst->setup = false;
516 517
}

518 519 520
static void dsa_tree_remove_switch(struct dsa_switch_tree *dst,
				   unsigned int index)
{
521 522
	dsa_tree_teardown(dst);

523 524 525 526 527 528 529 530
	dst->ds[index] = NULL;
	dsa_tree_put(dst);
}

static int dsa_tree_add_switch(struct dsa_switch_tree *dst,
			       struct dsa_switch *ds)
{
	unsigned int index = ds->index;
531
	int err;
532 533 534 535 536 537 538

	if (dst->ds[index])
		return -EBUSY;

	dsa_tree_get(dst);
	dst->ds[index] = ds;

539 540 541 542 543
	err = dsa_tree_setup(dst);
	if (err)
		dsa_tree_remove_switch(dst, index);

	return err;
544 545
}

546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565
static int dsa_port_parse_user(struct dsa_port *dp, const char *name)
{
	if (!name)
		name = "eth%d";

	dp->type = DSA_PORT_TYPE_USER;
	dp->name = name;

	return 0;
}

static int dsa_port_parse_dsa(struct dsa_port *dp)
{
	dp->type = DSA_PORT_TYPE_DSA;

	return 0;
}

static int dsa_port_parse_cpu(struct dsa_port *dp, struct net_device *master)
{
566 567 568 569 570
	struct dsa_switch *ds = dp->ds;
	struct dsa_switch_tree *dst = ds->dst;
	const struct dsa_device_ops *tag_ops;
	enum dsa_tag_protocol tag_protocol;

571
	tag_protocol = ds->ops->get_tag_protocol(ds, dp->index);
572 573 574 575 576 577
	tag_ops = dsa_resolve_tag_protocol(tag_protocol);
	if (IS_ERR(tag_ops)) {
		dev_warn(ds->dev, "No tagger for this switch\n");
		return PTR_ERR(tag_ops);
	}

578
	dp->type = DSA_PORT_TYPE_CPU;
579 580
	dp->rcv = tag_ops->rcv;
	dp->tag_ops = tag_ops;
581
	dp->master = master;
582
	dp->dst = dst;
583 584 585 586

	return 0;
}

587 588
static int dsa_port_parse_of(struct dsa_port *dp, struct device_node *dn)
{
589
	struct device_node *ethernet = of_parse_phandle(dn, "ethernet", 0);
590
	const char *name = of_get_property(dn, "label", NULL);
591
	bool link = of_property_read_bool(dn, "link");
592

593 594
	dp->dn = dn;

595
	if (ethernet) {
596 597 598 599 600 601
		struct net_device *master;

		master = of_find_net_device_by_node(ethernet);
		if (!master)
			return -EPROBE_DEFER;

602
		return dsa_port_parse_cpu(dp, master);
603 604
	}

605 606
	if (link)
		return dsa_port_parse_dsa(dp);
607

608
	return dsa_port_parse_user(dp, name);
609 610
}

V
Vivien Didelot 已提交
611 612
static int dsa_switch_parse_ports_of(struct dsa_switch *ds,
				     struct device_node *dn)
613
{
614
	struct device_node *ports, *port;
615
	struct dsa_port *dp;
616
	int err = 0;
617
	u32 reg;
618 619 620 621 622 623

	ports = of_get_child_by_name(dn, "ports");
	if (!ports) {
		dev_err(ds->dev, "no ports child node found\n");
		return -EINVAL;
	}
624 625 626 627

	for_each_available_child_of_node(ports, port) {
		err = of_property_read_u32(port, "reg", &reg);
		if (err)
628
			goto out_put_node;
629

630 631 632 633
		if (reg >= ds->num_ports) {
			err = -EINVAL;
			goto out_put_node;
		}
634

635 636 637 638
		dp = &ds->ports[reg];

		err = dsa_port_parse_of(dp, port);
		if (err)
639
			goto out_put_node;
640 641
	}

642 643 644
out_put_node:
	of_node_put(ports);
	return err;
645 646
}

V
Vivien Didelot 已提交
647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679
static int dsa_switch_parse_member_of(struct dsa_switch *ds,
				      struct device_node *dn)
{
	u32 m[2] = { 0, 0 };
	int sz;

	/* Don't error out if this optional property isn't found */
	sz = of_property_read_variable_u32_array(dn, "dsa,member", m, 2, 2);
	if (sz < 0 && sz != -EINVAL)
		return sz;

	ds->index = m[1];
	if (ds->index >= DSA_MAX_SWITCHES)
		return -EINVAL;

	ds->dst = dsa_tree_touch(m[0]);
	if (!ds->dst)
		return -ENOMEM;

	return 0;
}

static int dsa_switch_parse_of(struct dsa_switch *ds, struct device_node *dn)
{
	int err;

	err = dsa_switch_parse_member_of(ds, dn);
	if (err)
		return err;

	return dsa_switch_parse_ports_of(ds, dn);
}

680 681 682
static int dsa_port_parse(struct dsa_port *dp, const char *name,
			  struct device *dev)
{
683
	if (!strcmp(name, "cpu")) {
684 685 686 687 688 689 690 691
		struct net_device *master;

		master = dsa_dev_to_net_device(dev);
		if (!master)
			return -EPROBE_DEFER;

		dev_put(master);

692
		return dsa_port_parse_cpu(dp, master);
693 694
	}

695 696
	if (!strcmp(name, "dsa"))
		return dsa_port_parse_dsa(dp);
697

698
	return dsa_port_parse_user(dp, name);
699 700
}

V
Vivien Didelot 已提交
701 702
static int dsa_switch_parse_ports(struct dsa_switch *ds,
				  struct dsa_chip_data *cd)
703 704
{
	bool valid_name_found = false;
705 706 707
	struct dsa_port *dp;
	struct device *dev;
	const char *name;
708
	unsigned int i;
709
	int err;
710 711

	for (i = 0; i < DSA_MAX_PORTS; i++) {
712 713 714 715 716
		name = cd->port_names[i];
		dev = cd->netdev[i];
		dp = &ds->ports[i];

		if (!name)
717 718
			continue;

719 720 721 722
		err = dsa_port_parse(dp, name, dev);
		if (err)
			return err;

723 724 725 726 727 728 729 730 731
		valid_name_found = true;
	}

	if (!valid_name_found && i == DSA_MAX_PORTS)
		return -EINVAL;

	return 0;
}

V
Vivien Didelot 已提交
732
static int dsa_switch_parse(struct dsa_switch *ds, struct dsa_chip_data *cd)
733
{
V
Vivien Didelot 已提交
734
	ds->cd = cd;
735

V
Vivien Didelot 已提交
736 737 738 739 740 741 742
	/* We don't support interconnected switches nor multiple trees via
	 * platform data, so this is the unique switch of the tree.
	 */
	ds->index = 0;
	ds->dst = dsa_tree_touch(0);
	if (!ds->dst)
		return -ENOMEM;
743

V
Vivien Didelot 已提交
744
	return dsa_switch_parse_ports(ds, cd);
745 746
}

747 748 749 750 751 752 753
static int dsa_switch_add(struct dsa_switch *ds)
{
	struct dsa_switch_tree *dst = ds->dst;

	return dsa_tree_add_switch(dst, ds);
}

754
static int dsa_switch_probe(struct dsa_switch *ds)
755
{
756 757
	struct dsa_chip_data *pdata = ds->dev->platform_data;
	struct device_node *np = ds->dev->of_node;
V
Vivien Didelot 已提交
758
	int err;
759

V
Vivien Didelot 已提交
760 761 762 763 764 765
	if (np)
		err = dsa_switch_parse_of(ds, np);
	else if (pdata)
		err = dsa_switch_parse(ds, pdata);
	else
		err = -ENODEV;
766

V
Vivien Didelot 已提交
767 768
	if (err)
		return err;
769

770
	return dsa_switch_add(ds);
771 772
}

773 774 775
struct dsa_switch *dsa_switch_alloc(struct device *dev, size_t n)
{
	struct dsa_switch *ds;
776
	int i;
777

778
	ds = devm_kzalloc(dev, struct_size(ds, ports, n), GFP_KERNEL);
779 780 781
	if (!ds)
		return NULL;

S
Salvatore Mesoraca 已提交
782 783 784 785 786 787 788 789 790 791 792 793 794 795
	/* We avoid allocating memory outside dsa_switch
	 * if it is not needed.
	 */
	if (n <= sizeof(ds->_bitmap) * 8) {
		ds->bitmap = &ds->_bitmap;
	} else {
		ds->bitmap = devm_kcalloc(dev,
					  BITS_TO_LONGS(n),
					  sizeof(unsigned long),
					  GFP_KERNEL);
		if (unlikely(!ds->bitmap))
			return NULL;
	}

796 797 798
	ds->dev = dev;
	ds->num_ports = n;

799 800 801 802 803
	for (i = 0; i < ds->num_ports; ++i) {
		ds->ports[i].index = i;
		ds->ports[i].ds = ds;
	}

804 805 806 807
	return ds;
}
EXPORT_SYMBOL_GPL(dsa_switch_alloc);

808
int dsa_register_switch(struct dsa_switch *ds)
809 810 811 812
{
	int err;

	mutex_lock(&dsa2_mutex);
813
	err = dsa_switch_probe(ds);
814
	dsa_tree_put(ds->dst);
815 816 817 818 819 820
	mutex_unlock(&dsa2_mutex);

	return err;
}
EXPORT_SYMBOL_GPL(dsa_register_switch);

821
static void dsa_switch_remove(struct dsa_switch *ds)
822 823
{
	struct dsa_switch_tree *dst = ds->dst;
824
	unsigned int index = ds->index;
825

826
	dsa_tree_remove_switch(dst, index);
827 828 829 830 831
}

void dsa_unregister_switch(struct dsa_switch *ds)
{
	mutex_lock(&dsa2_mutex);
832
	dsa_switch_remove(ds);
833 834 835
	mutex_unlock(&dsa2_mutex);
}
EXPORT_SYMBOL_GPL(dsa_unregister_switch);